| #!/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) |