Mercurial > hg > nginx
annotate src/http/modules/ngx_http_map_module.c @ 3789:19c931be8b98
allow expressions in the first "map" parameter
| author | Igor Sysoev <igor@sysoev.ru> |
|---|---|
| date | Thu, 25 Nov 2010 15:22:43 +0000 |
| parents | ac33852faaac |
| children | d8300807c4f3 |
| rev | line source |
|---|---|
| 589 | 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 | |
| 11 | |
| 12 typedef struct { | |
| 13 ngx_uint_t hash_max_size; | |
| 14 ngx_uint_t hash_bucket_size; | |
| 15 } ngx_http_map_conf_t; | |
| 16 | |
| 17 | |
| 18 typedef struct { | |
| 593 | 19 ngx_hash_keys_arrays_t keys; |
| 589 | 20 |
| 21 ngx_array_t *values_hash; | |
| 22 | |
| 23 ngx_http_variable_value_t *default_value; | |
| 24 ngx_uint_t hostnames; /* unsigned hostnames:1 */ | |
| 25 } ngx_http_map_conf_ctx_t; | |
| 26 | |
| 27 | |
| 28 typedef struct { | |
|
1253
8ef04207c84f
the "www.example.*" wildcard hash support
Igor Sysoev <igor@sysoev.ru>
parents:
797
diff
changeset
|
29 ngx_hash_combined_t hash; |
|
3789
19c931be8b98
allow expressions in the first "map" parameter
Igor Sysoev <igor@sysoev.ru>
parents:
3642
diff
changeset
|
30 ngx_http_complex_value_t value; |
| 589 | 31 ngx_http_variable_value_t *default_value; |
| 32 ngx_uint_t hostnames; /* unsigned hostnames:1 */ | |
| 33 } ngx_http_map_ctx_t; | |
| 34 | |
| 35 | |
| 36 static int ngx_libc_cdecl ngx_http_map_cmp_dns_wildcards(const void *one, | |
| 37 const void *two); | |
| 38 static void *ngx_http_map_create_conf(ngx_conf_t *cf); | |
| 39 static char *ngx_http_map_block(ngx_conf_t *cf, ngx_command_t *cmd, void *conf); | |
| 40 static char *ngx_http_map(ngx_conf_t *cf, ngx_command_t *dummy, void *conf); | |
| 41 | |
| 42 | |
| 43 static ngx_command_t ngx_http_map_commands[] = { | |
| 44 | |
| 45 { ngx_string("map"), | |
| 46 NGX_HTTP_MAIN_CONF|NGX_CONF_BLOCK|NGX_CONF_TAKE2, | |
| 47 ngx_http_map_block, | |
| 48 NGX_HTTP_MAIN_CONF_OFFSET, | |
| 49 0, | |
| 50 NULL }, | |
| 51 | |
| 52 { ngx_string("map_hash_max_size"), | |
| 53 NGX_HTTP_MAIN_CONF|NGX_CONF_TAKE1, | |
| 54 ngx_conf_set_num_slot, | |
| 55 NGX_HTTP_MAIN_CONF_OFFSET, | |
| 56 offsetof(ngx_http_map_conf_t, hash_max_size), | |
| 57 NULL }, | |
| 58 | |
| 59 { ngx_string("map_hash_bucket_size"), | |
| 60 NGX_HTTP_MAIN_CONF|NGX_CONF_TAKE1, | |
| 61 ngx_conf_set_num_slot, | |
| 62 NGX_HTTP_MAIN_CONF_OFFSET, | |
| 63 offsetof(ngx_http_map_conf_t, hash_bucket_size), | |
| 64 NULL }, | |
| 65 | |
| 66 ngx_null_command | |
| 67 }; | |
| 68 | |
| 69 | |
| 70 static ngx_http_module_t ngx_http_map_module_ctx = { | |
| 71 NULL, /* preconfiguration */ | |
| 72 NULL, /* postconfiguration */ | |
| 73 | |
| 74 ngx_http_map_create_conf, /* create main configuration */ | |
| 75 NULL, /* init main configuration */ | |
| 76 | |
| 77 NULL, /* create server configuration */ | |
| 78 NULL, /* merge server configuration */ | |
| 79 | |
| 80 NULL, /* create location configuration */ | |
| 81 NULL /* merge location configuration */ | |
| 82 }; | |
| 83 | |
| 84 | |
| 85 ngx_module_t ngx_http_map_module = { | |
| 86 NGX_MODULE_V1, | |
| 87 &ngx_http_map_module_ctx, /* module context */ | |
| 88 ngx_http_map_commands, /* module directives */ | |
| 89 NGX_HTTP_MODULE, /* module type */ | |
| 90 NULL, /* init master */ | |
| 91 NULL, /* init module */ | |
| 92 NULL, /* init process */ | |
| 93 NULL, /* init thread */ | |
| 94 NULL, /* exit thread */ | |
| 95 NULL, /* exit process */ | |
| 96 NULL, /* exit master */ | |
| 97 NGX_MODULE_V1_PADDING | |
| 98 }; | |
| 99 | |
| 100 | |
| 101 static ngx_int_t | |
| 102 ngx_http_map_variable(ngx_http_request_t *r, ngx_http_variable_value_t *v, | |
| 103 uintptr_t data) | |
| 104 { | |
| 105 ngx_http_map_ctx_t *map = (ngx_http_map_ctx_t *) data; | |
| 106 | |
| 107 size_t len; | |
|
3789
19c931be8b98
allow expressions in the first "map" parameter
Igor Sysoev <igor@sysoev.ru>
parents:
3642
diff
changeset
|
108 ngx_str_t val; |
| 2136 | 109 ngx_uint_t key; |
|
3789
19c931be8b98
allow expressions in the first "map" parameter
Igor Sysoev <igor@sysoev.ru>
parents:
3642
diff
changeset
|
110 ngx_http_variable_value_t *value; |
| 589 | 111 |
| 112 ngx_log_debug0(NGX_LOG_DEBUG_HTTP, r->connection->log, 0, | |
| 113 "http map started"); | |
| 114 | |
|
3789
19c931be8b98
allow expressions in the first "map" parameter
Igor Sysoev <igor@sysoev.ru>
parents:
3642
diff
changeset
|
115 if (ngx_http_complex_value(r, &map->value, &val) != NGX_OK) { |
|
19c931be8b98
allow expressions in the first "map" parameter
Igor Sysoev <igor@sysoev.ru>
parents:
3642
diff
changeset
|
116 return NGX_ERROR; |
|
797
36f7b549f481
fix segfault if $server_addr failed
Igor Sysoev <igor@sysoev.ru>
parents:
637
diff
changeset
|
117 } |
|
36f7b549f481
fix segfault if $server_addr failed
Igor Sysoev <igor@sysoev.ru>
parents:
637
diff
changeset
|
118 |
|
3789
19c931be8b98
allow expressions in the first "map" parameter
Igor Sysoev <igor@sysoev.ru>
parents:
3642
diff
changeset
|
119 len = val.len; |
| 589 | 120 |
|
3789
19c931be8b98
allow expressions in the first "map" parameter
Igor Sysoev <igor@sysoev.ru>
parents:
3642
diff
changeset
|
121 if (len && map->hostnames && val.data[len - 1] == '.') { |
| 589 | 122 len--; |
| 123 } | |
| 124 | |
| 125 if (len == 0) { | |
| 126 *v = *map->default_value; | |
| 127 return NGX_OK; | |
| 128 } | |
| 129 | |
|
3789
19c931be8b98
allow expressions in the first "map" parameter
Igor Sysoev <igor@sysoev.ru>
parents:
3642
diff
changeset
|
130 key = ngx_hash_strlow(val.data, val.data, len); |
| 589 | 131 |
|
3789
19c931be8b98
allow expressions in the first "map" parameter
Igor Sysoev <igor@sysoev.ru>
parents:
3642
diff
changeset
|
132 value = ngx_hash_find_combined(&map->hash, key, val.data, len); |
| 589 | 133 |
| 134 if (value) { | |
| 135 *v = *value; | |
| 136 | |
| 137 } else { | |
|
1253
8ef04207c84f
the "www.example.*" wildcard hash support
Igor Sysoev <igor@sysoev.ru>
parents:
797
diff
changeset
|
138 *v = *map->default_value; |
| 589 | 139 } |
| 140 | |
| 141 ngx_log_debug2(NGX_LOG_DEBUG_HTTP, r->connection->log, 0, | |
|
3789
19c931be8b98
allow expressions in the first "map" parameter
Igor Sysoev <igor@sysoev.ru>
parents:
3642
diff
changeset
|
142 "http map: \"%v\" \"%v\"", &val, v); |
| 589 | 143 |
| 144 return NGX_OK; | |
| 145 } | |
| 146 | |
| 147 | |
| 148 static void * | |
| 149 ngx_http_map_create_conf(ngx_conf_t *cf) | |
| 150 { | |
| 151 ngx_http_map_conf_t *mcf; | |
| 152 | |
| 153 mcf = ngx_palloc(cf->pool, sizeof(ngx_http_map_conf_t)); | |
| 154 if (mcf == NULL) { | |
|
2912
c7d57b539248
return NULL instead of NGX_CONF_ERROR on a create conf failure
Igor Sysoev <igor@sysoev.ru>
parents:
2536
diff
changeset
|
155 return NULL; |
| 589 | 156 } |
| 157 | |
| 158 mcf->hash_max_size = NGX_CONF_UNSET_UINT; | |
| 159 mcf->hash_bucket_size = NGX_CONF_UNSET_UINT; | |
| 160 | |
| 161 return mcf; | |
| 162 } | |
| 163 | |
| 164 | |
| 165 static char * | |
| 166 ngx_http_map_block(ngx_conf_t *cf, ngx_command_t *cmd, void *conf) | |
| 167 { | |
| 168 ngx_http_map_conf_t *mcf = conf; | |
| 169 | |
|
3789
19c931be8b98
allow expressions in the first "map" parameter
Igor Sysoev <igor@sysoev.ru>
parents:
3642
diff
changeset
|
170 char *rv; |
|
19c931be8b98
allow expressions in the first "map" parameter
Igor Sysoev <igor@sysoev.ru>
parents:
3642
diff
changeset
|
171 ngx_str_t *value, name; |
|
19c931be8b98
allow expressions in the first "map" parameter
Igor Sysoev <igor@sysoev.ru>
parents:
3642
diff
changeset
|
172 ngx_conf_t save; |
|
19c931be8b98
allow expressions in the first "map" parameter
Igor Sysoev <igor@sysoev.ru>
parents:
3642
diff
changeset
|
173 ngx_pool_t *pool; |
|
19c931be8b98
allow expressions in the first "map" parameter
Igor Sysoev <igor@sysoev.ru>
parents:
3642
diff
changeset
|
174 ngx_hash_init_t hash; |
|
19c931be8b98
allow expressions in the first "map" parameter
Igor Sysoev <igor@sysoev.ru>
parents:
3642
diff
changeset
|
175 ngx_http_map_ctx_t *map; |
|
19c931be8b98
allow expressions in the first "map" parameter
Igor Sysoev <igor@sysoev.ru>
parents:
3642
diff
changeset
|
176 ngx_http_variable_t *var; |
|
19c931be8b98
allow expressions in the first "map" parameter
Igor Sysoev <igor@sysoev.ru>
parents:
3642
diff
changeset
|
177 ngx_http_map_conf_ctx_t ctx; |
|
19c931be8b98
allow expressions in the first "map" parameter
Igor Sysoev <igor@sysoev.ru>
parents:
3642
diff
changeset
|
178 ngx_http_compile_complex_value_t ccv; |
| 589 | 179 |
| 180 if (mcf->hash_max_size == NGX_CONF_UNSET_UINT) { | |
| 181 mcf->hash_max_size = 2048; | |
| 182 } | |
| 183 | |
| 184 if (mcf->hash_bucket_size == NGX_CONF_UNSET_UINT) { | |
| 185 mcf->hash_bucket_size = ngx_cacheline_size; | |
| 186 | |
| 187 } else { | |
| 188 mcf->hash_bucket_size = ngx_align(mcf->hash_bucket_size, | |
| 189 ngx_cacheline_size); | |
| 190 } | |
| 191 | |
| 192 map = ngx_pcalloc(cf->pool, sizeof(ngx_http_map_ctx_t)); | |
| 193 if (map == NULL) { | |
| 194 return NGX_CONF_ERROR; | |
| 195 } | |
| 196 | |
| 197 value = cf->args->elts; | |
| 198 | |
|
3789
19c931be8b98
allow expressions in the first "map" parameter
Igor Sysoev <igor@sysoev.ru>
parents:
3642
diff
changeset
|
199 ngx_memzero(&ccv, sizeof(ngx_http_compile_complex_value_t)); |
| 589 | 200 |
|
3789
19c931be8b98
allow expressions in the first "map" parameter
Igor Sysoev <igor@sysoev.ru>
parents:
3642
diff
changeset
|
201 ccv.cf = cf; |
|
19c931be8b98
allow expressions in the first "map" parameter
Igor Sysoev <igor@sysoev.ru>
parents:
3642
diff
changeset
|
202 ccv.value = &value[1]; |
|
19c931be8b98
allow expressions in the first "map" parameter
Igor Sysoev <igor@sysoev.ru>
parents:
3642
diff
changeset
|
203 ccv.complex_value = &map->value; |
| 589 | 204 |
|
3789
19c931be8b98
allow expressions in the first "map" parameter
Igor Sysoev <igor@sysoev.ru>
parents:
3642
diff
changeset
|
205 if (ngx_http_compile_complex_value(&ccv) != NGX_OK) { |
| 589 | 206 return NGX_CONF_ERROR; |
| 207 } | |
| 208 | |
| 209 name = value[2]; | |
| 210 name.len--; | |
| 211 name.data++; | |
| 212 | |
| 1565 | 213 var = ngx_http_add_variable(cf, &name, NGX_HTTP_VAR_CHANGEABLE); |
| 589 | 214 if (var == NULL) { |
| 215 return NGX_CONF_ERROR; | |
| 216 } | |
| 217 | |
| 637 | 218 var->get_handler = ngx_http_map_variable; |
| 589 | 219 var->data = (uintptr_t) map; |
| 220 | |
| 221 pool = ngx_create_pool(16384, cf->log); | |
| 222 if (pool == NULL) { | |
| 223 return NGX_CONF_ERROR; | |
| 224 } | |
| 225 | |
| 593 | 226 ctx.keys.pool = cf->pool; |
| 227 ctx.keys.temp_pool = pool; | |
| 589 | 228 |
| 593 | 229 if (ngx_hash_keys_array_init(&ctx.keys, NGX_HASH_LARGE) != NGX_OK) { |
| 589 | 230 ngx_destroy_pool(pool); |
| 231 return NGX_CONF_ERROR; | |
| 232 } | |
| 233 | |
| 593 | 234 ctx.values_hash = ngx_pcalloc(pool, sizeof(ngx_array_t) * ctx.keys.hsize); |
| 589 | 235 if (ctx.values_hash == NULL) { |
| 236 ngx_destroy_pool(pool); | |
| 237 return NGX_CONF_ERROR; | |
| 238 } | |
| 239 | |
| 240 ctx.default_value = NULL; | |
| 241 ctx.hostnames = 0; | |
| 242 | |
| 243 save = *cf; | |
| 244 cf->pool = pool; | |
| 245 cf->ctx = &ctx; | |
| 246 cf->handler = ngx_http_map; | |
| 247 cf->handler_conf = conf; | |
| 248 | |
| 249 rv = ngx_conf_parse(cf, NULL); | |
| 250 | |
| 251 *cf = save; | |
| 252 | |
| 253 if (rv != NGX_CONF_OK) { | |
| 254 ngx_destroy_pool(pool); | |
| 255 return rv; | |
| 256 } | |
| 257 | |
|
1253
8ef04207c84f
the "www.example.*" wildcard hash support
Igor Sysoev <igor@sysoev.ru>
parents:
797
diff
changeset
|
258 map->default_value = ctx.default_value ? ctx.default_value: |
|
8ef04207c84f
the "www.example.*" wildcard hash support
Igor Sysoev <igor@sysoev.ru>
parents:
797
diff
changeset
|
259 &ngx_http_variable_null_value; |
|
8ef04207c84f
the "www.example.*" wildcard hash support
Igor Sysoev <igor@sysoev.ru>
parents:
797
diff
changeset
|
260 |
| 589 | 261 hash.key = ngx_hash_key_lc; |
| 262 hash.max_size = mcf->hash_max_size; | |
| 263 hash.bucket_size = mcf->hash_bucket_size; | |
| 264 hash.name = "map_hash"; | |
| 265 hash.pool = cf->pool; | |
| 266 | |
| 593 | 267 if (ctx.keys.keys.nelts) { |
|
1253
8ef04207c84f
the "www.example.*" wildcard hash support
Igor Sysoev <igor@sysoev.ru>
parents:
797
diff
changeset
|
268 hash.hash = &map->hash.hash; |
| 589 | 269 hash.temp_pool = NULL; |
| 270 | |
| 593 | 271 if (ngx_hash_init(&hash, ctx.keys.keys.elts, ctx.keys.keys.nelts) |
| 591 | 272 != NGX_OK) |
| 273 { | |
| 593 | 274 ngx_destroy_pool(pool); |
| 589 | 275 return NGX_CONF_ERROR; |
| 276 } | |
| 277 } | |
| 278 | |
|
1253
8ef04207c84f
the "www.example.*" wildcard hash support
Igor Sysoev <igor@sysoev.ru>
parents:
797
diff
changeset
|
279 if (ctx.keys.dns_wc_head.nelts) { |
| 589 | 280 |
|
1253
8ef04207c84f
the "www.example.*" wildcard hash support
Igor Sysoev <igor@sysoev.ru>
parents:
797
diff
changeset
|
281 ngx_qsort(ctx.keys.dns_wc_head.elts, |
|
8ef04207c84f
the "www.example.*" wildcard hash support
Igor Sysoev <igor@sysoev.ru>
parents:
797
diff
changeset
|
282 (size_t) ctx.keys.dns_wc_head.nelts, |
| 589 | 283 sizeof(ngx_hash_key_t), ngx_http_map_cmp_dns_wildcards); |
| 284 | |
| 285 hash.hash = NULL; | |
| 286 hash.temp_pool = pool; | |
| 287 | |
|
1253
8ef04207c84f
the "www.example.*" wildcard hash support
Igor Sysoev <igor@sysoev.ru>
parents:
797
diff
changeset
|
288 if (ngx_hash_wildcard_init(&hash, ctx.keys.dns_wc_head.elts, |
|
8ef04207c84f
the "www.example.*" wildcard hash support
Igor Sysoev <igor@sysoev.ru>
parents:
797
diff
changeset
|
289 ctx.keys.dns_wc_head.nelts) |
| 589 | 290 != NGX_OK) |
| 291 { | |
| 593 | 292 ngx_destroy_pool(pool); |
| 589 | 293 return NGX_CONF_ERROR; |
| 294 } | |
| 295 | |
|
1253
8ef04207c84f
the "www.example.*" wildcard hash support
Igor Sysoev <igor@sysoev.ru>
parents:
797
diff
changeset
|
296 map->hash.wc_head = (ngx_hash_wildcard_t *) hash.hash; |
|
8ef04207c84f
the "www.example.*" wildcard hash support
Igor Sysoev <igor@sysoev.ru>
parents:
797
diff
changeset
|
297 } |
|
8ef04207c84f
the "www.example.*" wildcard hash support
Igor Sysoev <igor@sysoev.ru>
parents:
797
diff
changeset
|
298 |
|
8ef04207c84f
the "www.example.*" wildcard hash support
Igor Sysoev <igor@sysoev.ru>
parents:
797
diff
changeset
|
299 if (ctx.keys.dns_wc_tail.nelts) { |
|
8ef04207c84f
the "www.example.*" wildcard hash support
Igor Sysoev <igor@sysoev.ru>
parents:
797
diff
changeset
|
300 |
|
8ef04207c84f
the "www.example.*" wildcard hash support
Igor Sysoev <igor@sysoev.ru>
parents:
797
diff
changeset
|
301 ngx_qsort(ctx.keys.dns_wc_tail.elts, |
|
8ef04207c84f
the "www.example.*" wildcard hash support
Igor Sysoev <igor@sysoev.ru>
parents:
797
diff
changeset
|
302 (size_t) ctx.keys.dns_wc_tail.nelts, |
|
8ef04207c84f
the "www.example.*" wildcard hash support
Igor Sysoev <igor@sysoev.ru>
parents:
797
diff
changeset
|
303 sizeof(ngx_hash_key_t), ngx_http_map_cmp_dns_wildcards); |
|
8ef04207c84f
the "www.example.*" wildcard hash support
Igor Sysoev <igor@sysoev.ru>
parents:
797
diff
changeset
|
304 |
|
8ef04207c84f
the "www.example.*" wildcard hash support
Igor Sysoev <igor@sysoev.ru>
parents:
797
diff
changeset
|
305 hash.hash = NULL; |
|
8ef04207c84f
the "www.example.*" wildcard hash support
Igor Sysoev <igor@sysoev.ru>
parents:
797
diff
changeset
|
306 hash.temp_pool = pool; |
|
8ef04207c84f
the "www.example.*" wildcard hash support
Igor Sysoev <igor@sysoev.ru>
parents:
797
diff
changeset
|
307 |
|
8ef04207c84f
the "www.example.*" wildcard hash support
Igor Sysoev <igor@sysoev.ru>
parents:
797
diff
changeset
|
308 if (ngx_hash_wildcard_init(&hash, ctx.keys.dns_wc_tail.elts, |
|
8ef04207c84f
the "www.example.*" wildcard hash support
Igor Sysoev <igor@sysoev.ru>
parents:
797
diff
changeset
|
309 ctx.keys.dns_wc_tail.nelts) |
|
8ef04207c84f
the "www.example.*" wildcard hash support
Igor Sysoev <igor@sysoev.ru>
parents:
797
diff
changeset
|
310 != NGX_OK) |
|
8ef04207c84f
the "www.example.*" wildcard hash support
Igor Sysoev <igor@sysoev.ru>
parents:
797
diff
changeset
|
311 { |
|
8ef04207c84f
the "www.example.*" wildcard hash support
Igor Sysoev <igor@sysoev.ru>
parents:
797
diff
changeset
|
312 ngx_destroy_pool(pool); |
|
8ef04207c84f
the "www.example.*" wildcard hash support
Igor Sysoev <igor@sysoev.ru>
parents:
797
diff
changeset
|
313 return NGX_CONF_ERROR; |
|
8ef04207c84f
the "www.example.*" wildcard hash support
Igor Sysoev <igor@sysoev.ru>
parents:
797
diff
changeset
|
314 } |
|
8ef04207c84f
the "www.example.*" wildcard hash support
Igor Sysoev <igor@sysoev.ru>
parents:
797
diff
changeset
|
315 |
|
8ef04207c84f
the "www.example.*" wildcard hash support
Igor Sysoev <igor@sysoev.ru>
parents:
797
diff
changeset
|
316 map->hash.wc_tail = (ngx_hash_wildcard_t *) hash.hash; |
| 589 | 317 } |
| 318 | |
| 319 ngx_destroy_pool(pool); | |
| 320 | |
| 321 return rv; | |
| 322 } | |
| 323 | |
| 324 | |
| 325 static int ngx_libc_cdecl | |
| 326 ngx_http_map_cmp_dns_wildcards(const void *one, const void *two) | |
| 327 { | |
| 328 ngx_hash_key_t *first, *second; | |
| 329 | |
| 330 first = (ngx_hash_key_t *) one; | |
| 331 second = (ngx_hash_key_t *) two; | |
| 332 | |
|
3116
98e288c6dac3
If .domain.com, .sub.domain.com, and .domain-some.com were defined,
Igor Sysoev <igor@sysoev.ru>
parents:
2912
diff
changeset
|
333 return ngx_dns_strcmp(first->key.data, second->key.data); |
| 589 | 334 } |
| 335 | |
| 336 | |
| 337 static char * | |
| 338 ngx_http_map(ngx_conf_t *cf, ngx_command_t *dummy, void *conf) | |
| 339 { | |
| 591 | 340 ngx_int_t rc; |
| 341 ngx_str_t *value, file; | |
|
1253
8ef04207c84f
the "www.example.*" wildcard hash support
Igor Sysoev <igor@sysoev.ru>
parents:
797
diff
changeset
|
342 ngx_uint_t i, key; |
| 589 | 343 ngx_http_map_conf_ctx_t *ctx; |
| 591 | 344 ngx_http_variable_value_t *var, **vp; |
| 589 | 345 |
| 346 ctx = cf->ctx; | |
| 347 | |
| 348 value = cf->args->elts; | |
| 349 | |
| 350 if (cf->args->nelts == 1 | |
| 351 && ngx_strcmp(value[0].data, "hostnames") == 0) | |
| 352 { | |
| 353 ctx->hostnames = 1; | |
| 354 return NGX_CONF_OK; | |
| 355 | |
| 356 } else if (cf->args->nelts != 2) { | |
| 357 ngx_conf_log_error(NGX_LOG_EMERG, cf, 0, | |
| 358 "invalid number of the map parameters"); | |
| 359 return NGX_CONF_ERROR; | |
| 360 | |
| 361 } else if (value[0].len == 0) { | |
| 362 ngx_conf_log_error(NGX_LOG_EMERG, cf, 0, | |
| 363 "invalid first parameter"); | |
| 364 return NGX_CONF_ERROR; | |
| 365 } | |
| 366 | |
| 367 if (ngx_strcmp(value[0].data, "include") == 0) { | |
| 368 file = value[1]; | |
| 369 | |
| 3642 | 370 if (ngx_conf_full_name(cf->cycle, &file, 1) != NGX_OK) { |
| 589 | 371 return NGX_CONF_ERROR; |
| 372 } | |
| 373 | |
| 374 ngx_log_debug1(NGX_LOG_DEBUG_CORE, cf->log, 0, "include %s", file.data); | |
| 375 | |
| 376 return ngx_conf_parse(cf, &file); | |
| 377 } | |
| 378 | |
| 379 key = 0; | |
| 380 | |
| 381 for (i = 0; i < value[1].len; i++) { | |
| 382 key = ngx_hash(key, value[1].data[i]); | |
| 383 } | |
| 384 | |
| 593 | 385 key %= ctx->keys.hsize; |
| 589 | 386 |
| 387 vp = ctx->values_hash[key].elts; | |
| 388 | |
| 389 if (vp) { | |
| 390 for (i = 0; i < ctx->values_hash[key].nelts; i++) { | |
| 391 if (value[1].len != (size_t) vp[i]->len) { | |
| 392 continue; | |
| 393 } | |
| 394 | |
| 395 if (ngx_strncmp(value[1].data, vp[i]->data, value[1].len) == 0) { | |
| 396 var = vp[i]; | |
| 397 goto found; | |
| 398 } | |
| 399 } | |
| 400 | |
| 401 } else { | |
| 402 if (ngx_array_init(&ctx->values_hash[key], cf->pool, 4, | |
| 403 sizeof(ngx_http_variable_value_t *)) | |
| 404 != NGX_OK) | |
| 405 { | |
| 406 return NGX_CONF_ERROR; | |
| 407 } | |
| 408 } | |
| 409 | |
| 593 | 410 var = ngx_palloc(ctx->keys.pool, sizeof(ngx_http_variable_value_t)); |
| 589 | 411 if (var == NULL) { |
| 412 return NGX_CONF_ERROR; | |
| 413 } | |
| 414 | |
| 415 var->len = value[1].len; | |
| 593 | 416 var->data = ngx_pstrdup(ctx->keys.pool, &value[1]); |
| 589 | 417 if (var->data == NULL) { |
| 418 return NGX_CONF_ERROR; | |
| 419 } | |
| 420 | |
| 421 var->valid = 1; | |
| 1565 | 422 var->no_cacheable = 0; |
| 589 | 423 var->not_found = 0; |
| 424 | |
| 425 vp = ngx_array_push(&ctx->values_hash[key]); | |
| 426 if (vp == NULL) { | |
| 427 return NGX_CONF_ERROR; | |
| 428 } | |
| 429 | |
| 430 *vp = var; | |
| 431 | |
| 432 found: | |
| 433 | |
|
1253
8ef04207c84f
the "www.example.*" wildcard hash support
Igor Sysoev <igor@sysoev.ru>
parents:
797
diff
changeset
|
434 if (ngx_strcmp(value[0].data, "default") == 0) { |
| 589 | 435 |
|
1253
8ef04207c84f
the "www.example.*" wildcard hash support
Igor Sysoev <igor@sysoev.ru>
parents:
797
diff
changeset
|
436 if (ctx->default_value) { |
| 589 | 437 ngx_conf_log_error(NGX_LOG_EMERG, cf, 0, |
|
1253
8ef04207c84f
the "www.example.*" wildcard hash support
Igor Sysoev <igor@sysoev.ru>
parents:
797
diff
changeset
|
438 "duplicate default map parameter"); |
| 589 | 439 return NGX_CONF_ERROR; |
| 440 } | |
| 441 | |
|
1253
8ef04207c84f
the "www.example.*" wildcard hash support
Igor Sysoev <igor@sysoev.ru>
parents:
797
diff
changeset
|
442 ctx->default_value = var; |
|
8ef04207c84f
the "www.example.*" wildcard hash support
Igor Sysoev <igor@sysoev.ru>
parents:
797
diff
changeset
|
443 |
|
8ef04207c84f
the "www.example.*" wildcard hash support
Igor Sysoev <igor@sysoev.ru>
parents:
797
diff
changeset
|
444 return NGX_CONF_OK; |
| 589 | 445 } |
| 446 | |
|
1253
8ef04207c84f
the "www.example.*" wildcard hash support
Igor Sysoev <igor@sysoev.ru>
parents:
797
diff
changeset
|
447 if (value[0].len && value[0].data[0] == '!') { |
|
8ef04207c84f
the "www.example.*" wildcard hash support
Igor Sysoev <igor@sysoev.ru>
parents:
797
diff
changeset
|
448 value[0].len--; |
|
8ef04207c84f
the "www.example.*" wildcard hash support
Igor Sysoev <igor@sysoev.ru>
parents:
797
diff
changeset
|
449 value[0].data++; |
|
8ef04207c84f
the "www.example.*" wildcard hash support
Igor Sysoev <igor@sysoev.ru>
parents:
797
diff
changeset
|
450 } |
|
8ef04207c84f
the "www.example.*" wildcard hash support
Igor Sysoev <igor@sysoev.ru>
parents:
797
diff
changeset
|
451 |
|
8ef04207c84f
the "www.example.*" wildcard hash support
Igor Sysoev <igor@sysoev.ru>
parents:
797
diff
changeset
|
452 rc = ngx_hash_add_key(&ctx->keys, &value[0], var, |
|
8ef04207c84f
the "www.example.*" wildcard hash support
Igor Sysoev <igor@sysoev.ru>
parents:
797
diff
changeset
|
453 (ctx->hostnames) ? NGX_HASH_WILDCARD_KEY : 0); |
| 589 | 454 |
| 591 | 455 if (rc == NGX_OK) { |
| 456 return NGX_CONF_OK; | |
| 457 } | |
| 589 | 458 |
|
1253
8ef04207c84f
the "www.example.*" wildcard hash support
Igor Sysoev <igor@sysoev.ru>
parents:
797
diff
changeset
|
459 if (rc == NGX_DECLINED) { |
|
8ef04207c84f
the "www.example.*" wildcard hash support
Igor Sysoev <igor@sysoev.ru>
parents:
797
diff
changeset
|
460 ngx_conf_log_error(NGX_LOG_EMERG, cf, 0, |
|
8ef04207c84f
the "www.example.*" wildcard hash support
Igor Sysoev <igor@sysoev.ru>
parents:
797
diff
changeset
|
461 "invalid hostname or wildcard \"%V\"", &value[0]); |
|
8ef04207c84f
the "www.example.*" wildcard hash support
Igor Sysoev <igor@sysoev.ru>
parents:
797
diff
changeset
|
462 } |
|
8ef04207c84f
the "www.example.*" wildcard hash support
Igor Sysoev <igor@sysoev.ru>
parents:
797
diff
changeset
|
463 |
| 591 | 464 if (rc == NGX_BUSY) { |
| 465 ngx_conf_log_error(NGX_LOG_EMERG, cf, 0, | |
| 466 "conflicting parameter \"%V\"", &value[0]); | |
| 467 } | |
| 589 | 468 |
| 591 | 469 return NGX_CONF_ERROR; |
| 589 | 470 } |