thread: New thread_setname() function

5 files changed, 37 insertions, 0 deletions

diff --git a/build/has/pthread-set-name-np.c b/build/has/pthread-set-name-np.c
new file mode 100644
index 0000000..324aab9
--- /dev/null
+++ b/build/has/pthread-set-name-np.c
@@ -0,0 +1,10 @@
+// Copyright © Tavian Barnes <tavianator@tavianator.com>
+// SPDX-License-Identifier: 0BSD
+
+#include <pthread.h>
+#include <pthread_np.h>
+
+int main(void) {
+	pthread_set_name_np(pthread_self(), "name");
+	return 0;
+}
diff --git a/build/has/pthread-setname-np.c b/build/has/pthread-setname-np.c
new file mode 100644
index 0000000..a3b94c1
--- /dev/null
+++ b/build/has/pthread-setname-np.c
@@ -0,0 +1,8 @@
+// Copyright © Tavian Barnes <tavianator@tavianator.com>
+// SPDX-License-Identifier: 0BSD
+
+#include <pthread.h>
+
+int main(void) {
+	return pthread_setname_np(pthread_self(), "name");
+}
diff --git a/build/header.mk b/build/header.mk
index f940e52..8b697d6 100644
--- a/build/header.mk
+++ b/build/header.mk
@@ -38,6 +38,8 @@ HEADERS := \
     gen/has/posix-getdents.h \
     gen/has/posix-spawn-addfchdir-np.h \
     gen/has/posix-spawn-addfchdir.h \
+    gen/has/pthread-set-name-np.h \
+    gen/has/pthread-setname-np.h \
     gen/has/st-acmtim.h \
     gen/has/st-acmtimespec.h \
     gen/has/st-birthtim.h \
diff --git a/src/thread.c b/src/thread.c
index d61ff8c..b3604f8 100644
--- a/src/thread.c
+++ b/src/thread.c
@@ -9,6 +9,10 @@
 #include <errno.h>
 #include <pthread.h>
 
+#if __has_include(<pthread_np.h>)
+#  include <pthread_np.h>
+#endif
+
 #define THREAD_FALLIBLE(expr) \
 	do { \
 		int err = expr; \
@@ -32,6 +36,14 @@ int thread_create(pthread_t *thread, const pthread_attr_t *attr, thread_fn *fn,
 	THREAD_FALLIBLE(pthread_create(thread, attr, fn, arg));
 }
 
+void thread_setname(pthread_t thread, const char *name) {
+#if BFS_HAS_PTHREAD_SETNAME_NP
+	pthread_setname_np(thread, name);
+#elif BFS_HAS_PTHREAD_SET_NAME_NP
+	pthread_set_name_np(thread, name);
+#endif
+}
+
 void thread_join(pthread_t thread, void **ret) {
 	THREAD_INFALLIBLE(pthread_join(thread, ret));
 }
diff --git a/src/thread.h b/src/thread.h
index 7d12468..3dd8422 100644
--- a/src/thread.h
+++ b/src/thread.h
@@ -22,6 +22,11 @@ typedef void *thread_fn(void *arg);
 int thread_create(pthread_t *thread, const pthread_attr_t *attr, thread_fn *fn, void *arg);
 
 /**
+ * Set the name of a thread.
+ */
+void thread_setname(pthread_t thread, const char *name);
+
+/**
  * Wrapper for pthread_join().
  */
 void thread_join(pthread_t thread, void **ret);