1 #include "base.h"
2 #include "log.h"
3
4 #include <sys/types.h>
5
6 #include <unistd.h>
7 #include <stdlib.h>
8 #include <string.h>
9 #include <errno.h>
10 #include <stdio.h>
11 #include <fcntl.h>
12 #include <assert.h>
13 /*----------------------------------------------------------------------------*/
14 fdevents *
fdevent_init(server * srv,size_t maxfds,fdevent_handler_t type)15 fdevent_init(server *srv, size_t maxfds, fdevent_handler_t type)
16 {
17 fdevents *ev;
18
19 ev = calloc(1, sizeof(*ev));
20 ev->srv = srv;
21 ev->fdarray = calloc(maxfds, sizeof(*ev->fdarray));
22 ev->maxfds = maxfds;
23
24 switch(type) {
25 case FDEVENT_HANDLER_POLL:
26 if (0 != fdevent_poll_init(ev)) {
27 log_error_write(ev->srv, __FILE__, __LINE__, "S",
28 "event-handler poll failed");
29
30 return NULL;
31 }
32 return ev;
33 case FDEVENT_HANDLER_SELECT:
34 if (0 != fdevent_select_init(ev)) {
35 log_error_write(ev->srv, __FILE__, __LINE__, "S",
36 "event-handler select failed");
37 return NULL;
38 }
39 return ev;
40 case FDEVENT_HANDLER_LINUX_SYSEPOLL:
41 if (0 != fdevent_linux_sysepoll_init(ev)) {
42 log_error_write(ev->srv, __FILE__, __LINE__, "S",
43 "event-handler linux-sysepoll failed, try to set server.event-handler = \"poll\" or \"select\"");
44 return NULL;
45 }
46 return ev;
47 case FDEVENT_HANDLER_LIBMTCP:
48 if (0 != fdevent_libmtcp_epoll_init(ev)) {
49 log_error_write(ev->srv, __FILE__, __LINE__, "S",
50 "event-handler linux-sysepoll failed, try to set server.event-handler = \"poll\" or \"select\"");
51 return NULL;
52 }
53 return ev;
54 case FDEVENT_HANDLER_SOLARIS_DEVPOLL:
55 if (0 != fdevent_solaris_devpoll_init(ev)) {
56 log_error_write(ev->srv, __FILE__, __LINE__, "S",
57 "event-handler solaris-devpoll failed, try to set server.event-handler = \"poll\" or \"select\"");
58 return NULL;
59 }
60 return ev;
61 case FDEVENT_HANDLER_SOLARIS_PORT:
62 if (0 != fdevent_solaris_port_init(ev)) {
63 log_error_write(ev->srv, __FILE__, __LINE__, "S",
64 "event-handler solaris-eventports failed, try to set server.event-handler = \"poll\" or \"select\"");
65 return NULL;
66 }
67 return ev;
68 case FDEVENT_HANDLER_FREEBSD_KQUEUE:
69 if (0 != fdevent_freebsd_kqueue_init(ev)) {
70 log_error_write(ev->srv, __FILE__, __LINE__, "S",
71 "event-handler freebsd-kqueue failed, try to set server.event-handler = \"poll\" or \"select\"");
72 return NULL;
73 }
74 return ev;
75 case FDEVENT_HANDLER_LIBEV:
76 if (0 != fdevent_libev_init(ev)) {
77 log_error_write(ev->srv, __FILE__, __LINE__, "S",
78 "event-handler libev failed, try to set server.event-handler = \"poll\" or \"select\"");
79 return NULL;
80 }
81 return ev;
82 case FDEVENT_HANDLER_UNSET:
83 break;
84 }
85
86 log_error_write(ev->srv, __FILE__, __LINE__, "S",
87 "event-handler is unknown, try to set server.event-handler = \"poll\" or \"select\"");
88 return NULL;
89 }
90 /*----------------------------------------------------------------------------*/
91 void
fdevent_free(fdevents * ev)92 fdevent_free(fdevents *ev)
93 {
94 size_t i;
95 if (!ev) return;
96
97 if (ev->free) ev->free(ev);
98
99 for (i = 0; i < ev->maxfds; i++) {
100 if (ev->fdarray[i]) free(ev->fdarray[i]);
101 }
102
103 free(ev->fdarray);
104 free(ev);
105 }
106 /*----------------------------------------------------------------------------*/
107 int
fdevent_reset(fdevents * ev)108 fdevent_reset(fdevents *ev)
109 {
110 if (ev->reset) return ev->reset(ev);
111
112 return 0;
113 }
114 /*----------------------------------------------------------------------------*/
115 static fdnode *
fdnode_init(void)116 fdnode_init(void)
117 {
118 fdnode *fdn;
119
120 fdn = calloc(1, sizeof(*fdn));
121 fdn->fd = -1;
122 return fdn;
123 }
124 /*----------------------------------------------------------------------------*/
125 static void
fdnode_free(fdnode * fdn)126 fdnode_free(fdnode *fdn)
127 {
128 free(fdn);
129 }
130 /*----------------------------------------------------------------------------*/
131 int
fdevent_register(fdevents * ev,int fd,fdevent_handler handler,void * ctx)132 fdevent_register(fdevents *ev, int fd, fdevent_handler handler, void *ctx)
133 {
134 fdnode *fdn;
135
136 fdn = fdnode_init();
137 fdn->handler = handler;
138 fdn->fd = fd;
139 fdn->ctx = ctx;
140 fdn->handler_ctx = NULL;
141 fdn->events = 0;
142
143 ev->fdarray[fd] = fdn;
144
145 return 0;
146 }
147 /*----------------------------------------------------------------------------*/
148 int
fdevent_unregister(fdevents * ev,int fd)149 fdevent_unregister(fdevents *ev, int fd)
150 {
151 fdnode *fdn;
152
153 if (!ev) return 0;
154 fdn = ev->fdarray[fd];
155
156 assert(fdn->events == 0);
157
158 fdnode_free(fdn);
159
160 ev->fdarray[fd] = NULL;
161
162 return 0;
163 }
164 /*----------------------------------------------------------------------------*/
165 int
fdevent_event_del(fdevents * ev,int * fde_ndx,int fd)166 fdevent_event_del(fdevents *ev, int *fde_ndx, int fd)
167 {
168 int fde = fde_ndx ? *fde_ndx : -1;
169
170 if (NULL == ev->fdarray[fd]) return 0;
171
172 if (ev->event_del) fde = ev->event_del(ev, fde, fd);
173 ev->fdarray[fd]->events = 0;
174
175 if (fde_ndx) *fde_ndx = fde;
176
177 return 0;
178 }
179 /*----------------------------------------------------------------------------*/
180 int
fdevent_event_set(fdevents * ev,int * fde_ndx,int fd,int events)181 fdevent_event_set(fdevents *ev, int *fde_ndx, int fd, int events)
182 {
183 int fde = fde_ndx ? *fde_ndx : -1;
184
185 if (ev->event_set) fde = ev->event_set(ev, fde, fd, events);
186 ev->fdarray[fd]->events = events;
187
188 if (fde_ndx) *fde_ndx = fde;
189
190 return 0;
191 }
192 /*----------------------------------------------------------------------------*/
193 int
fdevent_poll(fdevents * ev,int timeout_ms)194 fdevent_poll(fdevents *ev, int timeout_ms)
195 {
196 if (ev->poll == NULL) SEGFAULT();
197 return ev->poll(ev, timeout_ms);
198 }
199 /*----------------------------------------------------------------------------*/
200 int
fdevent_event_get_revent(fdevents * ev,size_t ndx)201 fdevent_event_get_revent(fdevents *ev, size_t ndx)
202 {
203 if (ev->event_get_revent == NULL) SEGFAULT();
204
205 return ev->event_get_revent(ev, ndx);
206 }
207 /*----------------------------------------------------------------------------*/
208 int
fdevent_event_get_fd(fdevents * ev,size_t ndx)209 fdevent_event_get_fd(fdevents *ev, size_t ndx)
210 {
211 if (ev->event_get_fd == NULL) SEGFAULT();
212
213 return ev->event_get_fd(ev, ndx);
214 }
215 /*----------------------------------------------------------------------------*/
216 fdevent_handler
fdevent_get_handler(fdevents * ev,int fd)217 fdevent_get_handler(fdevents *ev, int fd)
218 {
219 if (ev->fdarray[fd] == NULL) SEGFAULT();
220 if (ev->fdarray[fd]->fd != fd) SEGFAULT();
221
222 return ev->fdarray[fd]->handler;
223 }
224 /*----------------------------------------------------------------------------*/
225 void *
fdevent_get_context(fdevents * ev,int fd)226 fdevent_get_context(fdevents *ev, int fd)
227 {
228 if (ev->fdarray[fd] == NULL) SEGFAULT();
229 if (ev->fdarray[fd]->fd != fd) SEGFAULT();
230
231 return ev->fdarray[fd]->ctx;
232 }
233 /*----------------------------------------------------------------------------*/
234 int
fdevent_fcntl_set(fdevents * ev,int fd)235 fdevent_fcntl_set(fdevents *ev, int fd)
236 {
237 #ifdef HAVE_LIBMTCP
238 /* Bailing out early, socket set as non-blocking */
239 if (mtcp_setsock_nonblock(ev->srv->mctx, fd) < 0) {
240 fprintf(stderr, "Failed to set socket (%d) in nonblocking mode.\n",
241 fd);
242 exit(EXIT_FAILURE);
243 }
244 return 0;
245 #endif
246 #ifdef FD_CLOEXEC
247 /* close fd on exec (cgi) */
248 fcntl(fd, F_SETFD, FD_CLOEXEC);
249 #endif
250 if ((ev) && (ev->fcntl_set)) return ev->fcntl_set(ev, fd);
251 #ifdef O_NONBLOCK
252 return fcntl(fd, F_SETFL, O_NONBLOCK | O_RDWR);
253 #else
254 return 0;
255 #endif
256 }
257 /*----------------------------------------------------------------------------*/
258 int
fdevent_event_next_fdndx(fdevents * ev,int ndx)259 fdevent_event_next_fdndx(fdevents *ev, int ndx)
260 {
261 if (ev->event_next_fdndx) return ev->event_next_fdndx(ev, ndx);
262
263 return -1;
264 }
265 /*----------------------------------------------------------------------------*/
266