hex/wstunnel
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Improve UDP server buffer allocation

Browse source
This commit is contained in:
Σrebe - Romain GERARD 2024-01-20 11:33:04 +01:00
parent c06e4e4945
commit 0943a7e7b5
No known key found for this signature in database
GPG key ID: 7A42B4B97E0332F4
2 changed files with 50 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

29
src/socks5_udp.rs
View file

@ -41,12 +41,32 @@ impl Socks5UdpServer {
let socket = socket2::SockRef::from(&listener); let socket = socket2::SockRef::from(&listener);
// Increase receive buffer // Increase receive buffer
if let Err(err) = socket.set_recv_buffer_size(64 * 1024 * 1024) { const BUF_SIZES: [usize; 7] = [64usize, 32usize, 16usize, 8usize, 4usize, 2usize, 1usize];
warn!("Cannot set UDP server recv buffer: {}", err); for size in BUF_SIZES.iter() {
if let Err(err) = socket.set_recv_buffer_size(size * 1024 * 1024) {
warn!("Cannot increase UDP server recv buffer to {} Mib: {}", size, err);
warn!("This is not fatal, but can lead to packet loss if you have too much throughput. You must monitor packet loss in this case");
continue;
}
if *size != BUF_SIZES[0] {
info!("Increased UDP server recv buffer to {} Mib", size);
}
break;
} }
if let Err(err) = socket.set_send_buffer_size(64 * 1024 * 1024) { for size in BUF_SIZES.iter() {
warn!("Cannot set UDP server recv buffer: {}", err); if let Err(err) = socket.set_send_buffer_size(size * 1024 * 1024) {
warn!("Cannot increase UDP server send buffer to {} Mib: {}", size, err);
warn!("This is not fatal, but can lead to packet loss if you have too much throughput. You must monitor packet loss in this case");
continue;
}
if *size != BUF_SIZES[0] {
info!("Increased UDP server send buffer to {} Mib", size);
}
break;
} }
Self { Self {
@ -56,6 +76,7 @@ impl Socks5UdpServer {
cnx_timeout: timeout, cnx_timeout: timeout,
} }
} }
#[inline] #[inline]
pub fn clean_dead_keys(&mut self) { pub fn clean_dead_keys(&mut self) {
let nb_key_to_delete = self.keys_to_delete.read().len(); let nb_key_to_delete = self.keys_to_delete.read().len();

29
src/udp.rs
View file

@ -40,12 +40,32 @@ impl UdpServer {
let socket = socket2::SockRef::from(&listener); let socket = socket2::SockRef::from(&listener);
// Increase receive buffer // Increase receive buffer
if let Err(err) = socket.set_recv_buffer_size(64 * 1024 * 1024) { const BUF_SIZES: [usize; 7] = [64usize, 32usize, 16usize, 8usize, 4usize, 2usize, 1usize];
warn!("Cannot set UDP server recv buffer: {}", err); for size in BUF_SIZES.iter() {
if let Err(err) = socket.set_recv_buffer_size(size * 1024 * 1024) {
warn!("Cannot increase UDP server recv buffer to {} Mib: {}", size, err);
warn!("This is not fatal, but can lead to packet loss if you have too much throughput. You must monitor packet loss in this case");
continue;
}
if *size != BUF_SIZES[0] {
info!("Increased UDP server recv buffer to {} Mib", size);
}
break;
} }
if let Err(err) = socket.set_send_buffer_size(64 * 1024 * 1024) { for size in BUF_SIZES.iter() {
warn!("Cannot set UDP server recv buffer: {}", err); if let Err(err) = socket.set_send_buffer_size(size * 1024 * 1024) {
warn!("Cannot increase UDP server send buffer to {} Mib: {}", size, err);
warn!("This is not fatal, but can lead to packet loss if you have too much throughput. You must monitor packet loss in this case");
continue;
}
if *size != BUF_SIZES[0] {
info!("Increased UDP server send buffer to {} Mib", size);
}
break;
} }
Self { Self {
@ -55,6 +75,7 @@ impl UdpServer {
cnx_timeout: timeout, cnx_timeout: timeout,
} }
} }
#[inline] #[inline]
pub fn clean_dead_keys(&mut self) { pub fn clean_dead_keys(&mut self) {
let nb_key_to_delete = self.keys_to_delete.read().len(); let nb_key_to_delete = self.keys_to_delete.read().len();