text: ﻿/* Copyright Joyent, Inc. and other Node contributors. All rights reserved.
   *
   * Permission is hereby granted, free of charge, to any person obtaining a copy
   * of this software and associated documentation files (the "Software"), to
   * deal in the Software without restriction, including without limitation the
   * rights to use, copy, modify, merge, publish, distribute, sublicense, and/or
   * sell copies of the Software, and to permit persons to whom the Software is
   * furnished to do so, subject to the following conditions:
   *
   * The above copyright notice and this permission notice shall be included in
   * all copies or substantial portions of the Software.
   *
   * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
   * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
   * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
   * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
   * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
   * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS
   * IN THE SOFTWARE.
   */
  
  #include "uv.h"
  #include "task.h"
  #include <stdio.h>
  #include <stdlib.h>
  
  
  static int connect_cb_called = 0;
  static int close_cb_called = 0;
  
  
  static void close_cb(uv_handle_t* handle) {
    ASSERT_NOT_NULL(handle);
    close_cb_called++;
  }
  
  
  static void connect_cb(uv_connect_t* req, int status) {
    ASSERT(status == UV_EADDRINUSE);
    uv_close((uv_handle_t*) req->handle, close_cb);
    connect_cb_called++;
  }
  
  
  TEST_IMPL(tcp_bind_error_addrinuse_connect) {
    struct sockaddr_in addr;
    int addrlen;
    uv_connect_t req;
    uv_tcp_t conn;
  
    /* 127.0.0.1:<TEST_PORT> is already taken by tcp4_echo_server running in
     * another process. uv_tcp_bind() and uv_tcp_connect() should still succeed
     * (greatest common denominator across platforms) but the connect callback
     * should receive an UV_EADDRINUSE error.
     */
    ASSERT(0 == uv_tcp_init(uv_default_loop(), &conn));
    ASSERT(0 == uv_ip4_addr("127.0.0.1", TEST_PORT, &addr));
    ASSERT(0 == uv_tcp_bind(&conn, (const struct sockaddr*) &addr, 0));
  
    ASSERT(0 == uv_ip4_addr("127.0.0.1", TEST_PORT + 1, &addr));
    ASSERT(0 == uv_tcp_connect(&req,
                               &conn,
                               (const struct sockaddr*) &addr,
                               connect_cb));
  
    addrlen = sizeof(addr);
    ASSERT(UV_EADDRINUSE == uv_tcp_getsockname(&conn,
                                               (struct sockaddr*) &addr,
                                               &addrlen));
  
    ASSERT(0 == uv_run(uv_default_loop(), UV_RUN_DEFAULT));
    ASSERT(connect_cb_called == 1);
    ASSERT(close_cb_called == 1);
  
    MAKE_VALGRIND_HAPPY();
    return 0;
  }
  
  
  TEST_IMPL(tcp_bind_error_addrinuse_listen) {
    struct sockaddr_in addr;
    uv_tcp_t server1, server2;
    int r;
  
    ASSERT(0 == uv_ip4_addr("0.0.0.0", TEST_PORT, &addr));
    r = uv_tcp_init(uv_default_loop(), &server1);
    ASSERT(r == 0);
    r = uv_tcp_bind(&server1, (const struct sockaddr*) &addr, 0);
    ASSERT(r == 0);
  
    r = uv_tcp_init(uv_default_loop(), &server2);
    ASSERT(r == 0);
    r = uv_tcp_bind(&server2, (const struct sockaddr*) &addr, 0);
    ASSERT(r == 0);
  
    r = uv_listen((uv_stream_t*)&server1, 128, NULL);
    ASSERT(r == 0);
    r = uv_listen((uv_stream_t*)&server2, 128, NULL);
    ASSERT(r == UV_EADDRINUSE);
  
    uv_close((uv_handle_t*)&server1, close_cb);
    uv_close((uv_handle_t*)&server2, close_cb);
  
    uv_run(uv_default_loop(), UV_RUN_DEFAULT);
  
    ASSERT(close_cb_called == 2);
  
    MAKE_VALGRIND_HAPPY();
    return 0;
  }
  
  
  TEST_IMPL(tcp_bind_error_addrnotavail_1) {
    struct sockaddr_in addr;
    uv_tcp_t server;
    int r;
  
    ASSERT(0 == uv_ip4_addr("127.255.255.255", TEST_PORT, &addr));
  
    r = uv_tcp_init(uv_default_loop(), &server);
    ASSERT(r == 0);
  
    /* It seems that Linux is broken here - bind succeeds. */
    r = uv_tcp_bind(&server, (const struct sockaddr*) &addr, 0);
    ASSERT(r == 0 || r == UV_EADDRNOTAVAIL);
  
    uv_close((uv_handle_t*)&server, close_cb);
  
    uv_run(uv_default_loop(), UV_RUN_DEFAULT);
  
    ASSERT(close_cb_called == 1);
  
    MAKE_VALGRIND_HAPPY();
    return 0;
  }
  
  
  TEST_IMPL(tcp_bind_error_addrnotavail_2) {
    struct sockaddr_in addr;
    uv_tcp_t server;
    int r;
  
    ASSERT(0 == uv_ip4_addr("4.4.4.4", TEST_PORT, &addr));
  
    r = uv_tcp_init(uv_default_loop(), &server);
    ASSERT(r == 0);
    r = uv_tcp_bind(&server, (const struct sockaddr*) &addr, 0);
    ASSERT(r == UV_EADDRNOTAVAIL);
  
    uv_close((uv_handle_t*)&server, close_cb);
  
    uv_run(uv_default_loop(), UV_RUN_DEFAULT);
  
    ASSERT(close_cb_called == 1);
  
    MAKE_VALGRIND_HAPPY();
    return 0;
  }
  
  
  TEST_IMPL(tcp_bind_error_fault) {
    char garbage[] =
        "blah blah blah blah blah blah blah blah blah blah blah blah";
    struct sockaddr_in* garbage_addr;
    uv_tcp_t server;
    int r;
  
    garbage_addr = (struct sockaddr_in*) &garbage;
  
    r = uv_tcp_init(uv_default_loop(), &server);
    ASSERT(r == 0);
    r = uv_tcp_bind(&server, (const struct sockaddr*) garbage_addr, 0);
    ASSERT(r == UV_EINVAL);
  
    uv_close((uv_handle_t*)&server, close_cb);
  
    uv_run(uv_default_loop(), UV_RUN_DEFAULT);
  
    ASSERT(close_cb_called == 1);
  
    MAKE_VALGRIND_HAPPY();
    return 0;
  }
  
  /* Notes: On Linux uv_bind(server, NULL) will segfault the program.  */
  
  TEST_IMPL(tcp_bind_error_inval) {
    struct sockaddr_in addr1;
    struct sockaddr_in addr2;
    uv_tcp_t server;
    int r;
  
    ASSERT(0 == uv_ip4_addr("0.0.0.0", TEST_PORT, &addr1));
    ASSERT(0 == uv_ip4_addr("0.0.0.0", TEST_PORT_2, &addr2));
  
    r = uv_tcp_init(uv_default_loop(), &server);
    ASSERT(r == 0);
    r = uv_tcp_bind(&server, (const struct sockaddr*) &addr1, 0);
    ASSERT(r == 0);
    r = uv_tcp_bind(&server, (const struct sockaddr*) &addr2, 0);
    ASSERT(r == UV_EINVAL);
  
    uv_close((uv_handle_t*)&server, close_cb);
  
    uv_run(uv_default_loop(), UV_RUN_DEFAULT);
  
    ASSERT(close_cb_called == 1);
  
    MAKE_VALGRIND_HAPPY();
    return 0;
  }
  
  
  TEST_IMPL(tcp_bind_localhost_ok) {
    struct sockaddr_in addr;
    uv_tcp_t server;
    int r;
  
    ASSERT(0 == uv_ip4_addr("127.0.0.1", TEST_PORT, &addr));
  
    r = uv_tcp_init(uv_default_loop(), &server);
    ASSERT(r == 0);
    r = uv_tcp_bind(&server, (const struct sockaddr*) &addr, 0);
    ASSERT(r == 0);
  
    MAKE_VALGRIND_HAPPY();
    return 0;
  }
  
  
  TEST_IMPL(tcp_bind_invalid_flags) {
    struct sockaddr_in addr;
    uv_tcp_t server;
    int r;
  
    ASSERT(0 == uv_ip4_addr("127.0.0.1", TEST_PORT, &addr));
  
    r = uv_tcp_init(uv_default_loop(), &server);
    ASSERT(r == 0);
    r = uv_tcp_bind(&server, (const struct sockaddr*) &addr, UV_TCP_IPV6ONLY);
    ASSERT(r == UV_EINVAL);
  
    MAKE_VALGRIND_HAPPY();
    return 0;
  }
  
  
  TEST_IMPL(tcp_listen_without_bind) {
    int r;
    uv_tcp_t server;
  
    r = uv_tcp_init(uv_default_loop(), &server);
    ASSERT(r == 0);
    r = uv_listen((uv_stream_t*)&server, 128, NULL);
    ASSERT(r == 0);
  
    MAKE_VALGRIND_HAPPY();
    return 0;
  }
  
  
  TEST_IMPL(tcp_bind_writable_flags) {
    struct sockaddr_in addr;
    uv_tcp_t server;
    uv_buf_t buf;
    uv_write_t write_req;
    uv_shutdown_t shutdown_req;
    int r;
  
    ASSERT(0 == uv_ip4_addr("0.0.0.0", TEST_PORT, &addr));
    r = uv_tcp_init(uv_default_loop(), &server);
    ASSERT(r == 0);
    r = uv_tcp_bind(&server, (const struct sockaddr*) &addr, 0);
    ASSERT(r == 0);
    r = uv_listen((uv_stream_t*)&server, 128, NULL);
    ASSERT(r == 0);
  
    ASSERT(0 == uv_is_writable((uv_stream_t*) &server));
    ASSERT(0 == uv_is_readable((uv_stream_t*) &server));
  
    buf = uv_buf_init("PING", 4);
    r = uv_write(&write_req, (uv_stream_t*) &server, &buf, 1, NULL);
    ASSERT(r == UV_EPIPE);
    r = uv_shutdown(&shutdown_req, (uv_stream_t*) &server, NULL);
    ASSERT(r == UV_ENOTCONN);
    r = uv_read_start((uv_stream_t*) &server,
                      (uv_alloc_cb) abort,
                      (uv_read_cb) abort);
    ASSERT(r == UV_ENOTCONN);
  
    uv_close((uv_handle_t*)&server, close_cb);
  
    uv_run(uv_default_loop(), UV_RUN_DEFAULT);
  
    ASSERT(close_cb_called == 1);
  
    MAKE_VALGRIND_HAPPY();
    return 0;
  }
  