fix: Add validation for country database

KitsuneLab-Development · Mar 28, 2024 · d6fda9c · d6fda9c
1 parent 442b815
commit d6fda9c
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/CHANGELOG b/CHANGELOG
@@ -1,6 +1,7 @@
 -- 2023.03.28 - V4.0.4
 
 - fix: Table generation problems because of lastseen default value
+- fix: Add validation for country database
 
 -- 2023.03.26 - V4.0.3
 

diff --git a/K4-System/src/Plugin/PluginStock.cs b/K4-System/src/Plugin/PluginStock.cs
@@ -589,6 +589,13 @@ public string GetPlayerCountryCode(CCSPlayerController player)
 			string[] parts = playerIp.Split(':');
 			string realIP = parts.Length == 2 ? parts[0] : playerIp;
 
+			string filePath = Path.Combine(ModuleDirectory, "GeoLite2-Country.mmdb");
+			if (!File.Exists(filePath))
+			{
+				Logger.LogError($"GeoLite2-Country.mmdb not found in {ModuleDirectory}. Download it from https://github.com/P3TERX/GeoLite.mmdb/releases and place it in the same directory as the plugin.");
+				return "??";
+			}
+
 			using DatabaseReader reader = new DatabaseReader(Path.Combine(ModuleDirectory, "GeoLite2-Country.mmdb"));
 			{
 				try