Use LinkedList as UDP packet cache

This commit is contained in:
B. Blechschmidt 2023-08-20 18:51:15 +02:00
parent 5301cf8d37
commit b0e275ec08

View file

@ -17,6 +17,7 @@ use std::{
rc::Rc, rc::Rc,
str::FromStr, str::FromStr,
}; };
use std::collections::LinkedList;
#[derive(Hash, Clone, Eq, PartialEq, PartialOrd, Ord, Debug)] #[derive(Hash, Clone, Eq, PartialEq, PartialOrd, Ord, Debug)]
pub(crate) struct ConnectionInfo { pub(crate) struct ConnectionInfo {
@ -180,7 +181,7 @@ struct TcpConnectState {
udp_socket: Option<UdpSocket>, udp_socket: Option<UdpSocket>,
udp_token: Option<Token>, udp_token: Option<Token>,
udp_origin_dst: Option<SocketAddr>, udp_origin_dst: Option<SocketAddr>,
udp_data_cache: Option<Vec<u8>>, udp_data_cache: LinkedList<Vec<u8>>,
} }
pub(crate) trait TcpProxy { pub(crate) trait TcpProxy {
@ -540,6 +541,7 @@ impl<'a> TunToProxy<'a> {
self.tunsocket_read_and_forward(&connection_info)?; self.tunsocket_read_and_forward(&connection_info)?;
self.write_to_server(&connection_info)?; self.write_to_server(&connection_info)?;
} else { } else {
log::info!("Subsequent udp packet {} ({})", connection_info, origin_dst);
// log::trace!("Subsequent udp packet {} ({})", connection_info, origin_dst); // log::trace!("Subsequent udp packet {} ({})", connection_info, origin_dst);
} }
@ -559,8 +561,9 @@ impl<'a> TunToProxy<'a> {
socket.send_to(&s5_udp_data, udp_associate)?; socket.send_to(&s5_udp_data, udp_associate)?;
} }
} else { } else {
log::info!("Cache udp packet {} ({})", connection_info, origin_dst);
// UDP associate tunnel not ready yet, we must cache the packet... // UDP associate tunnel not ready yet, we must cache the packet...
state.udp_data_cache = Some(s5_udp_data); state.udp_data_cache.push_back(s5_udp_data);
} }
} }
} else { } else {
@ -621,7 +624,7 @@ impl<'a> TunToProxy<'a> {
udp_socket, udp_socket,
udp_token, udp_token,
udp_origin_dst: None, udp_origin_dst: None,
udp_data_cache: None, udp_data_cache: LinkedList::new(),
}; };
Ok(state) Ok(state)
} }
@ -871,7 +874,7 @@ impl<'a> TunToProxy<'a> {
if let Some(udp_socket) = state.udp_socket.as_ref() { if let Some(udp_socket) = state.udp_socket.as_ref() {
if let Some(addr) = state.tcp_proxy_handler.get_udp_associate() { if let Some(addr) = state.tcp_proxy_handler.get_udp_associate() {
// Take ownership of udp_data_cache // Take ownership of udp_data_cache
if let Some(buf) = state.udp_data_cache.take() { while let Some(buf) = state.udp_data_cache.pop_front(){
udp_socket.send_to(&buf, addr)?; udp_socket.send_to(&buf, addr)?;
} }
} }