From 2fba2e31879de69687e55f6f5c2a9e6d0afd4378 Mon Sep 17 00:00:00 2001 From: Christoph Date: Wed, 3 Nov 2021 13:19:21 +0100 Subject: [PATCH] Add script 'check_postgrey.sh'. --- check_postgrey.sh | 256 ++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 256 insertions(+) create mode 100755 check_postgrey.sh diff --git a/check_postgrey.sh b/check_postgrey.sh new file mode 100755 index 0000000..b11e3db --- /dev/null +++ b/check_postgrey.sh @@ -0,0 +1,256 @@ +#!/usr/bin/env bash + +service_name=postgrey + +#--------------------------------------- +#----------------------------- +# Base Function(s) +#----------------------------- +#--------------------------------------- + +fatal(){ + echo "" + if $terminal ; then + echo -e " [ \033[31m\033[1mFatal\033[m ] $*" + else + echo -e " [ Fatal ] $*" + fi + echo "" + if $terminal ; then + echo -e " \033[1mScript terminated\033[m.." + else + echo -e " Script terminated.." + fi + echo "" + rm -rf $LOCK_DIR + exit 1 +} + +error (){ + echo "" + if $terminal ; then + echo -e " [ \033[31m\033[1mError\033[m ] $*" + else + echo " [ Error ] $*" + fi + echo "" +} + +warn (){ + echo "" + if $terminal ; then + echo -e " [ \033[33m\033[1mWarn\033[m ] $*" + else + echo " [ Warn ] $*" + fi + echo "" +} + +info (){ + echo "" + if $terminal ; then + echo -e " [ \033[32m\033[1mInfo\033[m ] $*" + else + echo " [ Info ] $*" + fi + echo "" +} + +ok (){ + echo "" + if $terminal ; then + echo -e " [ \033[32m\033[1mOk\033[m ] $*" + else + echo " [ Ok ] $*" + fi + echo "" +} + +trim() { + local var="$*" + var="${var#"${var%%[![:space:]]*}"}" # remove leading whitespace characters + var="${var%"${var##*[![:space:]]}"}" # remove trailing whitespace characters + echo -n "$var" +} + +trim_double_quotes() { + local var="$*" + var="${var#"${var%%[!\"]*}"}" # remove leading whitespace characters + var="${var%"${var##*[!\"]}"}" # remove trailing whitespace characters + echo -n "$var" +} + +read_env_from_file() { + + _file="$*" + + while read _ENV _REST ; do + + _ENV="$(trim $_ENV)" + + if [[ "$_ENV" =~ .{2,}=.+ ]]; then + _key="${_ENV%=*}" + _val="${_ENV##*=}" + + + # - Remove leading / trailling double quotes + # - _val="${_val%\"}" + # - _val="${_val#\"}" + # - + _val="$(trim_double_quotes $_val)" + + if [[ -n "$(trim $_key)" ]] && [[ -n "$(trim $_val)" ]] ; then + export $_ENV + fi + fi + + done < "$_file" + +} + + +#--------------------------------------- +#----------------------------- +# Check some prerequisites +#----------------------------- +#--------------------------------------- + +LOCK_DIR=`mktemp -d` + +# - Running in a terminal? +# - +if [[ -t 1 ]] ; then + terminal=true + LOGGING=true +else + terminal=false + LOGGING=false +fi + + +# - Systemd supported ? +# - +systemd=$(which systemd) +systemctl=$(which systemctl) + +systemd_supported=false +if [[ -n "$systemd" ]] && [[ -n "$systemctl" ]] ; then + systemd_supported=true +fi + +SYSTEMD_SERVICE= +SYSV_INIT_SCRIPT= + +if $systemd_supported ; then + if systemctl -t service list-unit-files \ + | grep -e "^${service_name,,}d" \ + | grep -q -E "(enabled|disabled)" 2> /devnull ; then + + SYSTEMD_SERVICE="$(systemctl -t service list-unit-files | grep -e "^${service_name,,}d" | awk '{print$1}' | head -1)" + elif systemctl -t service list-unit-files \ + | grep -e "^${service_name,,}" \ + | grep -q -E "(enabled|disabled)" 2> /devnull ; then + + SYSTEMD_SERVICE="$(systemctl -t service list-unit-files | grep -e "^${service_name,,}" | awk '{print$1}' | head -1)" + fi +fi + +if [[ -z "$SYSTEMD_SERVICE" ]]; then + if [[ -x "/etc/init.d/${service_name,,}" ]]; then + SYSV_INIT_SCRIPT="/etc/init.d/${service_name,,}" + elif [[ -x "/etc/init.d/${service_name,,}d" ]]; then + SYSV_INIT_SCRIPT="/etc/init.d/${service_name,,}d" + fi +fi + +if [[ -z "$SYSTEMD_SERVICE" ]] && [[ -z "$SYSV_INIT_SCRIPT" ]] ; then + fatal "Neither an init-script nor a service file for $service_name found!" +fi + +# - Determin check_string_ps .. +# - +if [[ -n "$SYSTEMD_SERVICE" ]] ; then + _systemd_service_file="" + if [[ -f "/etc/systemd/system/$SYSTEMD_SERVICE" ]] ; then + _systemd_service_file="/etc/systemd/system/$SYSTEMD_SERVICE" + elif [[ -f "/lib/systemd/system/$SYSTEMD_SERVICE" ]] ; then + _systemd_service_file="/lib/systemd/system/$SYSTEMD_SERVICE" + fi + if [[ -n "$_systemd_service_file" ]]; then + check_string_ps="$(trim $(cat "$_systemd_service_file" | grep ExecStart | grep -o -E "ExecStart\s*=\s*[^[:space:]]+" | cut -d "=" -f2))" + fi +else + + read_env_from_file $SYSV_INIT_SCRIPT + + _check_string_ps="$(trim $(cat "$SYSV_INIT_SCRIPT" | grep DAEMON | grep -o -E "DAEMON\s*=\s*[^[:space:]]+" | cut -d "=" -f2))" + check_string_ps="$(eval echo "$_check_string_ps")" +fi + + +#--------------------------------------- +#----------------------------- +# Check if service is running +#----------------------------- +#--------------------------------------- + +if $LOGGING ; then + echo -e "\n Check if $service_name service is running.." + echo -e " =====================================" +fi +if ! ps -e f | grep -E "[[:digit:]]\ (${check_string_ps}|$(basename ${check_string_ps}))" | grep -v grep > /dev/null ; then + error "$service_name service seems to be down! Trying to restart service now.." + + if [[ -n "$SYSTEMD_SERVICE" ]] ; then + $systemctl daemon-reload > /dev/null 2> ${LOCK_DIR}/err_msg.log + if [[ $? -ne 0 ]]; then + error "$(cat ${LOCK_DIR}/err_msg.log)" + fi + sleep 2 + $systemctl stop $SYSTEMD_SERVICE > /dev/null 2> ${LOCK_DIR}/err_msg.log + if [[ $? -ne 0 ]]; then + error "$(cat ${LOCK_DIR}/err_msg.log)" + fi + sleep 10 + $systemctl start $SYSTEMD_SERVICE > /dev/null 2> ${LOCK_DIR}/err_msg.log + if [[ $? -ne 0 ]]; then + error "$(cat ${LOCK_DIR}/err_msg.log)" + fi + else + $SYSV_INIT_SCRIPT stop > /dev/null 2>&1 + if [[ $? -ne 0 ]]; then + error "Stopping $service_name service failed!" + fi + sleep 10 + $SYSV_INIT_SCRIPT start > /dev/null 2>&1 + if [[ $? -ne 0 ]]; then + error "Starting $service_name service failed!" + fi + fi + + declare -i counter=0 + PID=$(ps -e f | grep -E "[[:digit:]]\ (${check_string_ps}|$(basename ${check_string_ps}))"| grep -v grep | awk '{print$2}') + while [[ "X${PID}" = "X" ]]; do + sleep 1 + PID=$(ps -e f | grep -E "[[:digit:]]\ (${check_string_ps}|$(basename ${check_string_ps}))"| grep -v grep | awk '{print$2}') + if [[ $counter -gt 10 ]]; then + break + else + ((counter++)) + fi + done + + if [[ "X${PID}" = "X" ]] ; then + error "Restarting $service_name service failed!" + else + ok "$service_name service is up and running." + fi + +else + if $LOGGING ; then + ok "$service_name service is up and running." + fi +fi + +rm -rf $LOCK_DIR +exit 0