WAF动作增加显示HTML内容

internal/iplibrary/action_base.go

@@ -3,6 +3,7 @@ package iplibrary
 import (
 	"encoding/json"
 	"github.com/iwind/TeaGo/maps"
+	"net/http"
 )
 
 type BaseAction struct {
@@ -12,6 +13,11 @@ func (this *BaseAction) Close() error {
 	return nil
 }
 
+// 处理HTTP请求
+func (this *BaseAction) DoHTTP(req *http.Request, resp http.ResponseWriter) (goNext bool, err error) {
+	return true, nil
+}
+
 func (this *BaseAction) convertParams(params maps.Map, ptr interface{}) error {
 	data, err := json.Marshal(params)
 	if err != nil {

internal/iplibrary/action_html.go

@@ -0,0 +1,55 @@
+package iplibrary
+
+import (
+	"github.com/TeaOSLab/EdgeCommon/pkg/rpc/pb"
+	"github.com/TeaOSLab/EdgeCommon/pkg/serverconfigs/firewallconfigs"
+	"net/http"
+)
+
+// HTML动作
+type HTMLAction struct {
+	BaseAction
+
+	config *firewallconfigs.FirewallActionHTMLConfig
+}
+
+// 获取新对象
+func NewHTMLAction() *HTMLAction {
+	return &HTMLAction{}
+}
+
+// 初始化
+func (this *HTMLAction) Init(config *firewallconfigs.FirewallActionConfig) error {
+	this.config = &firewallconfigs.FirewallActionHTMLConfig{}
+	err := this.convertParams(config.Params, this.config)
+	if err != nil {
+		return err
+	}
+	return nil
+}
+
+// 添加
+func (this *HTMLAction) AddItem(listType IPListType, item *pb.IPItem) error {
+	return nil
+}
+
+// 删除
+func (this *HTMLAction) DeleteItem(listType IPListType, item *pb.IPItem) error {
+	return nil
+}
+
+// 关闭
+func (this *HTMLAction) Close() error {
+	return nil
+}
+
+// 处理HTTP请求
+func (this *HTMLAction) DoHTTP(req *http.Request, resp http.ResponseWriter) (goNext bool, err error) {
+	if this.config == nil {
+		goNext = true
+		return
+	}
+	resp.WriteHeader(http.StatusForbidden) // TODO改成可以配置
+	_, _ = resp.Write([]byte(this.config.Content))
+	return false, nil
+}

internal/iplibrary/action_interface.go

@@ -3,6 +3,7 @@ package iplibrary
 import (
 	"github.com/TeaOSLab/EdgeCommon/pkg/rpc/pb"
 	"github.com/TeaOSLab/EdgeCommon/pkg/serverconfigs/firewallconfigs"
+	"net/http"
 )
 
 type ActionInterface interface {
@@ -17,4 +18,7 @@ type ActionInterface interface {
 
 	// 关闭
 	Close() error
+
+	// 处理HTTP请求
+	DoHTTP(req *http.Request, resp http.ResponseWriter) (goNext bool, err error)
 }

internal/iplibrary/action_manager.go

@@ -108,6 +108,13 @@ func (this *ActionManager) UpdateActions(actions []*firewallconfigs.FirewallActi
 	}
 }
 
+// 查找事件对应的动作
+func (this *ActionManager) FindEventActions(eventLevel string) []ActionInterface {
+	this.locker.Lock()
+	defer this.locker.Unlock()
+	return this.eventMap[eventLevel]
+}
+
 // 执行添加IP动作
 func (this *ActionManager) AddItem(listType IPListType, item *pb.IPItem) {
 	instances, ok := this.eventMap[item.EventLevel]
@@ -147,6 +154,8 @@ func (this *ActionManager) createInstance(config *firewallconfigs.FirewallAction
 		instance = NewScriptAction()
 	case firewallconfigs.FirewallActionTypeHTTPAPI:
 		instance = NewHTTPAPIAction()
+	case firewallconfigs.FirewallActionTypeHTML:
+		instance = NewHTMLAction()
 	}
 	if instance == nil {
 		return nil, errors.New("can not create instance for type '" + config.Type + "'")

internal/iplibrary/ip_list.go

@@ -110,27 +110,40 @@ func (this *IPList) Contains(ip uint64) bool {
 }
 
 // 是否包含一组IP
-func (this *IPList) ContainsIPStrings(ipStrings []string) bool {
+func (this *IPList) ContainsIPStrings(ipStrings []string) (found bool, item *IPItem) {
 	if len(ipStrings) == 0 {
-		return false
+		return
 	}
 	this.locker.RLock()
 	if this.isAll {
+		itemIds := this.ipMap[0]
+		if len(itemIds) > 0 {
+			itemId := itemIds[0]
+			item = this.itemsMap[itemId]
+		}
+
 		this.locker.RUnlock()
-		return true
+		found = true
+		return
 	}
 	for _, ipString := range ipStrings {
 		if len(ipString) == 0 {
 			continue
 		}
-		_, ok := this.ipMap[utils.IP2Long(ipString)]
+		itemIds, ok := this.ipMap[utils.IP2Long(ipString)]
 		if ok {
+			if len(itemIds) > 0 {
+				itemId := itemIds[0]
+				item = this.itemsMap[itemId]
+			}
+
 			this.locker.RUnlock()
-			return true
+			found = true
+			return
 		}
 	}
 	this.locker.RUnlock()
-	return false
+	return
 }
 
 // 在不加锁的情况下删除某个Item

internal/nodes/http_request_waf.go

@@ -49,18 +49,39 @@ func (this *HTTPRequest) checkWAFRequest(firewallPolicy *firewallconfigs.HTTPFir
 	inbound := firewallPolicy.Inbound
 	if inbound.AllowListRef != nil && inbound.AllowListRef.IsOn && inbound.AllowListRef.ListId > 0 {
 		list := iplibrary.SharedIPListManager.FindList(inbound.AllowListRef.ListId)
-		if list != nil && list.ContainsIPStrings(remoteAddrs) {
+		if list != nil {
+			found, _ := list.ContainsIPStrings(remoteAddrs)
+			if found {
 				breakChecking = true
 				return
 			}
 		}
+	}
 
 	// 检查IP黑名单
 	if inbound.DenyListRef != nil && inbound.DenyListRef.IsOn && inbound.DenyListRef.ListId > 0 {
 		list := iplibrary.SharedIPListManager.FindList(inbound.DenyListRef.ListId)
-		if list != nil && list.ContainsIPStrings(remoteAddrs) {
-			// TODO 可以配置对封禁的处理方式等
+		if list != nil {
+			found, item := list.ContainsIPStrings(remoteAddrs)
+			if found {
+				// 触发事件
+				if item != nil && len(item.EventLevel) > 0 {
+					actions := iplibrary.SharedActionManager.FindEventActions(item.EventLevel)
+					for _, action := range actions {
+						goNext, err := action.DoHTTP(this.RawReq, this.RawWriter)
+						if err != nil {
+							remotelogs.Error("REQUEST", "do action '"+err.Error()+"' failed: "+err.Error())
+							return true, false
+						}
+						if !goNext {
+							this.disableLog = true
+							return true, false
+						}
+					}
+				}
+
 				// TODO 需要记录日志信息
+
 				this.writer.WriteHeader(http.StatusForbidden)
 				this.writer.Close()
 
@@ -70,6 +91,7 @@ func (this *HTTPRequest) checkWAFRequest(firewallPolicy *firewallconfigs.HTTPFir
 				return true, false
 			}
 		}
+	}
 
 	// 检查地区封禁
 	if iplibrary.SharedLibrary != nil {