edge set unauthorized on tcp disconnected

src/bin/sdlan/main.rs

@@ -1,9 +1,9 @@
 use sdlan_rs::get_edge;
+use sdlan_rs::run_sdlan;
+use sdlan_rs::CommandLine;
 use sdlan_sn_rs::log;
 use std::time::Duration;
 use tokio::sync::mpsc::Sender;
-use sdlan_rs::run_sdlan;
-use sdlan_rs::CommandLine;
 
 #[tokio::main]
 async fn main() {
@@ -23,6 +23,22 @@ async fn main() {
     let edge = get_edge();
     edge.start("0".to_owned()).await;
 
+    /*
     tokio::time::sleep(Duration::from_secs(20)).await;
     edge.stop().await;
+    */
+
+    let mut stream =
+        tokio::signal::unix::signal(tokio::signal::unix::SignalKind::user_defined1()).unwrap();
+
+    let mut started = true;
+    loop {
+        let sig = stream.recv().await;
+        if started {
+            edge.stop().await;
+        } else {
+            edge.start("0".to_owned()).await;
+        }
+        started = !started;
+    }
 }

src/network/async_main.rs

@@ -207,6 +207,9 @@ pub async fn async_main(
                 }
             })
         },
+        || async {
+            edge.set_authorized(false, vec![]);
+        },
         |msg| handle_tcp_message(msg),
         edge.tcp_pong.clone(),
         // tcp_pong,

src/tcp/tcp_conn.rs

@@ -52,15 +52,18 @@ impl ReadWriteActor {
         }
     }
 
-    pub async fn run<'a, T>(
+    pub async fn run<'a, T, T2, F>(
         &self,
         keep_reconnect: bool,
         mut to_tcp: Receiver<Vec<u8>>,
         on_connected: T,
+        on_disconnected: T2,
         mut start_stop_chan: Receiver<bool>,
         // cancel: CancellationToken,
     ) where
         T: for<'b> Fn(&'b mut TcpStream) -> BoxFuture<'b, ()>,
+        T2: Fn() -> F,
+        F: Future<Output = ()>,
     {
         // let (tx, rx) = channel(20);
         let mut started = false;
@@ -155,7 +158,7 @@ impl ReadWriteActor {
                     started = false;
                 }
             }
-
+            on_disconnected().await;
             println!("connect retrying");
             // future::select(read_from_tcp, write_to_tcp).await;
         }
@@ -185,9 +188,10 @@ impl ReadWriterHandle {
         Ok(())
     }
 
-    fn new<'a, T, T2, F>(
+    fn new<'a, T, T3, T2, F, F2>(
         addr: &str,
         on_connected: T,
+        on_disconnected: T3,
         on_message: T2,
         pong_time: Arc<AtomicU64>,
         start_stop_chan: Receiver<bool>,
@@ -195,15 +199,21 @@ impl ReadWriterHandle {
     ) -> Self
     where
         T: for<'b> Fn(&'b mut TcpStream) -> BoxFuture<'b, ()> + Send + 'static,
+        T3: Fn() -> F2 + Send + 'static,
         T2: Fn(SdlanTcp) -> F + Send + 'static,
         F: Future<Output = ()> + Send,
+        F2: Future<Output = ()> + Send,
     {
         let (send_to_tcp, to_tcp) = channel(20);
         let (from_tcp, mut data_from_tcp) = channel(20);
 
         let connected = Arc::new(AtomicBool::new(false));
         let actor = ReadWriteActor::new(addr, from_tcp, connected.clone(), pong_time);
-        tokio::spawn(async move { actor.run(true, to_tcp, on_connected, start_stop_chan).await });
+        tokio::spawn(async move {
+            actor
+                .run(true, to_tcp, on_connected, on_disconnected, start_stop_chan)
+                .await
+        });
         tokio::spawn(async move {
             loop {
                 if let Some(msg) = data_from_tcp.recv().await {
@@ -223,20 +233,29 @@ impl ReadWriterHandle {
     }
 }
 
-pub fn init_tcp_conn<'a, T, T2, F>(
+pub fn init_tcp_conn<'a, T, T3, T2, F, F2>(
     addr: &str,
     on_connected: T,
+    on_disconnected: T3,
     on_message: T2,
     pong_time: Arc<AtomicU64>,
     // cancel: CancellationToken,
     start_stop_chan: Receiver<bool>,
 ) where
     T: for<'b> Fn(&'b mut TcpStream) -> BoxFuture<'b, ()> + Send + 'static,
+    T3: Fn() -> F2 + Send + 'static,
     T2: Fn(SdlanTcp) -> F + Send + 'static,
     F: Future<Output = ()> + Send,
+    F2: Future<Output = ()> + Send,
 {
-    let tcp_handle =
-        ReadWriterHandle::new(addr, on_connected, on_message, pong_time, start_stop_chan);
+    let tcp_handle = ReadWriterHandle::new(
+        addr,
+        on_connected,
+        on_disconnected,
+        on_message,
+        pong_time,
+        start_stop_chan,
+    );
 
     GLOBAL_TCP_HANDLE
         .set(tcp_handle)