diff options
author | hozan23 <hozan23@karyontech.net> | 2024-07-16 08:16:57 +0200 |
---|---|---|
committer | hozan23 <hozan23@karyontech.net> | 2024-07-16 08:19:34 +0200 |
commit | cae0c15d10235bf0ec0bd6f8b20814dc7b63dfd5 (patch) | |
tree | a0724e160cdc5c556d132b07639c0225226b761a /p2p/src/connector.rs | |
parent | 6795c2a8c8a580575d107f596961e221faad69cf (diff) |
Diffstat (limited to 'p2p/src/connector.rs')
-rw-r--r-- | p2p/src/connector.rs | 8 |
1 files changed, 8 insertions, 0 deletions
diff --git a/p2p/src/connector.rs b/p2p/src/connector.rs index cfa661b..98cdfc7 100644 --- a/p2p/src/connector.rs +++ b/p2p/src/connector.rs @@ -148,6 +148,10 @@ impl Connector { async fn dial(&self, endpoint: &Endpoint, peer_id: &Option<PeerID>) -> Result<Conn<NetMsg>> { if self.enable_tls { + if !endpoint.is_tcp() && !endpoint.is_tls() { + return Err(Error::UnsupportedEndpoint(endpoint.to_string())); + } + let tls_config = tls::ClientTlsConfig { tcp_config: Default::default(), client_config: tls_client_config(&self.key_pair, peer_id.clone())?, @@ -157,6 +161,10 @@ impl Connector { .await .map(|l| Box::new(l) as karyon_net::Conn<NetMsg>) } else { + if !endpoint.is_tcp() { + return Err(Error::UnsupportedEndpoint(endpoint.to_string())); + } + tcp::dial(endpoint, tcp::TcpConfig::default(), NetMsgCodec::new()) .await .map(|l| Box::new(l) as karyon_net::Conn<NetMsg>) |