From 4f5059fca43000a19f1aaf90b27ee934976774ab Mon Sep 17 00:00:00 2001
From: Hannah Ward <Hannah.ward9001@gmail.com>
Date: Fri, 12 Aug 2016 14:45:28 +0100
Subject: [PATCH] Added lookup by country code

---
 misp_modules/modules/expansion/countrycode.py | 59 +++++++++++++++++++
 1 file changed, 59 insertions(+)
 create mode 100755 misp_modules/modules/expansion/countrycode.py

diff --git a/misp_modules/modules/expansion/countrycode.py b/misp_modules/modules/expansion/countrycode.py
new file mode 100755
index 0000000..9fc0bfd
--- /dev/null
+++ b/misp_modules/modules/expansion/countrycode.py
@@ -0,0 +1,59 @@
+import json
+import requests
+
+misperrors = {'error': 'Error'}
+mispattributes = {'input': ['hostname', 'domain'], 'output': ['ip-src',
+                                                              'ip-dst']}
+
+# possible module-types: 'expansion', 'hover' or both
+moduleinfo = {'version': '1', 'author': 'Hannah Ward',
+              'description': 'Expand Country Codes',
+              'module-type': ['expansion', '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)"
+              }
+
+def handler(q=False):
+    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
+      codes = requests.get("http://www.geognos.com/api/en/countries/info/all.json").json()
+
+      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
+