#!/bin/bash

# ============================================================
# USER SETTINGS
# Change these to suit your own setup
# ============================================================

IFACE="wlan1" # WiFi adapter used for monitor mode
CAPTURE_DIR="$HOME/wifi/captures" # Folder to save captures

DEFAULT_WAIT_TIME=8 # Seconds before deauth in option 6
DEFAULT_DEAUTH_COUNT=10 # Default number of deauth packets

SCRIPT_TITLE="uConsole WiFi Toolkit"

# ============================================================
# SCRIPT START
# ============================================================

pause() {
echo
read -p "Press Enter to continue..."
}

while true; do
clear
echo "$SCRIPT_TITLE"
echo "===================="
echo "Current interface: $IFACE"
echo "Capture folder: $CAPTURE_DIR"
echo
echo "1) Set $IFACE to monitor mode using iw"
echo "2) Set $IFACE to monitor mode using airmon-ng"
echo "3) Scan access points"
echo "4) Capture BSSID/channel only"
echo "5) Deauth test only"
echo "6) Capture and deauth"
echo "7) Show wireless interfaces"
echo "0) Exit"
echo
read -p "Choose an option: " choice

case "$choice" in
1)
sudo ip link set "$IFACE" down
sudo iw dev "$IFACE" set type monitor
sudo ip link set "$IFACE" up
iw dev
pause
;;

2)
sudo airmon-ng start "$IFACE"
pause
;;

3)
echo "Scanning on $IFACE. Press Ctrl+C to stop."
sleep 2
sudo airodump-ng "$IFACE"
pause
;;

4)
read -p "Enter BSSID: " BSSID
read -p "Enter channel: " CHANNEL

mkdir -p "$CAPTURE_DIR"

sudo airodump-ng \
-w "$CAPTURE_DIR/capture" \
-c "$CHANNEL" \
--bssid "$BSSID" \
"$IFACE"

pause
;;

5)
read -p "Enter BSSID: " BSSID
read -p "How many deauth packets? Default $DEFAULT_DEAUTH_COUNT: " DEAUTH_COUNT
DEAUTH_COUNT=${DEAUTH_COUNT:-$DEFAULT_DEAUTH_COUNT}

echo
echo "Only continue if this is your own network or an authorised test."
read -p "Type YES to continue: " confirm

if [ "$confirm" = "YES" ]; then
sudo aireplay-ng --deauth "$DEAUTH_COUNT" -a "$BSSID" "$IFACE"
else
echo "Cancelled."
fi

pause
;;

6)
read -p "Enter BSSID: " BSSID
read -p "Enter channel: " CHANNEL
read -p "Seconds before deauth? Default $DEFAULT_WAIT_TIME: " WAIT_TIME
read -p "How many deauth packets? Default $DEFAULT_DEAUTH_COUNT: " DEAUTH_COUNT

WAIT_TIME=${WAIT_TIME:-$DEFAULT_WAIT_TIME}
DEAUTH_COUNT=${DEAUTH_COUNT:-$DEFAULT_DEAUTH_COUNT}

echo
echo "Only continue if this is your own network or an authorised test."
read -p "Type YES to continue: " confirm

if [ "$confirm" != "YES" ]; then
echo "Cancelled."
pause
continue
fi

mkdir -p "$CAPTURE_DIR"

echo
echo "[+] Starting capture on $IFACE..."
echo "[+] Press Ctrl+C to stop capture when done."
echo

sudo airodump-ng \
-w "$CAPTURE_DIR/capture" \
-c "$CHANNEL" \
--bssid "$BSSID" \
"$IFACE" &

AIRO_PID=$!

sleep "$WAIT_TIME"

echo
echo "[+] Sending deauth packets..."
sudo aireplay-ng --deauth "$DEAUTH_COUNT" -a "$BSSID" "$IFACE"

echo
echo "[+] Capture is still running."
echo "[+] Press Ctrl+C to stop it."

wait "$AIRO_PID"
pause
;;

7)
iw dev
pause
;;

0)
exit 0
;;

*)
echo "Invalid option."
sleep 1
;;
esac
done