libpthread-218.1.3.tar.gz

[apple/libpthread.git] / tests / bsdthread_set_self.c

diff --git a/tests/bsdthread_set_self.c b/tests/bsdthread_set_self.c
new file mode 100644 (file)
index 0000000..dd22660
--- /dev/null
+++ b/tests/bsdthread_set_self.c
@@ -0,0 +1,87 @@
+#include <assert.h>
+#include <errno.h>
+#include <stdatomic.h>
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+#include <unistd.h>
+#include <time.h>
+#include <sys/types.h>
+#include <sys/sysctl.h>
+#include <sys/qos.h>
+
+#include <pthread.h>
+#include <pthread/tsd_private.h>
+#include <pthread/qos_private.h>
+#include <pthread/workqueue_private.h>
+
+#include <dispatch/dispatch.h>
+
+#include <darwintest.h>
+
+T_DECL(bsdthread_set_self_constrained_transition, "bsdthread_ctl(SET_SELF) with overcommit change",
+               T_META_ALL_VALID_ARCHS(YES))
+{
+       dispatch_async(dispatch_get_global_queue(0, 0), ^{
+               pthread_priority_t overcommit = (pthread_priority_t)_pthread_getspecific_direct(_PTHREAD_TSD_SLOT_PTHREAD_QOS_CLASS) |
+                       _PTHREAD_PRIORITY_OVERCOMMIT_FLAG;
+               pthread_priority_t constrained = overcommit & (~_PTHREAD_PRIORITY_OVERCOMMIT_FLAG);
+
+               T_ASSERT_POSIX_ZERO(_pthread_set_properties_self(_PTHREAD_SET_SELF_QOS_FLAG, overcommit, 0), NULL);
+               T_ASSERT_POSIX_ZERO(_pthread_set_properties_self(_PTHREAD_SET_SELF_QOS_FLAG, constrained, 0), NULL);
+               T_ASSERT_POSIX_ZERO(_pthread_set_properties_self(_PTHREAD_SET_SELF_QOS_FLAG, overcommit, 0), NULL);
+               T_ASSERT_POSIX_ZERO(_pthread_set_properties_self(_PTHREAD_SET_SELF_QOS_FLAG, constrained, 0), NULL);
+               T_ASSERT_POSIX_ZERO(_pthread_set_properties_self(_PTHREAD_SET_SELF_QOS_FLAG, overcommit, 0), NULL);
+               T_ASSERT_POSIX_ZERO(_pthread_set_properties_self(_PTHREAD_SET_SELF_QOS_FLAG, constrained, 0), NULL);
+
+               T_END;
+       });
+
+       dispatch_main();
+}
+
+T_DECL(bsdthread_set_self_constrained_threads, "bsdthread_ctl(SET_SELF) with overcommit change",
+               T_META_CHECK_LEAKS(NO), T_META_ALL_VALID_ARCHS(YES))
+{
+       static const int THREADS = 128;
+       static atomic_int threads_started;
+       dispatch_queue_t q = dispatch_queue_create("my queue", DISPATCH_QUEUE_CONCURRENT);
+       dispatch_set_target_queue(q, dispatch_get_global_queue(0, 0));
+       for (int i = 0; i < THREADS; i++) {
+               dispatch_async(q, ^{
+                       int thread_id = ++threads_started;
+                       T_PASS("Thread %d started successfully", thread_id);
+                       if (thread_id == THREADS){
+                               T_PASS("All threads started successfully");
+                               T_END;
+                       }
+
+                       pthread_priority_t overcommit = (pthread_priority_t)_pthread_getspecific_direct(_PTHREAD_TSD_SLOT_PTHREAD_QOS_CLASS) |
+                               _PTHREAD_PRIORITY_OVERCOMMIT_FLAG;
+                       T_ASSERT_POSIX_ZERO(_pthread_set_properties_self(_PTHREAD_SET_SELF_QOS_FLAG, overcommit, 0), NULL);
+
+                       uint64_t t = clock_gettime_nsec_np(CLOCK_MONOTONIC_RAW);
+                       while (t > clock_gettime_nsec_np(CLOCK_MONOTONIC_RAW) - (thread_id == 1 ? 30 : 60) * NSEC_PER_SEC) {
+                               sleep(1);
+                       }
+                       if (thread_id == 1) {
+                               T_FAIL("Where are my threads?");
+                               T_END;
+                       }
+               });
+       }
+
+       dispatch_main();
+}
+
+T_DECL(bsdthread_set_self_unbind, "bsdthread_ctl(SET_SELF) with kevent unbind",
+               T_META_ALL_VALID_ARCHS(YES))
+{
+       dispatch_async(dispatch_get_global_queue(0, 0), ^{
+               T_ASSERT_POSIX_ZERO(_pthread_set_properties_self(_PTHREAD_SET_SELF_WQ_KEVENT_UNBIND, 0, 0), NULL);
+
+               T_END;
+       });
+
+       dispatch_main();
+}