From 2a7b9be01da20ce36f7c6fc7e0f3e00eca42ea0f Mon Sep 17 00:00:00 2001
From: Bert Belder <bertbelder@gmail.com>
Date: Thu, 14 Sep 2017 01:18:22 +0200
Subject: [PATCH] test: add test-ctl-fuzz

---
 test/test-ctl-fuzz.c | 94 ++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 94 insertions(+)
 create mode 100644 test/test-ctl-fuzz.c

diff --git a/test/test-ctl-fuzz.c b/test/test-ctl-fuzz.c
new file mode 100644
index 0000000..4a8cb56
--- /dev/null
+++ b/test/test-ctl-fuzz.c
@@ -0,0 +1,94 @@
+#include <assert.h>
+#include <stdbool.h>
+#include <stdint.h>
+#include <stdio.h>
+#include <stdlib.h>
+
+#include "epoll.h"
+#include "init.h"
+#include "util.h"
+#include "win.h"
+
+#include <WS2tcpip.h>
+
+#define NUM_SOCKETS 10000
+#define RUN_TIME 10000
+#define PRINT_INTERVAL 500
+
+static SOCKET create_and_add_socket(epoll_t epfd) {
+  SOCKET sock;
+  unsigned long one;
+  int r;
+  struct epoll_event ev;
+
+  sock = socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP);
+  assert(sock > 0);
+
+  one = 1;
+  r = ioctlsocket(sock, FIONBIO, &one);
+  assert(r == 0);
+
+  ev.events = 0;
+  ev.data.u64 = 42;
+  r = epoll_ctl(epfd, EPOLL_CTL_ADD, sock, &ev);
+  assert(r == 0);
+
+  return sock;
+}
+
+int main(void) {
+  uint64_t total_events = 0;
+  uint64_t start_time, last_print_time, now, total_time;
+  SOCKET sockets[NUM_SOCKETS];
+  int r;
+  epoll_t epfd;
+
+  r = init();
+  assert(r == 0);
+
+  epfd = epoll_create();
+  assert(epfd != NULL);
+
+  for (size_t i = 0; i < NUM_SOCKETS; i++)
+    sockets[i] = create_and_add_socket(epfd);
+
+  start_time = GetTickCount64();
+  last_print_time = 0;
+
+  do {
+    struct epoll_event ev_out[64];
+    uint64_t count;
+
+    for (size_t i = 0; i < NUM_SOCKETS; i++) {
+      SOCKET sock = sockets[i];
+      struct epoll_event ev_in;
+
+      ev_in.events = rand() & 0xff | EPOLLONESHOT;
+      ev_in.data.u64 = 42;
+
+      r = epoll_ctl(epfd, EPOLL_CTL_MOD, sock, &ev_in);
+      assert(r == 0);
+    }
+
+    count = 0;
+    do {
+      r = epoll_wait(epfd, ev_out, array_count(ev_out), count > 0 ? 0 : -1);
+      assert(r >= 0);
+      count += r;
+    } while (r > 0);
+
+    total_events += count;
+
+    now = GetTickCount64();
+    total_time = now - start_time;
+
+    if (now - last_print_time > PRINT_INTERVAL) {
+      last_print_time = now;
+      printf("%f events (%f events/sec)\n",
+             (double) total_events,
+             (double) total_events / total_time * 1000);
+    }
+  } while (total_time < RUN_TIME);
+
+  return 0;
+}