?  search:
summary | log | graphiclog1 | graphiclog2 | commit | commitdiff | tree | refs | edit | fork
blame | history | raw | HEAD
nginx 0.8.36
[nginx-catap.git] / src / http / ngx_http_variables.c
blobc8d1c8dcbc637070d51658b9a6cc9a28f4ca8d17
1
2 /*
3 * Copyright (C) Igor Sysoev
4 */
5
6
7 #include <ngx_config.h>
8 #include <ngx_core.h>
9 #include <ngx_http.h>
10 #include <nginx.h>
11
12
13 static ngx_int_t ngx_http_variable_request(ngx_http_request_t *r,
14 ngx_http_variable_value_t *v, uintptr_t data);
15 static void ngx_http_variable_request_set(ngx_http_request_t *r,
16 ngx_http_variable_value_t *v, uintptr_t data);
17 static ngx_int_t ngx_http_variable_request_get_size(ngx_http_request_t *r,
18 ngx_http_variable_value_t *v, uintptr_t data);
19 static void ngx_http_variable_request_set_size(ngx_http_request_t *r,
20 ngx_http_variable_value_t *v, uintptr_t data);
21 static ngx_int_t ngx_http_variable_header(ngx_http_request_t *r,
22 ngx_http_variable_value_t *v, uintptr_t data);
23 static ngx_int_t ngx_http_variable_headers(ngx_http_request_t *r,
24 ngx_http_variable_value_t *v, uintptr_t data);
25
26 static ngx_int_t ngx_http_variable_unknown_header_in(ngx_http_request_t *r,
27 ngx_http_variable_value_t *v, uintptr_t data);
28 static ngx_int_t ngx_http_variable_unknown_header_out(ngx_http_request_t *r,
29 ngx_http_variable_value_t *v, uintptr_t data);
30 static ngx_int_t ngx_http_variable_request_line(ngx_http_request_t *r,
31 ngx_http_variable_value_t *v, uintptr_t data);
32 static ngx_int_t ngx_http_variable_cookie(ngx_http_request_t *r,
33 ngx_http_variable_value_t *v, uintptr_t data);
34 static ngx_int_t ngx_http_variable_argument(ngx_http_request_t *r,
35 ngx_http_variable_value_t *v, uintptr_t data);
36
37 static ngx_int_t ngx_http_variable_host(ngx_http_request_t *r,
38 ngx_http_variable_value_t *v, uintptr_t data);
39 static ngx_int_t ngx_http_variable_binary_remote_addr(ngx_http_request_t *r,
40 ngx_http_variable_value_t *v, uintptr_t data);
41 static ngx_int_t ngx_http_variable_remote_addr(ngx_http_request_t *r,
42 ngx_http_variable_value_t *v, uintptr_t data);
43 static ngx_int_t ngx_http_variable_remote_port(ngx_http_request_t *r,
44 ngx_http_variable_value_t *v, uintptr_t data);
45 static ngx_int_t ngx_http_variable_server_addr(ngx_http_request_t *r,
46 ngx_http_variable_value_t *v, uintptr_t data);
47 static ngx_int_t ngx_http_variable_server_port(ngx_http_request_t *r,
48 ngx_http_variable_value_t *v, uintptr_t data);
49 static ngx_int_t ngx_http_variable_scheme(ngx_http_request_t *r,
50 ngx_http_variable_value_t *v, uintptr_t data);
51 static ngx_int_t ngx_http_variable_is_args(ngx_http_request_t *r,
52 ngx_http_variable_value_t *v, uintptr_t data);
53 static ngx_int_t ngx_http_variable_document_root(ngx_http_request_t *r,
54 ngx_http_variable_value_t *v, uintptr_t data);
55 static ngx_int_t ngx_http_variable_realpath_root(ngx_http_request_t *r,
56 ngx_http_variable_value_t *v, uintptr_t data);
57 static ngx_int_t ngx_http_variable_request_filename(ngx_http_request_t *r,
58 ngx_http_variable_value_t *v, uintptr_t data);
59 static ngx_int_t ngx_http_variable_server_name(ngx_http_request_t *r,
60 ngx_http_variable_value_t *v, uintptr_t data);
61 static ngx_int_t ngx_http_variable_request_method(ngx_http_request_t *r,
62 ngx_http_variable_value_t *v, uintptr_t data);
63 static ngx_int_t ngx_http_variable_remote_user(ngx_http_request_t *r,
64 ngx_http_variable_value_t *v, uintptr_t data);
65 static ngx_int_t ngx_http_variable_body_bytes_sent(ngx_http_request_t *r,
66 ngx_http_variable_value_t *v, uintptr_t data);
67 static ngx_int_t ngx_http_variable_request_completion(ngx_http_request_t *r,
68 ngx_http_variable_value_t *v, uintptr_t data);
69 static ngx_int_t ngx_http_variable_request_body(ngx_http_request_t *r,
70 ngx_http_variable_value_t *v, uintptr_t data);
71 static ngx_int_t ngx_http_variable_request_body_file(ngx_http_request_t *r,
72 ngx_http_variable_value_t *v, uintptr_t data);
73
74 static ngx_int_t ngx_http_variable_sent_content_type(ngx_http_request_t *r,
75 ngx_http_variable_value_t *v, uintptr_t data);
76 static ngx_int_t ngx_http_variable_sent_content_length(ngx_http_request_t *r,
77 ngx_http_variable_value_t *v, uintptr_t data);
78 static ngx_int_t ngx_http_variable_sent_location(ngx_http_request_t *r,
79 ngx_http_variable_value_t *v, uintptr_t data);
80 static ngx_int_t ngx_http_variable_sent_last_modified(ngx_http_request_t *r,
81 ngx_http_variable_value_t *v, uintptr_t data);
82 static ngx_int_t ngx_http_variable_sent_connection(ngx_http_request_t *r,
83 ngx_http_variable_value_t *v, uintptr_t data);
84 static ngx_int_t ngx_http_variable_sent_keep_alive(ngx_http_request_t *r,
85 ngx_http_variable_value_t *v, uintptr_t data);
86 static ngx_int_t ngx_http_variable_sent_transfer_encoding(ngx_http_request_t *r,
87 ngx_http_variable_value_t *v, uintptr_t data);
88
89 static ngx_int_t ngx_http_variable_nginx_version(ngx_http_request_t *r,
90 ngx_http_variable_value_t *v, uintptr_t data);
91 static ngx_int_t ngx_http_variable_hostname(ngx_http_request_t *r,
92 ngx_http_variable_value_t *v, uintptr_t data);
93 static ngx_int_t ngx_http_variable_pid(ngx_http_request_t *r,
94 ngx_http_variable_value_t *v, uintptr_t data);
95
96 /*
97 * TODO:
98 * Apache CGI: AUTH_TYPE, PATH_INFO (null), PATH_TRANSLATED
99 * REMOTE_HOST (null), REMOTE_IDENT (null),
100 * SERVER_SOFTWARE
101 *
102 * Apache SSI: DOCUMENT_NAME, LAST_MODIFIED, USER_NAME (file owner)
103 */
104
105 /*
106 * the $http_host, $http_user_agent, $http_referer, $http_via,
107 * and $http_x_forwarded_for variables may be handled by generic
108 * ngx_http_variable_unknown_header_in(), but for perfomance reasons
109 * they are handled using dedicated entries
110 */
111
112 static ngx_http_variable_t ngx_http_core_variables[] = {
113
114 { ngx_string("http_host"), NULL, ngx_http_variable_header,
115 offsetof(ngx_http_request_t, headers_in.host), 0, 0 },
116
117 { ngx_string("http_user_agent"), NULL, ngx_http_variable_header,
118 offsetof(ngx_http_request_t, headers_in.user_agent), 0, 0 },
119
120 { ngx_string("http_referer"), NULL, ngx_http_variable_header,
121 offsetof(ngx_http_request_t, headers_in.referer), 0, 0 },
122
123 #if (NGX_HTTP_GZIP)
124 { ngx_string("http_via"), NULL, ngx_http_variable_header,
125 offsetof(ngx_http_request_t, headers_in.via), 0, 0 },
126 #endif
127
128 #if (NGX_HTTP_PROXY || NGX_HTTP_REALIP)
129 { ngx_string("http_x_forwarded_for"), NULL, ngx_http_variable_header,
130 offsetof(ngx_http_request_t, headers_in.x_forwarded_for), 0, 0 },
131 #endif
132
133 { ngx_string("http_cookie"), NULL, ngx_http_variable_headers,
134 offsetof(ngx_http_request_t, headers_in.cookies), 0, 0 },
135
136 { ngx_string("content_length"), NULL, ngx_http_variable_header,
137 offsetof(ngx_http_request_t, headers_in.content_length), 0, 0 },
138
139 { ngx_string("content_type"), NULL, ngx_http_variable_header,
140 offsetof(ngx_http_request_t, headers_in.content_type), 0, 0 },
141
142 { ngx_string("host"), NULL, ngx_http_variable_host, 0, 0, 0 },
143
144 { ngx_string("binary_remote_addr"), NULL,
145 ngx_http_variable_binary_remote_addr, 0, 0, 0 },
146
147 { ngx_string("remote_addr"), NULL, ngx_http_variable_remote_addr, 0, 0, 0 },
148
149 { ngx_string("remote_port"), NULL, ngx_http_variable_remote_port, 0, 0, 0 },
150
151 { ngx_string("server_addr"), NULL, ngx_http_variable_server_addr, 0, 0, 0 },
152
153 { ngx_string("server_port"), NULL, ngx_http_variable_server_port, 0, 0, 0 },
154
155 { ngx_string("server_protocol"), NULL, ngx_http_variable_request,
156 offsetof(ngx_http_request_t, http_protocol), 0, 0 },
157
158 { ngx_string("scheme"), NULL, ngx_http_variable_scheme, 0, 0, 0 },
159
160 { ngx_string("request_uri"), NULL, ngx_http_variable_request,
161 offsetof(ngx_http_request_t, unparsed_uri), 0, 0 },
162
163 { ngx_string("uri"), NULL, ngx_http_variable_request,
164 offsetof(ngx_http_request_t, uri),
165 NGX_HTTP_VAR_NOCACHEABLE, 0 },
166
167 { ngx_string("document_uri"), NULL, ngx_http_variable_request,
168 offsetof(ngx_http_request_t, uri),
169 NGX_HTTP_VAR_NOCACHEABLE, 0 },
170
171 { ngx_string("request"), NULL, ngx_http_variable_request_line, 0, 0, 0 },
172
173 { ngx_string("document_root"), NULL,
174 ngx_http_variable_document_root, 0, NGX_HTTP_VAR_NOCACHEABLE, 0 },
175
176 { ngx_string("realpath_root"), NULL,
177 ngx_http_variable_realpath_root, 0, NGX_HTTP_VAR_NOCACHEABLE, 0 },
178
179 { ngx_string("query_string"), NULL, ngx_http_variable_request,
180 offsetof(ngx_http_request_t, args),
181 NGX_HTTP_VAR_NOCACHEABLE, 0 },
182
183 { ngx_string("args"),
184 ngx_http_variable_request_set,
185 ngx_http_variable_request,
186 offsetof(ngx_http_request_t, args),
187 NGX_HTTP_VAR_CHANGEABLE|NGX_HTTP_VAR_NOCACHEABLE, 0 },
188
189 { ngx_string("is_args"), NULL, ngx_http_variable_is_args,
190 0, NGX_HTTP_VAR_NOCACHEABLE, 0 },
191
192 { ngx_string("request_filename"), NULL,
193 ngx_http_variable_request_filename, 0,
194 NGX_HTTP_VAR_NOCACHEABLE, 0 },
195
196 { ngx_string("server_name"), NULL, ngx_http_variable_server_name, 0, 0, 0 },
197
198 { ngx_string("request_method"), NULL,
199 ngx_http_variable_request_method, 0,
200 NGX_HTTP_VAR_NOCACHEABLE, 0 },
201
202 { ngx_string("remote_user"), NULL, ngx_http_variable_remote_user, 0, 0, 0 },
203
204 { ngx_string("body_bytes_sent"), NULL, ngx_http_variable_body_bytes_sent,
205 0, 0, 0 },
206
207 { ngx_string("request_completion"), NULL,
208 ngx_http_variable_request_completion,
209 0, 0, 0 },
210
211 { ngx_string("request_body"), NULL,
212 ngx_http_variable_request_body,
213 0, 0, 0 },
214
215 { ngx_string("request_body_file"), NULL,
216 ngx_http_variable_request_body_file,
217 0, 0, 0 },
218
219 { ngx_string("sent_http_content_type"), NULL,
220 ngx_http_variable_sent_content_type, 0, 0, 0 },
221
222 { ngx_string("sent_http_content_length"), NULL,
223 ngx_http_variable_sent_content_length, 0, 0, 0 },
224
225 { ngx_string("sent_http_location"), NULL,
226 ngx_http_variable_sent_location, 0, 0, 0 },
227
228 { ngx_string("sent_http_last_modified"), NULL,
229 ngx_http_variable_sent_last_modified, 0, 0, 0 },
230
231 { ngx_string("sent_http_connection"), NULL,
232 ngx_http_variable_sent_connection, 0, 0, 0 },
233
234 { ngx_string("sent_http_keep_alive"), NULL,
235 ngx_http_variable_sent_keep_alive, 0, 0, 0 },
236
237 { ngx_string("sent_http_transfer_encoding"), NULL,
238 ngx_http_variable_sent_transfer_encoding, 0, 0, 0 },
239
240 { ngx_string("sent_http_cache_control"), NULL, ngx_http_variable_headers,
241 offsetof(ngx_http_request_t, headers_out.cache_control), 0, 0 },
242
243 { ngx_string("limit_rate"), ngx_http_variable_request_set_size,
244 ngx_http_variable_request_get_size,
245 offsetof(ngx_http_request_t, limit_rate),
246 NGX_HTTP_VAR_CHANGEABLE|NGX_HTTP_VAR_NOCACHEABLE, 0 },
247
248 { ngx_string("nginx_version"), NULL, ngx_http_variable_nginx_version,
249 0, 0, 0 },
250
251 { ngx_string("hostname"), NULL, ngx_http_variable_hostname,
252 0, 0, 0 },
253
254 { ngx_string("pid"), NULL, ngx_http_variable_pid,
255 0, 0, 0 },
256
257 { ngx_null_string, NULL, NULL, 0, 0, 0 }
258 };
259
260
261 ngx_http_variable_value_t ngx_http_variable_null_value =
262 ngx_http_variable("");
263 ngx_http_variable_value_t ngx_http_variable_true_value =
264 ngx_http_variable("1");
265
266
267 ngx_http_variable_t *
268 ngx_http_add_variable(ngx_conf_t *cf, ngx_str_t *name, ngx_uint_t flags)
269 {
270 ngx_int_t rc;
271 ngx_uint_t i;
272 ngx_hash_key_t *key;
273 ngx_http_variable_t *v;
274 ngx_http_core_main_conf_t *cmcf;
275
276 cmcf = ngx_http_conf_get_module_main_conf(cf, ngx_http_core_module);
277
278 key = cmcf->variables_keys->keys.elts;
279 for (i = 0; i < cmcf->variables_keys->keys.nelts; i++) {
280 if (name->len != key[i].key.len
281 || ngx_strncasecmp(name->data, key[i].key.data, name->len) != 0)
282 {
283 continue;
284 }
285
286 v = key[i].value;
287
288 if (!(v->flags & NGX_HTTP_VAR_CHANGEABLE)) {
289 ngx_conf_log_error(NGX_LOG_EMERG, cf, 0,
290 "the duplicate \"%V\" variable", name);
291 return NULL;
292 }
293
294 return v;
295 }
296
297 v = ngx_palloc(cf->pool, sizeof(ngx_http_variable_t));
298 if (v == NULL) {
299 return NULL;
300 }
301
302 v->name.len = name->len;
303 v->name.data = ngx_pnalloc(cf->pool, name->len);
304 if (v->name.data == NULL) {
305 return NULL;
306 }
307
308 ngx_strlow(v->name.data, name->data, name->len);
309
310 v->set_handler = NULL;
311 v->get_handler = NULL;
312 v->data = 0;
313 v->flags = flags;
314 v->index = 0;
315
316 rc = ngx_hash_add_key(cmcf->variables_keys, &v->name, v, 0);
317
318 if (rc == NGX_ERROR) {
319 return NULL;
320 }
321
322 if (rc == NGX_BUSY) {
323 ngx_conf_log_error(NGX_LOG_EMERG, cf, 0,
324 "conflicting variable name \"%V\"", name);
325 return NULL;
326 }
327
328 return v;
329 }
330
331
332 ngx_int_t
333 ngx_http_get_variable_index(ngx_conf_t *cf, ngx_str_t *name)
334 {
335 ngx_uint_t i;
336 ngx_http_variable_t *v;
337 ngx_http_core_main_conf_t *cmcf;
338
339 cmcf = ngx_http_conf_get_module_main_conf(cf, ngx_http_core_module);
340
341 v = cmcf->variables.elts;
342
343 if (v == NULL) {
344 if (ngx_array_init(&cmcf->variables, cf->pool, 4,
345 sizeof(ngx_http_variable_t))
346 != NGX_OK)
347 {
348 return NGX_ERROR;
349 }
350
351 } else {
352 for (i = 0; i < cmcf->variables.nelts; i++) {
353 if (name->len != v[i].name.len
354 || ngx_strncasecmp(name->data, v[i].name.data, name->len) != 0)
355 {
356 continue;
357 }
358
359 return i;
360 }
361 }
362
363 v = ngx_array_push(&cmcf->variables);
364 if (v == NULL) {
365 return NGX_ERROR;
366 }
367
368 v->name.len = name->len;
369 v->name.data = ngx_pnalloc(cf->pool, name->len);
370 if (v->name.data == NULL) {
371 return NGX_ERROR;
372 }
373
374 ngx_strlow(v->name.data, name->data, name->len);
375
376 v->set_handler = NULL;
377 v->get_handler = NULL;
378 v->data = 0;
379 v->flags = 0;
380 v->index = cmcf->variables.nelts - 1;
381
382 return cmcf->variables.nelts - 1;
383 }
384
385
386 ngx_http_variable_value_t *
387 ngx_http_get_indexed_variable(ngx_http_request_t *r, ngx_uint_t index)
388 {
389 ngx_http_variable_t *v;
390 ngx_http_core_main_conf_t *cmcf;
391
392 cmcf = ngx_http_get_module_main_conf(r, ngx_http_core_module);
393
394 if (cmcf->variables.nelts <= index) {
395 ngx_log_error(NGX_LOG_ALERT, r->connection->log, 0,
396 "unknown variable index: %d", index);
397 return NULL;
398 }
399
400 if (r->variables[index].not_found || r->variables[index].valid) {
401 return &r->variables[index];
402 }
403
404 v = cmcf->variables.elts;
405
406 if (v[index].get_handler(r, &r->variables[index], v[index].data)
407 == NGX_OK)
408 {
409 if (v[index].flags & NGX_HTTP_VAR_NOCACHEABLE) {
410 r->variables[index].no_cacheable = 1;
411 }
412
413 return &r->variables[index];
414 }
415
416 r->variables[index].valid = 0;
417 r->variables[index].not_found = 1;
418
419 return NULL;
420 }
421
422
423 ngx_http_variable_value_t *
424 ngx_http_get_flushed_variable(ngx_http_request_t *r, ngx_uint_t index)
425 {
426 ngx_http_variable_value_t *v;
427
428 v = &r->variables[index];
429
430 if (v->valid) {
431 if (!v->no_cacheable) {
432 return v;
433 }
434
435 v->valid = 0;
436 v->not_found = 0;
437 }
438
439 return ngx_http_get_indexed_variable(r, index);
440 }
441
442
443 ngx_http_variable_value_t *
444 ngx_http_get_variable(ngx_http_request_t *r, ngx_str_t *name, ngx_uint_t key)
445 {
446 ngx_http_variable_t *v;
447 ngx_http_variable_value_t *vv;
448 ngx_http_core_main_conf_t *cmcf;
449
450 cmcf = ngx_http_get_module_main_conf(r, ngx_http_core_module);
451
452 v = ngx_hash_find(&cmcf->variables_hash, key, name->data, name->len);
453
454 if (v) {
455 if (v->flags & NGX_HTTP_VAR_INDEXED) {
456 return ngx_http_get_flushed_variable(r, v->index);
457
458 } else {
459
460 vv = ngx_palloc(r->pool, sizeof(ngx_http_variable_value_t));
461
462 if (vv && v->get_handler(r, vv, v->data) == NGX_OK) {
463 return vv;
464 }
465
466 return NULL;
467 }
468 }
469
470 vv = ngx_palloc(r->pool, sizeof(ngx_http_variable_value_t));
471 if (vv == NULL) {
472 return NULL;
473 }
474
475 if (ngx_strncmp(name->data, "http_", 5) == 0) {
476
477 if (ngx_http_variable_unknown_header_in(r, vv, (uintptr_t) name)
478 == NGX_OK)
479 {
480 return vv;
481 }
482
483 return NULL;
484 }
485
486 if (ngx_strncmp(name->data, "sent_http_", 10) == 0) {
487
488 if (ngx_http_variable_unknown_header_out(r, vv, (uintptr_t) name)
489 == NGX_OK)
490 {
491 return vv;
492 }
493
494 return NULL;
495 }
496
497 if (ngx_strncmp(name->data, "upstream_http_", 14) == 0) {
498
499 if (ngx_http_upstream_header_variable(r, vv, (uintptr_t) name)
500 == NGX_OK)
501 {
502 return vv;
503 }
504
505 return NULL;
506 }
507
508 if (ngx_strncmp(name->data, "cookie_", 7) == 0) {
509
510 if (ngx_http_variable_cookie(r, vv, (uintptr_t) name) == NGX_OK) {
511 return vv;
512 }
513
514 return NULL;
515 }
516
517 if (ngx_strncmp(name->data, "arg_", 4) == 0) {
518
519 if (ngx_http_variable_argument(r, vv, (uintptr_t) name) == NGX_OK) {
520 return vv;
521 }
522
523 return NULL;
524 }
525
526 vv->not_found = 1;
527
528 return vv;
529 }
530
531
532 static ngx_int_t
533 ngx_http_variable_request(ngx_http_request_t *r, ngx_http_variable_value_t *v,
534 uintptr_t data)
535 {
536 ngx_str_t *s;
537
538 s = (ngx_str_t *) ((char *) r + data);
539
540 if (s->data) {
541 v->len = s->len;
542 v->valid = 1;
543 v->no_cacheable = 0;
544 v->not_found = 0;
545 v->data = s->data;
546
547 } else {
548 v->not_found = 1;
549 }
550
551 return NGX_OK;
552 }
553
554
555 static void
556 ngx_http_variable_request_set(ngx_http_request_t *r,
557 ngx_http_variable_value_t *v, uintptr_t data)
558 {
559 ngx_str_t *s;
560
561 s = (ngx_str_t *) ((char *) r + data);
562
563 s->len = v->len;
564 s->data = v->data;
565 }
566
567
568 static ngx_int_t
569 ngx_http_variable_request_get_size(ngx_http_request_t *r,
570 ngx_http_variable_value_t *v, uintptr_t data)
571 {
572 size_t *sp;
573
574 sp = (size_t *) ((char *) r + data);
575
576 v->data = ngx_pnalloc(r->pool, NGX_SIZE_T_LEN);
577 if (v->data == NULL) {
578 return NGX_ERROR;
579 }
580
581 v->len = ngx_sprintf(v->data, "%uz", *sp) - v->data;
582 v->valid = 1;
583 v->no_cacheable = 0;
584 v->not_found = 0;
585
586 return NGX_OK;
587 }
588
589
590 static void
591 ngx_http_variable_request_set_size(ngx_http_request_t *r,
592 ngx_http_variable_value_t *v, uintptr_t data)
593 {
594 ssize_t s, *sp;
595 ngx_str_t val;
596
597 val.len = v->len;
598 val.data = v->data;
599
600 s = ngx_parse_size(&val);
601
602 if (s == NGX_ERROR) {
603 ngx_log_error(NGX_LOG_ERR, r->connection->log, 0,
604 "invalid size \"%V\"", &val);
605 return;
606 }
607
608 sp = (ssize_t *) ((char *) r + data);
609
610 *sp = s;
611
612 return;
613 }
614
615
616 static ngx_int_t
617 ngx_http_variable_header(ngx_http_request_t *r, ngx_http_variable_value_t *v,
618 uintptr_t data)
619 {
620 ngx_table_elt_t *h;
621
622 h = *(ngx_table_elt_t **) ((char *) r + data);
623
624 if (h) {
625 v->len = h->value.len;
626 v->valid = 1;
627 v->no_cacheable = 0;
628 v->not_found = 0;
629 v->data = h->value.data;
630
631 } else {
632 v->not_found = 1;
633 }
634
635 return NGX_OK;
636 }
637
638
639 static ngx_int_t
640 ngx_http_variable_headers(ngx_http_request_t *r, ngx_http_variable_value_t *v,
641 uintptr_t data)
642 {
643 ssize_t len;
644 u_char *p;
645 ngx_uint_t i, n;
646 ngx_array_t *a;
647 ngx_table_elt_t **h;
648
649 a = (ngx_array_t *) ((char *) r + data);
650
651 n = a->nelts;
652
653 if (n == 0) {
654 v->not_found = 1;
655 return NGX_OK;
656 }
657
658 v->valid = 1;
659 v->no_cacheable = 0;
660 v->not_found = 0;
661
662 h = a->elts;
663
664 if (n == 1) {
665 v->len = (*h)->value.len;
666 v->data = (*h)->value.data;
667
668 return NGX_OK;
669 }
670
671 len = - (ssize_t) (sizeof("; ") - 1);
672
673 for (i = 0; i < n; i++) {
674 len += h[i]->value.len + sizeof("; ") - 1;
675 }
676
677 p = ngx_pnalloc(r->pool, len);
678 if (p == NULL) {
679 return NGX_ERROR;
680 }
681
682 v->len = len;
683 v->data = p;
684
685 for (i = 0; /* void */ ; i++) {
686 p = ngx_copy(p, h[i]->value.data, h[i]->value.len);
687
688 if (i == n - 1) {
689 break;
690 }
691
692 *p++ = ';'; *p++ = ' ';
693 }
694
695 return NGX_OK;
696 }
697
698
699 static ngx_int_t
700 ngx_http_variable_unknown_header_in(ngx_http_request_t *r,
701 ngx_http_variable_value_t *v, uintptr_t data)
702 {
703 return ngx_http_variable_unknown_header(v, (ngx_str_t *) data,
704 &r->headers_in.headers.part,
705 sizeof("http_") - 1);
706 }
707
708
709 static ngx_int_t
710 ngx_http_variable_unknown_header_out(ngx_http_request_t *r,
711 ngx_http_variable_value_t *v, uintptr_t data)
712 {
713 return ngx_http_variable_unknown_header(v, (ngx_str_t *) data,
714 &r->headers_out.headers.part,
715 sizeof("sent_http_") - 1);
716 }
717
718
719 ngx_int_t
720 ngx_http_variable_unknown_header(ngx_http_variable_value_t *v, ngx_str_t *var,
721 ngx_list_part_t *part, size_t prefix)
722 {
723 u_char ch;
724 ngx_uint_t i, n;
725 ngx_table_elt_t *header;
726
727 header = part->elts;
728
729 for (i = 0; /* void */ ; i++) {
730
731 if (i >= part->nelts) {
732 if (part->next == NULL) {
733 break;
734 }
735
736 part = part->next;
737 header = part->elts;
738 i = 0;
739 }
740
741 for (n = 0; n + prefix < var->len && n < header[i].key.len; n++) {
742 ch = header[i].key.data[n];
743
744 if (ch >= 'A' && ch <= 'Z') {
745 ch |= 0x20;
746
747 } else if (ch == '-') {
748 ch = '_';
749 }
750
751 if (var->data[n + prefix] != ch) {
752 break;
753 }
754 }
755
756 if (n + prefix == var->len && n == header[i].key.len) {
757 v->len = header[i].value.len;
758 v->valid = 1;
759 v->no_cacheable = 0;
760 v->not_found = 0;
761 v->data = header[i].value.data;
762
763 return NGX_OK;
764 }
765 }
766
767 v->not_found = 1;
768
769 return NGX_OK;
770 }
771
772
773 static ngx_int_t
774 ngx_http_variable_request_line(ngx_http_request_t *r,
775 ngx_http_variable_value_t *v, uintptr_t data)
776 {
777 u_char *p, *s;
778
779 s = r->request_line.data;
780
781 if (s == NULL) {
782 s = r->request_start;
783
784 if (s == NULL) {
785 v->not_found = 1;
786 return NGX_OK;
787 }
788
789 for (p = s; p < r->header_in->last; p++) {
790 if (*p == CR || *p == LF) {
791 break;
792 }
793 }
794
795 r->request_line.len = p - s;
796 r->request_line.data = s;
797 }
798
799 v->len = r->request_line.len;
800 v->valid = 1;
801 v->no_cacheable = 0;
802 v->not_found = 0;
803 v->data = s;
804
805 return NGX_OK;
806 }
807
808
809 static ngx_int_t
810 ngx_http_variable_cookie(ngx_http_request_t *r, ngx_http_variable_value_t *v,
811 uintptr_t data)
812 {
813 ngx_str_t *name = (ngx_str_t *) data;
814
815 ngx_str_t cookie, s;
816
817 s.len = name->len - (sizeof("cookie_") - 1);
818 s.data = name->data + sizeof("cookie_") - 1;
819
820 if (ngx_http_parse_multi_header_lines(&r->headers_in.cookies, &s, &cookie)
821 == NGX_DECLINED)
822 {
823 v->not_found = 1;
824 return NGX_OK;
825 }
826
827 v->len = cookie.len;
828 v->valid = 1;
829 v->no_cacheable = 0;
830 v->not_found = 0;
831 v->data = cookie.data;
832
833 return NGX_OK;
834 }
835
836
837 static ngx_int_t
838 ngx_http_variable_argument(ngx_http_request_t *r, ngx_http_variable_value_t *v,
839 uintptr_t data)
840 {
841 ngx_str_t *name = (ngx_str_t *) data;
842
843 u_char *arg;
844 size_t len;
845 ngx_str_t value;
846
847 len = name->len - (sizeof("arg_") - 1);
848 arg = name->data + sizeof("arg_") - 1;
849
850 if (ngx_http_arg(r, arg, len, &value) != NGX_OK) {
851 v->not_found = 1;
852 return NGX_OK;
853 }
854
855 v->data = value.data;
856 v->len = value.len;
857 v->valid = 1;
858 v->no_cacheable = 0;
859 v->not_found = 0;
860
861 return NGX_OK;
862 }
863
864
865 static ngx_int_t
866 ngx_http_variable_host(ngx_http_request_t *r, ngx_http_variable_value_t *v,
867 uintptr_t data)
868 {
869 ngx_http_core_srv_conf_t *cscf;
870
871 if (r->headers_in.server.len) {
872 v->len = r->headers_in.server.len;
873 v->data = r->headers_in.server.data;
874
875 } else {
876 cscf = ngx_http_get_module_srv_conf(r, ngx_http_core_module);
877
878 v->len = cscf->server_name.len;
879 v->data = cscf->server_name.data;
880 }
881
882 v->valid = 1;
883 v->no_cacheable = 0;
884 v->not_found = 0;
885
886 return NGX_OK;
887 }
888
889
890 static ngx_int_t
891 ngx_http_variable_binary_remote_addr(ngx_http_request_t *r,
892 ngx_http_variable_value_t *v, uintptr_t data)
893 {
894 struct sockaddr_in *sin;
895 #if (NGX_HAVE_INET6)
896 struct sockaddr_in6 *sin6;
897 #endif
898
899 switch (r->connection->sockaddr->sa_family) {
900
901 #if (NGX_HAVE_INET6)
902 case AF_INET6:
903 sin6 = (struct sockaddr_in6 *) r->connection->sockaddr;
904
905 v->len = sizeof(struct in6_addr);
906 v->valid = 1;
907 v->no_cacheable = 0;
908 v->not_found = 0;
909 v->data = sin6->sin6_addr.s6_addr;
910
911 break;
912 #endif
913
914 default: /* AF_INET */
915 sin = (struct sockaddr_in *) r->connection->sockaddr;
916
917 v->len = sizeof(in_addr_t);
918 v->valid = 1;
919 v->no_cacheable = 0;
920 v->not_found = 0;
921 v->data = (u_char *) &sin->sin_addr;
922
923 break;
924 }
925
926 return NGX_OK;
927 }
928
929
930 static ngx_int_t
931 ngx_http_variable_remote_addr(ngx_http_request_t *r,
932 ngx_http_variable_value_t *v, uintptr_t data)
933 {
934 v->len = r->connection->addr_text.len;
935 v->valid = 1;
936 v->no_cacheable = 0;
937 v->not_found = 0;
938 v->data = r->connection->addr_text.data;
939
940 return NGX_OK;
941 }
942
943
944 static ngx_int_t
945 ngx_http_variable_remote_port(ngx_http_request_t *r,
946 ngx_http_variable_value_t *v, uintptr_t data)
947 {
948 ngx_uint_t port;
949 struct sockaddr_in *sin;
950 #if (NGX_HAVE_INET6)
951 struct sockaddr_in6 *sin6;
952 #endif
953
954 v->len = 0;
955 v->valid = 1;
956 v->no_cacheable = 0;
957 v->not_found = 0;
958
959 v->data = ngx_pnalloc(r->pool, sizeof("65535") - 1);
960 if (v->data == NULL) {
961 return NGX_ERROR;
962 }
963
964 switch (r->connection->sockaddr->sa_family) {
965
966 #if (NGX_HAVE_INET6)
967 case AF_INET6:
968 sin6 = (struct sockaddr_in6 *) r->connection->sockaddr;
969 port = ntohs(sin6->sin6_port);
970 break;
971 #endif
972
973 default: /* AF_INET */
974 sin = (struct sockaddr_in *) r->connection->sockaddr;
975 port = ntohs(sin->sin_port);
976 break;
977 }
978
979 if (port > 0 && port < 65536) {
980 v->len = ngx_sprintf(v->data, "%ui", port) - v->data;
981 }
982
983 return NGX_OK;
984 }
985
986
987 static ngx_int_t
988 ngx_http_variable_server_addr(ngx_http_request_t *r,
989 ngx_http_variable_value_t *v, uintptr_t data)
990 {
991 ngx_str_t s;
992 u_char addr[NGX_SOCKADDR_STRLEN];
993
994 s.len = NGX_SOCKADDR_STRLEN;
995 s.data = addr;
996
997 if (ngx_connection_local_sockaddr(r->connection, &s, 0) != NGX_OK) {
998 return NGX_ERROR;
999 }
1000
1001 s.data = ngx_pnalloc(r->pool, s.len);
1002 if (s.data == NULL) {
1003 return NGX_ERROR;
1004 }
1005
1006 ngx_memcpy(s.data, addr, s.len);
1007
1008 v->len = s.len;
1009 v->valid = 1;
1010 v->no_cacheable = 0;
1011 v->not_found = 0;
1012 v->data = s.data;
1013
1014 return NGX_OK;
1015 }
1016
1017
1018 static ngx_int_t
1019 ngx_http_variable_server_port(ngx_http_request_t *r,
1020 ngx_http_variable_value_t *v, uintptr_t data)
1021 {
1022 ngx_uint_t port;
1023 struct sockaddr_in *sin;
1024 #if (NGX_HAVE_INET6)
1025 struct sockaddr_in6 *sin6;
1026 #endif
1027
1028 v->len = 0;
1029 v->valid = 1;
1030 v->no_cacheable = 0;
1031 v->not_found = 0;
1032
1033 if (ngx_connection_local_sockaddr(r->connection, NULL, 0) != NGX_OK) {
1034 return NGX_ERROR;
1035 }
1036
1037 v->data = ngx_pnalloc(r->pool, sizeof("65535") - 1);
1038 if (v->data == NULL) {
1039 return NGX_ERROR;
1040 }
1041
1042 switch (r->connection->local_sockaddr->sa_family) {
1043
1044 #if (NGX_HAVE_INET6)
1045 case AF_INET6:
1046 sin6 = (struct sockaddr_in6 *) r->connection->local_sockaddr;
1047 port = ntohs(sin6->sin6_port);
1048 break;
1049 #endif
1050
1051 default: /* AF_INET */
1052 sin = (struct sockaddr_in *) r->connection->local_sockaddr;
1053 port = ntohs(sin->sin_port);
1054 break;
1055 }
1056
1057 if (port > 0 && port < 65536) {
1058 v->len = ngx_sprintf(v->data, "%ui", port) - v->data;
1059 }
1060
1061 return NGX_OK;
1062 }
1063
1064
1065 static ngx_int_t
1066 ngx_http_variable_scheme(ngx_http_request_t *r,
1067 ngx_http_variable_value_t *v, uintptr_t data)
1068 {
1069 #if (NGX_HTTP_SSL)
1070
1071 if (r->connection->ssl) {
1072 v->len = sizeof("https") - 1;
1073 v->valid = 1;
1074 v->no_cacheable = 0;
1075 v->not_found = 0;
1076 v->data = (u_char *) "https";
1077
1078 return NGX_OK;
1079 }
1080
1081 #endif
1082
1083 v->len = sizeof("http") - 1;
1084 v->valid = 1;
1085 v->no_cacheable = 0;
1086 v->not_found = 0;
1087 v->data = (u_char *) "http";
1088
1089 return NGX_OK;
1090 }
1091
1092
1093 static ngx_int_t
1094 ngx_http_variable_is_args(ngx_http_request_t *r,
1095 ngx_http_variable_value_t *v, uintptr_t data)
1096 {
1097 v->valid = 1;
1098 v->no_cacheable = 0;
1099 v->not_found = 0;
1100
1101 if (r->args.len == 0) {
1102 v->len = 0;
1103 v->data = NULL;
1104 return NGX_OK;
1105 }
1106
1107 v->len = 1;
1108 v->data = (u_char *) "?";
1109
1110 return NGX_OK;
1111 }
1112
1113
1114 static ngx_int_t
1115 ngx_http_variable_document_root(ngx_http_request_t *r,
1116 ngx_http_variable_value_t *v, uintptr_t data)
1117 {
1118 ngx_str_t path;
1119 ngx_http_core_loc_conf_t *clcf;
1120
1121 clcf = ngx_http_get_module_loc_conf(r, ngx_http_core_module);
1122
1123 if (clcf->root_lengths == NULL) {
1124 v->len = clcf->root.len;
1125 v->valid = 1;
1126 v->no_cacheable = 0;
1127 v->not_found = 0;
1128 v->data = clcf->root.data;
1129
1130 } else {
1131 if (ngx_http_script_run(r, &path, clcf->root_lengths->elts, 0,
1132 clcf->root_values->elts)
1133 == NULL)
1134 {
1135 return NGX_ERROR;
1136 }
1137
1138 if (ngx_conf_full_name((ngx_cycle_t *) ngx_cycle, &path, 0) != NGX_OK) {
1139 return NGX_ERROR;
1140 }
1141
1142 v->len = path.len;
1143 v->valid = 1;
1144 v->no_cacheable = 0;
1145 v->not_found = 0;
1146 v->data = path.data;
1147 }
1148
1149 return NGX_OK;
1150 }
1151
1152
1153 static ngx_int_t
1154 ngx_http_variable_realpath_root(ngx_http_request_t *r,
1155 ngx_http_variable_value_t *v, uintptr_t data)
1156 {
1157 size_t len;
1158 ngx_str_t path;
1159 ngx_http_core_loc_conf_t *clcf;
1160 u_char real[NGX_MAX_PATH];
1161
1162 clcf = ngx_http_get_module_loc_conf(r, ngx_http_core_module);
1163
1164 if (clcf->root_lengths == NULL) {
1165 path = clcf->root;
1166
1167 } else {
1168 if (ngx_http_script_run(r, &path, clcf->root_lengths->elts, 1,
1169 clcf->root_values->elts)
1170 == NULL)
1171 {
1172 return NGX_ERROR;
1173 }
1174
1175 path.data[path.len - 1] = '\0';
1176
1177 if (ngx_conf_full_name((ngx_cycle_t *) ngx_cycle, &path, 0) != NGX_OK) {
1178 return NGX_ERROR;
1179 }
1180 }
1181
1182 if (ngx_realpath(path.data, real) == NULL) {
1183 ngx_log_error(NGX_LOG_CRIT, r->connection->log, ngx_errno,
1184 ngx_realpath_n " \"%s\" failed", path.data);
1185 return NGX_ERROR;
1186 }
1187
1188 len = ngx_strlen(real);
1189
1190 v->data = ngx_pnalloc(r->pool, len);
1191 if (v->data == NULL) {
1192 return NGX_ERROR;
1193 }
1194
1195 v->len = len;
1196 v->valid = 1;
1197 v->no_cacheable = 0;
1198 v->not_found = 0;
1199
1200 ngx_memcpy(v->data, real, len);
1201
1202 return NGX_OK;
1203 }
1204
1205
1206 static ngx_int_t
1207 ngx_http_variable_request_filename(ngx_http_request_t *r,
1208 ngx_http_variable_value_t *v, uintptr_t data)
1209 {
1210 size_t root;
1211 ngx_str_t path;
1212
1213 if (ngx_http_map_uri_to_path(r, &path, &root, 0) == NULL) {
1214 return NGX_ERROR;
1215 }
1216
1217 /* ngx_http_map_uri_to_path() allocates memory for terminating '\0' */
1218
1219 v->len = path.len - 1;
1220 v->valid = 1;
1221 v->no_cacheable = 0;
1222 v->not_found = 0;
1223 v->data = path.data;
1224
1225 return NGX_OK;
1226 }
1227
1228
1229 static ngx_int_t
1230 ngx_http_variable_server_name(ngx_http_request_t *r,
1231 ngx_http_variable_value_t *v, uintptr_t data)
1232 {
1233 ngx_http_core_srv_conf_t *cscf;
1234
1235 cscf = ngx_http_get_module_srv_conf(r, ngx_http_core_module);
1236
1237 v->len = cscf->server_name.len;
1238 v->valid = 1;
1239 v->no_cacheable = 0;
1240 v->not_found = 0;
1241 v->data = cscf->server_name.data;
1242
1243 return NGX_OK;
1244 }
1245
1246
1247 static ngx_int_t
1248 ngx_http_variable_request_method(ngx_http_request_t *r,
1249 ngx_http_variable_value_t *v, uintptr_t data)
1250 {
1251 if (r->main->method_name.data) {
1252 v->len = r->main->method_name.len;
1253 v->valid = 1;
1254 v->no_cacheable = 0;
1255 v->not_found = 0;
1256 v->data = r->main->method_name.data;
1257
1258 } else {
1259 v->not_found = 1;
1260 }
1261
1262 return NGX_OK;
1263 }
1264
1265
1266 static ngx_int_t
1267 ngx_http_variable_remote_user(ngx_http_request_t *r,
1268 ngx_http_variable_value_t *v, uintptr_t data)
1269 {
1270 ngx_int_t rc;
1271
1272 rc = ngx_http_auth_basic_user(r);
1273
1274 if (rc == NGX_DECLINED) {
1275 v->not_found = 1;
1276 return NGX_OK;
1277 }
1278
1279 if (rc == NGX_ERROR) {
1280 return NGX_ERROR;
1281 }
1282
1283 v->len = r->headers_in.user.len;
1284 v->valid = 1;
1285 v->no_cacheable = 0;
1286 v->not_found = 0;
1287 v->data = r->headers_in.user.data;
1288
1289 return NGX_OK;
1290 }
1291
1292
1293 static ngx_int_t
1294 ngx_http_variable_body_bytes_sent(ngx_http_request_t *r,
1295 ngx_http_variable_value_t *v, uintptr_t data)
1296 {
1297 off_t sent;
1298 u_char *p;
1299
1300 sent = r->connection->sent - r->header_size;
1301
1302 if (sent < 0) {
1303 sent = 0;
1304 }
1305
1306 p = ngx_pnalloc(r->pool, NGX_OFF_T_LEN);
1307 if (p == NULL) {
1308 return NGX_ERROR;
1309 }
1310
1311 v->len = ngx_sprintf(p, "%O", sent) - p;
1312 v->valid = 1;
1313 v->no_cacheable = 0;
1314 v->not_found = 0;
1315 v->data = p;
1316
1317 return NGX_OK;
1318 }
1319
1320
1321 static ngx_int_t
1322 ngx_http_variable_sent_content_type(ngx_http_request_t *r,
1323 ngx_http_variable_value_t *v, uintptr_t data)
1324 {
1325 if (r->headers_out.content_type.len) {
1326 v->len = r->headers_out.content_type.len;
1327 v->valid = 1;
1328 v->no_cacheable = 0;
1329 v->not_found = 0;
1330 v->data = r->headers_out.content_type.data;
1331
1332 } else {
1333 v->not_found = 1;
1334 }
1335
1336 return NGX_OK;
1337 }
1338
1339
1340 static ngx_int_t
1341 ngx_http_variable_sent_content_length(ngx_http_request_t *r,
1342 ngx_http_variable_value_t *v, uintptr_t data)
1343 {
1344 u_char *p;
1345
1346 if (r->headers_out.content_length) {
1347 v->len = r->headers_out.content_length->value.len;
1348 v->valid = 1;
1349 v->no_cacheable = 0;
1350 v->not_found = 0;
1351 v->data = r->headers_out.content_length->value.data;
1352
1353 return NGX_OK;
1354 }
1355
1356 if (r->headers_out.content_length_n >= 0) {
1357 p = ngx_pnalloc(r->pool, NGX_OFF_T_LEN);
1358 if (p == NULL) {
1359 return NGX_ERROR;
1360 }
1361
1362 v->len = ngx_sprintf(p, "%O", r->headers_out.content_length_n) - p;
1363 v->valid = 1;
1364 v->no_cacheable = 0;
1365 v->not_found = 0;
1366 v->data = p;
1367
1368 return NGX_OK;
1369 }
1370
1371 v->not_found = 1;
1372
1373 return NGX_OK;
1374 }
1375
1376
1377 static ngx_int_t
1378 ngx_http_variable_sent_location(ngx_http_request_t *r,
1379 ngx_http_variable_value_t *v, uintptr_t data)
1380 {
1381 ngx_str_t name;
1382
1383 if (r->headers_out.location) {
1384 v->len = r->headers_out.location->value.len;
1385 v->valid = 1;
1386 v->no_cacheable = 0;
1387 v->not_found = 0;
1388 v->data = r->headers_out.location->value.data;
1389
1390 return NGX_OK;
1391 }
1392
1393 name.len = sizeof("sent_http_location") - 1;
1394 name.data = (u_char *) "sent_http_location";
1395
1396 return ngx_http_variable_unknown_header(v, &name,
1397 &r->headers_out.headers.part,
1398 sizeof("sent_http_") - 1);
1399 }
1400
1401
1402 static ngx_int_t
1403 ngx_http_variable_sent_last_modified(ngx_http_request_t *r,
1404 ngx_http_variable_value_t *v, uintptr_t data)
1405 {
1406 u_char *p;
1407
1408 if (r->headers_out.last_modified) {
1409 v->len = r->headers_out.last_modified->value.len;
1410 v->valid = 1;
1411 v->no_cacheable = 0;
1412 v->not_found = 0;
1413 v->data = r->headers_out.last_modified->value.data;
1414
1415 return NGX_OK;
1416 }
1417
1418 if (r->headers_out.last_modified_time >= 0) {
1419 p = ngx_pnalloc(r->pool,
1420 sizeof("Last-Modified: Mon, 28 Sep 1970 06:00:00 GMT") - 1);
1421 if (p == NULL) {
1422 return NGX_ERROR;
1423 }
1424
1425 v->len = ngx_http_time(p, r->headers_out.last_modified_time) - p;
1426 v->valid = 1;
1427 v->no_cacheable = 0;
1428 v->not_found = 0;
1429 v->data = p;
1430
1431 return NGX_OK;
1432 }
1433
1434 v->not_found = 1;
1435
1436 return NGX_OK;
1437 }
1438
1439
1440 static ngx_int_t
1441 ngx_http_variable_sent_connection(ngx_http_request_t *r,
1442 ngx_http_variable_value_t *v, uintptr_t data)
1443 {
1444 size_t len;
1445 char *p;
1446
1447 if (r->keepalive) {
1448 len = sizeof("keep-alive") - 1;
1449 p = "keep-alive";
1450
1451 } else {
1452 len = sizeof("close") - 1;
1453 p = "close";
1454 }
1455
1456 v->len = len;
1457 v->valid = 1;
1458 v->no_cacheable = 0;
1459 v->not_found = 0;
1460 v->data = (u_char *) p;
1461
1462 return NGX_OK;
1463 }
1464
1465
1466 static ngx_int_t
1467 ngx_http_variable_sent_keep_alive(ngx_http_request_t *r,
1468 ngx_http_variable_value_t *v, uintptr_t data)
1469 {
1470 u_char *p;
1471 ngx_http_core_loc_conf_t *clcf;
1472
1473 if (r->keepalive) {
1474 clcf = ngx_http_get_module_loc_conf(r, ngx_http_core_module);
1475
1476 if (clcf->keepalive_header) {
1477
1478 p = ngx_pnalloc(r->pool, sizeof("timeout=") - 1 + NGX_TIME_T_LEN);
1479 if (p == NULL) {
1480 return NGX_ERROR;
1481 }
1482
1483 v->len = ngx_sprintf(p, "timeout=%T", clcf->keepalive_header) - p;
1484 v->valid = 1;
1485 v->no_cacheable = 0;
1486 v->not_found = 0;
1487 v->data = p;
1488
1489 return NGX_OK;
1490 }
1491 }
1492
1493 v->not_found = 1;
1494
1495 return NGX_OK;
1496 }
1497
1498
1499 static ngx_int_t
1500 ngx_http_variable_sent_transfer_encoding(ngx_http_request_t *r,
1501 ngx_http_variable_value_t *v, uintptr_t data)
1502 {
1503 if (r->chunked) {
1504 v->len = sizeof("chunked") - 1;
1505 v->valid = 1;
1506 v->no_cacheable = 0;
1507 v->not_found = 0;
1508 v->data = (u_char *) "chunked";
1509
1510 } else {
1511 v->not_found = 1;
1512 }
1513
1514 return NGX_OK;
1515 }
1516
1517
1518 static ngx_int_t
1519 ngx_http_variable_request_completion(ngx_http_request_t *r,
1520 ngx_http_variable_value_t *v, uintptr_t data)
1521 {
1522 if (r->request_complete) {
1523 v->len = 2;
1524 v->valid = 1;
1525 v->no_cacheable = 0;
1526 v->not_found = 0;
1527 v->data = (u_char *) "OK";
1528
1529 return NGX_OK;
1530 }
1531
1532 v->len = 0;
1533 v->valid = 1;
1534 v->no_cacheable = 0;
1535 v->not_found = 0;
1536 v->data = (u_char *) "";
1537
1538 return NGX_OK;
1539 }
1540
1541
1542 static ngx_int_t
1543 ngx_http_variable_request_body(ngx_http_request_t *r,
1544 ngx_http_variable_value_t *v, uintptr_t data)
1545 {
1546 u_char *p;
1547 size_t len;
1548 ngx_buf_t *buf, *next;
1549 ngx_chain_t *cl;
1550
1551 if (r->request_body == NULL
1552 || r->request_body->bufs == NULL
1553 || r->request_body->temp_file)
1554 {
1555 v->not_found = 1;
1556
1557 return NGX_OK;
1558 }
1559
1560 cl = r->request_body->bufs;
1561 buf = cl->buf;
1562
1563 if (cl->next == NULL) {
1564 v->len = buf->last - buf->pos;
1565 v->valid = 1;
1566 v->no_cacheable = 0;
1567 v->not_found = 0;
1568 v->data = buf->pos;
1569
1570 return NGX_OK;
1571 }
1572
1573 next = cl->next->buf;
1574 len = (buf->last - buf->pos) + (next->last - next->pos);
1575
1576 p = ngx_pnalloc(r->pool, len);
1577 if (p == NULL) {
1578 return NGX_ERROR;
1579 }
1580
1581 v->data = p;
1582
1583 p = ngx_cpymem(p, buf->pos, buf->last - buf->pos);
1584 ngx_memcpy(p, next->pos, next->last - next->pos);
1585
1586 v->len = len;
1587 v->valid = 1;
1588 v->no_cacheable = 0;
1589 v->not_found = 0;
1590
1591 return NGX_OK;
1592 }
1593
1594
1595 static ngx_int_t
1596 ngx_http_variable_request_body_file(ngx_http_request_t *r,
1597 ngx_http_variable_value_t *v, uintptr_t data)
1598 {
1599 if (r->request_body == NULL || r->request_body->temp_file == NULL) {
1600 v->not_found = 1;
1601
1602 return NGX_OK;
1603 }
1604
1605 v->len = r->request_body->temp_file->file.name.len;
1606 v->valid = 1;
1607 v->no_cacheable = 0;
1608 v->not_found = 0;
1609 v->data = r->request_body->temp_file->file.name.data;
1610
1611 return NGX_OK;
1612 }
1613
1614
1615 static ngx_int_t
1616 ngx_http_variable_nginx_version(ngx_http_request_t *r,
1617 ngx_http_variable_value_t *v, uintptr_t data)
1618 {
1619 v->len = sizeof(NGINX_VERSION) - 1;
1620 v->valid = 1;
1621 v->no_cacheable = 0;
1622 v->not_found = 0;
1623 v->data = (u_char *) NGINX_VERSION;
1624
1625 return NGX_OK;
1626 }
1627
1628
1629 static ngx_int_t
1630 ngx_http_variable_hostname(ngx_http_request_t *r,
1631 ngx_http_variable_value_t *v, uintptr_t data)
1632 {
1633 v->len = ngx_cycle->hostname.len;
1634 v->valid = 1;
1635 v->no_cacheable = 0;
1636 v->not_found = 0;
1637 v->data = ngx_cycle->hostname.data;
1638
1639 return NGX_OK;
1640 }
1641
1642
1643 static ngx_int_t
1644 ngx_http_variable_pid(ngx_http_request_t *r,
1645 ngx_http_variable_value_t *v, uintptr_t data)
1646 {
1647 u_char *p;
1648
1649 p = ngx_pnalloc(r->pool, NGX_INT64_LEN);
1650 if (p == NULL) {
1651 return NGX_ERROR;
1652 }
1653
1654 v->len = ngx_sprintf(p, "%P", ngx_pid) - p;
1655 v->valid = 1;
1656 v->no_cacheable = 0;
1657 v->not_found = 0;
1658 v->data = p;
1659
1660 return NGX_OK;
1661 }
1662
1663
1664 #if (NGX_PCRE)
1665
1666 static ngx_int_t
1667 ngx_http_variable_not_found(ngx_http_request_t *r, ngx_http_variable_value_t *v,
1668 uintptr_t data)
1669 {
1670 v->not_found = 1;
1671 return NGX_OK;
1672 }
1673
1674
1675 ngx_http_regex_t *
1676 ngx_http_regex_compile(ngx_conf_t *cf, ngx_regex_compile_t *rc)
1677 {
1678 u_char *p;
1679 size_t size;
1680 ngx_str_t name;
1681 ngx_uint_t i, n;
1682 ngx_http_variable_t *v;
1683 ngx_http_regex_t *re;
1684 ngx_http_regex_variable_t *rv;
1685 ngx_http_core_main_conf_t *cmcf;
1686
1687 rc->pool = cf->pool;
1688
1689 if (ngx_regex_compile(rc) != NGX_OK) {
1690 ngx_conf_log_error(NGX_LOG_EMERG, cf, 0, "%V", &rc->err);
1691 return NULL;
1692 }
1693
1694 re = ngx_pcalloc(cf->pool, sizeof(ngx_http_regex_t));
1695 if (re == NULL) {
1696 return NULL;
1697 }
1698
1699 re->regex = rc->regex;
1700 re->ncaptures = rc->captures;
1701
1702 cmcf = ngx_http_conf_get_module_main_conf(cf, ngx_http_core_module);
1703 cmcf->ncaptures = ngx_max(cmcf->ncaptures, re->ncaptures);
1704
1705 n = (ngx_uint_t) rc->named_captures;
1706
1707 if (n == 0) {
1708 return re;
1709 }
1710
1711 rv = ngx_palloc(rc->pool, n * sizeof(ngx_http_regex_variable_t));
1712 if (rv == NULL) {
1713 return NULL;
1714 }
1715
1716 re->variables = rv;
1717 re->nvariables = n;
1718 re->name = rc->pattern;
1719
1720 size = rc->name_size;
1721 p = rc->names;
1722
1723 for (i = 0; i < n; i++) {
1724 rv[i].capture = 2 * ((p[0] << 8) + p[1]);
1725
1726 name.data = &p[2];
1727 name.len = ngx_strlen(name.data);
1728
1729 v = ngx_http_add_variable(cf, &name, NGX_HTTP_VAR_CHANGEABLE);
1730 if (v == NULL) {
1731 return NULL;
1732 }
1733
1734 rv[i].index = ngx_http_get_variable_index(cf, &name);
1735 if (rv[i].index == NGX_ERROR) {
1736 return NULL;
1737 }
1738
1739 v->get_handler = ngx_http_variable_not_found;
1740
1741 p += size;
1742 }
1743
1744 return re;
1745 }
1746
1747
1748 ngx_int_t
1749 ngx_http_regex_exec(ngx_http_request_t *r, ngx_http_regex_t *re, ngx_str_t *s)
1750 {
1751 ngx_int_t rc, index;
1752 ngx_uint_t i, n, len;
1753 ngx_http_variable_value_t *vv;
1754 ngx_http_core_main_conf_t *cmcf;
1755
1756 cmcf = ngx_http_get_module_main_conf(r, ngx_http_core_module);
1757
1758 if (re->ncaptures) {
1759 len = cmcf->ncaptures;
1760
1761 if (r->captures == NULL) {
1762 r->captures = ngx_palloc(r->pool, len * sizeof(int));
1763 if (r->captures == NULL) {
1764 return NGX_ERROR;
1765 }
1766 }
1767
1768 } else {
1769 len = 0;
1770 }
1771
1772 rc = ngx_regex_exec(re->regex, s, r->captures, len);
1773
1774 if (rc == NGX_REGEX_NO_MATCHED) {
1775 return NGX_DECLINED;
1776 }
1777
1778 if (rc < 0) {
1779 ngx_log_error(NGX_LOG_ALERT, r->connection->log, 0,
1780 ngx_regex_exec_n " failed: %i on \"%V\" using \"%V\"",
1781 rc, s, &re->name);
1782 return NGX_ERROR;
1783 }
1784
1785 for (i = 0; i < re->nvariables; i++) {
1786
1787 n = re->variables[i].capture;
1788 index = re->variables[i].index;
1789 vv = &r->variables[index];
1790
1791 vv->len = r->captures[n + 1] - r->captures[n];
1792 vv->valid = 1;
1793 vv->no_cacheable = 0;
1794 vv->not_found = 0;
1795 vv->data = &s->data[r->captures[n]];
1796
1797 #if (NGX_DEBUG)
1798 {
1799 ngx_http_variable_t *v;
1800
1801 v = cmcf->variables.elts;
1802
1803 ngx_log_debug3(NGX_LOG_DEBUG_HTTP, r->connection->log, 0,
1804 "http regex set $%V to \"%*s\"",
1805 &v[index].name, vv->len, vv->data);
1806 }
1807 #endif
1808 }
1809
1810 r->ncaptures = rc * 2;
1811 r->captures_data = s->data;
1812
1813 return NGX_OK;
1814 }
1815
1816 #endif
1817
1818
1819 ngx_int_t
1820 ngx_http_variables_add_core_vars(ngx_conf_t *cf)
1821 {
1822 ngx_int_t rc;
1823 ngx_http_variable_t *v;
1824 ngx_http_core_main_conf_t *cmcf;
1825
1826 cmcf = ngx_http_conf_get_module_main_conf(cf, ngx_http_core_module);
1827
1828 cmcf->variables_keys = ngx_pcalloc(cf->temp_pool,
1829 sizeof(ngx_hash_keys_arrays_t));
1830 if (cmcf->variables_keys == NULL) {
1831 return NGX_ERROR;
1832 }
1833
1834 cmcf->variables_keys->pool = cf->pool;
1835 cmcf->variables_keys->temp_pool = cf->pool;
1836
1837 if (ngx_hash_keys_array_init(cmcf->variables_keys, NGX_HASH_SMALL)
1838 != NGX_OK)
1839 {
1840 return NGX_ERROR;
1841 }
1842
1843 for (v = ngx_http_core_variables; v->name.len; v++) {
1844 rc = ngx_hash_add_key(cmcf->variables_keys, &v->name, v,
1845 NGX_HASH_READONLY_KEY);
1846
1847 if (rc == NGX_OK) {
1848 continue;
1849 }
1850
1851 if (rc == NGX_BUSY) {
1852 ngx_conf_log_error(NGX_LOG_EMERG, cf, 0,
1853 "conflicting variable name \"%V\"", &v->name);
1854 }
1855
1856 return NGX_ERROR;
1857 }
1858
1859 return NGX_OK;
1860 }
1861
1862
1863 ngx_int_t
1864 ngx_http_variables_init_vars(ngx_conf_t *cf)
1865 {
1866 ngx_uint_t i, n;
1867 ngx_hash_key_t *key;
1868 ngx_hash_init_t hash;
1869 ngx_http_variable_t *v, *av;
1870 ngx_http_core_main_conf_t *cmcf;
1871
1872 /* set the handlers for the indexed http variables */
1873
1874 cmcf = ngx_http_conf_get_module_main_conf(cf, ngx_http_core_module);
1875
1876 v = cmcf->variables.elts;
1877 key = cmcf->variables_keys->keys.elts;
1878
1879 for (i = 0; i < cmcf->variables.nelts; i++) {
1880
1881 for (n = 0; n < cmcf->variables_keys->keys.nelts; n++) {
1882
1883 av = key[n].value;
1884
1885 if (av->get_handler
1886 && v[i].name.len == key[n].key.len
1887 && ngx_strncmp(v[i].name.data, key[n].key.data, v[i].name.len)
1888 == 0)
1889 {
1890 v[i].get_handler = av->get_handler;
1891 v[i].data = av->data;
1892
1893 av->flags |= NGX_HTTP_VAR_INDEXED;
1894 v[i].flags = av->flags;
1895
1896 av->index = i;
1897
1898 goto next;
1899 }
1900 }
1901
1902 if (ngx_strncmp(v[i].name.data, "http_", 5) == 0) {
1903 v[i].get_handler = ngx_http_variable_unknown_header_in;
1904 v[i].data = (uintptr_t) &v[i].name;
1905
1906 continue;
1907 }
1908
1909 if (ngx_strncmp(v[i].name.data, "sent_http_", 10) == 0) {
1910 v[i].get_handler = ngx_http_variable_unknown_header_out;
1911 v[i].data = (uintptr_t) &v[i].name;
1912
1913 continue;
1914 }
1915
1916 if (ngx_strncmp(v[i].name.data, "upstream_http_", 14) == 0) {
1917 v[i].get_handler = ngx_http_upstream_header_variable;
1918 v[i].data = (uintptr_t) &v[i].name;
1919 v[i].flags = NGX_HTTP_VAR_NOCACHEABLE;
1920
1921 continue;
1922 }
1923
1924 if (ngx_strncmp(v[i].name.data, "cookie_", 7) == 0) {
1925 v[i].get_handler = ngx_http_variable_cookie;
1926 v[i].data = (uintptr_t) &v[i].name;
1927
1928 continue;
1929 }
1930
1931 if (ngx_strncmp(v[i].name.data, "arg_", 4) == 0) {
1932 v[i].get_handler = ngx_http_variable_argument;
1933 v[i].data = (uintptr_t) &v[i].name;
1934 v[i].flags = NGX_HTTP_VAR_NOCACHEABLE;
1935
1936 continue;
1937 }
1938
1939 ngx_log_error(NGX_LOG_EMERG, cf->log, 0,
1940 "unknown \"%V\" variable", &v[i].name);
1941
1942 return NGX_ERROR;
1943
1944 next:
1945 continue;
1946 }
1947
1948
1949 for (n = 0; n < cmcf->variables_keys->keys.nelts; n++) {
1950 av = key[n].value;
1951
1952 if (av->flags & NGX_HTTP_VAR_NOHASH) {
1953 key[n].key.data = NULL;
1954 }
1955 }
1956
1957
1958 hash.hash = &cmcf->variables_hash;
1959 hash.key = ngx_hash_key;
1960 hash.max_size = cmcf->variables_hash_max_size;
1961 hash.bucket_size = cmcf->variables_hash_bucket_size;
1962 hash.name = "variables_hash";
1963 hash.pool = cf->pool;
1964 hash.temp_pool = NULL;
1965
1966 if (ngx_hash_init(&hash, cmcf->variables_keys->keys.elts,
1967 cmcf->variables_keys->keys.nelts)
1968 != NGX_OK)
1969 {
1970 return NGX_ERROR;
1971 }
1972
1973 cmcf->variables_keys = NULL;
1974
1975 return NGX_OK;
1976 }
1977
1978
1979 void
1980 ngx_http_variable_value_rbtree_insert(ngx_rbtree_node_t *temp,
1981 ngx_rbtree_node_t *node, ngx_rbtree_node_t *sentinel)
1982 {
1983 ngx_rbtree_node_t **p;
1984 ngx_http_variable_value_node_t *vvn, *vvt;
1985
1986 for ( ;; ) {
1987
1988 vvn = (ngx_http_variable_value_node_t *) node;
1989 vvt = (ngx_http_variable_value_node_t *) temp;
1990
1991 if (node->key != temp->key) {
1992
1993 p = (node->key < temp->key) ? &temp->left : &temp->right;
1994
1995 } else if (vvn->len != vvt->len) {
1996
1997 p = (vvn->len < vvt->len) ? &temp->left : &temp->right;
1998
1999 } else {
2000 p = (ngx_memcmp(vvn->value->data, vvt->value->data, vvn->len) < 0)
2001 ? &temp->left : &temp->right;
2002 }
2003
2004 if (*p == sentinel) {
2005 break;
2006 }
2007
2008 temp = *p;
2009 }
2010
2011 *p = node;
2012 node->parent = temp;
2013 node->left = sentinel;
2014 node->right = sentinel;
2015 ngx_rbt_red(node);
2016 }
2017
2018
2019 ngx_http_variable_value_t *
2020 ngx_http_variable_value_lookup(ngx_rbtree_t *rbtree, ngx_str_t *val,
2021 uint32_t hash)
2022 {
2023 ngx_int_t rc;
2024 ngx_rbtree_node_t *node, *sentinel;
2025 ngx_http_variable_value_node_t *vvn;
2026
2027 node = rbtree->root;
2028 sentinel = rbtree->sentinel;
2029
2030 while (node != sentinel) {
2031
2032 vvn = (ngx_http_variable_value_node_t *) node;
2033
2034 if (hash != node->key) {
2035 node = (hash < node->key) ? node->left : node->right;
2036 continue;
2037 }
2038
2039 if (val->len != vvn->len) {
2040 node = (val->len < vvn->len) ? node->left : node->right;
2041 continue;
2042 }
2043
2044 rc = ngx_memcmp(val->data, vvn->value->data, val->len);
2045
2046 if (rc < 0) {
2047 node = node->left;
2048 continue;
2049 }
2050
2051 if (rc > 0) {
2052 node = node->right;
2053 continue;
2054 }
2055
2056 return vvn->value;
2057 }
2058
2059 return NULL;
2060 }
catap's nginx repository