Git of dcc - anni/blob - lib/pleroma/web/plugs/remote_ip.ex

   1 # Pleroma: A lightweight social networking server
   2 # Copyright © 2017-2022 Pleroma Authors <https://pleroma.social/>
   3 # SPDX-License-Identifier: AGPL-3.0-only
   4
   5 defmodule Pleroma.Web.Plugs.RemoteIp do
   6   @moduledoc """
   7   This is a shim to call [`RemoteIp`](https://git.pleroma.social/pleroma/remote_ip) but with runtime configuration.
   8   """
   9
  10   alias Pleroma.Config
  11   import Plug.Conn
  12
  13   @behaviour Plug
  14
  15   def init(_), do: nil
  16
  17   def call(%{remote_ip: original_remote_ip} = conn, _) do
  18     if Config.get([__MODULE__, :enabled]) do
  19       %{remote_ip: new_remote_ip} = conn = RemoteIp.call(conn, remote_ip_opts())
  20       assign(conn, :remote_ip_found, original_remote_ip != new_remote_ip)
  21     else
  22       conn
  23     end
  24   end
  25
  26   defp remote_ip_opts do
  27     headers = Config.get([__MODULE__, :headers], []) |> MapSet.new()
  28     reserved = Config.get([__MODULE__, :reserved], [])
  29
  30     proxies =
  31       Config.get([__MODULE__, :proxies], [])
  32       |> Enum.concat(reserved)
  33       |> Enum.map(&maybe_add_cidr/1)
  34
  35     {headers, proxies}
  36   end
  37
  38   defp maybe_add_cidr(proxy) when is_binary(proxy) do
  39     proxy =
  40       cond do
  41         "/" in String.codepoints(proxy) -> proxy
  42         InetCidr.v4?(InetCidr.parse_address!(proxy)) -> proxy <> "/32"
  43         InetCidr.v6?(InetCidr.parse_address!(proxy)) -> proxy <> "/128"
  44       end
  45
  46     InetCidr.parse_cidr!(proxy, true)
  47   end
  48 end