1 /*
2 * Event loop based on Windows events and WaitForMultipleObjects
3 * Copyright (c) 2002-2009, Jouni Malinen <j@w1.fi>
4 *
5 * This software may be distributed under the terms of the BSD license.
6 * See README for more details.
7 */
8
9 #include "includes.h"
10 #include <winsock2.h>
11
12 #include "common.h"
13 #include "list.h"
14 #include "eloop.h"
15
16
17 struct eloop_sock {
18 int sock;
19 void *eloop_data;
20 void *user_data;
21 eloop_sock_handler handler;
22 WSAEVENT event;
23 };
24
25 struct eloop_event {
26 void *eloop_data;
27 void *user_data;
28 eloop_event_handler handler;
29 HANDLE event;
30 };
31
32 struct eloop_timeout {
33 struct dl_list list;
34 struct os_reltime time;
35 void *eloop_data;
36 void *user_data;
37 eloop_timeout_handler handler;
38 };
39
40 struct eloop_signal {
41 int sig;
42 void *user_data;
43 eloop_signal_handler handler;
44 int signaled;
45 };
46
47 struct eloop_data {
48 int max_sock;
49 size_t reader_count;
50 struct eloop_sock *readers;
51
52 size_t event_count;
53 struct eloop_event *events;
54
55 struct dl_list timeout;
56
57 int signal_count;
58 struct eloop_signal *signals;
59 int signaled;
60 int pending_terminate;
61
62 int terminate;
63 int reader_table_changed;
64
65 struct eloop_signal term_signal;
66 HANDLE term_event;
67
68 HANDLE *handles;
69 size_t num_handles;
70 };
71
72 static struct eloop_data eloop;
73
74
eloop_init(void)75 int eloop_init(void)
76 {
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)
83 return -1;
84
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);
90 return -1;
91 }
92
93 return 0;
94 }
95
96
eloop_prepare_handles(void)97 static int eloop_prepare_handles(void)
98 {
99 HANDLE *n;
100
101 if (eloop.num_handles > eloop.reader_count + eloop.event_count + 8)
102 return 0;
103 n = os_realloc_array(eloop.handles, eloop.num_handles * 2,
104 sizeof(eloop.handles[0]));
105 if (n == NULL)
106 return -1;
107 eloop.handles = n;
108 eloop.num_handles *= 2;
109 return 0;
110 }
111
112
eloop_register_read_sock(int sock,eloop_sock_handler handler,void * eloop_data,void * user_data)113 int eloop_register_read_sock(int sock, eloop_sock_handler handler,
114 void *eloop_data, void *user_data)
115 {
116 WSAEVENT event;
117 struct eloop_sock *tmp;
118
119 if (eloop_prepare_handles())
120 return -1;
121
122 event = WSACreateEvent();
123 if (event == WSA_INVALID_EVENT) {
124 printf("WSACreateEvent() failed: %d\n", WSAGetLastError());
125 return -1;
126 }
127
128 if (WSAEventSelect(sock, event, FD_READ)) {
129 printf("WSAEventSelect() failed: %d\n", WSAGetLastError());
130 WSACloseEvent(event);
131 return -1;
132 }
133 tmp = os_realloc_array(eloop.readers, eloop.reader_count + 1,
134 sizeof(struct eloop_sock));
135 if (tmp == NULL) {
136 WSAEventSelect(sock, event, 0);
137 WSACloseEvent(event);
138 return -1;
139 }
140
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++;
147 eloop.readers = tmp;
148 if (sock > eloop.max_sock)
149 eloop.max_sock = sock;
150 eloop.reader_table_changed = 1;
151
152 return 0;
153 }
154
155
eloop_unregister_read_sock(int sock)156 void eloop_unregister_read_sock(int sock)
157 {
158 size_t i;
159
160 if (eloop.readers == NULL || eloop.reader_count == 0)
161 return;
162
163 for (i = 0; i < eloop.reader_count; i++) {
164 if (eloop.readers[i].sock == sock)
165 break;
166 }
167 if (i == eloop.reader_count)
168 return;
169
170 WSAEventSelect(eloop.readers[i].sock, eloop.readers[i].event, 0);
171 WSACloseEvent(eloop.readers[i].event);
172
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));
177 }
178 eloop.reader_count--;
179 eloop.reader_table_changed = 1;
180 }
181
182
eloop_register_event(void * event,size_t event_size,eloop_event_handler handler,void * eloop_data,void * user_data)183 int eloop_register_event(void *event, size_t event_size,
184 eloop_event_handler handler,
185 void *eloop_data, void *user_data)
186 {
187 struct eloop_event *tmp;
188 HANDLE h = event;
189
190 if (event_size != sizeof(HANDLE) || h == INVALID_HANDLE_VALUE)
191 return -1;
192
193 if (eloop_prepare_handles())
194 return -1;
195
196 tmp = os_realloc_array(eloop.events, eloop.event_count + 1,
197 sizeof(struct eloop_event));
198 if (tmp == NULL)
199 return -1;
200
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;
205 eloop.event_count++;
206 eloop.events = tmp;
207
208 return 0;
209 }
210
211
eloop_unregister_event(void * event,size_t event_size)212 void eloop_unregister_event(void *event, size_t event_size)
213 {
214 size_t i;
215 HANDLE h = event;
216
217 if (eloop.events == NULL || eloop.event_count == 0 ||
218 event_size != sizeof(HANDLE))
219 return;
220
221 for (i = 0; i < eloop.event_count; i++) {
222 if (eloop.events[i].event == h)
223 break;
224 }
225 if (i == eloop.event_count)
226 return;
227
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));
232 }
233 eloop.event_count--;
234 }
235
236
eloop_register_timeout(unsigned int secs,unsigned int usecs,eloop_timeout_handler handler,void * eloop_data,void * user_data)237 int eloop_register_timeout(unsigned int secs, unsigned int usecs,
238 eloop_timeout_handler handler,
239 void *eloop_data, void *user_data)
240 {
241 struct eloop_timeout *timeout, *tmp;
242 os_time_t now_sec;
243
244 timeout = os_zalloc(sizeof(*timeout));
245 if (timeout == NULL)
246 return -1;
247 if (os_get_reltime(&timeout->time) < 0) {
248 os_free(timeout);
249 return -1;
250 }
251 now_sec = timeout->time.sec;
252 timeout->time.sec += secs;
253 if (timeout->time.sec < now_sec) {
254 /*
255 * Integer overflow - assume long enough timeout to be assumed
256 * to be infinite, i.e., the timeout would never happen.
257 */
258 wpa_printf(MSG_DEBUG, "ELOOP: Too long timeout (secs=%u) to "
259 "ever happen - ignore it", secs);
260 os_free(timeout);
261 return 0;
262 }
263 timeout->time.usec += usecs;
264 while (timeout->time.usec >= 1000000) {
265 timeout->time.sec++;
266 timeout->time.usec -= 1000000;
267 }
268 timeout->eloop_data = eloop_data;
269 timeout->user_data = user_data;
270 timeout->handler = handler;
271
272 /* Maintain timeouts in order of increasing time */
273 dl_list_for_each(tmp, &eloop.timeout, struct eloop_timeout, list) {
274 if (os_reltime_before(&timeout->time, &tmp->time)) {
275 dl_list_add(tmp->list.prev, &timeout->list);
276 return 0;
277 }
278 }
279 dl_list_add_tail(&eloop.timeout, &timeout->list);
280
281 return 0;
282 }
283
284
eloop_remove_timeout(struct eloop_timeout * timeout)285 static void eloop_remove_timeout(struct eloop_timeout *timeout)
286 {
287 dl_list_del(&timeout->list);
288 os_free(timeout);
289 }
290
291
eloop_cancel_timeout(eloop_timeout_handler handler,void * eloop_data,void * user_data)292 int eloop_cancel_timeout(eloop_timeout_handler handler,
293 void *eloop_data, void *user_data)
294 {
295 struct eloop_timeout *timeout, *prev;
296 int removed = 0;
297
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);
306 removed++;
307 }
308 }
309
310 return removed;
311 }
312
313
eloop_cancel_timeout_one(eloop_timeout_handler handler,void * eloop_data,void * user_data,struct os_reltime * remaining)314 int eloop_cancel_timeout_one(eloop_timeout_handler handler,
315 void *eloop_data, void *user_data,
316 struct os_reltime *remaining)
317 {
318 struct eloop_timeout *timeout, *prev;
319 int removed = 0;
320 struct os_reltime now;
321
322 os_get_reltime(&now);
323 remaining->sec = remaining->usec = 0;
324
325 dl_list_for_each_safe(timeout, prev, &eloop.timeout,
326 struct eloop_timeout, list) {
327 if (timeout->handler == handler &&
328 (timeout->eloop_data == eloop_data) &&
329 (timeout->user_data == user_data)) {
330 removed = 1;
331 if (os_reltime_before(&now, &timeout->time))
332 os_reltime_sub(&timeout->time, &now, remaining);
333 eloop_remove_timeout(timeout);
334 break;
335 }
336 }
337 return removed;
338 }
339
340
eloop_is_timeout_registered(eloop_timeout_handler handler,void * eloop_data,void * user_data)341 int eloop_is_timeout_registered(eloop_timeout_handler handler,
342 void *eloop_data, void *user_data)
343 {
344 struct eloop_timeout *tmp;
345
346 dl_list_for_each(tmp, &eloop.timeout, struct eloop_timeout, list) {
347 if (tmp->handler == handler &&
348 tmp->eloop_data == eloop_data &&
349 tmp->user_data == user_data)
350 return 1;
351 }
352
353 return 0;
354 }
355
356
eloop_deplete_timeout(unsigned int req_secs,unsigned int req_usecs,eloop_timeout_handler handler,void * eloop_data,void * user_data)357 int eloop_deplete_timeout(unsigned int req_secs, unsigned int req_usecs,
358 eloop_timeout_handler handler, void *eloop_data,
359 void *user_data)
360 {
361 struct os_reltime now, requested, remaining;
362 struct eloop_timeout *tmp;
363
364 dl_list_for_each(tmp, &eloop.timeout, struct eloop_timeout, list) {
365 if (tmp->handler == handler &&
366 tmp->eloop_data == eloop_data &&
367 tmp->user_data == user_data) {
368 requested.sec = req_secs;
369 requested.usec = req_usecs;
370 os_get_reltime(&now);
371 os_reltime_sub(&tmp->time, &now, &remaining);
372 if (os_reltime_before(&requested, &remaining)) {
373 eloop_cancel_timeout(handler, eloop_data,
374 user_data);
375 eloop_register_timeout(requested.sec,
376 requested.usec,
377 handler, eloop_data,
378 user_data);
379 return 1;
380 }
381 return 0;
382 }
383 }
384
385 return -1;
386 }
387
388
eloop_replenish_timeout(unsigned int req_secs,unsigned int req_usecs,eloop_timeout_handler handler,void * eloop_data,void * user_data)389 int eloop_replenish_timeout(unsigned int req_secs, unsigned int req_usecs,
390 eloop_timeout_handler handler, void *eloop_data,
391 void *user_data)
392 {
393 struct os_reltime now, requested, remaining;
394 struct eloop_timeout *tmp;
395
396 dl_list_for_each(tmp, &eloop.timeout, struct eloop_timeout, list) {
397 if (tmp->handler == handler &&
398 tmp->eloop_data == eloop_data &&
399 tmp->user_data == user_data) {
400 requested.sec = req_secs;
401 requested.usec = req_usecs;
402 os_get_reltime(&now);
403 os_reltime_sub(&tmp->time, &now, &remaining);
404 if (os_reltime_before(&remaining, &requested)) {
405 eloop_cancel_timeout(handler, eloop_data,
406 user_data);
407 eloop_register_timeout(requested.sec,
408 requested.usec,
409 handler, eloop_data,
410 user_data);
411 return 1;
412 }
413 return 0;
414 }
415 }
416
417 return -1;
418 }
419
420
421 /* TODO: replace with suitable signal handler */
422 #if 0
423 static void eloop_handle_signal(int sig)
424 {
425 int i;
426
427 eloop.signaled++;
428 for (i = 0; i < eloop.signal_count; i++) {
429 if (eloop.signals[i].sig == sig) {
430 eloop.signals[i].signaled++;
431 break;
432 }
433 }
434 }
435 #endif
436
437
eloop_process_pending_signals(void)438 static void eloop_process_pending_signals(void)
439 {
440 int i;
441
442 if (eloop.signaled == 0)
443 return;
444 eloop.signaled = 0;
445
446 if (eloop.pending_terminate) {
447 eloop.pending_terminate = 0;
448 }
449
450 for (i = 0; i < eloop.signal_count; i++) {
451 if (eloop.signals[i].signaled) {
452 eloop.signals[i].signaled = 0;
453 eloop.signals[i].handler(eloop.signals[i].sig,
454 eloop.signals[i].user_data);
455 }
456 }
457
458 if (eloop.term_signal.signaled) {
459 eloop.term_signal.signaled = 0;
460 eloop.term_signal.handler(eloop.term_signal.sig,
461 eloop.term_signal.user_data);
462 }
463 }
464
465
eloop_register_signal(int sig,eloop_signal_handler handler,void * user_data)466 int eloop_register_signal(int sig, eloop_signal_handler handler,
467 void *user_data)
468 {
469 struct eloop_signal *tmp;
470
471 tmp = os_realloc_array(eloop.signals, eloop.signal_count + 1,
472 sizeof(struct eloop_signal));
473 if (tmp == NULL)
474 return -1;
475
476 tmp[eloop.signal_count].sig = sig;
477 tmp[eloop.signal_count].user_data = user_data;
478 tmp[eloop.signal_count].handler = handler;
479 tmp[eloop.signal_count].signaled = 0;
480 eloop.signal_count++;
481 eloop.signals = tmp;
482
483 /* TODO: register signal handler */
484
485 return 0;
486 }
487
488
489 #ifndef _WIN32_WCE
eloop_handle_console_ctrl(DWORD type)490 static BOOL eloop_handle_console_ctrl(DWORD type)
491 {
492 switch (type) {
493 case CTRL_C_EVENT:
494 case CTRL_BREAK_EVENT:
495 eloop.signaled++;
496 eloop.term_signal.signaled++;
497 SetEvent(eloop.term_event);
498 return TRUE;
499 default:
500 return FALSE;
501 }
502 }
503 #endif /* _WIN32_WCE */
504
505
eloop_register_signal_terminate(eloop_signal_handler handler,void * user_data)506 int eloop_register_signal_terminate(eloop_signal_handler handler,
507 void *user_data)
508 {
509 #ifndef _WIN32_WCE
510 if (SetConsoleCtrlHandler((PHANDLER_ROUTINE) eloop_handle_console_ctrl,
511 TRUE) == 0) {
512 printf("SetConsoleCtrlHandler() failed: %d\n",
513 (int) GetLastError());
514 return -1;
515 }
516 #endif /* _WIN32_WCE */
517
518 eloop.term_signal.handler = handler;
519 eloop.term_signal.user_data = user_data;
520
521 return 0;
522 }
523
524
eloop_register_signal_reconfig(eloop_signal_handler handler,void * user_data)525 int eloop_register_signal_reconfig(eloop_signal_handler handler,
526 void *user_data)
527 {
528 /* TODO */
529 return 0;
530 }
531
532
eloop_run(void)533 void eloop_run(void)
534 {
535 struct os_reltime tv, now;
536 DWORD count, ret, timeout_val, err;
537 size_t i;
538
539 while (!eloop.terminate &&
540 (!dl_list_empty(&eloop.timeout) || eloop.reader_count > 0 ||
541 eloop.event_count > 0)) {
542 struct eloop_timeout *timeout;
543 tv.sec = tv.usec = 0;
544 timeout = dl_list_first(&eloop.timeout, struct eloop_timeout,
545 list);
546 if (timeout) {
547 os_get_reltime(&now);
548 if (os_reltime_before(&now, &timeout->time))
549 os_reltime_sub(&timeout->time, &now, &tv);
550 }
551
552 count = 0;
553 for (i = 0; i < eloop.event_count; i++)
554 eloop.handles[count++] = eloop.events[i].event;
555
556 for (i = 0; i < eloop.reader_count; i++)
557 eloop.handles[count++] = eloop.readers[i].event;
558
559 if (eloop.term_event)
560 eloop.handles[count++] = eloop.term_event;
561
562 if (timeout)
563 timeout_val = tv.sec * 1000 + tv.usec / 1000;
564 else
565 timeout_val = INFINITE;
566
567 if (count > MAXIMUM_WAIT_OBJECTS) {
568 printf("WaitForMultipleObjects: Too many events: "
569 "%d > %d (ignoring extra events)\n",
570 (int) count, MAXIMUM_WAIT_OBJECTS);
571 count = MAXIMUM_WAIT_OBJECTS;
572 }
573 #ifdef _WIN32_WCE
574 ret = WaitForMultipleObjects(count, eloop.handles, FALSE,
575 timeout_val);
576 #else /* _WIN32_WCE */
577 ret = WaitForMultipleObjectsEx(count, eloop.handles, FALSE,
578 timeout_val, TRUE);
579 #endif /* _WIN32_WCE */
580 err = GetLastError();
581
582 eloop_process_pending_signals();
583
584 /* check if some registered timeouts have occurred */
585 timeout = dl_list_first(&eloop.timeout, struct eloop_timeout,
586 list);
587 if (timeout) {
588 os_get_reltime(&now);
589 if (!os_reltime_before(&now, &timeout->time)) {
590 void *eloop_data = timeout->eloop_data;
591 void *user_data = timeout->user_data;
592 eloop_timeout_handler handler =
593 timeout->handler;
594 eloop_remove_timeout(timeout);
595 handler(eloop_data, user_data);
596 }
597
598 }
599
600 if (ret == WAIT_FAILED) {
601 printf("WaitForMultipleObjects(count=%d) failed: %d\n",
602 (int) count, (int) err);
603 os_sleep(1, 0);
604 continue;
605 }
606
607 #ifndef _WIN32_WCE
608 if (ret == WAIT_IO_COMPLETION)
609 continue;
610 #endif /* _WIN32_WCE */
611
612 if (ret == WAIT_TIMEOUT)
613 continue;
614
615 while (ret >= WAIT_OBJECT_0 &&
616 ret < WAIT_OBJECT_0 + eloop.event_count) {
617 eloop.events[ret].handler(
618 eloop.events[ret].eloop_data,
619 eloop.events[ret].user_data);
620 ret = WaitForMultipleObjects(eloop.event_count,
621 eloop.handles, FALSE, 0);
622 }
623
624 eloop.reader_table_changed = 0;
625 for (i = 0; i < eloop.reader_count; i++) {
626 WSANETWORKEVENTS events;
627 if (WSAEnumNetworkEvents(eloop.readers[i].sock,
628 eloop.readers[i].event,
629 &events) == 0 &&
630 (events.lNetworkEvents & FD_READ)) {
631 eloop.readers[i].handler(
632 eloop.readers[i].sock,
633 eloop.readers[i].eloop_data,
634 eloop.readers[i].user_data);
635 if (eloop.reader_table_changed)
636 break;
637 }
638 }
639 }
640 }
641
642
eloop_terminate(void)643 void eloop_terminate(void)
644 {
645 eloop.terminate = 1;
646 SetEvent(eloop.term_event);
647 }
648
649
eloop_destroy(void)650 void eloop_destroy(void)
651 {
652 struct eloop_timeout *timeout, *prev;
653
654 dl_list_for_each_safe(timeout, prev, &eloop.timeout,
655 struct eloop_timeout, list) {
656 eloop_remove_timeout(timeout);
657 }
658 os_free(eloop.readers);
659 os_free(eloop.signals);
660 if (eloop.term_event)
661 CloseHandle(eloop.term_event);
662 os_free(eloop.handles);
663 eloop.handles = NULL;
664 os_free(eloop.events);
665 eloop.events = NULL;
666 }
667
668
eloop_terminated(void)669 int eloop_terminated(void)
670 {
671 return eloop.terminate;
672 }
673
674
eloop_wait_for_read_sock(int sock)675 void eloop_wait_for_read_sock(int sock)
676 {
677 WSAEVENT event;
678
679 event = WSACreateEvent();
680 if (event == WSA_INVALID_EVENT) {
681 printf("WSACreateEvent() failed: %d\n", WSAGetLastError());
682 return;
683 }
684
685 if (WSAEventSelect(sock, event, FD_READ)) {
686 printf("WSAEventSelect() failed: %d\n", WSAGetLastError());
687 WSACloseEvent(event);
688 return ;
689 }
690
691 WaitForSingleObject(event, INFINITE);
692 WSAEventSelect(sock, event, 0);
693 WSACloseEvent(event);
694 }
695