1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
|
/*
* Copyright (c) 2008-2009 Apple Inc. All rights reserved.
*
* @APPLE_APACHE_LICENSE_HEADER_START@
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*
* @APPLE_APACHE_LICENSE_HEADER_END@
*/
#include "config/config.h"
#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;
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.
//
// Simple child process that sleeps 2 seconds.
//
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 completion = dispatch_queue_create("completion", NULL);
assert(pid > 0);
//
// Suspend the "completion" queue when each observer is created.
// Each observer resumes the queue when the child process exits.
// If the queue is resumed too few times (indicating that not all
// observers received the exit event) then the test case will not exit
// within the alloted time and result in failure.
//
int i;
for (i = 0; i < PID_CNT; ++i) {
dispatch_suspend(completion);
proc = dispatch_source_create(DISPATCH_SOURCE_TYPE_PROC, pid, DISPATCH_PROC_EXIT, dispatch_get_main_queue());
test_ptr_notnull("DISPATCH_SOURCE_TYPE_PROC", proc);
dispatch_source_set_event_handler(proc, ^{
long flags = dispatch_source_get_data(proc);
test_long("DISPATCH_PROC_EXIT", flags, DISPATCH_PROC_EXIT);
event_cnt++;
dispatch_release(proc);
dispatch_resume(completion);
});
dispatch_resume(proc);
}
//
// The completion block will be pending on the completion queue until it
// has been fully resumed, at which point the test will exit successfully.
//
dispatch_async(completion, ^{
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_stop();
});
kill(pid, SIGCONT);
dispatch_main();
return 0;
}
|
