librewolf
/
bunkerized-nginx
mirror of https://github.com/bunkerity/bunkerized-nginx
4
0
Fork 1
Code Issues Projects Releases Wiki Activity

lua - use pcall with mmdb functions

This commit is contained in:
bunkerity 2022-07-11 08:59:53 +02:00
parent 658ab75049
commit e62523d1dc
No known key found for this signature in database
GPG Key ID: 3D80806F12602A7C
1 changed files with 8 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
lua/utils.lua
View File

@ -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