TRKJ/EdgeNode
2
0
Fork 0
mirror of https://github.com/TeaOSLab/EdgeNode.git synced 2026-01-03 22:06:37 +08:00
Code Issues Packages Projects Releases Wiki Activity

优化连接关闭速度

Browse Source
This commit is contained in:
GoEdgeLab
2021-10-25 19:00:42 +08:00
parent 22ca557691
commit d8943f3816
5 changed files with 51 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
internal/nodes/http_request_waf.go
View File

@@ -20,22 +20,18 @@ func (this *HTTPRequest) doWAFRequest() (blocked bool) {
// 当前连接是否已关闭
var conn = this.RawReq.Context().Value(HTTPConnContextKey)
if conn != nil {
trafficConn, ok := conn.(*ClientConn)
if ok && trafficConn.IsClosed() {
if isClientConnClosed(conn.(net.Conn)) {
this.disableLog = true
return true
}
}
// 检查是否在临时黑名单中
if waf.SharedIPBlackList.Contains(waf.IPTypeAll, firewallconfigs.FirewallScopeService, this.Server.Id, this.WAFRemoteIP()) {
var remoteAddr = this.WAFRemoteIP()
if waf.SharedIPBlackList.Contains(waf.IPTypeAll, firewallconfigs.FirewallScopeService, this.Server.Id, remoteAddr) || waf.SharedIPBlackList.Contains(waf.IPTypeAll, firewallconfigs.FirewallScopeGlobal, 0, remoteAddr) {
this.disableLog = true
if conn != nil {
trafficConn, ok := conn.(*ClientConn)
if ok && !trafficConn.IsClosed() {
_ = trafficConn.Close()
}
_ = conn.(net.Conn).Close()
}
return true