|
160
|
1 /* Copyright Joyent, Inc. and other Node contributors. All rights reserved.
|
|
|
2 *
|
|
|
3 * Permission is hereby granted, free of charge, to any person obtaining a copy
|
|
|
4 * of this software and associated documentation files (the "Software"), to
|
|
|
5 * deal in the Software without restriction, including without limitation the
|
|
|
6 * rights to use, copy, modify, merge, publish, distribute, sublicense, and/or
|
|
|
7 * sell copies of the Software, and to permit persons to whom the Software is
|
|
|
8 * furnished to do so, subject to the following conditions:
|
|
|
9 *
|
|
|
10 * The above copyright notice and this permission notice shall be included in
|
|
|
11 * all copies or substantial portions of the Software.
|
|
|
12 *
|
|
|
13 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
|
|
14 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
|
|
15 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
|
|
16 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
|
|
17 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
|
|
|
18 * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS
|
|
|
19 * IN THE SOFTWARE.
|
|
|
20 */
|
|
|
21
|
|
|
22 #include "uv.h"
|
|
|
23 #include "task.h"
|
|
|
24
|
|
|
25 #ifndef _WIN32
|
|
|
26 #include <unistd.h>
|
|
|
27 #endif
|
|
|
28
|
|
|
29 TEST_IMPL(kill_invalid_signum) {
|
|
|
30 uv_pid_t pid;
|
|
|
31
|
|
|
32 pid = uv_os_getpid();
|
|
|
33
|
|
|
34 ASSERT_EQ(uv_kill(pid, -1), UV_EINVAL);
|
|
|
35 #ifdef _WIN32
|
|
|
36 /* NSIG is not available on all platforms. */
|
|
|
37 ASSERT_EQ(uv_kill(pid, NSIG), UV_EINVAL);
|
|
|
38 #endif
|
|
|
39 ASSERT_EQ(uv_kill(pid, 4096), UV_EINVAL);
|
|
|
40
|
|
|
41 MAKE_VALGRIND_HAPPY(uv_default_loop());
|
|
|
42 return 0;
|
|
|
43 }
|
|
|
44
|
|
|
45 /* For Windows we test only signum handling */
|
|
|
46 #ifdef _WIN32
|
|
|
47 static void signum_test_cb(uv_signal_t* handle, int signum) {
|
|
|
48 FATAL("signum_test_cb should not be called");
|
|
|
49 }
|
|
|
50
|
|
|
51 TEST_IMPL(win32_signum_number) {
|
|
|
52 uv_signal_t signal;
|
|
|
53 uv_loop_t* loop;
|
|
|
54
|
|
|
55 loop = uv_default_loop();
|
|
|
56 uv_signal_init(loop, &signal);
|
|
|
57
|
|
|
58 ASSERT_EQ(uv_signal_start(&signal, signum_test_cb, 0), UV_EINVAL);
|
|
|
59 ASSERT_OK(uv_signal_start(&signal, signum_test_cb, SIGINT));
|
|
|
60 ASSERT_OK(uv_signal_start(&signal, signum_test_cb, SIGBREAK));
|
|
|
61 ASSERT_OK(uv_signal_start(&signal, signum_test_cb, SIGHUP));
|
|
|
62 ASSERT_OK(uv_signal_start(&signal, signum_test_cb, SIGWINCH));
|
|
|
63 ASSERT_OK(uv_signal_start(&signal, signum_test_cb, SIGILL));
|
|
|
64 ASSERT_OK(uv_signal_start(&signal, signum_test_cb, SIGABRT_COMPAT));
|
|
|
65 ASSERT_OK(uv_signal_start(&signal, signum_test_cb, SIGFPE));
|
|
|
66 ASSERT_OK(uv_signal_start(&signal, signum_test_cb, SIGSEGV));
|
|
|
67 ASSERT_OK(uv_signal_start(&signal, signum_test_cb, SIGTERM));
|
|
|
68 ASSERT_OK(uv_signal_start(&signal, signum_test_cb, SIGABRT));
|
|
|
69 ASSERT_EQ(uv_signal_start(&signal, signum_test_cb, -1), UV_EINVAL);
|
|
|
70 ASSERT_EQ(uv_signal_start(&signal, signum_test_cb, NSIG), UV_EINVAL);
|
|
|
71 ASSERT_EQ(uv_signal_start(&signal, signum_test_cb, 1024), UV_EINVAL);
|
|
|
72 MAKE_VALGRIND_HAPPY(loop);
|
|
|
73 return 0;
|
|
|
74 }
|
|
|
75 #else
|
|
|
76 #include <errno.h>
|
|
|
77 #include <signal.h>
|
|
|
78 #include <stdarg.h>
|
|
|
79 #include <stdio.h>
|
|
|
80 #include <stdlib.h>
|
|
|
81 #include <string.h>
|
|
|
82 #include <unistd.h>
|
|
|
83
|
|
|
84 #define NSIGNALS 10
|
|
|
85
|
|
|
86 struct timer_ctx {
|
|
|
87 unsigned int ncalls;
|
|
|
88 uv_timer_t handle;
|
|
|
89 int signum;
|
|
|
90 };
|
|
|
91
|
|
|
92 struct signal_ctx {
|
|
|
93 enum { CLOSE, STOP, NOOP } stop_or_close;
|
|
|
94 unsigned int ncalls;
|
|
|
95 uv_signal_t handle;
|
|
|
96 int signum;
|
|
|
97 int one_shot;
|
|
|
98 };
|
|
|
99
|
|
|
100
|
|
|
101 static void signal_cb(uv_signal_t* handle, int signum) {
|
|
|
102 struct signal_ctx* ctx = container_of(handle, struct signal_ctx, handle);
|
|
|
103 ASSERT_EQ(signum, ctx->signum);
|
|
|
104 if (++ctx->ncalls == NSIGNALS) {
|
|
|
105 if (ctx->stop_or_close == STOP)
|
|
|
106 uv_signal_stop(handle);
|
|
|
107 else if (ctx->stop_or_close == CLOSE)
|
|
|
108 uv_close((uv_handle_t*)handle, NULL);
|
|
|
109 else
|
|
|
110 ASSERT(0);
|
|
|
111 }
|
|
|
112 }
|
|
|
113
|
|
|
114 static void signal_cb_one_shot(uv_signal_t* handle, int signum) {
|
|
|
115 struct signal_ctx* ctx = container_of(handle, struct signal_ctx, handle);
|
|
|
116 ASSERT_EQ(signum, ctx->signum);
|
|
|
117 ASSERT_EQ(1, ++ctx->ncalls);
|
|
|
118 if (ctx->stop_or_close == CLOSE)
|
|
|
119 uv_close((uv_handle_t*)handle, NULL);
|
|
|
120 }
|
|
|
121
|
|
|
122
|
|
|
123 static void timer_cb(uv_timer_t* handle) {
|
|
|
124 struct timer_ctx* ctx = container_of(handle, struct timer_ctx, handle);
|
|
|
125
|
|
|
126 raise(ctx->signum);
|
|
|
127
|
|
|
128 if (++ctx->ncalls == NSIGNALS)
|
|
|
129 uv_close((uv_handle_t*)handle, NULL);
|
|
|
130 }
|
|
|
131
|
|
|
132
|
|
|
133 static void start_watcher(uv_loop_t* loop,
|
|
|
134 int signum,
|
|
|
135 struct signal_ctx* ctx,
|
|
|
136 int one_shot) {
|
|
|
137 ctx->ncalls = 0;
|
|
|
138 ctx->signum = signum;
|
|
|
139 ctx->stop_or_close = CLOSE;
|
|
|
140 ctx->one_shot = one_shot;
|
|
|
141 ASSERT_OK(uv_signal_init(loop, &ctx->handle));
|
|
|
142 if (one_shot)
|
|
|
143 ASSERT_OK(uv_signal_start_oneshot(&ctx->handle, signal_cb_one_shot, signum));
|
|
|
144 else
|
|
|
145 ASSERT_OK(uv_signal_start(&ctx->handle, signal_cb, signum));
|
|
|
146 }
|
|
|
147
|
|
|
148 static void start_timer(uv_loop_t* loop, int signum, struct timer_ctx* ctx) {
|
|
|
149 ctx->ncalls = 0;
|
|
|
150 ctx->signum = signum;
|
|
|
151 ASSERT_OK(uv_timer_init(loop, &ctx->handle));
|
|
|
152 ASSERT_OK(uv_timer_start(&ctx->handle, timer_cb, 5, 5));
|
|
|
153 }
|
|
|
154
|
|
|
155
|
|
|
156 TEST_IMPL(we_get_signal) {
|
|
|
157 struct signal_ctx sc;
|
|
|
158 struct timer_ctx tc;
|
|
|
159 uv_loop_t* loop;
|
|
|
160
|
|
|
161 loop = uv_default_loop();
|
|
|
162 start_timer(loop, SIGCHLD, &tc);
|
|
|
163 start_watcher(loop, SIGCHLD, &sc, 0);
|
|
|
164 sc.stop_or_close = STOP; /* stop, don't close the signal handle */
|
|
|
165 ASSERT_OK(uv_run(loop, UV_RUN_DEFAULT));
|
|
|
166 ASSERT_EQ(tc.ncalls, NSIGNALS);
|
|
|
167 ASSERT_EQ(sc.ncalls, NSIGNALS);
|
|
|
168
|
|
|
169 start_timer(loop, SIGCHLD, &tc);
|
|
|
170 ASSERT_OK(uv_run(loop, UV_RUN_DEFAULT));
|
|
|
171 ASSERT_EQ(tc.ncalls, NSIGNALS);
|
|
|
172 ASSERT_EQ(sc.ncalls, NSIGNALS);
|
|
|
173
|
|
|
174 sc.ncalls = 0;
|
|
|
175 sc.stop_or_close = CLOSE; /* now close it when it's done */
|
|
|
176 uv_signal_start(&sc.handle, signal_cb, SIGCHLD);
|
|
|
177
|
|
|
178 start_timer(loop, SIGCHLD, &tc);
|
|
|
179 ASSERT_OK(uv_run(loop, UV_RUN_DEFAULT));
|
|
|
180 ASSERT_EQ(tc.ncalls, NSIGNALS);
|
|
|
181 ASSERT_EQ(sc.ncalls, NSIGNALS);
|
|
|
182
|
|
|
183 MAKE_VALGRIND_HAPPY(loop);
|
|
|
184 return 0;
|
|
|
185 }
|
|
|
186
|
|
|
187
|
|
|
188 TEST_IMPL(we_get_signals) {
|
|
|
189 struct signal_ctx sc[4];
|
|
|
190 struct timer_ctx tc[2];
|
|
|
191 uv_loop_t* loop;
|
|
|
192 unsigned int i;
|
|
|
193
|
|
|
194 loop = uv_default_loop();
|
|
|
195 start_watcher(loop, SIGUSR1, sc + 0, 0);
|
|
|
196 start_watcher(loop, SIGUSR1, sc + 1, 0);
|
|
|
197 start_watcher(loop, SIGUSR2, sc + 2, 0);
|
|
|
198 start_watcher(loop, SIGUSR2, sc + 3, 0);
|
|
|
199 start_timer(loop, SIGUSR1, tc + 0);
|
|
|
200 start_timer(loop, SIGUSR2, tc + 1);
|
|
|
201 ASSERT_OK(uv_run(loop, UV_RUN_DEFAULT));
|
|
|
202
|
|
|
203 for (i = 0; i < ARRAY_SIZE(sc); i++)
|
|
|
204 ASSERT_EQ(sc[i].ncalls, NSIGNALS);
|
|
|
205
|
|
|
206 for (i = 0; i < ARRAY_SIZE(tc); i++)
|
|
|
207 ASSERT_EQ(tc[i].ncalls, NSIGNALS);
|
|
|
208
|
|
|
209 MAKE_VALGRIND_HAPPY(loop);
|
|
|
210 return 0;
|
|
|
211 }
|
|
|
212
|
|
|
213 TEST_IMPL(we_get_signal_one_shot) {
|
|
|
214 struct signal_ctx sc;
|
|
|
215 struct timer_ctx tc;
|
|
|
216 uv_loop_t* loop;
|
|
|
217
|
|
|
218 loop = uv_default_loop();
|
|
|
219 start_timer(loop, SIGCHLD, &tc);
|
|
|
220 start_watcher(loop, SIGCHLD, &sc, 1);
|
|
|
221 sc.stop_or_close = NOOP;
|
|
|
222 ASSERT_OK(uv_run(loop, UV_RUN_DEFAULT));
|
|
|
223 ASSERT_EQ(tc.ncalls, NSIGNALS);
|
|
|
224 ASSERT_EQ(1, sc.ncalls);
|
|
|
225
|
|
|
226 start_timer(loop, SIGCHLD, &tc);
|
|
|
227 ASSERT_OK(uv_run(loop, UV_RUN_DEFAULT));
|
|
|
228 ASSERT_EQ(1, sc.ncalls);
|
|
|
229
|
|
|
230 sc.ncalls = 0;
|
|
|
231 sc.stop_or_close = CLOSE; /* now close it when it's done */
|
|
|
232 uv_signal_start_oneshot(&sc.handle, signal_cb_one_shot, SIGCHLD);
|
|
|
233 start_timer(loop, SIGCHLD, &tc);
|
|
|
234 ASSERT_OK(uv_run(loop, UV_RUN_DEFAULT));
|
|
|
235 ASSERT_EQ(tc.ncalls, NSIGNALS);
|
|
|
236 ASSERT_EQ(1, sc.ncalls);
|
|
|
237
|
|
|
238 MAKE_VALGRIND_HAPPY(loop);
|
|
|
239 return 0;
|
|
|
240 }
|
|
|
241
|
|
|
242 TEST_IMPL(we_get_signals_mixed) {
|
|
|
243 struct signal_ctx sc[4];
|
|
|
244 struct timer_ctx tc;
|
|
|
245 uv_loop_t* loop;
|
|
|
246
|
|
|
247 loop = uv_default_loop();
|
|
|
248
|
|
|
249 /* 2 one-shot */
|
|
|
250 start_timer(loop, SIGCHLD, &tc);
|
|
|
251 start_watcher(loop, SIGCHLD, sc + 0, 1);
|
|
|
252 start_watcher(loop, SIGCHLD, sc + 1, 1);
|
|
|
253 sc[0].stop_or_close = CLOSE;
|
|
|
254 sc[1].stop_or_close = CLOSE;
|
|
|
255 ASSERT_OK(uv_run(loop, UV_RUN_DEFAULT));
|
|
|
256 ASSERT_EQ(tc.ncalls, NSIGNALS);
|
|
|
257 ASSERT_EQ(1, sc[0].ncalls);
|
|
|
258 ASSERT_EQ(1, sc[1].ncalls);
|
|
|
259
|
|
|
260 /* 2 one-shot, 1 normal then remove normal */
|
|
|
261 start_timer(loop, SIGCHLD, &tc);
|
|
|
262 start_watcher(loop, SIGCHLD, sc + 0, 1);
|
|
|
263 start_watcher(loop, SIGCHLD, sc + 1, 1);
|
|
|
264 sc[0].stop_or_close = CLOSE;
|
|
|
265 sc[1].stop_or_close = CLOSE;
|
|
|
266 start_watcher(loop, SIGCHLD, sc + 2, 0);
|
|
|
267 uv_close((uv_handle_t*)&(sc[2]).handle, NULL);
|
|
|
268 ASSERT_OK(uv_run(loop, UV_RUN_DEFAULT));
|
|
|
269 ASSERT_EQ(tc.ncalls, NSIGNALS);
|
|
|
270 ASSERT_EQ(1, sc[0].ncalls);
|
|
|
271 ASSERT_EQ(1, sc[1].ncalls);
|
|
|
272 ASSERT_OK(sc[2].ncalls);
|
|
|
273
|
|
|
274 /* 2 normal, 1 one-shot then remove one-shot */
|
|
|
275 start_timer(loop, SIGCHLD, &tc);
|
|
|
276 start_watcher(loop, SIGCHLD, sc + 0, 0);
|
|
|
277 start_watcher(loop, SIGCHLD, sc + 1, 0);
|
|
|
278 sc[0].stop_or_close = CLOSE;
|
|
|
279 sc[1].stop_or_close = CLOSE;
|
|
|
280 start_watcher(loop, SIGCHLD, sc + 2, 1);
|
|
|
281 uv_close((uv_handle_t*)&(sc[2]).handle, NULL);
|
|
|
282 ASSERT_OK(uv_run(loop, UV_RUN_DEFAULT));
|
|
|
283 ASSERT_EQ(tc.ncalls, NSIGNALS);
|
|
|
284 ASSERT_EQ(sc[0].ncalls, NSIGNALS);
|
|
|
285 ASSERT_EQ(sc[1].ncalls, NSIGNALS);
|
|
|
286 ASSERT_OK(sc[2].ncalls);
|
|
|
287
|
|
|
288 /* 2 normal, 2 one-shot then remove 2 normal */
|
|
|
289 start_timer(loop, SIGCHLD, &tc);
|
|
|
290 start_watcher(loop, SIGCHLD, sc + 0, 0);
|
|
|
291 start_watcher(loop, SIGCHLD, sc + 1, 0);
|
|
|
292 start_watcher(loop, SIGCHLD, sc + 2, 1);
|
|
|
293 start_watcher(loop, SIGCHLD, sc + 3, 1);
|
|
|
294 sc[2].stop_or_close = CLOSE;
|
|
|
295 sc[3].stop_or_close = CLOSE;
|
|
|
296 uv_close((uv_handle_t*)&(sc[0]).handle, NULL);
|
|
|
297 uv_close((uv_handle_t*)&(sc[1]).handle, NULL);
|
|
|
298 ASSERT_OK(uv_run(loop, UV_RUN_DEFAULT));
|
|
|
299 ASSERT_EQ(tc.ncalls, NSIGNALS);
|
|
|
300 ASSERT_OK(sc[0].ncalls);
|
|
|
301 ASSERT_OK(sc[1].ncalls);
|
|
|
302 ASSERT_EQ(1, sc[2].ncalls);
|
|
|
303 ASSERT_EQ(1, sc[2].ncalls);
|
|
|
304
|
|
|
305 /* 1 normal, 1 one-shot, 2 normal then remove 1st normal, 2nd normal */
|
|
|
306 start_timer(loop, SIGCHLD, &tc);
|
|
|
307 start_watcher(loop, SIGCHLD, sc + 0, 0);
|
|
|
308 start_watcher(loop, SIGCHLD, sc + 1, 1);
|
|
|
309 start_watcher(loop, SIGCHLD, sc + 2, 0);
|
|
|
310 start_watcher(loop, SIGCHLD, sc + 3, 0);
|
|
|
311 sc[3].stop_or_close = CLOSE;
|
|
|
312 uv_close((uv_handle_t*)&(sc[0]).handle, NULL);
|
|
|
313 uv_close((uv_handle_t*)&(sc[2]).handle, NULL);
|
|
|
314 ASSERT_OK(uv_run(loop, UV_RUN_DEFAULT));
|
|
|
315 ASSERT_EQ(tc.ncalls, NSIGNALS);
|
|
|
316 ASSERT_OK(sc[0].ncalls);
|
|
|
317 ASSERT_EQ(1, sc[1].ncalls);
|
|
|
318 ASSERT_OK(sc[2].ncalls);
|
|
|
319 ASSERT_EQ(sc[3].ncalls, NSIGNALS);
|
|
|
320
|
|
|
321 MAKE_VALGRIND_HAPPY(loop);
|
|
|
322 return 0;
|
|
|
323 }
|
|
|
324
|
|
|
325 #endif /* _WIN32 */
|
