From eb2dcca12b853ead10856b7de15b4d0fff4209f2 Mon Sep 17 00:00:00 2001
From: Sascha Rommelfangen <sascha.rommelfangen@circl.lu>
Date: Thu, 14 Mar 2019 14:39:58 +0100
Subject: [PATCH] fixed a bug when checking malformed BTC addresses

---
 misp_modules/modules/expansion/btc_steroids.py | 16 +++++++++++++---
 1 file changed, 13 insertions(+), 3 deletions(-)

diff --git a/misp_modules/modules/expansion/btc_steroids.py b/misp_modules/modules/expansion/btc_steroids.py
index 7011eda..5958502 100755
--- a/misp_modules/modules/expansion/btc_steroids.py
+++ b/misp_modules/modules/expansion/btc_steroids.py
@@ -91,6 +91,7 @@ def mprint(input):
 def handler(q=False):
     global result_text
     global conversion_rates
+    result_text = ""
     # start_time = time.time()
     # now = time.time()
     if q is False:
@@ -105,7 +106,6 @@ def handler(q=False):
         btc = request['btc']
     else:
         return False
-
     mprint("\nAddress:\t" + btc)
     try:
         req = requests.get(blockchain_all.format(btc, "&limit=50"))
@@ -113,8 +113,18 @@ def handler(q=False):
     except Exception:
         # print(e)
         print(req.text)
-        result_text = ""
-        sys.exit(1)
+        result_text = "Not a valid BTC address"
+        r = {
+            'results': [
+                {
+                    'types': ['text'],
+                    'values':[
+                        str(result_text)
+                    ]
+                }
+            ]
+        }
+        return r
 
     n_tx = jreq['n_tx']
     balance = float(jreq['final_balance'] / 100000000)