libdispatch-84.5.3.tar.gz

[apple/libdispatch.git] / testing / dispatch_proc.c

diff --git a/testing/dispatch_proc.c b/testing/dispatch_proc.c
new file mode 100644 (file)
index 0000000..6f5ef02
--- /dev/null
+++ b/testing/dispatch_proc.c
@@ -0,0 +1,89 @@
+#include <dispatch/dispatch.h>
+#include <stdio.h>
+#include <unistd.h>
+#include <stdlib.h>
+#include <assert.h>
+#include <spawn.h>
+#include <signal.h>
+#include <libkern/OSAtomic.h>
+
+#include "dispatch_test.h"
+
+#define PID_CNT 5
+
+static long event_cnt;
+static long cancel_cnt;
+
+int
+main(void)
+{
+       dispatch_source_t proc;
+       int res;
+       pid_t pid;
+
+       test_start("Dispatch Proc");
+       
+       // Creates a process and register multiple observers.  Send a signal,
+       // exit the process, etc., and verify all observers were notified.
+       
+       posix_spawnattr_t attr;
+       res = posix_spawnattr_init(&attr);
+       assert(res == 0);
+       res = posix_spawnattr_setflags(&attr, POSIX_SPAWN_START_SUSPENDED);
+       assert(res == 0);
+
+       char* args[] = {
+               "/bin/sleep", "2", NULL
+       };
+       
+       res = posix_spawnp(&pid, args[0], NULL, &attr, args, NULL);
+       if (res < 0) {
+               perror(args[0]);
+               exit(127);
+       }
+
+       res = posix_spawnattr_destroy(&attr);
+       assert(res == 0);
+
+       dispatch_queue_t semaphore = dispatch_queue_create("semaphore", NULL);
+       
+       assert(pid > 0);
+
+       int i;
+       for (i = 0; i < PID_CNT; ++i) {
+               dispatch_suspend(semaphore);
+               proc = dispatch_source_proc_create(pid, DISPATCH_PROC_EXIT, NULL, dispatch_get_main_queue(),
+                       ^(dispatch_event_t ev) {
+                               long err_dom, err_val;
+                               if ((err_dom = dispatch_event_get_error(ev, &err_val))) {
+                                       test_long("PROC error domain", err_dom, DISPATCH_ERROR_DOMAIN_POSIX);
+                                       test_long("PROC error value", err_val, ECANCELED);
+                                       cancel_cnt++;
+                                       dispatch_resume(semaphore);
+                               } else {
+                                       long flags = dispatch_event_get_flags(ev);
+                                       test_long("DISPATCH_PROC_EXIT", flags, DISPATCH_PROC_EXIT);
+                                       event_cnt++;
+                                       dispatch_release(dispatch_event_get_source(ev));
+                               }
+               });
+               test_ptr_notnull("dispatch_source_proc_create", proc);
+       }
+
+       dispatch_async(semaphore, ^{
+               int status;
+               int res2 = waitpid(pid, &status, 0);
+               assert(res2 != -1);
+               //int passed = (WIFEXITED(status) && WEXITSTATUS(status) == 0);
+               test_long("Sub-process exited", WEXITSTATUS(status) | WTERMSIG(status), 0);
+               test_long("Event count", event_cnt, PID_CNT);
+               test_long("Cancel count", cancel_cnt, PID_CNT);
+               test_stop();
+       });
+
+       kill(pid, SIGCONT);
+
+       dispatch_main();
+
+       return 0;
+}