diff --git a/main.go b/main.go index 78ae39e..2ef667c 100644 --- a/main.go +++ b/main.go @@ -47,9 +47,12 @@ func handleRoot(w http.ResponseWriter, r *http.Request) { return } - host, _, err := net.SplitHostPort(r.RemoteAddr) - if err != nil { - host = r.RemoteAddr + host := r.Header.Get("X-Real-IP") + if host == "" { + host = r.Header.Get("X-Forwarded-For") + } + if host == "" { + host, _, _ = net.SplitHostPort(r.RemoteAddr) } dox, err := callApi(host)