Mercurial > hg > nginx-quic
annotate src/http/ngx_http_variables.c @ 982:664afe2c1480
$binary_remote_addr
author | Igor Sysoev <igor@sysoev.ru> |
---|---|
date | Sun, 07 Jan 2007 13:35:08 +0000 |
parents | a6fe6bedb9e3 |
children | 7a8ca436d611 |
rev | line source |
---|---|
499 | 1 |
2 /* | |
3 * Copyright (C) Igor Sysoev | |
4 */ | |
5 | |
6 | |
7 #include <ngx_config.h> | |
8 #include <ngx_core.h> | |
9 #include <ngx_event.h> | |
10 #include <ngx_http.h> | |
11 | |
12 | |
573 | 13 static ngx_int_t ngx_http_variable_request(ngx_http_request_t *r, |
14 ngx_http_variable_value_t *v, uintptr_t data); | |
637 | 15 static void ngx_http_variable_request_set_size(ngx_http_request_t *r, |
16 ngx_http_variable_value_t *v, uintptr_t data); | |
573 | 17 static ngx_int_t ngx_http_variable_header(ngx_http_request_t *r, |
18 ngx_http_variable_value_t *v, uintptr_t data); | |
19 static ngx_int_t ngx_http_variable_headers(ngx_http_request_t *r, | |
20 ngx_http_variable_value_t *v, uintptr_t data); | |
577 | 21 |
22 static ngx_int_t ngx_http_variable_unknown_header_in(ngx_http_request_t *r, | |
573 | 23 ngx_http_variable_value_t *v, uintptr_t data); |
577 | 24 static ngx_int_t ngx_http_variable_unknown_header_out(ngx_http_request_t *r, |
25 ngx_http_variable_value_t *v, uintptr_t data); | |
26 static ngx_int_t ngx_http_variable_unknown_header(ngx_http_variable_value_t *v, | |
27 ngx_str_t *var, ngx_list_part_t *part, size_t prefix); | |
28 | |
573 | 29 static ngx_int_t ngx_http_variable_host(ngx_http_request_t *r, |
30 ngx_http_variable_value_t *v, uintptr_t data); | |
31 static ngx_int_t ngx_http_variable_remote_addr(ngx_http_request_t *r, | |
32 ngx_http_variable_value_t *v, uintptr_t data); | |
33 static ngx_int_t ngx_http_variable_remote_port(ngx_http_request_t *r, | |
34 ngx_http_variable_value_t *v, uintptr_t data); | |
35 static ngx_int_t ngx_http_variable_server_addr(ngx_http_request_t *r, | |
36 ngx_http_variable_value_t *v, uintptr_t data); | |
37 static ngx_int_t ngx_http_variable_server_port(ngx_http_request_t *r, | |
38 ngx_http_variable_value_t *v, uintptr_t data); | |
731 | 39 static ngx_int_t ngx_http_variable_scheme(ngx_http_request_t *r, |
40 ngx_http_variable_value_t *v, uintptr_t data); | |
573 | 41 static ngx_int_t ngx_http_variable_document_root(ngx_http_request_t *r, |
42 ngx_http_variable_value_t *v, uintptr_t data); | |
43 static ngx_int_t ngx_http_variable_request_filename(ngx_http_request_t *r, | |
44 ngx_http_variable_value_t *v, uintptr_t data); | |
45 static ngx_int_t ngx_http_variable_request_method(ngx_http_request_t *r, | |
46 ngx_http_variable_value_t *v, uintptr_t data); | |
47 static ngx_int_t ngx_http_variable_remote_user(ngx_http_request_t *r, | |
48 ngx_http_variable_value_t *v, uintptr_t data); | |
611 | 49 static ngx_int_t ngx_http_variable_body_bytes_sent(ngx_http_request_t *r, |
50 ngx_http_variable_value_t *v, uintptr_t data); | |
629 | 51 static ngx_int_t ngx_http_variable_request_completion(ngx_http_request_t *r, |
52 ngx_http_variable_value_t *v, uintptr_t data); | |
759 | 53 static ngx_int_t ngx_http_variable_request_body_file(ngx_http_request_t *r, |
54 ngx_http_variable_value_t *v, uintptr_t data); | |
499 | 55 |
641 | 56 static ngx_int_t ngx_http_variable_sent_content_type(ngx_http_request_t *r, |
57 ngx_http_variable_value_t *v, uintptr_t data); | |
58 static ngx_int_t ngx_http_variable_sent_content_length(ngx_http_request_t *r, | |
59 ngx_http_variable_value_t *v, uintptr_t data); | |
60 static ngx_int_t ngx_http_variable_sent_last_modified(ngx_http_request_t *r, | |
61 ngx_http_variable_value_t *v, uintptr_t data); | |
62 static ngx_int_t ngx_http_variable_sent_connection(ngx_http_request_t *r, | |
63 ngx_http_variable_value_t *v, uintptr_t data); | |
64 static ngx_int_t ngx_http_variable_sent_keep_alive(ngx_http_request_t *r, | |
65 ngx_http_variable_value_t *v, uintptr_t data); | |
66 static ngx_int_t ngx_http_variable_sent_transfer_encoding(ngx_http_request_t *r, | |
67 ngx_http_variable_value_t *v, uintptr_t data); | |
68 | |
499 | 69 |
509 | 70 /* |
71 * TODO: | |
72 * Apache CGI: AUTH_TYPE, PATH_INFO (null), PATH_TRANSLATED | |
73 * REMOTE_HOST (null), REMOTE_IDENT (null), | |
74 * SERVER_SOFTWARE | |
75 * | |
571 | 76 * Apache SSI: DOCUMENT_NAME, LAST_MODIFIED, USER_NAME (file owner) |
509 | 77 */ |
499 | 78 |
641 | 79 /* |
80 * the $http_host, $http_user_agent, $http_referer, $http_via, | |
81 * and $http_x_forwarded_for variables may be handled by generic | |
82 * ngx_http_variable_unknown_header_in(), but for perfomance reasons | |
83 * they are handled using dedicated entries | |
84 */ | |
85 | |
509 | 86 static ngx_http_variable_t ngx_http_core_variables[] = { |
87 | |
637 | 88 { ngx_string("http_host"), NULL, ngx_http_variable_header, |
533 | 89 offsetof(ngx_http_request_t, headers_in.host), 0, 0 }, |
499 | 90 |
637 | 91 { ngx_string("http_user_agent"), NULL, ngx_http_variable_header, |
533 | 92 offsetof(ngx_http_request_t, headers_in.user_agent), 0, 0 }, |
509 | 93 |
637 | 94 { ngx_string("http_referer"), NULL, ngx_http_variable_header, |
533 | 95 offsetof(ngx_http_request_t, headers_in.referer), 0, 0 }, |
499 | 96 |
97 #if (NGX_HTTP_GZIP) | |
637 | 98 { ngx_string("http_via"), NULL, ngx_http_variable_header, |
533 | 99 offsetof(ngx_http_request_t, headers_in.via), 0, 0 }, |
499 | 100 #endif |
101 | |
102 #if (NGX_HTTP_PROXY) | |
637 | 103 { ngx_string("http_x_forwarded_for"), NULL, ngx_http_variable_header, |
533 | 104 offsetof(ngx_http_request_t, headers_in.x_forwarded_for), 0, 0 }, |
499 | 105 #endif |
106 | |
637 | 107 { ngx_string("http_cookie"), NULL, ngx_http_variable_headers, |
533 | 108 offsetof(ngx_http_request_t, headers_in.cookies), 0, 0 }, |
515 | 109 |
637 | 110 { ngx_string("content_length"), NULL, ngx_http_variable_header, |
533 | 111 offsetof(ngx_http_request_t, headers_in.content_length), 0, 0 }, |
509 | 112 |
637 | 113 { ngx_string("content_type"), NULL, ngx_http_variable_header, |
533 | 114 offsetof(ngx_http_request_t, headers_in.content_type), 0, 0 }, |
509 | 115 |
637 | 116 { ngx_string("host"), NULL, ngx_http_variable_host, 0, 0, 0 }, |
509 | 117 |
982 | 118 { ngx_string("binary_remote_addr"), NULL, |
119 ngx_http_variable_remote_addr, 1, 0, 0 }, | |
120 | |
637 | 121 { ngx_string("remote_addr"), NULL, ngx_http_variable_remote_addr, 0, 0, 0 }, |
499 | 122 |
637 | 123 { ngx_string("remote_port"), NULL, ngx_http_variable_remote_port, 0, 0, 0 }, |
509 | 124 |
637 | 125 { ngx_string("server_addr"), NULL, ngx_http_variable_server_addr, 0, 0, 0 }, |
509 | 126 |
637 | 127 { ngx_string("server_port"), NULL, ngx_http_variable_server_port, 0, 0, 0 }, |
509 | 128 |
637 | 129 { ngx_string("server_protocol"), NULL, ngx_http_variable_request, |
533 | 130 offsetof(ngx_http_request_t, http_protocol), 0, 0 }, |
509 | 131 |
731 | 132 { ngx_string("scheme"), NULL, ngx_http_variable_scheme, 0, 0, 0 }, |
133 | |
637 | 134 { ngx_string("request_uri"), NULL, ngx_http_variable_request, |
533 | 135 offsetof(ngx_http_request_t, unparsed_uri), 0, 0 }, |
499 | 136 |
637 | 137 { ngx_string("uri"), NULL, ngx_http_variable_request, |
138 offsetof(ngx_http_request_t, uri), | |
139 NGX_HTTP_VAR_NOCACHABLE, 0 }, | |
140 | |
141 { ngx_string("document_uri"), NULL, ngx_http_variable_request, | |
573 | 142 offsetof(ngx_http_request_t, uri), |
143 NGX_HTTP_VAR_NOCACHABLE, 0 }, | |
144 | |
637 | 145 { ngx_string("request"), NULL, ngx_http_variable_request, |
569 | 146 offsetof(ngx_http_request_t, request_line), 0, 0 }, |
147 | |
637 | 148 { ngx_string("document_root"), NULL, |
671 | 149 ngx_http_variable_document_root, 0, NGX_HTTP_VAR_NOCACHABLE, 0 }, |
509 | 150 |
637 | 151 { ngx_string("query_string"), NULL, ngx_http_variable_request, |
509 | 152 offsetof(ngx_http_request_t, args), |
533 | 153 NGX_HTTP_VAR_NOCACHABLE, 0 }, |
499 | 154 |
637 | 155 { ngx_string("args"), NULL, ngx_http_variable_request, |
589 | 156 offsetof(ngx_http_request_t, args), |
157 NGX_HTTP_VAR_NOCACHABLE, 0 }, | |
158 | |
637 | 159 { ngx_string("request_filename"), NULL, |
160 ngx_http_variable_request_filename, 0, | |
533 | 161 NGX_HTTP_VAR_NOCACHABLE, 0 }, |
509 | 162 |
637 | 163 { ngx_string("server_name"), NULL, ngx_http_variable_request, |
533 | 164 offsetof(ngx_http_request_t, server_name), 0, 0 }, |
509 | 165 |
637 | 166 { ngx_string("request_method"), NULL, |
167 ngx_http_variable_request_method, 0, 0, 0 }, | |
509 | 168 |
637 | 169 { ngx_string("remote_user"), NULL, ngx_http_variable_remote_user, 0, 0, 0 }, |
509 | 170 |
637 | 171 { ngx_string("body_bytes_sent"), NULL, ngx_http_variable_body_bytes_sent, |
611 | 172 0, 0, 0 }, |
173 | |
637 | 174 { ngx_string("request_completion"), NULL, |
175 ngx_http_variable_request_completion, | |
629 | 176 0, 0, 0 }, |
177 | |
759 | 178 { ngx_string("request_body_file"), NULL, |
179 ngx_http_variable_request_body_file, | |
180 0, 0, 0 }, | |
181 | |
641 | 182 { ngx_string("sent_http_content_type"), NULL, |
183 ngx_http_variable_sent_content_type, 0, 0, 0 }, | |
184 | |
185 { ngx_string("sent_http_content_length"), NULL, | |
186 ngx_http_variable_sent_content_length, 0, 0, 0 }, | |
187 | |
188 { ngx_string("sent_http_last_modified"), NULL, | |
189 ngx_http_variable_sent_last_modified, 0, 0, 0 }, | |
190 | |
191 { ngx_string("sent_http_connection"), NULL, | |
192 ngx_http_variable_sent_connection, 0, 0, 0 }, | |
193 | |
194 { ngx_string("sent_http_keep_alive"), NULL, | |
195 ngx_http_variable_sent_keep_alive, 0, 0, 0 }, | |
196 | |
197 { ngx_string("sent_http_transfer_encoding"), NULL, | |
198 ngx_http_variable_sent_transfer_encoding, 0, 0, 0 }, | |
199 | |
200 { ngx_string("sent_http_cache_control"), NULL, ngx_http_variable_headers, | |
201 offsetof(ngx_http_request_t, headers_out.cache_control), 0, 0 }, | |
202 | |
637 | 203 { ngx_string("limit_rate"), ngx_http_variable_request_set_size, |
204 ngx_http_variable_request, | |
205 offsetof(ngx_http_request_t, limit_rate), | |
206 NGX_HTTP_VAR_CHANGABLE|NGX_HTTP_VAR_NOCACHABLE, 0 }, | |
207 | |
208 { ngx_null_string, NULL, NULL, 0, 0, 0 } | |
499 | 209 }; |
210 | |
211 | |
577 | 212 ngx_http_variable_value_t ngx_http_variable_null_value = |
213 ngx_http_variable(""); | |
214 ngx_http_variable_value_t ngx_http_variable_true_value = | |
215 ngx_http_variable("1"); | |
216 | |
217 | |
499 | 218 ngx_http_variable_t * |
509 | 219 ngx_http_add_variable(ngx_conf_t *cf, ngx_str_t *name, ngx_uint_t flags) |
220 { | |
611 | 221 ngx_int_t rc; |
509 | 222 ngx_uint_t i; |
611 | 223 ngx_hash_key_t *key; |
509 | 224 ngx_http_variable_t *v; |
225 ngx_http_core_main_conf_t *cmcf; | |
226 | |
227 cmcf = ngx_http_conf_get_module_main_conf(cf, ngx_http_core_module); | |
228 | |
611 | 229 key = cmcf->variables_keys->keys.elts; |
230 for (i = 0; i < cmcf->variables_keys->keys.nelts; i++) { | |
231 if (name->len != key[i].key.len | |
232 || ngx_strncasecmp(name->data, key[i].key.data, name->len) != 0) | |
509 | 233 { |
234 continue; | |
235 } | |
236 | |
611 | 237 v = key[i].value; |
238 | |
239 if (!(v->flags & NGX_HTTP_VAR_CHANGABLE)) { | |
509 | 240 ngx_conf_log_error(NGX_LOG_EMERG, cf, 0, |
241 "the duplicate \"%V\" variable", name); | |
242 return NULL; | |
243 } | |
244 | |
611 | 245 return v; |
509 | 246 } |
247 | |
611 | 248 v = ngx_palloc(cf->pool, sizeof(ngx_http_variable_t)); |
509 | 249 if (v == NULL) { |
250 return NULL; | |
251 } | |
252 | |
253 v->name.len = name->len; | |
254 v->name.data = ngx_palloc(cf->pool, name->len); | |
255 if (v->name.data == NULL) { | |
256 return NULL; | |
257 } | |
258 | |
259 for (i = 0; i < name->len; i++) { | |
260 v->name.data[i] = ngx_tolower(name->data[i]); | |
261 } | |
262 | |
637 | 263 v->set_handler = NULL; |
264 v->get_handler = NULL; | |
509 | 265 v->data = 0; |
266 v->flags = flags; | |
533 | 267 v->index = 0; |
509 | 268 |
611 | 269 rc = ngx_hash_add_key(cmcf->variables_keys, &v->name, v, 0); |
270 | |
271 if (rc == NGX_ERROR) { | |
272 return NULL; | |
273 } | |
274 | |
275 if (rc == NGX_BUSY) { | |
276 ngx_conf_log_error(NGX_LOG_EMERG, cf, 0, | |
277 "conflicting variable name \"%V\"", name); | |
278 return NULL; | |
279 } | |
280 | |
509 | 281 return v; |
282 } | |
283 | |
284 | |
285 ngx_int_t | |
286 ngx_http_get_variable_index(ngx_conf_t *cf, ngx_str_t *name) | |
499 | 287 { |
501 | 288 ngx_uint_t i; |
289 ngx_http_variable_t *v; | |
499 | 290 ngx_http_core_main_conf_t *cmcf; |
291 | |
292 cmcf = ngx_http_conf_get_module_main_conf(cf, ngx_http_core_module); | |
293 | |
501 | 294 v = cmcf->variables.elts; |
295 | |
296 if (v == NULL) { | |
499 | 297 if (ngx_array_init(&cmcf->variables, cf->pool, 4, |
298 sizeof(ngx_http_variable_t)) == NGX_ERROR) | |
299 { | |
509 | 300 return NGX_ERROR; |
499 | 301 } |
501 | 302 |
303 } else { | |
304 for (i = 0; i < cmcf->variables.nelts; i++) { | |
305 if (name->len != v[i].name.len | |
306 || ngx_strncasecmp(name->data, v[i].name.data, name->len) != 0) | |
307 { | |
308 continue; | |
309 } | |
310 | |
509 | 311 return i; |
501 | 312 } |
499 | 313 } |
314 | |
501 | 315 v = ngx_array_push(&cmcf->variables); |
316 if (v == NULL) { | |
509 | 317 return NGX_ERROR; |
499 | 318 } |
319 | |
501 | 320 v->name.len = name->len; |
321 v->name.data = ngx_palloc(cf->pool, name->len); | |
322 if (v->name.data == NULL) { | |
509 | 323 return NGX_ERROR; |
501 | 324 } |
499 | 325 |
501 | 326 for (i = 0; i < name->len; i++) { |
509 | 327 v->name.data[i] = ngx_tolower(name->data[i]); |
501 | 328 } |
329 | |
637 | 330 v->set_handler = NULL; |
331 v->get_handler = NULL; | |
501 | 332 v->data = 0; |
509 | 333 v->flags = 0; |
533 | 334 v->index = cmcf->variables.nelts - 1; |
501 | 335 |
509 | 336 return cmcf->variables.nelts - 1; |
499 | 337 } |
338 | |
339 | |
340 ngx_http_variable_value_t * | |
341 ngx_http_get_indexed_variable(ngx_http_request_t *r, ngx_uint_t index) | |
342 { | |
501 | 343 ngx_http_variable_t *v; |
499 | 344 ngx_http_core_main_conf_t *cmcf; |
345 | |
346 cmcf = ngx_http_get_module_main_conf(r, ngx_http_core_module); | |
347 | |
553 | 348 if (cmcf->variables.nelts <= index) { |
499 | 349 ngx_log_error(NGX_LOG_ALERT, r->connection->log, 0, |
350 "unknown variable index: %d", index); | |
351 return NULL; | |
352 } | |
353 | |
573 | 354 if (r->variables[index].not_found || r->variables[index].valid) { |
355 return &r->variables[index]; | |
501 | 356 } |
499 | 357 |
501 | 358 v = cmcf->variables.elts; |
499 | 359 |
637 | 360 if (v[index].get_handler(r, &r->variables[index], v[index].data) |
361 == NGX_OK) | |
362 { | |
573 | 363 if (v[index].flags & NGX_HTTP_VAR_NOCACHABLE) { |
364 r->variables[index].no_cachable = 1; | |
499 | 365 } |
573 | 366 |
367 return &r->variables[index]; | |
499 | 368 } |
369 | |
657 | 370 r->variables[index].valid = 0; |
371 r->variables[index].not_found = 1; | |
372 | |
798 | 373 return NULL; |
573 | 374 } |
375 | |
376 | |
377 ngx_http_variable_value_t * | |
378 ngx_http_get_flushed_variable(ngx_http_request_t *r, ngx_uint_t index) | |
379 { | |
380 ngx_http_variable_value_t *v; | |
381 | |
382 v = &r->variables[index]; | |
383 | |
384 if (v->valid) { | |
385 if (!v->no_cachable) { | |
386 return v; | |
387 } | |
388 | |
389 v->valid = 0; | |
390 v->not_found = 0; | |
509 | 391 } |
501 | 392 |
573 | 393 return ngx_http_get_indexed_variable(r, index); |
499 | 394 } |
395 | |
396 | |
397 ngx_http_variable_value_t * | |
635 | 398 ngx_http_get_variable(ngx_http_request_t *r, ngx_str_t *name, ngx_uint_t key, |
399 ngx_uint_t nowarn) | |
499 | 400 { |
501 | 401 ngx_http_variable_t *v; |
573 | 402 ngx_http_variable_value_t *vv; |
499 | 403 ngx_http_core_main_conf_t *cmcf; |
404 | |
405 cmcf = ngx_http_get_module_main_conf(r, ngx_http_core_module); | |
406 | |
611 | 407 v = ngx_hash_find(&cmcf->variables_hash, key, name->data, name->len); |
499 | 408 |
611 | 409 if (v) { |
410 if (v->flags & NGX_HTTP_VAR_INDEXED) { | |
411 return ngx_http_get_indexed_variable(r, v->index); | |
519 | 412 |
413 } else { | |
499 | 414 |
573 | 415 vv = ngx_palloc(r->pool, sizeof(ngx_http_variable_value_t)); |
501 | 416 |
637 | 417 if (vv && v->get_handler(r, vv, v->data) == NGX_OK) { |
573 | 418 return vv; |
419 } | |
509 | 420 |
573 | 421 return NULL; |
422 } | |
509 | 423 } |
424 | |
425 vv = ngx_palloc(r->pool, sizeof(ngx_http_variable_value_t)); | |
426 if (vv == NULL) { | |
427 return NULL; | |
428 } | |
429 | |
573 | 430 if (ngx_strncmp(name->data, "http_", 5) == 0) { |
499 | 431 |
577 | 432 if (ngx_http_variable_unknown_header_in(r, vv, (uintptr_t) name) |
433 == NGX_OK) | |
434 { | |
435 return vv; | |
436 } | |
437 | |
438 return NULL; | |
439 } | |
440 | |
441 if (ngx_strncmp(name->data, "sent_http_", 10) == 0) { | |
442 | |
443 if (ngx_http_variable_unknown_header_out(r, vv, (uintptr_t) name) | |
444 == NGX_OK) | |
573 | 445 { |
446 return vv; | |
447 } | |
499 | 448 |
449 return NULL; | |
450 } | |
451 | |
635 | 452 vv->not_found = 1; |
573 | 453 |
635 | 454 if (nowarn == 0) { |
455 ngx_log_error(NGX_LOG_ERR, r->connection->log, 0, | |
456 "unknown \"%V\" variable", name); | |
457 } | |
499 | 458 |
501 | 459 return vv; |
499 | 460 } |
461 | |
462 | |
573 | 463 static ngx_int_t |
464 ngx_http_variable_request(ngx_http_request_t *r, ngx_http_variable_value_t *v, | |
465 uintptr_t data) | |
466 { | |
467 ngx_str_t *s; | |
468 | |
469 s = (ngx_str_t *) ((char *) r + data); | |
470 | |
471 if (s->data) { | |
472 v->len = s->len; | |
473 v->valid = 1; | |
474 v->no_cachable = 0; | |
475 v->not_found = 0; | |
476 v->data = s->data; | |
477 | |
478 } else { | |
479 v->not_found = 1; | |
480 } | |
481 | |
482 return NGX_OK; | |
483 } | |
484 | |
485 | |
637 | 486 static void |
487 ngx_http_variable_request_set_size(ngx_http_request_t *r, | |
488 ngx_http_variable_value_t *v, uintptr_t data) | |
489 { | |
490 ssize_t s, *sp; | |
491 ngx_str_t val; | |
492 | |
493 val.len = v->len & 0xffff; | |
494 val.data = v->data; | |
495 | |
496 s = ngx_parse_size(&val); | |
497 | |
498 if (s == NGX_ERROR) { | |
499 ngx_log_error(NGX_LOG_ERR, r->connection->log, 0, | |
500 "invalid size \"%V\"", &val); | |
501 return; | |
502 } | |
503 | |
504 sp = (ssize_t *) ((char *) r + data); | |
505 | |
506 *sp = s; | |
507 | |
508 return; | |
509 } | |
510 | |
511 | |
573 | 512 static ngx_int_t |
513 ngx_http_variable_header(ngx_http_request_t *r, ngx_http_variable_value_t *v, | |
514 uintptr_t data) | |
515 | 515 { |
573 | 516 ngx_table_elt_t *h; |
517 | |
518 h = *(ngx_table_elt_t **) ((char *) r + data); | |
519 | |
520 if (h) { | |
521 v->len = h->value.len; | |
522 v->valid = 1; | |
523 v->no_cachable = 0; | |
524 v->not_found = 0; | |
525 v->data = h->value.data; | |
526 | |
527 } else { | |
528 v->not_found = 1; | |
529 } | |
530 | |
531 return NGX_OK; | |
532 } | |
533 | |
534 | |
535 static ngx_int_t | |
536 ngx_http_variable_headers(ngx_http_request_t *r, ngx_http_variable_value_t *v, | |
537 uintptr_t data) | |
538 { | |
667 | 539 ssize_t len; |
573 | 540 u_char *p; |
667 | 541 ngx_uint_t i, n; |
573 | 542 ngx_array_t *a; |
543 ngx_table_elt_t **h; | |
515 | 544 |
545 a = (ngx_array_t *) ((char *) r + data); | |
546 | |
667 | 547 n = a->nelts; |
548 | |
549 if (n == 0) { | |
573 | 550 v->not_found = 1; |
551 return NGX_OK; | |
515 | 552 } |
553 | |
573 | 554 v->valid = 1; |
555 v->no_cachable = 0; | |
556 v->not_found = 0; | |
515 | 557 |
558 h = a->elts; | |
559 | |
667 | 560 if (n == 1) { |
573 | 561 v->len = (*h)->value.len; |
562 v->data = (*h)->value.data; | |
515 | 563 |
573 | 564 return NGX_OK; |
515 | 565 } |
566 | |
667 | 567 len = - (ssize_t) (sizeof("; ") - 1); |
573 | 568 |
667 | 569 for (i = 0; i < n; i++) { |
573 | 570 len += h[i]->value.len + sizeof("; ") - 1; |
515 | 571 } |
572 | |
573 | 573 p = ngx_palloc(r->pool, len); |
574 if (p == NULL) { | |
575 return NGX_ERROR; | |
576 } | |
577 | |
578 v->len = len; | |
579 v->data = p; | |
515 | 580 |
581 for (i = 0; /* void */ ; i++) { | |
573 | 582 p = ngx_copy(p, h[i]->value.data, h[i]->value.len); |
515 | 583 |
667 | 584 if (i == n - 1) { |
515 | 585 break; |
586 } | |
587 | |
588 *p++ = ';'; *p++ = ' '; | |
589 } | |
590 | |
573 | 591 return NGX_OK; |
515 | 592 } |
593 | |
594 | |
573 | 595 static ngx_int_t |
577 | 596 ngx_http_variable_unknown_header_in(ngx_http_request_t *r, |
597 ngx_http_variable_value_t *v, uintptr_t data) | |
598 { | |
599 return ngx_http_variable_unknown_header(v, (ngx_str_t *) data, | |
600 &r->headers_in.headers.part, | |
601 sizeof("http_") - 1); | |
602 } | |
603 | |
604 | |
605 static ngx_int_t | |
606 ngx_http_variable_unknown_header_out(ngx_http_request_t *r, | |
573 | 607 ngx_http_variable_value_t *v, uintptr_t data) |
499 | 608 { |
577 | 609 return ngx_http_variable_unknown_header(v, (ngx_str_t *) data, |
610 &r->headers_out.headers.part, | |
611 sizeof("sent_http_") - 1); | |
612 } | |
501 | 613 |
577 | 614 |
615 static ngx_int_t | |
616 ngx_http_variable_unknown_header(ngx_http_variable_value_t *v, ngx_str_t *var, | |
617 ngx_list_part_t *part, size_t prefix) | |
618 { | |
573 | 619 u_char ch; |
620 ngx_uint_t i, n; | |
621 ngx_table_elt_t *header; | |
499 | 622 |
623 header = part->elts; | |
624 | |
625 for (i = 0; /* void */ ; i++) { | |
626 | |
627 if (i >= part->nelts) { | |
628 if (part->next == NULL) { | |
629 break; | |
630 } | |
631 | |
632 part = part->next; | |
633 header = part->elts; | |
634 i = 0; | |
635 } | |
636 | |
577 | 637 for (n = 0; n + prefix < var->len && n < header[i].key.len; n++) { |
499 | 638 ch = header[i].key.data[n]; |
639 | |
509 | 640 if (ch >= 'A' && ch <= 'Z') { |
641 ch |= 0x20; | |
499 | 642 |
643 } else if (ch == '-') { | |
644 ch = '_'; | |
645 } | |
646 | |
577 | 647 if (var->data[n + prefix] != ch) { |
499 | 648 break; |
649 } | |
650 } | |
651 | |
742
75d767d32624
nonexistent $sent_http_content has value of $sent_http_content_length or so
Igor Sysoev <igor@sysoev.ru>
parents:
732
diff
changeset
|
652 if (n + prefix == var->len && n == header[i].key.len) { |
573 | 653 v->len = header[i].value.len; |
654 v->valid = 1; | |
655 v->no_cachable = 0; | |
656 v->not_found = 0; | |
657 v->data = header[i].value.data; | |
499 | 658 |
573 | 659 return NGX_OK; |
499 | 660 } |
661 } | |
662 | |
573 | 663 v->not_found = 1; |
664 | |
665 return NGX_OK; | |
509 | 666 } |
667 | |
668 | |
573 | 669 static ngx_int_t |
670 ngx_http_variable_host(ngx_http_request_t *r, ngx_http_variable_value_t *v, | |
671 uintptr_t data) | |
509 | 672 { |
864 | 673 if (r->host_start == NULL) { |
674 | |
675 if (r->headers_in.host) { | |
676 v->len = r->headers_in.host_name_len; | |
677 v->data = r->headers_in.host->value.data; | |
678 | |
679 } else { | |
680 v->len = r->server_name.len; | |
681 v->data = r->server_name.data; | |
682 } | |
509 | 683 |
928
a6fe6bedb9e3
fix segfault when $host is used and
Igor Sysoev <igor@sysoev.ru>
parents:
864
diff
changeset
|
684 } else if (r->host_end) { |
864 | 685 v->len = r->host_end - r->host_start; |
686 v->data = r->host_start; | |
928
a6fe6bedb9e3
fix segfault when $host is used and
Igor Sysoev <igor@sysoev.ru>
parents:
864
diff
changeset
|
687 |
a6fe6bedb9e3
fix segfault when $host is used and
Igor Sysoev <igor@sysoev.ru>
parents:
864
diff
changeset
|
688 } else { |
a6fe6bedb9e3
fix segfault when $host is used and
Igor Sysoev <igor@sysoev.ru>
parents:
864
diff
changeset
|
689 v->not_found = 1; |
a6fe6bedb9e3
fix segfault when $host is used and
Igor Sysoev <igor@sysoev.ru>
parents:
864
diff
changeset
|
690 return NGX_OK; |
509 | 691 } |
692 | |
573 | 693 v->valid = 1; |
694 v->no_cachable = 0; | |
695 v->not_found = 0; | |
509 | 696 |
573 | 697 return NGX_OK; |
499 | 698 } |
699 | |
700 | |
573 | 701 static ngx_int_t |
702 ngx_http_variable_remote_addr(ngx_http_request_t *r, | |
703 ngx_http_variable_value_t *v, uintptr_t data) | |
499 | 704 { |
982 | 705 struct sockaddr_in *sin; |
706 | |
707 if (data == 0) { | |
708 v->len = r->connection->addr_text.len; | |
709 v->valid = 1; | |
710 v->no_cachable = 0; | |
711 v->not_found = 0; | |
712 v->data = r->connection->addr_text.data; | |
713 | |
714 } else { | |
715 | |
716 /* AF_INET only */ | |
717 | |
718 sin = (struct sockaddr_in *) r->connection->sockaddr; | |
719 | |
720 v->len = sizeof(in_addr_t); | |
721 v->valid = 1; | |
722 v->no_cachable = 0; | |
723 v->not_found = 0; | |
724 v->data = (u_char *) &sin->sin_addr.s_addr; | |
725 } | |
499 | 726 |
573 | 727 return NGX_OK; |
499 | 728 } |
729 | |
730 | |
573 | 731 static ngx_int_t |
732 ngx_http_variable_remote_port(ngx_http_request_t *r, | |
733 ngx_http_variable_value_t *v, uintptr_t data) | |
499 | 734 { |
573 | 735 ngx_uint_t port; |
736 struct sockaddr_in *sin; | |
509 | 737 |
573 | 738 v->len = 0; |
739 v->valid = 1; | |
740 v->no_cachable = 0; | |
741 v->not_found = 0; | |
509 | 742 |
573 | 743 v->data = ngx_palloc(r->pool, sizeof("65535") - 1); |
744 if (v->data == NULL) { | |
745 return NGX_ERROR; | |
509 | 746 } |
747 | |
748 /* AF_INET only */ | |
577 | 749 |
509 | 750 if (r->connection->sockaddr->sa_family == AF_INET) { |
751 sin = (struct sockaddr_in *) r->connection->sockaddr; | |
577 | 752 |
509 | 753 port = ntohs(sin->sin_port); |
577 | 754 |
509 | 755 if (port > 0 && port < 65536) { |
573 | 756 v->len = ngx_sprintf(v->data, "%ui", port) - v->data; |
509 | 757 } |
758 } | |
759 | |
573 | 760 return NGX_OK; |
509 | 761 } |
762 | |
763 | |
573 | 764 static ngx_int_t |
765 ngx_http_variable_server_addr(ngx_http_request_t *r, | |
766 ngx_http_variable_value_t *v, uintptr_t data) | |
509 | 767 { |
573 | 768 socklen_t len; |
769 ngx_connection_t *c; | |
770 struct sockaddr_in sin; | |
499 | 771 |
573 | 772 v->data = ngx_palloc(r->pool, INET_ADDRSTRLEN); |
773 if (v->data == NULL) { | |
774 return NGX_ERROR; | |
509 | 775 } |
776 | |
777 c = r->connection; | |
778 | |
779 if (r->in_addr == 0) { | |
780 len = sizeof(struct sockaddr_in); | |
781 if (getsockname(c->fd, (struct sockaddr *) &sin, &len) == -1) { | |
788
ff16489f1eac
downgrade gravity of getsockname() errors
Igor Sysoev <igor@sysoev.ru>
parents:
773
diff
changeset
|
782 ngx_connection_error(c, ngx_socket_errno, "getsockname() failed"); |
573 | 783 return NGX_ERROR; |
509 | 784 } |
785 | |
786 r->in_addr = sin.sin_addr.s_addr; | |
787 } | |
788 | |
573 | 789 v->len = ngx_inet_ntop(c->listening->family, &r->in_addr, |
790 v->data, INET_ADDRSTRLEN); | |
791 v->valid = 1; | |
792 v->no_cachable = 0; | |
793 v->not_found = 0; | |
499 | 794 |
573 | 795 return NGX_OK; |
499 | 796 } |
797 | |
798 | |
573 | 799 static ngx_int_t |
800 ngx_http_variable_server_port(ngx_http_request_t *r, | |
801 ngx_http_variable_value_t *v, uintptr_t data) | |
509 | 802 { |
573 | 803 v->len = r->port_text->len - 1; |
804 v->valid = 1; | |
805 v->no_cachable = 0; | |
806 v->not_found = 0; | |
807 v->data = r->port_text->data + 1; | |
509 | 808 |
573 | 809 return NGX_OK; |
509 | 810 } |
811 | |
812 | |
573 | 813 static ngx_int_t |
731 | 814 ngx_http_variable_scheme(ngx_http_request_t *r, |
815 ngx_http_variable_value_t *v, uintptr_t data) | |
816 { | |
817 #if (NGX_HTTP_SSL) | |
818 | |
819 if (r->connection->ssl) { | |
820 v->len = sizeof("https") - 1; | |
821 v->valid = 1; | |
822 v->no_cachable = 0; | |
823 v->not_found = 0; | |
732 | 824 v->data = (u_char *) "https"; |
731 | 825 |
826 return NGX_OK; | |
827 } | |
828 | |
829 #endif | |
830 | |
831 v->len = sizeof("http") - 1; | |
832 v->valid = 1; | |
833 v->no_cachable = 0; | |
834 v->not_found = 0; | |
732 | 835 v->data = (u_char *) "http"; |
731 | 836 |
837 return NGX_OK; | |
838 } | |
839 | |
840 | |
841 static ngx_int_t | |
573 | 842 ngx_http_variable_document_root(ngx_http_request_t *r, |
843 ngx_http_variable_value_t *v, uintptr_t data) | |
499 | 844 { |
671 | 845 ngx_str_t path; |
573 | 846 ngx_http_core_loc_conf_t *clcf; |
509 | 847 |
848 clcf = ngx_http_get_module_loc_conf(r, ngx_http_core_module); | |
849 | |
671 | 850 if (clcf->root_lengths == NULL) { |
851 v->len = clcf->root.len; | |
852 v->valid = 1; | |
853 v->no_cachable = 0; | |
854 v->not_found = 0; | |
855 v->data = clcf->root.data; | |
856 | |
857 } else { | |
858 if (ngx_http_script_run(r, &path, clcf->root_lengths->elts, 0, | |
859 clcf->root_values->elts) | |
860 == NULL) | |
861 { | |
862 return NGX_ERROR; | |
863 } | |
864 | |
865 if (ngx_conf_full_name((ngx_cycle_t *) ngx_cycle, &path) == NGX_ERROR) { | |
866 return NGX_ERROR; | |
867 } | |
868 | |
869 v->len = path.len; | |
870 v->valid = 1; | |
871 v->no_cachable = 0; | |
872 v->not_found = 0; | |
873 v->data = path.data; | |
874 } | |
509 | 875 |
573 | 876 return NGX_OK; |
509 | 877 } |
878 | |
879 | |
573 | 880 static ngx_int_t |
881 ngx_http_variable_request_filename(ngx_http_request_t *r, | |
882 ngx_http_variable_value_t *v, uintptr_t data) | |
509 | 883 { |
773 | 884 size_t root; |
573 | 885 ngx_str_t path; |
499 | 886 |
773 | 887 if (ngx_http_map_uri_to_path(r, &path, &root, 0) == NULL) { |
573 | 888 return NGX_ERROR; |
557 | 889 } |
509 | 890 |
557 | 891 /* ngx_http_map_uri_to_path() allocates memory for terminating '\0' */ |
509 | 892 |
573 | 893 v->len = path.len - 1; |
894 v->valid = 1; | |
895 v->no_cachable = 0; | |
896 v->not_found = 0; | |
897 v->data = path.data; | |
499 | 898 |
573 | 899 return NGX_OK; |
499 | 900 } |
901 | |
902 | |
573 | 903 static ngx_int_t |
904 ngx_http_variable_request_method(ngx_http_request_t *r, | |
905 ngx_http_variable_value_t *v, uintptr_t data) | |
561 | 906 { |
647 | 907 if (r->main->method_name.data) { |
908 v->len = r->main->method_name.len; | |
573 | 909 v->valid = 1; |
910 v->no_cachable = 0; | |
911 v->not_found = 0; | |
647 | 912 v->data = r->main->method_name.data; |
573 | 913 |
914 } else { | |
915 v->not_found = 1; | |
561 | 916 } |
917 | |
573 | 918 return NGX_OK; |
561 | 919 } |
920 | |
921 | |
573 | 922 static ngx_int_t |
923 ngx_http_variable_remote_user(ngx_http_request_t *r, | |
924 ngx_http_variable_value_t *v, uintptr_t data) | |
539 | 925 { |
573 | 926 ngx_int_t rc; |
539 | 927 |
928 rc = ngx_http_auth_basic_user(r); | |
929 | |
930 if (rc == NGX_DECLINED) { | |
573 | 931 v->not_found = 1; |
932 return NGX_OK; | |
539 | 933 } |
934 | |
935 if (rc == NGX_ERROR) { | |
573 | 936 return NGX_ERROR; |
539 | 937 } |
938 | |
573 | 939 v->len = r->headers_in.user.len; |
940 v->valid = 1; | |
941 v->no_cachable = 0; | |
942 v->not_found = 0; | |
943 v->data = r->headers_in.user.data; | |
571 | 944 |
573 | 945 return NGX_OK; |
571 | 946 } |
947 | |
948 | |
611 | 949 static ngx_int_t |
950 ngx_http_variable_body_bytes_sent(ngx_http_request_t *r, | |
951 ngx_http_variable_value_t *v, uintptr_t data) | |
952 { | |
953 off_t sent; | |
954 u_char *p; | |
955 | |
956 sent = r->connection->sent - r->header_size; | |
957 | |
958 if (sent < 0) { | |
959 sent = 0; | |
960 } | |
961 | |
962 p = ngx_palloc(r->pool, NGX_OFF_T_LEN); | |
963 if (p == NULL) { | |
964 return NGX_ERROR; | |
965 } | |
966 | |
967 v->len = ngx_sprintf(p, "%O", sent) - p; | |
968 v->valid = 1; | |
969 v->no_cachable = 0; | |
970 v->not_found = 0; | |
971 v->data = p; | |
972 | |
973 return NGX_OK; | |
974 } | |
975 | |
976 | |
629 | 977 static ngx_int_t |
641 | 978 ngx_http_variable_sent_content_type(ngx_http_request_t *r, |
979 ngx_http_variable_value_t *v, uintptr_t data) | |
980 { | |
981 if (r->headers_out.content_type.len) { | |
982 v->len = r->headers_out.content_type.len; | |
983 v->valid = 1; | |
984 v->no_cachable = 0; | |
985 v->not_found = 0; | |
986 v->data = r->headers_out.content_type.data; | |
987 | |
988 } else { | |
989 v->not_found = 1; | |
990 } | |
991 | |
992 return NGX_OK; | |
993 } | |
994 | |
995 | |
996 static ngx_int_t | |
997 ngx_http_variable_sent_content_length(ngx_http_request_t *r, | |
998 ngx_http_variable_value_t *v, uintptr_t data) | |
999 { | |
1000 u_char *p; | |
1001 | |
1002 if (r->headers_out.content_length) { | |
1003 v->len = r->headers_out.content_length->value.len; | |
1004 v->valid = 1; | |
1005 v->no_cachable = 0; | |
1006 v->not_found = 0; | |
1007 v->data = r->headers_out.content_length->value.data; | |
1008 | |
1009 return NGX_OK; | |
1010 } | |
1011 | |
1012 if (r->headers_out.content_length_n >= 0) { | |
1013 p = ngx_palloc(r->pool, NGX_OFF_T_LEN); | |
1014 if (p == NULL) { | |
1015 return NGX_ERROR; | |
1016 } | |
1017 | |
1018 v->len = ngx_sprintf(p, "%O", r->headers_out.content_length_n) - p; | |
1019 v->valid = 1; | |
1020 v->no_cachable = 0; | |
1021 v->not_found = 0; | |
1022 v->data = p; | |
1023 | |
1024 return NGX_OK; | |
1025 } | |
1026 | |
1027 v->not_found = 1; | |
1028 | |
1029 return NGX_OK; | |
1030 } | |
1031 | |
1032 | |
1033 static ngx_int_t | |
1034 ngx_http_variable_sent_last_modified(ngx_http_request_t *r, | |
1035 ngx_http_variable_value_t *v, uintptr_t data) | |
1036 { | |
1037 u_char *p; | |
1038 | |
1039 if (r->headers_out.last_modified) { | |
1040 v->len = r->headers_out.last_modified->value.len; | |
1041 v->valid = 1; | |
1042 v->no_cachable = 0; | |
1043 v->not_found = 0; | |
1044 v->data = r->headers_out.last_modified->value.data; | |
1045 | |
1046 return NGX_OK; | |
1047 } | |
1048 | |
1049 if (r->headers_out.last_modified_time >= 0) { | |
1050 p = ngx_palloc(r->pool, | |
1051 sizeof("Last-Modified: Mon, 28 Sep 1970 06:00:00 GMT") - 1); | |
1052 if (p == NULL) { | |
1053 return NGX_ERROR; | |
1054 } | |
1055 | |
1056 v->len = ngx_http_time(p, r->headers_out.last_modified_time) - p; | |
1057 v->valid = 1; | |
1058 v->no_cachable = 0; | |
1059 v->not_found = 0; | |
1060 v->data = p; | |
1061 | |
1062 return NGX_OK; | |
1063 } | |
1064 | |
1065 v->not_found = 1; | |
1066 | |
1067 return NGX_OK; | |
1068 } | |
1069 | |
1070 | |
1071 static ngx_int_t | |
1072 ngx_http_variable_sent_connection(ngx_http_request_t *r, | |
1073 ngx_http_variable_value_t *v, uintptr_t data) | |
1074 { | |
1075 size_t len; | |
1076 char *p; | |
1077 | |
1078 if (r->keepalive) { | |
1079 len = sizeof("keep-alive") - 1; | |
1080 p = "keep-alive"; | |
1081 | |
1082 } else { | |
1083 len = sizeof("close") - 1; | |
1084 p = "close"; | |
1085 } | |
1086 | |
1087 v->len = len; | |
1088 v->valid = 1; | |
1089 v->no_cachable = 0; | |
1090 v->not_found = 0; | |
1091 v->data = (u_char *) p; | |
1092 | |
1093 return NGX_OK; | |
1094 } | |
1095 | |
1096 | |
1097 static ngx_int_t | |
1098 ngx_http_variable_sent_keep_alive(ngx_http_request_t *r, | |
1099 ngx_http_variable_value_t *v, uintptr_t data) | |
1100 { | |
1101 u_char *p; | |
1102 ngx_http_core_loc_conf_t *clcf; | |
1103 | |
1104 if (r->keepalive) { | |
1105 clcf = ngx_http_get_module_loc_conf(r, ngx_http_core_module); | |
1106 | |
1107 if (clcf->keepalive_header) { | |
1108 | |
1109 p = ngx_palloc(r->pool, sizeof("timeout=") - 1 + NGX_TIME_T_LEN); | |
1110 if (p == NULL) { | |
1111 return NGX_ERROR; | |
1112 } | |
1113 | |
1114 v->len = ngx_sprintf(p, "timeout=%T", clcf->keepalive_header) - p; | |
1115 v->valid = 1; | |
1116 v->no_cachable = 0; | |
1117 v->not_found = 0; | |
1118 v->data = p; | |
1119 | |
1120 return NGX_OK; | |
1121 } | |
1122 } | |
1123 | |
1124 v->not_found = 1; | |
1125 | |
1126 return NGX_OK; | |
1127 } | |
1128 | |
1129 | |
1130 static ngx_int_t | |
1131 ngx_http_variable_sent_transfer_encoding(ngx_http_request_t *r, | |
1132 ngx_http_variable_value_t *v, uintptr_t data) | |
1133 { | |
1134 if (r->chunked) { | |
1135 v->len = sizeof("chunked") - 1; | |
1136 v->valid = 1; | |
1137 v->no_cachable = 0; | |
1138 v->not_found = 0; | |
1139 v->data = (u_char *) "chunked"; | |
1140 | |
1141 } else { | |
1142 v->not_found = 1; | |
1143 } | |
1144 | |
1145 return NGX_OK; | |
1146 } | |
1147 | |
1148 | |
1149 static ngx_int_t | |
629 | 1150 ngx_http_variable_request_completion(ngx_http_request_t *r, |
1151 ngx_http_variable_value_t *v, uintptr_t data) | |
1152 { | |
1153 if (r->request_complete) { | |
1154 v->len = 2; | |
1155 v->valid = 1; | |
1156 v->no_cachable = 0; | |
1157 v->not_found = 0; | |
1158 v->data = (u_char *) "OK"; | |
1159 | |
1160 return NGX_OK; | |
1161 } | |
1162 | |
1163 v->len = 0; | |
1164 v->valid = 1; | |
1165 v->no_cachable = 0; | |
1166 v->not_found = 0; | |
1167 v->data = (u_char *) ""; | |
1168 | |
1169 return NGX_OK; | |
1170 } | |
1171 | |
1172 | |
759 | 1173 static ngx_int_t |
1174 ngx_http_variable_request_body_file(ngx_http_request_t *r, | |
1175 ngx_http_variable_value_t *v, uintptr_t data) | |
1176 { | |
1177 if (r->request_body == NULL || r->request_body->temp_file == NULL) { | |
763
0b0f3d4854c0
variable should not be found if no request body file
Igor Sysoev <igor@sysoev.ru>
parents:
759
diff
changeset
|
1178 v->not_found = 1; |
759 | 1179 |
1180 return NGX_OK; | |
1181 } | |
1182 | |
1183 v->len = r->request_body->temp_file->file.name.len; | |
1184 v->valid = 1; | |
1185 v->no_cachable = 0; | |
1186 v->not_found = 0; | |
1187 v->data = r->request_body->temp_file->file.name.data; | |
1188 | |
1189 return NGX_OK; | |
1190 } | |
1191 | |
1192 | |
499 | 1193 ngx_int_t |
509 | 1194 ngx_http_variables_add_core_vars(ngx_conf_t *cf) |
499 | 1195 { |
611 | 1196 ngx_int_t rc; |
1197 ngx_http_variable_t *v; | |
501 | 1198 ngx_http_core_main_conf_t *cmcf; |
499 | 1199 |
509 | 1200 cmcf = ngx_http_conf_get_module_main_conf(cf, ngx_http_core_module); |
1201 | |
611 | 1202 cmcf->variables_keys = ngx_pcalloc(cf->temp_pool, |
1203 sizeof(ngx_hash_keys_arrays_t)); | |
1204 if (cmcf->variables_keys == NULL) { | |
1205 return NGX_ERROR; | |
1206 } | |
1207 | |
1208 cmcf->variables_keys->pool = cf->pool; | |
1209 cmcf->variables_keys->temp_pool = cf->pool; | |
1210 | |
1211 if (ngx_hash_keys_array_init(cmcf->variables_keys, NGX_HASH_SMALL) | |
1212 != NGX_OK) | |
509 | 1213 { |
499 | 1214 return NGX_ERROR; |
1215 } | |
1216 | |
611 | 1217 for (v = ngx_http_core_variables; v->name.len; v++) { |
1218 rc = ngx_hash_add_key(cmcf->variables_keys, &v->name, v, | |
1219 NGX_HASH_READONLY_KEY); | |
1220 | |
1221 if (rc == NGX_OK) { | |
1222 continue; | |
499 | 1223 } |
1224 | |
611 | 1225 if (rc == NGX_BUSY) { |
1226 ngx_conf_log_error(NGX_LOG_EMERG, cf, 0, | |
1227 "conflicting variable name \"%V\"", &v->name); | |
1228 } | |
1229 | |
1230 return NGX_ERROR; | |
499 | 1231 } |
1232 | |
1233 return NGX_OK; | |
1234 } | |
509 | 1235 |
1236 | |
1237 ngx_int_t | |
1238 ngx_http_variables_init_vars(ngx_conf_t *cf) | |
1239 { | |
1240 ngx_uint_t i, n; | |
611 | 1241 ngx_hash_key_t *key; |
1242 ngx_hash_init_t hash; | |
509 | 1243 ngx_http_variable_t *v, *av; |
1244 ngx_http_core_main_conf_t *cmcf; | |
1245 | |
1246 /* set the handlers for the indexed http variables */ | |
1247 | |
1248 cmcf = ngx_http_conf_get_module_main_conf(cf, ngx_http_core_module); | |
1249 | |
1250 v = cmcf->variables.elts; | |
611 | 1251 key = cmcf->variables_keys->keys.elts; |
509 | 1252 |
1253 for (i = 0; i < cmcf->variables.nelts; i++) { | |
1254 | |
611 | 1255 for (n = 0; n < cmcf->variables_keys->keys.nelts; n++) { |
509 | 1256 |
611 | 1257 if (v[i].name.len == key[n].key.len |
1258 && ngx_strncmp(v[i].name.data, key[n].key.data, v[i].name.len) | |
509 | 1259 == 0) |
1260 { | |
611 | 1261 av = key[n].value; |
1262 | |
637 | 1263 v[i].get_handler = av->get_handler; |
611 | 1264 v[i].data = av->data; |
527 | 1265 |
611 | 1266 av->flags |= NGX_HTTP_VAR_INDEXED; |
1267 v[i].flags = av->flags; | |
509 | 1268 |
611 | 1269 av->index = i; |
533 | 1270 |
509 | 1271 goto next; |
1272 } | |
1273 } | |
1274 | |
1275 if (ngx_strncmp(v[i].name.data, "http_", 5) == 0) { | |
637 | 1276 v[i].get_handler = ngx_http_variable_unknown_header_in; |
577 | 1277 v[i].data = (uintptr_t) &v[i].name; |
1278 | |
1279 continue; | |
1280 } | |
1281 | |
1282 if (ngx_strncmp(v[i].name.data, "sent_http_", 10) == 0) { | |
637 | 1283 v[i].get_handler = ngx_http_variable_unknown_header_out; |
509 | 1284 v[i].data = (uintptr_t) &v[i].name; |
1285 | |
1286 continue; | |
1287 } | |
1288 | |
1289 ngx_log_error(NGX_LOG_EMERG, cf->log, 0, | |
1290 "unknown \"%V\" variable", &v[i].name); | |
1291 | |
1292 return NGX_ERROR; | |
1293 | |
1294 next: | |
1295 continue; | |
1296 } | |
1297 | |
1298 | |
611 | 1299 for (n = 0; n < cmcf->variables_keys->keys.nelts; n++) { |
1300 av = key[n].value; | |
1301 | |
1302 if (av->flags & NGX_HTTP_VAR_NOHASH) { | |
1303 key[n].key.data = NULL; | |
583 | 1304 } |
1305 } | |
1306 | |
1307 | |
611 | 1308 hash.hash = &cmcf->variables_hash; |
1309 hash.key = ngx_hash_key; | |
1310 hash.max_size = cmcf->variables_hash_max_size; | |
1311 hash.bucket_size = cmcf->variables_hash_bucket_size; | |
1312 hash.name = "variables_hash"; | |
1313 hash.pool = cf->pool; | |
1314 hash.temp_pool = NULL; | |
509 | 1315 |
611 | 1316 if (ngx_hash_init(&hash, cmcf->variables_keys->keys.elts, |
1317 cmcf->variables_keys->keys.nelts) | |
581 | 1318 != NGX_OK) |
509 | 1319 { |
1320 return NGX_ERROR; | |
1321 } | |
1322 | |
611 | 1323 cmcf->variables_keys = NULL; |
509 | 1324 |
1325 return NGX_OK; | |
1326 } |