scripts/replay-dump.py

   1 #!/usr/bin/env python3
   2 # -*- coding: utf-8 -*-
   3 #
   4 # Dump the contents of a recorded execution stream
   5 #
   6 #  Copyright (c) 2017 Alex Bennée <alex.bennee@linaro.org>
   7 #
   8 # This library is free software; you can redistribute it and/or
   9 # modify it under the terms of the GNU Lesser General Public
  10 # License as published by the Free Software Foundation; either
  11 # version 2 of the License, or (at your option) any later version.
  12 #
  13 # This library is distributed in the hope that it will be useful,
  14 # but WITHOUT ANY WARRANTY; without even the implied warranty of
  15 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  16 # Lesser General Public License for more details.
  17 #
  18 # You should have received a copy of the GNU Lesser General Public
  19 # License along with this library; if not, see <http://www.gnu.org/licenses/>.
  20
  21 import argparse
  22 import struct
  23 from collections import namedtuple
  24
  25 # This mirrors some of the global replay state which some of the
  26 # stream loading refers to. Some decoders may read the next event so
  27 # we need handle that case. Calling reuse_event will ensure the next
  28 # event is read from the cache rather than advancing the file.
  29
  30 class ReplayState(object):
  31     def __init__(self):
  32         self.event = -1
  33         self.event_count = 0
  34         self.already_read = False
  35         self.current_checkpoint = 0
  36         self.checkpoint = 0
  37
  38     def set_event(self, ev):
  39         self.event = ev
  40         self.event_count += 1
  41
  42     def get_event(self):
  43         self.already_read = False
  44         return self.event
  45
  46     def reuse_event(self, ev):
  47         self.event = ev
  48         self.already_read = True
  49
  50     def set_checkpoint(self):
  51         self.checkpoint = self.event - self.checkpoint_start
  52
  53     def get_checkpoint(self):
  54         return self.checkpoint
  55
  56 replay_state = ReplayState()
  57
  58 # Simple read functions that mirror replay-internal.c
  59 # The file-stream is big-endian and manually written out a byte at a time.
  60
  61 def read_byte(fin):
  62     "Read a single byte"
  63     return struct.unpack('>B', fin.read(1))[0]
  64
  65 def read_event(fin):
  66     "Read a single byte event, but save some state"
  67     if replay_state.already_read:
  68         return replay_state.get_event()
  69     else:
  70         replay_state.set_event(read_byte(fin))
  71         return replay_state.event
  72
  73 def read_word(fin):
  74     "Read a 16 bit word"
  75     return struct.unpack('>H', fin.read(2))[0]
  76
  77 def read_dword(fin):
  78     "Read a 32 bit word"
  79     return struct.unpack('>I', fin.read(4))[0]
  80
  81 def read_qword(fin):
  82     "Read a 64 bit word"
  83     return struct.unpack('>Q', fin.read(8))[0]
  84
  85 # Generic decoder structure
  86 Decoder = namedtuple("Decoder", "eid name fn")
  87
  88 def call_decode(table, index, dumpfile):
  89     "Search decode table for next step"
  90     decoder = next((d for d in table if d.eid == index), None)
  91     if not decoder:
  92         print("Could not decode index: %d" % (index))
  93         print("Entry is: %s" % (decoder))
  94         print("Decode Table is:\n%s" % (table))
  95         return False
  96     else:
  97         return decoder.fn(decoder.eid, decoder.name, dumpfile)
  98
  99 # Print event
 100 def print_event(eid, name, string=None, event_count=None):
 101     "Print event with count"
 102     if not event_count:
 103         event_count = replay_state.event_count
 104
 105     if string:
 106         print("%d:%s(%d) %s" % (event_count, name, eid, string))
 107     else:
 108         print("%d:%s(%d)" % (event_count, name, eid))
 109
 110
 111 # Decoders for each event type
 112
 113 def decode_unimp(eid, name, _unused_dumpfile):
 114     "Unimplimented decoder, will trigger exit"
 115     print("%s not handled - will now stop" % (name))
 116     return False
 117
 118 # Checkpoint decoder
 119 def swallow_async_qword(eid, name, dumpfile):
 120     "Swallow a qword of data without looking at it"
 121     step_id = read_qword(dumpfile)
 122     print("  %s(%d) @ %d" % (name, eid, step_id))
 123     return True
 124
 125 async_decode_table = [ Decoder(0, "REPLAY_ASYNC_EVENT_BH", swallow_async_qword),
 126                        Decoder(1, "REPLAY_ASYNC_INPUT", decode_unimp),
 127                        Decoder(2, "REPLAY_ASYNC_INPUT_SYNC", decode_unimp),
 128                        Decoder(3, "REPLAY_ASYNC_CHAR_READ", decode_unimp),
 129                        Decoder(4, "REPLAY_ASYNC_EVENT_BLOCK", decode_unimp),
 130                        Decoder(5, "REPLAY_ASYNC_EVENT_NET", decode_unimp),
 131 ]
 132 # See replay_read_events/replay_read_event
 133 def decode_async(eid, name, dumpfile):
 134     """Decode an ASYNC event"""
 135
 136     print_event(eid, name)
 137
 138     async_event_kind = read_byte(dumpfile)
 139     async_event_checkpoint = read_byte(dumpfile)
 140
 141     if async_event_checkpoint != replay_state.current_checkpoint:
 142         print("  mismatch between checkpoint %d and async data %d" % (
 143             replay_state.current_checkpoint, async_event_checkpoint))
 144         return True
 145
 146     return call_decode(async_decode_table, async_event_kind, dumpfile)
 147
 148
 149 def decode_instruction(eid, name, dumpfile):
 150     ins_diff = read_dword(dumpfile)
 151     print_event(eid, name, "0x%x" % (ins_diff))
 152     return True
 153
 154 def decode_audio_out(eid, name, dumpfile):
 155     audio_data = read_dword(dumpfile)
 156     print_event(eid, name, "%d" % (audio_data))
 157     return True
 158
 159 def decode_checkpoint(eid, name, dumpfile):
 160     """Decode a checkpoint.
 161
 162     Checkpoints contain a series of async events with their own specific data.
 163     """
 164     replay_state.set_checkpoint()
 165     # save event count as we peek ahead
 166     event_number = replay_state.event_count
 167     next_event = read_event(dumpfile)
 168
 169     # if the next event is EVENT_ASYNC there are a bunch of
 170     # async events to read, otherwise we are done
 171     if next_event != 3:
 172         print_event(eid, name, "no additional data", event_number)
 173     else:
 174         print_event(eid, name, "more data follows", event_number)
 175
 176     replay_state.reuse_event(next_event)
 177     return True
 178
 179 def decode_checkpoint_init(eid, name, dumpfile):
 180     print_event(eid, name)
 181     return True
 182
 183 def decode_interrupt(eid, name, dumpfile):
 184     print_event(eid, name)
 185     return True
 186
 187 def decode_clock(eid, name, dumpfile):
 188     clock_data = read_qword(dumpfile)
 189     print_event(eid, name, "0x%x" % (clock_data))
 190     return True
 191
 192
 193 # pre-MTTCG merge
 194 v5_event_table = [Decoder(0, "EVENT_INSTRUCTION", decode_instruction),
 195                   Decoder(1, "EVENT_INTERRUPT", decode_interrupt),
 196                   Decoder(2, "EVENT_EXCEPTION", decode_unimp),
 197                   Decoder(3, "EVENT_ASYNC", decode_async),
 198                   Decoder(4, "EVENT_SHUTDOWN", decode_unimp),
 199                   Decoder(5, "EVENT_CHAR_WRITE", decode_unimp),
 200                   Decoder(6, "EVENT_CHAR_READ_ALL", decode_unimp),
 201                   Decoder(7, "EVENT_CHAR_READ_ALL_ERROR", decode_unimp),
 202                   Decoder(8, "EVENT_CLOCK_HOST", decode_clock),
 203                   Decoder(9, "EVENT_CLOCK_VIRTUAL_RT", decode_clock),
 204                   Decoder(10, "EVENT_CP_CLOCK_WARP_START", decode_checkpoint),
 205                   Decoder(11, "EVENT_CP_CLOCK_WARP_ACCOUNT", decode_checkpoint),
 206                   Decoder(12, "EVENT_CP_RESET_REQUESTED", decode_checkpoint),
 207                   Decoder(13, "EVENT_CP_SUSPEND_REQUESTED", decode_checkpoint),
 208                   Decoder(14, "EVENT_CP_CLOCK_VIRTUAL", decode_checkpoint),
 209                   Decoder(15, "EVENT_CP_CLOCK_HOST", decode_checkpoint),
 210                   Decoder(16, "EVENT_CP_CLOCK_VIRTUAL_RT", decode_checkpoint),
 211                   Decoder(17, "EVENT_CP_INIT", decode_checkpoint_init),
 212                   Decoder(18, "EVENT_CP_RESET", decode_checkpoint),
 213 ]
 214
 215 # post-MTTCG merge, AUDIO support added
 216 v6_event_table = [Decoder(0, "EVENT_INSTRUCTION", decode_instruction),
 217                   Decoder(1, "EVENT_INTERRUPT", decode_interrupt),
 218                   Decoder(2, "EVENT_EXCEPTION", decode_unimp),
 219                   Decoder(3, "EVENT_ASYNC", decode_async),
 220                   Decoder(4, "EVENT_SHUTDOWN", decode_unimp),
 221                   Decoder(5, "EVENT_CHAR_WRITE", decode_unimp),
 222                   Decoder(6, "EVENT_CHAR_READ_ALL", decode_unimp),
 223                   Decoder(7, "EVENT_CHAR_READ_ALL_ERROR", decode_unimp),
 224                   Decoder(8, "EVENT_AUDIO_OUT", decode_audio_out),
 225                   Decoder(9, "EVENT_AUDIO_IN", decode_unimp),
 226                   Decoder(10, "EVENT_CLOCK_HOST", decode_clock),
 227                   Decoder(11, "EVENT_CLOCK_VIRTUAL_RT", decode_clock),
 228                   Decoder(12, "EVENT_CP_CLOCK_WARP_START", decode_checkpoint),
 229                   Decoder(13, "EVENT_CP_CLOCK_WARP_ACCOUNT", decode_checkpoint),
 230                   Decoder(14, "EVENT_CP_RESET_REQUESTED", decode_checkpoint),
 231                   Decoder(15, "EVENT_CP_SUSPEND_REQUESTED", decode_checkpoint),
 232                   Decoder(16, "EVENT_CP_CLOCK_VIRTUAL", decode_checkpoint),
 233                   Decoder(17, "EVENT_CP_CLOCK_HOST", decode_checkpoint),
 234                   Decoder(18, "EVENT_CP_CLOCK_VIRTUAL_RT", decode_checkpoint),
 235                   Decoder(19, "EVENT_CP_INIT", decode_checkpoint_init),
 236                   Decoder(20, "EVENT_CP_RESET", decode_checkpoint),
 237 ]
 238
 239 # Shutdown cause added
 240 v7_event_table = [Decoder(0, "EVENT_INSTRUCTION", decode_instruction),
 241                   Decoder(1, "EVENT_INTERRUPT", decode_interrupt),
 242                   Decoder(2, "EVENT_EXCEPTION", decode_unimp),
 243                   Decoder(3, "EVENT_ASYNC", decode_async),
 244                   Decoder(4, "EVENT_SHUTDOWN", decode_unimp),
 245                   Decoder(5, "EVENT_SHUTDOWN_HOST_ERR", decode_unimp),
 246                   Decoder(6, "EVENT_SHUTDOWN_HOST_QMP", decode_unimp),
 247                   Decoder(7, "EVENT_SHUTDOWN_HOST_SIGNAL", decode_unimp),
 248                   Decoder(8, "EVENT_SHUTDOWN_HOST_UI", decode_unimp),
 249                   Decoder(9, "EVENT_SHUTDOWN_GUEST_SHUTDOWN", decode_unimp),
 250                   Decoder(10, "EVENT_SHUTDOWN_GUEST_RESET", decode_unimp),
 251                   Decoder(11, "EVENT_SHUTDOWN_GUEST_PANIC", decode_unimp),
 252                   Decoder(12, "EVENT_SHUTDOWN___MAX", decode_unimp),
 253                   Decoder(13, "EVENT_CHAR_WRITE", decode_unimp),
 254                   Decoder(14, "EVENT_CHAR_READ_ALL", decode_unimp),
 255                   Decoder(15, "EVENT_CHAR_READ_ALL_ERROR", decode_unimp),
 256                   Decoder(16, "EVENT_AUDIO_OUT", decode_audio_out),
 257                   Decoder(17, "EVENT_AUDIO_IN", decode_unimp),
 258                   Decoder(18, "EVENT_CLOCK_HOST", decode_clock),
 259                   Decoder(19, "EVENT_CLOCK_VIRTUAL_RT", decode_clock),
 260                   Decoder(20, "EVENT_CP_CLOCK_WARP_START", decode_checkpoint),
 261                   Decoder(21, "EVENT_CP_CLOCK_WARP_ACCOUNT", decode_checkpoint),
 262                   Decoder(22, "EVENT_CP_RESET_REQUESTED", decode_checkpoint),
 263                   Decoder(23, "EVENT_CP_SUSPEND_REQUESTED", decode_checkpoint),
 264                   Decoder(24, "EVENT_CP_CLOCK_VIRTUAL", decode_checkpoint),
 265                   Decoder(25, "EVENT_CP_CLOCK_HOST", decode_checkpoint),
 266                   Decoder(26, "EVENT_CP_CLOCK_VIRTUAL_RT", decode_checkpoint),
 267                   Decoder(27, "EVENT_CP_INIT", decode_checkpoint_init),
 268                   Decoder(28, "EVENT_CP_RESET", decode_checkpoint),
 269 ]
 270
 271 def parse_arguments():
 272     "Grab arguments for script"
 273     parser = argparse.ArgumentParser()
 274     parser.add_argument("-f", "--file", help='record/replay dump to read from',
 275                         required=True)
 276     return parser.parse_args()
 277
 278 def decode_file(filename):
 279     "Decode a record/replay dump"
 280     dumpfile = open(filename, "rb")
 281
 282     # read and throwaway the header
 283     version = read_dword(dumpfile)
 284     junk = read_qword(dumpfile)
 285
 286     print("HEADER: version 0x%x" % (version))
 287
 288     if version == 0xe02007:
 289         event_decode_table = v7_event_table
 290         replay_state.checkpoint_start = 12
 291     elif version == 0xe02006:
 292         event_decode_table = v6_event_table
 293         replay_state.checkpoint_start = 12
 294     else:
 295         event_decode_table = v5_event_table
 296         replay_state.checkpoint_start = 10
 297
 298     try:
 299         decode_ok = True
 300         while decode_ok:
 301             event = read_event(dumpfile)
 302             decode_ok = call_decode(event_decode_table, event, dumpfile)
 303     finally:
 304         dumpfile.close()
 305
 306 if __name__ == "__main__":
 307     args = parse_arguments()
 308     decode_file(args.file)