From 5f085c5eb1f0a88645a4fedbdc6a8bce451551bd Mon Sep 17 00:00:00 2001 From: John W Kerns Date: Fri, 1 Sep 2017 17:34:46 -0700 Subject: [PATCH] Added logging --- checkmyip.py | 242 ++++++++++++++++++++++++++++++++------------------- 1 file changed, 153 insertions(+), 89 deletions(-) diff --git a/checkmyip.py b/checkmyip.py index 6d5bcf5..f346ef8 100644 --- a/checkmyip.py +++ b/checkmyip.py @@ -1,120 +1,184 @@ +import os import sys +import time import socket import jinja2 import paramiko import threading -j2log = "Connection from: {{ ip }} ({{ port }})" -j2send = "\n\n\nYour IP Address is {{ ip }} ({{ port }})\n\n\n\n" +j2log = "Connection from: {{ ip }} ({{ port }}) ({{ proto }})" +#j2send = "\n\n\nYour IP Address is {{ ip }} ({{ port }})\n\n\n\n" -paramiko.util.log_to_file('demo_server.log') +j2send = """{ + + +"comment": "## Your IP Address is {{ ip }} ({{ port }}) ##", + + +"family": "{{ family }}", +"ip": "{{ ip }}", +"port": "{{ port }}" +}""" + + + + + +class log_management: + def __init__(self): + self.logpath = "/etc/checkmyip/" + self.logfile = "/etc/checkmyip/checkmyip.log" + self._publish_methods() + self.can_log = True + paramiko.util.log_to_file('/etc/checkmyip/ssh.log') + def _logger(self, data): + logdata = time.strftime("%Y-%m-%d %H:%M:%S") + ": " + data + "\n" + if self.can_log: + try: + f = open(self.logfile, 'a') + f.write(logdata) + f.close() + except IOError: + self._console("Unable to log to logfile %s. Creating log directory" % self.logfile) + self.can_log = False + self._create_log_dir() + self._console(logdata) + def _console(self, data, timestamp=False): + if timestamp: + logdata = time.strftime("%Y-%m-%d %H:%M:%S") + ": " + data + "\n" + else: + logdata = data + print(logdata) + def _publish_methods(self): + global log + global console + log = self._logger + console = self._console + def _create_log_dir(self): + os.system('mkdir -p ' + self.logpath) + self._console("Logpath (%s) created" % self.logpath) + self.can_log = True class rsa_key: - data = """-----BEGIN RSA PRIVATE KEY----- - MIICWgIBAAKBgQDTj1bqB4WmayWNPB+8jVSYpZYk80Ujvj680pOTh2bORBjbIAyz - oWGW+GUjzKxTiiPvVmxFgx5wdsFvF03v34lEVVhMpouqPAYQ15N37K/ir5XY+9m/ - d8ufMCkjeXsQkKqFbAlQcnWMCRnOoPHS3I4vi6hmnDDeeYTSRvfLbW0fhwIBIwKB - gBIiOqZYaoqbeD9OS9z2K9KR2atlTxGxOJPXiP4ESqP3NVScWNwyZ3NXHpyrJLa0 - EbVtzsQhLn6rF+TzXnOlcipFvjsem3iYzCpuChfGQ6SovTcOjHV9z+hnpXvQ/fon - soVRZY65wKnF7IAoUwTmJS9opqgrN6kRgCd3DASAMd1bAkEA96SBVWFt/fJBNJ9H - tYnBKZGw0VeHOYmVYbvMSstssn8un+pQpUm9vlG/bp7Oxd/m+b9KWEh2xPfv6zqU - avNwHwJBANqzGZa/EpzF4J8pGti7oIAPUIDGMtfIcmqNXVMckrmzQ2vTfqtkEZsA - 4rE1IERRyiJQx6EJsz21wJmGV9WJQ5kCQQDwkS0uXqVdFzgHO6S++tjmjYcxwr3g - H0CoFYSgbddOT6miqRskOQF3DZVkJT3kyuBgU2zKygz52ukQZMqxCb1fAkASvuTv - qfpH87Qq5kQhNKdbbwbmd2NxlNabazPijWuphGTdW0VfJdWfklyS2Kr+iqrs/5wV - HhathJt636Eg7oIjAkA8ht3MQ+XSl9yIJIS8gVpbPxSw5OMfw0PjVE7tBdQruiSc - nvuQES5C9BMHjF39LZiGH1iLQy7FgdHyoP+eodI7 - -----END RSA PRIVATE KEY----- - """ - def open(self): - pass - def close(self): - pass - def readlines(self): - return self.data.split("\n") - def __call__(self): - return paramiko.RSAKey.from_private_key(self) + data = """-----BEGIN RSA PRIVATE KEY----- + MIICWgIBAAKBgQDTj1bqB4WmayWNPB+8jVSYpZYk80Ujvj680pOTh2bORBjbIAyz + oWGW+GUjzKxTiiPvVmxFgx5wdsFvF03v34lEVVhMpouqPAYQ15N37K/ir5XY+9m/ + d8ufMCkjeXsQkKqFbAlQcnWMCRnOoPHS3I4vi6hmnDDeeYTSRvfLbW0fhwIBIwKB + gBIiOqZYaoqbeD9OS9z2K9KR2atlTxGxOJPXiP4ESqP3NVScWNwyZ3NXHpyrJLa0 + EbVtzsQhLn6rF+TzXnOlcipFvjsem3iYzCpuChfGQ6SovTcOjHV9z+hnpXvQ/fon + soVRZY65wKnF7IAoUwTmJS9opqgrN6kRgCd3DASAMd1bAkEA96SBVWFt/fJBNJ9H + tYnBKZGw0VeHOYmVYbvMSstssn8un+pQpUm9vlG/bp7Oxd/m+b9KWEh2xPfv6zqU + avNwHwJBANqzGZa/EpzF4J8pGti7oIAPUIDGMtfIcmqNXVMckrmzQ2vTfqtkEZsA + 4rE1IERRyiJQx6EJsz21wJmGV9WJQ5kCQQDwkS0uXqVdFzgHO6S++tjmjYcxwr3g + H0CoFYSgbddOT6miqRskOQF3DZVkJT3kyuBgU2zKygz52ukQZMqxCb1fAkASvuTv + qfpH87Qq5kQhNKdbbwbmd2NxlNabazPijWuphGTdW0VfJdWfklyS2Kr+iqrs/5wV + HhathJt636Eg7oIjAkA8ht3MQ+XSl9yIJIS8gVpbPxSw5OMfw0PjVE7tBdQruiSc + nvuQES5C9BMHjF39LZiGH1iLQy7FgdHyoP+eodI7 + -----END RSA PRIVATE KEY----- + """ + def open(self): + pass + def close(self): + pass + def readlines(self): + return self.data.split("\n") + def __call__(self): + return paramiko.RSAKey.from_private_key(self) class ssh_server (paramiko.ServerInterface): - def __init__(self): - self.event = threading.Event() - def check_channel_request(self, kind, chanid): - if kind == 'session': - return paramiko.OPEN_SUCCEEDED - return paramiko.OPEN_FAILED_ADMINISTRATIVELY_PROHIBITED - def check_auth_none(self, username): - return paramiko.AUTH_SUCCESSFUL - def get_allowed_auths(self, username): - return 'none' - def check_channel_shell_request(self, channel): - self.event.set() - return True - def check_channel_pty_request(self, channel, term, width, height, - pixelwidth, pixelheight, modes): - return True + def __init__(self): + self.event = threading.Event() + def check_channel_request(self, kind, chanid): + if kind == 'session': + return paramiko.OPEN_SUCCEEDED + return paramiko.OPEN_FAILED_ADMINISTRATIVELY_PROHIBITED + def check_auth_none(self, username): + return paramiko.AUTH_SUCCESSFUL + def get_allowed_auths(self, username): + return 'none' + def check_channel_shell_request(self, channel): + self.event.set() + return True + def check_channel_pty_request(self, channel, term, width, height, + pixelwidth, pixelheight, modes): + return True def j2format(j2tmp, valdict): - template = jinja2.Template(j2tmp) - return template.render(valdict) + template = jinja2.Template(j2tmp) + return template.render(valdict) -def cleanip(ip): - if len(ip) > 6: # If this IP is not a super short v6 address - if ip[:7] == "::ffff:": # If this is a prefixed IPv4 address - return ip.replace("::ffff:", "") # Return the cleaned IP - return ip # Return the uncleaned IP if not matched +def cleanip(addr): + ip = addr[0] + port = addr[1] + family = "ipv6" + if len(ip) > 6: # If this IP is not a super short v6 address + if ip[:7] == "::ffff:": # If this is a prefixed IPv4 address + ip = ip.replace("::ffff:", "") # Return the cleaned IP + family = "ipv4" + return (ip, port, family) # Return the uncleaned IP if not matched def listener(port, talker): - listen_ip = '' - listen_port = port - buffer_size = 100 # Normally 1024, but we want fast response - while True: - sock = socket.socket(socket.AF_INET6, socket.SOCK_STREAM) - sock.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1) - sock.bind((listen_ip, listen_port)) - sock.listen(buffer_size) - client, addr = sock.accept() - thread = threading.Thread(target=talker, args=(client, addr)) - thread.start() + listen_ip = '' + listen_port = port + buffer_size = 100 # Normally 1024, but we want fast response + while True: + sock = socket.socket(socket.AF_INET6, socket.SOCK_STREAM) + sock.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1) + sock.bind((listen_ip, listen_port)) + sock.listen(buffer_size) + client, addr = sock.accept() + ip, port, family = cleanip(addr) + valdict = {"ip": ip, "port": port, "family": family} + thread = threading.Thread(target=talker, args=(client, valdict)) + thread.start() -def telnet_talker(client, addr): - valdict = {"ip": cleanip(addr[0]), "port": addr[1]} - print(j2format(j2log, valdict)) - client.send(j2format(j2send, valdict)) # echo - client.close() +def telnet_talker(client, valdict): + valdict.update({"proto": "telnet"}) + log(j2format(j2log, valdict)) + client.send(j2format(j2send, valdict)) # echo + client.close() + #quit() -def ssh_talker(client, addr): - t = paramiko.Transport(client, gss_kex=True) - t.set_gss_host(socket.getfqdn("")) - t.load_server_moduli() - t.add_server_key(rsa_key()()) - server = ssh_server() - t.start_server(server=server) - chan = t.accept(20) - server.event.wait(10) - valdict = {"ip": cleanip(addr[0]), "port": addr[1]} - #chan.send(j2format(j2log, valdict)) - chan.send('%s' % j2format(j2send, valdict)) - chan.makefile('rU').readline().strip('\r\n') - chan.close() +def ssh_talker(client, valdict): + valdict.update({"proto": "ssh"}) + t = paramiko.Transport(client, gss_kex=True) + t.set_gss_host(socket.getfqdn("")) + t.load_server_moduli() + t.add_server_key(rsa_key()()) + server = ssh_server() + t.start_server(server=server) + chan = t.accept(20) + if chan: + server.event.wait(10) + #chan.send(j2format(j2log, valdict)) + log(j2format(j2log, valdict)) + chan.send('%s' % j2format(j2send, valdict)) + chan.makefile('rU').readline().strip('\r\n') + chan.close() + #quit() def start(): - talkers = {2200: ssh_talker, 23: telnet_talker} - for talker in talkers: - thread = threading.Thread(target=listener, args=(talker, talkers[talker])) - thread.daemon = False - thread.start() + talkers = {22: ssh_talker, 23: telnet_talker} + for talker in talkers: + thread = threading.Thread(target=listener, args=(talker, talkers[talker])) + thread.daemon = True + thread.start() + while True: + try: + time.sleep(1) + except KeyboardInterrupt: + quit() + if __name__ == "__main__": - start() - - - + logging = log_management() + start() \ No newline at end of file