138
|
1
|
|
2 /*
|
|
3 * Copyright (C) Igor Sysoev
|
644
|
4 * Copyright (C) Nginx, Inc.
|
138
|
5 */
|
|
6
|
|
7
|
|
8 #include <ngx_config.h>
|
|
9 #include <ngx_core.h>
|
|
10 #include <ngx_http.h>
|
|
11
|
|
12
|
|
13 typedef struct {
|
|
14 ngx_uint_t hash_max_size;
|
|
15 ngx_uint_t hash_bucket_size;
|
|
16 } ngx_http_map_conf_t;
|
|
17
|
|
18
|
|
19 typedef struct {
|
142
|
20 ngx_hash_keys_arrays_t keys;
|
138
|
21
|
|
22 ngx_array_t *values_hash;
|
604
|
23 ngx_array_t var_values;
|
616
|
24 #if (NGX_PCRE)
|
|
25 ngx_array_t regexes;
|
|
26 #endif
|
138
|
27
|
|
28 ngx_http_variable_value_t *default_value;
|
604
|
29 ngx_conf_t *cf;
|
138
|
30 ngx_uint_t hostnames; /* unsigned hostnames:1 */
|
|
31 } ngx_http_map_conf_ctx_t;
|
|
32
|
|
33
|
|
34 typedef struct {
|
616
|
35 ngx_http_map_t map;
|
604
|
36 ngx_http_complex_value_t value;
|
138
|
37 ngx_http_variable_value_t *default_value;
|
|
38 ngx_uint_t hostnames; /* unsigned hostnames:1 */
|
|
39 } ngx_http_map_ctx_t;
|
|
40
|
|
41
|
|
42 static int ngx_libc_cdecl ngx_http_map_cmp_dns_wildcards(const void *one,
|
|
43 const void *two);
|
|
44 static void *ngx_http_map_create_conf(ngx_conf_t *cf);
|
|
45 static char *ngx_http_map_block(ngx_conf_t *cf, ngx_command_t *cmd, void *conf);
|
|
46 static char *ngx_http_map(ngx_conf_t *cf, ngx_command_t *dummy, void *conf);
|
|
47
|
|
48
|
|
49 static ngx_command_t ngx_http_map_commands[] = {
|
|
50
|
|
51 { ngx_string("map"),
|
|
52 NGX_HTTP_MAIN_CONF|NGX_CONF_BLOCK|NGX_CONF_TAKE2,
|
|
53 ngx_http_map_block,
|
|
54 NGX_HTTP_MAIN_CONF_OFFSET,
|
|
55 0,
|
|
56 NULL },
|
|
57
|
|
58 { ngx_string("map_hash_max_size"),
|
|
59 NGX_HTTP_MAIN_CONF|NGX_CONF_TAKE1,
|
|
60 ngx_conf_set_num_slot,
|
|
61 NGX_HTTP_MAIN_CONF_OFFSET,
|
|
62 offsetof(ngx_http_map_conf_t, hash_max_size),
|
|
63 NULL },
|
|
64
|
|
65 { ngx_string("map_hash_bucket_size"),
|
|
66 NGX_HTTP_MAIN_CONF|NGX_CONF_TAKE1,
|
|
67 ngx_conf_set_num_slot,
|
|
68 NGX_HTTP_MAIN_CONF_OFFSET,
|
|
69 offsetof(ngx_http_map_conf_t, hash_bucket_size),
|
|
70 NULL },
|
|
71
|
|
72 ngx_null_command
|
|
73 };
|
|
74
|
|
75
|
|
76 static ngx_http_module_t ngx_http_map_module_ctx = {
|
|
77 NULL, /* preconfiguration */
|
|
78 NULL, /* postconfiguration */
|
|
79
|
|
80 ngx_http_map_create_conf, /* create main configuration */
|
|
81 NULL, /* init main configuration */
|
|
82
|
|
83 NULL, /* create server configuration */
|
|
84 NULL, /* merge server configuration */
|
|
85
|
|
86 NULL, /* create location configuration */
|
|
87 NULL /* merge location configuration */
|
|
88 };
|
|
89
|
|
90
|
|
91 ngx_module_t ngx_http_map_module = {
|
|
92 NGX_MODULE_V1,
|
|
93 &ngx_http_map_module_ctx, /* module context */
|
|
94 ngx_http_map_commands, /* module directives */
|
|
95 NGX_HTTP_MODULE, /* module type */
|
|
96 NULL, /* init master */
|
|
97 NULL, /* init module */
|
|
98 NULL, /* init process */
|
|
99 NULL, /* init thread */
|
|
100 NULL, /* exit thread */
|
|
101 NULL, /* exit process */
|
|
102 NULL, /* exit master */
|
|
103 NGX_MODULE_V1_PADDING
|
|
104 };
|
|
105
|
|
106
|
|
107 static ngx_int_t
|
|
108 ngx_http_map_variable(ngx_http_request_t *r, ngx_http_variable_value_t *v,
|
|
109 uintptr_t data)
|
|
110 {
|
|
111 ngx_http_map_ctx_t *map = (ngx_http_map_ctx_t *) data;
|
|
112
|
|
113 size_t len;
|
604
|
114 ngx_str_t val;
|
|
115 ngx_http_variable_value_t *value;
|
138
|
116
|
|
117 ngx_log_debug0(NGX_LOG_DEBUG_HTTP, r->connection->log, 0,
|
|
118 "http map started");
|
|
119
|
604
|
120 if (ngx_http_complex_value(r, &map->value, &val) != NGX_OK) {
|
|
121 return NGX_ERROR;
|
250
|
122 }
|
|
123
|
604
|
124 len = val.len;
|
138
|
125
|
604
|
126 if (len && map->hostnames && val.data[len - 1] == '.') {
|
138
|
127 len--;
|
|
128 }
|
|
129
|
628
|
130 value = ngx_http_map_find(r, &map->map, &val);
|
604
|
131
|
|
132 if (value == NULL) {
|
|
133 value = map->default_value;
|
138
|
134 }
|
|
135
|
604
|
136 if (!value->valid) {
|
|
137 value = ngx_http_get_flushed_variable(r, (ngx_uint_t) value->data);
|
138
|
138
|
604
|
139 if (value == NULL || value->not_found) {
|
|
140 value = &ngx_http_variable_null_value;
|
|
141 }
|
138
|
142 }
|
|
143
|
604
|
144 *v = *value;
|
|
145
|
138
|
146 ngx_log_debug2(NGX_LOG_DEBUG_HTTP, r->connection->log, 0,
|
604
|
147 "http map: \"%v\" \"%v\"", &val, v);
|
138
|
148
|
|
149 return NGX_OK;
|
|
150 }
|
|
151
|
|
152
|
|
153 static void *
|
|
154 ngx_http_map_create_conf(ngx_conf_t *cf)
|
|
155 {
|
|
156 ngx_http_map_conf_t *mcf;
|
|
157
|
|
158 mcf = ngx_palloc(cf->pool, sizeof(ngx_http_map_conf_t));
|
|
159 if (mcf == NULL) {
|
496
|
160 return NULL;
|
138
|
161 }
|
|
162
|
|
163 mcf->hash_max_size = NGX_CONF_UNSET_UINT;
|
|
164 mcf->hash_bucket_size = NGX_CONF_UNSET_UINT;
|
|
165
|
|
166 return mcf;
|
|
167 }
|
|
168
|
|
169
|
|
170 static char *
|
|
171 ngx_http_map_block(ngx_conf_t *cf, ngx_command_t *cmd, void *conf)
|
|
172 {
|
|
173 ngx_http_map_conf_t *mcf = conf;
|
|
174
|
604
|
175 char *rv;
|
|
176 ngx_str_t *value, name;
|
|
177 ngx_conf_t save;
|
|
178 ngx_pool_t *pool;
|
|
179 ngx_hash_init_t hash;
|
|
180 ngx_http_map_ctx_t *map;
|
|
181 ngx_http_variable_t *var;
|
|
182 ngx_http_map_conf_ctx_t ctx;
|
|
183 ngx_http_compile_complex_value_t ccv;
|
138
|
184
|
|
185 if (mcf->hash_max_size == NGX_CONF_UNSET_UINT) {
|
|
186 mcf->hash_max_size = 2048;
|
|
187 }
|
|
188
|
|
189 if (mcf->hash_bucket_size == NGX_CONF_UNSET_UINT) {
|
|
190 mcf->hash_bucket_size = ngx_cacheline_size;
|
|
191
|
|
192 } else {
|
|
193 mcf->hash_bucket_size = ngx_align(mcf->hash_bucket_size,
|
|
194 ngx_cacheline_size);
|
|
195 }
|
|
196
|
|
197 map = ngx_pcalloc(cf->pool, sizeof(ngx_http_map_ctx_t));
|
|
198 if (map == NULL) {
|
|
199 return NGX_CONF_ERROR;
|
|
200 }
|
|
201
|
|
202 value = cf->args->elts;
|
|
203
|
604
|
204 ngx_memzero(&ccv, sizeof(ngx_http_compile_complex_value_t));
|
138
|
205
|
604
|
206 ccv.cf = cf;
|
|
207 ccv.value = &value[1];
|
|
208 ccv.complex_value = &map->value;
|
138
|
209
|
604
|
210 if (ngx_http_compile_complex_value(&ccv) != NGX_OK) {
|
138
|
211 return NGX_CONF_ERROR;
|
|
212 }
|
|
213
|
|
214 name = value[2];
|
|
215 name.len--;
|
|
216 name.data++;
|
|
217
|
340
|
218 var = ngx_http_add_variable(cf, &name, NGX_HTTP_VAR_CHANGEABLE);
|
138
|
219 if (var == NULL) {
|
|
220 return NGX_CONF_ERROR;
|
|
221 }
|
|
222
|
186
|
223 var->get_handler = ngx_http_map_variable;
|
138
|
224 var->data = (uintptr_t) map;
|
|
225
|
|
226 pool = ngx_create_pool(16384, cf->log);
|
|
227 if (pool == NULL) {
|
|
228 return NGX_CONF_ERROR;
|
|
229 }
|
|
230
|
142
|
231 ctx.keys.pool = cf->pool;
|
|
232 ctx.keys.temp_pool = pool;
|
138
|
233
|
142
|
234 if (ngx_hash_keys_array_init(&ctx.keys, NGX_HASH_LARGE) != NGX_OK) {
|
138
|
235 ngx_destroy_pool(pool);
|
|
236 return NGX_CONF_ERROR;
|
|
237 }
|
|
238
|
142
|
239 ctx.values_hash = ngx_pcalloc(pool, sizeof(ngx_array_t) * ctx.keys.hsize);
|
138
|
240 if (ctx.values_hash == NULL) {
|
|
241 ngx_destroy_pool(pool);
|
|
242 return NGX_CONF_ERROR;
|
|
243 }
|
|
244
|
604
|
245 if (ngx_array_init(&ctx.var_values, cf->pool, 2,
|
|
246 sizeof(ngx_http_variable_value_t))
|
|
247 != NGX_OK)
|
|
248 {
|
|
249 ngx_destroy_pool(pool);
|
|
250 return NGX_CONF_ERROR;
|
|
251 }
|
|
252
|
616
|
253 #if (NGX_PCRE)
|
|
254 if (ngx_array_init(&ctx.regexes, cf->pool, 2, sizeof(ngx_http_map_regex_t))
|
|
255 != NGX_OK)
|
|
256 {
|
|
257 ngx_destroy_pool(pool);
|
|
258 return NGX_CONF_ERROR;
|
|
259 }
|
|
260 #endif
|
|
261
|
138
|
262 ctx.default_value = NULL;
|
604
|
263 ctx.cf = &save;
|
138
|
264 ctx.hostnames = 0;
|
|
265
|
|
266 save = *cf;
|
|
267 cf->pool = pool;
|
|
268 cf->ctx = &ctx;
|
|
269 cf->handler = ngx_http_map;
|
|
270 cf->handler_conf = conf;
|
|
271
|
|
272 rv = ngx_conf_parse(cf, NULL);
|
|
273
|
|
274 *cf = save;
|
|
275
|
|
276 if (rv != NGX_CONF_OK) {
|
|
277 ngx_destroy_pool(pool);
|
|
278 return rv;
|
|
279 }
|
|
280
|
312
|
281 map->default_value = ctx.default_value ? ctx.default_value:
|
|
282 &ngx_http_variable_null_value;
|
|
283
|
138
|
284 hash.key = ngx_hash_key_lc;
|
|
285 hash.max_size = mcf->hash_max_size;
|
|
286 hash.bucket_size = mcf->hash_bucket_size;
|
|
287 hash.name = "map_hash";
|
|
288 hash.pool = cf->pool;
|
|
289
|
142
|
290 if (ctx.keys.keys.nelts) {
|
616
|
291 hash.hash = &map->map.hash.hash;
|
138
|
292 hash.temp_pool = NULL;
|
|
293
|
142
|
294 if (ngx_hash_init(&hash, ctx.keys.keys.elts, ctx.keys.keys.nelts)
|
140
|
295 != NGX_OK)
|
|
296 {
|
142
|
297 ngx_destroy_pool(pool);
|
138
|
298 return NGX_CONF_ERROR;
|
|
299 }
|
|
300 }
|
|
301
|
312
|
302 if (ctx.keys.dns_wc_head.nelts) {
|
138
|
303
|
312
|
304 ngx_qsort(ctx.keys.dns_wc_head.elts,
|
|
305 (size_t) ctx.keys.dns_wc_head.nelts,
|
138
|
306 sizeof(ngx_hash_key_t), ngx_http_map_cmp_dns_wildcards);
|
|
307
|
|
308 hash.hash = NULL;
|
|
309 hash.temp_pool = pool;
|
|
310
|
312
|
311 if (ngx_hash_wildcard_init(&hash, ctx.keys.dns_wc_head.elts,
|
|
312 ctx.keys.dns_wc_head.nelts)
|
138
|
313 != NGX_OK)
|
|
314 {
|
142
|
315 ngx_destroy_pool(pool);
|
138
|
316 return NGX_CONF_ERROR;
|
|
317 }
|
|
318
|
616
|
319 map->map.hash.wc_head = (ngx_hash_wildcard_t *) hash.hash;
|
312
|
320 }
|
|
321
|
|
322 if (ctx.keys.dns_wc_tail.nelts) {
|
|
323
|
|
324 ngx_qsort(ctx.keys.dns_wc_tail.elts,
|
|
325 (size_t) ctx.keys.dns_wc_tail.nelts,
|
|
326 sizeof(ngx_hash_key_t), ngx_http_map_cmp_dns_wildcards);
|
|
327
|
|
328 hash.hash = NULL;
|
|
329 hash.temp_pool = pool;
|
|
330
|
|
331 if (ngx_hash_wildcard_init(&hash, ctx.keys.dns_wc_tail.elts,
|
|
332 ctx.keys.dns_wc_tail.nelts)
|
|
333 != NGX_OK)
|
|
334 {
|
|
335 ngx_destroy_pool(pool);
|
|
336 return NGX_CONF_ERROR;
|
|
337 }
|
|
338
|
616
|
339 map->map.hash.wc_tail = (ngx_hash_wildcard_t *) hash.hash;
|
138
|
340 }
|
|
341
|
616
|
342 #if (NGX_PCRE)
|
|
343
|
|
344 if (ctx.regexes.nelts) {
|
|
345 map->map.regex = ctx.regexes.elts;
|
|
346 map->map.nregex = ctx.regexes.nelts;
|
|
347 }
|
|
348
|
|
349 #endif
|
|
350
|
138
|
351 ngx_destroy_pool(pool);
|
|
352
|
|
353 return rv;
|
|
354 }
|
|
355
|
|
356
|
|
357 static int ngx_libc_cdecl
|
|
358 ngx_http_map_cmp_dns_wildcards(const void *one, const void *two)
|
|
359 {
|
|
360 ngx_hash_key_t *first, *second;
|
|
361
|
|
362 first = (ngx_hash_key_t *) one;
|
|
363 second = (ngx_hash_key_t *) two;
|
|
364
|
526
|
365 return ngx_dns_strcmp(first->key.data, second->key.data);
|
138
|
366 }
|
|
367
|
|
368
|
|
369 static char *
|
|
370 ngx_http_map(ngx_conf_t *cf, ngx_command_t *dummy, void *conf)
|
|
371 {
|
604
|
372 ngx_int_t rc, index;
|
|
373 ngx_str_t *value, file, name;
|
312
|
374 ngx_uint_t i, key;
|
138
|
375 ngx_http_map_conf_ctx_t *ctx;
|
140
|
376 ngx_http_variable_value_t *var, **vp;
|
138
|
377
|
|
378 ctx = cf->ctx;
|
|
379
|
|
380 value = cf->args->elts;
|
|
381
|
|
382 if (cf->args->nelts == 1
|
|
383 && ngx_strcmp(value[0].data, "hostnames") == 0)
|
|
384 {
|
|
385 ctx->hostnames = 1;
|
|
386 return NGX_CONF_OK;
|
|
387
|
|
388 } else if (cf->args->nelts != 2) {
|
|
389 ngx_conf_log_error(NGX_LOG_EMERG, cf, 0,
|
|
390 "invalid number of the map parameters");
|
|
391 return NGX_CONF_ERROR;
|
|
392 }
|
|
393
|
|
394 if (ngx_strcmp(value[0].data, "include") == 0) {
|
|
395 file = value[1];
|
|
396
|
582
|
397 if (ngx_conf_full_name(cf->cycle, &file, 1) != NGX_OK) {
|
138
|
398 return NGX_CONF_ERROR;
|
|
399 }
|
|
400
|
|
401 ngx_log_debug1(NGX_LOG_DEBUG_CORE, cf->log, 0, "include %s", file.data);
|
|
402
|
|
403 return ngx_conf_parse(cf, &file);
|
|
404 }
|
|
405
|
604
|
406 if (value[1].data[0] == '$') {
|
|
407 name = value[1];
|
|
408 name.len--;
|
|
409 name.data++;
|
|
410
|
|
411 index = ngx_http_get_variable_index(ctx->cf, &name);
|
|
412 if (index == NGX_ERROR) {
|
|
413 return NGX_CONF_ERROR;
|
|
414 }
|
|
415
|
|
416 var = ctx->var_values.elts;
|
|
417
|
|
418 for (i = 0; i < ctx->var_values.nelts; i++) {
|
|
419 if (index == (ngx_int_t) var[i].data) {
|
|
420 goto found;
|
|
421 }
|
|
422 }
|
|
423
|
|
424 var = ngx_palloc(ctx->keys.pool, sizeof(ngx_http_variable_value_t));
|
|
425 if (var == NULL) {
|
|
426 return NGX_CONF_ERROR;
|
|
427 }
|
|
428
|
|
429 var->valid = 0;
|
|
430 var->no_cacheable = 0;
|
|
431 var->not_found = 0;
|
|
432 var->len = 0;
|
|
433 var->data = (u_char *) index;
|
|
434
|
|
435 vp = ngx_array_push(&ctx->var_values);
|
|
436 if (vp == NULL) {
|
|
437 return NGX_CONF_ERROR;
|
|
438 }
|
|
439
|
|
440 *vp = var;
|
|
441
|
|
442 goto found;
|
|
443 }
|
|
444
|
138
|
445 key = 0;
|
|
446
|
|
447 for (i = 0; i < value[1].len; i++) {
|
|
448 key = ngx_hash(key, value[1].data[i]);
|
|
449 }
|
|
450
|
142
|
451 key %= ctx->keys.hsize;
|
138
|
452
|
|
453 vp = ctx->values_hash[key].elts;
|
|
454
|
|
455 if (vp) {
|
|
456 for (i = 0; i < ctx->values_hash[key].nelts; i++) {
|
|
457 if (value[1].len != (size_t) vp[i]->len) {
|
|
458 continue;
|
|
459 }
|
|
460
|
|
461 if (ngx_strncmp(value[1].data, vp[i]->data, value[1].len) == 0) {
|
|
462 var = vp[i];
|
|
463 goto found;
|
|
464 }
|
|
465 }
|
|
466
|
|
467 } else {
|
|
468 if (ngx_array_init(&ctx->values_hash[key], cf->pool, 4,
|
|
469 sizeof(ngx_http_variable_value_t *))
|
|
470 != NGX_OK)
|
|
471 {
|
|
472 return NGX_CONF_ERROR;
|
|
473 }
|
|
474 }
|
|
475
|
142
|
476 var = ngx_palloc(ctx->keys.pool, sizeof(ngx_http_variable_value_t));
|
138
|
477 if (var == NULL) {
|
|
478 return NGX_CONF_ERROR;
|
|
479 }
|
|
480
|
|
481 var->len = value[1].len;
|
142
|
482 var->data = ngx_pstrdup(ctx->keys.pool, &value[1]);
|
138
|
483 if (var->data == NULL) {
|
|
484 return NGX_CONF_ERROR;
|
|
485 }
|
|
486
|
|
487 var->valid = 1;
|
340
|
488 var->no_cacheable = 0;
|
138
|
489 var->not_found = 0;
|
|
490
|
|
491 vp = ngx_array_push(&ctx->values_hash[key]);
|
|
492 if (vp == NULL) {
|
|
493 return NGX_CONF_ERROR;
|
|
494 }
|
|
495
|
|
496 *vp = var;
|
|
497
|
|
498 found:
|
|
499
|
312
|
500 if (ngx_strcmp(value[0].data, "default") == 0) {
|
138
|
501
|
312
|
502 if (ctx->default_value) {
|
138
|
503 ngx_conf_log_error(NGX_LOG_EMERG, cf, 0,
|
312
|
504 "duplicate default map parameter");
|
138
|
505 return NGX_CONF_ERROR;
|
|
506 }
|
|
507
|
312
|
508 ctx->default_value = var;
|
|
509
|
|
510 return NGX_CONF_OK;
|
138
|
511 }
|
|
512
|
616
|
513 #if (NGX_PCRE)
|
|
514
|
|
515 if (value[0].len && value[0].data[0] == '~') {
|
|
516 ngx_regex_compile_t rc;
|
|
517 ngx_http_map_regex_t *regex;
|
|
518 u_char errstr[NGX_MAX_CONF_ERRSTR];
|
|
519
|
|
520 regex = ngx_array_push(&ctx->regexes);
|
|
521 if (regex == NULL) {
|
|
522 return NGX_CONF_ERROR;
|
|
523 }
|
|
524
|
|
525 value[0].len--;
|
|
526 value[0].data++;
|
|
527
|
|
528 ngx_memzero(&rc, sizeof(ngx_regex_compile_t));
|
|
529
|
628
|
530 if (value[0].data[0] == '*') {
|
|
531 value[0].len--;
|
|
532 value[0].data++;
|
|
533 rc.options = NGX_REGEX_CASELESS;
|
|
534 }
|
|
535
|
616
|
536 rc.pattern = value[0];
|
|
537 rc.err.len = NGX_MAX_CONF_ERRSTR;
|
|
538 rc.err.data = errstr;
|
|
539
|
|
540 regex->regex = ngx_http_regex_compile(ctx->cf, &rc);
|
|
541 if (regex->regex == NULL) {
|
|
542 return NGX_CONF_ERROR;
|
|
543 }
|
|
544
|
|
545 regex->value = var;
|
|
546
|
|
547 return NGX_CONF_OK;
|
|
548 }
|
|
549
|
|
550 #endif
|
|
551
|
604
|
552 if (value[0].len && value[0].data[0] == '\\') {
|
312
|
553 value[0].len--;
|
|
554 value[0].data++;
|
|
555 }
|
|
556
|
|
557 rc = ngx_hash_add_key(&ctx->keys, &value[0], var,
|
|
558 (ctx->hostnames) ? NGX_HASH_WILDCARD_KEY : 0);
|
138
|
559
|
140
|
560 if (rc == NGX_OK) {
|
|
561 return NGX_CONF_OK;
|
|
562 }
|
138
|
563
|
312
|
564 if (rc == NGX_DECLINED) {
|
|
565 ngx_conf_log_error(NGX_LOG_EMERG, cf, 0,
|
|
566 "invalid hostname or wildcard \"%V\"", &value[0]);
|
|
567 }
|
|
568
|
140
|
569 if (rc == NGX_BUSY) {
|
|
570 ngx_conf_log_error(NGX_LOG_EMERG, cf, 0,
|
|
571 "conflicting parameter \"%V\"", &value[0]);
|
|
572 }
|
138
|
573
|
140
|
574 return NGX_CONF_ERROR;
|
138
|
575 }
|