diff options
Diffstat (limited to 'src/event/modules/ngx_poll_module.c')
| -rw-r--r-- | src/event/modules/ngx_poll_module.c | 258 |
1 files changed, 258 insertions, 0 deletions
diff --git a/src/event/modules/ngx_poll_module.c b/src/event/modules/ngx_poll_module.c new file mode 100644 index 000000000..e10e61538 --- /dev/null +++ b/src/event/modules/ngx_poll_module.c @@ -0,0 +1,258 @@ + +#include <ngx_config.h> +#include <ngx_core.h> +#include <ngx_types.h> +#include <ngx_log.h> +#include <ngx_time.h> +#include <ngx_connection.h> +#include <ngx_event.h> +#include <ngx_poll_module.h> + + +/* should be per-thread */ +static struct pollfd *event_list; +static int nevents; + +static ngx_event_t **event_index; +static ngx_event_t timer_queue; +/* */ + +int ngx_poll_init(int max_connections, ngx_log_t *log) +{ + ngx_test_null(event_list, + ngx_alloc(sizeof(struct pollfd) * max_connections, log), + NGX_ERROR); + + ngx_test_null(event_index, + ngx_alloc(sizeof(ngx_event_t *) * max_connections, log), + NGX_ERROR); + + nevents = 0; + + timer_queue.timer_prev = &timer_queue; + timer_queue.timer_next = &timer_queue; + + ngx_event_actions.add = ngx_poll_add_event; + ngx_event_actions.del = ngx_poll_del_event; + ngx_event_actions.timer = ngx_poll_add_timer; + ngx_event_actions.process = ngx_poll_process_events; + + return NGX_OK; +} + +int ngx_poll_add_event(ngx_event_t *ev, int event, u_int flags) +{ + ngx_event_t *e; + ngx_connection_t *c; + + c = (ngx_connection_t *) ev->data; + + ev->oneshot = (flags & NGX_ONESHOT_EVENT) ? 1: 0; + + if (event == NGX_READ_EVENT) { + e = c->write; +#if (NGX_READ_EVENT != POLLIN) + event = POLLIN; +#endif + + } else { + e = c->read; +#if (NGX_WRITE_EVENT != POLLOUT) + event = POLLOUT; +#endif + } + + ngx_log_debug(ev->log, "poll fd:%d event:%d" _ c->fd _ event); + + if (e == NULL || e->index == NGX_INVALID_INDEX) { + event_list[nevents].fd = c->fd; + event_list[nevents].events = event; + event_list[nevents].revents = 0; + + event_index[nevents] = ev; + ev->index = nevents; + nevents++; + + } else { + event_list[e->index].events |= event; + ev->index = e->index; + } + + return NGX_OK; +} + +int ngx_poll_del_event(ngx_event_t *ev, int event, u_int flags) +{ + ngx_event_t *e; + ngx_connection_t *c; + + c = (ngx_connection_t *) ev->data; + + if (ev->index == NGX_INVALID_INDEX) + return NGX_OK; + + if (event == NGX_READ_EVENT) { + e = c->write; +#if (NGX_READ_EVENT != POLLIN) + event = POLLIN; +#endif + + } else { + e = c->read; +#if (NGX_WRITE_EVENT != POLLOUT) + event = POLLOUT; +#endif + } + + ngx_log_debug(c->log, "del event: %d, %d" _ c->fd _ event); + + if (e == NULL || e->index == NGX_INVALID_INDEX) { + if (ev->index < --nevents) { + event_index[ev->index] = event_index[nevents]; + event_index[ev->index]->index = ev->index; + } + + } else { + event_list[e->index].events &= ~event; + } + + ev->index = NGX_INVALID_INDEX; + + return NGX_OK; +} + +int ngx_poll_process_events(ngx_log_t *log) +{ + int i, ready, found; + u_int timer, delta; + ngx_event_t *ev; + ngx_connection_t *c; + + if (timer_queue.timer_next != &timer_queue) { + timer = timer_queue.timer_next->timer_delta; + delta = ngx_msec(); + + } else { + timer = INFTIM; + delta = 0; + } + +#if 1 + /* DEBUG */ + for (i = 0; i < nevents; i++) { + ngx_log_debug(log, "poll: %d, %d" _ + event_list[i].fd _ event_list[i].events); + } +#endif + + ngx_log_debug(log, "poll timer: %d" _ timer); + + if ((ready = poll(event_list, nevents, timer)) == -1) { + ngx_log_error(NGX_LOG_ALERT, log, ngx_errno, "poll() failed"); + return NGX_ERROR; + } + + ngx_log_debug(log, "poll ready %d" _ ready); + + if (timer != INFTIM) { + delta = ngx_msec() - delta; + + } else { + ngx_assert((ready != 0), return NGX_ERROR, log, + "poll() returns no events without timeout"); + } + + ngx_log_debug(log, "poll timer: %d, delta: %d" _ timer _ delta); + + if (timer != INFTIM) { + if (delta >= timer) { + for ( ;; ) { + ev = timer_queue.timer_next; + + if (ev == &timer_queue || delta < ev->timer_delta) { + break; + } + + delta -= ev->timer_delta; + ngx_del_timer(ev); + ev->timedout = 1; + + if (ev->event_handler(ev) == NGX_ERROR) { + ev->close_handler(ev); + } + } + + } else { + timer_queue.timer_next->timer_delta -= delta; + } + } + + for (i = 0; ready; i++) { + c = &ngx_connections[event_list[i].fd]; + + ngx_log_debug(log, "poll: fd:%d, ev:%d, rev:%d" _ + event_list[i].fd _ + event_list[i].events _ event_list[i].revents); + + found = 0; + + if (event_list[i].revents & POLLIN) { + found = 1; + c->read->ready = 1; + + if (c->read->oneshot) { + ngx_del_timer(c->read); + ngx_select_del_event(c->read, NGX_READ_EVENT, 0); + } + + if (c->read->event_handler(c->read) == NGX_ERROR) { + c->read->close_handler(c->read); + } + } + + if (event_list[i].revents & POLLOUT) { + found = 1; + c->write->ready = 1; + + if (c->write->oneshot) { + ngx_del_timer(c->write); + ngx_select_del_event(c->write, NGX_WRITE_EVENT, 0); + } + + if (c->write->event_handler(c->write) == NGX_ERROR) { + c->write->close_handler(c->write); + } + } + + if (event_list[i].revents & (POLLERR|POLLHUP|POLLNVAL)) { + found = 1; + ngx_log_error(NGX_LOG_ERR, log, ngx_errno, + "poll() error on %d:%d", + event_list[i].fd, event_list[i].revents); + } + + if (found) { + ready--; + } + } + + return NGX_OK; +} + +void ngx_poll_add_timer(ngx_event_t *ev, ngx_msec_t timer) +{ + ngx_event_t *e; + + for (e = timer_queue.timer_next; + e != &timer_queue && timer > e->timer_delta; + e = e->timer_next) + timer -= e->timer_delta; + + ev->timer_delta = timer; + + ev->timer_next = e; + ev->timer_prev = e->timer_prev; + + e->timer_prev->timer_next = ev; + e->timer_prev = ev; +} |