2 * Event loop based on Windows events and WaitForMultipleObjects
3 * Copyright (c) 2002-2006, Jouni Malinen <j@w1.fi>
5 * This software may be distributed under the terms of the BSD license.
6 * See README for more details.
20 eloop_sock_handler handler;
27 eloop_event_handler handler;
31 struct eloop_timeout {
35 eloop_timeout_handler handler;
36 struct eloop_timeout *next;
42 eloop_signal_handler handler;
49 struct eloop_sock *readers;
52 struct eloop_event *events;
54 struct eloop_timeout *timeout;
57 struct eloop_signal *signals;
59 int pending_terminate;
62 int reader_table_changed;
64 struct eloop_signal term_signal;
71 static struct eloop_data eloop;
76 os_memset(&eloop, 0, sizeof(eloop));
77 eloop.num_handles = 1;
78 eloop.handles = os_malloc(eloop.num_handles *
79 sizeof(eloop.handles[0]));
80 if (eloop.handles == NULL)
83 eloop.term_event = CreateEvent(NULL, FALSE, FALSE, NULL);
84 if (eloop.term_event == NULL) {
85 printf("CreateEvent() failed: %d\n",
86 (int) GetLastError());
87 os_free(eloop.handles);
95 static int eloop_prepare_handles(void)
99 if (eloop.num_handles > eloop.reader_count + eloop.event_count + 8)
101 n = os_realloc(eloop.handles,
102 eloop.num_handles * 2 * sizeof(eloop.handles[0]));
106 eloop.num_handles *= 2;
111 int eloop_register_read_sock(int sock, eloop_sock_handler handler,
112 void *eloop_data, void *user_data)
115 struct eloop_sock *tmp;
117 if (eloop_prepare_handles())
120 event = WSACreateEvent();
121 if (event == WSA_INVALID_EVENT) {
122 printf("WSACreateEvent() failed: %d\n", WSAGetLastError());
126 if (WSAEventSelect(sock, event, FD_READ)) {
127 printf("WSAEventSelect() failed: %d\n", WSAGetLastError());
128 WSACloseEvent(event);
131 tmp = os_realloc(eloop.readers,
132 (eloop.reader_count + 1) * sizeof(struct eloop_sock));
134 WSAEventSelect(sock, event, 0);
135 WSACloseEvent(event);
139 tmp[eloop.reader_count].sock = sock;
140 tmp[eloop.reader_count].eloop_data = eloop_data;
141 tmp[eloop.reader_count].user_data = user_data;
142 tmp[eloop.reader_count].handler = handler;
143 tmp[eloop.reader_count].event = event;
144 eloop.reader_count++;
146 if (sock > eloop.max_sock)
147 eloop.max_sock = sock;
148 eloop.reader_table_changed = 1;
154 void eloop_unregister_read_sock(int sock)
158 if (eloop.readers == NULL || eloop.reader_count == 0)
161 for (i = 0; i < eloop.reader_count; i++) {
162 if (eloop.readers[i].sock == sock)
165 if (i == eloop.reader_count)
168 WSAEventSelect(eloop.readers[i].sock, eloop.readers[i].event, 0);
169 WSACloseEvent(eloop.readers[i].event);
171 if (i != eloop.reader_count - 1) {
172 os_memmove(&eloop.readers[i], &eloop.readers[i + 1],
173 (eloop.reader_count - i - 1) *
174 sizeof(struct eloop_sock));
176 eloop.reader_count--;
177 eloop.reader_table_changed = 1;
181 int eloop_register_event(void *event, size_t event_size,
182 eloop_event_handler handler,
183 void *eloop_data, void *user_data)
185 struct eloop_event *tmp;
188 if (event_size != sizeof(HANDLE) || h == INVALID_HANDLE_VALUE)
191 if (eloop_prepare_handles())
194 tmp = os_realloc(eloop.events,
195 (eloop.event_count + 1) * sizeof(struct eloop_event));
199 tmp[eloop.event_count].eloop_data = eloop_data;
200 tmp[eloop.event_count].user_data = user_data;
201 tmp[eloop.event_count].handler = handler;
202 tmp[eloop.event_count].event = h;
210 void eloop_unregister_event(void *event, size_t event_size)
215 if (eloop.events == NULL || eloop.event_count == 0 ||
216 event_size != sizeof(HANDLE))
219 for (i = 0; i < eloop.event_count; i++) {
220 if (eloop.events[i].event == h)
223 if (i == eloop.event_count)
226 if (i != eloop.event_count - 1) {
227 os_memmove(&eloop.events[i], &eloop.events[i + 1],
228 (eloop.event_count - i - 1) *
229 sizeof(struct eloop_event));
235 int eloop_register_timeout(unsigned int secs, unsigned int usecs,
236 eloop_timeout_handler handler,
237 void *eloop_data, void *user_data)
239 struct eloop_timeout *timeout, *tmp, *prev;
242 timeout = os_malloc(sizeof(*timeout));
245 os_get_time(&timeout->time);
246 now_sec = timeout->time.sec;
247 timeout->time.sec += secs;
248 if (timeout->time.sec < now_sec) {
250 * Integer overflow - assume long enough timeout to be assumed
251 * to be infinite, i.e., the timeout would never happen.
253 wpa_printf(MSG_DEBUG, "ELOOP: Too long timeout (secs=%u) to "
254 "ever happen - ignore it", secs);
258 timeout->time.usec += usecs;
259 while (timeout->time.usec >= 1000000) {
261 timeout->time.usec -= 1000000;
263 timeout->eloop_data = eloop_data;
264 timeout->user_data = user_data;
265 timeout->handler = handler;
266 timeout->next = NULL;
268 if (eloop.timeout == NULL) {
269 eloop.timeout = timeout;
275 while (tmp != NULL) {
276 if (os_time_before(&timeout->time, &tmp->time))
283 timeout->next = eloop.timeout;
284 eloop.timeout = timeout;
286 timeout->next = prev->next;
287 prev->next = timeout;
294 int eloop_cancel_timeout(eloop_timeout_handler handler,
295 void *eloop_data, void *user_data)
297 struct eloop_timeout *timeout, *prev, *next;
301 timeout = eloop.timeout;
302 while (timeout != NULL) {
303 next = timeout->next;
305 if (timeout->handler == handler &&
306 (timeout->eloop_data == eloop_data ||
307 eloop_data == ELOOP_ALL_CTX) &&
308 (timeout->user_data == user_data ||
309 user_data == ELOOP_ALL_CTX)) {
311 eloop.timeout = next;
326 int eloop_is_timeout_registered(eloop_timeout_handler handler,
327 void *eloop_data, void *user_data)
329 struct eloop_timeout *tmp;
332 while (tmp != NULL) {
333 if (tmp->handler == handler &&
334 tmp->eloop_data == eloop_data &&
335 tmp->user_data == user_data)
345 /* TODO: replace with suitable signal handler */
347 static void eloop_handle_signal(int sig)
352 for (i = 0; i < eloop.signal_count; i++) {
353 if (eloop.signals[i].sig == sig) {
354 eloop.signals[i].signaled++;
362 static void eloop_process_pending_signals(void)
366 if (eloop.signaled == 0)
370 if (eloop.pending_terminate) {
371 eloop.pending_terminate = 0;
374 for (i = 0; i < eloop.signal_count; i++) {
375 if (eloop.signals[i].signaled) {
376 eloop.signals[i].signaled = 0;
377 eloop.signals[i].handler(eloop.signals[i].sig,
378 eloop.signals[i].user_data);
382 if (eloop.term_signal.signaled) {
383 eloop.term_signal.signaled = 0;
384 eloop.term_signal.handler(eloop.term_signal.sig,
385 eloop.term_signal.user_data);
390 int eloop_register_signal(int sig, eloop_signal_handler handler,
393 struct eloop_signal *tmp;
395 tmp = os_realloc(eloop.signals,
396 (eloop.signal_count + 1) *
397 sizeof(struct eloop_signal));
401 tmp[eloop.signal_count].sig = sig;
402 tmp[eloop.signal_count].user_data = user_data;
403 tmp[eloop.signal_count].handler = handler;
404 tmp[eloop.signal_count].signaled = 0;
405 eloop.signal_count++;
408 /* TODO: register signal handler */
415 static BOOL eloop_handle_console_ctrl(DWORD type)
419 case CTRL_BREAK_EVENT:
421 eloop.term_signal.signaled++;
422 SetEvent(eloop.term_event);
428 #endif /* _WIN32_WCE */
431 int eloop_register_signal_terminate(eloop_signal_handler handler,
435 if (SetConsoleCtrlHandler((PHANDLER_ROUTINE) eloop_handle_console_ctrl,
437 printf("SetConsoleCtrlHandler() failed: %d\n",
438 (int) GetLastError());
441 #endif /* _WIN32_WCE */
443 eloop.term_signal.handler = handler;
444 eloop.term_signal.user_data = user_data;
450 int eloop_register_signal_reconfig(eloop_signal_handler handler,
460 struct os_time tv, now;
461 DWORD count, ret, timeout, err;
464 while (!eloop.terminate &&
465 (eloop.timeout || eloop.reader_count > 0 ||
466 eloop.event_count > 0)) {
467 tv.sec = tv.usec = 0;
470 if (os_time_before(&now, &eloop.timeout->time))
471 os_time_sub(&eloop.timeout->time, &now, &tv);
475 for (i = 0; i < eloop.event_count; i++)
476 eloop.handles[count++] = eloop.events[i].event;
478 for (i = 0; i < eloop.reader_count; i++)
479 eloop.handles[count++] = eloop.readers[i].event;
481 if (eloop.term_event)
482 eloop.handles[count++] = eloop.term_event;
485 timeout = tv.sec * 1000 + tv.usec / 1000;
489 if (count > MAXIMUM_WAIT_OBJECTS) {
490 printf("WaitForMultipleObjects: Too many events: "
491 "%d > %d (ignoring extra events)\n",
492 (int) count, MAXIMUM_WAIT_OBJECTS);
493 count = MAXIMUM_WAIT_OBJECTS;
496 ret = WaitForMultipleObjects(count, eloop.handles, FALSE,
498 #else /* _WIN32_WCE */
499 ret = WaitForMultipleObjectsEx(count, eloop.handles, FALSE,
501 #endif /* _WIN32_WCE */
502 err = GetLastError();
504 eloop_process_pending_signals();
506 /* check if some registered timeouts have occurred */
508 struct eloop_timeout *tmp;
511 if (!os_time_before(&now, &eloop.timeout->time)) {
513 eloop.timeout = eloop.timeout->next;
514 tmp->handler(tmp->eloop_data,
521 if (ret == WAIT_FAILED) {
522 printf("WaitForMultipleObjects(count=%d) failed: %d\n",
523 (int) count, (int) err);
529 if (ret == WAIT_IO_COMPLETION)
531 #endif /* _WIN32_WCE */
533 if (ret == WAIT_TIMEOUT)
536 while (ret >= WAIT_OBJECT_0 &&
537 ret < WAIT_OBJECT_0 + eloop.event_count) {
538 eloop.events[ret].handler(
539 eloop.events[ret].eloop_data,
540 eloop.events[ret].user_data);
541 ret = WaitForMultipleObjects(eloop.event_count,
542 eloop.handles, FALSE, 0);
545 eloop.reader_table_changed = 0;
546 for (i = 0; i < eloop.reader_count; i++) {
547 WSANETWORKEVENTS events;
548 if (WSAEnumNetworkEvents(eloop.readers[i].sock,
549 eloop.readers[i].event,
551 (events.lNetworkEvents & FD_READ)) {
552 eloop.readers[i].handler(
553 eloop.readers[i].sock,
554 eloop.readers[i].eloop_data,
555 eloop.readers[i].user_data);
556 if (eloop.reader_table_changed)
564 void eloop_terminate(void)
567 SetEvent(eloop.term_event);
571 void eloop_destroy(void)
573 struct eloop_timeout *timeout, *prev;
575 timeout = eloop.timeout;
576 while (timeout != NULL) {
578 timeout = timeout->next;
581 os_free(eloop.readers);
582 os_free(eloop.signals);
583 if (eloop.term_event)
584 CloseHandle(eloop.term_event);
585 os_free(eloop.handles);
586 eloop.handles = NULL;
587 os_free(eloop.events);
592 int eloop_terminated(void)
594 return eloop.terminate;
598 void eloop_wait_for_read_sock(int sock)
602 event = WSACreateEvent();
603 if (event == WSA_INVALID_EVENT) {
604 printf("WSACreateEvent() failed: %d\n", WSAGetLastError());
608 if (WSAEventSelect(sock, event, FD_READ)) {
609 printf("WSAEventSelect() failed: %d\n", WSAGetLastError());
610 WSACloseEvent(event);
614 WaitForSingleObject(event, INFINITE);
615 WSAEventSelect(sock, event, 0);
616 WSACloseEvent(event);