2
0
mirror of https://github.com/deajan/osync synced 2024-11-11 07:10:40 +00:00
osync/dev/ofunctions.sh

1507 lines
43 KiB
Bash
Raw Normal View History

2016-08-06 11:17:56 +00:00
#### MINIMAL-FUNCTION-SET BEGIN ####
## FUNC_BUILD=2016091601
2016-08-28 09:32:25 +00:00
## BEGIN Generic bash functions written in 2013-2016 by Orsiris de Jong - http://www.netpower.fr - ozy@netpower.fr
## To use in a program, define the following variables:
## PROGRAM=program-name
## INSTANCE_ID=program-instance-name
## _DEBUG=yes/no
2015-11-12 00:37:43 +00:00
2016-09-05 20:06:57 +00:00
#TODO: Windows checks, check sendmail & mailsend
2015-11-12 00:37:43 +00:00
if ! type "$BASH" > /dev/null; then
echo "Please run this script only with bash shell. Tested on bash >= 3.2"
exit 127
fi
2016-03-19 09:48:46 +00:00
## Correct output of sort command (language agnostic sorting)
export LC_ALL=C
# Standard alert mail body
MAIL_ALERT_MSG="Execution of $PROGRAM instance $INSTANCE_ID on $(date) has warnings/errors."
2016-08-06 11:17:56 +00:00
# Environment variables that can be overriden by programs
_DRYRUN=false
_SILENT=false
_VERBOSE=false
2016-08-06 11:17:56 +00:00
_LOGGER_PREFIX="date"
_LOGGER_STDERR=false
if [ "$KEEP_LOGGING" == "" ]; then
KEEP_LOGGING=1801
fi
2015-11-12 00:37:43 +00:00
# Initial error status, logging 'WARN', 'ERROR' or 'CRITICAL' will enable alerts flags
ERROR_ALERT=false
WARN_ALERT=false
2015-11-12 00:37:43 +00:00
# Log from current run
2016-09-02 19:43:14 +00:00
CURRENT_LOG=""
2016-08-28 09:32:25 +00:00
2015-11-12 00:37:43 +00:00
## allow function call checks #__WITH_PARANOIA_DEBUG
if [ "$_PARANOIA_DEBUG" == "yes" ];then #__WITH_PARANOIA_DEBUG
_DEBUG=yes #__WITH_PARANOIA_DEBUG
fi #__WITH_PARANOIA_DEBUG
## allow debugging from command line with _DEBUG=yes
if [ ! "$_DEBUG" == "yes" ]; then
_DEBUG=no
2016-08-03 08:02:07 +00:00
SLEEP_TIME=.05 # Tested under linux and FreeBSD bash, #TODO tests on cygwin / msys
_VERBOSE=false
2015-11-12 00:37:43 +00:00
else
SLEEP_TIME=1
trap 'TrapError ${LINENO} $?' ERR
_VERBOSE=true
2015-11-12 00:37:43 +00:00
fi
SCRIPT_PID=$$
LOCAL_USER=$(whoami)
LOCAL_HOST=$(hostname)
2016-08-28 09:32:25 +00:00
if [ "$PROGRAM" == "" ]; then
PROGRAM="ofunctions"
fi
2015-11-12 00:37:43 +00:00
## Default log file until config file is loaded
if [ -w /var/log ]; then
LOG_FILE="/var/log/$PROGRAM.log"
elif ([ "$HOME" != "" ] && [ -w "$HOME" ]); then
LOG_FILE="$HOME/$PROGRAM.log"
2015-11-12 00:37:43 +00:00
else
LOG_FILE="./$PROGRAM.log"
fi
## Default directory where to store temporary run files
if [ -w /tmp ]; then
RUN_DIR=/tmp
elif [ -w /var/tmp ]; then
RUN_DIR=/var/tmp
else
RUN_DIR=.
fi
# Default alert attachment filename
ALERT_LOG_FILE="$RUN_DIR/$PROGRAM.last.log"
# Set error exit code if a piped command fails
set -o pipefail
set -o errtrace
function Dummy {
2016-02-18 19:54:07 +00:00
__CheckArguments 0 $# ${FUNCNAME[0]} "$@" #__WITH_PARANOIA_DEBUG
2016-08-06 11:17:56 +00:00
sleep $SLEEP_TIME
2015-11-12 00:37:43 +00:00
}
2016-05-26 20:11:00 +00:00
# Sub function of Logger
2015-11-12 00:37:43 +00:00
function _Logger {
local svalue="${1}" # What to log to stdout
2015-11-12 00:37:43 +00:00
local lvalue="${2:-$svalue}" # What to log to logfile, defaults to screen value
local evalue="${3}" # What to log to stderr
2016-08-06 11:17:56 +00:00
2015-11-12 00:37:43 +00:00
echo -e "$lvalue" >> "$LOG_FILE"
CURRENT_LOG="$CURRENT_LOG"$'\n'"$lvalue" #WIP
2015-11-12 00:37:43 +00:00
if [ $_LOGGER_STDERR == true ] && [ "$evalue" != "" ]; then
2016-08-03 21:58:24 +00:00
cat <<< "$evalue" 1>&2
elif [ "$_SILENT" == false ]; then
2015-11-12 00:37:43 +00:00
echo -e "$svalue"
fi
}
2016-08-29 15:37:48 +00:00
# General log function with log levels:
# CRITICAL, ERROR, WARN are colored in stdout, prefixed in stderr
# NOTICE is standard level
# VERBOSE is only sent to stdout / stderr if _VERBOSE=true
# DEBUG & PARANOIA_DEBUG are only sent if _DEBUG=yes
2015-11-12 00:37:43 +00:00
function Logger {
local value="${1}" # Sentence to log (in double quotes)
local level="${2}" # Log level: PARANOIA_DEBUG, DEBUG, VERBOSE, NOTICE, WARN, ERROR, CRITIAL
2015-11-12 00:37:43 +00:00
2016-08-03 21:58:24 +00:00
if [ "$_LOGGER_PREFIX" == "time" ]; then
prefix="TIME: $SECONDS - "
elif [ "$_LOGGER_PREFIX" == "date" ]; then
2015-11-12 00:37:43 +00:00
prefix="$(date) - "
else
2016-08-03 21:58:24 +00:00
prefix=""
2015-11-12 00:37:43 +00:00
fi
if [ "$level" == "CRITICAL" ]; then
_Logger "$prefix\e[41m$value\e[0m" "$prefix$level:$value" "$level:$value"
ERROR_ALERT=true
2015-11-12 00:37:43 +00:00
return
elif [ "$level" == "ERROR" ]; then
_Logger "$prefix\e[91m$value\e[0m" "$prefix$level:$value" "$level:$value"
ERROR_ALERT=true
2015-11-12 00:37:43 +00:00
return
elif [ "$level" == "WARN" ]; then
_Logger "$prefix\e[93m$value\e[0m" "$prefix$level:$value" "$level:$value"
WARN_ALERT=true
2015-11-12 00:37:43 +00:00
return
elif [ "$level" == "NOTICE" ]; then
_Logger "$prefix$value"
return
2016-08-29 17:03:19 +00:00
elif [ "$level" == "VERBOSE" ]; then
if [ $_VERBOSE == true ]; then
_Logger "$prefix$value"
fi
return
2015-11-12 00:37:43 +00:00
elif [ "$level" == "DEBUG" ]; then
if [ "$_DEBUG" == "yes" ]; then
_Logger "$prefix$value"
return
fi
elif [ "$level" == "PARANOIA_DEBUG" ]; then #__WITH_PARANOIA_DEBUG
if [ "$_PARANOIA_DEBUG" == "yes" ]; then #__WITH_PARANOIA_DEBUG
_Logger "$prefix$value" #__WITH_PARANOIA_DEBUG
return #__WITH_PARANOIA_DEBUG
fi #__WITH_PARANOIA_DEBUG
else
2016-08-29 17:03:19 +00:00
_Logger "\e[41mLogger function called without proper loglevel [$level].\e[0m"
2016-09-05 20:06:57 +00:00
_Logger "Value was: $prefix$value"
2015-11-12 00:37:43 +00:00
fi
}
2016-05-26 20:11:00 +00:00
# QuickLogger subfunction, can be called directly
function _QuickLogger {
local value="${1}"
local destination="${2}" # Destination: stdout, log, both
2016-08-06 11:17:56 +00:00
__CheckArguments 2 $# ${FUNCNAME[0]} "$@" #__WITH_PARANOIA_DEBUG
2016-05-26 20:11:00 +00:00
if ([ "$destination" == "log" ] || [ "$destination" == "both" ]); then
echo -e "$(date) - $value" >> "$LOG_FILE"
elif ([ "$destination" == "stdout" ] || [ "$destination" == "both" ]); then
echo -e "$value"
fi
}
# Generic quick logging function
function QuickLogger {
local value="${1}"
2016-08-06 11:17:56 +00:00
__CheckArguments 1 $# ${FUNCNAME[0]} "$@" #__WITH_PARANOIA_DEBUG
if [ $_SILENT == true ]; then
2016-05-26 20:11:00 +00:00
_QuickLogger "$value" "log"
else
_QuickLogger "$value" "stdout"
fi
}
2015-11-12 00:37:43 +00:00
# Portable child (and grandchild) kill function tester under Linux, BSD and MacOS X
function KillChilds {
local pid="${1}" # Parent pid to kill childs
local self="${2:-false}" # Should parent be killed too ?
2015-11-12 00:37:43 +00:00
2016-08-03 08:02:07 +00:00
2016-08-03 21:58:24 +00:00
if children="$(pgrep -P "$pid")"; then
for child in $children; do
Logger "Launching KillChilds \"$child\" true" "DEBUG" #__WITH_PARANOIA_DEBUG
KillChilds "$child" true
done
fi
2016-08-03 08:02:07 +00:00
# Try to kill nicely, if not, wait 15 seconds to let Trap actions happen before killing
2016-08-03 21:58:24 +00:00
if ( [ "$self" == true ] && kill -0 $pid > /dev/null 2>&1); then
Logger "Sending SIGTERM to process [$pid]." "DEBUG"
2016-08-22 05:55:03 +00:00
kill -s TERM "$pid"
2016-08-03 21:58:24 +00:00
if [ $? != 0 ]; then
sleep 15
Logger "Sending SIGTERM to process [$pid] failed." "DEBUG"
kill -9 "$pid"
2016-04-12 08:16:21 +00:00
if [ $? != 0 ]; then
2016-08-03 21:58:24 +00:00
Logger "Sending SIGKILL to process [$pid] failed." "DEBUG"
return 1
2016-04-12 08:16:21 +00:00
fi
2016-08-03 08:02:07 +00:00
else
2016-08-03 21:58:24 +00:00
return 0
fi
2016-08-03 21:58:24 +00:00
else
return 0
fi
}
function KillAllChilds {
local pids="${1}" # List of parent pids to kill separated by semi-colon
local self="${2:-false}" # Should parent be killed too ?
2016-08-03 21:58:24 +00:00
2016-08-06 11:17:56 +00:00
__CheckArguments 1 $# ${FUNCNAME[0]} "$@" #__WITH_PARANOIA_DEBUG
2016-08-03 21:58:24 +00:00
local errorcount=0
IFS=';' read -a pidsArray <<< "$pids"
for pid in "${pidsArray[@]}"; do
KillChilds $pid $self
2016-08-03 21:58:24 +00:00
if [ $? != 0 ]; then
errorcount=$((errorcount+1))
fi
2016-08-03 08:02:07 +00:00
done
return $errorcount
2015-11-12 00:37:43 +00:00
}
# osync/obackup/pmocr script specific mail alert function, use SendEmail function for generic mail sending
2015-11-12 00:37:43 +00:00
function SendAlert {
2016-08-28 09:32:25 +00:00
local runAlert="${1:-false}" # Specifies if current message is sent while running or at the end of a run
__CheckArguments 0-1 $# ${FUNCNAME[0]} "$@" #__WITH_PARANOIA_DEBUG
2016-02-18 19:54:07 +00:00
2016-02-18 22:28:57 +00:00
local mail_no_attachment=
2016-02-18 19:54:07 +00:00
local attachment_command=
local subject=
2016-08-28 09:32:25 +00:00
local body=
2015-11-12 00:37:43 +00:00
# Windows specific settings
local encryption_string=
local auth_string=
if [ "$DESTINATION_MAILS" == "" ]; then
return 0
fi
2015-11-12 00:37:43 +00:00
if [ "$_DEBUG" == "yes" ]; then
Logger "Debug mode, no warning mail will be sent." "NOTICE"
2015-11-12 00:37:43 +00:00
return 0
fi
# <OSYNC SPECIFIC>
if [ "$_QUICK_SYNC" -eq 2 ]; then
2015-11-12 00:37:43 +00:00
Logger "Current task is a quicksync task. Will not send any alert." "NOTICE"
return 0
fi
# </OSYNC SPECIFIC>
eval "cat \"$LOG_FILE\" $COMPRESSION_PROGRAM > $ALERT_LOG_FILE"
2016-02-16 21:15:44 +00:00
if [ $? != 0 ]; then
Logger "Cannot create [$ALERT_LOG_FILE]" "WARN"
mail_no_attachment=1
else
mail_no_attachment=0
fi
2016-08-28 09:32:25 +00:00
body="$MAIL_ALERT_MSG"$'\n\n'"$CURRENT_LOG"
if [ $ERROR_ALERT == true ]; then
2015-11-12 00:37:43 +00:00
subject="Error alert for $INSTANCE_ID"
elif [ $WARN_ALERT == true ]; then
2015-11-12 00:37:43 +00:00
subject="Warning alert for $INSTANCE_ID"
else
subject="Alert for $INSTANCE_ID"
fi
2016-08-28 09:32:25 +00:00
if [ $runAlert == true ]; then
subject="Currently runing - $subject"
else
subject="Fnished run - $subject"
fi
2016-02-18 19:54:07 +00:00
if [ "$mail_no_attachment" -eq 0 ]; then
2016-02-16 21:15:44 +00:00
attachment_command="-a $ALERT_LOG_FILE"
fi
2015-11-12 00:37:43 +00:00
if type mutt > /dev/null 2>&1 ; then
2016-08-28 09:32:25 +00:00
echo "$body" | $(type -p mutt) -x -s "$subject" $DESTINATION_MAILS $attachment_command
2015-11-12 00:37:43 +00:00
if [ $? != 0 ]; then
Logger "Cannot send alert mail via $(type -p mutt) !!!" "WARN"
2015-11-12 00:37:43 +00:00
else
Logger "Sent alert mail using mutt." "NOTICE"
return 0
fi
fi
if type mail > /dev/null 2>&1 ; then
2016-02-18 19:54:07 +00:00
if [ "$mail_no_attachment" -eq 0 ] && $(type -p mail) -V | grep "GNU" > /dev/null; then
attachment_command="-A $ALERT_LOG_FILE"
2016-02-18 19:54:07 +00:00
elif [ "$mail_no_attachment" -eq 0 ] && $(type -p mail) -V > /dev/null; then
2016-03-31 10:01:10 +00:00
attachment_command="-a$ALERT_LOG_FILE"
else
attachment_command=""
2015-12-15 12:45:11 +00:00
fi
2016-08-28 09:32:25 +00:00
echo "$body" | $(type -p mail) $attachment_command -s "$subject" $DESTINATION_MAILS
2015-11-12 00:37:43 +00:00
if [ $? != 0 ]; then
Logger "Cannot send alert mail via $(type -p mail) with attachments !!!" "WARN"
2016-08-28 09:32:25 +00:00
echo "$body" | $(type -p mail) -s "$subject" $DESTINATION_MAILS
2015-11-12 00:37:43 +00:00
if [ $? != 0 ]; then
Logger "Cannot send alert mail via $(type -p mail) without attachments !!!" "WARN"
2015-11-12 00:37:43 +00:00
else
Logger "Sent alert mail using mail command without attachment." "NOTICE"
return 0
fi
else
Logger "Sent alert mail using mail command." "NOTICE"
return 0
fi
fi
if type sendmail > /dev/null 2>&1 ; then
2016-08-28 09:32:25 +00:00
echo -e "Subject:$subject\r\n$body" | $(type -p sendmail) $DESTINATION_MAILS
2015-11-12 00:37:43 +00:00
if [ $? != 0 ]; then
Logger "Cannot send alert mail via $(type -p sendmail) !!!" "WARN"
2015-11-12 00:37:43 +00:00
else
Logger "Sent alert mail using sendmail command without attachment." "NOTICE"
return 0
fi
fi
# Windows specific
2016-08-08 16:06:31 +00:00
if type "mailsend.exe" > /dev/null 2>&1 ; then
if [ "$SMTP_ENCRYPTION" != "tls" ] && [ "$SMTP_ENCRYPTION" != "ssl" ] && [ "$SMTP_ENCRYPTION" != "none" ]; then
Logger "Bogus smtp encryption, assuming none." "WARN"
encryption_string=
elif [ "$SMTP_ENCRYPTION" == "tls" ]; then
encryption_string=-starttls
elif [ "$SMTP_ENCRYPTION" == "ssl" ]:; then
encryption_string=-ssl
fi
if [ "$SMTP_USER" != "" ] && [ "$SMTP_USER" != "" ]; then
auth_string="-auth -user \"$SMTP_USER\" -pass \"$SMTP_PASSWORD\""
fi
2016-08-28 09:32:25 +00:00
$(type mailsend.exe) -f $SENDER_MAIL -t "$DESTINATION_MAILS" -sub "$subject" -M "$body" -attach "$attachment" -smtp "$SMTP_SERVER" -port "$SMTP_PORT" $encryption_string $auth_string
2016-08-08 16:06:31 +00:00
if [ $? != 0 ]; then
Logger "Cannot send mail via $(type mailsend.exe) !!!" "WARN"
else
Logger "Sent mail using mailsend.exe command with attachment." "NOTICE"
return 0
fi
fi
# Windows specific, kept for compatibility (sendemail from http://caspian.dotconf.net/menu/Software/SendEmail/)
2015-11-12 00:37:43 +00:00
if type sendemail > /dev/null 2>&1 ; then
if [ "$SMTP_USER" != "" ] && [ "$SMTP_PASSWORD" != "" ]; then
SMTP_OPTIONS="-xu $SMTP_USER -xp $SMTP_PASSWORD"
else
SMTP_OPTIONS=""
fi
2016-08-28 09:32:25 +00:00
$(type -p sendemail) -f $SENDER_MAIL -t "$DESTINATION_MAILS" -u "$subject" -m "$body" -s $SMTP_SERVER $SMTP_OPTIONS > /dev/null 2>&1
2015-11-12 00:37:43 +00:00
if [ $? != 0 ]; then
Logger "Cannot send alert mail via $(type -p sendemail) !!!" "WARN"
2015-11-12 00:37:43 +00:00
else
Logger "Sent alert mail using sendemail command without attachment." "NOTICE"
return 0
fi
fi
# pfSense specific
if [ -f /usr/local/bin/mail.php ]; then
2016-08-28 09:32:25 +00:00
echo "$body" | /usr/local/bin/mail.php -s="$subject"
if [ $? != 0 ]; then
Logger "Cannot send alert mail via /usr/local/bin/mail.php (pfsense) !!!" "WARN"
else
Logger "Sent alert mail using pfSense mail.php." "NOTICE"
return 0
fi
fi
# If function has not returned 0 yet, assume it is critical that no alert can be sent
Logger "Cannot send alert (neither mutt, mail, sendmail, mailsend, sendemail or pfSense mail.php could be used)." "ERROR" # Is not marked critical because execution must continue
2015-11-12 00:37:43 +00:00
# Delete tmp log file
if [ -f "$ALERT_LOG_FILE" ]; then
rm "$ALERT_LOG_FILE"
fi
}
# Generic email sending function.
# Usage (linux / BSD), attachment is optional, can be "/path/to/my.file" or ""
# SendEmail "subject" "Body text" "receiver@example.com receiver2@otherdomain.com" "/path/to/attachment.file"
# Usage (Windows, make sure you have mailsend.exe in executable path, see http://github.com/muquit/mailsend)
# attachment is optional but must be in windows format like "c:\\some\path\\my.file", or ""
# smtp_server.domain.tld is mandatory, as is smtp_port (should be 25, 465 or 587)
# encryption can be set to tls, ssl or none
# smtp_user and smtp_password are optional
2016-07-19 08:52:43 +00:00
# SendEmail "subject" "Body text" "receiver@example.com receiver2@otherdomain.com" "/path/to/attachment.file" "sender_email@example.com" "smtp_server.domain.tld" "smtp_port" "encryption" "smtp_user" "smtp_password"
2016-04-14 16:01:56 +00:00
function SendEmail {
local subject="${1}"
local message="${2}"
local destination_mails="${3}"
local attachment="${4}"
local sender_email="${5}"
local smtp_server="${6}"
local smtp_port="${7}"
2016-07-19 08:52:43 +00:00
local encryption="${8}"
local smtp_user="${9}"
local smtp_password="${10}"
# CheckArguments will report a warning that can be ignored if used in Windows with paranoia debug enabled
2016-04-14 16:01:56 +00:00
__CheckArguments 4 $# ${FUNCNAME[0]} "$@" #__WITH_PARANOIA_DEBUG
local mail_no_attachment=
local attachment_command=
local encryption_string=
local auth_string=
2016-04-14 16:01:56 +00:00
if [ ! -f "$attachment" ]; then
attachment_command="-a $ALERT_LOG_FILE"
mail_no_attachment=1
else
mail_no_attachment=0
fi
if type mutt > /dev/null 2>&1 ; then
echo "$message" | $(type -p mutt) -x -s "$subject" "$destination_mails" $attachment_command
if [ $? != 0 ]; then
Logger "Cannot send mail via $(type -p mutt) !!!" "WARN"
2016-04-14 16:01:56 +00:00
else
Logger "Sent mail using mutt." "NOTICE"
2016-04-14 16:01:56 +00:00
return 0
fi
fi
if type mail > /dev/null 2>&1 ; then
if [ "$mail_no_attachment" -eq 0 ] && $(type -p mail) -V | grep "GNU" > /dev/null; then
attachment_command="-A $attachment"
elif [ "$mail_no_attachment" -eq 0 ] && $(type -p mail) -V > /dev/null; then
attachment_command="-a$attachment"
else
attachment_command=""
fi
echo "$message" | $(type -p mail) $attachment_command -s "$subject" "$destination_mails"
if [ $? != 0 ]; then
Logger "Cannot send mail via $(type -p mail) with attachments !!!" "WARN"
2016-04-14 16:01:56 +00:00
echo "$message" | $(type -p mail) -s "$subject" "$destination_mails"
if [ $? != 0 ]; then
Logger "Cannot send mail via $(type -p mail) without attachments !!!" "WARN"
2016-04-14 16:01:56 +00:00
else
Logger "Sent mail using mail command without attachment." "NOTICE"
2016-04-14 16:01:56 +00:00
return 0
fi
else
Logger "Sent mail using mail command." "NOTICE"
2016-04-14 16:01:56 +00:00
return 0
fi
fi
if type sendmail > /dev/null 2>&1 ; then
echo -e "Subject:$subject\r\n$message" | $(type -p sendmail) "$destination_mails"
if [ $? != 0 ]; then
Logger "Cannot send mail via $(type -p sendmail) !!!" "WARN"
2016-04-14 16:01:56 +00:00
else
Logger "Sent mail using sendmail command without attachment." "NOTICE"
2016-04-14 16:01:56 +00:00
return 0
fi
fi
# Windows specific
2016-08-08 16:06:31 +00:00
if type "mailsend.exe" > /dev/null 2>&1 ; then
if [ "$sender_email" == "" ]; then
Logger "Missing sender email." "ERROR"
return 1
fi
if [ "$smtp_server" == "" ]; then
Logger "Missing smtp port." "ERROR"
return 1
fi
if [ "$smtp_port" == "" ]; then
Logger "Missing smtp port, assuming 25." "WARN"
smtp_port=25
fi
if [ "$encryption" != "tls" ] && [ "$encryption" != "ssl" ] && [ "$encryption" != "none" ]; then
Logger "Bogus smtp encryption, assuming none." "WARN"
encryption_string=
elif [ "$encryption" == "tls" ]; then
encryption_string=-starttls
elif [ "$encryption" == "ssl" ]:; then
encryption_string=-ssl
fi
if [ "$smtp_user" != "" ] && [ "$smtp_password" != "" ]; then
auth_string="-auth -user \"$smtp_user\" -pass \"$smtp_password\""
fi
2016-08-08 16:06:31 +00:00
$(type mailsend.exe) -f "$sender_email" -t "$destination_mails" -sub "$subject" -M "$message" -attach "$attachment" -smtp "$smtp_server" -port "$smtp_port" $encryption_string $auth_string
if [ $? != 0 ]; then
Logger "Cannot send mail via $(type mailsend.exe) !!!" "WARN"
else
Logger "Sent mail using mailsend.exe command with attachment." "NOTICE"
return 0
fi
fi
2016-04-14 16:01:56 +00:00
# pfSense specific
if [ -f /usr/local/bin/mail.php ]; then
echo "$message" | /usr/local/bin/mail.php -s="$subject"
if [ $? != 0 ]; then
Logger "Cannot send mail via /usr/local/bin/mail.php (pfsense) !!!" "WARN"
2016-04-14 16:01:56 +00:00
else
Logger "Sent mail using pfSense mail.php." "NOTICE"
2016-04-14 16:01:56 +00:00
return 0
fi
fi
# If function has not returned 0 yet, assume it is critical that no alert can be sent
Logger "Cannot send mail (neither mutt, mail, sendmail, sendemail, mailsend (windows) or pfSense mail.php could be used)." "ERROR" # Is not marked critical because execution must continue
2016-04-14 16:01:56 +00:00
}
function TrapError {
local job="$0"
local line="$1"
local code="${2:-1}"
if [ $_SILENT == false ]; then
echo -e " /!\ ERROR in ${job}: Near line ${line}, exit code ${code}"
fi
}
2016-03-19 09:48:46 +00:00
function LoadConfigFile {
2016-09-06 17:08:53 +00:00
local configFile="${1}"
2016-03-19 09:48:46 +00:00
__CheckArguments 1 $# ${FUNCNAME[0]} "$@" #__WITH_PARANOIA_DEBUG
2016-09-06 17:08:53 +00:00
if [ ! -f "$configFile" ]; then
Logger "Cannot load configuration file [$configFile]. Cannot start." "CRITICAL"
2016-03-19 09:48:46 +00:00
exit 1
2016-09-06 17:08:53 +00:00
elif [[ "$configFile" != *".conf" ]]; then
Logger "Wrong configuration file supplied [$configFile]. Cannot start." "CRITICAL"
2016-03-19 09:48:46 +00:00
exit 1
else
2016-09-06 17:08:53 +00:00
# Remove everything that is not a variable assignation
grep '^[^ ]*=[^;&]*' "$configFile" > "$RUN_DIR/$PROGRAM.${FUNCNAME[0]}.$SCRIPT_PID"
2016-03-19 09:48:46 +00:00
source "$RUN_DIR/$PROGRAM.${FUNCNAME[0]}.$SCRIPT_PID"
fi
2016-09-06 17:08:53 +00:00
CONFIG_FILE="$configFile"
2016-03-19 09:48:46 +00:00
}
function Spinner {
if [ $_SILENT == true ]; then
return 0
fi
case $toggle
in
1)
echo -n " \ "
echo -ne "\r"
toggle="2"
;;
2)
echo -n " | "
echo -ne "\r"
toggle="3"
;;
3)
echo -n " / "
echo -ne "\r"
toggle="4"
;;
*)
echo -n " - "
echo -ne "\r"
toggle="1"
;;
esac
}
2016-08-08 16:06:31 +00:00
# Array to string converter, see http://stackoverflow.com/questions/1527049/bash-join-elements-of-an-array
# usage: joinString separaratorChar Array
function joinString {
local IFS="$1"; shift; echo "$*";
}
2016-08-09 20:14:17 +00:00
# Time control function for background processes, suitable for multiple synchronous processes
# Fills a global variable called WAIT_FOR_TASK_COMPLETION that contains list of failed pids in format pid1:result1;pid2:result2
# Warning: Don't imbricate this function into another run if you plan to use the global variable output
2016-08-28 09:32:25 +00:00
2016-08-06 11:17:56 +00:00
function WaitForTaskCompletion {
local pids="${1}" # pids to wait for, separated by semi-colon
local soft_max_time="${2}" # If program with pid $pid takes longer than $soft_max_time seconds, will log a warning, unless $soft_max_time equals 0.
local hard_max_time="${3}" # If program with pid $pid takes longer than $hard_max_time seconds, will stop execution, unless $hard_max_time equals 0.
local caller_name="${4}" # Who called this function
local counting="${5:-true}" # Count time since function has been launched if true, since script has been launched if false
local keep_logging="${6:-0}" # Log a standby message every X seconds. Set to zero to disable logging
2016-08-06 11:17:56 +00:00
Logger "${FUNCNAME[0]} called by [$caller_name]." "PARANOIA_DEBUG" #__WITH_PARANOIA_DEBUG
2016-08-07 22:16:38 +00:00
__CheckArguments 6 $# ${FUNCNAME[0]} "$@" #__WITH_PARANOIA_DEBUG
2016-08-06 11:17:56 +00:00
2016-08-30 09:49:21 +00:00
local soft_alert=false # Does a soft alert need to be triggered, if yes, send an alert once
2016-08-06 11:17:56 +00:00
local log_ttime=0 # local time instance for comparaison
local seconds_begin=$SECONDS # Seconds since the beginning of the script
local exec_time=0 # Seconds since the beginning of this function
local retval=0 # return value of monitored pid process
local errorcount=0 # Number of pids that finished with errors
2016-08-28 15:54:20 +00:00
local pid # Current pid working on
2016-08-06 11:17:56 +00:00
local pidCount # number of given pids
local pidState # State of the process
2016-08-06 11:17:56 +00:00
2016-08-28 15:54:20 +00:00
local pidsArray # Array of currently running pids
local newPidsArray # New array of currently running pids
2016-08-30 09:49:21 +00:00
local hasPids=false # Are any valable pids given to function ? #__WITH_PARANOIA_DEBUG
2016-08-06 11:17:56 +00:00
IFS=';' read -a pidsArray <<< "$pids"
pidCount=${#pidsArray[@]}
2016-08-09 20:14:17 +00:00
WAIT_FOR_TASK_COMPLETION=""
2016-08-06 11:17:56 +00:00
while [ ${#pidsArray[@]} -gt 0 ]; do
newPidsArray=()
Spinner
2016-08-07 21:54:46 +00:00
if [ $counting == true ]; then
exec_time=$(($SECONDS - $seconds_begin))
else
exec_time=$SECONDS
fi
if [ $keep_logging -ne 0 ]; then
if [ $((($exec_time + 1) % $keep_logging)) -eq 0 ]; then
if [ $log_ttime -ne $exec_time ]; then # Fix when sleep time lower than 1s
log_ttime=$exec_time
Logger "Current tasks still running with pids [$(joinString , ${pidsArray[@]})]." "NOTICE"
fi
2016-08-06 11:17:56 +00:00
fi
fi
if [ $exec_time -gt $soft_max_time ]; then
2016-08-30 09:49:21 +00:00
if [ $soft_alert == true ] && [ $soft_max_time -ne 0 ]; then
2016-08-08 16:06:31 +00:00
Logger "Max soft execution time exceeded for task [$caller_name] with pids [$(joinString , ${pidsArray[@]})]." "WARN"
2016-08-30 09:49:21 +00:00
soft_alert=true
2016-08-28 09:32:25 +00:00
SendAlert true
2016-08-06 11:17:56 +00:00
fi
if [ $exec_time -gt $hard_max_time ] && [ $hard_max_time -ne 0 ]; then
2016-08-08 16:06:31 +00:00
Logger "Max hard execution time exceeded for task [$caller_name] with pids [$(joinString , ${pidsArray[@]})]. Stopping task execution." "ERROR"
for pid in "${pidsArray[@]}"; do
KillChilds $pid true
if [ $? == 0 ]; then
Logger "Task with pid [$pid] stopped successfully." "NOTICE"
else
Logger "Could not stop task with pid [$pid]." "ERROR"
fi
done
2016-08-28 09:32:25 +00:00
SendAlert true
2016-08-06 11:17:56 +00:00
fi
fi
for pid in "${pidsArray[@]}"; do
2016-09-05 20:06:57 +00:00
if [ $(IsInteger $pid) -eq 1 ]; then
if kill -0 $pid > /dev/null 2>&1; then
# Handle uninterruptible sleep state or zombies by ommiting them from running process array (How to kill that is already dead ? :)
#TODO(high): have this tested on *BSD, Mac & Win
pidState=$(ps -p$pid -o state= 2 > /dev/null)
if [ "$pidState" != "D" ] && [ "$pidState" != "Z" ]; then
newPidsArray+=($pid)
fi
else
# pid is dead, get it's exit code from wait command
wait $pid
retval=$?
if [ $retval -ne 0 ]; then
errorcount=$((errorcount+1))
Logger "${FUNCNAME[0]} called by [$caller_name] finished monitoring [$pid] with exitcode [$retval]." "DEBUG"
if [ "$WAIT_FOR_TASK_COMPLETION" == "" ]; then
WAIT_FOR_TASK_COMPLETION="$pid:$retval"
else
WAIT_FOR_TASK_COMPLETION=";$pid:$retval"
fi
fi
fi
2016-08-30 09:49:21 +00:00
hasPids=true ##__WITH_PARANOIA_DEBUG
fi
done
2016-08-30 09:49:21 +00:00
if [ $hasPids == false ]; then ##__WITH_PARANOIA_DEBUG
Logger "No valable pids given." "ERROR" ##__WITH_PARANOIA_DEBUG
fi ##__WITH_PARANOIA_DEBUG
2016-08-06 11:17:56 +00:00
pidsArray=("${newPidsArray[@]}")
2016-08-28 09:32:25 +00:00
# Trivial wait time for bash to not eat up all CPU
2016-08-06 11:17:56 +00:00
sleep $SLEEP_TIME
done
Logger "${FUNCNAME[0]} ended for [$caller_name] using [$pidCount] subprocesses with [$errorcount] errors." "PARANOIA_DEBUG" #__WITH_PARANOIA_DEBUG
# Return exit code if only one process was monitored, else return number of errors
if [ $pidCount -eq 1 ] && [ $errorcount -eq 0 ]; then
return $errorcount
2016-08-06 11:17:56 +00:00
else
return $errorcount
fi
}
2016-08-28 09:32:25 +00:00
# Take a list of commands to run, runs them sequentially with numberOfProcesses commands simultaneously runs
# Returns the number of non zero exit codes from commands
2016-09-06 17:27:45 +00:00
# Use cmd1;cmd2;cmd3 syntax for small sets, use file for large command sets
2016-08-28 09:32:25 +00:00
function ParallelExec {
local numberOfProcesses="${1}" # Number of simultaneous commands to run
2016-09-06 17:08:53 +00:00
local commandsArg="${2}" # Semi-colon separated list of commands, or file containing one command per line
2016-09-08 20:49:51 +00:00
local readFromFile="${3:-false}" # Is commandsArg a file or a string ?
__CheckArguments 3 $# ${FUNCNAME[0]} "$@" #__WITH_PARANOIA_DEBUG
2016-08-28 09:32:25 +00:00
2016-09-06 17:08:53 +00:00
local commandCount
local command
2016-08-28 09:32:25 +00:00
local pid
local counter=0
local commandsArray
local pidsArray
local newPidsArray
local retval
2016-09-05 20:06:57 +00:00
local errorCount=0
2016-08-28 09:32:25 +00:00
local pidState
local commandsArrayPid
2016-08-30 09:49:21 +00:00
local hasPids=false # Are any valable pids given to function ? #__WITH_PARANOIA_DEBUG
2016-09-08 20:49:51 +00:00
if [ $readFromFile == true ];then
if [ -f "$commandsArg" ]; then
commandCount=$(wc -l < "$commandsArg")
else
commandCount=0
fi
2016-09-06 17:08:53 +00:00
else
IFS=';' read -r -a commandsArray <<< "$commandsArg"
commandCount=${#commandsArray[@]}
fi
2016-08-28 09:32:25 +00:00
2016-09-06 17:08:53 +00:00
Logger "Runnning $commandCount commands in $numberOfProcesses simultaneous processes." "DEBUG"
2016-08-28 09:32:25 +00:00
2016-09-06 17:08:53 +00:00
while [ $counter -lt "$commandCount" ] || [ ${#pidsArray[@]} -gt 0 ]; do
2016-08-28 09:32:25 +00:00
2016-09-06 17:08:53 +00:00
while [ $counter -lt "$commandCount" ] && [ ${#pidsArray[@]} -lt $numberOfProcesses ]; do
if [ $readFromFile == true ]; then
2016-09-08 20:49:51 +00:00
#TODO: Checked on FreeBSD 10, also check on Win
2016-09-06 17:08:53 +00:00
command=$(awk 'NR == num_line {print; exit}' num_line=$((counter+1)) "$commandsArg")
else
command="${commandsArray[$counter]}"
fi
Logger "Running command [$command]." "DEBUG"
eval "$command" &
2016-08-28 09:32:25 +00:00
pid=$!
pidsArray+=($pid)
2016-09-06 17:08:53 +00:00
commandsArrayPid[$pid]="$command"
2016-08-28 09:32:25 +00:00
counter=$((counter+1))
done
newPidsArray=()
for pid in "${pidsArray[@]}"; do
2016-09-05 20:06:57 +00:00
if [ $(IsInteger $pid) -eq 1 ]; then
# Handle uninterruptible sleep state or zombies by ommiting them from running process array (How to kill that is already dead ? :)
if kill -0 $pid > /dev/null 2>&1; then
pidState=$(ps -p$pid -o state= 2 > /dev/null)
if [ "$pidState" != "D" ] && [ "$pidState" != "Z" ]; then
newPidsArray+=($pid)
fi
else
# pid is dead, get it's exit code from wait command
wait $pid
retval=$?
if [ $retval -ne 0 ]; then
Logger "Command [${commandsArrayPid[$pid]}] failed with exit code [$retval]." "ERROR"
2016-09-05 20:06:57 +00:00
errorCount=$((errorCount+1))
fi
2016-08-28 09:32:25 +00:00
fi
2016-08-30 09:49:21 +00:00
hasPids=true ##__WITH_PARANOIA_DEBUG
2016-08-28 09:32:25 +00:00
fi
done
2016-08-30 09:49:21 +00:00
if [ $hasPids == false ]; then ##__WITH_PARANOIA_DEBUG
Logger "No valable pids given." "ERROR" ##__WITH_PARANOIA_DEBUG
fi ##__WITH_PARANOIA_DEBUG
2016-08-28 09:32:25 +00:00
pidsArray=("${newPidsArray[@]}")
# Trivial wait time for bash to not eat up all CPU
sleep $SLEEP_TIME
done
2016-09-05 20:06:57 +00:00
return $errorCount
2016-08-28 09:32:25 +00:00
}
2016-08-07 21:54:46 +00:00
function CleanUp {
__CheckArguments 0 $# ${FUNCNAME[0]} "$@" #__WITH_PARANOIA_DEBUG
2016-08-06 11:17:56 +00:00
2016-08-07 21:54:46 +00:00
if [ "$_DEBUG" != "yes" ]; then
rm -f "$RUN_DIR/$PROGRAM."*".$SCRIPT_PID"
# Fix for sed -i requiring backup extension for BSD & Mac (see all sed -i statements)
rm -f "$RUN_DIR/$PROGRAM."*".$SCRIPT_PID.tmp"
fi
2016-08-06 11:17:56 +00:00
}
2016-04-14 16:01:56 +00:00
# obsolete, use StripQuotes
function SedStripQuotes {
2016-08-08 16:06:31 +00:00
echo $(echo $1 | sed "s/^\([\"']\)\(.*\)\1\$/\2/g")
}
2016-04-14 16:01:56 +00:00
# Usage: var=$(StripSingleQuotes "$var")
function StripSingleQuotes {
local string="${1}"
string="${string/#\'/}" # Remove singlequote if it begins string
string="${string/%\'/}" # Remove singlequote if it ends string
echo "$string"
}
2016-04-14 16:01:56 +00:00
# Usage: var=$(StripDoubleQuotes "$var")
function StripDoubleQuotes {
local string="${1}"
string="${string/#\"/}"
string="${string/%\"/}"
echo "$string"
}
function StripQuotes {
local string="${1}"
echo "$(StripSingleQuotes $(StripDoubleQuotes $string))"
}
2016-09-05 20:06:57 +00:00
# Usage var=$(EscapeSpaces "$var") or var="$(EscapeSpaces "$var")"
function EscapeSpaces {
local string="${1}" # String on which spaces will be escaped
2016-09-05 20:06:57 +00:00
echo "${string// /\\ }"
}
2016-09-05 20:06:57 +00:00
function IsNumericExpand {
eval "local value=\"${1}\"" # Needed eval so variable variables can be processed
local re="^-?[0-9]+([.][0-9]+)?$"
if [[ $value =~ $re ]]; then
echo 1
else
echo 0
fi
}
2016-09-05 20:06:57 +00:00
function IsNumeric {
local value="${1}"
if [[ $value =~ ^[0-9]+([.][0-9]+)?$ ]]; then
echo 1
else
echo 0
fi
}
function IsInteger {
local value="${1}"
if [[ $value =~ ^[0-9]+$ ]]; then
echo 1
else
echo 0
fi
}
2016-05-26 20:11:00 +00:00
## from https://gist.github.com/cdown/1163649
function urlEncode {
local length="${#1}"
local LANG=C
for (( i = 0; i < length; i++ )); do
local c="${1:i:1}"
case $c in
[a-zA-Z0-9.~_-])
printf "$c"
;;
*)
printf '%%%02X' "'$c"
;;
esac
done
}
function urlDecode {
local url_encoded="${1//+/ }"
2016-05-26 20:11:00 +00:00
printf '%b' "${url_encoded//%/\\x}"
2016-05-26 20:11:00 +00:00
}
2015-11-12 00:37:43 +00:00
function GetLocalOS {
2016-02-18 19:54:07 +00:00
__CheckArguments 0 $# ${FUNCNAME[0]} "$@" #__WITH_PARANOIA_DEBUG
2015-11-12 00:37:43 +00:00
2016-02-18 19:54:07 +00:00
local local_os_var=
2016-02-18 22:28:57 +00:00
local_os_var="$(uname -spio 2>&1)"
2015-11-12 00:37:43 +00:00
if [ $? != 0 ]; then
2016-02-18 22:28:57 +00:00
local_os_var="$(uname -v 2>&1)"
2015-11-12 00:37:43 +00:00
if [ $? != 0 ]; then
2016-02-18 22:28:57 +00:00
local_os_var="$(uname)"
2015-11-12 00:37:43 +00:00
fi
fi
case $local_os_var in
*"Linux"*)
LOCAL_OS="Linux"
;;
*"BSD"*)
LOCAL_OS="BSD"
;;
*"MINGW32"*|*"CYGWIN"*)
2015-11-12 00:37:43 +00:00
LOCAL_OS="msys"
;;
*"Darwin"*)
LOCAL_OS="MacOSX"
;;
*)
2016-03-19 09:48:46 +00:00
if [ "$IGNORE_OS_TYPE" == "yes" ]; then #DOC: Undocumented option
2016-03-20 15:53:18 +00:00
Logger "Running on unknown local OS [$local_os_var]." "WARN"
2016-03-19 09:48:46 +00:00
return
fi
2015-11-12 00:37:43 +00:00
Logger "Running on >> $local_os_var << not supported. Please report to the author." "ERROR"
exit 1
;;
esac
Logger "Local OS: [$local_os_var]." "DEBUG"
}
#### MINIMAL-FUNCTION-SET END ####
2015-11-12 00:37:43 +00:00
function GetRemoteOS {
2016-02-18 19:54:07 +00:00
__CheckArguments 0 $# ${FUNCNAME[0]} "$@" #__WITH_PARANOIA_DEBUG
local cmd=
local remote_os_var=
2015-11-12 00:37:43 +00:00
if [ "$REMOTE_OPERATION" == "yes" ]; then
CheckConnectivity3rdPartyHosts
CheckConnectivityRemoteHost
2016-02-18 19:54:07 +00:00
cmd=$SSH_CMD' "uname -spio" > "'$RUN_DIR/$PROGRAM.${FUNCNAME[0]}.$SCRIPT_PID'" 2>&1'
2015-11-12 00:37:43 +00:00
Logger "cmd: $cmd" "DEBUG"
eval "$cmd" &
WaitForTaskCompletion $! 120 240 ${FUNCNAME[0]}"-1" true $KEEP_LOGGING
2015-11-12 00:37:43 +00:00
retval=$?
if [ $retval != 0 ]; then
2016-02-18 19:54:07 +00:00
cmd=$SSH_CMD' "uname -v" > "'$RUN_DIR/$PROGRAM.${FUNCNAME[0]}.$SCRIPT_PID'" 2>&1'
2015-11-12 00:37:43 +00:00
Logger "cmd: $cmd" "DEBUG"
eval "$cmd" &
WaitForTaskCompletion $! 120 240 ${FUNCNAME[0]}"-2" true $KEEP_LOGGING
2015-11-12 00:37:43 +00:00
retval=$?
if [ $retval != 0 ]; then
2016-02-18 19:54:07 +00:00
cmd=$SSH_CMD' "uname" > "'$RUN_DIR/$PROGRAM.${FUNCNAME[0]}.$SCRIPT_PID'" 2>&1'
2015-11-12 00:37:43 +00:00
Logger "cmd: $cmd" "DEBUG"
eval "$cmd" &
WaitForTaskCompletion $! 120 240 ${FUNCNAME[0]}"-3" true $KEEP_LOGGING
2015-11-12 00:37:43 +00:00
retval=$?
if [ $retval != 0 ]; then
Logger "Cannot Get remote OS type." "ERROR"
fi
fi
fi
2016-02-18 22:28:57 +00:00
remote_os_var=$(cat "$RUN_DIR/$PROGRAM.${FUNCNAME[0]}.$SCRIPT_PID")
2015-11-12 00:37:43 +00:00
case $remote_os_var in
*"Linux"*)
REMOTE_OS="Linux"
;;
*"BSD"*)
REMOTE_OS="BSD"
;;
*"MINGW32"*|*"CYGWIN"*)
2015-11-12 00:37:43 +00:00
REMOTE_OS="msys"
;;
*"Darwin"*)
REMOTE_OS="MacOSX"
;;
*"ssh"*|*"SSH"*)
Logger "Cannot connect to remote system." "CRITICAL"
exit 1
;;
*)
2016-03-31 08:17:50 +00:00
if [ "$IGNORE_OS_TYPE" == "yes" ]; then #DOC: Undocumented option
2016-03-20 15:53:18 +00:00
Logger "Running on unknown remote OS [$remote_os_var]." "WARN"
2016-03-19 09:48:46 +00:00
return
fi
2015-11-12 00:37:43 +00:00
Logger "Running on remote OS failed. Please report to the author if the OS is not supported." "CRITICAL"
Logger "Remote OS said:\n$remote_os_var" "CRITICAL"
exit 1
esac
Logger "Remote OS: [$remote_os_var]." "DEBUG"
fi
}
function RunLocalCommand {
local command="${1}" # Command to run
local hard_max_time="${2}" # Max time to wait for command to compleet
2016-02-18 19:54:07 +00:00
__CheckArguments 2 $# ${FUNCNAME[0]} "$@" #__WITH_PARANOIA_DEBUG
2015-11-12 00:37:43 +00:00
if [ $_DRYRUN == true ]; then
2015-11-12 00:37:43 +00:00
Logger "Dryrun: Local command [$command] not run." "NOTICE"
2015-11-19 21:39:31 +00:00
return 0
2015-11-12 00:37:43 +00:00
fi
2015-11-19 21:39:31 +00:00
2015-11-12 00:37:43 +00:00
Logger "Running command [$command] on local host." "NOTICE"
2016-02-18 19:54:07 +00:00
eval "$command" > "$RUN_DIR/$PROGRAM.${FUNCNAME[0]}.$SCRIPT_PID" 2>&1 &
WaitForTaskCompletion $! 0 $hard_max_time ${FUNCNAME[0]} true $KEEP_LOGGING
2015-11-12 00:37:43 +00:00
retval=$?
if [ $retval -eq 0 ]; then
Logger "Command succeded." "NOTICE"
else
Logger "Command failed." "ERROR"
fi
if [ $_VERBOSE == true ] || [ $retval -ne 0 ]; then
2016-02-18 19:54:07 +00:00
Logger "Command output:\n$(cat $RUN_DIR/$PROGRAM.${FUNCNAME[0]}.$SCRIPT_PID)" "NOTICE"
2015-11-12 00:37:43 +00:00
fi
if [ "$STOP_ON_CMD_ERROR" == "yes" ] && [ $retval -ne 0 ]; then
Logger "Stopping on command execution error." "CRITICAL"
exit 1
fi
}
## Runs remote command $1 and waits for completition in $2 seconds
function RunRemoteCommand {
local command="${1}" # Command to run
local hard_max_time="${2}" # Max time to wait for command to compleet
2016-02-18 19:54:07 +00:00
__CheckArguments 2 $# ${FUNCNAME[0]} "$@" #__WITH_PARANOIA_DEBUG
2015-11-12 00:37:43 +00:00
CheckConnectivity3rdPartyHosts
CheckConnectivityRemoteHost
if [ $_DRYRUN == true ]; then
2015-11-12 00:37:43 +00:00
Logger "Dryrun: Local command [$command] not run." "NOTICE"
2015-11-19 21:39:31 +00:00
return 0
2015-11-12 00:37:43 +00:00
fi
2015-11-19 21:39:31 +00:00
2015-11-12 00:37:43 +00:00
Logger "Running command [$command] on remote host." "NOTICE"
2016-02-18 19:54:07 +00:00
cmd=$SSH_CMD' "$command" > "'$RUN_DIR/$PROGRAM.${FUNCNAME[0]}.$SCRIPT_PID'" 2>&1'
2015-11-12 00:37:43 +00:00
Logger "cmd: $cmd" "DEBUG"
eval "$cmd" &
WaitForTaskCompletion $! 0 $hard_max_time ${FUNCNAME[0]} true $KEEP_LOGGING
2015-11-12 00:37:43 +00:00
retval=$?
if [ $retval -eq 0 ]; then
Logger "Command succeded." "NOTICE"
else
Logger "Command failed." "ERROR"
fi
if [ -f "$RUN_DIR/$PROGRAM.${FUNCNAME[0]}.$SCRIPT_PID" ] && ([ $_VERBOSE == true ] || [ $retval -ne 0 ])
2015-11-12 00:37:43 +00:00
then
2016-02-18 19:54:07 +00:00
Logger "Command output:\n$(cat $RUN_DIR/$PROGRAM.${FUNCNAME[0]}.$SCRIPT_PID)" "NOTICE"
2015-11-12 00:37:43 +00:00
fi
if [ "$STOP_ON_CMD_ERROR" == "yes" ] && [ $retval -ne 0 ]; then
Logger "Stopping on command execution error." "CRITICAL"
exit 1
fi
}
function RunBeforeHook {
2016-02-18 19:54:07 +00:00
__CheckArguments 0 $# ${FUNCNAME[0]} "$@" #__WITH_PARANOIA_DEBUG
2015-11-12 00:37:43 +00:00
2016-08-03 22:52:05 +00:00
local pids=
2015-11-12 00:37:43 +00:00
if [ "$LOCAL_RUN_BEFORE_CMD" != "" ]; then
RunLocalCommand "$LOCAL_RUN_BEFORE_CMD" $MAX_EXEC_TIME_PER_CMD_BEFORE &
pids="$!"
2015-11-12 00:37:43 +00:00
fi
if [ "$REMOTE_RUN_BEFORE_CMD" != "" ]; then
RunRemoteCommand "$REMOTE_RUN_BEFORE_CMD" $MAX_EXEC_TIME_PER_CMD_BEFORE &
pids="$pids;$!"
2015-11-12 00:37:43 +00:00
fi
2016-08-03 22:52:05 +00:00
if [ "$pids" != "" ]; then
WaitForTaskCompletion $pids 0 0 ${FUNCNAME[0]} true $KEEP_LOGGING
2016-08-03 22:52:05 +00:00
fi
2015-11-12 00:37:43 +00:00
}
function RunAfterHook {
2016-02-18 19:54:07 +00:00
__CheckArguments 0 $# ${FUNCNAME[0]} "$@" #__WITH_PARANOIA_DEBUG
2015-11-12 00:37:43 +00:00
2016-08-03 22:52:05 +00:00
local pids
2015-11-12 00:37:43 +00:00
if [ "$LOCAL_RUN_AFTER_CMD" != "" ]; then
RunLocalCommand "$LOCAL_RUN_AFTER_CMD" $MAX_EXEC_TIME_PER_CMD_AFTER &
pids="$!"
2015-11-12 00:37:43 +00:00
fi
if [ "$REMOTE_RUN_AFTER_CMD" != "" ]; then
RunRemoteCommand "$REMOTE_RUN_AFTER_CMD" $MAX_EXEC_TIME_PER_CMD_AFTER &
pids="$pids;$!"
2015-11-12 00:37:43 +00:00
fi
2016-08-03 22:52:05 +00:00
if [ "$pids" != "" ]; then
WaitForTaskCompletion $pids 0 0 ${FUNCNAME[0]} true $KEEP_LOGGING
2016-08-03 22:52:05 +00:00
fi
2015-11-12 00:37:43 +00:00
}
function CheckConnectivityRemoteHost {
2016-02-18 19:54:07 +00:00
__CheckArguments 0 $# ${FUNCNAME[0]} "$@" #__WITH_PARANOIA_DEBUG
2015-11-12 00:37:43 +00:00
2016-09-05 20:06:57 +00:00
local retval
2015-11-12 00:37:43 +00:00
if [ "$_PARANOIA_DEBUG" != "yes" ]; then # Do not loose time in paranoia debug
if [ "$REMOTE_HOST_PING" != "no" ] && [ "$REMOTE_OPERATION" != "no" ]; then
eval "$PING_CMD $REMOTE_HOST > /dev/null 2>&1" &
2016-08-28 09:32:25 +00:00
WaitForTaskCompletion $! 60 180 ${FUNCNAME[0]} true $KEEP_LOGGING
2016-09-05 20:06:57 +00:00
retval=$?
if [ $retval != 0 ]; then
Logger "Cannot ping [$REMOTE_HOST]. Return code [$retval]." "WARN"
2016-09-05 20:06:57 +00:00
return $retval
2015-11-12 00:37:43 +00:00
fi
fi
fi
}
function CheckConnectivity3rdPartyHosts {
2016-02-18 19:54:07 +00:00
__CheckArguments 0 $# ${FUNCNAME[0]} "$@" #__WITH_PARANOIA_DEBUG
2015-11-12 00:37:43 +00:00
2016-08-08 16:06:31 +00:00
local remote_3rd_party_success
2016-09-05 20:06:57 +00:00
local retval
2016-08-08 16:06:31 +00:00
2015-11-12 00:37:43 +00:00
if [ "$_PARANOIA_DEBUG" != "yes" ]; then # Do not loose time in paranoia debug
if [ "$REMOTE_3RD_PARTY_HOSTS" != "" ]; then
remote_3rd_party_success=false
2015-11-12 00:37:43 +00:00
for i in $REMOTE_3RD_PARTY_HOSTS
do
eval "$PING_CMD $i > /dev/null 2>&1" &
WaitForTaskCompletion $! 180 360 ${FUNCNAME[0]} true $KEEP_LOGGING
2016-09-05 20:06:57 +00:00
retval=$?
if [ $retval != 0 ]; then
Logger "Cannot ping 3rd party host [$i]. Return code [$retval]." "NOTICE"
2015-11-12 00:37:43 +00:00
else
remote_3rd_party_success=true
2015-11-12 00:37:43 +00:00
fi
done
2016-08-08 16:06:31 +00:00
if [ $remote_3rd_party_success == false ]; then
Logger "No remote 3rd party host responded to ping. No internet ?" "WARN"
2015-11-12 00:37:43 +00:00
return 1
2016-08-08 16:06:31 +00:00
else
return 0
2015-11-12 00:37:43 +00:00
fi
fi
fi
}
#__BEGIN_WITH_PARANOIA_DEBUG
function __CheckArguments {
# Checks the number of arguments of a function and raises an error if some are missing
if [ "$_DEBUG" == "yes" ]; then
2016-08-28 09:32:25 +00:00
local numberOfArguments="${1}" # Number of arguments the tested function should have, can be a number of a range, eg 0-2 for zero to two arguments
local numberOfGivenArguments="${2}" # Number of arguments that have been passed
local functionName="${3}" # Function name that called __CheckArguments
local minArgs
local maxArgs
2015-11-12 00:37:43 +00:00
if [ "$_PARANOIA_DEBUG" == "yes" ]; then
2016-08-28 09:32:25 +00:00
Logger "Entering function [$functionName]." "DEBUG"
2015-11-12 00:37:43 +00:00
fi
2016-08-08 16:06:31 +00:00
# All arguments of the function to check are passed as array in ${4} (the function call waits for $@)
# If any of the arguments contains spaces, bash things there are two aguments
# In order to avoid this, we need to iterate over ${4} and count
local iterate=4
local fetchArguments=true
2016-08-28 09:32:25 +00:00
local argList=""
local countedArguments
while [ $fetchArguments == true ]; do
2016-08-08 16:06:31 +00:00
cmd='argument=${'$iterate'}'
eval $cmd
if [ "$argument" = "" ]; then
fetchArguments=false
2016-08-08 16:06:31 +00:00
else
2016-08-28 09:32:25 +00:00
argList="$arg_list [Argument $(($iterate-3)): $argument]"
2016-08-08 16:06:31 +00:00
iterate=$(($iterate+1))
fi
done
2016-08-28 09:32:25 +00:00
countedArguments=$((iterate-4))
if [ $(IsNumeric "$numberOfArguments") -eq 1 ]; then
minArgs=$numberOfArguments
maxArgs=$numberOfArguments
else
IFS='-' read minArgs maxArgs <<< "$numberOfArguments"
fi
2016-08-08 16:06:31 +00:00
2016-08-28 09:32:25 +00:00
if ! ([ $countedArguments -ge $minArgs ] && [ $countedArguments -le $maxArgs ]); then
Logger "Function $functionName may have inconsistent number of arguments. Expected min: $minArgs, max: $maxArgs, count: $countedArguments, bash seen: $numberOfGivenArguments. see log file." "ERROR"
Logger "Arguments passed: $argList" "ERROR"
2016-08-08 16:06:31 +00:00
fi
2015-11-12 00:37:43 +00:00
fi
}
#__END_WITH_PARANOIA_DEBUG
2015-11-12 00:37:43 +00:00
function RsyncPatternsAdd {
2016-04-01 19:29:33 +00:00
local pattern_type="${1}" # exclude or include
local pattern="${2}"
__CheckArguments 2 $# ${FUNCNAME[0]} "$@" #__WITH_PARANOIA_DEBUG
2015-11-12 00:37:43 +00:00
2016-08-08 16:06:31 +00:00
local rest
2015-11-12 00:37:43 +00:00
# Disable globbing so wildcards from exclusions do not get expanded
set -f
rest="$pattern"
while [ -n "$rest" ]
do
# Take the string until first occurence until $PATH_SEPARATOR_CHAR
str=${rest%%;*}
# Handle the last case
if [ "$rest" = "${rest/$PATH_SEPARATOR_CHAR/}" ]; then
rest=
else
# Cut everything before the first occurence of $PATH_SEPARATOR_CHAR
rest=${rest#*$PATH_SEPARATOR_CHAR}
2015-11-12 00:37:43 +00:00
fi
if [ "$RSYNC_PATTERNS" == "" ]; then
RSYNC_PATTERNS="--"$pattern_type"=\"$str\""
else
RSYNC_PATTERNS="$RSYNC_PATTERNS --"$pattern_type"=\"$str\""
2015-11-12 00:37:43 +00:00
fi
done
set +f
}
2015-11-12 00:37:43 +00:00
function RsyncPatternsFromAdd {
2016-08-08 16:06:31 +00:00
local pattern_type="${1}"
local pattern_from="${2}"
__CheckArguments 2 $# ${FUNCNAME[0]} "$@" #__WITH_PARANOIA_DEBUG
2016-08-08 16:06:31 +00:00
## Check if the exclude list has a full path, and if not, add the config file path if there is one
if [ "$(basename $pattern_from)" == "$pattern_from" ]; then
pattern_from="$(dirname $CONFIG_FILE)/$pattern_from"
fi
2016-08-08 16:06:31 +00:00
if [ -e "$pattern_from" ]; then
RSYNC_PATTERNS="$RSYNC_PATTERNS --"$pattern_type"-from=\"$pattern_from\""
fi
}
function RsyncPatterns {
2016-08-08 16:06:31 +00:00
__CheckArguments 0 $# ${FUNCNAME[0]} "$@" #__WITH_PARANOIA_DEBUG
if [ "$RSYNC_PATTERN_FIRST" == "exclude" ]; then
if [ "$RSYNC_EXCLUDE_PATTERN" != "" ]; then
RsyncPatternsAdd "exclude" "$RSYNC_EXCLUDE_PATTERN"
fi
if [ "$RSYNC_EXCLUDE_FROM" != "" ]; then
RsyncPatternsFromAdd "exclude" "$RSYNC_EXCLUDE_FROM"
fi
if [ "$RSYNC_INCLUDE_PATTERN" != "" ]; then
RsyncPatternsAdd "$RSYNC_INCLUDE_PATTERN" "include"
fi
if [ "$RSYNC_INCLUDE_FROM" != "" ]; then
RsyncPatternsFromAdd "include" "$RSYNC_INCLUDE_FROM"
fi
# Use default include first for quicksync runs
elif [ "$RSYNC_PATTERN_FIRST" == "include" ] || [ $_QUICK_SYNC -eq 2 ]; then
2016-08-08 16:06:31 +00:00
if [ "$RSYNC_INCLUDE_PATTERN" != "" ]; then
RsyncPatternsAdd "include" "$RSYNC_INCLUDE_PATTERN"
fi
if [ "$RSYNC_INCLUDE_FROM" != "" ]; then
RsyncPatternsFromAdd "include" "$RSYNC_INCLUDE_FROM"
fi
if [ "$RSYNC_EXCLUDE_PATTERN" != "" ]; then
RsyncPatternsAdd "exclude" "$RSYNC_EXCLUDE_PATTERN"
fi
if [ "$RSYNC_EXCLUDE_FROM" != "" ]; then
RsyncPatternsFromAdd "exclude" "$RSYNC_EXCLUDE_FROM"
fi
else
Logger "Bogus RSYNC_PATTERN_FIRST value in config file. Will not use rsync patterns." "WARN"
fi
2015-11-12 00:37:43 +00:00
}
function PreInit {
2016-02-18 19:54:07 +00:00
__CheckArguments 0 $# ${FUNCNAME[0]} "$@" #__WITH_PARANOIA_DEBUG
2015-11-12 00:37:43 +00:00
## SSH compression
2016-08-08 16:06:31 +00:00
if [ "$SSH_COMPRESSION" != "no" ]; then
SSH_COMP=-C
else
SSH_COMP=
fi
2015-11-12 00:37:43 +00:00
## Ignore SSH known host verification
if [ "$SSH_IGNORE_KNOWN_HOSTS" == "yes" ]; then
SSH_OPTS="-o UserKnownHostsFile=/dev/null -o StrictHostKeyChecking=no"
fi
2016-08-08 16:06:31 +00:00
## Support for older config files without RSYNC_EXECUTABLE option
if [ "$RSYNC_EXECUTABLE" == "" ]; then
RSYNC_EXECUTABLE=rsync
fi
## Sudo execution option
if [ "$SUDO_EXEC" == "yes" ]; then
if [ "$RSYNC_REMOTE_PATH" != "" ]; then
RSYNC_PATH="sudo $RSYNC_REMOTE_PATH/$RSYNC_EXECUTABLE"
else
RSYNC_PATH="sudo $RSYNC_EXECUTABLE"
fi
COMMAND_SUDO="sudo"
else
if [ "$RSYNC_REMOTE_PATH" != "" ]; then
RSYNC_PATH="$RSYNC_REMOTE_PATH/$RSYNC_EXECUTABLE"
else
RSYNC_PATH="$RSYNC_EXECUTABLE"
fi
COMMAND_SUDO=""
fi
2015-11-12 00:37:43 +00:00
## Set rsync default arguments
2016-08-08 16:06:31 +00:00
RSYNC_ARGS="-rltD"
if [ "$_DRYRUN" == true ]; then
2016-02-19 20:44:52 +00:00
RSYNC_DRY_ARG="-n"
2016-08-08 16:06:31 +00:00
DRY_WARNING="/!\ DRY RUN"
2016-02-19 20:44:52 +00:00
else
RSYNC_DRY_ARG=""
fi
2015-11-12 00:37:43 +00:00
2016-08-09 11:44:43 +00:00
RSYNC_ATTR_ARGS=""
if [ "$PRESERVE_PERMISSIONS" != "no" ]; then
RSYNC_ATTR_ARGS=$RSYNC_ATTR_ARGS" -p"
fi
if [ "$PRESERVE_OWNER" != "no" ]; then
RSYNC_ATTR_ARGS=$RSYNC_ATTR_ARGS" -o"
fi
if [ "$PRESERVE_GROUP" != "no" ]; then
RSYNC_ATTR_ARGS=$RSYNC_ATTR_ARGS" -g"
2016-08-09 11:48:21 +00:00
fi
2016-08-08 16:06:31 +00:00
if [ "$PRESERVE_ACL" == "yes" ]; then
RSYNC_ATTR_ARGS=$RSYNC_ATTR_ARGS" -A"
fi
if [ "$PRESERVE_XATTR" == "yes" ]; then
RSYNC_ATTR_ARGS=$RSYNC_ATTR_ARGS" -X"
fi
if [ "$RSYNC_COMPRESS" == "yes" ]; then
RSYNC_ARGS=$RSYNC_ARGS" -z"
fi
if [ "$COPY_SYMLINKS" == "yes" ]; then
RSYNC_ARGS=$RSYNC_ARGS" -L"
fi
if [ "$KEEP_DIRLINKS" == "yes" ]; then
RSYNC_ARGS=$RSYNC_ARGS" -K"
fi
if [ "$PRESERVE_HARDLINKS" == "yes" ]; then
RSYNC_ARGS=$RSYNC_ARGS" -H"
fi
if [ "$CHECKSUM" == "yes" ]; then
RSYNC_TYPE_ARGS=$RSYNC_TYPE_ARGS" --checksum"
fi
if [ "$BANDWIDTH" != "" ] && [ "$BANDWIDTH" != "0" ]; then
RSYNC_ARGS=$RSYNC_ARGS" --bwlimit=$BANDWIDTH"
fi
if [ "$PARTIAL" == "yes" ]; then
RSYNC_ARGS=$RSYNC_ARGS" --partial --partial-dir=\"$PARTIAL_DIR\""
RSYNC_PARTIAL_EXCLUDE="--exclude=\"$PARTIAL_DIR\""
fi
2016-02-16 19:52:53 +00:00
if [ "$DELTA_COPIES" != "no" ]; then
2016-08-08 16:06:31 +00:00
RSYNC_ARGS=$RSYNC_ARGS" --no-whole-file"
else
RSYNC_ARGS=$RSYNC_ARGS" --whole-file"
fi
2016-02-16 19:52:53 +00:00
2015-11-12 00:37:43 +00:00
## Set compression executable and extension
2016-08-08 16:06:31 +00:00
COMPRESSION_LEVEL=3
if type xz > /dev/null 2>&1
then
COMPRESSION_PROGRAM="| xz -$COMPRESSION_LEVEL"
COMPRESSION_EXTENSION=.xz
elif type lzma > /dev/null 2>&1
then
COMPRESSION_PROGRAM="| lzma -$COMPRESSION_LEVEL"
COMPRESSION_EXTENSION=.lzma
elif type pigz > /dev/null 2>&1
then
COMPRESSION_PROGRAM="| pigz -$COMPRESSION_LEVEL"
COMPRESSION_EXTENSION=.gz
2016-02-18 22:28:57 +00:00
# obackup specific
2016-08-08 16:06:31 +00:00
COMPRESSION_OPTIONS=--rsyncable
elif type gzip > /dev/null 2>&1
then
COMPRESSION_PROGRAM="| gzip -$COMPRESSION_LEVEL"
COMPRESSION_EXTENSION=.gz
2016-02-18 22:28:57 +00:00
# obackup specific
2016-08-08 16:06:31 +00:00
COMPRESSION_OPTIONS=--rsyncable
else
COMPRESSION_PROGRAM=
COMPRESSION_EXTENSION=
fi
ALERT_LOG_FILE="$ALERT_LOG_FILE$COMPRESSION_EXTENSION"
2015-11-12 00:37:43 +00:00
}
function PostInit {
2016-08-08 16:06:31 +00:00
__CheckArguments 0 $# ${FUNCNAME[0]} "$@" #__WITH_PARANOIA_DEBUG
2015-11-12 00:37:43 +00:00
# Define remote commands
2016-08-08 16:06:31 +00:00
SSH_CMD="$(type -p ssh) $SSH_COMP -i $SSH_RSA_PRIVATE_KEY $SSH_OPTS $REMOTE_USER@$REMOTE_HOST -p $REMOTE_PORT"
SCP_CMD="$(type -p scp) $SSH_COMP -i $SSH_RSA_PRIVATE_KEY -P $REMOTE_PORT"
RSYNC_SSH_CMD="$(type -p ssh) $SSH_COMP -i $SSH_RSA_PRIVATE_KEY $SSH_OPTS -p $REMOTE_PORT"
2015-11-12 00:37:43 +00:00
}
function InitLocalOSSettings {
2016-08-08 16:06:31 +00:00
__CheckArguments 0 $# ${FUNCNAME[0]} "$@" #__WITH_PARANOIA_DEBUG
## If running under Msys, some commands do not run the same way
## Using mingw version of find instead of windows one
## Getting running processes is quite different
## Ping command is not the same
if [ "$LOCAL_OS" == "msys" ]; then
FIND_CMD=$(dirname $BASH)/find
PING_CMD='$SYSTEMROOT\system32\ping -n 2'
else
FIND_CMD=find
PING_CMD="ping -c 2 -i .2"
fi
## Stat command has different syntax on Linux and FreeBSD/MacOSX
if [ "$LOCAL_OS" == "MacOSX" ] || [ "$LOCAL_OS" == "BSD" ]; then
STAT_CMD="stat -f \"%Sm\""
STAT_CTIME_MTIME_CMD="stat -f %N;%c;%m"
2016-08-08 16:06:31 +00:00
else
STAT_CMD="stat --format %y"
STAT_CTIME_MTIME_CMD="stat -c %n;%Z;%Y"
2016-08-08 16:06:31 +00:00
fi
2015-11-12 00:37:43 +00:00
}
function InitRemoteOSSettings {
2016-08-08 16:06:31 +00:00
__CheckArguments 0 $# ${FUNCNAME[0]} "$@" #__WITH_PARANOIA_DEBUG
## MacOSX does not use the -E parameter like Linux or BSD does (-E is mapped to extended attrs instead of preserve executability)
2016-08-13 09:56:54 +00:00
if [ "$PRESERVE_EXECUTABILITY" != "no" ];then
if [ "$LOCAL_OS" != "MacOSX" ] && [ "$REMOTE_OS" != "MacOSX" ]; then
RSYNC_ATTR_ARGS=$RSYNC_ATTR_ARGS" -E"
fi
fi
2016-08-08 16:06:31 +00:00
if [ "$REMOTE_OS" == "msys" ]; then
REMOTE_FIND_CMD=$(dirname $BASH)/find
else
REMOTE_FIND_CMD=find
fi
## Stat command has different syntax on Linux and FreeBSD/MacOSX
if [ "$LOCAL_OS" == "MacOSX" ] || [ "$LOCAL_OS" == "BSD" ]; then
REMOTE_STAT_CMD="stat -f \"%Sm\""
REMOTE_STAT_CTIME_MTIME_CMD="stat -f \\\"%N;%c;%m\\\""
2016-08-08 16:06:31 +00:00
else
REMOTE_STAT_CMD="stat --format %y"
REMOTE_STAT_CTIME_MTIME_CMD="stat -c \\\"%n;%Z;%Y\\\""
2016-08-08 16:06:31 +00:00
fi
2015-11-12 00:37:43 +00:00
}
## IFS debug function
function PrintIFS {
printf "IFS is: %q" "$IFS"
}
# Process debugging
# Recursive function to get all parents from a pid
function ParentPid {
local pid="${1}" # Pid to analyse
local parent
parent=$(ps -p $pid -o ppid=)
echo "$pid is a child of $parent"
if [ $parent -gt 0 ]; then
ParentPid $parent
fi
}
2015-11-12 00:37:43 +00:00
## END Generic functions