Modified file for BSD Cafe Fork

This commit is contained in:
Stefano Marinelli 2024-05-09 13:16:00 +02:00
parent 795c04e6df
commit ebd866659e

View file

@ -1,10 +1,11 @@
#!/usr/bin/python #!/usr/local/bin/python3.9
##### CheckMyIP Server ##### ##### BSD Cafe CheckMyIP Server #####
##### Written by John W Kerns ##### ##### Originally written by John W Kerns #####
##### http://blog.packetsar.com ##### ##### http://blog.packetsar.com #####
##### https://github.com/packetsar/checkmyip ##### ##### Forked to BSD Cafe by Stefano Marinelli #####
##### https://brew.bsd.cafe/BSDCafe/checkmyip #####
##### Inform version here ##### ##### Inform version here #####
@ -37,18 +38,18 @@ j2send = """{
"port": "{{ port }}", "port": "{{ port }}",
"protocol": "{{ proto }}", "protocol": "{{ proto }}",
"version": "%s", "version": "%s",
"website": "https://github.com/packetsar/checkmyip", "website": "https://brew.bsd.cafe/BSDCafe/checkmyip",
"sponsor": "Sponsored by ConvergeOne, https://www.convergeone.com/" "sponsor": "Served by BSD Cafe, https://bsd.cafe/"
}""" % version }""" % version
##### Handles all prnting to console and logging to the logfile ##### ##### Handles all prnting to console and logging to the logfile #####
class log_management: class log_management:
def __init__(self): def __init__(self):
self.logpath = "/etc/checkmyip/" # Log file directory path self.logpath = "/var/log/checkmyip/" # Log file directory path
self.logfile = "/etc/checkmyip/%scheckmyip.log" % \ self.logfile = "/var/log/checkmyip/%scheckmyip.log" % \
time.strftime("%Y-%m-%d_") # Log file full path time.strftime("%Y-%m-%d_") # Log file full path
self.paramikolog = "/etc/checkmyip/%sssh.log" % \ self.paramikolog = "/var/log/checkmyip/%sssh.log" % \
time.strftime("%Y-%m-%d_") # SSH log file path time.strftime("%Y-%m-%d_") # SSH log file path
self.thread = threading.Thread(target=self._change_logfiles) self.thread = threading.Thread(target=self._change_logfiles)
self.thread.daemon = True self.thread.daemon = True
@ -89,9 +90,9 @@ class log_management:
def _change_logfiles(self, thread=True): def _change_logfiles(self, thread=True):
while True: while True:
time.sleep(10) time.sleep(10)
self.logfile = "/etc/checkmyip/%scheckmyip.log" % \ self.logfile = "/var/log/checkmyip/%scheckmyip.log" % \
time.strftime("%Y-%m-%d_") # Log file full path time.strftime("%Y-%m-%d_") # Log file full path
self.paramikolog = "/etc/checkmyip/%sssh.log" % \ self.paramikolog = "/var/log/checkmyip/%sssh.log" % \
time.strftime("%Y-%m-%d_") # SSH log file path time.strftime("%Y-%m-%d_") # SSH log file path
paramiko.util.log_to_file(self.paramikolog) paramiko.util.log_to_file(self.paramikolog)
@ -148,14 +149,14 @@ def j2format(j2tmp, valdict):
##### Cleans IP addresses coming from socket library ##### ##### Cleans IP addresses coming from socket library #####
def cleanip(addr): def cleanip(addr):
ip = addr[0] ip, port = addr[:2]
port = addr[1] family = "ipv6" # Default to IPv6
family = "ipv6" if ip.startswith("::ffff:"): # Check if this is a prefixed IPv4 address
if len(ip) > 6: # If this IP is not a super short v6 address ip = ip.replace("::ffff:", "") # Clean the IP
if ip[:7] == "::ffff:": # If this is a prefixed IPv4 address
ip = ip.replace("::ffff:", "") # Return the cleaned IP
family = "ipv4" family = "ipv4"
return (ip, port, family) # Return the uncleaned IP if not matched elif ":" not in ip: # Simple check to recognize IPv4
family = "ipv4"
return ip, port, family # Return cleaned IP and family
##### TCP listener methods. Gets used once for each listening port ##### ##### TCP listener methods. Gets used once for each listening port #####
@ -164,17 +165,19 @@ def listener(port, talker):
listen_port = port listen_port = port
buffer_size = 1024 buffer_size = 1024
while True: while True:
sock = socket.socket(socket.AF_INET6, socket.SOCK_STREAM) # v6 family # Use AF_INET6 but also handle IPv4 connections
sock = socket.socket(socket.AF_INET6, socket.SOCK_STREAM)
sock.setsockopt(socket.IPPROTO_IPV6, socket.IPV6_V6ONLY, 0) # Allow IPv4-mapped IPv6 addresses
sock.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1) sock.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
sock.bind((listen_ip, listen_port)) sock.bind((listen_ip, listen_port))
sock.listen(buffer_size) sock.listen(buffer_size)
while True:
client, addr = sock.accept() client, addr = sock.accept()
ip, port, family = cleanip(addr) # Get all cleaned IP info ip, port, family = cleanip(addr) # Get cleaned IP info including family
valdict = {"ip": ip, "port": port, "family": family} # Put in dict valdict = {"ip": ip, "port": port, "family": family, "proto": ""}
thread = threading.Thread(target=talker, args=(client, valdict)) thread = threading.Thread(target=talker, args=(client, valdict))
thread.start() # Start talker thread to listen to port thread.start() # Start talker thread to listen to port
##### Telnet responder method. Is run in own thread for each telnet query ##### ##### Telnet responder method. Is run in own thread for each telnet query #####
def telnet_talker(client, valdict, proto="telnet"): def telnet_talker(client, valdict, proto="telnet"):
valdict.update({"proto": proto}) # Add the protocol to the value dict valdict.update({"proto": proto}) # Add the protocol to the value dict
@ -209,15 +212,26 @@ def ssh_talker(client, valdict, proto="ssh"):
##### HTTP responder method. Gets run in own thread for each HTTP query ##### ##### HTTP responder method. Gets run in own thread for each HTTP query #####
##### Automatically detects if client is a browser or a telnet client ##### ##### Automatically detects if client is a browser or a telnet client #####
##### HTTP responder method. Gets run in own thread for each HTTP query #####
def http_talker(client, valdict, proto="http"): def http_talker(client, valdict, proto="http"):
time.sleep(.1) # Sleep to allow the client to send some data time.sleep(.1) # Sleep to allow the client to send some data
client.setblocking(0) # Set the socket recv as non-blocking client.setblocking(0) # Set the socket recv as non-blocking
browser = False # Is the client using a browser? browser = False # Is the client using a browser?
try: # client.recv() will raise an error if the buffer is empty headers = {}
data = client.recv(2048) # Recieve data from the buffer (if any) try:
print(data) # Print to stdout data = client.recv(2048) # Receive data from the buffer (if any)
request_lines = data.decode().split('\r\n')
for line in request_lines:
if ": " in line:
key, value = line.split(": ", 1)
headers[key] = value
# Extract real IP and adjust family based on headers if present
if 'X-Real-IP' in headers or 'X-Forwarded-For' in headers:
real_ip = headers.get('X-Real-IP', headers.get('X-Forwarded-For').split(',')[0])
ip, port, family = cleanip((real_ip, valdict['port']))
valdict.update({"ip": ip, "family": family})
browser = True # Set client browser to True browser = True # Set client browser to True
except: # If buffer was empty, then like a telnet client on TCP80 except socket.error: # If buffer was empty, then like a telnet client on TCP80
browser = False # Set client browser to False browser = False # Set client browser to False
if not browser: # If the client is not a browser if not browser: # If the client is not a browser
telnet_talker(client, valdict, "http-telnet") # Hand to telnet_talker telnet_talker(client, valdict, "http-telnet") # Hand to telnet_talker
@ -234,10 +248,11 @@ Connection: close""" % str(len(response_body_raw)) # Response with headers
client.close() client.close()
##### Server startup method. Starts a listener thread for each TCP port ##### ##### Server startup method. Starts a listener thread for each TCP port #####
def start(): def start():
talkers = {22: ssh_talker, 23: telnet_talker, talkers = {2222: ssh_talker, 2223: telnet_talker,
80: http_talker} # Three listeners on different ports 2280: http_talker} # Three listeners on different ports
for talker in talkers: for talker in talkers:
# Launch a thread for each listener # Launch a thread for each listener
thread = threading.Thread(target=listener, thread = threading.Thread(target=listener,