beenull/Pi-hole_in_High_Availability
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions
Pi-hole_in_High_Availability/pihole-gemini_master_active_script

276 lines
15 KiB
Text
Raw Permalink Normal View History

Create pihole-gemini_master_active_script
2022-10-14 21:02:03 +00:00
#! /bin/bash
# Script was last updated 03-12-2019
SCRIPTVER="pihole-gemini v0.0.2.2a"
# --------------------------------------------------------------------------------------------------------
# This script was written to try and keep the white and black lists in sync between two Pi-holes.
# The script should be launched from /opt/pihole/gravity.sh
# It should be called in the second to last line of the gravity.sh file and should be run as the user
# configured for the sync process. In my case, I'm using user pi, so you would change the user from 'pi'
# to the username you're using to sync the files (the user you're assigning to HAUSER in this script.)
# Example of the line to add as the second to last line of the /opt/pihole/gravity.sh file:
# su -c '/usr/local/bin/pihole-gemini' - pi
# In the above example, you change the 'pi' at the end to your sync user, and it should appear directly
# above the line that reads:
# "${PIHOLE_COMMAND}" status
# in the /opt/pihole/gravity.sh file.
# --------------------------------------------------------------------------------------------------------
# USER-DEFINED VARIABLES - START - Edit the values in this section to configure the script to your install.
# User account - Should have sudo and ssh access, and a "matching" account should be set up on both Pi-hole systems
HAUSER=pi
# Pi-hole directory (this path is expected to be the same on both Pi-holes, and should be where the FILES are located)
PIHOLEDIR=/etc/pihole
# IP Addresses - Note that removing the value or the variable definition of either SSHPORT will cause the script to
# use the default SSH Port of 22 for connections to that ip address.
# Primary Pi-hole ip and ssh port
PIHOLE1=192.168.1.11
SSHPORT1=22
# Secondary Pi-hole ip and ssh port
PIHOLE2=192.168.1.12
SSHPORT2=22
# Define the local path to store the logfile, DO NOT include a trailing "/". Make sure this path is writeable by the user
# that runs the sync. Change this from /tmp to another folder to prevent logs from being cleared on system restart
LOGFILEPATH="/tmp"
# How long to keep old log files (in addition to the current day's log, so a value of 6 means 7 days)
LOGKEEPDAYS=6
# USER-DEFINED VARIABLES - END - Edits made below this line may break the script.
# ---------------------------------------------------------------------------------------------------------
# SCRIPT VARIABLES - START - These values are used by the script and should not be changed manually.
# List of files to sync
FILES=(
'black.list'
'blacklist.txt'
'regex.list'
'whitelist.txt'
'lan.list'
'adlists.list'
'gravity.list'
)
# Full logfile path and name, uncluding timestamp
LOGFILE="${LOGFILEPATH}/pihole-gemini_`date +\%Y\%m\%d`.log"
# Flags used to determine script blocks to run.
# RUNUPDATE - 0 = Don't update; 1 = update Pi-holes
RUNUPDATE=0
# RESTART - 0 = Don't restart; 1 = restart
RESTART=0
# RUNGRAVITY - 0 = Don't update gravity; 1 = Update gravity (don't download lists)
RUNGRAVITY=0
# DEFAULT SSHPORT - This value is only provided here to use the default port of 22 if a port is not defined for the specified ip address
DEFAULTSSHPORT=22
# INITIALIZE PORT VARIABLES - This segment keeps the SSHPORT variables as set in the user-defined variables section. They are included here
# mainly as a fallback in case the variables are accidentally removed from the user-defined variables section.
SSHPORT1=$SSHPORT1
SSHPORT2=$SSHPORT2
# DETERMINE IP AND PORTS TO USE - START
# This section checks the interface used to connect to the local gateway, then determines the current ip address
# based on that interface. Finally, it checks the current ip address against the two addresses the script has
# configured to determine which one is the remote Pi-hole to send commands to, and configures the appropriate port.
LOCALIP=$PIHOLE1
case $LOCALIP in
$PIHOLE1)
# echo "Local ip address matches PIHOLE1 ip, so the remote Pi-hole is PIHOLE2."
REMOTEPI=$PIHOLE2
if [ -z $SSHPORT2 ]; then
# SSHPORT2 value is empty, use the default ssh port.
SSHPORT=$DEFAULTSSHPORT
else
# SSHPORT2 value exists, use that for connections to PIHOLE2.
SSHPORT=$SSHPORT2
fi
;;
$PIHOLE2)
# echo "Local ip address matches PIHOLE2 ip, so the remote Pi-hole is PIHOLE1."
REMOTEPI=$PIHOLE1
if [ -z $SSHPORT1 ]; then
# SSHPORT1 value is empty, use the default ssh port.
SSHPORT=$DEFAULTSSHPORT
else
# SSHPORT1 value exists, use that for connections to PIHOLE1.
SSHPORT=$SSHPORT1
fi
;;
*)
echo "--------------------------------------------------------------------------------------------" 2>&1 | tee -a $LOGFILE
echo "* `date '+%Y-%m-%d %H:%M:%S'` - ERROR! - $SCRIPTVER was unable to determine the local ip address." 2>&1 | tee -a $LOGFILE
echo " This is a fatal error. Script can not continue. Please check your" 2>&1 | tee -a $LOGFILE
echo " configuration for the script, and make sure you are connected to" 2>&1 | tee -a $LOGFILE
echo " your router and have an ip address assigned." 2>&1 | tee -a $LOGFILE
echo "--------------------------------------------------------------------------------------------" 2>&1 | tee -a $LOGFILE
REMOTEPI="Unknown"
exit 1
;;
esac
# DETERMINE IP AND PORTS TO USE - END
# SCRIPT VARIABLES - END
# LOG HEADER START - The following echo commands are for formatting purposes in the log file.
echo "--------------------------------------------------------------------------------------------" 2>&1 | tee -a $LOGFILE
echo "`date '+%Y-%m-%d %H:%M:%S'` - $SCRIPTVER was successfully launched as user: $USER" 2>&1 | tee -a $LOGFILE
echo " $LOCALIP is updating Pi-hole on $REMOTEPI"
echo "--------------------------------------------------------------------------------------------" 2>&1 | tee -a $LOGFILE
# LOG HEADER - END
# SSH CONNECTION TEST - START
echo "`date '+%Y-%m-%d %H:%M:%S'` - Testing SSH Connection to $HAUSER@$REMOTEPI. Please wait." 2>&1 | tee -a $LOGFILE
ssh -q -p $SSHPORT $HAUSER@$REMOTEPI exit
SSHSTATUS=$?
if [ $SSHSTATUS -eq 0 ]; then
# SSH is up
echo "`date '+%Y-%m-%d %H:%M:%S'` - SSH Connection to $HAUSER@$REMOTEPI was tested successfully." 2>&1 | tee -a $LOGFILE
RUNUPDATE=1
else
# SSH is down
echo "* `date '+%Y-%m-%d %H:%M:%S'` - ERROR! Unable to establish SSH connection as $HAUSER@$REMOTEPI on port $SSHPORT." 2>&1 | tee -a $LOGFILE
echo "* This is a fatal error as SSH is required for this script. Unable to update $REMOTEPI on port $SSHPORT." 2>&1 | tee -a $LOGFILE
echo "* $REMOTEPI may be offline, you may have specified the wrong port, or you have not correctly configured your SSH keys." 2>&1 | tee -a $LOGFILE
echo "* $REMOTEPI has NOT been updated."
RUNUPDATE=0
fi
# SSH CONNECTION TEST - END
# UPDATE REMOTE PIHOLE - START
if [ $RUNUPDATE -eq 1 ]; then
echo "--------------------------------------------------------------------------------------------" 2>&1 | tee -a $LOGFILE
# FILE SYNC - START - This section handles compares the local and remote versions of the files specified in the $FILES variable
# and updates the remote files if neccesary.
for FILE in ${FILES[@]}
do
if [[ -f $PIHOLEDIR/$FILE ]]; then
echo "`date '+%Y-%m-%d %H:%M:%S'` - Comparing local to remote $FILE and updating if neccesary." 2>&1 | tee -a $LOGFILE
RSYNC_COMMAND=$(rsync --rsync-path='/usr/bin/sudo /usr/bin/rsync' -aiu -e "ssh -l $HAUSER@$REMOTEPI -p$SSHPORT" $PIHOLEDIR/$FILE $HAUSER@$REMOTEPI:$PIHOLEDIR)
if [[ -n "${RSYNC_COMMAND}" ]]; then
# rsync copied changes so restart
case $FILE in
adlists.list)
# Updating adlists.list requires only a gravity update
echo "`date '+%Y-%m-%d %H:%M:%S'` - Updated $FILE on $REMOTEPI. Gravity will be updated on $REMOTEPI." 2>&1 | tee -a $LOGFILE
RUNGRAVITY=1
;;
gravity.list)
# Updating gravity.list requires only a gravity update
echo "`date '+%Y-%m-%d %H:%M:%S'` - Updated $FILE on $REMOTEPI. Gravity will be updated on $REMOTEPI." 2>&1 | tee -a $LOGFILE
RUNGRAVITY=1
;;
*)
# Updating white and/or black lists (or other files) requires a remote restart but not a gravity update
echo "`date '+%Y-%m-%d %H:%M:%S'` - Updated $FILE on $REMOTEPI. $REMOTEPI will be restarted." 2>&1 | tee -a $LOGFILE
RESTART=1
;;
esac
else
# no changes so do nothing
echo "`date '+%Y-%m-%d %H:%M:%S'` - Local file $FILE matches $FILE on $REMOTEPI. Remote file was not updated." 2>&1 | tee -a $LOGFILE
fi
else
# file does not exist, skipping
echo "`date '+%Y-%m-%d %H:%M:%S'` - Local file $FILE does not exist. Skipping." 2>&1 | tee -a $LOGFILE
fi
done
# FILE SYNC - END
# RESTART REMOTEPI (IF NEEDED) - START
if [ $RESTART -eq 1 ]; then
echo "--------------------------------------------------------------------------------------------" 2>&1 | tee -a $LOGFILE
echo "`date '+%Y-%m-%d %H:%M:%S'` - Updated files have been copied to $REMOTEPI. Restarting..." 2>&1 | tee -a $LOGFILE
echo "`date '+%Y-%m-%d %H:%M:%S'` - Sending restart command to pihole-FTL on $REMOTEPI." 2>&1 | tee -a $LOGFILE
ssh $HAUSER@$REMOTEPI -p $SSHPORT "sudo -S service pihole-FTL restart"
if [ $? -ne 0 ]; then
echo "* `date '+%Y-%m-%d %H:%M:%S'` - ERROR! - Unable to restart pihole-FTL service on $REMOTEPI." 2>&1 | tee -a $LOGFILE
else
echo "`date '+%Y-%m-%d %H:%M:%S'` - Successfully restarted pihole-FTL service on $REMOTEPI." 2>&1 | tee -a $LOGFILE
fi
fi
# RESTART REMOTEPI - END
# UPDATE REMOTE GRAVITY (IF NEEDED) - START
echo "--------------------------------------------------------------------------------------------" 2>&1 | tee -a $LOGFILE
case $RUNGRAVITY in
0)
# Gravity did not need updating - do nothing
echo "`date '+%Y-%m-%d %H:%M:%S'` - Gravity on $REMOTEPI did not need updating." 2>&1 | tee -a $LOGFILE
;;
1)
# Gravity needs refreshing, but not a full update - Update gravity without redownloading lists
echo "`date '+%Y-%m-%d %H:%M:%S'` - Refreshing gravity on $REMOTEPI. Lists will not be redownloaded." 2>&1 | tee -a $LOGFILE
ssh $HAUSER@$REMOTEPI -p $SSHPORT "sudo -S pihole -g --skip-download"
if [ $? -ne 0 ]; then
echo "* `date '+%Y-%m-%d %H:%M:%S'` - ERROR! - Unable to refresh gravity on $REMOTEPI." 2>&1 | tee -a $LOGFILE
else
echo "`date '+%Y-%m-%d %H:%M:%S'` - Success! Successfully refreshed gravity on $REMOTEPI." 2>&1 | tee -a $LOGFILE
fi
;;
esac
# UPDATE REMOTE GRAVITY - END
# CLEAN OLD LOG FILES - START
# Check the LOGFILEPATH for outdated log files and delete old logs
echo "--------------------------------------------------------------------------------------------" 2>&1 | tee -a $LOGFILE
echo "`date '+%Y-%m-%d %H:%M:%S'` - Checking ${LOGFILEPATH} for outdated log files." 2>&1 | tee -a $LOGFILE
look_in="${LOGFILEPATH}/pihole-gemini_*.log"
del_file=`find $look_in ! -wholename $LOGFILE -daystart -mtime +$LOGKEEPDAYS`
if [ -z "$del_file" ]; then
echo "`date '+%Y-%m-%d %H:%M:%S'` - There were no outdated log files to remove." 2>&1 | tee -a $LOGFILE
else
echo "`date '+%Y-%m-%d %H:%M:%S'` - Outdated log file(s) found. Removing..." 2>&1 | tee -a $LOGFILE
sudo find $look_in ! -wholename $LOGFILE -daystart -mtime +$LOGKEEPDAYS -prune -exec rm -rv "{}" \;
if [ $? -ne 0 ]; then
echo "`date '+%Y-%m-%d %H:%M:%S'` - ERROR! Unable to remove outdated log files from $LOGFILEPATH." 2>&1 | tee -a $LOGFILE
else
echo "`date '+%Y-%m-%d %H:%M:%S'` - Outdated log files successfully cleaned." 2>&1 | tee -a $LOGFILE
fi
fi
# CLEAN OLD LOG FILES - END
fi
# UPDATE REMOTE PIHOLE - END
# LOG FOOTER - START
echo "--------------------------------------------------------------------------------------------" 2>&1 | tee -a $LOGFILE
echo "`date '+%Y-%m-%d %H:%M:%S'` - $SCRIPTVER completed update of $REMOTEPI from $LOCALIP." 2>&1 | tee -a $LOGFILE
echo "--------------------------------------------------------------------------------------------" 2>&1 | tee -a $LOGFILE
# LOG FOOTER - END
# END OF SCRIPT
Reference in a new issue Copy permalink