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 static uv_timer_t timer1_handle; 26 static uv_timer_t timer2_handle; 27 static uv_tcp_t tcp_handle; 28 29 static int connect_cb_called; 30 static int timer1_cb_called; 31 static int close_cb_called; 32 static int netunreach_errors; 33 34 35 static void close_cb(uv_handle_t* handle) { 36 close_cb_called++; 37 } 38 39 40 static void connect_cb(uv_connect_t* req, int status) { 41 /* The expected error is UV_ECANCELED but the test tries to connect to what 42 * is basically an arbitrary address in the expectation that no network path 43 * exists, so UV_ENETUNREACH is an equally plausible outcome. 44 */ 45 ASSERT(status == UV_ECANCELED || status == UV_ENETUNREACH); 46 uv_timer_stop(&timer2_handle); 47 connect_cb_called++; 48 if (status == UV_ENETUNREACH) 49 netunreach_errors++; 50 } 51 52 53 static void timer1_cb(uv_timer_t* handle) { 54 uv_close((uv_handle_t*)handle, close_cb); 55 uv_close((uv_handle_t*)&tcp_handle, close_cb); 56 timer1_cb_called++; 57 } 58 59 60 static void timer2_cb(uv_timer_t* handle) { 61 ASSERT(0 && "should not be called"); 62 } 63 64 65 TEST_IMPL(tcp_close_while_connecting) { 66 uv_connect_t connect_req; 67 struct sockaddr_in addr; 68 uv_loop_t* loop; 69 int r; 70 71 loop = uv_default_loop(); 72 ASSERT(0 == uv_ip4_addr("1.2.3.4", TEST_PORT, &addr)); 73 ASSERT(0 == uv_tcp_init(loop, &tcp_handle)); 74 r = uv_tcp_connect(&connect_req, 75 &tcp_handle, 76 (const struct sockaddr*) &addr, 77 connect_cb); 78 if (r == UV_ENETUNREACH) 79 RETURN_SKIP("Network unreachable."); 80 ASSERT(r == 0); 81 ASSERT(0 == uv_timer_init(loop, &timer1_handle)); 82 ASSERT(0 == uv_timer_start(&timer1_handle, timer1_cb, 1, 0)); 83 ASSERT(0 == uv_timer_init(loop, &timer2_handle)); 84 ASSERT(0 == uv_timer_start(&timer2_handle, timer2_cb, 86400 * 1000, 0)); 85 ASSERT(0 == uv_run(loop, UV_RUN_DEFAULT)); 86 87 ASSERT(connect_cb_called == 1); 88 ASSERT(timer1_cb_called == 1); 89 ASSERT(close_cb_called == 2); 90 91 MAKE_VALGRIND_HAPPY(); 92 93 if (netunreach_errors > 0) 94 RETURN_SKIP("Network unreachable."); 95 96 return 0; 97 } 98