annotate src/http/modules/ngx_http_limit_zone_module.c @ 644:ad25218fd14b NGINX_1_0_12

nginx 1.0.12 *) Feature: the "TLSv1.1" and "TLSv1.2" parameters of the "ssl_protocols" directive. *) Feature: the "if" SSI command supports captures in regular expressions. *) Bugfix: the "if" SSI command did not work inside the "block" command. *) Bugfix: in AIO error handling on FreeBSD. *) Bugfix: in the OpenSSL library initialization. *) Bugfix: the "worker_cpu_affinity" directive might not work. *) Bugfix: the "limit_conn_log_level" and "limit_req_log_level" directives might not work. *) Bugfix: the "read_ahead" directive might not work combined with "try_files" and "open_file_cache". *) Bugfix: the "proxy_cache_use_stale" directive with "error" parameter did not return answer from cache if there were no live upstreams. *) Bugfix: a segmentation fault might occur in a worker process if small time was used in the "inactive" parameter of the "proxy_cache_path" directive. *) Bugfix: responses from cache might hang. *) Bugfix: in error handling while connecting to a backend. Thanks to Piotr Sikora. *) Bugfix: in the "epoll" event method. Thanks to Yichun Zhang. *) Bugfix: the $sent_http_cache_control variable might contain a wrong value if the "expires" directive was used. Thanks to Yichun Zhang. *) Bugfix: the "limit_rate" directive did not allow to use full throughput, even if limit value was very high. *) Bugfix: the "sendfile_max_chunk" directive did not work, if the "limit_rate" directive was used. *) Bugfix: nginx could not be built on Solaris; the bug had appeared in 1.0.11. *) Bugfix: in the ngx_http_scgi_module. *) Bugfix: in the ngx_http_mp4_module.
author Igor Sysoev <http://sysoev.ru>
date Mon, 06 Feb 2012 00:00:00 +0400
parents f7ec98e3caeb
children 09a689c5e494
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
272
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
1
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
2 /*
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
3 * Copyright (C) Igor Sysoev
644
ad25218fd14b nginx 1.0.12
Igor Sysoev <http://sysoev.ru>
parents: 532
diff changeset
4 * Copyright (C) Nginx, Inc.
272
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
5 */
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
6
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
7
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
8 #include <ngx_config.h>
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
9 #include <ngx_core.h>
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
10 #include <ngx_http.h>
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
11
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
12
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
13 typedef struct {
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
14 u_char color;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
15 u_char len;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
16 u_short conn;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
17 u_char data[1];
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
18 } ngx_http_limit_zone_node_t;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
19
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
20
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
21 typedef struct {
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
22 ngx_shm_zone_t *shm_zone;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
23 ngx_rbtree_node_t *node;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
24 } ngx_http_limit_zone_cleanup_t;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
25
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
26
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
27 typedef struct {
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
28 ngx_rbtree_t *rbtree;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
29 ngx_int_t index;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
30 ngx_str_t var;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
31 } ngx_http_limit_zone_ctx_t;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
32
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
33
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
34 typedef struct {
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
35 ngx_shm_zone_t *shm_zone;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
36 ngx_uint_t conn;
532
f7ec98e3caeb nginx 0.8.18
Igor Sysoev <http://sysoev.ru>
parents: 496
diff changeset
37 ngx_uint_t log_level;
272
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
38 } ngx_http_limit_zone_conf_t;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
39
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
40
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
41 static void ngx_http_limit_zone_cleanup(void *data);
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
42
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
43 static void *ngx_http_limit_zone_create_conf(ngx_conf_t *cf);
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
44 static char *ngx_http_limit_zone_merge_conf(ngx_conf_t *cf, void *parent,
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
45 void *child);
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
46 static char *ngx_http_limit_zone(ngx_conf_t *cf, ngx_command_t *cmd,
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
47 void *conf);
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
48 static char *ngx_http_limit_conn(ngx_conf_t *cf, ngx_command_t *cmd,
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
49 void *conf);
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
50 static ngx_int_t ngx_http_limit_zone_init(ngx_conf_t *cf);
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
51
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
52
532
f7ec98e3caeb nginx 0.8.18
Igor Sysoev <http://sysoev.ru>
parents: 496
diff changeset
53 static ngx_conf_enum_t ngx_http_limit_conn_log_levels[] = {
f7ec98e3caeb nginx 0.8.18
Igor Sysoev <http://sysoev.ru>
parents: 496
diff changeset
54 { ngx_string("info"), NGX_LOG_INFO },
f7ec98e3caeb nginx 0.8.18
Igor Sysoev <http://sysoev.ru>
parents: 496
diff changeset
55 { ngx_string("notice"), NGX_LOG_NOTICE },
f7ec98e3caeb nginx 0.8.18
Igor Sysoev <http://sysoev.ru>
parents: 496
diff changeset
56 { ngx_string("warn"), NGX_LOG_WARN },
f7ec98e3caeb nginx 0.8.18
Igor Sysoev <http://sysoev.ru>
parents: 496
diff changeset
57 { ngx_string("error"), NGX_LOG_ERR },
f7ec98e3caeb nginx 0.8.18
Igor Sysoev <http://sysoev.ru>
parents: 496
diff changeset
58 { ngx_null_string, 0 }
f7ec98e3caeb nginx 0.8.18
Igor Sysoev <http://sysoev.ru>
parents: 496
diff changeset
59 };
f7ec98e3caeb nginx 0.8.18
Igor Sysoev <http://sysoev.ru>
parents: 496
diff changeset
60
f7ec98e3caeb nginx 0.8.18
Igor Sysoev <http://sysoev.ru>
parents: 496
diff changeset
61
272
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
62 static ngx_command_t ngx_http_limit_zone_commands[] = {
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
63
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
64 { ngx_string("limit_zone"),
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
65 NGX_HTTP_MAIN_CONF|NGX_CONF_TAKE3,
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
66 ngx_http_limit_zone,
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
67 0,
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
68 0,
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
69 NULL },
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
70
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
71 { ngx_string("limit_conn"),
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
72 NGX_HTTP_MAIN_CONF|NGX_HTTP_SRV_CONF|NGX_HTTP_LOC_CONF|NGX_CONF_TAKE2,
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
73 ngx_http_limit_conn,
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
74 NGX_HTTP_LOC_CONF_OFFSET,
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
75 0,
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
76 NULL },
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
77
532
f7ec98e3caeb nginx 0.8.18
Igor Sysoev <http://sysoev.ru>
parents: 496
diff changeset
78 { ngx_string("limit_conn_log_level"),
f7ec98e3caeb nginx 0.8.18
Igor Sysoev <http://sysoev.ru>
parents: 496
diff changeset
79 NGX_HTTP_MAIN_CONF|NGX_HTTP_SRV_CONF|NGX_HTTP_LOC_CONF|NGX_CONF_TAKE1,
f7ec98e3caeb nginx 0.8.18
Igor Sysoev <http://sysoev.ru>
parents: 496
diff changeset
80 ngx_conf_set_enum_slot,
f7ec98e3caeb nginx 0.8.18
Igor Sysoev <http://sysoev.ru>
parents: 496
diff changeset
81 NGX_HTTP_LOC_CONF_OFFSET,
f7ec98e3caeb nginx 0.8.18
Igor Sysoev <http://sysoev.ru>
parents: 496
diff changeset
82 offsetof(ngx_http_limit_zone_conf_t, log_level),
f7ec98e3caeb nginx 0.8.18
Igor Sysoev <http://sysoev.ru>
parents: 496
diff changeset
83 &ngx_http_limit_conn_log_levels },
f7ec98e3caeb nginx 0.8.18
Igor Sysoev <http://sysoev.ru>
parents: 496
diff changeset
84
272
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
85 ngx_null_command
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
86 };
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
87
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
88
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
89 static ngx_http_module_t ngx_http_limit_zone_module_ctx = {
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
90 NULL, /* preconfiguration */
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
91 ngx_http_limit_zone_init, /* postconfiguration */
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
92
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
93 NULL, /* create main configuration */
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
94 NULL, /* init main configuration */
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
95
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
96 NULL, /* create server configuration */
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
97 NULL, /* merge server configuration */
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
98
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
99 ngx_http_limit_zone_create_conf, /* create location configration */
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
100 ngx_http_limit_zone_merge_conf /* merge location configration */
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
101 };
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
102
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
103
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
104 ngx_module_t ngx_http_limit_zone_module = {
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
105 NGX_MODULE_V1,
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
106 &ngx_http_limit_zone_module_ctx, /* module context */
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
107 ngx_http_limit_zone_commands, /* module directives */
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
108 NGX_HTTP_MODULE, /* module type */
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
109 NULL, /* init master */
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
110 NULL, /* init module */
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
111 NULL, /* init process */
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
112 NULL, /* init thread */
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
113 NULL, /* exit thread */
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
114 NULL, /* exit process */
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
115 NULL, /* exit master */
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
116 NGX_MODULE_V1_PADDING
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
117 };
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
118
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
119
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
120 static ngx_int_t
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
121 ngx_http_limit_zone_handler(ngx_http_request_t *r)
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
122 {
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
123 size_t len, n;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
124 uint32_t hash;
274
052a7b1d40e5 nginx 0.5.7
Igor Sysoev <http://sysoev.ru>
parents: 272
diff changeset
125 ngx_int_t rc;
272
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
126 ngx_slab_pool_t *shpool;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
127 ngx_rbtree_node_t *node, *sentinel;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
128 ngx_pool_cleanup_t *cln;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
129 ngx_http_variable_value_t *vv;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
130 ngx_http_limit_zone_ctx_t *ctx;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
131 ngx_http_limit_zone_node_t *lz;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
132 ngx_http_limit_zone_conf_t *lzcf;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
133 ngx_http_limit_zone_cleanup_t *lzcln;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
134
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
135 if (r->main->limit_zone_set) {
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
136 return NGX_DECLINED;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
137 }
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
138
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
139 lzcf = ngx_http_get_module_loc_conf(r, ngx_http_limit_zone_module);
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
140
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
141 if (lzcf->shm_zone == NULL) {
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
142 return NGX_DECLINED;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
143 }
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
144
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
145 ctx = lzcf->shm_zone->data;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
146
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
147 vv = ngx_http_get_indexed_variable(r, ctx->index);
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
148
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
149 if (vv == NULL || vv->not_found) {
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
150 return NGX_DECLINED;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
151 }
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
152
274
052a7b1d40e5 nginx 0.5.7
Igor Sysoev <http://sysoev.ru>
parents: 272
diff changeset
153 len = vv->len;
052a7b1d40e5 nginx 0.5.7
Igor Sysoev <http://sysoev.ru>
parents: 272
diff changeset
154
052a7b1d40e5 nginx 0.5.7
Igor Sysoev <http://sysoev.ru>
parents: 272
diff changeset
155 if (len == 0) {
052a7b1d40e5 nginx 0.5.7
Igor Sysoev <http://sysoev.ru>
parents: 272
diff changeset
156 return NGX_DECLINED;
052a7b1d40e5 nginx 0.5.7
Igor Sysoev <http://sysoev.ru>
parents: 272
diff changeset
157 }
272
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
158
274
052a7b1d40e5 nginx 0.5.7
Igor Sysoev <http://sysoev.ru>
parents: 272
diff changeset
159 if (len > 255) {
052a7b1d40e5 nginx 0.5.7
Igor Sysoev <http://sysoev.ru>
parents: 272
diff changeset
160 ngx_log_error(NGX_LOG_ERR, r->connection->log, 0,
052a7b1d40e5 nginx 0.5.7
Igor Sysoev <http://sysoev.ru>
parents: 272
diff changeset
161 "the value of the \"%V\" variable "
328
390b8f8309d6 nginx 0.6.8
Igor Sysoev <http://sysoev.ru>
parents: 276
diff changeset
162 "is more than 255 bytes: \"%v\"",
274
052a7b1d40e5 nginx 0.5.7
Igor Sysoev <http://sysoev.ru>
parents: 272
diff changeset
163 &ctx->var, vv);
052a7b1d40e5 nginx 0.5.7
Igor Sysoev <http://sysoev.ru>
parents: 272
diff changeset
164 return NGX_DECLINED;
052a7b1d40e5 nginx 0.5.7
Igor Sysoev <http://sysoev.ru>
parents: 272
diff changeset
165 }
052a7b1d40e5 nginx 0.5.7
Igor Sysoev <http://sysoev.ru>
parents: 272
diff changeset
166
052a7b1d40e5 nginx 0.5.7
Igor Sysoev <http://sysoev.ru>
parents: 272
diff changeset
167 r->main->limit_zone_set = 1;
272
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
168
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
169 hash = ngx_crc32_short(vv->data, len);
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
170
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
171 cln = ngx_pool_cleanup_add(r->pool, sizeof(ngx_http_limit_zone_cleanup_t));
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
172 if (cln == NULL) {
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
173 return NGX_HTTP_INTERNAL_SERVER_ERROR;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
174 }
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
175
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
176 shpool = (ngx_slab_pool_t *) lzcf->shm_zone->shm.addr;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
177
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
178 ngx_shmtx_lock(&shpool->mutex);
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
179
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
180 node = ctx->rbtree->root;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
181 sentinel = ctx->rbtree->sentinel;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
182
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
183 while (node != sentinel) {
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
184
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
185 if (hash < node->key) {
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
186 node = node->left;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
187 continue;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
188 }
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
189
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
190 if (hash > node->key) {
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
191 node = node->right;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
192 continue;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
193 }
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
194
274
052a7b1d40e5 nginx 0.5.7
Igor Sysoev <http://sysoev.ru>
parents: 272
diff changeset
195 /* hash == node->key */
052a7b1d40e5 nginx 0.5.7
Igor Sysoev <http://sysoev.ru>
parents: 272
diff changeset
196
052a7b1d40e5 nginx 0.5.7
Igor Sysoev <http://sysoev.ru>
parents: 272
diff changeset
197 do {
272
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
198 lz = (ngx_http_limit_zone_node_t *) &node->color;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
199
276
c5c2b2883984 nginx 0.5.8
Igor Sysoev <http://sysoev.ru>
parents: 274
diff changeset
200 rc = ngx_memn2cmp(vv->data, lz->data, len, (size_t) lz->len);
274
052a7b1d40e5 nginx 0.5.7
Igor Sysoev <http://sysoev.ru>
parents: 272
diff changeset
201
052a7b1d40e5 nginx 0.5.7
Igor Sysoev <http://sysoev.ru>
parents: 272
diff changeset
202 if (rc == 0) {
272
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
203 if ((ngx_uint_t) lz->conn < lzcf->conn) {
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
204 lz->conn++;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
205 goto done;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
206 }
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
207
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
208 ngx_shmtx_unlock(&shpool->mutex);
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
209
532
f7ec98e3caeb nginx 0.8.18
Igor Sysoev <http://sysoev.ru>
parents: 496
diff changeset
210 ngx_log_error(lzcf->log_level, r->connection->log, 0,
426
e7dbea1ee115 nginx 0.7.25
Igor Sysoev <http://sysoev.ru>
parents: 356
diff changeset
211 "limiting connections by zone \"%V\"",
480
549994537f15 nginx 0.7.52
Igor Sysoev <http://sysoev.ru>
parents: 468
diff changeset
212 &lzcf->shm_zone->shm.name);
426
e7dbea1ee115 nginx 0.7.25
Igor Sysoev <http://sysoev.ru>
parents: 356
diff changeset
213
272
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
214 return NGX_HTTP_SERVICE_UNAVAILABLE;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
215 }
274
052a7b1d40e5 nginx 0.5.7
Igor Sysoev <http://sysoev.ru>
parents: 272
diff changeset
216
052a7b1d40e5 nginx 0.5.7
Igor Sysoev <http://sysoev.ru>
parents: 272
diff changeset
217 node = (rc < 0) ? node->left : node->right;
052a7b1d40e5 nginx 0.5.7
Igor Sysoev <http://sysoev.ru>
parents: 272
diff changeset
218
052a7b1d40e5 nginx 0.5.7
Igor Sysoev <http://sysoev.ru>
parents: 272
diff changeset
219 } while (node != sentinel && hash == node->key);
052a7b1d40e5 nginx 0.5.7
Igor Sysoev <http://sysoev.ru>
parents: 272
diff changeset
220
052a7b1d40e5 nginx 0.5.7
Igor Sysoev <http://sysoev.ru>
parents: 272
diff changeset
221 break;
272
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
222 }
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
223
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
224 n = offsetof(ngx_rbtree_node_t, color)
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
225 + offsetof(ngx_http_limit_zone_node_t, data)
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
226 + len;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
227
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
228 node = ngx_slab_alloc_locked(shpool, n);
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
229 if (node == NULL) {
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
230 ngx_shmtx_unlock(&shpool->mutex);
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
231 return NGX_HTTP_SERVICE_UNAVAILABLE;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
232 }
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
233
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
234 lz = (ngx_http_limit_zone_node_t *) &node->color;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
235
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
236 node->key = hash;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
237 lz->len = (u_char) len;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
238 lz->conn = 1;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
239 ngx_memcpy(lz->data, vv->data, len);
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
240
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
241 ngx_rbtree_insert(ctx->rbtree, node);
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
242
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
243 done:
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
244
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
245 ngx_log_debug2(NGX_LOG_DEBUG_HTTP, r->connection->log, 0,
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
246 "limit zone: %08XD %d", node->key, lz->conn);
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
247
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
248 ngx_shmtx_unlock(&shpool->mutex);
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
249
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
250 cln->handler = ngx_http_limit_zone_cleanup;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
251 lzcln = cln->data;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
252
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
253 lzcln->shm_zone = lzcf->shm_zone;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
254 lzcln->node = node;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
255
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
256 return NGX_DECLINED;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
257 }
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
258
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
259
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
260 static void
274
052a7b1d40e5 nginx 0.5.7
Igor Sysoev <http://sysoev.ru>
parents: 272
diff changeset
261 ngx_http_limit_zone_rbtree_insert_value(ngx_rbtree_node_t *temp,
052a7b1d40e5 nginx 0.5.7
Igor Sysoev <http://sysoev.ru>
parents: 272
diff changeset
262 ngx_rbtree_node_t *node, ngx_rbtree_node_t *sentinel)
052a7b1d40e5 nginx 0.5.7
Igor Sysoev <http://sysoev.ru>
parents: 272
diff changeset
263 {
356
b743d290eb3b nginx 0.6.22
Igor Sysoev <http://sysoev.ru>
parents: 328
diff changeset
264 ngx_rbtree_node_t **p;
b743d290eb3b nginx 0.6.22
Igor Sysoev <http://sysoev.ru>
parents: 328
diff changeset
265 ngx_http_limit_zone_node_t *lzn, *lznt;
274
052a7b1d40e5 nginx 0.5.7
Igor Sysoev <http://sysoev.ru>
parents: 272
diff changeset
266
052a7b1d40e5 nginx 0.5.7
Igor Sysoev <http://sysoev.ru>
parents: 272
diff changeset
267 for ( ;; ) {
052a7b1d40e5 nginx 0.5.7
Igor Sysoev <http://sysoev.ru>
parents: 272
diff changeset
268
052a7b1d40e5 nginx 0.5.7
Igor Sysoev <http://sysoev.ru>
parents: 272
diff changeset
269 if (node->key < temp->key) {
052a7b1d40e5 nginx 0.5.7
Igor Sysoev <http://sysoev.ru>
parents: 272
diff changeset
270
356
b743d290eb3b nginx 0.6.22
Igor Sysoev <http://sysoev.ru>
parents: 328
diff changeset
271 p = &temp->left;
274
052a7b1d40e5 nginx 0.5.7
Igor Sysoev <http://sysoev.ru>
parents: 272
diff changeset
272
052a7b1d40e5 nginx 0.5.7
Igor Sysoev <http://sysoev.ru>
parents: 272
diff changeset
273 } else if (node->key > temp->key) {
052a7b1d40e5 nginx 0.5.7
Igor Sysoev <http://sysoev.ru>
parents: 272
diff changeset
274
356
b743d290eb3b nginx 0.6.22
Igor Sysoev <http://sysoev.ru>
parents: 328
diff changeset
275 p = &temp->right;
274
052a7b1d40e5 nginx 0.5.7
Igor Sysoev <http://sysoev.ru>
parents: 272
diff changeset
276
052a7b1d40e5 nginx 0.5.7
Igor Sysoev <http://sysoev.ru>
parents: 272
diff changeset
277 } else { /* node->key == temp->key */
052a7b1d40e5 nginx 0.5.7
Igor Sysoev <http://sysoev.ru>
parents: 272
diff changeset
278
052a7b1d40e5 nginx 0.5.7
Igor Sysoev <http://sysoev.ru>
parents: 272
diff changeset
279 lzn = (ngx_http_limit_zone_node_t *) &node->color;
052a7b1d40e5 nginx 0.5.7
Igor Sysoev <http://sysoev.ru>
parents: 272
diff changeset
280 lznt = (ngx_http_limit_zone_node_t *) &temp->color;
052a7b1d40e5 nginx 0.5.7
Igor Sysoev <http://sysoev.ru>
parents: 272
diff changeset
281
356
b743d290eb3b nginx 0.6.22
Igor Sysoev <http://sysoev.ru>
parents: 328
diff changeset
282 p = (ngx_memn2cmp(lzn->data, lznt->data, lzn->len, lznt->len) < 0)
b743d290eb3b nginx 0.6.22
Igor Sysoev <http://sysoev.ru>
parents: 328
diff changeset
283 ? &temp->left : &temp->right;
b743d290eb3b nginx 0.6.22
Igor Sysoev <http://sysoev.ru>
parents: 328
diff changeset
284 }
274
052a7b1d40e5 nginx 0.5.7
Igor Sysoev <http://sysoev.ru>
parents: 272
diff changeset
285
356
b743d290eb3b nginx 0.6.22
Igor Sysoev <http://sysoev.ru>
parents: 328
diff changeset
286 if (*p == sentinel) {
b743d290eb3b nginx 0.6.22
Igor Sysoev <http://sysoev.ru>
parents: 328
diff changeset
287 break;
b743d290eb3b nginx 0.6.22
Igor Sysoev <http://sysoev.ru>
parents: 328
diff changeset
288 }
274
052a7b1d40e5 nginx 0.5.7
Igor Sysoev <http://sysoev.ru>
parents: 272
diff changeset
289
356
b743d290eb3b nginx 0.6.22
Igor Sysoev <http://sysoev.ru>
parents: 328
diff changeset
290 temp = *p;
274
052a7b1d40e5 nginx 0.5.7
Igor Sysoev <http://sysoev.ru>
parents: 272
diff changeset
291 }
052a7b1d40e5 nginx 0.5.7
Igor Sysoev <http://sysoev.ru>
parents: 272
diff changeset
292
356
b743d290eb3b nginx 0.6.22
Igor Sysoev <http://sysoev.ru>
parents: 328
diff changeset
293 *p = node;
274
052a7b1d40e5 nginx 0.5.7
Igor Sysoev <http://sysoev.ru>
parents: 272
diff changeset
294 node->parent = temp;
052a7b1d40e5 nginx 0.5.7
Igor Sysoev <http://sysoev.ru>
parents: 272
diff changeset
295 node->left = sentinel;
052a7b1d40e5 nginx 0.5.7
Igor Sysoev <http://sysoev.ru>
parents: 272
diff changeset
296 node->right = sentinel;
052a7b1d40e5 nginx 0.5.7
Igor Sysoev <http://sysoev.ru>
parents: 272
diff changeset
297 ngx_rbt_red(node);
052a7b1d40e5 nginx 0.5.7
Igor Sysoev <http://sysoev.ru>
parents: 272
diff changeset
298 }
052a7b1d40e5 nginx 0.5.7
Igor Sysoev <http://sysoev.ru>
parents: 272
diff changeset
299
052a7b1d40e5 nginx 0.5.7
Igor Sysoev <http://sysoev.ru>
parents: 272
diff changeset
300
052a7b1d40e5 nginx 0.5.7
Igor Sysoev <http://sysoev.ru>
parents: 272
diff changeset
301 static void
272
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
302 ngx_http_limit_zone_cleanup(void *data)
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
303 {
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
304 ngx_http_limit_zone_cleanup_t *lzcln = data;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
305
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
306 ngx_slab_pool_t *shpool;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
307 ngx_rbtree_node_t *node;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
308 ngx_http_limit_zone_ctx_t *ctx;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
309 ngx_http_limit_zone_node_t *lz;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
310
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
311 ctx = lzcln->shm_zone->data;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
312 shpool = (ngx_slab_pool_t *) lzcln->shm_zone->shm.addr;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
313 node = lzcln->node;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
314 lz = (ngx_http_limit_zone_node_t *) &node->color;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
315
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
316 ngx_shmtx_lock(&shpool->mutex);
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
317
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
318 ngx_log_debug2(NGX_LOG_DEBUG_HTTP, lzcln->shm_zone->shm.log, 0,
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
319 "limit zone cleanup: %08XD %d", node->key, lz->conn);
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
320
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
321 lz->conn--;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
322
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
323 if (lz->conn == 0) {
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
324 ngx_rbtree_delete(ctx->rbtree, node);
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
325 ngx_slab_free_locked(shpool, node);
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
326 }
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
327
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
328 ngx_shmtx_unlock(&shpool->mutex);
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
329 }
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
330
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
331
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
332 static ngx_int_t
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
333 ngx_http_limit_zone_init_zone(ngx_shm_zone_t *shm_zone, void *data)
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
334 {
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
335 ngx_http_limit_zone_ctx_t *octx = data;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
336
468
56baf312c1b5 nginx 0.7.46
Igor Sysoev <http://sysoev.ru>
parents: 426
diff changeset
337 size_t len;
272
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
338 ngx_slab_pool_t *shpool;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
339 ngx_rbtree_node_t *sentinel;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
340 ngx_http_limit_zone_ctx_t *ctx;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
341
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
342 ctx = shm_zone->data;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
343
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
344 if (octx) {
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
345 if (ngx_strcmp(ctx->var.data, octx->var.data) != 0) {
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
346 ngx_log_error(NGX_LOG_EMERG, shm_zone->shm.log, 0,
274
052a7b1d40e5 nginx 0.5.7
Igor Sysoev <http://sysoev.ru>
parents: 272
diff changeset
347 "limit_zone \"%V\" uses the \"%V\" variable "
272
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
348 "while previously it used the \"%V\" variable",
480
549994537f15 nginx 0.7.52
Igor Sysoev <http://sysoev.ru>
parents: 468
diff changeset
349 &shm_zone->shm.name, &ctx->var, &octx->var);
272
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
350 return NGX_ERROR;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
351 }
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
352
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
353 ctx->rbtree = octx->rbtree;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
354
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
355 return NGX_OK;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
356 }
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
357
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
358 shpool = (ngx_slab_pool_t *) shm_zone->shm.addr;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
359
480
549994537f15 nginx 0.7.52
Igor Sysoev <http://sysoev.ru>
parents: 468
diff changeset
360 if (shm_zone->shm.exists) {
549994537f15 nginx 0.7.52
Igor Sysoev <http://sysoev.ru>
parents: 468
diff changeset
361 ctx->rbtree = shpool->data;
549994537f15 nginx 0.7.52
Igor Sysoev <http://sysoev.ru>
parents: 468
diff changeset
362
549994537f15 nginx 0.7.52
Igor Sysoev <http://sysoev.ru>
parents: 468
diff changeset
363 return NGX_OK;
549994537f15 nginx 0.7.52
Igor Sysoev <http://sysoev.ru>
parents: 468
diff changeset
364 }
549994537f15 nginx 0.7.52
Igor Sysoev <http://sysoev.ru>
parents: 468
diff changeset
365
272
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
366 ctx->rbtree = ngx_slab_alloc(shpool, sizeof(ngx_rbtree_t));
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
367 if (ctx->rbtree == NULL) {
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
368 return NGX_ERROR;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
369 }
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
370
480
549994537f15 nginx 0.7.52
Igor Sysoev <http://sysoev.ru>
parents: 468
diff changeset
371 shpool->data = ctx->rbtree;
549994537f15 nginx 0.7.52
Igor Sysoev <http://sysoev.ru>
parents: 468
diff changeset
372
272
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
373 sentinel = ngx_slab_alloc(shpool, sizeof(ngx_rbtree_node_t));
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
374 if (sentinel == NULL) {
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
375 return NGX_ERROR;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
376 }
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
377
356
b743d290eb3b nginx 0.6.22
Igor Sysoev <http://sysoev.ru>
parents: 328
diff changeset
378 ngx_rbtree_init(ctx->rbtree, sentinel,
b743d290eb3b nginx 0.6.22
Igor Sysoev <http://sysoev.ru>
parents: 328
diff changeset
379 ngx_http_limit_zone_rbtree_insert_value);
272
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
380
480
549994537f15 nginx 0.7.52
Igor Sysoev <http://sysoev.ru>
parents: 468
diff changeset
381 len = sizeof(" in limit_zone \"\"") + shm_zone->shm.name.len;
468
56baf312c1b5 nginx 0.7.46
Igor Sysoev <http://sysoev.ru>
parents: 426
diff changeset
382
56baf312c1b5 nginx 0.7.46
Igor Sysoev <http://sysoev.ru>
parents: 426
diff changeset
383 shpool->log_ctx = ngx_slab_alloc(shpool, len);
56baf312c1b5 nginx 0.7.46
Igor Sysoev <http://sysoev.ru>
parents: 426
diff changeset
384 if (shpool->log_ctx == NULL) {
56baf312c1b5 nginx 0.7.46
Igor Sysoev <http://sysoev.ru>
parents: 426
diff changeset
385 return NGX_ERROR;
56baf312c1b5 nginx 0.7.46
Igor Sysoev <http://sysoev.ru>
parents: 426
diff changeset
386 }
56baf312c1b5 nginx 0.7.46
Igor Sysoev <http://sysoev.ru>
parents: 426
diff changeset
387
480
549994537f15 nginx 0.7.52
Igor Sysoev <http://sysoev.ru>
parents: 468
diff changeset
388 ngx_sprintf(shpool->log_ctx, " in limit_zone \"%V\"%Z",
549994537f15 nginx 0.7.52
Igor Sysoev <http://sysoev.ru>
parents: 468
diff changeset
389 &shm_zone->shm.name);
468
56baf312c1b5 nginx 0.7.46
Igor Sysoev <http://sysoev.ru>
parents: 426
diff changeset
390
272
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
391 return NGX_OK;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
392 }
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
393
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
394
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
395 static void *
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
396 ngx_http_limit_zone_create_conf(ngx_conf_t *cf)
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
397 {
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
398 ngx_http_limit_zone_conf_t *conf;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
399
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
400 conf = ngx_pcalloc(cf->pool, sizeof(ngx_http_limit_zone_conf_t));
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
401 if (conf == NULL) {
496
f39b9e29530d nginx 0.8.0
Igor Sysoev <http://sysoev.ru>
parents: 480
diff changeset
402 return NULL;
272
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
403 }
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
404
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
405 /*
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
406 * set by ngx_pcalloc():
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
407 *
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
408 * conf->shm_zone = NULL;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
409 * conf->conn = 0;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
410 */
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
411
532
f7ec98e3caeb nginx 0.8.18
Igor Sysoev <http://sysoev.ru>
parents: 496
diff changeset
412 conf->log_level = NGX_CONF_UNSET_UINT;
f7ec98e3caeb nginx 0.8.18
Igor Sysoev <http://sysoev.ru>
parents: 496
diff changeset
413
272
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
414 return conf;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
415 }
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
416
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
417
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
418 static char *
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
419 ngx_http_limit_zone_merge_conf(ngx_conf_t *cf, void *parent, void *child)
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
420 {
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
421 ngx_http_limit_zone_conf_t *prev = parent;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
422 ngx_http_limit_zone_conf_t *conf = child;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
423
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
424 if (conf->shm_zone == NULL) {
644
ad25218fd14b nginx 1.0.12
Igor Sysoev <http://sysoev.ru>
parents: 532
diff changeset
425 conf->shm_zone = prev->shm_zone;
ad25218fd14b nginx 1.0.12
Igor Sysoev <http://sysoev.ru>
parents: 532
diff changeset
426 conf->conn = prev->conn;
272
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
427 }
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
428
532
f7ec98e3caeb nginx 0.8.18
Igor Sysoev <http://sysoev.ru>
parents: 496
diff changeset
429 ngx_conf_merge_uint_value(conf->log_level, prev->log_level, NGX_LOG_ERR);
f7ec98e3caeb nginx 0.8.18
Igor Sysoev <http://sysoev.ru>
parents: 496
diff changeset
430
272
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
431 return NGX_CONF_OK;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
432 }
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
433
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
434
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
435 static char *
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
436 ngx_http_limit_zone(ngx_conf_t *cf, ngx_command_t *cmd, void *conf)
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
437 {
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
438 ssize_t n;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
439 ngx_str_t *value;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
440 ngx_shm_zone_t *shm_zone;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
441 ngx_http_limit_zone_ctx_t *ctx;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
442
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
443 value = cf->args->elts;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
444
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
445 if (value[2].data[0] != '$') {
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
446 ngx_conf_log_error(NGX_LOG_EMERG, cf, 0,
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
447 "invalid variable name \"%V\"", &value[2]);
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
448 return NGX_CONF_ERROR;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
449 }
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
450
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
451 value[2].len--;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
452 value[2].data++;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
453
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
454 ctx = ngx_pcalloc(cf->pool, sizeof(ngx_http_limit_zone_ctx_t));
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
455 if (ctx == NULL) {
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
456 return NGX_CONF_ERROR;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
457 }
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
458
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
459 ctx->index = ngx_http_get_variable_index(cf, &value[2]);
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
460 if (ctx->index == NGX_ERROR) {
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
461 return NGX_CONF_ERROR;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
462 }
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
463
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
464 ctx->var = value[2];
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
465
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
466 n = ngx_parse_size(&value[3]);
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
467
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
468 if (n == NGX_ERROR) {
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
469 ngx_conf_log_error(NGX_LOG_EMERG, cf, 0,
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
470 "invalid size of limit_zone \"%V\"", &value[3]);
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
471 return NGX_CONF_ERROR;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
472 }
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
473
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
474 if (n < (ngx_int_t) (8 * ngx_pagesize)) {
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
475 ngx_conf_log_error(NGX_LOG_EMERG, cf, 0,
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
476 "limit_zone \"%V\" is too small", &value[1]);
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
477 return NGX_CONF_ERROR;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
478 }
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
479
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
480
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
481 shm_zone = ngx_shared_memory_add(cf, &value[1], n,
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
482 &ngx_http_limit_zone_module);
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
483 if (shm_zone == NULL) {
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
484 return NGX_CONF_ERROR;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
485 }
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
486
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
487 if (shm_zone->data) {
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
488 ctx = shm_zone->data;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
489
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
490 ngx_conf_log_error(NGX_LOG_EMERG, cf, 0,
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
491 "limit_zone \"%V\" is already bound to variable \"%V\"",
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
492 &value[1], &ctx->var);
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
493 return NGX_CONF_ERROR;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
494 }
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
495
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
496 shm_zone->init = ngx_http_limit_zone_init_zone;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
497 shm_zone->data = ctx;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
498
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
499 return NGX_CONF_OK;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
500 }
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
501
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
502
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
503 static char *
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
504 ngx_http_limit_conn(ngx_conf_t *cf, ngx_command_t *cmd, void *conf)
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
505 {
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
506 ngx_http_limit_zone_conf_t *lzcf = conf;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
507
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
508 ngx_int_t n;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
509 ngx_str_t *value;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
510
426
e7dbea1ee115 nginx 0.7.25
Igor Sysoev <http://sysoev.ru>
parents: 356
diff changeset
511 if (lzcf->shm_zone) {
e7dbea1ee115 nginx 0.7.25
Igor Sysoev <http://sysoev.ru>
parents: 356
diff changeset
512 return "is duplicate";
e7dbea1ee115 nginx 0.7.25
Igor Sysoev <http://sysoev.ru>
parents: 356
diff changeset
513 }
e7dbea1ee115 nginx 0.7.25
Igor Sysoev <http://sysoev.ru>
parents: 356
diff changeset
514
272
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
515 value = cf->args->elts;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
516
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
517 lzcf->shm_zone = ngx_shared_memory_add(cf, &value[1], 0,
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
518 &ngx_http_limit_zone_module);
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
519 if (lzcf->shm_zone == NULL) {
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
520 return NGX_CONF_ERROR;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
521 }
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
522
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
523 n = ngx_atoi(value[2].data, value[2].len);
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
524 if (n <= 0) {
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
525 ngx_conf_log_error(NGX_LOG_EMERG, cf, 0,
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
526 "invalid number of connections \"%V\"", &value[2]);
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
527 return NGX_CONF_ERROR;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
528 }
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
529
274
052a7b1d40e5 nginx 0.5.7
Igor Sysoev <http://sysoev.ru>
parents: 272
diff changeset
530 if (n > 65535) {
052a7b1d40e5 nginx 0.5.7
Igor Sysoev <http://sysoev.ru>
parents: 272
diff changeset
531 ngx_conf_log_error(NGX_LOG_EMERG, cf, 0,
052a7b1d40e5 nginx 0.5.7
Igor Sysoev <http://sysoev.ru>
parents: 272
diff changeset
532 "connection limit must be less 65536");
052a7b1d40e5 nginx 0.5.7
Igor Sysoev <http://sysoev.ru>
parents: 272
diff changeset
533 return NGX_CONF_ERROR;
052a7b1d40e5 nginx 0.5.7
Igor Sysoev <http://sysoev.ru>
parents: 272
diff changeset
534 }
052a7b1d40e5 nginx 0.5.7
Igor Sysoev <http://sysoev.ru>
parents: 272
diff changeset
535
272
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
536 lzcf->conn = n;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
537
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
538 return NGX_CONF_OK;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
539 }
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
540
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
541
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
542 static ngx_int_t
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
543 ngx_http_limit_zone_init(ngx_conf_t *cf)
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
544 {
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
545 ngx_http_handler_pt *h;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
546 ngx_http_core_main_conf_t *cmcf;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
547
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
548 cmcf = ngx_http_conf_get_module_main_conf(cf, ngx_http_core_module);
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
549
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
550 h = ngx_array_push(&cmcf->phases[NGX_HTTP_PREACCESS_PHASE].handlers);
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
551 if (h == NULL) {
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
552 return NGX_ERROR;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
553 }
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
554
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
555 *h = ngx_http_limit_zone_handler;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
556
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
557 return NGX_OK;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
558 }