TRKJ/EdgeNode
2
0
Fork 0
mirror of https://github.com/TeaOSLab/EdgeNode.git synced 2025-11-04 07:40:56 +08:00
Code Issues Packages Projects Releases Wiki Activity
Files
bbc892ad4b8e57bccfdb63191693c66db33960da
EdgeNode/internal/waf/action_js_cookie.go

160 lines
4.7 KiB
Go
Raw Normal View History

增加Javascript Cookie验证
2022-08-25 15:35:32 +08:00
// Copyright 2022 Liuxiangchao iwind.liu@gmail.com. All rights reserved. Official site: https://goedge.cn .
package waf
import (
"crypto/md5"
"fmt"
"github.com/TeaOSLab/EdgeCommon/pkg/nodeconfigs"
"github.com/TeaOSLab/EdgeCommon/pkg/serverconfigs/firewallconfigs"
合并多个计数器，便于统一的内存控制
2023-10-05 09:45:46 +08:00
"github.com/TeaOSLab/EdgeNode/internal/utils/counters"
增加Javascript Cookie验证
2022-08-25 15:35:32 +08:00
"github.com/TeaOSLab/EdgeNode/internal/waf/requests"
"github.com/iwind/TeaGo/types"
"net/http"
WAF优化captcha和js_cookie的失败计数器/增强js_cookie的安全性
2022-08-25 17:06:52 +08:00
"strings"
增加Javascript Cookie验证
2022-08-25 15:35:32 +08:00
"time"
)
type JSCookieAction struct {
BaseAction
Life int32 `yaml:"life" json:"life"`
MaxFails int `yaml:"maxFails" json:"maxFails"` // 最大失败次数
FailBlockTimeout int `yaml:"failBlockTimeout" json:"failBlockTimeout"` // 失败拦截时间
Scope string `yaml:"scope" json:"scope"`
优化WAF * 信息加密使用struct代替map，以缩短加密后内容长度 * 拦截动作、人机识别动作增加是否尝试全局封禁选项 * JSCookie识别动作增加默认设置选项 * 人机识别中传入info参数异常时，尝试跳转到来源地址，避免直接提示invalid request
2024-04-07 14:31:22 +08:00
FailBlockScopeAll bool `yaml:"failBlockScopeAll" json:"failBlockScopeAll"`
增加Javascript Cookie验证
2022-08-25 15:35:32 +08:00
}
func (this *JSCookieAction) Init(waf *WAF) error {
优化WAF * 信息加密使用struct代替map，以缩短加密后内容长度 * 拦截动作、人机识别动作增加是否尝试全局封禁选项 * JSCookie识别动作增加默认设置选项 * 人机识别中传入info参数异常时，尝试跳转到来源地址，避免直接提示invalid request
2024-04-07 14:31:22 +08:00
if waf.DefaultJSCookieAction != nil {
if this.Life <= 0 {
this.Life = waf.DefaultJSCookieAction.Life
}
if this.MaxFails <= 0 {
this.MaxFails = waf.DefaultJSCookieAction.MaxFails
}
if this.FailBlockTimeout <= 0 {
this.FailBlockTimeout = waf.DefaultJSCookieAction.FailBlockTimeout
}
if len(this.Scope) == 0 {
this.Scope = waf.DefaultJSCookieAction.Scope
}
this.FailBlockScopeAll = waf.DefaultJSCookieAction.FailBlockScopeAll
}
if len(this.Scope) == 0 {
this.Scope = firewallconfigs.FirewallScopeGlobal
}
增加Javascript Cookie验证
2022-08-25 15:35:32 +08:00
return nil
}
func (this *JSCookieAction) Code() string {
return ActionJavascriptCookie
}
func (this *JSCookieAction) IsAttack() bool {
return false
}
func (this *JSCookieAction) WillChange() bool {
return true
}
WAF允许动作默认跳过所有规则
2024-01-20 20:54:41 +08:00
func (this *JSCookieAction) Perform(waf *WAF, group *RuleGroup, set *RuleSet, req requests.Request, writer http.ResponseWriter) PerformResult {
增加Javascript Cookie验证
2022-08-25 15:35:32 +08:00
// 是否在白名单中
if SharedIPWhiteList.Contains("set:"+types.String(set.Id), this.Scope, req.WAFServerId(), req.WAFRemoteIP()) {
WAF允许动作默认跳过所有规则
2024-01-20 20:54:41 +08:00
return PerformResult{
ContinueRequest: true,
}
增加Javascript Cookie验证
2022-08-25 15:35:32 +08:00
}
nodeConfig, err := nodeconfigs.SharedNodeConfig()
if err != nil {
WAF允许动作默认跳过所有规则
2024-01-20 20:54:41 +08:00
return PerformResult{
ContinueRequest: true,
}
增加Javascript Cookie验证
2022-08-25 15:35:32 +08:00
}
var life = this.Life
if life <= 0 {
life = 3600
}
// 检查Cookie
var cookieName = "ge_js_validator_" + types.String(set.Id)
cookie, err := req.WAFRaw().Cookie(cookieName)
if err == nil && cookie != nil {
var cookieValue = cookie.Value
if len(cookieValue) > 10 {
WAF优化captcha和js_cookie的失败计数器/增强js_cookie的安全性
2022-08-25 17:06:52 +08:00
var pieces = strings.Split(cookieValue, "@")
if len(pieces) == 3 {
var timestamp = pieces[0]
var sum = pieces[2]
if types.Int64(timestamp) >= time.Now().Unix()-int64(life) && fmt.Sprintf("%x", md5.Sum([]byte(timestamp+"@"+types.String(set.Id)+"@"+nodeConfig.NodeId))) == sum {
WAF允许动作默认跳过所有规则
2024-01-20 20:54:41 +08:00
return PerformResult{
ContinueRequest: true,
}
WAF优化captcha和js_cookie的失败计数器/增强js_cookie的安全性
2022-08-25 17:06:52 +08:00
}
增加Javascript Cookie验证
2022-08-25 15:35:32 +08:00
}
}
}
部分WAF动作输出内容时增加自定义报头
2023-06-12 18:07:07 +08:00
req.ProcessResponseHeaders(writer.Header(), http.StatusOK)
增加Javascript Cookie验证
2022-08-25 15:35:32 +08:00
writer.Header().Set("Content-Type", "text/html; charset=utf-8")
writer.Header().Set("Cache-Control", "no-cache")
var timestamp = types.String(time.Now().Unix())
WAF优化captcha和js_cookie的失败计数器/增强js_cookie的安全性
2022-08-25 17:06:52 +08:00
var cookieValue = timestamp + "@" + types.String(set.Id) + "@" + fmt.Sprintf("%x", md5.Sum([]byte(timestamp+"@"+types.String(set.Id)+"@"+nodeConfig.NodeId)))
部分WAF动作输出内容时增加自定义报头
2023-06-12 18:07:07 +08:00
var respHTML = `<!DOCTYPE html>
增加Javascript Cookie验证
2022-08-25 15:35:32 +08:00
<html>
<head>
<title></title>
<meta charset="UTF-8"/>
<script type="text/javascript">
document.cookie = "` + cookieName + `=` + cookieValue + `; path=/; max-age=` + types.String(life) + `;";
window.location.reload();
</script>
</head>
<body>
</body>
部分WAF动作输出内容时增加自定义报头
2023-06-12 18:07:07 +08:00
</html>`
writer.Header().Set("Content-Length", types.String(len(respHTML)))
writer.WriteHeader(http.StatusOK)
_, _ = writer.Write([]byte(respHTML))
增加Javascript Cookie验证
2022-08-25 15:35:32 +08:00
// 记录失败次数
优化WAF * 信息加密使用struct代替map，以缩短加密后内容长度 * 拦截动作、人机识别动作增加是否尝试全局封禁选项 * JSCookie识别动作增加默认设置选项 * 人机识别中传入info参数异常时，尝试跳转到来源地址，避免直接提示invalid request
2024-04-07 14:31:22 +08:00
this.increaseFails(req, waf.Id, group.Id, set.Id, waf.UseLocalFirewall && (this.FailBlockScopeAll || this.Scope == firewallconfigs.FirewallScopeGlobal))
增加Javascript Cookie验证
2022-08-25 15:35:32 +08:00
WAF允许动作默认跳过所有规则
2024-01-20 20:54:41 +08:00
return PerformResult{}
增加Javascript Cookie验证
2022-08-25 15:35:32 +08:00
}
优化WAF * 信息加密使用struct代替map，以缩短加密后内容长度 * 拦截动作、人机识别动作增加是否尝试全局封禁选项 * JSCookie识别动作增加默认设置选项 * 人机识别中传入info参数异常时，尝试跳转到来源地址，避免直接提示invalid request
2024-04-07 14:31:22 +08:00
func (this *JSCookieAction) increaseFails(req requests.Request, policyId int64, groupId int64, setId int64, useLocalFirewall bool) (goNext bool) {
增加Javascript Cookie验证
2022-08-25 15:35:32 +08:00
var maxFails = this.MaxFails
var failBlockTimeout = this.FailBlockTimeout
if maxFails <= 0 {
maxFails = 10 // 默认10次
优化WAF * 信息加密使用struct代替map，以缩短加密后内容长度 * 拦截动作、人机识别动作增加是否尝试全局封禁选项 * JSCookie识别动作增加默认设置选项 * 人机识别中传入info参数异常时，尝试跳转到来源地址，避免直接提示invalid request
2024-04-07 14:31:22 +08:00
} else if maxFails <= 5 {
maxFails = 5 // 不能小于3，防止意外刷新出现
增加Javascript Cookie验证
2022-08-25 15:35:32 +08:00
}
if failBlockTimeout <= 0 {
failBlockTimeout = 1800 // 默认1800s
}
合并多个计数器，便于统一的内存控制
2023-10-05 09:45:46 +08:00
var key = "WAF:JS_COOKIE:FAILS:" + req.WAFRemoteIP() + ":" + types.String(req.WAFServerId()) + ":" + req.WAFRaw().URL.String()
增加Javascript Cookie验证
2022-08-25 15:35:32 +08:00
合并多个计数器，便于统一的内存控制
2023-10-05 09:45:46 +08:00
var countFails = counters.SharedCounter.IncreaseKey(key, 300)
增加Javascript Cookie验证
2022-08-25 15:35:32 +08:00
if int(countFails) >= maxFails {
优化WAF * 信息加密使用struct代替map，以缩短加密后内容长度 * 拦截动作、人机识别动作增加是否尝试全局封禁选项 * JSCookie识别动作增加默认设置选项 * 人机识别中传入info参数异常时，尝试跳转到来源地址，避免直接提示invalid request
2024-04-07 14:31:22 +08:00
SharedIPBlackList.RecordIP(IPTypeAll, firewallconfigs.FirewallScopeServer, req.WAFServerId(), req.WAFRemoteIP(), time.Now().Unix()+int64(failBlockTimeout), policyId, useLocalFirewall, groupId, setId, "JS_COOKIE验证连续失败超过"+types.String(maxFails)+"次")
增加Javascript Cookie验证
2022-08-25 15:35:32 +08:00
return false
}
return true
}
Reference in New Issue Copy Permalink