git.squeep.com Git - akkoma/blob - lib/pleroma/web/media_proxy/controller.ex

   1 defmodule Pleroma.Web.MediaProxy.MediaProxyController do
   2   use Pleroma.Web, :controller
   3   alias Pleroma.{Web.MediaProxy, ReverseProxy}
   4
   5   @default_proxy_opts [max_body_length: 25 * 1_048_576, http: [follow_redirect: true]]
   6
   7   def remote(conn, params = %{"sig" => sig64, "url" => url64}) do
   8     with config <- Pleroma.Config.get([:media_proxy], []),
   9          true <- Keyword.get(config, :enabled, false),
  10          {:ok, url} <- MediaProxy.decode_url(sig64, url64),
  11          :ok <- filename_matches(Map.has_key?(params, "filename"), conn.request_path, url) do
  12       ReverseProxy.call(conn, url, Keyword.get(config, :proxy_opts, @default_proxy_opts))
  13     else
  14       false ->
  15         send_resp(conn, 404, Plug.Conn.Status.reason_phrase(404))
  16
  17       {:error, :invalid_signature} ->
  18         send_resp(conn, 403, Plug.Conn.Status.reason_phrase(403))
  19
  20       {:wrong_filename, filename} ->
  21         redirect(conn, external: MediaProxy.build_url(sig64, url64, filename))
  22     end
  23   end
  24
  25   def filename_matches(has_filename, path, url) do
  26     filename =
  27       url
  28       |> MediaProxy.filename()
  29       |> URI.decode()
  30
  31     path = URI.decode(path)
  32
  33     if has_filename && filename && Path.basename(path) != filename do
  34       {:wrong_filename, filename}
  35     else
  36       :ok
  37     end
  38   end
  39 end