1*0e552da7Schristos /* Copyright Joyent, Inc. and other Node contributors. All rights reserved.
2*0e552da7Schristos *
3*0e552da7Schristos * Permission is hereby granted, free of charge, to any person obtaining a copy
4*0e552da7Schristos * of this software and associated documentation files (the "Software"), to
5*0e552da7Schristos * deal in the Software without restriction, including without limitation the
6*0e552da7Schristos * rights to use, copy, modify, merge, publish, distribute, sublicense, and/or
7*0e552da7Schristos * sell copies of the Software, and to permit persons to whom the Software is
8*0e552da7Schristos * furnished to do so, subject to the following conditions:
9*0e552da7Schristos *
10*0e552da7Schristos * The above copyright notice and this permission notice shall be included in
11*0e552da7Schristos * all copies or substantial portions of the Software.
12*0e552da7Schristos *
13*0e552da7Schristos * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
14*0e552da7Schristos * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
15*0e552da7Schristos * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
16*0e552da7Schristos * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
17*0e552da7Schristos * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
18*0e552da7Schristos * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS
19*0e552da7Schristos * IN THE SOFTWARE.
20*0e552da7Schristos */
21*0e552da7Schristos
22*0e552da7Schristos #include "uv.h"
23*0e552da7Schristos #include "task.h"
24*0e552da7Schristos
25*0e552da7Schristos static uv_timer_t timer1_handle;
26*0e552da7Schristos static uv_timer_t timer2_handle;
27*0e552da7Schristos static uv_tcp_t tcp_handle;
28*0e552da7Schristos
29*0e552da7Schristos static int connect_cb_called;
30*0e552da7Schristos static int timer1_cb_called;
31*0e552da7Schristos static int close_cb_called;
32*0e552da7Schristos static int netunreach_errors;
33*0e552da7Schristos
34*0e552da7Schristos
close_cb(uv_handle_t * handle)35*0e552da7Schristos static void close_cb(uv_handle_t* handle) {
36*0e552da7Schristos close_cb_called++;
37*0e552da7Schristos }
38*0e552da7Schristos
39*0e552da7Schristos
connect_cb(uv_connect_t * req,int status)40*0e552da7Schristos static void connect_cb(uv_connect_t* req, int status) {
41*0e552da7Schristos /* The expected error is UV_ECANCELED but the test tries to connect to what
42*0e552da7Schristos * is basically an arbitrary address in the expectation that no network path
43*0e552da7Schristos * exists, so UV_ENETUNREACH is an equally plausible outcome.
44*0e552da7Schristos */
45*0e552da7Schristos ASSERT(status == UV_ECANCELED || status == UV_ENETUNREACH);
46*0e552da7Schristos uv_timer_stop(&timer2_handle);
47*0e552da7Schristos connect_cb_called++;
48*0e552da7Schristos if (status == UV_ENETUNREACH)
49*0e552da7Schristos netunreach_errors++;
50*0e552da7Schristos }
51*0e552da7Schristos
52*0e552da7Schristos
timer1_cb(uv_timer_t * handle)53*0e552da7Schristos static void timer1_cb(uv_timer_t* handle) {
54*0e552da7Schristos uv_close((uv_handle_t*)handle, close_cb);
55*0e552da7Schristos uv_close((uv_handle_t*)&tcp_handle, close_cb);
56*0e552da7Schristos timer1_cb_called++;
57*0e552da7Schristos }
58*0e552da7Schristos
59*0e552da7Schristos
timer2_cb(uv_timer_t * handle)60*0e552da7Schristos static void timer2_cb(uv_timer_t* handle) {
61*0e552da7Schristos ASSERT(0 && "should not be called");
62*0e552da7Schristos }
63*0e552da7Schristos
64*0e552da7Schristos
TEST_IMPL(tcp_close_while_connecting)65*0e552da7Schristos TEST_IMPL(tcp_close_while_connecting) {
66*0e552da7Schristos uv_connect_t connect_req;
67*0e552da7Schristos struct sockaddr_in addr;
68*0e552da7Schristos uv_loop_t* loop;
69*0e552da7Schristos int r;
70*0e552da7Schristos
71*0e552da7Schristos loop = uv_default_loop();
72*0e552da7Schristos ASSERT(0 == uv_ip4_addr("1.2.3.4", TEST_PORT, &addr));
73*0e552da7Schristos ASSERT(0 == uv_tcp_init(loop, &tcp_handle));
74*0e552da7Schristos r = uv_tcp_connect(&connect_req,
75*0e552da7Schristos &tcp_handle,
76*0e552da7Schristos (const struct sockaddr*) &addr,
77*0e552da7Schristos connect_cb);
78*0e552da7Schristos if (r == UV_ENETUNREACH)
79*0e552da7Schristos RETURN_SKIP("Network unreachable.");
80*0e552da7Schristos ASSERT(r == 0);
81*0e552da7Schristos ASSERT(0 == uv_timer_init(loop, &timer1_handle));
82*0e552da7Schristos ASSERT(0 == uv_timer_start(&timer1_handle, timer1_cb, 1, 0));
83*0e552da7Schristos ASSERT(0 == uv_timer_init(loop, &timer2_handle));
84*0e552da7Schristos ASSERT(0 == uv_timer_start(&timer2_handle, timer2_cb, 86400 * 1000, 0));
85*0e552da7Schristos ASSERT(0 == uv_run(loop, UV_RUN_DEFAULT));
86*0e552da7Schristos
87*0e552da7Schristos ASSERT(connect_cb_called == 1);
88*0e552da7Schristos ASSERT(timer1_cb_called == 1);
89*0e552da7Schristos ASSERT(close_cb_called == 2);
90*0e552da7Schristos
91*0e552da7Schristos MAKE_VALGRIND_HAPPY();
92*0e552da7Schristos
93*0e552da7Schristos if (netunreach_errors > 0)
94*0e552da7Schristos RETURN_SKIP("Network unreachable.");
95*0e552da7Schristos
96*0e552da7Schristos return 0;
97*0e552da7Schristos }
98