| |
| /* |
| * Copyright (C) Igor Sysoev |
| */ |
| |
| |
| #include <ngx_config.h> |
| #include <ngx_core.h> |
| #include <ngx_http.h> |
| |
| |
| #if (NGX_HTTP_CACHE) |
| static ngx_int_t ngx_http_upstream_cache(ngx_http_request_t *r, |
| ngx_http_upstream_t *u); |
| static ngx_int_t ngx_http_upstream_cache_send(ngx_http_request_t *r, |
| ngx_http_upstream_t *u); |
| static ngx_int_t ngx_http_upstream_cache_status(ngx_http_request_t *r, |
| ngx_http_variable_value_t *v, uintptr_t data); |
| #endif |
| |
| static void ngx_http_upstream_init_request(ngx_http_request_t *r); |
| static void ngx_http_upstream_resolve_handler(ngx_resolver_ctx_t *ctx); |
| static void ngx_http_upstream_rd_check_broken_connection(ngx_http_request_t *r); |
| static void ngx_http_upstream_wr_check_broken_connection(ngx_http_request_t *r); |
| static void ngx_http_upstream_check_broken_connection(ngx_http_request_t *r, |
| ngx_event_t *ev); |
| static void ngx_http_upstream_connect(ngx_http_request_t *r, |
| ngx_http_upstream_t *u); |
| static ngx_int_t ngx_http_upstream_reinit(ngx_http_request_t *r, |
| ngx_http_upstream_t *u); |
| static void ngx_http_upstream_send_request(ngx_http_request_t *r, |
| ngx_http_upstream_t *u); |
| static void ngx_http_upstream_send_request_handler(ngx_http_request_t *r, |
| ngx_http_upstream_t *u); |
| static void ngx_http_upstream_process_header(ngx_http_request_t *r, |
| ngx_http_upstream_t *u); |
| static ngx_int_t ngx_http_upstream_test_next(ngx_http_request_t *r, |
| ngx_http_upstream_t *u); |
| static ngx_int_t ngx_http_upstream_intercept_errors(ngx_http_request_t *r, |
| ngx_http_upstream_t *u); |
| static ngx_int_t ngx_http_upstream_test_connect(ngx_connection_t *c); |
| static ngx_int_t ngx_http_upstream_process_headers(ngx_http_request_t *r, |
| ngx_http_upstream_t *u); |
| static void ngx_http_upstream_process_body_in_memory(ngx_http_request_t *r, |
| ngx_http_upstream_t *u); |
| static void ngx_http_upstream_send_response(ngx_http_request_t *r, |
| ngx_http_upstream_t *u); |
| static void |
| ngx_http_upstream_process_non_buffered_downstream(ngx_http_request_t *r); |
| static void |
| ngx_http_upstream_process_non_buffered_upstream(ngx_http_request_t *r, |
| ngx_http_upstream_t *u); |
| static void |
| ngx_http_upstream_process_non_buffered_request(ngx_http_request_t *r, |
| ngx_uint_t do_write); |
| static ngx_int_t ngx_http_upstream_non_buffered_filter_init(void *data); |
| static ngx_int_t ngx_http_upstream_non_buffered_filter(void *data, |
| ssize_t bytes); |
| static void ngx_http_upstream_process_downstream(ngx_http_request_t *r); |
| static void ngx_http_upstream_process_upstream(ngx_http_request_t *r, |
| ngx_http_upstream_t *u); |
| static void ngx_http_upstream_process_request(ngx_http_request_t *r); |
| static void ngx_http_upstream_store(ngx_http_request_t *r, |
| ngx_http_upstream_t *u); |
| static void ngx_http_upstream_dummy_handler(ngx_http_request_t *r, |
| ngx_http_upstream_t *u); |
| static void ngx_http_upstream_next(ngx_http_request_t *r, |
| ngx_http_upstream_t *u, ngx_uint_t ft_type); |
| static void ngx_http_upstream_cleanup(void *data); |
| static void ngx_http_upstream_finalize_request(ngx_http_request_t *r, |
| ngx_http_upstream_t *u, ngx_int_t rc); |
| |
| static ngx_int_t ngx_http_upstream_process_header_line(ngx_http_request_t *r, |
| ngx_table_elt_t *h, ngx_uint_t offset); |
| static ngx_int_t |
| ngx_http_upstream_process_cache_control(ngx_http_request_t *r, |
| ngx_table_elt_t *h, ngx_uint_t offset); |
| static ngx_int_t ngx_http_upstream_ignore_header_line(ngx_http_request_t *r, |
| ngx_table_elt_t *h, ngx_uint_t offset); |
| static ngx_int_t ngx_http_upstream_process_expires(ngx_http_request_t *r, |
| ngx_table_elt_t *h, ngx_uint_t offset); |
| static ngx_int_t ngx_http_upstream_process_accel_expires(ngx_http_request_t *r, |
| ngx_table_elt_t *h, ngx_uint_t offset); |
| static ngx_int_t ngx_http_upstream_process_limit_rate(ngx_http_request_t *r, |
| ngx_table_elt_t *h, ngx_uint_t offset); |
| static ngx_int_t ngx_http_upstream_process_buffering(ngx_http_request_t *r, |
| ngx_table_elt_t *h, ngx_uint_t offset); |
| static ngx_int_t ngx_http_upstream_process_charset(ngx_http_request_t *r, |
| ngx_table_elt_t *h, ngx_uint_t offset); |
| static ngx_int_t ngx_http_upstream_copy_header_line(ngx_http_request_t *r, |
| ngx_table_elt_t *h, ngx_uint_t offset); |
| static ngx_int_t |
| ngx_http_upstream_copy_multi_header_lines(ngx_http_request_t *r, |
| ngx_table_elt_t *h, ngx_uint_t offset); |
| static ngx_int_t ngx_http_upstream_copy_content_type(ngx_http_request_t *r, |
| ngx_table_elt_t *h, ngx_uint_t offset); |
| static ngx_int_t ngx_http_upstream_copy_content_length(ngx_http_request_t *r, |
| ngx_table_elt_t *h, ngx_uint_t offset); |
| static ngx_int_t ngx_http_upstream_copy_last_modified(ngx_http_request_t *r, |
| ngx_table_elt_t *h, ngx_uint_t offset); |
| static ngx_int_t ngx_http_upstream_rewrite_location(ngx_http_request_t *r, |
| ngx_table_elt_t *h, ngx_uint_t offset); |
| static ngx_int_t ngx_http_upstream_rewrite_refresh(ngx_http_request_t *r, |
| ngx_table_elt_t *h, ngx_uint_t offset); |
| static ngx_int_t ngx_http_upstream_copy_allow_ranges(ngx_http_request_t *r, |
| ngx_table_elt_t *h, ngx_uint_t offset); |
| |
| #if (NGX_HTTP_GZIP) |
| static ngx_int_t ngx_http_upstream_copy_content_encoding(ngx_http_request_t *r, |
| ngx_table_elt_t *h, ngx_uint_t offset); |
| #endif |
| |
| static ngx_int_t ngx_http_upstream_add_variables(ngx_conf_t *cf); |
| static ngx_int_t ngx_http_upstream_addr_variable(ngx_http_request_t *r, |
| ngx_http_variable_value_t *v, uintptr_t data); |
| static ngx_int_t ngx_http_upstream_status_variable(ngx_http_request_t *r, |
| ngx_http_variable_value_t *v, uintptr_t data); |
| static ngx_int_t ngx_http_upstream_response_time_variable(ngx_http_request_t *r, |
| ngx_http_variable_value_t *v, uintptr_t data); |
| static ngx_int_t ngx_http_upstream_response_length_variable( |
| ngx_http_request_t *r, ngx_http_variable_value_t *v, uintptr_t data); |
| |
| static char *ngx_http_upstream(ngx_conf_t *cf, ngx_command_t *cmd, void *dummy); |
| static char *ngx_http_upstream_server(ngx_conf_t *cf, ngx_command_t *cmd, |
| void *conf); |
| |
| static void *ngx_http_upstream_create_main_conf(ngx_conf_t *cf); |
| static char *ngx_http_upstream_init_main_conf(ngx_conf_t *cf, void *conf); |
| |
| #if (NGX_HTTP_SSL) |
| static void ngx_http_upstream_ssl_init_connection(ngx_http_request_t *, |
| ngx_http_upstream_t *u, ngx_connection_t *c); |
| static void ngx_http_upstream_ssl_handshake(ngx_connection_t *c); |
| #endif |
| |
| |
| ngx_http_upstream_header_t ngx_http_upstream_headers_in[] = { |
| |
| { ngx_string("Status"), |
| ngx_http_upstream_process_header_line, |
| offsetof(ngx_http_upstream_headers_in_t, status), |
| ngx_http_upstream_copy_header_line, 0, 0 }, |
| |
| { ngx_string("Content-Type"), |
| ngx_http_upstream_process_header_line, |
| offsetof(ngx_http_upstream_headers_in_t, content_type), |
| ngx_http_upstream_copy_content_type, 0, 1 }, |
| |
| { ngx_string("Content-Length"), |
| ngx_http_upstream_process_header_line, |
| offsetof(ngx_http_upstream_headers_in_t, content_length), |
| ngx_http_upstream_copy_content_length, 0, 0 }, |
| |
| { ngx_string("Date"), |
| ngx_http_upstream_process_header_line, |
| offsetof(ngx_http_upstream_headers_in_t, date), |
| ngx_http_upstream_copy_header_line, |
| offsetof(ngx_http_headers_out_t, date), 0 }, |
| |
| { ngx_string("Last-Modified"), |
| ngx_http_upstream_process_header_line, |
| offsetof(ngx_http_upstream_headers_in_t, last_modified), |
| ngx_http_upstream_copy_last_modified, 0, 0 }, |
| |
| { ngx_string("ETag"), |
| ngx_http_upstream_process_header_line, |
| offsetof(ngx_http_upstream_headers_in_t, etag), |
| ngx_http_upstream_copy_header_line, |
| offsetof(ngx_http_headers_out_t, etag), 0 }, |
| |
| { ngx_string("Server"), |
| ngx_http_upstream_process_header_line, |
| offsetof(ngx_http_upstream_headers_in_t, server), |
| ngx_http_upstream_copy_header_line, |
| offsetof(ngx_http_headers_out_t, server), 0 }, |
| |
| { ngx_string("WWW-Authenticate"), |
| ngx_http_upstream_process_header_line, |
| offsetof(ngx_http_upstream_headers_in_t, www_authenticate), |
| ngx_http_upstream_copy_header_line, 0, 0 }, |
| |
| { ngx_string("Location"), |
| ngx_http_upstream_process_header_line, |
| offsetof(ngx_http_upstream_headers_in_t, location), |
| ngx_http_upstream_rewrite_location, 0, 0 }, |
| |
| { ngx_string("Refresh"), |
| ngx_http_upstream_ignore_header_line, 0, |
| ngx_http_upstream_rewrite_refresh, 0, 0 }, |
| |
| { ngx_string("Set-Cookie"), |
| ngx_http_upstream_ignore_header_line, 0, |
| ngx_http_upstream_copy_header_line, 0, 1 }, |
| |
| { ngx_string("Content-Disposition"), |
| ngx_http_upstream_ignore_header_line, 0, |
| ngx_http_upstream_copy_header_line, 0, 1 }, |
| |
| { ngx_string("Cache-Control"), |
| ngx_http_upstream_process_cache_control, 0, |
| ngx_http_upstream_copy_multi_header_lines, |
| offsetof(ngx_http_headers_out_t, cache_control), 1 }, |
| |
| { ngx_string("Expires"), |
| ngx_http_upstream_process_expires, 0, |
| ngx_http_upstream_copy_header_line, |
| offsetof(ngx_http_headers_out_t, expires), 1 }, |
| |
| { ngx_string("Accept-Ranges"), |
| ngx_http_upstream_process_header_line, |
| offsetof(ngx_http_upstream_headers_in_t, accept_ranges), |
| ngx_http_upstream_copy_allow_ranges, |
| offsetof(ngx_http_headers_out_t, accept_ranges), 1 }, |
| |
| { ngx_string("Connection"), |
| ngx_http_upstream_ignore_header_line, 0, |
| ngx_http_upstream_ignore_header_line, 0, 0 }, |
| |
| { ngx_string("Keep-Alive"), |
| ngx_http_upstream_ignore_header_line, 0, |
| ngx_http_upstream_ignore_header_line, 0, 0 }, |
| |
| { ngx_string("X-Powered-By"), |
| ngx_http_upstream_ignore_header_line, 0, |
| ngx_http_upstream_copy_header_line, 0, 0 }, |
| |
| { ngx_string("X-Accel-Expires"), |
| ngx_http_upstream_process_accel_expires, 0, |
| ngx_http_upstream_copy_header_line, 0, 0 }, |
| |
| { ngx_string("X-Accel-Redirect"), |
| ngx_http_upstream_process_header_line, |
| offsetof(ngx_http_upstream_headers_in_t, x_accel_redirect), |
| ngx_http_upstream_copy_header_line, 0, 0 }, |
| |
| { ngx_string("X-Accel-Limit-Rate"), |
| ngx_http_upstream_process_limit_rate, 0, |
| ngx_http_upstream_copy_header_line, 0, 0 }, |
| |
| { ngx_string("X-Accel-Buffering"), |
| ngx_http_upstream_process_buffering, 0, |
| ngx_http_upstream_copy_header_line, 0, 0 }, |
| |
| { ngx_string("X-Accel-Charset"), |
| ngx_http_upstream_process_charset, 0, |
| ngx_http_upstream_copy_header_line, 0, 0 }, |
| |
| #if (NGX_HTTP_GZIP) |
| { ngx_string("Content-Encoding"), |
| ngx_http_upstream_process_header_line, |
| offsetof(ngx_http_upstream_headers_in_t, content_encoding), |
| ngx_http_upstream_copy_content_encoding, 0, 0 }, |
| #endif |
| |
| { ngx_null_string, NULL, 0, NULL, 0, 0 } |
| }; |
| |
| |
| static ngx_command_t ngx_http_upstream_commands[] = { |
| |
| { ngx_string("upstream"), |
| NGX_HTTP_MAIN_CONF|NGX_CONF_BLOCK|NGX_CONF_TAKE1, |
| ngx_http_upstream, |
| 0, |
| 0, |
| NULL }, |
| |
| { ngx_string("server"), |
| NGX_HTTP_UPS_CONF|NGX_CONF_1MORE, |
| ngx_http_upstream_server, |
| NGX_HTTP_SRV_CONF_OFFSET, |
| 0, |
| NULL }, |
| |
| ngx_null_command |
| }; |
| |
| |
| static ngx_http_module_t ngx_http_upstream_module_ctx = { |
| ngx_http_upstream_add_variables, /* preconfiguration */ |
| NULL, /* postconfiguration */ |
| |
| ngx_http_upstream_create_main_conf, /* create main configuration */ |
| ngx_http_upstream_init_main_conf, /* init main configuration */ |
| |
| NULL, /* create server configuration */ |
| NULL, /* merge server configuration */ |
| |
| NULL, /* create location configuration */ |
| NULL /* merge location configuration */ |
| }; |
| |
| |
| ngx_module_t ngx_http_upstream_module = { |
| NGX_MODULE_V1, |
| &ngx_http_upstream_module_ctx, /* module context */ |
| ngx_http_upstream_commands, /* module directives */ |
| NGX_HTTP_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_http_variable_t ngx_http_upstream_vars[] = { |
| |
| { ngx_string("upstream_addr"), NULL, |
| ngx_http_upstream_addr_variable, 0, |
| NGX_HTTP_VAR_NOHASH|NGX_HTTP_VAR_NOCACHEABLE, 0 }, |
| |
| { ngx_string("upstream_status"), NULL, |
| ngx_http_upstream_status_variable, 0, |
| NGX_HTTP_VAR_NOHASH|NGX_HTTP_VAR_NOCACHEABLE, 0 }, |
| |
| { ngx_string("upstream_response_time"), NULL, |
| ngx_http_upstream_response_time_variable, 0, |
| NGX_HTTP_VAR_NOHASH|NGX_HTTP_VAR_NOCACHEABLE, 0 }, |
| |
| { ngx_string("upstream_response_length"), NULL, |
| ngx_http_upstream_response_length_variable, 0, |
| NGX_HTTP_VAR_NOHASH|NGX_HTTP_VAR_NOCACHEABLE, 0 }, |
| |
| #if (NGX_HTTP_CACHE) |
| |
| { ngx_string("upstream_cache_status"), NULL, |
| ngx_http_upstream_cache_status, 0, |
| NGX_HTTP_VAR_NOHASH|NGX_HTTP_VAR_NOCACHEABLE, 0 }, |
| |
| #endif |
| |
| { ngx_null_string, NULL, NULL, 0, 0, 0 } |
| }; |
| |
| |
| static ngx_http_upstream_next_t ngx_http_upstream_next_errors[] = { |
| { 500, NGX_HTTP_UPSTREAM_FT_HTTP_500 }, |
| { 502, NGX_HTTP_UPSTREAM_FT_HTTP_502 }, |
| { 503, NGX_HTTP_UPSTREAM_FT_HTTP_503 }, |
| { 504, NGX_HTTP_UPSTREAM_FT_HTTP_504 }, |
| { 404, NGX_HTTP_UPSTREAM_FT_HTTP_404 }, |
| { 0, 0 } |
| }; |
| |
| |
| ngx_conf_bitmask_t ngx_http_upstream_cache_method_mask[] = { |
| { ngx_string("GET"), NGX_HTTP_GET}, |
| { ngx_string("HEAD"), NGX_HTTP_HEAD }, |
| { ngx_string("POST"), NGX_HTTP_POST }, |
| { ngx_null_string, 0 } |
| }; |
| |
| |
| ngx_int_t |
| ngx_http_upstream_create(ngx_http_request_t *r) |
| { |
| ngx_http_upstream_t *u; |
| |
| u = r->upstream; |
| |
| if (u && u->cleanup) { |
| ngx_http_upstream_cleanup(r); |
| *u->cleanup = NULL; |
| } |
| |
| u = ngx_pcalloc(r->pool, sizeof(ngx_http_upstream_t)); |
| if (u == NULL) { |
| return NGX_ERROR; |
| } |
| |
| r->upstream = u; |
| |
| u->peer.log = r->connection->log; |
| u->peer.log_error = NGX_ERROR_ERR; |
| #if (NGX_THREADS) |
| u->peer.lock = &r->connection->lock; |
| #endif |
| |
| #if (NGX_HTTP_CACHE) |
| r->cache = NULL; |
| #endif |
| |
| return NGX_OK; |
| } |
| |
| |
| void |
| ngx_http_upstream_init(ngx_http_request_t *r) |
| { |
| ngx_connection_t *c; |
| |
| c = r->connection; |
| |
| ngx_log_debug1(NGX_LOG_DEBUG_HTTP, c->log, 0, |
| "http init upstream, client timer: %d", c->read->timer_set); |
| |
| if (c->read->timer_set) { |
| ngx_del_timer(c->read); |
| } |
| |
| if (ngx_event_flags & NGX_USE_CLEAR_EVENT) { |
| |
| if (!c->write->active) { |
| if (ngx_add_event(c->write, NGX_WRITE_EVENT, NGX_CLEAR_EVENT) |
| == NGX_ERROR) |
| { |
| ngx_http_finalize_request(r, NGX_HTTP_INTERNAL_SERVER_ERROR); |
| return; |
| } |
| } |
| } |
| |
| ngx_http_upstream_init_request(r); |
| } |
| |
| |
| static void |
| ngx_http_upstream_init_request(ngx_http_request_t *r) |
| { |
| ngx_str_t *host; |
| ngx_uint_t i; |
| ngx_resolver_ctx_t *ctx, temp; |
| ngx_http_cleanup_t *cln; |
| ngx_http_upstream_t *u; |
| ngx_http_core_loc_conf_t *clcf; |
| ngx_http_upstream_srv_conf_t *uscf, **uscfp; |
| ngx_http_upstream_main_conf_t *umcf; |
| |
| if (r->aio) { |
| return; |
| } |
| |
| u = r->upstream; |
| |
| #if (NGX_HTTP_CACHE) |
| |
| if (u->conf->cache) { |
| ngx_int_t rc; |
| |
| rc = ngx_http_upstream_cache(r, u); |
| |
| if (rc == NGX_BUSY) { |
| r->write_event_handler = ngx_http_upstream_init_request; |
| return; |
| } |
| |
| r->write_event_handler = ngx_http_request_empty_handler; |
| |
| if (rc == NGX_DONE) { |
| return; |
| } |
| |
| if (rc != NGX_DECLINED) { |
| ngx_http_finalize_request(r, rc); |
| return; |
| } |
| } |
| |
| #endif |
| |
| u->store = (u->conf->store || u->conf->store_lengths); |
| |
| if (!u->store && !r->post_action && !u->conf->ignore_client_abort) { |
| r->read_event_handler = ngx_http_upstream_rd_check_broken_connection; |
| r->write_event_handler = ngx_http_upstream_wr_check_broken_connection; |
| } |
| |
| if (r->request_body) { |
| u->request_bufs = r->request_body->bufs; |
| } |
| |
| if (u->create_request(r) != NGX_OK) { |
| ngx_http_finalize_request(r, NGX_HTTP_INTERNAL_SERVER_ERROR); |
| return; |
| } |
| |
| clcf = ngx_http_get_module_loc_conf(r, ngx_http_core_module); |
| |
| u->output.alignment = clcf->directio_alignment; |
| u->output.pool = r->pool; |
| u->output.bufs.num = 1; |
| u->output.bufs.size = clcf->client_body_buffer_size; |
| u->output.output_filter = ngx_chain_writer; |
| u->output.filter_ctx = &u->writer; |
| |
| u->writer.pool = r->pool; |
| |
| if (r->upstream_states == NULL) { |
| |
| r->upstream_states = ngx_array_create(r->pool, 1, |
| sizeof(ngx_http_upstream_state_t)); |
| if (r->upstream_states == NULL) { |
| ngx_http_finalize_request(r, NGX_HTTP_INTERNAL_SERVER_ERROR); |
| return; |
| } |
| |
| } else { |
| |
| u->state = ngx_array_push(r->upstream_states); |
| if (u->state == NULL) { |
| ngx_http_upstream_finalize_request(r, u, |
| NGX_HTTP_INTERNAL_SERVER_ERROR); |
| return; |
| } |
| |
| ngx_memzero(u->state, sizeof(ngx_http_upstream_state_t)); |
| } |
| |
| cln = ngx_http_cleanup_add(r, 0); |
| if (cln == NULL) { |
| ngx_http_finalize_request(r, NGX_HTTP_INTERNAL_SERVER_ERROR); |
| return; |
| } |
| |
| cln->handler = ngx_http_upstream_cleanup; |
| cln->data = r; |
| u->cleanup = &cln->handler; |
| |
| if (u->resolved == NULL) { |
| |
| uscf = u->conf->upstream; |
| |
| } else { |
| |
| if (u->resolved->sockaddr) { |
| |
| if (ngx_http_upstream_create_round_robin_peer(r, u->resolved) |
| != NGX_OK) |
| { |
| ngx_http_finalize_request(r, NGX_HTTP_INTERNAL_SERVER_ERROR); |
| return; |
| } |
| |
| ngx_http_upstream_connect(r, u); |
| |
| return; |
| } |
| |
| host = &u->resolved->host; |
| |
| umcf = ngx_http_get_module_main_conf(r, ngx_http_upstream_module); |
| |
| uscfp = umcf->upstreams.elts; |
| |
| for (i = 0; i < umcf->upstreams.nelts; i++) { |
| |
| uscf = uscfp[i]; |
| |
| if (uscf->host.len == host->len |
| && ((uscf->port == 0 && u->resolved->no_port) |
| || uscf->port == u->resolved->port) |
| && ngx_memcmp(uscf->host.data, host->data, host->len) == 0) |
| { |
| goto found; |
| } |
| } |
| |
| temp.name = *host; |
| |
| ctx = ngx_resolve_start(clcf->resolver, &temp); |
| if (ctx == NULL) { |
| ngx_http_finalize_request(r, NGX_HTTP_INTERNAL_SERVER_ERROR); |
| return; |
| } |
| |
| if (ctx == NGX_NO_RESOLVER) { |
| ngx_log_error(NGX_LOG_ERR, r->connection->log, 0, |
| "no resolver defined to resolve %V", host); |
| |
| ngx_http_finalize_request(r, NGX_HTTP_BAD_GATEWAY); |
| return; |
| } |
| |
| ctx->name = *host; |
| ctx->type = NGX_RESOLVE_A; |
| ctx->handler = ngx_http_upstream_resolve_handler; |
| ctx->data = r; |
| ctx->timeout = clcf->resolver_timeout; |
| |
| u->resolved->ctx = ctx; |
| |
| if (ngx_resolve_name(ctx) != NGX_OK) { |
| u->resolved->ctx = NULL; |
| ngx_http_finalize_request(r, NGX_HTTP_INTERNAL_SERVER_ERROR); |
| return; |
| } |
| |
| return; |
| } |
| |
| found: |
| |
| if (uscf->peer.init(r, uscf) != NGX_OK) { |
| ngx_http_finalize_request(r, NGX_HTTP_INTERNAL_SERVER_ERROR); |
| return; |
| } |
| |
| ngx_http_upstream_connect(r, u); |
| } |
| |
| |
| #if (NGX_HTTP_CACHE) |
| |
| static ngx_int_t |
| ngx_http_upstream_cache(ngx_http_request_t *r, ngx_http_upstream_t *u) |
| { |
| ngx_int_t rc; |
| ngx_http_cache_t *c; |
| |
| c = r->cache; |
| |
| if (c == NULL) { |
| |
| if (!(r->method & u->conf->cache_methods)) { |
| return NGX_DECLINED; |
| } |
| |
| if (r->method & NGX_HTTP_HEAD) { |
| u->method = ngx_http_core_get_method; |
| } |
| |
| c = ngx_pcalloc(r->pool, sizeof(ngx_http_cache_t)); |
| if (c == NULL) { |
| return NGX_ERROR; |
| } |
| |
| if (ngx_array_init(&c->keys, r->pool, 4, sizeof(ngx_str_t)) != NGX_OK) { |
| return NGX_ERROR; |
| } |
| |
| r->cache = c; |
| c->file.log = r->connection->log; |
| |
| if (u->create_key(r) != NGX_OK) { |
| return NGX_ERROR; |
| } |
| |
| /* TODO: add keys */ |
| |
| ngx_http_file_cache_create_key(r); |
| |
| u->cacheable = 1; |
| |
| c->min_uses = u->conf->cache_min_uses; |
| c->body_start = u->conf->buffer_size; |
| c->file_cache = u->conf->cache->data; |
| |
| u->cache_status = NGX_HTTP_CACHE_MISS; |
| } |
| |
| rc = ngx_http_file_cache_open(r); |
| |
| ngx_log_debug1(NGX_LOG_DEBUG_HTTP, r->connection->log, 0, |
| "http upstream cache: %i", rc); |
| |
| switch (rc) { |
| |
| case NGX_HTTP_CACHE_UPDATING: |
| |
| if (u->conf->cache_use_stale & NGX_HTTP_UPSTREAM_FT_UPDATING) { |
| u->cache_status = rc; |
| rc = NGX_OK; |
| |
| } else { |
| rc = NGX_HTTP_CACHE_STALE; |
| } |
| |
| break; |
| |
| case NGX_OK: |
| u->cache_status = NGX_HTTP_CACHE_HIT; |
| } |
| |
| switch (rc) { |
| |
| case NGX_OK: |
| |
| rc = ngx_http_upstream_cache_send(r, u); |
| |
| if (rc != NGX_HTTP_UPSTREAM_INVALID_HEADER) { |
| return rc; |
| } |
| |
| break; |
| |
| case NGX_HTTP_CACHE_STALE: |
| |
| c->valid_sec = 0; |
| u->buffer.start = NULL; |
| u->cache_status = NGX_HTTP_CACHE_EXPIRED; |
| |
| break; |
| |
| case NGX_DECLINED: |
| |
| if ((size_t) (u->buffer.end - u->buffer.start) < u->conf->buffer_size) { |
| u->buffer.start = NULL; |
| |
| } else { |
| u->buffer.pos = u->buffer.start + c->header_start; |
| u->buffer.last = u->buffer.pos; |
| } |
| |
| break; |
| |
| case NGX_HTTP_CACHE_SCARCE: |
| |
| u->cacheable = 0; |
| |
| break; |
| |
| case NGX_AGAIN: |
| |
| return NGX_BUSY; |
| |
| case NGX_ERROR: |
| |
| return NGX_ERROR; |
| |
| default: |
| |
| /* cached NGX_HTTP_BAD_GATEWAY, NGX_HTTP_GATEWAY_TIME_OUT, etc. */ |
| |
| u->cache_status = NGX_HTTP_CACHE_HIT; |
| |
| return rc; |
| } |
| |
| r->cached = 0; |
| |
| return NGX_DECLINED; |
| } |
| |
| |
| static ngx_int_t |
| ngx_http_upstream_cache_send(ngx_http_request_t *r, ngx_http_upstream_t *u) |
| { |
| ngx_int_t rc; |
| ngx_http_cache_t *c; |
| |
| r->cached = 1; |
| c = r->cache; |
| |
| /* TODO: cache stack */ |
| |
| u->buffer = *c->buf; |
| u->buffer.pos += c->header_start; |
| |
| ngx_memzero(&u->headers_in, sizeof(ngx_http_upstream_headers_in_t)); |
| |
| if (ngx_list_init(&u->headers_in.headers, r->pool, 8, |
| sizeof(ngx_table_elt_t)) |
| != NGX_OK) |
| { |
| return NGX_ERROR; |
| } |
| |
| rc = u->process_header(r); |
| |
| if (rc == NGX_OK) { |
| |
| if (ngx_http_upstream_process_headers(r, u) != NGX_OK) { |
| return NGX_DONE; |
| } |
| |
| return ngx_http_cache_send(r); |
| } |
| |
| if (rc == NGX_ERROR) { |
| return NGX_ERROR; |
| } |
| |
| /* rc == NGX_HTTP_UPSTREAM_INVALID_HEADER */ |
| |
| /* TODO: delete file */ |
| |
| return rc; |
| } |
| |
| #endif |
| |
| |
| static void |
| ngx_http_upstream_resolve_handler(ngx_resolver_ctx_t *ctx) |
| { |
| ngx_http_request_t *r; |
| ngx_http_upstream_resolved_t *ur; |
| |
| r = ctx->data; |
| |
| r->upstream->resolved->ctx = NULL; |
| |
| if (ctx->state) { |
| ngx_log_error(NGX_LOG_ERR, r->connection->log, 0, |
| "%V could not be resolved (%i: %s)", |
| &ctx->name, ctx->state, |
| ngx_resolver_strerror(ctx->state)); |
| |
| ngx_resolve_name_done(ctx); |
| ngx_http_finalize_request(r, NGX_HTTP_BAD_GATEWAY); |
| return; |
| } |
| |
| ur = r->upstream->resolved; |
| ur->naddrs = ctx->naddrs; |
| ur->addrs = ctx->addrs; |
| |
| #if (NGX_DEBUG) |
| { |
| in_addr_t addr; |
| ngx_uint_t i; |
| |
| for (i = 0; i < ctx->naddrs; i++) { |
| addr = ntohl(ur->addrs[i]); |
| |
| ngx_log_debug4(NGX_LOG_DEBUG_HTTP, r->connection->log, 0, |
| "name was resolved to %ud.%ud.%ud.%ud", |
| (addr >> 24) & 0xff, (addr >> 16) & 0xff, |
| (addr >> 8) & 0xff, addr & 0xff); |
| } |
| } |
| #endif |
| |
| if (ngx_http_upstream_create_round_robin_peer(r, ur) != NGX_OK) { |
| ngx_resolve_name_done(ctx); |
| ngx_http_finalize_request(r, NGX_HTTP_INTERNAL_SERVER_ERROR); |
| return; |
| } |
| |
| ngx_resolve_name_done(ctx); |
| |
| ngx_http_upstream_connect(r, r->upstream); |
| } |
| |
| |
| static void |
| ngx_http_upstream_handler(ngx_event_t *ev) |
| { |
| ngx_connection_t *c; |
| ngx_http_request_t *r; |
| ngx_http_log_ctx_t *ctx; |
| ngx_http_upstream_t *u; |
| |
| c = ev->data; |
| r = c->data; |
| |
| u = r->upstream; |
| c = r->connection; |
| |
| ctx = c->log->data; |
| ctx->current_request = r; |
| |
| ngx_log_debug2(NGX_LOG_DEBUG_HTTP, c->log, 0, |
| "http upstream request: \"%V?%V\"", &r->uri, &r->args); |
| |
| if (ev->write) { |
| u->write_event_handler(r, u); |
| |
| } else { |
| u->read_event_handler(r, u); |
| } |
| |
| ngx_http_run_posted_requests(c); |
| } |
| |
| |
| static void |
| ngx_http_upstream_rd_check_broken_connection(ngx_http_request_t *r) |
| { |
| ngx_http_upstream_check_broken_connection(r, r->connection->read); |
| } |
| |
| |
| static void |
| ngx_http_upstream_wr_check_broken_connection(ngx_http_request_t *r) |
| { |
| ngx_http_upstream_check_broken_connection(r, r->connection->write); |
| } |
| |
| |
| static void |
| ngx_http_upstream_check_broken_connection(ngx_http_request_t *r, |
| ngx_event_t *ev) |
| { |
| int n; |
| char buf[1]; |
| ngx_err_t err; |
| ngx_int_t event; |
| ngx_connection_t *c; |
| ngx_http_upstream_t *u; |
| |
| ngx_log_debug2(NGX_LOG_DEBUG_HTTP, ev->log, 0, |
| "http upstream check client, write event:%d, \"%V\"", |
| ev->write, &r->uri); |
| |
| c = r->connection; |
| u = r->upstream; |
| |
| if (c->error) { |
| if ((ngx_event_flags & NGX_USE_LEVEL_EVENT) && ev->active) { |
| |
| event = ev->write ? NGX_WRITE_EVENT : NGX_READ_EVENT; |
| |
| if (ngx_del_event(ev, event, 0) != NGX_OK) { |
| ngx_http_upstream_finalize_request(r, u, |
| NGX_HTTP_INTERNAL_SERVER_ERROR); |
| return; |
| } |
| } |
| |
| if (!u->cacheable) { |
| ngx_http_upstream_finalize_request(r, u, |
| NGX_HTTP_CLIENT_CLOSED_REQUEST); |
| } |
| |
| return; |
| } |
| |
| if (u->peer.connection == NULL) { |
| return; |
| } |
| |
| #if (NGX_HAVE_KQUEUE) |
| |
| if (ngx_event_flags & NGX_USE_KQUEUE_EVENT) { |
| |
| if (!ev->pending_eof) { |
| return; |
| } |
| |
| ev->eof = 1; |
| c->error = 1; |
| |
| if (ev->kq_errno) { |
| ev->error = 1; |
| } |
| |
| if (!u->cacheable && u->peer.connection) { |
| ngx_log_error(NGX_LOG_INFO, ev->log, ev->kq_errno, |
| "kevent() reported that client closed prematurely " |
| "connection, so upstream connection is closed too"); |
| ngx_http_upstream_finalize_request(r, u, |
| NGX_HTTP_CLIENT_CLOSED_REQUEST); |
| return; |
| } |
| |
| ngx_log_error(NGX_LOG_INFO, ev->log, ev->kq_errno, |
| "kevent() reported that client closed " |
| "prematurely connection"); |
| |
| if (u->peer.connection == NULL) { |
| ngx_http_upstream_finalize_request(r, u, |
| NGX_HTTP_CLIENT_CLOSED_REQUEST); |
| return; |
| } |
| |
| return; |
| } |
| |
| #endif |
| |
| n = recv(c->fd, buf, 1, MSG_PEEK); |
| |
| err = ngx_socket_errno; |
| |
| ngx_log_debug1(NGX_LOG_DEBUG_HTTP, ev->log, err, |
| "http upstream recv(): %d", n); |
| |
| if (ev->write && (n >= 0 || err == NGX_EAGAIN)) { |
| return; |
| } |
| |
| if ((ngx_event_flags & NGX_USE_LEVEL_EVENT) && ev->active) { |
| |
| event = ev->write ? NGX_WRITE_EVENT : NGX_READ_EVENT; |
| |
| if (ngx_del_event(ev, event, 0) != NGX_OK) { |
| ngx_http_upstream_finalize_request(r, u, |
| NGX_HTTP_INTERNAL_SERVER_ERROR); |
| return; |
| } |
| } |
| |
| if (n > 0) { |
| return; |
| } |
| |
| if (n == -1) { |
| if (err == NGX_EAGAIN) { |
| return; |
| } |
| |
| ev->error = 1; |
| |
| } else { /* n == 0 */ |
| err = 0; |
| } |
| |
| ev->eof = 1; |
| c->error = 1; |
| |
| if (!u->cacheable && u->peer.connection) { |
| ngx_log_error(NGX_LOG_INFO, ev->log, err, |
| "client closed prematurely connection, " |
| "so upstream connection is closed too"); |
| ngx_http_upstream_finalize_request(r, u, |
| NGX_HTTP_CLIENT_CLOSED_REQUEST); |
| return; |
| } |
| |
| ngx_log_error(NGX_LOG_INFO, ev->log, err, |
| "client closed prematurely connection"); |
| |
| if (u->peer.connection == NULL) { |
| ngx_http_upstream_finalize_request(r, u, |
| NGX_HTTP_CLIENT_CLOSED_REQUEST); |
| return; |
| } |
| } |
| |
| |
| static void |
| ngx_http_upstream_connect(ngx_http_request_t *r, ngx_http_upstream_t *u) |
| { |
| ngx_int_t rc; |
| ngx_time_t *tp; |
| ngx_connection_t *c; |
| |
| r->connection->log->action = "connecting to upstream"; |
| |
| r->connection->single_connection = 0; |
| |
| if (u->state && u->state->response_sec) { |
| tp = ngx_timeofday(); |
| u->state->response_sec = tp->sec - u->state->response_sec; |
| u->state->response_msec = tp->msec - u->state->response_msec; |
| } |
| |
| u->state = ngx_array_push(r->upstream_states); |
| if (u->state == NULL) { |
| ngx_http_upstream_finalize_request(r, u, |
| NGX_HTTP_INTERNAL_SERVER_ERROR); |
| return; |
| } |
| |
| ngx_memzero(u->state, sizeof(ngx_http_upstream_state_t)); |
| |
| tp = ngx_timeofday(); |
| u->state->response_sec = tp->sec; |
| u->state->response_msec = tp->msec; |
| |
| rc = ngx_event_connect_peer(&u->peer); |
| |
| ngx_log_debug1(NGX_LOG_DEBUG_HTTP, r->connection->log, 0, |
| "http upstream connect: %i", rc); |
| |
| if (rc == NGX_ERROR) { |
| ngx_http_upstream_finalize_request(r, u, |
| NGX_HTTP_INTERNAL_SERVER_ERROR); |
| return; |
| } |
| |
| u->state->peer = u->peer.name; |
| |
| if (rc == NGX_BUSY) { |
| ngx_log_error(NGX_LOG_ERR, r->connection->log, 0, "no live upstreams"); |
| ngx_http_upstream_next(r, u, NGX_HTTP_UPSTREAM_FT_NOLIVE); |
| return; |
| } |
| |
| if (rc == NGX_DECLINED) { |
| ngx_http_upstream_next(r, u, NGX_HTTP_UPSTREAM_FT_ERROR); |
| return; |
| } |
| |
| /* rc == NGX_OK || rc == NGX_AGAIN */ |
| |
| c = u->peer.connection; |
| |
| c->data = r; |
| |
| c->write->handler = ngx_http_upstream_handler; |
| c->read->handler = ngx_http_upstream_handler; |
| |
| u->write_event_handler = ngx_http_upstream_send_request_handler; |
| u->read_event_handler = ngx_http_upstream_process_header; |
| |
| c->sendfile &= r->connection->sendfile; |
| u->output.sendfile = c->sendfile; |
| |
| c->pool = r->pool; |
| c->log = r->connection->log; |
| c->read->log = c->log; |
| c->write->log = c->log; |
| |
| /* init or reinit the ngx_output_chain() and ngx_chain_writer() contexts */ |
| |
| u->writer.out = NULL; |
| u->writer.last = &u->writer.out; |
| u->writer.connection = c; |
| u->writer.limit = 0; |
| |
| if (u->request_sent) { |
| if (ngx_http_upstream_reinit(r, u) != NGX_OK) { |
| ngx_http_upstream_finalize_request(r, u, |
| NGX_HTTP_INTERNAL_SERVER_ERROR); |
| return; |
| } |
| } |
| |
| if (r->request_body |
| && r->request_body->buf |
| && r->request_body->temp_file |
| && r == r->main) |
| { |
| /* |
| * the r->request_body->buf can be reused for one request only, |
| * the subrequests should allocate their own temporay bufs |
| */ |
| |
| u->output.free = ngx_alloc_chain_link(r->pool); |
| if (u->output.free == NULL) { |
| ngx_http_upstream_finalize_request(r, u, |
| NGX_HTTP_INTERNAL_SERVER_ERROR); |
| return; |
| } |
| |
| u->output.free->buf = r->request_body->buf; |
| u->output.free->next = NULL; |
| u->output.allocated = 1; |
| |
| r->request_body->buf->pos = r->request_body->buf->start; |
| r->request_body->buf->last = r->request_body->buf->start; |
| r->request_body->buf->tag = u->output.tag; |
| } |
| |
| u->request_sent = 0; |
| |
| if (rc == NGX_AGAIN) { |
| ngx_add_timer(c->write, u->conf->connect_timeout); |
| return; |
| } |
| |
| #if (NGX_HTTP_SSL) |
| |
| if (u->ssl && c->ssl == NULL) { |
| ngx_http_upstream_ssl_init_connection(r, u, c); |
| return; |
| } |
| |
| #endif |
| |
| ngx_http_upstream_send_request(r, u); |
| } |
| |
| |
| #if (NGX_HTTP_SSL) |
| |
| static void |
| ngx_http_upstream_ssl_init_connection(ngx_http_request_t *r, |
| ngx_http_upstream_t *u, ngx_connection_t *c) |
| { |
| ngx_int_t rc; |
| |
| if (ngx_ssl_create_connection(u->conf->ssl, c, |
| NGX_SSL_BUFFER|NGX_SSL_CLIENT) |
| != NGX_OK) |
| { |
| ngx_http_upstream_finalize_request(r, u, |
| NGX_HTTP_INTERNAL_SERVER_ERROR); |
| return; |
| } |
| |
| c->sendfile = 0; |
| u->output.sendfile = 0; |
| |
| if (u->conf->ssl_session_reuse) { |
| if (u->peer.set_session(&u->peer, u->peer.data) != NGX_OK) { |
| ngx_http_upstream_finalize_request(r, u, |
| NGX_HTTP_INTERNAL_SERVER_ERROR); |
| return; |
| } |
| } |
| |
| r->connection->log->action = "SSL handshaking to upstream"; |
| |
| rc = ngx_ssl_handshake(c); |
| |
| if (rc == NGX_AGAIN) { |
| c->ssl->handler = ngx_http_upstream_ssl_handshake; |
| return; |
| } |
| |
| ngx_http_upstream_ssl_handshake(c); |
| } |
| |
| |
| static void |
| ngx_http_upstream_ssl_handshake(ngx_connection_t *c) |
| { |
| ngx_http_request_t *r; |
| ngx_http_upstream_t *u; |
| |
| r = c->data; |
| u = r->upstream; |
| |
| if (c->ssl->handshaked) { |
| |
| if (u->conf->ssl_session_reuse) { |
| u->peer.save_session(&u->peer, u->peer.data); |
| } |
| |
| c->write->handler = ngx_http_upstream_handler; |
| c->read->handler = ngx_http_upstream_handler; |
| |
| ngx_http_upstream_send_request(r, u); |
| |
| return; |
| } |
| |
| ngx_http_upstream_next(r, u, NGX_HTTP_UPSTREAM_FT_ERROR); |
| |
| } |
| |
| #endif |
| |
| |
| static ngx_int_t |
| ngx_http_upstream_reinit(ngx_http_request_t *r, ngx_http_upstream_t *u) |
| { |
| ngx_chain_t *cl; |
| |
| if (u->reinit_request(r) != NGX_OK) { |
| return NGX_ERROR; |
| } |
| |
| ngx_memzero(&u->headers_in, sizeof(ngx_http_upstream_headers_in_t)); |
| |
| if (ngx_list_init(&u->headers_in.headers, r->pool, 8, |
| sizeof(ngx_table_elt_t)) |
| != NGX_OK) |
| { |
| return NGX_ERROR; |
| } |
| |
| /* reinit the request chain */ |
| |
| for (cl = u->request_bufs; cl; cl = cl->next) { |
| cl->buf->pos = cl->buf->start; |
| cl->buf->file_pos = 0; |
| } |
| |
| /* reinit the subrequest's ngx_output_chain() context */ |
| |
| if (r->request_body && r->request_body->temp_file |
| && r != r->main && u->output.buf) |
| { |
| u->output.free = ngx_alloc_chain_link(r->pool); |
| if (u->output.free == NULL) { |
| return NGX_ERROR; |
| } |
| |
| u->output.free->buf = u->output.buf; |
| u->output.free->next = NULL; |
| |
| u->output.buf->pos = u->output.buf->start; |
| u->output.buf->last = u->output.buf->start; |
| } |
| |
| u->output.buf = NULL; |
| u->output.in = NULL; |
| u->output.busy = NULL; |
| |
| /* reinit u->buffer */ |
| |
| u->buffer.pos = u->buffer.start; |
| |
| #if (NGX_HTTP_CACHE) |
| |
| if (r->cache) { |
| u->buffer.pos += r->cache->header_start; |
| } |
| |
| #endif |
| |
| u->buffer.last = u->buffer.pos; |
| |
| return NGX_OK; |
| } |
| |
| |
| static void |
| ngx_http_upstream_send_request(ngx_http_request_t *r, ngx_http_upstream_t *u) |
| { |
| ngx_int_t rc; |
| ngx_connection_t *c; |
| |
| c = u->peer.connection; |
| |
| ngx_log_debug0(NGX_LOG_DEBUG_HTTP, c->log, 0, |
| "http upstream send request"); |
| |
| if (!u->request_sent && ngx_http_upstream_test_connect(c) != NGX_OK) { |
| ngx_http_upstream_next(r, u, NGX_HTTP_UPSTREAM_FT_ERROR); |
| return; |
| } |
| |
| c->log->action = "sending request to upstream"; |
| |
| rc = ngx_output_chain(&u->output, u->request_sent ? NULL : u->request_bufs); |
| |
| u->request_sent = 1; |
| |
| if (rc == NGX_ERROR) { |
| ngx_http_upstream_next(r, u, NGX_HTTP_UPSTREAM_FT_ERROR); |
| return; |
| } |
| |
| if (c->write->timer_set) { |
| ngx_del_timer(c->write); |
| } |
| |
| if (rc == NGX_AGAIN) { |
| ngx_add_timer(c->write, u->conf->send_timeout); |
| |
| if (ngx_handle_write_event(c->write, u->conf->send_lowat) != NGX_OK) { |
| ngx_http_upstream_finalize_request(r, u, |
| NGX_HTTP_INTERNAL_SERVER_ERROR); |
| return; |
| } |
| |
| return; |
| } |
| |
| /* rc == NGX_OK */ |
| |
| if (c->tcp_nopush == NGX_TCP_NOPUSH_SET) { |
| 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_upstream_finalize_request(r, u, |
| NGX_HTTP_INTERNAL_SERVER_ERROR); |
| return; |
| } |
| |
| c->tcp_nopush = NGX_TCP_NOPUSH_UNSET; |
| } |
| |
| ngx_add_timer(c->read, u->conf->read_timeout); |
| |
| #if 1 |
| if (c->read->ready) { |
| |
| /* post aio operation */ |
| |
| /* |
| * TODO comment |
| * although we can post aio operation just in the end |
| * of ngx_http_upstream_connect() CHECK IT !!! |
| * it's better to do here because we postpone header buffer allocation |
| */ |
| |
| ngx_http_upstream_process_header(r, u); |
| return; |
| } |
| #endif |
| |
| u->write_event_handler = ngx_http_upstream_dummy_handler; |
| |
| if (ngx_handle_write_event(c->write, 0) != NGX_OK) { |
| ngx_http_upstream_finalize_request(r, u, |
| NGX_HTTP_INTERNAL_SERVER_ERROR); |
| return; |
| } |
| } |
| |
| |
| static void |
| ngx_http_upstream_send_request_handler(ngx_http_request_t *r, |
| ngx_http_upstream_t *u) |
| { |
| ngx_connection_t *c; |
| |
| c = u->peer.connection; |
| |
| ngx_log_debug0(NGX_LOG_DEBUG_HTTP, r->connection->log, 0, |
| "http upstream send request handler"); |
| |
| if (c->write->timedout) { |
| ngx_http_upstream_next(r, u, NGX_HTTP_UPSTREAM_FT_TIMEOUT); |
| return; |
| } |
| |
| #if (NGX_HTTP_SSL) |
| |
| if (u->ssl && c->ssl == NULL) { |
| ngx_http_upstream_ssl_init_connection(r, u, c); |
| return; |
| } |
| |
| #endif |
| |
| if (u->header_sent) { |
| u->write_event_handler = ngx_http_upstream_dummy_handler; |
| |
| (void) ngx_handle_write_event(c->write, 0); |
| |
| return; |
| } |
| |
| ngx_http_upstream_send_request(r, u); |
| } |
| |
| |
| static void |
| ngx_http_upstream_process_header(ngx_http_request_t *r, ngx_http_upstream_t *u) |
| { |
| ssize_t n; |
| ngx_int_t rc; |
| ngx_connection_t *c; |
| |
| c = u->peer.connection; |
| |
| ngx_log_debug0(NGX_LOG_DEBUG_HTTP, c->log, 0, |
| "http upstream process header"); |
| |
| c->log->action = "reading response header from upstream"; |
| |
| if (c->read->timedout) { |
| ngx_http_upstream_next(r, u, NGX_HTTP_UPSTREAM_FT_TIMEOUT); |
| return; |
| } |
| |
| if (!u->request_sent && ngx_http_upstream_test_connect(c) != NGX_OK) { |
| ngx_http_upstream_next(r, u, NGX_HTTP_UPSTREAM_FT_ERROR); |
| return; |
| } |
| |
| if (u->buffer.start == NULL) { |
| u->buffer.start = ngx_palloc(r->pool, u->conf->buffer_size); |
| if (u->buffer.start == NULL) { |
| ngx_http_upstream_finalize_request(r, u, |
| NGX_HTTP_INTERNAL_SERVER_ERROR); |
| return; |
| } |
| |
| u->buffer.pos = u->buffer.start; |
| u->buffer.last = u->buffer.start; |
| u->buffer.end = u->buffer.start + u->conf->buffer_size; |
| u->buffer.temporary = 1; |
| |
| u->buffer.tag = u->output.tag; |
| |
| if (ngx_list_init(&u->headers_in.headers, r->pool, 8, |
| sizeof(ngx_table_elt_t)) |
| != NGX_OK) |
| { |
| ngx_http_upstream_finalize_request(r, u, |
| NGX_HTTP_INTERNAL_SERVER_ERROR); |
| return; |
| } |
| |
| #if (NGX_HTTP_CACHE) |
| |
| if (r->cache) { |
| u->buffer.pos += r->cache->header_start; |
| u->buffer.last = u->buffer.pos; |
| } |
| #endif |
| } |
| |
| for ( ;; ) { |
| |
| n = c->recv(c, u->buffer.last, u->buffer.end - u->buffer.last); |
| |
| if (n == NGX_AGAIN) { |
| #if 0 |
| ngx_add_timer(rev, u->read_timeout); |
| #endif |
| |
| if (ngx_handle_read_event(c->read, 0) != NGX_OK) { |
| ngx_http_upstream_finalize_request(r, u, |
| NGX_HTTP_INTERNAL_SERVER_ERROR); |
| return; |
| } |
| |
| return; |
| } |
| |
| if (n == 0) { |
| ngx_log_error(NGX_LOG_ERR, c->log, 0, |
| "upstream prematurely closed connection"); |
| } |
| |
| if (n == NGX_ERROR || n == 0) { |
| ngx_http_upstream_next(r, u, NGX_HTTP_UPSTREAM_FT_ERROR); |
| return; |
| } |
| |
| u->buffer.last += n; |
| |
| #if 0 |
| u->valid_header_in = 0; |
| |
| u->peer.cached = 0; |
| #endif |
| |
| rc = u->process_header(r); |
| |
| if (rc == NGX_AGAIN) { |
| |
| if (u->buffer.pos == u->buffer.end) { |
| ngx_log_error(NGX_LOG_ERR, c->log, 0, |
| "upstream sent too big header"); |
| |
| ngx_http_upstream_next(r, u, |
| NGX_HTTP_UPSTREAM_FT_INVALID_HEADER); |
| return; |
| } |
| |
| continue; |
| } |
| |
| break; |
| } |
| |
| if (rc == NGX_HTTP_UPSTREAM_INVALID_HEADER) { |
| ngx_http_upstream_next(r, u, NGX_HTTP_UPSTREAM_FT_INVALID_HEADER); |
| return; |
| } |
| |
| if (rc == NGX_ERROR) { |
| ngx_http_upstream_finalize_request(r, u, |
| NGX_HTTP_INTERNAL_SERVER_ERROR); |
| return; |
| } |
| |
| /* rc == NGX_OK */ |
| |
| if (u->headers_in.status_n >= NGX_HTTP_BAD_REQUEST) { |
| |
| if (r->subrequest_in_memory) { |
| u->buffer.last = u->buffer.pos; |
| } |
| |
| if (ngx_http_upstream_test_next(r, u) == NGX_OK) { |
| return; |
| } |
| |
| if (ngx_http_upstream_intercept_errors(r, u) == NGX_OK) { |
| return; |
| } |
| } |
| |
| if (ngx_http_upstream_process_headers(r, u) != NGX_OK) { |
| return; |
| } |
| |
| if (!r->subrequest_in_memory) { |
| ngx_http_upstream_send_response(r, u); |
| return; |
| } |
| |
| /* subrequest content in memory */ |
| |
| if (u->input_filter == NULL) { |
| u->input_filter_init = ngx_http_upstream_non_buffered_filter_init; |
| u->input_filter = ngx_http_upstream_non_buffered_filter; |
| u->input_filter_ctx = r; |
| } |
| |
| if (u->input_filter_init(u->input_filter_ctx) == NGX_ERROR) { |
| ngx_http_upstream_finalize_request(r, u, |
| NGX_HTTP_INTERNAL_SERVER_ERROR); |
| return; |
| } |
| |
| n = u->buffer.last - u->buffer.pos; |
| |
| if (n) { |
| u->buffer.last -= n; |
| |
| u->state->response_length += n; |
| |
| if (u->input_filter(u->input_filter_ctx, n) == NGX_ERROR) { |
| ngx_http_upstream_finalize_request(r, u, NGX_ERROR); |
| return; |
| } |
| |
| if (u->length == 0) { |
| ngx_http_upstream_finalize_request(r, u, 0); |
| return; |
| } |
| } |
| |
| u->read_event_handler = ngx_http_upstream_process_body_in_memory; |
| |
| ngx_http_upstream_process_body_in_memory(r, u); |
| } |
| |
| |
| static ngx_int_t |
| ngx_http_upstream_test_next(ngx_http_request_t *r, ngx_http_upstream_t *u) |
| { |
| ngx_uint_t status; |
| ngx_http_upstream_next_t *un; |
| |
| status = u->headers_in.status_n; |
| |
| for (un = ngx_http_upstream_next_errors; un->status; un++) { |
| |
| if (status != un->status) { |
| continue; |
| } |
| |
| if (u->peer.tries > 1 && (u->conf->next_upstream & un->mask)) { |
| ngx_http_upstream_next(r, u, un->mask); |
| return NGX_OK; |
| } |
| |
| #if (NGX_HTTP_CACHE) |
| |
| if (u->cache_status == NGX_HTTP_CACHE_EXPIRED |
| && (u->conf->cache_use_stale & un->mask)) |
| { |
| ngx_int_t rc; |
| |
| rc = u->reinit_request(r); |
| |
| if (rc == NGX_OK) { |
| u->cache_status = NGX_HTTP_CACHE_STALE; |
| rc = ngx_http_upstream_cache_send(r, u); |
| } |
| |
| ngx_http_upstream_finalize_request(r, u, rc); |
| return NGX_OK; |
| } |
| |
| #endif |
| } |
| |
| return NGX_DECLINED; |
| } |
| |
| |
| static ngx_int_t |
| ngx_http_upstream_intercept_errors(ngx_http_request_t *r, |
| ngx_http_upstream_t *u) |
| { |
| ngx_int_t status; |
| ngx_uint_t i; |
| ngx_table_elt_t *h; |
| ngx_http_err_page_t *err_page; |
| ngx_http_core_loc_conf_t *clcf; |
| |
| status = u->headers_in.status_n; |
| |
| if (status == NGX_HTTP_NOT_FOUND && u->conf->intercept_404) { |
| ngx_http_upstream_finalize_request(r, u, NGX_HTTP_NOT_FOUND); |
| return NGX_OK; |
| } |
| |
| if (!u->conf->intercept_errors) { |
| return NGX_DECLINED; |
| } |
| |
| clcf = ngx_http_get_module_loc_conf(r, ngx_http_core_module); |
| |
| if (clcf->error_pages == NULL) { |
| return NGX_DECLINED; |
| } |
| |
| err_page = clcf->error_pages->elts; |
| for (i = 0; i < clcf->error_pages->nelts; i++) { |
| |
| if (err_page[i].status == status) { |
| |
| if (status == NGX_HTTP_UNAUTHORIZED |
| && u->headers_in.www_authenticate) |
| { |
| h = ngx_list_push(&r->headers_out.headers); |
| |
| if (h == NULL) { |
| ngx_http_upstream_finalize_request(r, u, |
| NGX_HTTP_INTERNAL_SERVER_ERROR); |
| return NGX_OK; |
| } |
| |
| *h = *u->headers_in.www_authenticate; |
| |
| r->headers_out.www_authenticate = h; |
| } |
| |
| ngx_http_upstream_finalize_request(r, u, status); |
| |
| return NGX_OK; |
| } |
| } |
| |
| return NGX_DECLINED; |
| } |
| |
| |
| static ngx_int_t |
| ngx_http_upstream_test_connect(ngx_connection_t *c) |
| { |
| int err; |
| socklen_t len; |
| |
| #if (NGX_HAVE_KQUEUE) |
| |
| if (ngx_event_flags & NGX_USE_KQUEUE_EVENT) { |
| if (c->write->pending_eof) { |
| c->log->action = "connecting to upstream"; |
| (void) ngx_connection_error(c, c->write->kq_errno, |
| "kevent() reported that connect() failed"); |
| return NGX_ERROR; |
| } |
| |
| } else |
| #endif |
| { |
| err = 0; |
| len = sizeof(int); |
| |
| /* |
| * BSDs and Linux return 0 and set a pending error in err |
| * Solaris returns -1 and sets errno |
| */ |
| |
| if (getsockopt(c->fd, SOL_SOCKET, SO_ERROR, (void *) &err, &len) |
| == -1) |
| { |
| err = ngx_errno; |
| } |
| |
| if (err) { |
| c->log->action = "connecting to upstream"; |
| (void) ngx_connection_error(c, err, "connect() failed"); |
| return NGX_ERROR; |
| } |
| } |
| |
| return NGX_OK; |
| } |
| |
| |
| static ngx_int_t |
| ngx_http_upstream_process_headers(ngx_http_request_t *r, ngx_http_upstream_t *u) |
| { |
| ngx_str_t *uri, args; |
| ngx_uint_t i, flags; |
| ngx_list_part_t *part; |
| ngx_table_elt_t *h; |
| ngx_http_upstream_header_t *hh; |
| ngx_http_upstream_main_conf_t *umcf; |
| |
| umcf = ngx_http_get_module_main_conf(r, ngx_http_upstream_module); |
| |
| if (u->headers_in.x_accel_redirect |
| && !(u->conf->ignore_headers & NGX_HTTP_UPSTREAM_IGN_XA_REDIRECT)) |
| { |
| ngx_http_upstream_finalize_request(r, u, NGX_DECLINED); |
| |
| part = &u->headers_in.headers.part; |
| h = part->elts; |
| |
| for (i = 0; /* void */; i++) { |
| |
| if (i >= part->nelts) { |
| if (part->next == NULL) { |
| break; |
| } |
| |
| part = part->next; |
| h = part->elts; |
| i = 0; |
| } |
| |
| hh = ngx_hash_find(&umcf->headers_in_hash, h[i].hash, |
| h[i].lowcase_key, h[i].key.len); |
| |
| if (hh && hh->redirect) { |
| if (hh->copy_handler(r, &h[i], hh->conf) != NGX_OK) { |
| ngx_http_finalize_request(r, |
| NGX_HTTP_INTERNAL_SERVER_ERROR); |
| return NGX_DONE; |
| } |
| } |
| } |
| |
| uri = &u->headers_in.x_accel_redirect->value; |
| args.len = 0; |
| args.data = NULL; |
| flags = 0; |
| |
| if (ngx_http_parse_unsafe_uri(r, uri, &args, &flags) != NGX_OK) { |
| ngx_http_finalize_request(r, NGX_HTTP_NOT_FOUND); |
| return NGX_DONE; |
| } |
| |
| if (flags & NGX_HTTP_ZERO_IN_URI) { |
| r->zero_in_uri = 1; |
| } |
| |
| if (r->method != NGX_HTTP_HEAD) { |
| r->method = NGX_HTTP_GET; |
| } |
| |
| r->valid_unparsed_uri = 0; |
| |
| ngx_http_internal_redirect(r, uri, &args); |
| return NGX_DONE; |
| } |
| |
| part = &u->headers_in.headers.part; |
| h = part->elts; |
| |
| for (i = 0; /* void */; i++) { |
| |
| if (i >= part->nelts) { |
| if (part->next == NULL) { |
| break; |
| } |
| |
| part = part->next; |
| h = part->elts; |
| i = 0; |
| } |
| |
| if (ngx_hash_find(&u->conf->hide_headers_hash, h[i].hash, |
| h[i].lowcase_key, h[i].key.len)) |
| { |
| continue; |
| } |
| |
| hh = ngx_hash_find(&umcf->headers_in_hash, h[i].hash, |
| h[i].lowcase_key, h[i].key.len); |
| |
| if (hh) { |
| if (hh->copy_handler(r, &h[i], hh->conf) != NGX_OK) { |
| ngx_http_upstream_finalize_request(r, u, |
| NGX_HTTP_INTERNAL_SERVER_ERROR); |
| return NGX_DONE; |
| } |
| |
| continue; |
| } |
| |
| if (ngx_http_upstream_copy_header_line(r, &h[i], 0) != NGX_OK) { |
| ngx_http_upstream_finalize_request(r, u, |
| NGX_HTTP_INTERNAL_SERVER_ERROR); |
| return NGX_DONE; |
| } |
| } |
| |
| if (r->headers_out.server && r->headers_out.server->value.data == NULL) { |
| r->headers_out.server->hash = 0; |
| } |
| |
| if (r->headers_out.date && r->headers_out.date->value.data == NULL) { |
| r->headers_out.date->hash = 0; |
| } |
| |
| r->headers_out.status = u->headers_in.status_n; |
| r->headers_out.status_line = u->headers_in.status_line; |
| |
| u->headers_in.content_length_n = r->headers_out.content_length_n; |
| |
| if (r->headers_out.content_length_n != -1) { |
| u->length = (size_t) r->headers_out.content_length_n; |
| |
| } else { |
| u->length = NGX_MAX_SIZE_T_VALUE; |
| } |
| |
| return NGX_OK; |
| } |
| |
| |
| static void |
| ngx_http_upstream_process_body_in_memory(ngx_http_request_t *r, |
| ngx_http_upstream_t *u) |
| { |
| size_t size; |
| ssize_t n; |
| ngx_buf_t *b; |
| ngx_event_t *rev; |
| ngx_connection_t *c; |
| |
| c = u->peer.connection; |
| rev = c->read; |
| |
| ngx_log_debug0(NGX_LOG_DEBUG_HTTP, c->log, 0, |
| "http upstream process body on memory"); |
| |
| if (rev->timedout) { |
| ngx_connection_error(c, NGX_ETIMEDOUT, "upstream timed out"); |
| ngx_http_upstream_finalize_request(r, u, NGX_ETIMEDOUT); |
| return; |
| } |
| |
| b = &u->buffer; |
| |
| for ( ;; ) { |
| |
| size = b->end - b->last; |
| |
| if (size == 0) { |
| ngx_log_error(NGX_LOG_ALERT, c->log, 0, |
| "upstream buffer is too small to read repsonse"); |
| ngx_http_upstream_finalize_request(r, u, NGX_ERROR); |
| return; |
| } |
| |
| n = c->recv(c, b->last, size); |
| |
| if (n == NGX_AGAIN) { |
| break; |
| } |
| |
| if (n == 0 || n == NGX_ERROR) { |
| ngx_http_upstream_finalize_request(r, u, n); |
| return; |
| } |
| |
| u->state->response_length += n; |
| |
| if (u->input_filter(u->input_filter_ctx, n) == NGX_ERROR) { |
| ngx_http_upstream_finalize_request(r, u, NGX_ERROR); |
| return; |
| } |
| |
| if (!rev->ready) { |
| break; |
| } |
| } |
| |
| if (ngx_handle_read_event(rev, 0) != NGX_OK) { |
| ngx_http_upstream_finalize_request(r, u, NGX_ERROR); |
| return; |
| } |
| |
| if (rev->active) { |
| ngx_add_timer(rev, u->conf->read_timeout); |
| |
| } else if (rev->timer_set) { |
| ngx_del_timer(rev); |
| } |
| } |
| |
| |
| static void |
| ngx_http_upstream_send_response(ngx_http_request_t *r, ngx_http_upstream_t *u) |
| { |
| int tcp_nodelay; |
| ssize_t n; |
| ngx_int_t rc; |
| ngx_event_pipe_t *p; |
| ngx_connection_t *c; |
| ngx_http_core_loc_conf_t *clcf; |
| |
| rc = ngx_http_send_header(r); |
| |
| if (rc == NGX_ERROR || rc > NGX_OK || r->post_action) { |
| ngx_http_upstream_finalize_request(r, u, rc); |
| return; |
| } |
| |
| c = r->connection; |
| |
| if (r->header_only) { |
| |
| if (u->cacheable || u->store) { |
| |
| if (ngx_shutdown_socket(c->fd, NGX_WRITE_SHUTDOWN) == -1) { |
| ngx_connection_error(c, ngx_socket_errno, |
| ngx_shutdown_socket_n " failed"); |
| } |
| |
| r->read_event_handler = ngx_http_request_empty_handler; |
| r->write_event_handler = ngx_http_request_empty_handler; |
| c->error = 1; |
| |
| } else { |
| ngx_http_upstream_finalize_request(r, u, rc); |
| return; |
| } |
| } |
| |
| u->header_sent = 1; |
| |
| if (r->request_body && r->request_body->temp_file) { |
| ngx_pool_run_cleanup_file(r->pool, r->request_body->temp_file->file.fd); |
| r->request_body->temp_file->file.fd = NGX_INVALID_FILE; |
| } |
| |
| clcf = ngx_http_get_module_loc_conf(r, ngx_http_core_module); |
| |
| if (!u->buffering) { |
| |
| if (u->input_filter == NULL) { |
| u->input_filter_init = ngx_http_upstream_non_buffered_filter_init; |
| u->input_filter = ngx_http_upstream_non_buffered_filter; |
| u->input_filter_ctx = r; |
| } |
| |
| u->read_event_handler = ngx_http_upstream_process_non_buffered_upstream; |
| r->write_event_handler = |
| ngx_http_upstream_process_non_buffered_downstream; |
| |
| r->limit_rate = 0; |
| |
| if (u->input_filter_init(u->input_filter_ctx) == NGX_ERROR) { |
| ngx_http_upstream_finalize_request(r, u, 0); |
| return; |
| } |
| |
| if (clcf->tcp_nodelay && c->tcp_nodelay == NGX_TCP_NODELAY_UNSET) { |
| ngx_log_debug0(NGX_LOG_DEBUG_HTTP, c->log, 0, "tcp_nodelay"); |
| |
| tcp_nodelay = 1; |
| |
| if (setsockopt(c->fd, IPPROTO_TCP, TCP_NODELAY, |
| (const void *) &tcp_nodelay, sizeof(int)) == -1) |
| { |
| ngx_connection_error(c, ngx_socket_errno, |
| "setsockopt(TCP_NODELAY) failed"); |
| ngx_http_upstream_finalize_request(r, u, 0); |
| return; |
| } |
| |
| c->tcp_nodelay = NGX_TCP_NODELAY_SET; |
| } |
| |
| n = u->buffer.last - u->buffer.pos; |
| |
| if (n) { |
| u->buffer.last = u->buffer.pos; |
| |
| u->state->response_length += n; |
| |
| if (u->input_filter(u->input_filter_ctx, n) == NGX_ERROR) { |
| ngx_http_upstream_finalize_request(r, u, 0); |
| return; |
| } |
| |
| ngx_http_upstream_process_non_buffered_downstream(r); |
| |
| } else { |
| u->buffer.pos = u->buffer.start; |
| u->buffer.last = u->buffer.start; |
| |
| if (ngx_http_send_special(r, NGX_HTTP_FLUSH) == NGX_ERROR) { |
| ngx_http_upstream_finalize_request(r, u, 0); |
| return; |
| } |
| |
| if (u->peer.connection->read->ready) { |
| ngx_http_upstream_process_non_buffered_upstream(r, u); |
| } |
| } |
| |
| return; |
| } |
| |
| /* TODO: preallocate event_pipe bufs, look "Content-Length" */ |
| |
| #if (NGX_HTTP_CACHE) |
| |
| if (r->cache && r->cache->file.fd != NGX_INVALID_FILE) { |
| ngx_pool_run_cleanup_file(r->pool, r->cache->file.fd); |
| r->cache->file.fd = NGX_INVALID_FILE; |
| } |
| |
| if (u->cacheable) { |
| time_t now, valid; |
| |
| now = ngx_time(); |
| |
| valid = r->cache->valid_sec; |
| |
| if (valid == 0) { |
| valid = ngx_http_file_cache_valid(u->conf->cache_valid, |
| u->headers_in.status_n); |
| if (valid) { |
| r->cache->valid_sec = now + valid; |
| } |
| } |
| |
| if (valid) { |
| r->cache->last_modified = r->headers_out.last_modified_time; |
| r->cache->date = now; |
| r->cache->body_start = (u_short) (u->buffer.pos - u->buffer.start); |
| |
| if (r->headers_out.content_length_n != -1) { |
| r->cache->length = r->cache->body_start |
| + r->headers_out.content_length_n; |
| } |
| |
| ngx_http_file_cache_set_header(r, u->buffer.start); |
| |
| } else { |
| u->cacheable = 0; |
| r->headers_out.last_modified_time = -1; |
| } |
| } |
| |
| ngx_log_debug1(NGX_LOG_DEBUG_HTTP, c->log, 0, |
| "http cacheable: %d", u->cacheable); |
| |
| if (u->cacheable == 0 && r->cache) { |
| ngx_http_file_cache_free(r, u->pipe->temp_file); |
| } |
| |
| #endif |
| |
| p = u->pipe; |
| |
| p->output_filter = (ngx_event_pipe_output_filter_pt) ngx_http_output_filter; |
| p->output_ctx = r; |
| p->tag = u->output.tag; |
| p->bufs = u->conf->bufs; |
| p->busy_size = u->conf->busy_buffers_size; |
| p->upstream = u->peer.connection; |
| p->downstream = c; |
| p->pool = r->pool; |
| p->log = c->log; |
| |
| p->cacheable = u->cacheable || u->store; |
| |
| p->temp_file = ngx_pcalloc(r->pool, sizeof(ngx_temp_file_t)); |
| if (p->temp_file == NULL) { |
| ngx_http_upstream_finalize_request(r, u, 0); |
| return; |
| } |
| |
| p->temp_file->file.fd = NGX_INVALID_FILE; |
| p->temp_file->file.log = c->log; |
| p->temp_file->path = u->conf->temp_path; |
| p->temp_file->pool = r->pool; |
| |
| if (p->cacheable) { |
| p->temp_file->persistent = 1; |
| |
| } else { |
| p->temp_file->log_level = NGX_LOG_WARN; |
| p->temp_file->warn = "an upstream response is buffered " |
| "to a temporary file"; |
| } |
| |
| p->max_temp_file_size = u->conf->max_temp_file_size; |
| p->temp_file_write_size = u->conf->temp_file_write_size; |
| |
| p->preread_bufs = ngx_alloc_chain_link(r->pool); |
| if (p->preread_bufs == NULL) { |
| ngx_http_upstream_finalize_request(r, u, 0); |
| return; |
| } |
| |
| p->preread_bufs->buf = &u->buffer; |
| p->preread_bufs->next = NULL; |
| u->buffer.recycled = 1; |
| |
| p->preread_size = u->buffer.last - u->buffer.pos; |
| |
| if (u->cacheable) { |
| |
| p->buf_to_file = ngx_calloc_buf(r->pool); |
| if (p->buf_to_file == NULL) { |
| ngx_http_upstream_finalize_request(r, u, 0); |
| return; |
| } |
| |
| p->buf_to_file->pos = u->buffer.start; |
| p->buf_to_file->last = u->buffer.pos; |
| p->buf_to_file->temporary = 1; |
| } |
| |
| if (ngx_event_flags & NGX_USE_AIO_EVENT) { |
| /* the posted aio operation may currupt a shadow buffer */ |
| p->single_buf = 1; |
| } |
| |
| /* TODO: p->free_bufs = 0 if use ngx_create_chain_of_bufs() */ |
| p->free_bufs = 1; |
| |
| /* |
| * event_pipe would do u->buffer.last += p->preread_size |
| * as though these bytes were read |
| */ |
| u->buffer.last = u->buffer.pos; |
| |
| if (u->conf->cyclic_temp_file) { |
| |
| /* |
| * we need to disable the use of sendfile() if we use cyclic temp file |
| * because the writing a new data may interfere with sendfile() |
| * that uses the same kernel file pages (at least on FreeBSD) |
| */ |
| |
| p->cyclic_temp_file = 1; |
| c->sendfile = 0; |
| |
| } else { |
| p->cyclic_temp_file = 0; |
| } |
| |
| p->read_timeout = u->conf->read_timeout; |
| p->send_timeout = clcf->send_timeout; |
| p->send_lowat = clcf->send_lowat; |
| |
| u->read_event_handler = ngx_http_upstream_process_upstream; |
| r->write_event_handler = ngx_http_upstream_process_downstream; |
| |
| ngx_http_upstream_process_upstream(r, u); |
| } |
| |
| |
| static void |
| ngx_http_upstream_process_non_buffered_downstream(ngx_http_request_t *r) |
| { |
| ngx_event_t *wev; |
| ngx_connection_t *c; |
| ngx_http_upstream_t *u; |
| |
| c = r->connection; |
| u = r->upstream; |
| wev = c->write; |
| |
| ngx_log_debug0(NGX_LOG_DEBUG_HTTP, c->log, 0, |
| "http upstream process non buffered downstream"); |
| |
| c->log->action = "sending to client"; |
| |
| if (wev->timedout) { |
| c->timedout = 1; |
| ngx_connection_error(c, NGX_ETIMEDOUT, "client timed out"); |
| ngx_http_upstream_finalize_request(r, u, 0); |
| return; |
| } |
| |
| ngx_http_upstream_process_non_buffered_request(r, 1); |
| } |
| |
| |
| static void |
| ngx_http_upstream_process_non_buffered_upstream(ngx_http_request_t *r, |
| ngx_http_upstream_t *u) |
| { |
| ngx_connection_t *c; |
| |
| c = u->peer.connection; |
| |
| ngx_log_debug0(NGX_LOG_DEBUG_HTTP, c->log, 0, |
| "http upstream process non buffered upstream"); |
| |
| c->log->action = "reading upstream"; |
| |
| if (c->read->timedout) { |
| ngx_connection_error(c, NGX_ETIMEDOUT, "upstream timed out"); |
| ngx_http_upstream_finalize_request(r, u, 0); |
| return; |
| } |
| |
| ngx_http_upstream_process_non_buffered_request(r, 0); |
| } |
| |
| |
| static void |
| ngx_http_upstream_process_non_buffered_request(ngx_http_request_t *r, |
| ngx_uint_t do_write) |
| { |
| size_t size; |
| ssize_t n; |
| ngx_buf_t *b; |
| ngx_int_t rc; |
| ngx_connection_t *downstream, *upstream; |
| ngx_http_upstream_t *u; |
| ngx_http_core_loc_conf_t *clcf; |
| |
| u = r->upstream; |
| downstream = r->connection; |
| upstream = u->peer.connection; |
| |
| b = &u->buffer; |
| |
| do_write = do_write || u->length == 0; |
| |
| for ( ;; ) { |
| |
| if (do_write) { |
| |
| if (u->out_bufs || u->busy_bufs) { |
| rc = ngx_http_output_filter(r, u->out_bufs); |
| |
| if (rc == NGX_ERROR) { |
| ngx_http_upstream_finalize_request(r, u, 0); |
| return; |
| } |
| |
| ngx_chain_update_chains(&u->free_bufs, &u->busy_bufs, |
| &u->out_bufs, u->output.tag); |
| } |
| |
| if (u->busy_bufs == NULL) { |
| |
| if (u->length == 0 |
| || upstream->read->eof |
| || upstream->read->error) |
| { |
| ngx_http_upstream_finalize_request(r, u, 0); |
| return; |
| } |
| |
| b->pos = b->start; |
| b->last = b->start; |
| } |
| } |
| |
| size = b->end - b->last; |
| |
| if (size > u->length) { |
| size = u->length; |
| } |
| |
| if (size && upstream->read->ready) { |
| |
| n = upstream->recv(upstream, b->last, size); |
| |
| if (n == NGX_AGAIN) { |
| break; |
| } |
| |
| if (n > 0) { |
| u->state->response_length += n; |
| |
| if (u->input_filter(u->input_filter_ctx, n) == NGX_ERROR) { |
| ngx_http_upstream_finalize_request(r, u, 0); |
| return; |
| } |
| } |
| |
| do_write = 1; |
| |
| continue; |
| } |
| |
| break; |
| } |
| |
| clcf = ngx_http_get_module_loc_conf(r, ngx_http_core_module); |
| |
| if (downstream->data == r) { |
| if (ngx_handle_write_event(downstream->write, clcf->send_lowat) |
| != NGX_OK) |
| { |
| ngx_http_upstream_finalize_request(r, u, 0); |
| return; |
| } |
| } |
| |
| if (downstream->write->active && !downstream->write->ready) { |
| ngx_add_timer(downstream->write, clcf->send_timeout); |
| |
| } else if (downstream->write->timer_set) { |
| ngx_del_timer(downstream->write); |
| } |
| |
| if (ngx_handle_read_event(upstream->read, 0) != NGX_OK) { |
| ngx_http_upstream_finalize_request(r, u, 0); |
| return; |
| } |
| |
| if (upstream->read->active && !upstream->read->ready) { |
| ngx_add_timer(upstream->read, u->conf->read_timeout); |
| |
| } else if (upstream->read->timer_set) { |
| ngx_del_timer(upstream->read); |
| } |
| } |
| |
| |
| static ngx_int_t |
| ngx_http_upstream_non_buffered_filter_init(void *data) |
| { |
| return NGX_OK; |
| } |
| |
| |
| static ngx_int_t |
| ngx_http_upstream_non_buffered_filter(void *data, ssize_t bytes) |
| { |
| ngx_http_request_t *r = data; |
| |
| ngx_buf_t *b; |
| ngx_chain_t *cl, **ll; |
| ngx_http_upstream_t *u; |
| |
| u = r->upstream; |
| |
| for (cl = u->out_bufs, ll = &u->out_bufs; cl; cl = cl->next) { |
| ll = &cl->next; |
| } |
| |
| cl = ngx_chain_get_free_buf(r->pool, &u->free_bufs); |
| if (cl == NULL) { |
| return NGX_ERROR; |
| } |
| |
| *ll = cl; |
| |
| cl->buf->flush = 1; |
| cl->buf->memory = 1; |
| |
| b = &u->buffer; |
| |
| cl->buf->pos = b->last; |
| b->last += bytes; |
| cl->buf->last = b->last; |
| cl->buf->tag = u->output.tag; |
| |
| if (u->length == NGX_MAX_SIZE_T_VALUE) { |
| return NGX_OK; |
| } |
| |
| u->length -= bytes; |
| |
| return NGX_OK; |
| } |
| |
| |
| static void |
| ngx_http_upstream_process_downstream(ngx_http_request_t *r) |
| { |
| ngx_event_t *wev; |
| ngx_connection_t *c; |
| ngx_event_pipe_t *p; |
| ngx_http_upstream_t *u; |
| |
| c = r->connection; |
| u = r->upstream; |
| p = u->pipe; |
| wev = c->write; |
| |
| ngx_log_debug0(NGX_LOG_DEBUG_HTTP, c->log, 0, |
| "http upstream process downstream"); |
| |
| c->log->action = "sending to client"; |
| |
| if (wev->timedout) { |
| |
| if (wev->delayed) { |
| |
| wev->timedout = 0; |
| wev->delayed = 0; |
| |
| if (!wev->ready) { |
| ngx_add_timer(wev, p->send_timeout); |
| |
| if (ngx_handle_write_event(wev, p->send_lowat) != NGX_OK) { |
| ngx_http_upstream_finalize_request(r, u, 0); |
| } |
| |
| return; |
| } |
| |
| if (ngx_event_pipe(p, wev->write) == NGX_ABORT) { |
| ngx_http_upstream_finalize_request(r, u, 0); |
| return; |
| } |
| |
| } else { |
| p->downstream_error = 1; |
| c->timedout = 1; |
| ngx_connection_error(c, NGX_ETIMEDOUT, "client timed out"); |
| } |
| |
| } else { |
| |
| if (wev->delayed) { |
| |
| ngx_log_debug0(NGX_LOG_DEBUG_HTTP, c->log, 0, |
| "http downstream delayed"); |
| |
| if (ngx_handle_write_event(wev, p->send_lowat) != NGX_OK) { |
| ngx_http_upstream_finalize_request(r, u, 0); |
| } |
| |
| return; |
| } |
| |
| if (ngx_event_pipe(p, 1) == NGX_ABORT) { |
| ngx_http_upstream_finalize_request(r, u, 0); |
| return; |
| } |
| } |
| |
| ngx_http_upstream_process_request(r); |
| } |
| |
| |
| static void |
| ngx_http_upstream_process_upstream(ngx_http_request_t *r, |
| ngx_http_upstream_t *u) |
| { |
| ngx_connection_t *c; |
| |
| c = u->peer.connection; |
| |
| ngx_log_debug0(NGX_LOG_DEBUG_HTTP, c->log, 0, |
| "http upstream process upstream"); |
| |
| c->log->action = "reading upstream"; |
| |
| if (c->read->timedout) { |
| u->pipe->upstream_error = 1; |
| ngx_connection_error(c, NGX_ETIMEDOUT, "upstream timed out"); |
| |
| } else { |
| if (ngx_event_pipe(u->pipe, 0) == NGX_ABORT) { |
| ngx_http_upstream_finalize_request(r, u, 0); |
| return; |
| } |
| } |
| |
| ngx_http_upstream_process_request(r); |
| } |
| |
| |
| static void |
| ngx_http_upstream_process_request(ngx_http_request_t *r) |
| { |
| ngx_uint_t del; |
| ngx_temp_file_t *tf; |
| ngx_event_pipe_t *p; |
| ngx_http_upstream_t *u; |
| |
| u = r->upstream; |
| p = u->pipe; |
| |
| if (u->peer.connection) { |
| |
| if (u->store) { |
| |
| del = p->upstream_error; |
| |
| tf = u->pipe->temp_file; |
| |
| if (p->upstream_eof || p->upstream_done) { |
| |
| if (u->headers_in.status_n == NGX_HTTP_OK |
| && (u->headers_in.content_length_n == -1 |
| || (u->headers_in.content_length_n == tf->offset))) |
| { |
| ngx_http_upstream_store(r, u); |
| |
| } else { |
| del = 1; |
| } |
| } |
| |
| if (del && tf->file.fd != NGX_INVALID_FILE) { |
| |
| if (ngx_delete_file(tf->file.name.data) == NGX_FILE_ERROR) { |
| |
| ngx_log_error(NGX_LOG_CRIT, r->connection->log, ngx_errno, |
| ngx_delete_file_n " \"%s\" failed", |
| u->pipe->temp_file->file.name.data); |
| } |
| } |
| } |
| |
| #if (NGX_HTTP_CACHE) |
| |
| if (u->cacheable) { |
| |
| if (p->upstream_done) { |
| ngx_http_file_cache_update(r, u->pipe->temp_file); |
| |
| } else if (p->upstream_eof) { |
| |
| /* TODO: check length & update cache */ |
| |
| ngx_http_file_cache_update(r, u->pipe->temp_file); |
| |
| } else if (p->upstream_error) { |
| ngx_http_file_cache_free(r, u->pipe->temp_file); |
| } |
| } |
| |
| #endif |
| |
| if (p->upstream_done || p->upstream_eof || p->upstream_error) { |
| ngx_log_debug1(NGX_LOG_DEBUG_HTTP, r->connection->log, 0, |
| "http upstream exit: %p", p->out); |
| #if 0 |
| ngx_http_busy_unlock(u->conf->busy_lock, &u->busy_lock); |
| #endif |
| ngx_http_upstream_finalize_request(r, u, 0); |
| return; |
| } |
| } |
| |
| if (p->downstream_error) { |
| ngx_log_debug0(NGX_LOG_DEBUG_HTTP, r->connection->log, 0, |
| "http upstream downstream error"); |
| |
| if (!u->cacheable && !u->store && u->peer.connection) { |
| ngx_http_upstream_finalize_request(r, u, 0); |
| } |
| } |
| } |
| |
| |
| static void |
| ngx_http_upstream_store(ngx_http_request_t *r, ngx_http_upstream_t *u) |
| { |
| size_t root; |
| time_t lm; |
| ngx_str_t path; |
| ngx_temp_file_t *tf; |
| ngx_ext_rename_file_t ext; |
| |
| tf = u->pipe->temp_file; |
| |
| if (tf->file.fd == NGX_INVALID_FILE) { |
| |
| /* create file for empty 200 response */ |
| |
| tf = ngx_pcalloc(r->pool, sizeof(ngx_temp_file_t)); |
| if (tf == NULL) { |
| return; |
| } |
| |
| tf->file.fd = NGX_INVALID_FILE; |
| tf->file.log = r->connection->log; |
| tf->path = u->conf->temp_path; |
| tf->pool = r->pool; |
| tf->persistent = 1; |
| |
| if (ngx_create_temp_file(&tf->file, tf->path, tf->pool, |
| tf->persistent, tf->clean, tf->access) |
| != NGX_OK) |
| { |
| return; |
| } |
| |
| u->pipe->temp_file = tf; |
| } |
| |
| ext.access = u->conf->store_access; |
| ext.path_access = u->conf->store_access; |
| ext.time = -1; |
| ext.create_path = 1; |
| ext.delete_file = 1; |
| ext.log = r->connection->log; |
| |
| if (u->headers_in.last_modified) { |
| |
| lm = ngx_http_parse_time(u->headers_in.last_modified->value.data, |
| u->headers_in.last_modified->value.len); |
| |
| if (lm != NGX_ERROR) { |
| ext.time = lm; |
| ext.fd = tf->file.fd; |
| } |
| } |
| |
| if (u->conf->store_lengths == NULL) { |
| |
| ngx_http_map_uri_to_path(r, &path, &root, 0); |
| |
| } else { |
| if (ngx_http_script_run(r, &path, u->conf->store_lengths->elts, 0, |
| u->conf->store_values->elts) |
| == NULL) |
| { |
| return; |
| } |
| } |
| |
| path.len--; |
| |
| ngx_log_debug2(NGX_LOG_DEBUG_HTTP, r->connection->log, 0, |
| "upstream stores \"%s\" to \"%s\"", |
| tf->file.name.data, path.data); |
| |
| (void) ngx_ext_rename_file(&tf->file.name, &path, &ext); |
| } |
| |
| |
| static void |
| ngx_http_upstream_dummy_handler(ngx_http_request_t *r, ngx_http_upstream_t *u) |
| { |
| ngx_log_debug0(NGX_LOG_DEBUG_HTTP, r->connection->log, 0, |
| "http upstream dummy handler"); |
| } |
| |
| |
| static void |
| ngx_http_upstream_next(ngx_http_request_t *r, ngx_http_upstream_t *u, |
| ngx_uint_t ft_type) |
| { |
| ngx_uint_t status, state; |
| |
| ngx_log_debug1(NGX_LOG_DEBUG_HTTP, r->connection->log, 0, |
| "http next upstream, %xi", ft_type); |
| |
| #if 0 |
| ngx_http_busy_unlock(u->conf->busy_lock, &u->busy_lock); |
| #endif |
| |
| if (ft_type == NGX_HTTP_UPSTREAM_FT_HTTP_404) { |
| state = NGX_PEER_NEXT; |
| } else { |
| state = NGX_PEER_FAILED; |
| } |
| |
| if (ft_type != NGX_HTTP_UPSTREAM_FT_NOLIVE) { |
| u->peer.free(&u->peer, u->peer.data, state); |
| } |
| |
| if (ft_type == NGX_HTTP_UPSTREAM_FT_TIMEOUT) { |
| ngx_log_error(NGX_LOG_ERR, r->connection->log, NGX_ETIMEDOUT, |
| "upstream timed out"); |
| } |
| |
| if (u->peer.cached && ft_type == NGX_HTTP_UPSTREAM_FT_ERROR) { |
| status = 0; |
| |
| } else { |
| switch(ft_type) { |
| |
| case NGX_HTTP_UPSTREAM_FT_TIMEOUT: |
| status = NGX_HTTP_GATEWAY_TIME_OUT; |
| break; |
| |
| case NGX_HTTP_UPSTREAM_FT_HTTP_500: |
| status = NGX_HTTP_INTERNAL_SERVER_ERROR; |
| break; |
| |
| case NGX_HTTP_UPSTREAM_FT_HTTP_404: |
| status = NGX_HTTP_NOT_FOUND; |
| break; |
| |
| /* |
| * NGX_HTTP_UPSTREAM_FT_BUSY_LOCK and NGX_HTTP_UPSTREAM_FT_MAX_WAITING |
| * never reach here |
| */ |
| |
| default: |
| status = NGX_HTTP_BAD_GATEWAY; |
| } |
| } |
| |
| if (r->connection->error) { |
| ngx_http_upstream_finalize_request(r, u, |
| NGX_HTTP_CLIENT_CLOSED_REQUEST); |
| return; |
| } |
| |
| if (status) { |
| u->state->status = status; |
| |
| if (u->peer.tries == 0 || !(u->conf->next_upstream & ft_type)) { |
| |
| #if (NGX_HTTP_CACHE) |
| |
| if (u->cache_status == NGX_HTTP_CACHE_EXPIRED |
| && (u->conf->cache_use_stale & ft_type)) |
| { |
| ngx_int_t rc; |
| |
| rc = u->reinit_request(r); |
| |
| if (rc == NGX_OK) { |
| u->cache_status = NGX_HTTP_CACHE_STALE; |
| rc = ngx_http_upstream_cache_send(r, u); |
| } |
| |
| ngx_http_upstream_finalize_request(r, u, rc); |
| return; |
| } |
| #endif |
| |
| ngx_http_upstream_finalize_request(r, u, status); |
| return; |
| } |
| } |
| |
| if (u->peer.connection) { |
| ngx_log_debug1(NGX_LOG_DEBUG_HTTP, r->connection->log, 0, |
| "close http upstream connection: %d", |
| u->peer.connection->fd); |
| #if (NGX_HTTP_SSL) |
| |
| if (u->peer.connection->ssl) { |
| u->peer.connection->ssl->no_wait_shutdown = 1; |
| u->peer.connection->ssl->no_send_shutdown = 1; |
| |
| (void) ngx_ssl_shutdown(u->peer.connection); |
| } |
| #endif |
| |
| ngx_close_connection(u->peer.connection); |
| } |
| |
| #if 0 |
| if (u->conf->busy_lock && !u->busy_locked) { |
| ngx_http_upstream_busy_lock(p); |
| return; |
| } |
| #endif |
| |
| ngx_http_upstream_connect(r, u); |
| } |
| |
| |
| static void |
| ngx_http_upstream_cleanup(void *data) |
| { |
| ngx_http_request_t *r = data; |
| |
| ngx_http_upstream_t *u; |
| |
| ngx_log_debug1(NGX_LOG_DEBUG_HTTP, r->connection->log, 0, |
| "cleanup http upstream request: \"%V\"", &r->uri); |
| |
| u = r->upstream; |
| |
| if (u->resolved && u->resolved->ctx) { |
| ngx_resolve_name_done(u->resolved->ctx); |
| } |
| |
| ngx_http_upstream_finalize_request(r, u, NGX_DONE); |
| } |
| |
| |
| static void |
| ngx_http_upstream_finalize_request(ngx_http_request_t *r, |
| ngx_http_upstream_t *u, ngx_int_t rc) |
| { |
| ngx_time_t *tp; |
| |
| ngx_log_debug1(NGX_LOG_DEBUG_HTTP, r->connection->log, 0, |
| "finalize http upstream request: %i", rc); |
| |
| if (u->cleanup) { |
| *u->cleanup = NULL; |
| } |
| |
| if (u->state && u->state->response_sec) { |
| tp = ngx_timeofday(); |
| u->state->response_sec = tp->sec - u->state->response_sec; |
| u->state->response_msec = tp->msec - u->state->response_msec; |
| |
| if (u->pipe) { |
| u->state->response_length = u->pipe->read_length; |
| } |
| } |
| |
| u->finalize_request(r, rc); |
| |
| if (u->peer.free) { |
| u->peer.free(&u->peer, u->peer.data, 0); |
| } |
| |
| if (u->peer.connection) { |
| |
| #if (NGX_HTTP_SSL) |
| |
| /* TODO: do not shutdown persistent connection */ |
| |
| if (u->peer.connection->ssl) { |
| |
| /* |
| * We send the "close notify" shutdown alert to the upstream only |
| * and do not wait its "close notify" shutdown alert. |
| * It is acceptable according to the TLS standard. |
| */ |
| |
| u->peer.connection->ssl->no_wait_shutdown = 1; |
| |
| (void) ngx_ssl_shutdown(u->peer.connection); |
| } |
| #endif |
| |
| ngx_log_debug1(NGX_LOG_DEBUG_HTTP, r->connection->log, 0, |
| "close http upstream connection: %d", |
| u->peer.connection->fd); |
| |
| ngx_close_connection(u->peer.connection); |
| } |
| |
| u->peer.connection = NULL; |
| |
| if (u->pipe && u->pipe->temp_file) { |
| ngx_log_debug1(NGX_LOG_DEBUG_HTTP, r->connection->log, 0, |
| "http upstream temp fd: %d", |
| u->pipe->temp_file->file.fd); |
| } |
| |
| #if (NGX_HTTP_CACHE) |
| |
| if (u->cacheable && r->cache) { |
| time_t valid; |
| |
| ngx_log_debug1(NGX_LOG_DEBUG_HTTP, r->connection->log, 0, |
| "http upstream cache fd: %d", |
| r->cache->file.fd); |
| |
| if (rc == NGX_HTTP_BAD_GATEWAY || rc == NGX_HTTP_GATEWAY_TIME_OUT) { |
| |
| valid = ngx_http_file_cache_valid(u->conf->cache_valid, rc); |
| |
| if (valid) { |
| r->cache->valid_sec = ngx_time() + valid; |
| r->cache->error = rc; |
| } |
| } |
| |
| ngx_http_file_cache_free(r, u->pipe->temp_file); |
| } |
| |
| #endif |
| |
| if (u->header_sent |
| && (rc == NGX_ERROR || rc >= NGX_HTTP_SPECIAL_RESPONSE)) |
| { |
| rc = 0; |
| } |
| |
| if (rc == NGX_DECLINED) { |
| return; |
| } |
| |
| r->connection->log->action = "sending to client"; |
| |
| if (rc == 0) { |
| rc = ngx_http_send_special(r, NGX_HTTP_LAST); |
| } |
| |
| ngx_http_finalize_request(r, rc); |
| } |
| |
| |
| static ngx_int_t |
| ngx_http_upstream_process_header_line(ngx_http_request_t *r, ngx_table_elt_t *h, |
| ngx_uint_t offset) |
| { |
| ngx_table_elt_t **ph; |
| |
| ph = (ngx_table_elt_t **) ((char *) &r->upstream->headers_in + offset); |
| |
| if (*ph == NULL) { |
| *ph = h; |
| } |
| |
| return NGX_OK; |
| } |
| |
| |
| static ngx_int_t |
| ngx_http_upstream_ignore_header_line(ngx_http_request_t *r, ngx_table_elt_t *h, |
| ngx_uint_t offset) |
| { |
| return NGX_OK; |
| } |
| |
| |
| static ngx_int_t |
| ngx_http_upstream_process_cache_control(ngx_http_request_t *r, |
| ngx_table_elt_t *h, ngx_uint_t offset) |
| { |
| ngx_array_t *pa; |
| ngx_table_elt_t **ph; |
| ngx_http_upstream_t *u; |
| |
| u = r->upstream; |
| pa = &u->headers_in.cache_control; |
| |
| if (pa->elts == NULL) { |
| if (ngx_array_init(pa, r->pool, 2, sizeof(ngx_table_elt_t *)) != NGX_OK) |
| { |
| return NGX_ERROR; |
| } |
| } |
| |
| ph = ngx_array_push(pa); |
| if (ph == NULL) { |
| return NGX_ERROR; |
| } |
| |
| *ph = h; |
| |
| #if (NGX_HTTP_CACHE) |
| { |
| u_char *p, *last; |
| ngx_int_t n; |
| |
| if (u->conf->ignore_headers & NGX_HTTP_UPSTREAM_IGN_CACHE_CONTROL) { |
| return NGX_OK; |
| } |
| |
| if (r->cache == NULL) { |
| return NGX_OK; |
| } |
| |
| if (r->cache->valid_sec != 0) { |
| return NGX_OK; |
| } |
| |
| last = h->value.data + h->value.len; |
| |
| if (ngx_strlcasestrn(h->value.data, last, (u_char *) "no-cache", 8 - 1) |
| != NULL) |
| { |
| u->cacheable = 0; |
| return NGX_OK; |
| } |
| |
| p = ngx_strlcasestrn(h->value.data, last, (u_char *) "max-age=", 8 - 1); |
| |
| if (p == NULL) { |
| return NGX_OK; |
| } |
| |
| n = 0; |
| |
| for (p += 8; p < last; p++) { |
| if (*p == ';' || *p == ' ') { |
| break; |
| } |
| |
| if (*p >= '0' && *p <= '9') { |
| n = n * 10 + *p - '0'; |
| continue; |
| } |
| |
| u->cacheable = 0; |
| return NGX_OK; |
| } |
| |
| if (n == 0) { |
| u->cacheable = 0; |
| return NGX_OK; |
| } |
| |
| r->cache->valid_sec = ngx_time() + n; |
| } |
| #endif |
| |
| return NGX_OK; |
| } |
| |
| |
| static ngx_int_t |
| ngx_http_upstream_process_expires(ngx_http_request_t *r, ngx_table_elt_t *h, |
| ngx_uint_t offset) |
| { |
| ngx_http_upstream_t *u; |
| |
| u = r->upstream; |
| u->headers_in.expires = h; |
| |
| #if (NGX_HTTP_CACHE) |
| { |
| time_t expires; |
| |
| if (u->conf->ignore_headers & NGX_HTTP_UPSTREAM_IGN_EXPIRES) { |
| return NGX_OK; |
| } |
| |
| if (r->cache == NULL) { |
| return NGX_OK; |
| } |
| |
| if (r->cache->valid_sec != 0) { |
| return NGX_OK; |
| } |
| |
| expires = ngx_http_parse_time(h->value.data, h->value.len); |
| |
| if (expires == NGX_ERROR || expires < ngx_time()) { |
| u->cacheable = 0; |
| return NGX_OK; |
| } |
| |
| r->cache->valid_sec = expires; |
| } |
| #endif |
| |
| return NGX_OK; |
| } |
| |
| |
| static ngx_int_t |
| ngx_http_upstream_process_accel_expires(ngx_http_request_t *r, |
| ngx_table_elt_t *h, ngx_uint_t offset) |
| { |
| ngx_http_upstream_t *u; |
| |
| u = r->upstream; |
| u->headers_in.x_accel_expires = h; |
| |
| #if (NGX_HTTP_CACHE) |
| { |
| u_char *p; |
| size_t len; |
| ngx_int_t n; |
| |
| if (u->conf->ignore_headers & NGX_HTTP_UPSTREAM_IGN_XA_EXPIRES) { |
| return NGX_OK; |
| } |
| |
| if (r->cache == NULL) { |
| return NGX_OK; |
| } |
| |
| len = h->value.len; |
| p = h->value.data; |
| |
| if (p[0] != '@') { |
| n = ngx_atoi(p, len); |
| |
| switch (n) { |
| case 0: |
| u->cacheable = 0; |
| case NGX_ERROR: |
| return NGX_OK; |
| |
| default: |
| r->cache->valid_sec = ngx_time() + n; |
| return NGX_OK; |
| } |
| } |
| |
| p++; |
| len--; |
| |
| n = ngx_atoi(p, len); |
| |
| if (n != NGX_ERROR) { |
| r->cache->valid_sec = n; |
| } |
| } |
| #endif |
| |
| return NGX_OK; |
| } |
| |
| |
| static ngx_int_t |
| ngx_http_upstream_process_limit_rate(ngx_http_request_t *r, ngx_table_elt_t *h, |
| ngx_uint_t offset) |
| { |
| ngx_int_t n; |
| |
| r->upstream->headers_in.x_accel_limit_rate = h; |
| |
| n = ngx_atoi(h->value.data, h->value.len); |
| |
| if (n != NGX_ERROR) { |
| r->limit_rate = (size_t) n; |
| } |
| |
| return NGX_OK; |
| } |
| |
| |
| static ngx_int_t |
| ngx_http_upstream_process_buffering(ngx_http_request_t *r, ngx_table_elt_t *h, |
| ngx_uint_t offset) |
| { |
| u_char c0, c1, c2; |
| |
| if (r->upstream->conf->change_buffering) { |
| |
| if (h->value.len == 2) { |
| c0 = ngx_tolower(h->value.data[0]); |
| c1 = ngx_tolower(h->value.data[1]); |
| |
| if (c0 == 'n' && c1 == 'o') { |
| r->upstream->buffering = 0; |
| } |
| |
| } else if (h->value.len == 3) { |
| c0 = ngx_tolower(h->value.data[0]); |
| c1 = ngx_tolower(h->value.data[1]); |
| c2 = ngx_tolower(h->value.data[2]); |
| |
| if (c0 == 'y' && c1 == 'e' && c2 == 's') { |
| r->upstream->buffering = 1; |
| } |
| } |
| } |
| |
| return NGX_OK; |
| } |
| |
| |
| static ngx_int_t |
| ngx_http_upstream_process_charset(ngx_http_request_t *r, ngx_table_elt_t *h, |
| ngx_uint_t offset) |
| { |
| r->headers_out.override_charset = &h->value; |
| |
| return NGX_OK; |
| } |
| |
| |
| static ngx_int_t |
| ngx_http_upstream_copy_header_line(ngx_http_request_t *r, ngx_table_elt_t *h, |
| ngx_uint_t offset) |
| { |
| ngx_table_elt_t *ho, **ph; |
| |
| ho = ngx_list_push(&r->headers_out.headers); |
| if (ho == NULL) { |
| return NGX_ERROR; |
| } |
| |
| *ho = *h; |
| |
| if (offset) { |
| ph = (ngx_table_elt_t **) ((char *) &r->headers_out + offset); |
| *ph = ho; |
| } |
| |
| return NGX_OK; |
| } |
| |
| |
| static ngx_int_t |
| ngx_http_upstream_copy_multi_header_lines(ngx_http_request_t *r, |
| ngx_table_elt_t *h, ngx_uint_t offset) |
| { |
| ngx_array_t *pa; |
| ngx_table_elt_t *ho, **ph; |
| |
| pa = (ngx_array_t *) ((char *) &r->headers_out + offset); |
| |
| if (pa->elts == NULL) { |
| if (ngx_array_init(pa, r->pool, 2, sizeof(ngx_table_elt_t *)) != NGX_OK) |
| { |
| return NGX_ERROR; |
| } |
| } |
| |
| ph = ngx_array_push(pa); |
| if (ph == NULL) { |
| return NGX_ERROR; |
| } |
| |
| ho = ngx_list_push(&r->headers_out.headers); |
| if (ho == NULL) { |
| return NGX_ERROR; |
| } |
| |
| *ho = *h; |
| *ph = ho; |
| |
| return NGX_OK; |
| } |
| |
| |
| static ngx_int_t |
| ngx_http_upstream_copy_content_type(ngx_http_request_t *r, ngx_table_elt_t *h, |
| ngx_uint_t offset) |
| { |
| u_char *p, *last; |
| |
| r->headers_out.content_type_len = h->value.len; |
| r->headers_out.content_type = h->value; |
| r->headers_out.content_type_lowcase = NULL; |
| |
| for (p = h->value.data; *p; p++) { |
| |
| if (*p != ';') { |
| continue; |
| } |
| |
| last = p; |
| |
| while (*++p == ' ') { /* void */ } |
| |
| if (*p == '\0') { |
| return NGX_OK; |
| } |
| |
| if (ngx_strncasecmp(p, (u_char *) "charset=", 8) != 0) { |
| continue; |
| } |
| |
| p += 8; |
| |
| r->headers_out.content_type_len = last - h->value.data; |
| |
| if (*p == '"') { |
| p++; |
| } |
| |
| last = h->value.data + h->value.len; |
| |
| if (*(last - 1) == '"') { |
| last--; |
| } |
| |
| r->headers_out.charset.len = last - p; |
| r->headers_out.charset.data = p; |
| |
| return NGX_OK; |
| } |
| |
| return NGX_OK; |
| } |
| |
| |
| static ngx_int_t |
| ngx_http_upstream_copy_content_length(ngx_http_request_t *r, ngx_table_elt_t *h, |
| ngx_uint_t offset) |
| { |
| ngx_table_elt_t *ho; |
| |
| ho = ngx_list_push(&r->headers_out.headers); |
| if (ho == NULL) { |
| return NGX_ERROR; |
| } |
| |
| *ho = *h; |
| |
| r->headers_out.content_length = ho; |
| r->headers_out.content_length_n = ngx_atoof(h->value.data, h->value.len); |
| |
| return NGX_OK; |
| } |
| |
| |
| static ngx_int_t |
| ngx_http_upstream_copy_last_modified(ngx_http_request_t *r, ngx_table_elt_t *h, |
| ngx_uint_t offset) |
| { |
| ngx_table_elt_t *ho; |
| |
| ho = ngx_list_push(&r->headers_out.headers); |
| if (ho == NULL) { |
| return NGX_ERROR; |
| } |
| |
| *ho = *h; |
| |
| r->headers_out.last_modified = ho; |
| |
| #if (NGX_HTTP_CACHE) |
| |
| if (r->upstream->cacheable) { |
| r->headers_out.last_modified_time = ngx_http_parse_time(h->value.data, |
| h->value.len); |
| } |
| |
| #endif |
| |
| return NGX_OK; |
| } |
| |
| |
| static ngx_int_t |
| ngx_http_upstream_rewrite_location(ngx_http_request_t *r, ngx_table_elt_t *h, |
| ngx_uint_t offset) |
| { |
| ngx_int_t rc; |
| ngx_table_elt_t *ho; |
| |
| ho = ngx_list_push(&r->headers_out.headers); |
| if (ho == NULL) { |
| return NGX_ERROR; |
| } |
| |
| *ho = *h; |
| |
| if (r->upstream->rewrite_redirect) { |
| rc = r->upstream->rewrite_redirect(r, ho, 0); |
| |
| if (rc == NGX_DECLINED) { |
| return NGX_OK; |
| } |
| |
| if (rc == NGX_OK) { |
| r->headers_out.location = ho; |
| |
| ngx_log_debug1(NGX_LOG_DEBUG_HTTP, r->connection->log, 0, |
| "rewritten location: \"%V\"", &ho->value); |
| } |
| |
| return rc; |
| } |
| |
| if (ho->value.data[0] != '/') { |
| r->headers_out.location = ho; |
| } |
| |
| /* |
| * we do not set r->headers_out.location here to avoid the handling |
| * the local redirects without a host name by ngx_http_header_filter() |
| */ |
| |
| return NGX_OK; |
| } |
| |
| |
| static ngx_int_t |
| ngx_http_upstream_rewrite_refresh(ngx_http_request_t *r, ngx_table_elt_t *h, |
| ngx_uint_t offset) |
| { |
| u_char *p; |
| ngx_int_t rc; |
| ngx_table_elt_t *ho; |
| |
| ho = ngx_list_push(&r->headers_out.headers); |
| if (ho == NULL) { |
| return NGX_ERROR; |
| } |
| |
| *ho = *h; |
| |
| if (r->upstream->rewrite_redirect) { |
| |
| p = ngx_strcasestrn(ho->value.data, "url=", 4 - 1); |
| |
| if (p) { |
| rc = r->upstream->rewrite_redirect(r, ho, p + 4 - ho->value.data); |
| |
| } else { |
| return NGX_OK; |
| } |
| |
| if (rc == NGX_DECLINED) { |
| return NGX_OK; |
| } |
| |
| if (rc == NGX_OK) { |
| r->headers_out.refresh = ho; |
| |
| ngx_log_debug1(NGX_LOG_DEBUG_HTTP, r->connection->log, 0, |
| "rewritten refresh: \"%V\"", &ho->value); |
| } |
| |
| return rc; |
| } |
| |
| r->headers_out.refresh = ho; |
| |
| return NGX_OK; |
| } |
| |
| |
| static ngx_int_t |
| ngx_http_upstream_copy_allow_ranges(ngx_http_request_t *r, |
| ngx_table_elt_t *h, ngx_uint_t offset) |
| { |
| ngx_table_elt_t *ho; |
| |
| #if (NGX_HTTP_CACHE) |
| |
| if (r->cached) { |
| r->allow_ranges = 1; |
| return NGX_OK; |
| |
| } |
| |
| #endif |
| |
| ho = ngx_list_push(&r->headers_out.headers); |
| if (ho == NULL) { |
| return NGX_ERROR; |
| } |
| |
| *ho = *h; |
| |
| r->headers_out.accept_ranges = ho; |
| |
| return NGX_OK; |
| } |
| |
| |
| #if (NGX_HTTP_GZIP) |
| |
| static ngx_int_t |
| ngx_http_upstream_copy_content_encoding(ngx_http_request_t *r, |
| ngx_table_elt_t *h, ngx_uint_t offset) |
| { |
| ngx_table_elt_t *ho; |
| |
| ho = ngx_list_push(&r->headers_out.headers); |
| if (ho == NULL) { |
| return NGX_ERROR; |
| } |
| |
| *ho = *h; |
| |
| r->headers_out.content_encoding = ho; |
| |
| return NGX_OK; |
| } |
| |
| #endif |
| |
| |
| static ngx_int_t |
| ngx_http_upstream_add_variables(ngx_conf_t *cf) |
| { |
| ngx_http_variable_t *var, *v; |
| |
| for (v = ngx_http_upstream_vars; v->name.len; v++) { |
| var = ngx_http_add_variable(cf, &v->name, v->flags); |
| if (var == NULL) { |
| return NGX_ERROR; |
| } |
| |
| var->get_handler = v->get_handler; |
| var->data = v->data; |
| } |
| |
| return NGX_OK; |
| } |
| |
| |
| static ngx_int_t |
| ngx_http_upstream_addr_variable(ngx_http_request_t *r, |
| ngx_http_variable_value_t *v, uintptr_t data) |
| { |
| u_char *p; |
| size_t len; |
| ngx_uint_t i; |
| ngx_http_upstream_state_t *state; |
| |
| v->valid = 1; |
| v->no_cacheable = 0; |
| v->not_found = 0; |
| |
| if (r->upstream_states == NULL || r->upstream_states->nelts == 0) { |
| v->not_found = 1; |
| return NGX_OK; |
| } |
| |
| len = 0; |
| state = r->upstream_states->elts; |
| |
| for (i = 0; i < r->upstream_states->nelts; i++) { |
| if (state[i].peer) { |
| len += state[i].peer->len + 2; |
| |
| } else { |
| len += 3; |
| } |
| } |
| |
| p = ngx_pnalloc(r->pool, len); |
| if (p == NULL) { |
| return NGX_ERROR; |
| } |
| |
| v->data = p; |
| |
| i = 0; |
| |
| for ( ;; ) { |
| if (state[i].peer) { |
| p = ngx_cpymem(p, state[i].peer->data, state[i].peer->len); |
| } |
| |
| if (++i == r->upstream_states->nelts) { |
| break; |
| } |
| |
| if (state[i].peer) { |
| *p++ = ','; |
| *p++ = ' '; |
| |
| } else { |
| *p++ = ' '; |
| *p++ = ':'; |
| *p++ = ' '; |
| |
| if (++i == r->upstream_states->nelts) { |
| break; |
| } |
| |
| continue; |
| } |
| } |
| |
| v->len = p - v->data; |
| |
| return NGX_OK; |
| } |
| |
| |
| static ngx_int_t |
| ngx_http_upstream_status_variable(ngx_http_request_t *r, |
| ngx_http_variable_value_t *v, uintptr_t data) |
| { |
| u_char *p; |
| size_t len; |
| ngx_uint_t i; |
| ngx_http_upstream_state_t *state; |
| |
| v->valid = 1; |
| v->no_cacheable = 0; |
| v->not_found = 0; |
| |
| if (r->upstream_states == NULL || r->upstream_states->nelts == 0) { |
| v->not_found = 1; |
| return NGX_OK; |
| } |
| |
| len = r->upstream_states->nelts * (3 + 2); |
| |
| p = ngx_pnalloc(r->pool, len); |
| if (p == NULL) { |
| return NGX_ERROR; |
| } |
| |
| v->data = p; |
| |
| i = 0; |
| state = r->upstream_states->elts; |
| |
| for ( ;; ) { |
| if (state[i].status) { |
| p = ngx_sprintf(p, "%ui", state[i].status); |
| |
| } else { |
| *p++ = '-'; |
| } |
| |
| if (++i == r->upstream_states->nelts) { |
| break; |
| } |
| |
| if (state[i].peer) { |
| *p++ = ','; |
| *p++ = ' '; |
| |
| } else { |
| *p++ = ' '; |
| *p++ = ':'; |
| *p++ = ' '; |
| |
| if (++i == r->upstream_states->nelts) { |
| break; |
| } |
| |
| continue; |
| } |
| } |
| |
| v->len = p - v->data; |
| |
| return NGX_OK; |
| } |
| |
| |
| static ngx_int_t |
| ngx_http_upstream_response_time_variable(ngx_http_request_t *r, |
| ngx_http_variable_value_t *v, uintptr_t data) |
| { |
| u_char *p; |
| size_t len; |
| ngx_uint_t i; |
| ngx_msec_int_t ms; |
| ngx_http_upstream_state_t *state; |
| |
| v->valid = 1; |
| v->no_cacheable = 0; |
| v->not_found = 0; |
| |
| if (r->upstream_states == NULL || r->upstream_states->nelts == 0) { |
| v->not_found = 1; |
| return NGX_OK; |
| } |
| |
| len = r->upstream_states->nelts * (NGX_TIME_T_LEN + 4 + 2); |
| |
| p = ngx_pnalloc(r->pool, len); |
| if (p == NULL) { |
| return NGX_ERROR; |
| } |
| |
| v->data = p; |
| |
| i = 0; |
| state = r->upstream_states->elts; |
| |
| for ( ;; ) { |
| if (state[i].status) { |
| ms = (ngx_msec_int_t) |
| (state[i].response_sec * 1000 + state[i].response_msec); |
| ms = (ms >= 0) ? ms : 0; |
| p = ngx_sprintf(p, "%d.%03d", ms / 1000, ms % 1000); |
| |
| } else { |
| *p++ = '-'; |
| } |
| |
| if (++i == r->upstream_states->nelts) { |
| break; |
| } |
| |
| if (state[i].peer) { |
| *p++ = ','; |
| *p++ = ' '; |
| |
| } else { |
| *p++ = ' '; |
| *p++ = ':'; |
| *p++ = ' '; |
| |
| if (++i == r->upstream_states->nelts) { |
| break; |
| } |
| |
| continue; |
| } |
| } |
| |
| v->len = p - v->data; |
| |
| return NGX_OK; |
| } |
| |
| |
| static ngx_int_t |
| ngx_http_upstream_response_length_variable(ngx_http_request_t *r, |
| ngx_http_variable_value_t *v, uintptr_t data) |
| { |
| u_char *p; |
| size_t len; |
| ngx_uint_t i; |
| ngx_http_upstream_state_t *state; |
| |
| v->valid = 1; |
| v->no_cacheable = 0; |
| v->not_found = 0; |
| |
| if (r->upstream_states == NULL || r->upstream_states->nelts == 0) { |
| v->not_found = 1; |
| return NGX_OK; |
| } |
| |
| len = r->upstream_states->nelts * (NGX_OFF_T_LEN + 2); |
| |
| p = ngx_pnalloc(r->pool, len); |
| if (p == NULL) { |
| return NGX_ERROR; |
| } |
| |
| v->data = p; |
| |
| i = 0; |
| state = r->upstream_states->elts; |
| |
| for ( ;; ) { |
| p = ngx_sprintf(p, "%O", state[i].response_length); |
| |
| if (++i == r->upstream_states->nelts) { |
| break; |
| } |
| |
| if (state[i].peer) { |
| *p++ = ','; |
| *p++ = ' '; |
| |
| } else { |
| *p++ = ' '; |
| *p++ = ':'; |
| *p++ = ' '; |
| |
| if (++i == r->upstream_states->nelts) { |
| break; |
| } |
| |
| continue; |
| } |
| } |
| |
| v->len = p - v->data; |
| |
| return NGX_OK; |
| } |
| |
| |
| ngx_int_t |
| ngx_http_upstream_header_variable(ngx_http_request_t *r, |
| ngx_http_variable_value_t *v, uintptr_t data) |
| { |
| if (r->upstream == NULL) { |
| v->not_found = 1; |
| return NGX_OK; |
| } |
| |
| return ngx_http_variable_unknown_header(v, (ngx_str_t *) data, |
| &r->upstream->headers_in.headers.part, |
| sizeof("upstream_http_") - 1); |
| } |
| |
| |
| #if (NGX_HTTP_CACHE) |
| |
| ngx_int_t |
| ngx_http_upstream_cache_status(ngx_http_request_t *r, |
| ngx_http_variable_value_t *v, uintptr_t data) |
| { |
| ngx_uint_t n; |
| |
| if (r->upstream == NULL || r->upstream->cache_status == 0) { |
| v->not_found = 1; |
| return NGX_OK; |
| } |
| |
| n = r->upstream->cache_status - 1; |
| |
| v->valid = 1; |
| v->no_cacheable = 0; |
| v->not_found = 0; |
| v->len = ngx_http_cache_status[n].len; |
| v->data = ngx_http_cache_status[n].data; |
| |
| return NGX_OK; |
| } |
| |
| #endif |
| |
| |
| static char * |
| ngx_http_upstream(ngx_conf_t *cf, ngx_command_t *cmd, void *dummy) |
| { |
| char *rv; |
| void *mconf; |
| ngx_str_t *value; |
| ngx_url_t u; |
| ngx_uint_t m; |
| ngx_conf_t pcf; |
| ngx_http_module_t *module; |
| ngx_http_conf_ctx_t *ctx, *http_ctx; |
| ngx_http_upstream_srv_conf_t *uscf; |
| |
| ngx_memzero(&u, sizeof(ngx_url_t)); |
| |
| value = cf->args->elts; |
| u.host = value[1]; |
| u.no_resolve = 1; |
| |
| uscf = ngx_http_upstream_add(cf, &u, NGX_HTTP_UPSTREAM_CREATE |
| |NGX_HTTP_UPSTREAM_WEIGHT |
| |NGX_HTTP_UPSTREAM_MAX_FAILS |
| |NGX_HTTP_UPSTREAM_FAIL_TIMEOUT |
| |NGX_HTTP_UPSTREAM_DOWN |
| |NGX_HTTP_UPSTREAM_BACKUP); |
| if (uscf == NULL) { |
| return NGX_CONF_ERROR; |
| } |
| |
| |
| ctx = ngx_pcalloc(cf->pool, sizeof(ngx_http_conf_ctx_t)); |
| if (ctx == NULL) { |
| return NGX_CONF_ERROR; |
| } |
| |
| http_ctx = cf->ctx; |
| ctx->main_conf = http_ctx->main_conf; |
| |
| /* the upstream{}'s srv_conf */ |
| |
| ctx->srv_conf = ngx_pcalloc(cf->pool, sizeof(void *) * ngx_http_max_module); |
| if (ctx->srv_conf == NULL) { |
| return NGX_CONF_ERROR; |
| } |
| |
| ctx->srv_conf[ngx_http_upstream_module.ctx_index] = uscf; |
| |
| uscf->srv_conf = ctx->srv_conf; |
| |
| |
| /* the upstream{}'s loc_conf */ |
| |
| ctx->loc_conf = ngx_pcalloc(cf->pool, sizeof(void *) * ngx_http_max_module); |
| if (ctx->loc_conf == NULL) { |
| return NGX_CONF_ERROR; |
| } |
| |
| for (m = 0; ngx_modules[m]; m++) { |
| if (ngx_modules[m]->type != NGX_HTTP_MODULE) { |
| continue; |
| } |
| |
| module = ngx_modules[m]->ctx; |
| |
| if (module->create_srv_conf) { |
| mconf = module->create_srv_conf(cf); |
| if (mconf == NULL) { |
| return NGX_CONF_ERROR; |
| } |
| |
| ctx->srv_conf[ngx_modules[m]->ctx_index] = mconf; |
| } |
| |
| if (module->create_loc_conf) { |
| mconf = module->create_loc_conf(cf); |
| if (mconf == NULL) { |
| return NGX_CONF_ERROR; |
| } |
| |
| ctx->loc_conf[ngx_modules[m]->ctx_index] = mconf; |
| } |
| } |
| |
| |
| /* parse inside upstream{} */ |
| |
| pcf = *cf; |
| cf->ctx = ctx; |
| cf->cmd_type = NGX_HTTP_UPS_CONF; |
| |
| rv = ngx_conf_parse(cf, NULL); |
| |
| *cf = pcf; |
| |
| if (rv != NGX_CONF_OK) { |
| return rv; |
| } |
| |
| if (uscf->servers == NULL) { |
| ngx_conf_log_error(NGX_LOG_EMERG, cf, 0, |
| "no servers are inside upstream"); |
| return NGX_CONF_ERROR; |
| } |
| |
| return rv; |
| } |
| |
| |
| static char * |
| ngx_http_upstream_server(ngx_conf_t *cf, ngx_command_t *cmd, void *conf) |
| { |
| ngx_http_upstream_srv_conf_t *uscf = conf; |
| |
| time_t fail_timeout; |
| ngx_str_t *value, s; |
| ngx_url_t u; |
| ngx_int_t weight, max_fails; |
| ngx_uint_t i; |
| ngx_http_upstream_server_t *us; |
| |
| if (uscf->servers == NULL) { |
| uscf->servers = ngx_array_create(cf->pool, 4, |
| sizeof(ngx_http_upstream_server_t)); |
| if (uscf->servers == NULL) { |
| return NGX_CONF_ERROR; |
| } |
| } |
| |
| us = ngx_array_push(uscf->servers); |
| if (us == NULL) { |
| return NGX_CONF_ERROR; |
| } |
| |
| ngx_memzero(us, sizeof(ngx_http_upstream_server_t)); |
| |
| value = cf->args->elts; |
| |
| ngx_memzero(&u, sizeof(ngx_url_t)); |
| |
| u.url = value[1]; |
| u.default_port = 80; |
| |
| if (ngx_parse_url(cf->pool, &u) != NGX_OK) { |
| if (u.err) { |
| ngx_conf_log_error(NGX_LOG_EMERG, cf, 0, |
| "%s in upstream \"%V\"", u.err, &u.url); |
| } |
| |
| return NGX_CONF_ERROR; |
| } |
| |
| weight = 1; |
| max_fails = 1; |
| fail_timeout = 10; |
| |
| for (i = 2; i < cf->args->nelts; i++) { |
| |
| if (ngx_strncmp(value[i].data, "weight=", 7) == 0) { |
| |
| if (!(uscf->flags & NGX_HTTP_UPSTREAM_WEIGHT)) { |
| goto invalid; |
| } |
| |
| weight = ngx_atoi(&value[i].data[7], value[i].len - 7); |
| |
| if (weight == NGX_ERROR || weight == 0) { |
| goto invalid; |
| } |
| |
| continue; |
| } |
| |
| if (ngx_strncmp(value[i].data, "max_fails=", 10) == 0) { |
| |
| if (!(uscf->flags & NGX_HTTP_UPSTREAM_MAX_FAILS)) { |
| goto invalid; |
| } |
| |
| max_fails = ngx_atoi(&value[i].data[10], value[i].len - 10); |
| |
| if (max_fails == NGX_ERROR) { |
| goto invalid; |
| } |
| |
| continue; |
| } |
| |
| if (ngx_strncmp(value[i].data, "fail_timeout=", 13) == 0) { |
| |
| if (!(uscf->flags & NGX_HTTP_UPSTREAM_FAIL_TIMEOUT)) { |
| goto invalid; |
| } |
| |
| s.len = value[i].len - 13; |
| s.data = &value[i].data[13]; |
| |
| fail_timeout = ngx_parse_time(&s, 1); |
| |
| if (fail_timeout == NGX_ERROR) { |
| goto invalid; |
| } |
| |
| continue; |
| } |
| |
| if (ngx_strncmp(value[i].data, "backup", 6) == 0) { |
| |
| if (!(uscf->flags & NGX_HTTP_UPSTREAM_BACKUP)) { |
| goto invalid; |
| } |
| |
| us->backup = 1; |
| |
| continue; |
| } |
| |
| if (ngx_strncmp(value[i].data, "down", 4) == 0) { |
| |
| if (!(uscf->flags & NGX_HTTP_UPSTREAM_DOWN)) { |
| goto invalid; |
| } |
| |
| us->down = 1; |
| |
| continue; |
| } |
| |
| goto invalid; |
| } |
| |
| us->addrs = u.addrs; |
| us->naddrs = u.naddrs; |
| us->weight = weight; |
| us->max_fails = max_fails; |
| us->fail_timeout = fail_timeout; |
| |
| return NGX_CONF_OK; |
| |
| invalid: |
| |
| ngx_conf_log_error(NGX_LOG_EMERG, cf, 0, |
| "invalid parameter \"%V\"", &value[i]); |
| |
| return NGX_CONF_ERROR; |
| } |
| |
| |
| ngx_http_upstream_srv_conf_t * |
| ngx_http_upstream_add(ngx_conf_t *cf, ngx_url_t *u, ngx_uint_t flags) |
| { |
| ngx_uint_t i; |
| ngx_http_upstream_server_t *us; |
| ngx_http_upstream_srv_conf_t *uscf, **uscfp; |
| ngx_http_upstream_main_conf_t *umcf; |
| |
| if (!(flags & NGX_HTTP_UPSTREAM_CREATE)) { |
| |
| if (ngx_parse_url(cf->pool, u) != NGX_OK) { |
| if (u->err) { |
| ngx_conf_log_error(NGX_LOG_EMERG, cf, 0, |
| "%s in upstream \"%V\"", u->err, &u->url); |
| } |
| |
| return NULL; |
| } |
| } |
| |
| umcf = ngx_http_conf_get_module_main_conf(cf, ngx_http_upstream_module); |
| |
| uscfp = umcf->upstreams.elts; |
| |
| for (i = 0; i < umcf->upstreams.nelts; i++) { |
| |
| if (uscfp[i]->host.len != u->host.len |
| || ngx_strncasecmp(uscfp[i]->host.data, u->host.data, u->host.len) |
| != 0) |
| { |
| continue; |
| } |
| |
| if ((flags & NGX_HTTP_UPSTREAM_CREATE) |
| && (uscfp[i]->flags & NGX_HTTP_UPSTREAM_CREATE)) |
| { |
| ngx_conf_log_error(NGX_LOG_EMERG, cf, 0, |
| "duplicate upstream \"%V\"", &u->host); |
| return NULL; |
| } |
| |
| if ((uscfp[i]->flags & NGX_HTTP_UPSTREAM_CREATE) && u->port) { |
| ngx_conf_log_error(NGX_LOG_WARN, cf, 0, |
| "upstream \"%V\" may not have port %d", |
| &u->host, u->port); |
| return NULL; |
| } |
| |
| if ((flags & NGX_HTTP_UPSTREAM_CREATE) && uscfp[i]->port) { |
| ngx_log_error(NGX_LOG_WARN, cf->log, 0, |
| "upstream \"%V\" may not have port %d in %s:%ui", |
| &u->host, uscfp[i]->port, |
| uscfp[i]->file_name, uscfp[i]->line); |
| return NULL; |
| } |
| |
| if (uscfp[i]->port != u->port) { |
| continue; |
| } |
| |
| if (uscfp[i]->default_port && u->default_port |
| && uscfp[i]->default_port != u->default_port) |
| { |
| continue; |
| } |
| |
| return uscfp[i]; |
| } |
| |
| uscf = ngx_pcalloc(cf->pool, sizeof(ngx_http_upstream_srv_conf_t)); |
| if (uscf == NULL) { |
| return NULL; |
| } |
| |
| uscf->flags = flags; |
| uscf->host = u->host; |
| uscf->file_name = cf->conf_file->file.name.data; |
| uscf->line = cf->conf_file->line; |
| uscf->port = u->port; |
| uscf->default_port = u->default_port; |
| |
| if (u->naddrs == 1) { |
| uscf->servers = ngx_array_create(cf->pool, 1, |
| sizeof(ngx_http_upstream_server_t)); |
| if (uscf->servers == NULL) { |
| return NGX_CONF_ERROR; |
| } |
| |
| us = ngx_array_push(uscf->servers); |
| if (us == NULL) { |
| return NGX_CONF_ERROR; |
| } |
| |
| ngx_memzero(us, sizeof(ngx_http_upstream_server_t)); |
| |
| us->addrs = u->addrs; |
| us->naddrs = u->naddrs; |
| } |
| |
| uscfp = ngx_array_push(&umcf->upstreams); |
| if (uscfp == NULL) { |
| return NULL; |
| } |
| |
| *uscfp = uscf; |
| |
| return uscf; |
| } |
| |
| |
| ngx_int_t |
| ngx_http_upstream_hide_headers_hash(ngx_conf_t *cf, |
| ngx_http_upstream_conf_t *conf, ngx_http_upstream_conf_t *prev, |
| ngx_str_t *default_hide_headers, ngx_hash_init_t *hash) |
| { |
| ngx_str_t *h; |
| ngx_uint_t i, j; |
| ngx_array_t hide_headers; |
| ngx_hash_key_t *hk; |
| |
| if (conf->hide_headers == NGX_CONF_UNSET_PTR |
| && conf->pass_headers == NGX_CONF_UNSET_PTR) |
| { |
| conf->hide_headers_hash = prev->hide_headers_hash; |
| |
| if (conf->hide_headers_hash.buckets |
| #if (NGX_HTTP_CACHE) |
| && ((conf->cache == NULL) == (prev->cache == NULL)) |
| #endif |
| ) |
| { |
| return NGX_OK; |
| } |
| |
| conf->hide_headers = prev->hide_headers; |
| conf->pass_headers = prev->pass_headers; |
| |
| } else { |
| if (conf->hide_headers == NGX_CONF_UNSET_PTR) { |
| conf->hide_headers = prev->hide_headers; |
| } |
| |
| if (conf->pass_headers == NGX_CONF_UNSET_PTR) { |
| conf->pass_headers = prev->pass_headers; |
| } |
| } |
| |
| if (ngx_array_init(&hide_headers, cf->temp_pool, 4, sizeof(ngx_hash_key_t)) |
| != NGX_OK) |
| { |
| return NGX_ERROR; |
| } |
| |
| for (h = default_hide_headers; h->len; h++) { |
| hk = ngx_array_push(&hide_headers); |
| if (hk == NULL) { |
| return NGX_ERROR; |
| } |
| |
| hk->key = *h; |
| hk->key_hash = ngx_hash_key_lc(h->data, h->len); |
| hk->value = (void *) 1; |
| } |
| |
| if (conf->hide_headers != NGX_CONF_UNSET_PTR) { |
| |
| h = conf->hide_headers->elts; |
| |
| for (i = 0; i < conf->hide_headers->nelts; i++) { |
| |
| hk = hide_headers.elts; |
| |
| for (j = 0; j < hide_headers.nelts; j++) { |
| if (ngx_strcasecmp(h[i].data, hk[j].key.data) == 0) { |
| goto exist; |
| } |
| } |
| |
| hk = ngx_array_push(&hide_headers); |
| if (hk == NULL) { |
| return NGX_ERROR; |
| } |
| |
| hk->key = h[i]; |
| hk->key_hash = ngx_hash_key_lc(h[i].data, h[i].len); |
| hk->value = (void *) 1; |
| |
| exist: |
| |
| continue; |
| } |
| } |
| |
| if (conf->pass_headers != NGX_CONF_UNSET_PTR) { |
| |
| h = conf->pass_headers->elts; |
| hk = hide_headers.elts; |
| |
| for (i = 0; i < conf->pass_headers->nelts; i++) { |
| for (j = 0; j < hide_headers.nelts; j++) { |
| |
| if (hk[j].key.data == NULL) { |
| continue; |
| } |
| |
| if (ngx_strcasecmp(h[i].data, hk[j].key.data) == 0) { |
| hk[j].key.data = NULL; |
| break; |
| } |
| } |
| } |
| } |
| |
| hash->hash = &conf->hide_headers_hash; |
| hash->key = ngx_hash_key_lc; |
| hash->pool = cf->pool; |
| hash->temp_pool = NULL; |
| |
| return ngx_hash_init(hash, hide_headers.elts, hide_headers.nelts); |
| } |
| |
| |
| static void * |
| ngx_http_upstream_create_main_conf(ngx_conf_t *cf) |
| { |
| ngx_http_upstream_main_conf_t *umcf; |
| |
| umcf = ngx_pcalloc(cf->pool, sizeof(ngx_http_upstream_main_conf_t)); |
| if (umcf == NULL) { |
| return NULL; |
| } |
| |
| if (ngx_array_init(&umcf->upstreams, cf->pool, 4, |
| sizeof(ngx_http_upstream_srv_conf_t *)) |
| != NGX_OK) |
| { |
| return NULL; |
| } |
| |
| return umcf; |
| } |
| |
| |
| static char * |
| ngx_http_upstream_init_main_conf(ngx_conf_t *cf, void *conf) |
| { |
| ngx_http_upstream_main_conf_t *umcf = conf; |
| |
| ngx_uint_t i; |
| ngx_array_t headers_in; |
| ngx_hash_key_t *hk; |
| ngx_hash_init_t hash; |
| ngx_http_upstream_init_pt init; |
| ngx_http_upstream_header_t *header; |
| ngx_http_upstream_srv_conf_t **uscfp; |
| |
| uscfp = umcf->upstreams.elts; |
| |
| for (i = 0; i < umcf->upstreams.nelts; i++) { |
| |
| init = uscfp[i]->peer.init_upstream ? uscfp[i]->peer.init_upstream: |
| ngx_http_upstream_init_round_robin; |
| |
| if (init(cf, uscfp[i]) != NGX_OK) { |
| return NGX_CONF_ERROR; |
| } |
| } |
| |
| |
| /* upstream_headers_in_hash */ |
| |
| if (ngx_array_init(&headers_in, cf->temp_pool, 32, sizeof(ngx_hash_key_t)) |
| != NGX_OK) |
| { |
| return NGX_CONF_ERROR; |
| } |
| |
| for (header = ngx_http_upstream_headers_in; header->name.len; header++) { |
| hk = ngx_array_push(&headers_in); |
| if (hk == NULL) { |
| return NGX_CONF_ERROR; |
| } |
| |
| hk->key = header->name; |
| hk->key_hash = ngx_hash_key_lc(header->name.data, header->name.len); |
| hk->value = header; |
| } |
| |
| hash.hash = &umcf->headers_in_hash; |
| hash.key = ngx_hash_key_lc; |
| hash.max_size = 512; |
| hash.bucket_size = ngx_align(64, ngx_cacheline_size); |
| hash.name = "upstream_headers_in_hash"; |
| hash.pool = cf->pool; |
| hash.temp_pool = NULL; |
| |
| if (ngx_hash_init(&hash, headers_in.elts, headers_in.nelts) != NGX_OK) { |
| return NGX_CONF_ERROR; |
| } |
| |
| return NGX_CONF_OK; |
| } |