From abb2d5049417bcb29d1093cdf4dae55f2ea722d9 Mon Sep 17 00:00:00 2001
From: Hugo <10965479+hugoghx@users.noreply.github.com>
Date: Fri, 16 Aug 2024 14:03:41 +0100
Subject: [PATCH] fix(worker): Use 16-bit int parsing for TCP port

---
 internal/daemon/worker/handler.go | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/internal/daemon/worker/handler.go b/internal/daemon/worker/handler.go
index 75bcd102e7..90e743e396 100644
--- a/internal/daemon/worker/handler.go
+++ b/internal/daemon/worker/handler.go
@@ -100,7 +100,7 @@ func (w *Worker) handleProxy(listenerCfg *listenerutil.ListenerConfig, sessionMa
 			wr.WriteHeader(http.StatusInternalServerError)
 			return
 		}
-		numPort, err := strconv.Atoi(clientPort)
+		numPort, err := strconv.ParseUint(clientPort, 10, 16)
 		if err != nil {
 			event.WriteError(ctx, op, err, event.WithInfoMsg("unable to understand remote port"))
 			wr.WriteHeader(http.StatusInternalServerError)
@@ -108,7 +108,7 @@ func (w *Worker) handleProxy(listenerCfg *listenerutil.ListenerConfig, sessionMa
 		}
 		clientAddr := &net.TCPAddr{
 			IP:   net.ParseIP(clientIp),
-			Port: numPort,
+			Port: int(numPort),
 		}
 
 		userClientIp, err := common.ClientIpFromRequest(ctx, listenerCfg, r)