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
|
||||
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…
Reference in New Issue
Block a user