1#!/bin/sh
2# SPDX-License-Identifier: GPL-2.0
3
4tenths=date\ +%s%1N
5
6# Wait for PID $1 to have $2 number of threads started
7# Time out after $3 tenths of a second or 5 seconds if $3 is ""
8wait_for_threads()
9{
10	tm_out=$3 ; [ -n "${tm_out}" ] || tm_out=50
11	start_time=$($tenths)
12	while [ -e "/proc/$1/task" ] ; do
13		th_cnt=$(find "/proc/$1/task" -mindepth 1 -maxdepth 1 -printf x | wc -c)
14		if [ "${th_cnt}" -ge "$2" ] ; then
15			return 0
16		fi
17		# Wait at most tm_out tenths of a second
18		if [ $(($($tenths) - start_time)) -ge $tm_out ] ; then
19			echo "PID $1 does not have $2 threads"
20			return 1
21		fi
22	done
23	return 1
24}
25
26# Wait for perf record -vvv 2>$2 with PID $1 to start by looking at file $2
27# It depends on capturing perf record debug message "perf record has started"
28# Time out after $3 tenths of a second or 5 seconds if $3 is ""
29wait_for_perf_to_start()
30{
31	tm_out=$3 ; [ -n "${tm_out}" ] || tm_out=50
32	echo "Waiting for \"perf record has started\" message"
33	start_time=$($tenths)
34	while [ -e "/proc/$1" ] ; do
35		if grep -q "perf record has started" "$2" ; then
36			echo OK
37			break
38		fi
39		# Wait at most tm_out tenths of a second
40		if [ $(($($tenths) - start_time)) -ge $tm_out ] ; then
41			echo "perf recording did not start"
42			return 1
43		fi
44	done
45	return 0
46}
47
48# Wait for process PID %1 to exit
49# Time out after $2 tenths of a second or 5 seconds if $2 is ""
50wait_for_process_to_exit()
51{
52	tm_out=$2 ; [ -n "${tm_out}" ] || tm_out=50
53	start_time=$($tenths)
54	while [ -e "/proc/$1" ] ; do
55		# Wait at most tm_out tenths of a second
56		if [ $(($($tenths) - start_time)) -ge $tm_out ] ; then
57			echo "PID $1 did not exit as expected"
58			return 1
59		fi
60	done
61	return 0
62}
63
64# Check if PID $1 is still running after $2 tenths of a second
65# or 0.3 seconds if $2 is ""
66is_running()
67{
68	tm_out=$2 ; [ -n "${tm_out}" ] || tm_out=3
69	start_time=$($tenths)
70	while [ -e "/proc/$1" ] ; do
71		# Check for at least tm_out tenths of a second
72		if [ $(($($tenths) - start_time)) -gt $tm_out ] ; then
73			return 0
74		fi
75	done
76	echo "PID $1 exited prematurely"
77	return 1
78}
79