nginx-quic: src/http/modules/ngx_http_fastcgi

comparison src/http/modules/ngx_http_fastcgi_module.c @ 3560:a36c63c6d26a

allow fastcgi_param to override client headers using HTTP_ parameters

author	Igor Sysoev <igor@sysoev.ru>
date	Thu, 03 Jun 2010 14:41:30 +0000
parents	519dcf4cb3de
children	e2d54eb7472e

comparison

equal deleted inserted replaced

-:519dcf4cb3de
+:a36c63c6d26a
 ngx_array_t                   *params_source;
 ngx_array_t                   *catch_stderr;
 ngx_array_t                   *fastcgi_lengths;
 ngx_array_t                   *fastcgi_values;
+ngx_hash_t                     headers_hash;
+ngx_uint_t                     header_params;
 #if (NGX_HTTP_CACHE)
 ngx_http_complex_value_t       cache_key;
 #endif
 static ngx_int_t
 ngx_http_fastcgi_create_request(ngx_http_request_t *r)
 {
 off_t                         file_pos;
-u_char                        ch, *pos;
+u_char                        ch, *pos, *lowcase_key;
-size_t                        size, len, key_len, val_len, padding;
+size_t                        size, len, key_len, val_len, padding,
-ngx_uint_t                    i, n, next;
+allocated;
+ngx_uint_t                    i, n, next, hash, header_params;
 ngx_buf_t                    *b;
 ngx_chain_t                  *cl, *body;
 ngx_list_part_t              *part;
-ngx_table_elt_t              *header;
+ngx_table_elt_t              *header, **ignored;
 ngx_http_script_code_pt       code;
 ngx_http_script_engine_t      e, le;
 ngx_http_fastcgi_header_t    *h;
 ngx_http_fastcgi_loc_conf_t  *flcf;
 ngx_http_script_len_code_pt   lcode;
 len = 0;
+header_params = 0;
+ignored = NULL;
 flcf = ngx_http_get_module_loc_conf(r, ngx_http_fastcgi_module);
 if (flcf->params_len) {
 ngx_memzero(&le, sizeof(ngx_http_script_engine_t));
 len += 1 + key_len + ((val_len > 127) ? 4 : 1) + val_len;
 }
 }
 if (flcf->upstream.pass_request_headers) {
+allocated = 0;
+lowcase_key = NULL;
+if (flcf->header_params) {
+ignored = ngx_palloc(r->pool, flcf->header_params * sizeof(void *));
+if (ignored == NULL) {
+return NGX_ERROR;
+}
+}
 part = &r->headers_in.headers.part;
 header = part->elts;
 for (i = 0; /* void */; i++) {
 part = part->next;
 header = part->elts;
 i = 0;
 }
-len += ((sizeof("HTTP_") - 1 + header[i].key.len > 127) ? 4 : 1)
+if (flcf->header_params) {
-+ ((header[i].value.len > 127) ? 4 : 1)
+if (allocated < header[i].key.len) {
-+ sizeof("HTTP_") - 1 + header[i].key.len + header[i].value.len;
+allocated = header[i].key.len + 16;
+lowcase_key = ngx_pnalloc(r->pool, allocated);
+if (lowcase_key == NULL) {
+return NGX_ERROR;
+}
+}
+hash = 0;
+for (n = 0; n < header[i].key.len; n++) {
+ch = header[i].key.data[n];
+if (ch >= 'A' && ch <= 'Z') {
+ch |= 0x20;
+} else if (ch == '-') {
+ch = '_';
+}
+hash = ngx_hash(hash, ch);
+lowcase_key[n] = ch;
+}
+if (ngx_hash_find(&flcf->headers_hash, hash, lowcase_key, n)) {
+ignored[header_params++] = &header[i];
+continue;
+}
+n += sizeof("HTTP_") - 1;
+} else {
+n = sizeof("HTTP_") - 1 + header[i].key.len;
+}
+len += ((n > 127) ? 4 : 1) + ((header[i].value.len > 127) ? 4 : 1)
++ n + header[i].value.len;
 }
 }
 if (len > 65535) {
 }
 part = part->next;
 header = part->elts;
 i = 0;
+}
+for (n = 0; n < header_params; n++) {
+if (&header[i] == ignored[n]) {
+goto next;
+}
 }
 key_len = sizeof("HTTP_") - 1 + header[i].key.len;
 if (key_len > 127) {
 *b->last++ = (u_char) (((key_len >> 24) & 0x7f) | 0x80);
 ngx_log_debug4(NGX_LOG_DEBUG_HTTP, r->connection->log, 0,
 "fastcgi param: \"%*s: %*s\"",
 key_len, b->last - (key_len + val_len),
 val_len, b->last - val_len);
+next:
+continue;
 }
 }
 if (padding) {
 u_char                       *p;
 size_t                        size;
 uintptr_t                    *code;
 ngx_str_t                    *h;
 ngx_uint_t                    i;
+ngx_array_t                   headers_names;
 ngx_keyval_t                 *src;
+ngx_hash_key_t               *hk;
 ngx_hash_init_t               hash;
 ngx_http_script_compile_t     sc;
 ngx_http_script_copy_code_t  *copy;
 if (conf->upstream.store != 0) {
 if (conf->params_source == NULL) {
 conf->flushes = prev->flushes;
 conf->params_len = prev->params_len;
 conf->params = prev->params;
 conf->params_source = prev->params_source;
+conf->headers_hash = prev->headers_hash;
 if (conf->params_source == NULL) {
 return NGX_CONF_OK;
 }
 }
 conf->params = ngx_array_create(cf->pool, 512, 1);
 if (conf->params == NULL) {
 return NGX_CONF_ERROR;
 }
+if (ngx_array_init(&headers_names, cf->temp_pool, 4, sizeof(ngx_hash_key_t))
+!= NGX_OK)
+{
+return NGX_CONF_ERROR;
+}
 src = conf->params_source->elts;
 for (i = 0; i < conf->params_source->nelts; i++) {
+if (src[i].key.len > sizeof("HTTP_") - 1
+&& ngx_strncmp(src[i].key.data, "HTTP_", sizeof("HTTP_") - 1) == 0)
+{
+hk = ngx_array_push(&headers_names);
+if (hk == NULL) {
+return NGX_CONF_ERROR;
+}
+hk->key.len = src[i].key.len - 5;
+hk->key.data = src[i].key.data + 5;
+hk->key_hash = ngx_hash_key_lc(hk->key.data, hk->key.len);
+hk->value = (void *) 1;
+if (src[i].value.len == 0) {
+continue;
+}
+}
 copy = ngx_array_push_n(conf->params_len,
 sizeof(ngx_http_script_copy_code_t));
 if (copy == NULL) {
 return NGX_CONF_ERROR;
 if (code == NULL) {
 return NGX_CONF_ERROR;
 }
 *code = (uintptr_t) NULL;
+conf->header_params = headers_names.nelts;
+hash.hash = &conf->headers_hash;
+hash.key = ngx_hash_key_lc;
+hash.max_size = 512;
+hash.bucket_size = 64;
+hash.name = "fastcgi_params_hash";
+hash.pool = cf->pool;
+hash.temp_pool = NULL;
+if (ngx_hash_init(&hash, headers_names.elts, headers_names.nelts) != NGX_OK)
+{
+return NGX_CONF_ERROR;
+}
 return NGX_CONF_OK;
 }

Mercurial > hg > nginx-quic

comparison src/http/modules/ngx_http_fastcgi_module.c @ 3560:a36c63c6d26a