1 files changed, 136 insertions, 0 deletions
diff --git a/app-emulation/kvm-init-script/files/kvm-init-script b/app-emulation/kvm-init-script/files/kvm-init-script
new file mode 100644
index 0000000..7bfeb9d
--- /dev/null
+++ b/app-emulation/kvm-init-script/files/kvm-init-script
@@ -0,0 +1,136 @@
+#!/sbin/runscript
+# Copyright 2009 Gentoo Foundation
+# Distributed under the terms of the GNU General Public License v2
+
+VMNAME=${SVCNAME#*.}
+PIDFILE=/var/run/vm/${VMNAME}.pid
+MONITOR=/var/run/vm/${VMNAME}.monitor
+QTAP_FILE=/var/run/vm/${VMNAME}.qtap
+
+# Default to qemu-kvm in the absense of a setting, failing back
+# to kvm if no binary is found, finally, back to qemu.
+[ -z "$VMSOFTWARE" ] && VMSOFTWARE=$(type -p qemu-kvm)
+[ -z "$VMSOFTWARE" ] && VMSOFTWARE=$(type -p kvm)
+[ -z "$VMSOFTWARE" ] && VMSOFTWARE=$(type -p qemu)
+
+DROP_USER=${DROP_USER:-nobody}
+MEMORY=${MEMORY:-512M}
+TIMEOUT=${TIMEOUT:-300}
+SMP=${SMP:-1}
+export KVM_USER=${KVM_USER:-"root"}
+
+extra_commands="reboot"
+
+depend() {
+	need net.br0
+}
+
+send_command() {
+	local command="socat -u - UNIX-CONNECT:${MONITOR}"
+	type -p nc6 > /dev/null && command="nc6 -U ${MONITOR} --send-only"
+	echo "$@" | ${command} > /dev/null 2>&1
+}
+
+sanity_check() {
+	if [ "${VMNAME}" = "${SVCNAME}" ]; then
+		eerror "You have to create an init script for each vm:"
+		eerror " ln -s vm /etc/init.d/vm.vmname"
+		return 1
+	elif [ ! -f "${DISKIMAGE}" -a ! -b "${DISKIMAGE}" ]; then
+		eerror "couldn't find \$DISKIMAGE '$DISKIMAGE'"
+		return 1;
+	fi
+}
+
+start() {
+	sanity_check
+
+	img=$(readlink -f "${DISKIMAGE}")
+	[ -z "$img" ] && {
+		eerror "couldn't find ${DISKIMAGE}"
+		return 1;
+	}
+
+	mkdir -p "${PIDFILE%/*}" "${MONITOR%/*}"
+	ebegin "creating qtap ${QTAP:-(auto allocating one)}"
+	if [ -n "$QTAP" ]; then
+		qtap-manipulate create_specific "${QTAP}" -u "${DROP_USER}"
+	else
+		QTAP=$(qtap-manipulate create -u "${DROP_USER}")
+		if [ 0 != $? ]; then
+			eerror "failed to create qtap interface"
+			return 1
+		fi
+	fi
+	echo "${QTAP}" > ${QTAP_FILE}
+	eend $?
+
+	ebegin "Starting ${VMSOFTWARE##*/} for ${VMNAME} at VNC port${VNC}"
+	start-stop-daemon --start "${VMSOFTWARE}" \
+		--pidfile ${PIDFILE} \
+		-- -daemonize -pidfile ${PIDFILE} -monitor unix:${MONITOR},server,nowait \
+		-runas ${DROP_USER} -name ${VMNAME} \
+		-drive file="$img",if=${DRIVE_MODEL:-virtio},cache=${DRIVE_CACHE:-none} \
+		-net nic,model=${NIC_MODEL:-virtio},macaddr=${MACADDR} -net tap,ifname=${QTAP},script=no \
+		${DISABLE_KVM:---enable-kvm}  \
+		${MEMORY:+-m ${MEMORY}} ${SMP:+-smp ${SMP}} ${VNC:+-vnc ${VNC}} ${OTHER_ARGS}
+	ret=$?
+	if [ "0" != "${ret}" ]; then
+		qtap-manipulate destroy ${QTAP}
+	fi
+	eend ${ret}
+}
+
+reboot() {
+	if [ ${VMNAME} = ${SVCNAME} ]; then
+		eerror "You have to create an init script for each vm:"
+		eerror " ln -s vm /etc/init.d/vm.vmname"
+		return 1
+	fi
+
+	ebegin "Rebooting ${VMNAME}"
+	send_command system_reset
+	eend $?
+}
+
+stop() {
+	sanity_check
+
+	ebegin "Powering off ${VMNAME}"
+	send_command system_powerdown
+	eend $?
+
+	ebegin "waiting up to ${TIMEOUT:-60} seconds for it to die"
+	local pid
+	[ -s "${PIDFILE}" ] && pid=$(cat "${PIDFILE}")
+	if [ -z "$pid" ]; then
+		eerror "Couldn't find stored pid at '$PIDFILE'; user will have to manually kill kvm"
+		eerror "Will attempt to destroy qtap despite."
+		eend 1
+	else
+		local ret=1
+		for x in $(seq 0 ${TIMEOUT:-60}); do
+			if kill -0 "${pid}" > /dev/null 2>&1; then
+				sleep 1s
+				continue
+			fi
+			ret=0
+			break
+		done
+		eend $ret
+	fi
+
+	ebegin "Stopping ${VMSOFTWARE##*/} for ${VMNAME}"
+	start-stop-daemon --stop "${VMSOFTWARE}" \
+		--user "${DROP_USER}" \
+		--pidfile "${PIDFILE}" \
+		--quiet
+	eend $?
+	local qtap
+	[ -s "${QTAP_FILE}" ] && qtap=$(cat "${QTAP_FILE}")
+	if [ -n "$qtap" ]; then
+		ebegin "destroying qtap ${qtap}"
+		qtap-manipulate destroy ${qtap}
+		eend $?
+	fi
+}