From 0965def6bfd9d4e6def23d8b442e7898238b534c Mon Sep 17 00:00:00 2001
From: Sebdraven <slarinier@gmail.com>
Date: Fri, 29 Jun 2018 17:22:19 +0200
Subject: [PATCH] add expand subdomains

---
 misp_modules/modules/expansion/dnstrails.py | 11 ++++++++++-
 1 file changed, 10 insertions(+), 1 deletion(-)

diff --git a/misp_modules/modules/expansion/dnstrails.py b/misp_modules/modules/expansion/dnstrails.py
index ffb4055..abef1e5 100644
--- a/misp_modules/modules/expansion/dnstrails.py
+++ b/misp_modules/modules/expansion/dnstrails.py
@@ -1,8 +1,9 @@
 import json
 import logging
 import sys
-from dnstrails import DnsTrails
+
 from dnstrails import APIError
+from dnstrails import DnsTrails
 
 log = logging.getLogger('dnstrails')
 log.setLevel(logging.DEBUG)
@@ -68,6 +69,14 @@ def handle_domain(api, domain, misperrors):
 
     r, status_ok = expand_domain_info(api, misperrors, domain)
 
+    if status_ok:
+        result_filtered['results'].extend(r)
+    else:
+        misperrors['error'] = 'Error dns result'
+        return misperrors
+
+    r, status_ok = expand_subdomains(api, domain)
+
     if status_ok:
         result_filtered['results'].extend(r)
     else: