| #!/usr/bin/env python |
| # |
| # tplist Display kernel tracepoints or USDT probes and their formats. |
| # |
| # USAGE: tplist [-p PID] [-l LIB] [-v] [filter] |
| # |
| # Licensed under the Apache License, Version 2.0 (the "License") |
| # Copyright (C) 2016 Sasha Goldshtein. |
| |
| import argparse |
| import fnmatch |
| import os |
| import re |
| import sys |
| |
| from bcc import USDT |
| |
| trace_root = "/sys/kernel/debug/tracing" |
| event_root = os.path.join(trace_root, "events") |
| |
| parser = argparse.ArgumentParser(description= |
| "Display kernel tracepoints or USDT probes and their formats.", |
| formatter_class=argparse.RawDescriptionHelpFormatter) |
| parser.add_argument("-p", "--pid", type=int, default=None, help= |
| "List USDT probes in the specified process") |
| parser.add_argument("-l", "--lib", default="", help= |
| "List USDT probes in the specified library or executable") |
| parser.add_argument("-v", dest="variables", action="store_true", help= |
| "Print the format (available variables)") |
| parser.add_argument(dest="filter", nargs="?", help= |
| "A filter that specifies which probes/tracepoints to print") |
| args = parser.parse_args() |
| |
| def print_tpoint_format(category, event): |
| fmt = open(os.path.join(event_root, category, event, "format") |
| ).readlines() |
| for line in fmt: |
| match = re.search(r'field:([^;]*);', line) |
| if match is None: |
| continue |
| parts = match.group(1).split() |
| field_name = parts[-1:][0] |
| field_type = " ".join(parts[:-1]) |
| if "__data_loc" in field_type: |
| continue |
| if field_name.startswith("common_"): |
| continue |
| print(" %s %s;" % (field_type, field_name)) |
| |
| def print_tpoint(category, event): |
| tpoint = "%s:%s" % (category, event) |
| if not args.filter or fnmatch.fnmatch(tpoint, args.filter): |
| print(tpoint) |
| if args.variables: |
| print_tpoint_format(category, event) |
| |
| def print_tracepoints(): |
| for category in os.listdir(event_root): |
| cat_dir = os.path.join(event_root, category) |
| if not os.path.isdir(cat_dir): |
| continue |
| for event in os.listdir(cat_dir): |
| evt_dir = os.path.join(cat_dir, event) |
| if os.path.isdir(evt_dir): |
| print_tpoint(category, event) |
| |
| def print_usdt(pid, lib): |
| reader = USDT(path=lib, pid=pid) |
| probes_seen = [] |
| for probe in reader.enumerate_probes(): |
| probe_name = probe.short_name() |
| if not args.filter or fnmatch.fnmatch(probe_name, args.filter): |
| if probe_name in probes_seen: |
| continue |
| probes_seen.append(probe_name) |
| if args.variables: |
| print(probe) |
| else: |
| print("%s %s:%s" % (probe.bin_path, |
| probe.provider, probe.name)) |
| |
| if __name__ == "__main__": |
| try: |
| if args.pid or args.lib != "": |
| print_usdt(args.pid, args.lib) |
| else: |
| print_tracepoints() |
| except: |
| if sys.exc_info()[0] is not SystemExit: |
| print(sys.exc_info()[1]) |
| |