From 2994e240b507be58850c7476e0dd67265eeb6cbd Mon Sep 17 00:00:00 2001
From: lilfade <lilfade.lor@gmail.com>
Date: Thu, 17 Aug 2023 22:29:18 -0500
Subject: [PATCH] updates

---
 entry.sh                             | 83 ++++++++++++++++------------
 entry.sh.old                         | 35 +++++++++---
 tools/imap_pop3_test.sh              | 59 ++++++++++++++++++++
 tools/load_time_test.sh              | 41 ++++++++++++++
 tools/smtp_test.sh                   |  4 +-
 wip/dns_query_analyzer.sh.unloaded   | 54 ++++++++++++++++++
 wip/subdomain_enumerator.sh.unloaded | 64 +++++++++++++++++++++
 7 files changed, 297 insertions(+), 43 deletions(-)
 create mode 100644 tools/imap_pop3_test.sh
 create mode 100644 tools/load_time_test.sh
 create mode 100644 wip/dns_query_analyzer.sh.unloaded
 create mode 100644 wip/subdomain_enumerator.sh.unloaded

diff --git a/entry.sh b/entry.sh
index e994079..06c1f54 100755
--- a/entry.sh
+++ b/entry.sh
@@ -15,22 +15,20 @@ for tool_script in $(ls "$TOOLS_DIR"/*.sh); do
     index=$((index + 1))
 done
 
-# Main menu loop
 while true; do
-    echo "Available tools:"
-    echo "[D] Add domain or file of domains"
-    for number in $(seq 1 ${#tool_scripts[@]}); do
-        source "${tool_scripts[$number]}"
-        tool_name=$(get_tool_name)
-        echo "[$number] $tool_name"
-    done
-    echo "[A] Run all tools against the loaded domain"
-    echo "[0] Exit and Print Output"
-
-    read -p "Enter the number of the tool to execute (0 to exit): " choice
+    if [ -z "$domain" ]; then
+        echo "[D] Add domain or file of domains"
+        echo "[Q] Exit and Print Output"
+        read -p "Enter an option (D/Q): " choice
+    else
+        echo "[T] Choose a tool to execute"
+        echo "[A] Run all tests"
+        echo "[Q] Exit and Print Output"
+        read -p "Enter an option (T/A/Q): " choice
+    fi
 
     if [ "$choice" == "D" ] || [ "$choice" == "d" ]; then
-	echo "Choose an option to load domains:"
+        echo "Choose an option to load domains:"
         echo "[1] Load domains from domains.txt"
         echo "[2] Load a single domain"
 
@@ -59,6 +57,29 @@ while true; do
         else
             echo "Invalid option: $load_choice"
         fi
+    elif [ "$choice" == "T" ] || [ "$choice" == "t" ]; then
+        if [ -z "$domain" ]; then
+            echo "No domain loaded. Please load a domain first."
+        else
+            echo "Available tools:"
+            for number in $(seq 1 ${#tool_scripts[@]}); do
+                source "${tool_scripts[$number]}"
+                tool_name=$(get_tool_name)
+                echo "[$number] $tool_name"
+            done
+
+            read -p "Enter the number of the tool to execute: " tool_choice
+
+            if [[ "$tool_choice" =~ ^[0-9]+$ ]] && [ "${tool_scripts[$tool_choice]}" ]; then
+                selected_script="${tool_scripts[$tool_choice]}"
+                source "$selected_script"
+                tool_output=$(execute_tool)
+                tool_outputs["$tool_choice"]="$tool_output"
+                echo "Tool $tool_choice executed."
+            else
+                echo "Invalid tool number: $tool_choice"
+            fi
+        fi
     elif [ "$choice" == "A" ] || [ "$choice" == "a" ]; then
         if [ -z "$domain" ]; then
             echo "No domain loaded. Please load a domain first."
@@ -72,20 +93,19 @@ while true; do
                     echo "Tool $number executed."
                 fi
             done
-            echo -e "\nResults:"
-            for number in $(seq 1 ${#tool_scripts[@]}); do
-                if [ "${tool_outputs["$number"]}" ]; then
-                    source "${tool_scripts[$number]}"
-                    tool_name=$(get_tool_name)
-                    echo "Tool: $tool_name"
-                    echo -e "${tool_outputs["$number"]}"
-                    echo "------------------------------------"
-		fi
-            done
-            exit
+            echo -e "\nAll tests executed."
         fi
-    elif [[ "$choice" =~ ^[0-9]+$ ]]; then
-	if [ "$choice" -eq 0 ]; then
+    elif [ "$choice" == "Q" ] || [ "$choice" == "q" ]; then
+        has_output=false
+        
+        for number in $(seq 1 ${#tool_scripts[@]}); do
+            if [ "${tool_outputs["$number"]}" ]; then
+                has_output=true
+                break
+            fi
+        done
+        
+        if [ "$has_output" = true ]; then
             echo -e "\nResults:"
             for number in $(seq 1 ${#tool_scripts[@]}); do
                 if [ "${tool_outputs["$number"]}" ]; then
@@ -94,16 +114,11 @@ while true; do
                     echo "------------------------------------"
                 fi
             done
-            exit
-        elif [ "${tool_scripts[$choice]}" ]; then
-            selected_script="${tool_scripts[$choice]}"
-            source "$selected_script"
-            tool_output=$(execute_tool)
-            tool_outputs["$choice"]+="$tool_output"
-            echo "Tool $choice executed."
         else
-            echo "Invalid tool number: $choice"
+            echo "No output available."
         fi
+        
+        exit
     else
         echo "Invalid input."
     fi
diff --git a/entry.sh.old b/entry.sh.old
index 2076bff..e994079 100755
--- a/entry.sh.old
+++ b/entry.sh.old
@@ -24,15 +24,15 @@ while true; do
         tool_name=$(get_tool_name)
         echo "[$number] $tool_name"
     done
+    echo "[A] Run all tools against the loaded domain"
     echo "[0] Exit and Print Output"
 
     read -p "Enter the number of the tool to execute (0 to exit): " choice
 
     if [ "$choice" == "D" ] || [ "$choice" == "d" ]; then
-        echo "Choose an option to load domains:"
+	echo "Choose an option to load domains:"
         echo "[1] Load domains from domains.txt"
         echo "[2] Load a single domain"
-        echo "[3] Load multiple single domains"
 
         read -p "Enter the number of the option: " load_choice
 
@@ -56,15 +56,36 @@ while true; do
             read -p "Enter a single domain: " single_domain
             domain="$single_domain"
             echo "Added single domain: $domain"
-        elif [ "$load_choice" -eq 3 ]; then
-            read -p "Enter multiple single domains (separated by spaces): " multiple_domains
-            domain="$multiple_domains"
-            echo "Added multiple single domains: $domain"
         else
             echo "Invalid option: $load_choice"
         fi
+    elif [ "$choice" == "A" ] || [ "$choice" == "a" ]; then
+        if [ -z "$domain" ]; then
+            echo "No domain loaded. Please load a domain first."
+        else
+            for number in $(seq 1 ${#tool_scripts[@]}); do
+                if [ "${tool_scripts[$number]}" ]; then
+                    selected_script="${tool_scripts[$number]}"
+                    source "$selected_script"
+                    tool_output=$(execute_tool)
+                    tool_outputs["$number"]="$tool_output"
+                    echo "Tool $number executed."
+                fi
+            done
+            echo -e "\nResults:"
+            for number in $(seq 1 ${#tool_scripts[@]}); do
+                if [ "${tool_outputs["$number"]}" ]; then
+                    source "${tool_scripts[$number]}"
+                    tool_name=$(get_tool_name)
+                    echo "Tool: $tool_name"
+                    echo -e "${tool_outputs["$number"]}"
+                    echo "------------------------------------"
+		fi
+            done
+            exit
+        fi
     elif [[ "$choice" =~ ^[0-9]+$ ]]; then
-        if [ "$choice" -eq 0 ]; then
+	if [ "$choice" -eq 0 ]; then
             echo -e "\nResults:"
             for number in $(seq 1 ${#tool_scripts[@]}); do
                 if [ "${tool_outputs["$number"]}" ]; then
diff --git a/tools/imap_pop3_test.sh b/tools/imap_pop3_test.sh
new file mode 100644
index 0000000..4787fb2
--- /dev/null
+++ b/tools/imap_pop3_test.sh
@@ -0,0 +1,59 @@
+#!/bin/bash
+
+get_tool_name() {
+    echo "IMAP and POP3 Accessibility Checker"
+}
+
+get_tool_description() {
+    echo "Check IMAP and POP3 accessibility for a domain"
+}
+
+execute_tool() {
+    result=""
+    if [ -n "$domain" ]; then
+        if [[ "$domain" == *$'\n'* ]]; then
+            for single_domain in $domain; do
+                process_domain "$single_domain"
+            done
+        else
+            process_domain "$domain"
+        fi
+    else
+        result="No domains specified."
+    fi
+
+    echo -e "$result"
+}
+
+process_domain() {
+    local single_domain="$1"
+    mx_record=$(dig +short MX "$single_domain" | sort -n | head -n 1 | awk '{print $2}')
+
+    if [ -z "$mx_record" ]; then
+        mx_record="$single_domain"
+    fi
+
+    if command -v telnet &> /dev/null; then
+        imap143_status=$(timeout 10 telnet "$mx_record" 143 <<< "quit" > /dev/null 2>&1 && echo "Connected" || echo "Failed")
+        imap993_status=$(timeout 10 telnet "$mx_record" 993 <<< "quit" > /dev/null 2>&1 && echo "Connected" || echo "Failed")
+        pop3110_status=$(timeout 10 telnet "$mx_record" 110 <<< "quit" > /dev/null 2>&1 && echo "Connected" || echo "Failed")
+        pop3995_status=$(timeout 10 telnet "$mx_record" 995 <<< "quit" > /dev/null 2>&1 && echo "Connected" || echo "Failed")
+    elif command -v nc &> /dev/null; then
+        imap143_status=$(timeout 10 nc -z "$mx_record" 143 > /dev/null 2>&1 && echo "Connected" || echo "Failed")
+        imap993_status=$(timeout 10 nc -z "$mx_record" 993 > /dev/null 2>&1 && echo "Connected" || echo "Failed")
+        pop3110_status=$(timeout 10 nc -z "$mx_record" 110 > /dev/null 2>&1 && echo "Connected" || echo "Failed")
+        pop3995_status=$(timeout 10 nc -z "$mx_record" 995 > /dev/null 2>&1 && echo "Connected" || echo "Failed")
+    else
+        result+="\e[1mDomain:\e[0m $single_domain\n"
+        result+="\e[1mIMAP Accessibility:\e[0m Netcat and Telnet not available.\n"
+        result+="\e[1mPOP3 Accessibility:\e[0m Netcat and Telnet not available.\n"
+        return
+    fi
+
+
+    result+="\e[1mDomain:\e[0m $single_domain\n"
+    result+="\e[1mIMAP Accessibility (Port 143):\e[0m $imap143_status\n"
+    result+="\e[1mIMAP Accessibility (Port 993):\e[0m $imap993_status\n"
+    result+="\e[1mPOP3 Accessibility (Port 110):\e[0m $pop3110_status\n"
+    result+="\e[1mPOP3 Accessibility (Port 995):\e[0m $pop3995_status\n"
+}
diff --git a/tools/load_time_test.sh b/tools/load_time_test.sh
new file mode 100644
index 0000000..cf9d58e
--- /dev/null
+++ b/tools/load_time_test.sh
@@ -0,0 +1,41 @@
+#!/bin/bash
+
+get_tool_name() {
+    echo "Website Load Time Tester"
+}
+
+get_tool_description() {
+    echo "Test website load times using HTTP requests"
+}
+
+execute_tool() {
+    result=""
+    if [ -n "$domain" ]; then
+        if [[ "$domain" == *$'\n'* ]]; then
+            for single_domain in $domain; do
+                process_domain "$single_domain"
+            done
+        else
+            process_domain "$domain"
+        fi
+    else
+        result="No domains specified."
+    fi
+
+    echo -e "$result"
+}
+
+process_domain() {
+    local single_domain="$1"
+    
+    result+="\e[1mDomain:\e[0m $single_domain\n"
+    
+    # Measure initial server response time using curl
+    response_time=$(curl -o /dev/null -s -w "%{time_starttransfer}\n" "$single_domain")
+    
+    # Measure fully loaded time using curl
+    fully_loaded_time=$(curl -o /dev/null -s -w "%{time_total}\n" "$single_domain")
+    
+    result+="\e[1mInitial Server Response Time:\e[0m $response_time seconds\n"
+    result+="\e[1mFully Loaded Time:\e[0m $fully_loaded_time seconds\n"
+}
diff --git a/tools/smtp_test.sh b/tools/smtp_test.sh
index 41b81be..3f6ffc1 100755
--- a/tools/smtp_test.sh
+++ b/tools/smtp_test.sh
@@ -33,9 +33,9 @@ process_domain() {
         ports_status=""
         for port in $smtp_ports; do
             if nc -z -w3 "$mx_record" "$port"; then
-                ports_status+="\e[32m$port: \e[1mSuccess\e[0m "
+                ports_status+="\e[0m$port: \e[32mSuccess\e[0m "
             else
-                ports_status+="\e[31m$port: \e[1mError\e[0m "
+                ports_status+="\e[0m$port: \e[31mError\e[0m "
             fi
         done
         result+="\e[1mMX Hostname Tested:\e[0m $mx_record \e[1mPorts:\e[0m $ports_status\n"
diff --git a/wip/dns_query_analyzer.sh.unloaded b/wip/dns_query_analyzer.sh.unloaded
new file mode 100644
index 0000000..3b780dd
--- /dev/null
+++ b/wip/dns_query_analyzer.sh.unloaded
@@ -0,0 +1,54 @@
+#!/bin/bash
+
+get_tool_name() {
+    echo "DNS Query Analyzer"
+}
+
+get_tool_description() {
+    echo "Analyze DNS queries and responses for slow responses"
+}
+
+execute_tool() {
+    if ! command -v tcpdump &> /dev/null || ! command -v tshark &> /dev/null; then
+        echo "Required tools (tcpdump and tshark) are not available."
+        exit 1
+    fi
+
+    result=""
+    if [ -n "$domain" ]; then
+        if [[ "$domain" == *$'\n'* ]]; then
+            for single_domain in $domain; do
+                analyze_dns "$single_domain"
+            done
+        else
+            analyze_dns "$domain"
+        fi
+    else
+        result="No domains specified."
+    fi
+
+    echo -e "$result"
+}
+
+analyze_dns() {
+    local single_domain="$1"
+    
+    result+="\e[1mAnalyzing DNS queries for $single_domain:\e[0m\n"
+    
+    # Capture DNS traffic with tcpdump
+    tcpdump -i any -n -s0 -w dns_traffic.pcap udp port 53 &> /dev/null &
+    sleep 5
+    pkill tcpdump
+    
+    # Analyze captured traffic with tshark
+    slow_queries=$(tshark -r dns_traffic.pcap -Y "dns.qry.name contains $single_domain && dns.a" -T fields -e dns.time -e dns.qry.name)
+    
+    if [ -n "$slow_queries" ]; then
+        result+="$slow_queries\n"
+    else
+        result+="No slow queries found.\n"
+    fi
+    
+    # Cleanup captured traffic file
+    rm -f dns_traffic.pcap
+}
diff --git a/wip/subdomain_enumerator.sh.unloaded b/wip/subdomain_enumerator.sh.unloaded
new file mode 100644
index 0000000..0949d8c
--- /dev/null
+++ b/wip/subdomain_enumerator.sh.unloaded
@@ -0,0 +1,64 @@
+#!/bin/bash
+
+get_tool_name() {
+    echo "Subdomain Enumerator"
+}
+
+get_tool_description() {
+    echo "Enumerate subdomains of a primary domain"
+}
+
+check_requirements() {
+    if ! command -v dig &> /dev/null; then
+        echo "dig is not available. Install it to continue."
+        exit 1
+    fi
+
+    if ! command -v amass &> /dev/null; then
+        echo "amass is not available. Install it to continue."
+        exit 1
+    fi
+}
+
+execute_tool() {
+    check_requirements
+
+    result=""
+    if [ -n "$domain" ]; then
+        if [[ "$domain" == *$'\n'* ]]; then
+            for single_domain in $domain; do
+                enumerate_subdomains "$single_domain"
+            done
+        else
+            enumerate_subdomains "$domain"
+        fi
+    else
+        result="No domains specified."
+    fi
+
+    echo -e "$result"
+}
+
+enumerate_subdomains() {
+    local primary_domain="$1"
+    
+    result+="\e[1mEnumerating subdomains of $primary_domain:\e[0m\n"
+    
+    # Perform DNS queries to enumerate subdomains
+    subdomains_dns=$(dig +short "$primary_domain" | grep -oE "([a-zA-Z0-9\-]+\.)*$primary_domain")
+    
+    # Perform web-based enumeration using Amass
+    subdomains_web=$(amass enum -d "$primary_domain" -o - 2>/dev/null)
+    
+    # Combine results from DNS and web-based enumeration
+    all_subdomains="$subdomains_dns\n$subdomains_web"
+    
+    # Remove duplicates and sort the list
+    unique_subdomains=$(echo -e "$all_subdomains" | sort -u)
+    
+    if [ -n "$unique_subdomains" ]; then
+        result+="$unique_subdomains\n"
+    else
+        result+="No subdomains found.\n"
+    fi
+}