|
| 1 | +extern crate native_tls; |
| 2 | + |
| 3 | +use native_tls::{Identity, TlsAcceptor, TlsStream}; |
| 4 | +use std::fs::File; |
| 5 | +use std::io::{Read, Write}; |
| 6 | +use std::net::{TcpListener, TcpStream}; |
| 7 | +use std::sync::Arc; |
| 8 | +use std::thread; |
| 9 | + |
| 10 | +fn main() { |
| 11 | + let mut cert_file = File::open("test/cert.pem").unwrap(); |
| 12 | + let mut certs = vec![]; |
| 13 | + cert_file.read_to_end(&mut certs).unwrap(); |
| 14 | + let mut key_file = File::open("test/key.pem").unwrap(); |
| 15 | + let mut key = vec![]; |
| 16 | + key_file.read_to_end(&mut key).unwrap(); |
| 17 | + let pkcs8 = Identity::from_pkcs8(&certs, &key).unwrap(); |
| 18 | + |
| 19 | + let acceptor = TlsAcceptor::new(pkcs8).unwrap(); |
| 20 | + let acceptor = Arc::new(acceptor); |
| 21 | + |
| 22 | + let listener = TcpListener::bind("0.0.0.0:8443").unwrap(); |
| 23 | + |
| 24 | + fn handle_client(mut stream: TlsStream<TcpStream>) { |
| 25 | + let mut buf = [0; 1024]; |
| 26 | + let read = stream.read(&mut buf).unwrap(); |
| 27 | + let received = std::str::from_utf8(&buf[0..read]).unwrap(); |
| 28 | + stream.write_all(format!("received '{}'", received).as_bytes()).unwrap(); |
| 29 | + } |
| 30 | + |
| 31 | + for stream in listener.incoming() { |
| 32 | + match stream { |
| 33 | + Ok(stream) => { |
| 34 | + let acceptor = acceptor.clone(); |
| 35 | + thread::spawn(move || { |
| 36 | + let stream = acceptor.accept(stream).unwrap(); |
| 37 | + handle_client(stream); |
| 38 | + }); |
| 39 | + } |
| 40 | + Err(_e) => { /* connection failed */ } |
| 41 | + } |
| 42 | + } |
| 43 | +} |
0 commit comments