From 3d02910124e8b5d9a7749340f4761a9c6d239e86 Mon Sep 17 00:00:00 2001
From: Jing Ling <shmilylty@gmail.com>
Date: Wed, 27 Jan 2021 22:33:09 +0800
Subject: [PATCH] fixed #175

---
 oneforall.py | 12 ++++++++----
 test.py      |  3 +++
 2 files changed, 11 insertions(+), 4 deletions(-)

diff --git a/oneforall.py b/oneforall.py
index 28d0ba8..dc78190 100644
--- a/oneforall.py
+++ b/oneforall.py
@@ -173,7 +173,9 @@ class OneForAll(object):
         utils.deal_data(self.domain)
         # Export results without resolve
         if not self.dns:
-            return self.export_data()
+            self.data = self.export_data()
+            self.datas.extend(self.data)
+            return self.data
 
         self.data = utils.get_data(self.domain)
 
@@ -185,7 +187,9 @@ class OneForAll(object):
 
         # Export results without HTTP request
         if not self.req:
-            return self.export_data()
+            self.data = self.export_data()
+            self.datas.extend(self.data)
+            return self.data
 
         if self.enable_wildcard:
             # deal wildcard
@@ -210,8 +214,8 @@ class OneForAll(object):
             enrich = Enrich(self.domain)
             enrich.run()
 
-        # Export
-        self.datas.extend(self.export_data())
+        self.data = self.export_data()
+        self.datas.extend(self.data)
 
         # Scan subdomain takeover
         if self.takeover:
diff --git a/test.py b/test.py
index 7c913d0..503bc62 100644
--- a/test.py
+++ b/test.py
@@ -10,10 +10,13 @@ from oneforall import OneForAll
 
 def oneforall(domain):
     test = OneForAll(target=domain)
+    test.dns = True
     test.brute = True
     test.req = True
     test.takeover = True
     test.run()
+    results = test.datas
+    print(results)
 
 
 if __name__ == '__main__':