优化代码结构

vnt-cli/src/callback.rs

@@ -2,8 +2,9 @@ use std::process;
 
 use console::style;
 
-use vnt::handle::callback::{ConnectInfo, ErrorType};
-use vnt::{DeviceInfo, ErrorInfo, HandshakeInfo, RegisterInfo, VntCallback};
+use vnt::{
+    ConnectInfo, DeviceInfo, ErrorInfo, ErrorType, HandshakeInfo, RegisterInfo, VntCallback,
+};
 
 #[derive(Clone)]
 pub struct VntHandler {}

vnt/src/channel/mod.rs

@@ -138,7 +138,7 @@ impl RouteKey {
     }
 }
 
-pub fn init_context(
+pub(crate) fn init_context(
     ports: Vec<u16>,
     use_channel_type: UseChannelType,
     first_latency: bool,
@@ -233,7 +233,7 @@ pub fn init_context(
     Ok((context, tcp_listener))
 }
 
-pub fn init_channel<H>(
+pub(crate) fn init_channel<H>(
     tcp_listener: mio::net::TcpListener,
     context: ChannelContext,
     stop_manager: StopManager,

vnt/src/handle/handshaker.rs

@@ -19,13 +19,6 @@ use crate::proto::message::SecretHandshakeRequest;
 use crate::protocol::body::RSA_ENCRYPTION_RESERVED;
 use crate::protocol::{service_packet, NetPacket, Protocol, MAX_TTL};
 
-pub enum HandshakeEnum {
-    NotSecret,
-    KeyError,
-    Timeout,
-    ServerError(String),
-    Other(String),
-}
 #[derive(Clone)]
 pub struct Handshake {
     time: Arc<AtomicCell<Instant>>,

vnt/src/lib.rs

@@ -3,16 +3,16 @@ pub const VNT_VERSION: &'static str = env!("CARGO_PKG_VERSION");
 pub mod channel;
 pub mod cipher;
 pub mod core;
-pub mod external_route;
-pub mod handle;
+mod external_route;
+mod handle;
 #[cfg(feature = "ip_proxy")]
-pub mod ip_proxy;
+mod ip_proxy;
 pub mod nat;
 #[cfg(feature = "port_mapping")]
-pub mod port_mapping;
-pub mod proto;
+mod port_mapping;
+mod proto;
 pub mod protocol;
-pub mod tun_tap_device;
+mod tun_tap_device;
 pub mod util;
 
 pub use handle::callback::*;