nginx: src/http/modules/ngx_http_limit_req

annotate src/http/modules/ngx_http_limit_req_module.c @ 8039:e64a1f32065b

Headers filter: improved memory allocation error handling.

author	Maxim Dounin <mdounin@mdounin.ru>
date	Mon, 30 May 2022 21:25:57 +0300
parents	1ebd78df4ce7
children

rev	line source
980 7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	1
7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	2 /*
7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	3 * Copyright (C) Igor Sysoev
4412 d620f497c50f Copyright updated. Maxim Konovalov <maxim@nginx.com> parents: 4399 diff changeset	4 * Copyright (C) Nginx, Inc.
980 7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	5 */
7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	6
7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	7
7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	8 #include <ngx_config.h>
7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	9 #include <ngx_core.h>
7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	10 #include <ngx_http.h>
7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	11
7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	12
7592 776d1bebdca2 Limit req: $limit_req_status variable. Roman Arutyunyan <arut@nginx.com> parents: 7515 diff changeset	13 #define NGX_HTTP_LIMIT_REQ_PASSED 1
776d1bebdca2 Limit req: $limit_req_status variable. Roman Arutyunyan <arut@nginx.com> parents: 7515 diff changeset	14 #define NGX_HTTP_LIMIT_REQ_DELAYED 2
776d1bebdca2 Limit req: $limit_req_status variable. Roman Arutyunyan <arut@nginx.com> parents: 7515 diff changeset	15 #define NGX_HTTP_LIMIT_REQ_REJECTED 3
776d1bebdca2 Limit req: $limit_req_status variable. Roman Arutyunyan <arut@nginx.com> parents: 7515 diff changeset	16 #define NGX_HTTP_LIMIT_REQ_DELAYED_DRY_RUN 4
776d1bebdca2 Limit req: $limit_req_status variable. Roman Arutyunyan <arut@nginx.com> parents: 7515 diff changeset	17 #define NGX_HTTP_LIMIT_REQ_REJECTED_DRY_RUN 5
776d1bebdca2 Limit req: $limit_req_status variable. Roman Arutyunyan <arut@nginx.com> parents: 7515 diff changeset	18
776d1bebdca2 Limit req: $limit_req_status variable. Roman Arutyunyan <arut@nginx.com> parents: 7515 diff changeset	19
980 7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	20 typedef struct {
2720 b3b8c66bd520 support attaching to an existent Win32 shared memory Igor Sysoev <igor@sysoev.ru> parents: 2716 diff changeset	21 u_char color;
b3b8c66bd520 support attaching to an existent Win32 shared memory Igor Sysoev <igor@sysoev.ru> parents: 2716 diff changeset	22 u_char dummy;
b3b8c66bd520 support attaching to an existent Win32 shared memory Igor Sysoev <igor@sysoev.ru> parents: 2716 diff changeset	23 u_short len;
b3b8c66bd520 support attaching to an existent Win32 shared memory Igor Sysoev <igor@sysoev.ru> parents: 2716 diff changeset	24 ngx_queue_t queue;
b3b8c66bd520 support attaching to an existent Win32 shared memory Igor Sysoev <igor@sysoev.ru> parents: 2716 diff changeset	25 ngx_msec_t last;
b3b8c66bd520 support attaching to an existent Win32 shared memory Igor Sysoev <igor@sysoev.ru> parents: 2716 diff changeset	26 /* integer value, 1 corresponds to 0.001 r/s */
b3b8c66bd520 support attaching to an existent Win32 shared memory Igor Sysoev <igor@sysoev.ru> parents: 2716 diff changeset	27 ngx_uint_t excess;
4420 9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	28 ngx_uint_t count;
2720 b3b8c66bd520 support attaching to an existent Win32 shared memory Igor Sysoev <igor@sysoev.ru> parents: 2716 diff changeset	29 u_char data[1];
2294 ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	30 } ngx_http_limit_req_node_t;
980 7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	31
7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	32
7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	33 typedef struct {
2720 b3b8c66bd520 support attaching to an existent Win32 shared memory Igor Sysoev <igor@sysoev.ru> parents: 2716 diff changeset	34 ngx_rbtree_t rbtree;
b3b8c66bd520 support attaching to an existent Win32 shared memory Igor Sysoev <igor@sysoev.ru> parents: 2716 diff changeset	35 ngx_rbtree_node_t sentinel;
b3b8c66bd520 support attaching to an existent Win32 shared memory Igor Sysoev <igor@sysoev.ru> parents: 2716 diff changeset	36 ngx_queue_t queue;
b3b8c66bd520 support attaching to an existent Win32 shared memory Igor Sysoev <igor@sysoev.ru> parents: 2716 diff changeset	37 } ngx_http_limit_req_shctx_t;
b3b8c66bd520 support attaching to an existent Win32 shared memory Igor Sysoev <igor@sysoev.ru> parents: 2716 diff changeset	38
b3b8c66bd520 support attaching to an existent Win32 shared memory Igor Sysoev <igor@sysoev.ru> parents: 2716 diff changeset	39
b3b8c66bd520 support attaching to an existent Win32 shared memory Igor Sysoev <igor@sysoev.ru> parents: 2716 diff changeset	40 typedef struct {
b3b8c66bd520 support attaching to an existent Win32 shared memory Igor Sysoev <igor@sysoev.ru> parents: 2716 diff changeset	41 ngx_http_limit_req_shctx_t *sh;
b3b8c66bd520 support attaching to an existent Win32 shared memory Igor Sysoev <igor@sysoev.ru> parents: 2716 diff changeset	42 ngx_slab_pool_t *shpool;
b3b8c66bd520 support attaching to an existent Win32 shared memory Igor Sysoev <igor@sysoev.ru> parents: 2716 diff changeset	43 /* integer value, 1 corresponds to 0.001 r/s */
b3b8c66bd520 support attaching to an existent Win32 shared memory Igor Sysoev <igor@sysoev.ru> parents: 2716 diff changeset	44 ngx_uint_t rate;
5862 ecbb99aa0e12 Limit req: use complex value in limit_req_zone. Valentin Bartenev <vbart@nginx.com> parents: 5846 diff changeset	45 ngx_http_complex_value_t key;
4420 9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	46 ngx_http_limit_req_node_t *node;
2294 ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	47 } ngx_http_limit_req_ctx_t;
987 14e68f471d02 new syntax Igor Sysoev <igor@sysoev.ru> parents: 984 diff changeset	48
14e68f471d02 new syntax Igor Sysoev <igor@sysoev.ru> parents: 984 diff changeset	49
14e68f471d02 new syntax Igor Sysoev <igor@sysoev.ru> parents: 984 diff changeset	50 typedef struct {
2720 b3b8c66bd520 support attaching to an existent Win32 shared memory Igor Sysoev <igor@sysoev.ru> parents: 2716 diff changeset	51 ngx_shm_zone_t *shm_zone;
b3b8c66bd520 support attaching to an existent Win32 shared memory Igor Sysoev <igor@sysoev.ru> parents: 2716 diff changeset	52 /* integer value, 1 corresponds to 0.001 r/s */
b3b8c66bd520 support attaching to an existent Win32 shared memory Igor Sysoev <igor@sysoev.ru> parents: 2716 diff changeset	53 ngx_uint_t burst;
7399 d6ca744c727e Limit req: "delay=" parameter. Maxim Dounin <mdounin@mdounin.ru> parents: 7398 diff changeset	54 ngx_uint_t delay;
4420 9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	55 } ngx_http_limit_req_limit_t;
9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	56
9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	57
9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	58 typedef struct {
9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	59 ngx_array_t limits;
3185 30570a5b9bb0 limit_req_log_level Igor Sysoev <igor@sysoev.ru> parents: 3183 diff changeset	60 ngx_uint_t limit_log_level;
30570a5b9bb0 limit_req_log_level Igor Sysoev <igor@sysoev.ru> parents: 3183 diff changeset	61 ngx_uint_t delay_log_level;
5117 00e4459739ed The limit_req_status and limit_conn_status directives. Maxim Dounin <mdounin@mdounin.ru> parents: 4832 diff changeset	62 ngx_uint_t status_code;
7515 2db68852d6a0 Limit req: limit_req_dry_run directive. Roman Arutyunyan <arut@nginx.com> parents: 7399 diff changeset	63 ngx_flag_t dry_run;
2294 ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	64 } ngx_http_limit_req_conf_t;
980 7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	65
7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	66
2294 ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	67 static void ngx_http_limit_req_delay(ngx_http_request_t *r);
4420 9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	68 static ngx_int_t ngx_http_limit_req_lookup(ngx_http_limit_req_limit_t *limit,
5863 102f85699420 Limit req: reduced number of parameters in the lookup function. Valentin Bartenev <vbart@nginx.com> parents: 5862 diff changeset	69 ngx_uint_t hash, ngx_str_t key, ngx_uint_t ep, ngx_uint_t account);
4420 9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	70 static ngx_msec_t ngx_http_limit_req_account(ngx_http_limit_req_limit_t *limits,
9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	71 ngx_uint_t n, ngx_uint_t ep, ngx_http_limit_req_limit_t *limit);
7726 559d19037984 Limit req: unlocking of nodes on complex value errors. Maxim Dounin <mdounin@mdounin.ru> parents: 7592 diff changeset	72 static void ngx_http_limit_req_unlock(ngx_http_limit_req_limit_t *limits,
559d19037984 Limit req: unlocking of nodes on complex value errors. Maxim Dounin <mdounin@mdounin.ru> parents: 7592 diff changeset	73 ngx_uint_t n);
2294 ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	74 static void ngx_http_limit_req_expire(ngx_http_limit_req_ctx_t *ctx,
ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	75 ngx_uint_t n);
980 7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	76
7592 776d1bebdca2 Limit req: $limit_req_status variable. Roman Arutyunyan <arut@nginx.com> parents: 7515 diff changeset	77 static ngx_int_t ngx_http_limit_req_status_variable(ngx_http_request_t *r,
776d1bebdca2 Limit req: $limit_req_status variable. Roman Arutyunyan <arut@nginx.com> parents: 7515 diff changeset	78 ngx_http_variable_value_t *v, uintptr_t data);
2294 ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	79 static void ngx_http_limit_req_create_conf(ngx_conf_t cf);
ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	80 static char ngx_http_limit_req_merge_conf(ngx_conf_t cf, void *parent,
980 7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	81 void *child);
2294 ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	82 static char ngx_http_limit_req_zone(ngx_conf_t cf, ngx_command_t *cmd,
980 7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	83 void *conf);
2294 ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	84 static char ngx_http_limit_req(ngx_conf_t cf, ngx_command_t *cmd,
980 7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	85 void *conf);
7592 776d1bebdca2 Limit req: $limit_req_status variable. Roman Arutyunyan <arut@nginx.com> parents: 7515 diff changeset	86 static ngx_int_t ngx_http_limit_req_add_variables(ngx_conf_t *cf);
2294 ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	87 static ngx_int_t ngx_http_limit_req_init(ngx_conf_t *cf);
980 7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	88
7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	89
3185 30570a5b9bb0 limit_req_log_level Igor Sysoev <igor@sysoev.ru> parents: 3183 diff changeset	90 static ngx_conf_enum_t ngx_http_limit_req_log_levels[] = {
30570a5b9bb0 limit_req_log_level Igor Sysoev <igor@sysoev.ru> parents: 3183 diff changeset	91 { ngx_string("info"), NGX_LOG_INFO },
30570a5b9bb0 limit_req_log_level Igor Sysoev <igor@sysoev.ru> parents: 3183 diff changeset	92 { ngx_string("notice"), NGX_LOG_NOTICE },
30570a5b9bb0 limit_req_log_level Igor Sysoev <igor@sysoev.ru> parents: 3183 diff changeset	93 { ngx_string("warn"), NGX_LOG_WARN },
30570a5b9bb0 limit_req_log_level Igor Sysoev <igor@sysoev.ru> parents: 3183 diff changeset	94 { ngx_string("error"), NGX_LOG_ERR },
30570a5b9bb0 limit_req_log_level Igor Sysoev <igor@sysoev.ru> parents: 3183 diff changeset	95 { ngx_null_string, 0 }
30570a5b9bb0 limit_req_log_level Igor Sysoev <igor@sysoev.ru> parents: 3183 diff changeset	96 };
30570a5b9bb0 limit_req_log_level Igor Sysoev <igor@sysoev.ru> parents: 3183 diff changeset	97
30570a5b9bb0 limit_req_log_level Igor Sysoev <igor@sysoev.ru> parents: 3183 diff changeset	98
5117 00e4459739ed The limit_req_status and limit_conn_status directives. Maxim Dounin <mdounin@mdounin.ru> parents: 4832 diff changeset	99 static ngx_conf_num_bounds_t ngx_http_limit_req_status_bounds = {
00e4459739ed The limit_req_status and limit_conn_status directives. Maxim Dounin <mdounin@mdounin.ru> parents: 4832 diff changeset	100 ngx_conf_check_num_bounds, 400, 599
00e4459739ed The limit_req_status and limit_conn_status directives. Maxim Dounin <mdounin@mdounin.ru> parents: 4832 diff changeset	101 };
00e4459739ed The limit_req_status and limit_conn_status directives. Maxim Dounin <mdounin@mdounin.ru> parents: 4832 diff changeset	102
00e4459739ed The limit_req_status and limit_conn_status directives. Maxim Dounin <mdounin@mdounin.ru> parents: 4832 diff changeset	103
2294 ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	104 static ngx_command_t ngx_http_limit_req_commands[] = {
980 7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	105
2294 ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	106 { ngx_string("limit_req_zone"),
987 14e68f471d02 new syntax Igor Sysoev <igor@sysoev.ru> parents: 984 diff changeset	107 NGX_HTTP_MAIN_CONF\|NGX_CONF_TAKE3,
2294 ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	108 ngx_http_limit_req_zone,
980 7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	109 0,
7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	110 0,
7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	111 NULL },
7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	112
2294 ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	113 { ngx_string("limit_req"),
ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	114 NGX_HTTP_MAIN_CONF\|NGX_HTTP_SRV_CONF\|NGX_HTTP_LOC_CONF\|NGX_CONF_TAKE123,
ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	115 ngx_http_limit_req,
980 7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	116 NGX_HTTP_LOC_CONF_OFFSET,
7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	117 0,
7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	118 NULL },
7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	119
3185 30570a5b9bb0 limit_req_log_level Igor Sysoev <igor@sysoev.ru> parents: 3183 diff changeset	120 { ngx_string("limit_req_log_level"),
30570a5b9bb0 limit_req_log_level Igor Sysoev <igor@sysoev.ru> parents: 3183 diff changeset	121 NGX_HTTP_MAIN_CONF\|NGX_HTTP_SRV_CONF\|NGX_HTTP_LOC_CONF\|NGX_CONF_TAKE1,
30570a5b9bb0 limit_req_log_level Igor Sysoev <igor@sysoev.ru> parents: 3183 diff changeset	122 ngx_conf_set_enum_slot,
30570a5b9bb0 limit_req_log_level Igor Sysoev <igor@sysoev.ru> parents: 3183 diff changeset	123 NGX_HTTP_LOC_CONF_OFFSET,
30570a5b9bb0 limit_req_log_level Igor Sysoev <igor@sysoev.ru> parents: 3183 diff changeset	124 offsetof(ngx_http_limit_req_conf_t, limit_log_level),
30570a5b9bb0 limit_req_log_level Igor Sysoev <igor@sysoev.ru> parents: 3183 diff changeset	125 &ngx_http_limit_req_log_levels },
30570a5b9bb0 limit_req_log_level Igor Sysoev <igor@sysoev.ru> parents: 3183 diff changeset	126
5117 00e4459739ed The limit_req_status and limit_conn_status directives. Maxim Dounin <mdounin@mdounin.ru> parents: 4832 diff changeset	127 { ngx_string("limit_req_status"),
00e4459739ed The limit_req_status and limit_conn_status directives. Maxim Dounin <mdounin@mdounin.ru> parents: 4832 diff changeset	128 NGX_HTTP_MAIN_CONF\|NGX_HTTP_SRV_CONF\|NGX_HTTP_LOC_CONF\|NGX_CONF_TAKE1,
00e4459739ed The limit_req_status and limit_conn_status directives. Maxim Dounin <mdounin@mdounin.ru> parents: 4832 diff changeset	129 ngx_conf_set_num_slot,
00e4459739ed The limit_req_status and limit_conn_status directives. Maxim Dounin <mdounin@mdounin.ru> parents: 4832 diff changeset	130 NGX_HTTP_LOC_CONF_OFFSET,
00e4459739ed The limit_req_status and limit_conn_status directives. Maxim Dounin <mdounin@mdounin.ru> parents: 4832 diff changeset	131 offsetof(ngx_http_limit_req_conf_t, status_code),
00e4459739ed The limit_req_status and limit_conn_status directives. Maxim Dounin <mdounin@mdounin.ru> parents: 4832 diff changeset	132 &ngx_http_limit_req_status_bounds },
00e4459739ed The limit_req_status and limit_conn_status directives. Maxim Dounin <mdounin@mdounin.ru> parents: 4832 diff changeset	133
7515 2db68852d6a0 Limit req: limit_req_dry_run directive. Roman Arutyunyan <arut@nginx.com> parents: 7399 diff changeset	134 { ngx_string("limit_req_dry_run"),
2db68852d6a0 Limit req: limit_req_dry_run directive. Roman Arutyunyan <arut@nginx.com> parents: 7399 diff changeset	135 NGX_HTTP_MAIN_CONF\|NGX_HTTP_SRV_CONF\|NGX_HTTP_LOC_CONF\|NGX_CONF_FLAG,
2db68852d6a0 Limit req: limit_req_dry_run directive. Roman Arutyunyan <arut@nginx.com> parents: 7399 diff changeset	136 ngx_conf_set_flag_slot,
2db68852d6a0 Limit req: limit_req_dry_run directive. Roman Arutyunyan <arut@nginx.com> parents: 7399 diff changeset	137 NGX_HTTP_LOC_CONF_OFFSET,
2db68852d6a0 Limit req: limit_req_dry_run directive. Roman Arutyunyan <arut@nginx.com> parents: 7399 diff changeset	138 offsetof(ngx_http_limit_req_conf_t, dry_run),
2db68852d6a0 Limit req: limit_req_dry_run directive. Roman Arutyunyan <arut@nginx.com> parents: 7399 diff changeset	139 NULL },
2db68852d6a0 Limit req: limit_req_dry_run directive. Roman Arutyunyan <arut@nginx.com> parents: 7399 diff changeset	140
980 7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	141 ngx_null_command
7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	142 };
7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	143
7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	144
2294 ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	145 static ngx_http_module_t ngx_http_limit_req_module_ctx = {
7592 776d1bebdca2 Limit req: $limit_req_status variable. Roman Arutyunyan <arut@nginx.com> parents: 7515 diff changeset	146 ngx_http_limit_req_add_variables, /* preconfiguration */
2294 ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	147 ngx_http_limit_req_init, /* postconfiguration */
980 7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	148
7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	149 NULL, /* create main configuration */
7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	150 NULL, /* init main configuration */
7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	151
7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	152 NULL, /* create server configuration */
7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	153 NULL, /* merge server configuration */
7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	154
4499 778ef9c3fd2d Fixed spelling in single-line comments. Ruslan Ermilov <ru@nginx.com> parents: 4497 diff changeset	155 ngx_http_limit_req_create_conf, /* create location configuration */
778ef9c3fd2d Fixed spelling in single-line comments. Ruslan Ermilov <ru@nginx.com> parents: 4497 diff changeset	156 ngx_http_limit_req_merge_conf /* merge location configuration */
980 7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	157 };
7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	158
7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	159
2294 ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	160 ngx_module_t ngx_http_limit_req_module = {
980 7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	161 NGX_MODULE_V1,
2294 ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	162 &ngx_http_limit_req_module_ctx, /* module context */
ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	163 ngx_http_limit_req_commands, /* module directives */
980 7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	164 NGX_HTTP_MODULE, /* module type */
7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	165 NULL, /* init master */
7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	166 NULL, /* init module */
7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	167 NULL, /* init process */
7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	168 NULL, /* init thread */
7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	169 NULL, /* exit thread */
7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	170 NULL, /* exit process */
7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	171 NULL, /* exit master */
7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	172 NGX_MODULE_V1_PADDING
7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	173 };
7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	174
7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	175
7592 776d1bebdca2 Limit req: $limit_req_status variable. Roman Arutyunyan <arut@nginx.com> parents: 7515 diff changeset	176 static ngx_http_variable_t ngx_http_limit_req_vars[] = {
776d1bebdca2 Limit req: $limit_req_status variable. Roman Arutyunyan <arut@nginx.com> parents: 7515 diff changeset	177
776d1bebdca2 Limit req: $limit_req_status variable. Roman Arutyunyan <arut@nginx.com> parents: 7515 diff changeset	178 { ngx_string("limit_req_status"), NULL,
776d1bebdca2 Limit req: $limit_req_status variable. Roman Arutyunyan <arut@nginx.com> parents: 7515 diff changeset	179 ngx_http_limit_req_status_variable, 0, NGX_HTTP_VAR_NOCACHEABLE, 0 },
776d1bebdca2 Limit req: $limit_req_status variable. Roman Arutyunyan <arut@nginx.com> parents: 7515 diff changeset	180
776d1bebdca2 Limit req: $limit_req_status variable. Roman Arutyunyan <arut@nginx.com> parents: 7515 diff changeset	181 ngx_http_null_variable
776d1bebdca2 Limit req: $limit_req_status variable. Roman Arutyunyan <arut@nginx.com> parents: 7515 diff changeset	182 };
776d1bebdca2 Limit req: $limit_req_status variable. Roman Arutyunyan <arut@nginx.com> parents: 7515 diff changeset	183
776d1bebdca2 Limit req: $limit_req_status variable. Roman Arutyunyan <arut@nginx.com> parents: 7515 diff changeset	184
776d1bebdca2 Limit req: $limit_req_status variable. Roman Arutyunyan <arut@nginx.com> parents: 7515 diff changeset	185 static ngx_str_t ngx_http_limit_req_status[] = {
776d1bebdca2 Limit req: $limit_req_status variable. Roman Arutyunyan <arut@nginx.com> parents: 7515 diff changeset	186 ngx_string("PASSED"),
776d1bebdca2 Limit req: $limit_req_status variable. Roman Arutyunyan <arut@nginx.com> parents: 7515 diff changeset	187 ngx_string("DELAYED"),
776d1bebdca2 Limit req: $limit_req_status variable. Roman Arutyunyan <arut@nginx.com> parents: 7515 diff changeset	188 ngx_string("REJECTED"),
776d1bebdca2 Limit req: $limit_req_status variable. Roman Arutyunyan <arut@nginx.com> parents: 7515 diff changeset	189 ngx_string("DELAYED_DRY_RUN"),
776d1bebdca2 Limit req: $limit_req_status variable. Roman Arutyunyan <arut@nginx.com> parents: 7515 diff changeset	190 ngx_string("REJECTED_DRY_RUN")
776d1bebdca2 Limit req: $limit_req_status variable. Roman Arutyunyan <arut@nginx.com> parents: 7515 diff changeset	191 };
776d1bebdca2 Limit req: $limit_req_status variable. Roman Arutyunyan <arut@nginx.com> parents: 7515 diff changeset	192
776d1bebdca2 Limit req: $limit_req_status variable. Roman Arutyunyan <arut@nginx.com> parents: 7515 diff changeset	193
980 7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	194 static ngx_int_t
2294 ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	195 ngx_http_limit_req_handler(ngx_http_request_t *r)
980 7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	196 {
4420 9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	197 uint32_t hash;
5862 ecbb99aa0e12 Limit req: use complex value in limit_req_zone. Valentin Bartenev <vbart@nginx.com> parents: 5846 diff changeset	198 ngx_str_t key;
4420 9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	199 ngx_int_t rc;
9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	200 ngx_uint_t n, excess;
9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	201 ngx_msec_t delay;
9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	202 ngx_http_limit_req_ctx_t *ctx;
9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	203 ngx_http_limit_req_conf_t *lrcf;
9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	204 ngx_http_limit_req_limit_t limit, limits;
980 7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	205
7592 776d1bebdca2 Limit req: $limit_req_status variable. Roman Arutyunyan <arut@nginx.com> parents: 7515 diff changeset	206 if (r->main->limit_req_status) {
984 dd128232e6ba count connection once per request Igor Sysoev <igor@sysoev.ru> parents: 981 diff changeset	207 return NGX_DECLINED;
dd128232e6ba count connection once per request Igor Sysoev <igor@sysoev.ru> parents: 981 diff changeset	208 }
dd128232e6ba count connection once per request Igor Sysoev <igor@sysoev.ru> parents: 981 diff changeset	209
2374 7b11f9a1bfe1 rename "lz" to "lr" in variable names Igor Sysoev <igor@sysoev.ru> parents: 2313 diff changeset	210 lrcf = ngx_http_get_module_loc_conf(r, ngx_http_limit_req_module);
4420 9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	211 limits = lrcf->limits.elts;
980 7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	212
4420 9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	213 excess = 0;
9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	214
9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	215 rc = NGX_DECLINED;
9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	216
4424 aacd7356c197 Limit req: unbreak compilation with MSVC. Maxim Dounin <mdounin@mdounin.ru> parents: 4420 diff changeset	217 #if (NGX_SUPPRESS_WARN)
aacd7356c197 Limit req: unbreak compilation with MSVC. Maxim Dounin <mdounin@mdounin.ru> parents: 4420 diff changeset	218 limit = NULL;
aacd7356c197 Limit req: unbreak compilation with MSVC. Maxim Dounin <mdounin@mdounin.ru> parents: 4420 diff changeset	219 #endif
aacd7356c197 Limit req: unbreak compilation with MSVC. Maxim Dounin <mdounin@mdounin.ru> parents: 4420 diff changeset	220
4420 9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	221 for (n = 0; n < lrcf->limits.nelts; n++) {
9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	222
9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	223 limit = &limits[n];
9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	224
9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	225 ctx = limit->shm_zone->data;
9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	226
5862 ecbb99aa0e12 Limit req: use complex value in limit_req_zone. Valentin Bartenev <vbart@nginx.com> parents: 5846 diff changeset	227 if (ngx_http_complex_value(r, &ctx->key, &key) != NGX_OK) {
7726 559d19037984 Limit req: unlocking of nodes on complex value errors. Maxim Dounin <mdounin@mdounin.ru> parents: 7592 diff changeset	228 ngx_http_limit_req_unlock(limits, n);
5862 ecbb99aa0e12 Limit req: use complex value in limit_req_zone. Valentin Bartenev <vbart@nginx.com> parents: 5846 diff changeset	229 return NGX_HTTP_INTERNAL_SERVER_ERROR;
4420 9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	230 }
9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	231
5862 ecbb99aa0e12 Limit req: use complex value in limit_req_zone. Valentin Bartenev <vbart@nginx.com> parents: 5846 diff changeset	232 if (key.len == 0) {
4420 9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	233 continue;
9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	234 }
9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	235
5862 ecbb99aa0e12 Limit req: use complex value in limit_req_zone. Valentin Bartenev <vbart@nginx.com> parents: 5846 diff changeset	236 if (key.len > 65535) {
4420 9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	237 ngx_log_error(NGX_LOG_ERR, r->connection->log, 0,
5862 ecbb99aa0e12 Limit req: use complex value in limit_req_zone. Valentin Bartenev <vbart@nginx.com> parents: 5846 diff changeset	238 "the value of the \"%V\" key "
ecbb99aa0e12 Limit req: use complex value in limit_req_zone. Valentin Bartenev <vbart@nginx.com> parents: 5846 diff changeset	239 "is more than 65535 bytes: \"%V\"",
ecbb99aa0e12 Limit req: use complex value in limit_req_zone. Valentin Bartenev <vbart@nginx.com> parents: 5846 diff changeset	240 &ctx->key.value, &key);
4420 9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	241 continue;
9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	242 }
9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	243
5862 ecbb99aa0e12 Limit req: use complex value in limit_req_zone. Valentin Bartenev <vbart@nginx.com> parents: 5846 diff changeset	244 hash = ngx_crc32_short(key.data, key.len);
4420 9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	245
9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	246 ngx_shmtx_lock(&ctx->shpool->mutex);
9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	247
5863 102f85699420 Limit req: reduced number of parameters in the lookup function. Valentin Bartenev <vbart@nginx.com> parents: 5862 diff changeset	248 rc = ngx_http_limit_req_lookup(limit, hash, &key, &excess,
4420 9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	249 (n == lrcf->limits.nelts - 1));
9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	250
9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	251 ngx_shmtx_unlock(&ctx->shpool->mutex);
9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	252
9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	253 ngx_log_debug4(NGX_LOG_DEBUG_HTTP, r->connection->log, 0,
9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	254 "limit_req[%ui]: %i %ui.%03ui",
9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	255 n, rc, excess / 1000, excess % 1000);
9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	256
9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	257 if (rc != NGX_AGAIN) {
9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	258 break;
9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	259 }
980 7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	260 }
7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	261
4420 9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	262 if (rc == NGX_DECLINED) {
1011 19118c44303f test length of variable and number of connections Igor Sysoev <igor@sysoev.ru> parents: 1002 diff changeset	263 return NGX_DECLINED;
19118c44303f test length of variable and number of connections Igor Sysoev <igor@sysoev.ru> parents: 1002 diff changeset	264 }
19118c44303f test length of variable and number of connections Igor Sysoev <igor@sysoev.ru> parents: 1002 diff changeset	265
4418 aac79fc948cc Limit req: allocation and initialization of a new node moved to the lookup Valentin Bartenev <vbart@nginx.com> parents: 4417 diff changeset	266 if (rc == NGX_BUSY \|\| rc == NGX_ERROR) {
aac79fc948cc Limit req: allocation and initialization of a new node moved to the lookup Valentin Bartenev <vbart@nginx.com> parents: 4417 diff changeset	267
aac79fc948cc Limit req: allocation and initialization of a new node moved to the lookup Valentin Bartenev <vbart@nginx.com> parents: 4417 diff changeset	268 if (rc == NGX_BUSY) {
aac79fc948cc Limit req: allocation and initialization of a new node moved to the lookup Valentin Bartenev <vbart@nginx.com> parents: 4417 diff changeset	269 ngx_log_error(lrcf->limit_log_level, r->connection->log, 0,
7515 2db68852d6a0 Limit req: limit_req_dry_run directive. Roman Arutyunyan <arut@nginx.com> parents: 7399 diff changeset	270 "limiting requests%s, excess: %ui.%03ui by zone \"%V\"",
2db68852d6a0 Limit req: limit_req_dry_run directive. Roman Arutyunyan <arut@nginx.com> parents: 7399 diff changeset	271 lrcf->dry_run ? ", dry run" : "",
2db68852d6a0 Limit req: limit_req_dry_run directive. Roman Arutyunyan <arut@nginx.com> parents: 7399 diff changeset	272 excess / 1000, excess % 1000,
2db68852d6a0 Limit req: limit_req_dry_run directive. Roman Arutyunyan <arut@nginx.com> parents: 7399 diff changeset	273 &limit->shm_zone->shm.name);
4420 9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	274 }
9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	275
7726 559d19037984 Limit req: unlocking of nodes on complex value errors. Maxim Dounin <mdounin@mdounin.ru> parents: 7592 diff changeset	276 ngx_http_limit_req_unlock(limits, n);
980 7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	277
7515 2db68852d6a0 Limit req: limit_req_dry_run directive. Roman Arutyunyan <arut@nginx.com> parents: 7399 diff changeset	278 if (lrcf->dry_run) {
7592 776d1bebdca2 Limit req: $limit_req_status variable. Roman Arutyunyan <arut@nginx.com> parents: 7515 diff changeset	279 r->main->limit_req_status = NGX_HTTP_LIMIT_REQ_REJECTED_DRY_RUN;
7515 2db68852d6a0 Limit req: limit_req_dry_run directive. Roman Arutyunyan <arut@nginx.com> parents: 7399 diff changeset	280 return NGX_DECLINED;
2db68852d6a0 Limit req: limit_req_dry_run directive. Roman Arutyunyan <arut@nginx.com> parents: 7399 diff changeset	281 }
2db68852d6a0 Limit req: limit_req_dry_run directive. Roman Arutyunyan <arut@nginx.com> parents: 7399 diff changeset	282
7592 776d1bebdca2 Limit req: $limit_req_status variable. Roman Arutyunyan <arut@nginx.com> parents: 7515 diff changeset	283 r->main->limit_req_status = NGX_HTTP_LIMIT_REQ_REJECTED;
776d1bebdca2 Limit req: $limit_req_status variable. Roman Arutyunyan <arut@nginx.com> parents: 7515 diff changeset	284
5117 00e4459739ed The limit_req_status and limit_conn_status directives. Maxim Dounin <mdounin@mdounin.ru> parents: 4832 diff changeset	285 return lrcf->status_code;
980 7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	286 }
7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	287
4420 9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	288 /* rc == NGX_AGAIN \|\| rc == NGX_OK */
2300 159136c9808d ) correct leaky bucket implementation Igor Sysoev <igor@sysoev.ru>* parents: 2294 diff changeset	289
4420 9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	290 if (rc == NGX_AGAIN) {
9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	291 excess = 0;
9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	292 }
9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	293
9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	294 delay = ngx_http_limit_req_account(limits, n, &excess, &limit);
9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	295
9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	296 if (!delay) {
7592 776d1bebdca2 Limit req: $limit_req_status variable. Roman Arutyunyan <arut@nginx.com> parents: 7515 diff changeset	297 r->main->limit_req_status = NGX_HTTP_LIMIT_REQ_PASSED;
3780 d94d7104f598 change order of limit_req lookup result processing Igor Sysoev <igor@sysoev.ru> parents: 3779 diff changeset	298 return NGX_DECLINED;
2294 ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	299 }
ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	300
3780 d94d7104f598 change order of limit_req lookup result processing Igor Sysoev <igor@sysoev.ru> parents: 3779 diff changeset	301 ngx_log_error(lrcf->delay_log_level, r->connection->log, 0,
7515 2db68852d6a0 Limit req: limit_req_dry_run directive. Roman Arutyunyan <arut@nginx.com> parents: 7399 diff changeset	302 "delaying request%s, excess: %ui.%03ui, by zone \"%V\"",
2db68852d6a0 Limit req: limit_req_dry_run directive. Roman Arutyunyan <arut@nginx.com> parents: 7399 diff changeset	303 lrcf->dry_run ? ", dry run" : "",
4420 9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	304 excess / 1000, excess % 1000, &limit->shm_zone->shm.name);
2294 ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	305
7515 2db68852d6a0 Limit req: limit_req_dry_run directive. Roman Arutyunyan <arut@nginx.com> parents: 7399 diff changeset	306 if (lrcf->dry_run) {
7592 776d1bebdca2 Limit req: $limit_req_status variable. Roman Arutyunyan <arut@nginx.com> parents: 7515 diff changeset	307 r->main->limit_req_status = NGX_HTTP_LIMIT_REQ_DELAYED_DRY_RUN;
7515 2db68852d6a0 Limit req: limit_req_dry_run directive. Roman Arutyunyan <arut@nginx.com> parents: 7399 diff changeset	308 return NGX_DECLINED;
2db68852d6a0 Limit req: limit_req_dry_run directive. Roman Arutyunyan <arut@nginx.com> parents: 7399 diff changeset	309 }
2db68852d6a0 Limit req: limit_req_dry_run directive. Roman Arutyunyan <arut@nginx.com> parents: 7399 diff changeset	310
7592 776d1bebdca2 Limit req: $limit_req_status variable. Roman Arutyunyan <arut@nginx.com> parents: 7515 diff changeset	311 r->main->limit_req_status = NGX_HTTP_LIMIT_REQ_DELAYED;
776d1bebdca2 Limit req: $limit_req_status variable. Roman Arutyunyan <arut@nginx.com> parents: 7515 diff changeset	312
7811 1ebd78df4ce7 Fixed handling of already closed connections. Maxim Dounin <mdounin@mdounin.ru> parents: 7726 diff changeset	313 if (r->connection->read->ready) {
1ebd78df4ce7 Fixed handling of already closed connections. Maxim Dounin <mdounin@mdounin.ru> parents: 7726 diff changeset	314 ngx_post_event(r->connection->read, &ngx_posted_events);
1ebd78df4ce7 Fixed handling of already closed connections. Maxim Dounin <mdounin@mdounin.ru> parents: 7726 diff changeset	315
1ebd78df4ce7 Fixed handling of already closed connections. Maxim Dounin <mdounin@mdounin.ru> parents: 7726 diff changeset	316 } else {
1ebd78df4ce7 Fixed handling of already closed connections. Maxim Dounin <mdounin@mdounin.ru> parents: 7726 diff changeset	317 if (ngx_handle_read_event(r->connection->read, 0) != NGX_OK) {
1ebd78df4ce7 Fixed handling of already closed connections. Maxim Dounin <mdounin@mdounin.ru> parents: 7726 diff changeset	318 return NGX_HTTP_INTERNAL_SERVER_ERROR;
1ebd78df4ce7 Fixed handling of already closed connections. Maxim Dounin <mdounin@mdounin.ru> parents: 7726 diff changeset	319 }
2294 ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	320 }
ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	321
3780 d94d7104f598 change order of limit_req lookup result processing Igor Sysoev <igor@sysoev.ru> parents: 3779 diff changeset	322 r->read_event_handler = ngx_http_test_reading;
d94d7104f598 change order of limit_req lookup result processing Igor Sysoev <igor@sysoev.ru> parents: 3779 diff changeset	323 r->write_event_handler = ngx_http_limit_req_delay;
6959 7fcf209d40c8 Limit req: fixed delaying subrequests. Maxim Dounin <mdounin@mdounin.ru> parents: 6653 diff changeset	324
7fcf209d40c8 Limit req: fixed delaying subrequests. Maxim Dounin <mdounin@mdounin.ru> parents: 6653 diff changeset	325 r->connection->write->delayed = 1;
4420 9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	326 ngx_add_timer(r->connection->write, delay);
2294 ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	327
3780 d94d7104f598 change order of limit_req lookup result processing Igor Sysoev <igor@sysoev.ru> parents: 3779 diff changeset	328 return NGX_AGAIN;
980 7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	329 }
7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	330
7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	331
7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	332 static void
2294 ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	333 ngx_http_limit_req_delay(ngx_http_request_t *r)
ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	334 {
2972 c5ad288f851d fix client write event handling in ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 2912 diff changeset	335 ngx_event_t *wev;
c5ad288f851d fix client write event handling in ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 2912 diff changeset	336
2294 ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	337 ngx_log_debug0(NGX_LOG_DEBUG_HTTP, r->connection->log, 0,
2973 70c8b2d28d1d style fix Igor Sysoev <igor@sysoev.ru> parents: 2972 diff changeset	338 "limit_req delay");
2294 ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	339
2972 c5ad288f851d fix client write event handling in ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 2912 diff changeset	340 wev = r->connection->write;
c5ad288f851d fix client write event handling in ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 2912 diff changeset	341
6961 903fb1ddc07f Moved handling of wev->delayed to the connection event handler. Maxim Dounin <mdounin@mdounin.ru> parents: 6959 diff changeset	342 if (wev->delayed) {
2972 c5ad288f851d fix client write event handling in ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 2912 diff changeset	343
c5ad288f851d fix client write event handling in ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 2912 diff changeset	344 if (ngx_handle_write_event(wev, 0) != NGX_OK) {
c5ad288f851d fix client write event handling in ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 2912 diff changeset	345 ngx_http_finalize_request(r, NGX_HTTP_INTERNAL_SERVER_ERROR);
c5ad288f851d fix client write event handling in ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 2912 diff changeset	346 }
c5ad288f851d fix client write event handling in ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 2912 diff changeset	347
c5ad288f851d fix client write event handling in ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 2912 diff changeset	348 return;
c5ad288f851d fix client write event handling in ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 2912 diff changeset	349 }
c5ad288f851d fix client write event handling in ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 2912 diff changeset	350
2294 ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	351 if (ngx_handle_read_event(r->connection->read, 0) != NGX_OK) {
ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	352 ngx_http_finalize_request(r, NGX_HTTP_INTERNAL_SERVER_ERROR);
ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	353 return;
ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	354 }
ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	355
ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	356 r->read_event_handler = ngx_http_block_reading;
ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	357 r->write_event_handler = ngx_http_core_run_phases;
ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	358
ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	359 ngx_http_core_run_phases(r);
ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	360 }
ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	361
ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	362
ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	363 static void
ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	364 ngx_http_limit_req_rbtree_insert_value(ngx_rbtree_node_t *temp,
1026 38be15c1379a fix duplicate rbtree keys case Igor Sysoev <igor@sysoev.ru> parents: 1012 diff changeset	365 ngx_rbtree_node_t node, ngx_rbtree_node_t sentinel)
38be15c1379a fix duplicate rbtree keys case Igor Sysoev <igor@sysoev.ru> parents: 1012 diff changeset	366 {
2294 ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	367 ngx_rbtree_node_t **p;
2374 7b11f9a1bfe1 rename "lz" to "lr" in variable names Igor Sysoev <igor@sysoev.ru> parents: 2313 diff changeset	368 ngx_http_limit_req_node_t lrn, lrnt;
1026 38be15c1379a fix duplicate rbtree keys case Igor Sysoev <igor@sysoev.ru> parents: 1012 diff changeset	369
38be15c1379a fix duplicate rbtree keys case Igor Sysoev <igor@sysoev.ru> parents: 1012 diff changeset	370 for ( ;; ) {
38be15c1379a fix duplicate rbtree keys case Igor Sysoev <igor@sysoev.ru> parents: 1012 diff changeset	371
38be15c1379a fix duplicate rbtree keys case Igor Sysoev <igor@sysoev.ru> parents: 1012 diff changeset	372 if (node->key < temp->key) {
38be15c1379a fix duplicate rbtree keys case Igor Sysoev <igor@sysoev.ru> parents: 1012 diff changeset	373
1743 4fc402c3ec73 optimize rbtree initialization and insert Igor Sysoev <igor@sysoev.ru> parents: 1406 diff changeset	374 p = &temp->left;
1026 38be15c1379a fix duplicate rbtree keys case Igor Sysoev <igor@sysoev.ru> parents: 1012 diff changeset	375
38be15c1379a fix duplicate rbtree keys case Igor Sysoev <igor@sysoev.ru> parents: 1012 diff changeset	376 } else if (node->key > temp->key) {
38be15c1379a fix duplicate rbtree keys case Igor Sysoev <igor@sysoev.ru> parents: 1012 diff changeset	377
1743 4fc402c3ec73 optimize rbtree initialization and insert Igor Sysoev <igor@sysoev.ru> parents: 1406 diff changeset	378 p = &temp->right;
1026 38be15c1379a fix duplicate rbtree keys case Igor Sysoev <igor@sysoev.ru> parents: 1012 diff changeset	379
38be15c1379a fix duplicate rbtree keys case Igor Sysoev <igor@sysoev.ru> parents: 1012 diff changeset	380 } else { /* node->key == temp->key */
38be15c1379a fix duplicate rbtree keys case Igor Sysoev <igor@sysoev.ru> parents: 1012 diff changeset	381
2374 7b11f9a1bfe1 rename "lz" to "lr" in variable names Igor Sysoev <igor@sysoev.ru> parents: 2313 diff changeset	382 lrn = (ngx_http_limit_req_node_t *) &node->color;
7b11f9a1bfe1 rename "lz" to "lr" in variable names Igor Sysoev <igor@sysoev.ru> parents: 2313 diff changeset	383 lrnt = (ngx_http_limit_req_node_t *) &temp->color;
1026 38be15c1379a fix duplicate rbtree keys case Igor Sysoev <igor@sysoev.ru> parents: 1012 diff changeset	384
2374 7b11f9a1bfe1 rename "lz" to "lr" in variable names Igor Sysoev <igor@sysoev.ru> parents: 2313 diff changeset	385 p = (ngx_memn2cmp(lrn->data, lrnt->data, lrn->len, lrnt->len) < 0)
1743 4fc402c3ec73 optimize rbtree initialization and insert Igor Sysoev <igor@sysoev.ru> parents: 1406 diff changeset	386 ? &temp->left : &temp->right;
4fc402c3ec73 optimize rbtree initialization and insert Igor Sysoev <igor@sysoev.ru> parents: 1406 diff changeset	387 }
1026 38be15c1379a fix duplicate rbtree keys case Igor Sysoev <igor@sysoev.ru> parents: 1012 diff changeset	388
1743 4fc402c3ec73 optimize rbtree initialization and insert Igor Sysoev <igor@sysoev.ru> parents: 1406 diff changeset	389 if (*p == sentinel) {
4fc402c3ec73 optimize rbtree initialization and insert Igor Sysoev <igor@sysoev.ru> parents: 1406 diff changeset	390 break;
4fc402c3ec73 optimize rbtree initialization and insert Igor Sysoev <igor@sysoev.ru> parents: 1406 diff changeset	391 }
1026 38be15c1379a fix duplicate rbtree keys case Igor Sysoev <igor@sysoev.ru> parents: 1012 diff changeset	392
1743 4fc402c3ec73 optimize rbtree initialization and insert Igor Sysoev <igor@sysoev.ru> parents: 1406 diff changeset	393 temp = *p;
1026 38be15c1379a fix duplicate rbtree keys case Igor Sysoev <igor@sysoev.ru> parents: 1012 diff changeset	394 }
38be15c1379a fix duplicate rbtree keys case Igor Sysoev <igor@sysoev.ru> parents: 1012 diff changeset	395
1743 4fc402c3ec73 optimize rbtree initialization and insert Igor Sysoev <igor@sysoev.ru> parents: 1406 diff changeset	396 *p = node;
1026 38be15c1379a fix duplicate rbtree keys case Igor Sysoev <igor@sysoev.ru> parents: 1012 diff changeset	397 node->parent = temp;
38be15c1379a fix duplicate rbtree keys case Igor Sysoev <igor@sysoev.ru> parents: 1012 diff changeset	398 node->left = sentinel;
38be15c1379a fix duplicate rbtree keys case Igor Sysoev <igor@sysoev.ru> parents: 1012 diff changeset	399 node->right = sentinel;
38be15c1379a fix duplicate rbtree keys case Igor Sysoev <igor@sysoev.ru> parents: 1012 diff changeset	400 ngx_rbt_red(node);
38be15c1379a fix duplicate rbtree keys case Igor Sysoev <igor@sysoev.ru> parents: 1012 diff changeset	401 }
38be15c1379a fix duplicate rbtree keys case Igor Sysoev <igor@sysoev.ru> parents: 1012 diff changeset	402
38be15c1379a fix duplicate rbtree keys case Igor Sysoev <igor@sysoev.ru> parents: 1012 diff changeset	403
2294 ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	404 static ngx_int_t
4420 9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	405 ngx_http_limit_req_lookup(ngx_http_limit_req_limit_t *limit, ngx_uint_t hash,
5863 102f85699420 Limit req: reduced number of parameters in the lookup function. Valentin Bartenev <vbart@nginx.com> parents: 5862 diff changeset	406 ngx_str_t key, ngx_uint_t ep, ngx_uint_t account)
980 7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	407 {
4418 aac79fc948cc Limit req: allocation and initialization of a new node moved to the lookup Valentin Bartenev <vbart@nginx.com> parents: 4417 diff changeset	408 size_t size;
2313 b56a9ba4824d use integer instead of float Igor Sysoev <igor@sysoev.ru> parents: 2312 diff changeset	409 ngx_int_t rc, excess;
2294 ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	410 ngx_msec_t now;
ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	411 ngx_msec_int_t ms;
ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	412 ngx_rbtree_node_t node, sentinel;
ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	413 ngx_http_limit_req_ctx_t *ctx;
2374 7b11f9a1bfe1 rename "lz" to "lr" in variable names Igor Sysoev <igor@sysoev.ru> parents: 2313 diff changeset	414 ngx_http_limit_req_node_t *lr;
2294 ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	415
6653 7a6456398fc3 Simplified extraction of current time. Ruslan Ermilov <ru@nginx.com> parents: 6117 diff changeset	416 now = ngx_current_msec;
4418 aac79fc948cc Limit req: allocation and initialization of a new node moved to the lookup Valentin Bartenev <vbart@nginx.com> parents: 4417 diff changeset	417
4420 9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	418 ctx = limit->shm_zone->data;
2294 ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	419
2720 b3b8c66bd520 support attaching to an existent Win32 shared memory Igor Sysoev <igor@sysoev.ru> parents: 2716 diff changeset	420 node = ctx->sh->rbtree.root;
b3b8c66bd520 support attaching to an existent Win32 shared memory Igor Sysoev <igor@sysoev.ru> parents: 2716 diff changeset	421 sentinel = ctx->sh->rbtree.sentinel;
980 7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	422
2294 ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	423 while (node != sentinel) {
ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	424
ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	425 if (hash < node->key) {
ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	426 node = node->left;
ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	427 continue;
ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	428 }
ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	429
ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	430 if (hash > node->key) {
ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	431 node = node->right;
ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	432 continue;
ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	433 }
ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	434
ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	435 /* hash == node->key */
ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	436
4497 95ab6658654a Fix of rbtree lookup on hash collisions. Maxim Dounin <mdounin@mdounin.ru> parents: 4424 diff changeset	437 lr = (ngx_http_limit_req_node_t *) &node->color;
2294 ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	438
5863 102f85699420 Limit req: reduced number of parameters in the lookup function. Valentin Bartenev <vbart@nginx.com> parents: 5862 diff changeset	439 rc = ngx_memn2cmp(key->data, lr->data, key->len, (size_t) lr->len);
3191 30f841e2536d fix r3184 Igor Sysoev <igor@sysoev.ru> parents: 3185 diff changeset	440
4497 95ab6658654a Fix of rbtree lookup on hash collisions. Maxim Dounin <mdounin@mdounin.ru> parents: 4424 diff changeset	441 if (rc == 0) {
95ab6658654a Fix of rbtree lookup on hash collisions. Maxim Dounin <mdounin@mdounin.ru> parents: 4424 diff changeset	442 ngx_queue_remove(&lr->queue);
95ab6658654a Fix of rbtree lookup on hash collisions. Maxim Dounin <mdounin@mdounin.ru> parents: 4424 diff changeset	443 ngx_queue_insert_head(&ctx->sh->queue, &lr->queue);
3183 b87542338ac3 make limit_req to conform to the leaky bucket algorithm Igor Sysoev <igor@sysoev.ru> parents: 2973 diff changeset	444
4497 95ab6658654a Fix of rbtree lookup on hash collisions. Maxim Dounin <mdounin@mdounin.ru> parents: 4424 diff changeset	445 ms = (ngx_msec_int_t) (now - lr->last);
980 7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	446
7281 bd6563e81cea Limit req: improved handling of negative times. Maxim Dounin <mdounin@mdounin.ru> parents: 6961 diff changeset	447 if (ms < -60000) {
bd6563e81cea Limit req: improved handling of negative times. Maxim Dounin <mdounin@mdounin.ru> parents: 6961 diff changeset	448 ms = 1;
bd6563e81cea Limit req: improved handling of negative times. Maxim Dounin <mdounin@mdounin.ru> parents: 6961 diff changeset	449
bd6563e81cea Limit req: improved handling of negative times. Maxim Dounin <mdounin@mdounin.ru> parents: 6961 diff changeset	450 } else if (ms < 0) {
bd6563e81cea Limit req: improved handling of negative times. Maxim Dounin <mdounin@mdounin.ru> parents: 6961 diff changeset	451 ms = 0;
bd6563e81cea Limit req: improved handling of negative times. Maxim Dounin <mdounin@mdounin.ru> parents: 6961 diff changeset	452 }
bd6563e81cea Limit req: improved handling of negative times. Maxim Dounin <mdounin@mdounin.ru> parents: 6961 diff changeset	453
bd6563e81cea Limit req: improved handling of negative times. Maxim Dounin <mdounin@mdounin.ru> parents: 6961 diff changeset	454 excess = lr->excess - ctx->rate * ms / 1000 + 1000;
4420 9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	455
4497 95ab6658654a Fix of rbtree lookup on hash collisions. Maxim Dounin <mdounin@mdounin.ru> parents: 4424 diff changeset	456 if (excess < 0) {
95ab6658654a Fix of rbtree lookup on hash collisions. Maxim Dounin <mdounin@mdounin.ru> parents: 4424 diff changeset	457 excess = 0;
2294 ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	458 }
ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	459
4497 95ab6658654a Fix of rbtree lookup on hash collisions. Maxim Dounin <mdounin@mdounin.ru> parents: 4424 diff changeset	460 *ep = excess;
95ab6658654a Fix of rbtree lookup on hash collisions. Maxim Dounin <mdounin@mdounin.ru> parents: 4424 diff changeset	461
95ab6658654a Fix of rbtree lookup on hash collisions. Maxim Dounin <mdounin@mdounin.ru> parents: 4424 diff changeset	462 if ((ngx_uint_t) excess > limit->burst) {
95ab6658654a Fix of rbtree lookup on hash collisions. Maxim Dounin <mdounin@mdounin.ru> parents: 4424 diff changeset	463 return NGX_BUSY;
95ab6658654a Fix of rbtree lookup on hash collisions. Maxim Dounin <mdounin@mdounin.ru> parents: 4424 diff changeset	464 }
2294 ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	465
4497 95ab6658654a Fix of rbtree lookup on hash collisions. Maxim Dounin <mdounin@mdounin.ru> parents: 4424 diff changeset	466 if (account) {
95ab6658654a Fix of rbtree lookup on hash collisions. Maxim Dounin <mdounin@mdounin.ru> parents: 4424 diff changeset	467 lr->excess = excess;
7281 bd6563e81cea Limit req: improved handling of negative times. Maxim Dounin <mdounin@mdounin.ru> parents: 6961 diff changeset	468
bd6563e81cea Limit req: improved handling of negative times. Maxim Dounin <mdounin@mdounin.ru> parents: 6961 diff changeset	469 if (ms) {
bd6563e81cea Limit req: improved handling of negative times. Maxim Dounin <mdounin@mdounin.ru> parents: 6961 diff changeset	470 lr->last = now;
bd6563e81cea Limit req: improved handling of negative times. Maxim Dounin <mdounin@mdounin.ru> parents: 6961 diff changeset	471 }
bd6563e81cea Limit req: improved handling of negative times. Maxim Dounin <mdounin@mdounin.ru> parents: 6961 diff changeset	472
4497 95ab6658654a Fix of rbtree lookup on hash collisions. Maxim Dounin <mdounin@mdounin.ru> parents: 4424 diff changeset	473 return NGX_OK;
95ab6658654a Fix of rbtree lookup on hash collisions. Maxim Dounin <mdounin@mdounin.ru> parents: 4424 diff changeset	474 }
2294 ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	475
4497 95ab6658654a Fix of rbtree lookup on hash collisions. Maxim Dounin <mdounin@mdounin.ru> parents: 4424 diff changeset	476 lr->count++;
95ab6658654a Fix of rbtree lookup on hash collisions. Maxim Dounin <mdounin@mdounin.ru> parents: 4424 diff changeset	477
95ab6658654a Fix of rbtree lookup on hash collisions. Maxim Dounin <mdounin@mdounin.ru> parents: 4424 diff changeset	478 ctx->node = lr;
95ab6658654a Fix of rbtree lookup on hash collisions. Maxim Dounin <mdounin@mdounin.ru> parents: 4424 diff changeset	479
95ab6658654a Fix of rbtree lookup on hash collisions. Maxim Dounin <mdounin@mdounin.ru> parents: 4424 diff changeset	480 return NGX_AGAIN;
95ab6658654a Fix of rbtree lookup on hash collisions. Maxim Dounin <mdounin@mdounin.ru> parents: 4424 diff changeset	481 }
95ab6658654a Fix of rbtree lookup on hash collisions. Maxim Dounin <mdounin@mdounin.ru> parents: 4424 diff changeset	482
95ab6658654a Fix of rbtree lookup on hash collisions. Maxim Dounin <mdounin@mdounin.ru> parents: 4424 diff changeset	483 node = (rc < 0) ? node->left : node->right;
980 7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	484 }
7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	485
3779 57aecfdcac3d an excess was logged as 0.000 if requests were limited without delay: Igor Sysoev <igor@sysoev.ru> parents: 3525 diff changeset	486 *ep = 0;
2294 ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	487
4418 aac79fc948cc Limit req: allocation and initialization of a new node moved to the lookup Valentin Bartenev <vbart@nginx.com> parents: 4417 diff changeset	488 size = offsetof(ngx_rbtree_node_t, color)
aac79fc948cc Limit req: allocation and initialization of a new node moved to the lookup Valentin Bartenev <vbart@nginx.com> parents: 4417 diff changeset	489 + offsetof(ngx_http_limit_req_node_t, data)
5863 102f85699420 Limit req: reduced number of parameters in the lookup function. Valentin Bartenev <vbart@nginx.com> parents: 5862 diff changeset	490 + key->len;
4418 aac79fc948cc Limit req: allocation and initialization of a new node moved to the lookup Valentin Bartenev <vbart@nginx.com> parents: 4417 diff changeset	491
4419 7084faa7a4b4 Limit req: number of cleanup calls reduced. Valentin Bartenev <vbart@nginx.com> parents: 4418 diff changeset	492 ngx_http_limit_req_expire(ctx, 1);
7084faa7a4b4 Limit req: number of cleanup calls reduced. Valentin Bartenev <vbart@nginx.com> parents: 4418 diff changeset	493
4418 aac79fc948cc Limit req: allocation and initialization of a new node moved to the lookup Valentin Bartenev <vbart@nginx.com> parents: 4417 diff changeset	494 node = ngx_slab_alloc_locked(ctx->shpool, size);
aac79fc948cc Limit req: allocation and initialization of a new node moved to the lookup Valentin Bartenev <vbart@nginx.com> parents: 4417 diff changeset	495
aac79fc948cc Limit req: allocation and initialization of a new node moved to the lookup Valentin Bartenev <vbart@nginx.com> parents: 4417 diff changeset	496 if (node == NULL) {
aac79fc948cc Limit req: allocation and initialization of a new node moved to the lookup Valentin Bartenev <vbart@nginx.com> parents: 4417 diff changeset	497 ngx_http_limit_req_expire(ctx, 0);
aac79fc948cc Limit req: allocation and initialization of a new node moved to the lookup Valentin Bartenev <vbart@nginx.com> parents: 4417 diff changeset	498
aac79fc948cc Limit req: allocation and initialization of a new node moved to the lookup Valentin Bartenev <vbart@nginx.com> parents: 4417 diff changeset	499 node = ngx_slab_alloc_locked(ctx->shpool, size);
aac79fc948cc Limit req: allocation and initialization of a new node moved to the lookup Valentin Bartenev <vbart@nginx.com> parents: 4417 diff changeset	500 if (node == NULL) {
5634 5024d29354f1 Core: slab log_nomem flag. Maxim Dounin <mdounin@mdounin.ru> parents: 5411 diff changeset	501 ngx_log_error(NGX_LOG_ALERT, ngx_cycle->log, 0,
5024d29354f1 Core: slab log_nomem flag. Maxim Dounin <mdounin@mdounin.ru> parents: 5411 diff changeset	502 "could not allocate node%s", ctx->shpool->log_ctx);
4418 aac79fc948cc Limit req: allocation and initialization of a new node moved to the lookup Valentin Bartenev <vbart@nginx.com> parents: 4417 diff changeset	503 return NGX_ERROR;
aac79fc948cc Limit req: allocation and initialization of a new node moved to the lookup Valentin Bartenev <vbart@nginx.com> parents: 4417 diff changeset	504 }
aac79fc948cc Limit req: allocation and initialization of a new node moved to the lookup Valentin Bartenev <vbart@nginx.com> parents: 4417 diff changeset	505 }
aac79fc948cc Limit req: allocation and initialization of a new node moved to the lookup Valentin Bartenev <vbart@nginx.com> parents: 4417 diff changeset	506
aac79fc948cc Limit req: allocation and initialization of a new node moved to the lookup Valentin Bartenev <vbart@nginx.com> parents: 4417 diff changeset	507 node->key = hash;
aac79fc948cc Limit req: allocation and initialization of a new node moved to the lookup Valentin Bartenev <vbart@nginx.com> parents: 4417 diff changeset	508
aac79fc948cc Limit req: allocation and initialization of a new node moved to the lookup Valentin Bartenev <vbart@nginx.com> parents: 4417 diff changeset	509 lr = (ngx_http_limit_req_node_t *) &node->color;
aac79fc948cc Limit req: allocation and initialization of a new node moved to the lookup Valentin Bartenev <vbart@nginx.com> parents: 4417 diff changeset	510
5863 102f85699420 Limit req: reduced number of parameters in the lookup function. Valentin Bartenev <vbart@nginx.com> parents: 5862 diff changeset	511 lr->len = (u_short) key->len;
4418 aac79fc948cc Limit req: allocation and initialization of a new node moved to the lookup Valentin Bartenev <vbart@nginx.com> parents: 4417 diff changeset	512 lr->excess = 0;
aac79fc948cc Limit req: allocation and initialization of a new node moved to the lookup Valentin Bartenev <vbart@nginx.com> parents: 4417 diff changeset	513
5863 102f85699420 Limit req: reduced number of parameters in the lookup function. Valentin Bartenev <vbart@nginx.com> parents: 5862 diff changeset	514 ngx_memcpy(lr->data, key->data, key->len);
4418 aac79fc948cc Limit req: allocation and initialization of a new node moved to the lookup Valentin Bartenev <vbart@nginx.com> parents: 4417 diff changeset	515
4832 949ea3d3cd1a Limit req: fix of rbtree node insertion on hash collisions. Valentin Bartenev <vbart@nginx.com> parents: 4811 diff changeset	516 ngx_rbtree_insert(&ctx->sh->rbtree, node);
949ea3d3cd1a Limit req: fix of rbtree node insertion on hash collisions. Valentin Bartenev <vbart@nginx.com> parents: 4811 diff changeset	517
949ea3d3cd1a Limit req: fix of rbtree node insertion on hash collisions. Valentin Bartenev <vbart@nginx.com> parents: 4811 diff changeset	518 ngx_queue_insert_head(&ctx->sh->queue, &lr->queue);
949ea3d3cd1a Limit req: fix of rbtree node insertion on hash collisions. Valentin Bartenev <vbart@nginx.com> parents: 4811 diff changeset	519
4420 9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	520 if (account) {
9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	521 lr->last = now;
9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	522 lr->count = 0;
9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	523 return NGX_OK;
9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	524 }
9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	525
9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	526 lr->last = 0;
9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	527 lr->count = 1;
9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	528
9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	529 ctx->node = lr;
9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	530
9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	531 return NGX_AGAIN;
9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	532 }
9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	533
9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	534
9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	535 static ngx_msec_t
9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	536 ngx_http_limit_req_account(ngx_http_limit_req_limit_t *limits, ngx_uint_t n,
9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	537 ngx_uint_t ep, ngx_http_limit_req_limit_t *limit)
9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	538 {
9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	539 ngx_int_t excess;
9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	540 ngx_msec_t now, delay, max_delay;
9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	541 ngx_msec_int_t ms;
9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	542 ngx_http_limit_req_ctx_t *ctx;
9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	543 ngx_http_limit_req_node_t *lr;
9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	544
9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	545 excess = *ep;
9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	546
7399 d6ca744c727e Limit req: "delay=" parameter. Maxim Dounin <mdounin@mdounin.ru> parents: 7398 diff changeset	547 if ((ngx_uint_t) excess <= (*limit)->delay) {
4420 9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	548 max_delay = 0;
9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	549
9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	550 } else {
9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	551 ctx = (*limit)->shm_zone->data;
7399 d6ca744c727e Limit req: "delay=" parameter. Maxim Dounin <mdounin@mdounin.ru> parents: 7398 diff changeset	552 max_delay = (excess - (limit)->delay) 1000 / ctx->rate;
4420 9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	553 }
9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	554
9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	555 while (n--) {
9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	556 ctx = limits[n].shm_zone->data;
9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	557 lr = ctx->node;
9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	558
9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	559 if (lr == NULL) {
9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	560 continue;
9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	561 }
9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	562
9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	563 ngx_shmtx_lock(&ctx->shpool->mutex);
9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	564
6653 7a6456398fc3 Simplified extraction of current time. Ruslan Ermilov <ru@nginx.com> parents: 6117 diff changeset	565 now = ngx_current_msec;
4420 9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	566 ms = (ngx_msec_int_t) (now - lr->last);
9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	567
7281 bd6563e81cea Limit req: improved handling of negative times. Maxim Dounin <mdounin@mdounin.ru> parents: 6961 diff changeset	568 if (ms < -60000) {
bd6563e81cea Limit req: improved handling of negative times. Maxim Dounin <mdounin@mdounin.ru> parents: 6961 diff changeset	569 ms = 1;
bd6563e81cea Limit req: improved handling of negative times. Maxim Dounin <mdounin@mdounin.ru> parents: 6961 diff changeset	570
bd6563e81cea Limit req: improved handling of negative times. Maxim Dounin <mdounin@mdounin.ru> parents: 6961 diff changeset	571 } else if (ms < 0) {
bd6563e81cea Limit req: improved handling of negative times. Maxim Dounin <mdounin@mdounin.ru> parents: 6961 diff changeset	572 ms = 0;
bd6563e81cea Limit req: improved handling of negative times. Maxim Dounin <mdounin@mdounin.ru> parents: 6961 diff changeset	573 }
bd6563e81cea Limit req: improved handling of negative times. Maxim Dounin <mdounin@mdounin.ru> parents: 6961 diff changeset	574
bd6563e81cea Limit req: improved handling of negative times. Maxim Dounin <mdounin@mdounin.ru> parents: 6961 diff changeset	575 excess = lr->excess - ctx->rate * ms / 1000 + 1000;
4420 9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	576
9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	577 if (excess < 0) {
9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	578 excess = 0;
9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	579 }
9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	580
7281 bd6563e81cea Limit req: improved handling of negative times. Maxim Dounin <mdounin@mdounin.ru> parents: 6961 diff changeset	581 if (ms) {
bd6563e81cea Limit req: improved handling of negative times. Maxim Dounin <mdounin@mdounin.ru> parents: 6961 diff changeset	582 lr->last = now;
bd6563e81cea Limit req: improved handling of negative times. Maxim Dounin <mdounin@mdounin.ru> parents: 6961 diff changeset	583 }
bd6563e81cea Limit req: improved handling of negative times. Maxim Dounin <mdounin@mdounin.ru> parents: 6961 diff changeset	584
4420 9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	585 lr->excess = excess;
9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	586 lr->count--;
9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	587
9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	588 ngx_shmtx_unlock(&ctx->shpool->mutex);
9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	589
9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	590 ctx->node = NULL;
9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	591
7399 d6ca744c727e Limit req: "delay=" parameter. Maxim Dounin <mdounin@mdounin.ru> parents: 7398 diff changeset	592 if ((ngx_uint_t) excess <= limits[n].delay) {
4420 9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	593 continue;
9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	594 }
9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	595
7399 d6ca744c727e Limit req: "delay=" parameter. Maxim Dounin <mdounin@mdounin.ru> parents: 7398 diff changeset	596 delay = (excess - limits[n].delay) * 1000 / ctx->rate;
4420 9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	597
9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	598 if (delay > max_delay) {
9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	599 max_delay = delay;
9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	600 *ep = excess;
9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	601 *limit = &limits[n];
9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	602 }
9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	603 }
9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	604
9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	605 return max_delay;
2294 ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	606 }
ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	607
ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	608
ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	609 static void
7726 559d19037984 Limit req: unlocking of nodes on complex value errors. Maxim Dounin <mdounin@mdounin.ru> parents: 7592 diff changeset	610 ngx_http_limit_req_unlock(ngx_http_limit_req_limit_t *limits, ngx_uint_t n)
559d19037984 Limit req: unlocking of nodes on complex value errors. Maxim Dounin <mdounin@mdounin.ru> parents: 7592 diff changeset	611 {
559d19037984 Limit req: unlocking of nodes on complex value errors. Maxim Dounin <mdounin@mdounin.ru> parents: 7592 diff changeset	612 ngx_http_limit_req_ctx_t *ctx;
559d19037984 Limit req: unlocking of nodes on complex value errors. Maxim Dounin <mdounin@mdounin.ru> parents: 7592 diff changeset	613
559d19037984 Limit req: unlocking of nodes on complex value errors. Maxim Dounin <mdounin@mdounin.ru> parents: 7592 diff changeset	614 while (n--) {
559d19037984 Limit req: unlocking of nodes on complex value errors. Maxim Dounin <mdounin@mdounin.ru> parents: 7592 diff changeset	615 ctx = limits[n].shm_zone->data;
559d19037984 Limit req: unlocking of nodes on complex value errors. Maxim Dounin <mdounin@mdounin.ru> parents: 7592 diff changeset	616
559d19037984 Limit req: unlocking of nodes on complex value errors. Maxim Dounin <mdounin@mdounin.ru> parents: 7592 diff changeset	617 if (ctx->node == NULL) {
559d19037984 Limit req: unlocking of nodes on complex value errors. Maxim Dounin <mdounin@mdounin.ru> parents: 7592 diff changeset	618 continue;
559d19037984 Limit req: unlocking of nodes on complex value errors. Maxim Dounin <mdounin@mdounin.ru> parents: 7592 diff changeset	619 }
559d19037984 Limit req: unlocking of nodes on complex value errors. Maxim Dounin <mdounin@mdounin.ru> parents: 7592 diff changeset	620
559d19037984 Limit req: unlocking of nodes on complex value errors. Maxim Dounin <mdounin@mdounin.ru> parents: 7592 diff changeset	621 ngx_shmtx_lock(&ctx->shpool->mutex);
559d19037984 Limit req: unlocking of nodes on complex value errors. Maxim Dounin <mdounin@mdounin.ru> parents: 7592 diff changeset	622
559d19037984 Limit req: unlocking of nodes on complex value errors. Maxim Dounin <mdounin@mdounin.ru> parents: 7592 diff changeset	623 ctx->node->count--;
559d19037984 Limit req: unlocking of nodes on complex value errors. Maxim Dounin <mdounin@mdounin.ru> parents: 7592 diff changeset	624
559d19037984 Limit req: unlocking of nodes on complex value errors. Maxim Dounin <mdounin@mdounin.ru> parents: 7592 diff changeset	625 ngx_shmtx_unlock(&ctx->shpool->mutex);
559d19037984 Limit req: unlocking of nodes on complex value errors. Maxim Dounin <mdounin@mdounin.ru> parents: 7592 diff changeset	626
559d19037984 Limit req: unlocking of nodes on complex value errors. Maxim Dounin <mdounin@mdounin.ru> parents: 7592 diff changeset	627 ctx->node = NULL;
559d19037984 Limit req: unlocking of nodes on complex value errors. Maxim Dounin <mdounin@mdounin.ru> parents: 7592 diff changeset	628 }
559d19037984 Limit req: unlocking of nodes on complex value errors. Maxim Dounin <mdounin@mdounin.ru> parents: 7592 diff changeset	629 }
559d19037984 Limit req: unlocking of nodes on complex value errors. Maxim Dounin <mdounin@mdounin.ru> parents: 7592 diff changeset	630
559d19037984 Limit req: unlocking of nodes on complex value errors. Maxim Dounin <mdounin@mdounin.ru> parents: 7592 diff changeset	631
559d19037984 Limit req: unlocking of nodes on complex value errors. Maxim Dounin <mdounin@mdounin.ru> parents: 7592 diff changeset	632 static void
2294 ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	633 ngx_http_limit_req_expire(ngx_http_limit_req_ctx_t *ctx, ngx_uint_t n)
ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	634 {
2313 b56a9ba4824d use integer instead of float Igor Sysoev <igor@sysoev.ru> parents: 2312 diff changeset	635 ngx_int_t excess;
2294 ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	636 ngx_msec_t now;
ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	637 ngx_queue_t *q;
ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	638 ngx_msec_int_t ms;
ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	639 ngx_rbtree_node_t *node;
2374 7b11f9a1bfe1 rename "lz" to "lr" in variable names Igor Sysoev <igor@sysoev.ru> parents: 2313 diff changeset	640 ngx_http_limit_req_node_t *lr;
2294 ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	641
6653 7a6456398fc3 Simplified extraction of current time. Ruslan Ermilov <ru@nginx.com> parents: 6117 diff changeset	642 now = ngx_current_msec;
2294 ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	643
ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	644 /*
ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	645 * n == 1 deletes one or two zero rate entries
ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	646 * n == 0 deletes oldest entry by force
ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	647 * and one or two zero rate entries
ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	648 */
ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	649
ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	650 while (n < 3) {
ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	651
2720 b3b8c66bd520 support attaching to an existent Win32 shared memory Igor Sysoev <igor@sysoev.ru> parents: 2716 diff changeset	652 if (ngx_queue_empty(&ctx->sh->queue)) {
2294 ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	653 return;
ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	654 }
ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	655
2720 b3b8c66bd520 support attaching to an existent Win32 shared memory Igor Sysoev <igor@sysoev.ru> parents: 2716 diff changeset	656 q = ngx_queue_last(&ctx->sh->queue);
2294 ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	657
2374 7b11f9a1bfe1 rename "lz" to "lr" in variable names Igor Sysoev <igor@sysoev.ru> parents: 2313 diff changeset	658 lr = ngx_queue_data(q, ngx_http_limit_req_node_t, queue);
2294 ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	659
4420 9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	660 if (lr->count) {
9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	661
9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	662 /*
9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	663 * There is not much sense in looking further,
9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	664 * because we bump nodes on the lookup stage.
9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	665 */
9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	666
9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	667 return;
9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	668 }
9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	669
2294 ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	670 if (n++ != 0) {
ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	671
2374 7b11f9a1bfe1 rename "lz" to "lr" in variable names Igor Sysoev <igor@sysoev.ru> parents: 2313 diff changeset	672 ms = (ngx_msec_int_t) (now - lr->last);
2294 ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	673 ms = ngx_abs(ms);
ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	674
ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	675 if (ms < 60000) {
ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	676 return;
ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	677 }
ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	678
2374 7b11f9a1bfe1 rename "lz" to "lr" in variable names Igor Sysoev <igor@sysoev.ru> parents: 2313 diff changeset	679 excess = lr->excess - ctx->rate * ms / 1000;
2294 ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	680
2313 b56a9ba4824d use integer instead of float Igor Sysoev <igor@sysoev.ru> parents: 2312 diff changeset	681 if (excess > 0) {
2294 ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	682 return;
ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	683 }
ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	684 }
ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	685
ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	686 ngx_queue_remove(q);
ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	687
ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	688 node = (ngx_rbtree_node_t *)
2374 7b11f9a1bfe1 rename "lz" to "lr" in variable names Igor Sysoev <igor@sysoev.ru> parents: 2313 diff changeset	689 ((u_char *) lr - offsetof(ngx_rbtree_node_t, color));
2294 ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	690
2720 b3b8c66bd520 support attaching to an existent Win32 shared memory Igor Sysoev <igor@sysoev.ru> parents: 2716 diff changeset	691 ngx_rbtree_delete(&ctx->sh->rbtree, node);
2294 ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	692
ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	693 ngx_slab_free_locked(ctx->shpool, node);
ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	694 }
980 7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	695 }
7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	696
7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	697
7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	698 static ngx_int_t
2294 ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	699 ngx_http_limit_req_init_zone(ngx_shm_zone_t shm_zone, void data)
980 7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	700 {
2294 ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	701 ngx_http_limit_req_ctx_t *octx = data;
993 1b9a4d92173f pass the inherited shm_zone data Igor Sysoev <igor@sysoev.ru> parents: 987 diff changeset	702
2611 2bce3f6416c6 improve ngx_slab_alloc() error logging Igor Sysoev <igor@sysoev.ru> parents: 2376 diff changeset	703 size_t len;
2294 ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	704 ngx_http_limit_req_ctx_t *ctx;
980 7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	705
993 1b9a4d92173f pass the inherited shm_zone data Igor Sysoev <igor@sysoev.ru> parents: 987 diff changeset	706 ctx = shm_zone->data;
1b9a4d92173f pass the inherited shm_zone data Igor Sysoev <igor@sysoev.ru> parents: 987 diff changeset	707
1b9a4d92173f pass the inherited shm_zone data Igor Sysoev <igor@sysoev.ru> parents: 987 diff changeset	708 if (octx) {
5862 ecbb99aa0e12 Limit req: use complex value in limit_req_zone. Valentin Bartenev <vbart@nginx.com> parents: 5846 diff changeset	709 if (ctx->key.value.len != octx->key.value.len
ecbb99aa0e12 Limit req: use complex value in limit_req_zone. Valentin Bartenev <vbart@nginx.com> parents: 5846 diff changeset	710 \|\| ngx_strncmp(ctx->key.value.data, octx->key.value.data,
ecbb99aa0e12 Limit req: use complex value in limit_req_zone. Valentin Bartenev <vbart@nginx.com> parents: 5846 diff changeset	711 ctx->key.value.len)
ecbb99aa0e12 Limit req: use complex value in limit_req_zone. Valentin Bartenev <vbart@nginx.com> parents: 5846 diff changeset	712 != 0)
ecbb99aa0e12 Limit req: use complex value in limit_req_zone. Valentin Bartenev <vbart@nginx.com> parents: 5846 diff changeset	713 {
993 1b9a4d92173f pass the inherited shm_zone data Igor Sysoev <igor@sysoev.ru> parents: 987 diff changeset	714 ngx_log_error(NGX_LOG_EMERG, shm_zone->shm.log, 0,
5862 ecbb99aa0e12 Limit req: use complex value in limit_req_zone. Valentin Bartenev <vbart@nginx.com> parents: 5846 diff changeset	715 "limit_req \"%V\" uses the \"%V\" key "
ecbb99aa0e12 Limit req: use complex value in limit_req_zone. Valentin Bartenev <vbart@nginx.com> parents: 5846 diff changeset	716 "while previously it used the \"%V\" key",
ecbb99aa0e12 Limit req: use complex value in limit_req_zone. Valentin Bartenev <vbart@nginx.com> parents: 5846 diff changeset	717 &shm_zone->shm.name, &ctx->key.value,
ecbb99aa0e12 Limit req: use complex value in limit_req_zone. Valentin Bartenev <vbart@nginx.com> parents: 5846 diff changeset	718 &octx->key.value);
993 1b9a4d92173f pass the inherited shm_zone data Igor Sysoev <igor@sysoev.ru> parents: 987 diff changeset	719 return NGX_ERROR;
1b9a4d92173f pass the inherited shm_zone data Igor Sysoev <igor@sysoev.ru> parents: 987 diff changeset	720 }
1b9a4d92173f pass the inherited shm_zone data Igor Sysoev <igor@sysoev.ru> parents: 987 diff changeset	721
2720 b3b8c66bd520 support attaching to an existent Win32 shared memory Igor Sysoev <igor@sysoev.ru> parents: 2716 diff changeset	722 ctx->sh = octx->sh;
2294 ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	723 ctx->shpool = octx->shpool;
993 1b9a4d92173f pass the inherited shm_zone data Igor Sysoev <igor@sysoev.ru> parents: 987 diff changeset	724
1b9a4d92173f pass the inherited shm_zone data Igor Sysoev <igor@sysoev.ru> parents: 987 diff changeset	725 return NGX_OK;
1b9a4d92173f pass the inherited shm_zone data Igor Sysoev <igor@sysoev.ru> parents: 987 diff changeset	726 }
1b9a4d92173f pass the inherited shm_zone data Igor Sysoev <igor@sysoev.ru> parents: 987 diff changeset	727
2294 ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	728 ctx->shpool = (ngx_slab_pool_t *) shm_zone->shm.addr;
980 7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	729
2720 b3b8c66bd520 support attaching to an existent Win32 shared memory Igor Sysoev <igor@sysoev.ru> parents: 2716 diff changeset	730 if (shm_zone->shm.exists) {
b3b8c66bd520 support attaching to an existent Win32 shared memory Igor Sysoev <igor@sysoev.ru> parents: 2716 diff changeset	731 ctx->sh = ctx->shpool->data;
b3b8c66bd520 support attaching to an existent Win32 shared memory Igor Sysoev <igor@sysoev.ru> parents: 2716 diff changeset	732
b3b8c66bd520 support attaching to an existent Win32 shared memory Igor Sysoev <igor@sysoev.ru> parents: 2716 diff changeset	733 return NGX_OK;
980 7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	734 }
7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	735
2720 b3b8c66bd520 support attaching to an existent Win32 shared memory Igor Sysoev <igor@sysoev.ru> parents: 2716 diff changeset	736 ctx->sh = ngx_slab_alloc(ctx->shpool, sizeof(ngx_http_limit_req_shctx_t));
b3b8c66bd520 support attaching to an existent Win32 shared memory Igor Sysoev <igor@sysoev.ru> parents: 2716 diff changeset	737 if (ctx->sh == NULL) {
980 7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	738 return NGX_ERROR;
7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	739 }
7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	740
2720 b3b8c66bd520 support attaching to an existent Win32 shared memory Igor Sysoev <igor@sysoev.ru> parents: 2716 diff changeset	741 ctx->shpool->data = ctx->sh;
b3b8c66bd520 support attaching to an existent Win32 shared memory Igor Sysoev <igor@sysoev.ru> parents: 2716 diff changeset	742
b3b8c66bd520 support attaching to an existent Win32 shared memory Igor Sysoev <igor@sysoev.ru> parents: 2716 diff changeset	743 ngx_rbtree_init(&ctx->sh->rbtree, &ctx->sh->sentinel,
2294 ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	744 ngx_http_limit_req_rbtree_insert_value);
ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	745
2720 b3b8c66bd520 support attaching to an existent Win32 shared memory Igor Sysoev <igor@sysoev.ru> parents: 2716 diff changeset	746 ngx_queue_init(&ctx->sh->queue);
980 7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	747
2716 d5896f6608e8 move zone name from ngx_shm_zone_t to ngx_shm_t to use Win32 shared memory Igor Sysoev <igor@sysoev.ru> parents: 2612 diff changeset	748 len = sizeof(" in limit_req zone \"\"") + shm_zone->shm.name.len;
2611 2bce3f6416c6 improve ngx_slab_alloc() error logging Igor Sysoev <igor@sysoev.ru> parents: 2376 diff changeset	749
2bce3f6416c6 improve ngx_slab_alloc() error logging Igor Sysoev <igor@sysoev.ru> parents: 2376 diff changeset	750 ctx->shpool->log_ctx = ngx_slab_alloc(ctx->shpool, len);
2bce3f6416c6 improve ngx_slab_alloc() error logging Igor Sysoev <igor@sysoev.ru> parents: 2376 diff changeset	751 if (ctx->shpool->log_ctx == NULL) {
2bce3f6416c6 improve ngx_slab_alloc() error logging Igor Sysoev <igor@sysoev.ru> parents: 2376 diff changeset	752 return NGX_ERROR;
2bce3f6416c6 improve ngx_slab_alloc() error logging Igor Sysoev <igor@sysoev.ru> parents: 2376 diff changeset	753 }
2bce3f6416c6 improve ngx_slab_alloc() error logging Igor Sysoev <igor@sysoev.ru> parents: 2376 diff changeset	754
2bce3f6416c6 improve ngx_slab_alloc() error logging Igor Sysoev <igor@sysoev.ru> parents: 2376 diff changeset	755 ngx_sprintf(ctx->shpool->log_ctx, " in limit_req zone \"%V\"%Z",
2716 d5896f6608e8 move zone name from ngx_shm_zone_t to ngx_shm_t to use Win32 shared memory Igor Sysoev <igor@sysoev.ru> parents: 2612 diff changeset	756 &shm_zone->shm.name);
2611 2bce3f6416c6 improve ngx_slab_alloc() error logging Igor Sysoev <igor@sysoev.ru> parents: 2376 diff changeset	757
5634 5024d29354f1 Core: slab log_nomem flag. Maxim Dounin <mdounin@mdounin.ru> parents: 5411 diff changeset	758 ctx->shpool->log_nomem = 0;
5024d29354f1 Core: slab log_nomem flag. Maxim Dounin <mdounin@mdounin.ru> parents: 5411 diff changeset	759
980 7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	760 return NGX_OK;
7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	761 }
7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	762
7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	763
7592 776d1bebdca2 Limit req: $limit_req_status variable. Roman Arutyunyan <arut@nginx.com> parents: 7515 diff changeset	764 static ngx_int_t
776d1bebdca2 Limit req: $limit_req_status variable. Roman Arutyunyan <arut@nginx.com> parents: 7515 diff changeset	765 ngx_http_limit_req_status_variable(ngx_http_request_t *r,
776d1bebdca2 Limit req: $limit_req_status variable. Roman Arutyunyan <arut@nginx.com> parents: 7515 diff changeset	766 ngx_http_variable_value_t *v, uintptr_t data)
776d1bebdca2 Limit req: $limit_req_status variable. Roman Arutyunyan <arut@nginx.com> parents: 7515 diff changeset	767 {
776d1bebdca2 Limit req: $limit_req_status variable. Roman Arutyunyan <arut@nginx.com> parents: 7515 diff changeset	768 if (r->main->limit_req_status == 0) {
776d1bebdca2 Limit req: $limit_req_status variable. Roman Arutyunyan <arut@nginx.com> parents: 7515 diff changeset	769 v->not_found = 1;
776d1bebdca2 Limit req: $limit_req_status variable. Roman Arutyunyan <arut@nginx.com> parents: 7515 diff changeset	770 return NGX_OK;
776d1bebdca2 Limit req: $limit_req_status variable. Roman Arutyunyan <arut@nginx.com> parents: 7515 diff changeset	771 }
776d1bebdca2 Limit req: $limit_req_status variable. Roman Arutyunyan <arut@nginx.com> parents: 7515 diff changeset	772
776d1bebdca2 Limit req: $limit_req_status variable. Roman Arutyunyan <arut@nginx.com> parents: 7515 diff changeset	773 v->valid = 1;
776d1bebdca2 Limit req: $limit_req_status variable. Roman Arutyunyan <arut@nginx.com> parents: 7515 diff changeset	774 v->no_cacheable = 0;
776d1bebdca2 Limit req: $limit_req_status variable. Roman Arutyunyan <arut@nginx.com> parents: 7515 diff changeset	775 v->not_found = 0;
776d1bebdca2 Limit req: $limit_req_status variable. Roman Arutyunyan <arut@nginx.com> parents: 7515 diff changeset	776 v->len = ngx_http_limit_req_status[r->main->limit_req_status - 1].len;
776d1bebdca2 Limit req: $limit_req_status variable. Roman Arutyunyan <arut@nginx.com> parents: 7515 diff changeset	777 v->data = ngx_http_limit_req_status[r->main->limit_req_status - 1].data;
776d1bebdca2 Limit req: $limit_req_status variable. Roman Arutyunyan <arut@nginx.com> parents: 7515 diff changeset	778
776d1bebdca2 Limit req: $limit_req_status variable. Roman Arutyunyan <arut@nginx.com> parents: 7515 diff changeset	779 return NGX_OK;
776d1bebdca2 Limit req: $limit_req_status variable. Roman Arutyunyan <arut@nginx.com> parents: 7515 diff changeset	780 }
776d1bebdca2 Limit req: $limit_req_status variable. Roman Arutyunyan <arut@nginx.com> parents: 7515 diff changeset	781
776d1bebdca2 Limit req: $limit_req_status variable. Roman Arutyunyan <arut@nginx.com> parents: 7515 diff changeset	782
980 7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	783 static void *
2294 ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	784 ngx_http_limit_req_create_conf(ngx_conf_t *cf)
980 7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	785 {
2294 ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	786 ngx_http_limit_req_conf_t *conf;
980 7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	787
2294 ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	788 conf = ngx_pcalloc(cf->pool, sizeof(ngx_http_limit_req_conf_t));
980 7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	789 if (conf == NULL) {
2912 c7d57b539248 return NULL instead of NGX_CONF_ERROR on a create conf failure Igor Sysoev <igor@sysoev.ru> parents: 2720 diff changeset	790 return NULL;
980 7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	791 }
7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	792
7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	793 /*
7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	794 * set by ngx_pcalloc():
7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	795 *
4420 9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	796 * conf->limits.elts = NULL;
980 7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	797 */
7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	798
3185 30570a5b9bb0 limit_req_log_level Igor Sysoev <igor@sysoev.ru> parents: 3183 diff changeset	799 conf->limit_log_level = NGX_CONF_UNSET_UINT;
5117 00e4459739ed The limit_req_status and limit_conn_status directives. Maxim Dounin <mdounin@mdounin.ru> parents: 4832 diff changeset	800 conf->status_code = NGX_CONF_UNSET_UINT;
7515 2db68852d6a0 Limit req: limit_req_dry_run directive. Roman Arutyunyan <arut@nginx.com> parents: 7399 diff changeset	801 conf->dry_run = NGX_CONF_UNSET;
3185 30570a5b9bb0 limit_req_log_level Igor Sysoev <igor@sysoev.ru> parents: 3183 diff changeset	802
980 7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	803 return conf;
7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	804 }
7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	805
7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	806
7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	807 static char *
2294 ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	808 ngx_http_limit_req_merge_conf(ngx_conf_t cf, void parent, void *child)
980 7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	809 {
2294 ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	810 ngx_http_limit_req_conf_t *prev = parent;
ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	811 ngx_http_limit_req_conf_t *conf = child;
980 7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	812
4420 9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	813 if (conf->limits.elts == NULL) {
9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	814 conf->limits = prev->limits;
980 7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	815 }
7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	816
3185 30570a5b9bb0 limit_req_log_level Igor Sysoev <igor@sysoev.ru> parents: 3183 diff changeset	817 ngx_conf_merge_uint_value(conf->limit_log_level, prev->limit_log_level,
30570a5b9bb0 limit_req_log_level Igor Sysoev <igor@sysoev.ru> parents: 3183 diff changeset	818 NGX_LOG_ERR);
30570a5b9bb0 limit_req_log_level Igor Sysoev <igor@sysoev.ru> parents: 3183 diff changeset	819
30570a5b9bb0 limit_req_log_level Igor Sysoev <igor@sysoev.ru> parents: 3183 diff changeset	820 conf->delay_log_level = (conf->limit_log_level == NGX_LOG_INFO) ?
30570a5b9bb0 limit_req_log_level Igor Sysoev <igor@sysoev.ru> parents: 3183 diff changeset	821 NGX_LOG_INFO : conf->limit_log_level + 1;
30570a5b9bb0 limit_req_log_level Igor Sysoev <igor@sysoev.ru> parents: 3183 diff changeset	822
5117 00e4459739ed The limit_req_status and limit_conn_status directives. Maxim Dounin <mdounin@mdounin.ru> parents: 4832 diff changeset	823 ngx_conf_merge_uint_value(conf->status_code, prev->status_code,
00e4459739ed The limit_req_status and limit_conn_status directives. Maxim Dounin <mdounin@mdounin.ru> parents: 4832 diff changeset	824 NGX_HTTP_SERVICE_UNAVAILABLE);
00e4459739ed The limit_req_status and limit_conn_status directives. Maxim Dounin <mdounin@mdounin.ru> parents: 4832 diff changeset	825
7515 2db68852d6a0 Limit req: limit_req_dry_run directive. Roman Arutyunyan <arut@nginx.com> parents: 7399 diff changeset	826 ngx_conf_merge_value(conf->dry_run, prev->dry_run, 0);
2db68852d6a0 Limit req: limit_req_dry_run directive. Roman Arutyunyan <arut@nginx.com> parents: 7399 diff changeset	827
980 7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	828 return NGX_CONF_OK;
7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	829 }
7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	830
7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	831
7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	832 static char *
2294 ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	833 ngx_http_limit_req_zone(ngx_conf_t cf, ngx_command_t cmd, void *conf)
980 7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	834 {
5862 ecbb99aa0e12 Limit req: use complex value in limit_req_zone. Valentin Bartenev <vbart@nginx.com> parents: 5846 diff changeset	835 u_char *p;
ecbb99aa0e12 Limit req: use complex value in limit_req_zone. Valentin Bartenev <vbart@nginx.com> parents: 5846 diff changeset	836 size_t len;
ecbb99aa0e12 Limit req: use complex value in limit_req_zone. Valentin Bartenev <vbart@nginx.com> parents: 5846 diff changeset	837 ssize_t size;
ecbb99aa0e12 Limit req: use complex value in limit_req_zone. Valentin Bartenev <vbart@nginx.com> parents: 5846 diff changeset	838 ngx_str_t *value, name, s;
ecbb99aa0e12 Limit req: use complex value in limit_req_zone. Valentin Bartenev <vbart@nginx.com> parents: 5846 diff changeset	839 ngx_int_t rate, scale;
ecbb99aa0e12 Limit req: use complex value in limit_req_zone. Valentin Bartenev <vbart@nginx.com> parents: 5846 diff changeset	840 ngx_uint_t i;
ecbb99aa0e12 Limit req: use complex value in limit_req_zone. Valentin Bartenev <vbart@nginx.com> parents: 5846 diff changeset	841 ngx_shm_zone_t *shm_zone;
ecbb99aa0e12 Limit req: use complex value in limit_req_zone. Valentin Bartenev <vbart@nginx.com> parents: 5846 diff changeset	842 ngx_http_limit_req_ctx_t *ctx;
ecbb99aa0e12 Limit req: use complex value in limit_req_zone. Valentin Bartenev <vbart@nginx.com> parents: 5846 diff changeset	843 ngx_http_compile_complex_value_t ccv;
980 7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	844
7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	845 value = cf->args->elts;
7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	846
5862 ecbb99aa0e12 Limit req: use complex value in limit_req_zone. Valentin Bartenev <vbart@nginx.com> parents: 5846 diff changeset	847 ctx = ngx_pcalloc(cf->pool, sizeof(ngx_http_limit_req_ctx_t));
ecbb99aa0e12 Limit req: use complex value in limit_req_zone. Valentin Bartenev <vbart@nginx.com> parents: 5846 diff changeset	848 if (ctx == NULL) {
ecbb99aa0e12 Limit req: use complex value in limit_req_zone. Valentin Bartenev <vbart@nginx.com> parents: 5846 diff changeset	849 return NGX_CONF_ERROR;
ecbb99aa0e12 Limit req: use complex value in limit_req_zone. Valentin Bartenev <vbart@nginx.com> parents: 5846 diff changeset	850 }
ecbb99aa0e12 Limit req: use complex value in limit_req_zone. Valentin Bartenev <vbart@nginx.com> parents: 5846 diff changeset	851
ecbb99aa0e12 Limit req: use complex value in limit_req_zone. Valentin Bartenev <vbart@nginx.com> parents: 5846 diff changeset	852 ngx_memzero(&ccv, sizeof(ngx_http_compile_complex_value_t));
ecbb99aa0e12 Limit req: use complex value in limit_req_zone. Valentin Bartenev <vbart@nginx.com> parents: 5846 diff changeset	853
ecbb99aa0e12 Limit req: use complex value in limit_req_zone. Valentin Bartenev <vbart@nginx.com> parents: 5846 diff changeset	854 ccv.cf = cf;
ecbb99aa0e12 Limit req: use complex value in limit_req_zone. Valentin Bartenev <vbart@nginx.com> parents: 5846 diff changeset	855 ccv.value = &value[1];
ecbb99aa0e12 Limit req: use complex value in limit_req_zone. Valentin Bartenev <vbart@nginx.com> parents: 5846 diff changeset	856 ccv.complex_value = &ctx->key;
ecbb99aa0e12 Limit req: use complex value in limit_req_zone. Valentin Bartenev <vbart@nginx.com> parents: 5846 diff changeset	857
ecbb99aa0e12 Limit req: use complex value in limit_req_zone. Valentin Bartenev <vbart@nginx.com> parents: 5846 diff changeset	858 if (ngx_http_compile_complex_value(&ccv) != NGX_OK) {
ecbb99aa0e12 Limit req: use complex value in limit_req_zone. Valentin Bartenev <vbart@nginx.com> parents: 5846 diff changeset	859 return NGX_CONF_ERROR;
ecbb99aa0e12 Limit req: use complex value in limit_req_zone. Valentin Bartenev <vbart@nginx.com> parents: 5846 diff changeset	860 }
ecbb99aa0e12 Limit req: use complex value in limit_req_zone. Valentin Bartenev <vbart@nginx.com> parents: 5846 diff changeset	861
2294 ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	862 size = 0;
ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	863 rate = 1;
ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	864 scale = 1;
ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	865 name.len = 0;
ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	866
5862 ecbb99aa0e12 Limit req: use complex value in limit_req_zone. Valentin Bartenev <vbart@nginx.com> parents: 5846 diff changeset	867 for (i = 2; i < cf->args->nelts; i++) {
2294 ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	868
ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	869 if (ngx_strncmp(value[i].data, "zone=", 5) == 0) {
ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	870
ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	871 name.data = value[i].data + 5;
ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	872
ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	873 p = (u_char *) ngx_strchr(name.data, ':');
ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	874
4417 9e9d2e06f933 Limit req: improved error handling when parsing "zone" parameter of Valentin Bartenev <vbart@nginx.com> parents: 4416 diff changeset	875 if (p == NULL) {
9e9d2e06f933 Limit req: improved error handling when parsing "zone" parameter of Valentin Bartenev <vbart@nginx.com> parents: 4416 diff changeset	876 ngx_conf_log_error(NGX_LOG_EMERG, cf, 0,
9e9d2e06f933 Limit req: improved error handling when parsing "zone" parameter of Valentin Bartenev <vbart@nginx.com> parents: 4416 diff changeset	877 "invalid zone size \"%V\"", &value[i]);
9e9d2e06f933 Limit req: improved error handling when parsing "zone" parameter of Valentin Bartenev <vbart@nginx.com> parents: 4416 diff changeset	878 return NGX_CONF_ERROR;
2294 ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	879 }
ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	880
4417 9e9d2e06f933 Limit req: improved error handling when parsing "zone" parameter of Valentin Bartenev <vbart@nginx.com> parents: 4416 diff changeset	881 name.len = p - name.data;
9e9d2e06f933 Limit req: improved error handling when parsing "zone" parameter of Valentin Bartenev <vbart@nginx.com> parents: 4416 diff changeset	882
9e9d2e06f933 Limit req: improved error handling when parsing "zone" parameter of Valentin Bartenev <vbart@nginx.com> parents: 4416 diff changeset	883 s.data = p + 1;
9e9d2e06f933 Limit req: improved error handling when parsing "zone" parameter of Valentin Bartenev <vbart@nginx.com> parents: 4416 diff changeset	884 s.len = value[i].data + value[i].len - s.data;
9e9d2e06f933 Limit req: improved error handling when parsing "zone" parameter of Valentin Bartenev <vbart@nginx.com> parents: 4416 diff changeset	885
9e9d2e06f933 Limit req: improved error handling when parsing "zone" parameter of Valentin Bartenev <vbart@nginx.com> parents: 4416 diff changeset	886 size = ngx_parse_size(&s);
9e9d2e06f933 Limit req: improved error handling when parsing "zone" parameter of Valentin Bartenev <vbart@nginx.com> parents: 4416 diff changeset	887
9e9d2e06f933 Limit req: improved error handling when parsing "zone" parameter of Valentin Bartenev <vbart@nginx.com> parents: 4416 diff changeset	888 if (size == NGX_ERROR) {
9e9d2e06f933 Limit req: improved error handling when parsing "zone" parameter of Valentin Bartenev <vbart@nginx.com> parents: 4416 diff changeset	889 ngx_conf_log_error(NGX_LOG_EMERG, cf, 0,
9e9d2e06f933 Limit req: improved error handling when parsing "zone" parameter of Valentin Bartenev <vbart@nginx.com> parents: 4416 diff changeset	890 "invalid zone size \"%V\"", &value[i]);
9e9d2e06f933 Limit req: improved error handling when parsing "zone" parameter of Valentin Bartenev <vbart@nginx.com> parents: 4416 diff changeset	891 return NGX_CONF_ERROR;
9e9d2e06f933 Limit req: improved error handling when parsing "zone" parameter of Valentin Bartenev <vbart@nginx.com> parents: 4416 diff changeset	892 }
9e9d2e06f933 Limit req: improved error handling when parsing "zone" parameter of Valentin Bartenev <vbart@nginx.com> parents: 4416 diff changeset	893
9e9d2e06f933 Limit req: improved error handling when parsing "zone" parameter of Valentin Bartenev <vbart@nginx.com> parents: 4416 diff changeset	894 if (size < (ssize_t) (8 * ngx_pagesize)) {
9e9d2e06f933 Limit req: improved error handling when parsing "zone" parameter of Valentin Bartenev <vbart@nginx.com> parents: 4416 diff changeset	895 ngx_conf_log_error(NGX_LOG_EMERG, cf, 0,
9e9d2e06f933 Limit req: improved error handling when parsing "zone" parameter of Valentin Bartenev <vbart@nginx.com> parents: 4416 diff changeset	896 "zone \"%V\" is too small", &value[i]);
9e9d2e06f933 Limit req: improved error handling when parsing "zone" parameter of Valentin Bartenev <vbart@nginx.com> parents: 4416 diff changeset	897 return NGX_CONF_ERROR;
9e9d2e06f933 Limit req: improved error handling when parsing "zone" parameter of Valentin Bartenev <vbart@nginx.com> parents: 4416 diff changeset	898 }
9e9d2e06f933 Limit req: improved error handling when parsing "zone" parameter of Valentin Bartenev <vbart@nginx.com> parents: 4416 diff changeset	899
9e9d2e06f933 Limit req: improved error handling when parsing "zone" parameter of Valentin Bartenev <vbart@nginx.com> parents: 4416 diff changeset	900 continue;
2294 ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	901 }
ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	902
ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	903 if (ngx_strncmp(value[i].data, "rate=", 5) == 0) {
ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	904
ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	905 len = value[i].len;
ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	906 p = value[i].data + len - 3;
987 14e68f471d02 new syntax Igor Sysoev <igor@sysoev.ru> parents: 984 diff changeset	907
2294 ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	908 if (ngx_strncmp(p, "r/s", 3) == 0) {
ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	909 scale = 1;
ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	910 len -= 3;
ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	911
ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	912 } else if (ngx_strncmp(p, "r/m", 3) == 0) {
ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	913 scale = 60;
ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	914 len -= 3;
ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	915 }
ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	916
ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	917 rate = ngx_atoi(value[i].data + 5, len - 5);
5166 fc595eeb6c54 Limit req: rate should be non-zero. Valentin Bartenev <vbart@nginx.com> parents: 5117 diff changeset	918 if (rate <= 0) {
2294 ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	919 ngx_conf_log_error(NGX_LOG_EMERG, cf, 0,
ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	920 "invalid rate \"%V\"", &value[i]);
ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	921 return NGX_CONF_ERROR;
ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	922 }
ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	923
ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	924 continue;
ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	925 }
ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	926
ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	927 ngx_conf_log_error(NGX_LOG_EMERG, cf, 0,
ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	928 "invalid parameter \"%V\"", &value[i]);
987 14e68f471d02 new syntax Igor Sysoev <igor@sysoev.ru> parents: 984 diff changeset	929 return NGX_CONF_ERROR;
14e68f471d02 new syntax Igor Sysoev <igor@sysoev.ru> parents: 984 diff changeset	930 }
14e68f471d02 new syntax Igor Sysoev <igor@sysoev.ru> parents: 984 diff changeset	931
4417 9e9d2e06f933 Limit req: improved error handling when parsing "zone" parameter of Valentin Bartenev <vbart@nginx.com> parents: 4416 diff changeset	932 if (name.len == 0) {
2294 ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	933 ngx_conf_log_error(NGX_LOG_EMERG, cf, 0,
ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	934 "\"%V\" must have \"zone\" parameter",
ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	935 &cmd->name);
987 14e68f471d02 new syntax Igor Sysoev <igor@sysoev.ru> parents: 984 diff changeset	936 return NGX_CONF_ERROR;
14e68f471d02 new syntax Igor Sysoev <igor@sysoev.ru> parents: 984 diff changeset	937 }
14e68f471d02 new syntax Igor Sysoev <igor@sysoev.ru> parents: 984 diff changeset	938
2313 b56a9ba4824d use integer instead of float Igor Sysoev <igor@sysoev.ru> parents: 2312 diff changeset	939 ctx->rate = rate * 1000 / scale;
980 7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	940
2294 ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	941 shm_zone = ngx_shared_memory_add(cf, &name, size,
ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	942 &ngx_http_limit_req_module);
980 7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	943 if (shm_zone == NULL) {
7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	944 return NGX_CONF_ERROR;
7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	945 }
7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	946
987 14e68f471d02 new syntax Igor Sysoev <igor@sysoev.ru> parents: 984 diff changeset	947 if (shm_zone->data) {
14e68f471d02 new syntax Igor Sysoev <igor@sysoev.ru> parents: 984 diff changeset	948 ctx = shm_zone->data;
14e68f471d02 new syntax Igor Sysoev <igor@sysoev.ru> parents: 984 diff changeset	949
14e68f471d02 new syntax Igor Sysoev <igor@sysoev.ru> parents: 984 diff changeset	950 ngx_conf_log_error(NGX_LOG_EMERG, cf, 0,
5862 ecbb99aa0e12 Limit req: use complex value in limit_req_zone. Valentin Bartenev <vbart@nginx.com> parents: 5846 diff changeset	951 "%V \"%V\" is already bound to key \"%V\"",
ecbb99aa0e12 Limit req: use complex value in limit_req_zone. Valentin Bartenev <vbart@nginx.com> parents: 5846 diff changeset	952 &cmd->name, &name, &ctx->key.value);
987 14e68f471d02 new syntax Igor Sysoev <igor@sysoev.ru> parents: 984 diff changeset	953 return NGX_CONF_ERROR;
14e68f471d02 new syntax Igor Sysoev <igor@sysoev.ru> parents: 984 diff changeset	954 }
14e68f471d02 new syntax Igor Sysoev <igor@sysoev.ru> parents: 984 diff changeset	955
2294 ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	956 shm_zone->init = ngx_http_limit_req_init_zone;
987 14e68f471d02 new syntax Igor Sysoev <igor@sysoev.ru> parents: 984 diff changeset	957 shm_zone->data = ctx;
980 7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	958
7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	959 return NGX_CONF_OK;
7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	960 }
7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	961
7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	962
7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	963 static char *
2294 ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	964 ngx_http_limit_req(ngx_conf_t cf, ngx_command_t cmd, void *conf)
980 7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	965 {
2374 7b11f9a1bfe1 rename "lz" to "lr" in variable names Igor Sysoev <igor@sysoev.ru> parents: 2313 diff changeset	966 ngx_http_limit_req_conf_t *lrcf = conf;
980 7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	967
7399 d6ca744c727e Limit req: "delay=" parameter. Maxim Dounin <mdounin@mdounin.ru> parents: 7398 diff changeset	968 ngx_int_t burst, delay;
4420 9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	969 ngx_str_t *value, s;
7399 d6ca744c727e Limit req: "delay=" parameter. Maxim Dounin <mdounin@mdounin.ru> parents: 7398 diff changeset	970 ngx_uint_t i;
4420 9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	971 ngx_shm_zone_t *shm_zone;
9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	972 ngx_http_limit_req_limit_t limit, limits;
980 7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	973
7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	974 value = cf->args->elts;
7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	975
4420 9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	976 shm_zone = NULL;
2294 ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	977 burst = 0;
7399 d6ca744c727e Limit req: "delay=" parameter. Maxim Dounin <mdounin@mdounin.ru> parents: 7398 diff changeset	978 delay = 0;
2294 ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	979
ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	980 for (i = 1; i < cf->args->nelts; i++) {
ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	981
ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	982 if (ngx_strncmp(value[i].data, "zone=", 5) == 0) {
ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	983
ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	984 s.len = value[i].len - 5;
ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	985 s.data = value[i].data + 5;
ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	986
4420 9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	987 shm_zone = ngx_shared_memory_add(cf, &s, 0,
9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	988 &ngx_http_limit_req_module);
9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	989 if (shm_zone == NULL) {
2294 ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	990 return NGX_CONF_ERROR;
ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	991 }
ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	992
ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	993 continue;
ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	994 }
ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	995
ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	996 if (ngx_strncmp(value[i].data, "burst=", 6) == 0) {
ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	997
2300 159136c9808d ) correct leaky bucket implementation Igor Sysoev <igor@sysoev.ru>* parents: 2294 diff changeset	998 burst = ngx_atoi(value[i].data + 6, value[i].len - 6);
2294 ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	999 if (burst <= 0) {
ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	1000 ngx_conf_log_error(NGX_LOG_EMERG, cf, 0,
7398 bca4dad0d3cb Limit req: fixed error message wording. Maxim Dounin <mdounin@mdounin.ru> parents: 7281 diff changeset	1001 "invalid burst value \"%V\"", &value[i]);
2294 ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	1002 return NGX_CONF_ERROR;
ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	1003 }
ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	1004
ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	1005 continue;
ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	1006 }
ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	1007
7399 d6ca744c727e Limit req: "delay=" parameter. Maxim Dounin <mdounin@mdounin.ru> parents: 7398 diff changeset	1008 if (ngx_strncmp(value[i].data, "delay=", 6) == 0) {
d6ca744c727e Limit req: "delay=" parameter. Maxim Dounin <mdounin@mdounin.ru> parents: 7398 diff changeset	1009
d6ca744c727e Limit req: "delay=" parameter. Maxim Dounin <mdounin@mdounin.ru> parents: 7398 diff changeset	1010 delay = ngx_atoi(value[i].data + 6, value[i].len - 6);
d6ca744c727e Limit req: "delay=" parameter. Maxim Dounin <mdounin@mdounin.ru> parents: 7398 diff changeset	1011 if (delay <= 0) {
d6ca744c727e Limit req: "delay=" parameter. Maxim Dounin <mdounin@mdounin.ru> parents: 7398 diff changeset	1012 ngx_conf_log_error(NGX_LOG_EMERG, cf, 0,
d6ca744c727e Limit req: "delay=" parameter. Maxim Dounin <mdounin@mdounin.ru> parents: 7398 diff changeset	1013 "invalid delay value \"%V\"", &value[i]);
d6ca744c727e Limit req: "delay=" parameter. Maxim Dounin <mdounin@mdounin.ru> parents: 7398 diff changeset	1014 return NGX_CONF_ERROR;
d6ca744c727e Limit req: "delay=" parameter. Maxim Dounin <mdounin@mdounin.ru> parents: 7398 diff changeset	1015 }
d6ca744c727e Limit req: "delay=" parameter. Maxim Dounin <mdounin@mdounin.ru> parents: 7398 diff changeset	1016
d6ca744c727e Limit req: "delay=" parameter. Maxim Dounin <mdounin@mdounin.ru> parents: 7398 diff changeset	1017 continue;
d6ca744c727e Limit req: "delay=" parameter. Maxim Dounin <mdounin@mdounin.ru> parents: 7398 diff changeset	1018 }
d6ca744c727e Limit req: "delay=" parameter. Maxim Dounin <mdounin@mdounin.ru> parents: 7398 diff changeset	1019
5411 5483d9e77b32 Limit req: fixed "nodelay" parsing. Maxim Dounin <mdounin@mdounin.ru> parents: 5166 diff changeset	1020 if (ngx_strcmp(value[i].data, "nodelay") == 0) {
7399 d6ca744c727e Limit req: "delay=" parameter. Maxim Dounin <mdounin@mdounin.ru> parents: 7398 diff changeset	1021 delay = NGX_MAX_INT_T_VALUE / 1000;
2294 ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	1022 continue;
ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	1023 }
ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	1024
ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	1025 ngx_conf_log_error(NGX_LOG_EMERG, cf, 0,
ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	1026 "invalid parameter \"%V\"", &value[i]);
980 7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	1027 return NGX_CONF_ERROR;
7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	1028 }
7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	1029
4420 9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	1030 if (shm_zone == NULL) {
980 7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	1031 ngx_conf_log_error(NGX_LOG_EMERG, cf, 0,
2294 ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	1032 "\"%V\" must have \"zone\" parameter",
ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	1033 &cmd->name);
980 7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	1034 return NGX_CONF_ERROR;
7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	1035 }
7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	1036
4420 9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	1037 limits = lrcf->limits.elts;
9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	1038
9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	1039 if (limits == NULL) {
9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	1040 if (ngx_array_init(&lrcf->limits, cf->pool, 1,
9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	1041 sizeof(ngx_http_limit_req_limit_t))
9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	1042 != NGX_OK)
9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	1043 {
9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	1044 return NGX_CONF_ERROR;
9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	1045 }
9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	1046 }
9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	1047
9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	1048 for (i = 0; i < lrcf->limits.nelts; i++) {
9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	1049 if (shm_zone == limits[i].shm_zone) {
9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	1050 return "is duplicate";
9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	1051 }
9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	1052 }
9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	1053
9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	1054 limit = ngx_array_push(&lrcf->limits);
4811 21d1e3bcb356 Added three missing checks for NULL after ngx_array_push() calls. Valentin Bartenev <vbart@nginx.com> parents: 4499 diff changeset	1055 if (limit == NULL) {
21d1e3bcb356 Added three missing checks for NULL after ngx_array_push() calls. Valentin Bartenev <vbart@nginx.com> parents: 4499 diff changeset	1056 return NGX_CONF_ERROR;
21d1e3bcb356 Added three missing checks for NULL after ngx_array_push() calls. Valentin Bartenev <vbart@nginx.com> parents: 4499 diff changeset	1057 }
4420 9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	1058
9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	1059 limit->shm_zone = shm_zone;
9ce48f9eb85b Limit req: support for multiple "limit_req" limits. Valentin Bartenev <vbart@nginx.com> parents: 4419 diff changeset	1060 limit->burst = burst * 1000;
7399 d6ca744c727e Limit req: "delay=" parameter. Maxim Dounin <mdounin@mdounin.ru> parents: 7398 diff changeset	1061 limit->delay = delay * 1000;
980 7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	1062
7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	1063 return NGX_CONF_OK;
7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	1064 }
7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	1065
7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	1066
7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	1067 static ngx_int_t
7592 776d1bebdca2 Limit req: $limit_req_status variable. Roman Arutyunyan <arut@nginx.com> parents: 7515 diff changeset	1068 ngx_http_limit_req_add_variables(ngx_conf_t *cf)
776d1bebdca2 Limit req: $limit_req_status variable. Roman Arutyunyan <arut@nginx.com> parents: 7515 diff changeset	1069 {
776d1bebdca2 Limit req: $limit_req_status variable. Roman Arutyunyan <arut@nginx.com> parents: 7515 diff changeset	1070 ngx_http_variable_t var, v;
776d1bebdca2 Limit req: $limit_req_status variable. Roman Arutyunyan <arut@nginx.com> parents: 7515 diff changeset	1071
776d1bebdca2 Limit req: $limit_req_status variable. Roman Arutyunyan <arut@nginx.com> parents: 7515 diff changeset	1072 for (v = ngx_http_limit_req_vars; v->name.len; v++) {
776d1bebdca2 Limit req: $limit_req_status variable. Roman Arutyunyan <arut@nginx.com> parents: 7515 diff changeset	1073 var = ngx_http_add_variable(cf, &v->name, v->flags);
776d1bebdca2 Limit req: $limit_req_status variable. Roman Arutyunyan <arut@nginx.com> parents: 7515 diff changeset	1074 if (var == NULL) {
776d1bebdca2 Limit req: $limit_req_status variable. Roman Arutyunyan <arut@nginx.com> parents: 7515 diff changeset	1075 return NGX_ERROR;
776d1bebdca2 Limit req: $limit_req_status variable. Roman Arutyunyan <arut@nginx.com> parents: 7515 diff changeset	1076 }
776d1bebdca2 Limit req: $limit_req_status variable. Roman Arutyunyan <arut@nginx.com> parents: 7515 diff changeset	1077
776d1bebdca2 Limit req: $limit_req_status variable. Roman Arutyunyan <arut@nginx.com> parents: 7515 diff changeset	1078 var->get_handler = v->get_handler;
776d1bebdca2 Limit req: $limit_req_status variable. Roman Arutyunyan <arut@nginx.com> parents: 7515 diff changeset	1079 var->data = v->data;
776d1bebdca2 Limit req: $limit_req_status variable. Roman Arutyunyan <arut@nginx.com> parents: 7515 diff changeset	1080 }
776d1bebdca2 Limit req: $limit_req_status variable. Roman Arutyunyan <arut@nginx.com> parents: 7515 diff changeset	1081
776d1bebdca2 Limit req: $limit_req_status variable. Roman Arutyunyan <arut@nginx.com> parents: 7515 diff changeset	1082 return NGX_OK;
776d1bebdca2 Limit req: $limit_req_status variable. Roman Arutyunyan <arut@nginx.com> parents: 7515 diff changeset	1083 }
776d1bebdca2 Limit req: $limit_req_status variable. Roman Arutyunyan <arut@nginx.com> parents: 7515 diff changeset	1084
776d1bebdca2 Limit req: $limit_req_status variable. Roman Arutyunyan <arut@nginx.com> parents: 7515 diff changeset	1085
776d1bebdca2 Limit req: $limit_req_status variable. Roman Arutyunyan <arut@nginx.com> parents: 7515 diff changeset	1086 static ngx_int_t
2294 ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	1087 ngx_http_limit_req_init(ngx_conf_t *cf)
980 7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	1088 {
7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	1089 ngx_http_handler_pt *h;
7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	1090 ngx_http_core_main_conf_t *cmcf;
7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	1091
7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	1092 cmcf = ngx_http_conf_get_module_main_conf(cf, ngx_http_core_module);
7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	1093
7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	1094 h = ngx_array_push(&cmcf->phases[NGX_HTTP_PREACCESS_PHASE].handlers);
7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	1095 if (h == NULL) {
7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	1096 return NGX_ERROR;
7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	1097 }
7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	1098
2294 ce574b0ffeec ngx_http_limit_req_module Igor Sysoev <igor@sysoev.ru> parents: 1743 diff changeset	1099 *h = ngx_http_limit_req_handler;
980 7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	1100
7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	1101 return NGX_OK;
7cb910b4a58a ngx_http_limit_zone_module Igor Sysoev <igor@sysoev.ru> parents: diff changeset	1102 }

Mercurial > hg > nginx

annotate src/http/modules/ngx_http_limit_req_module.c @ 8039:e64a1f32065b