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
29TEST_IMPL(kill_invalid_signum) {
30 uv_pid_t pid;
31
32 pid = uv_os_getpid();
33
34 ASSERT(uv_kill(pid, -1) == UV_EINVAL);
35#ifdef _WIN32
36 /* NSIG is not available on all platforms. */
37 ASSERT(uv_kill(pid, NSIG) == UV_EINVAL);
38#endif
39 ASSERT(uv_kill(pid, 4096) == UV_EINVAL);
40
41 MAKE_VALGRIND_HAPPY();
42 return 0;
43}
44
45/* For Windows we test only signum handling */
46#ifdef _WIN32
47static void signum_test_cb(uv_signal_t* handle, int signum) {
48 FATAL("signum_test_cb should not be called");
49}
50
51TEST_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(uv_signal_start(&signal, signum_test_cb, 0) == UV_EINVAL);
59 ASSERT(uv_signal_start(&signal, signum_test_cb, SIGINT) == 0);
60 ASSERT(uv_signal_start(&signal, signum_test_cb, SIGBREAK) == 0);
61 ASSERT(uv_signal_start(&signal, signum_test_cb, SIGHUP) == 0);
62 ASSERT(uv_signal_start(&signal, signum_test_cb, SIGWINCH) == 0);
63 ASSERT(uv_signal_start(&signal, signum_test_cb, SIGILL) == 0);
64 ASSERT(uv_signal_start(&signal, signum_test_cb, SIGABRT_COMPAT) == 0);
65 ASSERT(uv_signal_start(&signal, signum_test_cb, SIGFPE) == 0);
66 ASSERT(uv_signal_start(&signal, signum_test_cb, SIGSEGV) == 0);
67 ASSERT(uv_signal_start(&signal, signum_test_cb, SIGTERM) == 0);
68 ASSERT(uv_signal_start(&signal, signum_test_cb, SIGABRT) == 0);
69 ASSERT(uv_signal_start(&signal, signum_test_cb, -1) == UV_EINVAL);
70 ASSERT(uv_signal_start(&signal, signum_test_cb, NSIG) == UV_EINVAL);
71 ASSERT(uv_signal_start(&signal, signum_test_cb, 1024) == UV_EINVAL);
72 MAKE_VALGRIND_HAPPY();
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
86struct timer_ctx {
87 unsigned int ncalls;
88 uv_timer_t handle;
89 int signum;
90};
91
92struct 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
101static void signal_cb(uv_signal_t* handle, int signum) {
102 struct signal_ctx* ctx = container_of(handle, struct signal_ctx, handle);
103 ASSERT(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
114static 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(signum == ctx->signum);
117 ASSERT(++ctx->ncalls == 1);
118 if (ctx->stop_or_close == CLOSE)
119 uv_close((uv_handle_t*)handle, NULL);
120}
121
122
123static 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
133static 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(0 == uv_signal_init(loop, &ctx->handle));
142 if (one_shot)
143 ASSERT(0 == uv_signal_start_oneshot(&ctx->handle, signal_cb_one_shot, signum));
144 else
145 ASSERT(0 == uv_signal_start(&ctx->handle, signal_cb, signum));
146}
147
148static void start_timer(uv_loop_t* loop, int signum, struct timer_ctx* ctx) {
149 ctx->ncalls = 0;
150 ctx->signum = signum;
151 ASSERT(0 == uv_timer_init(loop, &ctx->handle));
152 ASSERT(0 == uv_timer_start(&ctx->handle, timer_cb, 5, 5));
153}
154
155
156TEST_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(0 == uv_run(loop, UV_RUN_DEFAULT));
166 ASSERT(tc.ncalls == NSIGNALS);
167 ASSERT(sc.ncalls == NSIGNALS);
168
169 start_timer(loop, SIGCHLD, &tc);
170 ASSERT(0 == uv_run(loop, UV_RUN_DEFAULT));
171 ASSERT(tc.ncalls == NSIGNALS);
172 ASSERT(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(0 == uv_run(loop, UV_RUN_DEFAULT));
180 ASSERT(tc.ncalls == NSIGNALS);
181 ASSERT(sc.ncalls == NSIGNALS);
182
183 MAKE_VALGRIND_HAPPY();
184 return 0;
185}
186
187
188TEST_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(0 == uv_run(loop, UV_RUN_DEFAULT));
202
203 for (i = 0; i < ARRAY_SIZE(sc); i++)
204 ASSERT(sc[i].ncalls == NSIGNALS);
205
206 for (i = 0; i < ARRAY_SIZE(tc); i++)
207 ASSERT(tc[i].ncalls == NSIGNALS);
208
209 MAKE_VALGRIND_HAPPY();
210 return 0;
211}
212
213TEST_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(0 == uv_run(loop, UV_RUN_DEFAULT));
223 ASSERT(tc.ncalls == NSIGNALS);
224 ASSERT(sc.ncalls == 1);
225
226 start_timer(loop, SIGCHLD, &tc);
227 ASSERT(0 == uv_run(loop, UV_RUN_DEFAULT));
228 ASSERT(sc.ncalls == 1);
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(0 == uv_run(loop, UV_RUN_DEFAULT));
235 ASSERT(tc.ncalls == NSIGNALS);
236 ASSERT(sc.ncalls == 1);
237
238 MAKE_VALGRIND_HAPPY();
239 return 0;
240}
241
242TEST_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(0 == uv_run(loop, UV_RUN_DEFAULT));
256 ASSERT(tc.ncalls == NSIGNALS);
257 ASSERT(sc[0].ncalls == 1);
258 ASSERT(sc[1].ncalls == 1);
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(0 == uv_run(loop, UV_RUN_DEFAULT));
269 ASSERT(tc.ncalls == NSIGNALS);
270 ASSERT(sc[0].ncalls == 1);
271 ASSERT(sc[1].ncalls == 1);
272 ASSERT(sc[2].ncalls == 0);
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(0 == uv_run(loop, UV_RUN_DEFAULT));
283 ASSERT(tc.ncalls == NSIGNALS);
284 ASSERT(sc[0].ncalls == NSIGNALS);
285 ASSERT(sc[1].ncalls == NSIGNALS);
286 ASSERT(sc[2].ncalls == 0);
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(0 == uv_run(loop, UV_RUN_DEFAULT));
299 ASSERT(tc.ncalls == NSIGNALS);
300 ASSERT(sc[0].ncalls == 0);
301 ASSERT(sc[1].ncalls == 0);
302 ASSERT(sc[2].ncalls == 1);
303 ASSERT(sc[2].ncalls == 1);
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(0 == uv_run(loop, UV_RUN_DEFAULT));
315 ASSERT(tc.ncalls == NSIGNALS);
316 ASSERT(sc[0].ncalls == 0);
317 ASSERT(sc[1].ncalls == 1);
318 ASSERT(sc[2].ncalls == 0);
319 ASSERT(sc[3].ncalls == NSIGNALS);
320
321 MAKE_VALGRIND_HAPPY();
322 return 0;
323}
324
325#endif /* _WIN32 */
326