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-cache', action='store_true')
 18ap.add_argument('--per-cluster', action='store_true')
 19ap.add_argument('--per-die', action='store_true')
 20ap.add_argument('--per-node', action='store_true')
 21ap.add_argument('--per-socket', action='store_true')
 22ap.add_argument('--file', type=argparse.FileType('r'), default=sys.stdin)
 23args = ap.parse_args()
 24
 25Lines = args.file.readlines()
 26
 27def isfloat(num):
 28  try:
 29    float(num)
 30    return True
 31  except ValueError:
 32    return False
 33
 34
 35def isint(num):
 36  try:
 37    int(num)
 38    return True
 39  except ValueError:
 40    return False
 41
 42def is_counter_value(num):
 43  return isfloat(num) or num == '<not counted>' or num == '<not supported>'
 44
 45def check_json_output(expected_items):
 46  checks = {
 47      'aggregate-number': lambda x: isfloat(x),
 48      'core': lambda x: True,
 49      'counter-value': lambda x: is_counter_value(x),
 50      'cgroup': lambda x: True,
 51      'cpu': lambda x: isint(x),
 52      'cache': lambda x: True,
 53      'cluster': lambda x: True,
 54      'die': lambda x: True,
 55      'event': lambda x: True,
 56      'event-runtime': lambda x: isfloat(x),
 57      'interval': lambda x: isfloat(x),
 58      'metric-unit': lambda x: True,
 59      'metric-value': lambda x: isfloat(x),
 60      'metric-threshold': lambda x: x in ['unknown', 'good', 'less good', 'nearly bad', 'bad'],
 61      'metricgroup': lambda x: True,
 62      'node': lambda x: True,
 63      'pcnt-running': lambda x: isfloat(x),
 64      'socket': lambda x: True,
 65      'thread': lambda x: True,
 66      'unit': lambda x: True,
 67  }
 68  input = '[\n' + ','.join(Lines) + '\n]'
 69  for item in json.loads(input):
 70    if expected_items != -1:
 71      count = len(item)
 72      if count not in expected_items and count >= 1 and count <= 7 and 'metric-value' in item:
 73        # Events that generate >1 metric may have isolated metric
 74        # values and possibly other prefixes like interval, core,
 75        # aggregate-number, or event-runtime/pcnt-running from multiplexing.
 76        pass
 77      elif count not in expected_items and count >= 1 and count <= 5 and 'metricgroup' in item:
 78        pass
 79      elif count - 1 in expected_items and 'metric-threshold' in item:
 80          pass
 81      elif count not in expected_items:
 82        raise RuntimeError(f'wrong number of fields. counted {count} expected {expected_items}'
 83                           f' in \'{item}\'')
 84    for key, value in item.items():
 85      if key not in checks:
 86        raise RuntimeError(f'Unexpected key: key={key} value={value}')
 87      if not checks[key](value):
 88        raise RuntimeError(f'Check failed for: key={key} value={value}')
 89
 90
 91try:
 92  if args.no_args or args.system_wide or args.event:
 93    expected_items = [5, 7]
 94  elif args.interval or args.per_thread or args.system_wide_no_aggr:
 95    expected_items = [6, 8]
 96  elif args.per_core or args.per_socket or args.per_node or args.per_die or args.per_cluster or args.per_cache:
 97    expected_items = [7, 9]
 98  else:
 99    # If no option is specified, don't check the number of items.
100    expected_items = -1
101  check_json_output(expected_items)
102except:
103  print('Test failed for input:\n' + '\n'.join(Lines))
104  raise