ntex-tls 3.4.1

An implementation of SSL streams for ntex backed by OpenSSL
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

//! An implementation of SSL streams for ntex backed by OpenSSL
use std::{any, cell::RefCell, io, sync::Arc};

use ntex_io::{Filter, FilterLayer, Io, Layer, ReadBuf, WriteBuf};
use tls_rustls::{ClientConfig, ClientConnection, pki_types::ServerName};

use super::Stream;

#[derive(Debug)]
/// An implementation of SSL streams
pub struct TlsClientFilter {
    session: RefCell<ClientConnection>,
}

impl FilterLayer for TlsClientFilter {
    fn query(&self, id: any::TypeId) -> Option<Box<dyn any::Any>> {
        Stream::new(&mut *self.session.borrow_mut()).query(id)
    }

    fn process_read_buf(&self, buf: &ReadBuf<'_>) -> io::Result<usize> {
        Stream::new(&mut *self.session.borrow_mut()).process_read_buf(buf)
    }

    fn process_write_buf(&self, buf: &WriteBuf<'_>) -> io::Result<()> {
        Stream::new(&mut *self.session.borrow_mut()).process_write_buf(buf)
    }
}

impl TlsClientFilter {
    pub async fn create<F: Filter>(
        io: Io<F>,
        cfg: Arc<ClientConfig>,
        domain: ServerName<'static>,
    ) -> Result<Io<Layer<TlsClientFilter, F>>, io::Error> {
        let session = ClientConnection::new(cfg, domain).map_err(io::Error::other)?;
        let io = io.add_filter(TlsClientFilter {
            session: RefCell::new(session),
        });
        super::stream::handshake(&io.filter().session, &io).await?;
        Ok(io)
    }
}