From f30b7ed9e6b7c0e557ff0b70902fa8a492052c34 Mon Sep 17 00:00:00 2001
From: Rory& <root@rory.gay>
Date: Mon, 26 Feb 2024 19:09:43 +0100
Subject: Use X-Real-IP and X-Forwarded-For if available

---
 MatrixMediaGate/Services/AuthValidator.cs | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/MatrixMediaGate/Services/AuthValidator.cs b/MatrixMediaGate/Services/AuthValidator.cs
index 4b74006..28127ae 100644
--- a/MatrixMediaGate/Services/AuthValidator.cs
+++ b/MatrixMediaGate/Services/AuthValidator.cs
@@ -80,6 +80,15 @@ public class AuthValidator(ILogger<AuthValidator> logger, ProxyConfiguration cfg
             Console.WriteLine($"Authorized (ignore me) - Headers: {key}: {value}");
         }
 
+        // X-Real-IP X-Forwarded-For
+        if (ctx.Request.Headers.TryGetValue("X-Real-IP", out var xRealIp)) {
+            return xRealIp.ToString();
+        }
+
+        if (ctx.Request.Headers.TryGetValue("X-Forwarded-For", out var xForwardedFor)) {
+            return xForwardedFor.ToString();
+        }
+
         return ctx.Connection.RemoteIpAddress?.ToString();
     }
 }
\ No newline at end of file
-- 
cgit 1.5.1