1 #!/bin/env python3 1 #!/bin/env python3
2 # SPDX-License-Identifier: GPL-2.0 2 # SPDX-License-Identifier: GPL-2.0
3 3
4 import subprocess 4 import subprocess
5 from shutil import which 5 from shutil import which
6 6
7 turbostat = which('turbostat') 7 turbostat = which('turbostat')
8 if turbostat is None: 8 if turbostat is None:
9 print('Could not find turbostat binary 9 print('Could not find turbostat binary')
10 exit(1) 10 exit(1)
11 11
12 timeout = which('timeout') 12 timeout = which('timeout')
13 if timeout is None: 13 if timeout is None:
14 print('Could not find timeout binary') 14 print('Could not find timeout binary')
15 exit(1) 15 exit(1)
16 16
17 proc_turbostat = subprocess.run([turbostat, '- 17 proc_turbostat = subprocess.run([turbostat, '--list'], capture_output = True)
18 if proc_turbostat.returncode != 0: 18 if proc_turbostat.returncode != 0:
19 print(f'turbostat failed with {proc_tu 19 print(f'turbostat failed with {proc_turbostat.returncode}')
20 exit(1) 20 exit(1)
21 21
22 # 22 #
23 # By default --list reports also "usec" and "T 23 # By default --list reports also "usec" and "Time_Of_Day_Seconds" columns
24 # which are only visible when running with --d 24 # which are only visible when running with --debug.
25 # 25 #
26 expected_columns_debug = proc_turbostat.stdout 26 expected_columns_debug = proc_turbostat.stdout.replace(b',', b'\t').strip()
27 expected_columns = expected_columns_debug.repl 27 expected_columns = expected_columns_debug.replace(b'usec\t', b'').replace(b'Time_Of_Day_Seconds\t', b'').replace(b'X2APIC\t', b'').replace(b'APIC\t', b'')
28 28
29 # 29 #
30 # Run turbostat with no options for 10 seconds 30 # Run turbostat with no options for 10 seconds and send SIGINT
31 # 31 #
32 timeout_argv = [timeout, '--preserve-status', 32 timeout_argv = [timeout, '--preserve-status', '-s', 'SIGINT', '-k', '3', '1s']
33 turbostat_argv = [turbostat, '-i', '0.250'] 33 turbostat_argv = [turbostat, '-i', '0.250']
34 34
35 print(f'Running turbostat with {turbostat_argv 35 print(f'Running turbostat with {turbostat_argv=}... ', end = '', flush = True)
36 proc_turbostat = subprocess.run(timeout_argv + 36 proc_turbostat = subprocess.run(timeout_argv + turbostat_argv, capture_output = True)
37 if proc_turbostat.returncode != 0: 37 if proc_turbostat.returncode != 0:
38 print(f'turbostat failed with {proc_tu 38 print(f'turbostat failed with {proc_turbostat.returncode}')
39 exit(1) 39 exit(1)
40 actual_columns = proc_turbostat.stdout.split(b 40 actual_columns = proc_turbostat.stdout.split(b'\n')[0]
41 if expected_columns != actual_columns: 41 if expected_columns != actual_columns:
42 print(f'turbostat column check failed\ 42 print(f'turbostat column check failed\n{expected_columns=}\n{actual_columns=}')
43 exit(1) 43 exit(1)
44 print('OK') 44 print('OK')
45 45
46 # 46 #
47 # Same, but with --debug 47 # Same, but with --debug
48 # 48 #
49 turbostat_argv.append('--debug') 49 turbostat_argv.append('--debug')
50 50
51 print(f'Running turbostat with {turbostat_argv 51 print(f'Running turbostat with {turbostat_argv=}... ', end = '', flush = True)
52 proc_turbostat = subprocess.run(timeout_argv + 52 proc_turbostat = subprocess.run(timeout_argv + turbostat_argv, capture_output = True)
53 if proc_turbostat.returncode != 0: 53 if proc_turbostat.returncode != 0:
54 print(f'turbostat failed with {proc_tu 54 print(f'turbostat failed with {proc_turbostat.returncode}')
55 exit(1) 55 exit(1)
56 actual_columns = proc_turbostat.stdout.split(b 56 actual_columns = proc_turbostat.stdout.split(b'\n')[0]
57 if expected_columns_debug != actual_columns: 57 if expected_columns_debug != actual_columns:
58 print(f'turbostat column check failed\ 58 print(f'turbostat column check failed\n{expected_columns_debug=}\n{actual_columns=}')
59 exit(1) 59 exit(1)
60 print('OK') 60 print('OK')
Linux® is a registered trademark of Linus Torvalds in the United States and other countries.
TOMOYO® is a registered trademark of NTT DATA CORPORATION.