1#!/bin/sh
2# perf stat --bpf-counters --for-each-cgroup test
3# SPDX-License-Identifier: GPL-2.0
4
5set -e
6
7test_cgroups=
8if [ "$1" = "-v" ]; then
9	verbose="1"
10fi
11
12# skip if --bpf-counters --for-each-cgroup is not supported
13check_bpf_counter()
14{
15	if ! perf stat -a --bpf-counters --for-each-cgroup / true > /dev/null 2>&1; then
16		if [ "${verbose}" = "1" ]; then
17			echo "Skipping: --bpf-counters --for-each-cgroup not supported"
18			perf --no-pager stat -a --bpf-counters --for-each-cgroup / true || true
19		fi
20		exit 2
21	fi
22}
23
24# find two cgroups to measure
25find_cgroups()
26{
27	# try usual systemd slices first
28	if [ -d /sys/fs/cgroup/system.slice ] && [ -d /sys/fs/cgroup/user.slice ]; then
29		test_cgroups="system.slice,user.slice"
30		return
31	fi
32
33	# try root and self cgroups
34	find_cgroups_self_cgrp=$(grep perf_event /proc/self/cgroup | cut -d: -f3)
35	if [ -z ${find_cgroups_self_cgrp} ]; then
36		# cgroup v2 doesn't specify perf_event
37		find_cgroups_self_cgrp=$(grep ^0: /proc/self/cgroup | cut -d: -f3)
38	fi
39
40	if [ -z ${find_cgroups_self_cgrp} ]; then
41		test_cgroups="/"
42	else
43		test_cgroups="/,${find_cgroups_self_cgrp}"
44	fi
45}
46
47# As cgroup events are cpu-wide, we cannot simply compare the result.
48# Just check if it runs without failure and has non-zero results.
49check_system_wide_counted()
50{
51	check_system_wide_counted_output=$(perf stat -a --bpf-counters --for-each-cgroup ${test_cgroups} -e cpu-clock -x, sleep 1  2>&1)
52	if echo ${check_system_wide_counted_output} | grep -q -F "<not "; then
53		echo "Some system-wide events are not counted"
54		if [ "${verbose}" = "1" ]; then
55			echo ${check_system_wide_counted_output}
56		fi
57		exit 1
58	fi
59}
60
61check_cpu_list_counted()
62{
63	check_cpu_list_counted_output=$(perf stat -C 0,1 --bpf-counters --for-each-cgroup ${test_cgroups} -e cpu-clock -x, taskset -c 1 sleep 1  2>&1)
64	if echo ${check_cpu_list_counted_output} | grep -q -F "<not "; then
65		echo "Some CPU events are not counted"
66		if [ "${verbose}" = "1" ]; then
67			echo ${check_cpu_list_counted_output}
68		fi
69		exit 1
70	fi
71}
72
73check_bpf_counter
74find_cgroups
75
76check_system_wide_counted
77check_cpu_list_counted
78
79exit 0
80