mirror of
https://github.com/AuxXxilium/arv-v-dsm.git
synced 2024-11-23 23:10:58 +07:00
85 lines
2.2 KiB
Bash
85 lines
2.2 KiB
Bash
#!/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"
|