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:
15: fdevents *fdevent_init(server *srv, size_t maxfds, fdevent_handler_t type) {
16: fdevents *ev;
17:
18: ev = calloc(1, sizeof(*ev));
19: ev->srv = srv;
20: ev->fdarray = calloc(maxfds, sizeof(*ev->fdarray));
21: ev->maxfds = maxfds;
22:
23: switch(type) {
24: case FDEVENT_HANDLER_POLL:
25: if (0 != fdevent_poll_init(ev)) {
26: log_error_write(srv, __FILE__, __LINE__, "S",
27: "event-handler poll failed");
28: goto error;
29: }
30: return ev;
31: case FDEVENT_HANDLER_SELECT:
32: if (0 != fdevent_select_init(ev)) {
33: log_error_write(srv, __FILE__, __LINE__, "S",
34: "event-handler select failed");
35: goto error;
36: }
37: return ev;
38: case FDEVENT_HANDLER_LINUX_SYSEPOLL:
39: if (0 != fdevent_linux_sysepoll_init(ev)) {
40: log_error_write(srv, __FILE__, __LINE__, "S",
41: "event-handler linux-sysepoll failed, try to set server.event-handler = \"poll\" or \"select\"");
42: goto error;
43: }
44: return ev;
45: case FDEVENT_HANDLER_SOLARIS_DEVPOLL:
46: if (0 != fdevent_solaris_devpoll_init(ev)) {
47: log_error_write(srv, __FILE__, __LINE__, "S",
48: "event-handler solaris-devpoll failed, try to set server.event-handler = \"poll\" or \"select\"");
49: goto error;
50: }
51: return ev;
52: case FDEVENT_HANDLER_SOLARIS_PORT:
53: if (0 != fdevent_solaris_port_init(ev)) {
54: log_error_write(srv, __FILE__, __LINE__, "S",
55: "event-handler solaris-eventports failed, try to set server.event-handler = \"poll\" or \"select\"");
56: goto error;
57: }
58: return ev;
59: case FDEVENT_HANDLER_FREEBSD_KQUEUE:
60: if (0 != fdevent_freebsd_kqueue_init(ev)) {
61: log_error_write(srv, __FILE__, __LINE__, "S",
62: "event-handler freebsd-kqueue failed, try to set server.event-handler = \"poll\" or \"select\"");
63: goto error;
64: }
65: return ev;
66: case FDEVENT_HANDLER_LIBEV:
67: if (0 != fdevent_libev_init(ev)) {
68: log_error_write(srv, __FILE__, __LINE__, "S",
69: "event-handler libev failed, try to set server.event-handler = \"poll\" or \"select\"");
70: goto error;
71: }
72: return ev;
73: case FDEVENT_HANDLER_UNSET:
74: break;
75: }
76:
77: error:
78: free(ev->fdarray);
79: free(ev);
80:
81: log_error_write(srv, __FILE__, __LINE__, "S",
82: "event-handler is unknown, try to set server.event-handler = \"poll\" or \"select\"");
83: return NULL;
84: }
85:
86: void fdevent_free(fdevents *ev) {
87: size_t i;
88: if (!ev) return;
89:
90: if (ev->free) ev->free(ev);
91:
92: for (i = 0; i < ev->maxfds; i++) {
93: if (ev->fdarray[i]) free(ev->fdarray[i]);
94: }
95:
96: free(ev->fdarray);
97: free(ev);
98: }
99:
100: int fdevent_reset(fdevents *ev) {
101: if (ev->reset) return ev->reset(ev);
102:
103: return 0;
104: }
105:
106: static fdnode *fdnode_init(void) {
107: fdnode *fdn;
108:
109: fdn = calloc(1, sizeof(*fdn));
110: fdn->fd = -1;
111: return fdn;
112: }
113:
114: static void fdnode_free(fdnode *fdn) {
115: free(fdn);
116: }
117:
118: int fdevent_register(fdevents *ev, int fd, fdevent_handler handler, void *ctx) {
119: fdnode *fdn;
120:
121: fdn = fdnode_init();
122: fdn->handler = handler;
123: fdn->fd = fd;
124: fdn->ctx = ctx;
125: fdn->handler_ctx = NULL;
126: fdn->events = 0;
127:
128: ev->fdarray[fd] = fdn;
129:
130: return 0;
131: }
132:
133: int fdevent_unregister(fdevents *ev, int fd) {
134: fdnode *fdn;
135:
136: if (!ev) return 0;
137: fdn = ev->fdarray[fd];
138:
139: force_assert(fdn->events == 0);
140:
141: fdnode_free(fdn);
142:
143: ev->fdarray[fd] = NULL;
144:
145: return 0;
146: }
147:
148: int fdevent_event_del(fdevents *ev, int *fde_ndx, int fd) {
149: int fde = fde_ndx ? *fde_ndx : -1;
150:
151: if (NULL == ev->fdarray[fd]) return 0;
152:
153: if (ev->event_del) fde = ev->event_del(ev, fde, fd);
154: ev->fdarray[fd]->events = 0;
155:
156: if (fde_ndx) *fde_ndx = fde;
157:
158: return 0;
159: }
160:
161: int fdevent_event_set(fdevents *ev, int *fde_ndx, int fd, int events) {
162: int fde = fde_ndx ? *fde_ndx : -1;
163:
164: if (ev->event_set) fde = ev->event_set(ev, fde, fd, events);
165: ev->fdarray[fd]->events = events;
166:
167: if (fde_ndx) *fde_ndx = fde;
168:
169: return 0;
170: }
171:
172: int fdevent_poll(fdevents *ev, int timeout_ms) {
173: if (ev->poll == NULL) SEGFAULT();
174: return ev->poll(ev, timeout_ms);
175: }
176:
177: int fdevent_event_get_revent(fdevents *ev, size_t ndx) {
178: if (ev->event_get_revent == NULL) SEGFAULT();
179:
180: return ev->event_get_revent(ev, ndx);
181: }
182:
183: int fdevent_event_get_fd(fdevents *ev, size_t ndx) {
184: if (ev->event_get_fd == NULL) SEGFAULT();
185:
186: return ev->event_get_fd(ev, ndx);
187: }
188:
189: fdevent_handler fdevent_get_handler(fdevents *ev, int fd) {
190: if (ev->fdarray[fd] == NULL) SEGFAULT();
191: if (ev->fdarray[fd]->fd != fd) SEGFAULT();
192:
193: return ev->fdarray[fd]->handler;
194: }
195:
196: void * fdevent_get_context(fdevents *ev, int fd) {
197: if (ev->fdarray[fd] == NULL) SEGFAULT();
198: if (ev->fdarray[fd]->fd != fd) SEGFAULT();
199:
200: return ev->fdarray[fd]->ctx;
201: }
202:
203: void fd_close_on_exec(int fd) {
204: #ifdef FD_CLOEXEC
205: if (fd < 0) return;
206: force_assert(-1 != fcntl(fd, F_SETFD, FD_CLOEXEC));
207: #else
208: UNUSED(fd);
209: #endif
210: }
211:
212: int fdevent_fcntl_set(fdevents *ev, int fd) {
213: fd_close_on_exec(fd);
214: if ((ev) && (ev->fcntl_set)) return ev->fcntl_set(ev, fd);
215: #ifdef O_NONBLOCK
216: return fcntl(fd, F_SETFL, O_NONBLOCK | O_RDWR);
217: #else
218: return 0;
219: #endif
220: }
221:
222:
223: int fdevent_event_next_fdndx(fdevents *ev, int ndx) {
224: if (ev->event_next_fdndx) return ev->event_next_fdndx(ev, ndx);
225:
226: return -1;
227: }
228:
FreeBSD-CVSweb <freebsd-cvsweb@FreeBSD.org>