github/dndx-phantun
1
0
Fork 0
mirror of https://github.com/dndx/phantun.git synced 2025-01-31 12:19:30 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

feat(server) add dead connection detection

Browse Source
This commit is contained in:
Datong Sun 2021-09-16 14:09:43 -07:00 committed by Datong Sun
parent 7c06c5b08b
commit e869255785
2 changed files with 22 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
src/bin/server.rs
View File

@ -1,9 +1,12 @@
use clap::{App, Arg};
use log::info;
use phantom::fake_tcp::packet::MAX_PACKET_LEN;
use phantom::fake_tcp::Stack;
use std::net::SocketAddrV4;
use tokio::net::UdpSocket;
use tokio::time::{self, Duration};
use tokio_tun::TunBuilder;
const UDP_TTL: Duration = Duration::from_secs(180);
#[tokio::main]
async fn main() {
@ -56,6 +59,7 @@ async fn main() {
//thread::sleep(time::Duration::from_secs(5));
let mut stack = Stack::new(tun);
stack.listen(local_port);
info!("Listening on {}", local_port);
let main_loop = tokio::spawn(async move {
let mut buf_udp = [0u8; MAX_PACKET_LEN];
@ -63,11 +67,15 @@ async fn main() {
loop {
let sock = stack.accept().await;
info!("New connection: {}", sock);
tokio::spawn(async move {
let udp_sock = UdpSocket::bind("0.0.0.0:0").await.unwrap();
udp_sock.connect(remote_addr).await.unwrap();
loop {
let read_timeout = time::sleep(UDP_TTL);
tokio::select! {
Ok(size) = udp_sock.recv(&mut buf_udp) => {
if let None = sock.send(&buf_udp[..size]).await {
@ -84,6 +92,10 @@ async fn main() {
None => { return; },
}
},
_ = read_timeout => {
info!("No traffic seen in the last {:?}, closing connection", UDP_TTL);
return;
}
};
}
});

10
src/fake_tcp/mod.rs
View File

@ -404,6 +404,16 @@ impl Stack {
);
shared.outgoing.try_send(buf).unwrap();
}
} else {
let buf = build_tcp_packet(
local_addr,
remote_addr,
0,
tcp_packet.get_acknowledgement() + 1,
tcp::TcpFlags::RST,
None,
);
shared.outgoing.try_send(buf).unwrap();
}
}
}