1 files changed, 69 insertions, 0 deletions

diff --git a/utils/fiio_m3k_tools/nand_patcher.py b/utils/fiio_m3k_tools/nand_patcher.py
new file mode 100755
index 0000000000..261a4de678
--- /dev/null
+++ b/utils/fiio_m3k_tools/nand_patcher.py
@@ -0,0 +1,69 @@
+#!/usr/bin/python3
+
+import sys
+
+IMAGE_SIZE = 128 * 1024 # image is an 128 KiB erase block
+SPL_SIZE   =  12 * 1024 # SPL is at most 12 KiB
+BOOT_SIZE  = 102 * 1024 # bootloader at most 102 KiB
+BOOT_OFF   =  26 * 1024 # offset of bootloader in image
+BOOT_END   = BOOT_OFF+BOOT_SIZE
+
+def patch(in_path, boot_path, spl_path, out_path):
+    # Open the input files
+    in_file = open(in_path, 'rb')
+    boot_file = open(boot_path, 'rb')
+    spl_file = open(spl_path, 'rb')
+
+    # Read the data
+    in_data = in_file.read()
+    boot_data = boot_file.read()
+    spl_data = spl_file.read()
+
+    # Close input files
+    in_file.close()
+    boot_file.close()
+    spl_file.close()
+
+    if len(in_data) != IMAGE_SIZE:
+        print("error: input image is %d bytes, expected %d" % (len(in_data), IMAGE_SIZE))
+        sys.exit(1)
+
+    if len(spl_data) > SPL_SIZE:
+        print("error: SPL is %d bytes, maximum is %d" % (len(spl_data), SPL_SIZE))
+        sys.exit(1)
+
+    if len(boot_data) > BOOT_SIZE:
+        print("error: bootloader is %d bytes, maximum is %d" % (len(boot_data), SPL_SIZE))
+        sys.exit(1)
+
+    print('Patching input image %s' % in_path)
+    print('- SPL size %d' % len(spl_data))
+    print('- Boot size %d' % len(boot_data))
+
+    # Construct output image
+    out_data = b''
+    out_data += spl_data
+    out_data += b'\xff' * (SPL_SIZE - len(spl_data))
+    out_data += in_data[SPL_SIZE:BOOT_OFF]
+    out_data += boot_data
+    out_data += b'\xff' * (BOOT_SIZE - len(boot_data))
+
+    # Sanity check
+    assert( len(out_data) == IMAGE_SIZE )
+
+    # Write output
+    print('Writing output image %s' % out_path)
+    out_file = open(out_path, 'wb')
+    out_file.write(out_data)
+    out_file.close()
+
+
+def main():
+    if len(sys.argv) != 5:
+        print("usage: nand_patcher.py IN_FILE BOOT_FILE SPL_FILE OUT_FILE")
+        sys.exit(1)
+
+    patch(sys.argv[1], sys.argv[2], sys.argv[3], sys.argv[4])
+
+if __name__ == '__main__':
+    main()