Skip to content

Commit

Permalink
remove some udp logs
Browse files Browse the repository at this point in the history
  • Loading branch information
ssrlive committed Jul 19, 2023
1 parent a19bec1 commit 4fb8dc5
Showing 1 changed file with 7 additions and 7 deletions.
14 changes: 7 additions & 7 deletions src/udprelay.rs
Original file line number Diff line number Diff line change
Expand Up @@ -96,7 +96,7 @@ async fn socks5_to_relay(
udp_tx: UdpRequestSender,
) -> Result<()> {
loop {
log::trace!("[UDP] waiting for incoming packet");
// log::trace!("[UDP] waiting for incoming packet");

let buf_size = MAX_UDP_RELAY_PACKET_SIZE - UdpHeader::max_serialized_len();
listen_udp.set_max_packet_size(buf_size);
Expand All @@ -110,7 +110,7 @@ async fn socks5_to_relay(
incoming.lock().await.clone_from(&src_addr);
incomings.lock().await.insert(src_addr);

log::trace!("[UDP] {src_addr} -> {dst_addr} incoming packet size {}", pkt.len());
// log::trace!("[UDP] {src_addr} -> {dst_addr} incoming packet size {}", pkt.len());
let src_addr = src_addr.into();
let _ = udp_tx.send((pkt, dst_addr, src_addr));
}
Expand All @@ -123,10 +123,10 @@ async fn relay_to_socks5(
incoming_addr: Arc<Mutex<SocketAddr>>,
mut udp_rx: UdpRequestReceiver,
) -> Result<()> {
while let Ok((pkt, addr, from_addr)) = udp_rx.recv().await {
while let Ok((pkt, addr, _from_addr)) = udp_rx.recv().await {
let to_addr = SocketAddr::try_from(addr.clone())?;
if *incoming_addr.lock().await == to_addr {
log::trace!("[UDP] {to_addr} <- {from_addr} feedback to incoming");
// log::trace!("[UDP] {to_addr} <- {_from_addr} feedback to incoming");
listen_udp.send_to(pkt, 0, addr, to_addr).await?;
}
}
Expand Down Expand Up @@ -168,8 +168,8 @@ async fn _run_udp_loop<S: AsyncRead + AsyncWrite + Unpin>(
loop {
let _res = tokio::select! {
Ok((pkt, dst_addr, src_addr)) = udp_rx.recv() => {
let flag = { incomings.lock().await.contains(&SocketAddr::try_from(dst_addr.clone())?) };
if !flag {
let direction = { incomings.lock().await.contains(&SocketAddr::try_from(dst_addr.clone())?) };
if !direction {
// packet send to remote server, format: dst_addr + src_addr + pkt
let mut buf = BytesMut::new();
dst_addr.write_to_buf(&mut buf);
Expand All @@ -189,7 +189,7 @@ async fn _run_udp_loop<S: AsyncRead + AsyncWrite + Unpin>(
let msg = Message::Binary(buf.freeze().to_vec());
ws_stream.send(msg).await?;
} else {
log::trace!("[UDP] {dst_addr} <- {src_addr} skip feedback packet");
// log::trace!("[UDP] {dst_addr} <- {src_addr} skip feedback packet");
}
Ok::<_, Error>(())
},
Expand Down

0 comments on commit 4fb8dc5

Please sign in to comment.