nginx: src/stream/ngx_stream_limit_conn

annotate src/stream/ngx_stream_limit_conn_module.c @ 6693:3908156a51fa

Stream: phases.

author	Roman Arutyunyan <arut@nginx.com>
date	Thu, 15 Sep 2016 14:55:54 +0300
parents	8ed51b02f655
children	359b0ea2b067

rev	line source
6197 0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	1
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	2 /*
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	3 * Copyright (C) Igor Sysoev
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	4 * Copyright (C) Nginx, Inc.
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	5 */
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	6
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	7
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	8 #include <ngx_config.h>
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	9 #include <ngx_core.h>
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	10 #include <ngx_stream.h>
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	11
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	12
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	13 typedef struct {
6610 d5b5866c06c4 Stream: got rid of pseudo variables. Vladimir Homutov <vl@nginx.com> parents: 6606 diff changeset	14 u_char color;
d5b5866c06c4 Stream: got rid of pseudo variables. Vladimir Homutov <vl@nginx.com> parents: 6606 diff changeset	15 u_char len;
d5b5866c06c4 Stream: got rid of pseudo variables. Vladimir Homutov <vl@nginx.com> parents: 6606 diff changeset	16 u_short conn;
d5b5866c06c4 Stream: got rid of pseudo variables. Vladimir Homutov <vl@nginx.com> parents: 6606 diff changeset	17 u_char data[1];
6197 0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	18 } ngx_stream_limit_conn_node_t;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	19
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	20
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	21 typedef struct {
6610 d5b5866c06c4 Stream: got rid of pseudo variables. Vladimir Homutov <vl@nginx.com> parents: 6606 diff changeset	22 ngx_shm_zone_t *shm_zone;
d5b5866c06c4 Stream: got rid of pseudo variables. Vladimir Homutov <vl@nginx.com> parents: 6606 diff changeset	23 ngx_rbtree_node_t *node;
6197 0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	24 } ngx_stream_limit_conn_cleanup_t;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	25
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	26
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	27 typedef struct {
6610 d5b5866c06c4 Stream: got rid of pseudo variables. Vladimir Homutov <vl@nginx.com> parents: 6606 diff changeset	28 ngx_rbtree_t *rbtree;
d5b5866c06c4 Stream: got rid of pseudo variables. Vladimir Homutov <vl@nginx.com> parents: 6606 diff changeset	29 ngx_stream_complex_value_t key;
6197 0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	30 } ngx_stream_limit_conn_ctx_t;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	31
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	32
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	33 typedef struct {
6610 d5b5866c06c4 Stream: got rid of pseudo variables. Vladimir Homutov <vl@nginx.com> parents: 6606 diff changeset	34 ngx_shm_zone_t *shm_zone;
d5b5866c06c4 Stream: got rid of pseudo variables. Vladimir Homutov <vl@nginx.com> parents: 6606 diff changeset	35 ngx_uint_t conn;
6197 0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	36 } ngx_stream_limit_conn_limit_t;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	37
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	38
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	39 typedef struct {
6610 d5b5866c06c4 Stream: got rid of pseudo variables. Vladimir Homutov <vl@nginx.com> parents: 6606 diff changeset	40 ngx_array_t limits;
d5b5866c06c4 Stream: got rid of pseudo variables. Vladimir Homutov <vl@nginx.com> parents: 6606 diff changeset	41 ngx_uint_t log_level;
6197 0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	42 } ngx_stream_limit_conn_conf_t;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	43
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	44
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	45 static ngx_rbtree_node_t ngx_stream_limit_conn_lookup(ngx_rbtree_t rbtree,
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	46 ngx_str_t *key, uint32_t hash);
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	47 static void ngx_stream_limit_conn_cleanup(void *data);
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	48 static ngx_inline void ngx_stream_limit_conn_cleanup_all(ngx_pool_t *pool);
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	49
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	50 static void ngx_stream_limit_conn_create_conf(ngx_conf_t cf);
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	51 static char ngx_stream_limit_conn_merge_conf(ngx_conf_t cf, void *parent,
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	52 void *child);
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	53 static char ngx_stream_limit_conn_zone(ngx_conf_t cf, ngx_command_t *cmd,
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	54 void *conf);
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	55 static char ngx_stream_limit_conn(ngx_conf_t cf, ngx_command_t *cmd,
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	56 void *conf);
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	57 static ngx_int_t ngx_stream_limit_conn_init(ngx_conf_t *cf);
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	58
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	59
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	60 static ngx_conf_enum_t ngx_stream_limit_conn_log_levels[] = {
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	61 { ngx_string("info"), NGX_LOG_INFO },
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	62 { ngx_string("notice"), NGX_LOG_NOTICE },
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	63 { ngx_string("warn"), NGX_LOG_WARN },
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	64 { ngx_string("error"), NGX_LOG_ERR },
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	65 { ngx_null_string, 0 }
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	66 };
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	67
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	68
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	69 static ngx_command_t ngx_stream_limit_conn_commands[] = {
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	70
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	71 { ngx_string("limit_conn_zone"),
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	72 NGX_STREAM_MAIN_CONF\|NGX_CONF_TAKE2,
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	73 ngx_stream_limit_conn_zone,
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	74 0,
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	75 0,
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	76 NULL },
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	77
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	78 { ngx_string("limit_conn"),
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	79 NGX_STREAM_MAIN_CONF\|NGX_STREAM_SRV_CONF\|NGX_CONF_TAKE2,
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	80 ngx_stream_limit_conn,
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	81 NGX_STREAM_SRV_CONF_OFFSET,
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	82 0,
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	83 NULL },
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	84
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	85 { ngx_string("limit_conn_log_level"),
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	86 NGX_STREAM_MAIN_CONF\|NGX_STREAM_SRV_CONF\|NGX_CONF_TAKE1,
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	87 ngx_conf_set_enum_slot,
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	88 NGX_STREAM_SRV_CONF_OFFSET,
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	89 offsetof(ngx_stream_limit_conn_conf_t, log_level),
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	90 &ngx_stream_limit_conn_log_levels },
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	91
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	92 ngx_null_command
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	93 };
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	94
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	95
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	96 static ngx_stream_module_t ngx_stream_limit_conn_module_ctx = {
6606 2f41d383c9c7 Stream: added preconfiguration step. Vladimir Homutov <vl@nginx.com> parents: 6601 diff changeset	97 NULL, /* preconfiguration */
6197 0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	98 ngx_stream_limit_conn_init, /* postconfiguration */
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	99
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	100 NULL, /* create main configuration */
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	101 NULL, /* init main configuration */
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	102
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	103 ngx_stream_limit_conn_create_conf, /* create server configuration */
6629 8ed51b02f655 Stream: style. Vladimir Homutov <vl@nginx.com> parents: 6610 diff changeset	104 ngx_stream_limit_conn_merge_conf /* merge server configuration */
6197 0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	105 };
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	106
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	107
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	108 ngx_module_t ngx_stream_limit_conn_module = {
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	109 NGX_MODULE_V1,
6601 d452cb27639f Style. Roman Arutyunyan <arut@nginx.com> parents: 6480 diff changeset	110 &ngx_stream_limit_conn_module_ctx, /* module context */
d452cb27639f Style. Roman Arutyunyan <arut@nginx.com> parents: 6480 diff changeset	111 ngx_stream_limit_conn_commands, /* module directives */
d452cb27639f Style. Roman Arutyunyan <arut@nginx.com> parents: 6480 diff changeset	112 NGX_STREAM_MODULE, /* module type */
d452cb27639f Style. Roman Arutyunyan <arut@nginx.com> parents: 6480 diff changeset	113 NULL, /* init master */
d452cb27639f Style. Roman Arutyunyan <arut@nginx.com> parents: 6480 diff changeset	114 NULL, /* init module */
d452cb27639f Style. Roman Arutyunyan <arut@nginx.com> parents: 6480 diff changeset	115 NULL, /* init process */
d452cb27639f Style. Roman Arutyunyan <arut@nginx.com> parents: 6480 diff changeset	116 NULL, /* init thread */
d452cb27639f Style. Roman Arutyunyan <arut@nginx.com> parents: 6480 diff changeset	117 NULL, /* exit thread */
d452cb27639f Style. Roman Arutyunyan <arut@nginx.com> parents: 6480 diff changeset	118 NULL, /* exit process */
d452cb27639f Style. Roman Arutyunyan <arut@nginx.com> parents: 6480 diff changeset	119 NULL, /* exit master */
6197 0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	120 NGX_MODULE_V1_PADDING
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	121 };
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	122
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	123
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	124 static ngx_int_t
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	125 ngx_stream_limit_conn_handler(ngx_stream_session_t *s)
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	126 {
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	127 size_t n;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	128 uint32_t hash;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	129 ngx_str_t key;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	130 ngx_uint_t i;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	131 ngx_slab_pool_t *shpool;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	132 ngx_rbtree_node_t *node;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	133 ngx_pool_cleanup_t *cln;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	134 ngx_stream_limit_conn_ctx_t *ctx;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	135 ngx_stream_limit_conn_node_t *lc;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	136 ngx_stream_limit_conn_conf_t *lccf;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	137 ngx_stream_limit_conn_limit_t *limits;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	138 ngx_stream_limit_conn_cleanup_t *lccln;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	139
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	140 lccf = ngx_stream_get_module_srv_conf(s, ngx_stream_limit_conn_module);
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	141 limits = lccf->limits.elts;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	142
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	143 for (i = 0; i < lccf->limits.nelts; i++) {
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	144 ctx = limits[i].shm_zone->data;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	145
6610 d5b5866c06c4 Stream: got rid of pseudo variables. Vladimir Homutov <vl@nginx.com> parents: 6606 diff changeset	146 if (ngx_stream_complex_value(s, &ctx->key, &key) != NGX_OK) {
d5b5866c06c4 Stream: got rid of pseudo variables. Vladimir Homutov <vl@nginx.com> parents: 6606 diff changeset	147 return NGX_ERROR;
d5b5866c06c4 Stream: got rid of pseudo variables. Vladimir Homutov <vl@nginx.com> parents: 6606 diff changeset	148 }
d5b5866c06c4 Stream: got rid of pseudo variables. Vladimir Homutov <vl@nginx.com> parents: 6606 diff changeset	149
d5b5866c06c4 Stream: got rid of pseudo variables. Vladimir Homutov <vl@nginx.com> parents: 6606 diff changeset	150 if (key.len == 0) {
d5b5866c06c4 Stream: got rid of pseudo variables. Vladimir Homutov <vl@nginx.com> parents: 6606 diff changeset	151 continue;
d5b5866c06c4 Stream: got rid of pseudo variables. Vladimir Homutov <vl@nginx.com> parents: 6606 diff changeset	152 }
d5b5866c06c4 Stream: got rid of pseudo variables. Vladimir Homutov <vl@nginx.com> parents: 6606 diff changeset	153
d5b5866c06c4 Stream: got rid of pseudo variables. Vladimir Homutov <vl@nginx.com> parents: 6606 diff changeset	154 if (key.len > 255) {
d5b5866c06c4 Stream: got rid of pseudo variables. Vladimir Homutov <vl@nginx.com> parents: 6606 diff changeset	155 ngx_log_error(NGX_LOG_ERR, s->connection->log, 0,
d5b5866c06c4 Stream: got rid of pseudo variables. Vladimir Homutov <vl@nginx.com> parents: 6606 diff changeset	156 "the value of the \"%V\" key "
d5b5866c06c4 Stream: got rid of pseudo variables. Vladimir Homutov <vl@nginx.com> parents: 6606 diff changeset	157 "is more than 255 bytes: \"%V\"",
d5b5866c06c4 Stream: got rid of pseudo variables. Vladimir Homutov <vl@nginx.com> parents: 6606 diff changeset	158 &ctx->key.value, &key);
d5b5866c06c4 Stream: got rid of pseudo variables. Vladimir Homutov <vl@nginx.com> parents: 6606 diff changeset	159 continue;
d5b5866c06c4 Stream: got rid of pseudo variables. Vladimir Homutov <vl@nginx.com> parents: 6606 diff changeset	160 }
d5b5866c06c4 Stream: got rid of pseudo variables. Vladimir Homutov <vl@nginx.com> parents: 6606 diff changeset	161
d5b5866c06c4 Stream: got rid of pseudo variables. Vladimir Homutov <vl@nginx.com> parents: 6606 diff changeset	162 hash = ngx_crc32_short(key.data, key.len);
d5b5866c06c4 Stream: got rid of pseudo variables. Vladimir Homutov <vl@nginx.com> parents: 6606 diff changeset	163
6197 0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	164 shpool = (ngx_slab_pool_t *) limits[i].shm_zone->shm.addr;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	165
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	166 ngx_shmtx_lock(&shpool->mutex);
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	167
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	168 node = ngx_stream_limit_conn_lookup(ctx->rbtree, &key, hash);
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	169
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	170 if (node == NULL) {
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	171
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	172 n = offsetof(ngx_rbtree_node_t, color)
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	173 + offsetof(ngx_stream_limit_conn_node_t, data)
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	174 + key.len;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	175
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	176 node = ngx_slab_alloc_locked(shpool, n);
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	177
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	178 if (node == NULL) {
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	179 ngx_shmtx_unlock(&shpool->mutex);
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	180 ngx_stream_limit_conn_cleanup_all(s->connection->pool);
6693 3908156a51fa Stream: phases. Roman Arutyunyan <arut@nginx.com> parents: 6629 diff changeset	181 return NGX_STREAM_SERVICE_UNAVAILABLE;
6197 0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	182 }
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	183
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	184 lc = (ngx_stream_limit_conn_node_t *) &node->color;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	185
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	186 node->key = hash;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	187 lc->len = (u_char) key.len;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	188 lc->conn = 1;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	189 ngx_memcpy(lc->data, key.data, key.len);
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	190
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	191 ngx_rbtree_insert(ctx->rbtree, node);
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	192
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	193 } else {
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	194
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	195 lc = (ngx_stream_limit_conn_node_t *) &node->color;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	196
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	197 if ((ngx_uint_t) lc->conn >= limits[i].conn) {
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	198
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	199 ngx_shmtx_unlock(&shpool->mutex);
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	200
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	201 ngx_log_error(lccf->log_level, s->connection->log, 0,
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	202 "limiting connections by zone \"%V\"",
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	203 &limits[i].shm_zone->shm.name);
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	204
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	205 ngx_stream_limit_conn_cleanup_all(s->connection->pool);
6693 3908156a51fa Stream: phases. Roman Arutyunyan <arut@nginx.com> parents: 6629 diff changeset	206 return NGX_STREAM_SERVICE_UNAVAILABLE;
6197 0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	207 }
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	208
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	209 lc->conn++;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	210 }
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	211
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	212 ngx_log_debug2(NGX_LOG_DEBUG_STREAM, s->connection->log, 0,
6480 f01ab2dbcfdc Fixed logging. Sergey Kandaurov <pluknet@nginx.com> parents: 6197 diff changeset	213 "limit conn: %08Xi %d", node->key, lc->conn);
6197 0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	214
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	215 ngx_shmtx_unlock(&shpool->mutex);
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	216
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	217 cln = ngx_pool_cleanup_add(s->connection->pool,
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	218 sizeof(ngx_stream_limit_conn_cleanup_t));
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	219 if (cln == NULL) {
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	220 return NGX_ERROR;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	221 }
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	222
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	223 cln->handler = ngx_stream_limit_conn_cleanup;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	224 lccln = cln->data;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	225
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	226 lccln->shm_zone = limits[i].shm_zone;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	227 lccln->node = node;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	228 }
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	229
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	230 return NGX_DECLINED;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	231 }
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	232
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	233
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	234 static void
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	235 ngx_stream_limit_conn_rbtree_insert_value(ngx_rbtree_node_t *temp,
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	236 ngx_rbtree_node_t node, ngx_rbtree_node_t sentinel)
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	237 {
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	238 ngx_rbtree_node_t **p;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	239 ngx_stream_limit_conn_node_t lcn, lcnt;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	240
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	241 for ( ;; ) {
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	242
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	243 if (node->key < temp->key) {
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	244
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	245 p = &temp->left;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	246
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	247 } else if (node->key > temp->key) {
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	248
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	249 p = &temp->right;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	250
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	251 } else { /* node->key == temp->key */
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	252
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	253 lcn = (ngx_stream_limit_conn_node_t *) &node->color;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	254 lcnt = (ngx_stream_limit_conn_node_t *) &temp->color;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	255
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	256 p = (ngx_memn2cmp(lcn->data, lcnt->data, lcn->len, lcnt->len) < 0)
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	257 ? &temp->left : &temp->right;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	258 }
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	259
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	260 if (*p == sentinel) {
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	261 break;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	262 }
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	263
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	264 temp = *p;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	265 }
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	266
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	267 *p = node;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	268 node->parent = temp;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	269 node->left = sentinel;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	270 node->right = sentinel;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	271 ngx_rbt_red(node);
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	272 }
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	273
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	274
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	275 static ngx_rbtree_node_t *
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	276 ngx_stream_limit_conn_lookup(ngx_rbtree_t rbtree, ngx_str_t key,
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	277 uint32_t hash)
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	278 {
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	279 ngx_int_t rc;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	280 ngx_rbtree_node_t node, sentinel;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	281 ngx_stream_limit_conn_node_t *lcn;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	282
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	283 node = rbtree->root;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	284 sentinel = rbtree->sentinel;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	285
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	286 while (node != sentinel) {
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	287
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	288 if (hash < node->key) {
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	289 node = node->left;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	290 continue;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	291 }
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	292
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	293 if (hash > node->key) {
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	294 node = node->right;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	295 continue;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	296 }
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	297
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	298 /* hash == node->key */
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	299
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	300 lcn = (ngx_stream_limit_conn_node_t *) &node->color;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	301
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	302 rc = ngx_memn2cmp(key->data, lcn->data, key->len, (size_t) lcn->len);
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	303
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	304 if (rc == 0) {
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	305 return node;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	306 }
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	307
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	308 node = (rc < 0) ? node->left : node->right;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	309 }
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	310
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	311 return NULL;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	312 }
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	313
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	314
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	315 static void
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	316 ngx_stream_limit_conn_cleanup(void *data)
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	317 {
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	318 ngx_stream_limit_conn_cleanup_t *lccln = data;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	319
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	320 ngx_slab_pool_t *shpool;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	321 ngx_rbtree_node_t *node;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	322 ngx_stream_limit_conn_ctx_t *ctx;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	323 ngx_stream_limit_conn_node_t *lc;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	324
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	325 ctx = lccln->shm_zone->data;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	326 shpool = (ngx_slab_pool_t *) lccln->shm_zone->shm.addr;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	327 node = lccln->node;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	328 lc = (ngx_stream_limit_conn_node_t *) &node->color;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	329
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	330 ngx_shmtx_lock(&shpool->mutex);
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	331
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	332 ngx_log_debug2(NGX_LOG_DEBUG_STREAM, lccln->shm_zone->shm.log, 0,
6480 f01ab2dbcfdc Fixed logging. Sergey Kandaurov <pluknet@nginx.com> parents: 6197 diff changeset	333 "limit conn cleanup: %08Xi %d", node->key, lc->conn);
6197 0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	334
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	335 lc->conn--;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	336
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	337 if (lc->conn == 0) {
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	338 ngx_rbtree_delete(ctx->rbtree, node);
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	339 ngx_slab_free_locked(shpool, node);
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	340 }
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	341
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	342 ngx_shmtx_unlock(&shpool->mutex);
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	343 }
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	344
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	345
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	346 static ngx_inline void
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	347 ngx_stream_limit_conn_cleanup_all(ngx_pool_t *pool)
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	348 {
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	349 ngx_pool_cleanup_t *cln;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	350
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	351 cln = pool->cleanup;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	352
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	353 while (cln && cln->handler == ngx_stream_limit_conn_cleanup) {
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	354 ngx_stream_limit_conn_cleanup(cln->data);
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	355 cln = cln->next;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	356 }
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	357
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	358 pool->cleanup = cln;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	359 }
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	360
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	361
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	362 static ngx_int_t
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	363 ngx_stream_limit_conn_init_zone(ngx_shm_zone_t shm_zone, void data)
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	364 {
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	365 ngx_stream_limit_conn_ctx_t *octx = data;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	366
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	367 size_t len;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	368 ngx_slab_pool_t *shpool;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	369 ngx_rbtree_node_t *sentinel;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	370 ngx_stream_limit_conn_ctx_t *ctx;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	371
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	372 ctx = shm_zone->data;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	373
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	374 if (octx) {
6610 d5b5866c06c4 Stream: got rid of pseudo variables. Vladimir Homutov <vl@nginx.com> parents: 6606 diff changeset	375 if (ctx->key.value.len != octx->key.value.len
d5b5866c06c4 Stream: got rid of pseudo variables. Vladimir Homutov <vl@nginx.com> parents: 6606 diff changeset	376 \|\| ngx_strncmp(ctx->key.value.data, octx->key.value.data,
d5b5866c06c4 Stream: got rid of pseudo variables. Vladimir Homutov <vl@nginx.com> parents: 6606 diff changeset	377 ctx->key.value.len)
d5b5866c06c4 Stream: got rid of pseudo variables. Vladimir Homutov <vl@nginx.com> parents: 6606 diff changeset	378 != 0)
d5b5866c06c4 Stream: got rid of pseudo variables. Vladimir Homutov <vl@nginx.com> parents: 6606 diff changeset	379 {
d5b5866c06c4 Stream: got rid of pseudo variables. Vladimir Homutov <vl@nginx.com> parents: 6606 diff changeset	380 ngx_log_error(NGX_LOG_EMERG, shm_zone->shm.log, 0,
d5b5866c06c4 Stream: got rid of pseudo variables. Vladimir Homutov <vl@nginx.com> parents: 6606 diff changeset	381 "limit_conn_zone \"%V\" uses the \"%V\" key "
d5b5866c06c4 Stream: got rid of pseudo variables. Vladimir Homutov <vl@nginx.com> parents: 6606 diff changeset	382 "while previously it used the \"%V\" key",
d5b5866c06c4 Stream: got rid of pseudo variables. Vladimir Homutov <vl@nginx.com> parents: 6606 diff changeset	383 &shm_zone->shm.name, &ctx->key.value,
d5b5866c06c4 Stream: got rid of pseudo variables. Vladimir Homutov <vl@nginx.com> parents: 6606 diff changeset	384 &octx->key.value);
d5b5866c06c4 Stream: got rid of pseudo variables. Vladimir Homutov <vl@nginx.com> parents: 6606 diff changeset	385 return NGX_ERROR;
d5b5866c06c4 Stream: got rid of pseudo variables. Vladimir Homutov <vl@nginx.com> parents: 6606 diff changeset	386 }
d5b5866c06c4 Stream: got rid of pseudo variables. Vladimir Homutov <vl@nginx.com> parents: 6606 diff changeset	387
6197 0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	388 ctx->rbtree = octx->rbtree;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	389
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	390 return NGX_OK;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	391 }
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	392
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	393 shpool = (ngx_slab_pool_t *) shm_zone->shm.addr;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	394
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	395 if (shm_zone->shm.exists) {
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	396 ctx->rbtree = shpool->data;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	397
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	398 return NGX_OK;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	399 }
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	400
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	401 ctx->rbtree = ngx_slab_alloc(shpool, sizeof(ngx_rbtree_t));
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	402 if (ctx->rbtree == NULL) {
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	403 return NGX_ERROR;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	404 }
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	405
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	406 shpool->data = ctx->rbtree;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	407
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	408 sentinel = ngx_slab_alloc(shpool, sizeof(ngx_rbtree_node_t));
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	409 if (sentinel == NULL) {
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	410 return NGX_ERROR;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	411 }
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	412
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	413 ngx_rbtree_init(ctx->rbtree, sentinel,
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	414 ngx_stream_limit_conn_rbtree_insert_value);
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	415
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	416 len = sizeof(" in limit_conn_zone \"\"") + shm_zone->shm.name.len;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	417
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	418 shpool->log_ctx = ngx_slab_alloc(shpool, len);
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	419 if (shpool->log_ctx == NULL) {
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	420 return NGX_ERROR;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	421 }
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	422
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	423 ngx_sprintf(shpool->log_ctx, " in limit_conn_zone \"%V\"%Z",
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	424 &shm_zone->shm.name);
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	425
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	426 return NGX_OK;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	427 }
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	428
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	429
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	430 static void *
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	431 ngx_stream_limit_conn_create_conf(ngx_conf_t *cf)
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	432 {
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	433 ngx_stream_limit_conn_conf_t *conf;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	434
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	435 conf = ngx_pcalloc(cf->pool, sizeof(ngx_stream_limit_conn_conf_t));
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	436 if (conf == NULL) {
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	437 return NULL;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	438 }
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	439
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	440 /*
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	441 * set by ngx_pcalloc():
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	442 *
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	443 * conf->limits.elts = NULL;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	444 */
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	445
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	446 conf->log_level = NGX_CONF_UNSET_UINT;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	447
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	448 return conf;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	449 }
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	450
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	451
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	452 static char *
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	453 ngx_stream_limit_conn_merge_conf(ngx_conf_t cf, void parent, void *child)
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	454 {
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	455 ngx_stream_limit_conn_conf_t *prev = parent;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	456 ngx_stream_limit_conn_conf_t *conf = child;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	457
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	458 if (conf->limits.elts == NULL) {
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	459 conf->limits = prev->limits;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	460 }
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	461
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	462 ngx_conf_merge_uint_value(conf->log_level, prev->log_level, NGX_LOG_ERR);
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	463
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	464 return NGX_CONF_OK;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	465 }
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	466
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	467
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	468 static char *
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	469 ngx_stream_limit_conn_zone(ngx_conf_t cf, ngx_command_t cmd, void *conf)
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	470 {
6610 d5b5866c06c4 Stream: got rid of pseudo variables. Vladimir Homutov <vl@nginx.com> parents: 6606 diff changeset	471 u_char *p;
d5b5866c06c4 Stream: got rid of pseudo variables. Vladimir Homutov <vl@nginx.com> parents: 6606 diff changeset	472 ssize_t size;
d5b5866c06c4 Stream: got rid of pseudo variables. Vladimir Homutov <vl@nginx.com> parents: 6606 diff changeset	473 ngx_str_t *value, name, s;
d5b5866c06c4 Stream: got rid of pseudo variables. Vladimir Homutov <vl@nginx.com> parents: 6606 diff changeset	474 ngx_uint_t i;
d5b5866c06c4 Stream: got rid of pseudo variables. Vladimir Homutov <vl@nginx.com> parents: 6606 diff changeset	475 ngx_shm_zone_t *shm_zone;
d5b5866c06c4 Stream: got rid of pseudo variables. Vladimir Homutov <vl@nginx.com> parents: 6606 diff changeset	476 ngx_stream_limit_conn_ctx_t *ctx;
d5b5866c06c4 Stream: got rid of pseudo variables. Vladimir Homutov <vl@nginx.com> parents: 6606 diff changeset	477 ngx_stream_compile_complex_value_t ccv;
6197 0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	478
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	479 value = cf->args->elts;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	480
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	481 ctx = ngx_pcalloc(cf->pool, sizeof(ngx_stream_limit_conn_ctx_t));
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	482 if (ctx == NULL) {
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	483 return NGX_CONF_ERROR;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	484 }
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	485
6610 d5b5866c06c4 Stream: got rid of pseudo variables. Vladimir Homutov <vl@nginx.com> parents: 6606 diff changeset	486 ngx_memzero(&ccv, sizeof(ngx_stream_compile_complex_value_t));
d5b5866c06c4 Stream: got rid of pseudo variables. Vladimir Homutov <vl@nginx.com> parents: 6606 diff changeset	487
d5b5866c06c4 Stream: got rid of pseudo variables. Vladimir Homutov <vl@nginx.com> parents: 6606 diff changeset	488 ccv.cf = cf;
d5b5866c06c4 Stream: got rid of pseudo variables. Vladimir Homutov <vl@nginx.com> parents: 6606 diff changeset	489 ccv.value = &value[1];
d5b5866c06c4 Stream: got rid of pseudo variables. Vladimir Homutov <vl@nginx.com> parents: 6606 diff changeset	490 ccv.complex_value = &ctx->key;
d5b5866c06c4 Stream: got rid of pseudo variables. Vladimir Homutov <vl@nginx.com> parents: 6606 diff changeset	491
d5b5866c06c4 Stream: got rid of pseudo variables. Vladimir Homutov <vl@nginx.com> parents: 6606 diff changeset	492 if (ngx_stream_compile_complex_value(&ccv) != NGX_OK) {
d5b5866c06c4 Stream: got rid of pseudo variables. Vladimir Homutov <vl@nginx.com> parents: 6606 diff changeset	493 return NGX_CONF_ERROR;
d5b5866c06c4 Stream: got rid of pseudo variables. Vladimir Homutov <vl@nginx.com> parents: 6606 diff changeset	494 }
d5b5866c06c4 Stream: got rid of pseudo variables. Vladimir Homutov <vl@nginx.com> parents: 6606 diff changeset	495
6197 0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	496 size = 0;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	497 name.len = 0;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	498
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	499 for (i = 2; i < cf->args->nelts; i++) {
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	500
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	501 if (ngx_strncmp(value[i].data, "zone=", 5) == 0) {
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	502
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	503 name.data = value[i].data + 5;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	504
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	505 p = (u_char *) ngx_strchr(name.data, ':');
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	506
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	507 if (p == NULL) {
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	508 ngx_conf_log_error(NGX_LOG_EMERG, cf, 0,
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	509 "invalid zone size \"%V\"", &value[i]);
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	510 return NGX_CONF_ERROR;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	511 }
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	512
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	513 name.len = p - name.data;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	514
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	515 s.data = p + 1;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	516 s.len = value[i].data + value[i].len - s.data;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	517
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	518 size = ngx_parse_size(&s);
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	519
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	520 if (size == NGX_ERROR) {
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	521 ngx_conf_log_error(NGX_LOG_EMERG, cf, 0,
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	522 "invalid zone size \"%V\"", &value[i]);
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	523 return NGX_CONF_ERROR;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	524 }
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	525
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	526 if (size < (ssize_t) (8 * ngx_pagesize)) {
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	527 ngx_conf_log_error(NGX_LOG_EMERG, cf, 0,
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	528 "zone \"%V\" is too small", &value[i]);
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	529 return NGX_CONF_ERROR;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	530 }
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	531
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	532 continue;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	533 }
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	534
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	535 ngx_conf_log_error(NGX_LOG_EMERG, cf, 0,
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	536 "invalid parameter \"%V\"", &value[i]);
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	537 return NGX_CONF_ERROR;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	538 }
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	539
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	540 if (name.len == 0) {
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	541 ngx_conf_log_error(NGX_LOG_EMERG, cf, 0,
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	542 "\"%V\" must have \"zone\" parameter",
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	543 &cmd->name);
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	544 return NGX_CONF_ERROR;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	545 }
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	546
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	547 shm_zone = ngx_shared_memory_add(cf, &name, size,
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	548 &ngx_stream_limit_conn_module);
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	549 if (shm_zone == NULL) {
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	550 return NGX_CONF_ERROR;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	551 }
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	552
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	553 if (shm_zone->data) {
6610 d5b5866c06c4 Stream: got rid of pseudo variables. Vladimir Homutov <vl@nginx.com> parents: 6606 diff changeset	554 ctx = shm_zone->data;
d5b5866c06c4 Stream: got rid of pseudo variables. Vladimir Homutov <vl@nginx.com> parents: 6606 diff changeset	555
6197 0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	556 ngx_conf_log_error(NGX_LOG_EMERG, cf, 0,
6610 d5b5866c06c4 Stream: got rid of pseudo variables. Vladimir Homutov <vl@nginx.com> parents: 6606 diff changeset	557 "%V \"%V\" is already bound to key \"%V\"",
d5b5866c06c4 Stream: got rid of pseudo variables. Vladimir Homutov <vl@nginx.com> parents: 6606 diff changeset	558 &cmd->name, &name, &ctx->key.value);
6197 0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	559 return NGX_CONF_ERROR;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	560 }
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	561
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	562 shm_zone->init = ngx_stream_limit_conn_init_zone;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	563 shm_zone->data = ctx;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	564
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	565 return NGX_CONF_OK;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	566 }
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	567
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	568
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	569 static char *
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	570 ngx_stream_limit_conn(ngx_conf_t cf, ngx_command_t cmd, void *conf)
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	571 {
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	572 ngx_shm_zone_t *shm_zone;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	573 ngx_stream_limit_conn_conf_t *lccf = conf;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	574 ngx_stream_limit_conn_limit_t limit, limits;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	575
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	576 ngx_str_t *value;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	577 ngx_int_t n;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	578 ngx_uint_t i;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	579
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	580 value = cf->args->elts;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	581
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	582 shm_zone = ngx_shared_memory_add(cf, &value[1], 0,
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	583 &ngx_stream_limit_conn_module);
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	584 if (shm_zone == NULL) {
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	585 return NGX_CONF_ERROR;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	586 }
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	587
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	588 limits = lccf->limits.elts;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	589
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	590 if (limits == NULL) {
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	591 if (ngx_array_init(&lccf->limits, cf->pool, 1,
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	592 sizeof(ngx_stream_limit_conn_limit_t))
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	593 != NGX_OK)
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	594 {
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	595 return NGX_CONF_ERROR;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	596 }
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	597 }
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	598
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	599 for (i = 0; i < lccf->limits.nelts; i++) {
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	600 if (shm_zone == limits[i].shm_zone) {
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	601 return "is duplicate";
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	602 }
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	603 }
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	604
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	605 n = ngx_atoi(value[2].data, value[2].len);
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	606 if (n <= 0) {
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	607 ngx_conf_log_error(NGX_LOG_EMERG, cf, 0,
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	608 "invalid number of connections \"%V\"", &value[2]);
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	609 return NGX_CONF_ERROR;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	610 }
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	611
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	612 if (n > 65535) {
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	613 ngx_conf_log_error(NGX_LOG_EMERG, cf, 0,
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	614 "connection limit must be less 65536");
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	615 return NGX_CONF_ERROR;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	616 }
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	617
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	618 limit = ngx_array_push(&lccf->limits);
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	619 if (limit == NULL) {
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	620 return NGX_CONF_ERROR;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	621 }
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	622
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	623 limit->conn = n;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	624 limit->shm_zone = shm_zone;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	625
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	626 return NGX_CONF_OK;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	627 }
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	628
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	629
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	630 static ngx_int_t
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	631 ngx_stream_limit_conn_init(ngx_conf_t *cf)
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	632 {
6693 3908156a51fa Stream: phases. Roman Arutyunyan <arut@nginx.com> parents: 6629 diff changeset	633 ngx_stream_handler_pt *h;
6197 0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	634 ngx_stream_core_main_conf_t *cmcf;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	635
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	636 cmcf = ngx_stream_conf_get_module_main_conf(cf, ngx_stream_core_module);
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	637
6693 3908156a51fa Stream: phases. Roman Arutyunyan <arut@nginx.com> parents: 6629 diff changeset	638 h = ngx_array_push(&cmcf->phases[NGX_STREAM_PREACCESS_PHASE].handlers);
3908156a51fa Stream: phases. Roman Arutyunyan <arut@nginx.com> parents: 6629 diff changeset	639 if (h == NULL) {
3908156a51fa Stream: phases. Roman Arutyunyan <arut@nginx.com> parents: 6629 diff changeset	640 return NGX_ERROR;
3908156a51fa Stream: phases. Roman Arutyunyan <arut@nginx.com> parents: 6629 diff changeset	641 }
3908156a51fa Stream: phases. Roman Arutyunyan <arut@nginx.com> parents: 6629 diff changeset	642
3908156a51fa Stream: phases. Roman Arutyunyan <arut@nginx.com> parents: 6629 diff changeset	643 *h = ngx_stream_limit_conn_handler;
6197 0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	644
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	645 return NGX_OK;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	646 }

Mercurial > hg > nginx

annotate src/stream/ngx_stream_limit_conn_module.c @ 6693:3908156a51fa