remote_ip.ex
No OneTemporary
Actions

Size

1 KB

Referenced Files

None

Subscribers

None

remote_ip.ex
View Options

	# 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