From 006247ca7ec618d2aff02052bac839ca769991a1 Mon Sep 17 00:00:00 2001
From: Konrad Zemek <konrad.zemek@gmail.com>
Date: Mon, 20 Nov 2023 13:12:13 +0000
Subject: [PATCH] Fix parsing addresses from PROXY headers.
---
tcp.go | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/tcp.go b/tcp.go
index 878cfde..2a297ca 100644
--- a/tcp.go
+++ b/tcp.go
@@ -52,10 +52,10 @@ func tcpHandleConnection(conn net.Conn, logger *slog.Logger) {
targetAddr := Opts.TargetAddr6
if saddr == nil {
- if netip.MustParseAddr(conn.RemoteAddr().String()).Is4() {
+ if netip.MustParseAddrPort(conn.RemoteAddr().String()).Addr().Is4() {
targetAddr = Opts.TargetAddr4
}
- } else if netip.MustParseAddr(saddr.String()).Is4() {
+ } else if netip.MustParseAddrPort(saddr.String()).Addr().Is4() {
targetAddr = Opts.TargetAddr4
}
--
GitLab