mirror of
https://gitee.com/samwaf/SamWaf.git
synced 2026-11-03 19:52:14 +08:00
190 lines
7.4 KiB
Go
190 lines
7.4 KiB
Go
package innerbean
|
|
|
|
import "strings"
|
|
|
|
type WebLog struct {
|
|
WafInnerDFlag string `json:"waf_inner_dflag"` //日志队列处理方式
|
|
HOST string `json:"host"`
|
|
URL string `json:"url"`
|
|
RawQuery string `json:"raw_query"` //原始URL查询
|
|
REFERER string `json:"referer"`
|
|
USER_AGENT string `json:"user_agent"`
|
|
METHOD string `json:"method"`
|
|
HEADER string `json:"header"`
|
|
SRC_IP string `json:"src_ip"`
|
|
SRC_PORT string `json:"src_port"`
|
|
COUNTRY string `json:"country"`
|
|
PROVINCE string `json:"province"`
|
|
CITY string `json:"city"`
|
|
CREATE_TIME string `gorm:"index:idx_weblog_time" json:"create_time"`
|
|
CONTENT_LENGTH int64 `json:"content_length"`
|
|
COOKIES string `json:"cookies"`
|
|
BODY string `json:"body"`
|
|
REQ_UUID string `json:"req_uuid"`
|
|
USER_CODE string `json:"user_code" gorm:"index"`
|
|
TenantId string `json:"tenant_id" gorm:"index"` //租户ID（主要键）
|
|
HOST_CODE string `json:"host_code" ` //主机ID （主要键）
|
|
Day int `json:"day"` //日 （主要键）
|
|
ACTION string `json:"action"`
|
|
RULE string `json:"rule"`
|
|
STATUS string `json:"status"` //状态
|
|
STATUS_CODE int `json:"status_code"` //状态编码
|
|
RES_BODY string `json:"res_body"` //返回信息
|
|
POST_FORM string `json:"post_form"` //提交的表单数据
|
|
TASK_FLAG int `json:"task_flag" gorm:"default:-1;index"` //任务处理标记 -1 等待处理；1 可以进行处理；2 处理完毕
|
|
UNIX_ADD_TIME int64 `json:"unix_add_time" gorm:"index"` //添加日期unix
|
|
RISK_LEVEL int `json:"risk_level"` //危险等级 0:正常 1:轻微 2:有害 3:严重 4:特别严重
|
|
GUEST_IDENTIFICATION string `json:"guest_identification"` //访客身份识别
|
|
IsBot int `json:"is_bot"` //是否是机器人 0 不是机器人 1 机器人
|
|
TimeSpent int64 `json:"time_spent"` //用时
|
|
NetSrcIp string `json:"net_src_ip"` //获取的原始IP
|
|
SrcByteBody []byte `json:"src_byte_body"` //原始body信息
|
|
SrcByteResBody []byte `json:"src_byte_res_body"` //返回body bytes信息
|
|
WebLogVersion int `json:"web_log_version"` //日志版本信息早期的是空和0，后期实时增加
|
|
Scheme string `json:"scheme"` //HTTP 协议
|
|
SrcURL []byte `json:"src_url"` //原始url信息
|
|
PreCheckCost int64 `json:"pre_check_cost"` // 前置检查耗时(ms)
|
|
ForwardCost int64 `json:"forward_cost"` // 转发耗时(ms)
|
|
BackendCheckCost int64 `json:"backend_check_cost"` // 后端处理耗时(ms)
|
|
ResHeader string `json:"res_header"` // 返回header情况
|
|
BodyHash string `json:"body_hash"` // body hash值
|
|
LogOnlyMode int `json:"log_only_mode"` //是否只记录日志 1 是 0 不是
|
|
IsBalance int `json:"is_balance"` //是否是负载均衡 1 是 0 不是
|
|
BalanceInfo string `json:"balance_info"` //负载均衡IP端口信息
|
|
}
|
|
|
|
// GetHeaderValue 从HEADER字段中提取指定header的值
|
|
// headerName: 要查找的header名称（不区分大小写）
|
|
// 返回: header的值，如果不存在则返回空字符串
|
|
func (w *WebLog) GetHeaderValue(headerName string) string {
|
|
if w.HEADER == "" || headerName == "" {
|
|
return ""
|
|
}
|
|
|
|
// 将header名称转换为小写进行比较
|
|
headerName = strings.ToLower(headerName)
|
|
|
|
// 按行分割header字符串
|
|
lines := strings.Split(w.HEADER, "\r\n")
|
|
|
|
for _, line := range lines {
|
|
// 跳过空行
|
|
if strings.TrimSpace(line) == "" {
|
|
continue
|
|
}
|
|
|
|
// 查找冒号分隔符
|
|
colonIndex := strings.Index(line, ":")
|
|
if colonIndex == -1 {
|
|
continue
|
|
}
|
|
|
|
// 提取header名称和值
|
|
key := strings.ToLower(strings.TrimSpace(line[:colonIndex]))
|
|
value := strings.TrimSpace(line[colonIndex+1:])
|
|
|
|
// 如果找到匹配的header，返回其值
|
|
if key == headerName {
|
|
return value
|
|
}
|
|
}
|
|
|
|
// 未找到指定的header，返回空字符串
|
|
return ""
|
|
}
|
|
|
|
// 在 GORM 的 Model 方法中定义复合索引
|
|
func (WebLog) TableName() string {
|
|
return "web_logs"
|
|
}
|
|
|
|
// IsSafeBot 判断是否是安全bot
|
|
// 返回: true表示是安全bot（是bot且风险等级为0），false表示不是
|
|
func (w *WebLog) IsSafeBot() bool {
|
|
return w.IsBot == 1 && w.RISK_LEVEL == 0
|
|
}
|
|
|
|
// GetIPFailureCount 获取IP在指定时间窗口内的失败次数（用于规则引擎）
|
|
// minutes: 时间窗口（分钟）
|
|
// 返回: 失败次数
|
|
func (w *WebLog) GetIPFailureCount(minutes int64) int64 {
|
|
if w.SRC_IP == "" {
|
|
return 0
|
|
}
|
|
|
|
// 如果是bot且危险程度是0，不统计失败次数
|
|
if w.IsSafeBot() {
|
|
return 0
|
|
}
|
|
|
|
// 如果是证书申请路径，不统计失败次数
|
|
if getSSLChallengePath != nil {
|
|
sslPath := getSSLChallengePath()
|
|
if sslPath != "" && strings.HasPrefix(w.URL, sslPath) {
|
|
return 0
|
|
}
|
|
}
|
|
|
|
// 直接调用IP失败管理器（延迟导入避免编译时循环依赖）
|
|
return getIPFailureCount(w.SRC_IP, minutes)
|
|
}
|
|
|
|
// RecordIPFailureThreshold 记录IP失败封禁的阈值信息（当规则匹配时调用）
|
|
// minutes: 触发封禁的时间窗口（分钟）
|
|
// count: 触发封禁的失败次数阈值
|
|
func (w *WebLog) RecordIPFailureThreshold(minutes int64, count int64) {
|
|
if w.SRC_IP == "" {
|
|
return
|
|
}
|
|
|
|
// 如果是bot且危险程度是0，不记录阈值
|
|
if w.IsSafeBot() {
|
|
return
|
|
}
|
|
|
|
// 如果是证书申请路径，不记录阈值
|
|
if getSSLChallengePath != nil {
|
|
sslPath := getSSLChallengePath()
|
|
if sslPath != "" && strings.HasPrefix(w.URL, sslPath) {
|
|
return
|
|
}
|
|
}
|
|
|
|
// 调用IP失败管理器记录阈值（延迟导入避免编译时循环依赖）
|
|
if recordIPFailureThreshold != nil {
|
|
recordIPFailureThreshold(w.SRC_IP, minutes, count)
|
|
}
|
|
}
|
|
|
|
// getSSLChallengePath 获取SSL证书验证路径（延迟导入避免循环依赖）
|
|
var getSSLChallengePath func() string
|
|
|
|
// SetSSLChallengePathGetter 设置SSL证书验证路径获取函数
|
|
func SetSSLChallengePathGetter(fn func() string) {
|
|
getSSLChallengePath = fn
|
|
}
|
|
|
|
// getIPFailureCount 获取IP失败次数（通过函数变量实现延迟导入）
|
|
var getIPFailureCount func(string, int64) int64
|
|
|
|
// SetIPFailureCountGetter 设置IP失败次数获取函数
|
|
func SetIPFailureCountGetter(fn func(string, int64) int64) {
|
|
getIPFailureCount = fn
|
|
}
|
|
|
|
// recordIPFailureThreshold 记录IP失败封禁阈值（通过函数变量实现延迟导入）
|
|
var recordIPFailureThreshold func(string, int64, int64)
|
|
|
|
// SetIPFailureThresholdRecorder 设置IP失败封禁阈值记录函数
|
|
func SetIPFailureThresholdRecorder(fn func(string, int64, int64)) {
|
|
recordIPFailureThreshold = fn
|
|
}
|
|
|
|
type WAFLog struct {
|
|
REQ_UUID string `json:"req_uuid"`
|
|
ACTION string `json:"action"`
|
|
RULE string `json:"rule"`
|
|
CREATE_TIME string `json:"create_time"`
|
|
USER_CODE string `json:"user_code"`
|
|
}
|