Fixed bug "too many colons in address".

2024-11-04 15:13:12 +07:00 · 2024-11-04 15:13:12 +07:00 · ffbd6b14fa
commit ffbd6b14fa
parent 13a9bd1aa1
1 changed files with 4 additions and 2 deletions
--- a/server.go
+++ b/server.go
@ -144,9 +144,11 @@ func (s *Server) handleConnection(ctx context.Context, conn net.Conn) {
 		if err != nil {
 			fields[log.FnError] = err.Error()
 			_ = s.logger.Warn("peekHTTP failed", fields)
-		} else {
-			if err == nil && host != "" {
+		} else if host != "" {
+			if !strings.Contains(host, ":") {
 				addr = host + addr[strings.Index(addr, ":"):]
+			} else {
+				addr = host
 			}
 		}
 		reader = reader_n3