xref: /freebsd/contrib/unbound/util/mini_event.h (revision 1f474190)
1 /*
2  * mini-event.h - micro implementation of libevent api, using select() only.
3  *
4  * Copyright (c) 2007, NLnet Labs. All rights reserved.
5  *
6  * This software is open source.
7  *
8  * Redistribution and use in source and binary forms, with or without
9  * modification, are permitted provided that the following conditions
10  * are met:
11  *
12  * Redistributions of source code must retain the above copyright notice,
13  * this list of conditions and the following disclaimer.
14  *
15  * Redistributions in binary form must reproduce the above copyright notice,
16  * this list of conditions and the following disclaimer in the documentation
17  * and/or other materials provided with the distribution.
18  *
19  * Neither the name of the NLNET LABS nor the names of its contributors may
20  * be used to endorse or promote products derived from this software without
21  * specific prior written permission.
22  *
23  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
24  * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
25  * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
26  * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
27  * HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
28  * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED
29  * TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
30  * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
31  * LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
32  * NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
33  * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
34  */
35 
36 /**
37  * \file
38  * This file implements part of the event(3) libevent api.
39  * The back end is only select. Max number of fds is limited.
40  * Max number of signals is limited, one handler per signal only.
41  * And one handler per fd.
42  *
43  * Although limited to select() and a max (1024) open fds, it
44  * is efficient:
45  * o dispatch call caches fd_sets to use.
46  * o handler calling takes time ~ to the number of fds.
47  * o timeouts are stored in a redblack tree, sorted, so take log(n).
48  * Timeouts are only accurate to the second (no subsecond accuracy).
49  * To avoid cpu hogging, fractional timeouts are rounded up to a whole second.
50  */
51 
52 #ifndef MINI_EVENT_H
53 #define MINI_EVENT_H
54 
55 #if defined(USE_MINI_EVENT) && !defined(USE_WINSOCK)
56 
57 #include <sys/time.h>
58 
59 #ifndef HAVE_EVENT_BASE_FREE
60 #define HAVE_EVENT_BASE_FREE
61 #endif
62 
63 /* redefine to use our own namespace so that on platforms where
64  * linkers crosslink library-private symbols with other symbols, it works */
65 #define event_init minievent_init
66 #define event_get_version minievent_get_version
67 #define event_get_method minievent_get_method
68 #define event_base_dispatch minievent_base_dispatch
69 #define event_base_loopexit minievent_base_loopexit
70 #define event_base_free minievent_base_free
71 #define event_set minievent_set
72 #define event_base_set minievent_base_set
73 #define event_add minievent_add
74 #define event_del minievent_del
75 #define signal_add minisignal_add
76 #define signal_del minisignal_del
77 
78 /** event timeout */
79 #define EV_TIMEOUT	0x01
80 /** event fd readable */
81 #define EV_READ		0x02
82 /** event fd writable */
83 #define EV_WRITE	0x04
84 /** event signal */
85 #define EV_SIGNAL	0x08
86 /** event must persist */
87 #define EV_PERSIST	0x10
88 
89 /* needs our redblack tree */
90 #include "rbtree.h"
91 
92 /** max number of file descriptors to support */
93 #define MAX_FDS 1024
94 /** max number of signals to support */
95 #define MAX_SIG 32
96 
97 /** event base */
98 struct event_base
99 {
100 	/** sorted by timeout (absolute), ptr */
101 	rbtree_type* times;
102 	/** array of 0 - maxfd of ptr to event for it */
103 	struct event** fds;
104 	/** max fd in use */
105 	int maxfd;
106 	/** capacity - size of the fds array */
107 	int capfd;
108 	/* fdset for read write, for fds ready, and added */
109 	fd_set
110 		/** fds for reading */
111 		reads,
112 		/** fds for writing */
113 		writes,
114 		/** fds determined ready for use */
115 		ready,
116 		/** ready plus newly added events. */
117 		content;
118 	/** array of 0 - maxsig of ptr to event for it */
119 	struct event** signals;
120 	/** if we need to exit */
121 	int need_to_exit;
122 	/** where to store time in seconds */
123 	time_t* time_secs;
124 	/** where to store time in microseconds */
125 	struct timeval* time_tv;
126 };
127 
128 /**
129  * Event structure. Has some of the event elements.
130  */
131 struct event {
132 	/** node in timeout rbtree */
133 	rbnode_type node;
134 	/** is event already added */
135 	int added;
136 
137 	/** event base it belongs to */
138 	struct event_base *ev_base;
139 	/** fd to poll or -1 for timeouts. signal number for sigs. */
140 	int ev_fd;
141 	/** what events this event is interested in, see EV_.. above. */
142 	short ev_events;
143 	/** timeout value */
144 	struct timeval ev_timeout;
145 
146 	/** callback to call: fd, eventbits, userarg */
147 	void (*ev_callback)(int, short, void *arg);
148 	/** callback user arg */
149 	void *ev_arg;
150 };
151 
152 /* function prototypes (some are as they appear in event.h) */
153 /** create event base */
154 void *event_init(time_t* time_secs, struct timeval* time_tv);
155 /** get version */
156 const char *event_get_version(void);
157 /** get polling method, select */
158 const char *event_get_method(void);
159 /** run select in a loop */
160 int event_base_dispatch(struct event_base *);
161 /** exit that loop */
162 int event_base_loopexit(struct event_base *, struct timeval *);
163 /** free event base. Free events yourself */
164 void event_base_free(struct event_base *);
165 /** set content of event */
166 void event_set(struct event *, int, short, void (*)(int, short, void *), void *);
167 /** add event to a base. You *must* call this for every event. */
168 int event_base_set(struct event_base *, struct event *);
169 /** add event to make it active. You may not change it with event_set anymore */
170 int event_add(struct event *, struct timeval *);
171 /** remove event. You may change it again */
172 int event_del(struct event *);
173 
174 /** add a timer */
175 #define evtimer_add(ev, tv)             event_add(ev, tv)
176 /** remove a timer */
177 #define evtimer_del(ev)                 event_del(ev)
178 
179 /* uses different implementation. Cannot mix fd/timeouts and signals inside
180  * the same struct event. create several event structs for that.  */
181 /** install signal handler */
182 int signal_add(struct event *, struct timeval *);
183 /** set signal event contents */
184 #define signal_set(ev, x, cb, arg)      \
185         event_set(ev, x, EV_SIGNAL|EV_PERSIST, cb, arg)
186 /** remove signal handler */
187 int signal_del(struct event *);
188 
189 #endif /* USE_MINI_EVENT and not USE_WINSOCK */
190 
191 /** compare events in tree, based on timevalue, ptr for uniqueness */
192 int mini_ev_cmp(const void* a, const void* b);
193 
194 #endif /* MINI_EVENT_H */
195