hex/wstunnel
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Add typealias socks5 udp

Browse source
This commit is contained in:
Σrebe - Romain GERARD 2024-01-09 21:04:46 +01:00
parent f183e648a7
commit db6f3ea3ce
No known key found for this signature in database
GPG key ID: 7A42B4B97E0332F4
1 changed files with 6 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
src/socks5_udp.rs
View file

@ -24,13 +24,15 @@ use tokio::time::Interval;
use tracing::{debug, error, info};
use url::Host;
type PeerMapKey = (SocketAddr, TargetAddr);
struct IoInner {
sender: mpsc::Sender<Bytes>,
}
struct Socks5UdpServer {
listener: Arc<UdpSocket>,
peers: HashMap<(SocketAddr, TargetAddr), Pin<Arc<IoInner>>, ahash::RandomState>,
keys_to_delete: Arc<RwLock<Vec<(SocketAddr, TargetAddr)>>>,
peers: HashMap<PeerMapKey, Pin<Arc<IoInner>>, ahash::RandomState>,
keys_to_delete: Arc<RwLock<Vec<PeerMapKey>>>,
cnx_timeout: Option<Duration>,
}
@ -82,7 +84,7 @@ pub struct Socks5UdpStream {
pub watchdog_deadline: Option<Interval>,
data_read_before_deadline: bool,
io: Pin<Arc<IoInner>>,
keys_to_delete: Weak<RwLock<Vec<(SocketAddr, TargetAddr)>>>,
keys_to_delete: Weak<RwLock<Vec<PeerMapKey>>>,
}
#[pinned_drop]
@ -100,7 +102,7 @@ impl Socks5UdpStream {
peer: SocketAddr,
destination: TargetAddr,
watchdog_deadline: Option<Duration>,
keys_to_delete: Weak<RwLock<Vec<(SocketAddr, TargetAddr)>>>,
keys_to_delete: Weak<RwLock<Vec<PeerMapKey>>>,
) -> (Self, Pin<Arc<IoInner>>) {
let (tx, rx) = mpsc::channel(1024);
let io = Arc::pin(IoInner { sender: tx });