mirrors/krata
1
0
Fork 0
mirror of https://github.com/edera-dev/krata.git synced 2025-08-05 14:11:32 +00:00
Code Issues Packages Projects Releases Wiki Activity

network: implement NAT table reclaim

Browse Source
This commit is contained in:
Alex Zenla
2024-02-10 14:02:54 +00:00
parent 8ee93c5752
commit d2136b5d8c
4 changed files with 98 additions and 36 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
network/src/proxynat/mod.rs
View File

@ -1,5 +1,3 @@
mod udp;
use async_trait::async_trait;
use log::{debug, warn};
@ -11,6 +9,8 @@ use crate::proxynat::udp::ProxyUdpHandler;
use crate::nat::{NatHandler, NatHandlerFactory, NatKey, NatKeyProtocol};
mod udp;
pub struct ProxyNatHandlerFactory {}
impl ProxyNatHandlerFactory {
@ -21,7 +21,12 @@ impl ProxyNatHandlerFactory {
#[async_trait]
impl NatHandlerFactory for ProxyNatHandlerFactory {
async fn nat(&self, key: NatKey, sender: Sender<Vec<u8>>) -> Option<Box<dyn NatHandler>> {
async fn nat(
&self,
key: NatKey,
tx_sender: Sender<Vec<u8>>,
reclaim_sender: Sender<NatKey>,
) -> Option<Box<dyn NatHandler>> {
debug!("creating proxy nat entry for key: {}", key);
match key.protocol {
@ -29,7 +34,7 @@ impl NatHandlerFactory for ProxyNatHandlerFactory {
let (rx_sender, rx_receiver) = channel::<Vec<u8>>(4);
let mut handler = ProxyUdpHandler::new(key, rx_sender);
if let Err(error) = handler.spawn(rx_receiver, sender.clone()).await {
if let Err(error) = handler.spawn(rx_receiver, tx_sender, reclaim_sender).await {
warn!("unable to spawn udp proxy handler: {}", error);
None
} else {
@ -45,5 +50,5 @@ impl NatHandlerFactory for ProxyNatHandlerFactory {
pub enum ProxyNatSelect {
External(usize),
Internal(Vec<u8>),
Closed,
Close,
}