Mercurial > hg > nginx-vendor-0-8
view src/event/ngx_event_accept.c @ 77:e6b3de2dc637
Added tag NGINX_0_1_38 for changeset da9a3b14312d
author | Igor Sysoev <http://sysoev.ru> |
---|---|
date | Fri, 08 Jul 2005 00:00:00 +0400 |
parents | b31656313b59 |
children | e916a291e9aa |
line wrap: on
line source
/* * Copyright (C) Igor Sysoev */ #include <ngx_config.h> #include <ngx_core.h> #include <ngx_event.h> static void ngx_close_accepted_socket(ngx_socket_t s, ngx_log_t *log); static u_char *ngx_accept_log_error(ngx_log_t *log, u_char *buf, size_t len); void ngx_event_accept(ngx_event_t *ev) { ngx_uint_t instance; #if 0 ngx_uint_t accepted; #endif socklen_t len; struct sockaddr *sa; ngx_err_t err; ngx_log_t *log; ngx_pool_t *pool; ngx_socket_t s; ngx_event_t *rev, *wev; ngx_connection_t *c, *ls; ngx_event_conf_t *ecf; ecf = ngx_event_get_conf(ngx_cycle->conf_ctx, ngx_event_core_module); if (ngx_event_flags & NGX_USE_RTSIG_EVENT) { ev->available = 1; } else if (!(ngx_event_flags & NGX_USE_KQUEUE_EVENT)) { ev->available = ecf->multi_accept; } ls = ev->data; ngx_log_debug2(NGX_LOG_DEBUG_EVENT, ev->log, 0, "accept on %V, ready: %d", &ls->listening->addr_text, ev->available); ev->ready = 0; pool = NULL; #if 0 accepted = 0; #endif do { if (pool == NULL) { /* * Create the pool before accept() to avoid the copying of * the sockaddr. Although accept() can fail it is uncommon * case and besides the pool can be got from the free pool list. */ pool = ngx_create_pool(ls->listening->pool_size, ev->log); if (pool == NULL) { return; } } sa = ngx_palloc(pool, ls->listening->socklen); if (sa == NULL) { ngx_destroy_pool(pool); return; } log = ngx_palloc(pool, sizeof(ngx_log_t)); if (log == NULL) { ngx_destroy_pool(pool); return; } ngx_memcpy(log, ls->log, sizeof(ngx_log_t)); pool->log = log; log->data = &ls->listening->addr_text; log->handler = ngx_accept_log_error; len = ls->listening->socklen; s = accept(ls->fd, sa, &len); if (s == -1) { err = ngx_socket_errno; if (err == NGX_EAGAIN) { #if 0 if (!(ngx_event_flags & NGX_USE_RTSIG_EVENT)) { ngx_log_error(NGX_LOG_NOTICE, log, err, "EAGAIN after %d accepted connection(s)", accepted); } #endif ngx_destroy_pool(pool); return; } ngx_log_error((err == NGX_ECONNABORTED) ? NGX_LOG_CRIT: NGX_LOG_ALERT, ev->log, err, "accept() on %V failed", &ls->listening->addr_text); if (err == NGX_ECONNABORTED) { if (ngx_event_flags & NGX_USE_KQUEUE_EVENT) { ev->available--; } if (ev->available) { /* reuse the previously allocated pool */ continue; } } ngx_destroy_pool(pool); return; } #if (NGX_STAT_STUB) ngx_atomic_inc(ngx_stat_accepted); ngx_atomic_inc(ngx_stat_active); #endif ngx_accept_disabled = (ngx_uint_t) s + NGX_ACCEPT_THRESHOLD - ecf->connections; /* disable warning: Win32 SOCKET is u_int while UNIX socket is int */ if ((ngx_uint_t) s >= ecf->connections) { ngx_log_error(NGX_LOG_ALERT, ev->log, 0, "accept() on %V returned socket #%d while " "only %d connections was configured, " "closing the connection", &ls->listening->addr_text, s, ecf->connections); ngx_close_accepted_socket(s, log); ngx_destroy_pool(pool); return; } /* set a blocking mode for aio and non-blocking mode for others */ if (ngx_inherited_nonblocking) { if ((ngx_event_flags & NGX_USE_AIO_EVENT)) { if (ngx_blocking(s) == -1) { ngx_log_error(NGX_LOG_ALERT, log, ngx_socket_errno, ngx_blocking_n " failed"); ngx_close_accepted_socket(s, log); ngx_destroy_pool(pool); return; } } } else { if (!(ngx_event_flags & (NGX_USE_AIO_EVENT|NGX_USE_RTSIG_EVENT))) { if (ngx_nonblocking(s) == -1) { ngx_log_error(NGX_LOG_ALERT, log, ngx_socket_errno, ngx_nonblocking_n " failed"); ngx_close_accepted_socket(s, log); ngx_destroy_pool(pool); return; } } } #if (NGX_WIN32) /* * Winsock assignes a socket number divisible by 4 * so to find a connection we divide a socket number by 4. */ if (s % 4) { ngx_log_error(NGX_LOG_EMERG, ev->log, 0, "accept() on %V returned socket #%d, " "not divisible by 4", &ls->listening->addr_text, s); exit(1); } c = &ngx_cycle->connections[s / 4]; rev = &ngx_cycle->read_events[s / 4]; wev = &ngx_cycle->write_events[s / 4]; #else c = &ngx_cycle->connections[s]; rev = &ngx_cycle->read_events[s]; wev = &ngx_cycle->write_events[s]; #endif instance = rev->instance; #if (NGX_THREADS) if (*(&c->lock)) { ngx_log_debug1(NGX_LOG_DEBUG_EVENT, ev->log, 0, "spinlock in accept, fd:%d", s); ngx_spinlock(&c->lock, 1000); ngx_unlock(&c->lock); } #endif ngx_memzero(rev, sizeof(ngx_event_t)); ngx_memzero(wev, sizeof(ngx_event_t)); ngx_memzero(c, sizeof(ngx_connection_t)); c->pool = pool; c->listening = ls->listening; c->sockaddr = sa; c->socklen = len; rev->instance = !instance; wev->instance = !instance; rev->index = NGX_INVALID_INDEX; wev->index = NGX_INVALID_INDEX; rev->data = c; wev->data = c; c->read = rev; c->write = wev; c->fd = s; c->unexpected_eof = 1; wev->write = 1; wev->ready = 1; if (ngx_event_flags & (NGX_USE_AIO_EVENT|NGX_USE_RTSIG_EVENT)) { /* rtsig, aio, iocp */ rev->ready = 1; } if (ev->deferred_accept) { rev->ready = 1; #if (NGX_HAVE_KQUEUE) rev->available = 1; #endif } c->ctx = ls->ctx; c->servers = ls->servers; c->recv = ngx_recv; c->send_chain = ngx_send_chain; c->log = log; rev->log = log; wev->log = log; /* * TODO: MT: - atomic increment (x86: lock xadd) * or protection by critical section or light mutex * * TODO: MP: - allocated in a shared memory * - atomic increment (x86: lock xadd) * or protection by critical section or light mutex */ c->number = ngx_atomic_inc(ngx_connection_counter); #if (NGX_STAT_STUB) ngx_atomic_inc(ngx_stat_handled); #endif #if (NGX_THREADS) rev->lock = &c->lock; wev->lock = &c->lock; rev->own_lock = &c->lock; wev->own_lock = &c->lock; #endif ngx_log_debug2(NGX_LOG_DEBUG_EVENT, ev->log, 0, "accept: fd:%d c:%d", s, c->number); if (c->listening->addr_ntop) { c->addr_text.data = ngx_palloc(c->pool, c->listening->addr_text_max_len); if (c->addr_text.data == NULL) { ngx_close_accepted_socket(s, log); ngx_destroy_pool(pool); return; } c->addr_text.len = ngx_sock_ntop(c->listening->family, c->sockaddr, c->addr_text.data, c->listening->addr_text_max_len); if (c->addr_text.len == 0) { ngx_close_accepted_socket(s, log); ngx_destroy_pool(pool); return; } } #if (NGX_DEBUG) { uint32_t *addr; in_addr_t i; struct sockaddr_in *sin; sin = (struct sockaddr_in *) sa; addr = ecf->debug_connection.elts; for (i = 0; i < ecf->debug_connection.nelts; i++) { if (addr[i] == sin->sin_addr.s_addr) { log->log_level = NGX_LOG_DEBUG_CONNECTION|NGX_LOG_DEBUG_ALL; break; } } } #endif if (ngx_add_conn && (ngx_event_flags & NGX_USE_EPOLL_EVENT) == 0) { if (ngx_add_conn(c) == NGX_ERROR) { ngx_close_accepted_socket(s, log); ngx_destroy_pool(pool); return; } } pool = NULL; log->data = NULL; log->handler = NULL; ls->listening->handler(c); if (ngx_event_flags & NGX_USE_KQUEUE_EVENT) { ev->available--; } #if 0 accepted++; #endif } while (ev->available); } ngx_int_t ngx_trylock_accept_mutex(ngx_cycle_t *cycle) { if (*ngx_accept_mutex == 0 && ngx_atomic_cmp_set(ngx_accept_mutex, 0, ngx_pid)) { ngx_log_debug0(NGX_LOG_DEBUG_EVENT, cycle->log, 0, "accept mutex locked"); if (!ngx_accept_mutex_held) { if (ngx_enable_accept_events(cycle) == NGX_ERROR) { *ngx_accept_mutex = 0; return NGX_ERROR; } ngx_accept_mutex_held = 1; } return NGX_OK; } if (ngx_accept_mutex_held) { if (ngx_disable_accept_events(cycle) == NGX_ERROR) { return NGX_ERROR; } ngx_accept_mutex_held = 0; } return NGX_OK; } ngx_int_t ngx_enable_accept_events(ngx_cycle_t *cycle) { ngx_uint_t i; ngx_listening_t *s; s = cycle->listening.elts; for (i = 0; i < cycle->listening.nelts; i++) { /* * we do not need to handle the Winsock sockets here (divide a socket * number by 4) because this function would never called * in the Winsock environment */ if (ngx_event_flags & NGX_USE_RTSIG_EVENT) { if (ngx_add_conn(&cycle->connections[s[i].fd]) == NGX_ERROR) { return NGX_ERROR; } } else { if (ngx_add_event(&cycle->read_events[s[i].fd], NGX_READ_EVENT, 0) == NGX_ERROR) { return NGX_ERROR; } } } return NGX_OK; } ngx_int_t ngx_disable_accept_events(ngx_cycle_t *cycle) { ngx_uint_t i; ngx_listening_t *s; s = cycle->listening.elts; for (i = 0; i < cycle->listening.nelts; i++) { /* * we do not need to handle the Winsock sockets here (divide a socket * number by 4) because this function would never called * in the Winsock environment */ if (ngx_event_flags & NGX_USE_RTSIG_EVENT) { if (!cycle->connections[s[i].fd].read->active) { continue; } if (ngx_del_conn(&cycle->connections[s[i].fd], NGX_DISABLE_EVENT) == NGX_ERROR) { return NGX_ERROR; } } else { if (!cycle->read_events[s[i].fd].active) { continue; } if (ngx_del_event(&cycle->read_events[s[i].fd], NGX_READ_EVENT, NGX_DISABLE_EVENT) == NGX_ERROR) { return NGX_ERROR; } } } return NGX_OK; } static void ngx_close_accepted_socket(ngx_socket_t s, ngx_log_t *log) { if (ngx_close_socket(s) == -1) { ngx_log_error(NGX_LOG_ALERT, log, ngx_socket_errno, ngx_close_socket_n " failed"); } #if (NGX_STAT_STUB) ngx_atomic_dec(ngx_stat_active); #endif } static u_char * ngx_accept_log_error(ngx_log_t *log, u_char *buf, size_t len) { return ngx_snprintf(buf, len, " while accept() on %V", log->data); }