1#!/bin/sh
2# Zstd perf.data compression/decompression
3
4# SPDX-License-Identifier: GPL-2.0
5
6trace_file=$(mktemp /tmp/perf.data.XXX)
7perf_tool=perf
8
9skip_if_no_z_record() {
10	$perf_tool record -h 2>&1 | grep -q '\-z, \-\-compression\-level'
11}
12
13collect_z_record() {
14	echo "Collecting compressed record file:"
15	$perf_tool record -o $trace_file -g -z -F 5000 -- \
16		dd count=500 if=/dev/urandom of=/dev/null
17}
18
19check_compressed_stats() {
20	echo "Checking compressed events stats:"
21	$perf_tool report -i $trace_file --header --stats | \
22		grep -E "(# compressed : Zstd,)|(COMPRESSED events:)"
23}
24
25check_compressed_output() {
26	$perf_tool inject -i $trace_file -o $trace_file.decomp &&
27	$perf_tool report -i $trace_file --stdio | head -n -3 > $trace_file.comp.output &&
28	$perf_tool report -i $trace_file.decomp --stdio | head -n -3 > $trace_file.decomp.output &&
29	diff $trace_file.comp.output $trace_file.decomp.output
30}
31
32skip_if_no_z_record || exit 2
33collect_z_record && check_compressed_stats && check_compressed_output
34err=$?
35rm -f $trace_file*
36exit $err
37