Fixed breakage due to changes introduced in bytecodealliance/wasmtime@e…

…55fa3c
bytecodealliance · Jun 21, 2024 · c2021ee · c2021ee
1 parent 9a43806
commit c2021ee
Showing 1 changed file with 9 additions and 5 deletions.
diff --git a/src/WasiConfiguration.cs b/src/WasiConfiguration.cs
@@ -299,7 +299,9 @@ private unsafe void SetConfigArgs(Handle config)
             {
                 fixed (byte** arrayOfStringsPtrNamedArgs = args)
                 {
-                    Native.wasi_config_set_argv(config, _args.Count, arrayOfStringsPtrNamedArgs);
+                    // This shouldn't ever fail - it would only return false if `ToUTF8PtrArray` creates invalid UTF8 bytes!
+                    if (!Native.wasi_config_set_argv(config, (nuint)_args.Count, arrayOfStringsPtrNamedArgs))
+                        throw new WasmtimeException("Failed to encode string to UTF8");
                 }
             }
             finally
@@ -329,7 +331,9 @@ private unsafe void SetEnvironmentVariables(Handle config)
 
             try
             {
-                Native.wasi_config_set_env(config, _vars.Count, names, values);
+                // This shouldn't ever fail - it would only return false if `ToUTF8PtrArray` creates invalid UTF8 bytes!
+                if (!Native.wasi_config_set_env(config, (nuint)_vars.Count, names, values))
+                    throw new WasmtimeException("Failed to encode string to UTF8");
             }
             finally
             {
@@ -450,12 +454,12 @@ private static class Native
             public static extern void wasi_config_delete(IntPtr config);
 
             [DllImport(Engine.LibraryName)]
-            public unsafe static extern void wasi_config_set_argv(Handle config, int argc, byte** argv);
+            public unsafe static extern bool wasi_config_set_argv(Handle config, nuint argc, byte** argv);
 
             [DllImport(Engine.LibraryName)]
-            public static extern unsafe void wasi_config_set_env(
+            public static extern unsafe bool wasi_config_set_env(
                 Handle config,
-                int envc,
+                nuint envc,
                 byte*[] names,
                 byte*[] values
             );