wip

Signed-off-by: Eval EXEC <[email protected]>

Author: eval-exec

Cargo.lock

@@ -35,6 +35,7 @@ bloom-filters = "0.1"
 ckb-spawn = { path = "../util/spawn", version = "= 0.120.0-pre" }
 bitflags = "1.0"
 p2p = { path = "/home/exec/Projects/github.com/nervosnetwork/tentacle/tentacle", package = "tentacle", default-features = false }
+url = "2.5.4"
 
 [target.'cfg(not(target_family = "wasm"))'.dependencies]
 p2p = { path = "/home/exec/Projects/github.com/nervosnetwork/tentacle/tentacle", package = "tentacle", default-features = false, features = [

network/Cargo.toml

@@ -27,6 +27,8 @@ pub enum Error {
     Dial(String),
     /// Peer store error
     PeerStore(PeerStoreError),
+    /// Config error
+    Config(String),
 }
 
 /// Error from tentacle

network/src/errors.rs

@@ -17,6 +17,7 @@ pub mod peer_registry;
 pub mod peer_store;
 mod protocols;
 mod services;
+mod proxy;
 
 #[cfg(test)]
 mod tests;

network/src/lib.rs

@@ -19,7 +19,7 @@ use crate::services::{
     dump_peer_store::DumpPeerStoreService, outbound_peer::OutboundPeerService,
     protocol_type_checker::ProtocolTypeCheckerService,
 };
-use crate::{Behaviour, CKBProtocol, Peer, PeerIndex, ProtocolId, ServiceControl};
+use crate::{proxy, Behaviour, CKBProtocol, Peer, PeerIndex, ProtocolId, ServiceControl};
 use ckb_app_config::{default_support_all_protocols, NetworkConfig, SupportProtocol};
 use ckb_logger::{debug, error, info, trace, warn};
 use ckb_spawn::Spawn;
@@ -122,6 +122,12 @@ impl NetworkState {
         let peer_store = Mutex::new(PeerStore::load_from_dir_or_default(
             config.peer_store_path(),
         ));
+        info!("Loaded the peer store.");
+        if config.proxy_config.enable {
+            proxy::check_proxy_url(&config.proxy_config.proxy_url)
+                .map_err(|reason| Error::Config(reason))?;
+        }
+
         let bootnodes = config.bootnodes();
 
         let peer_registry = PeerRegistry::new(
@@ -1018,11 +1024,11 @@ impl NetworkService {
                                 service_builder = service_builder.tcp_config(bind_fn);
 
                                 if config.proxy_config.enable {
-                                    let proxy_config = Some(ProxyConfig {
+                                    let proxy_config = ProxyConfig {
                                         proxy_url: config.proxy_config.proxy_url.clone(),
-                                    });
+                                    };
                                     service_builder =
-                                        service_builder.tcp_proxy_config(proxy_config);
+                                        service_builder.tcp_proxy_config(Some(proxy_config));
                                 }
                             }
                         }

network/src/network.rs

@@ -0,0 +1,28 @@
+pub(crate) fn check_proxy_url(proxy_url: &str) -> Result<(), String> {
+    let parsed_url = url::Url::parse(proxy_url).map_err(|e| e.to_string())?;
+    if parsed_url.host_str().is_none() {
+        return Err( format!("missing host in proxy url: {}", proxy_url) );
+    }
+    let scheme = parsed_url.scheme();
+    if scheme.ne("socks5") {
+        return Err(format!("CKB doesn't support proxy scheme: {}", scheme));
+    }
+    Ok(())
+}
+
+#[test]
+fn parse_socks5_url() {
+    let result = url::Url::parse("socks5://username:password@localhost:1080");
+    assert!(result.is_ok());
+    let parsed_url = result.unwrap();
+    dbg!(&parsed_url);
+    assert_eq!(parsed_url.scheme(), "socks5");
+    // username
+    assert_eq!(parsed_url.username(), "username");
+    // password
+    assert_eq!(parsed_url.password(), Some("password"));
+    // host
+    assert_eq!(parsed_url.host_str(), Some("localhost"));
+    // port
+    assert_eq!(parsed_url.port(), Some(1080));
+}

network/src/proxy.rs

@@ -101,6 +101,7 @@ pub struct Config {
 #[derive(Clone, Debug, Serialize, Deserialize, Default)]
 pub struct ProxyConfig {
     pub enable: bool,
+    // like: socks5://username:[email protected]:1080
     pub proxy_url: String,
 }