/* 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 close_cb_called = 0;

static void close_cb(uv_handle_t* handle) {

  ASSERT(handle != NULL);

  close_cb_called++;

}

TEST_IMPL(tcp_bind_error_addrinuse) {

  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, NULL, NULL);

  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;

}