/*
 * Nsock regression test suite
 * Same license as nmap -- see https://nmap.org/book/man-legal.html
 */

#include "test-common.h"


struct connect_test_data {
  nsock_pool nsp;
  nsock_iod  nsi;
  int connect_result;
};


static void connect_handler(nsock_pool nsp, nsock_event nse, void *udata) {
  struct connect_test_data *ctd;

  ctd = (struct connect_test_data *)nsock_pool_get_udata(nsp);

  switch(nse_status(nse)) {
    case NSE_STATUS_SUCCESS:
      ctd->connect_result = 0;
      break;

    case NSE_STATUS_ERROR:
      ctd->connect_result = -(nse_errorcode(nse));
      break;

    case NSE_STATUS_TIMEOUT:
      ctd->connect_result = -ETIMEDOUT;
      break;

    default:
      ctd->connect_result = -EINVAL;
      break;
  }
}

static int connect_setup(void **tdata) {
  struct connect_test_data *ctd;

  ctd = calloc(1, sizeof(struct connect_test_data));
  if (ctd == NULL)
    return -ENOMEM;

  ctd->nsp = nsock_pool_new(ctd);
  AssertNonNull(ctd->nsp);

  ctd->nsi = nsock_iod_new(ctd->nsp, NULL);
  AssertNonNull(ctd->nsi);

  *tdata = ctd;
  return 0;
}

static int connect_teardown(void *tdata) {
  struct connect_test_data *ctd = (struct connect_test_data *)tdata;

  if (tdata) {
    nsock_iod_delete(ctd->nsi, NSOCK_PENDING_SILENT); /* nsock_pool_delete would also handle it */
    nsock_pool_delete(ctd->nsp);
    free(tdata);
  }
  return 0;
}

static int connect_tcp(void *tdata) {
  struct connect_test_data *ctd = (struct connect_test_data *)tdata;
  struct sockaddr_in peer;

  memset(&peer, 0, sizeof(peer));
  peer.sin_family = AF_INET;
  inet_aton("127.0.0.1", &peer.sin_addr);

  nsock_connect_tcp(ctd->nsp, ctd->nsi, connect_handler, 4000, NULL,
                    (struct sockaddr *)&peer, sizeof(peer), PORT_TCP);

  nsock_loop(ctd->nsp, 4000);
  return ctd->connect_result;
}

static int connect_tcp_failure(void *tdata) {
  struct connect_test_data *ctd = (struct connect_test_data *)tdata;
  struct sockaddr_in peer;

  memset(&peer, 0, sizeof(peer));
  peer.sin_family = AF_INET;
  inet_aton("127.0.0.1", &peer.sin_addr);

  /* pass in addrlen == 0 to force connect(2) to fail */
  nsock_connect_tcp(ctd->nsp, ctd->nsi, connect_handler, 4000, NULL,
                    (struct sockaddr *)&peer, 0, PORT_TCP);

  nsock_loop(ctd->nsp, 4000);
  AssertEqual(ctd->connect_result, -EINVAL);
  return 0;
}


const struct test_case TestConnectTCP = {
  .t_name     = "simple tcp connection",
  .t_setup    = connect_setup,
  .t_run      = connect_tcp,
  .t_teardown = connect_teardown
};

const struct test_case TestConnectFailure = {
  .t_name     = "tcp connection failure case",
  .t_setup    = connect_setup,
  .t_run      = connect_tcp_failure,
  .t_teardown = connect_teardown
};