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
TEST_IMPL(kill_invalid_signum)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
signum_test_cb(uv_signal_t * handle,int signum)47 static void signum_test_cb(uv_signal_t* handle, int signum) {
48 FATAL("signum_test_cb should not be called");
49 }
50
TEST_IMPL(win32_signum_number)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
signal_cb(uv_signal_t * handle,int signum)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
signal_cb_one_shot(uv_signal_t * handle,int signum)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
timer_cb(uv_timer_t * handle)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
start_watcher(uv_loop_t * loop,int signum,struct signal_ctx * ctx,int one_shot)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
start_timer(uv_loop_t * loop,int signum,struct timer_ctx * ctx)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
TEST_IMPL(we_get_signal)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
TEST_IMPL(we_get_signals)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
TEST_IMPL(we_get_signal_one_shot)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
TEST_IMPL(we_get_signals_mixed)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 */
326
