Honor HTTP header 'X-Real-IP' when available

Fix #5.
tmate-io · Mar 2, 2021 · 62f8cfd · 62f8cfd
1 parent 628cd91
commit 62f8cfd
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/lib/tmate/ws_api/websocket.ex b/lib/tmate/ws_api/websocket.ex
@@ -26,6 +26,9 @@ defmodule Tmate.WsApi.WebSocket do
               :ok ->
                 ip = case req do
                   %{proxy_header: %{src_address: ip}} -> ip
+                  %{headers: %{"x-real-ip" => ipstring}} ->
+                    {_, ip} = :inet.parse_address(ipstring |> to_charlist)
+                    ip
                   %{peer: {ip, _port}} -> ip
                 end
                 ip = :inet_parse.ntoa(ip) |> to_string