test: Move context temporary files to state dir

Currently the context command dispatch subsystem creates a bunch of
temporary files in $LOGDIR, which is messy.  Store them in $STATEDIR which
is for precisely this purpose.  The logs from each context still go into
$LOGDIR.

Signed-off-by: David Gibson <david@gibson.dropbear.id.au>
This commit is contained in:
David Gibson 2022-09-13 14:35:21 +10:00 committed by Stefano Brivio
parent e4ecb6d795
commit 54c5addeaa

View file

@ -17,17 +17,17 @@
# $1: Context name # $1: Context name
context_setup_common() { context_setup_common() {
__name="$1" __name="$1"
__prefix="${LOGDIR}/context_${__name}" __log="${LOGDIR}/context_${__name}.log"
echo -n "${__name}$ " > "${__prefix}.log" echo -n "${__name}$ " > "${__log}"
} }
# context_setup_host() - Create a new context for running commands on the host # context_setup_host() - Create a new context for running commands on the host
# $1: Context name # $1: Context name
context_setup_host() { context_setup_host() {
__name="$1" __name="$1"
__prefix="${LOGDIR}/context_${__name}" __enter="${STATESETUP}/context_${__name}.enter"
context_setup_common "${__name}" context_setup_common "${__name}"
echo sh -c > "${__prefix}.enter" echo sh -c > "${__enter}"
} }
# context_setup_nsenter() - Create a new context for running commands in a namespace # context_setup_nsenter() - Create a new context for running commands in a namespace
@ -36,9 +36,9 @@ context_setup_host() {
context_setup_nsenter() { context_setup_nsenter() {
__name="$1" __name="$1"
shift shift
__prefix="${LOGDIR}/context_${__name}" __enter="${STATESETUP}/context_${__name}.enter"
context_setup_common "${__name}" context_setup_common "${__name}"
echo "nsenter $@ sh -c" > "${__prefix}.enter" echo "nsenter $@ sh -c" > "${__enter}"
} }
# context_setup_guest() - Create a new context for running commands in a guest # context_setup_guest() - Create a new context for running commands in a guest
@ -47,22 +47,23 @@ context_setup_nsenter() {
context_setup_guest() { context_setup_guest() {
__name="$1" __name="$1"
__cid="$2" __cid="$2"
__prefix="${LOGDIR}/context_${__name}" __enter="${STATESETUP}/context_${__name}.enter"
__ssh="${STATESETUP}/context_${__name}.ssh"
context_setup_common "${__name}" context_setup_common "${__name}"
cat > "${__prefix}.ssh" <<EOF cat > "${__ssh}" <<EOF
Host ${__name} Host ${__name}
User root User root
UserKnownHostsFile ${__prefix}.hosts UserKnownHostsFile ${STATESETUP}/context_${__name}.hosts
StrictHostKeyChecking no StrictHostKeyChecking no
IdentityFile ${BASEPATH}/guest-key IdentityFile ${BASEPATH}/guest-key
IdentityAgent none IdentityAgent none
ProxyCommand socat - VSOCK-CONNECT:${__cid}:22 ProxyCommand socat - VSOCK-CONNECT:${__cid}:22
EOF EOF
echo "ssh -F ${__prefix}.ssh ${__name}" > "${__prefix}.enter" echo "ssh -F ${__ssh} ${__name}" > "${__enter}"
# Wait for the guest to be booted and accepting connections # Wait for the guest to be booted and accepting connections
while ! ssh -F "${__prefix}.ssh" "${__name}" :; do while ! ssh -F "${__ssh}" "${__name}" :; do
sleep 0.1 sleep 0.1
done done
} }
@ -71,7 +72,7 @@ EOF
# $1: Context name # $1: Context name
context_teardown() { context_teardown() {
__name="$1" __name="$1"
__prefix="${LOGDIR}/context_${__name}" __prefix="${STATESETUP}/context_${__name}"
rm -f "${__prefix}.enter" "${__prefix}.ssh" "${__prefix}.hosts" rm -f "${__prefix}.enter" "${__prefix}.ssh" "${__prefix}.hosts"
} }
@ -79,8 +80,8 @@ context_teardown() {
# $1: Context name # $1: Context name
context_exists() { context_exists() {
__name="$1" __name="$1"
__prefix="${LOGDIR}/context_${__name}" __enter="${STATESETUP}/context_${__name}.enter"
[ -f "${__prefix}.enter" ] [ -f "${__enter}" ]
} }
# context_run() - Run a shell command in a context, and wait for it to finish # context_run() - Run a shell command in a context, and wait for it to finish
@ -88,18 +89,20 @@ context_exists() {
# $*: Command to start # $*: Command to start
context_run() { context_run() {
__name="$1" __name="$1"
__prefix="${LOGDIR}/context_${__name}" __log="${LOGDIR}/context_${__name}.log"
__enter="$(cat "${__prefix}.enter")" __enter="${STATESETUP}/context_${__name}.enter"
__stdout="${STATESETUP}/context_${__name}.stdout"
__stderr="${STATESETUP}/context_${__name}.stderr"
shift shift
echo "$*" >> "${__prefix}.log" echo "$*" >> "${__log}"
mkfifo "${__prefix}.stdout" "${__prefix}.stderr" mkfifo "${__stdout}" "${__stderr}"
tee -a "${__prefix}.log" < "${__prefix}.stdout" & tee -a "${__log}" < "${__stdout}" &
tee -a "${__prefix}.log" < "${__prefix}.stderr" >&2 & tee -a "${__log}" < "${__stderr}" >&2 &
${__enter} "$*" >> "${__prefix}.stdout" 2>> "${__prefix}.stderr" $(cat ${__enter}) "$*" >> "${__stdout}" 2>> "${__stderr}"
rc=$? rc=$?
rm "${__prefix}.stdout" "${__prefix}.stderr" rm "${__stdout}" "${__stderr}"
[ ${DEBUG} -eq 1 ] && echo "[Exit code: $rc]" >> "${__prefix}.log" [ ${DEBUG} -eq 1 ] && echo "[Exit code: $rc]" >> "${__log}"
echo -n "${__name}$ " >> "${__prefix}.log" echo -n "${__name}$ " >> "${__log}"
return $rc return $rc
} }
@ -108,9 +111,9 @@ context_run() {
# $*: Command to start # $*: Command to start
context_run_bg() { context_run_bg() {
__name="$1" __name="$1"
__prefix="${LOGDIR}/context_${__name}" __pidfile="${STATESETUP}/context_${__name}.pid"
context_run "$@" & context_run "$@" &
echo $! > "${__prefix}.pid" echo $! > "${__pidfile}"
} }
# context_wait() - Wait for background command in a context to complete # context_wait() - Wait for background command in a context to complete
@ -118,8 +121,8 @@ context_run_bg() {
# Returns the status of the completed command # Returns the status of the completed command
context_wait() { context_wait() {
__name="$1" __name="$1"
__prefix="${LOGDIR}/context_${__name}" __pidfile="${STATESETUP}/context_${__name}.pid"
__pid=$(cat "${__prefix}.pid") __pid=$(cat "${__pidfile}")
rm "${__prefix}.pid" rm "${__pidfile}"
wait ${__pid} wait ${__pid}
} }