#! /bin/bash
#
# This is special script for start,stop,restart of quagga daemons
#

# debug
# echo $* | logger -p local7.debug -t quagga-manager

usage() {
    echo "Usage: $0 {check|start|stop|restart} {bgpd|ospfd|ripd|ospf6d|ripngd}"
    exit 1
}

if [ $# -lt 1 ]; then
    usage
fi

if [ $EUID -ne 0 ]; then
    echo "must be root!"
    exit 1
fi

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

. /lib/lsb/init-functions

status() {
    local daemon=$1
    local pidfile=${pid_dir}/${daemon}.pid
    local binpath=/usr/sbin/vyatta-$daemon
    status_of_proc -p $pidfile $binpath $daemon && exit 0 || exit $?
}

start() {
    local daemon=$1
    local pidfile=${pid_dir}/${daemon}.pid
    local binpath=/usr/sbin/vyatta-$daemon

    if [ ! -x $binpath ]; then
	echo "Unknown daemon $daemon"
	return 1
    fi

    local -a args=( -d -P 0 -i $pidfile )
    if [ "$daemon" = "zebra" ]; then
	args+=( -l -S -s 1048576 )
    fi

    exec start-stop-daemon --start --oknodo --quiet --name $daemon  \
		--chdir $log_dir --exec $binpath --pidfile $pidfile \
		-- ${args[@]}
}

stop() {
    local daemon=$1
    local pidfile=${pid_dir}/${daemon}.pid
    local binpath=/usr/sbin/vyatta-$daemon

    start-stop-daemon --stop --quiet --oknodo --retry 5 \
	--name $daemon --exec $binpath --pidfile=$pidfile
    rm -f $pidfile
}

get_protocol() {
    local daemon=$1
    case $daemon in
    bgpd|ospfd|ripd|ripngd|isisd)   echo ${daemon/%d/};;
    ospf6d)			    echo "ospfv3";;
    *)  echo "Unknown daemon $daemon" 1>&2; exit 1;;
    esac
}

reload_config() {
    local daemon=$1
    local proto=$(get_protocol $daemon)

    local vyatta_cfg=/opt/vyatta/config/active
    local path=$vyatta_cfg/protocols/$proto

    # If daemon does not have config nothing to do.
    [ -d $path ] || return;

    # Begin reloading transaction
    /opt/vyatta/sbin/vyatta-cfg-cmd-wrapper begin || exit 1

    # In case of error undo
    trap "/opt/vyatta/sbin/vyatta-cfg-cmd-wrapper end" EXIT HUP INT QUIT TERM
    
    # Save current configuration
    tmp=/tmp/${daemon}-restart.$$
    /opt/vyatta/sbin/vyatta-save-config.pl $tmp || exit 1

    # Erase portion of active configuration for that protocol
    rm -fr $path

    # special case for interface, remove parameters force reload
    case $proto in
    rip|ospf)
	    find $vyatta_cfg/interfaces -type d -path "*/ip/$proto" \
		-exec rm -fr '{}' \; ;;
    ripng|ospfv3)
	    find $vyatta_cfg/interfaces -type d -path "*/ipv6/$proto" \
		-exec rm -fr '{}' \; ;;
    esac

    # Reload causing configuration to activate - implies commit
    /opt/vyatta/sbin/vyatta-load-config.pl $tmp || exit 1

    # remove tmp file if successful
    rm $tmp
    trap "" EXIT HUP INT QUIT TERM
}

case "$1" in
    start)   start $2;;
    stop)    stop $2;;
    status)  status $2;;
    reload)  start $2; reload_config $2;;
    restart) stop $2; start $2;;
    *)       usage;;
esac