allow tunnel listener alloc port after listen

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
+19
View File
@@ -45,6 +45,10 @@ impl TunnelListener for TcpTunnelListener {
// socket.set_reuseport(true)?;
socket.bind(addr)?;
self.addr
.set_port(Some(socket.local_addr()?.port()))
.unwrap();
self.listener = Some(socket.listen(1024)?);
Ok(())
}
@@ -232,4 +236,19 @@ mod tests {
connector.set_ip_version(IpVersion::V4);
_tunnel_pingpong(listener, connector).await;
}
#[tokio::test]
async fn test_alloc_port() {
// v4
let mut listener = TcpTunnelListener::new("tcp://0.0.0.0:0".parse().unwrap());
listener.listen().await.unwrap();
let port = listener.local_url().port().unwrap();
assert!(port > 0);
// v6
let mut listener = TcpTunnelListener::new("tcp://[::]:0".parse().unwrap());
listener.listen().await.unwrap();
let port = listener.local_url().port().unwrap();
assert!(port > 0);
}
}