via 2.0.0-gm.12

An async multi-threaded web framework for people who appreciate simplicity.
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
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133

use http::Version;
use std::error::Error;
use std::io;
use tokio::io::{AsyncRead, AsyncWrite};
use tokio::net::TcpStream;

#[cfg(feature = "native-tls")]
pub use native::NativeTlsAcceptor;

#[cfg(feature = "rustls")]
pub use rustls::RustlsAcceptor;

#[derive(Eq, PartialEq)]
pub struct Alpn(Version);

pub struct TcpAcceptor;

pub trait Acceptor {
    type Io: AsyncRead + AsyncWrite;
    type Error: Error;

    #[cfg_attr(not(any(feature = "native-tls", feature = "rustls")), allow(dead_code))]
    fn accept(
        &self,
        io: TcpStream,
    ) -> impl Future<Output = Result<(Self::Io, Alpn), Self::Error>> + Send + 'static;
}

impl Acceptor for TcpAcceptor {
    type Io = TcpStream;
    type Error = io::Error;

    #[allow(clippy::manual_async_fn)]
    fn accept(
        &self,
        _: TcpStream,
    ) -> impl Future<Output = Result<(Self::Io, Alpn), Self::Error>> + Send + 'static {
        async { unreachable!() }
    }
}

#[allow(dead_code)]
impl Alpn {
    pub const H2: Self = Self(Version::HTTP_2);
    pub const HTTP_11: Self = Self(Version::HTTP_11);
}

#[cfg(feature = "native-tls")]
mod native {
    use native_tls::{Identity, Protocol};
    use std::sync::Arc;
    use tokio::net::TcpStream;
    use tokio_native_tls::{TlsAcceptor, TlsStream};

    use super::{Acceptor, Alpn};

    pub struct NativeTlsAcceptor(Arc<TlsAcceptor>);

    impl NativeTlsAcceptor {
        pub fn new(identity: Identity) -> Self {
            Self(Arc::new(TlsAcceptor::from(
                native_tls::TlsAcceptor::builder(identity)
                    .min_protocol_version(Some(Protocol::Tlsv12))
                    .build()
                    .expect("tls config is invalid or missing"),
            )))
        }
    }

    impl Acceptor for NativeTlsAcceptor {
        type Io = TlsStream<TcpStream>;
        type Error = native_tls::Error;

        fn accept(
            &self,
            io: TcpStream,
        ) -> impl Future<Output = Result<(Self::Io, Alpn), Self::Error>> + Send + 'static {
            let acceptor = Arc::clone(&self.0);

            async move {
                let io = acceptor.accept(io).await?;
                let stream = io.get_ref();
                let alpn = stream
                    .negotiated_alpn()?
                    .and_then(|negotiated| (negotiated == b"h2").then_some(Alpn::H2))
                    .unwrap_or(Alpn::HTTP_11);

                Ok((io, alpn))
            }
        }
    }
}

#[cfg(feature = "rustls")]
mod rustls {
    use rustls::ServerConfig;
    use std::{io, sync::Arc};
    use tokio::net::TcpStream;
    use tokio_rustls::server::{TlsAcceptor, TlsStream};

    use super::{Acceptor, Alpn};

    pub struct RustlsAcceptor(TlsAcceptor);

    impl RustlsAcceptor {
        pub fn new(rustls_config: ServerConfig) -> Self {
            Self(TlsAcceptor::from(Arc::new(rustls_config)))
        }
    }

    impl Acceptor for RustlsAcceptor {
        type Io = TlsStream<TcpStream>;
        type Error = io::Error;

        fn accept(
            &self,
            io: TcpStream,
        ) -> impl Future<Output = Result<(Self::Io, Alpn), Self::Error>> + Send + 'static {
            let future = self.0.accept(io);

            async move {
                let io = future.await?;
                let (_, conn) = io.get_ref();
                let alpn = conn
                    .alpn_protocol()
                    .and_then(|negotiated| (negotiated == b"h2").then_some(Alpn::H2))
                    .unwrap_or(Alpn::HTTP_11);

                Ok((io, alpn))
            }
        }
    }
}