contrib/wpa/src/utils/eloop_win.c

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