Commit c470016743474cd4686ddb523652b4ad7c7ef42f | zig

author	Frank Denis <124872+jedisct1@users.noreply.github.com> 2024-03-18 00:39:32+01:00
committer	Frank Denis <124872+jedisct1@users.noreply.github.com> 2024-03-18 16:51:41+01:00
log	c470016743474cd4686ddb523652b4ad7c7ef42f
tree	edff409066bf71d0d375c04c6d0382a32c627924
parent	8e7d9afdacef9d3a9443fc4b70cfe6aa229ecee5

Unbreak support for WASI threads

1 files changed, 36 insertions(+), 6 deletions(-)

lib
- std
  - Thread.zig

lib/std/Thread.zig+36-6

 @@ -324,6 +324,9 @@ pub const SpawnError = error{
     /// would exceed the limit.
     LockedMemoryLimitExceeded,
+    /// An allocator is required to spawn a thread
+    AllocatorRequired,
     Unexpected,
 };
 @@ -819,7 +822,7 @@ const WasiThreadImpl = struct {
                 \\ memory.atomic.wait32 0
                 \\ local.set %[ret]
                 : [ret] "=r" (-> u32),
-                : [ptr] "r" (&self.thread.tid.value),
+                : [ptr] "r" (&self.thread.tid.raw),
                   [expected] "r" (tid),
             );
             switch (result) {
 @@ -831,15 +834,42 @@ const WasiThreadImpl = struct {
         }
     }
-    fn spawn(config: std.Thread.SpawnConfig, comptime f: anytype, args: anytype) !WasiThreadImpl {
+    fn spawn(config: std.Thread.SpawnConfig, comptime f: anytype, args: anytype) SpawnError!WasiThreadImpl {
-        if (config.allocator == null) return error.OutOfMemory; // an allocator is required to spawn a WASI-thread
+        if (config.allocator == null) {
+            return error.AllocatorRequired; // an allocator is required to spawn a WASI thread
+        }
         // Wrapping struct required to hold the user-provided function arguments.
         const Wrapper = struct {
             args: @TypeOf(args),
             fn entry(ptr: usize) void {
                 const w: *@This() = @ptrFromInt(ptr);
-                @call(.auto, f, w.args);
+                const bad_fn_ret = "expected return type of startFn to be 'u8', 'noreturn', 'void', or '!void'";
+                switch (@typeInfo(@typeInfo(@TypeOf(f)).Fn.return_type.?)) {
+                    .NoReturn, .Void => {
+                        @call(.auto, w, args);
+                    },
+                    .Int => |info| {
+                        if (info.bits != 8) {
+                            @compileError(bad_fn_ret);
+                        }
+                        _ = @call(.auto, w, args); // WASI threads don't support exit status, ignore value
+                    },
+                    .ErrorUnion => |info| {
+                        if (info.payload != void) {
+                            @compileError(bad_fn_ret);
+                        }
+                        @call(.auto, f, args) catch |err| {
+                            std.debug.print("error: {s}\n", .{@errorName(err)});
+                            if (@errorReturnTrace()) |trace| {
+                                std.debug.dumpStackTrace(trace.*);
+                            }
+                        };
+                    },
+                    else => {
+                        @compileError(bad_fn_ret);
+                    },
+                }
             }
         };
 @@ -927,7 +957,7 @@ const WasiThreadImpl = struct {
                     \\ i32.const 0
                     \\ i32.atomic.store 0
                     :
-                    : [ptr] "r" (&arg.thread.tid.value),
+                    : [ptr] "r" (&arg.thread.tid.raw),
                 );
                 // Wake the main thread listening to this thread
 @@ -937,7 +967,7 @@ const WasiThreadImpl = struct {
                     \\ memory.atomic.notify 0
                     \\ drop # no need to know the waiters
                     :
-                    : [ptr] "r" (&arg.thread.tid.value),
+                    : [ptr] "r" (&arg.thread.tid.raw),
                 );
             },
             .completed => unreachable,