1#!/usr/bin/python
2# SPDX-License-Identifier: (LGPL-2.1 OR BSD-2-Clause)
3# Basic sanity check of perf JSON output as specified in the man page.
4
5import argparse
6import sys
7import json
8
9ap = argparse.ArgumentParser()
10ap.add_argument('--no-args', action='store_true')
11ap.add_argument('--interval', action='store_true')
12ap.add_argument('--system-wide-no-aggr', action='store_true')
13ap.add_argument('--system-wide', action='store_true')
14ap.add_argument('--event', action='store_true')
15ap.add_argument('--per-core', action='store_true')
16ap.add_argument('--per-thread', action='store_true')
17ap.add_argument('--per-die', action='store_true')
18ap.add_argument('--per-node', action='store_true')
19ap.add_argument('--per-socket', action='store_true')
20args = ap.parse_args()
21
22Lines = sys.stdin.readlines()
23
24def isfloat(num):
25  try:
26    float(num)
27    return True
28  except ValueError:
29    return False
30
31
32def isint(num):
33  try:
34    int(num)
35    return True
36  except ValueError:
37    return False
38
39def is_counter_value(num):
40  return isfloat(num) or num == '<not counted>' or num == '<not supported>'
41
42def check_json_output(expected_items):
43  if expected_items != -1:
44    for line in Lines:
45      if 'failed' not in line:
46        count = 0
47        count = line.count(',')
48        if count != expected_items and count >= 1 and count <= 3 and 'metric-value' in line:
49          # Events that generate >1 metric may have isolated metric
50          # values and possibly other prefixes like interval, core and
51          # aggregate-number.
52          continue
53        if count != expected_items:
54          raise RuntimeError(f'wrong number of fields. counted {count} expected {expected_items}'
55                             f' in \'{line}\'')
56  checks = {
57      'aggregate-number': lambda x: isfloat(x),
58      'core': lambda x: True,
59      'counter-value': lambda x: is_counter_value(x),
60      'cgroup': lambda x: True,
61      'cpu': lambda x: isint(x),
62      'die': lambda x: True,
63      'event': lambda x: True,
64      'event-runtime': lambda x: isfloat(x),
65      'interval': lambda x: isfloat(x),
66      'metric-unit': lambda x: True,
67      'metric-value': lambda x: isfloat(x),
68      'node': lambda x: True,
69      'pcnt-running': lambda x: isfloat(x),
70      'socket': lambda x: True,
71      'thread': lambda x: True,
72      'unit': lambda x: True,
73  }
74  input = '[\n' + ','.join(Lines) + '\n]'
75  for item in json.loads(input):
76    for key, value in item.items():
77      if key not in checks:
78        raise RuntimeError(f'Unexpected key: key={key} value={value}')
79      if not checks[key](value):
80        raise RuntimeError(f'Check failed for: key={key} value={value}')
81
82
83try:
84  if args.no_args or args.system_wide or args.event:
85    expected_items = 6
86  elif args.interval or args.per_thread or args.system_wide_no_aggr:
87    expected_items = 7
88  elif args.per_core or args.per_socket or args.per_node or args.per_die:
89    expected_items = 8
90  else:
91    # If no option is specified, don't check the number of items.
92    expected_items = -1
93  check_json_output(expected_items)
94except:
95  print('Test failed for input:\n' + '\n'.join(Lines))
96  raise
97