File:  [ELWIX - Embedded LightWeight unIX -] / embedaddon / lighttpd / src / fdevent.c
Revision 1.1.1.2 (vendor branch): download - view: text, annotated - select for diffs - revision graph
Sun Jun 15 20:20:06 2014 UTC (10 years ago) by misho
Branches: lighttpd, MAIN
CVS tags: v1_4_35p0, v1_4_35, HEAD
lighttpd 1.4.35

    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>