Rename power.sh to run/power.sh

run/power.sh

@@ -0,0 +1,84 @@
+#!/usr/bin/env bash
+set -eu
+
+# Configure QEMU for graceful shutdown
+
+QEMU_MONPORT=7100
+QEMU_POWERDOWN_TIMEOUT=50
+
+_QEMU_PID=/run/qemu.pid
+_QEMU_SHUTDOWN_COUNTER=/run/qemu.counter
+
+rm -f "${_QEMU_PID}"
+rm -f "${_QEMU_SHUTDOWN_COUNTER}"
+
+# Allows for troubleshooting signals sent to the process
+_trap(){
+    func="$1" ; shift
+    for sig ; do
+        trap "$func $sig" "$sig"
+    done
+}
+
+_graceful_shutdown(){
+
+  local QEMU_MONPORT="${QEMU_MONPORT:-7100}"
+  local QEMU_POWERDOWN_TIMEOUT="${QEMU_POWERDOWN_TIMEOUT:-50}"
+
+  [ -f "${_QEMU_SHUTDOWN_COUNTER}" ] && return
+
+  set +e
+  echo
+  echo "Received $1 signal, shutting down..."
+  echo 0 > "${_QEMU_SHUTDOWN_COUNTER}"
+
+  # Don't send the powerdown signal because vDSM ignores ACPI signals
+  # echo 'system_powerdown' | nc -q 1 -w 1 localhost "${QEMU_MONPORT}">/dev/null
+
+  # Send shutdown command to guest agent tools instead via serial port
+  RESPONSE=$(curl -s -m 2 -S http://127.0.0.1:2210/write?command=6 2>&1)
+
+  if [[ ! "${RESPONSE}" =~ "\"success\"" ]] ; then
+
+    echo
+    echo "Could not send shutdown command to guest, error: $RESPONSE"
+
+    FILE="${IMG}/${BASE}.agent"
+    [ ! -f "$FILE" ] && echo "1" > "$FILE"
+    AGENT_VERSION=$(cat "${FILE}")
+
+    if ((AGENT_VERSION < 2)); then
+      echo
+      echo "Please update the agent to allow gracefull shutdowns..."
+      pkill -f qemu-system-x86_64
+    else
+      # Send a NMI interrupt which will be detected by the kernel
+      echo 'nmi' | nc -q 1 -w 1 localhost "${QEMU_MONPORT}">/dev/null
+    fi
+
+  fi
+
+  while [ "$(cat ${_QEMU_SHUTDOWN_COUNTER})" -lt "${QEMU_POWERDOWN_TIMEOUT}" ]; do
+
+    # Increase the counter
+    echo $(($(cat ${_QEMU_SHUTDOWN_COUNTER})+1)) > ${_QEMU_SHUTDOWN_COUNTER}
+
+    # Try to connect to qemu
+    if echo 'info version'| nc -q 1 -w 1 localhost "${QEMU_MONPORT:-7100}">/dev/null; then
+
+      sleep 1
+      #echo "Shutting down, waiting... ($(cat ${_QEMU_SHUTDOWN_COUNTER})/${QEMU_POWERDOWN_TIMEOUT})"
+
+    fi
+  done
+
+  echo
+  echo "Quitting..."
+  echo 'quit' | nc -q 1 -w 1 localhost "${QEMU_MONPORT:-7100}">/dev/null || true
+
+  return
+}
+
+_trap _graceful_shutdown SIGTERM SIGHUP SIGINT SIGABRT SIGQUIT
+
+KVM_MON_OPTS="-monitor telnet:localhost:${QEMU_MONPORT:-7100},server,nowait,nodelay"