fix: Folder structure (#379)

* Moved to src * Moved to src * Moved to src * Moved to src * Moved to src * Moved to src * Moved to src * Moved to src * Moved to src * Moved to src * Moved to src * Moved to src * fix: Check entrypoint * Moved to src * Moved to src * fix: Relative paths * fix: Relative paths * fix: Shellcheck * fix: Relative paths * Test shellcheck * Test shellcheck
2025-11-07 18:43:41 +08:00 · 2023-11-15 20:54:51 +01:00
parent 0cd1ddf0a1
commit a024294e19
14 changed files with 14 additions and 13 deletions
--- a/src/power.sh
+++ b/src/power.sh
@@ -0,0 +1,74 @@
+#!/usr/bin/env bash
+set -Eeuo pipefail
+
+# Configure QEMU for graceful shutdown
+
+QEMU_PORT=7100
+QEMU_TIMEOUT=50
+
+QEMU_PID=/run/qemu.pid
+QEMU_COUNT=/run/qemu.count
+
+rm -f "${QEMU_PID}"
+rm -f "${QEMU_COUNT}"
+
+_trap(){
+    func="$1" ; shift
+    for sig ; do
+        trap "$func $sig" "$sig"
+    done
+}
+
+_graceful_shutdown() {
+
+  set +e
+
+  [ ! -f "${QEMU_PID}" ] && exit 130
+  [ -f "${QEMU_COUNT}" ] && return
+
+  echo && info "Received $1 signal, shutting down..."
+  echo 0 > "${QEMU_COUNT}"
+
+  # Don't send the powerdown signal because vDSM ignores ACPI signals
+  # echo 'system_powerdown' | nc -q 1 -w 1 localhost "${QEMU_PORT}" > /dev/null
+
+  # Send shutdown command to guest agent via serial port
+  RESPONSE=$(curl -s -m 5 -S http://127.0.0.1:2210/read?command=6 2>&1)
+
+  if [[ ! "${RESPONSE}" =~ "\"success\"" ]] ; then
+
+    echo && error "Could not send shutdown command to the guest ($RESPONSE)"
+
+    kill -15 "$(cat "${QEMU_PID}")"
+    pkill -f qemu-system-x86_64 || true
+
+  fi
+
+  while [ "$(cat ${QEMU_COUNT})" -lt "${QEMU_TIMEOUT}" ]; do
+
+    # Increase the counter
+    echo $(($(cat ${QEMU_COUNT})+1)) > ${QEMU_COUNT}
+
+    # Try to connect to qemu
+    if echo 'info version'| nc -q 1 -w 1 localhost "${QEMU_PORT}" >/dev/null 2>&1 ; then
+
+      sleep 1
+
+      CNT="$(cat ${QEMU_COUNT})/${QEMU_TIMEOUT}"
+      [[ "${DEBUG}" == [Yy1]* ]] && info "Shutting down, waiting... (${CNT})"
+
+    fi
+
+  done
+
+  echo && echo "❯ Quitting..."
+  echo 'quit' | nc -q 1 -w 1 localhost "${QEMU_PORT}" >/dev/null 2>&1 || true
+
+  closeNetwork
+
+  return
+}
+
+_trap _graceful_shutdown SIGTERM SIGHUP SIGINT SIGABRT SIGQUIT
+
+MON_OPTS="-monitor telnet:localhost:${QEMU_PORT},server,nowait,nodelay"