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