#!/bin/bash
#
#   /etc/rc.d/init.d/frr
#
#   Start/Stop the FRR Routing daemons
#   <any general comments about this init script>
#
# chkconfig: 2345 15 85
#
# description: FRRouting (FRR) is a routing suite for IP routing protocols 
#              like BGP, OSPF, RIP and others. This script contols the main
#              daemon "frr" as well as the individual protocol daemons.
#
### BEGIN INIT INFO
# Provides: frr
# Required-Start: $local_fs $network $syslog
# Required-Stop: $local_fs $syslog
# Should-Start: $syslog
# Should-Stop: $network $syslog
# Default-Start: 2 3 4 5
# Default-Stop: 0 1 6
# Short-Description: Start/Stop the FRR Routing daemons
# Description: FRRouting (FRR) is a routing suite for IP routing protocols 
#              like BGP, OSPF, RIP and others. This script contols the main
#              daemon "frr" as well as the individual protocol daemons.
### END INIT INFO

PATH=/bin:/usr/bin:/sbin:/usr/sbin
D_PATH=/usr/lib/frr
C_PATH=/etc/frr
V_PATH=/var/run/frr

# Local Daemon selection may be done by using /etc/frr/daemons.
# See /usr/share/doc/frr/README.Debian.gz for further information.
# Keep zebra first and do not list watchfrr!
DAEMONS="zebra bgpd ripd ripngd ospfd ospf6d isisd pimd ldpd nhrpd eigrpd babeld"
MAX_INSTANCES=5
RELOAD_SCRIPT=/usr/lib/frr/frr-reload.py

. /etc/init.d/functions

# Print the name of the pidfile.
pidfile()
{
    echo "$V_PATH/$1.pid"
}

# Print the name of the vtysh.
vtyfile()
{
    echo "$V_PATH/$1.vty"
}

# Check if daemon is started by using the pidfile.
started()
{
    [ ! -e `pidfile $1` ] && return 3
    if [ -n "$2" ] && [ "$2" == "log" ]; then
        status -p `pidfile $1` $1 && return 0 || return $?
    else
        kill -0 `cat \`pidfile $1\`` 2> /dev/null || return 1
        return 0
    fi
}

# Loads the config via vtysh -b if configured to do so.
vtysh_b ()
{
        # Rember, that all variables have been incremented by 1 in convert_daemon_prios()
        if [ "$vtysh_enable" = 2 -a -f $C_PATH/frr.conf ]; then
                /usr/bin/vtysh -b -n
        fi
}

# Check if the daemon is activated and if its executable and config files
# are in place.
# params:       daemon name
# returns:      0=ok, 1=error
check_daemon()
{
    # If the integrated config file is used the others are not checked.
    if [ -r "$C_PATH/frr.conf" ]; then
        return 0
    fi

    # vtysh_enable has no config file nor binary so skip check.
    # (Not sure why vtysh_enable is in this list but does not hurt)
    if [ $1 != "watchfrr" -a $1 != "vtysh_enable" ]; then
        # check for daemon binary
        if [ ! -x "$D_PATH/$1" ]; then return 1; fi

        # check for config file
        if [ -n "$2" ]; then
            if [ ! -r "$C_PATH/$1-$2.conf" ]; then
                touch "$C_PATH/$1-$2.conf"
                chown frr:frr "$C_PATH/$1-$2.conf"
            fi
        elif [ ! -r "$C_PATH/$1.conf" ]; then
            touch "$C_PATH/$1.conf"
            chown frr:frr "$C_PATH/$1.conf"
        fi
    fi
    return 0
}

# Starts the server if it's not already running according to the pid file.
# The Frr daemons creates the pidfile when starting.
start()
{
    ulimit -n $MAX_FDS > /dev/null 2> /dev/null
    if [ "$1" = "watchfrr" ]; then

        # We may need to restart watchfrr if new daemons are added and/or
        # removed
        if started "$1" ; then
            stop watchfrr
        else
            # Echo only once. watchfrr is printed in the stop above
            echo -n " $1"
        fi

        if [ -e /var/run/frr/watchfrr.started ] ; then
            rm /var/run/frr/watchfrr.started
        fi
        daemon --pidfile=`pidfile $1` "$D_PATH/$1" -d "${watchfrr_options[@]}"
        RETVAL=$?
        [ $RETVAL -ne 0 ] && break
        for i in `seq 1 10`;
        do
            if [ -e /var/run/frr/watchfrr.started ] ; then
                RETVAL=0
                break
            else
                sleep 1
            fi
        done
        RETVAL=1
    elif [ -n "$2" ]; then
        echo -n " $1-$2"
        if ! check_daemon $1 $2 ; then
            echo -n " (binary does not exist)"
            return;
        fi
        daemon --pidfile=`pidfile $1-$2` "$D_PATH/$1" -d `eval echo "$""$1""_options"` -n "$2"
        RETVAL=$?
    else
        echo -n " $1 "
        if ! check_daemon $1; then
            echo " (binary does not exist)"
            return;
        fi
        daemon --pidfile=`pidfile $1` "$D_PATH/$1" -d `eval echo "$""$1""_options"`
        RETVAL=$?
    fi
    echo
    [ $RETVAL -eq 0 ] && touch /var/lock/subsys/$1
    return $RETVAL
}

# Stop the daemon given in the parameter, printing its name to the terminal.
stop()
{
    local inst

    if [ -n "$2" ]; then
       inst="$1-$2"
    else
       inst="$1"
    fi

    if ! started "$inst" ; then
        # echo -n " ($inst)"
        return 0
    else
        echo -n " $inst"
        PIDFILE=`pidfile $inst`
        PID=`cat $PIDFILE 2>/dev/null`
        killproc -p "$PIDFILE" "$D_PATH/$1"
        RETVAL=$?
        [ $RETVAL -eq 0 ] && rm -f $lockfile
        rm -f `pidfile $inst`
        rm -f `vtyfile $inst`
        echo
        return $RETVAL
    fi
}

# Converts values from /etc/frr/daemons to all-numeric values.
convert_daemon_prios()
{
    for name in $DAEMONS zebra vtysh_enable watchfrr_enable; do
        # First, assign the value set by the user to $value
        eval value=\${${name}:0:3}

        # Daemon not activated or entry missing?
        if [ "$value" = "no" -o "$value" = "" ]; then value=0; fi

        # These strings parsed for backwards compatibility.
        if [ "$value" = "yes"  -o  "$value" = "true" ]; then
            value=1;
        fi

        # Zebra is threatened special. It must be between 0=off and the first
        # user assigned value "1" so we increase all other enabled daemons' values.
        if [ "$name" != "zebra" -a "$value" -gt 0 ]; then value=`expr "$value" + 1`; fi

        # If e.g. name is zebra then we set "zebra=yes".
        eval $name=$value
    done
}

# Starts watchfrr for all wanted daemons.
start_watchfrr()
{
    local daemon_name
    local daemon_prio
    local found_one
    local daemon_inst

    # Start the monitor daemon only if desired.
    if [ 0 -eq "$watchfrr_enable" ]; then
        return
    fi

    # Check variable type
    if ! declare -p watchfrr_options | grep -q '^declare \-a'; then
        echo
        echo "ERROR: The variable watchfrr_options from /etc/frr/daemons must be a BASH array!"
        echo "ERROR: Please convert config file and restart!"
        exit 1
    fi

    # Which daemons have been started?
    found_one=0
    for daemon_name in $DAEMONS; do
        eval daemon_prio=\$$daemon_name
        if [ "$daemon_prio" -gt 0 ]; then
            eval "daemon_inst=\${${daemon_name}_instances//,/ }"
            if [ -n "$daemon_inst" ]; then
                for inst in ${daemon_inst}; do
                    eval "inst_disable=\${${daemon_name}_${inst}}"
                    if [ -z ${inst_disable} ] || [ ${inst_disable} != 0 ]; then
                        if check_daemon $daemon_name $inst; then
                            watchfrr_options+=("${daemon_name}-${inst}")
                        fi
                    fi
                done
            else
                if check_daemon $daemon_name; then
                    watchfrr_options+=($daemon_name)
                fi
            fi
            found_one=1
        fi
    done

    # Start if at least one daemon is activated.
    if [ $found_one -eq 1 ]; then
        echo "Starting FRRouting monitor daemon:"
        start watchfrr
    fi
}

# Stopps watchfrr.
stop_watchfrr()
{
    echo "Stopping FRRouting monitor daemon:"
    stop watchfrr
}

# Stops all daemons that have a lower level of priority than the given.
# (technically if daemon_prio >= wanted_prio)
stop_prio()
{
    local wanted_prio
    local daemon_prio
    local daemon_list
    local daemon_inst
    local inst

    if [ -n "$2" ] && [[ "$2" =~ (.*)-(.*) ]]; then
        daemon=${BASH_REMATCH[1]}
        inst=${BASH_REMATCH[2]}
    else
        daemon="$2"
    fi

    wanted_prio=$1
    daemon_list=${daemon:-$DAEMONS}

    echo "Stopping FRRouting daemons (prio:$wanted_prio):"

    for prio_i in `seq 10 -1 $wanted_prio`; do
        for daemon_name in $daemon_list; do
            eval daemon_prio=\${${daemon_name}:0:3}
            daemon_inst=""
            if [ $daemon_prio -eq $prio_i ]; then
                eval "daemon_inst=\${${daemon_name}_instances//,/ }"
                if [ -n "$daemon_inst" ]; then
                    for i in ${daemon_inst}; do
                        if [ -n "$inst" ] && [ "$i" == "$inst" ]; then
                            stop "$daemon_name" "$inst"
                        elif [ x"$inst"  == x ]; then
                            stop "$daemon_name" "$i"
                        fi
                    done
                else
                    stop "$daemon_name"
                fi
            fi
        done
    done

    if [ -z "$inst" ]; then
    # Now stop other daemons that're prowling, coz the daemons file changed
        echo "Stopping other FRRouting daemons"
        if [ -n "$daemon" ]; then
            eval "file_list_suffix="$V_PATH"/"$daemon*""
        else
            eval "file_list_suffix="$V_PATH/*""
        fi
        for pidfile in $file_list_suffix.pid; do
            if [ -f "$pidfile" ]; then
                filename=${pidfile##*/}
                daemon=${filename%.*}
                echo -n " $daemon"
                killproc -p "$pidfile" "$daemon"
                RETVAL=$?
                [ $RETVAL -eq 0 ] && rm -f $lockfile
                rm -f "$pidfile"
                echo
            fi
        done
        echo -n "Removing remaining .vty files"
        for vtyfile in $file_list_suffix.vty; do
            rm -rf "$vtyfile"
        done
        echo
    fi
}

# Starts all daemons that have a higher level of priority than the given.
# (technically if daemon_prio <= wanted_prio)
start_prio()
{
    local wanted_prio
    local daemon_prio
    local daemon_list
    local daemon_name
    local daemon_inst
    local inst

    if [ -n "$2" ] && [[ "$2" =~ (.*)-(.*) ]]; then
        daemon=${BASH_REMATCH[1]}
        inst=${BASH_REMATCH[2]}
    else
        daemon="$2"
    fi

    wanted_prio=$1
    daemon_list=${daemon:-$DAEMONS}

    echo "Starting FRRouting daemons (prio:$wanted_prio):"

    for prio_i in `seq 1 $wanted_prio`; do
        for daemon_name in $daemon_list; do
            eval daemon_prio=\$${daemon_name}
            daemon_inst=""
            if [ $daemon_prio -eq $prio_i ]; then
                eval "daemon_inst=\${${daemon_name}_instances//,/ }"
                if [ -n "$daemon_inst" ]; then
                    if [ `echo "$daemon_inst" | wc -w` -gt ${MAX_INSTANCES} ]; then
                        echo "Max instances supported is ${MAX_INSTANCES}. Aborting"
                        exit 1
                    fi
                    # Check if we're starting again by switching from single instance
                    # to MI version
                    if started "$daemon_name"; then
                        PIDFILE=`pidfile $daemon_name`
                        killproc -p "$PIDFILE" "$daemon_name"
                        rm -f `pidfile $1`
                        rm -f `vtyfile $1`
                    fi

                    for i in ${daemon_inst}; do
                        if [ -n "$inst" ] && [ "$i" == "$inst" ]; then
                            start "$daemon_name" "$inst"
                        elif [ x"$inst" == x ]; then
                            start "$daemon_name" "$i"
                        fi
                    done
                else
                    # Check if we're starting again by switching from
                    # single instance to MI version
                    eval "file_list_suffix="$V_PATH"/"$daemon_name-*""
                    for pidfile in $file_list_suffix.pid; do
                        if [ -f "$pidfile" ]; then
                            killproc -p "$pidfile" "$daemon_name"
                            rm -rf "$pidfile"
                        fi
                    done
                    for vtyfile in $file_list_suffix.vty; do
                        rm -rf "$vtyfile"
                    done

                    start "$daemon_name"
                fi
            fi
        done
    done
}

check_status()
{
    local daemon_name
    local daemon_prio
    local daemon_inst
    local failed_status=0

    if [ -n "$1" ] && [[ "$1" =~ (.*)-(.*) ]]; then
        daemon=${BASH_REMATCH[1]}
        inst=${BASH_REMATCH[2]}
    else
        daemon="$1"
    fi

    daemon_list=${daemon:-$DAEMONS}

    # Which daemons have been started?
    for daemon_name in $daemon_list; do
        eval daemon_prio=\$$daemon_name
        if [ "$daemon_prio" -gt 0 ]; then
            eval "daemon_inst=\${${daemon_name}_instances//,/ }"
            if [ -n "$daemon_inst" ]; then
                for i in ${daemon_inst}; do
                    if [ -n "$inst" -a "$inst" = "$i" ]; then
                        started "$1" "log" || failed_status=$?
                    elif [ -z "$inst" ]; then
                        started "$daemon_name-$i" "log" || failed_status=$?
                    fi
                done
            else
                started "$daemon_name" "log" || failed_status=$?
            fi
        fi
    done

    # All daemons that need to have been started are up and running
    return $failed_status
}

#########################################################
#               Main program                            #
#########################################################

# Config broken but script must exit silently.
[ ! -r "$C_PATH/daemons" ] && exit 0

# Load configuration
. "$C_PATH/daemons"

# Read configuration variable file if it is present
[ -r /etc/sysconfig/frr ] && . /etc/sysconfig/frr

MAX_INSTANCES=${MAX_INSTANCES:=5}

# Set priority of un-startable daemons to 'no' and substitute 'yes' to '0'
convert_daemon_prios

if [ ! -d $V_PATH ]; then
    echo "Creating $V_PATH"
    mkdir -p $V_PATH
    chown frr:frr $V_PATH
    chmod 755 /$V_PATH
fi

if [ -n "$3" ] && [ "$3" != "all" ]; then
    dmn="$2"-"$3"
elif [ -n "$2" ] && [ "$2" != "all" ]; then
    dmn="$2"
fi

case "$1" in
    start)
        # Try to load this necessary (at least for 2.6) module.
        if [ -d /lib/modules/`uname -r` ] ; then
            echo "Loading capability module if not yet done."
            LC_ALL=C modprobe -a capability 2>&1 | egrep -v "(not found|Can't locate)"
        fi

        # Start all daemons
        cd $C_PATH/
        if [ "$2" != "watchfrr" ]; then
          start_prio 10 $dmn
        fi
        start_watchfrr
        vtysh_b
        ;;

    1|2|3|4|5|6|7|8|9|10)
        # Stop/start daemons for the appropriate priority level
        stop_prio $1
        start_prio $1
        vtysh_b
        ;;

    stop|0)
        # Stop all daemons at level '0' or 'stop'
        stop_watchfrr
        if [ "$dmn" != "watchfrr" ]; then
            [ -n "${dmn}" ] && eval "${dmn/-/_}=0"
            stop_prio 0 $dmn
        fi

        if [ -z "$dmn" -o "$dmn" = "zebra" ]; then
            echo "Removing all routes made by zebra."
            ip route flush proto zebra
            # At least in CentOS/RHEL 6, iproute2 doesn't know
            # about the new protocol names, so we have to flush them
            # by number (it also doesn't support rt_protos.d
            ip route flush proto 186
            ip route flush proto 187
            ip route flush proto 188
            ip route flush proto 189
            ip route flush proto 190
            ip route flush proto 191
            ip route flush proto 192
            ip route flush proto 193
            ip route flush proto 194
        else
            [ -n "$dmn" ] && eval "${dmn/-/_}=0"
            start_watchfrr
        fi
        ;;

    reload)
        # Just apply the commands that have changed, no restart necessary
        if [ ! -x "$RELOAD_SCRIPT" ]; then
            echo "frr-reload - reload not supported. Use restart or install frr-pythontools package"
            exit 1
        fi
        NEW_CONFIG_FILE="${2:-$C_PATH/frr.conf}"
        if [ ! -r $NEW_CONFIG_FILE ]; then
            echo "Unable to read configuration file $NEW_CONFIG_FILE. Only supporting integrated config"
            exit 1
        fi
        echo "Applying only incremental changes to running configuration from frr.conf"
        "$RELOAD_SCRIPT" --reload /etc/frr/frr.conf
        exit $?
        ;;

    status)
        check_status $dmn
        exit $?
        ;;

    restart|force-reload)
        $0 stop $dmn
        sleep 1
        $0 start $dmn
        ;;

    *)
        echo "Usage: /etc/init.d/frr {start|stop|status|reload|restart|force-reload|<priority>} [daemon]"
        echo "       E.g. '/etc/init.d/frr 5' would start all daemons with a prio 1-5."
        echo "       reload applies only modifications from the running config to all daemons."
        echo "       reload neither restarts starts any daemon nor starts any new ones."
        echo "       Read /usr/share/doc/frr/README.Debian for details."
        exit 1
        ;;
esac

exit 0
