Browse Source

Remove port when using headers in c.RealIP

Co-authored-by: Elwin Tamminga <elwintamminga@gmail.com>
Co-committed-by: Elwin Tamminga <elwintamminga@gmail.com>
master v0.2.1
Elwin Tamminga 5 months ago
committed by Nise Void
parent
commit
283c75b32e
  1. 15
      context.go

15
context.go

@ -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
}

Loading…
Cancel
Save