distributed-state-network 0.6.2__tar.gz → 0.6.4__tar.gz

{distributed_state_network-0.6.2 → distributed_state_network-0.6.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: distributed-state-network
-Version: 0.6.2
+Version: 0.6.4
 Summary: A tool to distribute the state of a network device to other devices on the network
 Project-URL: Homepage, https://github.com/erinclemmer/distributed_state_network
 Project-URL: Issues, https://github.com/erinclemmer/distributed_state_network/issues

{distributed_state_network-0.6.2 → distributed_state_network-0.6.4}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "distributed-state-network"
-version = "0.6.2"
+version = "0.6.4"
 authors = [{name="Erin Clemmer", email="erin.c.clemmer@gmail.com"}]
 description = "A tool to distribute the state of a network device to other devices on the network"
 keywords = ["distributed", "networking"]

{distributed_state_network-0.6.2 → distributed_state_network-0.6.4}/src/distributed_state_network/handler.py

@@ -8,7 +8,7 @@ from distributed_state_network.objects.config import DSNodeConfig
 from distributed_state_network.util.aes import generate_aes_key
 from distributed_state_network.util import stop_thread
 
-VERSION = "0.6.2"
+VERSION = "0.6.4"
 logging.basicConfig(level=logging.INFO)
 
 # Silence Flask and Werkzeug logging
@@ -19,6 +19,7 @@ MSG_HELLO = 1
 MSG_PEERS = 2
 MSG_UPDATE = 3
 MSG_PING = 4
+MSG_DATA = 5
 
 class DSNodeServer:
     def __init__(
@@ -60,6 +61,11 @@ class DSNodeServer:
         def handle_ping_route():
             return self._handle_request(MSG_PING, request.data, request.remote_addr)
 
+        @self.app.route('/data', methods=['POST'])
+        def handle_data_route():
+            return self._handle_request(MSG_DATA, request.data, request.remote_addr)
+
+
     def _handle_request(self, msg_type: int, data: bytes, remote_addr: str) -> Response:
         if not self.running:
             return Response(status=500)
@@ -94,6 +100,9 @@ class DSNodeServer:
                 
             elif msg_type == MSG_PING:
                 response_data = b''
+
+            elif msg_type == MSG_DATA:
+                response_data = self.node.receive_data(body)
             
             # Send response if handler returned data
             if response_data is not None:
@@ -180,4 +189,7 @@ class DSNodeServer:
         self.node.update_cb = cb
 
     def set_disconnect_cb(self, cb: Callable):
-        self.node.disconnect_cb = cb
+        self.node.disconnect_cb = cb
+
+    def receive_data(self, data: bytes):
+        self.node.receive_cb(data)

{distributed_state_network-0.6.2 → distributed_state_network-0.6.4}/tests/test.py

@@ -13,6 +13,7 @@ from distributed_state_network import DSNodeServer, Endpoint, DSNodeConfig
 
 from distributed_state_network.objects.state_packet import StatePacket
 from distributed_state_network.objects.hello_packet import HelloPacket
+from distributed_state_network.objects.data_packet import DataPacket
 
 from distributed_state_network.util.aes import generate_aes_key
 
@@ -481,7 +482,7 @@ class TestNode(unittest.TestCase):
         n = spawn_node("http-test-node")
         
         # Test that all endpoints exist
-        endpoints = ['/hello', '/peers', '/update', '/ping']
+        endpoints = ['/hello', '/peers', '/update', '/ping', '/data']
         
         for endpoint in endpoints:
             # Send a request to each endpoint (will fail due to encryption, but should return 400/500, not 404)
@@ -497,5 +498,56 @@ class TestNode(unittest.TestCase):
             except Exception as e:
                 print(f"Endpoint {endpoint} test failed: {e}")
 
+    def test_send_to_node_success(self):
+        bootstrap = spawn_node("bootstrap")
+        connector = spawn_node("connector", [bootstrap.node.my_con().to_json()])
+
+        # Direct DSNode API
+        resp = connector.node.send_to_node("bootstrap", b"Hello, world!")
+        self.assertEqual(resp, "OK")
+
+    def test_send_to_node_wrapper(self):
+        bootstrap = spawn_node("bootstrap")
+        connector = spawn_node("connector", [bootstrap.node.my_con().to_json()])
+
+        # DSNodeServer wrapper
+        resp = connector.send_to_node("bootstrap", b"Hello via wrapper")
+        self.assertEqual(resp, "OK")
+
+    def test_data_route_unknown_sender(self):
+        # Start a single node; it should reject data packets from unknown senders
+        n = spawn_node("node")
+        time.sleep(0.5)  # Allow server to start
+
+        # Craft a DataPacket with a node_id that's not in address book
+        pkt = DataPacket("unknown", b"", b"payload")
+        encrypted = n.node.encrypt_data(bytes([5]) + pkt.to_bytes())  # MSG_DATA = 5
+
+        response = requests.post(
+            f'http://127.0.0.1:{n.config.port}/data',
+            data=encrypted,
+            headers={'Content-Type': 'application/octet-stream'},
+            timeout=2
+        )
+        self.assertEqual(response.status_code, 401)
+
+    def test_data_route_bad_signature(self):
+        # Connect two nodes so bootstrap knows connector's public key
+        bootstrap = spawn_node("bootstrap")
+        connector = spawn_node("connector", [bootstrap.node.my_con().to_json()])
+
+        # Create a packet claiming to be from connector but without a valid signature
+        pkt = DataPacket("connector", b"", b"payload")
+        encrypted = bootstrap.node.encrypt_data(bytes([5]) + pkt.to_bytes())  # MSG_DATA = 5
+
+        response = requests.post(
+            f'http://127.0.0.1:{bootstrap.config.port}/data',
+            data=encrypted,
+            headers={'Content-Type': 'application/octet-stream'},
+            timeout=2
+        )
+        # Should be unauthorized due to bad signature
+        self.assertEqual(response.status_code, 401)
+
 if __name__ == "__main__":
     unittest.main()