arampoire/onetun
1
0
Fork 0
mirror of https://github.com/arampoire/onetun.git synced 2025-12-01 03:00:25 -05:00
Code Issues Projects Releases Packages Wiki Activity Actions

Implement event-based UDP interface

Browse source
This commit is contained in:
Aram 🍐 2022-01-08 02:25:56 -05:00
parent 51788c9557
commit abd9df6be4
7 changed files with 218 additions and 50 deletions
Download patch file Download diff file Expand all files Collapse all files

8
src/tunnel/udp.rs
View file

@ -48,7 +48,7 @@ pub async fn udp_proxy_server(
to_send_result = next_udp_datagram(&socket, &mut buffer, port_pool.clone()) => {
match to_send_result {
Ok(Some((port, data))) => {
endpoint.send(Event::LocalData(port, data));
endpoint.send(Event::LocalData(port_forward, port, data));
}
Ok(None) => {
continue;
@ -64,12 +64,12 @@ pub async fn udp_proxy_server(
}
event = endpoint.recv() => {
if let Event::RemoteData(port, data) = event {
if let Some(peer_addr) = port_pool.get_peer_addr(port).await {
if let Err(e) = socket.send_to(&data, peer_addr).await {
if let Some(peer) = port_pool.get_peer_addr(port).await {
if let Err(e) = socket.send_to(&data, peer).await {
error!(
"[{}] Failed to send UDP datagram to real client ({}): {:?}",
port,
peer_addr,
peer,
e,
);
}