diff --git a/internal/waf/checkpoints/request_header_max_length.go b/internal/waf/checkpoints/request_header_max_length.go new file mode 100644 index 0000000..3d5c6a6 --- /dev/null +++ b/internal/waf/checkpoints/request_header_max_length.go @@ -0,0 +1,37 @@ +package checkpoints + +import ( + "github.com/TeaOSLab/EdgeNode/internal/waf/requests" + "github.com/TeaOSLab/EdgeNode/internal/waf/utils" + "github.com/iwind/TeaGo/maps" +) + +type RequestHeaderMaxLengthCheckpoint struct { + Checkpoint +} + +func (this *RequestHeaderMaxLengthCheckpoint) RequestValue(req requests.Request, param string, options maps.Map, ruleId int64) (value any, hasRequestBody bool, sysErr error, userErr error) { + var maxLen int + for _, v := range req.WAFRaw().Header { + for _, subV := range v { + var l = len(subV) + if l > maxLen { + maxLen = l + } + } + } + value = maxLen + + return +} + +func (this *RequestHeaderMaxLengthCheckpoint) ResponseValue(req requests.Request, resp *requests.Response, param string, options maps.Map, ruleId int64) (value any, hasRequestBody bool, sysErr error, userErr error) { + if this.IsRequest() { + return this.RequestValue(req, param, options, ruleId) + } + return +} + +func (this *RequestHeaderMaxLengthCheckpoint) CacheLife() utils.CacheLife { + return utils.CacheShortLife +} diff --git a/internal/waf/checkpoints/utils.go b/internal/waf/checkpoints/utils.go index f65301c..8cd4651 100644 --- a/internal/waf/checkpoints/utils.go +++ b/internal/waf/checkpoints/utils.go @@ -243,13 +243,21 @@ var AllCheckpoints = []*CheckpointDefinition{ Priority: 100, }, { - Name: "单个Header值", + Name: "单个报头值", Prefix: "header", - Description: "单个Header值", + Description: "单个报头值", HasParams: true, Instance: new(RequestHeaderCheckpoint), Priority: 100, }, + { + Name: "请求报头最大长度", + Prefix: "headerMaxLength", + Description: "最长的请求报头的长度。", + HasParams: false, + Instance: new(RequestHeaderMaxLengthCheckpoint), + Priority: 100, + }, { Name: "国家/地区名称", Prefix: "geoCountryName",