48
|
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
|
122
|
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 ngx_int_t ngx_http_variable_header(ngx_http_request_t *r,
|
|
16 ngx_http_variable_value_t *v, uintptr_t data);
|
|
17 static ngx_int_t ngx_http_variable_headers(ngx_http_request_t *r,
|
|
18 ngx_http_variable_value_t *v, uintptr_t data);
|
126
|
19
|
|
20 static ngx_int_t ngx_http_variable_unknown_header_in(ngx_http_request_t *r,
|
122
|
21 ngx_http_variable_value_t *v, uintptr_t data);
|
126
|
22 static ngx_int_t ngx_http_variable_unknown_header_out(ngx_http_request_t *r,
|
|
23 ngx_http_variable_value_t *v, uintptr_t data);
|
|
24 static ngx_int_t ngx_http_variable_unknown_header(ngx_http_variable_value_t *v,
|
|
25 ngx_str_t *var, ngx_list_part_t *part, size_t prefix);
|
|
26
|
122
|
27 static ngx_int_t ngx_http_variable_host(ngx_http_request_t *r,
|
|
28 ngx_http_variable_value_t *v, uintptr_t data);
|
|
29 static ngx_int_t ngx_http_variable_remote_addr(ngx_http_request_t *r,
|
|
30 ngx_http_variable_value_t *v, uintptr_t data);
|
|
31 static ngx_int_t ngx_http_variable_remote_port(ngx_http_request_t *r,
|
|
32 ngx_http_variable_value_t *v, uintptr_t data);
|
|
33 static ngx_int_t ngx_http_variable_server_addr(ngx_http_request_t *r,
|
|
34 ngx_http_variable_value_t *v, uintptr_t data);
|
|
35 static ngx_int_t ngx_http_variable_server_port(ngx_http_request_t *r,
|
|
36 ngx_http_variable_value_t *v, uintptr_t data);
|
|
37 static ngx_int_t ngx_http_variable_document_root(ngx_http_request_t *r,
|
|
38 ngx_http_variable_value_t *v, uintptr_t data);
|
|
39 static ngx_int_t ngx_http_variable_request_filename(ngx_http_request_t *r,
|
|
40 ngx_http_variable_value_t *v, uintptr_t data);
|
|
41 static ngx_int_t ngx_http_variable_request_method(ngx_http_request_t *r,
|
|
42 ngx_http_variable_value_t *v, uintptr_t data);
|
|
43 static ngx_int_t ngx_http_variable_remote_user(ngx_http_request_t *r,
|
|
44 ngx_http_variable_value_t *v, uintptr_t data);
|
160
|
45 static ngx_int_t ngx_http_variable_body_bytes_sent(ngx_http_request_t *r,
|
|
46 ngx_http_variable_value_t *v, uintptr_t data);
|
178
|
47 static ngx_int_t ngx_http_variable_request_completion(ngx_http_request_t *r,
|
|
48 ngx_http_variable_value_t *v, uintptr_t data);
|
48
|
49
|
|
50
|
58
|
51 /*
|
|
52 * TODO:
|
|
53 * Apache CGI: AUTH_TYPE, PATH_INFO (null), PATH_TRANSLATED
|
|
54 * REMOTE_HOST (null), REMOTE_IDENT (null),
|
|
55 * SERVER_SOFTWARE
|
|
56 *
|
120
|
57 * Apache SSI: DOCUMENT_NAME, LAST_MODIFIED, USER_NAME (file owner)
|
58
|
58 */
|
48
|
59
|
58
|
60 static ngx_http_variable_t ngx_http_core_variables[] = {
|
|
61
|
|
62 { ngx_string("http_host"), ngx_http_variable_header,
|
82
|
63 offsetof(ngx_http_request_t, headers_in.host), 0, 0 },
|
48
|
64
|
58
|
65 { ngx_string("http_user_agent"), ngx_http_variable_header,
|
82
|
66 offsetof(ngx_http_request_t, headers_in.user_agent), 0, 0 },
|
58
|
67
|
|
68 { ngx_string("http_referer"), ngx_http_variable_header,
|
82
|
69 offsetof(ngx_http_request_t, headers_in.referer), 0, 0 },
|
48
|
70
|
|
71 #if (NGX_HTTP_GZIP)
|
58
|
72 { ngx_string("http_via"), ngx_http_variable_header,
|
82
|
73 offsetof(ngx_http_request_t, headers_in.via), 0, 0 },
|
48
|
74 #endif
|
|
75
|
|
76 #if (NGX_HTTP_PROXY)
|
58
|
77 { ngx_string("http_x_forwarded_for"), ngx_http_variable_header,
|
82
|
78 offsetof(ngx_http_request_t, headers_in.x_forwarded_for), 0, 0 },
|
48
|
79 #endif
|
|
80
|
64
|
81 { ngx_string("http_cookie"), ngx_http_variable_headers,
|
82
|
82 offsetof(ngx_http_request_t, headers_in.cookies), 0, 0 },
|
64
|
83
|
58
|
84 { ngx_string("content_length"), ngx_http_variable_header,
|
82
|
85 offsetof(ngx_http_request_t, headers_in.content_length), 0, 0 },
|
58
|
86
|
|
87 { ngx_string("content_type"), ngx_http_variable_header,
|
82
|
88 offsetof(ngx_http_request_t, headers_in.content_type), 0, 0 },
|
58
|
89
|
82
|
90 { ngx_string("host"), ngx_http_variable_host, 0, 0, 0 },
|
58
|
91
|
82
|
92 { ngx_string("remote_addr"), ngx_http_variable_remote_addr, 0, 0, 0 },
|
48
|
93
|
82
|
94 { ngx_string("remote_port"), ngx_http_variable_remote_port, 0, 0, 0 },
|
58
|
95
|
82
|
96 { ngx_string("server_addr"), ngx_http_variable_server_addr, 0, 0, 0 },
|
58
|
97
|
82
|
98 { ngx_string("server_port"), ngx_http_variable_server_port, 0, 0, 0 },
|
58
|
99
|
|
100 { ngx_string("server_protocol"), ngx_http_variable_request,
|
82
|
101 offsetof(ngx_http_request_t, http_protocol), 0, 0 },
|
58
|
102
|
|
103 { ngx_string("request_uri"), ngx_http_variable_request,
|
82
|
104 offsetof(ngx_http_request_t, unparsed_uri), 0, 0 },
|
48
|
105
|
122
|
106 { ngx_string("uri"), ngx_http_variable_request,
|
|
107 offsetof(ngx_http_request_t, uri),
|
|
108 NGX_HTTP_VAR_NOCACHABLE, 0 },
|
|
109
|
58
|
110 { ngx_string("document_uri"), ngx_http_variable_request,
|
122
|
111 offsetof(ngx_http_request_t, uri),
|
|
112 NGX_HTTP_VAR_NOCACHABLE, 0 },
|
58
|
113
|
118
|
114 { ngx_string("request"), ngx_http_variable_request,
|
|
115 offsetof(ngx_http_request_t, request_line), 0, 0 },
|
|
116
|
82
|
117 { ngx_string("document_root"), ngx_http_variable_document_root, 0, 0, 0 },
|
58
|
118
|
|
119 { ngx_string("query_string"), ngx_http_variable_request,
|
|
120 offsetof(ngx_http_request_t, args),
|
82
|
121 NGX_HTTP_VAR_NOCACHABLE, 0 },
|
48
|
122
|
138
|
123 { ngx_string("args"), ngx_http_variable_request,
|
|
124 offsetof(ngx_http_request_t, args),
|
|
125 NGX_HTTP_VAR_NOCACHABLE, 0 },
|
|
126
|
58
|
127 { ngx_string("request_filename"), ngx_http_variable_request_filename, 0,
|
82
|
128 NGX_HTTP_VAR_NOCACHABLE, 0 },
|
58
|
129
|
|
130 { ngx_string("server_name"), ngx_http_variable_request,
|
82
|
131 offsetof(ngx_http_request_t, server_name), 0, 0 },
|
58
|
132
|
110
|
133 { ngx_string("request_method"), ngx_http_variable_request_method, 0, 0, 0 },
|
58
|
134
|
88
|
135 { ngx_string("remote_user"), ngx_http_variable_remote_user, 0, 0, 0 },
|
58
|
136
|
160
|
137 { ngx_string("body_bytes_sent"), ngx_http_variable_body_bytes_sent,
|
|
138 0, 0, 0 },
|
|
139
|
178
|
140 { ngx_string("request_completion"), ngx_http_variable_request_completion,
|
|
141 0, 0, 0 },
|
|
142
|
82
|
143 { ngx_null_string, NULL, 0, 0, 0 }
|
48
|
144 };
|
|
145
|
|
146
|
126
|
147 ngx_http_variable_value_t ngx_http_variable_null_value =
|
|
148 ngx_http_variable("");
|
|
149 ngx_http_variable_value_t ngx_http_variable_true_value =
|
|
150 ngx_http_variable("1");
|
|
151
|
|
152
|
48
|
153 ngx_http_variable_t *
|
58
|
154 ngx_http_add_variable(ngx_conf_t *cf, ngx_str_t *name, ngx_uint_t flags)
|
|
155 {
|
160
|
156 ngx_int_t rc;
|
58
|
157 ngx_uint_t i;
|
160
|
158 ngx_hash_key_t *key;
|
58
|
159 ngx_http_variable_t *v;
|
|
160 ngx_http_core_main_conf_t *cmcf;
|
|
161
|
|
162 cmcf = ngx_http_conf_get_module_main_conf(cf, ngx_http_core_module);
|
|
163
|
160
|
164 key = cmcf->variables_keys->keys.elts;
|
|
165 for (i = 0; i < cmcf->variables_keys->keys.nelts; i++) {
|
|
166 if (name->len != key[i].key.len
|
|
167 || ngx_strncasecmp(name->data, key[i].key.data, name->len) != 0)
|
58
|
168 {
|
|
169 continue;
|
|
170 }
|
|
171
|
160
|
172 v = key[i].value;
|
|
173
|
|
174 if (!(v->flags & NGX_HTTP_VAR_CHANGABLE)) {
|
58
|
175 ngx_conf_log_error(NGX_LOG_EMERG, cf, 0,
|
|
176 "the duplicate \"%V\" variable", name);
|
|
177 return NULL;
|
|
178 }
|
|
179
|
160
|
180 return v;
|
58
|
181 }
|
|
182
|
160
|
183 v = ngx_palloc(cf->pool, sizeof(ngx_http_variable_t));
|
58
|
184 if (v == NULL) {
|
|
185 return NULL;
|
|
186 }
|
|
187
|
|
188 v->name.len = name->len;
|
|
189 v->name.data = ngx_palloc(cf->pool, name->len);
|
|
190 if (v->name.data == NULL) {
|
|
191 return NULL;
|
|
192 }
|
|
193
|
|
194 for (i = 0; i < name->len; i++) {
|
|
195 v->name.data[i] = ngx_tolower(name->data[i]);
|
|
196 }
|
|
197
|
|
198 v->handler = NULL;
|
|
199 v->data = 0;
|
|
200 v->flags = flags;
|
82
|
201 v->index = 0;
|
58
|
202
|
160
|
203 rc = ngx_hash_add_key(cmcf->variables_keys, &v->name, v, 0);
|
|
204
|
|
205 if (rc == NGX_ERROR) {
|
|
206 return NULL;
|
|
207 }
|
|
208
|
|
209 if (rc == NGX_BUSY) {
|
|
210 ngx_conf_log_error(NGX_LOG_EMERG, cf, 0,
|
|
211 "conflicting variable name \"%V\"", name);
|
|
212 return NULL;
|
|
213 }
|
|
214
|
58
|
215 return v;
|
|
216 }
|
|
217
|
|
218
|
|
219 ngx_int_t
|
|
220 ngx_http_get_variable_index(ngx_conf_t *cf, ngx_str_t *name)
|
48
|
221 {
|
50
|
222 ngx_uint_t i;
|
|
223 ngx_http_variable_t *v;
|
48
|
224 ngx_http_core_main_conf_t *cmcf;
|
|
225
|
|
226 cmcf = ngx_http_conf_get_module_main_conf(cf, ngx_http_core_module);
|
|
227
|
50
|
228 v = cmcf->variables.elts;
|
|
229
|
|
230 if (v == NULL) {
|
48
|
231 if (ngx_array_init(&cmcf->variables, cf->pool, 4,
|
|
232 sizeof(ngx_http_variable_t)) == NGX_ERROR)
|
|
233 {
|
58
|
234 return NGX_ERROR;
|
48
|
235 }
|
50
|
236
|
|
237 } else {
|
|
238 for (i = 0; i < cmcf->variables.nelts; i++) {
|
|
239 if (name->len != v[i].name.len
|
|
240 || ngx_strncasecmp(name->data, v[i].name.data, name->len) != 0)
|
|
241 {
|
|
242 continue;
|
|
243 }
|
|
244
|
58
|
245 return i;
|
50
|
246 }
|
48
|
247 }
|
|
248
|
50
|
249 v = ngx_array_push(&cmcf->variables);
|
|
250 if (v == NULL) {
|
58
|
251 return NGX_ERROR;
|
48
|
252 }
|
|
253
|
50
|
254 v->name.len = name->len;
|
|
255 v->name.data = ngx_palloc(cf->pool, name->len);
|
|
256 if (v->name.data == NULL) {
|
58
|
257 return NGX_ERROR;
|
50
|
258 }
|
48
|
259
|
50
|
260 for (i = 0; i < name->len; i++) {
|
58
|
261 v->name.data[i] = ngx_tolower(name->data[i]);
|
50
|
262 }
|
|
263
|
|
264 v->handler = NULL;
|
|
265 v->data = 0;
|
58
|
266 v->flags = 0;
|
82
|
267 v->index = cmcf->variables.nelts - 1;
|
50
|
268
|
58
|
269 return cmcf->variables.nelts - 1;
|
48
|
270 }
|
|
271
|
|
272
|
|
273 ngx_http_variable_value_t *
|
|
274 ngx_http_get_indexed_variable(ngx_http_request_t *r, ngx_uint_t index)
|
|
275 {
|
50
|
276 ngx_http_variable_t *v;
|
48
|
277 ngx_http_core_main_conf_t *cmcf;
|
|
278
|
|
279 cmcf = ngx_http_get_module_main_conf(r, ngx_http_core_module);
|
|
280
|
102
|
281 if (cmcf->variables.nelts <= index) {
|
48
|
282 ngx_log_error(NGX_LOG_ALERT, r->connection->log, 0,
|
|
283 "unknown variable index: %d", index);
|
|
284 return NULL;
|
|
285 }
|
|
286
|
122
|
287 if (r->variables[index].not_found || r->variables[index].valid) {
|
|
288 return &r->variables[index];
|
50
|
289 }
|
48
|
290
|
50
|
291 v = cmcf->variables.elts;
|
48
|
292
|
122
|
293 if (v[index].handler(r, &r->variables[index], v[index].data) == NGX_OK) {
|
48
|
294
|
122
|
295 if (v[index].flags & NGX_HTTP_VAR_NOCACHABLE) {
|
|
296 r->variables[index].no_cachable = 1;
|
48
|
297 }
|
122
|
298
|
|
299 return &r->variables[index];
|
48
|
300 }
|
|
301
|
122
|
302 return NULL;
|
|
303 }
|
|
304
|
|
305
|
|
306 ngx_http_variable_value_t *
|
|
307 ngx_http_get_flushed_variable(ngx_http_request_t *r, ngx_uint_t index)
|
|
308 {
|
|
309 ngx_http_variable_value_t *v;
|
|
310
|
|
311 v = &r->variables[index];
|
|
312
|
|
313 if (v->valid) {
|
|
314 if (!v->no_cachable) {
|
|
315 return v;
|
|
316 }
|
|
317
|
|
318 v->valid = 0;
|
|
319 v->not_found = 0;
|
58
|
320 }
|
50
|
321
|
122
|
322 return ngx_http_get_indexed_variable(r, index);
|
48
|
323 }
|
|
324
|
|
325
|
|
326 ngx_http_variable_value_t *
|
160
|
327 ngx_http_get_variable(ngx_http_request_t *r, ngx_str_t *name, ngx_uint_t key)
|
48
|
328 {
|
50
|
329 ngx_http_variable_t *v;
|
122
|
330 ngx_http_variable_value_t *vv;
|
48
|
331 ngx_http_core_main_conf_t *cmcf;
|
|
332
|
|
333 cmcf = ngx_http_get_module_main_conf(r, ngx_http_core_module);
|
|
334
|
160
|
335 v = ngx_hash_find(&cmcf->variables_hash, key, name->data, name->len);
|
48
|
336
|
160
|
337 if (v) {
|
|
338 if (v->flags & NGX_HTTP_VAR_INDEXED) {
|
|
339 return ngx_http_get_indexed_variable(r, v->index);
|
68
|
340
|
|
341 } else {
|
48
|
342
|
122
|
343 vv = ngx_palloc(r->pool, sizeof(ngx_http_variable_value_t));
|
50
|
344
|
160
|
345 if (vv && v->handler(r, vv, v->data) == NGX_OK) {
|
122
|
346 return vv;
|
|
347 }
|
58
|
348
|
122
|
349 return NULL;
|
|
350 }
|
58
|
351 }
|
|
352
|
|
353 vv = ngx_palloc(r->pool, sizeof(ngx_http_variable_value_t));
|
|
354 if (vv == NULL) {
|
|
355 return NULL;
|
|
356 }
|
|
357
|
122
|
358 if (ngx_strncmp(name->data, "http_", 5) == 0) {
|
48
|
359
|
126
|
360 if (ngx_http_variable_unknown_header_in(r, vv, (uintptr_t) name)
|
|
361 == NGX_OK)
|
|
362 {
|
|
363 return vv;
|
|
364 }
|
|
365
|
|
366 return NULL;
|
|
367 }
|
|
368
|
|
369 if (ngx_strncmp(name->data, "sent_http_", 10) == 0) {
|
|
370
|
|
371 if (ngx_http_variable_unknown_header_out(r, vv, (uintptr_t) name)
|
|
372 == NGX_OK)
|
122
|
373 {
|
|
374 return vv;
|
|
375 }
|
48
|
376
|
|
377 return NULL;
|
|
378 }
|
|
379
|
122
|
380 ngx_log_error(NGX_LOG_ERR, r->connection->log, 0,
|
|
381 "unknown \"%V\" variable", name);
|
|
382
|
|
383 vv->not_found = 1;
|
48
|
384
|
50
|
385 return vv;
|
48
|
386 }
|
|
387
|
|
388
|
122
|
389 static ngx_int_t
|
|
390 ngx_http_variable_request(ngx_http_request_t *r, ngx_http_variable_value_t *v,
|
|
391 uintptr_t data)
|
|
392 {
|
|
393 ngx_str_t *s;
|
|
394
|
|
395 s = (ngx_str_t *) ((char *) r + data);
|
|
396
|
|
397 if (s->data) {
|
|
398 v->len = s->len;
|
|
399 v->valid = 1;
|
|
400 v->no_cachable = 0;
|
|
401 v->not_found = 0;
|
|
402 v->data = s->data;
|
|
403
|
|
404 } else {
|
|
405 v->not_found = 1;
|
|
406 }
|
|
407
|
|
408 return NGX_OK;
|
|
409 }
|
|
410
|
|
411
|
|
412 static ngx_int_t
|
|
413 ngx_http_variable_header(ngx_http_request_t *r, ngx_http_variable_value_t *v,
|
|
414 uintptr_t data)
|
64
|
415 {
|
122
|
416 ngx_table_elt_t *h;
|
|
417
|
|
418 h = *(ngx_table_elt_t **) ((char *) r + data);
|
|
419
|
|
420 if (h) {
|
|
421 v->len = h->value.len;
|
|
422 v->valid = 1;
|
|
423 v->no_cachable = 0;
|
|
424 v->not_found = 0;
|
|
425 v->data = h->value.data;
|
|
426
|
|
427 } else {
|
|
428 v->not_found = 1;
|
|
429 }
|
|
430
|
|
431 return NGX_OK;
|
|
432 }
|
|
433
|
|
434
|
|
435 static ngx_int_t
|
|
436 ngx_http_variable_headers(ngx_http_request_t *r, ngx_http_variable_value_t *v,
|
|
437 uintptr_t data)
|
|
438 {
|
|
439 size_t len;
|
|
440 u_char *p;
|
|
441 ngx_uint_t i;
|
|
442 ngx_array_t *a;
|
|
443 ngx_table_elt_t **h;
|
64
|
444
|
|
445 a = (ngx_array_t *) ((char *) r + data);
|
|
446
|
|
447 if (a->nelts == 0) {
|
122
|
448 v->not_found = 1;
|
|
449 return NGX_OK;
|
64
|
450 }
|
|
451
|
122
|
452 v->valid = 1;
|
|
453 v->no_cachable = 0;
|
|
454 v->not_found = 0;
|
64
|
455
|
|
456 h = a->elts;
|
|
457
|
|
458 if (a->nelts == 1) {
|
122
|
459 v->len = (*h)->value.len;
|
|
460 v->data = (*h)->value.data;
|
64
|
461
|
122
|
462 return NGX_OK;
|
64
|
463 }
|
|
464
|
122
|
465 len = (size_t) - (ssize_t) (sizeof("; ") - 1);
|
|
466
|
|
467 for (i = 0; i < a->nelts; i++) {
|
|
468 len += h[i]->value.len + sizeof("; ") - 1;
|
64
|
469 }
|
|
470
|
122
|
471 p = ngx_palloc(r->pool, len);
|
|
472 if (p == NULL) {
|
|
473 return NGX_ERROR;
|
|
474 }
|
|
475
|
|
476 v->len = len;
|
|
477 v->data = p;
|
64
|
478
|
|
479 for (i = 0; /* void */ ; i++) {
|
122
|
480 p = ngx_copy(p, h[i]->value.data, h[i]->value.len);
|
64
|
481
|
|
482 if (i == a->nelts - 1) {
|
|
483 break;
|
|
484 }
|
|
485
|
|
486 *p++ = ';'; *p++ = ' ';
|
|
487 }
|
|
488
|
122
|
489 return NGX_OK;
|
64
|
490 }
|
|
491
|
|
492
|
122
|
493 static ngx_int_t
|
126
|
494 ngx_http_variable_unknown_header_in(ngx_http_request_t *r,
|
|
495 ngx_http_variable_value_t *v, uintptr_t data)
|
|
496 {
|
|
497 return ngx_http_variable_unknown_header(v, (ngx_str_t *) data,
|
|
498 &r->headers_in.headers.part,
|
|
499 sizeof("http_") - 1);
|
|
500 }
|
|
501
|
|
502
|
|
503 static ngx_int_t
|
|
504 ngx_http_variable_unknown_header_out(ngx_http_request_t *r,
|
122
|
505 ngx_http_variable_value_t *v, uintptr_t data)
|
48
|
506 {
|
126
|
507 return ngx_http_variable_unknown_header(v, (ngx_str_t *) data,
|
|
508 &r->headers_out.headers.part,
|
|
509 sizeof("sent_http_") - 1);
|
|
510 }
|
50
|
511
|
126
|
512
|
|
513 static ngx_int_t
|
|
514 ngx_http_variable_unknown_header(ngx_http_variable_value_t *v, ngx_str_t *var,
|
|
515 ngx_list_part_t *part, size_t prefix)
|
|
516 {
|
122
|
517 u_char ch;
|
|
518 ngx_uint_t i, n;
|
|
519 ngx_table_elt_t *header;
|
48
|
520
|
|
521 header = part->elts;
|
|
522
|
|
523 for (i = 0; /* void */ ; i++) {
|
|
524
|
|
525 if (i >= part->nelts) {
|
|
526 if (part->next == NULL) {
|
|
527 break;
|
|
528 }
|
|
529
|
|
530 part = part->next;
|
|
531 header = part->elts;
|
|
532 i = 0;
|
|
533 }
|
|
534
|
126
|
535 for (n = 0; n + prefix < var->len && n < header[i].key.len; n++) {
|
48
|
536 ch = header[i].key.data[n];
|
|
537
|
58
|
538 if (ch >= 'A' && ch <= 'Z') {
|
|
539 ch |= 0x20;
|
48
|
540
|
|
541 } else if (ch == '-') {
|
|
542 ch = '_';
|
|
543 }
|
|
544
|
126
|
545 if (var->data[n + prefix] != ch) {
|
48
|
546 break;
|
|
547 }
|
|
548 }
|
|
549
|
126
|
550 if (n + prefix == var->len) {
|
122
|
551 v->len = header[i].value.len;
|
|
552 v->valid = 1;
|
|
553 v->no_cachable = 0;
|
|
554 v->not_found = 0;
|
|
555 v->data = header[i].value.data;
|
48
|
556
|
122
|
557 return NGX_OK;
|
48
|
558 }
|
|
559 }
|
|
560
|
122
|
561 v->not_found = 1;
|
|
562
|
|
563 return NGX_OK;
|
58
|
564 }
|
|
565
|
|
566
|
122
|
567 static ngx_int_t
|
|
568 ngx_http_variable_host(ngx_http_request_t *r, ngx_http_variable_value_t *v,
|
|
569 uintptr_t data)
|
58
|
570 {
|
122
|
571 if (r->headers_in.host) {
|
|
572 v->len = r->headers_in.host_name_len;
|
|
573 v->data = r->headers_in.host->value.data;
|
58
|
574
|
122
|
575 } else {
|
|
576 v->len = r->server_name.len;
|
|
577 v->data = r->server_name.data;
|
58
|
578 }
|
|
579
|
122
|
580 v->valid = 1;
|
|
581 v->no_cachable = 0;
|
|
582 v->not_found = 0;
|
58
|
583
|
122
|
584 return NGX_OK;
|
48
|
585 }
|
|
586
|
|
587
|
122
|
588 static ngx_int_t
|
|
589 ngx_http_variable_remote_addr(ngx_http_request_t *r,
|
|
590 ngx_http_variable_value_t *v, uintptr_t data)
|
48
|
591 {
|
122
|
592 v->len = r->connection->addr_text.len;
|
|
593 v->valid = 1;
|
|
594 v->no_cachable = 0;
|
|
595 v->not_found = 0;
|
|
596 v->data = r->connection->addr_text.data;
|
48
|
597
|
122
|
598 return NGX_OK;
|
48
|
599 }
|
|
600
|
|
601
|
122
|
602 static ngx_int_t
|
|
603 ngx_http_variable_remote_port(ngx_http_request_t *r,
|
|
604 ngx_http_variable_value_t *v, uintptr_t data)
|
48
|
605 {
|
122
|
606 ngx_uint_t port;
|
|
607 struct sockaddr_in *sin;
|
58
|
608
|
122
|
609 v->len = 0;
|
|
610 v->valid = 1;
|
|
611 v->no_cachable = 0;
|
|
612 v->not_found = 0;
|
58
|
613
|
122
|
614 v->data = ngx_palloc(r->pool, sizeof("65535") - 1);
|
|
615 if (v->data == NULL) {
|
|
616 return NGX_ERROR;
|
58
|
617 }
|
|
618
|
|
619 /* AF_INET only */
|
126
|
620
|
58
|
621 if (r->connection->sockaddr->sa_family == AF_INET) {
|
|
622 sin = (struct sockaddr_in *) r->connection->sockaddr;
|
126
|
623
|
58
|
624 port = ntohs(sin->sin_port);
|
126
|
625
|
58
|
626 if (port > 0 && port < 65536) {
|
122
|
627 v->len = ngx_sprintf(v->data, "%ui", port) - v->data;
|
58
|
628 }
|
|
629 }
|
|
630
|
122
|
631 return NGX_OK;
|
58
|
632 }
|
|
633
|
|
634
|
122
|
635 static ngx_int_t
|
|
636 ngx_http_variable_server_addr(ngx_http_request_t *r,
|
|
637 ngx_http_variable_value_t *v, uintptr_t data)
|
58
|
638 {
|
122
|
639 socklen_t len;
|
|
640 ngx_connection_t *c;
|
|
641 struct sockaddr_in sin;
|
48
|
642
|
122
|
643 v->data = ngx_palloc(r->pool, INET_ADDRSTRLEN);
|
|
644 if (v->data == NULL) {
|
|
645 return NGX_ERROR;
|
58
|
646 }
|
|
647
|
|
648 c = r->connection;
|
|
649
|
|
650 if (r->in_addr == 0) {
|
|
651 len = sizeof(struct sockaddr_in);
|
|
652 if (getsockname(c->fd, (struct sockaddr *) &sin, &len) == -1) {
|
|
653 ngx_log_error(NGX_LOG_CRIT, c->log,
|
|
654 ngx_socket_errno, "getsockname() failed");
|
122
|
655 return NGX_ERROR;
|
58
|
656 }
|
|
657
|
|
658 r->in_addr = sin.sin_addr.s_addr;
|
|
659 }
|
|
660
|
122
|
661 v->len = ngx_inet_ntop(c->listening->family, &r->in_addr,
|
|
662 v->data, INET_ADDRSTRLEN);
|
|
663 v->valid = 1;
|
|
664 v->no_cachable = 0;
|
|
665 v->not_found = 0;
|
48
|
666
|
122
|
667 return NGX_OK;
|
48
|
668 }
|
|
669
|
|
670
|
122
|
671 static ngx_int_t
|
|
672 ngx_http_variable_server_port(ngx_http_request_t *r,
|
|
673 ngx_http_variable_value_t *v, uintptr_t data)
|
58
|
674 {
|
122
|
675 v->len = r->port_text->len - 1;
|
|
676 v->valid = 1;
|
|
677 v->no_cachable = 0;
|
|
678 v->not_found = 0;
|
|
679 v->data = r->port_text->data + 1;
|
58
|
680
|
122
|
681 return NGX_OK;
|
58
|
682 }
|
|
683
|
|
684
|
122
|
685 static ngx_int_t
|
|
686 ngx_http_variable_document_root(ngx_http_request_t *r,
|
|
687 ngx_http_variable_value_t *v, uintptr_t data)
|
48
|
688 {
|
122
|
689 ngx_http_core_loc_conf_t *clcf;
|
58
|
690
|
|
691 clcf = ngx_http_get_module_loc_conf(r, ngx_http_core_module);
|
|
692
|
122
|
693 v->len = clcf->root.len;
|
|
694 v->valid = 1;
|
|
695 v->no_cachable = 0;
|
|
696 v->not_found = 0;
|
|
697 v->data = clcf->root.data;
|
58
|
698
|
122
|
699 return NGX_OK;
|
58
|
700 }
|
|
701
|
|
702
|
122
|
703 static ngx_int_t
|
|
704 ngx_http_variable_request_filename(ngx_http_request_t *r,
|
|
705 ngx_http_variable_value_t *v, uintptr_t data)
|
58
|
706 {
|
122
|
707 ngx_str_t path;
|
48
|
708
|
122
|
709 if (ngx_http_map_uri_to_path(r, &path, 0) == NULL) {
|
|
710 return NGX_ERROR;
|
106
|
711 }
|
58
|
712
|
106
|
713 /* ngx_http_map_uri_to_path() allocates memory for terminating '\0' */
|
58
|
714
|
122
|
715 v->len = path.len - 1;
|
|
716 v->valid = 1;
|
|
717 v->no_cachable = 0;
|
|
718 v->not_found = 0;
|
|
719 v->data = path.data;
|
48
|
720
|
122
|
721 return NGX_OK;
|
48
|
722 }
|
|
723
|
|
724
|
122
|
725 static ngx_int_t
|
|
726 ngx_http_variable_request_method(ngx_http_request_t *r,
|
|
727 ngx_http_variable_value_t *v, uintptr_t data)
|
110
|
728 {
|
122
|
729 if (r->method_name.data) {
|
|
730 if (r->upstream && r->upstream->method.len) {
|
|
731 v->len = r->upstream->method.len;
|
|
732 v->data = r->upstream->method.data;
|
110
|
733
|
122
|
734 } else {
|
|
735 v->len = r->method_name.len;
|
|
736 v->data = r->method_name.data;
|
|
737 }
|
|
738
|
|
739 v->valid = 1;
|
|
740 v->no_cachable = 0;
|
|
741 v->not_found = 0;
|
|
742
|
|
743 } else {
|
|
744 v->not_found = 1;
|
110
|
745 }
|
|
746
|
122
|
747 return NGX_OK;
|
110
|
748 }
|
|
749
|
|
750
|
122
|
751 static ngx_int_t
|
|
752 ngx_http_variable_remote_user(ngx_http_request_t *r,
|
|
753 ngx_http_variable_value_t *v, uintptr_t data)
|
88
|
754 {
|
122
|
755 ngx_int_t rc;
|
88
|
756
|
|
757 rc = ngx_http_auth_basic_user(r);
|
|
758
|
|
759 if (rc == NGX_DECLINED) {
|
122
|
760 v->not_found = 1;
|
|
761 return NGX_OK;
|
88
|
762 }
|
|
763
|
|
764 if (rc == NGX_ERROR) {
|
122
|
765 return NGX_ERROR;
|
88
|
766 }
|
|
767
|
122
|
768 v->len = r->headers_in.user.len;
|
|
769 v->valid = 1;
|
|
770 v->no_cachable = 0;
|
|
771 v->not_found = 0;
|
|
772 v->data = r->headers_in.user.data;
|
120
|
773
|
122
|
774 return NGX_OK;
|
120
|
775 }
|
|
776
|
|
777
|
160
|
778 static ngx_int_t
|
|
779 ngx_http_variable_body_bytes_sent(ngx_http_request_t *r,
|
|
780 ngx_http_variable_value_t *v, uintptr_t data)
|
|
781 {
|
|
782 off_t sent;
|
|
783 u_char *p;
|
|
784
|
|
785 sent = r->connection->sent - r->header_size;
|
|
786
|
|
787 if (sent < 0) {
|
|
788 sent = 0;
|
|
789 }
|
|
790
|
|
791 p = ngx_palloc(r->pool, NGX_OFF_T_LEN);
|
|
792 if (p == NULL) {
|
|
793 return NGX_ERROR;
|
|
794 }
|
|
795
|
|
796 v->len = ngx_sprintf(p, "%O", sent) - p;
|
|
797 v->valid = 1;
|
|
798 v->no_cachable = 0;
|
|
799 v->not_found = 0;
|
|
800 v->data = p;
|
|
801
|
|
802 return NGX_OK;
|
|
803 }
|
|
804
|
|
805
|
178
|
806 static ngx_int_t
|
|
807 ngx_http_variable_request_completion(ngx_http_request_t *r,
|
|
808 ngx_http_variable_value_t *v, uintptr_t data)
|
|
809 {
|
|
810 if (r->request_complete) {
|
|
811 v->len = 2;
|
|
812 v->valid = 1;
|
|
813 v->no_cachable = 0;
|
|
814 v->not_found = 0;
|
|
815 v->data = (u_char *) "OK";
|
|
816
|
|
817 return NGX_OK;
|
|
818 }
|
|
819
|
|
820 v->len = 0;
|
|
821 v->valid = 1;
|
|
822 v->no_cachable = 0;
|
|
823 v->not_found = 0;
|
|
824 v->data = (u_char *) "";
|
|
825
|
|
826 return NGX_OK;
|
|
827 }
|
|
828
|
|
829
|
48
|
830 ngx_int_t
|
58
|
831 ngx_http_variables_add_core_vars(ngx_conf_t *cf)
|
48
|
832 {
|
160
|
833 ngx_int_t rc;
|
|
834 ngx_http_variable_t *v;
|
50
|
835 ngx_http_core_main_conf_t *cmcf;
|
48
|
836
|
58
|
837 cmcf = ngx_http_conf_get_module_main_conf(cf, ngx_http_core_module);
|
|
838
|
160
|
839 cmcf->variables_keys = ngx_pcalloc(cf->temp_pool,
|
|
840 sizeof(ngx_hash_keys_arrays_t));
|
|
841 if (cmcf->variables_keys == NULL) {
|
|
842 return NGX_ERROR;
|
|
843 }
|
|
844
|
|
845 cmcf->variables_keys->pool = cf->pool;
|
|
846 cmcf->variables_keys->temp_pool = cf->pool;
|
|
847
|
|
848 if (ngx_hash_keys_array_init(cmcf->variables_keys, NGX_HASH_SMALL)
|
|
849 != NGX_OK)
|
58
|
850 {
|
48
|
851 return NGX_ERROR;
|
|
852 }
|
|
853
|
160
|
854 for (v = ngx_http_core_variables; v->name.len; v++) {
|
|
855 rc = ngx_hash_add_key(cmcf->variables_keys, &v->name, v,
|
|
856 NGX_HASH_READONLY_KEY);
|
|
857
|
|
858 if (rc == NGX_OK) {
|
|
859 continue;
|
48
|
860 }
|
|
861
|
160
|
862 if (rc == NGX_BUSY) {
|
|
863 ngx_conf_log_error(NGX_LOG_EMERG, cf, 0,
|
|
864 "conflicting variable name \"%V\"", &v->name);
|
|
865 }
|
|
866
|
|
867 return NGX_ERROR;
|
48
|
868 }
|
|
869
|
|
870 return NGX_OK;
|
|
871 }
|
58
|
872
|
|
873
|
|
874 ngx_int_t
|
|
875 ngx_http_variables_init_vars(ngx_conf_t *cf)
|
|
876 {
|
|
877 ngx_uint_t i, n;
|
160
|
878 ngx_hash_key_t *key;
|
|
879 ngx_hash_init_t hash;
|
58
|
880 ngx_http_variable_t *v, *av;
|
|
881 ngx_http_core_main_conf_t *cmcf;
|
|
882
|
|
883 /* set the handlers for the indexed http variables */
|
|
884
|
|
885 cmcf = ngx_http_conf_get_module_main_conf(cf, ngx_http_core_module);
|
|
886
|
|
887 v = cmcf->variables.elts;
|
160
|
888 key = cmcf->variables_keys->keys.elts;
|
58
|
889
|
|
890 for (i = 0; i < cmcf->variables.nelts; i++) {
|
|
891
|
160
|
892 for (n = 0; n < cmcf->variables_keys->keys.nelts; n++) {
|
58
|
893
|
160
|
894 if (v[i].name.len == key[n].key.len
|
|
895 && ngx_strncmp(v[i].name.data, key[n].key.data, v[i].name.len)
|
58
|
896 == 0)
|
|
897 {
|
160
|
898 av = key[n].value;
|
|
899
|
|
900 v[i].handler = av->handler;
|
|
901 v[i].data = av->data;
|
76
|
902
|
160
|
903 av->flags |= NGX_HTTP_VAR_INDEXED;
|
|
904 v[i].flags = av->flags;
|
58
|
905
|
160
|
906 av->index = i;
|
82
|
907
|
58
|
908 goto next;
|
|
909 }
|
|
910 }
|
|
911
|
|
912 if (ngx_strncmp(v[i].name.data, "http_", 5) == 0) {
|
126
|
913 v[i].handler = ngx_http_variable_unknown_header_in;
|
|
914 v[i].data = (uintptr_t) &v[i].name;
|
|
915
|
|
916 continue;
|
|
917 }
|
|
918
|
|
919 if (ngx_strncmp(v[i].name.data, "sent_http_", 10) == 0) {
|
|
920 v[i].handler = ngx_http_variable_unknown_header_out;
|
58
|
921 v[i].data = (uintptr_t) &v[i].name;
|
|
922
|
|
923 continue;
|
|
924 }
|
|
925
|
|
926 ngx_log_error(NGX_LOG_EMERG, cf->log, 0,
|
|
927 "unknown \"%V\" variable", &v[i].name);
|
|
928
|
|
929 return NGX_ERROR;
|
|
930
|
|
931 next:
|
|
932 continue;
|
|
933 }
|
|
934
|
|
935
|
160
|
936 for (n = 0; n < cmcf->variables_keys->keys.nelts; n++) {
|
|
937 av = key[n].value;
|
|
938
|
|
939 if (av->flags & NGX_HTTP_VAR_NOHASH) {
|
|
940 key[n].key.data = NULL;
|
132
|
941 }
|
|
942 }
|
|
943
|
|
944
|
160
|
945 hash.hash = &cmcf->variables_hash;
|
|
946 hash.key = ngx_hash_key;
|
|
947 hash.max_size = cmcf->variables_hash_max_size;
|
|
948 hash.bucket_size = cmcf->variables_hash_bucket_size;
|
|
949 hash.name = "variables_hash";
|
|
950 hash.pool = cf->pool;
|
|
951 hash.temp_pool = NULL;
|
58
|
952
|
160
|
953 if (ngx_hash_init(&hash, cmcf->variables_keys->keys.elts,
|
|
954 cmcf->variables_keys->keys.nelts)
|
130
|
955 != NGX_OK)
|
58
|
956 {
|
|
957 return NGX_ERROR;
|
|
958 }
|
|
959
|
160
|
960 cmcf->variables_keys = NULL;
|
58
|
961
|
|
962 return NGX_OK;
|
|
963 }
|