From 97b6db348516a62de986eb0301e62e7019eb5053 Mon Sep 17 00:00:00 2001
From: lubeilin <1791778603@qq.com>
Date: Thu, 7 Mar 2024 12:25:52 +0800
Subject: [PATCH] =?UTF-8?q?=E8=BE=93=E5=87=BA=E5=A2=9E=E5=8A=A0=E9=A2=9C?=
 =?UTF-8?q?=E8=89=B2?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

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

diff --git a/vnt-cli/src/callback.rs b/vnt-cli/src/callback.rs
index 111bc7e..72698c7 100644
--- a/vnt-cli/src/callback.rs
+++ b/vnt-cli/src/callback.rs
@@ -1,5 +1,7 @@
 use std::process;
 
+use console::style;
+
 use vnt::handle::callback::ConnectInfo;
 use vnt::{DeviceInfo, ErrorInfo, HandshakeInfo, RegisterInfo, VntCallback};
 
@@ -8,7 +10,7 @@ pub struct VntHandler {}
 
 impl VntCallback for VntHandler {
     fn success(&self) {
-        println!(" ====== Connect Successfully ====== ")
+        println!(" {} ", style("====== Connect Successfully ======").green())
     }
     fn create_tun(&self, info: DeviceInfo) {
         println!("create_tun {}", info)
@@ -24,12 +26,12 @@ impl VntCallback for VntHandler {
     }
 
     fn register(&self, info: RegisterInfo) -> bool {
-        println!("register {}", info);
+        println!("register {}", style(info).green());
         true
     }
 
     fn error(&self, info: ErrorInfo) {
-        println!("error {}", info);
+        println!("{}", style(format!("error {}", info)).red());
     }
 
     fn stop(&self) {
diff --git a/vnt/src/handle/callback.rs b/vnt/src/handle/callback.rs
index 8a432e7..aeac7d5 100644
--- a/vnt/src/handle/callback.rs
+++ b/vnt/src/handle/callback.rs
@@ -165,6 +165,7 @@ pub enum ErrorType {
     AddressExhausted,
     IpAlreadyExists,
     InvalidIp,
+    LocalIpExists,
     Unknown,
 }
 
@@ -176,7 +177,8 @@ impl Into<u8> for ErrorType {
             ErrorType::AddressExhausted => 3,
             ErrorType::IpAlreadyExists => 4,
             ErrorType::InvalidIp => 5,
-            ErrorType::Unknown => 6,
+            ErrorType::LocalIpExists => 6,
+            ErrorType::Unknown => 255,
         }
     }
 }
diff --git a/vnt/src/handle/recv_data/server.rs b/vnt/src/handle/recv_data/server.rs
index 2207db9..bbe2cc7 100644
--- a/vnt/src/handle/recv_data/server.rs
+++ b/vnt/src/handle/recv_data/server.rs
@@ -262,7 +262,7 @@ impl<Call: VntCallback> ServerPacketHandler<Call> {
                         }
                         if let Err(e) = self.device.set_ip(virtual_ip, virtual_netmask) {
                             self.callback.error(ErrorInfo::new_msg(
-                                ErrorType::Unknown,
+                                ErrorType::LocalIpExists,
                                 format!("set_ip {:?}", e),
                             ));
                             return Ok(());