Upload files to "/"

patch_bp.py

@@ -0,0 +1,91 @@
+import argparse
+import os
+import struct
+
+def parse_arguments():
+    parser = argparse.ArgumentParser(description='Replace text in a binary file based on a mapping file.')
+    parser.add_argument('bp_file', type=str, help='The path to the binary file.')
+    parser.add_argument('txt_file', type=str, help='The path to the text file containing replacements.')
+    parser.add_argument('out_bp_file', type=str, nargs='?', help='The output binary file path. If not specified, will add "_out" to bp_file.')
+    return parser.parse_args()
+
+def read_replacements(txt_file):
+    replacements = {}
+    with open(txt_file, 'r', encoding='utf-8') as f:
+        while True:
+            original_line = f.readline()
+            if not original_line:
+                break
+            original_line = original_line.strip()
+            if not original_line:
+                continue
+            translation_line = f.readline().rstrip('\n')
+            if translation_line:  # Skip if translation is empty
+                # Extract the hex address and original text
+                hex_address, original_text = original_line.split(']', 1)
+                hex_address = hex_address.strip('[')
+                address = int(hex_address, 16)
+                translation_line = translation_line.split(']', 1)[1]
+                if not translation_line:
+                    continue
+                replacements[address] = (original_text.encode('cp932'), translation_line.encode('utf-8'))
+    return replacements
+
+def replace_text_in_bp(bp_file, replacements):
+    with open(bp_file, 'rb') as f:
+        data = bytearray(f.read())
+
+    for address, (original_text, translation_text) in replacements.items():
+        olen = len(original_text)
+        tlen = len(translation_text)
+        data[address:address + olen] = translation_text + b' ' * (olen - tlen)
+        print(address, olen, tlen, original_text.decode('cp932'), translation_text.decode(), data[address:address + olen])
+
+    return data
+
+# def replace_text_in_bp(bp_file, replacements):
+#     with open(bp_file, 'rb') as f:
+#         data = f.read()
+    
+#     offset = 0
+
+#     for address, (original_text, translation_text) in replacements.items():
+#         olen = len(original_text)
+#         tlen = len(translation_text)
+#         address += offset
+#         data = data[:address] + translation_text + data[address+olen:]
+#         print(address, offset, olen, tlen, original_text.decode('cp932'), translation_text.decode(), data[address:address + tlen])
+#         offset += tlen - olen
+
+#     data = bytearray(data)
+#     header = struct.unpack('<I', data[0:4])[0]
+#     size = struct.unpack('<I', data[4:8])[0]
+#     print(header, size)
+#     data[4:8] = struct.pack('<I', len(data) - header)
+#     size = struct.unpack('<I', data[4:8])[0]
+#     print(len(data) - header, size)
+#     return data
+
+def main():
+    args = parse_arguments()
+    
+    # Determine output file path
+    if args.out_bp_file:
+        out_bp_file = args.out_bp_file
+    else:
+        out_bp_file = f"{os.path.splitext(args.bp_file)[0]}_out{os.path.splitext(args.bp_file)[1]}"
+
+    # Read replacements from the txt file
+    replacements = read_replacements(args.txt_file)
+
+    # Replace text in the binary file
+    modified_data = replace_text_in_bp(args.bp_file, replacements)
+
+    # Write the modified data to the output file
+    with open(out_bp_file, 'wb') as f:
+        f.write(modified_data)
+
+    print(f'Modified binary file written to: {out_bp_file}')
+
+if __name__ == '__main__':
+    main()