From e62523d1dc550cbd13a72da4cbd7fd40e1a8dda9 Mon Sep 17 00:00:00 2001
From: bunkerity <contact@bunkerity.com>
Date: Mon, 11 Jul 2022 08:59:53 +0200
Subject: [PATCH] lua - use pcall with mmdb functions

---
 lua/utils.lua | 10 ++++++++--
 1 file changed, 8 insertions(+), 2 deletions(-)

diff --git a/lua/utils.lua b/lua/utils.lua
index d36586e28..c91615d81 100644
--- a/lua/utils.lua
+++ b/lua/utils.lua
@@ -317,7 +317,10 @@ utils.get_country = function(ip)
 	if not mmdb.country_db then
 		return false, "mmdb country not loaded"
 	end
-	local result, err = mmdb.country_db:lookup(ip)
+	local ok, result, err = pcall(mmdb.country_db:lookup(ip))
+	if not ok then
+		return nil, result
+	end
 	if not result then
 		return nil, err
 	end
@@ -328,7 +331,10 @@ utils.get_asn = function(ip)
 	if not mmdb.asn_db then
 		return false, "mmdb asn not loaded"
 	end
-	local result, err = mmdb.asn_db:lookup(ip)
+	local ok, result, err = pcall(mmdb.asn_db:lookup(ip))
+	if not ok then
+		return nil, result
+	end
 	if not result then
 		return nil, err
 	end