fix(socks5 udp): correctly reserve enough space for new packets

erebe · Jan 23, 2024 · b71787c · b71787c
1 parent 362fa92
commit b71787c
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/src/socks5_udp.rs b/src/socks5_udp.rs
@@ -229,9 +229,7 @@ pub async fn run_server(
     let stream = stream::unfold((udp_server, buffer), |(mut server, mut buf)| async move {
         loop {
             server.clean_dead_keys();
-            if buf.remaining_mut() < MAX_PACKET_LENGTH {
-                buf.reserve(MAX_PACKET_LENGTH);
-            }
+            buf.reserve(MAX_PACKET_LENGTH);
 
             let peer_addr = match server.listener.recv_buf_from(&mut buf).await {
                 Ok((_read_len, peer_addr)) => peer_addr,