compio-btls 0.5.4

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

use btls::ssl::{Ssl, SslAcceptor, SslFiletype};
use btls::ssl::{SslConnector, SslMethod};
use compio_btls::SslStream;
use compio_io::AsyncWrite;
use compio_io::{AsyncReadExt, AsyncWriteExt};
use compio_net::TcpListener;
use compio_net::TcpStream;
use futures::future;
use std::net::ToSocketAddrs;
use std::pin::Pin;

#[compio::test]
async fn google() {
    let addr = "google.com:443".to_socket_addrs().unwrap().next().unwrap();
    let stream = TcpStream::connect(&addr).await.unwrap();

    let config = SslConnector::builder(SslMethod::tls())
        .unwrap()
        .build()
        .configure()
        .unwrap();

    let ssl = config.into_ssl("google.com").unwrap();
    let mut stream = SslStream::new(ssl, stream).unwrap();
    Pin::new(&mut stream).connect().await.unwrap();

    stream.write(b"GET / HTTP/1.0\r\n\r\n").await.unwrap();
    stream.flush().await.unwrap();
    let (_, buf) = stream.read_to_end(vec![]).await.unwrap();
    stream.shutdown().await.unwrap();
    let response = String::from_utf8_lossy(&buf);

    // any response code is fine
    assert!(response.starts_with("HTTP/1.0 "));
    assert!(response.ends_with("</html>") || response.ends_with("</HTML>"));
}

#[compio::test]
async fn server() {
    let listener = TcpListener::bind("127.0.0.1:0").await.unwrap();
    let addr = listener.local_addr().unwrap();

    let server = async move {
        let mut acceptor = SslAcceptor::mozilla_intermediate(SslMethod::tls()).unwrap();
        acceptor
            .set_private_key_file("tests/key.pem", SslFiletype::PEM)
            .unwrap();
        acceptor
            .set_certificate_chain_file("tests/cert.pem")
            .unwrap();
        let acceptor = acceptor.build();

        let ssl = Ssl::new(acceptor.context()).unwrap();
        let stream = listener.accept().await.unwrap().0;
        let mut stream = SslStream::new(ssl, stream).unwrap();

        Pin::new(&mut stream).accept().await.unwrap();

        let (_, buf) = stream.read_exact(vec![0u8; 4]).await.unwrap();
        assert_eq!(&buf, b"asdf");

        stream.write(b"jkl;").await.unwrap();
        stream.flush().await.unwrap();

        Pin::new(&mut stream).shutdown().await.unwrap()
    };

    let client = async {
        let mut connector = SslConnector::builder(SslMethod::tls()).unwrap();
        connector.set_ca_file("tests/cert.pem").unwrap();
        let ssl = connector
            .build()
            .configure()
            .unwrap()
            .into_ssl("localhost")
            .unwrap();

        let stream = TcpStream::connect(&addr).await.unwrap();
        let mut stream = SslStream::new(ssl, stream).unwrap();

        Pin::new(&mut stream).connect().await.unwrap();

        stream.write_all(b"asdf").await.unwrap();
        stream.flush().await.unwrap();

        let (_, buf) = stream.read_to_end(vec![]).await.unwrap();
        assert_eq!(buf, b"jkl;");
    };

    future::join(server, client).await;
}