From c565dd5ece1620d4a0666c33c709c8cc492bca25 Mon Sep 17 00:00:00 2001 From: ston1th Date: Thu, 23 Mar 2023 00:14:26 +0100 Subject: [PATCH] fixed client ip --- pkg/api/types/clientip.go | 8 +++++++- pkg/api/types/clientip_openbsd.go | 5 ++++- 2 files changed, 11 insertions(+), 2 deletions(-) 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 }