Add function to downcast a stream error to a network-related error. S…

…imilar to the existing filesystem-error-code and http-error-code functions. (#9276)

crates/cli-flags/src/lib.rs

@@ -340,6 +340,8 @@ wasmtime_option_group! {
         pub tcp: Option<bool>,
         /// Indicates whether `wasi:sockets` UDP support is enabled or not.
         pub udp: Option<bool>,
+        /// Enable WASI APIs marked as: @unstable(feature = network-error-code)
+        pub network_error_code: Option<bool>,
         /// Allows imports from the `wasi_unstable` core wasm module.
         pub preview0: Option<bool>,
         /// Inherit all environment variables from the parent process.

crates/wasi/src/host/network.rs

@@ -19,7 +19,11 @@ where
 
     fn network_error_code(&mut self, err: Resource<Error>) -> anyhow::Result<Option<ErrorCode>> {
         let err = self.table().get(&err)?;
-        let _ = err; // TODO: should fill in this implementation
+
+        if let Some(err) = err.downcast_ref::<std::io::Error>() {
+            return Ok(Some(ErrorCode::from(err)));
+        }
+
         Ok(None)
     }
 }
@@ -39,8 +43,14 @@ where
 
 impl From<io::Error> for ErrorCode {
     fn from(value: io::Error) -> Self {
+        (&value).into()
+    }
+}
+
+impl From<&io::Error> for ErrorCode {
+    fn from(value: &io::Error) -> Self {
         // Attempt the more detailed native error code first:
-        if let Some(errno) = Errno::from_io_error(&value) {
+        if let Some(errno) = Errno::from_io_error(value) {
             return errno.into();
         }
 
@@ -69,7 +79,13 @@ impl From<io::Error> for ErrorCode {
 
 impl From<Errno> for ErrorCode {
     fn from(value: Errno) -> Self {
-        match value {
+        (&value).into()
+    }
+}
+
+impl From<&Errno> for ErrorCode {
+    fn from(value: &Errno) -> Self {
+        match *value {
             Errno::WOULDBLOCK => ErrorCode::WouldBlock,
             #[allow(unreachable_patterns)] // EWOULDBLOCK and EAGAIN can have the same value.
             Errno::AGAIN => ErrorCode::WouldBlock,

src/common.rs

@@ -337,6 +337,7 @@ impl RunCommon {
     pub fn compute_wasi_features(&self) -> LinkOptions {
         let mut options = LinkOptions::default();
         options.cli_exit_with_code(self.common.wasi.cli_exit_with_code.unwrap_or(false));
+        options.network_error_code(self.common.wasi.network_error_code.unwrap_or(false));
         options
     }
 }