2238dcc393
This is an ongoing series of commits that are reformatting our Python code. Reformatting is done with `black` (23.1.0). If you end up having problems merging this commit because you have made changes to a python file, the best way to handle that is to run `git checkout --ours <yourfile>` and then reformat it with black. RFC: https://discourse.llvm.org/t/rfc-document-and-standardize-python-code-style Differential revision: https://reviews.llvm.org/D151460
81 lines
2.5 KiB
Python
81 lines
2.5 KiB
Python
#!/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)
|