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

allow tunnel listener alloc port after listen

Browse Source
This commit is contained in:
sijie.sun
2024-04-27 21:20:23 +08:00
committed by Sijie.Sun
parent a3e85a1270
commit 3a965efab2
4 changed files with 79 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
easytier/src/tunnel/wireguard.rs
View File

@@ -537,6 +537,10 @@ impl TunnelListener for WgTunnelListener {
}
self.udp = Some(Arc::new(UdpSocket::from_std(socket2_socket.into())?));
self.addr
.set_port(Some(self.udp.as_ref().unwrap().local_addr()?.port()))
.unwrap();
self.tasks.spawn(Self::handle_udp_incoming(
self.get_udp_socket(),
self.config.clone(),
@@ -868,4 +872,21 @@ pub mod tests {
connector.set_ip_version(IpVersion::V4);
_tunnel_pingpong(listener, connector).await;
}
#[tokio::test]
async fn test_alloc_port() {
// v4
let (server_cfg, _client_cfg) = create_wg_config();
let mut listener = WgTunnelListener::new("wg://0.0.0.0:0".parse().unwrap(), server_cfg);
listener.listen().await.unwrap();
let port = listener.local_url().port().unwrap();
assert!(port > 0);
// v6
let (server_cfg, _client_cfg) = create_wg_config();
let mut listener = WgTunnelListener::new("wg://[::]:0".parse().unwrap(), server_cfg);
listener.listen().await.unwrap();
let port = listener.local_url().port().unwrap();
assert!(port > 0);
}
}