From: href <href@random.sh>
Date: Mon, 4 Feb 2019 16:44:41 +0000 (+0100)
Subject: Mime: detect RIFF formats (wave, webp, avi)
X-Git-Url: https://git.squeep.com/?a=commitdiff_plain;h=127f99ae18eea7732e8cb7756a102ca4c5beacca;p=akkoma

Mime: detect RIFF formats (wave, webp, avi)
---

diff --git a/lib/pleroma/mime.ex b/lib/pleroma/mime.ex
index 84fb536e0..36771533f 100644
--- a/lib/pleroma/mime.ex
+++ b/lib/pleroma/mime.ex
@@ -102,10 +102,18 @@ defmodule Pleroma.MIME do
     "audio/ogg"
   end
 
-  defp check_mime_type(<<0x52, 0x49, 0x46, 0x46, _::binary>>) do
+  defp check_mime_type(<<"RIFF", _::binary-size(4), "WAVE", _::binary>>) do
     "audio/wav"
   end
 
+  defp check_mime_type(<<"RIFF", _::binary-size(4), "WEBP", _::binary>>) do
+    "image/webp"
+  end
+
+  defp check_mime_type(<<"RIFF", _::binary-size(4), "AVI.", _::binary>>) do
+    "video/avi"
+  end
+
   defp check_mime_type(_) do
     @default
   end