| |
| #include <ngx_config.h> |
| #include <ngx_core.h> |
| #include <ngx_event.h> |
| #include <ngx_http.h> |
| |
| |
| static void ngx_http_init_request(ngx_event_t *ev); |
| static void ngx_http_process_request_line(ngx_event_t *rev); |
| static void ngx_http_process_request_headers(ngx_event_t *rev); |
| static ssize_t ngx_http_read_request_header(ngx_http_request_t *r); |
| |
| static void ngx_http_writer(ngx_event_t *ev); |
| |
| static void ngx_http_block_read(ngx_event_t *ev); |
| static void ngx_http_read_discarded_body_event(ngx_event_t *rev); |
| static int ngx_http_read_discarded_body(ngx_http_request_t *r); |
| |
| static void ngx_http_set_keepalive(ngx_http_request_t *r); |
| static void ngx_http_keepalive_handler(ngx_event_t *ev); |
| static void ngx_http_set_lingering_close(ngx_http_request_t *r); |
| static void ngx_http_lingering_close_handler(ngx_event_t *ev); |
| static void ngx_http_empty_handler(ngx_event_t *wev); |
| |
| static void ngx_http_header_parse_error(ngx_http_request_t *r, int parse_err); |
| static size_t ngx_http_log_error(void *data, char *buf, size_t len); |
| |
| |
| /* NGX_HTTP_PARSE_ ... errors */ |
| |
| static char *header_errors[] = { |
| "client %s sent invalid method", |
| "client %s sent invalid request", |
| "client %s sent too long URI", |
| "client %s sent invalid method in HTTP/0.9 request", |
| |
| "client %s sent invalid header, URL: %s", |
| "client %s sent too long header line, URL: %s", |
| "client %s sent HTTP/1.1 request without \"Host\" header, URL: %s", |
| "client %s sent invalid \"Content-Length\" header, URL: %s" |
| }; |
| |
| |
| |
| static ngx_http_header_t headers_in[] = { |
| { ngx_string("Host"), offsetof(ngx_http_headers_in_t, host) }, |
| { ngx_string("Connection"), offsetof(ngx_http_headers_in_t, connection) }, |
| { ngx_string("If-Modified-Since"), |
| offsetof(ngx_http_headers_in_t, if_modified_since) }, |
| { ngx_string("Content-Length"), |
| offsetof(ngx_http_headers_in_t, content_length) }, |
| |
| { ngx_string("Range"), offsetof(ngx_http_headers_in_t, range) }, |
| #if 0 |
| { ngx_string("If-Range"), offsetof(ngx_http_headers_in_t, if_range) }, |
| #endif |
| |
| { ngx_string("User-Agent"), offsetof(ngx_http_headers_in_t, user_agent) }, |
| |
| { ngx_null_string, 0 } |
| }; |
| |
| |
| void ngx_http_init_connection(ngx_connection_t *c) |
| { |
| int event; |
| ngx_event_t *rev; |
| ngx_http_log_ctx_t *lcx; |
| |
| c->addr_text.data = ngx_palloc(c->pool, c->listening->addr_text_max_len); |
| if (c->addr_text.data == NULL) { |
| ngx_http_close_connection(c); |
| 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_http_close_connection(c); |
| return; |
| } |
| |
| lcx = ngx_pcalloc(c->pool, sizeof(ngx_http_log_ctx_t)); |
| if (lcx == NULL) { |
| ngx_http_close_connection(c); |
| return; |
| } |
| |
| lcx->client = c->addr_text.data; |
| lcx->action = "reading client request line"; |
| c->log->data = lcx; |
| c->log->handler = ngx_http_log_error; |
| |
| rev = c->read; |
| rev->event_handler = ngx_http_init_request; |
| |
| if (rev->ready) { |
| /* deferred accept */ |
| ngx_http_init_request(rev); |
| return; |
| } |
| |
| ngx_add_timer(rev, c->listening->post_accept_timeout); |
| rev->timer_set = 1; |
| |
| if (ngx_event_flags & (NGX_HAVE_AIO_EVENT|NGX_HAVE_EDGE_EVENT)) { |
| /* aio, iocp, epoll */ |
| ngx_http_init_request(rev); |
| return; |
| } |
| |
| if (ngx_event_flags & NGX_HAVE_CLEAR_EVENT) { |
| /* kqueue */ |
| event = NGX_CLEAR_EVENT; |
| |
| } else { |
| /* select, poll, /dev/poll */ |
| event = NGX_LEVEL_EVENT; |
| } |
| |
| if (ngx_add_event(rev, NGX_READ_EVENT, event) == NGX_ERROR) { |
| ngx_http_close_connection(c); |
| } |
| } |
| |
| |
| static void ngx_http_init_request(ngx_event_t *rev) |
| { |
| int i; |
| socklen_t len; |
| struct sockaddr_in addr_in; |
| ngx_connection_t *c; |
| ngx_http_request_t *r; |
| ngx_http_in_port_t *in_port; |
| ngx_http_in_addr_t *in_addr; |
| ngx_http_server_name_t *server_name; |
| ngx_http_core_srv_conf_t *cscf; |
| |
| c = rev->data; |
| |
| r = ngx_pcalloc(c->pool, sizeof(ngx_http_request_t)); |
| if (r == NULL) { |
| ngx_http_close_connection(c); |
| return; |
| } |
| |
| /* find the server configuration for the address:port */ |
| |
| /* AF_INET only */ |
| |
| in_port = c->servers; |
| in_addr = in_port->addrs.elts; |
| |
| r->port = in_port->port; |
| r->port_name = &in_port->port_name; |
| |
| i = 0; |
| |
| if (in_port->addrs.nelts > 1) { |
| |
| /* |
| * there're the several addresses on this port and one of them |
| * is "*:port" so getsockname() is needed to determine |
| * the server address. |
| * AcceptEx() already gave this address. |
| */ |
| |
| #if (WIN32) |
| if (c->local_sockaddr) { |
| r->in_addr = |
| ((struct sockaddr_in *) c->local_sockaddr)->sin_addr.s_addr; |
| |
| } else { |
| #endif |
| len = sizeof(struct sockaddr_in); |
| if (getsockname(c->fd, (struct sockaddr *) &addr_in, &len) == -1) { |
| ngx_log_error(NGX_LOG_CRIT, rev->log, ngx_socket_errno, |
| "getsockname() failed"); |
| ngx_http_close_connection(c); |
| return; |
| } |
| #if (WIN32) |
| } |
| #endif |
| |
| r->in_addr = addr_in.sin_addr.s_addr; |
| |
| /* the last in_port->addrs address is "*" */ |
| |
| for ( /* void */ ; i < in_port->addrs.nelts - 1; i++) { |
| if (in_addr[i].addr == r->in_addr) { |
| break; |
| } |
| } |
| |
| } else { |
| r->in_addr = in_addr[0].addr; |
| } |
| |
| r->virtual_names = &in_addr[i].names; |
| |
| /* the default server configuration for the address:port */ |
| cscf = in_addr[i].core_srv_conf; |
| |
| r->main_conf = cscf->ctx->main_conf; |
| r->srv_conf = cscf->ctx->srv_conf; |
| r->loc_conf = cscf->ctx->loc_conf; |
| |
| server_name = cscf->server_names.elts; |
| r->server_name = &server_name->name; |
| |
| if (c->buffer == NULL) { |
| c->buffer = ngx_create_temp_hunk(c->pool, |
| cscf->client_header_buffer_size, |
| 0, 0); |
| if (c->buffer == NULL) { |
| ngx_http_close_connection(c); |
| return; |
| } |
| } |
| |
| r->pool = ngx_create_pool(cscf->request_pool_size, c->log); |
| if (r->pool == NULL) { |
| ngx_http_close_connection(c); |
| return; |
| } |
| |
| r->headers_out.headers = ngx_create_table(r->pool, 10); |
| if (r->headers_out.headers == NULL) { |
| ngx_http_close_request(r, NGX_HTTP_INTERNAL_SERVER_ERROR); |
| ngx_http_close_connection(c); |
| return; |
| } |
| |
| r->ctx = ngx_pcalloc(r->pool, sizeof(void *) * ngx_http_max_module); |
| if (r->ctx == NULL) { |
| ngx_http_close_request(r, NGX_HTTP_INTERNAL_SERVER_ERROR); |
| ngx_http_close_connection(c); |
| return; |
| } |
| |
| c->sent = 0; |
| c->data = r; |
| r->connection = c; |
| r->pipeline = c->pipeline; |
| r->header_in = c->buffer; |
| |
| r->file.fd = NGX_INVALID_FILE; |
| |
| r->headers_in.content_length_n = -1; |
| r->headers_out.content_length = -1; |
| r->headers_out.last_modified_time = -1; |
| |
| rev->event_handler = ngx_http_process_request_line; |
| ngx_http_process_request_line(rev); |
| } |
| |
| |
| static void ngx_http_process_request_line(ngx_event_t *rev) |
| { |
| int rc, offset; |
| ssize_t n; |
| ngx_connection_t *c; |
| ngx_http_request_t *r; |
| ngx_http_log_ctx_t *lcx; |
| ngx_http_core_srv_conf_t *cscf; |
| |
| c = rev->data; |
| r = c->data; |
| |
| ngx_log_debug(rev->log, "http process request line"); |
| |
| if (rev->timedout) { |
| ngx_http_close_request(r, NGX_HTTP_REQUEST_TIME_OUT); |
| ngx_http_close_connection(c); |
| return; |
| } |
| |
| n = ngx_http_read_request_header(r); |
| |
| if (n == NGX_AGAIN || n == NGX_ERROR) { |
| return; |
| } |
| |
| rc = ngx_parse_http_request_line(r); |
| |
| if (rc == NGX_OK) { |
| |
| /* the request line has been parsed successfully */ |
| |
| /* STUB: we need to handle such URIs */ |
| if (r->complex_uri || r->unusual_uri) { |
| r->request_line.len = r->request_end - r->request_start; |
| r->request_line.data = r->request_start; |
| r->request_line.data[r->request_line.len] = '\0'; |
| |
| ngx_http_header_parse_error(r, NGX_HTTP_PARSE_INVALID_REQUEST); |
| ngx_http_finalize_request(r, NGX_HTTP_BAD_REQUEST); |
| return; |
| } |
| |
| cscf = ngx_http_get_module_srv_conf(r, ngx_http_core_module); |
| |
| if (r->http_version >= NGX_HTTP_VERSION_10 |
| && cscf->large_client_header == 0 |
| && r->header_in->pos == r->header_in->end) |
| { |
| /* no space for "\r\n" at the end of the header */ |
| |
| ngx_http_header_parse_error(r, NGX_HTTP_PARSE_TOO_LONG_URI); |
| ngx_http_finalize_request(r, NGX_HTTP_REQUEST_URI_TOO_LARGE); |
| return; |
| } |
| |
| /* copy URI */ |
| |
| if (r->args_start) { |
| r->uri.len = r->args_start - 1 - r->uri_start; |
| } else { |
| r->uri.len = r->uri_end - r->uri_start; |
| } |
| |
| r->uri.data = ngx_palloc(r->pool, r->uri.len + 1); |
| if (r->uri.data == NULL) { |
| ngx_http_close_request(r, NGX_HTTP_INTERNAL_SERVER_ERROR); |
| ngx_http_close_connection(c); |
| return; |
| } |
| |
| ngx_cpystrn(r->uri.data, r->uri_start, r->uri.len + 1); |
| |
| #if 1 /* THINK: needed to log url on errors in proxy only ? */ |
| |
| /* copy unparsed URI */ |
| |
| r->unparsed_uri.len = r->uri_end - r->uri_start; |
| r->unparsed_uri.data = ngx_palloc(r->pool, r->unparsed_uri.len + 1); |
| if (r->unparsed_uri.data == NULL) { |
| ngx_http_close_request(r, NGX_HTTP_INTERNAL_SERVER_ERROR); |
| ngx_http_close_connection(c); |
| return; |
| } |
| |
| ngx_cpystrn(r->unparsed_uri.data, r->uri_start, |
| r->unparsed_uri.len + 1); |
| |
| #endif |
| |
| r->request_line.len = r->request_end - r->request_start; |
| |
| /* if the large client headers are enabled then |
| we need to copy a request line */ |
| |
| if (cscf->large_client_header) { |
| |
| r->request_line.data = ngx_palloc(r->pool, r->request_line.len + 1); |
| if (r->request_line.data == NULL) { |
| ngx_http_close_request(r, NGX_HTTP_INTERNAL_SERVER_ERROR); |
| ngx_http_close_connection(c); |
| return; |
| } |
| |
| ngx_cpystrn(r->request_line.data, r->request_start, |
| r->request_line.len + 1); |
| |
| } else { |
| r->request_line.data = r->request_start; |
| r->request_line.data[r->request_line.len] = '\0'; |
| } |
| |
| /* copy URI extention if it exists */ |
| |
| if (r->uri_ext) { |
| if (r->args_start) { |
| r->exten.len = r->args_start - 1 - r->uri_ext; |
| } else { |
| r->exten.len = r->uri_end - r->uri_ext; |
| } |
| |
| r->exten.data = ngx_palloc(r->pool, r->exten.len + 1); |
| if (r->exten.data == NULL) { |
| ngx_http_close_request(r, NGX_HTTP_INTERNAL_SERVER_ERROR); |
| ngx_http_close_connection(c); |
| return; |
| } |
| |
| ngx_cpystrn(r->exten.data, r->uri_ext, r->exten.len + 1); |
| } |
| |
| /* copy URI arguments if they exist */ |
| |
| if (r->args_start && r->uri_end > r->args_start) { |
| r->args.len = r->uri_end - r->args_start; |
| |
| r->args.data = ngx_palloc(r->pool, r->args.len + 1); |
| if (r->args.data == NULL) { |
| ngx_http_close_request(r, NGX_HTTP_INTERNAL_SERVER_ERROR); |
| ngx_http_close_connection(c); |
| return; |
| } |
| |
| ngx_cpystrn(r->args.data, r->args_start, r->args.len + 1); |
| } |
| |
| #if 1 /* DEBUG */ |
| if (r->exten.data == NULL) { r->exten.data = ""; } |
| if (r->args.data == NULL) { r->args.data = ""; } |
| ngx_log_debug(c->log, "HTTP: %d, %d, '%s', '%s', '%s'" _ |
| r->method _ r->http_version _ |
| r->uri.data _ r->exten.data _ r->args.data); |
| if (r->exten.data[0] == '\0') { r->exten.data = NULL; } |
| if (r->args.data[0] == '\0') { r->args.data = NULL; } |
| #endif |
| |
| if (r->http_version < NGX_HTTP_VERSION_10) { |
| rev->event_handler = ngx_http_block_read; |
| ngx_http_handler(r); |
| return; |
| } |
| |
| lcx = c->log->data; |
| lcx->action = "reading client request headers"; |
| lcx->url = r->unparsed_uri.data; |
| r->headers_in.headers = ngx_create_table(r->pool, 10); |
| |
| if (cscf->large_client_header |
| && r->header_in->pos == r->header_in->last) |
| { |
| r->header_in->pos = r->header_in->last = r->header_in->start; |
| } |
| |
| rev->event_handler = ngx_http_process_request_headers; |
| ngx_http_process_request_headers(rev); |
| |
| return; |
| |
| } else if (rc != NGX_AGAIN) { |
| |
| /* there was error while a request line parsing */ |
| |
| ngx_http_header_parse_error(r, rc); |
| ngx_http_finalize_request(r, NGX_HTTP_BAD_REQUEST); |
| |
| return; |
| } |
| |
| /* NGX_AGAIN: a request line parsing is still not complete */ |
| |
| if (r->header_in->last == r->header_in->end) { |
| |
| /* If it's a pipelined request and a request line is not complete |
| then we need to copy it to the start of the r->header_in hunk. |
| We need to copy it here only if the large client headers |
| are enabled otherwise a request line had been already copied |
| to the start of the r->header_in hunk in ngx_http_set_keepalive() */ |
| |
| cscf = ngx_http_get_module_srv_conf(r, ngx_http_core_module); |
| |
| if (cscf->large_client_header) { |
| offset = r->request_start - r->header_in->start; |
| |
| if (offset == 0) { |
| ngx_http_header_parse_error(r, NGX_HTTP_PARSE_TOO_LONG_URI); |
| ngx_http_finalize_request(r, NGX_HTTP_REQUEST_URI_TOO_LARGE); |
| |
| return; |
| } |
| |
| ngx_memcpy(r->header_in->start, r->request_start, |
| r->header_in->last - r->request_start); |
| |
| r->header_in->pos -= offset; |
| r->header_in->last -= offset; |
| r->request_start = r->header_in->start; |
| r->request_end -= offset; |
| r->uri_start -= offset; |
| r->uri_end -= offset; |
| if (r->uri_ext) { |
| r->uri_ext -= offset; |
| } |
| if (r->args_start) { |
| r->args_start -= offset; |
| } |
| |
| } else { |
| ngx_http_header_parse_error(r, NGX_HTTP_PARSE_TOO_LONG_URI); |
| ngx_http_finalize_request(r, NGX_HTTP_REQUEST_URI_TOO_LARGE); |
| } |
| } |
| |
| return; |
| } |
| |
| |
| static void ngx_http_process_request_headers(ngx_event_t *rev) |
| { |
| int rc, i, offset; |
| size_t len; |
| ssize_t n; |
| ngx_table_elt_t *h; |
| ngx_connection_t *c; |
| ngx_http_request_t *r; |
| ngx_http_server_name_t *name; |
| ngx_http_core_srv_conf_t *cscf; |
| |
| c = rev->data; |
| r = c->data; |
| |
| ngx_log_debug(rev->log, "http process request header line"); |
| |
| if (rev->timedout) { |
| ngx_http_close_request(r, NGX_HTTP_REQUEST_TIME_OUT); |
| ngx_http_close_connection(c); |
| return; |
| } |
| |
| cscf = ngx_http_get_module_srv_conf(r, ngx_http_core_module); |
| |
| rc = NGX_AGAIN; |
| |
| for ( ;; ) { |
| if (rc == NGX_AGAIN) { |
| n = ngx_http_read_request_header(r); |
| |
| if (n == NGX_AGAIN || n == NGX_ERROR) { |
| return; |
| } |
| } |
| |
| rc = ngx_parse_http_header_line(r, r->header_in); |
| |
| if (rc == NGX_OK) { |
| |
| /* a header line has been parsed successfully */ |
| |
| h = ngx_push_table(r->headers_in.headers); |
| if (h == NULL) { |
| ngx_http_close_request(r, NGX_HTTP_INTERNAL_SERVER_ERROR); |
| ngx_http_close_connection(c); |
| return; |
| } |
| |
| h->key.len = r->header_name_end - r->header_name_start; |
| h->value.len = r->header_end - r->header_start; |
| |
| /* if the large client headers are enabled then |
| we need to copy the header name and value */ |
| |
| if (cscf->large_client_header) { |
| h->key.data = ngx_palloc(r->pool, |
| h->key.len + 1 + h->value.len + 1); |
| if (h->key.data == NULL) { |
| ngx_http_close_request(r, NGX_HTTP_INTERNAL_SERVER_ERROR); |
| ngx_http_close_connection(c); |
| return; |
| } |
| |
| h->value.data = h->key.data + h->key.len + 1; |
| ngx_cpystrn(h->key.data, r->header_name_start, h->key.len + 1); |
| ngx_cpystrn(h->value.data, r->header_start, h->value.len + 1); |
| |
| } else { |
| h->key.data = r->header_name_start; |
| h->key.data[h->key.len] = '\0'; |
| h->value.data = r->header_start; |
| h->value.data[h->value.len] = '\0'; |
| } |
| |
| for (i = 0; headers_in[i].name.len != 0; i++) { |
| if (headers_in[i].name.len != h->key.len) { |
| continue; |
| } |
| |
| if (ngx_strcasecmp(headers_in[i].name.data, h->key.data) == 0) { |
| *((ngx_table_elt_t **) |
| ((char *) &r->headers_in + headers_in[i].offset)) = h; |
| } |
| } |
| |
| ngx_log_debug(r->connection->log, "HTTP header: '%s: %s'" _ |
| h->key.data _ h->value.data); |
| |
| if (cscf->large_client_header |
| && r->header_in->pos == r->header_in->last) |
| { |
| r->header_in->pos = r->header_in->last = r->header_in->start; |
| } |
| |
| continue; |
| |
| } else if (rc == NGX_HTTP_PARSE_HEADER_DONE) { |
| |
| /* a whole header has been parsed successfully */ |
| |
| ngx_log_debug(r->connection->log, "HTTP header done"); |
| |
| if (r->headers_in.host) { |
| for (len = 0; len < r->headers_in.host->value.len; len++) { |
| if (r->headers_in.host->value.data[len] == ':') { |
| break; |
| } |
| } |
| r->headers_in.host_name_len = len; |
| |
| /* find the name based server configuration */ |
| |
| name = r->virtual_names->elts; |
| for (i = 0; i < r->virtual_names->nelts; i++) { |
| if (r->headers_in.host_name_len != name[i].name.len) { |
| continue; |
| } |
| |
| if (ngx_strncasecmp(r->headers_in.host->value.data, |
| name[i].name.data, |
| r->headers_in.host_name_len) == 0) |
| { |
| r->srv_conf = name[i].core_srv_conf->ctx->srv_conf; |
| r->loc_conf = name[i].core_srv_conf->ctx->loc_conf; |
| break; |
| } |
| } |
| |
| } else { |
| if (r->http_version > NGX_HTTP_VERSION_10) { |
| ngx_http_header_parse_error(r, |
| NGX_HTTP_PARSE_NO_HOST_HEADER); |
| ngx_http_finalize_request(r, NGX_HTTP_BAD_REQUEST); |
| return; |
| } |
| r->headers_in.host_name_len = 0; |
| } |
| |
| if (r->headers_in.content_length) { |
| r->headers_in.content_length_n = |
| ngx_atoi(r->headers_in.content_length->value.data, |
| r->headers_in.content_length->value.len); |
| if (r->headers_in.content_length_n == NGX_ERROR) { |
| ngx_http_header_parse_error(r, |
| NGX_HTTP_PARSE_INVALID_CL_HEADER); |
| ngx_http_finalize_request(r, NGX_HTTP_BAD_REQUEST); |
| return; |
| } |
| } |
| |
| rev->event_handler = ngx_http_block_read; |
| ngx_http_handler(r); |
| return; |
| |
| } else if (rc != NGX_AGAIN) { |
| |
| /* there was error while a header line parsing */ |
| |
| ngx_http_header_parse_error(r, rc); |
| ngx_http_finalize_request(r, NGX_HTTP_BAD_REQUEST); |
| |
| return; |
| } |
| |
| /* NGX_AGAIN: a header line parsing is still not complete */ |
| |
| if (r->header_in->last == r->header_in->end) { |
| |
| /* if the large client headers are enabled then |
| we need to compact r->header_in hunk */ |
| |
| if (cscf->large_client_header) { |
| offset = r->header_name_start - r->header_in->start; |
| |
| if (offset == 0) { |
| ngx_http_header_parse_error(r, |
| NGX_HTTP_PARSE_TOO_LONG_HEADER); |
| ngx_http_finalize_request(r, NGX_HTTP_BAD_REQUEST); |
| return; |
| } |
| |
| ngx_memcpy(r->header_in->start, r->header_name_start, |
| r->header_in->last - r->header_name_start); |
| |
| r->header_in->last -= offset; |
| r->header_in->pos -= offset; |
| r->header_name_start = r->header_in->start; |
| r->header_name_end -= offset; |
| r->header_start -= offset; |
| r->header_end -= offset; |
| |
| } else { |
| ngx_http_header_parse_error(r, NGX_HTTP_PARSE_TOO_LONG_HEADER); |
| ngx_http_finalize_request(r, NGX_HTTP_BAD_REQUEST); |
| return; |
| } |
| } |
| } |
| } |
| |
| |
| static ssize_t ngx_http_read_request_header(ngx_http_request_t *r) |
| { |
| int event; |
| ssize_t n; |
| ngx_event_t *rev; |
| ngx_http_core_srv_conf_t *cscf; |
| |
| rev = r->connection->read; |
| |
| n = r->header_in->last - r->header_in->pos; |
| |
| if (n > 0) { |
| rev->ready = 0; |
| return n; |
| } |
| |
| n = ngx_recv(r->connection, r->header_in->last, |
| r->header_in->end - r->header_in->last); |
| |
| if (n == NGX_AGAIN) { |
| if (!r->header_timeout_set) { |
| if (rev->timer_set) { |
| ngx_del_timer(rev); |
| } else { |
| rev->timer_set = 1; |
| } |
| |
| cscf = ngx_http_get_module_srv_conf(r, ngx_http_core_module); |
| |
| ngx_add_timer(rev, cscf->client_header_timeout); |
| r->header_timeout_set = 1; |
| } |
| |
| if (!rev->active) { |
| if (ngx_event_flags & NGX_HAVE_CLEAR_EVENT) { |
| /* kqueue */ |
| event = NGX_CLEAR_EVENT; |
| |
| } else { |
| /* select, poll, /dev/poll */ |
| event = NGX_LEVEL_EVENT; |
| } |
| |
| if (ngx_add_event(rev, NGX_READ_EVENT, event) == NGX_ERROR) { |
| ngx_http_close_request(r, NGX_HTTP_INTERNAL_SERVER_ERROR); |
| ngx_http_close_connection(r->connection); |
| return NGX_ERROR; |
| } |
| } |
| |
| return NGX_AGAIN; |
| } |
| |
| if (n == 0) { |
| ngx_log_error(NGX_LOG_INFO, r->connection->log, 0, |
| "client closed prematurely connection"); |
| } |
| |
| if (n == 0 || n == NGX_ERROR) { |
| ngx_http_close_request(r, NGX_HTTP_BAD_REQUEST); |
| ngx_http_close_connection(r->connection); |
| return NGX_ERROR; |
| } |
| |
| r->header_in->last += n; |
| |
| return n; |
| } |
| |
| |
| void ngx_http_finalize_request(ngx_http_request_t *r, int error) |
| { |
| int rc; |
| ngx_event_t *rev, *wev; |
| |
| rc = error; |
| |
| if (rc >= NGX_HTTP_SPECIAL_RESPONSE) { |
| |
| rev = r->connection->read; |
| if (rev->timer_set) { |
| ngx_del_timer(rev); |
| rev->timer_set = 0; |
| } |
| |
| wev = r->connection->write; |
| if (wev->timer_set) { |
| ngx_del_timer(wev); |
| wev->timer_set = 0; |
| } |
| |
| rc = ngx_http_special_response_handler(r, rc); |
| |
| if (rc == NGX_AGAIN) { |
| return; |
| } |
| |
| if (rc == NGX_ERROR) { |
| ngx_http_close_request(r, 0); |
| ngx_http_close_connection(r->connection); |
| return; |
| } |
| |
| } else if (rc == NGX_ERROR) { |
| r->keepalive = 0; |
| r->lingering_close = 0; |
| } |
| |
| rev = r->connection->read; |
| if (rev->timer_set) { |
| ngx_del_timer(rev); |
| rev->timer_set = 0; |
| } |
| |
| wev = r->connection->write; |
| if (wev->timer_set) { |
| ngx_del_timer(wev); |
| wev->timer_set = 0; |
| } |
| |
| if (r->keepalive != 0) { |
| ngx_http_set_keepalive(r); |
| |
| } else if (r->lingering_close) { |
| ngx_http_set_lingering_close(r); |
| |
| } else { |
| ngx_http_close_request(r, 0); |
| ngx_http_close_connection(r->connection); |
| } |
| } |
| |
| |
| void ngx_http_set_write_handler(ngx_http_request_t *r) |
| { |
| int event; |
| ngx_event_t *wev; |
| ngx_http_core_loc_conf_t *clcf; |
| |
| wev = r->connection->write; |
| wev->event_handler = ngx_http_writer; |
| |
| if (wev->delayed && wev->ready) { |
| return; |
| } |
| |
| clcf = ngx_http_get_module_loc_conf(r->main ? r->main : r, |
| ngx_http_core_module); |
| ngx_add_timer(wev, clcf->send_timeout); |
| wev->timer_set = 1; |
| |
| if (ngx_event_flags & (NGX_HAVE_AIO_EVENT|NGX_HAVE_EDGE_EVENT)) { |
| /* aio, iocp, epoll */ |
| return; |
| } |
| |
| #if (HAVE_LOWAT_EVENT) /* kqueue's NOTE_LOWAT */ |
| |
| if (ngx_event_flags & NGX_HAVE_LOWAT_EVENT) { |
| wev->lowat = clcf->send_lowat; |
| } |
| |
| #endif |
| |
| if (ngx_event_flags & NGX_HAVE_CLEAR_EVENT) { |
| /* kqueue */ |
| event = NGX_CLEAR_EVENT; |
| |
| } else { |
| /* select, poll, /dev/poll */ |
| event = NGX_LEVEL_EVENT; |
| } |
| |
| if (ngx_add_event(wev, NGX_WRITE_EVENT, event) == NGX_ERROR) { |
| ngx_http_close_request(r, 0); |
| ngx_http_close_connection(r->connection); |
| } |
| } |
| |
| |
| static void ngx_http_writer(ngx_event_t *wev) |
| { |
| int rc; |
| ngx_event_t *rev; |
| ngx_connection_t *c; |
| ngx_http_request_t *r; |
| ngx_http_core_loc_conf_t *clcf; |
| |
| c = wev->data; |
| r = c->data; |
| |
| rc = ngx_http_output_filter(r, NULL); |
| |
| ngx_log_debug(c->log, "writer output filter: %d" _ rc); |
| |
| if (rc == NGX_AGAIN) { |
| |
| clcf = ngx_http_get_module_loc_conf(r->main ? r->main : r, |
| ngx_http_core_module); |
| if (wev->timer_set) { |
| ngx_del_timer(wev); |
| } else { |
| wev->timer_set = 1; |
| } |
| |
| ngx_add_timer(wev, clcf->send_timeout); |
| |
| return; |
| } |
| |
| if (rc == NGX_ERROR) { |
| ngx_http_close_request(r, 0); |
| ngx_http_close_connection(c); |
| return; |
| } |
| |
| /* rc == NGX_OK */ |
| |
| ngx_log_debug(c->log, "http writer done"); |
| |
| rev = r->connection->read; |
| if (rev->timer_set) { |
| ngx_del_timer(rev); |
| rev->timer_set = 0; |
| } |
| |
| if (wev->timer_set) { |
| ngx_del_timer(wev); |
| wev->timer_set = 0; |
| } |
| |
| if (r->keepalive != 0) { |
| ngx_http_set_keepalive(r); |
| |
| } else if (r->lingering_close) { |
| ngx_http_set_lingering_close(r); |
| |
| } else { |
| ngx_http_close_request(r, 0); |
| ngx_http_close_connection(r->connection); |
| } |
| } |
| |
| |
| static void ngx_http_block_read(ngx_event_t *rev) |
| { |
| ngx_connection_t *c; |
| ngx_http_request_t *r; |
| |
| ngx_log_debug(rev->log, "http read blocked"); |
| |
| /* aio does not call this handler */ |
| |
| if (ngx_event_flags & NGX_USE_LEVEL_EVENT) { |
| |
| /* select, poll, /dev/poll */ |
| |
| rev->blocked = 1; |
| |
| if (ngx_del_event(rev, NGX_READ_EVENT, 0) == NGX_ERROR) { |
| c = (ngx_connection_t *) rev->data; |
| r = (ngx_http_request_t *) c->data; |
| ngx_http_close_request(r, 0); |
| ngx_http_close_connection(c); |
| } |
| } |
| |
| /* kqueue, epoll */ |
| |
| return; |
| } |
| |
| |
| int ngx_http_discard_body(ngx_http_request_t *r) |
| { |
| ssize_t size; |
| ngx_event_t *rev; |
| |
| rev = r->connection->read; |
| |
| ngx_log_debug(rev->log, "set discard body"); |
| |
| if (rev->timer_set) { |
| ngx_del_timer(rev); |
| rev->timer_set = 0; |
| } |
| |
| if (r->headers_in.content_length_n > 0) { |
| |
| size = r->header_in->last - r->header_in->pos; |
| |
| if (size) { |
| if (r->headers_in.content_length_n > size) { |
| r->headers_in.content_length_n -= size; |
| |
| } else { |
| r->header_in->pos += r->headers_in.content_length_n; |
| r->headers_in.content_length_n = 0; |
| return NGX_OK; |
| } |
| } |
| |
| rev->event_handler = ngx_http_read_discarded_body_event; |
| |
| if (rev->blocked) { |
| if (ngx_event_flags & NGX_USE_LEVEL_EVENT) { |
| if (ngx_add_event(rev, NGX_READ_EVENT, NGX_LEVEL_EVENT) |
| == NGX_ERROR) { |
| return NGX_HTTP_INTERNAL_SERVER_ERROR; |
| } |
| } |
| |
| rev->blocked = 0; |
| return ngx_http_read_discarded_body(r); |
| } |
| } |
| |
| return NGX_OK; |
| } |
| |
| |
| static void ngx_http_read_discarded_body_event(ngx_event_t *rev) |
| { |
| int rc; |
| ngx_connection_t *c; |
| ngx_http_request_t *r; |
| |
| c = rev->data; |
| r = c->data; |
| |
| rc = ngx_http_read_discarded_body(r); |
| |
| if (rc != NGX_OK) { |
| ngx_http_close_request(r, rc); |
| ngx_http_close_connection(c); |
| } |
| } |
| |
| |
| static int ngx_http_read_discarded_body(ngx_http_request_t *r) |
| { |
| size_t size; |
| ssize_t n; |
| ngx_http_core_loc_conf_t *clcf; |
| |
| ngx_log_debug(r->connection->log, "http read discarded body"); |
| |
| clcf = ngx_http_get_module_loc_conf(r, ngx_http_core_module); |
| |
| if (r->discarded_buffer == NULL) { |
| r->discarded_buffer = ngx_palloc(r->pool, clcf->discarded_buffer_size); |
| if (r->discarded_buffer == NULL) { |
| return NGX_HTTP_INTERNAL_SERVER_ERROR; |
| } |
| } |
| |
| size = r->headers_in.content_length_n; |
| if (size > clcf->discarded_buffer_size) { |
| size = clcf->discarded_buffer_size; |
| } |
| |
| n = ngx_recv(r->connection, r->discarded_buffer, size); |
| if (n == NGX_ERROR) { |
| return NGX_HTTP_BAD_REQUEST; |
| } |
| |
| if (n == NGX_AGAIN) { |
| return NGX_OK; |
| } |
| |
| r->headers_in.content_length_n -= n; |
| |
| return NGX_OK; |
| } |
| |
| |
| static void ngx_http_set_keepalive(ngx_http_request_t *r) |
| { |
| int len, blocked; |
| ngx_hunk_t *h; |
| ngx_event_t *rev, *wev; |
| ngx_connection_t *c; |
| ngx_http_log_ctx_t *ctx; |
| ngx_http_core_srv_conf_t *cscf; |
| ngx_http_core_loc_conf_t *clcf; |
| |
| c = r->connection; |
| rev = c->read; |
| |
| ngx_log_debug(c->log, "set http keepalive handler"); |
| |
| ctx = (ngx_http_log_ctx_t *) c->log->data; |
| ctx->action = "closing request"; |
| ngx_http_close_request(r, 0); |
| |
| if (rev->timer_set) { |
| ngx_del_timer(rev); |
| } else { |
| rev->timer_set = 1; |
| } |
| |
| clcf = ngx_http_get_module_loc_conf(r, ngx_http_core_module); |
| |
| ngx_add_timer(rev, clcf->keepalive_timeout); |
| |
| if (rev->blocked && (ngx_event_flags & NGX_USE_LEVEL_EVENT)) { |
| if (ngx_add_event(rev, NGX_READ_EVENT, NGX_LEVEL_EVENT) == NGX_ERROR) { |
| ngx_http_close_connection(c); |
| return; |
| } |
| |
| blocked = 1; |
| rev->blocked = 0; |
| |
| } else { |
| blocked = 0; |
| } |
| |
| h = c->buffer; |
| |
| /* pipelined request */ |
| if (h->pos < h->last) { |
| |
| /* We do not know here whether a pipelined request is complete |
| so if the large client headers are not enabled |
| we need to copy the data to the start of c->buffer. |
| This copy should be rare because clients that support |
| pipelined requests (Mozilla 1.x, Opera 6.x) are still rare */ |
| |
| cscf = ngx_http_get_module_srv_conf(r, ngx_http_core_module); |
| |
| if (!cscf->large_client_header) { |
| len = h->last - h->pos; |
| ngx_memcpy(h->start, h->pos, len); |
| h->pos = h->start; |
| h->last = h->start + len; |
| } |
| |
| ngx_log_debug(c->log, "pipelined request"); |
| |
| c->pipeline = 1; |
| ctx->action = "reading client pipelined request line"; |
| ngx_http_init_request(rev); |
| return; |
| } |
| |
| c->pipeline = 0; |
| |
| h->pos = h->last = h->start; |
| rev->event_handler = ngx_http_keepalive_handler; |
| wev = c->write; |
| |
| if (wev->active) { |
| if (ngx_event_flags & NGX_USE_LEVEL_EVENT) { |
| if (ngx_del_event(wev, NGX_WRITE_EVENT, 0) == NGX_ERROR) { |
| ngx_http_close_connection(c); |
| return; |
| } |
| |
| } else if ((ngx_event_flags & NGX_HAVE_AIO_EVENT) == 0) { |
| wev->event_handler = ngx_http_empty_handler; |
| } |
| } |
| |
| ctx->action = "keepalive"; |
| |
| if (c->tcp_nopush) { |
| if (ngx_tcp_push(c->fd) == NGX_ERROR) { |
| ngx_log_error(NGX_LOG_CRIT, c->log, ngx_socket_errno, |
| ngx_tcp_push_n " failed"); |
| ngx_http_close_connection(c); |
| return; |
| } |
| c->tcp_nopush = 0; |
| } |
| |
| if ((ngx_event_flags & NGX_HAVE_AIO_EVENT) || blocked) { |
| ngx_http_keepalive_handler(rev); |
| } |
| } |
| |
| |
| static void ngx_http_keepalive_handler(ngx_event_t *rev) |
| { |
| ssize_t n; |
| ngx_connection_t *c; |
| ngx_http_log_ctx_t *lctx; |
| |
| c = (ngx_connection_t *) rev->data; |
| |
| ngx_log_debug(c->log, "http keepalive handler"); |
| |
| if (rev->timedout) { |
| ngx_http_close_connection(c); |
| return; |
| } |
| |
| /* MSIE closes a keepalive connection with RST flag |
| so we ignore ECONNRESET here */ |
| |
| rev->ignore_econnreset = 1; |
| ngx_set_socket_errno(0); |
| n = ngx_recv(c, c->buffer->last, c->buffer->end - c->buffer->last); |
| rev->ignore_econnreset = 0; |
| |
| if (n == NGX_AGAIN) { |
| return; |
| } |
| |
| if (n == NGX_ERROR) { |
| ngx_http_close_connection(c); |
| return; |
| } |
| |
| lctx = (ngx_http_log_ctx_t *) rev->log->data; |
| rev->log->handler = NULL; |
| |
| if (n == 0) { |
| ngx_log_error(NGX_LOG_INFO, c->log, ngx_socket_errno, |
| "client %s closed keepalive connection", lctx->client); |
| ngx_http_close_connection(c); |
| return; |
| } |
| |
| c->buffer->last += n; |
| rev->log->handler = ngx_http_log_error; |
| lctx->action = "reading client request line"; |
| |
| ngx_http_init_request(rev); |
| } |
| |
| |
| static void ngx_http_set_lingering_close(ngx_http_request_t *r) |
| { |
| ngx_event_t *rev; |
| ngx_connection_t *c; |
| ngx_http_core_loc_conf_t *clcf; |
| |
| c = r->connection; |
| rev = c->read; |
| |
| clcf = ngx_http_get_module_loc_conf(r, ngx_http_core_module); |
| |
| r->lingering_time = ngx_time() + clcf->lingering_time / 1000; |
| rev->event_handler = ngx_http_lingering_close_handler; |
| |
| if (rev->timer_set) { |
| ngx_del_timer(rev); |
| } else { |
| rev->timer_set = 1; |
| } |
| |
| ngx_add_timer(rev, clcf->lingering_timeout); |
| |
| if (rev->blocked && (ngx_event_flags & NGX_USE_LEVEL_EVENT)) { |
| if (ngx_add_event(rev, NGX_READ_EVENT, NGX_LEVEL_EVENT) == NGX_ERROR) { |
| ngx_http_close_request(r, 0); |
| ngx_http_close_connection(c); |
| return; |
| } |
| rev->blocked = 0; |
| } |
| |
| if (c->write->active) { |
| if (ngx_event_flags & NGX_USE_LEVEL_EVENT) { |
| if (ngx_del_event(c->write, NGX_WRITE_EVENT, 0) == NGX_ERROR) { |
| ngx_http_close_request(r, 0); |
| ngx_http_close_connection(c); |
| return; |
| } |
| |
| } else if ((ngx_event_flags & NGX_HAVE_AIO_EVENT) == 0) { |
| c->write->event_handler = ngx_http_empty_handler; |
| } |
| } |
| |
| if (ngx_shutdown_socket(c->fd, NGX_WRITE_SHUTDOWN) == -1) { |
| ngx_log_error(NGX_LOG_CRIT, c->log, ngx_socket_errno, |
| ngx_shutdown_socket_n " failed"); |
| ngx_http_close_request(r, 0); |
| ngx_http_close_connection(c); |
| return; |
| } |
| |
| if (rev->ready || (ngx_event_flags & NGX_HAVE_AIO_EVENT)) { |
| ngx_http_lingering_close_handler(rev); |
| } |
| } |
| |
| |
| static void ngx_http_lingering_close_handler(ngx_event_t *rev) |
| { |
| ssize_t n; |
| ngx_msec_t timer; |
| ngx_connection_t *c; |
| ngx_http_request_t *r; |
| ngx_http_core_loc_conf_t *clcf; |
| |
| c = rev->data; |
| r = c->data; |
| |
| ngx_log_debug(c->log, "http lingering close handler"); |
| |
| if (rev->timedout) { |
| ngx_http_close_request(r, 0); |
| ngx_http_close_connection(c); |
| return; |
| } |
| |
| timer = r->lingering_time - ngx_time(); |
| if (timer <= 0) { |
| ngx_http_close_request(r, 0); |
| ngx_http_close_connection(c); |
| return; |
| } |
| |
| clcf = ngx_http_get_module_loc_conf(r, ngx_http_core_module); |
| |
| if (r->discarded_buffer == NULL) { |
| |
| /* TODO: r->header_in->start (if large headers are enabled) |
| or the end of parsed header (otherwise) |
| instead of r->header_in->last */ |
| |
| if ((size_t)(r->header_in->end - r->header_in->last) |
| >= clcf->discarded_buffer_size) { |
| r->discarded_buffer = r->header_in->last; |
| |
| } else { |
| r->discarded_buffer = ngx_palloc(c->pool, |
| clcf->discarded_buffer_size); |
| if (r->discarded_buffer) { |
| ngx_http_close_request(r, 0); |
| ngx_http_close_connection(c); |
| return; |
| } |
| } |
| } |
| |
| do { |
| n = ngx_recv(c, r->discarded_buffer, clcf->discarded_buffer_size); |
| |
| ngx_log_debug(c->log, "lingering read: %d" _ n); |
| |
| if (n == NGX_ERROR || n == 0) { |
| ngx_http_close_request(r, 0); |
| ngx_http_close_connection(c); |
| return; |
| } |
| |
| } while (rev->ready); |
| |
| timer *= 1000; |
| if (timer > clcf->lingering_timeout) { |
| timer = clcf->lingering_timeout; |
| } |
| |
| if (rev->timer_set) { |
| ngx_del_timer(rev); |
| } else { |
| rev->timer_set = 1; |
| } |
| ngx_add_timer(rev, timer); |
| |
| return; |
| } |
| |
| |
| static void ngx_http_empty_handler(ngx_event_t *wev) |
| { |
| ngx_log_debug(wev->log, "http empty handler"); |
| |
| return; |
| } |
| |
| |
| void ngx_http_close_request(ngx_http_request_t *r, int error) |
| { |
| ngx_http_log_ctx_t *ctx; |
| |
| ngx_log_debug(r->connection->log, "close http request"); |
| |
| if (r->pool == NULL) { |
| ngx_log_error(NGX_LOG_ALERT, r->connection->log, 0, |
| "http request already closed"); |
| return; |
| } |
| |
| if (error) { |
| r->headers_out.status = error; |
| } |
| |
| ngx_http_log_handler(r); |
| |
| if (r->file.fd != NGX_INVALID_FILE) { |
| if (ngx_close_file(r->file.fd) == NGX_FILE_ERROR) { |
| ngx_log_error(NGX_LOG_ALERT, r->connection->log, ngx_errno, |
| ngx_close_file_n " \"%s\" failed", r->file.name.data); |
| } |
| } |
| |
| /* ctx->url was allocated from r->pool */ |
| ctx = (ngx_http_log_ctx_t *) r->connection->log->data; |
| ctx->url = NULL; |
| |
| ngx_destroy_pool(r->pool); |
| } |
| |
| |
| void ngx_http_close_connection(ngx_connection_t *c) |
| { |
| ngx_log_debug(c->log, "close connection: %d" _ c->fd); |
| |
| if (c->pool == NULL) { |
| ngx_log_error(NGX_LOG_ALERT, c->log, 0, "connection already closed"); |
| return; |
| } |
| |
| if (c->read->timer_set) { |
| ngx_del_timer(c->read); |
| c->read->timer_set = 0; |
| } |
| |
| if (c->write->timer_set) { |
| ngx_del_timer(c->write); |
| c->write->timer_set = 0; |
| } |
| |
| if (ngx_del_conn) { |
| ngx_del_conn(c); |
| |
| } else { |
| if (c->read->active) { |
| ngx_del_event(c->read, NGX_READ_EVENT, NGX_CLOSE_EVENT); |
| } |
| |
| if (c->write->active) { |
| ngx_del_event(c->write, NGX_WRITE_EVENT, NGX_CLOSE_EVENT); |
| } |
| } |
| |
| if (ngx_close_socket(c->fd) == -1) { |
| ngx_log_error(NGX_LOG_ALERT, c->log, ngx_socket_errno, |
| ngx_close_socket_n " failed"); |
| } |
| |
| c->fd = -1; |
| |
| ngx_destroy_pool(c->pool); |
| } |
| |
| |
| static void ngx_http_header_parse_error(ngx_http_request_t *r, int parse_err) |
| { |
| ngx_http_log_ctx_t *ctx; |
| |
| ctx = r->connection->log->data; |
| r->connection->log->handler = NULL; |
| |
| if (ctx->url) { |
| ngx_log_error(NGX_LOG_ERR, r->connection->log, 0, |
| header_errors[parse_err - NGX_HTTP_PARSE_INVALID_METHOD], |
| ctx->client, ctx->url); |
| |
| } else { |
| ngx_log_error(NGX_LOG_ERR, r->connection->log, 0, |
| header_errors[parse_err - NGX_HTTP_PARSE_INVALID_METHOD], |
| ctx->client); |
| } |
| |
| r->connection->log->handler = ngx_http_log_error; |
| } |
| |
| |
| static size_t ngx_http_log_error(void *data, char *buf, size_t len) |
| { |
| ngx_http_log_ctx_t *ctx = (ngx_http_log_ctx_t *) data; |
| |
| if (ctx->action && ctx->url) { |
| return ngx_snprintf(buf, len, " while %s, client: %s, URL: %s", |
| ctx->action, ctx->client, ctx->url); |
| |
| } else if (ctx->action == NULL && ctx->url) { |
| return ngx_snprintf(buf, len, ", client: %s, URL: %s", |
| ctx->client, ctx->url); |
| |
| } else { |
| return ngx_snprintf(buf, len, " while %s, client: %s", |
| ctx->action, ctx->client); |
| } |
| } |
