mirror of
https://github.com/jokob-sk/NetAlertX.git
synced 2025-12-07 09:36:05 -08:00
306 lines
11 KiB
Bash
306 lines
11 KiB
Bash
#!/bin/bash
|
|
# ------------------------------------------------------------------------------
|
|
# Pi.Alert
|
|
# Open Source Network Guard / WIFI & LAN intrusion detector
|
|
#
|
|
# pialert_update.sh - Update script
|
|
# ------------------------------------------------------------------------------
|
|
# Puche 2021 pi.alert.application@gmail.com GNU GPLv3
|
|
# ------------------------------------------------------------------------------
|
|
|
|
# ------------------------------------------------------------------------------
|
|
# Variables
|
|
# ------------------------------------------------------------------------------
|
|
INSTALL_DIR=~
|
|
PIALERT_HOME="$INSTALL_DIR/pialert"
|
|
LOG="pialert_update_`date +"%Y-%m-%d_%H-%M"`.log"
|
|
PYTHON_BIN=python
|
|
|
|
|
|
# ------------------------------------------------------------------------------
|
|
# Main
|
|
# ------------------------------------------------------------------------------
|
|
main() {
|
|
print_superheader "Pi.Alert Update"
|
|
log "`date`"
|
|
log "Logfile: $LOG"
|
|
log ""
|
|
|
|
set -e
|
|
|
|
check_pialert_home
|
|
check_python_version
|
|
|
|
create_backup
|
|
move_files
|
|
clean_files
|
|
|
|
check_packages
|
|
download_pialert
|
|
update_config
|
|
update_db
|
|
|
|
test_pialert
|
|
|
|
print_header "Update process finished"
|
|
print_msg ""
|
|
|
|
move_logfile
|
|
}
|
|
|
|
# ------------------------------------------------------------------------------
|
|
# Create backup
|
|
# ------------------------------------------------------------------------------
|
|
create_backup() {
|
|
# Previous backups are not deleted
|
|
# print_msg "- Deleting previous Pi.Alert backups..."
|
|
# rm "$INSTALL_DIR/"pialert_update_backup_*.tar 2>/dev/null || :
|
|
|
|
print_msg "- Creating new Pi.Alert backup..."
|
|
cd "$INSTALL_DIR"
|
|
tar cvf "$INSTALL_DIR"/pialert_update_backup_`date +"%Y-%m-%d_%H-%M"`.tar pialert --checkpoint=100 --checkpoint-action="ttyout=." 2>&1 >> "$LOG"
|
|
echo ""
|
|
}
|
|
|
|
# ------------------------------------------------------------------------------
|
|
# Move files to the new directory
|
|
# ------------------------------------------------------------------------------
|
|
move_files() {
|
|
if [ -e "$PIALERT_HOME/back/pialert.conf" ] ; then
|
|
print_msg "- Moving pialert.conf to the new directory..."
|
|
mkdir -p "$PIALERT_HOME/config"
|
|
mv "$PIALERT_HOME/back/pialert.conf" "$PIALERT_HOME/config"
|
|
fi
|
|
}
|
|
|
|
# ------------------------------------------------------------------------------
|
|
# Move files to the new directory
|
|
# ------------------------------------------------------------------------------
|
|
clean_files() {
|
|
print_msg "- Cleaning previous version..."
|
|
rm -r "$PIALERT_HOME/back" 2>/dev/null || :
|
|
rm -r "$PIALERT_HOME/doc" 2>/dev/null || :
|
|
rm -r "$PIALERT_HOME/docs" 2>/dev/null || :
|
|
rm -r "$PIALERT_HOME/front" 2>/dev/null || :
|
|
rm -r "$PIALERT_HOME/install" 2>/dev/null || :
|
|
rm -r "$PIALERT_HOME/"*.txt 2>/dev/null || :
|
|
rm -r "$PIALERT_HOME/"*.md 2>/dev/null || :
|
|
}
|
|
|
|
# ------------------------------------------------------------------------------
|
|
# Check packages
|
|
# ------------------------------------------------------------------------------
|
|
check_packages() {
|
|
print_msg "- Checking package apt-utils..."
|
|
sudo apt-get install apt-utils -y 2>&1 >> "$LOG"
|
|
|
|
print_msg "- Checking package sqlite3..."
|
|
sudo apt-get install sqlite3 -y 2>&1 >> "$LOG"
|
|
|
|
print_msg "- Checking packages dnsutils & net-tools..."
|
|
sudo apt-get install dnsutils net-tools -y 2>&1 >> "$LOG"
|
|
}
|
|
|
|
|
|
# ------------------------------------------------------------------------------
|
|
# Download and uncompress Pi.Alert
|
|
# ------------------------------------------------------------------------------
|
|
download_pialert() {
|
|
if [ -f "$INSTALL_DIR/pialert_latest.tar" ] ; then
|
|
print_msg "- Deleting previous downloaded tar file"
|
|
rm -r "$INSTALL_DIR/pialert_latest.tar"
|
|
fi
|
|
|
|
print_msg "- Downloading update file..."
|
|
curl -Lo "$INSTALL_DIR/pialert_latest.tar" https://github.com/pucherot/Pi.Alert/raw/main/tar/pialert_latest.tar
|
|
echo ""
|
|
|
|
print_msg "- Uncompressing tar file"
|
|
tar xf "$INSTALL_DIR/pialert_latest.tar" -C "$INSTALL_DIR" \
|
|
--exclude='pialert/config/pialert.conf' \
|
|
--exclude='pialert/db/pialert.db' \
|
|
--exclude='pialert/log/*' \
|
|
--checkpoint=100 --checkpoint-action="ttyout=." 2>&1 >> "$LOG"
|
|
echo ""
|
|
|
|
print_msg "- Deleting downloaded tar file..."
|
|
rm -r "$INSTALL_DIR/pialert_latest.tar"
|
|
}
|
|
|
|
# ------------------------------------------------------------------------------
|
|
# Update conf file
|
|
# ------------------------------------------------------------------------------
|
|
update_config() {
|
|
print_msg "- Config backup..."
|
|
cp "$PIALERT_HOME/config/pialert.conf" "$PIALERT_HOME/config/pialert.conf.back" 2>&1 >> "$LOG"
|
|
|
|
print_msg "- Updating config file..."
|
|
sed -i '/VERSION/d' "$PIALERT_HOME/config/pialert.conf" 2>&1 >> "$LOG"
|
|
sed -i 's/PA_FRONT_URL/REPORT_DEVICE_URL/g' "$PIALERT_HOME/config/pialert.conf" 2>&1 >> "$LOG"
|
|
|
|
if ! grep -Fq PIALERT_PATH "$PIALERT_HOME/config/pialert.conf" ; then
|
|
echo "PIALERT_PATH = '$PIALERT_HOME'" >> "$PIALERT_HOME/config/pialert.conf"
|
|
fi
|
|
|
|
if ! grep -Fq QUERY_MYIP_SERVER "$PIALERT_HOME/config/pialert.conf" ; then
|
|
echo "QUERY_MYIP_SERVER = 'http://ipv4.icanhazip.com'" >> "$PIALERT_HOME/config/pialert.conf"
|
|
fi
|
|
|
|
if ! grep -Fq SCAN_SUBNETS "$PIALERT_HOME/config/pialert.conf" ; then
|
|
echo "SCAN_SUBNETS = '--localnet'" >> "$PIALERT_HOME/config/pialert.conf"
|
|
fi
|
|
}
|
|
|
|
# ------------------------------------------------------------------------------
|
|
# DB DDL
|
|
# ------------------------------------------------------------------------------
|
|
update_db() {
|
|
print_msg "- Updating DB permissions..."
|
|
sudo chgrp -R www-data $PIALERT_HOME/db 2>&1 >> "$LOG"
|
|
chmod -R 770 $PIALERT_HOME/db 2>&1 >> "$LOG"
|
|
|
|
print_msg "- Installing sqlite3..."
|
|
sudo apt-get install sqlite3 -y 2>&1 >> "$LOG"
|
|
|
|
print_msg "- Checking 'Parameters' table..."
|
|
TAB=`sqlite3 $PIALERT_HOME/db/pialert.db "SELECT COUNT(*) FROM sqlite_master WHERE type='table' AND name='Parameters' COLLATE NOCASE;"` 2>&1 >> "$LOG"
|
|
if [ "$TAB" == "0" ] ; then
|
|
print_msg " - Creating 'Parameters' table..."
|
|
sqlite3 $PIALERT_HOME/db/pialert.db "CREATE TABLE Parameters (par_ID STRING (50) PRIMARY KEY NOT NULL COLLATE NOCASE, par_Value STRING (250) );" 2>&1 >> "$LOG"
|
|
sqlite3 $PIALERT_HOME/db/pialert.db "CREATE INDEX IDX_par_ID ON Parameters (par_ID COLLATE NOCASE);" 2>&1 >> "$LOG"
|
|
fi
|
|
|
|
print_msg "- Checking Devices new columns..."
|
|
COL=`sqlite3 $PIALERT_HOME/db/pialert.db "SELECT COUNT(*) FROM PRAGMA_TABLE_INFO ('Devices') WHERE name='dev_NewDevice' COLLATE NOCASE";` 2>&1 >> "$LOG"
|
|
if [ "$COL" == "0" ] ; then
|
|
print_msg " - Adding column 'NewDevice' to 'Devices'..."
|
|
sqlite3 $PIALERT_HOME/db/pialert.db "ALTER TABLE Devices ADD COLUMN dev_NewDevice BOOLEAN NOT NULL DEFAULT (1) CHECK (dev_NewDevice IN (0, 1) );" 2>&1 >> "$LOG"
|
|
sqlite3 $PIALERT_HOME/db/pialert.db "CREATE INDEX IDX_dev_NewDevice ON Devices (dev_NewDevice);"
|
|
fi
|
|
|
|
COL=`sqlite3 $PIALERT_HOME/db/pialert.db "SELECT COUNT(*) FROM PRAGMA_TABLE_INFO ('Devices') WHERE name='dev_Location' COLLATE NOCASE";` 2>&1 >> "$LOG"
|
|
if [ "$COL" == "0" ] ; then
|
|
print_msg " - Adding column 'Location' to 'Devices'..."
|
|
sqlite3 $PIALERT_HOME/db/pialert.db "ALTER TABLE Devices ADD COLUMN dev_Location STRING(250) COLLATE NOCASE;" 2>&1 >> "$LOG"
|
|
fi
|
|
|
|
COL=`sqlite3 $PIALERT_HOME/db/pialert.db "SELECT COUNT(*) FROM PRAGMA_TABLE_INFO ('Devices') WHERE name='dev_Archived' COLLATE NOCASE";` 2>&1 >> "$LOG"
|
|
if [ "$COL" == "0" ] ; then
|
|
print_msg " - Adding column 'Archived / Hidden' to 'Devices'..."
|
|
sqlite3 $PIALERT_HOME/db/pialert.db "ALTER TABLE Devices ADD COLUMN dev_Archived BOOLEAN NOT NULL DEFAULT (0) CHECK (dev_Archived IN (0, 1) );" 2>&1 >> "$LOG"
|
|
sqlite3 $PIALERT_HOME/db/pialert.db "CREATE INDEX IDX_dev_Archived ON Devices (dev_Archived);" 2>&1 >> "$LOG"
|
|
fi
|
|
|
|
print_msg "- Cheking Internet scancycle..."
|
|
sqlite3 $PIALERT_HOME/db/pialert.db "UPDATE Devices set dev_ScanCycle=1, dev_AlertEvents=1, dev_AlertDeviceDown=1 WHERE dev_MAC='Internet' AND dev_ScanCycle=0;" 2>&1 >> "$LOG"
|
|
}
|
|
|
|
# ------------------------------------------------------------------------------
|
|
# Test Pi.Alert
|
|
# ------------------------------------------------------------------------------
|
|
test_pialert() {
|
|
print_msg "- Testing Pi.Alert HW vendors database update process..."
|
|
print_msg "*** PLEASE WAIT A COUPLE OF MINUTES..."
|
|
stdbuf -i0 -o0 -e0 $PYTHON_BIN $PIALERT_HOME/back/pialert.py update_vendors_silent 2>&1 | tee -ai "$LOG"
|
|
|
|
echo ""
|
|
print_msg "- Testing Pi.Alert Internet IP Lookup..."
|
|
stdbuf -i0 -o0 -e0 $PYTHON_BIN $PIALERT_HOME/back/pialert.py internet_IP 2>&1 | tee -ai "$LOG"
|
|
|
|
echo ""
|
|
print_msg "- Testing Pi.Alert Network scan..."
|
|
print_msg "*** PLEASE WAIT A COUPLE OF MINUTES..."
|
|
stdbuf -i0 -o0 -e0 $PYTHON_BIN $PIALERT_HOME/back/pialert.py 1 2>&1 | tee -ai "$LOG"
|
|
}
|
|
|
|
# ------------------------------------------------------------------------------
|
|
# Check Pi.Alert Installation Path
|
|
# ------------------------------------------------------------------------------
|
|
check_pialert_home() {
|
|
if [ ! -e "$PIALERT_HOME" ] ; then
|
|
process_error "Pi.Alert directory dosn't exists: $PIALERT_HOME"
|
|
fi
|
|
}
|
|
|
|
# ------------------------------------------------------------------------------
|
|
# Check Python versions available
|
|
# ------------------------------------------------------------------------------
|
|
check_python_version() {
|
|
print_msg "- Checking Python..."
|
|
if [ -f /usr/bin/python ] ; then
|
|
PYTHON_BIN="python"
|
|
elif [ -f /usr/bin/python3 ] ; then
|
|
PYTHON_BIN="python3"
|
|
else
|
|
process_error "Python NOT installed"
|
|
fi
|
|
}
|
|
|
|
|
|
# ------------------------------------------------------------------------------
|
|
# Move Logfile
|
|
# ------------------------------------------------------------------------------
|
|
move_logfile() {
|
|
NEWLOG="$PIALERT_HOME/log/$LOG"
|
|
|
|
mkdir -p "$PIALERT_HOME/log"
|
|
mv $LOG $NEWLOG
|
|
|
|
LOG="$NEWLOG"
|
|
NEWLOG=""
|
|
}
|
|
|
|
# ------------------------------------------------------------------------------
|
|
# Log
|
|
# ------------------------------------------------------------------------------
|
|
log() {
|
|
echo "$1" | tee -a "$LOG"
|
|
}
|
|
|
|
log_no_screen () {
|
|
echo "$1" >> "$LOG"
|
|
}
|
|
|
|
log_only_screen () {
|
|
echo "$1"
|
|
}
|
|
|
|
print_msg() {
|
|
log_no_screen ""
|
|
log "$1"
|
|
}
|
|
|
|
print_superheader() {
|
|
log ""
|
|
log "############################################################"
|
|
log " $1"
|
|
log "############################################################"
|
|
}
|
|
|
|
print_header() {
|
|
log ""
|
|
log "------------------------------------------------------------"
|
|
log " $1"
|
|
log "------------------------------------------------------------"
|
|
}
|
|
|
|
process_error() {
|
|
log ""
|
|
log "************************************************************"
|
|
log "************************************************************"
|
|
log "** ERROR UPDATING PI.ALERT **"
|
|
log "************************************************************"
|
|
log "************************************************************"
|
|
log ""
|
|
log "$1"
|
|
log ""
|
|
log "Use 'cat $LOG' to view update log"
|
|
log ""
|
|
|
|
exit 1
|
|
}
|
|
|
|
# ------------------------------------------------------------------------------
|
|
main
|
|
exit 0
|