suhoan/EasyTier
1
0
Fork 0
mirror of https://mirror.suhoan.cn/https://github.com/EasyTier/EasyTier.git synced 2025-12-13 21:27:25 +08:00
Code Issues Packages Projects Releases Wiki Activity

websocket support bind addr (#129)

Browse Source
This commit is contained in:
Sijie.Sun
2024-06-02 21:48:16 +08:00
committed by GitHub
parent 360691276c
commit c1b725e64e
2 changed files with 113 additions and 36 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
easytier/src/connector/mod.rs
View File

@@ -11,7 +11,7 @@ use crate::{
common::{error::Error, global_ctx::ArcGlobalCtx, network::IPCollector},
tunnel::{
check_scheme_and_get_socket_addr, ring::RingTunnelConnector, tcp::TcpTunnelConnector,
udp::UdpTunnelConnector, TunnelConnector,
udp::UdpTunnelConnector, FromUrl, IpVersion, TunnelConnector,
},
};
@@ -107,7 +107,14 @@ pub async fn create_connector_by_url(
}
#[cfg(feature = "websocket")]
"ws" | "wss" => {
let connector = crate::tunnel::websocket::WSTunnelConnector::new(url);
let dst_addr = SocketAddr::from_url(url.clone(), IpVersion::Both)?;
let mut connector = crate::tunnel::websocket::WSTunnelConnector::new(url);
set_bind_addr_for_peer_connector(
&mut connector,
dst_addr.is_ipv4(),
&global_ctx.get_ip_collector(),
)
.await;
return Ok(Box::new(connector));
}
_ => {