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

use std::io;
use std::thread;

use tls_api::runtime::AsyncReadExt;
use tls_api::runtime::AsyncWriteExt;
use tls_api::TlsAcceptorType;
use tls_api::TlsConnectorType;

use crate::block_on;
use crate::new_acceptor_dyn;
use crate::new_connector_builder_dyn_with_root_ca;
use crate::AcceptorKeyKind;
use crate::TcpListener;
use crate::TcpStream;
use crate::BIND_HOST;

async fn test_client_server_dyn_impl(
    connector: &dyn TlsConnectorType,
    acceptor: &dyn TlsAcceptorType,
    key: AcceptorKeyKind,
) {
    drop(env_logger::try_init());

    if !connector.implemented() {
        eprintln!("connector {} is not implemented; skipping", connector);
        return;
    }

    if !acceptor.implemented() {
        eprintln!("acceptor {} is not implemented; skipping", acceptor);
        return;
    }

    let acceptor = new_acceptor_dyn(acceptor, Some(key));

    let acceptor = acceptor.build().expect("acceptor build");
    #[allow(unused_mut)]
    let mut listener = t!(TcpListener::bind((BIND_HOST, 0)).await);
    let port = listener.local_addr().expect("local_addr").port();

    let server_thread_name = format!("{}-server", thread::current().name().unwrap_or("test"));
    let j = thread::Builder::new()
        .name(server_thread_name)
        .spawn(move || {
            let future = async {
                let socket = t!(listener.accept().await).0;
                let mut socket = t!(acceptor.accept(socket).await);

                let mut buf = [0; 5];
                t!(socket.read_exact(&mut buf).await);
                assert_eq!(&buf, b"hello");

                t!(socket.write_all(b"world").await);
            };
            block_on(future);
        })
        .unwrap();

    let socket = t!(TcpStream::connect((BIND_HOST, port)).await);

    let connector = new_connector_builder_dyn_with_root_ca(connector);
    let connector = connector.build().expect("acceptor build");
    let mut socket = t!(connector.connect("localhost", socket).await);

    t!(socket.write_all(b"hello").await);
    let mut buf = vec![];
    match socket.read_to_end(&mut buf).await {
        Ok(_) => {}
        Err(e) if e.kind() == io::ErrorKind::ConnectionReset => {
            // rustls on Windows does that
        }
        Err(e) => panic!("{}", e),
    }
    assert_eq!(buf, b"world");

    j.join().expect("thread join");
}

pub fn test_client_server_dyn_der(
    connector: &dyn TlsConnectorType,
    acceptor: &dyn TlsAcceptorType,
) {
    block_on(test_client_server_dyn_impl(
        connector,
        acceptor,
        AcceptorKeyKind::Der,
    ))
}

pub fn test_client_server_dyn_pkcs12(
    connector: &dyn TlsConnectorType,
    acceptor: &dyn TlsAcceptorType,
) {
    block_on(test_client_server_dyn_impl(
        connector,
        acceptor,
        AcceptorKeyKind::Pkcs12,
    ))
}