3c47b65a64
For the tests, we need to run commands in various contexts: in the host, in a guest or in a namespace. Currently we do this by running each context in a tmux pane, and using tmux commands to type the commands into the relevant pane, then screen-scrape the output for the results if we need them. This is very fragile, because we have to make various assumptions to parse the output. Those can break if a shell doesn't have the prompt we expect, if the tmux pane is too small or in various other conditions. This starts some library functions for a new "context" system, that provides a common way to invoke commands in a given context, in a way that properly preserves stdout, stderr and the process return code. Signed-off-by: David Gibson <david@gibson.dropbear.id.au>
80 lines
2.1 KiB
Bash
80 lines
2.1 KiB
Bash
#! /bin/sh
|
|
#
|
|
# SPDX-License-Identifier: AGPL-3.0-or-later
|
|
#
|
|
# PASST - Plug A Simple Socket Transport
|
|
# for qemu/UNIX domain socket mode
|
|
#
|
|
# PASTA - Pack A Subtle Tap Abstraction
|
|
# for network namespace/tap device mode
|
|
#
|
|
# test/lib/context - Run commands in different contexts (host, guest, namespace etc.)
|
|
#
|
|
# Copyright Red Hat
|
|
# Author: David Gibson <david@gibson.dropbear.id.au>
|
|
|
|
# context_setup_host() - Create a new context for running commands on the host
|
|
# $1: Context name
|
|
context_setup_host() {
|
|
__name="$1"
|
|
__prefix="${LOGDIR}/context_${__name}"
|
|
echo sh -c > "${__prefix}.enter"
|
|
echo -n "${__name}$ " > "${__prefix}.log"
|
|
}
|
|
|
|
# context_teardown() - Remove a context (leave log files intact)
|
|
# $1: Context name
|
|
context_teardown() {
|
|
__name="$1"
|
|
__prefix="${LOGDIR}/context_${__name}"
|
|
rm -f "${__prefix}.enter"
|
|
}
|
|
|
|
# context_exists() - Test if a context currently exists
|
|
# $1: Context name
|
|
context_exists() {
|
|
__name="$1"
|
|
__prefix="${LOGDIR}/context_${__name}"
|
|
[ -f "${__prefix}.enter" ]
|
|
}
|
|
|
|
# context_run() - Run a shell command in a context, and wait for it to finish
|
|
# $1: Context name
|
|
# $*: Command to start
|
|
context_run() {
|
|
__name="$1"
|
|
__prefix="${LOGDIR}/context_${__name}"
|
|
__enter="$(cat "${__prefix}.enter")"
|
|
shift
|
|
echo "$*" >> "${__prefix}.log"
|
|
mkfifo "${__prefix}.stdout" "${__prefix}.stderr"
|
|
tee -a "${__prefix}.log" < "${__prefix}.stdout" &
|
|
tee -a "${__prefix}.log" < "${__prefix}.stderr" >&2 &
|
|
${__enter} "$*" >> "${__prefix}.stdout" 2>> "${__prefix}.stderr"
|
|
rc=$?
|
|
rm "${__prefix}.stdout" "${__prefix}.stderr"
|
|
[ ${DEBUG} -eq 1 ] && echo "[Exit code: $rc]" >> "${__prefix}.log"
|
|
echo -n "${__name}$ " >> "${__prefix}.log"
|
|
return $rc
|
|
}
|
|
|
|
# context_run_bg() - Start a shell command in a context
|
|
# $1: Context name
|
|
# $*: Command to start
|
|
context_run_bg() {
|
|
__name="$1"
|
|
__prefix="${LOGDIR}/context_${__name}"
|
|
context_run "$@" &
|
|
echo $! > "${__prefix}.pid"
|
|
}
|
|
|
|
# context_wait() - Wait for background command in a context to complete
|
|
# $1: Context name
|
|
# Returns the status of the completed command
|
|
context_wait() {
|
|
__name="$1"
|
|
__prefix="${LOGDIR}/context_${__name}"
|
|
__pid=$(cat "${__prefix}.pid")
|
|
rm "${__prefix}.pid"
|
|
wait ${__pid}
|
|
}
|