From 6295560216a1ec5e7d9d9660c76204b289b5f503 Mon Sep 17 00:00:00 2001
From: Terrtia <or1994@hotmail.fr>
Date: Tue, 23 Jul 2019 17:09:29 +0200
Subject: [PATCH] chg: [misp_modules] add enrichment request

---
 bin/packages/MispModules.py | 14 ++++++++++----
 1 file changed, 10 insertions(+), 4 deletions(-)

diff --git a/bin/packages/MispModules.py b/bin/packages/MispModules.py
index 214ba925..f57caa70 100755
--- a/bin/packages/MispModules.py
+++ b/bin/packages/MispModules.py
@@ -31,10 +31,16 @@ def init_module_config(module_json, config, config_path=default_config_path):
             print(module_json['name'])
     return config
 
-def misp_module_enrichement(misp_module_url, misp_module_port, request_content):
+def build_enrichment_request_json(module_name, var_name, var_value):
+    request_dict = {'module': module_name, var_name: var_value}
+    # # TODO: add error handler
+    return json.dumps(request_dict)
+
+def misp_module_enrichement_request(misp_module_url, misp_module_port, request_content):
     endpoint_url = '{}:{}/query'.format(misp_module_url, misp_module_port)
     req = requests.post(endpoint_url, headers={'Content-Type': 'application/json'}, data=request_content)
-    print(req.json())
+    if req.status_code == 200:
+        print(req.json())
 
 if __name__ == "__main__":
     req = requests.get('{}/modules'.format(misp_module_url))
@@ -60,8 +66,8 @@ if __name__ == "__main__":
 
         misp_module_url = 'http://localhost'
         misp_module_port = 6666
-        test_content = json.dumps({'module': 'btc_steroids', 'btc': '1hmZdUYHyqH3DmWyduRRW3HT8Vm6PHsD1'})
-        misp_module_enrichement(misp_module_url, misp_module_port, test_content)
+        test_content = build_enrichment_request_json('btc_steroids', 'btc', 'btc address')
+        misp_module_enrichement_request(misp_module_url, misp_module_port, test_content)
 
 
     else: