From e8dfc1fcfd5a7705f6a53a23a2e0ca078d7c76f9 Mon Sep 17 00:00:00 2001
From: samwaf <samwafgo@gmail.com>
Date: Mon, 14 Apr 2025 09:41:02 +0800
Subject: [PATCH] fix: ws log Strategy

---
 wafenginecore/wafengine.go | 18 +++++++++++++++++-
 1 file changed, 17 insertions(+), 1 deletion(-)

diff --git a/wafenginecore/wafengine.go b/wafenginecore/wafengine.go
index 13cd1ee..c825487 100644
--- a/wafenginecore/wafengine.go
+++ b/wafenginecore/wafengine.go
@@ -595,7 +595,7 @@ func (waf *WafEngine) modifyResponse() func(*http.Response) error {
 
 			if wafHttpContext, ok := r.Context().Value("waf_context").(innerbean.WafHttpContextData); ok {
 				weblogfrist := wafHttpContext.Weblog
-
+				host := waf.HostCode[wafHttpContext.HostCode]
 				weblogfrist.ACTION = "放行"
 				weblogfrist.STATUS = resp.Status
 				weblogfrist.STATUS_CODE = resp.StatusCode
@@ -614,6 +614,22 @@ func (waf *WafEngine) modifyResponse() func(*http.Response) error {
 
 				// 记录日志
 				if global.GWAF_RUNTIME_RECORD_LOG_TYPE == "all" {
+					if waf.HostTarget[host].Host.EXCLUDE_URL_LOG == "" {
+						global.GQEQUE_LOG_DB.Enqueue(weblogfrist)
+					} else {
+						lines := strings.Split(waf.HostTarget[host].Host.EXCLUDE_URL_LOG, "\n")
+						isRecordLog := true
+						// 检查每一行
+						for _, line := range lines {
+							if strings.HasPrefix(weblogfrist.URL, line) {
+								isRecordLog = false
+							}
+						}
+						if isRecordLog {
+							global.GQEQUE_LOG_DB.Enqueue(weblogfrist)
+						}
+					}
+				} else if global.GWAF_RUNTIME_RECORD_LOG_TYPE == "abnormal" && weblogfrist.ACTION != "放行" {
 					global.GQEQUE_LOG_DB.Enqueue(weblogfrist)
 				}
 			}