feat: new stuff

.idea/.gitignore

@@ -0,0 +1,8 @@
+# Default ignored files
+/shelf/
+/workspace.xml
+# Editor-based HTTP Client requests
+/httpRequests/
+# Datasource local storage ignored files
+/dataSources/
+/dataSources.local.xml

.idea/domo.iml

@@ -0,0 +1,14 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<module type="JAVA_MODULE" version="4">
+  <component name="NewModuleRootManager" inherit-compiler-output="true">
+    <exclude-output />
+    <content url="file://$MODULE_DIR$">
+      <sourceFolder url="file://$MODULE_DIR$/domo_node/src" isTestSource="false" />
+      <sourceFolder url="file://$MODULE_DIR$/domo_proto/src" isTestSource="false" />
+      <excludeFolder url="file://$MODULE_DIR$/domo_node/target" />
+      <excludeFolder url="file://$MODULE_DIR$/domo_proto/target" />
+    </content>
+    <orderEntry type="inheritedJdk" />
+    <orderEntry type="sourceFolder" forTests="false" />
+  </component>
+</module>

.idea/encodings.xml

@@ -0,0 +1,6 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project version="4">
+  <component name="Encoding">
+    <file url="file://$PROJECT_DIR$/domo_node/packet" charset="ISO-8859-1" />
+  </component>
+</project>

.idea/misc.xml

@@ -0,0 +1,10 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project version="4">
+  <component name="DiscordProjectSettings">
+    <option name="show" value="ASK" />
+    <option name="description" value="" />
+  </component>
+  <component name="ProjectRootManager">
+    <output url="file://$PROJECT_DIR$/out" />
+  </component>
+</project>

.idea/modules.xml

@@ -0,0 +1,8 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project version="4">
+  <component name="ProjectModuleManager">
+    <modules>
+      <module fileurl="file://$PROJECT_DIR$/.idea/domo.iml" filepath="$PROJECT_DIR$/.idea/domo.iml" />
+    </modules>
+  </component>
+</project>

.idea/vcs.xml

@@ -0,0 +1,6 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project version="4">
+  <component name="VcsDirectoryMappings">
+    <mapping directory="$PROJECT_DIR$" vcs="Git" />
+  </component>
+</project>

doc/domo_node/node_types.md

@@ -8,7 +8,7 @@ The job of a master node is rather complex.
 ## Forwarding packets
 The most straight forward job of the node.  
 It works by essentially having a hashmap with a reference to a socket.  
-When a packet comes in the node gets forwarded to the right socket.
+When a packet comes in the packet gets forwarded to the right node and socket.
 
 ## Handling network state
 The master node ensures there are no duplicate identifiers and therefore nodes.  
@@ -16,7 +16,7 @@ Also, it will make sure no invalid updates can be sent.
 Most packets go through the master node before getting to the source.  
 Most because the subnet node can have private nodes and handle those themselves.
 
-# The Bridge node (`relay`)
+# The Bridge node (`bridge`)
 The bridge node is simple.  
 In config you define where to forward the packets to, and they get delivered to there.  
 

domo_node/config.toml

@@ -1,5 +1,5 @@
 [node.type.master]
-bind = "127.0.0.1:4480"
+bind = "0.0.0.0:4480"
 
 [node]
 device_id = "ffffffff"

domo_node/src/app/master.rs

@@ -0,0 +1,72 @@
+use std::collections::HashMap;
+use std::io;
+use std::net::SocketAddr;
+use log::{info, trace};
+use domo_proto::data_types::DataType;
+use domo_proto::identifier::Identifier;
+
+pub struct Device {
+    /// Device identifier
+    pub device_id: Identifier,
+    /// Device socket address
+    pub socket_addr: SocketAddr,
+    /// List of properties and last state.
+    pub properties: HashMap<String, DataType>
+}
+
+pub struct Devices {
+    devices: HashMap<Identifier, Device>
+}
+
+impl Devices {
+    pub fn new() -> Self {
+        Self {
+            devices: HashMap::new()
+        }
+    }
+
+    pub fn add_device_random(&mut self, socket_addr: SocketAddr) -> Identifier {
+        let random_id = loop {
+            // TODO: random should avoid 0xFFFF----
+            let id = Identifier::random();
+            trace!("checking if id is taken: {}", id);
+            if !self.devices.contains_key(&id) { break id; }
+        };
+        self.devices.insert(random_id, Device {
+            device_id: random_id,
+            socket_addr,
+            properties: HashMap::new()
+        });
+        info!("Registered new device: {}", random_id);
+        random_id
+    }
+
+    pub fn add_device(&mut self, socket_addr: SocketAddr, identifier: Identifier) -> Identifier {
+        self.devices.insert(identifier, Device {
+            device_id: identifier,
+            socket_addr,
+            properties: HashMap::new()
+        });
+        identifier
+    }
+
+    pub fn add_device_auto(&mut self, socket_addr: SocketAddr, identifier: Option<Identifier>) -> Identifier {
+        let identifier = if identifier == Some(Identifier::default()) {
+            None
+        } else {
+            identifier
+        };
+        if let Some(identifier) = identifier {
+            match (identifier, self.devices.contains_key(&identifier)) {
+                (identifier, false) => self.add_device(socket_addr, identifier),
+                _ => self.add_device_random(socket_addr)
+            }
+        } else {
+            self.add_device_random(socket_addr)
+        }
+    }
+
+    pub fn get_device(&self, identifier: Identifier) -> Option<&Device> {
+        self.devices.get(&identifier)
+    }
+}

domo_node/src/app/mod.rs

@@ -1,20 +1,21 @@
 use std::io;
-use log::{error, trace, warn};
-use tokio::sync::mpsc::channel;
+use log::{error, info, trace, warn};
 use domo_proto::commands::node_management::NodeManagementCommand;
 use domo_proto::identifier::Identifier;
-use domo_proto::packet::{Packet, ToPacket};
+use domo_proto::packet::{ToPacket};
 use domo_proto::packet::packet_data::PacketData;
-use domo_proto::packet::raw_packet::RawPacket;
 use crate::{CONFIG, prelude};
+use crate::app::master::Devices;
 use crate::config::node::NodeType;
 use crate::connection::server::Server;
 
+pub mod master;
+
 impl NodeType {
     pub async fn start(self) -> io::Result<()> {
         match self {
             NodeType::Master { bind } => {
-                let server = Server::new(
+                let mut server = Server::new(
                     bind,
                     Some(
                         Identifier::from(
@@ -24,25 +25,51 @@ impl NodeType {
                     ),
                 ).await?;
 
+                info!("Started server at {}", server.sock().local_addr()?);
+
+                let mut devices = Devices::new();
+
                 {
-                    let p = NodeManagementCommand::Ping.to_packet(
+                    std::fs::write("./register_packet", Into::<Vec<u8>>::into(NodeManagementCommand::RegisterNode {
+                        device_id: Identifier::from(0x000000AA)
+                    }.to_packet(
                         Identifier::random(),
                         server.device_id(),
                         Identifier::random(),
                         Identifier::default(),
-                    );
-                    std::fs::write("./packet", Into::<Vec<u8>>::into(p))?;
+                    )))?;
+                    std::fs::write("./ping_packet", Into::<Vec<u8>>::into(NodeManagementCommand::Ping.to_packet(
+                        Identifier::from(0x000000AA),
+                        server.device_id(),
+                        Identifier::random(),
+                        Identifier::default(),
+                    )))?;
                 }
 
                 loop {
                     match server.recv_packet().await {
                         (Ok(p), Some(s)) => {
-                            if p.dest != server.device_id() {
-                                if let Some(d) = server.devices().get(&p.dest) {
-                                    let _ = &server.send(d.socket_addr, p).await;
-                                }
+                            if devices.get_device(p.src).is_none() {
+                                if let PacketData::NodeManagement(NodeManagementCommand::RegisterNode { device_id }) = p.data {
+                                    let device_id = devices.add_device_auto(s, Some(device_id));
+                                    let _ = server.send(s, NodeManagementCommand::RegisterNode {
+                                        device_id
+                                    }.to_packet(
+                                        server.device_id(),
+                                        device_id,
+                                        Identifier::random(),
+                                        p.packet_id
+                                    )).await;
+                                } else {
+                                    warn!("{s} is unregistered and tried to send a packet {p:#?}.");
+                                    let _ = server.send(s, prelude::quick_err::errc_not_registered(
+                                        server.device_id(),
+                                        p.src,
+                                        p.packet_id
+                                    )).await;
                                     continue;
                                 }
+                            }
                             match p.data {
                                 PacketData::NodeManagement(NodeManagementCommand::Ping) => {
                                     trace!("ping from: [{s}->{}]", p.src);
@@ -53,7 +80,7 @@ impl NodeType {
                                         p.packet_id,
                                     )).await;
                                 }
-                                _ => {}
+                                _ => warn!("dropped packet")
                             }
                         }
                         (Err(e), Some(source)) => {

domo_node/src/connection/server.rs

@@ -5,20 +5,15 @@ use std::net::SocketAddr;
 use std::io;
 use domo_proto::packet::{FULL_PACKET_SIZE, Packet};
 use std::io::ErrorKind;
-use log::error;
+use log::{error, trace};
+use domo_proto::data_types::DataType;
 use domo_proto::packet::raw_packet::RawPacket;
 
-pub struct Device {
-    /// Device identifier
-    pub device_id: Identifier,
-    /// Device socket address
-    pub socket_addr: SocketAddr,
-}
+
 
 pub struct Server {
     sock: UdpSocket,
     device_id: Identifier,
-    devices: HashMap<Identifier, Device>
 }
 
 impl Server {
@@ -26,7 +21,6 @@ impl Server {
         Ok(Self {
             sock: UdpSocket::bind(socket_addr).await?,
             device_id: device_id.unwrap_or(Identifier::random()),
-            devices: HashMap::new()
         })
     }
 
@@ -38,6 +32,8 @@ impl Server {
             Err(e) => return (Err(e), None)
         };
 
+        trace!("received: {} bytes from {}", size, socket_addr);
+
         (
             Packet::try_from(buf[..size].to_vec()),
             Some(socket_addr)
@@ -59,8 +55,4 @@ impl Server {
     pub fn sock(&self) -> &UdpSocket {
         &self.sock
     }
-
-    pub fn devices(&self) -> &HashMap<Identifier, Device> {
-        &self.devices
-    }
 }

domo_node/src/prelude.rs

@@ -39,4 +39,16 @@ pub mod quick_err {
             reply_to
         )
     }
+
+    pub fn errc_not_registered(src: Identifier, dest: Identifier, reply_to: Identifier) -> Packet {
+        NodeManagementCommand::Error {
+            error_code: NodeManagementError::errc_not_registered.into(),
+            metadata: "this node is not registered".to_string().into_bytes()
+        }.to_packet(
+            src,
+            dest,
+            Identifier::random(),
+            reply_to
+        )
+    }
 }

domo_node/test.py

@@ -0,0 +1,36 @@
+import socket
+import sys
+
+# Create a UDP socket
+udp_socket = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
+
+def send_file_contents(filename):
+    try:
+        print(f"reading {filename}...")
+        with open(filename, 'rb') as file:
+            file_contents = file.read()
+    except FileNotFoundError:
+        print(f"File '{filename}' not found.")
+        return
+
+    print(f"sending content len: {len(file_contents)}...")
+    # Send the file contents to the UDP socket
+    udp_socket.sendto(file_contents, ('0.0.0.0', 4480))
+
+    print("waiting for response...")
+    # Receive and display the response
+    response, server_address = udp_socket.recvfrom(1024)
+    print(f"Received response from {server_address}:")
+    print(response.hex())
+
+if __name__ == "__main__":
+    try:
+        while True:
+            filename = input("Enter a file name (or 'exit' to quit): ")
+            if filename == 'exit':
+                break
+            send_file_contents(filename)
+    except KeyboardInterrupt:
+        print("\nExiting the program.")
+    finally:
+        udp_socket.close()

domo_proto/src/commands/node_management/vec.rs

@@ -5,7 +5,9 @@ impl Into<Vec<u8>> for NodeManagementCommand {
     fn into(self) -> Vec<u8> {
         match self {
             NodeManagementCommand::Ping => vec![],
-            NodeManagementCommand::RegisterNode { device_id } => device_id.bytes.to_vec(),
+            NodeManagementCommand::RegisterNode { device_id } => {
+                vec![device_id.bytes[0], device_id.bytes[1], device_id.bytes[2], device_id.bytes[3]]
+            },
             NodeManagementCommand::RemoveNode => vec![],
             NodeManagementCommand::RegisterProperty { property_name, data_type, read_only } => {
                 let mut vec = vec![];

domo_proto/src/packet/packet_data.rs

@@ -31,7 +31,9 @@ impl PacketData {
 impl TryFrom<RawPacket> for PacketData {
     type Error = io::Error;
     fn try_from(raw_packet: RawPacket) -> io::Result<Self> {
-        match raw_packet.command {
+        match {
+            raw_packet.command >> 4
+        } {
             0x0 => Ok(PacketData::NodeManagement(NodeManagementCommand::try_from(raw_packet)?)),
             0x1 => Ok(PacketData::PropertyControl(PropertyControlCommand::try_from(raw_packet)?)),
             _ => Err(io::Error::new(io::ErrorKind::InvalidInput, "command group is unsupported"))

domo_proto/src/packet/raw_packet/packet.rs

@@ -40,6 +40,10 @@ impl Into<Packet> for RawPacket {
             reply_to: Identifier::from(self.reply_to),
             command: self.command,
             data: PacketData::try_from(self.clone())
+                .map_err(|e| {
+                    println!("{}", e);
+                    e
+                })
                 .unwrap_or(PacketData::Raw(self.data)),
         }
     }

domo_proto/src/packet/raw_packet/vec.rs

@@ -10,10 +10,11 @@ impl TryFrom<Vec<u8>> for RawPacket {
             return Err(io::Error::new(io::ErrorKind::InvalidData, "Can't parse data into RawPacket"))
         }
         let data_length = (((data[0x12] as u16) << 8) | data[0x13] as u16) as usize;
+        println!("LENGTH {data_length}");
         if data.len() < 0x14 + data_length {
             return Err(io::Error::new(io::ErrorKind::InvalidData, "Can't parse data into RawPacket"))
         }
-        let checksum = as_u32_be(data[(data.len() - 4)..].as_ref());
+        let checksum = as_u32_be(data[(0x14 + data_length)..].as_ref());
         let built_checksum = crc32fast::hash(data[..(data.len() - 4)].as_ref());
         if checksum != built_checksum {
             return Err(io::Error::new(io::ErrorKind::InvalidData, format!("Checksum does not match {checksum:X?} != {built_checksum:X?}")))

recv_server.py

@@ -1,45 +0,0 @@
-
-import socket, zlib
-
-HOST = "127.0.0.1"  # Standard loopback interface address (localhost)
-PORT = 2000  # Port to listen on (non-privileged ports are > 1023)
-
-def calculate_crc(data):
-    # Calculate the CRC32 checksum of the data
-    crc = zlib.crc32(data)
-    # Convert the CRC32 value to a 4-byte big-endian byte array
-    crc_bytes = crc.to_bytes(4, byteorder='big')
-    return crc_bytes
-
-with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as s:
-    s.bind((HOST, PORT))
-    s.listen()
-    while True:
-        try:
-            conn, addr = s.accept()
-            with conn:
-                print(f"Connected by {addr}")
-                while True:
-                    data = conn.recv(1024)
-                    if not data:
-                        break
-                    print(f'RECV: {data.hex()} [{len(data)}]')
-                    if data[0x11] == 0x01:
-                        pckt = bytearray([0x01])
-                        pckt.extend([0x00, 0x00, 0x00, 0x00]) # src
-                        pckt.extend([0x00, 0x00, 0x00, 0xFF]) # dest
-                        pckt.extend([0x00, 0x00, 0x00, 0x00]) # packet_id
-                        pckt.extend(data[0x09:0x0D]) # reply_to
-                        pckt.extend([0x01]) # command
-                        pckt.extend([0x00, 0x04]) # data_length
-                        pckt.extend([0x00, 0x00, 0x00, 0xFF])
-
-                        # Calculate CRC for the received data
-                        crc = calculate_crc(pckt)
-                        # Append the CRC value to the data
-                        data_with_crc = pckt + crc
-                        # Send the modified data (with CRC) back to the client
-                        conn.sendall(data_with_crc)
-                        print(f'SEND: {data_with_crc.hex()} [{len(data_with_crc)}]')
-        except:
-            s.close()