125 lines
2.5 KiB
C
125 lines
2.5 KiB
C
#include <assert.h>
|
|
#include <process.h>
|
|
#include <stdint.h>
|
|
#include <stdio.h>
|
|
|
|
#include "init.h"
|
|
#include "util.h"
|
|
#include "wepoll.h"
|
|
#include "win.h"
|
|
|
|
#include <WS2tcpip.h>
|
|
|
|
#define POLL_THREAD_COUNT 20
|
|
#define LISTEN_PORT 12345
|
|
|
|
static SOCKET create_socket(unsigned short port) {
|
|
SOCKET sock;
|
|
struct sockaddr_in address;
|
|
unsigned long one;
|
|
int r;
|
|
|
|
sock = socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP);
|
|
assert(sock > 0);
|
|
|
|
address.sin_family = AF_INET;
|
|
address.sin_port = htons(port);
|
|
address.sin_addr.s_addr = htonl(INADDR_LOOPBACK);
|
|
r = bind(sock, (struct sockaddr*) &address, sizeof address);
|
|
assert(r == 0);
|
|
|
|
one = 1;
|
|
r = ioctlsocket(sock, FIONBIO, &one);
|
|
assert(r == 0);
|
|
|
|
return sock;
|
|
}
|
|
|
|
static void send_message(SOCKET sock, unsigned short port) {
|
|
char hello[] = "hello";
|
|
struct sockaddr_in address;
|
|
WSABUF buf;
|
|
DWORD bytes_sent;
|
|
int r;
|
|
|
|
address.sin_family = AF_INET;
|
|
address.sin_port = htons(port);
|
|
address.sin_addr.s_addr = htonl(INADDR_LOOPBACK);
|
|
|
|
buf.buf = hello;
|
|
buf.len = sizeof(hello);
|
|
|
|
r = WSASendTo(sock,
|
|
&buf,
|
|
1,
|
|
&bytes_sent,
|
|
0,
|
|
(struct sockaddr*) &address,
|
|
sizeof address,
|
|
NULL,
|
|
NULL);
|
|
assert(r >= 0);
|
|
}
|
|
|
|
static unsigned int __stdcall poll_thread(void* arg) {
|
|
HANDLE epfd;
|
|
SOCKET sock;
|
|
struct epoll_event ev_in;
|
|
struct epoll_event ev_out;
|
|
int r;
|
|
|
|
sock = (SOCKET) arg;
|
|
|
|
epfd = epoll_create1(0);
|
|
assert(epfd != NULL);
|
|
|
|
ev_in.events = EPOLLIN;
|
|
ev_in.data.u64 = 42;
|
|
r = epoll_ctl(epfd, EPOLL_CTL_ADD, sock, &ev_in);
|
|
assert(r == 0);
|
|
|
|
memset(&ev_out, 0, sizeof ev_out);
|
|
r = epoll_wait(epfd, &ev_out, 1, -1);
|
|
assert(r == 1);
|
|
|
|
assert(ev_out.data.u64 == ev_in.data.u64);
|
|
assert(ev_out.events == ev_in.events);
|
|
|
|
puts("Got event.");
|
|
|
|
r = epoll_close(epfd);
|
|
assert(r == 0);
|
|
|
|
return 0;
|
|
}
|
|
|
|
int main(void) {
|
|
HANDLE threads[POLL_THREAD_COUNT];
|
|
|
|
int r = init();
|
|
assert(r == 0);
|
|
|
|
SOCKET send_sock = create_socket(0);
|
|
SOCKET recv_sock = create_socket(LISTEN_PORT);
|
|
|
|
for (size_t i = 0; i < array_count(threads); i++) {
|
|
HANDLE thread = (HANDLE) _beginthreadex(
|
|
NULL, 0, poll_thread, (void*) recv_sock, 0, NULL);
|
|
assert(thread != INVALID_HANDLE_VALUE);
|
|
threads[i] = thread;
|
|
}
|
|
|
|
Sleep(500);
|
|
|
|
send_message(send_sock, LISTEN_PORT);
|
|
|
|
for (size_t i = 0; i < array_count(threads); i++) {
|
|
HANDLE thread = threads[i];
|
|
DWORD wr = WaitForSingleObject(thread, INFINITE);
|
|
assert(wr == WAIT_OBJECT_0);
|
|
CloseHandle(thread);
|
|
}
|
|
|
|
return 0;
|
|
}
|