From 1493feb1844fd93bbf6419d095fbd3905625b630 Mon Sep 17 00:00:00 2001 From: Aram Peres Date: Tue, 26 Oct 2021 01:20:02 -0400 Subject: [PATCH] Reduce udp client socket meta buffer --- src/main.rs | 11 ++++------- src/virtual_iface/udp.rs | 4 ++-- 2 files changed, 6 insertions(+), 9 deletions(-) diff --git a/src/main.rs b/src/main.rs index d20138d..146a65b 100644 --- a/src/main.rs +++ b/src/main.rs @@ -57,13 +57,10 @@ async fn main() -> anyhow::Result<()> { .into_iter() .map(|pf| (pf, wg.clone(), tcp_port_pool.clone(), udp_port_pool.clone())) .for_each(move |(pf, wg, tcp_port_pool, udp_port_pool)| { - std::thread::spawn(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) - .await - .unwrap_or_else(|e| error!("Port-forward failed for {} : {}", pf, e)) - }); + tokio::spawn(async move { + tunnel::port_forward(pf, source_peer_ip, tcp_port_pool, udp_port_pool, wg) + .await + .unwrap_or_else(|e| error!("Port-forward failed for {} : {}", pf, e)) }); }); } diff --git a/src/virtual_iface/udp.rs b/src/virtual_iface/udp.rs index 5b275de..212cb8d 100644 --- a/src/virtual_iface/udp.rs +++ b/src/virtual_iface/udp.rs @@ -161,8 +161,8 @@ impl VirtualInterfacePoll for UdpVirtualInterface { }); let client_socket_handle = client_sockets.entry(client_port).or_insert_with(|| { - let rx_meta = vec![UdpPacketMetadata::EMPTY; MAX_PACKET]; - let tx_meta = vec![UdpPacketMetadata::EMPTY; MAX_PACKET]; + let rx_meta = vec![UdpPacketMetadata::EMPTY; 10]; + let tx_meta = vec![UdpPacketMetadata::EMPTY; 10]; let rx_data = vec![0u8; MAX_PACKET]; let tx_data = vec![0u8; MAX_PACKET]; let udp_rx_buffer = UdpSocketBuffer::new(rx_meta, rx_data);