cyper-core 0.1.0-beta.1

Core adaptors and services for compio running on hyper.
Documentation 
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

use std::io;

use compio_tls::TlsConnector;

/// Represents TLS backend options
#[derive(Debug, Clone, Copy)]
pub enum TlsBackend {
    /// Use [`native_tls`] as TLS backend.
    #[cfg(feature = "native-tls")]
    NativeTls,
    /// Use [`rustls`] as TLS backend.
    #[cfg(feature = "rustls")]
    Rustls,
}

impl Default for TlsBackend {
    fn default() -> Self {
        cfg_if::cfg_if! {
            if #[cfg(feature = "native-tls")] {
                Self::NativeTls
            } else if #[cfg(feature = "rustls")] {
                Self::Rustls
            } else {
                compile_error!("You must choose at least one of these features: [\"native-tls\", \"rustls\"]")
            }
        }
    }
}

impl TlsBackend {
    pub(crate) fn create_connector(&self) -> io::Result<TlsConnector> {
        match self {
            #[cfg(feature = "native-tls")]
            Self::NativeTls => Ok(TlsConnector::from(
                native_tls::TlsConnector::new()
                    .map_err(|e| io::Error::new(io::ErrorKind::Other, e))?,
            )),
            #[cfg(feature = "rustls")]
            Self::Rustls => {
                let mut store = rustls::RootCertStore::empty();
                for cert in rustls_native_certs::load_native_certs().unwrap() {
                    store.add(cert).unwrap();
                }

                Ok(TlsConnector::from(std::sync::Arc::new(
                    rustls::ClientConfig::builder()
                        .with_safe_defaults()
                        .with_root_certificates(store)
                        .with_no_client_auth(),
                )))
            }
        }
    }
}