clawbot
f7e3190654
check / check (push) Successful in 5m58s
The command is now `sww waf`, and the package is `internal/waf`. Types, functions, files, tests, comments, docs, the Dockerfile, the runit script, `script/`, compose examples, the log field `waf_line` and the `X-From-Waf` test header follow. No behaviour changes, and no setting changes name: rproxy already finds the waf through `SWRP_WAF_BACKEND`. Done with one `sed -i` (Discriminator and DISCRIMINATOR to WAF, discriminator to waf) over the files `git grep -i discriminator` listed, as sneak allowed for this rename, then `git mv` for the package and files. By hand afterwards: the header's spelling, which the linter requires, and the padding after `waf` in `sww`'s usage text. Model: opus-5-5
42 lines
1.4 KiB
Go
42 lines
1.4 KiB
Go
package waf
|
|
|
|
// The heuristic byte_limits. It reads SWW_BYTES_COUNT, and the byte limits
|
|
// and biased thresholds through the limiter and limitPercentages.
|
|
|
|
import (
|
|
"context"
|
|
"time"
|
|
)
|
|
|
|
// byteLimits counts the request's bytes, as countedBytes gives them, for the
|
|
// byte limits, once its report has come, and notes the client's byte totals for
|
|
// the log line; its requests stay there as the rate limits counted them. Only
|
|
// an allowed request has them counted, and only one the rate limits counted; in
|
|
// observe mode, not one that enforce mode would have refused. Bytes that take
|
|
// the client over a byte limit, as its limit percentage for the byte limits
|
|
// lowers it, break it, which bans the client; the response was passed on whole.
|
|
// Its figures are the window, the count and the limit.
|
|
type byteLimits struct{}
|
|
|
|
func (byteLimits) name() string { return "byte_limits" }
|
|
|
|
func (byteLimits) check(_ context.Context, rq *request, now time.Time) *verdict {
|
|
if !rq.counted || rq.line.WouldAction != "" {
|
|
return nil
|
|
}
|
|
|
|
counts, hit, over := rq.h.limiter.CountBytes(rq.h.clientGroup(rq.client), now,
|
|
rq.countedBytes(), rq.bytesLimitPercent.percent)
|
|
rq.line.Counts.MinuteBytes = counts.MinuteBytes
|
|
rq.line.Counts.HourBytes = counts.HourBytes
|
|
rq.line.Counts.DayBytes = counts.DayBytes
|
|
|
|
if !over {
|
|
return nil
|
|
}
|
|
|
|
rq.banForLimit(now, hit, rq.out.status)
|
|
|
|
return &verdict{figures: limitFigures(hit)}
|
|
}
|