初始提交

build/script/param_packet.py

@@ -0,0 +1,84 @@
+#!/usr/bin/env python3
+# encoding=utf-8
+# Copyright (c) HiSilicon (Shanghai) Technologies Co., Ltd. 2022-2022. All rights reserved.
+
+import os
+import sys
+import struct
+sys.path.append(os.path.dirname(__file__))
+from conf_parser import BuildConfParser, ParserError
+from utils.build_utils import fn_str_to_int
+
+##
+##  Params Area Stucture
+##
+##  Image_ID:       4B
+##  Struct_Version: 4B
+##  Version:        4B
+##  Item_Offset:    2B
+##  Item_Count:     1B
+##  ItemID_List:    Item_Count
+##  Padding_Area:   Item_Offset - 15 - Item_Count
+##  Item_List:      (4B + 4B) * Item_Count
+##
+def param_area(dst_file, image_id, stru_ver, version, param_info) -> None:
+
+    item_num = len(param_info)
+    item_offset = 15 + item_num
+    item_offset = int((item_offset + 3) / 4) * 4
+
+    ## image_id, stru_ver, version, item_offset, item_num
+    bin_size = 1024
+
+    binary = struct.pack("<IIIHB", image_id, stru_ver, version, item_offset, item_num)
+
+    ## item id list
+    for item in param_info:
+        binary += struct.pack("<B", item[0])
+
+    ## pad for 4-byte-align
+    for i in range(0, item_offset - 15 - item_num):
+        binary += struct.pack("<B", 0x00)
+
+    ## item list
+    for item in param_info:
+        binary += struct.pack("<II", item[1], item[2])
+
+    binary += bytearray(bin_size - len(binary))
+    # print("len", len(binary), binary)
+
+    ## write to file
+    with open(dst_file, "wb") as f:
+        f.write(binary)
+
+    print("Build params success!")
+
+def gen_flash_part_bin(param_file, dst_file):
+    partition_cfg = BuildConfParser(param_file).get_conf_data().get("Partition_Tbl")
+    if partition_cfg is None:
+        msg = "[error] flash partiton config NULL:[%s]!" % param_file
+        raise ParserError(msg)
+    image_id = partition_cfg.get("image_id")
+    stru_ver = partition_cfg.get("stru_ver")
+    version  = partition_cfg.get("version")
+    param_info = partition_cfg.get("param_info")
+    if image_id is None or stru_ver is None or version is None or param_info is None:
+        msg = "[error] flash partiton config err:[%s]!" % param_file
+        raise ParserError(msg)
+
+    table = []
+    for param in param_info:
+        if len(param) != 3:
+            msg = "[error] flash partiton table err:[%s]!" % param_file
+            raise ParserError(msg)
+        table.append([fn_str_to_int(x) for x in param])
+    param_area(dst_file, fn_str_to_int(image_id), fn_str_to_int(stru_ver), fn_str_to_int(version), table)
+
+## python3 param_packet.py XXX.json YYY.bin
+def test():
+    param_file = sys.argv[1]
+    dst_file = sys.argv[2]
+    gen_flash_part_bin(param_file, dst_file)
+
+if __name__ == '__main__':
+    test()