#!/usr/bin/bash
#
# Copyright (C) 1997-2003 Sistina Software, Inc.  All rights reserved.
# Copyright (C) 2004-2011 Red Hat, Inc.  All rights reserved.
#
# This program is free software; you can redistribute it and/or
# modify it under the terms of the GNU General Public License
# as published by the Free Software Foundation; either version 2
# of the License, or (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
#

PATH=/bin:/sbin:/usr/bin:/usr/sbin

# Only allow pid status checks during monitor operations.
# Otherwise we want proceed with failing if virsh is not available.
ALLOW_PID_STATUS_CHECK=0
if [ "$1" = "monitor" ] || [ "$1" = "status" ]; then
	ALLOW_PID_STATUS_CHECK=1
fi

export PATH

. $(dirname $0)/ocf-shellfuncs || exit 1

#
# Virtual Machine start/stop script (requires the virsh command)
#

EMULATOR_STATE="/var/run/vm-${OCF_RESKEY_name}-emu.state"

# Indeterminate state: xend/libvirtd is down.
export OCF_APP_ERR_INDETERMINATE=150

meta_data()
{
    cat <<EOT
<?xml version="1.0"?>
<!DOCTYPE resource-agent SYSTEM "ra-api-1-modified.dtd">
<resource-agent version="rgmanager 2.0" name="vm">
    <version>1.0</version>

    <longdesc lang="en">
	Defines a Virtual Machine
    </longdesc>
    <shortdesc lang="en">
        Defines a Virtual Machine
    </shortdesc>

    <parameters>
        <parameter name="name" primary="1">
            <longdesc lang="en">
                This is the name of the virtual machine.
            </longdesc>
            <shortdesc lang="en">
                Name
            </shortdesc>
            <content type="string"/>
        </parameter>
    
        <parameter name="domain" reconfig="1">
            <longdesc lang="en">
                Failover domains define lists of cluster members
                to try in the event that the host of the virtual machine
		fails.
            </longdesc>
            <shortdesc lang="en">
                Cluster failover Domain
            </shortdesc>
            <content type="string"/>
        </parameter>

        <parameter name="autostart" reconfig="1">
            <longdesc lang="en">
	    	If set to yes, this resource group will automatically be started
		after the cluster forms a quorum.  If set to no, this virtual
		machine will start in the 'disabled' state after the cluster
		forms a quorum.
            </longdesc>
            <shortdesc lang="en">
	    	Automatic start after quorum formation
            </shortdesc>
            <content type="boolean" default="1"/>
        </parameter>

        <parameter name="exclusive" reconfig="1">
            <longdesc lang="en">
	    	If set, this resource group will only relocate to
		nodes which have no other resource groups running in the
		event of a failure.  If no empty nodes are available,
		this resource group will not be restarted after a failure.
		Additionally, resource groups will not automatically
		relocate to the node running this resource group.  This
		option can be overridden by manual start and/or relocate
		operations.
            </longdesc>
            <shortdesc lang="en">
	        Exclusive resource group
            </shortdesc>
            <content type="boolean" default="0"/>
        </parameter>

        <parameter name="recovery" reconfig="1">
            <longdesc lang="en">
	        This currently has three possible options: "restart" tries
		to restart this virtual machine locally before
		attempting to relocate (default); "relocate" does not bother
		trying to restart the VM locally; "disable" disables
		the VM if it fails.
            </longdesc>
            <shortdesc lang="en">
	    	Failure recovery policy
            </shortdesc>
            <content type="string"/>
        </parameter>

       <parameter name="migration_mapping" reconfig="1">
           <longdesc lang="en">
               Mapping of the hostname of a target cluster member to a different hostname
           </longdesc>
           <shortdesc lang="en">
               memberhost:targethost,memberhost:targethost ..
           </shortdesc>
            <content type="string"/>
        </parameter>

	<parameter name="use_virsh">
	    <longdesc lang="en">
	    	Force use of virsh instead of xm on Xen machines.
	    </longdesc>
	    <shortdesc lang="en">
	    	If set to 1, vm.sh will use the virsh command to manage
		virtual machines instead of xm.  This is required when
		using non-Xen virtual machines (e.g. qemu / KVM).
	    </shortdesc>
            <content type="integer" default=""/>
        </parameter>

	<parameter name="xmlfile">
	    <longdesc lang="en">
	    	Full path to libvirt XML file describing the domain.
	    </longdesc>
	    <shortdesc lang="en">
	    	Full path to libvirt XML file describing the domain.
	    </shortdesc>
            <content type="string"/>
	</parameter>

	<parameter name="migrate">
	    <longdesc lang="en">
	    	Migration type (live or pause, default = live).
	    </longdesc>
	    <shortdesc lang="en">
	    	Migration type (live or pause, default = live).
	    </shortdesc>
            <content type="string" default="live"/>
        </parameter>

	<parameter name="migrate_options">
	    <longdesc lang="en">
	    	Extra options for the guest live migration.
	    </longdesc>
	    <shortdesc lang="en">
	    	Extra options for the guest live migration.
	    </shortdesc>
            <content type="string"/>
        </parameter>

	<parameter name="tunnelled">
	    <longdesc lang="en">
	    	Tunnel data over ssh to securely migrate virtual machines.
	    </longdesc>
	    <shortdesc lang="en">
	    	Tunnel data over ssh to securely migrate virtual machines.
	    </shortdesc>
            <content type="string" default=""/>
        </parameter>

	<parameter name="path">
	    <longdesc lang="en">
		Path specification vm.sh will search for the specified
 		VM configuration file.  /path1:/path2:...
	    </longdesc>
	    <shortdesc lang="en">
		Path to virtual machine configuration files.
 	    </shortdesc>
	    <content type="string"/>
 	</parameter>

	<parameter name="snapshot">
	    <longdesc lang="en">
	    	Path to the snapshot directory where the virtual machine
		image will be stored.
	    </longdesc>
	    <shortdesc lang="en">
	    	Path to the snapshot directory where the virtual machine
		image will be stored.
	    </shortdesc>
            <content type="string" default=""/>
        </parameter>

        <parameter name="depend">
            <longdesc lang="en">
		Service dependency; will not start without the specified
		service running.
            </longdesc>
            <shortdesc lang="en">
		Top-level service this depends on, in service:name format.
            </shortdesc>
            <content type="string"/>
        </parameter>

        <parameter name="depend_mode">
            <longdesc lang="en">
		Service dependency mode.
		hard - This service is stopped/started if its dependency
		       is stopped/started
		soft - This service only depends on the other service for
		       initial startip.  If the other service stops, this
		       service is not stopped.
            </longdesc>
            <shortdesc lang="en">
	    	Service dependency mode (soft or hard).
            </shortdesc>
            <content type="string" default="hard"/>
        </parameter>

        <parameter name="max_restarts" reconfig="1">
            <longdesc lang="en">
	    	Maximum restarts for this service.
            </longdesc>
            <shortdesc lang="en">
	    	Maximum restarts for this service.
            </shortdesc>
            <content type="string" default="0"/>
        </parameter>

        <parameter name="restart_expire_time" reconfig="1">
            <longdesc lang="en">
	    	Restart expiration time.  A restart is forgotten
		after this time.  When combined with the max_restarts
		option, this lets administrators specify a threshold
		for when to fail over services.  If max_restarts
		is exceeded in this given expiration time, the service
		is relocated instead of restarted again.
            </longdesc>
            <shortdesc lang="en">
	    	Restart expiration time; amount of time before a restart
		is forgotten.
            </shortdesc>
            <content type="string" default="0"/>
        </parameter>

        <parameter name="status_program" reconfig="1">
            <longdesc lang="en">
	    	Ordinarily, only the presence/health of a virtual machine
		is checked.  If specified, the status_program value is
		executed during a depth 10 check.  The intent of this 
		program is to ascertain the status of critical services
		within a virtual machine.
            </longdesc>
            <shortdesc lang="en">
	    	Additional status check program
            </shortdesc>
            <content type="string" default=""/>
        </parameter>

	<parameter name="hypervisor">
            <longdesc lang="en">
		Specify hypervisor tricks to use.  Default = auto.
		Other supported options are xen and qemu.
            </longdesc>
            <shortdesc lang="en">
		Hypervisor
            </shortdesc >
	    <content type="string" default="auto"/>
	</parameter>

	<parameter name="hypervisor_uri">
            <longdesc lang="en">
		Hypervisor URI.  Generally, this is keyed off of the
		hypervisor and does not need to be set.
            </longdesc>
            <shortdesc lang="en">
		Hypervisor URI (normally automatic).
            </shortdesc >
	    <content type="string" default="auto" />
	</parameter>

	<parameter name="migration_uri">
            <longdesc lang="en">
		Migration URI.  Generally, this is keyed off of the
		hypervisor and does not need to be set.
            </longdesc>
            <shortdesc lang="en">
		Migration URI (normally automatic).
            </shortdesc >
	    <content type="string" default="auto" />
	</parameter>

	<parameter name="no_kill">
            <longdesc lang="en">
		Do not force kill vm during stop, instead
		fail after the timeout expires.
            </longdesc>
            <shortdesc lang="en">
		Don't force kill vm on stop.
            </shortdesc >
	    <content type="boolean" default="false" />
	</parameter>

    </parameters>

    <actions>
        <action name="start" timeout="300"/>
        <action name="stop" timeout="120"/>
	
        <action name="status" timeout="10" interval="30"/>
        <action name="monitor" timeout="10" interval="30"/>

	<!-- depth 10 calls the status_program -->
        <action name="status" depth="10" timeout="20" interval="60"/>
        <action name="monitor" depth="10" timeout="20" interval="60"/>

	<!-- reconfigure - reconfigure with new OCF parameters.
	     NOT OCF COMPATIBLE AT ALL -->
	<action name="reconfig" timeout="10"/>

	<action name="migrate" timeout="10m"/>

        <action name="meta-data" timeout="5"/>
        <action name="validate-all" timeout="5"/>

    </actions>
    
    <special tag="rgmanager">
    	<!-- Destroy_on_delete / init_on_add are currently only
	     supported for migratory resources (no children
	     and the 'migrate' action; see above.  Do not try this
	     with normal services -->
        <attributes maxinstances="1" destroy_on_delete="0" init_on_add="0"/>
    </special>
</resource-agent>
EOT
}


build_virsh_cmdline()
{
	declare cmdline=""
	declare operation=$1

	if [ -n "$OCF_RESKEY_hypervisor_uri" ]; then
		cmdline="$cmdline -c $OCF_RESKEY_hypervisor_uri"
	fi

	cmdline="$cmdline $operation $OCF_RESKEY_name"
		
	echo $cmdline
}


# this is only used on startup
build_xm_cmdline()
{
	declare operation=$1
	#
	# Virtual domains should never restart themselves when 
	# controlled externally; the external monitoring app
	# should.
	#
	declare cmdline="on_shutdown=\"destroy\" on_reboot=\"destroy\" on_crash=\"destroy\""

	if [ -n "$OCF_RESKEY_path" ]; then
		operation="$operation --path=\"$OCF_RESKEY_path\""
	fi

	if [ -n "$OCF_RESKEY_name" ]; then
		cmdline="$operation $OCF_RESKEY_name $cmdline"
	fi

	echo $cmdline
}


do_xm_start()
{
	# Use /dev/null for the configuration file, if xmdefconfig
	# doesn't exist...
	#
	declare cmdline

	echo -n "Virtual machine $OCF_RESKEY_name is "
	do_status && return 0

	cmdline="`build_xm_cmdline create`"

	ocf_log debug "xm $cmdline"

	eval xm $cmdline
	return $?
}


get_timeout()
{
	declare -i default_timeout=60
	declare -i tout=60

	if [ -n "$OCF_RESKEY_RGMANAGER_meta_timeout" ]; then
		tout=$OCF_RESKEY_RGMANAGER_meta_timeout
	elif [ -n "$OCF_RESKEY_CRM_meta_timeout" ]; then
		tout=$OCF_RESKEY_CRM_meta_timeout
	fi

	if [ $tout -eq 0 ]; then
		echo $default_timeout
		return 0
	fi
	if [ $tout -lt 0 ]; then
		echo $default_timeout
		return 0
	fi

	echo $tout
	return 0
}

get_emulator()
{
	local emulator=""

	emulator=$(virsh $VIRSH_OPTIONS dumpxml $OCF_RESKEY_name 2>/dev/null | sed -n -e 's/^.*<emulator>\(.*\)<\/emulator>.*$/\1/p')
	if [ -z "$emulator" ] && [ -a "$EMULATOR_STATE" ]; then
		emulator=$(cat $EMULATOR_STATE)
	fi
	if [ -z "$emulator" ]; then
		emulator=$(cat ${OCF_RESKEY_config} | sed -n -e 's/^.*<emulator>\(.*\)<\/emulator>.*$/\1/p')
	fi

	if [ -n "$emulator" ]; then
		basename $emulator
	else 
		ocf_log error "Unable to determine emulator for $OCF_RESKEY_name" 
	fi
}

update_emulator_cache()
{
	local emulator

	emulator=$(get_emulator)
	if [ -n "$emulator" ]; then
		echo $emulator > $EMULATOR_STATE
	fi
}

#
# Start a virtual machine given the parameters from
# the environment.
#
do_virsh_start()
{
	declare cmdline
	declare snapshotimage
	declare rc

	echo -n "Virtual machine $OCF_RESKEY_name is "
	do_status && return 0

	snapshotimage="$OCF_RESKEY_snapshot/$OCF_RESKEY_name"

        if [ -n "$OCF_RESKEY_snapshot" -a -f "$snapshotimage" ]; then
		eval virsh restore $snapshotimage
		if [ $? -eq 0 ]; then
			rm -f $snapshotimage
			return 0
		fi
		return 1
	fi

	if [ -n "$OCF_RESKEY_xmlfile" -a -f "$OCF_RESKEY_xmlfile" ]; then
		# TODO: try to use build_virsh_cmdline for the hypervisor_uri
		cmdline="virsh create $OCF_RESKEY_xmlfile"
	else
		cmdline="virsh $(build_virsh_cmdline start)"
	fi

	ocf_log debug "$cmdline"

	$cmdline
	rc=$?

	update_emulator_cache

	return $rc
}

do_xm_stop()
{
	declare -i timeout=60
	declare -i ret=1
	declare st

	for op in $*; do
		echo "CMD: xm $op $OCF_RESKEY_name"
		xm $op $OCF_RESKEY_name

		timeout=60
		while [ $timeout -gt 0 ]; do
			sleep 5
			((timeout -= 5))
			do_status&>/dev/null || return 0
			while read dom state; do
				#
				# State is "stopped".  Kill it.
				#
				if [ "$dom" != "$OCF_RESKEY_name" ]; then
					continue
				fi
				if [ "$state" != "---s-" ]; then
					continue
				fi
				xm destroy $OCF_RESKEY_name
			done < <(xm list | awk '{print $1, $5}')
		done
	done

	return 1
}


#
# Stop a VM.  Try to shut it down.  Wait a bit, and if it
# doesn't shut down, destroy it.
#
do_virsh_stop()
{
	declare -i timeout=$(get_timeout)
	declare -i ret=1
	declare state

	state=$(do_status)
	[ $? -eq 0 ] || return 0

	if [ -n "$OCF_RESKEY_snapshot" ]; then
		virsh save $OCF_RESKEY_name "$OCF_RESKEY_snapshot/$OCF_RESKEY_name"
	fi

	for op in $*; do
		echo virsh $op $OCF_RESKEY_name ...
		virsh $op $OCF_RESKEY_name

		timeout=$(get_timeout)
		while [ $timeout -gt 0 ]; do
			sleep 5
			((timeout -= 5))
			state=$(do_status)
			[ $? -eq 0 ] || return 0

			if [ "$state" = "paused" ]; then
				virsh destroy $OCF_RESKEY_name
			fi
		done
	done

	ocf_log err "Stop operation timed out for vm '$OCF_RESKEY_name'"
	return 1
}


do_start()
{
	if [ "$OCF_RESKEY_use_virsh" = "1" ]; then
		do_virsh_start $*
		return $?
	fi

	do_xm_start $*
	return $?
}


do_stop()
{
	declare domstate rv

	domstate=$(do_status)
	rv=$?
	ocf_log debug "Virtual machine $OCF_RESKEY_name is $domstate"
	if [ $rv -eq $OCF_APP_ERR_INDETERMINATE ]; then
		ocf_log crit "xend/libvirtd is dead; cannot stop $OCF_RESKEY_name"
		return 1
	fi

	if [ "$OCF_RESKEY_use_virsh" = "1" ]; then
		do_virsh_stop $*
		return $?
	fi

	do_xm_stop $*
	return $?
}


#
# Reconfigure a running VM.
#
reconfigure()
{
	return 0
}


xm_status()
{
	service xend status &> /dev/null
	if [ $? -ne 0 ]; then 
		# if xend died
		echo indeterminate
		return $OCF_APP_ERR_INDETERMINATE
	fi

	xm list $OCF_RESKEY_name &> /dev/null
	if [ $? -eq 0 ]; then
		echo "running"
		return 0
	fi
	xm list migrating-$OCF_RESKEY_name &> /dev/null
	if [ $? -eq 0 ]; then
		echo "running"
		return 0
	fi
	echo "not running"
	return $OCF_NOT_RUNNING
}

# attempt to check domain status outside of libvirt using the emulator process
vm_pid_status()
{
	local rc=$OCF_ERR_GENERIC
	local emulator

	if [ $ALLOW_PID_STATUS_CHECK -eq 0 ]; then
		echo "indeterminate"
		return $OCF_APP_ERR_INDETERMINATE
	fi

	emulator=$(get_emulator)
	case "$emulator" in
		qemu-kvm|qemu-system-*)
			rc=$OCF_NOT_RUNNING
			ps awx | grep -E "[q]emu-(kvm|system).*-name $OCF_RESKEY_name " > /dev/null 2>&1
			if [ $? -eq 0 ]; then
				rc=$OCF_SUCCESS
			fi
			;;
		# This can be expanded to check for additional emulators
		*)
			echo "indeterminate"
			return $OCF_APP_ERR_INDETERMINATE
			;;
	esac

	if [ $rc -eq $OCF_SUCCESS ]; then
		echo "running"
	elif [ $rc -eq $OCF_NOT_RUNNING ]; then
		echo "shut off"
	fi

	return $rc
}

virsh_status()
{
	declare state pid

	if [ "$OCF_RESKEY_hypervisor" = "xen" ]; then
		service xend status &> /dev/null
		if [ $? -ne 0 ]; then 
			echo indeterminate
			return $OCF_APP_ERR_INDETERMINATE
		fi
	fi

	#
	# libvirtd is required when using virsh even though
	# not specifically when also using Xen.  This is because
	# libvirtd is required for migration.
	#
	pid=$(pidof libvirtd)
	if [ -z "$pid" ]; then
		# attempt to determine if vm is running from pid file
		vm_pid_status
		return $?
	fi

	state=$(virsh domstate $OCF_RESKEY_name)

	echo $state

	if [ "$state" = "running" ] || [ "$state" = "paused" ] || [ "$state" = "no state" ] || 
	   [ "$state" = "idle" ]; then
		update_emulator_cache
		return 0
	fi

	if [ "$state" = "shut off" ]; then
		return $OCF_NOT_RUNNING
	fi

	return $OCF_ERR_GENERIC
}


#
# Simple status check: Find the VM in the list of running
# VMs
#
do_status()
{
	if [ "$OCF_RESKEY_use_virsh" = "1" ]; then
		virsh_status
		return $?
	fi

	xm_status
	return $?
}


#
# virsh "path" attribute support
#
check_config_file()
{
	declare path=$1

	if [ -f "$path/$OCF_RESKEY_name" ]; then
		echo $path/$OCF_RESKEY_name
		return 2
	elif [ -f "$path/$OCF_RESKEY_name.xml" ]; then
		echo $path/$OCF_RESKEY_name.xml
		return 2
	fi

	return 0
}


parse_input()
{
	declare delim=$1
	declare input=$2
	declare func=$3
	declare inp
	declare value

	while [ -n "$input" ]; do
		value=${input/$delim*/}
		if [ -n "$value" ]; then
			eval $func $value
			if [ $? -eq 2 ]; then
				return 0
			fi
		fi
		inp=${input/$value$delim/}
		if [ "$input" = "$inp" ]; then
			inp=${input/$value/}
		fi
		input=$inp
	done
}


search_config_path()
{
	declare config_file=$(parse_input ":" "$OCF_RESKEY_path" check_config_file)

	if [ -n "$config_file" ]; then
		export OCF_RESKEY_xmlfile=$config_file
		return 0
	fi

	return 1
}


choose_management_tool()
{
	declare -i is_xml

	#
	# Don't override user value for use_virsh if one is given
	#
	if [ -n "$OCF_RESKEY_use_virsh" ]; then
		return 0
	fi

	which xmllint &> /dev/null
	if [ $? -ne 0 ]; then
		ocf_log warning "Could not find xmllint; assuming virsh mode"
		export OCF_RESKEY_use_virsh=1
		unset OCF_RESKEY_path
		return 0
	fi

	xmllint $OCF_RESKEY_xmlfile &> /dev/null
	is_xml=$?

	if [ $is_xml -eq 0 ]; then
		ocf_log debug "$OCF_RESKEY_xmlfile is XML; using virsh"
		export OCF_RESKEY_use_virsh=1
		unset OCF_RESKEY_path
	else
		ocf_log debug "$OCF_RESKEY_xmlfile is not XML; using xm"
		export OCF_RESKEY_use_virsh=0
		unset OCF_RESKEY_xmlfile
	fi

	return 0
}

get_hypervisor()
{
	local hypervisor="`virsh version | grep \"Running hypervisor:\" | awk '{print $3}' | tr A-Z a-z`"
	# if virsh gives us nothing (likely because libvirt is down), we can attempt
	# to determine auto detect the hypervisor is qemu if a qemu emulator is used
	# for this domain.
	if [ -z "$hypervisor" ]; then
		get_emulator | grep "qemu" > /dev/null 2>&1
		if [ $? -eq 0 ]; then
			hypervisor="qemu"
		fi
	fi
	echo $hypervisor
}

validate_all()
{
	if [ "$(id -u)" != "0" ]; then
	       ocf_log err "Cannot control VMs. as non-root user."
	       return 1
	fi

	#
	# If someone selects a hypervisor, honor it.
	# Otherwise, ask virsh what the hypervisor is.
	#
	if [ -z "$OCF_RESKEY_hypervisor" ] ||
	   [ "$OCF_RESKEY_hypervisor" = "auto" ]; then
		export OCF_RESKEY_hypervisor="`get_hypervisor`"
		if [ -z "$OCF_RESKEY_hypervisor" ]; then
			ocf_log err "Could not determine Hypervisor"
			return $OCF_ERR_ARGS
		fi
		echo Hypervisor: $OCF_RESKEY_hypervisor 
	fi

	#
	# Xen hypervisor only for when use_virsh = 0.
	#
	if [ "$OCF_RESKEY_use_virsh" = "0" ]; then
		if [ "$OCF_RESKEY_hypervisor" != "xen" ]; then
			ocf_log err "Cannot use $OCF_RESKEY_hypervisor hypervisor without using virsh"
			return $OCF_ERR_ARGS
		fi

		if [ -n "$OCF_RESKEY_xmlfile" ]; then
			ocf_log err "Cannot use xmlfile if use_virsh is set to 0"
			return $OCF_ERR_ARGS
		fi
	else

		#
		# Virsh path support.
		#
		if [ -n "$OCF_RESKEY_path" ] &&
		   [ "$OCF_RESKEY_path" != "/etc/xen" ]; then
			if [ -n "$OCF_RESKEY_xmlfile" ]; then
				ocf_log warning "Using $OCF_RESKEY_xmlfile instead of searching $OCF_RESKEY_path"
			else
				search_config_path
				if [ $? -ne 0 ]; then
					ocf_log warning "Could not find $OCF_RESKEY_name or $OCF_RESKEY_name.xml in search path $OCF_RESKEY_path"
					unset OCF_RESKEY_xmlfile
				else
					ocf_log debug "Using $OCF_RESKEY_xmlfile"
				fi
				choose_management_tool
			fi
		else
			export OCF_RESKEY_use_virsh=1
		fi
	fi

	if [ "$OCF_RESKEY_use_virsh" = "0" ]; then

		echo "Management tool: xm"
		which xm &> /dev/null
		if [ $? -ne 0 ]; then
			ocf_log err "Cannot find 'xm'; is it installed?"
			return $OCF_ERR_INSTALLED
		fi

		if [ "$OCF_RESKEY_hypervisor" != "xen" ]; then
			ocf_log err "Cannot use $OCF_RESKEY_hypervisor hypervisor without using virsh"
			return $OCF_ERR_ARGS
		fi
	else
		echo "Management tool: virsh"
		which virsh &> /dev/null
		if [ $? -ne 0 ]; then
			ocf_log err "Cannot find 'virsh'; is it installed?"
			return $OCF_ERR_INSTALLED
		fi
	fi

	#
	# Set the hypervisor URI
	#
	if [ -z "$OCF_RESKEY_hypervisor_uri" -o "$OCF_RESKEY_hypervisor_uri" = "auto" ] &&
	   [ "$OCF_RESKEY_use_virsh" = "1" ]; then

		# Virsh makes it easier to do this.  Really.
		if [ "$OCF_RESKEY_hypervisor" = "qemu" ]; then
			OCF_RESKEY_hypervisor_uri="qemu:///system"
		fi

		# I just need to believe in it more.
		if [ "$OCF_RESKEY_hypervisor" = "xen" ]; then
			OCF_RESKEY_hypervisor_uri="xen:///"
		fi

		echo Hypervisor URI: $OCF_RESKEY_hypervisor_uri
	fi

	#
	# Set the migration URI
	#
	if [ -z "$OCF_RESKEY_migration_uri" -o "$OCF_RESKEY_migration_uri" = "auto" ] &&
	   [ "$OCF_RESKEY_use_virsh" = "1" ]; then

		# Virsh makes it easier to do this.  Really.
		if [ "$OCF_RESKEY_hypervisor" = "qemu" ]; then
			export OCF_RESKEY_migration_uri="qemu+ssh://%s/system"
		fi

		# I just need to believe in it more.
		if [ "$OCF_RESKEY_hypervisor" = "xen" ]; then
			export OCF_RESKEY_migration_uri="xenmigr://%s/"
		fi

		[ -n "$OCF_RESKEY_migration_uri" ] && echo Migration URI format: $(printf $OCF_RESKEY_migration_uri target_host)
	fi

	if [ -z "$OCF_RESKEY_name" ]; then
		echo No domain name specified
		return $OCF_ERR_ARGS
	fi

	if [ "$OCF_RESKEY_hypervisor" = "qemu" ]; then
		export migrateuriopt="tcp:%s"
	fi

	case "$OCF_RESKEY_no_kill" in
		yes|true|1|YES|TRUE|on|ON)
			OCF_RESKEY_no_kill=1
		;;
	esac
	#virsh list --all | awk '{print $2}' | grep -q "^$OCF_RESKEY_name\$"
	return $?
}


virsh_migrate()
{
	declare target=$1
	declare rv=1

	#
	# Xen and qemu have different migration mechanisms
	#
	if [ "$OCF_RESKEY_hypervisor" = "xen" ]; then 
		cmd="virsh migrate $migrate_opt $OCF_RESKEY_migrate_options $OCF_RESKEY_name $OCF_RESKEY_hypervisor_uri $(printf $OCF_RESKEY_migration_uri $target)"
		ocf_log debug "$cmd"
		
		err=$($cmd 2>&1 | head -1; exit ${PIPESTATUS[0]})
		rv=$?
	elif [ "$OCF_RESKEY_hypervisor" = "qemu" ]; then
		if [ -z "$tunnelled_opt" ]; then
			cmd="virsh migrate $tunnelled_opt $migrate_opt $OCF_RESKEY_migrate_options $OCF_RESKEY_name $(printf $OCF_RESKEY_migration_uri $target) $(printf $migrateuriopt $target)"
		else
			cmd="virsh migrate $tunnelled_opt $migrate_opt $OCF_RESKEY_migrate_options $OCF_RESKEY_name $(printf $OCF_RESKEY_migration_uri $target)"
		fi
		ocf_log debug "$cmd"
		
		err=$($cmd 2>&1 | head -1; exit ${PIPESTATUS[0]})
		rv=$?
	fi

	if [ $rv -ne 0 ]; then
		ocf_log err "Migrate $OCF_RESKEY_name to $target failed:"
		ocf_log err "$err"

		if [ "$err" != "${err/does not exist/}" ]; then
			return $OCF_ERR_CONFIGURED
		fi
		if [ "$err" != "${err/Domain not found/}" ]; then
			return $OCF_ERR_CONFIGURED
		fi

		return $OCF_ERR_GENERIC
	fi

	return $rv
}


#
# XM migrate
#
xm_migrate()
{
	declare target=$1
	declare errstr rv migrate_opt cmd

	rv=1

	if [ "$OCF_RESKEY_migrate" = "live" ]; then
		migrate_opt="-l"
	fi

	# migrate() function  sets target using migration_mapping;
	# no need to do it here anymore
	cmd="xm migrate $migrate_opt $OCF_RESKEY_migrate_options $OCF_RESKEY_name $target"
	ocf_log debug "$cmd"

	err=$($cmd 2>&1 | head -1; exit ${PIPESTATUS[0]})
	rv=$?

	if [ $rv -ne 0 ]; then
		ocf_log err "Migrate $OCF_RESKEY_name to $target failed:"
		ocf_log err "$err"

		if [ "$err" != "${err/does not exist/}" ]; then
			return $OCF_NOT_RUNNING
		fi
		if [ "$err" != "${err/Connection refused/}" ]; then
			return $OCF_ERR_CONFIGURED
		fi

		return $OCF_ERR_GENERIC
	fi

	return $?
}

# 
# Virsh migrate
#
migrate()
{
	declare target=$1
	declare rv migrate_opt
	declare tunnelled_opt

	if [ "$OCF_RESKEY_migrate" = "live" ]; then
		migrate_opt="--live"
	fi

	case "$OCF_RESKEY_tunnelled" in
		yes|true|1|YES|TRUE|on|ON)
			tunnelled_opt="--tunnelled --p2p"
		;;
	esac

	# Patch from Marcelo Azevedo to migrate over private
	# LANs instead of public LANs
        if [ -n "$OCF_RESKEY_migration_mapping" ] ; then
                target=${OCF_RESKEY_migration_mapping#*$target:} target=${target%%,*}
        fi

	if [ "$OCF_RESKEY_use_virsh" = "1" ]; then
		virsh_migrate $target
		rv=$?
	else
		xm_migrate $target
		rv=$?
	fi

	return $rv
}


wait_start()
{
	declare -i timeout_remaining=$(get_timeout)
	declare -i start_time
	declare -i end_time
	declare -i delta
	declare -i sleep_time

	if [ -z "$OCF_RESKEY_status_program" ]; then
		return 0
	fi

	while [ $timeout_remaining -gt 0 ]; do
		start_time=$(date +%s)
		bash -c "$OCF_RESKEY_status_program"
		if [ $? -eq 0 ]; then
			return 0
		fi
		end_time=$(date +%s)
		delta=$(((end_time - start_time)))
		sleep_time=$(((5 - delta)))

		((timeout_remaining -= $delta))
		if [ $sleep_time -gt 0 ]; then
			sleep $sleep_time
			((timeout_remaining -= $sleep_time))
		fi
	done

	ocf_log err "Start of $OCF_RESOURCE_INSTANCE has failed"
	ocf_log err "Timeout exceeded while waiting for \"$OCF_RESKEY_status_program\""

	return 1
}

#
#
#

case $1 in
	start)
		validate_all || exit $OCF_ERR_ARGS
		do_start
		rv=$?
		if [ $rv -ne 0 ]; then
			exit $rv
		fi

		wait_start
		exit $?
		;;
	stop)
		validate_all || exit $OCF_ERR_ARGS
		if [ $OCF_RESKEY_no_kill -eq 1 ]; then
			do_stop shutdown
		else
			do_stop shutdown destroy
		fi
		exit $?
		;;
	kill)
		validate_all || exit $OCF_ERR_ARGS
		do_stop destroy
		exit $?
		;;
	recover|restart)
		exit 0
		;;
	status|monitor)
		validate_all || exit $OCF_ERR_ARGS
		echo -n "Virtual machine $OCF_RESKEY_name is "
		do_status
		rv=$?
		if [ $rv -ne 0 ]; then
			exit $rv
		fi
		[ -z "$OCF_RESKEY_status_program" ] && exit 0
		[ -z "$OCF_CHECK_LEVEL" ] && exit 0
		[ $OCF_CHECK_LEVEL -lt 10 ] && exit 0

		bash -c "$OCF_RESKEY_status_program" &> /dev/null
		exit $?
		;;
	migrate)
		validate_all || exit $OCF_ERR_ARGS
		migrate $2 # Send VM to this node
		rv=$?
		if [ $rv -eq $OCF_ERR_GENERIC ]; then
			# Catch-all: If migration failed with
			# an unhandled error, do a status check
			# to see if the VM is really dead.
			#
			# If the VM is still in good health, return
			# a value to rgmanager to indicate the 
			# non-critical error
			#
			# OCF states that codes 150-199 are reserved
			# for application use, so we'll use 150
			#
			do_status > /dev/null
			if [ $? -eq 0 ]; then
				rv=150
			fi
		fi
		exit $rv
		;;
	reload)
		exit 0
		;;
	reconfig)
		validate_all || exit $OCF_ERR_ARGS
		echo "$0 RECONFIGURING $OCF_RESKEY_memory"
		reconfigure
		exit $?
		;;
	meta-data)
		meta_data
		exit 0
		;;
	validate-all)
		validate_all
		exit $?
		;;
	*)
		echo "usage: $0 {start|stop|restart|status|reload|reconfig|meta-data|validate-all}"
		exit 1
		;;
esac
