Remove port when using headers in c.RealIP

Co-authored-by: Elwin Tamminga <elwintamminga@gmail.com>
Co-committed-by: Elwin Tamminga <elwintamminga@gmail.com>
This commit is contained in:
Elwin Tamminga 2021-07-20 10:49:55 +02:00 committed by Nise Void
parent 66ae2a435c
commit 283c75b32e
1 changed files with 9 additions and 6 deletions

View File

@ -105,14 +105,17 @@ func (c *Context) Get(key string) interface{} {
// RealIP uses proxy headers for the real ip, if none exist the IP of the current connection is returned
func (c *Context) RealIP() string {
reqIP := c.Request.RemoteAddr
if ip := c.Request.Header.Get(`X-Forwarded-For`); ip != `` {
return strings.Split(ip, `, `)[0]
reqIP = strings.Split(ip, `, `)[0]
} else if ip := c.Request.Header.Get(`X-Real-IP`); ip != `` {
reqIP = ip
}
if ip := c.Request.Header.Get(`X-Real-IP`); ip != `` {
return ip
ra, _, _ := net.SplitHostPort(reqIP)
if ra != `` {
reqIP = ra
}
ra, _, _ := net.SplitHostPort(c.Request.RemoteAddr)
return ra
return reqIP
}