Mercurial > hg > nginx
annotate src/http/modules/ngx_http_geo_module.c @ 2334:6e30d64f919b
use value rbtree instead of array in geo configuration
| author | Igor Sysoev <igor@sysoev.ru> |
|---|---|
| date | Sat, 22 Nov 2008 20:42:51 +0000 |
| parents | 4c43e25d11ea |
| children | 22d2b6609853 |
| rev | line source |
|---|---|
| 485 | 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 { | |
|
2334
6e30d64f919b
use value rbtree instead of array in geo configuration
Igor Sysoev <igor@sysoev.ru>
parents:
1565
diff
changeset
|
13 ngx_radix_tree_t *tree; |
|
6e30d64f919b
use value rbtree instead of array in geo configuration
Igor Sysoev <igor@sysoev.ru>
parents:
1565
diff
changeset
|
14 ngx_rbtree_t rbtree; |
|
6e30d64f919b
use value rbtree instead of array in geo configuration
Igor Sysoev <igor@sysoev.ru>
parents:
1565
diff
changeset
|
15 ngx_rbtree_node_t sentinel; |
|
6e30d64f919b
use value rbtree instead of array in geo configuration
Igor Sysoev <igor@sysoev.ru>
parents:
1565
diff
changeset
|
16 ngx_pool_t *pool; |
| 589 | 17 } ngx_http_geo_conf_ctx_t; |
| 485 | 18 |
| 19 | |
| 20 static char *ngx_http_geo_block(ngx_conf_t *cf, ngx_command_t *cmd, void *conf); | |
| 21 static char *ngx_http_geo(ngx_conf_t *cf, ngx_command_t *dummy, void *conf); | |
| 22 | |
| 23 | |
| 24 static ngx_command_t ngx_http_geo_commands[] = { | |
| 25 | |
| 26 { ngx_string("geo"), | |
| 27 NGX_HTTP_MAIN_CONF|NGX_CONF_BLOCK|NGX_CONF_TAKE1, | |
| 28 ngx_http_geo_block, | |
| 29 NGX_HTTP_MAIN_CONF_OFFSET, | |
| 30 0, | |
| 31 NULL }, | |
| 32 | |
| 33 ngx_null_command | |
| 34 }; | |
| 35 | |
| 36 | |
| 37 static ngx_http_module_t ngx_http_geo_module_ctx = { | |
| 509 | 38 NULL, /* preconfiguration */ |
| 39 NULL, /* postconfiguration */ | |
| 485 | 40 |
| 41 NULL, /* create main configuration */ | |
| 42 NULL, /* init main configuration */ | |
| 43 | |
| 44 NULL, /* create server configuration */ | |
| 45 NULL, /* merge server configuration */ | |
| 46 | |
| 47 NULL, /* create location configuration */ | |
| 48 NULL /* merge location configuration */ | |
| 49 }; | |
| 50 | |
| 51 | |
| 52 ngx_module_t ngx_http_geo_module = { | |
| 509 | 53 NGX_MODULE_V1, |
| 485 | 54 &ngx_http_geo_module_ctx, /* module context */ |
| 55 ngx_http_geo_commands, /* module directives */ | |
| 56 NGX_HTTP_MODULE, /* module type */ | |
| 541 | 57 NULL, /* init master */ |
| 485 | 58 NULL, /* init module */ |
| 541 | 59 NULL, /* init process */ |
| 60 NULL, /* init thread */ | |
| 61 NULL, /* exit thread */ | |
| 62 NULL, /* exit process */ | |
| 63 NULL, /* exit master */ | |
| 64 NGX_MODULE_V1_PADDING | |
| 485 | 65 }; |
| 66 | |
| 67 | |
| 68 /* AF_INET only */ | |
| 69 | |
| 573 | 70 static ngx_int_t |
| 71 ngx_http_geo_variable(ngx_http_request_t *r, ngx_http_variable_value_t *v, | |
| 72 uintptr_t data) | |
| 485 | 73 { |
| 501 | 74 ngx_radix_tree_t *tree = (ngx_radix_tree_t *) data; |
| 485 | 75 |
| 489 | 76 struct sockaddr_in *sin; |
| 573 | 77 ngx_http_variable_value_t *vv; |
| 485 | 78 |
| 79 sin = (struct sockaddr_in *) r->connection->sockaddr; | |
| 80 | |
| 489 | 81 ngx_log_debug0(NGX_LOG_DEBUG_HTTP, r->connection->log, 0, |
| 82 "http geo started"); | |
| 83 | |
| 573 | 84 vv = (ngx_http_variable_value_t *) |
| 485 | 85 ngx_radix32tree_find(tree, ntohl(sin->sin_addr.s_addr)); |
| 489 | 86 |
| 573 | 87 *v = *vv; |
| 88 | |
| 489 | 89 ngx_log_debug2(NGX_LOG_DEBUG_HTTP, r->connection->log, 0, |
|
1406
03341711f9a2
use %v for ngx_variable_value_t in ngx_sprintf(),
Igor Sysoev <igor@sysoev.ru>
parents:
1380
diff
changeset
|
90 "http geo: %V %v", &r->connection->addr_text, v); |
| 489 | 91 |
| 573 | 92 return NGX_OK; |
| 485 | 93 } |
| 94 | |
| 95 | |
| 489 | 96 static char * |
| 97 ngx_http_geo_block(ngx_conf_t *cf, ngx_command_t *cmd, void *conf) | |
| 485 | 98 { |
| 589 | 99 char *rv; |
| 100 ngx_str_t *value, name; | |
| 101 ngx_conf_t save; | |
| 102 ngx_pool_t *pool; | |
| 103 ngx_radix_tree_t *tree; | |
|
1406
03341711f9a2
use %v for ngx_variable_value_t in ngx_sprintf(),
Igor Sysoev <igor@sysoev.ru>
parents:
1380
diff
changeset
|
104 ngx_http_variable_t *var; |
| 589 | 105 ngx_http_geo_conf_ctx_t ctx; |
| 485 | 106 |
| 501 | 107 value = cf->args->elts; |
| 108 | |
| 109 name = value[1]; | |
| 110 | |
| 111 if (name.data[0] != '$') { | |
| 112 ngx_conf_log_error(NGX_LOG_WARN, cf, 0, | |
| 113 "\"%V\" variable name should start with '$'", | |
| 114 &value[1]); | |
| 115 } else { | |
| 116 name.len--; | |
| 117 name.data++; | |
| 118 } | |
| 119 | |
| 1565 | 120 var = ngx_http_add_variable(cf, &name, NGX_HTTP_VAR_CHANGEABLE); |
| 501 | 121 if (var == NULL) { |
| 485 | 122 return NGX_CONF_ERROR; |
| 123 } | |
| 124 | |
| 501 | 125 tree = ngx_radix_tree_create(cf->pool, -1); |
| 563 | 126 |
| 501 | 127 if (tree == NULL) { |
| 485 | 128 return NGX_CONF_ERROR; |
| 129 } | |
| 130 | |
| 637 | 131 var->get_handler = ngx_http_geo_variable; |
| 501 | 132 var->data = (uintptr_t) tree; |
| 485 | 133 |
| 583 | 134 pool = ngx_create_pool(16384, cf->log); |
| 501 | 135 if (pool == NULL) { |
| 485 | 136 return NGX_CONF_ERROR; |
| 137 } | |
| 138 | |
|
2334
6e30d64f919b
use value rbtree instead of array in geo configuration
Igor Sysoev <igor@sysoev.ru>
parents:
1565
diff
changeset
|
139 ngx_rbtree_init(&ctx.rbtree, &ctx.sentinel, |
|
6e30d64f919b
use value rbtree instead of array in geo configuration
Igor Sysoev <igor@sysoev.ru>
parents:
1565
diff
changeset
|
140 ngx_http_variable_value_rbtree_insert); |
| 485 | 141 |
| 589 | 142 ctx.tree = tree; |
| 143 ctx.pool = cf->pool; | |
| 485 | 144 |
| 145 save = *cf; | |
| 146 cf->pool = pool; | |
| 589 | 147 cf->ctx = &ctx; |
| 485 | 148 cf->handler = ngx_http_geo; |
| 149 cf->handler_conf = conf; | |
| 150 | |
| 151 rv = ngx_conf_parse(cf, NULL); | |
| 152 | |
| 153 *cf = save; | |
| 154 | |
| 155 ngx_destroy_pool(pool); | |
| 156 | |
| 157 if (ngx_radix32tree_find(tree, 0) != NGX_RADIX_NO_VALUE) { | |
| 158 return rv; | |
| 159 } | |
| 160 | |
| 161 if (ngx_radix32tree_insert(tree, 0, 0, | |
| 577 | 162 (uintptr_t) &ngx_http_variable_null_value) |
| 163 == NGX_ERROR) | |
| 485 | 164 { |
| 165 return NGX_CONF_ERROR; | |
| 166 } | |
| 167 | |
| 168 return rv; | |
| 169 } | |
| 170 | |
| 171 | |
| 172 /* AF_INET only */ | |
| 173 | |
| 489 | 174 static char * |
| 175 ngx_http_geo(ngx_conf_t *cf, ngx_command_t *dummy, void *conf) | |
| 485 | 176 { |
|
2334
6e30d64f919b
use value rbtree instead of array in geo configuration
Igor Sysoev <igor@sysoev.ru>
parents:
1565
diff
changeset
|
177 uint32_t hash; |
|
6e30d64f919b
use value rbtree instead of array in geo configuration
Igor Sysoev <igor@sysoev.ru>
parents:
1565
diff
changeset
|
178 ngx_int_t rc; |
|
6e30d64f919b
use value rbtree instead of array in geo configuration
Igor Sysoev <igor@sysoev.ru>
parents:
1565
diff
changeset
|
179 ngx_str_t *value, file; |
|
6e30d64f919b
use value rbtree instead of array in geo configuration
Igor Sysoev <igor@sysoev.ru>
parents:
1565
diff
changeset
|
180 ngx_uint_t i; |
|
6e30d64f919b
use value rbtree instead of array in geo configuration
Igor Sysoev <igor@sysoev.ru>
parents:
1565
diff
changeset
|
181 ngx_inet_cidr_t cidrin; |
|
6e30d64f919b
use value rbtree instead of array in geo configuration
Igor Sysoev <igor@sysoev.ru>
parents:
1565
diff
changeset
|
182 ngx_http_geo_conf_ctx_t *ctx; |
|
6e30d64f919b
use value rbtree instead of array in geo configuration
Igor Sysoev <igor@sysoev.ru>
parents:
1565
diff
changeset
|
183 ngx_http_variable_value_t *val, *old; |
|
6e30d64f919b
use value rbtree instead of array in geo configuration
Igor Sysoev <igor@sysoev.ru>
parents:
1565
diff
changeset
|
184 ngx_http_variable_value_node_t *vvn; |
| 485 | 185 |
| 589 | 186 ctx = cf->ctx; |
| 485 | 187 |
| 188 if (cf->args->nelts != 2) { | |
| 189 ngx_conf_log_error(NGX_LOG_EMERG, cf, 0, | |
| 190 "invalid number of the geo parameters"); | |
| 191 return NGX_CONF_ERROR; | |
| 192 } | |
| 193 | |
| 194 value = cf->args->elts; | |
| 195 | |
| 196 if (ngx_strcmp(value[0].data, "include") == 0) { | |
| 197 file = value[1]; | |
| 198 | |
| 1352 | 199 if (ngx_conf_full_name(cf->cycle, &file, 1) == NGX_ERROR){ |
| 485 | 200 return NGX_CONF_ERROR; |
| 201 } | |
| 202 | |
| 203 ngx_log_debug1(NGX_LOG_DEBUG_CORE, cf->log, 0, "include %s", file.data); | |
| 204 | |
| 205 return ngx_conf_parse(cf, &file); | |
| 206 } | |
| 207 | |
| 208 if (ngx_strcmp(value[0].data, "default") == 0) { | |
| 209 cidrin.addr = 0; | |
| 210 cidrin.mask = 0; | |
| 211 | |
| 212 } else { | |
|
1380
b590a528fd41
ignore meaningless bits in CIDR and warn about them
Igor Sysoev <igor@sysoev.ru>
parents:
1352
diff
changeset
|
213 rc = ngx_ptocidr(&value[0], &cidrin); |
|
b590a528fd41
ignore meaningless bits in CIDR and warn about them
Igor Sysoev <igor@sysoev.ru>
parents:
1352
diff
changeset
|
214 |
|
b590a528fd41
ignore meaningless bits in CIDR and warn about them
Igor Sysoev <igor@sysoev.ru>
parents:
1352
diff
changeset
|
215 if (rc == NGX_ERROR) { |
| 485 | 216 ngx_conf_log_error(NGX_LOG_EMERG, cf, 0, |
| 217 "invalid parameter \"%V\"", &value[0]); | |
| 218 return NGX_CONF_ERROR; | |
| 219 } | |
| 220 | |
|
1380
b590a528fd41
ignore meaningless bits in CIDR and warn about them
Igor Sysoev <igor@sysoev.ru>
parents:
1352
diff
changeset
|
221 if (rc == NGX_DONE) { |
|
b590a528fd41
ignore meaningless bits in CIDR and warn about them
Igor Sysoev <igor@sysoev.ru>
parents:
1352
diff
changeset
|
222 ngx_conf_log_error(NGX_LOG_WARN, cf, 0, |
|
b590a528fd41
ignore meaningless bits in CIDR and warn about them
Igor Sysoev <igor@sysoev.ru>
parents:
1352
diff
changeset
|
223 "low address bits of %V are meaningless", |
|
b590a528fd41
ignore meaningless bits in CIDR and warn about them
Igor Sysoev <igor@sysoev.ru>
parents:
1352
diff
changeset
|
224 &value[0]); |
|
b590a528fd41
ignore meaningless bits in CIDR and warn about them
Igor Sysoev <igor@sysoev.ru>
parents:
1352
diff
changeset
|
225 } |
|
b590a528fd41
ignore meaningless bits in CIDR and warn about them
Igor Sysoev <igor@sysoev.ru>
parents:
1352
diff
changeset
|
226 |
| 485 | 227 cidrin.addr = ntohl(cidrin.addr); |
| 228 cidrin.mask = ntohl(cidrin.mask); | |
| 229 } | |
| 230 | |
|
2334
6e30d64f919b
use value rbtree instead of array in geo configuration
Igor Sysoev <igor@sysoev.ru>
parents:
1565
diff
changeset
|
231 hash = ngx_crc32_long(value[1].data, value[1].len); |
| 485 | 232 |
|
2334
6e30d64f919b
use value rbtree instead of array in geo configuration
Igor Sysoev <igor@sysoev.ru>
parents:
1565
diff
changeset
|
233 val = ngx_http_variable_value_lookup(&ctx->rbtree, &value[1], hash); |
| 485 | 234 |
|
2334
6e30d64f919b
use value rbtree instead of array in geo configuration
Igor Sysoev <igor@sysoev.ru>
parents:
1565
diff
changeset
|
235 if (val == NULL) { |
|
6e30d64f919b
use value rbtree instead of array in geo configuration
Igor Sysoev <igor@sysoev.ru>
parents:
1565
diff
changeset
|
236 val = ngx_palloc(ctx->pool, sizeof(ngx_http_variable_value_t)); |
|
6e30d64f919b
use value rbtree instead of array in geo configuration
Igor Sysoev <igor@sysoev.ru>
parents:
1565
diff
changeset
|
237 if (val == NULL) { |
| 485 | 238 return NGX_CONF_ERROR; |
| 239 } | |
| 240 | |
|
2334
6e30d64f919b
use value rbtree instead of array in geo configuration
Igor Sysoev <igor@sysoev.ru>
parents:
1565
diff
changeset
|
241 val->len = value[1].len; |
|
6e30d64f919b
use value rbtree instead of array in geo configuration
Igor Sysoev <igor@sysoev.ru>
parents:
1565
diff
changeset
|
242 val->data = ngx_pstrdup(ctx->pool, &value[1]); |
|
6e30d64f919b
use value rbtree instead of array in geo configuration
Igor Sysoev <igor@sysoev.ru>
parents:
1565
diff
changeset
|
243 if (val->data == NULL) { |
| 485 | 244 return NGX_CONF_ERROR; |
| 245 } | |
| 246 | |
|
2334
6e30d64f919b
use value rbtree instead of array in geo configuration
Igor Sysoev <igor@sysoev.ru>
parents:
1565
diff
changeset
|
247 val->valid = 1; |
|
6e30d64f919b
use value rbtree instead of array in geo configuration
Igor Sysoev <igor@sysoev.ru>
parents:
1565
diff
changeset
|
248 val->no_cacheable = 0; |
|
6e30d64f919b
use value rbtree instead of array in geo configuration
Igor Sysoev <igor@sysoev.ru>
parents:
1565
diff
changeset
|
249 val->not_found = 0; |
| 485 | 250 |
|
2334
6e30d64f919b
use value rbtree instead of array in geo configuration
Igor Sysoev <igor@sysoev.ru>
parents:
1565
diff
changeset
|
251 vvn = ngx_palloc(cf->pool, sizeof(ngx_http_variable_value_node_t)); |
|
6e30d64f919b
use value rbtree instead of array in geo configuration
Igor Sysoev <igor@sysoev.ru>
parents:
1565
diff
changeset
|
252 if (vvn == NULL) { |
| 485 | 253 return NGX_CONF_ERROR; |
| 254 } | |
| 255 | |
|
2334
6e30d64f919b
use value rbtree instead of array in geo configuration
Igor Sysoev <igor@sysoev.ru>
parents:
1565
diff
changeset
|
256 vvn->node.key = hash; |
|
6e30d64f919b
use value rbtree instead of array in geo configuration
Igor Sysoev <igor@sysoev.ru>
parents:
1565
diff
changeset
|
257 vvn->len = val->len; |
|
6e30d64f919b
use value rbtree instead of array in geo configuration
Igor Sysoev <igor@sysoev.ru>
parents:
1565
diff
changeset
|
258 vvn->value = val; |
|
6e30d64f919b
use value rbtree instead of array in geo configuration
Igor Sysoev <igor@sysoev.ru>
parents:
1565
diff
changeset
|
259 |
|
6e30d64f919b
use value rbtree instead of array in geo configuration
Igor Sysoev <igor@sysoev.ru>
parents:
1565
diff
changeset
|
260 ngx_rbtree_insert(&ctx->rbtree, &vvn->node); |
| 485 | 261 } |
| 262 | |
| 553 | 263 for (i = 2; i; i--) { |
| 589 | 264 rc = ngx_radix32tree_insert(ctx->tree, cidrin.addr, cidrin.mask, |
|
2334
6e30d64f919b
use value rbtree instead of array in geo configuration
Igor Sysoev <igor@sysoev.ru>
parents:
1565
diff
changeset
|
265 (uintptr_t) val); |
| 553 | 266 if (rc == NGX_OK) { |
| 267 return NGX_CONF_OK; | |
| 268 } | |
| 269 | |
| 270 if (rc == NGX_ERROR) { | |
| 271 return NGX_CONF_ERROR; | |
| 272 } | |
| 273 | |
| 274 /* rc == NGX_BUSY */ | |
| 275 | |
| 276 old = (ngx_http_variable_value_t *) | |
| 589 | 277 ngx_radix32tree_find(ctx->tree, cidrin.addr & cidrin.mask); |
| 553 | 278 |
| 279 ngx_conf_log_error(NGX_LOG_WARN, cf, 0, | |
|
1406
03341711f9a2
use %v for ngx_variable_value_t in ngx_sprintf(),
Igor Sysoev <igor@sysoev.ru>
parents:
1380
diff
changeset
|
280 "duplicate parameter \"%V\", value: \"%v\", old value: \"%v\"", |
|
2334
6e30d64f919b
use value rbtree instead of array in geo configuration
Igor Sysoev <igor@sysoev.ru>
parents:
1565
diff
changeset
|
281 &value[0], val, old); |
| 553 | 282 |
| 589 | 283 rc = ngx_radix32tree_delete(ctx->tree, cidrin.addr, cidrin.mask); |
| 553 | 284 |
| 285 if (rc == NGX_ERROR) { | |
| 286 return NGX_CONF_ERROR; | |
| 287 } | |
| 485 | 288 } |
| 289 | |
| 553 | 290 return NGX_CONF_ERROR; |
| 485 | 291 } |