1#!/usr/bin/env python3 2# SPDX-License-Identifier: GPL-2.0-only 3# 4# Copyright (C) 2019-2022 Red Hat, Inc. Daniel Bristot de Oliveira <bristot@kernel.org> 5# 6# dot2k: transform dot files into a monitor for the Linux kernel. 7# 8# For further information, see: 9# Documentation/trace/rv/monitor_synthesis.rst 10 11if __name__ == '__main__': 12 from rvgen.dot2k import da2k, ha2k 13 from rvgen.generator import Monitor 14 from rvgen.container import Container 15 from rvgen.ltl2k import ltl2k 16 from rvgen.kunit import KUnit, KUnitError 17 from rvgen.automata import AutomataError 18 from rvgen.ltl2ba import LTLError 19 import argparse 20 import sys 21 22 parser = argparse.ArgumentParser(description='Generate kernel rv monitor') 23 24 parent_parser = argparse.ArgumentParser(add_help=False) 25 parent_parser.add_argument("-D", "--description", dest="description", required=False) 26 parent_parser.add_argument("-a", "--auto_patch", dest="auto_patch", 27 action="store_true", required=False, 28 help="Patch the kernel in place") 29 30 subparsers = parser.add_subparsers(dest="subcmd", required=True) 31 32 monitor_parser = subparsers.add_parser("monitor", parents=[parent_parser]) 33 monitor_parser.add_argument('-n', "--model_name", dest="model_name") 34 monitor_parser.add_argument("-p", "--parent", dest="parent", 35 required=False, help="Create a monitor nested to parent") 36 monitor_parser.add_argument('-c', "--class", dest="monitor_class", required=True, 37 help="Monitor class, either \"da\", \"ha\" or \"ltl\"") 38 monitor_parser.add_argument('-s', "--spec", dest="spec", required=True, 39 help="Monitor specification file") 40 monitor_parser.add_argument('-t', "--monitor_type", dest="monitor_type", required=True, 41 help=f"Available options: {', '.join(Monitor.monitor_types.keys())}") 42 43 container_parser = subparsers.add_parser("container", parents=[parent_parser]) 44 container_parser.add_argument('-n', "--model_name", dest="model_name", required=True) 45 46 kunit_parser = subparsers.add_parser("kunit", parents=[parent_parser]) 47 kunit_parser.add_argument('-n', "--model_name", dest="model_name", required=True) 48 kunit_parser.add_argument('-l', "--local", dest="local", action="store_true", required=False, 49 help="Force looking for the monitor in the current directory only") 50 51 params = parser.parse_args() 52 53 try: 54 if params.subcmd == "monitor": 55 print(f"Opening and parsing the specification file {params.spec}") 56 if params.monitor_class == "da": 57 monitor = da2k(params.spec, params.monitor_type, vars(params)) 58 elif params.monitor_class == "ha": 59 monitor = ha2k(params.spec, params.monitor_type, vars(params)) 60 elif params.monitor_class == "ltl": 61 monitor = ltl2k(params.spec, params.monitor_type, vars(params)) 62 else: 63 print("Unknown monitor class:", params.monitor_class) 64 sys.exit(1) 65 elif params.subcmd == "container": 66 monitor = Container(vars(params)) 67 elif params.subcmd == "kunit": 68 monitor = KUnit(vars(params)) 69 monitor.print_files() 70 sys.exit(0) 71 except (AutomataError, LTLError) as e: 72 print(f"There was an error processing {params.spec}:\n{e}", file=sys.stderr) 73 sys.exit(1) 74 except KUnitError as e: 75 print(f"There was an error generating KUnit files:\n{e}", file=sys.stderr) 76 sys.exit(1) 77 78 print(f"Writing the monitor into the directory {monitor.name}") 79 monitor.print_files() 80 print("Almost done, checklist") 81 if params.subcmd == "monitor": 82 print(f" - Edit the {monitor.name}/{monitor.name}.c to add the instrumentation") 83 print(monitor.fill_tracepoint_tooltip()) 84 print(monitor.fill_makefile_tooltip()) 85 print(monitor.fill_kconfig_tooltip()) 86 print(monitor.fill_monitor_tooltip()) 87