mirror of
https://github.com/isledecomp/isle.git
synced 2024-11-22 15:48:09 -05:00
bd85abaf2a
* Use python3 features * Use `with` statement for file access * Use f-strings instead of modulo string formatting * Single quotes in most places Fix typo in 'with' statement * Add files into missing messages * Fix can_resolve_register_differences and round percentages * Return modified value instead of relying on in-place modification
76 lines
2.1 KiB
Python
Executable file
76 lines
2.1 KiB
Python
Executable file
#!/usr/bin/env python3
|
|
|
|
import argparse
|
|
import colorama
|
|
import difflib
|
|
import subprocess
|
|
import os
|
|
import sys
|
|
|
|
parser = argparse.ArgumentParser(allow_abbrev=False,
|
|
description='Verify Exports: Compare the exports of two DLLs.')
|
|
parser.add_argument('original', metavar='original-binary', help='The original binary')
|
|
parser.add_argument('recompiled', metavar='recompiled-binary', help='The recompiled binary')
|
|
parser.add_argument('--no-color', '-n', action='store_true', help='Do not color the output')
|
|
|
|
args = parser.parse_args()
|
|
|
|
if not os.path.isfile(args.original):
|
|
parser.error(f'Original binary file {args.original} does not exist')
|
|
|
|
if not os.path.isfile(args.recompiled):
|
|
parser.error(f'Recompiled binary {args.recompiled} does not exist')
|
|
|
|
def get_file_in_script_dir(fn):
|
|
return os.path.join(os.path.dirname(os.path.abspath(sys.argv[0])), fn)
|
|
|
|
def get_exports(file):
|
|
call = [get_file_in_script_dir('DUMPBIN.EXE'), '/EXPORTS']
|
|
|
|
if os.name != 'nt':
|
|
call.insert(0, 'wine')
|
|
file = subprocess.check_output(['winepath', '-w', file]).decode('utf-8').strip()
|
|
|
|
call.append(file)
|
|
|
|
raw = subprocess.check_output(call).decode('utf-8').split('\r\n')
|
|
exports = []
|
|
|
|
start = False
|
|
|
|
for line in raw:
|
|
if not start:
|
|
if line == ' ordinal hint name':
|
|
start = True
|
|
else:
|
|
if line:
|
|
exports.append(line[27:line.rindex(' (')])
|
|
elif exports:
|
|
break
|
|
|
|
return exports
|
|
|
|
og_exp = get_exports(args.original)
|
|
re_exp = get_exports(args.recompiled)
|
|
|
|
udiff = difflib.unified_diff(og_exp, re_exp)
|
|
has_diff = False
|
|
|
|
for line in udiff:
|
|
has_diff = True
|
|
color = ''
|
|
if line.startswith('++') or line.startswith('@@') or line.startswith('--'):
|
|
# Skip unneeded parts of the diff for the brief view
|
|
continue
|
|
# Work out color if we are printing color
|
|
if not args.no_color:
|
|
if line.startswith('+'):
|
|
color = colorama.Fore.GREEN
|
|
elif line.startswith('-'):
|
|
color = colorama.Fore.RED
|
|
print(color + line)
|
|
# Reset color if we're printing in color
|
|
if not args.no_color:
|
|
print(colorama.Style.RESET_ALL, end='')
|
|
|
|
sys.exit(1 if has_diff else 0)
|