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)
|