nginx: src/stream/ngx_stream_limit_conn

annotate src/stream/ngx_stream_limit_conn_module.c @ 6606:2f41d383c9c7

Stream: added preconfiguration step.

author	Vladimir Homutov <vl@nginx.com>
date	Wed, 15 Jun 2016 15:10:24 +0300
parents	d452cb27639f
children	d5b5866c06c4

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 {
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	14 u_char color;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	15 u_char len;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	16 u_short conn;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	17 u_char data[1];
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 {
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	22 ngx_shm_zone_t *shm_zone;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	23 ngx_rbtree_node_t *node;
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 {
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	28 ngx_rbtree_t *rbtree;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	29 } ngx_stream_limit_conn_ctx_t;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	30
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 typedef struct {
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	33 ngx_shm_zone_t *shm_zone;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	34 ngx_uint_t conn;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	35 } ngx_stream_limit_conn_limit_t;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	36
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 typedef struct {
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	39 ngx_array_t limits;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	40 ngx_uint_t log_level;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	41 } ngx_stream_limit_conn_conf_t;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	42
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 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	45 ngx_str_t *key, uint32_t hash);
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	46 static void ngx_stream_limit_conn_cleanup(void *data);
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	47 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	48
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	49 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	50 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	51 void *child);
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	52 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	53 void *conf);
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	54 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	55 void *conf);
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	56 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	57
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 static ngx_conf_enum_t ngx_stream_limit_conn_log_levels[] = {
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	60 { ngx_string("info"), NGX_LOG_INFO },
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	61 { ngx_string("notice"), NGX_LOG_NOTICE },
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	62 { ngx_string("warn"), NGX_LOG_WARN },
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	63 { ngx_string("error"), NGX_LOG_ERR },
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	64 { ngx_null_string, 0 }
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	65 };
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 static ngx_command_t ngx_stream_limit_conn_commands[] = {
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	69
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	70 { ngx_string("limit_conn_zone"),
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	71 NGX_STREAM_MAIN_CONF\|NGX_CONF_TAKE2,
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	72 ngx_stream_limit_conn_zone,
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	73 0,
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 NULL },
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	76
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	77 { ngx_string("limit_conn"),
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	78 NGX_STREAM_MAIN_CONF\|NGX_STREAM_SRV_CONF\|NGX_CONF_TAKE2,
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	79 ngx_stream_limit_conn,
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	80 NGX_STREAM_SRV_CONF_OFFSET,
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	81 0,
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	82 NULL },
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	83
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	84 { ngx_string("limit_conn_log_level"),
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	85 NGX_STREAM_MAIN_CONF\|NGX_STREAM_SRV_CONF\|NGX_CONF_TAKE1,
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	86 ngx_conf_set_enum_slot,
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	87 NGX_STREAM_SRV_CONF_OFFSET,
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	88 offsetof(ngx_stream_limit_conn_conf_t, log_level),
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	89 &ngx_stream_limit_conn_log_levels },
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	90
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	91 ngx_null_command
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	92 };
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 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	96 NULL, /* preconfiguration */
6197 0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	97 ngx_stream_limit_conn_init, /* postconfiguration */
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	98
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	99 NULL, /* create main configuration */
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	100 NULL, /* init main configuration */
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	101
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	102 ngx_stream_limit_conn_create_conf, /* create server configuration */
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	103 ngx_stream_limit_conn_merge_conf, /* merge server configuration */
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	104 };
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 ngx_module_t ngx_stream_limit_conn_module = {
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	108 NGX_MODULE_V1,
6601 d452cb27639f Style. Roman Arutyunyan <arut@nginx.com> parents: 6480 diff changeset	109 &ngx_stream_limit_conn_module_ctx, /* module context */
d452cb27639f Style. Roman Arutyunyan <arut@nginx.com> parents: 6480 diff changeset	110 ngx_stream_limit_conn_commands, /* module directives */
d452cb27639f Style. Roman Arutyunyan <arut@nginx.com> parents: 6480 diff changeset	111 NGX_STREAM_MODULE, /* module type */
d452cb27639f Style. Roman Arutyunyan <arut@nginx.com> parents: 6480 diff changeset	112 NULL, /* init master */
d452cb27639f Style. Roman Arutyunyan <arut@nginx.com> parents: 6480 diff changeset	113 NULL, /* init module */
d452cb27639f Style. Roman Arutyunyan <arut@nginx.com> parents: 6480 diff changeset	114 NULL, /* init process */
d452cb27639f Style. Roman Arutyunyan <arut@nginx.com> parents: 6480 diff changeset	115 NULL, /* init thread */
d452cb27639f Style. Roman Arutyunyan <arut@nginx.com> parents: 6480 diff changeset	116 NULL, /* exit thread */
d452cb27639f Style. Roman Arutyunyan <arut@nginx.com> parents: 6480 diff changeset	117 NULL, /* exit process */
d452cb27639f Style. Roman Arutyunyan <arut@nginx.com> parents: 6480 diff changeset	118 NULL, /* exit master */
6197 0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	119 NGX_MODULE_V1_PADDING
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	120 };
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 static ngx_int_t
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	124 ngx_stream_limit_conn_handler(ngx_stream_session_t *s)
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	125 {
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	126 size_t n;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	127 uint32_t hash;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	128 ngx_str_t key;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	129 ngx_uint_t i;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	130 ngx_slab_pool_t *shpool;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	131 ngx_rbtree_node_t *node;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	132 ngx_pool_cleanup_t *cln;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	133 struct sockaddr_in *sin;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	134 #if (NGX_HAVE_INET6)
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	135 struct sockaddr_in6 *sin6;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	136 #endif
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	137 ngx_stream_limit_conn_ctx_t *ctx;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	138 ngx_stream_limit_conn_node_t *lc;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	139 ngx_stream_limit_conn_conf_t *lccf;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	140 ngx_stream_limit_conn_limit_t *limits;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	141 ngx_stream_limit_conn_cleanup_t *lccln;
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 switch (s->connection->sockaddr->sa_family) {
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	144
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	145 case AF_INET:
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	146 sin = (struct sockaddr_in *) s->connection->sockaddr;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	147
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	148 key.len = sizeof(in_addr_t);
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	149 key.data = (u_char *) &sin->sin_addr;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	150
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	151 break;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	152
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	153 #if (NGX_HAVE_INET6)
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	154 case AF_INET6:
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	155 sin6 = (struct sockaddr_in6 *) s->connection->sockaddr;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	156
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	157 key.len = sizeof(struct in6_addr);
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	158 key.data = sin6->sin6_addr.s6_addr;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	159
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	160 break;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	161 #endif
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	162
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	163 default:
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	164 return NGX_DECLINED;
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
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	167 hash = ngx_crc32_short(key.data, key.len);
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	168
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	169 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	170 limits = lccf->limits.elts;
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 for (i = 0; i < lccf->limits.nelts; i++) {
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	173 ctx = limits[i].shm_zone->data;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	174
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	175 shpool = (ngx_slab_pool_t *) limits[i].shm_zone->shm.addr;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	176
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	177 ngx_shmtx_lock(&shpool->mutex);
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	178
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	179 node = ngx_stream_limit_conn_lookup(ctx->rbtree, &key, hash);
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	180
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	181 if (node == NULL) {
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 n = offsetof(ngx_rbtree_node_t, color)
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	184 + offsetof(ngx_stream_limit_conn_node_t, data)
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	185 + key.len;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	186
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	187 node = ngx_slab_alloc_locked(shpool, n);
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	188
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	189 if (node == NULL) {
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	190 ngx_shmtx_unlock(&shpool->mutex);
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	191 ngx_stream_limit_conn_cleanup_all(s->connection->pool);
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	192 return NGX_ABORT;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	193 }
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 node->key = hash;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	198 lc->len = (u_char) key.len;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	199 lc->conn = 1;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	200 ngx_memcpy(lc->data, key.data, key.len);
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	201
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	202 ngx_rbtree_insert(ctx->rbtree, node);
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	203
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	204 } else {
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	205
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	206 lc = (ngx_stream_limit_conn_node_t *) &node->color;
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 if ((ngx_uint_t) lc->conn >= limits[i].conn) {
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	209
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	210 ngx_shmtx_unlock(&shpool->mutex);
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_error(lccf->log_level, s->connection->log, 0,
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	213 "limiting connections by zone \"%V\"",
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	214 &limits[i].shm_zone->shm.name);
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	215
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	216 ngx_stream_limit_conn_cleanup_all(s->connection->pool);
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	217 return NGX_ABORT;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	218 }
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	219
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	220 lc->conn++;
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 ngx_log_debug2(NGX_LOG_DEBUG_STREAM, s->connection->log, 0,
6480 f01ab2dbcfdc Fixed logging. Sergey Kandaurov <pluknet@nginx.com> parents: 6197 diff changeset	224 "limit conn: %08Xi %d", node->key, lc->conn);
6197 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 ngx_shmtx_unlock(&shpool->mutex);
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	227
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	228 cln = ngx_pool_cleanup_add(s->connection->pool,
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	229 sizeof(ngx_stream_limit_conn_cleanup_t));
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	230 if (cln == NULL) {
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	231 return NGX_ERROR;
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 cln->handler = ngx_stream_limit_conn_cleanup;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	235 lccln = cln->data;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	236
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	237 lccln->shm_zone = limits[i].shm_zone;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	238 lccln->node = node;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	239 }
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 return NGX_DECLINED;
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
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 static void
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	246 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	247 ngx_rbtree_node_t node, ngx_rbtree_node_t sentinel)
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 ngx_rbtree_node_t **p;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	250 ngx_stream_limit_conn_node_t lcn, lcnt;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	251
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	252 for ( ;; ) {
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	253
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	254 if (node->key < temp->key) {
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 = &temp->left;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	257
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	258 } else if (node->key > temp->key) {
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 p = &temp->right;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	261
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	262 } else { /* node->key == temp->key */
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 lcn = (ngx_stream_limit_conn_node_t *) &node->color;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	265 lcnt = (ngx_stream_limit_conn_node_t *) &temp->color;
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 = (ngx_memn2cmp(lcn->data, lcnt->data, lcn->len, lcnt->len) < 0)
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	268 ? &temp->left : &temp->right;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	269 }
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	270
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	271 if (*p == sentinel) {
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	272 break;
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 temp = *p;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	276 }
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	277
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	278 *p = node;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	279 node->parent = temp;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	280 node->left = sentinel;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	281 node->right = sentinel;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	282 ngx_rbt_red(node);
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	283 }
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	284
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 static ngx_rbtree_node_t *
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	287 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	288 uint32_t hash)
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	289 {
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	290 ngx_int_t rc;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	291 ngx_rbtree_node_t node, sentinel;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	292 ngx_stream_limit_conn_node_t *lcn;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	293
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	294 node = rbtree->root;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	295 sentinel = rbtree->sentinel;
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 while (node != sentinel) {
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	298
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	299 if (hash < node->key) {
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	300 node = node->left;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	301 continue;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	302 }
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 (hash > node->key) {
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	305 node = node->right;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	306 continue;
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
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	309 /* hash == node->key */
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 lcn = (ngx_stream_limit_conn_node_t *) &node->color;
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 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	314
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	315 if (rc == 0) {
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	316 return node;
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
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	319 node = (rc < 0) ? node->left : node->right;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	320 }
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	321
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	322 return NULL;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	323 }
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
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	326 static void
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	327 ngx_stream_limit_conn_cleanup(void *data)
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	328 {
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	329 ngx_stream_limit_conn_cleanup_t *lccln = data;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	330
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	331 ngx_slab_pool_t *shpool;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	332 ngx_rbtree_node_t *node;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	333 ngx_stream_limit_conn_ctx_t *ctx;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	334 ngx_stream_limit_conn_node_t *lc;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	335
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	336 ctx = lccln->shm_zone->data;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	337 shpool = (ngx_slab_pool_t *) lccln->shm_zone->shm.addr;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	338 node = lccln->node;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	339 lc = (ngx_stream_limit_conn_node_t *) &node->color;
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 ngx_shmtx_lock(&shpool->mutex);
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	342
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	343 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	344 "limit conn cleanup: %08Xi %d", node->key, lc->conn);
6197 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 lc->conn--;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	347
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	348 if (lc->conn == 0) {
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	349 ngx_rbtree_delete(ctx->rbtree, node);
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	350 ngx_slab_free_locked(shpool, node);
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	351 }
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 ngx_shmtx_unlock(&shpool->mutex);
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	354 }
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	355
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 static ngx_inline void
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	358 ngx_stream_limit_conn_cleanup_all(ngx_pool_t *pool)
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 ngx_pool_cleanup_t *cln;
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 cln = pool->cleanup;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	363
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	364 while (cln && cln->handler == ngx_stream_limit_conn_cleanup) {
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	365 ngx_stream_limit_conn_cleanup(cln->data);
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	366 cln = cln->next;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	367 }
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	368
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	369 pool->cleanup = cln;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	370 }
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
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	373 static ngx_int_t
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	374 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	375 {
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	376 ngx_stream_limit_conn_ctx_t *octx = data;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	377
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	378 size_t len;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	379 ngx_slab_pool_t *shpool;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	380 ngx_rbtree_node_t *sentinel;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	381 ngx_stream_limit_conn_ctx_t *ctx;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	382
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	383 ctx = shm_zone->data;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	384
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	385 if (octx) {
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	386 ctx->rbtree = octx->rbtree;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	387
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	388 return NGX_OK;
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
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	391 shpool = (ngx_slab_pool_t *) shm_zone->shm.addr;
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 if (shm_zone->shm.exists) {
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	394 ctx->rbtree = shpool->data;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	395
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	396 return NGX_OK;
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
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	399 ctx->rbtree = ngx_slab_alloc(shpool, sizeof(ngx_rbtree_t));
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	400 if (ctx->rbtree == NULL) {
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	401 return NGX_ERROR;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	402 }
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	403
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	404 shpool->data = ctx->rbtree;
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 sentinel = ngx_slab_alloc(shpool, sizeof(ngx_rbtree_node_t));
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	407 if (sentinel == NULL) {
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	408 return NGX_ERROR;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	409 }
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	410
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	411 ngx_rbtree_init(ctx->rbtree, sentinel,
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	412 ngx_stream_limit_conn_rbtree_insert_value);
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	413
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	414 len = sizeof(" in limit_conn_zone \"\"") + shm_zone->shm.name.len;
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 shpool->log_ctx = ngx_slab_alloc(shpool, len);
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	417 if (shpool->log_ctx == NULL) {
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	418 return NGX_ERROR;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	419 }
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	420
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	421 ngx_sprintf(shpool->log_ctx, " in limit_conn_zone \"%V\"%Z",
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	422 &shm_zone->shm.name);
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	423
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	424 return NGX_OK;
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
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 static void *
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	429 ngx_stream_limit_conn_create_conf(ngx_conf_t *cf)
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	430 {
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	431 ngx_stream_limit_conn_conf_t *conf;
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 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	434 if (conf == NULL) {
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	435 return NULL;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	436 }
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	437
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 * set by ngx_pcalloc():
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 * conf->limits.elts = NULL;
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
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	444 conf->log_level = NGX_CONF_UNSET_UINT;
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 return conf;
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
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 static char *
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	451 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	452 {
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	453 ngx_stream_limit_conn_conf_t *prev = parent;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	454 ngx_stream_limit_conn_conf_t *conf = child;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	455
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	456 if (conf->limits.elts == NULL) {
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	457 conf->limits = prev->limits;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	458 }
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	459
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	460 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	461
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	462 return NGX_CONF_OK;
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
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 static char *
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	467 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	468 {
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	469 u_char *p;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	470 ssize_t size;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	471 ngx_str_t *value, name, s;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	472 ngx_uint_t i;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	473 ngx_shm_zone_t *shm_zone;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	474 ngx_stream_limit_conn_ctx_t *ctx;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	475
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	476 value = cf->args->elts;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	477
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	478 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	479 if (ctx == NULL) {
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	480 return NGX_CONF_ERROR;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	481 }
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	482
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	483 size = 0;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	484 name.len = 0;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	485
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	486 for (i = 2; i < cf->args->nelts; i++) {
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	487
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	488 if (ngx_strncmp(value[i].data, "zone=", 5) == 0) {
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	489
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	490 name.data = value[i].data + 5;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	491
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	492 p = (u_char *) ngx_strchr(name.data, ':');
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	493
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	494 if (p == NULL) {
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	495 ngx_conf_log_error(NGX_LOG_EMERG, cf, 0,
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	496 "invalid zone size \"%V\"", &value[i]);
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	497 return NGX_CONF_ERROR;
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
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	500 name.len = p - name.data;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	501
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	502 s.data = p + 1;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	503 s.len = value[i].data + value[i].len - s.data;
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 size = ngx_parse_size(&s);
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 (size == NGX_ERROR) {
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 if (size < (ssize_t) (8 * ngx_pagesize)) {
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	514 ngx_conf_log_error(NGX_LOG_EMERG, cf, 0,
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	515 "zone \"%V\" is too small", &value[i]);
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	516 return NGX_CONF_ERROR;
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
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	519 continue;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	520 }
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	521
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	522 ngx_conf_log_error(NGX_LOG_EMERG, cf, 0,
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	523 "invalid parameter \"%V\"", &value[i]);
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	524 return NGX_CONF_ERROR;
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
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	527 if (name.len == 0) {
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	528 ngx_conf_log_error(NGX_LOG_EMERG, cf, 0,
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	529 "\"%V\" must have \"zone\" parameter",
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	530 &cmd->name);
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	531 return NGX_CONF_ERROR;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	532 }
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 shm_zone = ngx_shared_memory_add(cf, &name, size,
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	535 &ngx_stream_limit_conn_module);
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	536 if (shm_zone == NULL) {
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 (shm_zone->data) {
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 \"%V\" is already bound to key "
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	543 "\"$binary_remote_addr\"",
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	544 &cmd->name, &name);
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	545 return NGX_CONF_ERROR;
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
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	548 if (ngx_strcmp(value[1].data, "$binary_remote_addr") != 0) {
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	549 ngx_conf_log_error(NGX_LOG_EMERG, cf, 0,
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	550 "unsupported key \"%V\", use "
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	551 "$binary_remote_addr", &value[1]);
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	552 return NGX_CONF_ERROR;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	553 }
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	554
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	555 shm_zone->init = ngx_stream_limit_conn_init_zone;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	556 shm_zone->data = ctx;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	557
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	558 return NGX_CONF_OK;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	559 }
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 static char *
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	563 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	564 {
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	565 ngx_shm_zone_t *shm_zone;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	566 ngx_stream_limit_conn_conf_t *lccf = conf;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	567 ngx_stream_limit_conn_limit_t limit, limits;
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 ngx_str_t *value;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	570 ngx_int_t n;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	571 ngx_uint_t i;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	572
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	573 value = cf->args->elts;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	574
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	575 shm_zone = ngx_shared_memory_add(cf, &value[1], 0,
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	576 &ngx_stream_limit_conn_module);
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	577 if (shm_zone == NULL) {
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	578 return NGX_CONF_ERROR;
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
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	581 limits = lccf->limits.elts;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	582
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	583 if (limits == NULL) {
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	584 if (ngx_array_init(&lccf->limits, cf->pool, 1,
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	585 sizeof(ngx_stream_limit_conn_limit_t))
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	586 != NGX_OK)
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 return NGX_CONF_ERROR;
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 }
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	591
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	592 for (i = 0; i < lccf->limits.nelts; i++) {
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	593 if (shm_zone == limits[i].shm_zone) {
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	594 return "is duplicate";
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	595 }
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 n = ngx_atoi(value[2].data, value[2].len);
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	599 if (n <= 0) {
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	600 ngx_conf_log_error(NGX_LOG_EMERG, cf, 0,
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	601 "invalid number of connections \"%V\"", &value[2]);
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	602 return NGX_CONF_ERROR;
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 if (n > 65535) {
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	606 ngx_conf_log_error(NGX_LOG_EMERG, cf, 0,
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	607 "connection limit must be less 65536");
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	608 return NGX_CONF_ERROR;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	609 }
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 limit = ngx_array_push(&lccf->limits);
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	612 if (limit == NULL) {
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	613 return NGX_CONF_ERROR;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	614 }
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	615
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	616 limit->conn = n;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	617 limit->shm_zone = shm_zone;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	618
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	619 return NGX_CONF_OK;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	620 }
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 static ngx_int_t
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	624 ngx_stream_limit_conn_init(ngx_conf_t *cf)
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 ngx_stream_core_main_conf_t *cmcf;
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 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	629
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	630 cmcf->limit_conn_handler = ngx_stream_limit_conn_handler;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	631
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	632 return NGX_OK;
0dcef374b8bb Stream: connection limiting module. Vladimir Homutov <vl@nginx.com> parents: diff changeset	633 }

Mercurial > hg > nginx

annotate src/stream/ngx_stream_limit_conn_module.c @ 6606:2f41d383c9c7