ratelimit

README

caddy-rate-limit

a rate limit plugin for caddy

Syntax

Excessive requests will be terminated with an error 429 (Too Many Requests)! And X-RateLimit-RetryAfter header will be returned.

For single resource:

ratelimit methods path rate burst unit

• methods are the request methods it will match (comma separately)

• path is the file or directory to apply rate limit

• rate is the limited request in every time unit (r/s, r/m, r/h, r/d, r/w) (e.g. 1)

• burst is the maximum burst size client can exceed; burst >= rate (e.g. 2)

• unit is the time interval (currently support: second, minute, hour, day, week)

For multiple resources:

ratelimit methods rate burst unit {
    whitelist CIDR,CIDR
    limit_by_header xxx
    status xxx,xxx
    resources
}

• whitelist is the keyword for whitelist your trusted ips (comma separately). CIDR is the IP range you don't want to perform rate limit. whitelist is a general rule, it won't target for specific resource.
• limit_by_header is the keyword for matching the request header. Like whitelist, it's also a general rule. Note: normally you shouldn't apply this rule unless the default limit by ip is not what you want and you want to limit by request header(e.g. Authorization).
• status is the keyword for matching the response status code (comma separately). If this rule is triggered, all subsequent requests from that client will be blocked regardless of which status code is returned or which resource is requested. Note: this won't block resources not defined in ratelimit's config.
• resources is a list of files/directories to apply rate limit, one per line

Note: If you don't want to apply rate limit on some special resources, add ^ in front of the path.

Examples

Limit clients to 2 requests per second (bursts of 3) to any methods and any resources under /r:

ratelimit * /r 2 3 second

Don't perform rate limit if requests come from 1.2.3.4 or 192.168.1.0/30(192.168.1.0 ~ 192.168.1.3), for the listed paths, limit clients to 2 requests per minute (bursts of 2) if the request method is GET or POST and always ignore /dist/app.js:

ratelimit get,post 2 2 minute {
    whitelist 1.2.3.4/32,192.168.1.0/30
    status *
    /foo.html
    /api
    ^/dist/app.js
}

Download

curl https://getcaddy.com | bash -s personal http.ratelimit

Docker

docker run -d -p 2016:2016 -v `pwd`/Caddyfile:/go/src/github.com/xuqingfeng/caddy-rate-limit/Caddyfile --name ratelimit xuqingfeng/caddy-rate-limit

---

Inspired by

http://nginx.org/en/docs/http/ngx_http_limit_req_module.html

https://github.com/didip/tollbooth

Documentation

Index

• Constants
• func GetRemoteIP(r *http.Request) (string, error)
• func IsWhitelistIPAddress(address string, localIPNets []*net.IPNet) bool
• func MatchMethod(methods, method string) bool
• func MatchStatus(status, s string) bool
• func ParseHTTPTime(value string) (time.Time, error)
• func SetRetryAfterHeader(header http.Header, retryAfter time.Duration)
• type CaddyLimiter
  • func NewCaddyLimiter() *CaddyLimiter
  • func (cl *CaddyLimiter) Allow(keys []string, rule Rule, limiter ...*rate.Limiter) bool
  • func (cl *CaddyLimiter) AllowN(keys []string, rule Rule, n int, limiter ...*rate.Limiter) bool
  • func (cl *CaddyLimiter) GetLimiter(key string) *rate.Limiter
  • func (cl *CaddyLimiter) GetLimiterOk(key string) (*rate.Limiter, bool)
  • func (cl *CaddyLimiter) HasLimiter(key string) bool
  • func (cl *CaddyLimiter) Reserve(keys []string) bool
  • func (cl *CaddyLimiter) RetryAfter(keys []string, limiter ...*rate.Limiter) time.Duration
  • func (cl *CaddyLimiter) SetLimiter(key string, limiter *rate.Limiter)
• type RateLimit
  • func (rl RateLimit) ServeHTTP(w http.ResponseWriter, r *http.Request) (nextResponseStatus int, err error)
• type Rule

Constants

const HTTPTimeLayout = "Mon, 02 Jan 2006 15:04:05 GMT"

Functions

func GetRemoteIP

func GetRemoteIP(r *http.Request) (string, error)

GetRemoteIP returns the ip of requester Doesn't care if the ip is real or not

func IsWhitelistIPAddress

func IsWhitelistIPAddress(address string, localIPNets []*net.IPNet) bool

IsWhitelistIPAddress check whether an ip is in whitelist

func MatchMethod

func MatchMethod(methods, method string) bool

MatchMethod check whether the request method is in the methods list

func MatchStatus

func MatchStatus(status, s string) bool

MatchStatus check whether the upstream response status code is in the status list

func ParseHTTPTime

func ParseHTTPTime(value string) (time.Time, error)

func SetRetryAfterHeader

func SetRetryAfterHeader(header http.Header, retryAfter time.Duration)

Types

type CaddyLimiter

type CaddyLimiter struct {
	sync.RWMutex
	// contains filtered or unexported fields
}

func NewCaddyLimiter

func NewCaddyLimiter() *CaddyLimiter

func (*CaddyLimiter) Allow

func (cl *CaddyLimiter) Allow(keys []string, rule Rule, limiter ...*rate.Limiter) bool

Allow is just a shortcut for AllowN

func (*CaddyLimiter) AllowN

func (cl *CaddyLimiter) AllowN(keys []string, rule Rule, n int, limiter ...*rate.Limiter) bool

AllowN check if n count are allowed for a specific key

func (*CaddyLimiter) GetLimiter

func (cl *CaddyLimiter) GetLimiter(key string) *rate.Limiter

func (*CaddyLimiter) GetLimiterOk

func (cl *CaddyLimiter) GetLimiterOk(key string) (*rate.Limiter, bool)

func (*CaddyLimiter) HasLimiter

func (cl *CaddyLimiter) HasLimiter(key string) bool

func (*CaddyLimiter) Reserve

func (cl *CaddyLimiter) Reserve(keys []string) bool

Reserve will consume 1 token from `token bucket`

func (*CaddyLimiter) RetryAfter

func (cl *CaddyLimiter) RetryAfter(keys []string, limiter ...*rate.Limiter) time.Duration

RetryAfter return a helper message for client

func (*CaddyLimiter) SetLimiter

func (cl *CaddyLimiter) SetLimiter(key string, limiter *rate.Limiter)

type RateLimit

type RateLimit struct {
	Next  httpserver.Handler
	Rules []Rule
}

RateLimit is an http.Handler that can limit request rate to specific paths or files

func (RateLimit) ServeHTTP

func (rl RateLimit) ServeHTTP(w http.ResponseWriter, r *http.Request) (nextResponseStatus int, err error)

ServeHTTP is the method handling every request

type Rule

type Rule struct {
	Methods       string
	Rate          int64
	Burst         int
	Unit          string
	Whitelist     []string
	LimitByHeader string
	Status        string
	Resources     []string
	// contains filtered or unexported fields
}

Rule is a configuration for ratelimit