diff options
| -rw-r--r-- | http.go | 10 |
1 files changed, 9 insertions, 1 deletions
diff --git a/http.go b/http.go index 8ba0442..98e0cb5 100644 --- a/http.go +++ b/http.go @@ -8,12 +8,20 @@ import ( "strings" ) -// Attaches a request's IP address to the request's context +// Attaches a request's IP address to the request's context. +// If getwtxt is behind a reverse proxy, get the last entry +// in the X-Forwarded-For HTTP header as the user IP. func newCtxUserIP(ctx context.Context, r *http.Request) context.Context { base := strings.Split(r.RemoteAddr, ":") uip := base[0] + if _, ok := r.Header["X-Forwarded-For"]; ok { + proxied := r.Header["X-Forwarded-For"] + base = strings.Split(proxied[len(proxied)-1], ":") + uip = base[0] + } + return context.WithValue(ctx, ctxKey, uip) } |