mirror of
https://github.com/jayofelony/pwnagotchi.git
synced 2025-07-01 18:37:27 -04:00
Backup and restore edits
Windows and linux files Signed-off-by: wpa-2 <9049886+wpa-2@users.noreply.github.com>
This commit is contained in:
wpa-2
171
scripts/BR.bat
Normal file
171
scripts/BR.bat
Normal file
@ -0,0 +1,171 @@
|
||||
@echo off
|
||||
setlocal
|
||||
|
||||
REM ---------------------------------------------------------------------------
|
||||
REM A simple .bat script that handles two subcommands: "backup" and "restore"
|
||||
REM USAGE:
|
||||
REM backup-restore.bat backup [ -n HOST ] [ -u USER ] [ -o OUTPUT ]
|
||||
REM backup-restore.bat restore [ -n HOST ] [ -u USER ] [ -b BACKUP_FILE ]
|
||||
REM ---------------------------------------------------------------------------
|
||||
|
||||
set "SCRIPTNAME=%~0"
|
||||
|
||||
REM ---------------------------------------------------------------------------
|
||||
REM If no arguments, show usage
|
||||
if "%~1"=="" goto usage
|
||||
|
||||
REM First argument is the subcommand: "backup" or "restore"
|
||||
set "SUBCOMMAND=%~1"
|
||||
shift
|
||||
|
||||
REM Default values
|
||||
set "UNIT_HOSTNAME=10.0.0.2"
|
||||
set "UNIT_USERNAME=pi"
|
||||
set "OUTPUT="
|
||||
set "BACKUP_FILE="
|
||||
|
||||
:parse_args
|
||||
if "%~1"=="" goto done_args
|
||||
|
||||
if "%~1"=="-n" (
|
||||
set "UNIT_HOSTNAME=%~2"
|
||||
shift
|
||||
shift
|
||||
goto parse_args
|
||||
)
|
||||
|
||||
if "%~1"=="-u" (
|
||||
set "UNIT_USERNAME=%~2"
|
||||
shift
|
||||
shift
|
||||
goto parse_args
|
||||
)
|
||||
|
||||
if "%SUBCOMMAND%"=="backup" (
|
||||
if "%~1"=="-o" (
|
||||
set "OUTPUT=%~2"
|
||||
shift
|
||||
shift
|
||||
goto parse_args
|
||||
)
|
||||
)
|
||||
|
||||
if "%SUBCOMMAND%"=="restore" (
|
||||
if "%~1"=="-b" (
|
||||
set "BACKUP_FILE=%~2"
|
||||
shift
|
||||
shift
|
||||
goto parse_args
|
||||
)
|
||||
)
|
||||
|
||||
echo "Unknown or invalid argument: %~1"
|
||||
goto usage
|
||||
|
||||
:done_args
|
||||
|
||||
REM ---------------------------------------------------------------------------
|
||||
REM Subcommand routing
|
||||
if "%SUBCOMMAND%"=="backup" goto do_backup
|
||||
if "%SUBCOMMAND%"=="restore" goto do_restore
|
||||
goto usage
|
||||
|
||||
REM ===========================================================================
|
||||
:do_backup
|
||||
REM If OUTPUT not specified, generate a name like HOST-backup-RANDOM.tgz
|
||||
if "%OUTPUT%"=="" (
|
||||
set /a RAND=%random%
|
||||
set "OUTPUT=%UNIT_HOSTNAME%-backup-%RAND%.tgz"
|
||||
)
|
||||
|
||||
echo @ Checking connectivity to %UNIT_HOSTNAME% ...
|
||||
ping -n 1 %UNIT_HOSTNAME% >nul 2>&1
|
||||
if errorlevel 1 (
|
||||
echo @ unit %UNIT_HOSTNAME% can't be reached.
|
||||
exit /b 1
|
||||
)
|
||||
|
||||
echo @ Backing up %UNIT_HOSTNAME% to %OUTPUT% ...
|
||||
|
||||
REM List of files to back up from the remote device:
|
||||
set "FILES_TO_BACKUP=/root/settings.yaml /root/client_secrets.json /root/.ssh /root/.api-report.json /root/.bashrc /root/.profile /home/pi/handshakes /root/peers /etc/pwnagotchi/ /usr/local/share/pwnagotchi/custom-plugins /etc/ssh/ /home/pi/.bashrc /home/pi/.profile /home/pi/.wpa_sec_uploads"
|
||||
|
||||
REM ---------------------------------------------------------------------------
|
||||
REM We'll have the REMOTE do the tar + gzip (since Windows cmd lacks gzip)
|
||||
REM so we capture the compressed stream directly into %OUTPUT%
|
||||
REM ---------------------------------------------------------------------------
|
||||
ssh %UNIT_USERNAME%@%UNIT_HOSTNAME% "sudo tar -cf - %FILES_TO_BACKUP% | gzip -9" > "%OUTPUT%"
|
||||
if errorlevel 1 (
|
||||
echo @ Backup failed
|
||||
exit /b 1
|
||||
)
|
||||
|
||||
echo @ Backup finished. Archive: %OUTPUT%
|
||||
exit /b 0
|
||||
|
||||
REM ===========================================================================
|
||||
:do_restore
|
||||
if "%BACKUP_FILE%"=="" (
|
||||
REM If user didn't specify -b, try to find the latest *.tgz with the hostname prefix
|
||||
for /f "delims=" %%A in ('dir /b /a-d /O-D "%UNIT_HOSTNAME%-backup-*.tgz" 2^>nul') do (
|
||||
set "BACKUP_FILE=%%A"
|
||||
goto found_file
|
||||
)
|
||||
echo @ Can't find backup file. Please specify one with '-b'.
|
||||
exit /b 1
|
||||
|
||||
:found_file
|
||||
echo @ Found backup file: %BACKUP_FILE%
|
||||
set /p CONTINUE="@ Continue restoring this file? (y/n) "
|
||||
if /i "%CONTINUE%"=="y" (
|
||||
echo (continuing...)
|
||||
) else (
|
||||
exit /b 1
|
||||
)
|
||||
)
|
||||
|
||||
echo @ Checking connectivity to %UNIT_HOSTNAME% ...
|
||||
ping -n 1 %UNIT_HOSTNAME% > nul 2>&1
|
||||
if errorlevel 1 (
|
||||
echo @ unit %UNIT_HOSTNAME% can't be reached.
|
||||
exit /b 1
|
||||
)
|
||||
|
||||
echo @ Restoring %BACKUP_FILE% to %UNIT_HOSTNAME% ...
|
||||
|
||||
REM We'll send the local file contents over SSH; the remote will do tar xzv
|
||||
type "%BACKUP_FILE%" | ssh %UNIT_USERNAME%@%UNIT_HOSTNAME% "tar xzv -C /"
|
||||
if errorlevel 1 (
|
||||
echo @ Restore failed.
|
||||
exit /b 1
|
||||
)
|
||||
|
||||
echo @ Restore finished.
|
||||
exit /b 0
|
||||
|
||||
REM ===========================================================================
|
||||
:usage
|
||||
echo Usage:
|
||||
echo %SCRIPTNAME% backup [ -n HOST ] [ -u USER ] [ -o OUTPUT ]
|
||||
echo %SCRIPTNAME% restore [ -n HOST ] [ -u USER ] [ -b BACKUP_FILE ]
|
||||
echo.
|
||||
echo Subcommands:
|
||||
echo backup Backup files from the remote device.
|
||||
echo restore Restore files to the remote device.
|
||||
echo.
|
||||
echo Common Options:
|
||||
echo -n HOST Hostname or IP of remote device (default: 10.0.0.2)
|
||||
echo -u USER Username for SSH (default: pi)
|
||||
echo.
|
||||
echo Options for 'backup':
|
||||
echo -o OUTPUT Path/name of the output archive (default: HOST-backup-RANDOM.tgz)
|
||||
echo.
|
||||
echo Options for 'restore':
|
||||
echo -b BACKUP Path to the local backup archive to restore.
|
||||
echo.
|
||||
echo Examples:
|
||||
echo %SCRIPTNAME% backup
|
||||
echo %SCRIPTNAME% backup -n 10.0.0.2 -u pi -o my-backup.tgz
|
||||
echo %SCRIPTNAME% restore -b my-backup.tgz
|
||||
echo.
|
||||
exit /b 1
|
||||
197
scripts/BR.sh
Normal file
197
scripts/BR.sh
Normal file
@ -0,0 +1,197 @@
|
||||
#!/usr/bin/env bash
|
||||
#
|
||||
# A Combined backup/restore script for linux
|
||||
# Usage:
|
||||
# backup-restore.sh backup [ -n HOST ] [ -u USER ] [ -o OUTPUT ]
|
||||
# backup-restore.sh restore [ -n HOST ] [ -u USER ] [ -b BACKUP_FILE ]
|
||||
#
|
||||
|
||||
###############################################################################
|
||||
# GLOBAL USAGE
|
||||
###############################################################################
|
||||
usage() {
|
||||
echo "Usage:"
|
||||
echo " $0 backup [ -n HOST ] [ -u USER ] [ -o OUTPUT ]"
|
||||
echo " $0 restore [ -n HOST ] [ -u USER ] [ -b BACKUP_FILE ]"
|
||||
echo
|
||||
echo "Subcommands:"
|
||||
echo " backup Backup files from the remote device."
|
||||
echo " restore Restore files to the remote device."
|
||||
echo
|
||||
echo "Common Options:"
|
||||
echo " -n HOST Hostname or IP of remote device (default: 10.0.0.2)"
|
||||
echo " -u USER Username for SSH (default: pi)"
|
||||
echo
|
||||
echo "Options for 'backup':"
|
||||
echo " -o OUTPUT Path/name of the output archive (default: \$HOST-backup-\$(date +%s).tgz)"
|
||||
echo
|
||||
echo "Options for 'restore':"
|
||||
echo " -b BACKUP Path to the local backup archive to restore."
|
||||
echo
|
||||
echo "Examples:"
|
||||
echo " $0 backup"
|
||||
echo " $0 backup -n 10.0.0.2 -u pi -o my-backup.tgz"
|
||||
echo " $0 restore -b 10.0.0.2-backup-123456789.tgz"
|
||||
echo
|
||||
}
|
||||
|
||||
###############################################################################
|
||||
# BACKUP FUNCTION
|
||||
###############################################################################
|
||||
do_backup() {
|
||||
# Defaults
|
||||
UNIT_HOSTNAME="10.0.0.2"
|
||||
UNIT_USERNAME="pi"
|
||||
OUTPUT=""
|
||||
|
||||
# Parse arguments specific to backup
|
||||
while getopts "hn:u:o:" arg; do
|
||||
case $arg in
|
||||
h)
|
||||
usage
|
||||
exit 0
|
||||
;;
|
||||
n)
|
||||
UNIT_HOSTNAME=$OPTARG
|
||||
;;
|
||||
u)
|
||||
UNIT_USERNAME=$OPTARG
|
||||
;;
|
||||
o)
|
||||
OUTPUT=$OPTARG
|
||||
;;
|
||||
*)
|
||||
usage
|
||||
exit 1
|
||||
;;
|
||||
esac
|
||||
done
|
||||
|
||||
# If OUTPUT was not specified, set a default
|
||||
if [[ -z "$OUTPUT" ]]; then
|
||||
OUTPUT="${UNIT_HOSTNAME}-backup-$(date +%s).tgz"
|
||||
fi
|
||||
|
||||
# List of files/directories to backup
|
||||
# (the same list you used in your backup.sh script)
|
||||
FILES_TO_BACKUP="
|
||||
/root/settings.yaml
|
||||
/root/client_secrets.json
|
||||
/root/.api-report.json
|
||||
/root/.ssh
|
||||
/root/.bashrc
|
||||
/root/.profile
|
||||
/home/pi/handshakes
|
||||
/root/peers
|
||||
/etc/pwnagotchi/
|
||||
/usr/local/share/pwnagotchi/custom-plugins
|
||||
/etc/ssh/
|
||||
/home/pi/.bashrc
|
||||
/home/pi/.profile
|
||||
/home/pi/.wpa_sec_uploads
|
||||
"
|
||||
# Convert multiline variable into a space-separated list
|
||||
FILES_TO_BACKUP=$(echo "$FILES_TO_BACKUP" | tr '\n' ' ')
|
||||
|
||||
echo "@ Checking connectivity to $UNIT_HOSTNAME ..."
|
||||
if ! ping -c 1 "$UNIT_HOSTNAME" &>/dev/null; then
|
||||
echo "@ unit ${UNIT_HOSTNAME} can't be reached, check network or USB interface IP."
|
||||
exit 1
|
||||
fi
|
||||
|
||||
echo "@ Backing up ${UNIT_HOSTNAME} to ${OUTPUT} ..."
|
||||
# -n => do not read from stdin
|
||||
ssh -n "${UNIT_USERNAME}@${UNIT_HOSTNAME}" \
|
||||
"sudo tar -cf - ${FILES_TO_BACKUP}" \
|
||||
| gzip -9 > "${OUTPUT}"
|
||||
|
||||
echo "@ Backup finished. Archive: ${OUTPUT}"
|
||||
}
|
||||
|
||||
###############################################################################
|
||||
# RESTORE FUNCTION
|
||||
###############################################################################
|
||||
do_restore() {
|
||||
# Defaults
|
||||
UNIT_HOSTNAME="10.0.0.2"
|
||||
UNIT_USERNAME="pi"
|
||||
BACKUP_FILE=""
|
||||
|
||||
# Parse arguments specific to restore
|
||||
while getopts "hn:u:b:" arg; do
|
||||
case $arg in
|
||||
h)
|
||||
usage
|
||||
exit 0
|
||||
;;
|
||||
n)
|
||||
UNIT_HOSTNAME=$OPTARG
|
||||
;;
|
||||
u)
|
||||
UNIT_USERNAME=$OPTARG
|
||||
;;
|
||||
b)
|
||||
BACKUP_FILE=$OPTARG
|
||||
;;
|
||||
*)
|
||||
usage
|
||||
exit 1
|
||||
;;
|
||||
esac
|
||||
done
|
||||
|
||||
# If no backup file given, try to find the latest
|
||||
if [[ -z "$BACKUP_FILE" ]]; then
|
||||
BACKUP_FILE=$(ls -rt "${UNIT_HOSTNAME}"-backup-*.tgz 2>/dev/null | tail -n1)
|
||||
if [[ -z "$BACKUP_FILE" ]]; then
|
||||
echo "@ Can't find backup file. Please specify one with '-b'."
|
||||
exit 1
|
||||
fi
|
||||
echo "@ Found backup file: $BACKUP_FILE"
|
||||
echo -n "@ Continue restoring this file? (y/n) "
|
||||
read -r CONTINUE
|
||||
CONTINUE=$(echo "${CONTINUE}" | tr "[:upper:]" "[:lower:]")
|
||||
if [[ "${CONTINUE}" != "y" ]]; then
|
||||
exit 1
|
||||
fi
|
||||
fi
|
||||
|
||||
echo "@ Checking connectivity to $UNIT_HOSTNAME ..."
|
||||
if ! ping -c 1 "$UNIT_HOSTNAME" &>/dev/null; then
|
||||
echo "@ unit ${UNIT_HOSTNAME} can't be reached, make sure it's connected and a static IP assigned."
|
||||
exit 1
|
||||
fi
|
||||
|
||||
echo "@ Restoring $BACKUP_FILE to $UNIT_HOSTNAME ..."
|
||||
cat "${BACKUP_FILE}" | ssh "${UNIT_USERNAME}@${UNIT_HOSTNAME}" "sudo tar xzv -C /"
|
||||
|
||||
echo "@ Restore finished."
|
||||
}
|
||||
|
||||
###############################################################################
|
||||
# MAIN ENTRY POINT
|
||||
###############################################################################
|
||||
# We expect the first argument to be either 'backup', 'restore', or '-h'.
|
||||
SUBCOMMAND=$1
|
||||
if [[ -z "$SUBCOMMAND" ]]; then
|
||||
usage
|
||||
exit 1
|
||||
fi
|
||||
shift # Shift past subcommand
|
||||
|
||||
case "$SUBCOMMAND" in
|
||||
backup)
|
||||
do_backup "$@"
|
||||
;;
|
||||
restore)
|
||||
do_restore "$@"
|
||||
;;
|
||||
-h|--help)
|
||||
usage
|
||||
exit 0
|
||||
;;
|
||||
*)
|
||||
usage
|
||||
exit 1
|
||||
;;
|
||||
esac
|
||||
Reference in New Issue
Block a user