1*9f20bfa6SDavid van Moolenbroek /* $NetBSD: eloop.h,v 1.9 2015/05/16 23:31:32 roy Exp $ */ 2*9f20bfa6SDavid van Moolenbroek 3*9f20bfa6SDavid van Moolenbroek /* 4*9f20bfa6SDavid van Moolenbroek * dhcpcd - DHCP client daemon 5*9f20bfa6SDavid van Moolenbroek * Copyright (c) 2006-2015 Roy Marples <roy@marples.name> 6*9f20bfa6SDavid van Moolenbroek * All rights reserved 7*9f20bfa6SDavid van Moolenbroek 8*9f20bfa6SDavid van Moolenbroek * Redistribution and use in source and binary forms, with or without 9*9f20bfa6SDavid van Moolenbroek * modification, are permitted provided that the following conditions 10*9f20bfa6SDavid van Moolenbroek * are met: 11*9f20bfa6SDavid van Moolenbroek * 1. Redistributions of source code must retain the above copyright 12*9f20bfa6SDavid van Moolenbroek * notice, this list of conditions and the following disclaimer. 13*9f20bfa6SDavid van Moolenbroek * 2. Redistributions in binary form must reproduce the above copyright 14*9f20bfa6SDavid van Moolenbroek * notice, this list of conditions and the following disclaimer in the 15*9f20bfa6SDavid van Moolenbroek * documentation and/or other materials provided with the distribution. 16*9f20bfa6SDavid van Moolenbroek * 17*9f20bfa6SDavid van Moolenbroek * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND 18*9f20bfa6SDavid van Moolenbroek * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 19*9f20bfa6SDavid van Moolenbroek * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 20*9f20bfa6SDavid van Moolenbroek * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE 21*9f20bfa6SDavid van Moolenbroek * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 22*9f20bfa6SDavid van Moolenbroek * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 23*9f20bfa6SDavid van Moolenbroek * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 24*9f20bfa6SDavid van Moolenbroek * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 25*9f20bfa6SDavid van Moolenbroek * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 26*9f20bfa6SDavid van Moolenbroek * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 27*9f20bfa6SDavid van Moolenbroek * SUCH DAMAGE. 28*9f20bfa6SDavid van Moolenbroek */ 29*9f20bfa6SDavid van Moolenbroek 30*9f20bfa6SDavid van Moolenbroek #ifndef ELOOP_H 31*9f20bfa6SDavid van Moolenbroek #define ELOOP_H 32*9f20bfa6SDavid van Moolenbroek 33*9f20bfa6SDavid van Moolenbroek #include <time.h> 34*9f20bfa6SDavid van Moolenbroek 35*9f20bfa6SDavid van Moolenbroek #ifdef HAVE_CONFIG_H 36*9f20bfa6SDavid van Moolenbroek #include "config.h" 37*9f20bfa6SDavid van Moolenbroek #else 38*9f20bfa6SDavid van Moolenbroek /* Attempt to autodetect kqueue or epoll. 39*9f20bfa6SDavid van Moolenbroek * If we can't, the system has to support pselect, which is a POSIX call. */ 40*9f20bfa6SDavid van Moolenbroek #if (defined(__unix__) || defined(unix)) && !defined(USG) 41*9f20bfa6SDavid van Moolenbroek #include <sys/param.h> 42*9f20bfa6SDavid van Moolenbroek #endif 43*9f20bfa6SDavid van Moolenbroek #if defined(BSD) 44*9f20bfa6SDavid van Moolenbroek /* Assume BSD has a working sys/queue.h and kqueue(2) interface */ 45*9f20bfa6SDavid van Moolenbroek #define HAVE_SYS_QUEUE_H 46*9f20bfa6SDavid van Moolenbroek #define HAVE_KQUEUE 47*9f20bfa6SDavid van Moolenbroek #elif defined(__linux__) 48*9f20bfa6SDavid van Moolenbroek /* Assume Linux has a working epoll(3) interface */ 49*9f20bfa6SDavid van Moolenbroek #define HAVE_EPOLL 50*9f20bfa6SDavid van Moolenbroek #endif 51*9f20bfa6SDavid van Moolenbroek #endif 52*9f20bfa6SDavid van Moolenbroek 53*9f20bfa6SDavid van Moolenbroek /* Our structures require TAILQ macros, which really every libc should 54*9f20bfa6SDavid van Moolenbroek * ship as they are useful beyond belief. 55*9f20bfa6SDavid van Moolenbroek * Sadly some libc's don't have sys/queue.h and some that do don't have 56*9f20bfa6SDavid van Moolenbroek * the TAILQ_FOREACH macro. For those that don't, the application using 57*9f20bfa6SDavid van Moolenbroek * this implementation will need to ship a working queue.h somewhere. 58*9f20bfa6SDavid van Moolenbroek * If we don't have sys/queue.h found in config.h, then 59*9f20bfa6SDavid van Moolenbroek * allow QUEUE_H to override loading queue.h in the current directory. */ 60*9f20bfa6SDavid van Moolenbroek #ifndef TAILQ_FOREACH 61*9f20bfa6SDavid van Moolenbroek #ifdef HAVE_SYS_QUEUE_H 62*9f20bfa6SDavid van Moolenbroek #include <sys/queue.h> 63*9f20bfa6SDavid van Moolenbroek #elif defined(QUEUE_H) 64*9f20bfa6SDavid van Moolenbroek #define __QUEUE_HEADER(x) #x 65*9f20bfa6SDavid van Moolenbroek #define _QUEUE_HEADER(x) __QUEUE_HEADER(x) 66*9f20bfa6SDavid van Moolenbroek #include _QUEUE_HEADER(QUEUE_H) 67*9f20bfa6SDavid van Moolenbroek #else 68*9f20bfa6SDavid van Moolenbroek #include "queue.h" 69*9f20bfa6SDavid van Moolenbroek #endif 70*9f20bfa6SDavid van Moolenbroek #endif 71*9f20bfa6SDavid van Moolenbroek 72*9f20bfa6SDavid van Moolenbroek /* Some systems don't define timespec macros */ 73*9f20bfa6SDavid van Moolenbroek #ifndef timespecclear 74*9f20bfa6SDavid van Moolenbroek #define timespecclear(tsp) (tsp)->tv_sec = (time_t)((tsp)->tv_nsec = 0L) 75*9f20bfa6SDavid van Moolenbroek #define timespecisset(tsp) ((tsp)->tv_sec || (tsp)->tv_nsec) 76*9f20bfa6SDavid van Moolenbroek #define timespeccmp(tsp, usp, cmp) \ 77*9f20bfa6SDavid van Moolenbroek (((tsp)->tv_sec == (usp)->tv_sec) ? \ 78*9f20bfa6SDavid van Moolenbroek ((tsp)->tv_nsec cmp (usp)->tv_nsec) : \ 79*9f20bfa6SDavid van Moolenbroek ((tsp)->tv_sec cmp (usp)->tv_sec)) 80*9f20bfa6SDavid van Moolenbroek #define timespecadd(tsp, usp, vsp) \ 81*9f20bfa6SDavid van Moolenbroek do { \ 82*9f20bfa6SDavid van Moolenbroek (vsp)->tv_sec = (tsp)->tv_sec + (usp)->tv_sec; \ 83*9f20bfa6SDavid van Moolenbroek (vsp)->tv_nsec = (tsp)->tv_nsec + (usp)->tv_nsec; \ 84*9f20bfa6SDavid van Moolenbroek if ((vsp)->tv_nsec >= 1000000000L) { \ 85*9f20bfa6SDavid van Moolenbroek (vsp)->tv_sec++; \ 86*9f20bfa6SDavid van Moolenbroek (vsp)->tv_nsec -= 1000000000L; \ 87*9f20bfa6SDavid van Moolenbroek } \ 88*9f20bfa6SDavid van Moolenbroek } while (/* CONSTCOND */ 0) 89*9f20bfa6SDavid van Moolenbroek #define timespecsub(tsp, usp, vsp) \ 90*9f20bfa6SDavid van Moolenbroek do { \ 91*9f20bfa6SDavid van Moolenbroek (vsp)->tv_sec = (tsp)->tv_sec - (usp)->tv_sec; \ 92*9f20bfa6SDavid van Moolenbroek (vsp)->tv_nsec = (tsp)->tv_nsec - (usp)->tv_nsec; \ 93*9f20bfa6SDavid van Moolenbroek if ((vsp)->tv_nsec < 0) { \ 94*9f20bfa6SDavid van Moolenbroek (vsp)->tv_sec--; \ 95*9f20bfa6SDavid van Moolenbroek (vsp)->tv_nsec += 1000000000L; \ 96*9f20bfa6SDavid van Moolenbroek } \ 97*9f20bfa6SDavid van Moolenbroek } while (/* CONSTCOND */ 0) 98*9f20bfa6SDavid van Moolenbroek #endif 99*9f20bfa6SDavid van Moolenbroek 100*9f20bfa6SDavid van Moolenbroek /* eloop queues are really only for deleting timeouts registered 101*9f20bfa6SDavid van Moolenbroek * for a function or object. 102*9f20bfa6SDavid van Moolenbroek * The idea being that one interface as different timeouts for 103*9f20bfa6SDavid van Moolenbroek * say DHCP and DHCPv6. */ 104*9f20bfa6SDavid van Moolenbroek #ifndef ELOOP_QUEUE 105*9f20bfa6SDavid van Moolenbroek #define ELOOP_QUEUE 1 106*9f20bfa6SDavid van Moolenbroek #endif 107*9f20bfa6SDavid van Moolenbroek 108*9f20bfa6SDavid van Moolenbroek struct eloop_event { 109*9f20bfa6SDavid van Moolenbroek TAILQ_ENTRY(eloop_event) next; 110*9f20bfa6SDavid van Moolenbroek int fd; 111*9f20bfa6SDavid van Moolenbroek void (*read_cb)(void *); 112*9f20bfa6SDavid van Moolenbroek void *read_cb_arg; 113*9f20bfa6SDavid van Moolenbroek void (*write_cb)(void *); 114*9f20bfa6SDavid van Moolenbroek void *write_cb_arg; 115*9f20bfa6SDavid van Moolenbroek #if !defined(HAVE_KQUEUE) && !defined(HAVE_EPOLL) 116*9f20bfa6SDavid van Moolenbroek struct pollfd *pollfd; 117*9f20bfa6SDavid van Moolenbroek #endif 118*9f20bfa6SDavid van Moolenbroek }; 119*9f20bfa6SDavid van Moolenbroek 120*9f20bfa6SDavid van Moolenbroek struct eloop_timeout { 121*9f20bfa6SDavid van Moolenbroek TAILQ_ENTRY(eloop_timeout) next; 122*9f20bfa6SDavid van Moolenbroek struct timespec when; 123*9f20bfa6SDavid van Moolenbroek void (*callback)(void *); 124*9f20bfa6SDavid van Moolenbroek void *arg; 125*9f20bfa6SDavid van Moolenbroek int queue; 126*9f20bfa6SDavid van Moolenbroek }; 127*9f20bfa6SDavid van Moolenbroek 128*9f20bfa6SDavid van Moolenbroek struct eloop { 129*9f20bfa6SDavid van Moolenbroek size_t events_len; 130*9f20bfa6SDavid van Moolenbroek TAILQ_HEAD (event_head, eloop_event) events; 131*9f20bfa6SDavid van Moolenbroek struct event_head free_events; 132*9f20bfa6SDavid van Moolenbroek 133*9f20bfa6SDavid van Moolenbroek TAILQ_HEAD (timeout_head, eloop_timeout) timeouts; 134*9f20bfa6SDavid van Moolenbroek struct timeout_head free_timeouts; 135*9f20bfa6SDavid van Moolenbroek 136*9f20bfa6SDavid van Moolenbroek void (*timeout0)(void *); 137*9f20bfa6SDavid van Moolenbroek void *timeout0_arg; 138*9f20bfa6SDavid van Moolenbroek const int *signals; 139*9f20bfa6SDavid van Moolenbroek size_t signals_len; 140*9f20bfa6SDavid van Moolenbroek void (*signal_cb)(int, void *); 141*9f20bfa6SDavid van Moolenbroek void *signal_cb_ctx; 142*9f20bfa6SDavid van Moolenbroek 143*9f20bfa6SDavid van Moolenbroek #if defined(HAVE_KQUEUE) || defined(HAVE_EPOLL) 144*9f20bfa6SDavid van Moolenbroek int poll_fd; 145*9f20bfa6SDavid van Moolenbroek #else 146*9f20bfa6SDavid van Moolenbroek struct pollfd *fds; 147*9f20bfa6SDavid van Moolenbroek size_t fds_len; 148*9f20bfa6SDavid van Moolenbroek #endif 149*9f20bfa6SDavid van Moolenbroek 150*9f20bfa6SDavid van Moolenbroek int exitnow; 151*9f20bfa6SDavid van Moolenbroek int exitcode; 152*9f20bfa6SDavid van Moolenbroek }; 153*9f20bfa6SDavid van Moolenbroek 154*9f20bfa6SDavid van Moolenbroek int eloop_event_add(struct eloop *, int, 155*9f20bfa6SDavid van Moolenbroek void (*)(void *), void *, 156*9f20bfa6SDavid van Moolenbroek void (*)(void *), void *); 157*9f20bfa6SDavid van Moolenbroek #define eloop_event_delete(eloop, fd) \ 158*9f20bfa6SDavid van Moolenbroek eloop_event_delete_write((eloop), (fd), 0) 159*9f20bfa6SDavid van Moolenbroek #define eloop_event_remove_writecb(eloop, fd) \ 160*9f20bfa6SDavid van Moolenbroek eloop_event_delete_write((eloop), (fd), 1) 161*9f20bfa6SDavid van Moolenbroek void eloop_event_delete_write(struct eloop *, int, int); 162*9f20bfa6SDavid van Moolenbroek 163*9f20bfa6SDavid van Moolenbroek #define eloop_timeout_add_tv(eloop, tv, cb, ctx) \ 164*9f20bfa6SDavid van Moolenbroek eloop_q_timeout_add_tv((eloop), ELOOP_QUEUE, (tv), (cb), (ctx)) 165*9f20bfa6SDavid van Moolenbroek #define eloop_timeout_add_sec(eloop, tv, cb, ctx) \ 166*9f20bfa6SDavid van Moolenbroek eloop_q_timeout_add_sec((eloop), ELOOP_QUEUE, (tv), (cb), (ctx)) 167*9f20bfa6SDavid van Moolenbroek #define eloop_timeout_add_msec(eloop, ms, cb, ctx) \ 168*9f20bfa6SDavid van Moolenbroek eloop_q_timeout_add_msec((eloop), ELOOP_QUEUE, (ms), (cb), (ctx)) 169*9f20bfa6SDavid van Moolenbroek #define eloop_timeout_delete(eloop, cb, ctx) \ 170*9f20bfa6SDavid van Moolenbroek eloop_q_timeout_delete((eloop), ELOOP_QUEUE, (cb), (ctx)) 171*9f20bfa6SDavid van Moolenbroek int eloop_q_timeout_add_tv(struct eloop *, int, 172*9f20bfa6SDavid van Moolenbroek const struct timespec *, void (*)(void *), void *); 173*9f20bfa6SDavid van Moolenbroek int eloop_q_timeout_add_sec(struct eloop *, int, 174*9f20bfa6SDavid van Moolenbroek time_t, void (*)(void *), void *); 175*9f20bfa6SDavid van Moolenbroek int eloop_q_timeout_add_msec(struct eloop *, int, 176*9f20bfa6SDavid van Moolenbroek long, void (*)(void *), void *); 177*9f20bfa6SDavid van Moolenbroek void eloop_q_timeout_delete(struct eloop *, int, void (*)(void *), void *); 178*9f20bfa6SDavid van Moolenbroek 179*9f20bfa6SDavid van Moolenbroek int eloop_signal_set_cb(struct eloop *, const int *, size_t, 180*9f20bfa6SDavid van Moolenbroek void (*)(int, void *), void *); 181*9f20bfa6SDavid van Moolenbroek int eloop_signal_mask(struct eloop *, sigset_t *oldset); 182*9f20bfa6SDavid van Moolenbroek 183*9f20bfa6SDavid van Moolenbroek struct eloop * eloop_new(void); 184*9f20bfa6SDavid van Moolenbroek #if defined(HAVE_KQUEUE) || defined(HAVE_EPOLL) 185*9f20bfa6SDavid van Moolenbroek int eloop_requeue(struct eloop *); 186*9f20bfa6SDavid van Moolenbroek #else 187*9f20bfa6SDavid van Moolenbroek #define eloop_requeue(eloop) (0) 188*9f20bfa6SDavid van Moolenbroek #endif 189*9f20bfa6SDavid van Moolenbroek void eloop_free(struct eloop *); 190*9f20bfa6SDavid van Moolenbroek void eloop_exit(struct eloop *, int); 191*9f20bfa6SDavid van Moolenbroek int eloop_start(struct eloop *, sigset_t *); 192*9f20bfa6SDavid van Moolenbroek 193*9f20bfa6SDavid van Moolenbroek #endif 194