test-tcp-connect-timeout.c 2.6 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889
  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. #include "uv.h"
  22. #include "task.h"
  23. #include <stdio.h>
  24. #include <stdlib.h>
  25. static int connect_cb_called;
  26. static int close_cb_called;
  27. static uv_connect_t connect_req;
  28. static uv_timer_t timer;
  29. static uv_tcp_t conn;
  30. static void connect_cb(uv_connect_t* req, int status);
  31. static void timer_cb(uv_timer_t* handle, int status);
  32. static void close_cb(uv_handle_t* handle);
  33. static void connect_cb(uv_connect_t* req, int status) {
  34. ASSERT(req == &connect_req);
  35. ASSERT(status == UV_ECANCELED);
  36. connect_cb_called++;
  37. }
  38. static void timer_cb(uv_timer_t* handle, int status) {
  39. ASSERT(handle == &timer);
  40. uv_close((uv_handle_t*)&conn, close_cb);
  41. uv_close((uv_handle_t*)&timer, close_cb);
  42. }
  43. static void close_cb(uv_handle_t* handle) {
  44. ASSERT(handle == (uv_handle_t*)&conn || handle == (uv_handle_t*)&timer);
  45. close_cb_called++;
  46. }
  47. /* Verify that connecting to an unreachable address or port doesn't hang
  48. * the event loop.
  49. */
  50. TEST_IMPL(tcp_connect_timeout) {
  51. struct sockaddr_in addr;
  52. int r;
  53. ASSERT(0 == uv_ip4_addr("8.8.8.8", 9999, &addr));
  54. r = uv_timer_init(uv_default_loop(), &timer);
  55. ASSERT(r == 0);
  56. r = uv_timer_start(&timer, timer_cb, 50, 0);
  57. ASSERT(r == 0);
  58. r = uv_tcp_init(uv_default_loop(), &conn);
  59. ASSERT(r == 0);
  60. r = uv_tcp_connect(&connect_req,
  61. &conn,
  62. (const struct sockaddr*) &addr,
  63. connect_cb);
  64. ASSERT(r == 0);
  65. r = uv_run(uv_default_loop(), UV_RUN_DEFAULT);
  66. ASSERT(r == 0);
  67. MAKE_VALGRIND_HAPPY();
  68. return 0;
  69. }