diff --git a/pkg/api/types/clientip.go b/pkg/api/types/clientip.go index c5eb6a2..9a68f63 100644 --- a/pkg/api/types/clientip.go +++ b/pkg/api/types/clientip.go @@ -5,6 +5,12 @@ package types +import "net" + func clientip(ip string) string { - return ip + if ip == "@" { + return ip + } + host, _, _ := net.SplitHostPort(ip) + return host } diff --git a/pkg/api/types/clientip_openbsd.go b/pkg/api/types/clientip_openbsd.go index 7f857d7..90c0812 100644 --- a/pkg/api/types/clientip_openbsd.go +++ b/pkg/api/types/clientip_openbsd.go @@ -5,9 +5,12 @@ package types +import "net" + func clientip(ip string) string { if ip == "" { return "@" } - return ip + host, _, _ := net.SplitHostPort(ip) + return host }