Annotation of qemu/scripts/simpletrace.py, revision 1.1.1.2

1.1       root        1: #!/usr/bin/env python
                      2: #
                      3: # Pretty-printer for simple trace backend binary trace files
                      4: #
                      5: # Copyright IBM, Corp. 2010
                      6: #
                      7: # This work is licensed under the terms of the GNU GPL, version 2.  See
                      8: # the COPYING file in the top-level directory.
                      9: #
                     10: # For help see docs/tracing.txt
                     11: 
                     12: import struct
                     13: import re
1.1.1.2 ! root       14: import inspect
1.1       root       15: 
                     16: header_event_id = 0xffffffffffffffff
                     17: header_magic    = 0xf2b177cb0aa429b4
                     18: header_version  = 0
1.1.1.2 ! root       19: dropped_event_id = 0xfffffffffffffffe
1.1       root       20: 
                     21: trace_fmt = '=QQQQQQQQ'
                     22: trace_len = struct.calcsize(trace_fmt)
                     23: event_re  = re.compile(r'(disable\s+)?([a-zA-Z0-9_]+)\(([^)]*)\).*')
                     24: 
                     25: def parse_events(fobj):
1.1.1.2 ! root       26:     """Parse a trace-events file into {event_num: (name, arg1, ...)}."""
1.1       root       27: 
                     28:     def get_argnames(args):
                     29:         """Extract argument names from a parameter list."""
                     30:         return tuple(arg.split()[-1].lstrip('*') for arg in args.split(','))
                     31: 
1.1.1.2 ! root       32:     events = {dropped_event_id: ('dropped', 'count')}
1.1       root       33:     event_num = 0
                     34:     for line in fobj:
                     35:         m = event_re.match(line.strip())
                     36:         if m is None:
                     37:             continue
                     38: 
                     39:         disable, name, args = m.groups()
                     40:         events[event_num] = (name,) + get_argnames(args)
                     41:         event_num += 1
                     42:     return events
                     43: 
                     44: def read_record(fobj):
1.1.1.2 ! root       45:     """Deserialize a trace record from a file into a tuple (event_num, timestamp, arg1, ..., arg6)."""
1.1       root       46:     s = fobj.read(trace_len)
                     47:     if len(s) != trace_len:
                     48:         return None
                     49:     return struct.unpack(trace_fmt, s)
                     50: 
                     51: def read_trace_file(fobj):
1.1.1.2 ! root       52:     """Deserialize trace records from a file, yielding record tuples (event_num, timestamp, arg1, ..., arg6)."""
1.1       root       53:     header = read_record(fobj)
                     54:     if header is None or \
                     55:        header[0] != header_event_id or \
                     56:        header[1] != header_magic or \
                     57:        header[2] != header_version:
1.1.1.2 ! root       58:         raise ValueError('not a trace file or incompatible version')
1.1       root       59: 
                     60:     while True:
                     61:         rec = read_record(fobj)
                     62:         if rec is None:
                     63:             break
                     64: 
                     65:         yield rec
                     66: 
1.1.1.2 ! root       67: class Analyzer(object):
        !            68:     """A trace file analyzer which processes trace records.
        !            69: 
        !            70:     An analyzer can be passed to run() or process().  The begin() method is
        !            71:     invoked, then each trace record is processed, and finally the end() method
        !            72:     is invoked.
        !            73: 
        !            74:     If a method matching a trace event name exists, it is invoked to process
        !            75:     that trace record.  Otherwise the catchall() method is invoked."""
        !            76: 
        !            77:     def begin(self):
        !            78:         """Called at the start of the trace."""
        !            79:         pass
        !            80: 
        !            81:     def catchall(self, event, rec):
        !            82:         """Called if no specific method for processing a trace event has been found."""
        !            83:         pass
        !            84: 
        !            85:     def end(self):
        !            86:         """Called at the end of the trace."""
        !            87:         pass
        !            88: 
        !            89: def process(events, log, analyzer):
        !            90:     """Invoke an analyzer on each event in a log."""
        !            91:     if isinstance(events, str):
        !            92:         events = parse_events(open(events, 'r'))
        !            93:     if isinstance(log, str):
        !            94:         log = open(log, 'rb')
        !            95: 
        !            96:     def build_fn(analyzer, event):
        !            97:         fn = getattr(analyzer, event[0], None)
        !            98:         if fn is None:
        !            99:             return analyzer.catchall
        !           100: 
        !           101:         event_argcount = len(event) - 1
        !           102:         fn_argcount = len(inspect.getargspec(fn)[0]) - 1
        !           103:         if fn_argcount == event_argcount + 1:
        !           104:             # Include timestamp as first argument
        !           105:             return lambda _, rec: fn(*rec[1:2 + fn_argcount])
        !           106:         else:
        !           107:             # Just arguments, no timestamp
        !           108:             return lambda _, rec: fn(*rec[2:2 + fn_argcount])
        !           109: 
        !           110:     analyzer.begin()
        !           111:     fn_cache = {}
        !           112:     for rec in read_trace_file(log):
        !           113:         event_num = rec[0]
        !           114:         event = events[event_num]
        !           115:         if event_num not in fn_cache:
        !           116:             fn_cache[event_num] = build_fn(analyzer, event)
        !           117:         fn_cache[event_num](event, rec)
        !           118:     analyzer.end()
        !           119: 
        !           120: def run(analyzer):
        !           121:     """Execute an analyzer on a trace file given on the command-line.
        !           122: 
        !           123:     This function is useful as a driver for simple analysis scripts.  More
        !           124:     advanced scripts will want to call process() instead."""
        !           125:     import sys
        !           126: 
        !           127:     if len(sys.argv) != 3:
        !           128:         sys.stderr.write('usage: %s <trace-events> <trace-file>\n' % sys.argv[0])
        !           129:         sys.exit(1)
        !           130: 
        !           131:     events = parse_events(open(sys.argv[1], 'r'))
        !           132:     process(events, sys.argv[2], analyzer)
        !           133: 
        !           134: if __name__ == '__main__':
        !           135:     class Formatter(Analyzer):
        !           136:         def __init__(self):
        !           137:             self.last_timestamp = None
        !           138: 
        !           139:         def catchall(self, event, rec):
        !           140:             timestamp = rec[1]
        !           141:             if self.last_timestamp is None:
        !           142:                 self.last_timestamp = timestamp
        !           143:             delta_ns = timestamp - self.last_timestamp
        !           144:             self.last_timestamp = timestamp
        !           145: 
        !           146:             fields = [event[0], '%0.3f' % (delta_ns / 1000.0)]
        !           147:             for i in xrange(1, len(event)):
        !           148:                 fields.append('%s=0x%x' % (event[i], rec[i + 1]))
        !           149:             print ' '.join(fields)
        !           150: 
        !           151:     run(Formatter())

unix.superglobalmegacorp.com

This archive runs on limited infrastructure. Preserving old code on modern bandwidth. Automated agents are requested to crawl responsibly.