tcpserver 2.2.1

rust tcp server frame.
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

use anyhow::{anyhow, Result};
use openssl::ssl::{SslConnector, SslFiletype, SslMethod};
use std::pin::Pin;
use tcpclient::SocketClientTrait;
use tokio::io::AsyncReadExt;
use tokio::sync::oneshot::channel;
use tokio_openssl::SslStream;

#[tokio::main]
async fn main() -> Result<()> {
    // TLS TEST CLIENT
    let (tx, rx) = channel();
    let client = tcpclient::TcpClient::connect_stream_type(
        "127.0.0.1:5555",
        |tcp_stream| async move {
            let mut connector = SslConnector::builder(SslMethod::tls())?;
            connector.set_ca_file("tests/chain.cert.pem")?;
            connector.set_private_key_file("tests/client-key.pem", SslFiletype::PEM)?;
            connector.set_certificate_chain_file("tests/client-cert.pem")?;
            connector.check_private_key()?;
            let ssl = connector.build().configure()?.into_ssl("localhost")?;
            let mut stream = SslStream::new(ssl, tcp_stream)?;
            Pin::new(&mut stream).connect().await?;
            Ok(stream)
        },
        |tx, _client, mut stream| async move {
            let mut buf = [0; 5];
            stream.read_exact(&mut buf).await?;
            assert_eq!(&buf, b"200\r\n");
            tx.send(()).map_err(|_| anyhow!("rx is close"))?;
            Ok(true)
        },
        tx,
    )
    .await?;

    client.send_all_ref(b"hello world\r\n").await?;
    rx.await?;
    println!("finish");
    Ok(())
}