| /* | |
| * Copyright (C) Igor Sysoev | |
| */ | |
| #include <ngx_config.h> | |
| #include <ngx_core.h> | |
| #include <ngx_event.h> | |
| #if (NGX_TEST_BUILD_DEVPOLL) | |
| /* Solaris declarations */ | |
| #define POLLREMOVE 0x0800 | |
| #define DP_POLL 0xD001 | |
| #define DP_ISPOLLED 0xD002 | |
| struct dvpoll { | |
| struct pollfd *dp_fds; | |
| int dp_nfds; | |
| int dp_timeout; | |
| }; | |
| #endif | |
| typedef struct { | |
| ngx_uint_t changes; | |
| ngx_uint_t events; | |
| } ngx_devpoll_conf_t; | |
| static ngx_int_t ngx_devpoll_init(ngx_cycle_t *cycle, ngx_msec_t timer); | |
| static void ngx_devpoll_done(ngx_cycle_t *cycle); | |
| static ngx_int_t ngx_devpoll_add_event(ngx_event_t *ev, ngx_int_t event, | |
| ngx_uint_t flags); | |
| static ngx_int_t ngx_devpoll_del_event(ngx_event_t *ev, ngx_int_t event, | |
| ngx_uint_t flags); | |
| static ngx_int_t ngx_devpoll_set_event(ngx_event_t *ev, ngx_int_t event, | |
| ngx_uint_t flags); | |
| static ngx_int_t ngx_devpoll_process_events(ngx_cycle_t *cycle, | |
| ngx_msec_t timer, ngx_uint_t flags); | |
| static void *ngx_devpoll_create_conf(ngx_cycle_t *cycle); | |
| static char *ngx_devpoll_init_conf(ngx_cycle_t *cycle, void *conf); | |
| static int dp = -1; | |
| static struct pollfd *change_list, *event_list; | |
| static ngx_uint_t nchanges, max_changes, nevents; | |
| static ngx_event_t **change_index; | |
| static ngx_str_t devpoll_name = ngx_string("/dev/poll"); | |
| static ngx_command_t ngx_devpoll_commands[] = { | |
| { ngx_string("devpoll_changes"), | |
| NGX_EVENT_CONF|NGX_CONF_TAKE1, | |
| ngx_conf_set_num_slot, | |
| 0, | |
| offsetof(ngx_devpoll_conf_t, changes), | |
| NULL }, | |
| { ngx_string("devpoll_events"), | |
| NGX_EVENT_CONF|NGX_CONF_TAKE1, | |
| ngx_conf_set_num_slot, | |
| 0, | |
| offsetof(ngx_devpoll_conf_t, events), | |
| NULL }, | |
| ngx_null_command | |
| }; | |
| ngx_event_module_t ngx_devpoll_module_ctx = { | |
| &devpoll_name, | |
| ngx_devpoll_create_conf, /* create configuration */ | |
| ngx_devpoll_init_conf, /* init configuration */ | |
| { | |
| ngx_devpoll_add_event, /* add an event */ | |
| ngx_devpoll_del_event, /* delete an event */ | |
| ngx_devpoll_add_event, /* enable an event */ | |
| ngx_devpoll_del_event, /* disable an event */ | |
| NULL, /* add an connection */ | |
| NULL, /* delete an connection */ | |
| NULL, /* process the changes */ | |
| ngx_devpoll_process_events, /* process the events */ | |
| ngx_devpoll_init, /* init the events */ | |
| ngx_devpoll_done, /* done the events */ | |
| } | |
| }; | |
| ngx_module_t ngx_devpoll_module = { | |
| NGX_MODULE_V1, | |
| &ngx_devpoll_module_ctx, /* module context */ | |
| ngx_devpoll_commands, /* module directives */ | |
| NGX_EVENT_MODULE, /* module type */ | |
| NULL, /* init master */ | |
| NULL, /* init module */ | |
| NULL, /* init process */ | |
| NULL, /* init thread */ | |
| NULL, /* exit thread */ | |
| NULL, /* exit process */ | |
| NULL, /* exit master */ | |
| NGX_MODULE_V1_PADDING | |
| }; | |
| static ngx_int_t | |
| ngx_devpoll_init(ngx_cycle_t *cycle, ngx_msec_t timer) | |
| { | |
| size_t n; | |
| ngx_devpoll_conf_t *dpcf; | |
| dpcf = ngx_event_get_conf(cycle->conf_ctx, ngx_devpoll_module); | |
| if (dp == -1) { | |
| dp = open("/dev/poll", O_RDWR); | |
| if (dp == -1) { | |
| ngx_log_error(NGX_LOG_EMERG, cycle->log, ngx_errno, | |
| "open(/dev/poll) failed"); | |
| return NGX_ERROR; | |
| } | |
| } | |
| if (max_changes < dpcf->changes) { | |
| if (nchanges) { | |
| n = nchanges * sizeof(struct pollfd); | |
| if (write(dp, change_list, n) != (ssize_t) n) { | |
| ngx_log_error(NGX_LOG_ALERT, cycle->log, ngx_errno, | |
| "write(/dev/poll) failed"); | |
| return NGX_ERROR; | |
| } | |
| nchanges = 0; | |
| } | |
| if (change_list) { | |
| ngx_free(change_list); | |
| } | |
| change_list = ngx_alloc(sizeof(struct pollfd) * dpcf->changes, | |
| cycle->log); | |
| if (change_list == NULL) { | |
| return NGX_ERROR; | |
| } | |
| if (change_index) { | |
| ngx_free(change_index); | |
| } | |
| change_index = ngx_alloc(sizeof(ngx_event_t *) * dpcf->changes, | |
| cycle->log); | |
| if (change_index == NULL) { | |
| return NGX_ERROR; | |
| } | |
| } | |
| max_changes = dpcf->changes; | |
| if (nevents < dpcf->events) { | |
| if (event_list) { | |
| ngx_free(event_list); | |
| } | |
| event_list = ngx_alloc(sizeof(struct pollfd) * dpcf->events, | |
| cycle->log); | |
| if (event_list == NULL) { | |
| return NGX_ERROR; | |
| } | |
| } | |
| nevents = dpcf->events; | |
| ngx_io = ngx_os_io; | |
| ngx_event_actions = ngx_devpoll_module_ctx.actions; | |
| ngx_event_flags = NGX_USE_LEVEL_EVENT|NGX_USE_FD_EVENT; | |
| return NGX_OK; | |
| } | |
| static void | |
| ngx_devpoll_done(ngx_cycle_t *cycle) | |
| { | |
| if (close(dp) == -1) { | |
| ngx_log_error(NGX_LOG_ALERT, cycle->log, ngx_errno, | |
| "close(/dev/poll) failed"); | |
| } | |
| dp = -1; | |
| ngx_free(change_list); | |
| ngx_free(event_list); | |
| ngx_free(change_index); | |
| change_list = NULL; | |
| event_list = NULL; | |
| change_index = NULL; | |
| max_changes = 0; | |
| nchanges = 0; | |
| nevents = 0; | |
| } | |
| static ngx_int_t | |
| ngx_devpoll_add_event(ngx_event_t *ev, ngx_int_t event, ngx_uint_t flags) | |
| { | |
| #if (NGX_DEBUG) | |
| ngx_connection_t *c; | |
| #endif | |
| #if (NGX_READ_EVENT != POLLIN) | |
| event = (event == NGX_READ_EVENT) ? POLLIN : POLLOUT; | |
| #endif | |
| #if (NGX_DEBUG) | |
| c = ev->data; | |
| ngx_log_debug2(NGX_LOG_DEBUG_EVENT, ev->log, 0, | |
| "devpoll add event: fd:%d ev:%04Xi", c->fd, event); | |
| #endif | |
| ev->active = 1; | |
| return ngx_devpoll_set_event(ev, event, 0); | |
| } | |
| static ngx_int_t | |
| ngx_devpoll_del_event(ngx_event_t *ev, ngx_int_t event, ngx_uint_t flags) | |
| { | |
| ngx_event_t *e; | |
| ngx_connection_t *c; | |
| c = ev->data; | |
| #if (NGX_READ_EVENT != POLLIN) | |
| event = (event == NGX_READ_EVENT) ? POLLIN : POLLOUT; | |
| #endif | |
| ngx_log_debug2(NGX_LOG_DEBUG_EVENT, ev->log, 0, | |
| "devpoll del event: fd:%d ev:%04Xi", c->fd, event); | |
| if (ngx_devpoll_set_event(ev, POLLREMOVE, flags) == NGX_ERROR) { | |
| return NGX_ERROR; | |
| } | |
| ev->active = 0; | |
| if (flags & NGX_CLOSE_EVENT) { | |
| e = (event == POLLIN) ? c->write : c->read; | |
| if (e) { | |
| e->active = 0; | |
| } | |
| return NGX_OK; | |
| } | |
| /* restore the pair event if it exists */ | |
| if (event == POLLIN) { | |
| e = c->write; | |
| event = POLLOUT; | |
| } else { | |
| e = c->read; | |
| event = POLLIN; | |
| } | |
| if (e && e->active) { | |
| return ngx_devpoll_set_event(e, event, 0); | |
| } | |
| return NGX_OK; | |
| } | |
| static ngx_int_t | |
| ngx_devpoll_set_event(ngx_event_t *ev, ngx_int_t event, ngx_uint_t flags) | |
| { | |
| size_t n; | |
| ngx_connection_t *c; | |
| c = ev->data; | |
| ngx_log_debug3(NGX_LOG_DEBUG_EVENT, ev->log, 0, | |
| "devpoll fd:%d ev:%04Xi fl:%04Xi", c->fd, event, flags); | |
| if (nchanges >= max_changes) { | |
| ngx_log_error(NGX_LOG_WARN, ev->log, 0, | |
| "/dev/pool change list is filled up"); | |
| n = nchanges * sizeof(struct pollfd); | |
| if (write(dp, change_list, n) != (ssize_t) n) { | |
| ngx_log_error(NGX_LOG_ALERT, ev->log, ngx_errno, | |
| "write(/dev/poll) failed"); | |
| return NGX_ERROR; | |
| } | |
| nchanges = 0; | |
| } | |
| change_list[nchanges].fd = c->fd; | |
| change_list[nchanges].events = (short) event; | |
| change_list[nchanges].revents = 0; | |
| change_index[nchanges] = ev; | |
| ev->index = nchanges; | |
| nchanges++; | |
| if (flags & NGX_CLOSE_EVENT) { | |
| n = nchanges * sizeof(struct pollfd); | |
| if (write(dp, change_list, n) != (ssize_t) n) { | |
| ngx_log_error(NGX_LOG_ALERT, ev->log, ngx_errno, | |
| "write(/dev/poll) failed"); | |
| return NGX_ERROR; | |
| } | |
| nchanges = 0; | |
| } | |
| return NGX_OK; | |
| } | |
| ngx_int_t | |
| ngx_devpoll_process_events(ngx_cycle_t *cycle, ngx_msec_t timer, | |
| ngx_uint_t flags) | |
| { | |
| int events, revents, rc; | |
| size_t n; | |
| ngx_fd_t fd; | |
| ngx_err_t err; | |
| ngx_int_t i; | |
| ngx_uint_t level; | |
| ngx_event_t *rev, *wev, **queue; | |
| ngx_connection_t *c; | |
| struct pollfd pfd; | |
| struct dvpoll dvp; | |
| /* NGX_TIMER_INFINITE == INFTIM */ | |
| ngx_log_debug1(NGX_LOG_DEBUG_EVENT, cycle->log, 0, | |
| "devpoll timer: %M", timer); | |
| if (nchanges) { | |
| n = nchanges * sizeof(struct pollfd); | |
| if (write(dp, change_list, n) != (ssize_t) n) { | |
| ngx_log_error(NGX_LOG_ALERT, cycle->log, ngx_errno, | |
| "write(/dev/poll) failed"); | |
| return NGX_ERROR; | |
| } | |
| nchanges = 0; | |
| } | |
| dvp.dp_fds = event_list; | |
| dvp.dp_nfds = (int) nevents; | |
| dvp.dp_timeout = timer; | |
| events = ioctl(dp, DP_POLL, &dvp); | |
| if (events == -1) { | |
| err = ngx_errno; | |
| } else { | |
| err = 0; | |
| } | |
| if (flags & NGX_UPDATE_TIME) { | |
| ngx_time_update(0, 0); | |
| } | |
| if (err) { | |
| if (err == NGX_EINTR) { | |
| if (ngx_event_timer_alarm) { | |
| ngx_event_timer_alarm = 0; | |
| return NGX_OK; | |
| } | |
| level = NGX_LOG_INFO; | |
| } else { | |
| level = NGX_LOG_ALERT; | |
| } | |
| ngx_log_error(level, cycle->log, err, "ioctl(DP_POLL) failed"); | |
| return NGX_ERROR; | |
| } | |
| if (events == 0) { | |
| if (timer != NGX_TIMER_INFINITE) { | |
| return NGX_OK; | |
| } | |
| ngx_log_error(NGX_LOG_ALERT, cycle->log, 0, | |
| "ioctl(DP_POLL) returned no events without timeout"); | |
| return NGX_ERROR; | |
| } | |
| ngx_mutex_lock(ngx_posted_events_mutex); | |
| for (i = 0; i < events; i++) { | |
| fd = event_list[i].fd; | |
| revents = event_list[i].revents; | |
| c = ngx_cycle->files[fd]; | |
| if (c == NULL || c->fd == -1) { | |
| pfd.fd = fd; | |
| pfd.events = 0; | |
| pfd.revents = 0; | |
| rc = ioctl(dp, DP_ISPOLLED, &pfd); | |
| switch (rc) { | |
| case -1: | |
| ngx_log_error(NGX_LOG_ALERT, cycle->log, ngx_errno, | |
| "ioctl(DP_ISPOLLED) failed for socket %d, event", | |
| fd, revents); | |
| break; | |
| case 0: | |
| ngx_log_error(NGX_LOG_ALERT, cycle->log, 0, | |
| "phantom event %04Xd for closed and removed socket %d", | |
| revents, fd); | |
| break; | |
| default: | |
| ngx_log_error(NGX_LOG_ALERT, cycle->log, 0, | |
| "unexpected event %04Xd for closed and removed socket %d, ", | |
| "ioctl(DP_ISPOLLED) returned rc:%d, fd:%d, event %04Xd", | |
| revents, fd, rc, pfd.fd, pfd.revents); | |
| pfd.fd = fd; | |
| pfd.events = POLLREMOVE; | |
| pfd.revents = 0; | |
| if (write(dp, &pfd, sizeof(struct pollfd)) | |
| != (ssize_t) sizeof(struct pollfd)) | |
| { | |
| ngx_log_error(NGX_LOG_ALERT, cycle->log, ngx_errno, | |
| "write(/dev/poll) for %d failed, fd"); | |
| } | |
| if (close(fd) == -1) { | |
| ngx_log_error(NGX_LOG_ALERT, cycle->log, ngx_errno, | |
| "close(%d) failed", fd); | |
| } | |
| break; | |
| } | |
| continue; | |
| } | |
| ngx_log_debug3(NGX_LOG_DEBUG_EVENT, cycle->log, 0, | |
| "devpoll: fd:%d, ev:%04Xd, rev:%04Xd", | |
| fd, event_list[i].events, revents); | |
| if (revents & (POLLERR|POLLHUP|POLLNVAL)) { | |
| ngx_log_debug3(NGX_LOG_DEBUG_EVENT, cycle->log, 0, | |
| "ioctl(DP_POLL) error fd:%d ev:%04Xd rev:%04Xd", | |
| fd, event_list[i].events, revents); | |
| } | |
| if (revents & ~(POLLIN|POLLOUT|POLLERR|POLLHUP|POLLNVAL)) { | |
| ngx_log_error(NGX_LOG_ALERT, cycle->log, 0, | |
| "strange ioctl(DP_POLL) events " | |
| "fd:%d ev:%04Xd rev:%04Xd", | |
| fd, event_list[i].events, revents); | |
| } | |
| if ((revents & (POLLERR|POLLHUP|POLLNVAL)) | |
| && (revents & (POLLIN|POLLOUT)) == 0) | |
| { | |
| /* | |
| * if the error events were returned without POLLIN or POLLOUT, | |
| * then add these flags to handle the events at least in one | |
| * active handler | |
| */ | |
| revents |= POLLIN|POLLOUT; | |
| } | |
| rev = c->read; | |
| if ((revents & POLLIN) && rev->active) { | |
| if ((flags & NGX_POST_THREAD_EVENTS) && !rev->accept) { | |
| rev->posted_ready = 1; | |
| } else { | |
| rev->ready = 1; | |
| } | |
| if (flags & NGX_POST_EVENTS) { | |
| queue = (ngx_event_t **) (rev->accept ? | |
| &ngx_posted_accept_events : &ngx_posted_events); | |
| ngx_locked_post_event(rev, queue); | |
| } else { | |
| rev->handler(rev); | |
| } | |
| } | |
| wev = c->write; | |
| if ((revents & POLLOUT) && wev->active) { | |
| if (flags & NGX_POST_THREAD_EVENTS) { | |
| wev->posted_ready = 1; | |
| } else { | |
| wev->ready = 1; | |
| } | |
| if (flags & NGX_POST_EVENTS) { | |
| ngx_locked_post_event(wev, &ngx_posted_events); | |
| } else { | |
| wev->handler(wev); | |
| } | |
| } | |
| } | |
| ngx_mutex_unlock(ngx_posted_events_mutex); | |
| return NGX_OK; | |
| } | |
| static void * | |
| ngx_devpoll_create_conf(ngx_cycle_t *cycle) | |
| { | |
| ngx_devpoll_conf_t *dpcf; | |
| dpcf = ngx_palloc(cycle->pool, sizeof(ngx_devpoll_conf_t)); | |
| if (dpcf == NULL) { | |
| return NGX_CONF_ERROR; | |
| } | |
| dpcf->changes = NGX_CONF_UNSET; | |
| dpcf->events = NGX_CONF_UNSET; | |
| return dpcf; | |
| } | |
| static char * | |
| ngx_devpoll_init_conf(ngx_cycle_t *cycle, void *conf) | |
| { | |
| ngx_devpoll_conf_t *dpcf = conf; | |
| ngx_conf_init_uint_value(dpcf->changes, 32); | |
| ngx_conf_init_uint_value(dpcf->events, 32); | |
| return NGX_CONF_OK; | |
| } |