Reduce udp client socket meta buffer

This commit is contained in:
Aram 🍐 2021-10-26 01:20:02 -04:00
parent 0da6fa51de
commit 1493feb184
2 changed files with 6 additions and 9 deletions

View file

@ -57,15 +57,12 @@ async fn main() -> anyhow::Result<()> {
.into_iter() .into_iter()
.map(|pf| (pf, wg.clone(), tcp_port_pool.clone(), udp_port_pool.clone())) .map(|pf| (pf, wg.clone(), tcp_port_pool.clone(), udp_port_pool.clone()))
.for_each(move |(pf, wg, tcp_port_pool, udp_port_pool)| { .for_each(move |(pf, wg, tcp_port_pool, udp_port_pool)| {
std::thread::spawn(move || { tokio::spawn(async move {
let cpu_pool = tokio::runtime::Runtime::new().unwrap();
cpu_pool.block_on(async move {
tunnel::port_forward(pf, source_peer_ip, tcp_port_pool, udp_port_pool, wg) tunnel::port_forward(pf, source_peer_ip, tcp_port_pool, udp_port_pool, wg)
.await .await
.unwrap_or_else(|e| error!("Port-forward failed for {} : {}", pf, e)) .unwrap_or_else(|e| error!("Port-forward failed for {} : {}", pf, e))
}); });
}); });
});
} }
futures::future::pending().await futures::future::pending().await

View file

@ -161,8 +161,8 @@ impl VirtualInterfacePoll for UdpVirtualInterface {
}); });
let client_socket_handle = client_sockets.entry(client_port).or_insert_with(|| { let client_socket_handle = client_sockets.entry(client_port).or_insert_with(|| {
let rx_meta = vec![UdpPacketMetadata::EMPTY; MAX_PACKET]; let rx_meta = vec![UdpPacketMetadata::EMPTY; 10];
let tx_meta = vec![UdpPacketMetadata::EMPTY; MAX_PACKET]; let tx_meta = vec![UdpPacketMetadata::EMPTY; 10];
let rx_data = vec![0u8; MAX_PACKET]; let rx_data = vec![0u8; MAX_PACKET];
let tx_data = vec![0u8; MAX_PACKET]; let tx_data = vec![0u8; MAX_PACKET];
let udp_rx_buffer = UdpSocketBuffer::new(rx_meta, rx_data); let udp_rx_buffer = UdpSocketBuffer::new(rx_meta, rx_data);