Mark-VIII/payload/user/remote_access/pager-webui/payload.sh

#!/bin/bash
# Title: Mark VIII
# Description: Mark VII-style web management UI for the WiFi Pineapple Pager
# Author: c4ch3c4d3
# Version: 1.3.1
# Category: Remote-Access
# Tags: remote-access, web-interface, device-management, pineap
# Firmware: Pineapple Pager 24.10.1

SCRIPT_SOURCE="${BASH_SOURCE[0]:-$0}"
SCRIPT_DIR="$(cd "$(dirname "$SCRIPT_SOURCE")" 2>/dev/null && pwd)"
[ -f "$SCRIPT_DIR/server.py" ] || SCRIPT_DIR="/root/payloads/user/remote_access/pager-webui"
[ -f "$SCRIPT_DIR/server.py" ] || SCRIPT_DIR="/mmc/root/payloads/user/remote_access/pager-webui"
PORT=8080
PID_FILE="/tmp/pagerwebui.pid"
INIT_SCRIPT="/etc/init.d/pagerwebui"

user_confirmed() {
    [ "$1" = "true" ] || [ "$1" = "$DUCKYSCRIPT_USER_CONFIRMED" ]
}

get_pager_ip() {
    for iface in br-lan eth0 wlan0 usb0; do
        IP=$(ip -4 addr show "$iface" 2>/dev/null | awk '/inet / {print $2}' | cut -d'/' -f1 | head -1)
        [ -n "$IP" ] && echo "$IP" && return
    done
    echo "172.16.52.1"
}

LOG "cyan" "+---------------------------+"
LOG "cyan" "|       Mark VIII v1.3.1     |"
LOG "cyan" "+---------------------------+"

if ! command -v python3 >/dev/null 2>&1; then
    LOG "red" "python3 required (present on current firmware)"
    exit 1
fi

run_env_check() {
    LOG "cyan" "Running environment check..."
    if ! python3 "$SCRIPT_DIR/server.py" --env-check; then
        LOG "red" "Environment check FAILED. Fix the issues above and re-run the payload."
        sleep 3
        exit 1
    fi
    LOG "green" "Environment check passed."
}

wait_for_server() {
    attempts="${1:-30}"
    while [ "$attempts" -gt 0 ]; do
        if curl -fsS "http://127.0.0.1:$PORT/" >/dev/null 2>&1; then
            return 0
        fi
        sleep 1
        attempts=$((attempts - 1))
    done
    return 1
}

wait_for_server_stop() {
    attempts="${1:-12}"
    while [ "$attempts" -gt 0 ]; do
        if ! curl -fsS "http://127.0.0.1:$PORT/" >/dev/null 2>&1; then
            return 0
        fi
        sleep 1
        attempts=$((attempts - 1))
    done
    return 1
}

remove_boot_service() {
    "$INIT_SCRIPT" stop 2>/dev/null || true
    "$INIT_SCRIPT" disable 2>/dev/null || true
    rm -f "$INIT_SCRIPT"
}

if [ -f "$INIT_SCRIPT" ] && "$INIT_SCRIPT" running 2>/dev/null; then
    PAGER_IP=$(get_pager_ip)
    LOG "green" "Mark VIII service is running"
    LOG "green" "http://$PAGER_IP:$PORT"
    resp=$(CONFIRMATION_DIALOG "Stop service?")
    if user_confirmed "$resp"; then
        LOG "yellow" "Stopping service..."
        remove_boot_service
        if ! wait_for_server_stop 12; then
            LOG "red" "Service is still listening on port $PORT"
            exit 1
        fi
        LOG "cyan" "Service stopped and removed from boot"
    fi
    exit 0
fi

if [ -f "$INIT_SCRIPT" ] && "$INIT_SCRIPT" enabled 2>/dev/null; then
    LOG "yellow" "Mark VIII boot service is installed but not running"
    resp=$(CONFIRMATION_DIALOG "Remove boot service?")
    if user_confirmed "$resp"; then
        remove_boot_service
        LOG "cyan" "Boot service removed"
        exit 0
    fi
fi

AUTO_MODE=$(PAYLOAD_GET_CONFIG pager_webui auto_mode 2>/dev/null)
RUN_MODE=$(PAYLOAD_GET_CONFIG pager_webui run_mode 2>/dev/null)

if [ "$AUTO_MODE" = "true" ]; then
    if [ "$RUN_MODE" = "background" ]; then
        resp="true"
        LOG "cyan" "Auto-starting background mode..."
    else
        resp=""
        LOG "cyan" "Auto-starting foreground mode..."
    fi
else
    resp=$(CONFIRMATION_DIALOG "Run as background service?")
fi

if user_confirmed "$resp"; then
    run_env_check
    LOG "cyan" "Starting as background service..."
    [ ! -f "$SCRIPT_DIR/server.py" ] && { LOG "red" "server.py not found!"; exit 1; }
    cp "$SCRIPT_DIR/pagerwebui.init" "$INIT_SCRIPT"
    chmod +x "$INIT_SCRIPT"
    "$INIT_SCRIPT" enable
    if ! "$INIT_SCRIPT" start; then
        LOG "red" "Service start command failed"
        exit 1
    fi
    if ! wait_for_server 90; then
        remove_boot_service
        LOG "red" "Service failed startup checks; inspect logread"
        exit 1
    fi
    PAGER_IP=$(get_pager_ip)
    LOG "green" "Service started!"
    LOG "green" "http://$PAGER_IP:$PORT"
    LOG "cyan" "Re-run payload to stop"
    sleep 3
    exit 0
fi

run_env_check
LOG "cyan" "Starting foreground mode..."
cleanup() {
    LOG "yellow" "Stopping Mark VIII..."
    [ -f "$PID_FILE" ] && kill "$(cat "$PID_FILE")" 2>/dev/null
    rm -f "$PID_FILE"
    LOG "cyan" "Stopped."
}
trap cleanup EXIT INT TERM

[ ! -f "$SCRIPT_DIR/server.py" ] && { LOG "red" "server.py not found!"; exit 1; }
python3 "$SCRIPT_DIR/server.py" >/tmp/pagerwebui.log 2>&1 &
echo $! > "$PID_FILE"

PAGER_IP=$(get_pager_ip)
if wait_for_server 90 && [ -f "$PID_FILE" ] &&
        kill -0 "$(cat "$PID_FILE")" 2>/dev/null; then
    LOG "green" "http://$PAGER_IP:$PORT"
    LOG ""
    LOG "magenta" "Press B to stop"
    while true; do
        BUTTON=$(WAIT_FOR_INPUT)
        if [ "$BUTTON" = "B" ] || [ "$BUTTON" = "Escape" ]; then
            break
        fi
    done
else
    LOG "red" "Failed to start server! Check /tmp/pagerwebui.log"
    exit 1
fi