him411/UA3F
1
0
Fork 0
mirror of https://github.com/SunBK201/UA3F.git synced 2025-12-16 08:44:29 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: ensure client connection is closed properly in HandleClient method

Browse Source
This commit is contained in:
SunBK201 2025-10-28 15:47:33 +08:00
parent 9f70d0f0ac
commit a3bf90130e
1 changed files with 2 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
src/internal/server/socks5/socks5.go
View File

@ -70,9 +70,10 @@ func (s *Server) Start() (err error) {
// handleClient performs SOCKS5 negotiation and dispatches TCP/UDP handling.
func (s *Server) HandleClient(client net.Conn) {
defer client.Close()
// Handshake (no auth)
if err := s.socks5Auth(client); err != nil {
_ = client.Close()
return
}
@ -81,12 +82,10 @@ func (s *Server) HandleClient(client net.Conn) {
if cmd == socksCmdUDP {
// UDP Associate
s.handleUDPAssociate(client)
_ = client.Close()
return
}
logrus.Debugf("[%s][%s] ParseSocks5Request failed: %s",
client.RemoteAddr().String(), destAddrPort, err.Error())
_ = client.Close()
return
}
@ -94,7 +93,6 @@ func (s *Server) HandleClient(client net.Conn) {
target, err := s.socks5Connect(client, destAddrPort)
if err != nil {
logrus.Debug("Connect failed: ", err)
_ = client.Close()
return
}
s.forwardTCP(client, target, destAddrPort)