|
| 1 | +import gdb |
| 2 | + |
| 3 | +def jitmap_raw(): |
| 4 | + pid = gdb.selected_inferior().pid |
| 5 | + jitmap_file = open("/tmp/perf-%d.map" % (pid,), "r") |
| 6 | + jitmap = jitmap_file.read() |
| 7 | + jitmap_file.close() |
| 8 | + return jitmap |
| 9 | + |
| 10 | +def jit_functions(): |
| 11 | + jitmap = jitmap_raw() |
| 12 | + |
| 13 | + functions = [] |
| 14 | + for line in jitmap.strip().split("\n"): |
| 15 | + [addr, size, name] = line.split(" ") |
| 16 | + functions.append((int(addr, 16), int(size, 16), name)) |
| 17 | + |
| 18 | + return functions |
| 19 | + |
| 20 | +class JitDecorator(gdb.FrameDecorator.FrameDecorator): |
| 21 | + def __init__(self, fobj, name): |
| 22 | + super(JitDecorator, self).__init__(fobj) |
| 23 | + self.name = name |
| 24 | + |
| 25 | + def function(self): |
| 26 | + return self.name |
| 27 | + |
| 28 | +class JitFilter: |
| 29 | + """ |
| 30 | + A backtrace filter which reads perf map files produced by cranelift-jit. |
| 31 | + """ |
| 32 | + |
| 33 | + def __init__(self): |
| 34 | + self.name = 'JitFilter' |
| 35 | + self.enabled = True |
| 36 | + self.priority = 0 |
| 37 | + |
| 38 | + gdb.current_progspace().frame_filters[self.name] = self |
| 39 | + |
| 40 | + # FIXME add an actual unwinder or somehow register JITed .eh_frame with gdb to avoid relying on |
| 41 | + # gdb unwinder heuristics. |
| 42 | + def filter(self, frame_iter): |
| 43 | + for frame in frame_iter: |
| 44 | + frame_addr = frame.inferior_frame().pc() |
| 45 | + for (addr, size, name) in jit_functions(): |
| 46 | + if frame_addr >= addr and frame_addr < addr + size: |
| 47 | + yield JitDecorator(frame, name) |
| 48 | + break |
| 49 | + else: |
| 50 | + yield frame |
| 51 | + |
| 52 | +JitFilter() |
0 commit comments