730fca46fc
JSON crashlogs normally start with a single line of meta data that we strip unconditionally. Some producers started omitting the meta data which tripped up crashlog. Be more resilient by only removing the first line when we know it really is meta data. rdar://82641662
80 lines
2.5 KiB
Python
80 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)
|