mirror of
https://github.com/EasyTier/EasyTier.git
synced 2026-05-10 16:05:37 +00:00
feat: add upnp support (#1449)
This commit is contained in:
@@ -88,7 +88,7 @@ impl UdpHolePunchRpc for UdpHolePunchServer {
|
||||
) -> rpc_types::error::Result<SelectPunchListenerResponse> {
|
||||
let (_, addr) = self
|
||||
.common
|
||||
.select_listener(input.force_new)
|
||||
.select_listener(input.force_new, input.prefer_port_mapping)
|
||||
.await
|
||||
.ok_or(anyhow::anyhow!("no listener available"))?;
|
||||
|
||||
@@ -584,6 +584,11 @@ impl UdpHolePunchConnector {
|
||||
|
||||
Ok(())
|
||||
}
|
||||
|
||||
#[cfg(test)]
|
||||
pub async fn run_immediately_for_test(&self) {
|
||||
self.client.run_immediately().await;
|
||||
}
|
||||
}
|
||||
|
||||
#[cfg(test)]
|
||||
@@ -614,6 +619,9 @@ pub mod tests {
|
||||
udp_nat_type: NatType,
|
||||
) -> Arc<PeerManager> {
|
||||
let p_a = create_mock_peer_manager().await;
|
||||
let mut flags = p_a.get_global_ctx().get_flags();
|
||||
flags.disable_upnp = true;
|
||||
p_a.get_global_ctx().set_flags(flags);
|
||||
replace_stun_info_collector(p_a.clone(), udp_nat_type);
|
||||
p_a
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user