mirror of
https://github.com/TeaOSLab/EdgeNode.git
synced 2025-11-03 23:20:25 +08:00
860 lines
23 KiB
Go
860 lines
23 KiB
Go
package waf
|
|
|
|
import (
|
|
"bytes"
|
|
"encoding/base64"
|
|
"encoding/binary"
|
|
"errors"
|
|
"github.com/TeaOSLab/EdgeCommon/pkg/configutils"
|
|
"github.com/TeaOSLab/EdgeCommon/pkg/serverconfigs/filterconfigs"
|
|
"github.com/TeaOSLab/EdgeNode/internal/re"
|
|
"github.com/TeaOSLab/EdgeNode/internal/remotelogs"
|
|
"github.com/TeaOSLab/EdgeNode/internal/utils/runes"
|
|
"github.com/TeaOSLab/EdgeNode/internal/waf/checkpoints"
|
|
"github.com/TeaOSLab/EdgeNode/internal/waf/injectionutils"
|
|
"github.com/TeaOSLab/EdgeNode/internal/waf/requests"
|
|
"github.com/TeaOSLab/EdgeNode/internal/waf/utils"
|
|
"github.com/TeaOSLab/EdgeNode/internal/waf/values"
|
|
"github.com/iwind/TeaGo/lists"
|
|
"github.com/iwind/TeaGo/maps"
|
|
"github.com/iwind/TeaGo/types"
|
|
stringutil "github.com/iwind/TeaGo/utils/string"
|
|
"net"
|
|
"reflect"
|
|
"regexp"
|
|
"sort"
|
|
"strings"
|
|
)
|
|
|
|
var singleParamRegexp = regexp.MustCompile(`^\${[\w.-]+}$`)
|
|
|
|
// Rule waf rule under rule set
|
|
type Rule struct {
|
|
Id int64
|
|
|
|
Description string `yaml:"description" json:"description"`
|
|
Param string `yaml:"param" json:"param"` // such as ${arg.name} or ${args}, can be composite as ${arg.firstName}${arg.lastName}
|
|
ParamFilters []*ParamFilter `yaml:"paramFilters" json:"paramFilters"`
|
|
Operator RuleOperator `yaml:"operator" json:"operator"` // such as contains, gt, ...
|
|
Value string `yaml:"value" json:"value"` // compared value
|
|
IsCaseInsensitive bool `yaml:"isCaseInsensitive" json:"isCaseInsensitive"`
|
|
CheckpointOptions map[string]any `yaml:"checkpointOptions" json:"checkpointOptions"`
|
|
Priority int `yaml:"priority" json:"priority"`
|
|
|
|
checkpointFinder func(prefix string) checkpoints.CheckpointInterface
|
|
|
|
singleParam string // real param after prefix
|
|
singleCheckpoint checkpoints.CheckpointInterface // if is single check point
|
|
|
|
multipleCheckpoints map[string]checkpoints.CheckpointInterface
|
|
|
|
isIP bool
|
|
ipValue net.IP
|
|
|
|
ipRangeListValue *values.IPRangeList
|
|
stringValues []string
|
|
stringValueRunes [][]rune
|
|
ipList *values.StringList
|
|
|
|
floatValue float64
|
|
|
|
reg *re.Regexp
|
|
cacheLife utils.CacheLife
|
|
}
|
|
|
|
func NewRule() *Rule {
|
|
return &Rule{}
|
|
}
|
|
|
|
func (this *Rule) Init() error {
|
|
// operator
|
|
switch this.Operator {
|
|
case RuleOperatorGt:
|
|
this.floatValue = types.Float64(this.Value)
|
|
case RuleOperatorGte:
|
|
this.floatValue = types.Float64(this.Value)
|
|
case RuleOperatorLt:
|
|
this.floatValue = types.Float64(this.Value)
|
|
case RuleOperatorLte:
|
|
this.floatValue = types.Float64(this.Value)
|
|
case RuleOperatorEq:
|
|
this.floatValue = types.Float64(this.Value)
|
|
case RuleOperatorNeq:
|
|
this.floatValue = types.Float64(this.Value)
|
|
case RuleOperatorContainsAny, RuleOperatorContainsAll, RuleOperatorContainsAnyWord, RuleOperatorContainsAllWords, RuleOperatorNotContainsAnyWord:
|
|
this.stringValues = []string{}
|
|
if len(this.Value) > 0 {
|
|
var lines = strings.Split(this.Value, "\n")
|
|
for _, line := range lines {
|
|
line = strings.TrimSpace(line)
|
|
if len(line) > 0 {
|
|
if this.IsCaseInsensitive {
|
|
this.stringValues = append(this.stringValues, strings.ToLower(line))
|
|
} else {
|
|
this.stringValues = append(this.stringValues, line)
|
|
}
|
|
}
|
|
}
|
|
if this.Operator == RuleOperatorContainsAnyWord || this.Operator == RuleOperatorContainsAllWords || this.Operator == RuleOperatorNotContainsAnyWord {
|
|
sort.Strings(this.stringValues)
|
|
}
|
|
|
|
this.stringValueRunes = [][]rune{}
|
|
for _, line := range this.stringValues {
|
|
this.stringValueRunes = append(this.stringValueRunes, []rune(line))
|
|
}
|
|
}
|
|
case RuleOperatorMatch:
|
|
var v = this.Value
|
|
if this.IsCaseInsensitive && !strings.HasPrefix(v, "(?i)") {
|
|
v = "(?i)" + v
|
|
}
|
|
|
|
v = this.unescape(v)
|
|
|
|
reg, err := re.Compile(v)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
this.reg = reg
|
|
case RuleOperatorNotMatch:
|
|
var v = this.Value
|
|
if this.IsCaseInsensitive && !strings.HasPrefix(v, "(?i)") {
|
|
v = "(?i)" + v
|
|
}
|
|
|
|
v = this.unescape(v)
|
|
|
|
reg, err := re.Compile(v)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
this.reg = reg
|
|
case RuleOperatorEqIP, RuleOperatorGtIP, RuleOperatorGteIP, RuleOperatorLtIP, RuleOperatorLteIP:
|
|
this.ipValue = net.ParseIP(this.Value)
|
|
this.isIP = this.ipValue != nil
|
|
|
|
if !this.isIP {
|
|
return errors.New("value should be a valid ip")
|
|
}
|
|
case RuleOperatorInIPList:
|
|
this.ipList = values.ParseStringList(this.Value, true)
|
|
case RuleOperatorIPRange, RuleOperatorNotIPRange:
|
|
this.ipRangeListValue = values.ParseIPRangeList(this.Value)
|
|
case RuleOperatorWildcardMatch, RuleOperatorWildcardNotMatch:
|
|
var pieces = strings.Split(this.Value, "*")
|
|
for index, piece := range pieces {
|
|
pieces[index] = regexp.QuoteMeta(piece)
|
|
}
|
|
var pattern = strings.Join(pieces, "(.*)")
|
|
var expr = "^" + pattern + "$"
|
|
if this.IsCaseInsensitive {
|
|
expr = "(?i)" + expr
|
|
}
|
|
reg, err := re.Compile(expr)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
this.reg = reg
|
|
}
|
|
|
|
if singleParamRegexp.MatchString(this.Param) {
|
|
var param = this.Param[2 : len(this.Param)-1]
|
|
var pieces = strings.SplitN(param, ".", 2)
|
|
var prefix = pieces[0]
|
|
if len(pieces) == 1 {
|
|
this.singleParam = ""
|
|
} else {
|
|
this.singleParam = pieces[1]
|
|
}
|
|
|
|
if this.checkpointFinder != nil {
|
|
var checkpoint = this.checkpointFinder(prefix)
|
|
if checkpoint == nil {
|
|
return errors.New("no check point '" + prefix + "' found")
|
|
}
|
|
this.singleCheckpoint = checkpoint
|
|
this.Priority = checkpoint.Priority()
|
|
|
|
this.cacheLife = checkpoint.CacheLife()
|
|
} else {
|
|
var checkpoint = checkpoints.FindCheckpoint(prefix)
|
|
if checkpoint == nil {
|
|
return errors.New("no check point '" + prefix + "' found")
|
|
}
|
|
checkpoint.Init()
|
|
this.singleCheckpoint = checkpoint
|
|
this.Priority = checkpoint.Priority()
|
|
|
|
this.cacheLife = checkpoint.CacheLife()
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
this.multipleCheckpoints = map[string]checkpoints.CheckpointInterface{}
|
|
var err error = nil
|
|
configutils.ParseVariables(this.Param, func(varName string) (value string) {
|
|
var pieces = strings.SplitN(varName, ".", 2)
|
|
var prefix = pieces[0]
|
|
if this.checkpointFinder != nil {
|
|
var checkpoint = this.checkpointFinder(prefix)
|
|
if checkpoint == nil {
|
|
err = errors.New("no check point '" + prefix + "' found")
|
|
} else {
|
|
this.multipleCheckpoints[prefix] = checkpoint
|
|
this.Priority = checkpoint.Priority()
|
|
|
|
if this.cacheLife <= 0 || checkpoint.CacheLife() < this.cacheLife {
|
|
this.cacheLife = checkpoint.CacheLife()
|
|
}
|
|
}
|
|
} else {
|
|
var checkpoint = checkpoints.FindCheckpoint(prefix)
|
|
if checkpoint == nil {
|
|
err = errors.New("no check point '" + prefix + "' found")
|
|
} else {
|
|
checkpoint.Init()
|
|
this.multipleCheckpoints[prefix] = checkpoint
|
|
this.Priority = checkpoint.Priority()
|
|
|
|
this.cacheLife = checkpoint.CacheLife()
|
|
}
|
|
}
|
|
|
|
return ""
|
|
})
|
|
|
|
return err
|
|
}
|
|
|
|
func (this *Rule) MatchRequest(req requests.Request) (b bool, hasRequestBody bool, err error) {
|
|
if this.singleCheckpoint != nil {
|
|
value, hasCheckedRequestBody, err, _ := this.singleCheckpoint.RequestValue(req, this.singleParam, this.CheckpointOptions, this.Id)
|
|
if hasCheckedRequestBody {
|
|
hasRequestBody = true
|
|
}
|
|
if err != nil {
|
|
return false, hasRequestBody, err
|
|
}
|
|
|
|
// execute filters
|
|
if len(this.ParamFilters) > 0 {
|
|
value = this.execFilter(value)
|
|
}
|
|
|
|
// if is composed checkpoint, we just returns true or false
|
|
if this.singleCheckpoint.IsComposed() {
|
|
return types.Bool(value), hasRequestBody, nil
|
|
}
|
|
|
|
return this.Test(value), hasRequestBody, nil
|
|
}
|
|
|
|
var value = configutils.ParseVariables(this.Param, func(varName string) (value string) {
|
|
var pieces = strings.SplitN(varName, ".", 2)
|
|
var prefix = pieces[0]
|
|
point, ok := this.multipleCheckpoints[prefix]
|
|
if !ok {
|
|
return ""
|
|
}
|
|
|
|
if len(pieces) == 1 {
|
|
value1, hasCheckRequestBody, err1, _ := point.RequestValue(req, "", this.CheckpointOptions, this.Id)
|
|
if hasCheckRequestBody {
|
|
hasRequestBody = true
|
|
}
|
|
if err1 != nil {
|
|
err = err1
|
|
}
|
|
return this.stringifyValue(value1)
|
|
}
|
|
|
|
value1, hasCheckRequestBody, err1, _ := point.RequestValue(req, pieces[1], this.CheckpointOptions, this.Id)
|
|
if hasCheckRequestBody {
|
|
hasRequestBody = true
|
|
}
|
|
if err1 != nil {
|
|
err = err1
|
|
}
|
|
return this.stringifyValue(value1)
|
|
})
|
|
|
|
if err != nil {
|
|
return false, hasRequestBody, err
|
|
}
|
|
|
|
return this.Test(value), hasRequestBody, nil
|
|
}
|
|
|
|
func (this *Rule) MatchResponse(req requests.Request, resp *requests.Response) (b bool, hasRequestBody bool, err error) {
|
|
if this.singleCheckpoint != nil {
|
|
// if is request param
|
|
if this.singleCheckpoint.IsRequest() {
|
|
value, hasCheckRequestBody, err, _ := this.singleCheckpoint.RequestValue(req, this.singleParam, this.CheckpointOptions, this.Id)
|
|
if hasCheckRequestBody {
|
|
hasRequestBody = true
|
|
}
|
|
if err != nil {
|
|
return false, hasRequestBody, err
|
|
}
|
|
|
|
// execute filters
|
|
if len(this.ParamFilters) > 0 {
|
|
value = this.execFilter(value)
|
|
}
|
|
|
|
return this.Test(value), hasRequestBody, nil
|
|
}
|
|
|
|
// response param
|
|
value, hasCheckRequestBody, err, _ := this.singleCheckpoint.ResponseValue(req, resp, this.singleParam, this.CheckpointOptions, this.Id)
|
|
if hasCheckRequestBody {
|
|
hasRequestBody = true
|
|
}
|
|
if err != nil {
|
|
return false, hasRequestBody, err
|
|
}
|
|
|
|
// if is composed checkpoint, we just returns true or false
|
|
if this.singleCheckpoint.IsComposed() {
|
|
return types.Bool(value), hasRequestBody, nil
|
|
}
|
|
|
|
return this.Test(value), hasRequestBody, nil
|
|
}
|
|
|
|
var value = configutils.ParseVariables(this.Param, func(varName string) (value string) {
|
|
var pieces = strings.SplitN(varName, ".", 2)
|
|
var prefix = pieces[0]
|
|
point, ok := this.multipleCheckpoints[prefix]
|
|
if !ok {
|
|
return ""
|
|
}
|
|
|
|
if len(pieces) == 1 {
|
|
if point.IsRequest() {
|
|
value1, hasCheckRequestBody, err1, _ := point.RequestValue(req, "", this.CheckpointOptions, this.Id)
|
|
if hasCheckRequestBody {
|
|
hasRequestBody = true
|
|
}
|
|
if err1 != nil {
|
|
err = err1
|
|
}
|
|
return this.stringifyValue(value1)
|
|
} else {
|
|
value1, hasCheckRequestBody, err1, _ := point.ResponseValue(req, resp, "", this.CheckpointOptions, this.Id)
|
|
if hasCheckRequestBody {
|
|
hasRequestBody = true
|
|
}
|
|
if err1 != nil {
|
|
err = err1
|
|
}
|
|
return this.stringifyValue(value1)
|
|
}
|
|
}
|
|
|
|
if point.IsRequest() {
|
|
value1, hasCheckRequestBody, err1, _ := point.RequestValue(req, pieces[1], this.CheckpointOptions, this.Id)
|
|
if hasCheckRequestBody {
|
|
hasRequestBody = true
|
|
}
|
|
if err1 != nil {
|
|
err = err1
|
|
}
|
|
return this.stringifyValue(value1)
|
|
} else {
|
|
value1, hasCheckRequestBody, err1, _ := point.ResponseValue(req, resp, pieces[1], this.CheckpointOptions, this.Id)
|
|
if hasCheckRequestBody {
|
|
hasRequestBody = true
|
|
}
|
|
if err1 != nil {
|
|
err = err1
|
|
}
|
|
return this.stringifyValue(value1)
|
|
}
|
|
})
|
|
|
|
if err != nil {
|
|
return false, hasRequestBody, err
|
|
}
|
|
|
|
return this.Test(value), hasRequestBody, nil
|
|
}
|
|
|
|
func (this *Rule) Test(value any) bool {
|
|
// operator
|
|
switch this.Operator {
|
|
case RuleOperatorGt:
|
|
return types.Float64(value) > this.floatValue
|
|
case RuleOperatorGte:
|
|
return types.Float64(value) >= this.floatValue
|
|
case RuleOperatorLt:
|
|
return types.Float64(value) < this.floatValue
|
|
case RuleOperatorLte:
|
|
return types.Float64(value) <= this.floatValue
|
|
case RuleOperatorEq:
|
|
return types.Float64(value) == this.floatValue
|
|
case RuleOperatorNeq:
|
|
return types.Float64(value) != this.floatValue
|
|
case RuleOperatorEqString:
|
|
if this.IsCaseInsensitive {
|
|
return strings.EqualFold(this.stringifyValue(value), this.Value)
|
|
} else {
|
|
return this.stringifyValue(value) == this.Value
|
|
}
|
|
case RuleOperatorNeqString:
|
|
if this.IsCaseInsensitive {
|
|
return !strings.EqualFold(this.stringifyValue(value), this.Value)
|
|
} else {
|
|
return this.stringifyValue(value) != this.Value
|
|
}
|
|
case RuleOperatorMatch, RuleOperatorWildcardMatch:
|
|
if value == nil {
|
|
value = ""
|
|
}
|
|
|
|
// strings
|
|
stringList, ok := value.([]string)
|
|
if ok {
|
|
for _, s := range stringList {
|
|
if utils.MatchStringCache(this.reg, s, this.cacheLife) {
|
|
return true
|
|
}
|
|
}
|
|
return false
|
|
}
|
|
|
|
// bytes list
|
|
byteSlices, ok := value.([][]byte)
|
|
if ok {
|
|
for _, byteSlice := range byteSlices {
|
|
if utils.MatchBytesCache(this.reg, byteSlice, this.cacheLife) {
|
|
return true
|
|
}
|
|
}
|
|
return false
|
|
}
|
|
|
|
// bytes
|
|
byteSlice, ok := value.([]byte)
|
|
if ok {
|
|
return utils.MatchBytesCache(this.reg, byteSlice, this.cacheLife)
|
|
}
|
|
|
|
// string
|
|
return utils.MatchStringCache(this.reg, this.stringifyValue(value), this.cacheLife)
|
|
case RuleOperatorNotMatch, RuleOperatorWildcardNotMatch:
|
|
if value == nil {
|
|
value = ""
|
|
}
|
|
stringList, ok := value.([]string)
|
|
if ok {
|
|
for _, s := range stringList {
|
|
if utils.MatchStringCache(this.reg, s, this.cacheLife) {
|
|
return false
|
|
}
|
|
}
|
|
return true
|
|
}
|
|
|
|
// bytes list
|
|
byteSlices, ok := value.([][]byte)
|
|
if ok {
|
|
for _, byteSlice := range byteSlices {
|
|
if utils.MatchBytesCache(this.reg, byteSlice, this.cacheLife) {
|
|
return false
|
|
}
|
|
}
|
|
return true
|
|
}
|
|
|
|
// bytes
|
|
byteSlice, ok := value.([]byte)
|
|
if ok {
|
|
return !utils.MatchBytesCache(this.reg, byteSlice, this.cacheLife)
|
|
}
|
|
|
|
return !utils.MatchStringCache(this.reg, this.stringifyValue(value), this.cacheLife)
|
|
case RuleOperatorContains:
|
|
if types.IsSlice(value) {
|
|
_, isBytes := value.([]byte)
|
|
if !isBytes {
|
|
var ok = false
|
|
lists.Each(value, func(k int, v any) {
|
|
if this.stringifyValue(v) == this.Value {
|
|
ok = true
|
|
}
|
|
})
|
|
return ok
|
|
}
|
|
}
|
|
if types.IsMap(value) {
|
|
var lowerValue = ""
|
|
if this.IsCaseInsensitive {
|
|
lowerValue = strings.ToLower(this.Value)
|
|
}
|
|
for _, v := range maps.NewMap(value) {
|
|
if this.IsCaseInsensitive {
|
|
if strings.ToLower(this.stringifyValue(v)) == lowerValue {
|
|
return true
|
|
}
|
|
} else {
|
|
if this.stringifyValue(v) == this.Value {
|
|
return true
|
|
}
|
|
}
|
|
}
|
|
return false
|
|
}
|
|
|
|
if this.IsCaseInsensitive {
|
|
return strings.Contains(strings.ToLower(this.stringifyValue(value)), strings.ToLower(this.Value))
|
|
} else {
|
|
return strings.Contains(this.stringifyValue(value), this.Value)
|
|
}
|
|
case RuleOperatorNotContains:
|
|
if this.IsCaseInsensitive {
|
|
return !strings.Contains(strings.ToLower(this.stringifyValue(value)), strings.ToLower(this.Value))
|
|
} else {
|
|
return !strings.Contains(this.stringifyValue(value), this.Value)
|
|
}
|
|
case RuleOperatorPrefix:
|
|
if this.IsCaseInsensitive {
|
|
var s = this.stringifyValue(value)
|
|
var sl = len(s)
|
|
var vl = len(this.Value)
|
|
if sl < vl {
|
|
return false
|
|
}
|
|
s = s[:vl]
|
|
return strings.HasPrefix(strings.ToLower(s), strings.ToLower(this.Value))
|
|
} else {
|
|
return strings.HasPrefix(this.stringifyValue(value), this.Value)
|
|
}
|
|
case RuleOperatorSuffix:
|
|
if this.IsCaseInsensitive {
|
|
var s = this.stringifyValue(value)
|
|
var sl = len(s)
|
|
var vl = len(this.Value)
|
|
if sl < vl {
|
|
return false
|
|
}
|
|
s = s[sl-vl:]
|
|
return strings.HasSuffix(strings.ToLower(s), strings.ToLower(this.Value))
|
|
} else {
|
|
return strings.HasSuffix(this.stringifyValue(value), this.Value)
|
|
}
|
|
case RuleOperatorContainsAny:
|
|
var stringValue = this.stringifyValue(value)
|
|
if this.IsCaseInsensitive {
|
|
stringValue = strings.ToLower(stringValue)
|
|
}
|
|
if len(stringValue) > 0 && len(this.stringValues) > 0 {
|
|
for _, v := range this.stringValues {
|
|
if strings.Contains(stringValue, v) {
|
|
return true
|
|
}
|
|
}
|
|
}
|
|
return false
|
|
case RuleOperatorContainsAll:
|
|
var stringValue = this.stringifyValue(value)
|
|
if this.IsCaseInsensitive {
|
|
stringValue = strings.ToLower(stringValue)
|
|
}
|
|
if len(stringValue) > 0 && len(this.stringValues) > 0 {
|
|
for _, v := range this.stringValues {
|
|
if !strings.Contains(stringValue, v) {
|
|
return false
|
|
}
|
|
}
|
|
return true
|
|
}
|
|
return false
|
|
case RuleOperatorContainsAnyWord:
|
|
return runes.ContainsAnyWordRunes(this.stringifyValue(value), this.stringValueRunes, this.IsCaseInsensitive)
|
|
case RuleOperatorContainsAllWords:
|
|
return runes.ContainsAllWords(this.stringifyValue(value), this.stringValues, this.IsCaseInsensitive)
|
|
case RuleOperatorNotContainsAnyWord:
|
|
return !runes.ContainsAnyWordRunes(this.stringifyValue(value), this.stringValueRunes, this.IsCaseInsensitive)
|
|
case RuleOperatorContainsSQLInjection:
|
|
if value == nil {
|
|
return false
|
|
}
|
|
switch xValue := value.(type) {
|
|
case []string:
|
|
for _, v := range xValue {
|
|
if injectionutils.DetectSQLInjectionCache(v, this.cacheLife) {
|
|
return true
|
|
}
|
|
}
|
|
return false
|
|
case [][]byte:
|
|
for _, v := range xValue {
|
|
if injectionutils.DetectSQLInjectionCache(string(v), this.cacheLife) {
|
|
return true
|
|
}
|
|
}
|
|
return false
|
|
default:
|
|
return injectionutils.DetectSQLInjectionCache(this.stringifyValue(value), this.cacheLife)
|
|
}
|
|
case RuleOperatorContainsXSS:
|
|
if value == nil {
|
|
return false
|
|
}
|
|
switch xValue := value.(type) {
|
|
case []string:
|
|
for _, v := range xValue {
|
|
if injectionutils.DetectXSSCache(v, this.cacheLife) {
|
|
return true
|
|
}
|
|
}
|
|
return false
|
|
case [][]byte:
|
|
for _, v := range xValue {
|
|
if injectionutils.DetectXSSCache(string(v), this.cacheLife) {
|
|
return true
|
|
}
|
|
}
|
|
return false
|
|
default:
|
|
return injectionutils.DetectXSSCache(this.stringifyValue(value), this.cacheLife)
|
|
}
|
|
case RuleOperatorContainsBinary:
|
|
data, _ := base64.StdEncoding.DecodeString(this.stringifyValue(this.Value))
|
|
if this.IsCaseInsensitive {
|
|
return bytes.Contains(bytes.ToUpper([]byte(this.stringifyValue(value))), bytes.ToUpper(data))
|
|
} else {
|
|
return bytes.Contains([]byte(this.stringifyValue(value)), data)
|
|
}
|
|
case RuleOperatorNotContainsBinary:
|
|
data, _ := base64.StdEncoding.DecodeString(this.stringifyValue(this.Value))
|
|
if this.IsCaseInsensitive {
|
|
return !bytes.Contains(bytes.ToUpper([]byte(this.stringifyValue(value))), bytes.ToUpper(data))
|
|
} else {
|
|
return !bytes.Contains([]byte(this.stringifyValue(value)), data)
|
|
}
|
|
case RuleOperatorHasKey:
|
|
if types.IsSlice(value) {
|
|
var index = types.Int(this.Value)
|
|
if index < 0 {
|
|
return false
|
|
}
|
|
return reflect.ValueOf(value).Len() > index
|
|
} else if types.IsMap(value) {
|
|
var m = maps.NewMap(value)
|
|
if this.IsCaseInsensitive {
|
|
var lowerValue = strings.ToLower(this.Value)
|
|
for k := range m {
|
|
if strings.ToLower(k) == lowerValue {
|
|
return true
|
|
}
|
|
}
|
|
} else {
|
|
return m.Has(this.Value)
|
|
}
|
|
} else {
|
|
return false
|
|
}
|
|
|
|
case RuleOperatorVersionGt:
|
|
return stringutil.VersionCompare(this.Value, types.String(value)) > 0
|
|
case RuleOperatorVersionLt:
|
|
return stringutil.VersionCompare(this.Value, types.String(value)) < 0
|
|
case RuleOperatorVersionRange:
|
|
if strings.Contains(this.Value, ",") {
|
|
var versions = strings.SplitN(this.Value, ",", 2)
|
|
var version1 = strings.TrimSpace(versions[0])
|
|
var version2 = strings.TrimSpace(versions[1])
|
|
if len(version1) > 0 && stringutil.VersionCompare(types.String(value), version1) < 0 {
|
|
return false
|
|
}
|
|
if len(version2) > 0 && stringutil.VersionCompare(types.String(value), version2) > 0 {
|
|
return false
|
|
}
|
|
return true
|
|
} else {
|
|
return stringutil.VersionCompare(types.String(value), this.Value) >= 0
|
|
}
|
|
case RuleOperatorEqIP:
|
|
var ip = net.ParseIP(types.String(value))
|
|
if ip == nil {
|
|
return false
|
|
}
|
|
return this.isIP && ip.Equal(this.ipValue)
|
|
case RuleOperatorGtIP:
|
|
var ip = net.ParseIP(types.String(value))
|
|
if ip == nil {
|
|
return false
|
|
}
|
|
return this.isIP && bytes.Compare(ip, this.ipValue) > 0
|
|
case RuleOperatorGteIP:
|
|
var ip = net.ParseIP(types.String(value))
|
|
if ip == nil {
|
|
return false
|
|
}
|
|
return this.isIP && bytes.Compare(ip, this.ipValue) >= 0
|
|
case RuleOperatorLtIP:
|
|
var ip = net.ParseIP(types.String(value))
|
|
if ip == nil {
|
|
return false
|
|
}
|
|
return this.isIP && bytes.Compare(ip, this.ipValue) < 0
|
|
case RuleOperatorLteIP:
|
|
var ip = net.ParseIP(types.String(value))
|
|
if ip == nil {
|
|
return false
|
|
}
|
|
return this.isIP && bytes.Compare(ip, this.ipValue) <= 0
|
|
case RuleOperatorIPRange:
|
|
return this.containsIP(value)
|
|
case RuleOperatorNotIPRange:
|
|
return !this.containsIP(value)
|
|
case RuleOperatorIPMod:
|
|
var pieces = strings.SplitN(this.Value, ",", 2)
|
|
if len(pieces) == 1 {
|
|
var rem = types.Int64(pieces[0])
|
|
return this.ipToInt64(net.ParseIP(types.String(value)))%10 == rem
|
|
}
|
|
var div = types.Int64(pieces[0])
|
|
if div == 0 {
|
|
return false
|
|
}
|
|
var rem = types.Int64(pieces[1])
|
|
return this.ipToInt64(net.ParseIP(types.String(value)))%div == rem
|
|
case RuleOperatorIPMod10:
|
|
return this.ipToInt64(net.ParseIP(types.String(value)))%10 == types.Int64(this.Value)
|
|
case RuleOperatorIPMod100:
|
|
return this.ipToInt64(net.ParseIP(types.String(value)))%100 == types.Int64(this.Value)
|
|
case RuleOperatorInIPList:
|
|
if this.ipList != nil {
|
|
return this.ipList.Contains(types.String(value))
|
|
}
|
|
return false
|
|
}
|
|
return false
|
|
}
|
|
|
|
func (this *Rule) IsSingleCheckpoint() bool {
|
|
return this.singleCheckpoint != nil
|
|
}
|
|
|
|
func (this *Rule) SetCheckpointFinder(finder func(prefix string) checkpoints.CheckpointInterface) {
|
|
this.checkpointFinder = finder
|
|
}
|
|
|
|
var unescapeChars = [][2]string{
|
|
{`\s`, `(\s|%09|%0A|\+)`},
|
|
{`\(`, `(\(|%28)`},
|
|
{`=`, `(=|%3D)`},
|
|
{`<`, `(<|%3C)`},
|
|
{`\*`, `(\*|%2A)`},
|
|
{`\\`, `(\\|%2F)`},
|
|
{`!`, `(!|%21)`},
|
|
{`/`, `(/|%2F)`},
|
|
{`;`, `(;|%3B)`},
|
|
{`\+`, `(\+|%20)`},
|
|
}
|
|
|
|
func (this *Rule) unescape(v string) string {
|
|
// replace urlencoded characters
|
|
|
|
for _, c := range unescapeChars {
|
|
if !strings.Contains(v, c[0]) {
|
|
continue
|
|
}
|
|
var pieces = strings.Split(v, c[0])
|
|
|
|
// 修复piece中错误的\
|
|
for pieceIndex, piece := range pieces {
|
|
var l = len(piece)
|
|
if l == 0 {
|
|
continue
|
|
}
|
|
if piece[l-1] != '\\' {
|
|
continue
|
|
}
|
|
|
|
// 计算\的数量
|
|
var countBackSlashes = 0
|
|
for i := l - 1; i >= 0; i-- {
|
|
if piece[i] == '\\' {
|
|
countBackSlashes++
|
|
} else {
|
|
break
|
|
}
|
|
}
|
|
if countBackSlashes%2 == 1 {
|
|
// 去掉最后一个
|
|
pieces[pieceIndex] = piece[:len(piece)-1]
|
|
}
|
|
}
|
|
|
|
v = strings.Join(pieces, c[1])
|
|
}
|
|
|
|
return v
|
|
}
|
|
|
|
func (this *Rule) containsIP(value any) bool {
|
|
if this.ipRangeListValue == nil {
|
|
return false
|
|
}
|
|
return this.ipRangeListValue.Contains(types.String(value))
|
|
}
|
|
|
|
func (this *Rule) ipToInt64(ip net.IP) int64 {
|
|
if len(ip) == 0 {
|
|
return 0
|
|
}
|
|
if len(ip) == 16 {
|
|
return int64(binary.BigEndian.Uint32(ip[12:16]))
|
|
}
|
|
return int64(binary.BigEndian.Uint32(ip))
|
|
}
|
|
|
|
func (this *Rule) execFilter(value any) any {
|
|
var goNext bool
|
|
var err error
|
|
|
|
for _, filter := range this.ParamFilters {
|
|
filterInstance := filterconfigs.FindFilter(filter.Code)
|
|
if filterInstance == nil {
|
|
continue
|
|
}
|
|
value, goNext, err = filterInstance.Do(value, filter.Options)
|
|
if err != nil {
|
|
remotelogs.Error("WAF", "filter error: "+err.Error())
|
|
break
|
|
}
|
|
if !goNext {
|
|
break
|
|
}
|
|
}
|
|
return value
|
|
}
|
|
|
|
func (this *Rule) stringifyValue(value any) string {
|
|
if value == nil {
|
|
return ""
|
|
}
|
|
switch v := value.(type) {
|
|
case string:
|
|
return v
|
|
case []string:
|
|
return strings.Join(v, "")
|
|
case []byte:
|
|
return string(v)
|
|
case [][]byte:
|
|
var b = &bytes.Buffer{}
|
|
for _, vb := range v {
|
|
b.Write(vb)
|
|
}
|
|
return b.String()
|
|
default:
|
|
return types.String(v)
|
|
}
|
|
}
|