Files
cubelinux-kernel/tools/testing/selftests/net/lib/sh/defer.sh
T
Greg Kroah-Hartman 598cf27219 Linux 6.19.3
Link: https://lore.kernel.org/r/20260217200002.683975158@linuxfoundation.org
Tested-by: Florian Fainelli <florian.fainelli@broadcom.com>
Tested-by: Takeshi Ogasawara <takeshi.ogasawara@futuring-girl.com>
Tested-by: Peter Schneider <pschneider1968@googlemail.com>
Tested-by: Jon Hunter <jonathanh@nvidia.com>
Tested-by: Salvatore Bonaccorso <carnil@debian.org>
Tested-by: Brett A C Sheffield <bacs@librecast.net>
Tested-by: Mark Brown <broonie@kernel.org>
Tested-by: Luna Jernberg <droidbittin@gmail.com>
Tested-by: Ronald Warsow <rwarsow@gmx.de>
Tested-by: Justin M. Forbes <jforbes@fedoraproject.org>
Tested-by: Ron Economos <re@w6rz.net>
Tested-by: Miguel Ojeda <ojeda@kernel.org>
Signed-off-by: Greg Kroah-Hartman <gregkh@linuxfoundation.org>
2026-02-19 16:33:27 +01:00

132 lines
2.2 KiB
Bash

#!/bin/bash
# SPDX-License-Identifier: GPL-2.0
# Whether to pause and allow debugging when an executed deferred command has a
# non-zero exit code.
: "${DEFER_PAUSE_ON_FAIL:=no}"
# map[(scope_id,track,cleanup_id) -> cleanup_command]
# track={d=default | p=priority}
declare -A __DEFER__JOBS
# map[(scope_id,track) -> # cleanup_commands]
declare -A __DEFER__NJOBS
# scope_id of the topmost scope.
__DEFER__SCOPE_ID=0
__defer__ndefer_key()
{
local track=$1; shift
echo $__DEFER__SCOPE_ID,$track
}
__defer__defer_key()
{
local track=$1; shift
local defer_ix=$1; shift
echo $__DEFER__SCOPE_ID,$track,$defer_ix
}
__defer__ndefers()
{
local track=$1; shift
echo ${__DEFER__NJOBS[$(__defer__ndefer_key $track)]}
}
__defer__run()
{
local track=$1; shift
local defer_ix=$1; shift
local defer_key=$(__defer__defer_key $track $defer_ix)
local ret
eval ${__DEFER__JOBS[$defer_key]}
ret=$?
if [[ "$DEFER_PAUSE_ON_FAIL" == yes && "$ret" -ne 0 ]]; then
echo "Deferred command (track $track index $defer_ix):"
echo " ${__DEFER__JOBS[$defer_key]}"
echo "... ended with an exit status of $ret"
echo "Hit enter to continue, 'q' to quit"
read a
[[ "$a" == q ]] && exit 1
fi
unset __DEFER__JOBS[$defer_key]
}
__defer__schedule()
{
local track=$1; shift
local ndefers=$(__defer__ndefers $track)
local ndefers_key=$(__defer__ndefer_key $track)
local defer_key=$(__defer__defer_key $track $ndefers)
local defer="${@@Q}"
__DEFER__JOBS[$defer_key]="$defer"
__DEFER__NJOBS[$ndefers_key]=$((ndefers + 1))
}
__defer__scope_wipe()
{
__DEFER__NJOBS[$(__defer__ndefer_key d)]=0
__DEFER__NJOBS[$(__defer__ndefer_key p)]=0
}
defer_scope_push()
{
((__DEFER__SCOPE_ID++))
__defer__scope_wipe
}
defer_scope_pop()
{
local defer_ix
for ((defer_ix=$(__defer__ndefers p); defer_ix-->0; )); do
__defer__run p $defer_ix
done
for ((defer_ix=$(__defer__ndefers d); defer_ix-->0; )); do
__defer__run d $defer_ix
done
__defer__scope_wipe
((__DEFER__SCOPE_ID--))
}
defer()
{
__defer__schedule d "$@"
}
defer_prio()
{
__defer__schedule p "$@"
}
defer_scopes_cleanup()
{
while ((__DEFER__SCOPE_ID >= 0)); do
defer_scope_pop
done
}
in_defer_scope()
{
local ret
defer_scope_push
"$@"
ret=$?
defer_scope_pop
return $ret
}
__defer__scope_wipe