From 5449bec562bbf5aa79cb5f143750fed511df4aef Mon Sep 17 00:00:00 2001
From: Yogesh Ojha <yogesh.ojha11@gmail.com>
Date: Wed, 15 Dec 2021 21:15:21 +0530
Subject: [PATCH] Default wordlists added

---
 web/celery-entrypoint.sh | 26 +++++++++++++++++++++-----
 web/reNgine/tasks.py     |  2 +-
 2 files changed, 22 insertions(+), 6 deletions(-)

diff --git a/web/celery-entrypoint.sh b/web/celery-entrypoint.sh
index 4020d09e..a9579312 100755
--- a/web/celery-entrypoint.sh
+++ b/web/celery-entrypoint.sh
@@ -22,6 +22,27 @@ python3 manage.py loaddata fixtures/default_keywords.yaml --app scanEngine.Inter
 # update whatportis
 yes | whatportis --update
 
+# clone dirsearch default wordlist
+if [ ! -d "/usr/src/wordlist" ]
+then
+  echo "Making Wordlist directory"
+  mkdir /usr/src/wordlist
+fi
+
+if [ ! -f "/usr/src/wordlist/" ]
+then
+  echo "Downloading Default Directory Bruteforce Wordlist"
+  wget https://raw.githubusercontent.com/maurosoria/dirsearch/master/db/dicc.txt -O /usr/src/wordlist/dicc.txt
+fi
+
+# check if default wordlist for amass exists
+if [ ! -f /usr/src/wordlist/deepmagic.com-prefixes-top50000.txt ];
+then
+  echo "Downloading Deepmagic top 50000 Wordlist"
+  wget https://raw.githubusercontent.com/danielmiessler/SecLists/master/Discovery/DNS/deepmagic.com-prefixes-top50000.txt -O /usr/src/wordlist/deepmagic.com-prefixes-top50000.txt
+fi
+
+
 # clone Sublist3r
 if [ ! -d "/usr/src/github/Sublist3r" ]
 then
@@ -79,11 +100,6 @@ then
   mkdir /usr/src/scan_results
 fi
 
-# check if default wordlist for amass exists
-if [ ! -f /usr/src/wordlist/deepmagic.com-prefixes-top50000.txt ]; then
-  wget https://raw.githubusercontent.com/danielmiessler/SecLists/master/Discovery/DNS/deepmagic.com-prefixes-top50000.txt -O /usr/src/wordlist/deepmagic.com-prefixes-top50000.txt
-fi
-
 # test tools, required for configuration
 naabu && subfinder && amass
 
diff --git a/web/reNgine/tasks.py b/web/reNgine/tasks.py
index ae881fc4..6ed50631 100644
--- a/web/reNgine/tasks.py
+++ b/web/reNgine/tasks.py
@@ -858,7 +858,7 @@ def directory_brute(task, domain, yaml_configuration, results_dir, activity_id):
     if (WORDLIST not in yaml_configuration[DIR_FILE_SEARCH] or
         not yaml_configuration[DIR_FILE_SEARCH][WORDLIST] or
             'default' in yaml_configuration[DIR_FILE_SEARCH][WORDLIST]):
-        wordlist_location = '/usr/src/github/dirsearch/db/dicc.txt'
+        wordlist_location = '/usr/src/wordlist/dicc.txt'
     else:
         wordlist_location = '/usr/src/wordlist/' + \
             yaml_configuration[DIR_FILE_SEARCH][WORDLIST] + '.txt'