blob: 320d9946e819ab9faf4197df5c75b4c87fccd985 (
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
|
#![warn(rust_2018_idioms)]
#![cfg(all(feature = "full", not(tokio_wasi)))] // Wasi doesn't support bind
use std::io::Read;
use std::io::Result;
use tokio::io::{AsyncReadExt, AsyncWriteExt};
use tokio::net::TcpListener;
use tokio::net::TcpStream;
#[tokio::test]
async fn tcp_into_std() -> Result<()> {
let mut data = [0u8; 12];
let listener = TcpListener::bind("127.0.0.1:0").await?;
let addr = listener.local_addr().unwrap().to_string();
let handle = tokio::spawn(async {
let stream: TcpStream = TcpStream::connect(addr).await.unwrap();
stream
});
let (tokio_tcp_stream, _) = listener.accept().await?;
let mut std_tcp_stream = tokio_tcp_stream.into_std()?;
std_tcp_stream
.set_nonblocking(false)
.expect("set_nonblocking call failed");
let mut client = handle.await.expect("The task being joined has panicked");
client.write_all(b"Hello world!").await?;
std_tcp_stream
.read_exact(&mut data)
.expect("std TcpStream read failed!");
assert_eq!(b"Hello world!", &data);
// test back to tokio stream
std_tcp_stream
.set_nonblocking(true)
.expect("set_nonblocking call failed");
let mut tokio_tcp_stream = TcpStream::from_std(std_tcp_stream)?;
client.write_all(b"Hello tokio!").await?;
let _size = tokio_tcp_stream.read_exact(&mut data).await?;
assert_eq!(b"Hello tokio!", &data);
Ok(())
}
|
