forked from Fuyu/router
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:
parent
66ae2a435c
commit
283c75b32e
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
|
// RealIP uses proxy headers for the real ip, if none exist the IP of the current connection is returned
|
||||||
func (c *Context) RealIP() string {
|
func (c *Context) RealIP() string {
|
||||||
|
reqIP := c.Request.RemoteAddr
|
||||||
|
|
||||||
if ip := c.Request.Header.Get(`X-Forwarded-For`); ip != `` {
|
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 != `` {
|
ra, _, _ := net.SplitHostPort(reqIP)
|
||||||
return ip
|
if ra != `` {
|
||||||
|
reqIP = ra
|
||||||
}
|
}
|
||||||
|
return reqIP
|
||||||
ra, _, _ := net.SplitHostPort(c.Request.RemoteAddr)
|
|
||||||
return ra
|
|
||||||
}
|
}
|
||||||
|
Loading…
Reference in New Issue
Block a user