1 files changed, 113 insertions, 0 deletions
diff --git a/nsock/tests/connect.c b/nsock/tests/connect.c
new file mode 100644
index 0000000..736b082
--- /dev/null
+++ b/nsock/tests/connect.c
@@ -0,0 +1,113 @@
+/*
+ * 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
+};