bagas/tunnel-please
Public Access
1
0
Fork 0
Code Issues 1 Pull Requests Actions Packages Projects Releases Wiki Activity

refactor: remove unuse variable and channel
All checks were successful
Docker Build and Push / build-and-push (push) Successful in 6m13s
Details

Browse Source
This commit is contained in:
bagas
2025-07-21 14:19:58 +07:00
parent 8c15da6131
commit 0ada07400d
1 changed files with 15 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

31
session/handler.go
View File

@ -177,22 +177,6 @@ func (s *Session) handleTCPIPForward(req *ssh.Request) {
return return
} }
if portToBind == 0 {
unassign, success := portUtil.Manager.GetUnassignedPort()
portToBind = unassign
if !success {
s.sendMessage(fmt.Sprintf("No available port\r\n", portToBind))
req.Reply(false, nil)
s.Close()
return
}
} else if isUse, isExist := portUtil.Manager.GetPortStatus(portToBind); !isExist || isUse {
s.sendMessage(fmt.Sprintf("Port %d is already in use or restricted. Please choose a different port. (03)\r\n", portToBind))
req.Reply(false, nil)
s.Close()
return
}
s.sendMessage("\033[H\033[2J") s.sendMessage("\033[H\033[2J")
showWelcomeMessage(s.ConnChannel) showWelcomeMessage(s.ConnChannel)
@ -202,6 +186,21 @@ func (s *Session) handleTCPIPForward(req *ssh.Request) {
s.handleHTTPForward(req, portToBind) s.handleHTTPForward(req, portToBind)
return return
} else { } else {
if portToBind == 0 {
unassign, success := portUtil.Manager.GetUnassignedPort()
portToBind = unassign
if !success {
s.sendMessage(fmt.Sprintf("No available port\r\n", portToBind))
req.Reply(false, nil)
s.Close()
return
}
} else if isUse, isExist := portUtil.Manager.GetPortStatus(portToBind); !isExist || isUse {
s.sendMessage(fmt.Sprintf("Port %d is already in use or restricted. Please choose a different port. (03)\r\n", portToBind))
req.Reply(false, nil)
s.Close()
return
}
portUtil.Manager.SetPortStatus(portToBind, true) portUtil.Manager.SetPortStatus(portToBind, true)
} }