#!/bin/sh
#		safe_sterisk 2007-06-02
#		Modified by Denke <denke@denkesys.com>

CLIARGS="$*"					# Grab any args passed to safe_asterisk
TTY=9						# TTY (if you want one) for Asterisk to run on
CONSOLE=yes					# Whether or not you want a console
#NOTIFY=ben@alkaloid.net			# Who to notify about crashes
#LOGFILE=/var/log/asterisk/safe_asterisk.log	# Where to place the normal logfile (disabled if blank)
#SYSLOG=local0					# Which syslog facility to use (disabled if blank)
#EXEC=/path/to/somescript			# Run this command if Asterisk crashes
MACHINE=`hostname`				# To specify which machine has crashed when getting the mail
DUMPDROP=/tmp
SLEEPSECS=2
ASTSBINDIR=/usr/sbin

# comment this line out to have this script _not_ kill all mpg123 processes when
# asterisk exits
KILLALLMPG123=1

# run asterisk with this priority
PRIORITY=0

# set system filemax on supported OSes if this variable is set
# SYSMAXFILES=262144

# set max files open with ulimit. On linux systems, this will be automatically
# set to the system's maximum files open devided by two, if not set here.
# MAXFILES=32768

#chech if we do have logger
if [ "$SYSLOG" != "" ] && [ ! -x /usr/bin/logger ]; then
    echo "Logger was not found, aborting syslog." >&2
    SYSLOG=
fi
# since we're going to change priority and open files limits, we need to be
# root. if running asterisk as other users, pass that to asterisk on the command
# line.
# if we're not root, fall back to standard everything.
if [ `id -u` != 0 ]
then
	echo "Ops. I'm not root. Exiting." >&2
	if [ "$SYSLOG" != "" ]; then
	    logger -p "${SYSLOG}.warn" -t safe_asterisk[$$] "safe_asterisk was tryed to be started by uid:`id -u`. Exiting."
	fi
	if [ "$LOGFILE" != "" ]; then
	    echo "safe_asterisk[$$]: WARNING: safe_asterisk was tryed to be started by uid:`id -u`. Exiting." >> "$LOGFILE"
	fi	
	exit 1;
else
	if `echo $OSTYPE | grep linux 2>&1 > /dev/null `; then
		# maximum number of open files is set to the system maximum divided by two if
		# MAXFILES is not set.
		if [ "$MAXFILES" = "" ]; then
			# just check if file-max is readable
			if [ -r /proc/sys/fs/file-max ]; then
				MAXFILES=$(( `cat /proc/sys/fs/file-max` / 2 ))
			fi
		fi
		SYSCTL_MAXFILES="fs.file-max"
	elif `echo $OSTYPE | grep darwin 2>&1 > /dev/null `; then
		SYSCTL_MAXFILES="kern.maxfiles"
	fi


	if [ "$SYSMAXFILES" != "" ]; then
		if [ "$SYSCTL_MAXFILES" != "" ]; then
			sysctl -w $SYSCTL_MAXFILES=$SYSMAXFILES
		fi
	fi

	# set the process's filemax to whatever set above
	ulimit -n $MAXFILES

fi

#
# Let Asterisk dump core
#
ulimit -c unlimited

#
# Don't fork when running "safely"
#
ASTARGS=""
if [ "$TTY" != "" ]; then
	if [ -c /dev/tty${TTY} ]; then
		TTY=tty${TTY}
	elif [ -c /dev/vc/${TTY} ]; then
		TTY=vc/${TTY}
	else
		echo "Cannot find your TTY (${TTY})" >&2
		if [ "$SYSLOG" != "" ]; then
		    logger -p "${SYSLOG}.warn" -t safe_asterisk[$$] "Cannot find specified TTY (${TTY})"
		fi
		if [ "$LOGFILE" != "" ]; then
		    echo "safe_asterisk[$$]: WARNING: Cannot find specified TTY (${TTY})" >> "$LOGFILE"
		fi
		exit 1
	fi
	ASTARGS="${ASTARGS} -vvvg"
	if [ "$CONSOLE" != "no" ]; then
		ASTARGS="${ASTARGS} -c"
	fi
fi
if [ ! -w ${DUMPDROP} ]; then	
	echo "Cannot write to ${DUMPDROP}" >&2
	if [ "$SYSLOG" != "" ]; then
	    logger -p "${SYSLOG}.warn" -t safe_asterisk[$$] "Cannot write to ${DUMPDROP}"
	fi
	if [ "$LOGFILE" != "" ]; then
	    echo "safe_asterisk[$$]: WARNING: Cannot write to ${DUMPDROP}" >> "$LOGFILE"
	fi
	exit 1
fi

#
# Run scripts to set any environment variables or do any other system-specific setup needed
#

if [ -d /etc/asterisk/startup.d ]; then
	for script in /etc/asterisk/startup.d/*.sh; do
		if [ -x ${script} ]; then
			source ${script}
		fi
	done
fi

run_asterisk()
{
	while :; do 

		if [ "$TTY" != "" ]&&[ "$CONSOLE" != "no" ]; then
			cd /tmp
			stty sane < /dev/${TTY}
			nice -n $PRIORITY ${ASTSBINDIR}/asterisk ${CLIARGS} ${ASTARGS} >& /dev/${TTY} < /dev/${TTY}
		else
			cd /tmp
			nice -n $PRIORITY ${ASTSBINDIR}/asterisk ${CLIARGS} ${ASTARGS}
		fi
		EXITSTATUS=$?
		echo "Asterisk ended with exit status $EXITSTATUS"
		if [ "$EXITSTATUS" = "0" ]; then
			# Properly shutdown....
			echo "Asterisk shutdown normally."
			if [ "$SYSLOG" != "" ]; then
			    logger -p "${SYSLOG}.notice" -t safe_asterisk[$$] "Asterisk shut down cleanly"
			fi
			if [ "$LOGFILE" != "" ]; then
			    echo "safe_asterisk[$$]: NOTICE: Asterisk shut down cleanly" >> "$LOGFILE"
			fi
			exit 0
		elif [ $EXITSTATUS -gt 128 ]; then
			let EXITSIGNAL=EXITSTATUS-128
			echo "Asterisk exited on signal $EXITSIGNAL."
			if [ "$SYSLOG" != "" ]; then
			    logger -p "${SYSLOG}.warn" -t safe_asterisk[$$] "Asterisk exited on signal $EXITSIGNAL."
			fi
			if [ "$LOGFILE" != "" ]; then
			    echo "safe_asterisk[$$]: WARNING: Asterisk exited on signal $EXITSIGNAL." >> "$LOGFILE"
			fi
			if [ "$NOTIFY" != "" ]; then
				echo "Asterisk on $MACHINE exited on signal $EXITSIGNAL.  Might want to take a peek." | \
				mail -s "Asterisk Died" $NOTIFY
			fi
			if [ "$EXEC" != "" ]; then
				$EXEC
			fi
			if [ -f /tmp/core ]; then
				mv /tmp/core ${DUMPDROP}/core.`hostname`-`date -Iseconds` &
			fi
		else
			echo "Asterisk died with code $EXITSTATUS."
			if [ "$SYSLOG" != "" ]; then
			    logger -p "${SYSLOG}.err" -t safe_asterisk[$$] "Asterisk died with code $EXITSTATUS."
			fi
			if [ "$LOGFILE" != "" ]; then
			    echo "safe_asterisk[$$]: ERROR: Asterisk died with code $EXITSTATUS." >> "$LOGFILE"
			fi
			if [ "$SYSLOG" != "" ]; then
			    logger -p "${SYSLOG}.err" -t safe_asterisk[$$] "Asterisk exited on signal $EXITSIGNAL."
			fi
			if [ "$LOGFILE" != "" ]; then
			    echo "safe_asterisk[$$]: ERROR: Asterisk exited on signal $EXITSIGNAL." >> "$LOGFILE"
			fi
			if [ -f /tmp/core ]; then
			    mv /tmp/core ${DUMPDROP}/core.`hostname`-`date -Iseconds` &
			fi
		fi
		sleep $SLEEPSECS
		echo "Automatically restarting Asterisk."
		if [ "$SYSLOG" != "" ]; then
		    logger -p "${SYSLOG}.notice" -t safe_asterisk[$$] "Automatically restarting Asterisk."
		fi
		if [ "$LOGFILE" != "" ]; then
		    echo "safe_asterisk[$$]: NOTICE: Automatically restarting Asterisk." >> "$LOGFILE"
		fi
		if [ $KILLALLMPG123 ]; then
			killall -9 mpg123
		fi
	done
}

run_asterisk &