From: rinpatch <rinpatch@sdf.org>
Date: Thu, 14 Mar 2019 22:36:29 +0000 (+0300)
Subject: MediaProxy: parse filename from content-disposition for non-whitelisted types
X-Git-Url: https://git.squeep.com/?a=commitdiff_plain;h=958227d5563d76f4f983b7cabb6948897d93bd4b;p=akkoma

MediaProxy: parse filename from content-disposition for non-whitelisted types
---

diff --git a/lib/pleroma/reverse_proxy.ex b/lib/pleroma/reverse_proxy.ex
index 6298b92f4..39ede8619 100644
--- a/lib/pleroma/reverse_proxy.ex
+++ b/lib/pleroma/reverse_proxy.ex
@@ -311,7 +311,25 @@ defmodule Pleroma.ReverseProxy do
       end
 
     if attachment? do
-      disposition = "attachment; filename=" <> Keyword.get(opts, :attachment_name, "attachment")
+      name =
+        try do
+          {{"content-disposition", content_disposition_string}, _} =
+            List.keytake(headers, "content-disposition", 0)
+
+          [name] =
+            Regex.run(
+              ~r/filename=\"(.*)\"/u,
+              content_disposition_string || "",
+              capture: :all_but_first
+            )
+
+          name
+        rescue
+          MatchError -> Keyword.get(opts, :attachment_name, "attachment")
+        end
+
+      disposition = "attachment; filename=" <> name
+
       List.keystore(headers, "content-disposition", 0, {"content-disposition", disposition})
     else
       headers