mirror of
https://github.com/isledecomp/isle.git
synced 2026-01-24 16:51:15 +00:00
61 lines
1.9 KiB
Python
61 lines
1.9 KiB
Python
#!/usr/bin/env python
|
|
|
|
import argparse
|
|
import hashlib
|
|
import pathlib
|
|
import shutil
|
|
|
|
ORIGINAL_C2_MD5 = "dcd69f1dd28b02dd03dd7ed02984299a" # original C2.EXE
|
|
|
|
C2_MD5 = (
|
|
ORIGINAL_C2_MD5,
|
|
"e70acde41802ddec06c4263bb357ac30", # patched C2.EXE
|
|
)
|
|
|
|
C2_SIZE = 549888
|
|
|
|
|
|
def main():
|
|
parser = argparse.ArgumentParser(allow_abbrev=False, description="Patch C2.EXE of Micrisoft Visual Studio 4.2.0 to disable C4786 warning")
|
|
parser.add_argument("path", type=pathlib.Path, help="Path of C2.EXE")
|
|
parser.add_argument("-f", dest="force", default=False, action="store_true", help="force")
|
|
args = parser.parse_args()
|
|
|
|
if not args.path.is_file():
|
|
parser.error("Input is not a file")
|
|
|
|
binary = bytearray(args.path.open("rb").read())
|
|
md5 = hashlib.md5(binary).hexdigest()
|
|
print(md5, C2_MD5)
|
|
|
|
msg_cb = parser.error if not args.force else print
|
|
if len(binary) != C2_SIZE:
|
|
msg_cb("file size is not correct")
|
|
if md5 not in C2_MD5:
|
|
msg_cb("md5 checksum does not match")
|
|
|
|
if md5 == ORIGINAL_C2_MD5:
|
|
backup = f"{args.path}.BAK"
|
|
print(f"Creating backup \"{backup}\"")
|
|
shutil.copyfile(args.path, backup)
|
|
|
|
def nop_patch(start, count, expected=None):
|
|
replacement = [0x90] * count
|
|
if expected:
|
|
current = list(binary[start:start + count])
|
|
assert len(expected) == count
|
|
assert current in (expected, replacement)
|
|
print(f"Nopping {count} bytes at 0x{start:08x}")
|
|
binary[start:start + count] = replacement
|
|
|
|
# Disable C4786 warning ('%Fs' : identifier was truncated to '%d' characters in the debug information)
|
|
nop_patch(0x52f07, 5, [0xe8, 0x4f, 0xb3, 0xfe, 0xff]) # 0x00453b07
|
|
nop_patch(0x74832, 5, [0xe8, 0x24, 0x9a, 0xfc, 0xff]) # 0x00475432
|
|
|
|
args.path.open("wb").write(binary)
|
|
print("done")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
raise SystemExit(main())
|