From 0ada07400d46578e56c87761064d21cdc8abfe9d Mon Sep 17 00:00:00 2001
From: bagas <bagas@fossy.my.id>
Date: Mon, 21 Jul 2025 14:19:58 +0700
Subject: [PATCH] refactor: remove unuse variable and channel

---
 session/handler.go | 31 +++++++++++++++----------------
 1 file changed, 15 insertions(+), 16 deletions(-)

diff --git a/session/handler.go b/session/handler.go
index 0547299..542c9a4 100644
--- a/session/handler.go
+++ b/session/handler.go
@@ -177,22 +177,6 @@ func (s *Session) handleTCPIPForward(req *ssh.Request) {
 		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")
 
 	showWelcomeMessage(s.ConnChannel)
@@ -202,6 +186,21 @@ func (s *Session) handleTCPIPForward(req *ssh.Request) {
 		s.handleHTTPForward(req, portToBind)
 		return
 	} 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)
 	}