summaryrefslogtreecommitdiffstats
path: root/tokio-tls/examples/echo.rs
blob: 697c14c634b23eb48136bcb07fab0e290fb5017f (plain)
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
// A tiny async TLS echo server with Tokio
extern crate native_tls;
extern crate tokio;
extern crate tokio_tls;

use native_tls::Identity;
use tokio::io;
use tokio::net::TcpListener;
use tokio::prelude::*;

fn main() -> Result<(), Box<std::error::Error>> {
    // Bind the server's socket
    let addr = "127.0.0.1:12345".parse()?;
    let tcp = TcpListener::bind(&addr)?;

    // Create the TLS acceptor.
    let der = include_bytes!("identity.p12");
    let cert = Identity::from_pkcs12(der, "mypass")?;
    let tls_acceptor = tokio_tls::TlsAcceptor::from(
        native_tls::TlsAcceptor::builder(cert).build()?);

    // Iterate incoming connections
    let server = tcp.incoming().for_each(move |tcp| {

        // Accept the TLS connection.
        let tls_accept = tls_acceptor.accept(tcp)
            .and_then(move |tls| {
                // Split up the read and write halves
                let (reader, writer) = tls.split();

                // Copy the data back to the client
                let conn = io::copy(reader, writer)
                    // print what happened
                    .map(|(n, _, _)| {
                        println!("wrote {} bytes", n)
                    })
                    // Handle any errors
                    .map_err(|err| {
                        println!("IO error {:?}", err)
                    });

                // Spawn the future as a concurrent task
                tokio::spawn(conn);

                Ok(())
            })
            .map_err(|err| {
                println!("TLS accept error: {:?}", err);
            });
        tokio::spawn(tls_accept);

        Ok(())
    }).map_err(|err| {
        println!("server error {:?}", err);
    });

    // Start the runtime and spin up the server
    tokio::run(server);
    Ok(())
}