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 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 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 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 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 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 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 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 285 static void eloop_remove_timeout(struct eloop_timeout *timeout) 286 { 287 dl_list_del(&timeout->list); 288 os_free(timeout); 289 } 290 291 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 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 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 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 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 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 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 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 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 525 int eloop_register_signal_reconfig(eloop_signal_handler handler, 526 void *user_data) 527 { 528 /* TODO */ 529 return 0; 530 } 531 532 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 643 void eloop_terminate(void) 644 { 645 eloop.terminate = 1; 646 SetEvent(eloop.term_event); 647 } 648 649 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 669 int eloop_terminated(void) 670 { 671 return eloop.terminate; 672 } 673 674 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 696 697 int eloop_sock_requeue(void) 698 { 699 return 0; 700 } 701