From c082492f22c8ea1355f42b436d63b02d39adb025 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E4=BD=95=E6=B3=BD=E9=9A=86?= <hezelong@topscomm.com>
Date: Wed, 8 May 2024 20:19:59 +0800
Subject: [PATCH] =?UTF-8?q?=E6=AF=8F=E6=97=A5=E6=9B=B4=E6=96=B0;?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 source/utl_upgrade.py | 14 +++++++++-----
 1 file changed, 9 insertions(+), 5 deletions(-)

diff --git a/source/utl_upgrade.py b/source/utl_upgrade.py
index 1dc3760..de5403d 100644
--- a/source/utl_upgrade.py
+++ b/source/utl_upgrade.py
@@ -196,7 +196,7 @@ def test2():
     buffer2_en = file_encryption(buffer2)
     pass
 
-def test3():
+def test3(path_bin):
     """ 完整升级包生成测试 """
     config = {
         'file_type':    [0x10, 0x01],       # Xilinx-Demo 自机升级文件
@@ -217,19 +217,23 @@ def test3():
         # 'flash_addr': 0x3E8020,               # 程序起始地址
         # 'flash_size': 0x005FC0,               # 程序空间大小
     }
-    path_bin = Path("F:\\Work\\FPGA\\Test\\Vivado\\test_update\\test_update.vitis\\upgrade_system\\Debug\\sd_card\\BOOT.BIN")
 
-    header, data_b, _ = make_datafile2(path_bin, config, header_len=512)
+    header, data_b, _ = make_datafile2(path_bin, config, header_len=128)
+    header_512, data_b_512, _ = make_datafile2(path_bin, config, header_len=512)
 
     print("Upgrade file generated successfully.")
     print(f"\t header_length={len(header)}, bin_length={len(data_b)}[{hex(len(data_b))}]")
     print(f"\t file md5: {trans_list_to_str(config['md5'])}")
-    file1 = path_bin.parent / (path_bin.stem + '_h512.dat')
+    file1 = path_bin.parent / (path_bin.stem + '.dat')
     file1.write_bytes(header + data_b)
+    file2 = path_bin.parent / (path_bin.stem + '_h512.dat')
+    file2.write_bytes(header_512 + data_b)
 
 
 if __name__ == "__main__":
+    path_bin = Path("F:\\Work\\FPGA\\Test\\Vivado\\test_update\\test_update.vitis\\upgrade_system\\Debug\\sd_card\\BOOT.BIN")
+
     # test1()
     # test2()
-    test3()
+    test3(path_bin)
     pass