From a9e78ad9b541da8d34d4d55030c0c2c954666ae1 Mon Sep 17 00:00:00 2001
From: Bert Belder <bertbelder@gmail.com>
Date: Fri, 16 Nov 2018 23:12:12 -0800
Subject: [PATCH] test: add test for events reported after connect() fails

---
 test/test-connect-fail-events.c | 69 +++++++++++++++++++++++++++++++++
 1 file changed, 69 insertions(+)
 create mode 100644 test/test-connect-fail-events.c

diff --git a/test/test-connect-fail-events.c b/test/test-connect-fail-events.c
new file mode 100644
index 0000000..1c28183
--- /dev/null
+++ b/test/test-connect-fail-events.c
@@ -0,0 +1,69 @@
+#include <errno.h>
+#include <stdlib.h>
+
+#include "test-util.h"
+#include "util.h"
+#include "wepoll.h"
+#include "win.h"
+
+#define FAIL_PORT 1
+
+int main(void) {
+  unsigned long one = 1;
+  HANDLE ephnd;
+  SOCKET sock;
+  struct epoll_event ev;
+  struct sockaddr_in addr;
+  int err, len;
+  int r;
+
+  /* Create epoll port. */
+  ephnd = epoll_create1(0);
+  check(ephnd != NULL);
+
+  /* Create a stream socket, switch it to non-blocking mode. */
+  sock = socket(AF_INET, SOCK_STREAM, 0);
+  check(sock != INVALID_SOCKET);
+  r = ioctlsocket(sock, (long) FIONBIO, &one);
+  check(r == 0);
+
+  /* Associate the socket with the epoll port; set it up such that we'll pick
+   * up any valid event. */
+  ev.events = EPOLLIN | EPOLLPRI | EPOLLOUT | EPOLLERR | EPOLLHUP |
+              EPOLLRDNORM | EPOLLRDBAND | EPOLLWRNORM | EPOLLWRBAND |
+              EPOLLMSG | EPOLLRDHUP;
+  ev.data.sock = sock;
+  r = epoll_ctl(ephnd, EPOLL_CTL_ADD, sock, &ev);
+  check(r == 0);
+
+  /* Attempt to connect to an address that we won't be able to connect to. */
+  addr.sin_family = AF_INET;
+  addr.sin_addr.s_addr = htonl(INADDR_LOOPBACK);
+  addr.sin_port = htons(FAIL_PORT);
+  r = connect(sock, (struct sockaddr*) &addr, sizeof addr);
+  check(r == SOCKET_ERROR && WSAGetLastError() == WSAEWOULDBLOCK);
+
+  /* Wait for the socket to report an event, and verify that the events
+   * reported match exactly what the linux kernel would report in case of a
+   * connect() failure. */
+  r = epoll_wait(ephnd, &ev, 1, -1);
+  check(r == 1);
+  check(ev.data.sock == sock);
+  check(ev.events == (EPOLLIN | EPOLLOUT | EPOLLERR | EPOLLRDNORM |
+                      EPOLLWRNORM | EPOLLRDHUP));
+
+  /* Retrieve the error associated with the socket, and verify that connect()
+   * failed because the server refused the connection. */
+  len = sizeof err;
+  r = getsockopt(sock, SOL_SOCKET, SO_ERROR, (char*) &err, &len);
+  check(r == 0);
+  check(err == WSAECONNREFUSED);
+
+  /* Clean up. */
+  r = closesocket(sock);
+  check(r == 0);
+  r = epoll_close(ephnd);
+  check(r == 0);
+
+  return 0;
+}