1 files changed, 134 insertions, 0 deletions
diff --git a/nsock/tests/cancel.c b/nsock/tests/cancel.c
new file mode 100644
index 0000000..49f98bf
--- /dev/null
+++ b/nsock/tests/cancel.c
@@ -0,0 +1,134 @@
+/*
+ * Nsock regression test suite
+ * Same license as nmap -- see https://nmap.org/book/man-legal.html
+ */
+
+#include "test-common.h"
+
+
+struct basic_test_data {
+  nsock_pool nsp;
+};
+
+
+static void cancel_handler(nsock_pool nsp, nsock_event nse, void *udata) {
+  int *ev_done = (int *)udata;
+
+  if (nse_status(nse) == NSE_STATUS_CANCELLED)
+    *ev_done = 1;
+}
+
+static int cancel_setup(void **tdata) {
+  struct basic_test_data *btd;
+
+  btd = calloc(1, sizeof(struct basic_test_data));
+  if (btd == NULL)
+    return -ENOMEM;
+
+  btd->nsp = nsock_pool_new(NULL);
+
+  *tdata = btd;
+  return 0;
+}
+
+static int cancel_teardown(void *tdata) {
+  struct basic_test_data *btd = (struct basic_test_data *)tdata;
+
+  if (tdata) {
+    nsock_pool_delete(btd->nsp);
+    free(tdata);
+  }
+  return 0;
+}
+
+static int cancel_tcp_run(void *tdata) {
+  struct basic_test_data *btd = (struct basic_test_data *)tdata;
+  struct sockaddr_in peer;
+  nsock_iod iod;
+  nsock_event_id id;
+  int done = 0;
+
+  iod = nsock_iod_new(btd->nsp, NULL);
+  AssertNonNull(iod);
+
+  memset(&peer, 0, sizeof(peer));
+  peer.sin_family = AF_INET;
+  inet_aton("127.0.0.1", &peer.sin_addr);
+
+  id = nsock_connect_tcp(btd->nsp, iod, cancel_handler, 4000, (void *)&done,
+                         (struct sockaddr *)&peer, sizeof(peer), PORT_TCP);
+  nsock_event_cancel(btd->nsp, id, 1);
+
+  nsock_iod_delete(iod, NSOCK_PENDING_SILENT);
+
+  return (done == 1) ? 0 : -ENOEXEC;
+}
+
+static int cancel_udp_run(void *tdata) {
+  struct basic_test_data *btd = (struct basic_test_data *)tdata;
+  struct sockaddr_in peer;
+  nsock_iod iod;
+  nsock_event_id id;
+  int done = 0;
+
+  iod = nsock_iod_new(btd->nsp, NULL);
+  AssertNonNull(iod);
+
+  memset(&peer, 0, sizeof(peer));
+  peer.sin_family = AF_INET;
+  inet_aton("127.0.0.1", &peer.sin_addr);
+
+  id = nsock_connect_udp(btd->nsp, iod, cancel_handler, (void *)&done,
+                         (struct sockaddr *)&peer, sizeof(peer), PORT_UDP);
+  nsock_event_cancel(btd->nsp, id, 1);
+
+  nsock_iod_delete(iod, NSOCK_PENDING_SILENT);
+
+  return (done == 1) ? 0 : -ENOEXEC;
+}
+
+static int cancel_ssl_run(void *tdata) {
+  struct basic_test_data *btd = (struct basic_test_data *)tdata;
+  struct sockaddr_in peer;
+  nsock_iod iod;
+  nsock_event_id id;
+  int done = 0;
+
+  iod = nsock_iod_new(btd->nsp, NULL);
+  AssertNonNull(iod);
+
+  memset(&peer, 0, sizeof(peer));
+  peer.sin_family = AF_INET;
+  inet_aton("127.0.0.1", &peer.sin_addr);
+
+  id = nsock_connect_ssl(btd->nsp, iod, cancel_handler, 4000, (void *)&done,
+                         (struct sockaddr *)&peer, sizeof(peer), IPPROTO_TCP,
+                         PORT_TCPSSL, NULL);
+  nsock_event_cancel(btd->nsp, id, 1);
+
+  nsock_iod_delete(iod, NSOCK_PENDING_SILENT);
+
+  return (done == 1) ? 0 : -ENOEXEC;
+}
+
+
+const struct test_case TestCancelTCP = {
+  .t_name     = "schedule and cancel TCP connect",
+  .t_setup    = cancel_setup,
+  .t_run      = cancel_tcp_run,
+  .t_teardown = cancel_teardown
+};
+
+const struct test_case TestCancelUDP = {
+  .t_name     = "schedule and cancel UDP pseudo-connect",
+  .t_setup    = cancel_setup,
+  .t_run      = cancel_udp_run,
+  .t_teardown = cancel_teardown
+};
+
+const struct test_case TestCancelSSL = {
+  .t_name     = "schedule and cancel SSL connect",
+  .t_setup    = cancel_setup,
+  .t_run      = cancel_ssl_run,
+  .t_teardown = cancel_teardown
+};