supernode related

Cargo.toml

@@ -12,6 +12,7 @@ byteorder = "1.5.0"
 cbc = "0.1.2"
 dashmap = "5.5.3"
 lazy_static = "1.4.0"
+once_cell = "1.19.0"
 rand = "0.8.5"
 rolling-file = { git = "https://git.asxalex.pw/rust/rolling-file" }
 rsa = "0.9.6"

src/bin/sdlan-sn/config/mod.rs

@@ -0,0 +1,3 @@
+pub const DEFAULT_MIN_AUTO_IP_NET: &'static str = "10.128.0.0";
+pub const DEFAULT_MAX_AUTO_IP_NET: &'static str = "10.255.255.0";
+pub const DEFAULT_IP_NET_BIT_LEN: u8 = 24;

src/bin/sdlan-sn/main.rs

@@ -8,6 +8,7 @@ use std::net::SocketAddr;
 use std::time::Duration;
 use tokio::net::UdpSocket;
 
+mod config;
 mod utils;
 
 use utils::license_ok;
@@ -43,16 +44,21 @@ async fn main() -> Result<()> {
 
     let listener = UdpSocket::bind(SERVER).await?;
 
+    let supernode = utils::SuperNode::new(listener);
+    utils::init_supernode(supernode);
+
     tokio::spawn(async {
         client(SERVER).await.unwrap();
     });
 
+    let listener = utils::get_supernode();
     loop {
         // let mut buffer = [0; 2048];
         let mut buffer = vec![0; 2048];
         // let mut data = Vec::with_capacity(2048);
-        let (n, addr) = listener.recv_from(&mut buffer).await?;
+        let (n, addr) = listener.sock.recv_from(&mut buffer).await?;
         buffer.truncate(n);
+
         tokio::spawn(async move {
             if let Err(e) = handle_packet(&buffer, addr).await {
                 error!("failed to handle packet: {:?}", e);

src/bin/sdlan-sn/utils/license.rs

@@ -0,0 +1,25 @@
+use sdlan_sn_rs::utils::Result;
+use serde::{Deserialize, Serialize};
+use tracing::error;
+
+#[derive(Serialize, Deserialize)]
+struct LicenseInfo<'a> {
+    // 由谁颁发的
+    #[serde(rename = "from")]
+    from: &'a str,
+    // 颁发给谁的
+    #[serde(rename = "to")]
+    to: &'a str,
+    // 有效起始时间
+    #[serde(rename = "starts")]
+    starts: i64,
+    // 有效结束时间
+    #[serde(rename = "ends")]
+    ends: i64,
+}
+
+pub fn license_ok() -> Result<()> {
+    // TODO: check license
+    error!("license expired");
+    Ok(())
+}

src/bin/sdlan-sn/utils/mod.rs

@@ -1,25 +1,5 @@
-use sdlan_sn_rs::utils::{rsa_decrypt, Result};
-use serde::{Deserialize, Serialize};
-use tracing::error;
+mod sn;
+pub use sn::*;
 
-#[derive(Serialize, Deserialize)]
-struct LicenseInfo<'a> {
-    // 由谁颁发的
-    #[serde(rename = "from")]
-    from: &'a str,
-    // 颁发给谁的
-    #[serde(rename = "to")]
-    to: &'a str,
-    // 有效起始时间
-    #[serde(rename = "starts")]
-    starts: i64,
-    // 有效结束时间
-    #[serde(rename = "ends")]
-    ends: i64,
-}
-
-pub fn license_ok() -> Result<()> {
-    // TODO: check license
-    error!("license expired");
-    Ok(())
-}
+mod license;
+pub use license::*;

src/bin/sdlan-sn/utils/sn.rs

@@ -0,0 +1,105 @@
+use dashmap::DashMap;
+use sdlan_sn_rs::peer::IpSubnet;
+use tokio::net::UdpSocket;
+
+use std::{
+    sync::atomic::{AtomicU32, AtomicU8},
+    time::{SystemTime, UNIX_EPOCH},
+};
+
+use sdlan_sn_rs::utils::{Result, SDLanError};
+
+use once_cell::sync::OnceCell;
+
+static SN: OnceCell<SuperNode> = OnceCell::new();
+
+pub fn init_supernode(sn: SuperNode) -> Result<()> {
+    if let Err(e) = SN.set(sn) {
+        return Err(SDLanError::NormalError("initialize sn error"));
+    }
+    Ok(())
+}
+
+pub fn get_supernode() -> &'static SuperNode {
+    SN.get().expect("Supernode has not been initialized")
+}
+
+pub struct SuperNode {
+    // supernode 启动时间
+    pub start_time: u64,
+    // 是否以daemon，后台运行。
+    pub daemon: bool,
+
+    // 本地监听的udp端口号
+    pub local_port: i16,
+
+    // 该SuperNode包含的所有的网络
+    // 在云端创建或者删除网络的时候，需要通知sn修改这个网络信息
+    pub networks: DashMap<String, Network>,
+
+    /// 把所有其他的supernode放在这个fedration里面，
+    /// 在广播或者收到消息发现本地不存在这个目标数据包的时候，
+    /// 需要把数据包广播到这个fedration中。
+    pub fedration: Network,
+
+    /// pending网络，所有未授权的放到这个网络里面
+    /// 当某个SN收到来自客户端的信息的时候，需要把客户端加到这个pending
+    /// 在数据库中保存一条记录，并且在redis中记录一条信息，表明这个客户端
+    /// 在我这个Supernode上
+    pub pending: Network,
+
+    pub ip_range: AutoIpAssign,
+
+    pub sock: UdpSocket,
+
+    pub auth_key: String,
+}
+
+impl SuperNode {
+    pub fn new(sock: UdpSocket) -> Self {
+        let now = SystemTime::now();
+        println!("now: {:?}", now);
+        let start_time = now
+            .duration_since(UNIX_EPOCH)
+            .expect("time went backwards")
+            .as_secs();
+
+        Self {
+            start_time,
+            daemon: false,
+            local_port: 7655,
+            networks: DashMap::new(),
+            fedration: Network::new("*fedration"),
+            pending: Network::new("*pending"),
+            ip_range: AutoIpAssign {
+                start_ip: IpSubnet {
+                    net_addr: AtomicU32::new(0),
+                    net_bit_len: AtomicU8::new(0),
+                },
+                end_ip: IpSubnet {
+                    net_addr: AtomicU32::new(0),
+                    net_bit_len: AtomicU8::new(0),
+                },
+            },
+            sock,
+            auth_key: String::from("encrypt!"),
+        }
+    }
+}
+
+pub struct AutoIpAssign {
+    pub start_ip: IpSubnet,
+    pub end_ip: IpSubnet,
+}
+
+pub struct Network {
+    name: String,
+}
+
+impl Network {
+    fn new(name: &str) -> Self {
+        Self {
+            name: name.to_string(),
+        }
+    }
+}

src/bin/sdlan/main.rs

@@ -1,2 +1,4 @@
+use tokio::net::UdpSocket;
+
 #[tokio::main]
 async fn main() {}

src/utils/mod.rs

@@ -14,7 +14,7 @@ pub mod test_utils {
 
     pub fn generate_info(n: i32) -> Vec<u8> {
         let mut result = vec![];
-        for i in 0..n {
+        for _ in 0..n {
             let v: u8 = thread_rng().gen_range(0..255);
             result.push(v);
         }