123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364 |
- #include "includes.h"
- #include "common.h"
- #include "list.h"
- #include "eloop.h"
- struct eloop_sock {
- int sock;
- void *eloop_data;
- void *user_data;
- eloop_sock_handler handler;
- };
- struct eloop_timeout {
- struct dl_list list;
- struct os_time time;
- void *eloop_data;
- void *user_data;
- eloop_timeout_handler handler;
- };
- struct eloop_signal {
- int sig;
- void *user_data;
- eloop_signal_handler handler;
- int signaled;
- };
- struct eloop_data {
- int max_sock, reader_count;
- struct eloop_sock *readers;
- struct dl_list timeout;
- int signal_count;
- struct eloop_signal *signals;
- int signaled;
- int pending_terminate;
- int terminate;
- int reader_table_changed;
- };
- static struct eloop_data eloop;
- int eloop_init(void)
- {
- memset(&eloop, 0, sizeof(eloop));
- dl_list_init(&eloop.timeout);
- return 0;
- }
- int eloop_register_read_sock(int sock, eloop_sock_handler handler,
- void *eloop_data, void *user_data)
- {
- struct eloop_sock *tmp;
- tmp = (struct eloop_sock *)
- realloc(eloop.readers,
- (eloop.reader_count + 1) * sizeof(struct eloop_sock));
- if (tmp == NULL)
- return -1;
- tmp[eloop.reader_count].sock = sock;
- tmp[eloop.reader_count].eloop_data = eloop_data;
- tmp[eloop.reader_count].user_data = user_data;
- tmp[eloop.reader_count].handler = handler;
- eloop.reader_count++;
- eloop.readers = tmp;
- if (sock > eloop.max_sock)
- eloop.max_sock = sock;
- eloop.reader_table_changed = 1;
- return 0;
- }
- void eloop_unregister_read_sock(int sock)
- {
- int i;
- if (eloop.readers == NULL || eloop.reader_count == 0)
- return;
- for (i = 0; i < eloop.reader_count; i++) {
- if (eloop.readers[i].sock == sock)
- break;
- }
- if (i == eloop.reader_count)
- return;
- if (i != eloop.reader_count - 1) {
- memmove(&eloop.readers[i], &eloop.readers[i + 1],
- (eloop.reader_count - i - 1) *
- sizeof(struct eloop_sock));
- }
- eloop.reader_count--;
- eloop.reader_table_changed = 1;
- }
- int eloop_register_timeout(unsigned int secs, unsigned int usecs,
- eloop_timeout_handler handler,
- void *eloop_data, void *user_data)
- {
- struct eloop_timeout *timeout, *tmp;
- timeout = (struct eloop_timeout *) malloc(sizeof(*timeout));
- if (timeout == NULL)
- return -1;
- os_get_time(&timeout->time);
- timeout->time.sec += secs;
- timeout->time.usec += usecs;
- while (timeout->time.usec >= 1000000) {
- timeout->time.sec++;
- timeout->time.usec -= 1000000;
- }
- timeout->eloop_data = eloop_data;
- timeout->user_data = user_data;
- timeout->handler = handler;
-
- dl_list_for_each(tmp, &eloop.timeout, struct eloop_timeout, list) {
- if (os_time_before(&timeout->time, &tmp->time)) {
- dl_list_add(tmp->list.prev, &timeout->list);
- return 0;
- }
- }
- dl_list_add_tail(&eloop.timeout, &timeout->list);
- return 0;
- }
- int eloop_cancel_timeout(eloop_timeout_handler handler,
- void *eloop_data, void *user_data)
- {
- struct eloop_timeout *timeout, *prev;
- int removed = 0;
- dl_list_for_each_safe(timeout, prev, &eloop.timeout,
- struct eloop_timeout, list) {
- if (timeout->handler == handler &&
- (timeout->eloop_data == eloop_data ||
- eloop_data == ELOOP_ALL_CTX) &&
- (timeout->user_data == user_data ||
- user_data == ELOOP_ALL_CTX)) {
- dl_list_del(&timeout->list);
- free(timeout);
- removed++;
- }
- }
- return removed;
- }
- int eloop_is_timeout_registered(eloop_timeout_handler handler,
- void *eloop_data, void *user_data)
- {
- struct eloop_timeout *tmp;
- dl_list_for_each(tmp, &eloop.timeout, struct eloop_timeout, list) {
- if (tmp->handler == handler &&
- tmp->eloop_data == eloop_data &&
- tmp->user_data == user_data)
- return 1;
- }
- return 0;
- }
- #if 0
- static void eloop_handle_signal(int sig)
- {
- int i;
- eloop.signaled++;
- for (i = 0; i < eloop.signal_count; i++) {
- if (eloop.signals[i].sig == sig) {
- eloop.signals[i].signaled++;
- break;
- }
- }
- }
- #endif
- static void eloop_process_pending_signals(void)
- {
- int i;
- if (eloop.signaled == 0)
- return;
- eloop.signaled = 0;
- if (eloop.pending_terminate) {
- eloop.pending_terminate = 0;
- }
- for (i = 0; i < eloop.signal_count; i++) {
- if (eloop.signals[i].signaled) {
- eloop.signals[i].signaled = 0;
- eloop.signals[i].handler(eloop.signals[i].sig,
- eloop.signals[i].user_data);
- }
- }
- }
- int eloop_register_signal(int sig, eloop_signal_handler handler,
- void *user_data)
- {
- struct eloop_signal *tmp;
- tmp = (struct eloop_signal *)
- realloc(eloop.signals,
- (eloop.signal_count + 1) *
- sizeof(struct eloop_signal));
- if (tmp == NULL)
- return -1;
- tmp[eloop.signal_count].sig = sig;
- tmp[eloop.signal_count].user_data = user_data;
- tmp[eloop.signal_count].handler = handler;
- tmp[eloop.signal_count].signaled = 0;
- eloop.signal_count++;
- eloop.signals = tmp;
-
- return 0;
- }
- int eloop_register_signal_terminate(eloop_signal_handler handler,
- void *user_data)
- {
- #if 0
-
- int ret = eloop_register_signal(SIGINT, handler, user_data);
- if (ret == 0)
- ret = eloop_register_signal(SIGTERM, handler, user_data);
- return ret;
- #endif
- return 0;
- }
- int eloop_register_signal_reconfig(eloop_signal_handler handler,
- void *user_data)
- {
- #if 0
-
- return eloop_register_signal(SIGHUP, handler, user_data);
- #endif
- return 0;
- }
- void eloop_run(void)
- {
- int i;
- struct os_time tv, now;
- while (!eloop.terminate &&
- (!dl_list_empty(&eloop.timeout) || eloop.reader_count > 0)) {
- struct eloop_timeout *timeout;
- timeout = dl_list_first(&eloop.timeout, struct eloop_timeout,
- list);
- if (timeout) {
- os_get_time(&now);
- if (os_time_before(&now, &timeout->time))
- os_time_sub(&timeout->time, &now, &tv);
- else
- tv.sec = tv.usec = 0;
- }
-
- os_sleep(1, 0);
- eloop_process_pending_signals();
-
- timeout = dl_list_first(&eloop.timeout, struct eloop_timeout,
- list);
- if (timeout) {
- os_get_time(&now);
- if (!os_time_before(&now, &timeout->time)) {
- dl_list_del(&timeout->list);
- timeout->handler(timeout->eloop_data,
- timeout->user_data);
- free(timeout);
- }
- }
- eloop.reader_table_changed = 0;
- for (i = 0; i < eloop.reader_count; i++) {
-
- if (0 ) {
- eloop.readers[i].handler(
- eloop.readers[i].sock,
- eloop.readers[i].eloop_data,
- eloop.readers[i].user_data);
- if (eloop.reader_table_changed)
- break;
- }
- }
- }
- }
- void eloop_terminate(void)
- {
- eloop.terminate = 1;
- }
- void eloop_destroy(void)
- {
- struct eloop_timeout *timeout, *prev;
- dl_list_for_each_safe(timeout, prev, &eloop.timeout,
- struct eloop_timeout, list) {
- dl_list_del(&timeout->list);
- free(timeout);
- }
- free(eloop.readers);
- free(eloop.signals);
- }
- int eloop_terminated(void)
- {
- return eloop.terminate;
- }
- void eloop_wait_for_read_sock(int sock)
- {
-
- }
|