Added a check for the port before calling the splitHostPort function (#3372)

Co-authored-by: Aziz Rmadi <azizrmadi@Azizs-MacBook-Air.local>
This commit is contained in:
armadi1809
2023-10-18 17:01:03 -07:00
committed by GitHub
co-authored by Aziz Rmadi
parent 47d5b3c92f
commit 6b27d850d8
+3 -1
View File
@@ -26,6 +26,7 @@ func GetIPAddressFromRequest(req *http.Request) string {
xForwardedFor := req.Header.Get("X-FORWARDED-FOR")
if xForwardedFor != "" {
clientIpString := strings.Split(xForwardedFor, ", ")[0]
if strings.Contains(clientIpString, ":") {
ip, _, err := net.SplitHostPort(clientIpString)
if err != nil {
log.Errorln(err)
@@ -33,7 +34,8 @@ func GetIPAddressFromRequest(req *http.Request) string {
}
return ip
}
return clientIpString
}
ip, _, err := net.SplitHostPort(ipAddressString)
if err != nil {
log.Errorln(err)