#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");
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);
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;
}