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 } |