dragonfly: port std.Thread.setname/getname

2 files changed, 25 insertions, 0 deletions

diff --git a/lib/std/Thread.zig b/lib/std/Thread.zig
index ae3a3e9d9e..1b91ab597b 100644
--- a/lib/std/Thread.zig
+++ b/lib/std/Thread.zig
@@ -41,6 +41,7 @@ pub const max_name_len = switch (target.os.tag) {
     .netbsd => 31,
     .freebsd => 15,
     .openbsd => 31,
+    .dragonfly => 1023,
     .solaris => 31,
     else => 0,
 };
@@ -130,6 +131,17 @@ pub fn setName(self: Thread, name: []const u8) SetNameError!void {
 
             std.c.pthread_set_name_np(self.getHandle(), name_with_terminator.ptr);
         },
+        .dragonfly => if (use_pthreads) {
+            const err = std.c.pthread_setname_np(self.getHandle(), name_with_terminator.ptr);
+            switch (err) {
+                .SUCCESS => return,
+                .INVAL => unreachable,
+                .FAULT => unreachable,
+                .NAMETOOLONG => unreachable, // already checked
+                .SRCH => unreachable,
+                else => |e| return os.unexpectedErrno(e),
+            }
+        },
         else => return error.Unsupported,
     }
 }
@@ -219,6 +231,16 @@ pub fn getName(self: Thread, buffer_ptr: *[max_name_len:0]u8) GetNameError!?[]co
             std.c.pthread_get_name_np(self.getHandle(), buffer.ptr, max_name_len + 1);
             return std.mem.sliceTo(buffer, 0);
         },
+        .dragonfly => if (use_pthreads) {
+            const err = std.c.pthread_getname_np(self.getHandle(), buffer.ptr, max_name_len + 1);
+            switch (err) {
+                .SUCCESS => return std.mem.sliceTo(buffer, 0),
+                .INVAL => unreachable,
+                .FAULT => unreachable,
+                .SRCH => unreachable,
+                else => |e| return os.unexpectedErrno(e),
+            }
+        },
         else => return error.Unsupported,
     }
 }
diff --git a/lib/std/c/dragonfly.zig b/lib/std/c/dragonfly.zig
index a2b7e31b4f..737cb70f40 100644
--- a/lib/std/c/dragonfly.zig
+++ b/lib/std/c/dragonfly.zig
@@ -33,6 +33,9 @@ pub const pthread_attr_t = extern struct { // copied from freebsd
 
 pub const sem_t = ?*opaque {};
 
+pub extern "c" fn pthread_setname_np(thread: std.c.pthread_t, name: [*:0]const u8) E;
+pub extern "c" fn pthread_getname_np(thread: std.c.pthread_t, name: [*:0]u8, len: usize) E;
+
 // See:
 // - https://gitweb.dragonflybsd.org/dragonfly.git/blob/HEAD:/include/unistd.h
 // - https://gitweb.dragonflybsd.org/dragonfly.git/blob/HEAD:/sys/sys/types.h