#! /bin/bash
### BEGIN INIT INFO
# Provides:       qhdevdmn
# Required-Start: $dcboot
# Should-Start: $ALL
# Required-Stop:
# Default-Start:     2 3 4 5
# Default-Stop:      S 0 1 6
# Short-Description: device control deamon
# Description:       Start and stop the device control deamon.
### END INIT INFO

#
# Revision History:
# Date:   June 30 2014.
# Author: Prashant SB.
# Desc:   Created.
#

# Do NOT "set -e"
PRODUCT_NAME="Seqrite"
QHEPSDIR="/usr/lib/${PRODUCT_NAME}/${PRODUCT_NAME}"
QHAVDIR="/usr/lib/${PRODUCT_NAME}"

# PATH should only include /usr/* if it runs after the mountnfs.sh script
PATH=/usr/sbin:/usr/bin:/sbin:/bin
DESC="${PRODUCT_NAME} Device control service"
CDCON_DESC="${PRODUCT_NAME} CD/DVD control service"

if [ -f ${QHEPSDIR}/QHfunctions ]; then
        . ${QHEPSDIR}/QHfunctions
        QHINSDIR=${QHEPSDIR}
else
        if [ -f ${QHAVDIR}/QHfunctions ]; then
                . ${QHAVDIR}/QHfunctions
                QHINSDIR=${QHAVDIR}
        fi
fi

NAME="qhdevx"
DAEMON="$QHINSDIR/$NAME"
DAEMON_ARGS="" #--options args"
PIDFILE=/var/run/$NAME.pid
SCRIPTNAME=/etc/init.d/qhdevdmn

CDCON_BINNAME="qhcdx"
CDCON_BINARY_PATH="$QHINSDIR/$CDCON_BINNAME"
QHSHDL_PIDFILE=/var/run/$CDCON_BINNAME.pid
# Check root user.

if [ $(id -u) -ne 0 ]
then
        if [ "$1" != "start" -a "$1" != "stop" -a "$1" != "restart" -a "$1" != "force-reload" -a "$1" != "status" ];
        then
                echo "Usage: $SCRIPTNAME {start|stop|restart|force-reload|status}" >&2
                exit 0
        fi

        if [ "$1" != "status" ]
        then
                echo "You are not root user. Please login as root user.";
                exit 0
        fi
fi

# Exit if the package is not installed
[ -x "$DAEMON" ] || exit 0
[ -x "$CDCON_BINARY_PATH" ] || exit 0

# Read configuration variable file if it is present
[ -r /etc/default/$NAME ] && . /etc/default/$NAME

# Load the VERBOSE setting and other rcS variables
[ -f /etc/default/rcS ] && . /etc/default/rcS

# Define LSB log_* functions.
# Depend on lsb-base (>= 3.0-6) to ensure that this file is present.
. /lib/lsb/init-functions

PATH="/bin:/usr/bin:/usr/local/bin:/sbin:/usr/sbin:/usr/local/sbin"

resolve_drivers()
{
     BT="btusb"
     ST="usb_storage"
     BT_STATE=`grep ${BT} /proc/modules`
     ST_STATE=`grep ${ST} /proc/modules`
 
     if [ "$BT_STATE" -a "$ST_STATE" ]; then 
         return 0
     fi  
         
     if [ "$BT_STATE" == "" ]; then
         `modprobe ${BT}` 
     fi  
 
     if [ "$ST_STATE" == "" ]; then
         `modprobe ${ST}` 
     fi  
 
     return 1
}

#
# Function that starts the daemon/service
#
start_cdcon()
{
	COUNT=0
	for PROC in $(ps axw -o "pid= command=" | grep ${CDCON_BINNAME} | grep -v grep)
	do
		if [ "${PROC%%[!0123456789]*}" = "${PROC}" ]
		then
			COUNT=$((${COUNT} + 1))
		fi
	done
	if [ ${COUNT} -ge 1 ]
	then
		echo "$CDCON_DESC is already running..."
		return 0
	fi

	[ "$VERBOSE" != no ] && log_daemon_msg "Starting $CDCON_DESC" 
	do_start_cdcon
	case "$?" in
		0|1) [ "$VERBOSE" != no ] && log_end_msg 0 ;;
		2) [ "$VERBOSE" != no ] && log_end_msg 1 ;;
	esac

}
start_devcon()
{

	# Check if already runnning;
	COUNT=0
	while true;
	do
		OUT=`ps axw -o "command="|grep "${NAME} --detach"|grep -v grep`
		if [ -n "$OUT" ];
		then
		   sleep 1
		else
		    break;
		fi
	done

	for PROC in $(ps axw -o "pid= command=" | grep ${NAME} | grep -v grep)
	do

		if [ "${PROC%%[!0123456789]*}" = "${PROC}" ]
		then
			COUNT=$((${COUNT} + 1))
		fi
	done
	if [ ${COUNT} -ge 1 ]
	then
		echo "$DESC $NAME is already running..."
		exit 1
	fi

	[ "$VERBOSE" != no ] && log_daemon_msg "Starting $DESC" 
	do_start_devcon
	case "$?" in
		0|1) [ "$VERBOSE" != no ] && log_end_msg 0 ;;
		2) [ "$VERBOSE" != no ] && log_end_msg 1 ;;
	esac
}

do_start()
{
    resolve_drivers	
    start_cdcon
	start_devcon
	
}

do_start_cdcon()
{
	start-stop-daemon --quiet --make-pidfile --pidfile $QHSHDL_PIDFILE --exec $CDCON_BINARY_PATH --start  || return 1
	sleep 2
    pkill -SIGHUP ${CDCON_BINNAME}
}

do_start_devcon()
{
	start-stop-daemon --quiet --make-pidfile --pidfile $PIDFILE --exec $DAEMON --start || return 1

	sleep 1
}


#
# Function that stops the daemon/service
#
do_stop()
{
	# Return
	#   0 if daemon has been stopped
	#   1 if daemon was already stopped
	#   2 if daemon could not be stopped
	#   other if a failure occurred

    [ "$VERBOSE" != no ] && log_daemon_msg "Stopping $CDCON_DESC"  
		do_stop_cdcon
		case "$?" in
			0|1) [ "$VERBOSE" != no ] && log_end_msg 0 ;;
			2) [ "$VERBOSE" != no ] && log_end_msg 1 ;;
		esac
	

	[ "$VERBOSE" != no ] && log_daemon_msg "Stopping $DESC"
	do_stop_devcon
	case "$?" in
		0|1) [ "$VERBOSE" != no ] && log_end_msg 0 ;;
		2) [ "$VERBOSE" != no ] && log_end_msg 1 ;;
	esac
	
}

do_stop_cdcon()
{
	#Category server stop

	start-stop-daemon --quiet --name $CDCON_BINNAME --retry TERM/30/KILL/5 --stop
	RETVAL="$?"
	[ "$RETVAL" = 2 ] && return 2
	# Wait for children to finish too if this is a daemon that forks
	# and if the daemon is only ever run from this initscript.
	# If the above conditions are not satisfied then add some other code
	# that waits for the process to drop all resources that could be
	# needed by services started subsequently.  A last resort is to
	# sleep for some time.
	start-stop-daemon --stop --quiet --oknodo --retry=0/30/KILL/5 --exec $CDCON_BINARY_PATH
	[ "$?" = 2 ] && return 2
	# Many daemons don't delete their pidfiles when they exit.
	rm -f $QHSHDL_PIDFILE
}

do_stop_devcon()
{

	start-stop-daemon --quiet --name $NAME --retry TERM/30/KILL/5 --stop
	RETVAL="$?"
	[ "$RETVAL" = 2 ] && return 2
	# Wait for children to finish too if this is a daemon that forks
	# and if the daemon is only ever run from this initscript.
	# If the above conditions are not satisfied then add some other code
	# that waits for the process to drop all resources that could be
	# needed by services started subsequently.  A last resort is to
	# sleep for some time.
	start-stop-daemon --stop --quiet --oknodo --retry=0/30/KILL/5 --exec $DAEMON
	[ "$?" = 2 ] && return 2
	# Many daemons don't delete their pidfiles when they exit.
	rm -f $PIDFILE

	return "$RETVAL"
}

#
# Function that sends a SIGHUP to the daemon/service
#
do_reload() 
{
	#
	# If the daemon can reload its configuration without
	# restarting (for example, when it is sent a SIGHUP),
	# then implement that here.
	#
    start-stop-daemon --stop --signal 1 --quiet --pidfile $QHSHDL_PIDFILE --name $CDCON_BINNAME
	
	start-stop-daemon --stop --signal 1 --quiet --pidfile $PIDFILE --name $NAME
	
	return 0
}


status_cdcon()
{

	COUNT=0
	for PROC in $(ps axw -o "pid= command=" | grep ${CDCON_BINNAME} | grep -v grep)
	do
		if [ "${PROC%%[!0123456789]*}" = "${PROC}" ]
		then
			COUNT=$((${COUNT} + 1))
		fi
	done
	if [ ${COUNT} -ge 1 ]
	then
		echo "$CDCON_DESC is running..."
	else
		echo "$CDCON_DESC is stopped..."
	fi

}

status_devcon()
{

	COUNT=0
	for PROC in $(ps axw -o "pid= command=" | grep ${NAME} | grep -v grep)
	do
		if [ "${PROC%%[!0123456789]*}" = "${PROC}" ]
		then
			COUNT=$((${COUNT} + 1))
		fi
	done
	if [ ${COUNT} -ge 1 ]
	then
		echo "$DESC is running..."
	else
		echo "$DESC is stopped..."
	fi

}

case "$1" in
  start)

	# Start URL categorization server
	do_start

	;;
  stop)
	
	do_stop
	;;
  #reload|force-reload)
	#
	# If do_reload() is not implemented then leave this commented out
	# and leave 'force-reload' as an alias for 'restart'.
	#
	#log_daemon_msg "Reloading $DESC" "$NAME"
	#do_reload
	#log_end_msg $?
	#;;
  restart|force-reload)
	#
	# If the "reload" option is implemented then remove the
	# 'force-reload' alias
	#
	#log_daemon_msg "Restarting $DESC" "$NAME"
	do_stop
	case "$?" in
	  0|1)
		do_start
		#case "$?" in
		#	0) log_end_msg 0 ;;
		#	1) log_end_msg 1 ;; # Old process is still running
		#	*) log_end_msg 1 ;; # Failed to start
		#esac
		;;
	  *)
	  	# Failed to stop
		log_end_msg 1
		;;
	esac
	;;

  status)
	# Check status;
	
    status_cdcon

	status_devcon

	;;

  *)
	#echo "Usage: $SCRIPTNAME {start|stop|restart|reload|force-reload}" >&2
	echo "Usage: $SCRIPTNAME {start|stop|restart|force-reload|status}" >&2
	exit 3
	;;
esac

: