From e76ee57c8a53c936ff529ffa96d055f6e998ed08 Mon Sep 17 00:00:00 2001
From: shmilylty <admin@hackfun.org>
Date: Sun, 4 Aug 2019 03:20:59 +0800
Subject: [PATCH] =?UTF-8?q?=E6=B7=BB=E5=8A=A0=E8=BF=90=E8=A1=8C=E8=BE=93?=
 =?UTF-8?q?=E5=87=BAbanner?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 oneforall/oneforall.py | 16 ++++++++++++++--
 1 file changed, 14 insertions(+), 2 deletions(-)

diff --git a/oneforall/oneforall.py b/oneforall/oneforall.py
index 91f4ecb..44ef687 100644
--- a/oneforall/oneforall.py
+++ b/oneforall/oneforall.py
@@ -12,12 +12,21 @@ import asyncio
 import fire
 import config
 import dbexport
+from datetime import datetime
 from config import logger
 from collect import Collect
 from aiobrute import AIOBrute
 from common import utils, database, resolve, request
 
 
+banner = """\033[01;33m
+             ___             _ _ 
+ ___ ___ ___|  _|___ ___ ___| | | \033[01;37m{\033[1;31mv0.0.2#dev\033[01;37m}\033[01;32m
+| . |   | -_|  _| . |  _| .'| | | \033[01;34m
+|___|_|_|___|_| |___|_| |__,|_|_| \033[0m\033[4;37mgit.io/fjHT1\033[0m\n
+        """
+
+
 class OneForAll(object):
     """
     OneForAll是一款功能强大的子域收集工具
@@ -57,6 +66,9 @@ class OneForAll(object):
         self.output = output
 
     def run(self):
+        print(banner)
+        dt = datetime.now().strftime('%Y-%m-%d %H:%M:%S')
+        print(f'[*] Starting OneForAll @ {dt}\n')
         logger.log('INFOR', f'开始运行OneForAll')
         self.domains = utils.get_domains(self.target)
         if self.domains:
@@ -89,5 +101,5 @@ class OneForAll(object):
 
 
 if __name__ == '__main__':
-    fire.Fire(OneForAll)
-    # OneForAll('example.com').run()
+    # fire.Fire(OneForAll)
+    OneForAll('example.com').run()