#!/bin/bash
#
### BEGIN INIT INFO
# Provides: vyatta-quagga
# Required-Start: $local_fs $network $remote_fs $syslog
# Required-Stop: $local_fs $network $remote_fs $syslog
# Default-Start:  2 3 4 5
# Default-Stop: 0 1 6
# Short-Description: start and stop the Quagga routing suite
# Description: Quagga is a routing suite for IP routing protocols like 
#              BGP, OSPF, RIP and others. 
#	       Only zebra is started here, others are started by protocols
#	       during config process.
### END INIT INFO
#

. /lib/lsb/init-functions

declare progname=${0##*/}
declare action=$1; shift

pid_dir=/var/run/vyatta/quagga
log_dir=/var/log/vyatta/quagga

for dir in $pid_dir $log_dir ; do
    if [ ! -d $dir ]; then
	mkdir -p $dir
	chown quagga:quagga $dir
	chmod 755 $dir
    fi
done

# Normally only start zebra here. other daemons started in vyatta config
declare -a common_args=( -d -P 0 )

vyatta_quagga_start ()
{
    local -a daemons

    if [ $# -gt 0 ] ; then
	daemons=( $* )
    else
	daemons=( zebra )
    fi

    log_daemon_msg "Starting routing daemons"
    for daemon in ${daemons[@]} ; do
	[ "$daemon" != zebra ] && log_action_cont_msg "$daemon"

	local -a args=( ${common_args[@]} )
	[ "$daemon" = zebra ] && args+=( -l -S -s 1048576 )
	args+=( -i ${pid_dir}/${daemon}.pid )

	start-stop-daemon --start --quiet --oknodo \
	    --chdir $log_dir \
	    --exec "/usr/sbin/vyatta-${daemon}" \
	    -- $args || \
	    ( log_action_end_msg 1 ; return 1 )
	log_progress_msg "$daemon"
    done
}

vyatta_quagga_stop ()
{
    local -a daemons
    if [ $# -gt 0 ] ; then
	daemons=( $* )
    else
	daemons=( watchquagga )
	daemons+=( bgpd ospfd ripd ripngd ospf6d isisd )
	daemons+=( zebra )
    fi

    log_action_begin_msg "Stopping routing services"
    for daemon in ${daemons[@]} ; do
	local pidfile=${pid_dir}/${daemon}.pid
	[ -f $pidfile && "$daemon" != zebra ] && log_action_cont_msg "$daemon"

	start-stop-daemon --stop --quiet --oknodo --retry 2 \
	    --exec /usr/sbin/vyatta-${daemon}
    done    

    log_action_end_msg $?

    if echo ${daemons[@]} | grep -q zebra ; then
	log_begin_msg "Removing all Quagga Routes"
	ip route flush proto zebra
	log_end_msg $?
    fi
}

vyatta_quagga_status ()
{
    local pidfile=$pid_dir/zebra.pid
    local binpath=/usr/sbin/vyatta-zebra

    status_of_proc -p $pidfile $binpath vyatta-zebra && exit 0 || exit $?
}

case "$action" in
    start)
	vyatta_quagga_start $*
	;;

    stop)
	vyatta_quagga_stop $*
   	;;

    status)
	vyatta_quagga_status
	;;

    restart|force-reload)
	vyatta_quagga_stop $*
	sleep 2
	vyatta_quagga_start $*
	;;

    *)
    	echo "Usage: $progname {start|stop|restart|force-reload} [daemon...]"
	exit 1
	;;
esac