155 lines
4.4 KiB
Python
Executable File
155 lines
4.4 KiB
Python
Executable File
#!/usr/bin/env python
|
|
# -*- coding: utf-8 -*-
|
|
|
|
"""
|
|
Command-line wrapper for the tracetool machinery.
|
|
"""
|
|
|
|
__author__ = "Lluís Vilanova <vilanova@ac.upc.edu>"
|
|
__copyright__ = "Copyright 2012-2014, Lluís Vilanova <vilanova@ac.upc.edu>"
|
|
__license__ = "GPL version 2 or (at your option) any later version"
|
|
|
|
__maintainer__ = "Stefan Hajnoczi"
|
|
__email__ = "stefanha@linux.vnet.ibm.com"
|
|
|
|
|
|
import sys
|
|
import getopt
|
|
import os.path
|
|
import re
|
|
|
|
from tracetool import error_write, out
|
|
import tracetool.backend
|
|
import tracetool.format
|
|
|
|
|
|
_SCRIPT = ""
|
|
|
|
def error_opt(msg = None):
|
|
if msg is not None:
|
|
error_write("Error: " + msg + "\n")
|
|
|
|
backend_descr = "\n".join([ " %-15s %s" % (n, d)
|
|
for n,d in tracetool.backend.get_list() ])
|
|
format_descr = "\n".join([ " %-15s %s" % (n, d)
|
|
for n,d in tracetool.format.get_list() ])
|
|
error_write("""\
|
|
Usage: %(script)s --format=<format> --backends=<backends> [<options>]
|
|
|
|
Backends:
|
|
%(backends)s
|
|
|
|
Formats:
|
|
%(formats)s
|
|
|
|
Options:
|
|
--help This help message.
|
|
--list-backends Print list of available backends.
|
|
--check-backends Check if the given backend is valid.
|
|
--binary <path> Full path to QEMU binary.
|
|
--target-type <type> QEMU emulator target type ('system' or 'user').
|
|
--target-name <name> QEMU emulator target name.
|
|
--group <name> Name of the event group
|
|
--probe-prefix <prefix> Prefix for dtrace probe names
|
|
(default: qemu-<target-type>-<target-name>).\
|
|
""" % {
|
|
"script" : _SCRIPT,
|
|
"backends" : backend_descr,
|
|
"formats" : format_descr,
|
|
})
|
|
|
|
if msg is None:
|
|
sys.exit(0)
|
|
else:
|
|
sys.exit(1)
|
|
|
|
def main(args):
|
|
global _SCRIPT
|
|
_SCRIPT = args[0]
|
|
|
|
long_opts = ["backends=", "format=", "help", "list-backends",
|
|
"check-backends", "group="]
|
|
long_opts += ["binary=", "target-type=", "target-name=", "probe-prefix="]
|
|
|
|
try:
|
|
opts, args = getopt.getopt(args[1:], "", long_opts)
|
|
except getopt.GetoptError as err:
|
|
error_opt(str(err))
|
|
|
|
check_backends = False
|
|
arg_backends = []
|
|
arg_format = ""
|
|
arg_group = None
|
|
binary = None
|
|
target_type = None
|
|
target_name = None
|
|
probe_prefix = None
|
|
for opt, arg in opts:
|
|
if opt == "--help":
|
|
error_opt()
|
|
|
|
elif opt == "--backends":
|
|
arg_backends = arg.split(",")
|
|
elif opt == "--group":
|
|
arg_group = arg
|
|
elif opt == "--format":
|
|
arg_format = arg
|
|
|
|
elif opt == "--list-backends":
|
|
public_backends = tracetool.backend.get_list(only_public = True)
|
|
out(", ".join([ b for b,_ in public_backends ]))
|
|
sys.exit(0)
|
|
elif opt == "--check-backends":
|
|
check_backends = True
|
|
|
|
elif opt == "--binary":
|
|
binary = arg
|
|
elif opt == '--target-type':
|
|
target_type = arg
|
|
elif opt == '--target-name':
|
|
target_name = arg
|
|
elif opt == '--probe-prefix':
|
|
probe_prefix = arg
|
|
|
|
else:
|
|
error_opt("unhandled option: %s" % opt)
|
|
|
|
if len(arg_backends) == 0:
|
|
error_opt("no backends specified")
|
|
|
|
if check_backends:
|
|
for backend in arg_backends:
|
|
if not tracetool.backend.exists(backend):
|
|
sys.exit(1)
|
|
sys.exit(0)
|
|
|
|
if arg_group is None:
|
|
error_opt("group name is required")
|
|
|
|
if arg_format == "stap":
|
|
if binary is None:
|
|
error_opt("--binary is required for SystemTAP tapset generator")
|
|
if probe_prefix is None and target_type is None:
|
|
error_opt("--target-type is required for SystemTAP tapset generator")
|
|
if probe_prefix is None and target_name is None:
|
|
error_opt("--target-name is required for SystemTAP tapset generator")
|
|
|
|
if probe_prefix is None:
|
|
probe_prefix = ".".join(["qemu", target_type, target_name])
|
|
|
|
if len(args) < 1:
|
|
error_opt("missing trace-events filepath")
|
|
events = []
|
|
for arg in args:
|
|
with open(arg, "r") as fh:
|
|
events.extend(tracetool.read_events(fh, arg))
|
|
|
|
try:
|
|
tracetool.generate(events, arg_group, arg_format, arg_backends,
|
|
binary=binary, probe_prefix=probe_prefix)
|
|
except tracetool.TracetoolError as e:
|
|
error_opt(str(e))
|
|
|
|
if __name__ == "__main__":
|
|
main(sys.argv)
|