508
|
1
|
|
2 /*
|
|
3 * Copyright (C) Igor Sysoev
|
660
|
4 * Copyright (C) Nginx, Inc.
|
508
|
5 */
|
|
6
|
|
7
|
|
8 #include <ngx_config.h>
|
|
9 #include <ngx_core.h>
|
|
10 #include <ngx_http.h>
|
|
11
|
|
12 #include <GeoIP.h>
|
|
13 #include <GeoIPCity.h>
|
|
14
|
|
15
|
|
16 typedef struct {
|
674
|
17 GeoIP *country;
|
|
18 GeoIP *org;
|
|
19 GeoIP *city;
|
|
20 ngx_array_t *proxies; /* array of ngx_cidr_t */
|
|
21 ngx_flag_t proxy_recursive;
|
508
|
22 } ngx_http_geoip_conf_t;
|
|
23
|
|
24
|
|
25 typedef struct {
|
674
|
26 ngx_str_t *name;
|
|
27 uintptr_t data;
|
508
|
28 } ngx_http_geoip_var_t;
|
|
29
|
|
30
|
|
31 typedef const char *(*ngx_http_geoip_variable_handler_pt)(GeoIP *, u_long addr);
|
|
32
|
674
|
33 static u_long ngx_http_geoip_addr(ngx_http_request_t *r,
|
|
34 ngx_http_geoip_conf_t *gcf);
|
508
|
35 static ngx_int_t ngx_http_geoip_country_variable(ngx_http_request_t *r,
|
|
36 ngx_http_variable_value_t *v, uintptr_t data);
|
626
|
37 static ngx_int_t ngx_http_geoip_org_variable(ngx_http_request_t *r,
|
|
38 ngx_http_variable_value_t *v, uintptr_t data);
|
508
|
39 static ngx_int_t ngx_http_geoip_city_variable(ngx_http_request_t *r,
|
|
40 ngx_http_variable_value_t *v, uintptr_t data);
|
592
|
41 static ngx_int_t ngx_http_geoip_region_name_variable(ngx_http_request_t *r,
|
|
42 ngx_http_variable_value_t *v, uintptr_t data);
|
558
|
43 static ngx_int_t ngx_http_geoip_city_float_variable(ngx_http_request_t *r,
|
|
44 ngx_http_variable_value_t *v, uintptr_t data);
|
592
|
45 static ngx_int_t ngx_http_geoip_city_int_variable(ngx_http_request_t *r,
|
|
46 ngx_http_variable_value_t *v, uintptr_t data);
|
558
|
47 static GeoIPRecord *ngx_http_geoip_get_city_record(ngx_http_request_t *r);
|
508
|
48
|
|
49 static ngx_int_t ngx_http_geoip_add_variables(ngx_conf_t *cf);
|
|
50 static void *ngx_http_geoip_create_conf(ngx_conf_t *cf);
|
674
|
51 static char *ngx_http_geoip_init_conf(ngx_conf_t *cf, void *conf);
|
508
|
52 static char *ngx_http_geoip_country(ngx_conf_t *cf, ngx_command_t *cmd,
|
|
53 void *conf);
|
626
|
54 static char *ngx_http_geoip_org(ngx_conf_t *cf, ngx_command_t *cmd,
|
|
55 void *conf);
|
508
|
56 static char *ngx_http_geoip_city(ngx_conf_t *cf, ngx_command_t *cmd,
|
|
57 void *conf);
|
674
|
58 static char *ngx_http_geoip_proxy(ngx_conf_t *cf, ngx_command_t *cmd,
|
|
59 void *conf);
|
|
60 static ngx_int_t ngx_http_geoip_cidr_value(ngx_conf_t *cf, ngx_str_t *net,
|
|
61 ngx_cidr_t *cidr);
|
508
|
62 static void ngx_http_geoip_cleanup(void *data);
|
|
63
|
|
64
|
|
65 static ngx_command_t ngx_http_geoip_commands[] = {
|
|
66
|
|
67 { ngx_string("geoip_country"),
|
614
|
68 NGX_HTTP_MAIN_CONF|NGX_CONF_TAKE12,
|
508
|
69 ngx_http_geoip_country,
|
|
70 NGX_HTTP_MAIN_CONF_OFFSET,
|
|
71 0,
|
|
72 NULL },
|
|
73
|
626
|
74 { ngx_string("geoip_org"),
|
|
75 NGX_HTTP_MAIN_CONF|NGX_CONF_TAKE12,
|
|
76 ngx_http_geoip_org,
|
|
77 NGX_HTTP_MAIN_CONF_OFFSET,
|
|
78 0,
|
|
79 NULL },
|
|
80
|
508
|
81 { ngx_string("geoip_city"),
|
614
|
82 NGX_HTTP_MAIN_CONF|NGX_CONF_TAKE12,
|
508
|
83 ngx_http_geoip_city,
|
|
84 NGX_HTTP_MAIN_CONF_OFFSET,
|
|
85 0,
|
|
86 NULL },
|
|
87
|
674
|
88 { ngx_string("geoip_proxy"),
|
|
89 NGX_HTTP_MAIN_CONF|NGX_CONF_TAKE1,
|
|
90 ngx_http_geoip_proxy,
|
|
91 NGX_HTTP_MAIN_CONF_OFFSET,
|
|
92 0,
|
|
93 NULL },
|
|
94
|
|
95 { ngx_string("geoip_proxy_recursive"),
|
|
96 NGX_HTTP_MAIN_CONF|NGX_CONF_FLAG,
|
|
97 ngx_conf_set_flag_slot,
|
|
98 NGX_HTTP_MAIN_CONF_OFFSET,
|
|
99 offsetof(ngx_http_geoip_conf_t, proxy_recursive),
|
|
100 NULL },
|
|
101
|
508
|
102 ngx_null_command
|
|
103 };
|
|
104
|
|
105
|
|
106 static ngx_http_module_t ngx_http_geoip_module_ctx = {
|
|
107 ngx_http_geoip_add_variables, /* preconfiguration */
|
|
108 NULL, /* postconfiguration */
|
|
109
|
|
110 ngx_http_geoip_create_conf, /* create main configuration */
|
674
|
111 ngx_http_geoip_init_conf, /* init main configuration */
|
508
|
112
|
|
113 NULL, /* create server configuration */
|
|
114 NULL, /* merge server configuration */
|
|
115
|
|
116 NULL, /* create location configuration */
|
|
117 NULL /* merge location configuration */
|
|
118 };
|
|
119
|
|
120
|
|
121 ngx_module_t ngx_http_geoip_module = {
|
|
122 NGX_MODULE_V1,
|
|
123 &ngx_http_geoip_module_ctx, /* module context */
|
|
124 ngx_http_geoip_commands, /* module directives */
|
|
125 NGX_HTTP_MODULE, /* module type */
|
|
126 NULL, /* init master */
|
|
127 NULL, /* init module */
|
|
128 NULL, /* init process */
|
|
129 NULL, /* init thread */
|
|
130 NULL, /* exit thread */
|
|
131 NULL, /* exit process */
|
|
132 NULL, /* exit master */
|
|
133 NGX_MODULE_V1_PADDING
|
|
134 };
|
|
135
|
|
136
|
|
137 static ngx_http_variable_t ngx_http_geoip_vars[] = {
|
|
138
|
558
|
139 { ngx_string("geoip_country_code"), NULL,
|
|
140 ngx_http_geoip_country_variable,
|
508
|
141 (uintptr_t) GeoIP_country_code_by_ipnum, 0, 0 },
|
|
142
|
558
|
143 { ngx_string("geoip_country_code3"), NULL,
|
|
144 ngx_http_geoip_country_variable,
|
508
|
145 (uintptr_t) GeoIP_country_code3_by_ipnum, 0, 0 },
|
|
146
|
558
|
147 { ngx_string("geoip_country_name"), NULL,
|
|
148 ngx_http_geoip_country_variable,
|
508
|
149 (uintptr_t) GeoIP_country_name_by_ipnum, 0, 0 },
|
|
150
|
626
|
151 { ngx_string("geoip_org"), NULL,
|
|
152 ngx_http_geoip_org_variable,
|
|
153 (uintptr_t) GeoIP_name_by_ipnum, 0, 0 },
|
|
154
|
558
|
155 { ngx_string("geoip_city_continent_code"), NULL,
|
|
156 ngx_http_geoip_city_variable,
|
|
157 offsetof(GeoIPRecord, continent_code), 0, 0 },
|
|
158
|
|
159 { ngx_string("geoip_city_country_code"), NULL,
|
|
160 ngx_http_geoip_city_variable,
|
508
|
161 offsetof(GeoIPRecord, country_code), 0, 0 },
|
|
162
|
|
163 { ngx_string("geoip_city_country_code3"), NULL,
|
|
164 ngx_http_geoip_city_variable,
|
|
165 offsetof(GeoIPRecord, country_code3), 0, 0 },
|
|
166
|
558
|
167 { ngx_string("geoip_city_country_name"), NULL,
|
|
168 ngx_http_geoip_city_variable,
|
508
|
169 offsetof(GeoIPRecord, country_name), 0, 0 },
|
|
170
|
|
171 { ngx_string("geoip_region"), NULL,
|
|
172 ngx_http_geoip_city_variable,
|
|
173 offsetof(GeoIPRecord, region), 0, 0 },
|
|
174
|
592
|
175 { ngx_string("geoip_region_name"), NULL,
|
|
176 ngx_http_geoip_region_name_variable,
|
|
177 0, 0, 0 },
|
|
178
|
508
|
179 { ngx_string("geoip_city"), NULL,
|
|
180 ngx_http_geoip_city_variable,
|
|
181 offsetof(GeoIPRecord, city), 0, 0 },
|
|
182
|
|
183 { ngx_string("geoip_postal_code"), NULL,
|
|
184 ngx_http_geoip_city_variable,
|
|
185 offsetof(GeoIPRecord, postal_code), 0, 0 },
|
|
186
|
558
|
187 { ngx_string("geoip_latitude"), NULL,
|
|
188 ngx_http_geoip_city_float_variable,
|
|
189 offsetof(GeoIPRecord, latitude), 0, 0 },
|
|
190
|
|
191 { ngx_string("geoip_longitude"), NULL,
|
|
192 ngx_http_geoip_city_float_variable,
|
|
193 offsetof(GeoIPRecord, longitude), 0, 0 },
|
|
194
|
592
|
195 { ngx_string("geoip_dma_code"), NULL,
|
|
196 ngx_http_geoip_city_int_variable,
|
|
197 offsetof(GeoIPRecord, dma_code), 0, 0 },
|
|
198
|
|
199 { ngx_string("geoip_area_code"), NULL,
|
|
200 ngx_http_geoip_city_int_variable,
|
|
201 offsetof(GeoIPRecord, area_code), 0, 0 },
|
|
202
|
508
|
203 { ngx_null_string, NULL, NULL, 0, 0, 0 }
|
|
204 };
|
|
205
|
|
206
|
626
|
207 static u_long
|
674
|
208 ngx_http_geoip_addr(ngx_http_request_t *r, ngx_http_geoip_conf_t *gcf)
|
626
|
209 {
|
674
|
210 ngx_addr_t addr;
|
|
211 ngx_table_elt_t *xfwd;
|
|
212 struct sockaddr_in *sin;
|
|
213
|
|
214 addr.sockaddr = r->connection->sockaddr;
|
|
215 addr.socklen = r->connection->socklen;
|
|
216 /* addr.name = r->connection->addr_text; */
|
626
|
217
|
674
|
218 xfwd = r->headers_in.x_forwarded_for;
|
626
|
219
|
674
|
220 if (xfwd != NULL && gcf->proxies != NULL) {
|
|
221 (void) ngx_http_get_forwarded_addr(r, &addr, xfwd->value.data,
|
|
222 xfwd->value.len, gcf->proxies,
|
|
223 gcf->proxy_recursive);
|
|
224 }
|
626
|
225
|
|
226 #if (NGX_HAVE_INET6)
|
|
227
|
674
|
228 if (addr.sockaddr->sa_family == AF_INET6) {
|
|
229 struct in6_addr *inaddr6;
|
|
230
|
|
231 inaddr6 = &((struct sockaddr_in6 *) addr.sockaddr)->sin6_addr;
|
626
|
232
|
674
|
233 if (IN6_IS_ADDR_V4MAPPED(inaddr6)) {
|
|
234 return ntohl(*(in_addr_t *) &inaddr6->s6_addr[12]);
|
626
|
235 }
|
674
|
236 }
|
626
|
237
|
|
238 #endif
|
674
|
239
|
|
240 if (addr.sockaddr->sa_family != AF_INET) {
|
|
241 return INADDR_NONE;
|
626
|
242 }
|
|
243
|
674
|
244 sin = (struct sockaddr_in *) addr.sockaddr;
|
|
245 return ntohl(sin->sin_addr.s_addr);
|
626
|
246 }
|
|
247
|
|
248
|
508
|
249 static ngx_int_t
|
|
250 ngx_http_geoip_country_variable(ngx_http_request_t *r,
|
|
251 ngx_http_variable_value_t *v, uintptr_t data)
|
|
252 {
|
|
253 ngx_http_geoip_variable_handler_pt handler =
|
|
254 (ngx_http_geoip_variable_handler_pt) data;
|
|
255
|
|
256 const char *val;
|
|
257 ngx_http_geoip_conf_t *gcf;
|
|
258
|
|
259 gcf = ngx_http_get_module_main_conf(r, ngx_http_geoip_module);
|
|
260
|
|
261 if (gcf->country == NULL) {
|
|
262 goto not_found;
|
|
263 }
|
|
264
|
674
|
265 val = handler(gcf->country, ngx_http_geoip_addr(r, gcf));
|
626
|
266
|
|
267 if (val == NULL) {
|
508
|
268 goto not_found;
|
|
269 }
|
|
270
|
626
|
271 v->len = ngx_strlen(val);
|
|
272 v->valid = 1;
|
|
273 v->no_cacheable = 0;
|
|
274 v->not_found = 0;
|
|
275 v->data = (u_char *) val;
|
|
276
|
|
277 return NGX_OK;
|
|
278
|
|
279 not_found:
|
|
280
|
|
281 v->not_found = 1;
|
|
282
|
|
283 return NGX_OK;
|
|
284 }
|
|
285
|
508
|
286
|
626
|
287 static ngx_int_t
|
|
288 ngx_http_geoip_org_variable(ngx_http_request_t *r,
|
|
289 ngx_http_variable_value_t *v, uintptr_t data)
|
|
290 {
|
|
291 ngx_http_geoip_variable_handler_pt handler =
|
|
292 (ngx_http_geoip_variable_handler_pt) data;
|
|
293
|
|
294 const char *val;
|
|
295 ngx_http_geoip_conf_t *gcf;
|
|
296
|
|
297 gcf = ngx_http_get_module_main_conf(r, ngx_http_geoip_module);
|
|
298
|
|
299 if (gcf->org == NULL) {
|
|
300 goto not_found;
|
|
301 }
|
|
302
|
674
|
303 val = handler(gcf->org, ngx_http_geoip_addr(r, gcf));
|
508
|
304
|
|
305 if (val == NULL) {
|
|
306 goto not_found;
|
|
307 }
|
|
308
|
|
309 v->len = ngx_strlen(val);
|
|
310 v->valid = 1;
|
|
311 v->no_cacheable = 0;
|
|
312 v->not_found = 0;
|
|
313 v->data = (u_char *) val;
|
|
314
|
|
315 return NGX_OK;
|
|
316
|
|
317 not_found:
|
|
318
|
|
319 v->not_found = 1;
|
|
320
|
|
321 return NGX_OK;
|
|
322 }
|
|
323
|
|
324
|
|
325 static ngx_int_t
|
|
326 ngx_http_geoip_city_variable(ngx_http_request_t *r,
|
|
327 ngx_http_variable_value_t *v, uintptr_t data)
|
|
328 {
|
558
|
329 char *val;
|
|
330 size_t len;
|
|
331 GeoIPRecord *gr;
|
508
|
332
|
558
|
333 gr = ngx_http_geoip_get_city_record(r);
|
508
|
334 if (gr == NULL) {
|
|
335 goto not_found;
|
|
336 }
|
|
337
|
|
338 val = *(char **) ((char *) gr + data);
|
|
339 if (val == NULL) {
|
516
|
340 goto no_value;
|
508
|
341 }
|
|
342
|
516
|
343 len = ngx_strlen(val);
|
|
344 v->data = ngx_pnalloc(r->pool, len);
|
|
345 if (v->data == NULL) {
|
|
346 GeoIPRecord_delete(gr);
|
|
347 return NGX_ERROR;
|
|
348 }
|
|
349
|
|
350 ngx_memcpy(v->data, val, len);
|
|
351
|
|
352 v->len = len;
|
508
|
353 v->valid = 1;
|
|
354 v->no_cacheable = 0;
|
|
355 v->not_found = 0;
|
516
|
356
|
|
357 GeoIPRecord_delete(gr);
|
508
|
358
|
|
359 return NGX_OK;
|
|
360
|
516
|
361 no_value:
|
|
362
|
|
363 GeoIPRecord_delete(gr);
|
|
364
|
508
|
365 not_found:
|
|
366
|
|
367 v->not_found = 1;
|
|
368
|
|
369 return NGX_OK;
|
|
370 }
|
|
371
|
|
372
|
|
373 static ngx_int_t
|
592
|
374 ngx_http_geoip_region_name_variable(ngx_http_request_t *r,
|
|
375 ngx_http_variable_value_t *v, uintptr_t data)
|
|
376 {
|
|
377 size_t len;
|
|
378 const char *val;
|
|
379 GeoIPRecord *gr;
|
|
380
|
|
381 gr = ngx_http_geoip_get_city_record(r);
|
|
382 if (gr == NULL) {
|
|
383 goto not_found;
|
|
384 }
|
|
385
|
|
386 val = GeoIP_region_name_by_code(gr->country_code, gr->region);
|
|
387
|
594
|
388 GeoIPRecord_delete(gr);
|
|
389
|
|
390 if (val == NULL) {
|
|
391 goto not_found;
|
|
392 }
|
|
393
|
592
|
394 len = ngx_strlen(val);
|
|
395 v->data = ngx_pnalloc(r->pool, len);
|
|
396 if (v->data == NULL) {
|
|
397 return NGX_ERROR;
|
|
398 }
|
|
399
|
|
400 ngx_memcpy(v->data, val, len);
|
|
401
|
|
402 v->len = len;
|
|
403 v->valid = 1;
|
|
404 v->no_cacheable = 0;
|
|
405 v->not_found = 0;
|
|
406
|
|
407 return NGX_OK;
|
|
408
|
|
409 not_found:
|
|
410
|
|
411 v->not_found = 1;
|
|
412
|
|
413 return NGX_OK;
|
|
414 }
|
|
415
|
|
416
|
|
417 static ngx_int_t
|
558
|
418 ngx_http_geoip_city_float_variable(ngx_http_request_t *r,
|
|
419 ngx_http_variable_value_t *v, uintptr_t data)
|
|
420 {
|
|
421 float val;
|
|
422 GeoIPRecord *gr;
|
|
423
|
|
424 gr = ngx_http_geoip_get_city_record(r);
|
|
425 if (gr == NULL) {
|
|
426 v->not_found = 1;
|
|
427 return NGX_OK;
|
|
428 }
|
|
429
|
|
430 v->data = ngx_pnalloc(r->pool, NGX_INT64_LEN + 5);
|
|
431 if (v->data == NULL) {
|
|
432 GeoIPRecord_delete(gr);
|
|
433 return NGX_ERROR;
|
|
434 }
|
|
435
|
|
436 val = *(float *) ((char *) gr + data);
|
|
437
|
|
438 v->len = ngx_sprintf(v->data, "%.4f", val) - v->data;
|
|
439
|
|
440 GeoIPRecord_delete(gr);
|
|
441
|
|
442 return NGX_OK;
|
|
443 }
|
|
444
|
|
445
|
592
|
446 static ngx_int_t
|
|
447 ngx_http_geoip_city_int_variable(ngx_http_request_t *r,
|
|
448 ngx_http_variable_value_t *v, uintptr_t data)
|
|
449 {
|
|
450 int val;
|
|
451 GeoIPRecord *gr;
|
|
452
|
|
453 gr = ngx_http_geoip_get_city_record(r);
|
|
454 if (gr == NULL) {
|
|
455 v->not_found = 1;
|
|
456 return NGX_OK;
|
|
457 }
|
|
458
|
|
459 v->data = ngx_pnalloc(r->pool, NGX_INT64_LEN);
|
|
460 if (v->data == NULL) {
|
|
461 GeoIPRecord_delete(gr);
|
|
462 return NGX_ERROR;
|
|
463 }
|
|
464
|
|
465 val = *(int *) ((char *) gr + data);
|
|
466
|
|
467 v->len = ngx_sprintf(v->data, "%d", val) - v->data;
|
|
468
|
|
469 GeoIPRecord_delete(gr);
|
|
470
|
|
471 return NGX_OK;
|
|
472 }
|
|
473
|
|
474
|
558
|
475 static GeoIPRecord *
|
|
476 ngx_http_geoip_get_city_record(ngx_http_request_t *r)
|
|
477 {
|
|
478 ngx_http_geoip_conf_t *gcf;
|
|
479
|
|
480 gcf = ngx_http_get_module_main_conf(r, ngx_http_geoip_module);
|
|
481
|
626
|
482 if (gcf->city) {
|
674
|
483 return GeoIP_record_by_ipnum(gcf->city, ngx_http_geoip_addr(r, gcf));
|
558
|
484 }
|
|
485
|
|
486 return NULL;
|
|
487 }
|
|
488
|
|
489
|
|
490 static ngx_int_t
|
508
|
491 ngx_http_geoip_add_variables(ngx_conf_t *cf)
|
|
492 {
|
|
493 ngx_http_variable_t *var, *v;
|
|
494
|
|
495 for (v = ngx_http_geoip_vars; v->name.len; v++) {
|
|
496 var = ngx_http_add_variable(cf, &v->name, v->flags);
|
|
497 if (var == NULL) {
|
|
498 return NGX_ERROR;
|
|
499 }
|
|
500
|
|
501 var->get_handler = v->get_handler;
|
|
502 var->data = v->data;
|
|
503 }
|
|
504
|
|
505 return NGX_OK;
|
|
506 }
|
|
507
|
|
508
|
|
509 static void *
|
|
510 ngx_http_geoip_create_conf(ngx_conf_t *cf)
|
|
511 {
|
|
512 ngx_pool_cleanup_t *cln;
|
|
513 ngx_http_geoip_conf_t *conf;
|
|
514
|
|
515 conf = ngx_pcalloc(cf->pool, sizeof(ngx_http_geoip_conf_t));
|
|
516 if (conf == NULL) {
|
|
517 return NULL;
|
|
518 }
|
|
519
|
674
|
520 conf->proxy_recursive = NGX_CONF_UNSET;
|
|
521
|
508
|
522 cln = ngx_pool_cleanup_add(cf->pool, 0);
|
|
523 if (cln == NULL) {
|
|
524 return NULL;
|
|
525 }
|
|
526
|
|
527 cln->handler = ngx_http_geoip_cleanup;
|
|
528 cln->data = conf;
|
|
529
|
|
530 return conf;
|
|
531 }
|
|
532
|
|
533
|
|
534 static char *
|
674
|
535 ngx_http_geoip_init_conf(ngx_conf_t *cf, void *conf)
|
|
536 {
|
|
537 ngx_http_geoip_conf_t *gcf = conf;
|
|
538
|
|
539 ngx_conf_init_value(gcf->proxy_recursive, 0);
|
|
540
|
|
541 return NGX_CONF_OK;
|
|
542 }
|
|
543
|
|
544
|
|
545 static char *
|
508
|
546 ngx_http_geoip_country(ngx_conf_t *cf, ngx_command_t *cmd, void *conf)
|
|
547 {
|
|
548 ngx_http_geoip_conf_t *gcf = conf;
|
|
549
|
|
550 ngx_str_t *value;
|
|
551
|
|
552 if (gcf->country) {
|
|
553 return "is duplicate";
|
|
554 }
|
|
555
|
|
556 value = cf->args->elts;
|
|
557
|
|
558 gcf->country = GeoIP_open((char *) value[1].data, GEOIP_MEMORY_CACHE);
|
|
559
|
|
560 if (gcf->country == NULL) {
|
|
561 ngx_conf_log_error(NGX_LOG_EMERG, cf, 0,
|
626
|
562 "GeoIP_open(\"%V\") failed", &value[1]);
|
508
|
563
|
|
564 return NGX_CONF_ERROR;
|
|
565 }
|
|
566
|
614
|
567 if (cf->args->nelts == 3) {
|
|
568 if (ngx_strcmp(value[2].data, "utf8") == 0) {
|
|
569 GeoIP_set_charset (gcf->country, GEOIP_CHARSET_UTF8);
|
|
570
|
|
571 } else {
|
|
572 ngx_conf_log_error(NGX_LOG_EMERG, cf, 0,
|
|
573 "invalid parameter \"%V\"", &value[2]);
|
|
574 return NGX_CONF_ERROR;
|
|
575 }
|
|
576 }
|
|
577
|
508
|
578 switch (gcf->country->databaseType) {
|
|
579
|
|
580 case GEOIP_COUNTRY_EDITION:
|
|
581 case GEOIP_PROXY_EDITION:
|
|
582 case GEOIP_NETSPEED_EDITION:
|
|
583
|
|
584 return NGX_CONF_OK;
|
|
585
|
|
586 default:
|
|
587 ngx_conf_log_error(NGX_LOG_EMERG, cf, 0,
|
|
588 "invalid GeoIP database \"%V\" type:%d",
|
|
589 &value[1], gcf->country->databaseType);
|
|
590 return NGX_CONF_ERROR;
|
|
591 }
|
|
592 }
|
|
593
|
|
594
|
|
595 static char *
|
626
|
596 ngx_http_geoip_org(ngx_conf_t *cf, ngx_command_t *cmd, void *conf)
|
|
597 {
|
|
598 ngx_http_geoip_conf_t *gcf = conf;
|
|
599
|
|
600 ngx_str_t *value;
|
|
601
|
|
602 if (gcf->org) {
|
|
603 return "is duplicate";
|
|
604 }
|
|
605
|
|
606 value = cf->args->elts;
|
|
607
|
|
608 gcf->org = GeoIP_open((char *) value[1].data, GEOIP_MEMORY_CACHE);
|
|
609
|
|
610 if (gcf->org == NULL) {
|
|
611 ngx_conf_log_error(NGX_LOG_EMERG, cf, 0,
|
|
612 "GeoIP_open(\"%V\") failed", &value[1]);
|
|
613
|
|
614 return NGX_CONF_ERROR;
|
|
615 }
|
|
616
|
|
617 if (cf->args->nelts == 3) {
|
|
618 if (ngx_strcmp(value[2].data, "utf8") == 0) {
|
|
619 GeoIP_set_charset (gcf->org, GEOIP_CHARSET_UTF8);
|
|
620
|
|
621 } else {
|
|
622 ngx_conf_log_error(NGX_LOG_EMERG, cf, 0,
|
|
623 "invalid parameter \"%V\"", &value[2]);
|
|
624 return NGX_CONF_ERROR;
|
|
625 }
|
|
626 }
|
|
627
|
|
628 switch (gcf->org->databaseType) {
|
|
629
|
|
630 case GEOIP_ISP_EDITION:
|
|
631 case GEOIP_ORG_EDITION:
|
|
632 case GEOIP_DOMAIN_EDITION:
|
|
633 case GEOIP_ASNUM_EDITION:
|
|
634
|
|
635 return NGX_CONF_OK;
|
|
636
|
|
637 default:
|
|
638 ngx_conf_log_error(NGX_LOG_EMERG, cf, 0,
|
|
639 "invalid GeoIP database \"%V\" type:%d",
|
|
640 &value[1], gcf->org->databaseType);
|
|
641 return NGX_CONF_ERROR;
|
|
642 }
|
|
643 }
|
|
644
|
|
645
|
|
646 static char *
|
508
|
647 ngx_http_geoip_city(ngx_conf_t *cf, ngx_command_t *cmd, void *conf)
|
|
648 {
|
|
649 ngx_http_geoip_conf_t *gcf = conf;
|
|
650
|
|
651 ngx_str_t *value;
|
|
652
|
|
653 if (gcf->city) {
|
|
654 return "is duplicate";
|
|
655 }
|
|
656
|
|
657 value = cf->args->elts;
|
|
658
|
|
659 gcf->city = GeoIP_open((char *) value[1].data, GEOIP_MEMORY_CACHE);
|
|
660
|
|
661 if (gcf->city == NULL) {
|
|
662 ngx_conf_log_error(NGX_LOG_EMERG, cf, 0,
|
626
|
663 "GeoIP_open(\"%V\") failed", &value[1]);
|
508
|
664
|
|
665 return NGX_CONF_ERROR;
|
|
666 }
|
|
667
|
614
|
668 if (cf->args->nelts == 3) {
|
|
669 if (ngx_strcmp(value[2].data, "utf8") == 0) {
|
|
670 GeoIP_set_charset (gcf->city, GEOIP_CHARSET_UTF8);
|
|
671
|
|
672 } else {
|
|
673 ngx_conf_log_error(NGX_LOG_EMERG, cf, 0,
|
|
674 "invalid parameter \"%V\"", &value[2]);
|
|
675 return NGX_CONF_ERROR;
|
|
676 }
|
|
677 }
|
|
678
|
508
|
679 switch (gcf->city->databaseType) {
|
|
680
|
|
681 case GEOIP_CITY_EDITION_REV0:
|
|
682 case GEOIP_CITY_EDITION_REV1:
|
|
683
|
|
684 return NGX_CONF_OK;
|
|
685
|
|
686 default:
|
|
687 ngx_conf_log_error(NGX_LOG_EMERG, cf, 0,
|
|
688 "invalid GeoIP City database \"%V\" type:%d",
|
|
689 &value[1], gcf->city->databaseType);
|
|
690 return NGX_CONF_ERROR;
|
|
691 }
|
|
692 }
|
|
693
|
|
694
|
674
|
695 static char *
|
|
696 ngx_http_geoip_proxy(ngx_conf_t *cf, ngx_command_t *cmd, void *conf)
|
|
697 {
|
|
698 ngx_http_geoip_conf_t *gcf = conf;
|
|
699
|
|
700 ngx_str_t *value;
|
|
701 ngx_cidr_t cidr, *c;
|
|
702
|
|
703 value = cf->args->elts;
|
|
704
|
|
705 if (ngx_http_geoip_cidr_value(cf, &value[1], &cidr) != NGX_OK) {
|
|
706 return NGX_CONF_ERROR;
|
|
707 }
|
|
708
|
|
709 if (gcf->proxies == NULL) {
|
|
710 gcf->proxies = ngx_array_create(cf->pool, 4, sizeof(ngx_cidr_t));
|
|
711 if (gcf->proxies == NULL) {
|
|
712 return NGX_CONF_ERROR;
|
|
713 }
|
|
714 }
|
|
715
|
|
716 c = ngx_array_push(gcf->proxies);
|
|
717 if (c == NULL) {
|
|
718 return NGX_CONF_ERROR;
|
|
719 }
|
|
720
|
|
721 *c = cidr;
|
|
722
|
|
723 return NGX_CONF_OK;
|
|
724 }
|
|
725
|
|
726 static ngx_int_t
|
|
727 ngx_http_geoip_cidr_value(ngx_conf_t *cf, ngx_str_t *net, ngx_cidr_t *cidr)
|
|
728 {
|
|
729 ngx_int_t rc;
|
|
730
|
|
731 if (ngx_strcmp(net->data, "255.255.255.255") == 0) {
|
|
732 cidr->family = AF_INET;
|
|
733 cidr->u.in.addr = 0xffffffff;
|
|
734 cidr->u.in.mask = 0xffffffff;
|
|
735
|
|
736 return NGX_OK;
|
|
737 }
|
|
738
|
|
739 rc = ngx_ptocidr(net, cidr);
|
|
740
|
|
741 if (rc == NGX_ERROR) {
|
|
742 ngx_conf_log_error(NGX_LOG_EMERG, cf, 0, "invalid network \"%V\"", net);
|
|
743 return NGX_ERROR;
|
|
744 }
|
|
745
|
|
746 if (rc == NGX_DONE) {
|
|
747 ngx_conf_log_error(NGX_LOG_WARN, cf, 0,
|
|
748 "low address bits of %V are meaningless", net);
|
|
749 }
|
|
750
|
|
751 return NGX_OK;
|
|
752 }
|
|
753
|
|
754
|
508
|
755 static void
|
|
756 ngx_http_geoip_cleanup(void *data)
|
|
757 {
|
|
758 ngx_http_geoip_conf_t *gcf = data;
|
|
759
|
|
760 if (gcf->country) {
|
|
761 GeoIP_delete(gcf->country);
|
|
762 }
|
|
763
|
626
|
764 if (gcf->org) {
|
|
765 GeoIP_delete(gcf->org);
|
|
766 }
|
|
767
|
508
|
768 if (gcf->city) {
|
|
769 GeoIP_delete(gcf->city);
|
|
770 }
|
|
771 }
|