tobaru 0.5.0

Port forwarding CLI utility written in Rust with allowlists, TLS, iptables support, round-robin forwarding (load balancing), and IP-to-destination mapping.
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24

use async_trait::async_trait;
use tokio::net::TcpStream;

use crate::async_stream::AsyncStream;

pub trait AsyncTlsFactory: Unpin + Send + Sync {
    fn create_acceptor(&self, cert_bytes: &[u8], key_bytes: &[u8]) -> Box<dyn AsyncTlsAcceptor>;
    // TODO: support different configs/certs.
    fn create_connector(&self, verify: bool) -> Box<dyn AsyncTlsConnector>;
}

#[async_trait]
pub trait AsyncTlsAcceptor: Unpin + Send + Sync {
    async fn accept(&self, stream: TcpStream) -> std::io::Result<Box<dyn AsyncStream>>;
}

#[async_trait]
pub trait AsyncTlsConnector: Unpin + Send + Sync {
    async fn connect(
        &self,
        domain: &str,
        stream: TcpStream,
    ) -> std::io::Result<Box<dyn AsyncStream>>;
}