add spaces & add References

Cargo.toml

@@ -62,6 +62,7 @@ stn_buf = { version = "*", path = "stn_buf" }
 
 [target.'cfg(any(target_os = "linux", target_os = "android"))'.dependencies]
 libc = "0.2"
+stn_tun = { version = "*", path = "stn_tun" }
 stn_tproxy = { version = "*", path = "stn_tproxy" }
 
 [target.'cfg(target_os = "linux")'.dependencies]

README.md

@@ -27,3 +27,8 @@ The compiled file is `target/release/stn`.
 ## Script
 
 - [bat script to set http proxy](doc/http_bat.md)
+
+## References
+
+- github.com/changlan/kytan
+- github.com/shadowsocks/shadowsocks-rust

src/dns/util.rs

@@ -31,7 +31,7 @@ pub(crate) fn get_server_and_refresh_system(
             .collect::<Vec<SocketAddr>>();
 
         // refresh system
-        if let Err(e) = refresh_system(server.clone(), shared_server.clone()) {
+        if let Err(e) = refresh_system(server.clone(), &shared_server) {
             warn!("{}", e);
         };
         let interval = Duration::from_nanos(
@@ -44,7 +44,7 @@ pub(crate) fn get_server_and_refresh_system(
                 async move {
                     loop {
                         sleep(interval).await;
-                        if let Err(e) = refresh_system(server.clone(), shared_server.clone()) {
+                        if let Err(e) = refresh_system(server.clone(), &shared_server) {
                             warn!("{}", e);
                         };
                     }
@@ -63,7 +63,7 @@ pub(crate) fn get_server_and_refresh_system(
 
 pub(crate) fn refresh_system(
     mut origin_server: Vec<SocketAddr>,
-    shared_server: Arc<RwLock<Vec<SocketAddr>>>,
+    shared_server: &RwLock<Vec<SocketAddr>>,
 ) -> Result<(), Box<dyn std::error::Error>> {
     cfg_if::cfg_if! {
         if #[cfg(target_os = "windows")] {

src/socks5/in.rs

@@ -227,10 +227,7 @@ impl In {
         //  o  BND.PORT       server bound port in network octet order
 
         // send
-        let write_buf = match socketaddr_to_string(&client.local_addr().unwrap())
-            .parse()
-            .unwrap()
-        {
+        let write_buf = match socketaddr_to_string(&client.local_addr()?).parse()? {
             std::net::SocketAddr::V4(addr) => {
                 let mut buf = vec![5, 0, 0, ATYP_IPV4];
                 buf.extend(addr.ip().octets());

stn_http_proxy_server/Cargo.toml

@@ -6,6 +6,6 @@ edition = "2018"
 # See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
 
 [dependencies]
-tokio = { version = "1", features=[] }
-httparse = { version = "1.4", features=[] }
+tokio = { version = "1", features = [] }
+httparse = { version = "1.4", features = [] }
 futures = "0.3"

stn_tproxy/Cargo.toml

@@ -6,5 +6,5 @@ edition = "2018"
 # See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
 
 [dependencies]
-tokio = { version = "1", features=["net"] }
+tokio = { version = "1", features = ["net"] }
 libc = "0.2"