| #!/usr/bin/env python |
| |
| import json |
| import os |
| import re |
| import subprocess |
| import sys |
| import argparse |
| |
| |
| class CrashLogPatcher: |
| |
| SYMBOL_REGEX = re.compile(r'^([0-9a-fA-F]+) T _(.*)$') |
| UUID_REGEX = re.compile(r'UUID: ([-0-9a-fA-F]+) \(([^\(]+)\) .*') |
| |
| def __init__(self, data, binary, offsets, json): |
| self.data = data |
| self.binary = binary |
| self.offsets = offsets |
| self.json = json |
| |
| def patch_executable(self): |
| self.data = self.data.replace("@EXEC@", self.binary) |
| self.data = self.data.replace("@NAME@", os.path.basename(self.binary)) |
| |
| def patch_uuid(self): |
| output = subprocess.check_output(['dwarfdump', '--uuid', self.binary]).decode("utf-8") |
| m = self.UUID_REGEX.match(output) |
| if m: |
| self.data = self.data.replace("@UUID@", m.group(1)) |
| |
| def patch_addresses(self): |
| if not self.offsets: |
| return |
| output = subprocess.check_output(['nm', self.binary]).decode("utf-8") |
| for line in output.splitlines(): |
| m = self.SYMBOL_REGEX.match(line) |
| if m: |
| address = m.group(1) |
| symbol = m.group(2) |
| if symbol in self.offsets: |
| patch_addr = int(m.group(1), 16) + int( |
| self.offsets[symbol]) |
| if self.json: |
| patch_addr = patch_addr - 0x100000000 |
| representation = int |
| else: |
| representation = hex |
| self.data = self.data.replace( |
| "@{}@".format(symbol), str(representation(patch_addr))) |
| |
| def remove_metadata(self): |
| self.data= self.data[self.data.index('\n') + 1:] |
| |
| |
| if __name__ == '__main__': |
| parser = argparse.ArgumentParser(description='Crashlog Patcher') |
| parser.add_argument('--binary', required=True) |
| parser.add_argument('--crashlog', required=True) |
| parser.add_argument('--offsets', required=True) |
| parser.add_argument('--json', default=False, action='store_true') |
| parser.add_argument('--no-metadata', default=False, action='store_true') |
| args = parser.parse_args() |
| |
| offsets = json.loads(args.offsets) |
| |
| with open(args.crashlog, 'r') as file: |
| data = file.read() |
| |
| p = CrashLogPatcher(data, args.binary, offsets, args.json) |
| p.patch_executable() |
| p.patch_uuid() |
| p.patch_addresses() |
| |
| if args.no_metadata: |
| p.remove_metadata() |
| |
| with open(args.crashlog, 'w') as file: |
| file.write(p.data) |