ospab
/
ostp
mirror of https://github.com/ospab/ostp.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fix(client): stabilize UDP sessions - prevent crashes on transient recv errors in udp_nat and proxy

This commit is contained in:
ospab 2026-05-30 02:12:15 +03:00
parent 3a39f19b45
commit 60b5d565e8
2 changed files with 47 additions and 35 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
ostp-client/src/tunnel/proxy.rs
View File

@ -371,11 +371,19 @@ async fn handle_udp_associate(
loop { loop {
tokio::select! { tokio::select! {
res = client_tcp.read(&mut tcp_buf) => { res = client_tcp.read(&mut tcp_buf) => {
let n = res?; match res {
if n == 0 { break; } Ok(0) | Err(_) => break,
Ok(_) => {}
}
} }
res = sock_rx.recv_from(&mut buf) => { res = sock_rx.recv_from(&mut buf) => {
let (len, addr) = res?; let (len, addr) = match res {
Ok(v) => v,
Err(e) => {
tracing::debug!("udp_associate recv_from error: {}", e);
continue; // transient error, don't kill the session
}
};
{ {
let mut guard = client_udp_addr.lock().unwrap(); let mut guard = client_udp_addr.lock().unwrap();
if guard.is_none() { if guard.is_none() {

28
ostp-client/src/tunnel/udp_nat.rs
View File

@ -96,13 +96,8 @@ async fn start_udp_session(
} }
} }
let udp = match relay_addr { // Local SOCKS5 proxy always returns 127.0.0.1 (IPv4), so always bind IPv4
SocketAddr::V4(_) => UdpSocket::bind("127.0.0.1:0").await?, let udp = UdpSocket::bind("127.0.0.1:0").await?;
SocketAddr::V6(_) => match UdpSocket::bind("[::1]:0").await {
Ok(sock) => sock,
Err(_) => UdpSocket::bind("[::]:0").await?,
},
};
let mut buf = vec![0u8; 65536]; let mut buf = vec![0u8; 65536];
@ -128,8 +123,13 @@ async fn start_udp_session(
} }
} }
res = udp.recv_from(&mut buf) => { res = udp.recv_from(&mut buf) => {
let (len, _peer) = res?; match res {
if len < 10 { continue; } // At least 10 bytes for SOCKS5 header Err(e) => {
tracing::debug!("udp_nat recv_from error: {}", e);
continue; // transient error, don't kill the session
}
Ok((len, _peer)) => {
if len < 4 { continue; }
let frag = buf[2]; let frag = buf[2];
if frag != 0 { continue; } // fragment not supported if frag != 0 { continue; } // fragment not supported
let atyp = buf[3]; let atyp = buf[3];
@ -148,16 +148,20 @@ async fn start_udp_session(
let port = u16::from_be_bytes([buf[20], buf[21]]); let port = u16::from_be_bytes([buf[20], buf[21]]);
(22, SocketAddr::new(std::net::IpAddr::V6(ip), port)) (22, SocketAddr::new(std::net::IpAddr::V6(ip), port))
} }
_ => continue, // Domain name not supported for incoming packets in typical UDP associate _ => continue,
}; };
let payload = buf[header_len..len].to_vec(); let payload = buf[header_len..len].to_vec();
use futures::SinkExt; use futures::SinkExt;
let _ = smoltcp_tx.lock().await.send((payload, remote_dst, client_src)).await; let _ = smoltcp_tx.lock().await.send((payload, remote_dst, client_src)).await;
} }
}
}
// If TCP drops, UDP association is over // If TCP drops, UDP association is over
res = tcp.read(&mut tcp_buf) => { res = tcp.read(&mut tcp_buf) => {
let n = res?; match res {
if n == 0 { break; } Ok(0) | Err(_) => break,
Ok(_) => {}
}
} }
} }
} }