#!/bin/bash

### BEGIN INIT INFO
# Provides:          cloud agent
# Required-Start:    $network $local_fs
# Required-Stop:     $network $local_fs
# Default-Start:     3 4 5
# Default-Stop:      0 1 2 6
# Short-Description: Start/stop Apache CloudStack Agent
# Description: This scripts Starts/Stops the Apache CloudStack agent
##  The CloudStack Agent is a part of the Apache CloudStack project and is used
##  for managing KVM-based Hypervisors and performing secondary storage tasks inside
##  the Secondary Storage System Virtual Machine.
## JSVC (Java daemonizing) is used for starting and stopping the agent
### END INIT INFO

# Licensed to the Apache Software Foundation (ASF) under one
# or more contributor license agreements.  See the NOTICE file
# distributed with this work for additional information
# regarding copyright ownership.  The ASF licenses this file
# to you under the Apache License, Version 2.0 (the
# "License"); you may not use this file except in compliance
# with the License.  You may obtain a copy of the License at
# 
#   http://www.apache.org/licenses/LICENSE-2.0
# 
# Unless required by applicable law or agreed to in writing,
# software distributed under the License is distributed on an
# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
# KIND, either express or implied.  See the License for the
# specific language governing permissions and limitations
# under the License.

. /lib/lsb/init-functions

SHORTNAME="cloud-agent"
PIDFILE=/var/run/"$SHORTNAME".pid
LOCKFILE=/var/lock/subsys/"$SHORTNAME"
LOGFILE=/var/log/cloud/agent/agent.log
PROGNAME="CloudStack Agent"
CLASS="com.cloud.agent.AgentShell"
PROG="jsvc"
DAEMON="/usr/bin/jsvc"
SHUTDOWN_WAIT="30"

unset OPTIONS
[ -r /etc/default/"$SHORTNAME" ] && source /etc/default/"$SHORTNAME"

# The first existing directory is used for JAVA_HOME (if JAVA_HOME is not defined in $DEFAULT)
JDK_DIRS="/usr/lib/jvm/java-6-openjdk /usr/lib/jvm/java-6-openjdk-i386 /usr/lib/jvm/java-6-openjdk-amd64 /usr/lib/jvm/java-6-sun /usr/lib/jvm/java-1.5.0-sun /usr/lib/j2sdk1.5-sun /usr/lib/j2sdk1.5-ibm"

for jdir in $JDK_DIRS; do
    if [ -r "$jdir/bin/java" -a -z "${JAVA_HOME}" ]; then
        JAVA_HOME="$jdir"
    fi
done
export JAVA_HOME

SCP="/usr/share/java/commons-collections.jar:/usr/share/java/commons-dbcp.jar:/usr/share/java/commons-codec.jar:/usr/share/java/commons-logging.jar:/usr/share/java/commons-logging-api.jar:/usr/share/java/commons-pool.jar:/usr/share/java/commons-httpclient.jar:/usr/share/java/ws-commons-util.jar:/usr/share/java/mysql-connector-java.jar:/usr/share/java/servlet-api-2.5.jar:/usr/share/java/jsp-api-2.1.jar:/usr/share/java/jasper.jar:/usr/share/java/el-api-2.1.jar:/usr/share/java/jasper-el.jar:/usr/share/java/ecj.jar:/usr/share/java/cglib.jar:/usr/share/java/asm3.jar:/usr/share/java/jsch.jar:/usr/share/java/backport-util-concurrent.jar:/usr/share/java/jetty.jar:/usr/share/java/jetty-util.jar:/usr/share/java/jetty-start-daemon.jar:/usr/share/java/jetty-sslengine.jar:/usr/share/java/jetty-start.jar:/usr/share/java/jna.jar:/usr/share/java/log4j-1.2.jar"
DCP="/usr/share/java/libvirt-0.4.8.jar:/usr/share/java/not-yet-commons-ssl-0.3.9.jar:/usr/share/java/geronimo-stax-api_1.0_spec-1.0.1.jar:/usr/share/java/esapi-2.0GA.jar:/usr/share/java/activation-1.1.jar:/usr/share/java/axiom-api-1.2.8.jar:/usr/share/java/xmlrpc-client-3.1.3.jar:/usr/share/java/javassist-3.9.0.GA.jar:/usr/share/java/annogen-0.1.0.jar:/usr/share/java/commons-logging-1.1.1.jar:/usr/share/java/gson-1.7.1.jar:/usr/share/java/wss4j-1.5.8.jar:/usr/share/java/jsr107cache-1.0.jar:/usr/share/java/ant-launcher-1.7.0.jar:/usr/share/java/commons-codec-1.6.jar:/usr/share/java/selenium-core-1.0-20080914.225453.jar:/usr/share/java/commons-lang-2.6.jar:/usr/share/java/oro-2.0.8.jar:/usr/share/java/httpcore-nio-4.0-beta1.jar:/usr/share/java/httpcore-4.0.jar:/usr/share/java/axis2-ant-plugin-1.4.1.jar:/usr/share/java/axiom-impl-1.2.7.jar:/usr/share/java/log4j-1.2.16.jar:/usr/share/java/commons-configuration-1.8.jar:/usr/share/java/commons-lang-2.1.jar:/usr/share/java/hibernate-jpa-2.0-api-1.0.0.Final.jar:/usr/share/java/xml-apis-1.3.04.jar:/usr/share/java/commons-pool-1.6.jar:/usr/share/java/xml-resolver-1.2.jar:/usr/share/java/jsch-0.1.42.jar:/usr/share/java/bcprov-jdk15-135.jar:/usr/share/java/axis2-adb-1.4.1.jar:/usr/share/java/mex-1.5.1-impl.jar:/usr/share/java/commons-collections-3.2.1.jar:/usr/share/java/slf4j-api-1.5.8.jar:/usr/share/java/jul-to-slf4j-1.6.1.jar:/usr/share/java/geronimo-activation_1.1_spec-1.0.1.jar:/usr/share/java/opensaml-2.5.1-1.jar:/usr/share/java/hibernate-annotations-3.5.1-Final.jar:/usr/share/java/axis2-1.5.1.jar:/usr/share/java/libvirt-0.4.9.jar:/usr/share/java/jetty-6.1.26.jar:/usr/share/java/wsdl4j-1.6.2.jar:/usr/share/java/woden-impl-dom-1.0M8.jar:/usr/share/java/bcprov-jdk14-140.jar:/usr/share/java/cglib-nodep-2.2.2.jar:/usr/share/java/asm-3.1.jar:/usr/share/java/dom4j-1.6.1.jar:/usr/share/java/jmdns-2.1.jar:/usr/share/java/commons-discovery-0.5.jar:/usr/share/java/selenium-server-coreless-1.0-20081010.060147.jar:/usr/share/java/joda-time-1.5.2.jar:/usr/share/java/log4j-over-slf4j-1.6.1.jar:/usr/share/java/ws-commons-util-1.0.2.jar:/usr/share/java/hamcrest-core-1.1.jar:/usr/share/java/jasypt-1.9.0.jar:/usr/share/java/xmlParserAPIs-2.6.0.jar:/usr/share/java/servlet-api-2.5-20081211.jar:/usr/share/java/commons-httpclient-3.1.jar:/usr/share/java/axiom-dom-1.2.7.jar:/usr/share/java/slf4j-api-1.5.11.jar:/usr/share/java/ant-1.6.5.jar:/usr/share/java/bcprov-jdk16-1.45.jar:/usr/share/java/jcl-over-slf4j-1.6.1.jar:/usr/share/java/mail-1.4.jar:/usr/share/java/servlet-api-2.3.jar:/usr/share/java/javax.persistence-2.0.0.jar:/usr/share/java/axis2-kernel-1.4.1.jar:/usr/share/java/ehcache-1.5.0.jar:/usr/share/java/axis2-mtompolicy-1.5.1.jar:/usr/share/java/commons-fileupload-1.2.jar:/usr/share/java/hibernate-entitymanager-3.5.1-Final.jar:/usr/share/java/junit-4.10.jar:/usr/share/java/ejb-api-3.0.jar:/usr/share/java/backport-util-concurrent-3.1.jar:/usr/share/java/xmlsec-1.4.2.jar:/usr/share/java/xercesImpl-2.8.1.jar:/usr/share/java/velocity-1.5.jar:/usr/share/java/xmltooling-1.3.1.jar:/usr/share/java/apache-log4j-extras-1.1.jar:/usr/share/java/XmlSchema-1.4.2.jar:/usr/share/java/axis2-java2wsdl-1.4.1.jar:/usr/share/java/selenium-java-client-driver-1.0.1.jar:/usr/share/java/jaxen-1.1.1.jar:/usr/share/java/axis2-adb-codegen-1.4.1.jar:/usr/share/java/woden-api-1.0M8.jar:/usr/share/java/xalan-2.7.0.jar:/usr/share/java/geronimo-jms_1.1_spec-1.1.jar:/usr/share/java/jstl-1.2.jar:/usr/share/java/hibernate-core-3.5.1-Final.jar:/usr/share/java/ant-1.7.0.jar:/usr/share/java/jta-1.1.jar:/usr/share/java/CAStorSDK-1.3.1-CS40.jar:/usr/share/java/antlr-2.7.6.jar:/usr/share/java/wstx-asl-3.2.4.jar:/usr/share/java/commons-io-1.4.jar:/usr/share/java/rampart-trust-1.5.jar:/usr/share/java/bcprov-jdk16-1.46.jar:/usr/share/java/cglib-2.2.jar:/usr/share/java/hibernate-commons-annotations-3.2.0.Final.jar:/usr/share/java/openws-1.4.1.jar:/usr/share/java/axis2-codegen-1.4.1.jar:/usr/share/java/commons-dbcp-1.4.jar:/usr/share/java/slf4j-jdk14-1.5.11.jar:/usr/share/java/json-simple-1.1.jar:/usr/share/java/bcprov-jdk15-1.45.jar:/usr/share/java/geronimo-javamail_1.4_spec-1.2.jar:/usr/share/java/xpp3_min-1.1.4c.jar:/usr/share/java/xmlrpc-common-3.1.3.jar:/usr/share/java/neethi-2.0.4.jar:/usr/share/java/jetty-util-6.1.26.jar:/usr/share/java/opensaml-1.1.jar:/usr/share/java/trilead-ssh2-build213-svnkit-1.3-patch.jar:/usr/share/java/xstream-1.3.1.jar:/usr/share/java/xapi-5.6.100-1-SNAPSHOT.jar:/usr/share/java/jcip-annotations-1.0.jar:/usr/share/java/org.mortbay.jetty-5.1.10.jar:/usr/share/java/selenium-server-1.0-20081010.060147.jar:/usr/share/java/rampart-policy-1.5.jar:/usr/share/java/mysql-connector-java-5.1.21.jar:/usr/share/java/rampart-core-1.5.jar"
ACP="/usr/share/java/cloud-utils.jar:/usr/share/java/cloud-api.jar:/usr/share/java/cloud-core.jar:/usr/share/java/cloud-server.jar:/usr/share/java/cloud-server-extras.jar:/usr/share/java/cloud-agent.jar:/usr/share/java/cloud-console-common.jar:/usr/share/java/cloud-console-proxy.jar:/usr/share/java/cloud-core-extras.jar:/usr/share/java/cloud-agent-extras.jar:/usr/share/java/cloud-plugin-hypervisor-kvm.jar:/usr/share/java/jna.jar"
JCP="/usr/share/java/commons-daemon.jar"

# We need to append the JSVC daemon JAR to the classpath
# AgentShell implements the JSVC daemon methods
export CLASSPATH="$SCP:$DCP:$ACP:$JCP:/etc/cloud/agent"

wait_for_network() {
    i=1
    while [ $i -lt 10 ]
    do
        # Under Ubuntu and Debian libvirt by default creates a bridge called virbr0.
        # That's why we want more then 3 lines back from brctl, so that there is a manually created bridge
        if [ "$(brctl show|wc -l)" -gt 2 ]; then
            break
        else
            sleep 1
            let i=$i+1
            continue
        fi
    done
}

start() {
    if [ -s "$PIDFILE" ] && kill -0 $(cat "$PIDFILE") >/dev/null 2>&1; then
        log_daemon_msg "$PROGNAME apparently already running"
        log_end_msg 0
        exit 0
    fi

    log_daemon_msg "Starting $PROGNAME" "$SHORTNAME"
    if hostname --fqdn >/dev/null 2>&1 ; then
        true
    else
        log_failure_msg "The host name does not resolve properly to an IP address. Cannot start $PROGNAME"
        log_end_msg 1
        exit 1
    fi

    wait_for_network

    if start_daemon -p $PIDFILE $DAEMON -cp "$CLASSPATH" -pidfile "$PIDFILE" $CLASS
        RETVAL=$?
    then
        rc=0
        sleep 1
        if ! kill -0 $(cat "$PIDFILE") >/dev/null 2>&1; then
            log_failure_msg "$PROG failed to start"
            rc=1
        fi
    else
        rc=1
    fi

    if [ $rc -eq 0 ]; then
        log_end_msg 0
    else
        log_end_msg 1
        rm -f "$PIDFILE"
    fi
}

stop() {
    count="0"

    log_daemon_msg "Stopping $PROGNAME" "$SHORTNAME"
    killproc -p $PIDFILE $DAEMON

    until [ "$count" -gt "$SHUTDOWN_WAIT" ]
    do
        agentPid=$(ps aux|grep [j]svc|grep $SHORTNAME)
        if [ "$?" -gt "0" ];then
            break
        fi
        sleep 1
        let count="${count}+1"
    done

    agentPid=$(ps aux|grep [j]svc|grep $SHORTNAME)
    if [ "$?" -eq "0" ]; then
        agentPid=$(ps aux|grep [j]svc|awk '{print $2}')
        if [ "$agentPid" != "" ]; then
            log_warning_msg "$PROG still running, forcing kill"
            kill -9 $agentPid
        fi
    fi

    log_end_msg $?
    rm -f "$PIDFILE"
}

case "$1" in
    start)
        start
        ;;
    stop)
        stop
        ;;
    status)
        status_of_proc -p "$PIDFILE" "$PROG" "$SHORTNAME"
        RETVAL=$?
        ;;
    restart | force-reload)
        stop
        sleep 3
        start
        ;;
    *)
    echo "Usage: $0 {start|stop|restart|force-reload|status}"
    RETVAL=3
esac

exit $RETVAL