mirror of https://github.com/MISP/misp-modules
62 lines
1.6 KiB
Python
Executable File
62 lines
1.6 KiB
Python
Executable File
import json
|
|
import requests
|
|
|
|
misperrors = {'error': 'Error'}
|
|
mispattributes = {'input': ['hostname', 'domain']}
|
|
|
|
# possible module-types: 'expansion', 'hover' or both
|
|
moduleinfo = {'version': '1', 'author': 'Hannah Ward',
|
|
'description': 'Expand Country Codes',
|
|
'module-type': ['hover']}
|
|
|
|
# config fields that your code expects from the site admin
|
|
moduleconfig = []
|
|
|
|
common_tlds = {"com":"Commercial (Worldwide)",
|
|
"org":"Organisation (Worldwide)",
|
|
"net":"Network (Worldwide)",
|
|
"int":"International (Worldwide)",
|
|
"edu":"Education (Usually USA)",
|
|
"gov":"Government (USA)"
|
|
}
|
|
|
|
codes = False
|
|
|
|
def handler(q=False):
|
|
global codes
|
|
if not codes:
|
|
codes = requests.get("http://www.geognos.com/api/en/countries/info/all.json").json()
|
|
if q is False:
|
|
return False
|
|
request = json.loads(q)
|
|
domain = request["domain"]
|
|
|
|
# Get the extension
|
|
ext = domain.split(".")[-1]
|
|
|
|
# Check if if's a common, non country one
|
|
if ext in common_tlds.keys():
|
|
val = common_tlds[ext]
|
|
else:
|
|
# Retrieve a json full of country info
|
|
if not codes["StatusMsg"] == "OK":
|
|
val = "Unknown"
|
|
else:
|
|
# Find our code based on TLD
|
|
codes = codes["Results"]
|
|
for code in codes.keys():
|
|
if codes[code]["CountryCodes"]["tld"] == ext:
|
|
val = codes[code]["Name"]
|
|
r = {'results': [{'types':['text'], 'values':[val]}]}
|
|
return r
|
|
|
|
|
|
def introspection():
|
|
return mispattributes
|
|
|
|
|
|
def version():
|
|
moduleinfo['config'] = moduleconfig
|
|
return moduleinfo
|
|
|