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

Basic UDP port pool

Browse source
This commit is contained in:
Aram 🍐 2021-10-20 16:49:24 -04:00
parent 11c5ec99fd
commit cc91cce169
4 changed files with 93 additions and 18 deletions
Download patch file Download diff file Expand all files Collapse all files

10
src/tunnel/tcp.rs
View file

@ -3,7 +3,7 @@ use crate::virtual_iface::tcp::TcpVirtualInterface;
use crate::virtual_iface::{VirtualInterfacePoll, VirtualPort};
use crate::wg::WireGuardTunnel;
use anyhow::Context;
use std::collections::{HashSet, VecDeque};
use std::collections::VecDeque;
use std::sync::atomic::{AtomicBool, Ordering};
use std::sync::Arc;
use tokio::net::{TcpListener, TcpStream};
@ -235,8 +235,7 @@ impl TcpPortPool {
let port = inner
.queue
.pop_front()
.with_context(|| "Virtual port pool is exhausted")?;
inner.taken.insert(port);
.with_context(|| "TCP virtual port pool is exhausted")?;
Ok(port)
}
@ -244,15 +243,12 @@ impl TcpPortPool {
pub async fn release(&self, port: u16) {
let mut inner = self.inner.write().await;
inner.queue.push_back(port);
inner.taken.remove(&port);
}
}
/// Non thread-safe inner logic for TCP port pool.
#[derive(Debug, Clone, Default)]
#[derive(Debug, Default)]
struct TcpPortPoolInner {
/// Remaining ports in the pool.
queue: VecDeque<u16>,
/// Ports taken out of the pool.
taken: HashSet<u16>,
}