3 * Copyright (c) 2002-2005, Jouni Malinen <jkmaline@cc.hut.fi>
5 * This program is free software; you can redistribute it and/or modify
6 * it under the terms of the GNU General Public License version 2 as
7 * published by the Free Software Foundation.
9 * Alternatively, this software may be distributed under the terms of BSD
12 * See README and COPYING for more details.
19 #include <sys/types.h>
24 #ifdef CONFIG_NATIVE_WINDOWS
26 #endif /* CONFIG_NATIVE_WINDOWS */
35 void (*handler)(int sock, void *eloop_ctx, void *sock_ctx);
38 struct eloop_timeout {
42 void (*handler)(void *eloop_ctx, void *sock_ctx);
43 struct eloop_timeout *next;
49 void (*handler)(int sig, void *eloop_ctx, void *signal_ctx);
56 int max_sock, reader_count;
57 struct eloop_sock *readers;
59 struct eloop_timeout *timeout;
62 struct eloop_signal *signals;
64 int pending_terminate;
69 static struct eloop_data eloop;
72 void eloop_init(void *user_data)
74 memset(&eloop, 0, sizeof(eloop));
75 eloop.user_data = user_data;
79 int eloop_register_read_sock(int sock,
80 void (*handler)(int sock, void *eloop_ctx,
82 void *eloop_data, void *user_data)
84 struct eloop_sock *tmp;
86 tmp = (struct eloop_sock *)
87 realloc(eloop.readers,
88 (eloop.reader_count + 1) * sizeof(struct eloop_sock));
92 tmp[eloop.reader_count].sock = sock;
93 tmp[eloop.reader_count].eloop_data = eloop_data;
94 tmp[eloop.reader_count].user_data = user_data;
95 tmp[eloop.reader_count].handler = handler;
98 if (sock > eloop.max_sock)
99 eloop.max_sock = sock;
105 void eloop_unregister_read_sock(int sock)
109 if (eloop.readers == NULL || eloop.reader_count == 0)
112 for (i = 0; i < eloop.reader_count; i++) {
113 if (eloop.readers[i].sock == sock)
116 if (i == eloop.reader_count)
118 if (i != eloop.reader_count - 1) {
119 memmove(&eloop.readers[i], &eloop.readers[i + 1],
120 (eloop.reader_count - i - 1) *
121 sizeof(struct eloop_sock));
123 eloop.reader_count--;
127 int eloop_register_timeout(unsigned int secs, unsigned int usecs,
128 void (*handler)(void *eloop_ctx, void *timeout_ctx),
129 void *eloop_data, void *user_data)
131 struct eloop_timeout *timeout, *tmp, *prev;
133 timeout = (struct eloop_timeout *) malloc(sizeof(*timeout));
136 gettimeofday(&timeout->time, NULL);
137 timeout->time.tv_sec += secs;
138 timeout->time.tv_usec += usecs;
139 while (timeout->time.tv_usec >= 1000000) {
140 timeout->time.tv_sec++;
141 timeout->time.tv_usec -= 1000000;
143 timeout->eloop_data = eloop_data;
144 timeout->user_data = user_data;
145 timeout->handler = handler;
146 timeout->next = NULL;
148 if (eloop.timeout == NULL) {
149 eloop.timeout = timeout;
155 while (tmp != NULL) {
156 if (timercmp(&timeout->time, &tmp->time, <))
163 timeout->next = eloop.timeout;
164 eloop.timeout = timeout;
166 timeout->next = prev->next;
167 prev->next = timeout;
174 int eloop_cancel_timeout(void (*handler)(void *eloop_ctx, void *sock_ctx),
175 void *eloop_data, void *user_data)
177 struct eloop_timeout *timeout, *prev, *next;
181 timeout = eloop.timeout;
182 while (timeout != NULL) {
183 next = timeout->next;
185 if (timeout->handler == handler &&
186 (timeout->eloop_data == eloop_data ||
187 eloop_data == ELOOP_ALL_CTX) &&
188 (timeout->user_data == user_data ||
189 user_data == ELOOP_ALL_CTX)) {
191 eloop.timeout = next;
206 #ifndef CONFIG_NATIVE_WINDOWS
207 static void eloop_handle_alarm(int sig)
209 fprintf(stderr, "eloop: could not process SIGINT or SIGTERM in two "
210 "seconds. Looks like there\n"
211 "is a bug that ends up in a busy loop that "
212 "prevents clean shutdown.\n"
213 "Killing program forcefully.\n");
216 #endif /* CONFIG_NATIVE_WINDOWS */
219 static void eloop_handle_signal(int sig)
223 #ifndef CONFIG_NATIVE_WINDOWS
224 if ((sig == SIGINT || sig == SIGTERM) && !eloop.pending_terminate) {
225 /* Use SIGALRM to break out from potential busy loops that
226 * would not allow the program to be killed. */
227 eloop.pending_terminate = 1;
228 signal(SIGALRM, eloop_handle_alarm);
231 #endif /* CONFIG_NATIVE_WINDOWS */
234 for (i = 0; i < eloop.signal_count; i++) {
235 if (eloop.signals[i].sig == sig) {
236 eloop.signals[i].signaled++;
243 static void eloop_process_pending_signals(void)
247 if (eloop.signaled == 0)
251 if (eloop.pending_terminate) {
252 #ifndef CONFIG_NATIVE_WINDOWS
254 #endif /* CONFIG_NATIVE_WINDOWS */
255 eloop.pending_terminate = 0;
258 for (i = 0; i < eloop.signal_count; i++) {
259 if (eloop.signals[i].signaled) {
260 eloop.signals[i].signaled = 0;
261 eloop.signals[i].handler(eloop.signals[i].sig,
263 eloop.signals[i].user_data);
269 int eloop_register_signal(int sig,
270 void (*handler)(int sig, void *eloop_ctx,
274 struct eloop_signal *tmp;
276 tmp = (struct eloop_signal *)
277 realloc(eloop.signals,
278 (eloop.signal_count + 1) *
279 sizeof(struct eloop_signal));
283 tmp[eloop.signal_count].sig = sig;
284 tmp[eloop.signal_count].user_data = user_data;
285 tmp[eloop.signal_count].handler = handler;
286 tmp[eloop.signal_count].signaled = 0;
287 eloop.signal_count++;
289 signal(sig, eloop_handle_signal);
299 struct timeval tv, now;
301 while (!eloop.terminate &&
302 (eloop.timeout || eloop.reader_count > 0)) {
304 gettimeofday(&now, NULL);
305 if (timercmp(&now, &eloop.timeout->time, <))
306 timersub(&eloop.timeout->time, &now, &tv);
308 tv.tv_sec = tv.tv_usec = 0;
310 printf("next timeout in %lu.%06lu sec\n",
311 tv.tv_sec, tv.tv_usec);
316 for (i = 0; i < eloop.reader_count; i++)
317 FD_SET(eloop.readers[i].sock, &rfds);
318 res = select(eloop.max_sock + 1, &rfds, NULL, NULL,
319 eloop.timeout ? &tv : NULL);
320 if (res < 0 && errno != EINTR) {
324 eloop_process_pending_signals();
326 /* check if some registered timeouts have occurred */
328 struct eloop_timeout *tmp;
330 gettimeofday(&now, NULL);
331 if (!timercmp(&now, &eloop.timeout->time, <)) {
333 eloop.timeout = eloop.timeout->next;
334 tmp->handler(tmp->eloop_data,
344 for (i = 0; i < eloop.reader_count; i++) {
345 if (FD_ISSET(eloop.readers[i].sock, &rfds)) {
346 eloop.readers[i].handler(
347 eloop.readers[i].sock,
348 eloop.readers[i].eloop_data,
349 eloop.readers[i].user_data);
356 void eloop_terminate(void)
362 void eloop_destroy(void)
364 struct eloop_timeout *timeout, *prev;
366 timeout = eloop.timeout;
367 while (timeout != NULL) {
369 timeout = timeout->next;
377 int eloop_terminated(void)
379 return eloop.terminate;