2 * Event loop based on Windows events and WaitForMultipleObjects
3 * Copyright (c) 2002-2009, Jouni Malinen <j@w1.fi>
5 * This software may be distributed under the terms of the BSD license.
6 * See README for more details.
21 eloop_sock_handler handler;
28 eloop_event_handler handler;
32 struct eloop_timeout {
37 eloop_timeout_handler handler;
43 eloop_signal_handler handler;
50 struct eloop_sock *readers;
53 struct eloop_event *events;
55 struct dl_list timeout;
58 struct eloop_signal *signals;
60 int pending_terminate;
63 int reader_table_changed;
65 struct eloop_signal term_signal;
72 static struct eloop_data eloop;
77 os_memset(&eloop, 0, sizeof(eloop));
78 dl_list_init(&eloop.timeout);
79 eloop.num_handles = 1;
80 eloop.handles = os_malloc(eloop.num_handles *
81 sizeof(eloop.handles[0]));
82 if (eloop.handles == NULL)
85 eloop.term_event = CreateEvent(NULL, FALSE, FALSE, NULL);
86 if (eloop.term_event == NULL) {
87 printf("CreateEvent() failed: %d\n",
88 (int) GetLastError());
89 os_free(eloop.handles);
97 static int eloop_prepare_handles(void)
101 if (eloop.num_handles > eloop.reader_count + eloop.event_count + 8)
103 n = os_realloc_array(eloop.handles, eloop.num_handles * 2,
104 sizeof(eloop.handles[0]));
108 eloop.num_handles *= 2;
113 int eloop_register_read_sock(int sock, eloop_sock_handler handler,
114 void *eloop_data, void *user_data)
117 struct eloop_sock *tmp;
119 if (eloop_prepare_handles())
122 event = WSACreateEvent();
123 if (event == WSA_INVALID_EVENT) {
124 printf("WSACreateEvent() failed: %d\n", WSAGetLastError());
128 if (WSAEventSelect(sock, event, FD_READ)) {
129 printf("WSAEventSelect() failed: %d\n", WSAGetLastError());
130 WSACloseEvent(event);
133 tmp = os_realloc_array(eloop.readers, eloop.reader_count + 1,
134 sizeof(struct eloop_sock));
136 WSAEventSelect(sock, event, 0);
137 WSACloseEvent(event);
141 tmp[eloop.reader_count].sock = sock;
142 tmp[eloop.reader_count].eloop_data = eloop_data;
143 tmp[eloop.reader_count].user_data = user_data;
144 tmp[eloop.reader_count].handler = handler;
145 tmp[eloop.reader_count].event = event;
146 eloop.reader_count++;
148 if (sock > eloop.max_sock)
149 eloop.max_sock = sock;
150 eloop.reader_table_changed = 1;
156 void eloop_unregister_read_sock(int sock)
160 if (eloop.readers == NULL || eloop.reader_count == 0)
163 for (i = 0; i < eloop.reader_count; i++) {
164 if (eloop.readers[i].sock == sock)
167 if (i == eloop.reader_count)
170 WSAEventSelect(eloop.readers[i].sock, eloop.readers[i].event, 0);
171 WSACloseEvent(eloop.readers[i].event);
173 if (i != eloop.reader_count - 1) {
174 os_memmove(&eloop.readers[i], &eloop.readers[i + 1],
175 (eloop.reader_count - i - 1) *
176 sizeof(struct eloop_sock));
178 eloop.reader_count--;
179 eloop.reader_table_changed = 1;
183 int eloop_register_event(void *event, size_t event_size,
184 eloop_event_handler handler,
185 void *eloop_data, void *user_data)
187 struct eloop_event *tmp;
190 if (event_size != sizeof(HANDLE) || h == INVALID_HANDLE_VALUE)
193 if (eloop_prepare_handles())
196 tmp = os_realloc_array(eloop.events, eloop.event_count + 1,
197 sizeof(struct eloop_event));
201 tmp[eloop.event_count].eloop_data = eloop_data;
202 tmp[eloop.event_count].user_data = user_data;
203 tmp[eloop.event_count].handler = handler;
204 tmp[eloop.event_count].event = h;
212 void eloop_unregister_event(void *event, size_t event_size)
217 if (eloop.events == NULL || eloop.event_count == 0 ||
218 event_size != sizeof(HANDLE))
221 for (i = 0; i < eloop.event_count; i++) {
222 if (eloop.events[i].event == h)
225 if (i == eloop.event_count)
228 if (i != eloop.event_count - 1) {
229 os_memmove(&eloop.events[i], &eloop.events[i + 1],
230 (eloop.event_count - i - 1) *
231 sizeof(struct eloop_event));
237 int eloop_register_timeout(unsigned int secs, unsigned int usecs,
238 eloop_timeout_handler handler,
239 void *eloop_data, void *user_data)
241 struct eloop_timeout *timeout, *tmp;
244 timeout = os_zalloc(sizeof(*timeout));
247 if (os_get_time(&timeout->time) < 0) {
251 now_sec = timeout->time.sec;
252 timeout->time.sec += secs;
253 if (timeout->time.sec < now_sec) {
255 * Integer overflow - assume long enough timeout to be assumed
256 * to be infinite, i.e., the timeout would never happen.
258 wpa_printf(MSG_DEBUG, "ELOOP: Too long timeout (secs=%u) to "
259 "ever happen - ignore it", secs);
263 timeout->time.usec += usecs;
264 while (timeout->time.usec >= 1000000) {
266 timeout->time.usec -= 1000000;
268 timeout->eloop_data = eloop_data;
269 timeout->user_data = user_data;
270 timeout->handler = handler;
272 /* Maintain timeouts in order of increasing time */
273 dl_list_for_each(tmp, &eloop.timeout, struct eloop_timeout, list) {
274 if (os_time_before(&timeout->time, &tmp->time)) {
275 dl_list_add(tmp->list.prev, &timeout->list);
279 dl_list_add_tail(&eloop.timeout, &timeout->list);
285 static void eloop_remove_timeout(struct eloop_timeout *timeout)
287 dl_list_del(&timeout->list);
292 int eloop_cancel_timeout(eloop_timeout_handler handler,
293 void *eloop_data, void *user_data)
295 struct eloop_timeout *timeout, *prev;
298 dl_list_for_each_safe(timeout, prev, &eloop.timeout,
299 struct eloop_timeout, list) {
300 if (timeout->handler == handler &&
301 (timeout->eloop_data == eloop_data ||
302 eloop_data == ELOOP_ALL_CTX) &&
303 (timeout->user_data == user_data ||
304 user_data == ELOOP_ALL_CTX)) {
305 eloop_remove_timeout(timeout);
314 int eloop_is_timeout_registered(eloop_timeout_handler handler,
315 void *eloop_data, void *user_data)
317 struct eloop_timeout *tmp;
319 dl_list_for_each(tmp, &eloop.timeout, struct eloop_timeout, list) {
320 if (tmp->handler == handler &&
321 tmp->eloop_data == eloop_data &&
322 tmp->user_data == user_data)
330 /* TODO: replace with suitable signal handler */
332 static void eloop_handle_signal(int sig)
337 for (i = 0; i < eloop.signal_count; i++) {
338 if (eloop.signals[i].sig == sig) {
339 eloop.signals[i].signaled++;
347 static void eloop_process_pending_signals(void)
351 if (eloop.signaled == 0)
355 if (eloop.pending_terminate) {
356 eloop.pending_terminate = 0;
359 for (i = 0; i < eloop.signal_count; i++) {
360 if (eloop.signals[i].signaled) {
361 eloop.signals[i].signaled = 0;
362 eloop.signals[i].handler(eloop.signals[i].sig,
363 eloop.signals[i].user_data);
367 if (eloop.term_signal.signaled) {
368 eloop.term_signal.signaled = 0;
369 eloop.term_signal.handler(eloop.term_signal.sig,
370 eloop.term_signal.user_data);
375 int eloop_register_signal(int sig, eloop_signal_handler handler,
378 struct eloop_signal *tmp;
380 tmp = os_realloc_array(eloop.signals, eloop.signal_count + 1,
381 sizeof(struct eloop_signal));
385 tmp[eloop.signal_count].sig = sig;
386 tmp[eloop.signal_count].user_data = user_data;
387 tmp[eloop.signal_count].handler = handler;
388 tmp[eloop.signal_count].signaled = 0;
389 eloop.signal_count++;
392 /* TODO: register signal handler */
399 static BOOL eloop_handle_console_ctrl(DWORD type)
403 case CTRL_BREAK_EVENT:
405 eloop.term_signal.signaled++;
406 SetEvent(eloop.term_event);
412 #endif /* _WIN32_WCE */
415 int eloop_register_signal_terminate(eloop_signal_handler handler,
419 if (SetConsoleCtrlHandler((PHANDLER_ROUTINE) eloop_handle_console_ctrl,
421 printf("SetConsoleCtrlHandler() failed: %d\n",
422 (int) GetLastError());
425 #endif /* _WIN32_WCE */
427 eloop.term_signal.handler = handler;
428 eloop.term_signal.user_data = user_data;
434 int eloop_register_signal_reconfig(eloop_signal_handler handler,
444 struct os_time tv, now;
445 DWORD count, ret, timeout_val, err;
448 while (!eloop.terminate &&
449 (!dl_list_empty(&eloop.timeout) || eloop.reader_count > 0 ||
450 eloop.event_count > 0)) {
451 struct eloop_timeout *timeout;
452 tv.sec = tv.usec = 0;
453 timeout = dl_list_first(&eloop.timeout, struct eloop_timeout,
457 if (os_time_before(&now, &timeout->time))
458 os_time_sub(&timeout->time, &now, &tv);
462 for (i = 0; i < eloop.event_count; i++)
463 eloop.handles[count++] = eloop.events[i].event;
465 for (i = 0; i < eloop.reader_count; i++)
466 eloop.handles[count++] = eloop.readers[i].event;
468 if (eloop.term_event)
469 eloop.handles[count++] = eloop.term_event;
472 timeout_val = tv.sec * 1000 + tv.usec / 1000;
474 timeout_val = INFINITE;
476 if (count > MAXIMUM_WAIT_OBJECTS) {
477 printf("WaitForMultipleObjects: Too many events: "
478 "%d > %d (ignoring extra events)\n",
479 (int) count, MAXIMUM_WAIT_OBJECTS);
480 count = MAXIMUM_WAIT_OBJECTS;
483 ret = WaitForMultipleObjects(count, eloop.handles, FALSE,
485 #else /* _WIN32_WCE */
486 ret = WaitForMultipleObjectsEx(count, eloop.handles, FALSE,
488 #endif /* _WIN32_WCE */
489 err = GetLastError();
491 eloop_process_pending_signals();
493 /* check if some registered timeouts have occurred */
494 timeout = dl_list_first(&eloop.timeout, struct eloop_timeout,
498 if (!os_time_before(&now, &timeout->time)) {
499 void *eloop_data = timeout->eloop_data;
500 void *user_data = timeout->user_data;
501 eloop_timeout_handler handler =
503 eloop_remove_timeout(timeout);
504 handler(eloop_data, user_data);
509 if (ret == WAIT_FAILED) {
510 printf("WaitForMultipleObjects(count=%d) failed: %d\n",
511 (int) count, (int) err);
517 if (ret == WAIT_IO_COMPLETION)
519 #endif /* _WIN32_WCE */
521 if (ret == WAIT_TIMEOUT)
524 while (ret >= WAIT_OBJECT_0 &&
525 ret < WAIT_OBJECT_0 + eloop.event_count) {
526 eloop.events[ret].handler(
527 eloop.events[ret].eloop_data,
528 eloop.events[ret].user_data);
529 ret = WaitForMultipleObjects(eloop.event_count,
530 eloop.handles, FALSE, 0);
533 eloop.reader_table_changed = 0;
534 for (i = 0; i < eloop.reader_count; i++) {
535 WSANETWORKEVENTS events;
536 if (WSAEnumNetworkEvents(eloop.readers[i].sock,
537 eloop.readers[i].event,
539 (events.lNetworkEvents & FD_READ)) {
540 eloop.readers[i].handler(
541 eloop.readers[i].sock,
542 eloop.readers[i].eloop_data,
543 eloop.readers[i].user_data);
544 if (eloop.reader_table_changed)
552 void eloop_terminate(void)
555 SetEvent(eloop.term_event);
559 void eloop_destroy(void)
561 struct eloop_timeout *timeout, *prev;
563 dl_list_for_each_safe(timeout, prev, &eloop.timeout,
564 struct eloop_timeout, list) {
565 eloop_remove_timeout(timeout);
567 os_free(eloop.readers);
568 os_free(eloop.signals);
569 if (eloop.term_event)
570 CloseHandle(eloop.term_event);
571 os_free(eloop.handles);
572 eloop.handles = NULL;
573 os_free(eloop.events);
578 int eloop_terminated(void)
580 return eloop.terminate;
584 void eloop_wait_for_read_sock(int sock)
588 event = WSACreateEvent();
589 if (event == WSA_INVALID_EVENT) {
590 printf("WSACreateEvent() failed: %d\n", WSAGetLastError());
594 if (WSAEventSelect(sock, event, FD_READ)) {
595 printf("WSAEventSelect() failed: %d\n", WSAGetLastError());
596 WSACloseEvent(event);
600 WaitForSingleObject(event, INFINITE);
601 WSAEventSelect(sock, event, 0);
602 WSACloseEvent(event);