Fuyu/router
3
1
Fork 2
Code Issues Pull Requests 2 Releases Wiki Activity

Remove port when using headers in c.RealIP

Browse Source 
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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
context.go
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 // 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
} }