tjh.dev
Linux kernel mirror (for testing)
git.kernel.org/pub/scm/linux/kernel/git/torvalds/linux.git
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 # Simple check: run rtla with given arguments and test exit code.
47 # If TEST_COUNT is set, run the test. Otherwise, just count.
48 ctr=$(($ctr + 1))
49 if [ -n "$TEST_COUNT" ]
50 then
51 # Reset osnoise options before running test.
52 [ "$NO_RESET_OSNOISE" == 1 ] || reset_osnoise
53 # Run rtla; in case of failure, include its output as comment
54 # in the test results.
55 result=$(eval stdbuf -oL $TIMEOUT "$RTLA" $2 2>&1); exitcode=$?
56 # Test if the results matches if requested
57 if [ -n "$expected_output" ]
58 then
59 grep -E "$expected_output" <<< "$result" > /dev/null; grep_result=$?
60 else
61 grep_result=0
62 fi
63
64 if [ $exitcode -eq $expected_exitcode ] && [ $grep_result -eq 0 ]
65 then
66 echo "ok $ctr - $1"
67 else
68 echo "not ok $ctr - $1"
69 # Add rtla output and exit code as comments in case of failure
70 echo "$result" | col -b | while read line; do echo "# $line"; done
71 printf "#\n# exit code %s\n" $exitcode
72 [ -n "$expected_output" ] && [ $grep_result -ne 0 ] && \
73 printf "# Output match failed: \"%s\"\n" "$expected_output"
74 fi
75 fi
76}
77
78check_with_osnoise_options() {
79 # Do the same as "check", but with pre-set osnoise options.
80 # Note: rtla should reset the osnoise options, this is used to test
81 # if it indeed does so.
82 # Save original arguments
83 arg1=$1
84 arg2=$2
85 arg3=$3
86
87 # Apply osnoise options (if not dry run)
88 if [ -n "$TEST_COUNT" ]
89 then
90 [ "$NO_RESET_OSNOISE" == 1 ] || reset_osnoise
91 shift
92 shift
93 while shift
94 do
95 [ "$1" == "" ] && continue
96 option=$(echo $1 | cut -d '=' -f 1)
97 value=$(echo $1 | cut -d '=' -f 2)
98 echo "option: $option, value: $value"
99 echo "$value" > "/sys/kernel/tracing/osnoise/$option" || return 1
100 done
101 fi
102
103 NO_RESET_OSNOISE=1 check "$arg1" "$arg2" "$arg3"
104}
105
106set_timeout() {
107 TIMEOUT="timeout -v -k 15s $1"
108}
109
110unset_timeout() {
111 unset TIMEOUT
112}
113
114set_no_reset_osnoise() {
115 NO_RESET_OSNOISE=1
116}
117
118unset_no_reset_osnoise() {
119 unset NO_RESET_OSNOISE
120}
121
122test_end() {
123 # If running without TEST_COUNT, tests are not actually run, just
124 # counted. In that case, re-run the test with the correct count.
125 [ -z "$TEST_COUNT" ] && TEST_COUNT=$ctr exec bash $0 || true
126}
127
128# Avoid any environmental discrepancies
129export LC_ALL=C
130unset_timeout