53 lines
1.1 KiB
Bash
53 lines
1.1 KiB
Bash
qemuDir=$(cd "${BASH_SOURCE[0]%/*}" && pwd)
|
|
|
|
# shellcheck disable=SC1091
|
|
source "$qemuDir/wait-until.sh"
|
|
|
|
tmpDir=/tmp/nix-bitcoin-qemu-vm
|
|
mkdir -p "$tmpDir"
|
|
|
|
# Cleanup on exit
|
|
cleanup() {
|
|
set +eu
|
|
if [[ $qemuPID ]]; then
|
|
kill -9 "$qemuPID"
|
|
fi
|
|
rm -rf "$tmpDir"
|
|
}
|
|
trap "cleanup" EXIT
|
|
|
|
identityFile=$qemuDir/id-vm
|
|
chmod 0600 "$identityFile"
|
|
|
|
runVM() {
|
|
vm=$1
|
|
vmNumCPUs=$2
|
|
vmMemoryMiB=$3
|
|
sshPort=$4
|
|
|
|
export NIX_DISK_IMAGE="$tmpDir/img"
|
|
export QEMU_NET_OPTS="hostfwd=tcp::${sshPort}-:22"
|
|
# shellcheck disable=SC2211
|
|
</dev/null "$vm"/bin/run-*-vm -m "$vmMemoryMiB" -smp "$vmNumCPUs" &>/dev/null &
|
|
qemuPID=$!
|
|
}
|
|
|
|
vmWaitForSSH() {
|
|
echo
|
|
printf "Waiting for SSH connection..."
|
|
waitUntil "c : 2>/dev/null" 500
|
|
echo
|
|
}
|
|
|
|
# Run command in VM
|
|
c() {
|
|
ssh -p "$sshPort" -i "$identityFile" -o ConnectTimeout=1 \
|
|
-o StrictHostKeyChecking=no -o UserKnownHostsFile=/dev/null -o LogLevel=ERROR \
|
|
-o ControlMaster=auto -o ControlPath=$tmpDir/ssh-connection -o ControlPersist=60 \
|
|
root@127.0.0.1 "$@"
|
|
}
|
|
export identityFile
|
|
export sshPort
|
|
export tmpDir
|
|
export -f c
|