Compare commits
19 Commits
0.2.0
...
8646388851
| Author | SHA1 | Date | |
|---|---|---|---|
|
8646388851
|
|||
|
b8333b6408
|
|||
| ae69e9e956 | |||
|
431d8bb3bb
|
|||
|
a00668ecf3
|
|||
|
9b077c8c48
|
|||
|
7ce32855c4
|
|||
|
b887525cf1
|
|||
|
3512608dd1
|
|||
|
5261ad8e9f
|
|||
|
b7562703dd
|
|||
|
cf78a4b05b
|
|||
|
3c86a5fefd
|
|||
|
014c4296e0
|
|||
|
1182e089bc
|
|||
|
cc1145a4ac
|
|||
|
21bc86fa8a
|
|||
|
b9d0fd92a6
|
|||
|
35b3af31be
|
@@ -3,7 +3,7 @@ from __future__ import annotations
|
||||
|
||||
import logging as lg
|
||||
|
||||
from judas_client.connector import Connector
|
||||
from judas_client.client import Client
|
||||
|
||||
if __name__ == "__main__":
|
||||
lg.basicConfig(
|
||||
@@ -12,5 +12,5 @@ if __name__ == "__main__":
|
||||
)
|
||||
logger = lg.getLogger(__name__)
|
||||
|
||||
connector = Connector("127.0.0.1", 3692)
|
||||
connector.run()
|
||||
client = Client(server_host="localhost", server_port=3692)
|
||||
client.run()
|
||||
|
||||
67
src/judas_client/client.py
Normal file
67
src/judas_client/client.py
Normal file
@@ -0,0 +1,67 @@
|
||||
# -*- coding: utf-8 -*-
|
||||
"""Client module for handling messages."""
|
||||
|
||||
from __future__ import annotations
|
||||
|
||||
import logging as lg
|
||||
import uuid
|
||||
|
||||
from judas_protocol import Message
|
||||
|
||||
from judas_client.connector import Connector
|
||||
|
||||
|
||||
class Client:
|
||||
"""Client class for handling messages."""
|
||||
|
||||
def __init__(self, server_host: str, server_port: int) -> None:
|
||||
"""Initialize the Client.
|
||||
|
||||
Args:
|
||||
server_host (str): The server host address.
|
||||
server_port (int): The server port number.
|
||||
"""
|
||||
self.logger: lg.Logger = lg.getLogger(
|
||||
f"{__name__}.{self.__class__.__name__}"
|
||||
)
|
||||
self.logger.debug("Initializing Client...")
|
||||
|
||||
self.server_host: str = server_host
|
||||
self.server_port: int = server_port
|
||||
|
||||
self.mac_address: str = self._get_mac_address()
|
||||
self.logger.debug(f"MAC address: {self.mac_address}")
|
||||
|
||||
self.connector: Connector = Connector(
|
||||
mac_address=self.mac_address,
|
||||
host=self.server_host,
|
||||
port=self.server_port,
|
||||
on_message=self.handle_message,
|
||||
)
|
||||
|
||||
def _get_mac_address(self) -> str:
|
||||
"""Get the MAC address of the client.
|
||||
|
||||
Returns:
|
||||
str: The MAC address in standard lowercase format (e.g., "0a:1b:2c:3d:4e:5f").
|
||||
"""
|
||||
mac_address: str = ":".join(
|
||||
[
|
||||
"{:02x}".format((uuid.getnode() >> ele) & 0xFF)
|
||||
for ele in range(0, 8 * 6, 8)
|
||||
][::-1]
|
||||
)
|
||||
return mac_address
|
||||
|
||||
def handle_message(self, message: Message) -> None:
|
||||
"""Handle incoming messages.
|
||||
|
||||
Args:
|
||||
message (Message): The incoming message to handle.
|
||||
"""
|
||||
self.logger.info(f"[v] Received message: {message}")
|
||||
|
||||
def run(self) -> None:
|
||||
"""Run the client."""
|
||||
self.logger.info(f"Starting Client ({self.mac_address})...")
|
||||
self.connector.run()
|
||||
@@ -4,7 +4,7 @@ from __future__ import annotations
|
||||
import logging as lg
|
||||
import socket
|
||||
import time
|
||||
import uuid
|
||||
from typing import Callable
|
||||
|
||||
from judas_protocol import Message
|
||||
|
||||
@@ -12,10 +12,13 @@ from judas_protocol import Message
|
||||
class Connector:
|
||||
def __init__(
|
||||
self,
|
||||
mac_address: str,
|
||||
host: str,
|
||||
port: int,
|
||||
*,
|
||||
connect_timeout: float = 5.0,
|
||||
ack_timeout: float = None,
|
||||
ack_timeout: float | None = None,
|
||||
on_message: Callable[[Message], None],
|
||||
) -> None:
|
||||
self.logger: lg.Logger = lg.getLogger(
|
||||
f"{__name__}.{self.__class__.__name__}"
|
||||
@@ -26,23 +29,15 @@ class Connector:
|
||||
self.port: int = port
|
||||
self.socket_timeout: None = None
|
||||
self.connect_timeout: float = connect_timeout
|
||||
self.ack_timeout: float = ack_timeout
|
||||
self.ack_timeout: float | None = ack_timeout
|
||||
|
||||
self.socket: socket.socket = socket.socket(
|
||||
socket.AF_INET, socket.SOCK_STREAM
|
||||
)
|
||||
|
||||
self.mac_address: str = self._get_mac_address()
|
||||
self.mac_address: str = mac_address
|
||||
|
||||
def _get_mac_address(self) -> str:
|
||||
mac_address: str = ":".join(
|
||||
[
|
||||
"{:02x}".format((uuid.getnode() >> ele) & 0xFF)
|
||||
for ele in range(0, 8 * 6, 8)
|
||||
][::-1]
|
||||
)
|
||||
self.logger.debug(f"MAC address: {mac_address}")
|
||||
return mac_address
|
||||
self.on_message: Callable[[Message], None] = on_message
|
||||
|
||||
def _send_ack(self) -> None:
|
||||
self.logger.debug("[>] Sending ACK...")
|
||||
@@ -73,29 +68,35 @@ class Connector:
|
||||
|
||||
return False
|
||||
|
||||
def connect(self, retry_interval: int = 1) -> None:
|
||||
self.logger.debug(
|
||||
f"Connecting to {self.host}:{self.port} with timeout {self.connect_timeout}s..."
|
||||
)
|
||||
try:
|
||||
self.socket.settimeout(self.connect_timeout)
|
||||
self.socket.connect((self.host, self.port))
|
||||
self.socket.settimeout(self.socket_timeout)
|
||||
self.logger.info(f"[+] Connected to {self.host}:{self.port}")
|
||||
self.send_hello()
|
||||
except (
|
||||
socket.timeout,
|
||||
ConnectionRefusedError,
|
||||
ConnectionAbortedError,
|
||||
) as e:
|
||||
self.logger.error(
|
||||
f"[!] Connection to {self.host}:{self.port} failed: {e}"
|
||||
def connect(self) -> None:
|
||||
retry_interval: int = 1
|
||||
connected: bool = False
|
||||
while not connected:
|
||||
self.logger.debug(
|
||||
f"[.] Connecting to {self.host}:{self.port} with timeout {self.connect_timeout}s..."
|
||||
)
|
||||
self.logger.info(
|
||||
f"[.] Retrying connection in {retry_interval} s..."
|
||||
)
|
||||
time.sleep(retry_interval)
|
||||
self.connect(retry_interval=min(30, retry_interval * 2))
|
||||
try:
|
||||
self.socket.settimeout(self.connect_timeout)
|
||||
self.socket.connect((self.host, self.port))
|
||||
self.socket.settimeout(self.socket_timeout)
|
||||
self.logger.info(f"[+] Connected to {self.host}:{self.port}")
|
||||
self.send_hello()
|
||||
connected = True
|
||||
except (
|
||||
socket.timeout,
|
||||
ConnectionRefusedError,
|
||||
ConnectionAbortedError,
|
||||
) as e:
|
||||
self.logger.error(
|
||||
f"[!] Connection to {self.host}:{self.port} failed: {e}"
|
||||
)
|
||||
self.logger.info(
|
||||
f"[.] Retrying connection in {retry_interval} s..."
|
||||
)
|
||||
time.sleep(retry_interval)
|
||||
retry_interval = min(
|
||||
retry_interval * 2, 30
|
||||
) # exponential backoff
|
||||
|
||||
def send(self, data: bytes, no_check_ack: bool = False) -> None:
|
||||
self.logger.debug(f"[>] Sending data: {data}")
|
||||
@@ -165,15 +166,28 @@ class Connector:
|
||||
time.sleep(1)
|
||||
|
||||
def _loop(self) -> None:
|
||||
self.logger.debug("Starting main loop...")
|
||||
self.logger.debug("Starting connector loop...")
|
||||
while True:
|
||||
time.sleep(1)
|
||||
time.sleep(0.1)
|
||||
data: bytes = self.receive()
|
||||
if not data:
|
||||
self.reconnect()
|
||||
continue
|
||||
message = Message.from_bytes(data.strip())
|
||||
self.logger.info(f"[<] Message received: {message}")
|
||||
for line in data.split(b"\n"):
|
||||
line: bytes = line.strip()
|
||||
|
||||
if not line:
|
||||
continue
|
||||
|
||||
self.logger.debug(f"[.] Raw message data: {line}")
|
||||
try:
|
||||
message: Message = Message.from_bytes(line)
|
||||
except ValueError as e:
|
||||
self.logger.error(f"[!] Failed to parse message: {e}")
|
||||
continue
|
||||
self.logger.info(f"[*] Message received: {message}")
|
||||
self.on_message(message)
|
||||
|
||||
# if self._check_ack():
|
||||
# self.logger.debug("[.] ACK verified")
|
||||
# else:
|
||||
|
||||
4
uv.lock
generated
4
uv.lock
generated
@@ -292,8 +292,8 @@ test = [
|
||||
|
||||
[[package]]
|
||||
name = "judas-protocol"
|
||||
version = "0.1.0"
|
||||
source = { git = "https://gitea.pufereq.pl/judas/judas_protocol.git#fd070b176347a0f7b81f937b189d8f50736f3514" }
|
||||
version = "0.3.0"
|
||||
source = { git = "https://gitea.pufereq.pl/judas/judas_protocol.git#c25ee1ebdfff8ff51bf00131732720091562e101" }
|
||||
|
||||
[[package]]
|
||||
name = "markdown-it-py"
|
||||
|
||||
Reference in New Issue
Block a user