From dc2786ff4d4cfc3a9a780451ef2a559e296c05a3 Mon Sep 17 00:00:00 2001
From: Simon Rettberg
Date: Thu, 2 May 2024 15:37:46 +0200
Subject: [run-virt] Add support for CoW-Mode

---
 .../data/opt/openslx/vmchooser/vmchooser-run_virt  | 37 ++++++++++++++++++++--
 1 file changed, 34 insertions(+), 3 deletions(-)

(limited to 'core/modules/run-virt/data/opt/openslx/vmchooser/vmchooser-run_virt')

diff --git a/core/modules/run-virt/data/opt/openslx/vmchooser/vmchooser-run_virt b/core/modules/run-virt/data/opt/openslx/vmchooser/vmchooser-run_virt
index ca881f29..dfebebe7 100755
--- a/core/modules/run-virt/data/opt/openslx/vmchooser/vmchooser-run_virt
+++ b/core/modules/run-virt/data/opt/openslx/vmchooser/vmchooser-run_virt
@@ -182,14 +182,45 @@ main() {
 
 	run_hooks "pre-exec" "$PLUGIN_ID" "$IMGUUID"
 
+	# Launch COWGUI in CoW-Edit-Mode
+	local cowpid cowurl vmpidfile
+	cowpid=
+	vmpidfile=
+	if [ -n "$DMSD_COW_SESSION" ]; then
+		vmpidfile="$( mktemp -p "/run/user/$( id -u )" )"
+		cowurl="${SLX_VMCHOOSER_BASE_URL//"/vmchooser/"/"/cow/"}"
+		(
+		cntr=0
+		while true; do
+			cowgui --session "$DMSD_COW_SESSION" --url "$cowurl" --pid "$DNBD3_PID" && break
+			# Unclean exit, let's see if it's worth relaunching
+			state="$( curl -m 3 -sS -L "$cowurl/status/$DMSD_COW_SESSION" | jq -r .state )"
+			[ -z "$state" ] && break
+			[ "$state" = "PROCESSING" ] && break
+			[ "$state" = "ERROR" ] && break
+			[ "$state" = "COMPLETELY_DONE" ] && break
+			(( cntr++ > 10 )) && break
+		done
+		[ -n "$vmpidfile" ] && kill "$( cat "$vmpidfile" )"
+		) &
+		cowpid=$!
+	fi
+
 	writelog "VM command: ${VIRTCMD} ${VIRTCMDOPTS[*]}"
-	# This will start the VM (no eval needed!)
+	# This will start the VM
 	writelog "---------- BEGIN VM command output ----------"
-	local retval
-	"${VIRTCMD}" "${VIRTCMDOPTS[@]}" &>> "${LOGFILE}"
+	local retval vmpid
+	"${VIRTCMD}" "${VIRTCMDOPTS[@]}" &>> "${LOGFILE}" &
+	vmpid=$!
+	[ -n "$vmpidfile" ] && echo "$vmpid" > "$vmpidfile"
+	wait "$vmpid"
 	retval=${?}
 	writelog "---------- END VM command output ----------"
 
+	[ -n "$DNBD3_PID" ] && kill "$DNBD3_PID"
+	# If cow, wait for GUI
+	[ -n "$cowpid" ] && wait "$cowpid"
+
 	run_hooks "post-exec" "$PLUGIN_ID" "$IMGUUID"
 
 	writelog "Virtualizer exited with '$retval'. Bye."
-- 
cgit v1.2.3-55-g7522