forked from AkkomaGang/akkoma
floatingghost
07a48b9293
Co-authored-by: FloatingGhost <hannah@coffee-and-dreams.uk> Reviewed-on: AkkomaGang/akkoma#371
50 lines
1.3 KiB
Elixir
50 lines
1.3 KiB
Elixir
# Pleroma: A lightweight social networking server
|
|
# Copyright © 2017-2021 Pleroma Authors <https://pleroma.social/>
|
|
# SPDX-License-Identifier: AGPL-3.0-only
|
|
|
|
defmodule Pleroma.Web.Plugs.RemoteIp do
|
|
@moduledoc """
|
|
This is a shim to call [`RemoteIp`](https://git.pleroma.social/pleroma/remote_ip) but with runtime configuration.
|
|
"""
|
|
|
|
alias Pleroma.Config
|
|
|
|
@behaviour Plug
|
|
|
|
def init(_), do: nil
|
|
|
|
def call(%{remote_ip: original_remote_ip} = conn, _) do
|
|
if Config.get([__MODULE__, :enabled]) do
|
|
{headers, proxies} = remote_ip_opts()
|
|
new_remote_ip = RemoteIp.from(conn.req_headers, headers: headers, proxies: proxies)
|
|
|
|
if new_remote_ip != original_remote_ip do
|
|
Map.put(conn, :remote_ip, new_remote_ip)
|
|
else
|
|
conn
|
|
end
|
|
else
|
|
conn
|
|
end
|
|
end
|
|
|
|
defp remote_ip_opts do
|
|
headers = Config.get([__MODULE__, :headers], [])
|
|
reserved = Config.get([__MODULE__, :reserved], [])
|
|
|
|
proxies =
|
|
Config.get([__MODULE__, :proxies], [])
|
|
|> Enum.concat(reserved)
|
|
|> Enum.map(&maybe_add_cidr/1)
|
|
|
|
{headers, proxies}
|
|
end
|
|
|
|
defp maybe_add_cidr(proxy) when is_binary(proxy) do
|
|
cond do
|
|
"/" in String.codepoints(proxy) -> proxy
|
|
InetCidr.v4?(InetCidr.parse_address!(proxy)) -> proxy <> "/32"
|
|
InetCidr.v6?(InetCidr.parse_address!(proxy)) -> proxy <> "/128"
|
|
end
|
|
end
|
|
end
|