mirror of
https://github.com/tokio-rs/tokio.git
synced 2026-08-29 00:00:11 +02:00
Joel Dice
43134f1e57
Motivation This adds networking support for the `wasm32-wasip2` target platform, which includes more extensive support for sockets than `wasm32-wasip1`. Solution The bulk of the changes are in https://github.com/tokio-rs/mio/pull/1931. This patch mainly tweaks a few `cfg` directives to indicate `wasm32-wasip2`'s additional capabilities. Note that this is a draft PR until until https://github.com/tokio-rs/mio/pull/1931 and https://github.com/rust-lang/socket2/pull/639 have been include in stable releases of their respective projects. Also note that I've added a `wasm32-wasip2` target to CI and triaged each test which was previously disabled for WASI into one of three categories: - Disabled on both WASIp1 and p2 due to not-yet-supported features such as multithreading - Disabled on p1 but enabled on p2 - Disabled on p1 and _temporarily_ disabled on p2 due to `wasi-libc` bugfixes which have been merged but not yet included in a Rust release. I'll open an issue to re-enable them when the fixes land in Rust. Future Work In the future, we could consider adding support for `tokio::net::lookup_host`. WASIp2 natively supports asynchronous DNS lookups and is single threaded, whereas Tokio currently assumes DNS lookups are blocking and require multithreading to emulate async lookups. A WASIp2-specific implementation could do the lookup directly without multithreading. WASIp2 also supports single-threaded, asynchronous file I/O, timers, etc. We could either support those directly or wait for WASIp3's multithreading support, in which case most of `tokio::fs` (as well as `tokio::net::lookup_host`, etc.) _should_ work unchanged via `wasi-libc` and worker threads. Currently, building for WASIp2 requires RUSTFLAGS="--cfg tokio_unstable"`. Once we have a solid maintenance plan, we can remove that requirement.
55 lines
1.5 KiB
Rust
55 lines
1.5 KiB
Rust
#![warn(rust_2018_idioms)]
|
|
// WASIp1 doesn't support bind
|
|
// No `socket` on miri.
|
|
#![cfg(all(
|
|
feature = "net",
|
|
feature = "macros",
|
|
feature = "rt",
|
|
feature = "io-util",
|
|
not(all(target_os = "wasi", target_env = "p1")),
|
|
not(miri)
|
|
))]
|
|
|
|
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(())
|
|
}
|