From 8c6c72499c033e7a27ecb3727fdc4ab49c433f75 Mon Sep 17 00:00:00 2001
From: bellma101 <bellma@0xfeed.io>
Date: Thu, 7 Mar 2019 11:56:20 -0600
Subject: [PATCH] Added use of built-in named word lists using config file.

---
 chomp-scan.sh | 58 ++++++++++++++++++++++++++++++++++++++++-----------
 1 file changed, 46 insertions(+), 12 deletions(-)

diff --git a/chomp-scan.sh b/chomp-scan.sh
index b4e8d68..abcdbde 100755
--- a/chomp-scan.sh
+++ b/chomp-scan.sh
@@ -256,12 +256,26 @@ function parse_config() {
 		fi
 
 		SUB_WORDLIST=$(grep '^SUBDOMAIN_WORDLIST' "$CONFIG_FILE" | cut -d '=' -f 2);
-		if [[ "$SUB_WORDLIST" != "" ]]; then
-				if [[ -w "$SUB_WORDLIST" ]]; then
-						SUBDOMAIN_WORDLIST="$SUB_WORDLIST";
-				else
-						echo -e "$RED""[!] Subdomain enumeration wordlist $SUB_WORDLIST does not exist or is not writable. Please check the configuration file.""$NC";
-						exit 1;
+		# Set to one of the defaults, else use provided wordlist
+		case "$SUB_WORDLIST" in
+				SHORT )
+						SUBDOMAIN_WORDLIST="$SHORT";
+						;;
+				LONG )
+						SUBDOMAIN_WORDLIST="$LONG";
+						;;
+				HUGE )
+						SUBDOMAIN_WORDLIST="$HUGE";
+						;;
+		esac
+		if [[ "$SUBDOMAIN_WORDLIST" == "" ]]; then
+				if [[ "$SUB_WORDLIST" != "" ]]; then
+						if [[ -w "$SUB_WORDLIST" ]]; then
+								SUBDOMAIN_WORDLIST="$SUB_WORDLIST";
+						else
+								echo -e "$RED""[!] Subdomain enumeration wordlist $SUB_WORDLIST does not exist or is not writable. Please check the configuration file.""$NC";
+								exit 1;
+						fi
 				fi
 		fi
 
@@ -294,12 +308,32 @@ function parse_config() {
 		fi
 
 		CON_WORDLIST=$(grep '^CONTENT_WORDLIST' "$CONFIG_FILE" | cut -d '=' -f 2);
-		if [[ "$CON_WORDLIST" != "" ]]; then
-				if [[ -w "$CON_WORDLIST" ]]; then
-						CONTENT_WORDLIST="$CON_WORDLIST";
-				else
-						echo -e "$RED""[!] Content discovery wordlist $CON_WORDLIST does not exist or is not writable. Please check the configuration file.""$NC";
-						exit 1;
+		# Set to one of the defaults, else use provided wordlist
+		case "$CON_WORDLIST" in
+				SMALL )
+						CONTENT_WORDLIST="$SMALL";
+						;;
+				MEDIUM )
+						CONTENT_WORDLIST="$MEDIUM";
+						;;
+				LARGE )
+						CONTENT_WORDLIST="$LARGE";
+						;;
+				XL )
+						CONTENT_WORDLIST="$XL";
+						;;
+				XXL )
+						CONTENT_WORDLIST="$XXL";
+						;;
+		esac
+		if [[ "$CONTENT_WORDLIST" == "" ]]; then
+				if [[ "$CON_WORDLIST" != "" ]]; then
+						if [[ -w "$CON_WORDLIST" ]]; then
+								CONTENT_WORDLIST="$CON_WORDLIST";
+						else
+								echo -e "$RED""[!] Content discovery wordlist $CON_WORDLIST does not exist or is not writable. Please check the configuration file.""$NC";
+								exit 1;
+						fi
 				fi
 		fi