|
160
|
1 /* Copyright libuv project 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 #ifndef _WIN32
|
|
|
22
|
|
|
23 #include "uv.h"
|
|
|
24 #include "task.h"
|
|
|
25
|
|
|
26 #include <string.h>
|
|
|
27 #include <unistd.h>
|
|
|
28
|
|
|
29 static uv_loop_t loop;
|
|
|
30 static uv_signal_t signal_hdl;
|
|
|
31 static uv_pipe_t pipe_hdl;
|
|
|
32 static uv_write_t write_req;
|
|
|
33 static char* buf;
|
|
|
34 static int close_cb_called;
|
|
|
35
|
|
|
36
|
|
|
37 static void stop_loop_cb(uv_signal_t* signal, int signum) {
|
|
|
38 ASSERT_EQ(signum, SIGPIPE);
|
|
|
39 uv_stop(signal->loop);
|
|
|
40 }
|
|
|
41
|
|
|
42 static void signal_cb(uv_signal_t* signal, int signum) {
|
|
|
43 ASSERT(0);
|
|
|
44 }
|
|
|
45
|
|
|
46 static void close_cb(uv_handle_t *handle) {
|
|
|
47 close_cb_called++;
|
|
|
48 }
|
|
|
49
|
|
|
50
|
|
|
51 static void write_cb(uv_write_t* req, int status) {
|
|
|
52 ASSERT_NOT_NULL(req);
|
|
|
53 ASSERT_EQ(status, UV_EPIPE);
|
|
|
54 free(buf);
|
|
|
55 uv_close((uv_handle_t *) &pipe_hdl, close_cb);
|
|
|
56 uv_close((uv_handle_t *) &signal_hdl, close_cb);
|
|
|
57 }
|
|
|
58
|
|
|
59
|
|
|
60 TEST_IMPL(signal_pending_on_close) {
|
|
|
61 int pipefds[2];
|
|
|
62 uv_buf_t buffer;
|
|
|
63 int r;
|
|
|
64
|
|
|
65 ASSERT_OK(uv_loop_init(&loop));
|
|
|
66
|
|
|
67 ASSERT_OK(uv_signal_init(&loop, &signal_hdl));
|
|
|
68
|
|
|
69 ASSERT_OK(uv_signal_start(&signal_hdl, signal_cb, SIGPIPE));
|
|
|
70
|
|
|
71 ASSERT_OK(pipe(pipefds));
|
|
|
72
|
|
|
73 ASSERT_OK(uv_pipe_init(&loop, &pipe_hdl, 0));
|
|
|
74
|
|
|
75 ASSERT_OK(uv_pipe_open(&pipe_hdl, pipefds[1]));
|
|
|
76
|
|
|
77 /* Write data large enough so it needs loop iteration */
|
|
|
78 buf = malloc(1<<24);
|
|
|
79 ASSERT_NOT_NULL(buf);
|
|
|
80 memset(buf, '.', 1<<24);
|
|
|
81 buffer = uv_buf_init(buf, 1<<24);
|
|
|
82
|
|
|
83 r = uv_write(&write_req, (uv_stream_t *) &pipe_hdl, &buffer, 1, write_cb);
|
|
|
84 ASSERT_OK(r);
|
|
|
85
|
|
|
86 /* cause a SIGPIPE on write in next iteration */
|
|
|
87 close(pipefds[0]);
|
|
|
88
|
|
|
89 ASSERT_OK(uv_run(&loop, UV_RUN_DEFAULT));
|
|
|
90
|
|
|
91 ASSERT_EQ(2, close_cb_called);
|
|
|
92
|
|
|
93 MAKE_VALGRIND_HAPPY(&loop);
|
|
|
94 return 0;
|
|
|
95 }
|
|
|
96
|
|
|
97
|
|
|
98 TEST_IMPL(signal_close_loop_alive) {
|
|
|
99 ASSERT_OK(uv_loop_init(&loop));
|
|
|
100 ASSERT_OK(uv_signal_init(&loop, &signal_hdl));
|
|
|
101 ASSERT_OK(uv_signal_start(&signal_hdl, stop_loop_cb, SIGPIPE));
|
|
|
102 uv_unref((uv_handle_t*) &signal_hdl);
|
|
|
103
|
|
|
104 ASSERT_OK(uv_kill(uv_os_getpid(), SIGPIPE));
|
|
|
105 ASSERT_OK(uv_run(&loop, UV_RUN_DEFAULT));
|
|
|
106 uv_close((uv_handle_t*) &signal_hdl, close_cb);
|
|
|
107 ASSERT_EQ(1, uv_loop_alive(&loop));
|
|
|
108
|
|
|
109 ASSERT_OK(uv_run(&loop, UV_RUN_DEFAULT));
|
|
|
110 ASSERT_EQ(1, close_cb_called);
|
|
|
111
|
|
|
112 MAKE_VALGRIND_HAPPY(&loop);
|
|
|
113 return 0;
|
|
|
114 }
|
|
|
115
|
|
|
116 #endif
|
