From 3f373e6ca3e799b26319940433e9bf167b3e5566 Mon Sep 17 00:00:00 2001
From: Kieran <kieran@harkin.me>
Date: Fri, 3 Nov 2023 01:24:55 +0900
Subject: [PATCH] Fix origin check

---
 VoidCat/Controllers/DownloadController.cs | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/VoidCat/Controllers/DownloadController.cs b/VoidCat/Controllers/DownloadController.cs
index 8b1a525..b1e5b54 100644
--- a/VoidCat/Controllers/DownloadController.cs
+++ b/VoidCat/Controllers/DownloadController.cs
@@ -111,7 +111,7 @@ public class DownloadController : Controller
 
     private async Task<VoidFileResponse?> SetupDownload(Guid id)
     {
-        var origin = Request.Headers.Origin.FirstOrDefault();
+        var origin = Request.Headers.Referer.FirstOrDefault() ?? Request.Headers.Origin.FirstOrDefault();
         if (!string.IsNullOrEmpty(origin) && Uri.TryCreate(origin, UriKind.RelativeOrAbsolute, out var u))
         {
             if (_settings.BlockedOrigins.Any(a => string.Equals(a, u.DnsSafeHost, StringComparison.InvariantCultureIgnoreCase)))