95 lines
1.9 KiB
C
95 lines
1.9 KiB
C
#include <assert.h>
|
|
#include <stdbool.h>
|
|
#include <stdint.h>
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
|
|
#include "init.h"
|
|
#include "util.h"
|
|
#include "wepoll.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(HANDLE 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;
|
|
HANDLE epfd;
|
|
|
|
r = init();
|
|
assert(r == 0);
|
|
|
|
epfd = epoll_create1(0);
|
|
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;
|
|
}
|