annotate src/http/modules/ngx_http_limit_zone_module.c @ 496:f39b9e29530d NGINX_0_8_0

nginx 0.8.0 *) Feature: the "keepalive_requests" directive. *) Feature: the "limit_rate_after" directive. Thanks to Ivan Debnar. *) Bugfix: XLST filter did not work in subrequests. *) Bugfix: in relative paths handling in nginx/Windows. *) Bugfix: in proxy_store, fastcgi_store, proxy_cache, and fastcgi_cache in nginx/Windows. *) Bugfix: in memory allocation error handling. Thanks to Maxim Dounin and Kirill A. Korinskiy.
author Igor Sysoev <http://sysoev.ru>
date Tue, 02 Jun 2009 00:00:00 +0400
parents 549994537f15
children f7ec98e3caeb
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
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
4 */
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 #include <ngx_config.h>
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
8 #include <ngx_core.h>
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
9 #include <ngx_http.h>
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
10
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 typedef struct {
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
13 u_char color;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
14 u_char len;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
15 u_short conn;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
16 u_char data[1];
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
17 } ngx_http_limit_zone_node_t;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
18
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 typedef struct {
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
21 ngx_shm_zone_t *shm_zone;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
22 ngx_rbtree_node_t *node;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
23 } ngx_http_limit_zone_cleanup_t;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
24
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 typedef struct {
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
27 ngx_rbtree_t *rbtree;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
28 ngx_int_t index;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
29 ngx_str_t var;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
30 } ngx_http_limit_zone_ctx_t;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
31
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 typedef struct {
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
34 ngx_shm_zone_t *shm_zone;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
35 ngx_uint_t conn;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
36 } ngx_http_limit_zone_conf_t;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
37
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
38
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
39 static void ngx_http_limit_zone_cleanup(void *data);
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_create_conf(ngx_conf_t *cf);
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
42 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
43 void *child);
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
44 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
45 void *conf);
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
46 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
47 void *conf);
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
48 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
49
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
50
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
51 static ngx_command_t ngx_http_limit_zone_commands[] = {
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
52
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
53 { ngx_string("limit_zone"),
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
54 NGX_HTTP_MAIN_CONF|NGX_CONF_TAKE3,
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
55 ngx_http_limit_zone,
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
56 0,
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
57 0,
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
58 NULL },
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
59
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
60 { ngx_string("limit_conn"),
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
61 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
62 ngx_http_limit_conn,
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
63 NGX_HTTP_LOC_CONF_OFFSET,
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
64 0,
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
65 NULL },
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
66
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
67 ngx_null_command
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
68 };
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
69
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 static ngx_http_module_t ngx_http_limit_zone_module_ctx = {
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
72 NULL, /* preconfiguration */
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
73 ngx_http_limit_zone_init, /* postconfiguration */
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
74
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
75 NULL, /* create main configuration */
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
76 NULL, /* init main configuration */
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
77
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
78 NULL, /* create server configuration */
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
79 NULL, /* merge server configuration */
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
80
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
81 ngx_http_limit_zone_create_conf, /* create location configration */
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
82 ngx_http_limit_zone_merge_conf /* merge location configration */
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
83 };
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
84
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
85
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
86 ngx_module_t ngx_http_limit_zone_module = {
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
87 NGX_MODULE_V1,
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
88 &ngx_http_limit_zone_module_ctx, /* module context */
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
89 ngx_http_limit_zone_commands, /* module directives */
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
90 NGX_HTTP_MODULE, /* module type */
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
91 NULL, /* init master */
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
92 NULL, /* init module */
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
93 NULL, /* init process */
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
94 NULL, /* init thread */
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
95 NULL, /* exit thread */
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
96 NULL, /* exit process */
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
97 NULL, /* exit master */
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
98 NGX_MODULE_V1_PADDING
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
99 };
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
100
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 static ngx_int_t
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
103 ngx_http_limit_zone_handler(ngx_http_request_t *r)
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
104 {
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
105 size_t len, n;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
106 uint32_t hash;
274
052a7b1d40e5 nginx 0.5.7
Igor Sysoev <http://sysoev.ru>
parents: 272
diff changeset
107 ngx_int_t rc;
272
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
108 ngx_slab_pool_t *shpool;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
109 ngx_rbtree_node_t *node, *sentinel;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
110 ngx_pool_cleanup_t *cln;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
111 ngx_http_variable_value_t *vv;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
112 ngx_http_limit_zone_ctx_t *ctx;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
113 ngx_http_limit_zone_node_t *lz;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
114 ngx_http_limit_zone_conf_t *lzcf;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
115 ngx_http_limit_zone_cleanup_t *lzcln;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
116
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
117 if (r->main->limit_zone_set) {
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
118 return NGX_DECLINED;
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
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
121 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
122
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
123 if (lzcf->shm_zone == NULL) {
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
124 return NGX_DECLINED;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
125 }
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
126
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
127 ctx = lzcf->shm_zone->data;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
128
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
129 vv = ngx_http_get_indexed_variable(r, ctx->index);
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
130
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
131 if (vv == NULL || vv->not_found) {
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
132 return NGX_DECLINED;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
133 }
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
134
274
052a7b1d40e5 nginx 0.5.7
Igor Sysoev <http://sysoev.ru>
parents: 272
diff changeset
135 len = vv->len;
052a7b1d40e5 nginx 0.5.7
Igor Sysoev <http://sysoev.ru>
parents: 272
diff changeset
136
052a7b1d40e5 nginx 0.5.7
Igor Sysoev <http://sysoev.ru>
parents: 272
diff changeset
137 if (len == 0) {
052a7b1d40e5 nginx 0.5.7
Igor Sysoev <http://sysoev.ru>
parents: 272
diff changeset
138 return NGX_DECLINED;
052a7b1d40e5 nginx 0.5.7
Igor Sysoev <http://sysoev.ru>
parents: 272
diff changeset
139 }
272
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
140
274
052a7b1d40e5 nginx 0.5.7
Igor Sysoev <http://sysoev.ru>
parents: 272
diff changeset
141 if (len > 255) {
052a7b1d40e5 nginx 0.5.7
Igor Sysoev <http://sysoev.ru>
parents: 272
diff changeset
142 ngx_log_error(NGX_LOG_ERR, r->connection->log, 0,
052a7b1d40e5 nginx 0.5.7
Igor Sysoev <http://sysoev.ru>
parents: 272
diff changeset
143 "the value of the \"%V\" variable "
328
390b8f8309d6 nginx 0.6.8
Igor Sysoev <http://sysoev.ru>
parents: 276
diff changeset
144 "is more than 255 bytes: \"%v\"",
274
052a7b1d40e5 nginx 0.5.7
Igor Sysoev <http://sysoev.ru>
parents: 272
diff changeset
145 &ctx->var, vv);
052a7b1d40e5 nginx 0.5.7
Igor Sysoev <http://sysoev.ru>
parents: 272
diff changeset
146 return NGX_DECLINED;
052a7b1d40e5 nginx 0.5.7
Igor Sysoev <http://sysoev.ru>
parents: 272
diff changeset
147 }
052a7b1d40e5 nginx 0.5.7
Igor Sysoev <http://sysoev.ru>
parents: 272
diff changeset
148
052a7b1d40e5 nginx 0.5.7
Igor Sysoev <http://sysoev.ru>
parents: 272
diff changeset
149 r->main->limit_zone_set = 1;
272
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
150
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
151 hash = ngx_crc32_short(vv->data, len);
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
152
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
153 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
154 if (cln == NULL) {
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
155 return NGX_HTTP_INTERNAL_SERVER_ERROR;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
156 }
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
157
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
158 shpool = (ngx_slab_pool_t *) lzcf->shm_zone->shm.addr;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
159
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
160 ngx_shmtx_lock(&shpool->mutex);
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
161
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
162 node = ctx->rbtree->root;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
163 sentinel = ctx->rbtree->sentinel;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
164
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
165 while (node != sentinel) {
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
166
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
167 if (hash < node->key) {
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
168 node = node->left;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
169 continue;
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
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
172 if (hash > node->key) {
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
173 node = node->right;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
174 continue;
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
274
052a7b1d40e5 nginx 0.5.7
Igor Sysoev <http://sysoev.ru>
parents: 272
diff changeset
177 /* hash == node->key */
052a7b1d40e5 nginx 0.5.7
Igor Sysoev <http://sysoev.ru>
parents: 272
diff changeset
178
052a7b1d40e5 nginx 0.5.7
Igor Sysoev <http://sysoev.ru>
parents: 272
diff changeset
179 do {
272
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
180 lz = (ngx_http_limit_zone_node_t *) &node->color;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
181
276
c5c2b2883984 nginx 0.5.8
Igor Sysoev <http://sysoev.ru>
parents: 274
diff changeset
182 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
183
052a7b1d40e5 nginx 0.5.7
Igor Sysoev <http://sysoev.ru>
parents: 272
diff changeset
184 if (rc == 0) {
272
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
185 if ((ngx_uint_t) lz->conn < lzcf->conn) {
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
186 lz->conn++;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
187 goto done;
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 ngx_shmtx_unlock(&shpool->mutex);
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
191
426
e7dbea1ee115 nginx 0.7.25
Igor Sysoev <http://sysoev.ru>
parents: 356
diff changeset
192 ngx_log_error(NGX_LOG_ERR, r->connection->log, 0,
e7dbea1ee115 nginx 0.7.25
Igor Sysoev <http://sysoev.ru>
parents: 356
diff changeset
193 "limiting connections by zone \"%V\"",
480
549994537f15 nginx 0.7.52
Igor Sysoev <http://sysoev.ru>
parents: 468
diff changeset
194 &lzcf->shm_zone->shm.name);
426
e7dbea1ee115 nginx 0.7.25
Igor Sysoev <http://sysoev.ru>
parents: 356
diff changeset
195
272
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
196 return NGX_HTTP_SERVICE_UNAVAILABLE;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
197 }
274
052a7b1d40e5 nginx 0.5.7
Igor Sysoev <http://sysoev.ru>
parents: 272
diff changeset
198
052a7b1d40e5 nginx 0.5.7
Igor Sysoev <http://sysoev.ru>
parents: 272
diff changeset
199 node = (rc < 0) ? node->left : node->right;
052a7b1d40e5 nginx 0.5.7
Igor Sysoev <http://sysoev.ru>
parents: 272
diff changeset
200
052a7b1d40e5 nginx 0.5.7
Igor Sysoev <http://sysoev.ru>
parents: 272
diff changeset
201 } while (node != sentinel && hash == node->key);
052a7b1d40e5 nginx 0.5.7
Igor Sysoev <http://sysoev.ru>
parents: 272
diff changeset
202
052a7b1d40e5 nginx 0.5.7
Igor Sysoev <http://sysoev.ru>
parents: 272
diff changeset
203 break;
272
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
204 }
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
205
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
206 n = offsetof(ngx_rbtree_node_t, color)
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
207 + offsetof(ngx_http_limit_zone_node_t, data)
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
208 + len;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
209
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
210 node = ngx_slab_alloc_locked(shpool, n);
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
211 if (node == NULL) {
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
212 ngx_shmtx_unlock(&shpool->mutex);
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
213 return NGX_HTTP_SERVICE_UNAVAILABLE;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
214 }
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
215
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
216 lz = (ngx_http_limit_zone_node_t *) &node->color;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
217
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
218 node->key = hash;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
219 lz->len = (u_char) len;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
220 lz->conn = 1;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
221 ngx_memcpy(lz->data, vv->data, len);
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 ngx_rbtree_insert(ctx->rbtree, node);
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
224
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
225 done:
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
226
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
227 ngx_log_debug2(NGX_LOG_DEBUG_HTTP, r->connection->log, 0,
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
228 "limit zone: %08XD %d", node->key, lz->conn);
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
229
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
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
232 cln->handler = ngx_http_limit_zone_cleanup;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
233 lzcln = cln->data;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
234
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
235 lzcln->shm_zone = lzcf->shm_zone;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
236 lzcln->node = node;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
237
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
238 return NGX_DECLINED;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
239 }
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
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
242 static void
274
052a7b1d40e5 nginx 0.5.7
Igor Sysoev <http://sysoev.ru>
parents: 272
diff changeset
243 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
244 ngx_rbtree_node_t *node, ngx_rbtree_node_t *sentinel)
052a7b1d40e5 nginx 0.5.7
Igor Sysoev <http://sysoev.ru>
parents: 272
diff changeset
245 {
356
b743d290eb3b nginx 0.6.22
Igor Sysoev <http://sysoev.ru>
parents: 328
diff changeset
246 ngx_rbtree_node_t **p;
b743d290eb3b nginx 0.6.22
Igor Sysoev <http://sysoev.ru>
parents: 328
diff changeset
247 ngx_http_limit_zone_node_t *lzn, *lznt;
274
052a7b1d40e5 nginx 0.5.7
Igor Sysoev <http://sysoev.ru>
parents: 272
diff changeset
248
052a7b1d40e5 nginx 0.5.7
Igor Sysoev <http://sysoev.ru>
parents: 272
diff changeset
249 for ( ;; ) {
052a7b1d40e5 nginx 0.5.7
Igor Sysoev <http://sysoev.ru>
parents: 272
diff changeset
250
052a7b1d40e5 nginx 0.5.7
Igor Sysoev <http://sysoev.ru>
parents: 272
diff changeset
251 if (node->key < temp->key) {
052a7b1d40e5 nginx 0.5.7
Igor Sysoev <http://sysoev.ru>
parents: 272
diff changeset
252
356
b743d290eb3b nginx 0.6.22
Igor Sysoev <http://sysoev.ru>
parents: 328
diff changeset
253 p = &temp->left;
274
052a7b1d40e5 nginx 0.5.7
Igor Sysoev <http://sysoev.ru>
parents: 272
diff changeset
254
052a7b1d40e5 nginx 0.5.7
Igor Sysoev <http://sysoev.ru>
parents: 272
diff changeset
255 } else if (node->key > temp->key) {
052a7b1d40e5 nginx 0.5.7
Igor Sysoev <http://sysoev.ru>
parents: 272
diff changeset
256
356
b743d290eb3b nginx 0.6.22
Igor Sysoev <http://sysoev.ru>
parents: 328
diff changeset
257 p = &temp->right;
274
052a7b1d40e5 nginx 0.5.7
Igor Sysoev <http://sysoev.ru>
parents: 272
diff changeset
258
052a7b1d40e5 nginx 0.5.7
Igor Sysoev <http://sysoev.ru>
parents: 272
diff changeset
259 } else { /* node->key == temp->key */
052a7b1d40e5 nginx 0.5.7
Igor Sysoev <http://sysoev.ru>
parents: 272
diff changeset
260
052a7b1d40e5 nginx 0.5.7
Igor Sysoev <http://sysoev.ru>
parents: 272
diff changeset
261 lzn = (ngx_http_limit_zone_node_t *) &node->color;
052a7b1d40e5 nginx 0.5.7
Igor Sysoev <http://sysoev.ru>
parents: 272
diff changeset
262 lznt = (ngx_http_limit_zone_node_t *) &temp->color;
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 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
265 ? &temp->left : &temp->right;
b743d290eb3b nginx 0.6.22
Igor Sysoev <http://sysoev.ru>
parents: 328
diff changeset
266 }
274
052a7b1d40e5 nginx 0.5.7
Igor Sysoev <http://sysoev.ru>
parents: 272
diff changeset
267
356
b743d290eb3b nginx 0.6.22
Igor Sysoev <http://sysoev.ru>
parents: 328
diff changeset
268 if (*p == sentinel) {
b743d290eb3b nginx 0.6.22
Igor Sysoev <http://sysoev.ru>
parents: 328
diff changeset
269 break;
b743d290eb3b nginx 0.6.22
Igor Sysoev <http://sysoev.ru>
parents: 328
diff changeset
270 }
274
052a7b1d40e5 nginx 0.5.7
Igor Sysoev <http://sysoev.ru>
parents: 272
diff changeset
271
356
b743d290eb3b nginx 0.6.22
Igor Sysoev <http://sysoev.ru>
parents: 328
diff changeset
272 temp = *p;
274
052a7b1d40e5 nginx 0.5.7
Igor Sysoev <http://sysoev.ru>
parents: 272
diff changeset
273 }
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 = node;
274
052a7b1d40e5 nginx 0.5.7
Igor Sysoev <http://sysoev.ru>
parents: 272
diff changeset
276 node->parent = temp;
052a7b1d40e5 nginx 0.5.7
Igor Sysoev <http://sysoev.ru>
parents: 272
diff changeset
277 node->left = sentinel;
052a7b1d40e5 nginx 0.5.7
Igor Sysoev <http://sysoev.ru>
parents: 272
diff changeset
278 node->right = sentinel;
052a7b1d40e5 nginx 0.5.7
Igor Sysoev <http://sysoev.ru>
parents: 272
diff changeset
279 ngx_rbt_red(node);
052a7b1d40e5 nginx 0.5.7
Igor Sysoev <http://sysoev.ru>
parents: 272
diff changeset
280 }
052a7b1d40e5 nginx 0.5.7
Igor Sysoev <http://sysoev.ru>
parents: 272
diff changeset
281
052a7b1d40e5 nginx 0.5.7
Igor Sysoev <http://sysoev.ru>
parents: 272
diff changeset
282
052a7b1d40e5 nginx 0.5.7
Igor Sysoev <http://sysoev.ru>
parents: 272
diff changeset
283 static void
272
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
284 ngx_http_limit_zone_cleanup(void *data)
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
285 {
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
286 ngx_http_limit_zone_cleanup_t *lzcln = data;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
287
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
288 ngx_slab_pool_t *shpool;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
289 ngx_rbtree_node_t *node;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
290 ngx_http_limit_zone_ctx_t *ctx;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
291 ngx_http_limit_zone_node_t *lz;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
292
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
293 ctx = lzcln->shm_zone->data;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
294 shpool = (ngx_slab_pool_t *) lzcln->shm_zone->shm.addr;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
295 node = lzcln->node;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
296 lz = (ngx_http_limit_zone_node_t *) &node->color;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
297
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
298 ngx_shmtx_lock(&shpool->mutex);
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
299
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
300 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
301 "limit zone cleanup: %08XD %d", node->key, lz->conn);
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
302
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
303 lz->conn--;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
304
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
305 if (lz->conn == 0) {
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
306 ngx_rbtree_delete(ctx->rbtree, node);
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
307 ngx_slab_free_locked(shpool, node);
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
308 }
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
309
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
310 ngx_shmtx_unlock(&shpool->mutex);
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
311 }
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
312
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
313
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
314 static ngx_int_t
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
315 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
316 {
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
317 ngx_http_limit_zone_ctx_t *octx = data;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
318
468
56baf312c1b5 nginx 0.7.46
Igor Sysoev <http://sysoev.ru>
parents: 426
diff changeset
319 size_t len;
272
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
320 ngx_slab_pool_t *shpool;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
321 ngx_rbtree_node_t *sentinel;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
322 ngx_http_limit_zone_ctx_t *ctx;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
323
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
324 ctx = shm_zone->data;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
325
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
326 if (octx) {
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
327 if (ngx_strcmp(ctx->var.data, octx->var.data) != 0) {
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
328 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
329 "limit_zone \"%V\" uses the \"%V\" variable "
272
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
330 "while previously it used the \"%V\" variable",
480
549994537f15 nginx 0.7.52
Igor Sysoev <http://sysoev.ru>
parents: 468
diff changeset
331 &shm_zone->shm.name, &ctx->var, &octx->var);
272
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
332 return NGX_ERROR;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
333 }
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 ctx->rbtree = octx->rbtree;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
336
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
337 return NGX_OK;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
338 }
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
339
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
340 shpool = (ngx_slab_pool_t *) shm_zone->shm.addr;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
341
480
549994537f15 nginx 0.7.52
Igor Sysoev <http://sysoev.ru>
parents: 468
diff changeset
342 if (shm_zone->shm.exists) {
549994537f15 nginx 0.7.52
Igor Sysoev <http://sysoev.ru>
parents: 468
diff changeset
343 ctx->rbtree = shpool->data;
549994537f15 nginx 0.7.52
Igor Sysoev <http://sysoev.ru>
parents: 468
diff changeset
344
549994537f15 nginx 0.7.52
Igor Sysoev <http://sysoev.ru>
parents: 468
diff changeset
345 return NGX_OK;
549994537f15 nginx 0.7.52
Igor Sysoev <http://sysoev.ru>
parents: 468
diff changeset
346 }
549994537f15 nginx 0.7.52
Igor Sysoev <http://sysoev.ru>
parents: 468
diff changeset
347
272
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
348 ctx->rbtree = ngx_slab_alloc(shpool, sizeof(ngx_rbtree_t));
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
349 if (ctx->rbtree == NULL) {
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
480
549994537f15 nginx 0.7.52
Igor Sysoev <http://sysoev.ru>
parents: 468
diff changeset
353 shpool->data = ctx->rbtree;
549994537f15 nginx 0.7.52
Igor Sysoev <http://sysoev.ru>
parents: 468
diff changeset
354
272
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
355 sentinel = ngx_slab_alloc(shpool, sizeof(ngx_rbtree_node_t));
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
356 if (sentinel == NULL) {
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
357 return NGX_ERROR;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
358 }
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
359
356
b743d290eb3b nginx 0.6.22
Igor Sysoev <http://sysoev.ru>
parents: 328
diff changeset
360 ngx_rbtree_init(ctx->rbtree, sentinel,
b743d290eb3b nginx 0.6.22
Igor Sysoev <http://sysoev.ru>
parents: 328
diff changeset
361 ngx_http_limit_zone_rbtree_insert_value);
272
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
362
480
549994537f15 nginx 0.7.52
Igor Sysoev <http://sysoev.ru>
parents: 468
diff changeset
363 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
364
56baf312c1b5 nginx 0.7.46
Igor Sysoev <http://sysoev.ru>
parents: 426
diff changeset
365 shpool->log_ctx = ngx_slab_alloc(shpool, len);
56baf312c1b5 nginx 0.7.46
Igor Sysoev <http://sysoev.ru>
parents: 426
diff changeset
366 if (shpool->log_ctx == NULL) {
56baf312c1b5 nginx 0.7.46
Igor Sysoev <http://sysoev.ru>
parents: 426
diff changeset
367 return NGX_ERROR;
56baf312c1b5 nginx 0.7.46
Igor Sysoev <http://sysoev.ru>
parents: 426
diff changeset
368 }
56baf312c1b5 nginx 0.7.46
Igor Sysoev <http://sysoev.ru>
parents: 426
diff changeset
369
480
549994537f15 nginx 0.7.52
Igor Sysoev <http://sysoev.ru>
parents: 468
diff changeset
370 ngx_sprintf(shpool->log_ctx, " in limit_zone \"%V\"%Z",
549994537f15 nginx 0.7.52
Igor Sysoev <http://sysoev.ru>
parents: 468
diff changeset
371 &shm_zone->shm.name);
468
56baf312c1b5 nginx 0.7.46
Igor Sysoev <http://sysoev.ru>
parents: 426
diff changeset
372
272
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
373 return NGX_OK;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
374 }
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
375
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 static void *
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
378 ngx_http_limit_zone_create_conf(ngx_conf_t *cf)
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
379 {
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
380 ngx_http_limit_zone_conf_t *conf;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
381
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
382 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
383 if (conf == NULL) {
496
f39b9e29530d nginx 0.8.0
Igor Sysoev <http://sysoev.ru>
parents: 480
diff changeset
384 return NULL;
272
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
385 }
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
386
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
387 /*
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
388 * set by ngx_pcalloc():
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
389 *
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
390 * conf->shm_zone = NULL;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
391 * conf->conn = 0;
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 return conf;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
395 }
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
396
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 static char *
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
399 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
400 {
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
401 ngx_http_limit_zone_conf_t *prev = parent;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
402 ngx_http_limit_zone_conf_t *conf = child;
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 if (conf->shm_zone == NULL) {
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
405 *conf = *prev;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
406 }
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 return NGX_CONF_OK;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
409 }
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
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
412 static char *
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
413 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
414 {
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
415 ssize_t n;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
416 ngx_str_t *value;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
417 ngx_shm_zone_t *shm_zone;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
418 ngx_http_limit_zone_ctx_t *ctx;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
419
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
420 value = cf->args->elts;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
421
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
422 if (value[2].data[0] != '$') {
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
423 ngx_conf_log_error(NGX_LOG_EMERG, cf, 0,
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
424 "invalid variable name \"%V\"", &value[2]);
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
425 return NGX_CONF_ERROR;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
426 }
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 value[2].len--;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
429 value[2].data++;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
430
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
431 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
432 if (ctx == NULL) {
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
433 return NGX_CONF_ERROR;
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
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
436 ctx->index = ngx_http_get_variable_index(cf, &value[2]);
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
437 if (ctx->index == NGX_ERROR) {
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
438 return NGX_CONF_ERROR;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
439 }
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
440
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
441 ctx->var = value[2];
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 n = ngx_parse_size(&value[3]);
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 (n == NGX_ERROR) {
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 size of limit_zone \"%V\"", &value[3]);
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 if (n < (ngx_int_t) (8 * ngx_pagesize)) {
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
452 ngx_conf_log_error(NGX_LOG_EMERG, cf, 0,
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
453 "limit_zone \"%V\" is too small", &value[1]);
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
454 return NGX_CONF_ERROR;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
455 }
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
456
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 shm_zone = ngx_shared_memory_add(cf, &value[1], n,
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
459 &ngx_http_limit_zone_module);
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
460 if (shm_zone == NULL) {
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 if (shm_zone->data) {
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
465 ctx = shm_zone->data;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
466
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
467 ngx_conf_log_error(NGX_LOG_EMERG, cf, 0,
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
468 "limit_zone \"%V\" is already bound to variable \"%V\"",
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
469 &value[1], &ctx->var);
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
470 return NGX_CONF_ERROR;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
471 }
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 shm_zone->init = ngx_http_limit_zone_init_zone;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
474 shm_zone->data = ctx;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
475
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
476 return NGX_CONF_OK;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
477 }
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 static char *
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
481 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
482 {
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
483 ngx_http_limit_zone_conf_t *lzcf = conf;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
484
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
485 ngx_int_t n;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
486 ngx_str_t *value;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
487
426
e7dbea1ee115 nginx 0.7.25
Igor Sysoev <http://sysoev.ru>
parents: 356
diff changeset
488 if (lzcf->shm_zone) {
e7dbea1ee115 nginx 0.7.25
Igor Sysoev <http://sysoev.ru>
parents: 356
diff changeset
489 return "is duplicate";
e7dbea1ee115 nginx 0.7.25
Igor Sysoev <http://sysoev.ru>
parents: 356
diff changeset
490 }
e7dbea1ee115 nginx 0.7.25
Igor Sysoev <http://sysoev.ru>
parents: 356
diff changeset
491
272
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
492 value = cf->args->elts;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
493
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
494 lzcf->shm_zone = ngx_shared_memory_add(cf, &value[1], 0,
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
495 &ngx_http_limit_zone_module);
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
496 if (lzcf->shm_zone == NULL) {
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
497 return NGX_CONF_ERROR;
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
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
500 n = ngx_atoi(value[2].data, value[2].len);
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
501 if (n <= 0) {
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
502 ngx_conf_log_error(NGX_LOG_EMERG, cf, 0,
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
503 "invalid number of connections \"%V\"", &value[2]);
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
504 return NGX_CONF_ERROR;
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
274
052a7b1d40e5 nginx 0.5.7
Igor Sysoev <http://sysoev.ru>
parents: 272
diff changeset
507 if (n > 65535) {
052a7b1d40e5 nginx 0.5.7
Igor Sysoev <http://sysoev.ru>
parents: 272
diff changeset
508 ngx_conf_log_error(NGX_LOG_EMERG, cf, 0,
052a7b1d40e5 nginx 0.5.7
Igor Sysoev <http://sysoev.ru>
parents: 272
diff changeset
509 "connection limit must be less 65536");
052a7b1d40e5 nginx 0.5.7
Igor Sysoev <http://sysoev.ru>
parents: 272
diff changeset
510 return NGX_CONF_ERROR;
052a7b1d40e5 nginx 0.5.7
Igor Sysoev <http://sysoev.ru>
parents: 272
diff changeset
511 }
052a7b1d40e5 nginx 0.5.7
Igor Sysoev <http://sysoev.ru>
parents: 272
diff changeset
512
272
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
513 lzcf->conn = n;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
514
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
515 return NGX_CONF_OK;
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
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
518
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
519 static ngx_int_t
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
520 ngx_http_limit_zone_init(ngx_conf_t *cf)
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 ngx_http_handler_pt *h;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
523 ngx_http_core_main_conf_t *cmcf;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
524
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
525 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
526
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
527 h = ngx_array_push(&cmcf->phases[NGX_HTTP_PREACCESS_PHASE].handlers);
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
528 if (h == NULL) {
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
529 return NGX_ERROR;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
530 }
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
531
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
532 *h = ngx_http_limit_zone_handler;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
533
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
534 return NGX_OK;
29a6403156b0 nginx 0.5.6
Igor Sysoev <http://sysoev.ru>
parents:
diff changeset
535 }