libpthread-137.1.1.tar.gz

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

diff --git a/tests/wq_event_manager.c b/tests/wq_event_manager.c
new file mode 100644 (file)
index 0000000..bed9faa
--- /dev/null
+++ b/tests/wq_event_manager.c
@@ -0,0 +1,95 @@
+#include <assert.h>
+#include <errno.h>
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+#include <unistd.h>
+#include <sys/types.h>
+#include <sys/sysctl.h>
+#include <sys/qos.h>
+
+#include <dispatch/dispatch.h>
+
+#include "../private/workqueue_private.h"
+#include "../private/qos_private.h"
+
+#include "wq_kevent.h"
+
+static dispatch_semaphore_t sema;
+static dispatch_time_t timeout;
+
+static int do_wait(int threads){
+       for (int i = 0; i < threads; i++){
+               int ret = dispatch_semaphore_wait(sema, timeout);
+               if (ret){
+                       fprintf(stderr, "timout waiting for thread %d.\n", i);
+                       return 1;
+               }
+       }
+       fprintf(stderr, "\tsuccessfully signaled by %d threads.\n", threads);
+       return 0;
+}
+
+static void workqueue_func(pthread_priority_t priority){
+       fprintf(stderr, "WARNING: workqueue_func called.\n");
+       dispatch_semaphore_signal(sema);
+}
+
+void (^cb)(void) = NULL;
+static void workqueue_func_kevent(void **buf, int *count){
+       pthread_priority_t p = (pthread_priority_t)pthread_getspecific(4);
+       fprintf(stderr, "\tthread with qos %s spawned.\n", describe_pri(p));
+
+       if (cb){
+               cb();
+       }
+
+       dispatch_semaphore_signal(sema);
+}
+
+int main(int argc, char *argv[]){
+       int ret = 0;
+       int exit_status = 0;
+
+       ret = _pthread_workqueue_init_with_kevent(workqueue_func, workqueue_func_kevent, 0, 0);
+       assert(ret == 0);
+
+       sema = dispatch_semaphore_create(0);
+       assert(sema != NULL);
+       timeout = dispatch_time(DISPATCH_TIME_NOW, 5LL * NSEC_PER_SEC);
+
+       // one event manager
+       requests[0].priority = _pthread_qos_class_encode(QOS_CLASS_UNSPECIFIED, 0, _PTHREAD_PRIORITY_EVENT_MANAGER_FLAG);
+       requests[0].count = 1;
+
+       if ((ret = do_req()) < 0) return ret;
+       if ((ret = do_wait(1)) < 0) return ret;
+
+       fprintf(stderr, "event_manager_priority -> USER_INITIATED\n");
+       _pthread_workqueue_set_event_manager_priority(_pthread_qos_class_encode(QOS_CLASS_USER_INITIATED, 0, 0));
+
+       if ((ret = do_req()) < 0) return ret;
+       if ((ret = do_wait(1)) < 0) return ret;
+
+       fprintf(stderr, "event_manager_priority -> UTILITY\n");
+       _pthread_workqueue_set_event_manager_priority(_pthread_qos_class_encode(QOS_CLASS_UTILITY, 0, 0));
+
+       if ((ret = do_req()) < 0) return ret;
+       if ((ret = do_wait(1)) < 0) return ret;
+
+       fprintf(stderr, "event_manager_priority -> 60\n");
+       _pthread_workqueue_set_event_manager_priority(_PTHREAD_PRIORITY_SCHED_PRI_FLAG | 60);;
+
+       cb = ^(void){sleep(2);};
+
+       if ((ret = do_req()) < 0) return ret;
+       if ((ret = do_wait(1)) < 0) return ret;
+
+       requests[0].priority = _pthread_qos_class_encode(QOS_CLASS_USER_INITIATED, 0, 0);
+       requests[0].count = 1;
+
+       if ((ret = do_req()) < 0) return ret;
+       if ((ret = do_wait(1)) < 0) return ret;
+       
+       return 0;
+}