From ae63ce80b5287a3e7facbd200af74735994bbf65 Mon Sep 17 00:00:00 2001
From: lubeilin <1791778603@qq.com>
Date: Wed, 6 Mar 2024 22:53:56 +0800
Subject: [PATCH] =?UTF-8?q?[mio]=20=E5=90=AF=E5=8A=A8=E6=88=90=E5=8A=9F?=
 =?UTF-8?q?=E8=BE=93=E5=87=BA?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 vnt-cli/src/callback.rs            | 3 +++
 vnt-jni/src/callback.rs            | 2 ++
 vnt/src/handle/callback.rs         | 3 +++
 vnt/src/handle/recv_data/server.rs | 1 +
 4 files changed, 9 insertions(+)

diff --git a/vnt-cli/src/callback.rs b/vnt-cli/src/callback.rs
index 4aeaa1a..111bc7e 100644
--- a/vnt-cli/src/callback.rs
+++ b/vnt-cli/src/callback.rs
@@ -7,6 +7,9 @@ use vnt::{DeviceInfo, ErrorInfo, HandshakeInfo, RegisterInfo, VntCallback};
 pub struct VntHandler {}
 
 impl VntCallback for VntHandler {
+    fn success(&self) {
+        println!(" ====== Connect Successfully ====== ")
+    }
     fn create_tun(&self, info: DeviceInfo) {
         println!("create_tun {}", info)
     }
diff --git a/vnt-jni/src/callback.rs b/vnt-jni/src/callback.rs
index 3c3d580..1812766 100644
--- a/vnt-jni/src/callback.rs
+++ b/vnt-jni/src/callback.rs
@@ -145,6 +145,8 @@ impl CallBack {
 }
 
 impl VntCallback for CallBack {
+    fn success(&self) {
+    }
     fn create_tun(&self, info: DeviceInfo) {
         if let Err(e) = self.create_tun0(info) {
             log::warn!("create_tun {:?}", e);
diff --git a/vnt/src/handle/callback.rs b/vnt/src/handle/callback.rs
index 0cc353d..8a432e7 100644
--- a/vnt/src/handle/callback.rs
+++ b/vnt/src/handle/callback.rs
@@ -182,6 +182,9 @@ impl Into<u8> for ErrorType {
 }
 
 pub trait VntCallback: Clone + Send + Sync + 'static {
+    /// 启动成功
+    fn success(&self) {}
+
     /// 创建网卡的信息
     fn create_tun(&self, _info: DeviceInfo) {}
     /// 连接
diff --git a/vnt/src/handle/recv_data/server.rs b/vnt/src/handle/recv_data/server.rs
index 3a6c7fe..2207db9 100644
--- a/vnt/src/handle/recv_data/server.rs
+++ b/vnt/src/handle/recv_data/server.rs
@@ -307,6 +307,7 @@ impl<Call: VntCallback> ServerPacketHandler<Call> {
                                 guard.push((dest, mask));
                             }
                         }
+                        self.callback.success();
                     }
                     self.set_device_info_list(response.device_info_list, response.epoch as _);
                 }