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
|
import logging as lg
|
||||||
|
|
||||||
from judas_client.connector import Connector
|
from judas_client.client import Client
|
||||||
|
|
||||||
if __name__ == "__main__":
|
if __name__ == "__main__":
|
||||||
lg.basicConfig(
|
lg.basicConfig(
|
||||||
@@ -12,5 +12,5 @@ if __name__ == "__main__":
|
|||||||
)
|
)
|
||||||
logger = lg.getLogger(__name__)
|
logger = lg.getLogger(__name__)
|
||||||
|
|
||||||
connector = Connector("127.0.0.1", 3692)
|
client = Client(server_host="localhost", server_port=3692)
|
||||||
connector.run()
|
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 logging as lg
|
||||||
import socket
|
import socket
|
||||||
import time
|
import time
|
||||||
import uuid
|
from typing import Callable
|
||||||
|
|
||||||
from judas_protocol import Message
|
from judas_protocol import Message
|
||||||
|
|
||||||
@@ -12,10 +12,13 @@ from judas_protocol import Message
|
|||||||
class Connector:
|
class Connector:
|
||||||
def __init__(
|
def __init__(
|
||||||
self,
|
self,
|
||||||
|
mac_address: str,
|
||||||
host: str,
|
host: str,
|
||||||
port: int,
|
port: int,
|
||||||
|
*,
|
||||||
connect_timeout: float = 5.0,
|
connect_timeout: float = 5.0,
|
||||||
ack_timeout: float = None,
|
ack_timeout: float | None = None,
|
||||||
|
on_message: Callable[[Message], None],
|
||||||
) -> None:
|
) -> None:
|
||||||
self.logger: lg.Logger = lg.getLogger(
|
self.logger: lg.Logger = lg.getLogger(
|
||||||
f"{__name__}.{self.__class__.__name__}"
|
f"{__name__}.{self.__class__.__name__}"
|
||||||
@@ -26,23 +29,15 @@ class Connector:
|
|||||||
self.port: int = port
|
self.port: int = port
|
||||||
self.socket_timeout: None = None
|
self.socket_timeout: None = None
|
||||||
self.connect_timeout: float = connect_timeout
|
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(
|
self.socket: socket.socket = socket.socket(
|
||||||
socket.AF_INET, socket.SOCK_STREAM
|
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:
|
self.on_message: Callable[[Message], None] = on_message
|
||||||
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
|
|
||||||
|
|
||||||
def _send_ack(self) -> None:
|
def _send_ack(self) -> None:
|
||||||
self.logger.debug("[>] Sending ACK...")
|
self.logger.debug("[>] Sending ACK...")
|
||||||
@@ -73,9 +68,12 @@ class Connector:
|
|||||||
|
|
||||||
return False
|
return False
|
||||||
|
|
||||||
def connect(self, retry_interval: int = 1) -> None:
|
def connect(self) -> None:
|
||||||
|
retry_interval: int = 1
|
||||||
|
connected: bool = False
|
||||||
|
while not connected:
|
||||||
self.logger.debug(
|
self.logger.debug(
|
||||||
f"Connecting to {self.host}:{self.port} with timeout {self.connect_timeout}s..."
|
f"[.] Connecting to {self.host}:{self.port} with timeout {self.connect_timeout}s..."
|
||||||
)
|
)
|
||||||
try:
|
try:
|
||||||
self.socket.settimeout(self.connect_timeout)
|
self.socket.settimeout(self.connect_timeout)
|
||||||
@@ -83,6 +81,7 @@ class Connector:
|
|||||||
self.socket.settimeout(self.socket_timeout)
|
self.socket.settimeout(self.socket_timeout)
|
||||||
self.logger.info(f"[+] Connected to {self.host}:{self.port}")
|
self.logger.info(f"[+] Connected to {self.host}:{self.port}")
|
||||||
self.send_hello()
|
self.send_hello()
|
||||||
|
connected = True
|
||||||
except (
|
except (
|
||||||
socket.timeout,
|
socket.timeout,
|
||||||
ConnectionRefusedError,
|
ConnectionRefusedError,
|
||||||
@@ -95,7 +94,9 @@ class Connector:
|
|||||||
f"[.] Retrying connection in {retry_interval} s..."
|
f"[.] Retrying connection in {retry_interval} s..."
|
||||||
)
|
)
|
||||||
time.sleep(retry_interval)
|
time.sleep(retry_interval)
|
||||||
self.connect(retry_interval=min(30, retry_interval * 2))
|
retry_interval = min(
|
||||||
|
retry_interval * 2, 30
|
||||||
|
) # exponential backoff
|
||||||
|
|
||||||
def send(self, data: bytes, no_check_ack: bool = False) -> None:
|
def send(self, data: bytes, no_check_ack: bool = False) -> None:
|
||||||
self.logger.debug(f"[>] Sending data: {data}")
|
self.logger.debug(f"[>] Sending data: {data}")
|
||||||
@@ -165,15 +166,28 @@ class Connector:
|
|||||||
time.sleep(1)
|
time.sleep(1)
|
||||||
|
|
||||||
def _loop(self) -> None:
|
def _loop(self) -> None:
|
||||||
self.logger.debug("Starting main loop...")
|
self.logger.debug("Starting connector loop...")
|
||||||
while True:
|
while True:
|
||||||
time.sleep(1)
|
time.sleep(0.1)
|
||||||
data: bytes = self.receive()
|
data: bytes = self.receive()
|
||||||
if not data:
|
if not data:
|
||||||
self.reconnect()
|
self.reconnect()
|
||||||
continue
|
continue
|
||||||
message = Message.from_bytes(data.strip())
|
for line in data.split(b"\n"):
|
||||||
self.logger.info(f"[<] Message received: {message}")
|
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():
|
# if self._check_ack():
|
||||||
# self.logger.debug("[.] ACK verified")
|
# self.logger.debug("[.] ACK verified")
|
||||||
# else:
|
# else:
|
||||||
|
|||||||
4
uv.lock
generated
4
uv.lock
generated
@@ -292,8 +292,8 @@ test = [
|
|||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "judas-protocol"
|
name = "judas-protocol"
|
||||||
version = "0.1.0"
|
version = "0.3.0"
|
||||||
source = { git = "https://gitea.pufereq.pl/judas/judas_protocol.git#fd070b176347a0f7b81f937b189d8f50736f3514" }
|
source = { git = "https://gitea.pufereq.pl/judas/judas_protocol.git#c25ee1ebdfff8ff51bf00131732720091562e101" }
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "markdown-it-py"
|
name = "markdown-it-py"
|
||||||
|
|||||||
Reference in New Issue
Block a user