96 lines
3.4 KiB
Python
96 lines
3.4 KiB
Python
#!/usr/local/bin/python3.7
|
||
# -*- coding: utf-8 -*-
|
||
|
||
import os
|
||
import sys
|
||
import time
|
||
import struct
|
||
import shutil
|
||
|
||
"""
|
||
介绍：
|
||
本脚本主要实现调用mkyaffs2image工具生成文件镜像
|
||
请注意使用本脚本时支持的python版本，本脚本只支持python3版本。
|
||
请根据实际情况配置对应信息，主要是nandflash的信息，以及分区信息
|
||
|
||
"""
|
||
|
||
root_path = os.path.join(os.path.split(os.path.realpath(__file__))[0], '..')
|
||
cur_dir = os.path.dirname(os.path.abspath(__file__))
|
||
image_tool_path = os.path.realpath(os.path.join(root_path, 'tools', 'bin', 'platform_tool', 'mkyaffs2image'))
|
||
|
||
# 配置生成文件镜像工具路径
|
||
mkyaffstool = os.path.join(image_tool_path, 'mkyaffs2image')
|
||
image_name = "file"
|
||
|
||
# 配置nandflash信息
|
||
cfg_page_size = 2048
|
||
cfg_obb_size = 64
|
||
|
||
# 配置分区信息，请根据实际情况进行配置，添加格式为[分区路径， 分区大小]
|
||
# 分区大小需要根据按照百分系统分配好后的blocks数进行计算，大小=blocks（文件块个数）*pages_per_block（每个文件块包含的页数）*bytes_per_page（每页包含字节数）
|
||
# 前面按照比例分好分区，最后分区置成null，即代表剩下所有blocks
|
||
partition_list = [
|
||
[os.path.join(cur_dir, 'fs_origin', 'system'), "0xcc0000"],
|
||
[os.path.join(cur_dir, 'fs_origin', 'update'), "0xcc0000"],
|
||
[os.path.join(cur_dir, 'fs_origin', 'music'), "0xcc0000"],
|
||
[os.path.join(cur_dir, 'fs_origin', 'user'), "null"]
|
||
]
|
||
|
||
# 将生成image文件转换大小端，保存成另外文件
|
||
def convert(src_file, dest_file):
|
||
fp_src = open(src_file, 'rb')
|
||
fp_dest = open(dest_file, 'wb')
|
||
src_size = os.path.getsize(src_file)
|
||
for i in range(int(src_size / 4)):
|
||
data = fp_src.read(4)
|
||
rw = struct.pack('BBBB', data[3], data[2], data[1], data[0])
|
||
fp_dest.write(rw)
|
||
fp_src.close()
|
||
fp_dest.close()
|
||
|
||
def get_page_type(page_size):
|
||
if page_size == 2048:
|
||
return " 1"
|
||
elif page_size == 4096:
|
||
return " 2"
|
||
else:
|
||
return " 0"
|
||
|
||
def get_obb_type(obb_size):
|
||
if obb_size == 64:
|
||
return " 2"
|
||
elif page_size == 128:
|
||
return " 4"
|
||
else:
|
||
return " 0"
|
||
|
||
if __name__ == "__main__":
|
||
image_file = os.path.join(cur_dir, "%s.bin" %image_name)
|
||
tmp_file = os.path.join(cur_dir, "%s_tmp.bin" %image_name)
|
||
|
||
if sys.platform == 'linux2' or sys.platform == 'linux':
|
||
partition_total = 0
|
||
page_type = get_page_type(cfg_page_size)
|
||
obb_type = get_obb_type(cfg_obb_size)
|
||
|
||
if os.path.exists(tmp_file):
|
||
os.remove(tmp_file)
|
||
if os.path.exists(image_file):
|
||
os.remove(image_file)
|
||
|
||
for partition in partition_list:
|
||
if (partition[1] != "null"):
|
||
partition_total = partition_total + int(partition[1][2:], 16)
|
||
command = mkyaffstool + " " + partition[0] + " " + tmp_file + page_type + obb_type + (" 0x%x" %partition_total)
|
||
else:
|
||
command = mkyaffstool + " " + partition[0] + " " + tmp_file + page_type + obb_type
|
||
print(command)
|
||
os.system(command)
|
||
convert(tmp_file, image_file)
|
||
|
||
if (len(sys.argv) == 2 and os.path.exists(image_file)):
|
||
out_path = os.path.join(sys.argv[1], "%s.bin" %image_name)
|
||
shutil.copyfile(image_file, out_path)
|
||
|
||
print("make yaffs image file over!") |