Linux kernel mirror (for testing)
git.kernel.org/pub/scm/linux/kernel/git/torvalds/linux.git
tjh.dev
kernel
os
linux
1#!/bin/bash
2# SPDX-License-Identifier: GPL-2.0
3test_begin() {
4 # Count tests to allow the test harness to double-check if all were
5 # included correctly.
6 ctr=0
7 [ -z "$RTLA" ] && RTLA="./rtla"
8 [ -n "$TEST_COUNT" ] && echo "1..$TEST_COUNT"
9}
10
11reset_osnoise() {
12 # Reset osnoise options to default and remove any dangling instances created
13 # by improperly exited rtla runs.
14 pushd /sys/kernel/tracing >/dev/null || return 1
15
16 # Remove dangling instances created by previous rtla run
17 echo 0 > tracing_thresh
18 cd instances
19 for i in osnoise_top osnoise_hist timerlat_top timerlat_hist
20 do
21 [ ! -d "$i" ] && continue
22 rmdir "$i"
23 done
24
25 # Reset options to default
26 # Note: those are copied from the default values of osnoise_data
27 # in kernel/trace/trace_osnoise.c
28 cd ../osnoise
29 echo all > cpus
30 echo DEFAULTS > options
31 echo 1000000 > period_us
32 echo 0 > print_stack
33 echo 1000000 > runtime_us
34 echo 0 > stop_tracing_total_us
35 echo 0 > stop_tracing_us
36 echo 1000 > timerlat_period_us
37
38 popd >/dev/null
39}
40
41check() {
42 test_name=$0
43 tested_command=$1
44 expected_exitcode=${3:-0}
45 expected_output=$4
46 unexpected_output=$5
47 # Simple check: run rtla with given arguments and test exit code.
48 # If TEST_COUNT is set, run the test. Otherwise, just count.
49 ctr=$(($ctr + 1))
50 if [ -n "$TEST_COUNT" ]
51 then
52 # Reset osnoise options before running test.
53 [ "$NO_RESET_OSNOISE" == 1 ] || reset_osnoise
54 # Run rtla; in case of failure, include its output as comment
55 # in the test results.
56 result=$(eval stdbuf -oL $TIMEOUT "$RTLA" $2 2>&1); exitcode=$?
57 failbuf=''
58 fail=0
59
60 # Test if the results matches if requested
61 if [ -n "$expected_output" ] && ! grep -qE "$expected_output" <<< "$result"
62 then
63 fail=1
64 failbuf+=$(printf "# Output match failed: \"%s\"" "$expected_output")
65 failbuf+=$'\n'
66 fi
67
68 if [ -n "$unexpected_output" ] && grep -qE "$unexpected_output" <<< "$result"
69 then
70 fail=1
71 failbuf+=$(printf "# Output non-match failed: \"%s\"" "$unexpected_output")
72 failbuf+=$'\n'
73 fi
74
75 if [ $exitcode -eq $expected_exitcode ] && [ $fail -eq 0 ]
76 then
77 echo "ok $ctr - $1"
78 else
79 # Add rtla output and exit code as comments in case of failure
80 echo "not ok $ctr - $1"
81 echo -n "$failbuf"
82 echo "$result" | col -b | while read line; do echo "# $line"; done
83 printf "#\n# exit code %s\n" $exitcode
84 fi
85 fi
86}
87
88check_with_osnoise_options() {
89 # Do the same as "check", but with pre-set osnoise options.
90 # Note: rtla should reset the osnoise options, this is used to test
91 # if it indeed does so.
92 # Save original arguments
93 arg1=$1
94 arg2=$2
95 arg3=$3
96
97 # Apply osnoise options (if not dry run)
98 if [ -n "$TEST_COUNT" ]
99 then
100 [ "$NO_RESET_OSNOISE" == 1 ] || reset_osnoise
101 shift
102 shift
103 while shift
104 do
105 [ "$1" == "" ] && continue
106 option=$(echo $1 | cut -d '=' -f 1)
107 value=$(echo $1 | cut -d '=' -f 2)
108 echo "option: $option, value: $value"
109 echo "$value" > "/sys/kernel/tracing/osnoise/$option" || return 1
110 done
111 fi
112
113 NO_RESET_OSNOISE=1 check "$arg1" "$arg2" "$arg3"
114}
115
116set_timeout() {
117 TIMEOUT="timeout -v -k 15s $1"
118}
119
120unset_timeout() {
121 unset TIMEOUT
122}
123
124set_no_reset_osnoise() {
125 NO_RESET_OSNOISE=1
126}
127
128unset_no_reset_osnoise() {
129 unset NO_RESET_OSNOISE
130}
131
132test_end() {
133 # If running without TEST_COUNT, tests are not actually run, just
134 # counted. In that case, re-run the test with the correct count.
135 [ -z "$TEST_COUNT" ] && TEST_COUNT=$ctr exec bash $0 || true
136}
137
138# Avoid any environmental discrepancies
139export LC_ALL=C
140unset_timeout