tobira 0.2.5

A VMess relay written in Rust.
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44

use std::future::Future;
use std::net::SocketAddr;
use std::pin::Pin;
use std::sync::Arc;

use anyhow::Result;
use tokio::io::{AsyncRead, AsyncWrite};

use crate::relay::runtime::RelayRuntime;
use crate::vmess::validator::Transport;
use crate::vmess::validator::Upstream;

pub mod grpc;
pub mod tcp;

pub trait InboundStream: AsyncRead + AsyncWrite + Unpin + Send + 'static {}

impl<T> InboundStream for T where T: AsyncRead + AsyncWrite + Unpin + Send + 'static {}

pub type OutboundFuture = Pin<Box<dyn Future<Output = Result<()>> + Send + 'static>>;

pub struct OutboundContext {
    pub upstream: Arc<Upstream>,
    /// The 16-byte VMess Auth ID already consumed from the inbound stream.
    /// Passed by value to avoid a heap copy on every connection.
    pub auth_id: [u8; 16],
    pub peer: SocketAddr,
    pub runtime: RelayRuntime,
}

pub trait Outbound: Send + 'static {
    fn relay(
        self: Box<Self>,
        inbound: Box<dyn InboundStream>,
        ctx: OutboundContext,
    ) -> OutboundFuture;
}

pub fn from_transport(transport: &Transport) -> Box<dyn Outbound> {
    match transport {
        Transport::Tcp => Box::new(tcp::TcpOutbound),
        Transport::Grpc { .. } => Box::new(grpc::GrpcOutbound),
    }
}