feat secure

examples/examples/benchmark_pubsub.rs

@@ -35,8 +35,9 @@ enum ImplSdkEvent {
 async fn run_node(node_id: NodeId, seeds: Vec<NodeAddr>) -> (PubsubSdk, NodeAddr) {
     log::info!("Run node {} connect to {:?}", node_id, seeds);
     let mut node_addr_builder = NodeAddrBuilder::new(node_id);
+    let secure = Arc::new(atm0s_sdn::StaticKeySecure::new("secure-token"));
     let socket = UdpTransport::prepare(0, &mut node_addr_builder).await;
-    let transport = Box::new(UdpTransport::new(node_addr_builder.addr(), socket));
+    let transport = Box::new(UdpTransport::new(node_addr_builder.addr(), socket, secure));
     let timer = Arc::new(SystemTimer());
 
     let router = SharedRouter::new(node_id);

examples/examples/benchmark_pubsub_local.rs

@@ -34,9 +34,10 @@ enum ImplSdkEvent {
 
 async fn run_node(node_id: NodeId, seeds: Vec<NodeAddr>) -> (PubsubSdk, NodeAddr) {
     log::info!("Run node {} connect to {:?}", node_id, seeds);
+    let secure = Arc::new(atm0s_sdn::StaticKeySecure::new("secure-token"));
     let mut node_addr_builder = NodeAddrBuilder::new(node_id);
     let socket = UdpTransport::prepare(0, &mut node_addr_builder).await;
-    let transport = Box::new(UdpTransport::new(node_addr_builder.addr(), socket));
+    let transport = Box::new(UdpTransport::new(node_addr_builder.addr(), socket, secure));
     let timer = Arc::new(SystemTimer());
 
     let router = SharedRouter::new(node_id);

examples/examples/benchmark_transport.rs

@@ -1,3 +1,4 @@
+use std::sync::Arc;
 use std::time::Duration;
 
 use atm0s_sdn::RouteRule;
@@ -7,13 +8,14 @@ use atm0s_sdn::{NodeAddrBuilder, UdpTransport};
 #[async_std::main]
 async fn main() {
     env_logger::builder().format_timestamp_millis().filter_level(log::LevelFilter::Info).init();
+    let secure = Arc::new(atm0s_sdn::StaticKeySecure::new("secure-token"));
     let mut node_addr_builder1 = NodeAddrBuilder::new(1);
     let socket1 = UdpTransport::prepare(0, &mut node_addr_builder1).await;
-    let mut transport1 = Box::new(UdpTransport::new(node_addr_builder1.addr(), socket1));
+    let mut transport1 = Box::new(UdpTransport::new(node_addr_builder1.addr(), socket1, secure.clone()));
 
     let mut node_addr_builder2 = NodeAddrBuilder::new(2);
     let socket2 = UdpTransport::prepare(0, &mut node_addr_builder2).await;
-    let mut transport2 = Box::new(UdpTransport::new(node_addr_builder2.addr(), socket2));
+    let mut transport2 = Box::new(UdpTransport::new(node_addr_builder2.addr(), socket2, secure));
 
     let task = async_std::task::spawn(async move {
         loop {

examples/examples/chat_example.rs

@@ -125,8 +125,9 @@ async fn main() {
     // In this example, we use the UDP transport layer.
     // There are also other transport layers, such as TCP and VNET, others are still in progress.
     // The port number is 50000 + node_id.
+    let secure = Arc::new(atm0s_sdn::StaticKeySecure::new("secure-token"));
     let socket = UdpTransport::prepare(50000 + args.node_id as u16, &mut node_addr_builder).await;
-    let transport = UdpTransport::new(node_addr_builder.addr(), socket);
+    let transport = UdpTransport::new(node_addr_builder.addr(), socket, secure);
     let node_addr = node_addr_builder.addr();
     println!("Listening on addr {}", node_addr);
 

examples/examples/discovery_manual_node.rs

@@ -51,7 +51,8 @@ async fn main() {
     let args: Args = Args::parse();
     let mut node_addr_builder = NodeAddrBuilder::new(args.node_id);
     let socket = UdpTransport::prepare(0, &mut node_addr_builder).await;
-    let transport = Box::new(UdpTransport::new(node_addr_builder.addr(), socket));
+    let secure = Arc::new(atm0s_sdn::StaticKeySecure::new("secure-token"));
+    let transport = Box::new(UdpTransport::new(node_addr_builder.addr(), socket, secure));
     let node_addr = node_addr_builder.addr();
     log::info!("Listen on addr {}", node_addr);
 

examples/examples/manual_node.rs

@@ -74,8 +74,9 @@ async fn main() {
     let udp_socket = UdpTransport::prepare(50000 + args.node_id as u16, &mut node_addr_builder).await;
     let tcp_listener = TcpTransport::prepare(50000 + args.node_id as u16, &mut node_addr_builder).await;
 
-    let udp = UdpTransport::new(node_addr_builder.addr(), udp_socket);
-    let tcp = TcpTransport::new(node_addr_builder.addr(), tcp_listener);
+    let secure = Arc::new(atm0s_sdn::StaticKeySecure::new("secure-token"));
+    let udp = UdpTransport::new(node_addr_builder.addr(), udp_socket, secure.clone());
+    let tcp = TcpTransport::new(node_addr_builder.addr(), tcp_listener, secure);
 
     let transport = UdpTcpTransport::new(udp, tcp);
 

packages/network/Cargo.toml

@@ -21,6 +21,7 @@ thiserror = { workspace = true }
 serde = { workspace = true }
 bytes = "1.5.0"
 bincode = "1.3.3"
+sha1 = "0.10.6"
 
 [dev-dependencies]
 env_logger = { workspace = true }

packages/network/src/lib.rs

@@ -1,6 +1,7 @@
 pub mod behaviour;
 pub mod msg;
 pub mod plane;
+pub mod secure;
 pub mod transport;
 pub mod transport_tests;