#!/bin/sh -e
#
# /etc/init.d/openvasmd
#
# Originally written by Miquel van Smoorenburg <miquels@drinkel.ow.org>.
# Modified for Debian GNU/Linux by Ian Murdock <imurdock@gnu.ai.mit.edu>.
# Modified for nessusd by Luca Andreucci <andrew@andrew.org>
# Further changes by Javier Fernandez-Sanguino <jfs@debian.org> for the 
# Debian GNU/Linux distribution 
#
### BEGIN INIT INFO
# Provides:          openvas-manager
# Required-Start:    $remote_fs $network $syslog
# Required-Stop:     $remote_fs $network $syslog
# Should-Start:      
# Should-Stop:       
# Default-Start:     
# Default-Stop:      0 6
# Short-Description: Start and stop the OpenVAS daemon manager
# Description:       Controls the main OpenVAS daemon "openvasmd".
### END INIT INFO

# daemon options
DAEMONOPTS=""
# time to wait for daemons death, in seconds
# don't set it too low or you might not let openvasmd die gracefully
DODTIME=5
[ -r /etc/default/openvas-manager ] && . /etc/default/openvas-manager

DAEMON=/usr/sbin/openvasmd
PIDFILE=/var/run/openvasmd.pid
NAME=openvasmd
LABEL="OpenVAS Manager"

test -x $DAEMON || exit 0

setup_db()
{
    if [ ! -e /var/lib/openvas/mgr/tasks.db ]; then
        echo -n "Creating the OpenVAS Manager SQL database (this takes some time)..."
        touch /var/lib/openvas/mgr/tasks.db 
        /usr/sbin/openvasmd -f --rebuild >/dev/null 2>&1 || :
        if [ $? -eq 0 ] ; then
            echo "..done."
        else
            echo "..ERROR (run 'openvasmd -f --rebuild' manually)."
            exit 1
        fi
     fi
}

running()
{
    # No pidfile, probably no daemon present
    #
    [ ! -f "$PIDFILE" ] && return 1
    pid=`cat $PIDFILE`

    # No pid, probably no daemon present
    [ -z "$pid" ] && return 1

    [ ! -d /proc/$pid ] &&  return 1
    cmd=`cat /proc/$pid/cmdline | tr "\000" "\n"|head -n 1 |cut -d : -f 1`
    # No openvasmd?
    [ "$cmd" != "$NAME" ] &&  return 1

    return 0
}


openvas_start() {
        setup_db
        start-stop-daemon --start --pidfile $PIDFILE --exec $DAEMON -- $DAEMONOPTS 2>&1 >/dev/null
	errcode=$?
# If we don't sleep then running() might not see the pidfile
	sleep $DODTIME
	return $errcode
}

force_stop() {
	[ ! -e "$PIDFILE" ] && return
	if running ; then
		kill -15 $pid
	# Is it really dead?
		sleep "$DODTIME"s
		if running ; then
			kill -9 $pid
			sleep "$DODTIME"s
			if running ; then
				echo "Cannot kill $LABEL (pid=$pid)!"
				exit 1
			fi
		fi
	fi
	rm -f $PIDFILE
}

case "$1" in
  start)
    echo -n "Starting $LABEL: "
    if openvas_start && running ;  then
	    echo "openvasmd."
    else
    	    echo "ERROR."
	    exit 1
    fi
    ;;
  stop)
    echo -n "Stopping $LABEL: "
    if running ; then
	start-stop-daemon --stop --pidfile $PIDFILE --quiet --oknodo --exec $DAEMON
    	sleep "$DODTIME"s
    fi
    if running; then
        force_stop
    fi
    echo "openvasmd."
    ;;
  restart)
    echo -n "Restarting $LABEL: "
    if running; then
    	start-stop-daemon --stop --pidfile $PIDFILE --quiet --oknodo --exec $DAEMON
    	sleep "$DODTIME"s
    fi
    if running; then
        force_stop
    fi
    if openvas_start && running ;  then
	    echo "openvasmd."
    else
    	    echo "ERROR."
	    exit 1
    fi
    ;;
  reload|force-reload)
    echo  -n "Reloading $LABEL configuration files: "
    start-stop-daemon --stop --pidfile $PIDFILE --signal 1 --exec $DAEMON
    sleep "$DODTIME"s
    if running ;  then
	    echo "done."
    else
    	    echo "ERROR."
	    exit 1
    fi
    ;;
  status)
    echo -n "$LABEL is "
    if running ;  then
	    echo "running"
    else
    	    echo "not running."
	    exit 1
    fi
    ;;
  *)
    echo "Usage: /etc/init.d/$NAME {start|stop|restart|reload|status}"
    exit 1
    ;;
esac

exit 0
