70 lines
2.0 KiB
Python
70 lines
2.0 KiB
Python
|
# -*- coding: utf-8 -*-
|
||
|
|
||
|
"""
|
||
|
Generic management for the 'vcpu' property.
|
||
|
|
||
|
"""
|
||
|
|
||
|
__author__ = "Lluís Vilanova <vilanova@ac.upc.edu>"
|
||
|
__copyright__ = "Copyright 2016, 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"
|
||
|
|
||
|
|
||
|
from tracetool import Arguments, try_import
|
||
|
|
||
|
|
||
|
def transform_event(event):
|
||
|
"""Transform event to comply with the 'vcpu' property (if present)."""
|
||
|
if "vcpu" in event.properties:
|
||
|
# events with 'tcg-trans' and 'tcg-exec' are auto-generated from
|
||
|
# already-patched events
|
||
|
assert "tcg-trans" not in event.properties
|
||
|
assert "tcg-exec" not in event.properties
|
||
|
|
||
|
event.args = Arguments([("CPUState *", "__cpu"), event.args])
|
||
|
if "tcg" in event.properties:
|
||
|
fmt = "\"cpu=%p \""
|
||
|
event.fmt = [fmt + event.fmt[0],
|
||
|
fmt + event.fmt[1]]
|
||
|
else:
|
||
|
fmt = "\"cpu=%p \""
|
||
|
event.fmt = fmt + event.fmt
|
||
|
return event
|
||
|
|
||
|
|
||
|
def transform_args(format, event, *args, **kwargs):
|
||
|
"""Transforms the arguments to suit the specified format.
|
||
|
|
||
|
The format module must implement function 'vcpu_args', which receives the
|
||
|
implicit arguments added by the 'vcpu' property, and must return suitable
|
||
|
arguments for the given format.
|
||
|
|
||
|
The function is only called for events with the 'vcpu' property.
|
||
|
|
||
|
Parameters
|
||
|
==========
|
||
|
format : str
|
||
|
Format module name.
|
||
|
event : Event
|
||
|
args, kwargs
|
||
|
Passed to 'vcpu_transform_args'.
|
||
|
|
||
|
Returns
|
||
|
=======
|
||
|
Arguments
|
||
|
The transformed arguments, including the non-implicit ones.
|
||
|
|
||
|
"""
|
||
|
if "vcpu" in event.properties:
|
||
|
ok, func = try_import("tracetool.format." + format,
|
||
|
"vcpu_transform_args")
|
||
|
assert ok
|
||
|
assert func
|
||
|
return Arguments([func(event.args[:1], *args, **kwargs),
|
||
|
event.args[1:]])
|
||
|
else:
|
||
|
return event.args
|