Merge pull request #4 from rylev/udp-override

UDP override
rylev · Jan 29, 2024 · 109efa1 · 109efa1
2 parents d796ad4 + cc1e62f
commit 109efa1
Show file tree

Hide file tree

Showing 7 changed files with 524 additions and 207 deletions.
diff --git a/crates/wasi/src/preview2/ctx.rs b/crates/wasi/src/preview2/ctx.rs
@@ -275,6 +275,7 @@ impl WasiCtxBuilder {
         let mut network = DefaultNetwork::new();
         network.allow_ip_name_lookup(allow_ip_name_lookup);
         network.allow_tcp(tcp_addr_check);
+        network.allow_udp(udp_addr_check);
 
         WasiCtx {
             stdin,
@@ -288,7 +289,6 @@ impl WasiCtxBuilder {
             insecure_random_seed,
             wall_clock,
             monotonic_clock,
-            udp_addr_check,
             network: Box::new(network),
         }
     }
@@ -313,6 +313,5 @@ pub struct WasiCtx {
     pub(crate) stdin: Box<dyn StdinStream>,
     pub(crate) stdout: Box<dyn StdoutStream>,
     pub(crate) stderr: Box<dyn StdoutStream>,
-    pub(crate) udp_addr_check: SocketAddrCheck,
     pub(crate) network: Box<dyn Network>,
 }
diff --git a/crates/wasi/src/preview2/host/mod.rs b/crates/wasi/src/preview2/host/mod.rs
@@ -8,5 +8,4 @@ pub(crate) mod ip_name_lookup;
 pub(crate) mod network;
 mod random;
 pub(crate) mod tcp;
-mod udp;
-mod udp_create_socket;
+pub(crate) mod udp;