nginx/src/http/modules/ngx_http_limit_req_module.c

/*
 * Copyright (C) Igor Sysoev
 * Copyright (C) Nginx, Inc.
 */


#include <ngx_config.h>
#include <ngx_core.h>
#include <ngx_http.h>


#define NGX_HTTP_LIMIT_REQ_PASSED …
#define NGX_HTTP_LIMIT_REQ_DELAYED …
#define NGX_HTTP_LIMIT_REQ_REJECTED …
#define NGX_HTTP_LIMIT_REQ_DELAYED_DRY_RUN …
#define NGX_HTTP_LIMIT_REQ_REJECTED_DRY_RUN …


ngx_http_limit_req_node_t;


ngx_http_limit_req_shctx_t;


ngx_http_limit_req_ctx_t;


ngx_http_limit_req_limit_t;


ngx_http_limit_req_conf_t;


static void ngx_http_limit_req_delay(ngx_http_request_t *r);
static ngx_int_t ngx_http_limit_req_lookup(ngx_http_limit_req_limit_t *limit,
    ngx_uint_t hash, ngx_str_t *key, ngx_uint_t *ep, ngx_uint_t account);
static ngx_msec_t ngx_http_limit_req_account(ngx_http_limit_req_limit_t *limits,
    ngx_uint_t n, ngx_uint_t *ep, ngx_http_limit_req_limit_t **limit);
static void ngx_http_limit_req_unlock(ngx_http_limit_req_limit_t *limits,
    ngx_uint_t n);
static void ngx_http_limit_req_expire(ngx_http_limit_req_ctx_t *ctx,
    ngx_uint_t n);

static ngx_int_t ngx_http_limit_req_status_variable(ngx_http_request_t *r,
    ngx_http_variable_value_t *v, uintptr_t data);
static void *ngx_http_limit_req_create_conf(ngx_conf_t *cf);
static char *ngx_http_limit_req_merge_conf(ngx_conf_t *cf, void *parent,
    void *child);
static char *ngx_http_limit_req_zone(ngx_conf_t *cf, ngx_command_t *cmd,
    void *conf);
static char *ngx_http_limit_req(ngx_conf_t *cf, ngx_command_t *cmd,
    void *conf);
static ngx_int_t ngx_http_limit_req_add_variables(ngx_conf_t *cf);
static ngx_int_t ngx_http_limit_req_init(ngx_conf_t *cf);


static ngx_conf_enum_t  ngx_http_limit_req_log_levels[] = …;


static ngx_conf_num_bounds_t  ngx_http_limit_req_status_bounds = …;


static ngx_command_t  ngx_http_limit_req_commands[] = …;


static ngx_http_module_t  ngx_http_limit_req_module_ctx = …;


ngx_module_t  ngx_http_limit_req_module = …;


static ngx_http_variable_t  ngx_http_limit_req_vars[] = …;


static ngx_str_t  ngx_http_limit_req_status[] = …;


static ngx_int_t
ngx_http_limit_req_handler(ngx_http_request_t *r)
{ … }


static void
ngx_http_limit_req_delay(ngx_http_request_t *r)
{ … }


static void
ngx_http_limit_req_rbtree_insert_value(ngx_rbtree_node_t *temp,
    ngx_rbtree_node_t *node, ngx_rbtree_node_t *sentinel)
{ … }


static ngx_int_t
ngx_http_limit_req_lookup(ngx_http_limit_req_limit_t *limit, ngx_uint_t hash,
    ngx_str_t *key, ngx_uint_t *ep, ngx_uint_t account)
{ … }


static ngx_msec_t
ngx_http_limit_req_account(ngx_http_limit_req_limit_t *limits, ngx_uint_t n,
    ngx_uint_t *ep, ngx_http_limit_req_limit_t **limit)
{ … }


static void
ngx_http_limit_req_unlock(ngx_http_limit_req_limit_t *limits, ngx_uint_t n)
{ … }


static void
ngx_http_limit_req_expire(ngx_http_limit_req_ctx_t *ctx, ngx_uint_t n)
{ … }


static ngx_int_t
ngx_http_limit_req_init_zone(ngx_shm_zone_t *shm_zone, void *data)
{ … }


static ngx_int_t
ngx_http_limit_req_status_variable(ngx_http_request_t *r,
    ngx_http_variable_value_t *v, uintptr_t data)
{ … }


static void *
ngx_http_limit_req_create_conf(ngx_conf_t *cf)
{ … }


static char *
ngx_http_limit_req_merge_conf(ngx_conf_t *cf, void *parent, void *child)
{ … }


static char *
ngx_http_limit_req_zone(ngx_conf_t *cf, ngx_command_t *cmd, void *conf)
{ … }


static char *
ngx_http_limit_req(ngx_conf_t *cf, ngx_command_t *cmd, void *conf)
{ … }


static ngx_int_t
ngx_http_limit_req_add_variables(ngx_conf_t *cf)
{ … }


static ngx_int_t
ngx_http_limit_req_init(ngx_conf_t *cf)
{ … }