diff options
| -rw-r--r-- | log_ssh_passwords.py | 9 |
1 files changed, 7 insertions, 2 deletions
diff --git a/log_ssh_passwords.py b/log_ssh_passwords.py index 1ad7c97..5d2a38b 100644 --- a/log_ssh_passwords.py +++ b/log_ssh_passwords.py @@ -1,6 +1,7 @@ #!/opt/public_ssh_password_stats/log_ssh_passwords/venv/bin/python3 import socket import time +import os import paramiko import db_handler @@ -38,8 +39,12 @@ def ssh_thread(client, address, port): # Create an SSH server host_key = paramiko.RSAKey.from_private_key_file("id_rsa") # paramiko.RSAKey.generate(2048) -server = '' -port = 2200 +if os.environ.get('DEBUG') == '1': # Local development + server = '127.0.0.1' + port = 2200 +else: # Production + server = '' + port = 22 # Start the server server_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM) |