feat: implement TCP transport using wasi:sockets

Signed-off-by: Roman Volosatovs <rvolosatovs@riseup.net>

Cargo.toml

@@ -127,6 +127,7 @@ tracing = { version = "0.1", default-features = false }
 tracing-subscriber = { version = "0.3", default-features = false }
 url = { version = "2" }
 uuid = { version = "1", default-features = false }
+wasi = { version = "0.13", default-features = false }
 wasi-preview1-component-adapter-provider = { version = "25", default-features = false }
 wasm-tokio = { version = "0.6", default-features = false }
 wasmparser = { version = "0.218", default-features = false }

crates/transport/Cargo.toml

@@ -27,6 +27,9 @@ tracing = { workspace = true, features = ["attributes"] }
 send-future = { workspace = true }
 wasm-tokio = { workspace = true, features = ["tracing"] }
 
+[target.'cfg(target_family = "wasm")'.dependencies]
+wasi = { workspace = true, features = ["std"] }
+
 [dev-dependencies]
 test-log = { workspace = true, features = ["color", "log", "trace"] }
 tokio = { workspace = true, features = ["macros", "rt-multi-thread"] }

crates/transport/src/frame/conn/mod.rs

@@ -477,5 +477,6 @@ async fn egress(
         trace!(?frame, "writing egress frame");
         tx.write_all_buf(&mut frame).await?;
     }
-    Ok(())
+    trace!("shutting down outgoing stream");
+    tx.shutdown().await
 }

crates/transport/src/frame/mod.rs

@@ -7,8 +7,9 @@ use bytes::Bytes;
 mod codec;
 mod conn;
 
-#[cfg(feature = "net")]
+#[cfg(any(target_family = "wasm", feature = "net"))]
 pub mod tcp;
+
 #[cfg(all(unix, feature = "net"))]
 pub mod unix;
 

crates/transport/src/frame/tcp/mod.rs

@@ -0,0 +1,11 @@
+//! wRPC TCP transport
+
+#[cfg(feature = "net")]
+pub mod tokio;
+#[cfg(feature = "net")]
+pub use tokio::*;
+
+#[cfg(target_family = "wasm")]
+pub mod wasi;
+#[cfg(all(target_family = "wasm", not(feature = "net")))]
+pub use wasi::*;

crates/transport/src/frame/tcp.rs → crates/transport/src/frame/tcp/tokio.rs

@@ -1,4 +1,4 @@
-//! TCP transport
+//! wRPC TCP transport using [tokio]
 
 use core::net::SocketAddr;
 
@@ -17,16 +17,16 @@ use crate::Invoke;
 /// repeated calls with return an error
 pub struct Invocation(std::sync::Mutex<Option<TcpStream>>);
 
-/// [Invoke] implementation of a TCP transport
+/// [Invoke] implementation of a TCP transport using [tokio]
 #[derive(Clone, Debug)]
 pub struct Client<T>(T);
 
 impl<T> From<T> for Client<T>
 where
     T: ToSocketAddrs + Clone,
 {
-    fn from(path: T) -> Self {
-        Self(path)
+    fn from(addr: T) -> Self {
+        Self(addr)
     }
 }