Mercurial > hg > nginx-quic
view src/event/modules/ngx_poll_module.c @ 182:3c49eaf3f522
nginx-0.0.1-2003-11-13-09:14:05 import
author | Igor Sysoev <igor@sysoev.ru> |
---|---|
date | Thu, 13 Nov 2003 06:14:05 +0000 |
parents | 9f3a78b06c48 |
children | 2357fa41738a |
line wrap: on
line source
/* * Copyright (C) 2002-2003 Igor Sysoev, http://sysoev.ru */ #include <ngx_config.h> #include <ngx_core.h> #include <ngx_event.h> static int ngx_poll_init(ngx_cycle_t *cycle); static void ngx_poll_done(ngx_cycle_t *cycle); static int ngx_poll_add_event(ngx_event_t *ev, int event, u_int flags); static int ngx_poll_del_event(ngx_event_t *ev, int event, u_int flags); static int ngx_poll_process_events(ngx_log_t *log); static struct pollfd *event_list; static u_int nevents; static ngx_event_t **event_index; static ngx_event_t **ready_index; static ngx_str_t poll_name = ngx_string("poll"); ngx_event_module_t ngx_poll_module_ctx = { &poll_name, NULL, /* create configuration */ NULL, /* init configuration */ { ngx_poll_add_event, /* add an event */ ngx_poll_del_event, /* delete an event */ ngx_poll_add_event, /* enable an event */ ngx_poll_del_event, /* disable an event */ NULL, /* add an connection */ NULL, /* delete an connection */ ngx_poll_process_events, /* process the events */ ngx_poll_init, /* init the events */ ngx_poll_done /* done the events */ } }; ngx_module_t ngx_poll_module = { NGX_MODULE, &ngx_poll_module_ctx, /* module context */ NULL, /* module directives */ NGX_EVENT_MODULE, /* module type */ NULL, /* init module */ NULL /* init child */ }; static int ngx_poll_init(ngx_cycle_t *cycle) { struct pollfd *list; ngx_event_t **index; if (event_list == NULL) { nevents = 0; } if (cycle->old_cycle == NULL || cycle->old_cycle->connection_n < cycle->connection_n) { ngx_test_null(list, ngx_alloc(sizeof(struct pollfd) * cycle->connection_n, cycle->log), NGX_ERROR); if (event_list) { ngx_memcpy(list, event_list, sizeof(ngx_event_t *) * nevents); ngx_free(event_list); } event_list = list; ngx_test_null(index, ngx_alloc(sizeof(ngx_event_t *) * cycle->connection_n, cycle->log), NGX_ERROR); if (event_index) { ngx_memcpy(index, event_index, sizeof(ngx_event_t *) * nevents); ngx_free(event_index); } event_index = index; if (ready_index) { ngx_free(ready_index); } ngx_test_null(ready_index, ngx_alloc(sizeof(ngx_event_t *) * 2 * cycle->connection_n, cycle->log), NGX_ERROR); } if (ngx_event_timer_init(cycle) == NGX_ERROR) { return NGX_ERROR; } ngx_io = ngx_os_io; ngx_event_actions = ngx_poll_module_ctx.actions; ngx_event_flags = NGX_USE_LEVEL_EVENT|NGX_USE_ONESHOT_EVENT; return NGX_OK; } static void ngx_poll_done(ngx_cycle_t *cycle) { ngx_event_timer_done(cycle); ngx_free(event_list); ngx_free(event_index); ngx_free(ready_index); event_list = NULL; event_index = NULL; ready_index = NULL; } static int ngx_poll_add_event(ngx_event_t *ev, int event, u_int flags) { ngx_event_t *e; ngx_connection_t *c; c = ev->data; ev->active = 1; 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 } #if (NGX_DEBUG_EVENT) ngx_log_debug(ev->log, "add event: %d:%d" _ c->fd _ event); #endif 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; } static 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 } #if (NGX_DEBUG_EVENT) ngx_log_debug(c->log, "del event: %d, %d" _ c->fd _ event); #endif if (e == NULL || e->index == NGX_INVALID_INDEX) { if (ev->index < --nevents) { event_list[ev->index] = event_list[nevents]; event_index[ev->index] = event_index[nevents]; event_index[ev->index]->index = ev->index; } } else { event_list[e->index].events &= ~event; } ev->active = 0; ev->index = NGX_INVALID_INDEX; return NGX_OK; } static int ngx_poll_process_events(ngx_log_t *log) { int ready, found, j; u_int nready, i; ngx_msec_t timer; ngx_err_t err; ngx_cycle_t **cycle; ngx_event_t *ev; ngx_epoch_msec_t delta; ngx_connection_t *c; struct timeval tv; timer = ngx_event_find_timer(); if (timer) { ngx_gettimeofday(&tv); delta = tv.tv_sec * 1000 + tv.tv_usec / 1000; } else { timer = INFTIM; delta = 0; } #if (NGX_DEBUG_EVENT) for (i = 0; i < nevents; i++) { ngx_log_debug(log, "poll: %d, %d" _ event_list[i].fd _ event_list[i].events); } ngx_log_debug(log, "poll timer: %d" _ timer); #endif ready = poll(event_list, nevents, timer); if (ready == -1) { err = ngx_errno; } else { err = 0; } ngx_log_debug(log, "poll ready %d" _ ready); ngx_gettimeofday(&tv); if (ngx_cached_time != tv.tv_sec) { ngx_cached_time = tv.tv_sec; ngx_time_update(); } if ((int) timer != INFTIM) { delta = tv.tv_sec * 1000 + tv.tv_usec / 1000 - delta; #if (NGX_DEBUG_EVENT) ngx_log_debug(log, "poll timer: %d, delta: %d" _ timer _ (int) delta); #endif ngx_event_expire_timers((ngx_msec_t) delta); } else { if (ready == 0) { ngx_log_error(NGX_LOG_ALERT, log, 0, "poll() returned no events without timeout"); return NGX_ERROR; } #if (NGX_DEBUG_EVENT) ngx_log_debug(log, "poll timer: %d, delta: %d" _ timer _ (int) delta); #endif } if (err) { ngx_log_error(NGX_LOG_ALERT, log, err, "poll() failed"); return NGX_ERROR; } nready = 0; for (i = 0; i < nevents && ready; i++) { c = &ngx_cycle->connections[event_list[i].fd]; if (c->fd == -1) { cycle = ngx_old_cycles.elts; for (j = 0; j < ngx_old_cycles.nelts; j++) { if (cycle[j] == NULL) { continue; } c = &cycle[j]->connections[event_list[i].fd]; if (c->fd != -1) { break; } } } if (c->fd == -1) { ngx_log_error(NGX_LOG_ALERT, log, 0, "unkonwn cycle"); exit(1); } #if (NGX_DEBUG_EVENT) ngx_log_debug(log, "poll: fd:%d, ev:%d, rev:%d" _ event_list[i].fd _ event_list[i].events _ event_list[i].revents); #endif found = 0; if (event_list[i].revents & POLLNVAL) { ngx_log_error(NGX_LOG_ALERT, log, EBADF, "poll() error on %d", event_list[i].fd); continue; } if (event_list[i].revents & POLLIN || (event_list[i].revents & (POLLERR|POLLHUP) && c->read->active)) { found = 1; ready_index[nready++] = c->read; } if (event_list[i].revents & POLLOUT || (event_list[i].revents & (POLLERR|POLLHUP) && c->write->active)) { found = 1; ready_index[nready++] = c->write; } if (found) { ready--; continue; } if (event_list[i].revents & (POLLERR|POLLHUP)) { ngx_log_error(NGX_LOG_ALERT, log, 0, "strange poll() error on %d:%d:%d", event_list[i].fd, event_list[i].events, event_list[i].revents); } } for (i = 0; i < nready; i++) { ev = ready_index[i]; if (!ev->active) { continue; } ev->ready = 1; if (ev->oneshot) { if (ev->timer_set) { ngx_del_timer(ev); } if (ev->write) { ngx_poll_del_event(ev, NGX_WRITE_EVENT, 0); } else { ngx_poll_del_event(ev, NGX_READ_EVENT, 0); } } ev->event_handler(ev); } if (ready != 0) { ngx_log_error(NGX_LOG_ALERT, log, 0, "poll ready != events"); } return NGX_OK; }