luckyimnotanigger/tun2proxy
1
0
Fork 0
mirror of https://github.com/tun2proxy/tun2proxy.git synced 2025-04-21 22:39:08 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

consume_cached_dns_over_tcp_packets

Browse source
This commit is contained in:
ssrlive 2023-08-22 18:55:44 +08:00
parent 2211ec6d7a
commit 9880741dc1
1 changed files with 12 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
src/tun2proxy.rs
View file

@ -526,6 +526,15 @@ impl<'a> TunToProxy<'a> {
Ok(()) Ok(())
} }
fn consume_cached_dns_over_tcp_packets(&mut self, info: &ConnectionInfo) -> Result<()> {
if let Some(state) = self.connection_map.get_mut(info) {
while let Some(buf) = state.udp_over_tcp_data_cache.pop_front() {
_ = state.mio_stream.write(&buf)?;
}
}
Ok(())
}
fn process_incoming_udp_packets( fn process_incoming_udp_packets(
&mut self, &mut self,
manager: &Rc<dyn ConnectionManager>, manager: &Rc<dyn ConnectionManager>,
@ -859,7 +868,7 @@ impl<'a> TunToProxy<'a> {
Ok(()) Ok(())
} }
fn comsume_cached_udp_packets(&mut self, info: &ConnectionInfo) -> Result<()> { fn consume_cached_udp_packets(&mut self, info: &ConnectionInfo) -> Result<()> {
// Try to send the first UDP packets to remote SOCKS5 server for UDP associate session // Try to send the first UDP packets to remote SOCKS5 server for UDP associate session
if let Some(state) = self.connection_map.get_mut(info) { if let Some(state) = self.connection_map.get_mut(info) {
if let Some(udp_socket) = state.udp_socket.as_ref() { if let Some(udp_socket) = state.udp_socket.as_ref() {
@ -958,7 +967,8 @@ impl<'a> TunToProxy<'a> {
// server. // server.
self.write_to_server(&conn_info)?; self.write_to_server(&conn_info)?;
self.comsume_cached_udp_packets(&conn_info)?; self.consume_cached_dns_over_tcp_packets(&conn_info)?;
self.consume_cached_udp_packets(&conn_info)?;
} }
if event.is_writable() { if event.is_writable() {