#! /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

check() {
    local daemon=$1
    [  -f ${pid_dir}/${daemon}.pid ] || start $daemon
}

start() {
    local daemon=$1
    local exe_file=/usr/sbin/vyatta-$daemon

    if [ ! -x $exe_file ]; then
	echo "Unknown daemon $daemon"
	exit 1
    fi
    
    local -a args=( -d -P 0 -i ${pid_dir}/${daemon}.pid )
    case $daemon in
    zebra) args+=( -l -S -s 1048576 ) ;;
    watchquagga)
	    args=( -dz -p ${pid_dir}/${daemon}.pid );
	    # Note: at present ospf/rip only monitored if actually active
	    # probably should get smarter in future here
	    local -a protocols=(`/opt/vyatta/bin/vyatta-show-protocols exists`)

	    if [ ${#protocols[*]} -eq 0 ]
	    then 
		# Only zebra: mode 1 simple restart
		args+=( -R "$0 restart zebra" zebra )
	    else
		# With routing protocols: phased restart
		args+=( -A -r "$0 restart %s" )
		args+=( -s "$0 reload %s" )
		args+=( -k "$0 stop %s" )
		args+=( zebra ${protocols[*]} )
	    fi
	    ;;
    esac

    start-stop-daemon --start --quiet --oknodo \
	--chdir $log_dir --exec $exe_file \
	-- "${args[@]}"
}

stop() {
    local daemon=$1
    local exe_file=/usr/sbin/vyatta-$daemon

    start-stop-daemon --stop --quiet --oknodo --retry 5 \
	--exec $exe_file --pidfile=$pid_dir/${daemon}.pid
    rm -f $pid_dir/${daemon}.pid
}

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
}

# Check if interface configuration exists for this protocol
okay_to_stop() {
    local daemon=$1
    local proto=$(get_protocol $daemon)
    local cfg=/opt/vyatta/config/active/interfaces

    case $proto in
    rip|ospf)
        return $(find $cfg -type d -path "*/ip/$proto" | wc -l);;
    ripng|ospfv3)
	return $(find $cfg -type d -path "*/ipv6/$proto" | wc -l);;
    *)
        return 0;;
    esac
}

update() {
    # get list of daemons deleted and added in current transaction
    local -a deleted=( `/opt/vyatta/bin/vyatta-show-protocols deleted` )
    local -a added=( `/opt/vyatta/bin/vyatta-show-protocols added` )

    # nothing lost, nothing gained
    [ ${#deleted[*]} -eq 0 -a ${#added[*]} -eq 0 ] && exit 0

    # Stop watcher (or it will restart daemons!)
    stop watchquagga

    # Cleanup any daemons no longer needed
    for p in ${deleted[*]}
    do
	# Need daemon to hold state of interface
	if okay_to_stop $p
	then stop $p
	fi
    done

    start watchquagga
}


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