EdgeNode/internal/waf/action_go_set.go

package waf

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

type GoSetAction struct {
	GroupId string `yaml:"groupId" json:"groupId"`
	SetId   string `yaml:"setId" json:"setId"`
}

func (this *GoSetAction) Perform(waf *WAF, request *requests.Request, writer http.ResponseWriter) (allow bool) {
	group := waf.FindRuleGroup(this.GroupId)
	if group == nil || !group.IsOn {
		return true
	}
	set := group.FindRuleSet(this.SetId)
	if set == nil || !set.IsOn {
		return true
	}

	b, err := set.MatchRequest(request)
	if err != nil {
		logs.Error(err)
		return true
	}
	if !b {
		return true
	}
	actionObject := FindActionInstance(set.Action, set.ActionOptions)
	if actionObject == nil {
		return true
	}
	return actionObject.Perform(waf, request, writer)
}