EdgeNode/internal/waf/action_go_set.go

package waf

import (
	"github.com/TeaOSLab/EdgeNode/internal/remotelogs"
	"github.com/TeaOSLab/EdgeNode/internal/waf/requests"
	"github.com/iwind/TeaGo/types"
	"net/http"
)

type GoSetAction struct {
	BaseAction

	GroupId string `yaml:"groupId" json:"groupId"`
	SetId   string `yaml:"setId" json:"setId"`
}

func (this *GoSetAction) Init(waf *WAF) error {
	return nil
}

func (this *GoSetAction) Code() string {
	return ActionGoSet
}

func (this *GoSetAction) IsAttack() bool {
	return false
}

func (this *GoSetAction) WillChange() bool {
	return true
}

func (this *GoSetAction) Perform(waf *WAF, group *RuleGroup, set *RuleSet, request requests.Request, writer http.ResponseWriter) PerformResult {
	var nextGroup = waf.FindRuleGroup(types.Int64(this.GroupId))
	if nextGroup == nil || !nextGroup.IsOn {
		return PerformResult{
			ContinueRequest: true,
			GoNextSet:       true,
		}
	}
	var nextSet = nextGroup.FindRuleSet(types.Int64(this.SetId))
	if nextSet == nil || !nextSet.IsOn {
		return PerformResult{
			ContinueRequest: true,
			GoNextSet:       true,
		}
	}

	b, _, err := nextSet.MatchRequest(request)
	if err != nil {
		remotelogs.Error("WAF", "GO_GROUP_ACTION: "+err.Error())
		return PerformResult{
			ContinueRequest: true,
			GoNextSet:       true,
		}
	}
	if !b {
		return PerformResult{
			ContinueRequest: true,
			GoNextSet:       true,
		}
	}
	return nextSet.PerformActions(waf, nextGroup, request, writer)
}