TRKJ/EdgeNode
2
0
Fork 0
mirror of https://github.com/TeaOSLab/EdgeNode.git synced 2025-11-03 23:20:25 +08:00
Code Issues Packages Projects Releases Wiki Activity

优化代码

Browse Source
This commit is contained in:
GoEdgeLab
2023-06-02 14:23:54 +08:00
parent 287783926e
commit 8f1c7581db
1 changed files with 6 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
internal/nodes/client_conn_base.go
View File

@@ -54,12 +54,13 @@ func (this *BaseClientConn) SetServerId(serverId int64) (goNext bool) {
goNext = true goNext = true
// 检查服务相关IP黑名单 // 检查服务相关IP黑名单
if serverId > 0 && len(this.rawIP) > 0 { var rawIP = this.RawIP()
if serverId > 0 && len(rawIP) > 0 {
// 是否在白名单中 // 是否在白名单中
ok, _, expiresAt := iplibrary.AllowIP(this.rawIP, serverId) ok, _, expiresAt := iplibrary.AllowIP(rawIP, serverId)
if !ok { if !ok {
_ = this.rawConn.Close() _ = this.rawConn.Close()
firewalls.DropTemporaryTo(this.rawIP, expiresAt) firewalls.DropTemporaryTo(rawIP, expiresAt)
return false return false
} }
} }
@@ -123,8 +124,8 @@ func (this *BaseClientConn) TCPConn() (tcpConn *net.TCPConn, ok bool) {
switch conn := this.rawConn.(type) { switch conn := this.rawConn.(type) {
case *tls.Conn: case *tls.Conn:
var internalConn = conn.NetConn() var internalConn = conn.NetConn()
clientConn, ok := internalConn.(*ClientConn) clientConn, isClientConn := internalConn.(*ClientConn)
if ok { if isClientConn {
return clientConn.TCPConn() return clientConn.TCPConn()
} }
tcpConn, ok = internalConn.(*net.TCPConn) tcpConn, ok = internalConn.(*net.TCPConn)