2206 lines
100 KiB
Python
2206 lines
100 KiB
Python
# decompyle3 version 3.9.0
|
|
# Python bytecode version base 3.7.0 (3394)
|
|
# Decompiled from: Python 3.7.16 (default, Mar 30 2023, 01:25:49)
|
|
# [GCC 12.2.1 20220924]
|
|
# Embedded file name: libs/base/bflb_base_eflash_loader.py
|
|
__doc__ = '\nCreated on 20220908\n\n@author: Dillon\n'
|
|
import os, lzma, ecdsa, config as gol
|
|
from libs.bflb_utils import *
|
|
from libs.bflb_configobj import BFConfigParser
|
|
from libs import bflb_ecdh
|
|
from libs import bflb_interface_jlink
|
|
from libs import bflb_interface_cklink
|
|
from libs import bflb_interface_openocd
|
|
from libs.base import bflb_img_loader
|
|
from libs.base import bflb_flash_select
|
|
from libs.base import bflb_efuse_boothd_create
|
|
from libs.base import bflb_img_create as img_create
|
|
from pickle import NONE
|
|
FLASH_LOAD_SHAKE_HAND = 'Flash load shake hand'
|
|
FLASH_ERASE_SHAKE_HAND = 'Flash erase shake hand'
|
|
|
|
class BaseEflashLoader(object):
|
|
__doc__ = '\n Load the flash base execution file.\n '
|
|
|
|
def __init__(self, chip_type, args, config, callback, macaddr_callback, create_simple_callback, create_img_callback, task_num):
|
|
self.chip_type = chip_type
|
|
self.chip_name = ''
|
|
self.args = args
|
|
self.config = config
|
|
self.callback = callback
|
|
self.macaddr_callback = macaddr_callback
|
|
self.create_simple_callback = create_simple_callback
|
|
self.create_img_callback = create_img_callback
|
|
self.temp_task_num = task_num
|
|
self.task_num = None
|
|
self.device = ''
|
|
self.speed = 0
|
|
self.boot_speed = 0
|
|
self.bflb_serial_object = None
|
|
self.start = ''
|
|
self.end = ''
|
|
self.file = ''
|
|
self.address = ''
|
|
self.macaddr = ''
|
|
self.bootinfo = None
|
|
self.flash_set = 0
|
|
self.read_flash_id = 0
|
|
self.id_valid_flag = '80'
|
|
self.read_flash2_id = 0
|
|
self.id2_valid_flag = '80'
|
|
self.do_reset = True
|
|
self.reset_hold_time = 100
|
|
self.shake_hand_delay = 100
|
|
self.reset_revert = True
|
|
self.cutoff_time = 0
|
|
self.shake_hand_retry = 2
|
|
self.flash_burn_retry = 1
|
|
self.ram_load = False
|
|
self.load_function = 1
|
|
self.macaddr_check = False
|
|
self.NUM_ERR = 5
|
|
self.cfg = ''
|
|
self.eflash_loader_file = ''
|
|
self.cpu_reset = False
|
|
self.retry_delay_after_cpu_reset = 0
|
|
self.input_macaddr = ''
|
|
self.isp_mode_sign = False
|
|
self.create_cfg = None
|
|
self._skip_addr = []
|
|
self._skip_len = []
|
|
self.address_list = []
|
|
self.flash_file_list = []
|
|
self.encrypt_key = None
|
|
self.encrypt_iv = None
|
|
self.public_key = None
|
|
self.private_key = None
|
|
self.load_file = ''
|
|
self._mac_addr = bytearray(0)
|
|
self._need_shake_hand = True
|
|
self._isp_shakehand_timeout = 0
|
|
self._macaddr_check = bytearray(0)
|
|
self._default_time_out = 2.0
|
|
self._flash2_en = False
|
|
self._flash1_size = 0
|
|
self._flash2_size = 0
|
|
self._flash2_select = False
|
|
self._efuse_bootheader_file = ''
|
|
self._img_create_file = ''
|
|
self._com_cmds = {'change_rate':{'cmd_id':'20',
|
|
'data_len':'0008',
|
|
'callback':None},
|
|
'reset':{'cmd_id':'21',
|
|
'data_len':'0000',
|
|
'callback':None},
|
|
'clk_set':{'cmd_id':'22',
|
|
'data_len':'0000',
|
|
'callback':None},
|
|
'opt_finish':{'cmd_id':'23',
|
|
'data_len':'0000',
|
|
'callback':None},
|
|
'flash_erase':{'cmd_id':'30',
|
|
'data_len':'0000',
|
|
'callback':None},
|
|
'flash_write':{'cmd_id':'31',
|
|
'data_len':'0100',
|
|
'callback':None},
|
|
'flash_read':{'cmd_id':'32',
|
|
'data_len':'0100',
|
|
'callback':None},
|
|
'flash_boot':{'cmd_id':'33',
|
|
'data_len':'0000',
|
|
'callback':None},
|
|
'flash_xip_read':{'cmd_id':'34',
|
|
'data_len':'0100',
|
|
'callback':None},
|
|
'flash_switch_bank':{'cmd_id':'35',
|
|
'data_len':'0100',
|
|
'callback':None},
|
|
'flash_read_jid':{'cmd_id':'36',
|
|
'data_len':'0000',
|
|
'callback':None},
|
|
'flash_read_status_reg':{'cmd_id':'37',
|
|
'data_len':'0000',
|
|
'callback':None},
|
|
'flash_write_status_reg':{'cmd_id':'38',
|
|
'data_len':'0000',
|
|
'callback':None},
|
|
'flash_write_check':{'cmd_id':'3a',
|
|
'data_len':'0000',
|
|
'callback':None},
|
|
'flash_set_para':{'cmd_id':'3b',
|
|
'data_len':'0000',
|
|
'callback':None},
|
|
'flash_chiperase':{'cmd_id':'3c',
|
|
'data_len':'0000',
|
|
'callback':None},
|
|
'flash_readSha':{'cmd_id':'3d',
|
|
'data_len':'0100',
|
|
'callback':None},
|
|
'flash_xip_readSha':{'cmd_id':'3e',
|
|
'data_len':'0100',
|
|
'callback':None},
|
|
'flash_decompress_write':{'cmd_id':'3f',
|
|
'data_len':'0100',
|
|
'callback':None},
|
|
'efuse_write':{'cmd_id':'40',
|
|
'data_len':'0080',
|
|
'callback':None},
|
|
'efuse_read':{'cmd_id':'41',
|
|
'data_len':'0000',
|
|
'callback':None},
|
|
'efuse_read_mac':{'cmd_id':'42',
|
|
'data_len':'0000',
|
|
'callback':None},
|
|
'efuse_write_mac':{'cmd_id':'43',
|
|
'data_len':'0006',
|
|
'callback':None},
|
|
'flash_xip_read_start':{'cmd_id':'60',
|
|
'data_len':'0080',
|
|
'callback':None},
|
|
'flash_xip_read_finish':{'cmd_id':'61',
|
|
'data_len':'0000',
|
|
'callback':None},
|
|
'log_read':{'cmd_id':'71',
|
|
'data_len':'0000',
|
|
'callback':None},
|
|
'efuse_security_write':{'cmd_id':'80',
|
|
'data_len':'0080',
|
|
'callback':None},
|
|
'efuse_security_read':{'cmd_id':'81',
|
|
'data_len':'0000',
|
|
'callback':None},
|
|
'ecdh_get_pk':{'cmd_id':'90',
|
|
'data_len':'0000',
|
|
'callback':None},
|
|
'ecdh_chanllenge':{'cmd_id':'91',
|
|
'data_len':'0000',
|
|
'callback':None}}
|
|
self._resp_cmds = [
|
|
'flash_read','flash_xip_read','efuse_read','efuse_read_mac','flash_readSha','flash_xip_readSha','flash_read_jid',
|
|
'flash_read_status_reg','log_read','ecdh_get_pk','ecdh_chanllenge','efuse_security_read']
|
|
|
|
def first_run_step_load_parameter(self):
|
|
"""
|
|
First step: init.
|
|
"""
|
|
if self.temp_task_num == None:
|
|
local_log_enable(True)
|
|
if self.temp_task_num != None:
|
|
if self.temp_task_num > 256:
|
|
self.task_num = self.temp_task_num - 256
|
|
else:
|
|
self.task_num = self.temp_task_num
|
|
else:
|
|
self.task_num = None
|
|
printf('========= eflash loader cmd arguments =========')
|
|
self.interface = self.config['param']['interface_type'].lower()
|
|
self.chip_name = self.config['param']['chip_name']
|
|
self.config_file = os.path.join(app_path, 'chips', self.chip_name.lower(), 'eflash_loader', 'eflash_loader_cfg.ini')
|
|
if not os.path.exists(self.config_file):
|
|
conf_file = self.config_file.replace('.ini', '.conf')
|
|
if os.path.exists(conf_file):
|
|
shutil.copy(conf_file, self.config_file)
|
|
if os.path.exists(self.config_file):
|
|
self.cfg = BFConfigParser()
|
|
self.cfg.read(self.config_file)
|
|
else:
|
|
printf('Config file "' + self.config_file + '" not found')
|
|
self.error_code_print('000B')
|
|
return (False, 0)
|
|
if self.interface == 'openocd':
|
|
self.device = self.cfg.get('LOAD_CFG', 'openocd_config')
|
|
self._bflb_sn_device = self.config['param']['comport_uart']
|
|
else:
|
|
if self.interface == 'cklink':
|
|
self.device = self.cfg.get('LOAD_CFG', 'cklink_vidpid')
|
|
self._bflb_sn_device = self.cfg.get('LOAD_CFG', 'cklink_type') + ' ' + self.config['param']['comport_uart']
|
|
else:
|
|
self.device = self.config['param']['comport_uart']
|
|
if 'aes_key' in self.config['param']:
|
|
if self.config['check_box']['encrypt']:
|
|
self.encrypt_key = self.config['param']['aes_key']
|
|
if 'aes_iv' in self.config['param']:
|
|
if self.config['check_box']['sign']:
|
|
self.encrypt_iv = self.config['param']['aes_iv']
|
|
if 'input_path' in self.config:
|
|
if 'publickey' in self.config['input_path']['publickey']:
|
|
self.public_key = self.config['input_path']['publickey']
|
|
if 'input_path' in self.config:
|
|
if 'privatekey' in self.config['input_path']['privatekey']:
|
|
self.private_key = self.config['input_path']['privatekey']
|
|
xtal_type = self.config['param']['chip_xtal']
|
|
if self.interface == 'uart':
|
|
temp_speed = int(self.config['param']['speed_uart'])
|
|
else:
|
|
temp_speed = int(self.config['param']['speed_jlink'])
|
|
printf('serial port is ', self.device)
|
|
printf('chiptype: ', self.chip_type)
|
|
try:
|
|
if self.args.start:
|
|
self.start = self.args.start
|
|
if self.args.end:
|
|
self.end = self.args.end
|
|
if self.args.file:
|
|
self.file = self.args.file
|
|
if self.args.addr:
|
|
self.address = self.args.addr
|
|
if self.args.mac:
|
|
self.macaddr = self.args.mac
|
|
if self.args.createcfg:
|
|
self.create_cfg = self.args.createcfg
|
|
if self.args.loadfile:
|
|
self.load_file = self.args.loadfile
|
|
if self.args.usage:
|
|
printf('-e --start=00000000 --end=0000FFFF -c config.ini')
|
|
printf('-w --flash -c config.ini')
|
|
printf('-w --flash --file=1.bin,2.bin --addr=00000000,00001000 -c config.ini')
|
|
printf('-r --flash --start=00000000 --end=0000FFFF --file=flash.bin -c config.ini')
|
|
except Exception as e:
|
|
try:
|
|
printf(e)
|
|
self.error_code_print('0002')
|
|
return (False, 0)
|
|
finally:
|
|
e = None
|
|
del e
|
|
|
|
if self.cfg.has_option('LOAD_CFG', 'verify'):
|
|
if self.cfg.get('LOAD_CFG', 'verify') == '1':
|
|
self.verify = 1
|
|
else:
|
|
self.verify = 0
|
|
else:
|
|
self.verify = 0
|
|
try:
|
|
self.erase = int(self.config['param']['erase'])
|
|
except:
|
|
self.erase = 1
|
|
|
|
try:
|
|
boot2_isp_mode = int(self.config['param']['boot2_isp_mode'])
|
|
if int(boot2_isp_mode) == 1:
|
|
self.isp_mode_sign = True
|
|
except:
|
|
pass
|
|
|
|
if not 'skip_mode' in self.config['param'] or self.config['param']['skip_mode']:
|
|
skip_para = self.config['param']['skip_mode'].replace(' ', '')
|
|
if skip_para[-1] == ';':
|
|
skip_para = skip_para[:-1]
|
|
skip_para_list = skip_para.split(';')
|
|
for temp_value in skip_para_list:
|
|
temp_list = temp_value.split(',')
|
|
if temp_list[0][0:2] == '0x':
|
|
self._skip_addr.append(int(temp_list[0][2:], 16))
|
|
else:
|
|
self._skip_addr.append(int(temp_list[0], 10))
|
|
if temp_list[1][0:2] == '0x':
|
|
self._skip_len.append(int(temp_list[1][2:], 16))
|
|
else:
|
|
self._skip_len.append(int(temp_list[1], 10))
|
|
|
|
if len(self._skip_len) > 1 or len(self._skip_len) == 1 and self._skip_len[0] > 0:
|
|
if self.erase == 2:
|
|
printf('Error: skip mode can not set flash chiperase!')
|
|
self.error_code_print('0044')
|
|
return (False, 0)
|
|
if self.cfg.has_option('LOAD_CFG', 'local_log'):
|
|
if self.cfg.get('LOAD_CFG', 'local_log') == 'true':
|
|
printf('local log enable')
|
|
local_log_enable(True)
|
|
self.input_macaddr = self.macaddr
|
|
else:
|
|
local_log_enable(False)
|
|
self.input_macaddr = ''
|
|
if self.interface == 'cklink':
|
|
self._bflb_com_tx_size = 14344
|
|
else:
|
|
self._bflb_com_tx_size = int(self.cfg.get('LOAD_CFG', 'tx_size'))
|
|
if self.cfg.has_option('LOAD_CFG', 'erase_time_out'):
|
|
self._erase_time_out = int(self.cfg.get('LOAD_CFG', 'erase_time_out'))
|
|
if self.cfg.has_option('LOAD_CFG', 'shake_hand_retry'):
|
|
self.shake_hand_retry = int(self.cfg.get('LOAD_CFG', 'shake_hand_retry'))
|
|
if self.cfg.has_option('LOAD_CFG', 'flash_burn_retry'):
|
|
self.flash_burn_retry = int(self.cfg.get('LOAD_CFG', 'flash_burn_retry'))
|
|
if self.cfg.has_option('LOAD_CFG', 'checksum_err_retry'):
|
|
self._checksum_err_retry_limit = int(self.cfg.get('LOAD_CFG', 'checksum_err_retry'))
|
|
if self.cfg.has_option('LOAD_CFG', 'cpu_reset_after_load'):
|
|
self.cpu_reset = self.cfg.get('LOAD_CFG', 'cpu_reset_after_load') == 'true'
|
|
if self.cfg.has_option('LOAD_CFG', 'retry_delay_after_cpu_reset'):
|
|
self.retry_delay_after_cpu_reset = int(self.cfg.get('LOAD_CFG', 'retry_delay_after_cpu_reset'))
|
|
printf('retry delay: ', self.retry_delay_after_cpu_reset)
|
|
if self.cfg.has_option('LOAD_CFG', 'eflash_loader_file'):
|
|
if self.eflash_loader_file is None:
|
|
self.eflash_loader_file = self.cfg.get('LOAD_CFG', 'eflash_loader_file')
|
|
printf('cpu_reset=', self.cpu_reset)
|
|
if xtal_type != '':
|
|
self.eflash_loader_file = 'chips/' + self.chip_name.lower() + '/eflash_loader/eflash_loader_' + xtal_type.replace('.', 'p').lower() + '.bin'
|
|
if self.load_file and not self.eflash_loader_file:
|
|
self.eflash_loader_file = self.load_file
|
|
else:
|
|
if self.eflash_loader_file is not None:
|
|
self.eflash_loader_file = os.path.join(app_path, self.eflash_loader_file)
|
|
self.load_function = 1
|
|
if self.cfg.has_option('LOAD_CFG', 'isp_shakehand_timeout'):
|
|
self._isp_shakehand_timeout = int(self.cfg.get('LOAD_CFG', 'isp_shakehand_timeout'))
|
|
result, address, flash_file = self.get_flash_file_and_address()
|
|
if not result:
|
|
return (False, self.flash_burn_retry)
|
|
temp_file_list = []
|
|
for one in flash_file:
|
|
if os.path.join(chip_path, self.chip_name) in one:
|
|
temp_file_list.append(one.replace(app_path, '')[1:])
|
|
else:
|
|
temp_file_list.append(os.path.join('chips', self.chip_name, 'img_create', os.path.basename(one)))
|
|
|
|
if 'erase' in self.config['param']:
|
|
if self.config['param']['erase']:
|
|
update_cfg(self.cfg, 'LOAD_CFG', 'erase', str(self.erase))
|
|
if 'skip_mode' in self.config['param']:
|
|
if self.config['param']['skip_mode']:
|
|
update_cfg(self.cfg, 'LOAD_CFG', 'skip_mode', self.config['param']['skip_mode'])
|
|
if 'boot2_isp_mode' in self.config['param']:
|
|
if self.config['param']['boot2_isp_mode']:
|
|
update_cfg(self.cfg, 'LOAD_CFG', 'boot2_isp_mode', self.config['param']['boot2_isp_mode'])
|
|
update_cfg(self.cfg, 'FLASH_CFG', 'file', ' '.join(temp_file_list))
|
|
update_cfg(self.cfg, 'FLASH_CFG', 'address', ' '.join(address))
|
|
self.cfg.write(self.config_file, 'w+')
|
|
with open((self.config_file), 'r', encoding='utf-8') as cf_file:
|
|
cf_context = cf_file.read().replace('"', '')
|
|
with open((self.config_file), 'w', encoding='utf-8') as cf_file:
|
|
cf_file.write(cf_context)
|
|
self.bl_write_flash_img(address, flash_file)
|
|
self.address_list = address[:]
|
|
self.flash_file_list = flash_file[:]
|
|
if self.args.efuse:
|
|
try:
|
|
if self.config['input_path']['efuse']:
|
|
if self.config['check_box']['efuse']:
|
|
efusefile = self.config['input_path']['efuse']
|
|
except:
|
|
efusefile = ''
|
|
|
|
if efusefile:
|
|
efuse_file = efusefile
|
|
mask_file = efuse_file.replace('.bin', '_mask.bin')
|
|
relpath_efuse_file = os.path.relpath(os.path.join('chips', self.chip_name, 'img_create', os.path.basename(efuse_file)))
|
|
relpath_mask_file = os.path.relpath(os.path.join('chips', self.chip_name, 'img_create', os.path.basename(mask_file)))
|
|
if self.cfg.has_option('EFUSE_CFG', 'file'):
|
|
update_cfg(self.cfg, 'EFUSE_CFG', 'file', relpath_efuse_file)
|
|
update_cfg(self.cfg, 'EFUSE_CFG', 'maskfile', relpath_mask_file)
|
|
self.cfg.write(self.config_file, 'w+')
|
|
with open((self.config_file), 'r', encoding='utf-8') as cf_file:
|
|
cf_context = cf_file.read().replace('"', '')
|
|
with open((self.config_file), 'w', encoding='utf-8') as cf_file:
|
|
cf_file.write(cf_context)
|
|
else:
|
|
efuse_file = self.cfg.get('EFUSE_CFG', 'file')
|
|
mask_file = self.cfg.get('EFUSE_CFG', 'maskfile')
|
|
if self.temp_task_num != None:
|
|
efuse_file = 'task' + str(self.temp_task_num) + '/' + efuse_file
|
|
ret = img_create.compress_dir(self.chip_name, 'img_create', self.args.efuse, address, flash_file, efuse_file, mask_file)
|
|
if ret is not True:
|
|
return errorcode_msg()
|
|
else:
|
|
ret = img_create.compress_dir(self.chip_name, 'img_create', False, address, flash_file)
|
|
if ret is not True:
|
|
return errorcode_msg()
|
|
if self.interface == 'uart' or self.interface == 'sdio':
|
|
if temp_speed:
|
|
self.speed = temp_speed
|
|
self.boot_speed = int(self.cfg.get('LOAD_CFG', 'speed_uart_boot'))
|
|
self.set_boot_speed()
|
|
if self.cfg.has_option('LOAD_CFG', 'reset_hold_time'):
|
|
self.reset_hold_time = int(self.cfg.get('LOAD_CFG', 'reset_hold_time'))
|
|
if self.cfg.has_option('LOAD_CFG', 'shake_hand_delay'):
|
|
self.shake_hand_delay = int(self.cfg.get('LOAD_CFG', 'shake_hand_delay'))
|
|
if self.cfg.has_option('LOAD_CFG', 'do_reset'):
|
|
self.do_reset = self.cfg.get('LOAD_CFG', 'do_reset') == 'true'
|
|
if self.cfg.has_option('LOAD_CFG', 'reset_revert'):
|
|
self.reset_revert = self.cfg.get('LOAD_CFG', 'reset_revert') == 'true'
|
|
if self.cfg.has_option('LOAD_CFG', 'cutoff_time'):
|
|
self.cutoff_time = int(self.cfg.get('LOAD_CFG', 'cutoff_time'))
|
|
printf('========= Interface is %s =========' % self.interface)
|
|
self._bflb_com_img_loader = bflb_img_loader.BflbImgLoader(self.device, self.speed, self.boot_speed, self.interface, self.chip_type, self.chip_name, self.eflash_loader_file, '', self.callback, self.do_reset, self.reset_hold_time, self.shake_hand_delay, self.reset_revert, self.cutoff_time, self.shake_hand_retry, self.isp_mode_sign, self._isp_shakehand_timeout, self.encrypt_key, self.encrypt_iv, self.public_key, self.private_key)
|
|
self.bflb_serial_object = self._bflb_com_img_loader.bflb_serial_object
|
|
if not self.cfg.has_option('LOAD_CFG', 'isp_mode_speed') or self.isp_mode_sign is True:
|
|
isp_mode_speed = int(self.cfg.get('LOAD_CFG', 'isp_mode_speed'))
|
|
self._bflb_com_img_loader.set_isp_baudrate(isp_mode_speed)
|
|
else:
|
|
if self.interface == 'jlink':
|
|
printf('========= Interface is JLink =========')
|
|
self.bflb_serial_object = bflb_interface_jlink.BflbJLinkPort()
|
|
if temp_speed:
|
|
self.speed = temp_speed
|
|
printf('com speed: %dk' % self.speed)
|
|
else:
|
|
self.speed = int(self.cfg.get('LOAD_CFG', 'speed_jlink'))
|
|
self.boot_speed = self.speed
|
|
else:
|
|
if self.interface == 'openocd':
|
|
printf('========= Interface is Openocd =========')
|
|
self.bflb_serial_object = bflb_interface_openocd.BflbOpenocdPort()
|
|
if temp_speed:
|
|
self.speed = temp_speed
|
|
printf('com speed: %dk' % self.speed)
|
|
else:
|
|
self.speed = int(self.cfg.get('LOAD_CFG', 'speed_jlink'))
|
|
self.boot_speed = self.speed
|
|
else:
|
|
if self.interface == 'cklink':
|
|
printf('========= Interface is CKLink =========')
|
|
self.bflb_serial_object = bflb_interface_cklink.BflbCKLinkPort()
|
|
if temp_speed:
|
|
self.speed = temp_speed
|
|
printf('com speed: %dk' % self.speed)
|
|
else:
|
|
self.speed = int(self.cfg.get('LOAD_CFG', 'speed_jlink'))
|
|
self.boot_speed = self.speed
|
|
else:
|
|
printf(self.interface + ' is not supported ')
|
|
return (
|
|
False, self.flash_burn_retry)
|
|
if self.args.chipid:
|
|
ret, self.bootinfo, res = self.get_boot_info()
|
|
if ret is False:
|
|
self.error_code_print('0003')
|
|
return (
|
|
False, self.flash_burn_retry)
|
|
return (
|
|
True, self.flash_burn_retry)
|
|
if self.cfg.has_option('LOAD_CFG', 'load_function'):
|
|
self.load_function = int(self.cfg.get('LOAD_CFG', 'load_function'))
|
|
if self.isp_mode_sign is True:
|
|
if self._isp_shakehand_timeout == 0:
|
|
self._isp_shakehand_timeout = 5
|
|
self.set_load_function()
|
|
return (True, 'continue')
|
|
|
|
def second_run_step_shake_hand(self):
|
|
"""
|
|
Second step: shake hand and load eflash_loader.bin.
|
|
"""
|
|
try:
|
|
if self.load_function == 0:
|
|
printf('No need load eflash_loader.bin')
|
|
else:
|
|
if self.load_function == 1:
|
|
load_bin_pass = False
|
|
printf('Eflash load bin file: ', self.eflash_loader_file)
|
|
ret, self.bootinfo, res = self.load_eflash_loader_bin()
|
|
if res == 'shake hand fail':
|
|
self.error_code_print('0050')
|
|
if res.startswith('repeat_burn') is True:
|
|
return ('repeat_burn', self.flash_burn_retry)
|
|
if res.startswith('error_shakehand') is True:
|
|
if self.cpu_reset is True:
|
|
self.error_code_print('0003')
|
|
return (
|
|
False, self.flash_burn_retry)
|
|
load_bin_pass = True
|
|
time.sleep(4.5)
|
|
if ret is False:
|
|
if load_bin_pass == False:
|
|
self.error_code_print('0003')
|
|
return (
|
|
False, self.flash_burn_retry)
|
|
if self.ram_load:
|
|
return (True, self.flash_burn_retry)
|
|
else:
|
|
if self.load_function == 2:
|
|
load_bin_pass = False
|
|
printf('Bootrom load')
|
|
ret, self.bootinfo, res = self.get_boot_info()
|
|
if res == 'shake hand fail':
|
|
self.error_code_print('0050')
|
|
if res.startswith('repeat_burn') is True:
|
|
self.error_code_print('000A')
|
|
return (
|
|
'repeat_burn', self.flash_burn_retry)
|
|
if res.startswith('error_shakehand') is True:
|
|
if self.cpu_reset is True:
|
|
self.error_code_print('0003')
|
|
return (
|
|
False, self.flash_burn_retry)
|
|
load_bin_pass = True
|
|
time.sleep(4.5)
|
|
if ret is False:
|
|
if load_bin_pass == False:
|
|
self.error_code_print('0050')
|
|
return (
|
|
False, self.flash_burn_retry)
|
|
self._need_shake_hand = False
|
|
clock_para = bytearray(0)
|
|
if self.cfg.has_option('LOAD_CFG', 'clock_para'):
|
|
clock_para_str = self.cfg.get('LOAD_CFG', 'clock_para')
|
|
if clock_para_str != '':
|
|
clock_para_file = os.path.join(app_path, clock_para_str)
|
|
printf('clock para file: ', clock_para_file)
|
|
clock_para = self.clock_para_update(clock_para_file)
|
|
printf('change bdrate: ', self.speed)
|
|
ret = self.clock_pll_set(self._need_shake_hand, True, clock_para)
|
|
if ret is False:
|
|
printf('pll set fail!!')
|
|
return (
|
|
False, self.flash_burn_retry)
|
|
return (True, 'continue')
|
|
except Exception as e:
|
|
try:
|
|
printf(e)
|
|
self.error_code_print('0003')
|
|
return (
|
|
False, self.flash_burn_retry)
|
|
finally:
|
|
e = None
|
|
del e
|
|
|
|
def third_run_step_read_mac_address(self):
|
|
"""
|
|
Third step: read mac address.
|
|
"""
|
|
time.sleep(0.1)
|
|
if self.isp_mode_sign is True:
|
|
if self.cpu_reset is True:
|
|
self.set_clear_boot_status(self._need_shake_hand)
|
|
if self.cfg.has_option('LOAD_CFG', 'check_mac'):
|
|
self.macaddr_check = self.cfg.get('LOAD_CFG', 'check_mac') == 'true'
|
|
if self.macaddr_check:
|
|
if self.isp_mode_sign is False:
|
|
ret, self._mac_addr = self.efuse_read_mac_addr_process()
|
|
if ret is False:
|
|
printf('read mac addr fail!!')
|
|
return (
|
|
False, self.flash_burn_retry)
|
|
if self._mac_addr == self._macaddr_check:
|
|
self.error_code_print('000A')
|
|
return (
|
|
False, self.flash_burn_retry)
|
|
self._need_shake_hand = False
|
|
self._macaddr_check_status = True
|
|
if self.macaddr_callback is not None:
|
|
ret, self._efuse_data, self._efuse_mask_data, macaddr = self.macaddr_callback(binascii.hexlify(self._mac_addr).decode('utf-8'))
|
|
if ret is False:
|
|
return (False, self.flash_burn_retry)
|
|
if self._efuse_data != bytearray(0) and self._efuse_mask_data != bytearray(0) or macaddr != '':
|
|
self.args.efuse = True
|
|
if self.callback:
|
|
self.callback(0, 100, 'running', 'blue')
|
|
return (True, 'continue')
|
|
|
|
def fourth_run_step_interact_chip(self):
|
|
"""
|
|
Fourth step: Interact with chip, read chip ID and update flash parameter.
|
|
"""
|
|
if self.args.flash:
|
|
flash_pin = 0
|
|
flash_clock_cfg = 0
|
|
flash_io_mode = 0
|
|
flash_clk_delay = 0
|
|
if self.cfg.has_option('FLASH_CFG', 'decompress_write'):
|
|
self.decompress_write = self.cfg.get('FLASH_CFG', 'decompress_write') == 'true'
|
|
self.set_decompress_write()
|
|
printf('flash set para')
|
|
if self.cfg.get('FLASH_CFG', 'flash_pin'):
|
|
flash_pin_cfg = self.cfg.get('FLASH_CFG', 'flash_pin')
|
|
if flash_pin_cfg.startswith('0x'):
|
|
flash_pin = int(flash_pin_cfg, 16)
|
|
else:
|
|
flash_pin = int(flash_pin_cfg, 10)
|
|
if flash_pin == 128:
|
|
flash_pin = self.get_flash_pin_from_bootinfo(self.chip_type, self.bootinfo)
|
|
printf('get flash pin cfg from bootinfo: 0x%02X' % flash_pin)
|
|
else:
|
|
flash_pin = self.get_flash_pin()
|
|
if self.cfg.has_option('FLASH_CFG', 'flash_clock_cfg'):
|
|
clock_div_cfg = self.cfg.get('FLASH_CFG', 'flash_clock_cfg')
|
|
if clock_div_cfg.startswith('0x'):
|
|
flash_clock_cfg = int(clock_div_cfg, 16)
|
|
else:
|
|
flash_clock_cfg = int(clock_div_cfg, 10)
|
|
if self.cfg.has_option('FLASH_CFG', 'flash_io_mode'):
|
|
io_mode_cfg = self.cfg.get('FLASH_CFG', 'flash_io_mode')
|
|
if io_mode_cfg.startswith('0x'):
|
|
flash_io_mode = int(io_mode_cfg, 16)
|
|
else:
|
|
flash_io_mode = int(io_mode_cfg, 10)
|
|
if self.cfg.has_option('FLASH_CFG', 'flash_clock_delay'):
|
|
clk_delay_cfg = self.cfg.get('FLASH_CFG', 'flash_clock_delay')
|
|
if clk_delay_cfg.startswith('0x'):
|
|
flash_clk_delay = int(clk_delay_cfg, 16)
|
|
else:
|
|
flash_clk_delay = int(clk_delay_cfg, 10)
|
|
self.flash_set = (flash_pin << 0) + (flash_clock_cfg << 8) + (flash_io_mode << 16) + (flash_clk_delay << 24)
|
|
if self.flash_set != 66047 or self.load_function == 2:
|
|
printf('set flash cfg: %X' % self.flash_set)
|
|
ret = self.flash_set_para_main_process(self.flash_set, bytearray(0))
|
|
self._need_shake_hand = False
|
|
if ret is False:
|
|
return (False, self.flash_burn_retry)
|
|
ret, data = self.flash_read_jedec_id_process()
|
|
if ret:
|
|
self._need_shake_hand = False
|
|
data = binascii.hexlify(data).decode('utf-8')
|
|
self.id_valid_flag = data[6:]
|
|
read_id = data[0:6]
|
|
self.read_flash_id = read_id
|
|
if self.cfg.has_option('FLASH_CFG', 'flash_para'):
|
|
flash_para_file = os.path.join(app_path, self.cfg.get('FLASH_CFG', 'flash_para'))
|
|
self.flash_para_update(flash_para_file, read_id)
|
|
if self.id_valid_flag != '80':
|
|
if not self.show_identify_fail():
|
|
return (False, self.flash_burn_retry)
|
|
if self.is_conf_exist(self.read_flash_id) is False:
|
|
self.error_code_print('003D')
|
|
return (
|
|
False, self.flash_burn_retry)
|
|
else:
|
|
self.error_code_print('0030')
|
|
return (
|
|
False, self.flash_burn_retry)
|
|
result, content = self.run_flash2()
|
|
return (
|
|
result, content)
|
|
return (True, 'continue')
|
|
|
|
def fifth_run_step_write_flash_and_check(self):
|
|
"""
|
|
Fifth step: write flash and check.
|
|
"""
|
|
if self.args.none:
|
|
return (True, self.flash_burn_retry)
|
|
if self.args.write:
|
|
if not self.args.flash:
|
|
if not self.args.efuse:
|
|
printf('No target select')
|
|
return (
|
|
False, self.flash_burn_retry)
|
|
printf('Program operation')
|
|
if self.args.flash:
|
|
flash_para_file = ''
|
|
flash2_para_file = ''
|
|
if self.cfg.has_option('FLASH_CFG', 'flash_para'):
|
|
flash_para_file = os.path.join(app_path, self.cfg.get('FLASH_CFG', 'flash_para'))
|
|
if self.cfg.has_option('FLASH2_CFG', 'flash2_para'):
|
|
flash2_para_file = os.path.join(app_path, self.cfg.get('FLASH2_CFG', 'flash2_para'))
|
|
address = self.address_list
|
|
flash_file = self.flash_file_list
|
|
if self.erase == 2:
|
|
ret = self.flash_chiperase_main_process()
|
|
if ret is False:
|
|
return (False, self.flash_burn_retry)
|
|
self._need_shake_hand = False
|
|
self.erase = 0
|
|
if len(flash_file) > 0:
|
|
size_before = 0
|
|
size_all = 0
|
|
i = 0
|
|
for item in flash_file:
|
|
if self.temp_task_num != None:
|
|
size_all += os.path.getsize(os.path.join(app_path, convert_path('task' + str(self.temp_task_num) + '/' + item)))
|
|
else:
|
|
size_all += os.path.getsize(os.path.join(app_path, convert_path(item)))
|
|
|
|
try:
|
|
ret = False
|
|
while i < len(flash_file):
|
|
if self.temp_task_num != None:
|
|
flash_file[i] = 'task' + str(self.temp_task_num) + '/' + flash_file[i]
|
|
size_current = os.path.getsize(os.path.join(app_path, convert_path(flash_file[i])))
|
|
else:
|
|
size_current = os.path.getsize(os.path.join(app_path, convert_path(flash_file[i])))
|
|
if self.callback:
|
|
self.callback(size_before, size_all, 'program1')
|
|
else:
|
|
if self.callback:
|
|
self.callback(size_current, size_all, 'program2')
|
|
printf('Dealing Index ', i)
|
|
if self.isp_mode_sign is True:
|
|
printf('========= programming ', convert_path(flash_file[i]))
|
|
else:
|
|
printf('========= programming ', convert_path(flash_file[i]), ' to 0x', address[i])
|
|
flash1_bin = ''
|
|
flash1_bin_len = 0
|
|
flash2_bin = ''
|
|
flash2_bin_len = 0
|
|
flash1_bin, flash1_bin_len, flash2_bin, flash2_bin_len = self.get_flash1_and_flash2(flash_file, address, size_current, i)
|
|
if flash1_bin != '' and flash2_bin != '':
|
|
ret = self.flash_cfg_option(self.read_flash_id, flash_para_file, self.flash_set, self.id_valid_flag, flash1_bin, self.config_file, self.cfg, self.create_img_callback, self.create_simple_callback)
|
|
if ret is False:
|
|
return (False, self.flash_burn_retry)
|
|
printf('========= programming ', convert_path(flash1_bin), ' to 0x', address[i])
|
|
ret = self.flash_load_specified(convert_path(flash1_bin), int(address[i], 16), self.callback)
|
|
if ret is False:
|
|
return (False, self.flash_burn_retry)
|
|
ret = self.flash_switch_bank_process(1)
|
|
self._need_shake_hand = False
|
|
if ret is False:
|
|
return (False, self.flash_burn_retry)
|
|
ret = self.flash_cfg_option(self.read_flash2_id, flash2_para_file, self.flash2_set, self.id2_valid_flag, flash_file[i], self.config_file, self.cfg, self.create_img_callback, self.create_simple_callback)
|
|
if ret is False:
|
|
return (False, self.flash_burn_retry)
|
|
printf('========= programming ', convert_path(flash2_bin), ' to 0x%08X' % (int(address[i], 16) + flash1_bin_len))
|
|
ret = self.flash_load_specified(convert_path(flash2_bin), int(address[i], 16) + flash1_bin_len, self.callback)
|
|
if ret is False:
|
|
return (False, self.flash_burn_retry)
|
|
else:
|
|
if not (self._flash2_en is False or self._flash2_select) is False or int(address[i], 16) < self._flash1_size:
|
|
ret = self.flash_cfg_option(self.read_flash_id, flash_para_file, self.flash_set, self.id_valid_flag, flash_file[i], self.config_file, self.cfg, self.create_img_callback, self.create_simple_callback)
|
|
if ret is False:
|
|
return (False, self.flash_burn_retry)
|
|
else:
|
|
if self._flash2_select is False:
|
|
if int(address[i], 16) >= self._flash1_size:
|
|
ret = self.flash_switch_bank_process(1)
|
|
self._need_shake_hand = False
|
|
if ret is False:
|
|
return (False, self.flash_burn_retry)
|
|
ret = self.flash_cfg_option(self.read_flash2_id, flash2_para_file, self.flash2_set, self.id2_valid_flag, flash_file[i], self.config_file, self.cfg, self.create_img_callback, self.create_simple_callback)
|
|
if ret is False:
|
|
return (False, self.flash_burn_retry)
|
|
if 'eflash_loader_cfg.ini' in flash_file[i]:
|
|
print('11111')
|
|
ret = self.flash_load_specified(convert_path(flash_file[i]), int(address[i], 16), self.callback)
|
|
if ret is False:
|
|
return (False, self.flash_burn_retry)
|
|
size_before += os.path.getsize(os.path.join(app_path, convert_path(flash_file[i])))
|
|
i += 1
|
|
if self.callback:
|
|
self.callback(i, len(flash_file), 'program')
|
|
self._need_shake_hand = False
|
|
|
|
if self._flash2_select is True:
|
|
ret = self.flash_switch_bank_process(0)
|
|
self._need_shake_hand = False
|
|
if ret is False:
|
|
return (False, self.flash_burn_retry)
|
|
printf('Program Finished')
|
|
except Exception as e:
|
|
try:
|
|
printf(e)
|
|
traceback.print_exc(limit=(self.NUM_ERR), file=(sys.stdout))
|
|
return (
|
|
False, self.flash_burn_retry)
|
|
finally:
|
|
e = None
|
|
del e
|
|
|
|
else:
|
|
printf('Warning: No input file to program to flash')
|
|
return (True, 'continue')
|
|
|
|
def sixth_run_step_write_efuse(self):
|
|
"""
|
|
Sixth step: write efuse.
|
|
"""
|
|
if self.args.efuse:
|
|
try:
|
|
if self.config['input_path']['efuse']:
|
|
efusefile = self.config['input_path']['efuse']
|
|
except:
|
|
efusefile = ''
|
|
|
|
if efusefile:
|
|
efuse_file = efusefile
|
|
mask_file = efuse_file.replace('.bin', '_mask.bin')
|
|
else:
|
|
efuse_file = self.cfg.get('EFUSE_CFG', 'file')
|
|
mask_file = self.cfg.get('EFUSE_CFG', 'maskfile')
|
|
if self.temp_task_num != None:
|
|
efuse_file = 'task' + str(self.temp_task_num) + '/' + efuse_file
|
|
efuse_load = True
|
|
efuse_verify = 0
|
|
if self.cfg.has_option('EFUSE_CFG', 'burn_en'):
|
|
efuse_load = self.cfg.get('EFUSE_CFG', 'burn_en') == 'true'
|
|
if self.cfg.has_option('EFUSE_CFG', 'factory_mode'):
|
|
if self.cfg.get('EFUSE_CFG', 'factory_mode') == 'true':
|
|
efuse_verify = 1
|
|
security_write = self.cfg.get('EFUSE_CFG', 'security_write') == 'true'
|
|
if efuse_load and self.isp_mode_sign is False:
|
|
ret = self.efuse_load_specified(efuse_file, mask_file, bytearray(0), bytearray(0), efuse_verify, security_write)
|
|
if self.callback:
|
|
self.callback(1, 1, 'APP_WR')
|
|
if ret is False:
|
|
return (False, self.flash_burn_retry)
|
|
else:
|
|
printf('efuse load disalbe')
|
|
self._need_shake_hand = False
|
|
return (True, 'continue')
|
|
|
|
def seventh_run_step_erase(self):
|
|
"""
|
|
Seventh step: erase.
|
|
"""
|
|
if self.args.erase:
|
|
printf('Erase flash operation')
|
|
if self.end == '0':
|
|
ret = self.flash_chiperase_main_process()
|
|
if ret is False:
|
|
return (False, self.flash_burn_retry)
|
|
else:
|
|
ret = self.flash_erase_main_process(int(self.start, 16), int(self.end, 16), self._need_shake_hand)
|
|
if ret is False:
|
|
return (False, self.flash_burn_retry)
|
|
printf('Erase flash OK')
|
|
return (True, 'continue')
|
|
|
|
def eighth_run_step_read(self):
|
|
"""
|
|
Eighth step: read.
|
|
"""
|
|
if self.args.read:
|
|
printf('Read operation')
|
|
if not self.args.flash:
|
|
if not self.args.efuse:
|
|
printf('No target select')
|
|
return (
|
|
False, self.flash_burn_retry)
|
|
if not (self.args.flash and self.start and self.end):
|
|
self.flash_read_jedec_id_process(self.callback)
|
|
else:
|
|
start_addr = int(self.start, 16)
|
|
end_addr = int(self.end, 16)
|
|
ret, readdata = self.flash_read_main_process(start_addr, end_addr - start_addr + 1, self._need_shake_hand, self.file, self.callback)
|
|
if ret is False:
|
|
return (False, self.flash_burn_retry)
|
|
if self.args.efuse:
|
|
start_addr = int(self.start, 16)
|
|
end_addr = int(self.end, 16)
|
|
if self.efuse_read_main_process(start_addr, end_addr - start_addr + 1, self._need_shake_hand, self.file) is False:
|
|
return (False, self.flash_burn_retry)
|
|
return (True, 'continue')
|
|
|
|
def ninth_run_step_end(self):
|
|
"""
|
|
Ninth step: run end.
|
|
"""
|
|
self.run_reset_cpu()
|
|
if self.macaddr_check is True:
|
|
self._bootinfo = self.bootinfo
|
|
self._macaddr_check = self._mac_addr
|
|
self._macaddr_check_status = False
|
|
return (
|
|
True, self.flash_burn_retry)
|
|
|
|
def set_clear_boot_status(self, shakehand=0):
|
|
pass
|
|
|
|
def set_boot_speed(self):
|
|
pass
|
|
|
|
def set_load_function(self):
|
|
self.load_function = 2
|
|
|
|
def set_decompress_write(self):
|
|
pass
|
|
|
|
def get_flash_pin(self):
|
|
return 0
|
|
|
|
def show_identify_fail(self):
|
|
return True
|
|
|
|
def run_flash2(self):
|
|
return (True, 'continue')
|
|
|
|
def get_flash1_and_flash2(self, flash_file, address, size_current, i):
|
|
return ('', 0, '', 0)
|
|
|
|
def run_reset_cpu(self):
|
|
pass
|
|
|
|
def get_mac_len(self):
|
|
return 6
|
|
|
|
def get_isp_sh_time(self):
|
|
return 0
|
|
|
|
def write_flash_data(self, file, start_addr, callback):
|
|
pass
|
|
|
|
def clear_boot_status(self, shakehand=0):
|
|
printf('Clear boot status at hbn rsvd register')
|
|
if shakehand != 0:
|
|
printf(FLASH_ERASE_SHAKE_HAND)
|
|
if self.img_load_shake_hand() is False:
|
|
return False
|
|
data = bytearray(12)
|
|
data[0] = 80
|
|
data[1] = 0
|
|
data[2] = 8
|
|
data[3] = 0
|
|
data[4] = 8
|
|
data[5] = 241
|
|
data[6] = 0
|
|
data[7] = 32
|
|
data[8] = 0
|
|
data[9] = 0
|
|
data[10] = 0
|
|
data[11] = 0
|
|
self.bflb_serial_object.write(data)
|
|
self.bflb_serial_object.deal_ack(dmy_data=False)
|
|
return True
|
|
|
|
def set_config_file(self, bootheaderFile, imgCreateFile):
|
|
self._efuse_bootheader_file = bootheaderFile
|
|
self._img_create_file = imgCreateFile
|
|
|
|
def load_eflash_loader_bin(self):
|
|
"""
|
|
This function is used to load eflash loader bin file.
|
|
"""
|
|
printf('========= load eflash_loader.bin =========')
|
|
bootinfo = None
|
|
if self.interface == 'jlink':
|
|
printf('Load eflash_loader.bin via jlink')
|
|
self.bflb_serial_object.if_init(self.device, self.speed, self.chip_type, self.chip_name)
|
|
self.bflb_serial_object.reset_cpu()
|
|
imge_fp = open_file(self.eflash_loader_file, 'rb')
|
|
fw_data = bytearray(imge_fp.read())[192:] + bytearray(0)
|
|
imge_fp.close()
|
|
sub_module = __import__(('libs.base.' + self.chip_type), fromlist=[self.chip_type])
|
|
load_addr = sub_module.jlink_load_cfg.jlink_load_addr
|
|
self.bflb_serial_object.if_raw_write(load_addr, fw_data)
|
|
pc = fw_data[4:8]
|
|
pc = bytes([pc[3], pc[2], pc[1], pc[0]])
|
|
msp = fw_data[0:4]
|
|
msp = bytes([msp[3], msp[2], msp[1], msp[0]])
|
|
self.bflb_serial_object.set_pc_msp(binascii.hexlify(pc), binascii.hexlify(msp).decode('utf-8'))
|
|
time.sleep(0.01)
|
|
self.bflb_serial_object.if_close()
|
|
return (
|
|
True, bootinfo, '')
|
|
if self.interface == 'openocd':
|
|
printf('Load eflash_loader.bin via openocd')
|
|
self.bflb_serial_object.if_init(self.device, self._bflb_sn_device, self.speed, self.chip_type, self.chip_name)
|
|
self.bflb_serial_object.halt_cpu()
|
|
imge_fp = open_file(self.eflash_loader_file, 'rb')
|
|
fw_data = bytearray(imge_fp.read())[192:] + bytearray(0)
|
|
imge_fp.close()
|
|
sub_module = __import__(('libs.base.' + self.chip_type), fromlist=[self.chip_type])
|
|
load_addr = sub_module.openocd_load_cfg.openocd_load_addr
|
|
self.bflb_serial_object.if_raw_write(load_addr, fw_data)
|
|
pc = fw_data[4:8]
|
|
pc = bytes([pc[3], pc[2], pc[1], pc[0]])
|
|
msp = fw_data[0:4]
|
|
msp = bytes([msp[3], msp[2], msp[1], msp[0]])
|
|
self.bflb_serial_object.set_pc_msp(binascii.hexlify(pc), binascii.hexlify(msp).decode('utf-8'))
|
|
return (
|
|
True, bootinfo, '')
|
|
if self.interface == 'cklink':
|
|
printf('Load eflash_loader.bin via cklink')
|
|
self.bflb_serial_object.if_init(self.device, self._bflb_sn_device, self.speed, self.chip_type, self.chip_name)
|
|
self.bflb_serial_object.halt_cpu()
|
|
imge_fp = open_file(self.eflash_loader_file, 'rb')
|
|
fw_data = bytearray(imge_fp.read())[192:] + bytearray(0)
|
|
imge_fp.close()
|
|
sub_module = __import__(('libs.base.' + self.chip_type), fromlist=[self.chip_type])
|
|
load_addr = sub_module.openocd_load_cfg.openocd_load_addr
|
|
self.bflb_serial_object.if_raw_write(load_addr, fw_data)
|
|
pc = fw_data[4:8]
|
|
pc = bytes([pc[3], pc[2], pc[1], pc[0]])
|
|
msp = fw_data[0:4]
|
|
msp = bytes([msp[3], msp[2], msp[1], msp[0]])
|
|
self.bflb_serial_object.set_pc_msp(binascii.hexlify(pc), binascii.hexlify(msp).decode('utf-8'))
|
|
self.bflb_serial_object.resume_cpu()
|
|
return (
|
|
True, bootinfo, '')
|
|
if self.interface.lower() == 'uart' or self.interface == 'sdio':
|
|
ret = True
|
|
printf('Load eflash_loader.bin via %s' % self.interface)
|
|
start_time = time.time() * 1000
|
|
ret, bootinfo, res = self._bflb_com_img_loader.img_load_process(self.boot_speed, self.boot_speed, None, self.do_reset, self.reset_hold_time, self.shake_hand_delay, self.reset_revert, self.cutoff_time, self.shake_hand_retry, self.isp_mode_sign, self._isp_shakehand_timeout, True, bootinfo)
|
|
printf('Load helper bin time cost(ms): ', time.time() * 1000 - start_time)
|
|
return (
|
|
ret, bootinfo, res)
|
|
|
|
def set_temp_timeout(self):
|
|
self.bflb_serial_object.set_timeout(self._erase_time_out / 1000)
|
|
|
|
def get_new_bh_data(self, section, bh_data, fp):
|
|
return b''
|
|
|
|
def get_chip_id(self, bootinfo):
|
|
chip_id = bootinfo[34:36] + bootinfo[32:34] + bootinfo[30:32] + bootinfo[28:30] + bootinfo[26:28] + bootinfo[24:26]
|
|
return chip_id
|
|
|
|
def get_flash_file_and_address(self):
|
|
result = True
|
|
temp_files = []
|
|
temp_addrs = []
|
|
temp_dict = {}
|
|
temp_set_addrs = []
|
|
for key, value_dict in self.config.items():
|
|
if isinstance(value_dict, dict):
|
|
if 'addr' in value_dict:
|
|
if value_dict['addr'] in temp_set_addrs:
|
|
printf('Error: %s has duplicate addresse %s!' % (key, value_dict['addr']))
|
|
return (
|
|
False, temp_addrs, temp_files)
|
|
else:
|
|
temp_dict[value_dict['addr']] = value_dict['path']
|
|
temp_set_addrs.append(value_dict['addr'])
|
|
|
|
if temp_dict:
|
|
temp_list = sorted((temp_dict.items()), key=(lambda x: int(x[0], 16)
|
|
), reverse=False)
|
|
for i in range(len(temp_list)):
|
|
temp_addrs.append(temp_list[i][0].replace('0x', ''))
|
|
temp_files.append(temp_list[i][1])
|
|
if i != 0:
|
|
temp_length = int(temp_list[i][0], 16) - int(temp_list[i - 1][0], 16)
|
|
file_length = os.path.getsize(temp_list[i - 1][1])
|
|
if temp_length < file_length:
|
|
result = False
|
|
printf('Error: path: %s size: %s range: %s' % (temp_list[i - 1][1], str(file_length), str(temp_length)))
|
|
printf('Error: The file size exceeds the address space size!')
|
|
|
|
return (
|
|
result, temp_addrs, temp_files)
|
|
|
|
def get_boot_info(self):
|
|
"""
|
|
This function is used to get boot information from chips. At the same time, it can get chip ID.
|
|
"""
|
|
printf('========= get_boot_info =========')
|
|
bootinfo = None
|
|
if self.interface == 'uart':
|
|
ret = True
|
|
start_time = time.time() * 1000
|
|
ret, bootinfo = self._bflb_com_img_loader.img_get_bootinfo(self.boot_speed, self.boot_speed, None, self.do_reset, self.reset_hold_time, self.shake_hand_delay, self.reset_revert, self.cutoff_time, self.shake_hand_retry, self.isp_mode_sign, self._isp_shakehand_timeout)
|
|
bootinfo = bootinfo.decode('utf-8')
|
|
chipid = ''
|
|
chipid = self.get_chip_id(bootinfo)
|
|
printf('========= ChipID: ', chipid, ' =========')
|
|
printf('Get bootinfo time cost(ms): ', time.time() * 1000 - start_time)
|
|
return (
|
|
ret, bootinfo, 'OK')
|
|
printf('interface not fit')
|
|
return (
|
|
False, bootinfo, '')
|
|
|
|
def error_code_print(self, code):
|
|
set_error_code(code, self.task_num)
|
|
printf('{"ErrorCode": "' + code + '","ErrorMsg":"' + eflash_loader_error_code[code] + '"}')
|
|
|
|
def clock_para_update(self, file):
|
|
if os.path.isfile(file) is False:
|
|
efuse_bootheader_path = os.path.join(chip_path, self.chip_name, 'efuse_bootheader')
|
|
efuse_bh_cfg = efuse_bootheader_path + '/efuse_bootheader_cfg.conf'
|
|
sub_module = __import__(('libs.base.' + self.chip_type), fromlist=[self.chip_type])
|
|
section = 'BOOTHEADER_GROUP0_CFG'
|
|
fp = open(efuse_bh_cfg, 'r')
|
|
data = fp.read()
|
|
fp.close()
|
|
if 'BOOTHEADER_CFG' in data:
|
|
section = 'BOOTHEADER_CFG'
|
|
else:
|
|
if 'BOOTHEADER_CPU0_CFG' in data:
|
|
section = 'BOOTHEADER_CPU0_CFG'
|
|
else:
|
|
if 'BOOTHEADER_GROUP0_CFG' in data:
|
|
section = 'BOOTHEADER_GROUP0_CFG'
|
|
bh_data, tmp = bflb_efuse_boothd_create.update_data_from_cfg(sub_module.bootheader_cfg_keys.bootheader_cfg_keys, efuse_bh_cfg, section)
|
|
bh_data = bflb_efuse_boothd_create.bootheader_update_flash_pll_crc(bh_data, self.chip_type)
|
|
fp = open(file, 'wb+')
|
|
self.get_new_bh_data(section, bh_data, fp)
|
|
fp.close()
|
|
fp = open_file(file, 'rb')
|
|
clock_para = bytearray(fp.read())
|
|
fp.close()
|
|
return clock_para
|
|
|
|
def clock_pll_set(self, shakehand, irq_en, clk_para):
|
|
printf('Clock PLL set')
|
|
if shakehand != 0:
|
|
printf('clock set shake hand')
|
|
if self.img_load_shake_hand() is False:
|
|
return False
|
|
start_time = time.time() * 1000
|
|
cmd_id = hexstr_to_bytearray(self._com_cmds.get('clk_set')['cmd_id'])
|
|
irq_enable = bytearray(4)
|
|
load_speed = bytearray(4)
|
|
if irq_en:
|
|
irq_enable = b'\x01\x00\x00\x00'
|
|
load_speed = int_to_4bytearray_l(int(self.speed))
|
|
data_send = irq_enable + load_speed + clk_para
|
|
try_cnt = 0
|
|
while True:
|
|
ret, dmy = self.com_process_one_cmd('clk_set', cmd_id, data_send)
|
|
if ret.startswith('OK'):
|
|
break
|
|
if try_cnt < self._checksum_err_retry_limit:
|
|
printf('Retry')
|
|
try_cnt += 1
|
|
else:
|
|
self.error_code_print('000C')
|
|
return False
|
|
|
|
printf('Set clock time cost(ms): ', time.time() * 1000 - start_time)
|
|
self.bflb_serial_object.repeat_init(self.device, self.speed, self.chip_type, self.chip_name)
|
|
self.bflb_serial_object.clear_buf()
|
|
time.sleep(0.01)
|
|
return True
|
|
|
|
def efuse_read_mac_addr_process(self, callback=None):
|
|
readdata = bytearray(0)
|
|
macLen = self.get_mac_len()
|
|
if self._need_shake_hand != 0:
|
|
printf(FLASH_LOAD_SHAKE_HAND)
|
|
if self.img_load_shake_hand() is False:
|
|
return (False, None)
|
|
cmd_id = hexstr_to_bytearray(self._com_cmds.get('efuse_read_mac')['cmd_id'])
|
|
printf('Read mac addr ')
|
|
ret, data_read = self.com_process_one_cmd('efuse_read_mac', cmd_id, bytearray(0))
|
|
if ret.startswith('OK') is False:
|
|
self.error_code_print('0023')
|
|
return (False, None)
|
|
readdata += data_read
|
|
crcarray = get_crc32_bytearray(readdata[:macLen])
|
|
if crcarray != readdata[macLen:macLen + 4]:
|
|
printf(binascii.hexlify(crcarray))
|
|
printf(binascii.hexlify(readdata[macLen:macLen + 4]))
|
|
self.error_code_print('0025')
|
|
return (False, None)
|
|
return (True, readdata[:macLen])
|
|
|
|
def flash_set_para_main_process(self, flash_pin, flash_para):
|
|
printf('Set flash config ')
|
|
if flash_para != bytearray(0):
|
|
if flash_para[13:14] == b'\xff':
|
|
printf('Skip set flash para due to flash id is 0xFF')
|
|
return True
|
|
if self._need_shake_hand != 0:
|
|
printf('Flash set para shake hand')
|
|
if self.img_load_shake_hand() is False:
|
|
return False
|
|
start_time = time.time() * 1000
|
|
cmd_id = hexstr_to_bytearray(self._com_cmds.get('flash_set_para')['cmd_id'])
|
|
data_send = int_to_4bytearray_l(flash_pin) + flash_para
|
|
try_cnt = 0
|
|
while True:
|
|
ret, dmy = self.com_process_one_cmd('flash_set_para', cmd_id, data_send)
|
|
if ret.startswith('OK'):
|
|
break
|
|
if try_cnt < self._checksum_err_retry_limit:
|
|
printf('Retry')
|
|
try_cnt += 1
|
|
else:
|
|
self.error_code_print('003B')
|
|
return False
|
|
|
|
printf('Set para time cost(ms): ', time.time() * 1000 - start_time)
|
|
return True
|
|
|
|
def flash_read_jedec_id_process(self, callback=None):
|
|
printf('========= flash read jedec ID =========')
|
|
readdata = bytearray(0)
|
|
if self._need_shake_hand is not False:
|
|
printf(FLASH_LOAD_SHAKE_HAND)
|
|
if self.img_load_shake_hand() is False:
|
|
return (False, None)
|
|
cmd_id = hexstr_to_bytearray(self._com_cmds.get('flash_read_jid')['cmd_id'])
|
|
ret, data_read = self.com_process_one_cmd('flash_read_jid', cmd_id, bytearray(0))
|
|
printf('Read flash jedec ID ')
|
|
if ret.startswith('OK') is False:
|
|
self.error_code_print('0030')
|
|
return (False, None)
|
|
readdata += data_read
|
|
printf('readdata: ')
|
|
printf(binascii.hexlify(readdata))
|
|
printf('Finished')
|
|
return (
|
|
True, readdata[:4])
|
|
|
|
def flash_para_update(self, file, jedec_id):
|
|
flash_para = bytearray(0)
|
|
if self.is_conf_exist(jedec_id) is True:
|
|
sub_module = __import__(('libs.base.' + self.chip_type), fromlist=[self.chip_type])
|
|
cfg_dir = app_path + '/utils/flash/' + self.chip_type + '/'
|
|
conf_name = sub_module.flash_select_do.get_suitable_file_name(cfg_dir, jedec_id)
|
|
offset, flashCfgLen, flash_para, flashCrcOffset, crcOffset = bflb_flash_select.update_flash_para_from_cfg(sub_module.bootheader_cfg_keys.bootheader_cfg_keys, cfg_dir + conf_name)
|
|
fp = open(os.path.join(app_path, file), 'wb+')
|
|
fp.write(flash_para)
|
|
fp.close()
|
|
return flash_para
|
|
|
|
def is_conf_exist(self, flash_id):
|
|
cfg_dir = app_path + '/utils/flash/' + self.chip_type + '/'
|
|
conf_name = self.get_suitable_conf_name(cfg_dir, flash_id)
|
|
if os.path.isfile(cfg_dir + conf_name) is False:
|
|
return False
|
|
return True
|
|
|
|
def flash_chiperase_main_process(self):
|
|
printf('Flash Chip Erase All')
|
|
if self._need_shake_hand != 0:
|
|
printf(FLASH_ERASE_SHAKE_HAND)
|
|
if self.img_load_shake_hand() is False:
|
|
printf('Shake hand fail')
|
|
return False
|
|
start_time = time.time() * 1000
|
|
self.set_temp_timeout()
|
|
cmd_id = hexstr_to_bytearray(self._com_cmds.get('flash_chiperase')['cmd_id'])
|
|
try_cnt = 0
|
|
while True:
|
|
ret, dmy = self.com_process_one_cmd('flash_chiperase', cmd_id, bytearray(0))
|
|
if ret.startswith('OK'):
|
|
break
|
|
else:
|
|
if ret.startswith('PD'):
|
|
printf('erase pending')
|
|
while True:
|
|
ret = self.bflb_serial_object.deal_ack()
|
|
if ret.startswith('PD'):
|
|
printf('erase pending')
|
|
else:
|
|
self.bflb_serial_object.set_timeout(0.02)
|
|
self.bflb_serial_object.read(1000)
|
|
break
|
|
if time.time() * 1000 - start_time > self._erase_time_out:
|
|
printf('erase timeout')
|
|
break
|
|
|
|
if ret.startswith('OK'):
|
|
break
|
|
if try_cnt < self._checksum_err_retry_limit:
|
|
printf('Retry')
|
|
try_cnt += 1
|
|
else:
|
|
printf('Erase Fail')
|
|
self.bflb_serial_object.set_timeout(self._default_time_out)
|
|
self.error_code_print('0033')
|
|
return False
|
|
|
|
printf('Chip erase time cost(ms): ', time.time() * 1000 - start_time)
|
|
self.bflb_serial_object.set_timeout(self._default_time_out)
|
|
return True
|
|
|
|
def flash_cfg_option(self, read_flash_id, flash_para_file, flash_set, id_valid_flag, binfile, cfgfile, cfg, create_img_callback=None, create_simple_callback=None):
|
|
ret = bflb_flash_select.flash_bootheader_config_check(self.chip_type, read_flash_id, convert_path(binfile), flash_para_file)
|
|
if ret is False:
|
|
printf('flashcfg not match first')
|
|
if self.is_conf_exist(read_flash_id) is True:
|
|
update_cfg(cfg, 'FLASH_CFG', 'flash_id', read_flash_id)
|
|
if isinstance(cfgfile, BFConfigParser) == False:
|
|
cfg.write(cfgfile, 'w+')
|
|
if create_img_callback is not None:
|
|
create_img_callback()
|
|
else:
|
|
if create_simple_callback is not None:
|
|
create_simple_callback()
|
|
else:
|
|
self.error_code_print('003D')
|
|
return False
|
|
ret = bflb_flash_select.flash_bootheader_config_check(self.chip_name, read_flash_id, convert_path(binfile), flash_para_file)
|
|
if ret is False:
|
|
printf('flashcfg not match again')
|
|
self.error_code_print('0040')
|
|
return False
|
|
if flash_para_file:
|
|
if id_valid_flag != '80':
|
|
printf('flash para file: ', flash_para_file)
|
|
fp = open_file(flash_para_file, 'rb')
|
|
flash_para = bytearray(fp.read())
|
|
fp.close()
|
|
ret = self.flash_set_para_main_process(flash_set, flash_para)
|
|
self._need_shake_hand = False
|
|
if ret is False:
|
|
return False
|
|
|
|
def flash_switch_bank_process(self, bank):
|
|
"""When the chip has two flashes, switch the flashes according to bank."""
|
|
printf('Flash Switch Bank')
|
|
if self._need_shake_hand != 0:
|
|
printf('Flash switch bank shake hand')
|
|
if self.img_load_shake_hand() is False:
|
|
printf('Shake hand fail')
|
|
return False
|
|
start_time = time.time() * 1000
|
|
self.bflb_serial_object.set_timeout(self._erase_time_out / 1000)
|
|
cmd_id = hexstr_to_bytearray(self._com_cmds.get('flash_switch_bank')['cmd_id'])
|
|
data_send = int_to_4bytearray_l(bank)
|
|
ret, dmy = self.com_process_one_cmd('flash_switch_bank', cmd_id, data_send)
|
|
if ret.startswith('OK') is False:
|
|
printf('Switch Fail')
|
|
self.bflb_serial_object.set_timeout(self._default_time_out)
|
|
self.error_code_print('0042')
|
|
return False
|
|
printf('Switch bank time cost(ms): ', time.time() * 1000 - start_time)
|
|
self.bflb_serial_object.set_timeout(self._default_time_out)
|
|
if bank == 0:
|
|
self._flash2_select = False
|
|
else:
|
|
self._flash2_select = True
|
|
return True
|
|
|
|
def flash_load_opt(self, file, start_addr, callback=None):
|
|
printf('========= flash load =========')
|
|
if self._need_shake_hand != 0:
|
|
printf(FLASH_LOAD_SHAKE_HAND)
|
|
if self.img_load_shake_hand() is False:
|
|
return False
|
|
if self._flash2_select is True:
|
|
start_addr -= self._flash1_size
|
|
self.write_flash_data(file, start_addr, callback)
|
|
ret = self.flash_load_main_process(file, start_addr, callback)
|
|
if ret is False:
|
|
printf('Flash load fail')
|
|
return ret
|
|
fw_sha256 = ''
|
|
fp = open_file(file, 'rb')
|
|
flash_data = fp.read()
|
|
fp.close()
|
|
flash_data_len = len(flash_data)
|
|
if flash_data_len > 2097152:
|
|
self.bflb_serial_object.set_timeout(2.0 * (flash_data_len / 2097152 + 1))
|
|
sh = hashlib.sha256()
|
|
sh.update(flash_data)
|
|
fw_sha256 = sh.hexdigest()
|
|
fw_sha256 = hexstr_to_bytearray(fw_sha256)
|
|
printf('Sha caled by host: ', binascii.hexlify(fw_sha256).decode('utf-8'))
|
|
del sh
|
|
printf('xip mode Verify')
|
|
ret, read_data = self.flash_xip_read_sha_main_process(start_addr, flash_data_len, 0, None, callback)
|
|
try:
|
|
printf('Sha caled by dev: ', binascii.hexlify(read_data).decode('utf-8'))
|
|
except:
|
|
printf('Sha caled by dev: ', binascii.hexlify(read_data))
|
|
|
|
if ret is True and read_data == fw_sha256:
|
|
printf('Verify success')
|
|
else:
|
|
printf('Verify fail')
|
|
self.flash_load_tips()
|
|
self.error_code_print('003E')
|
|
ret = False
|
|
if self.verify > 0:
|
|
fp = open_file(file, 'rb')
|
|
flash_data = bytearray(fp.read())
|
|
fp.close()
|
|
flash_data_len = len(flash_data)
|
|
ret, read_data = self.flash_read_main_process(start_addr, flash_data_len, 0, None, callback)
|
|
if ret is True and read_data == flash_data:
|
|
printf('Verify success')
|
|
else:
|
|
printf('Verify fail')
|
|
self.flash_load_tips()
|
|
self.error_code_print('003E')
|
|
ret = False
|
|
printf('sbus mode Verify')
|
|
ret, read_data = self.flash_read_sha_main_process(start_addr, flash_data_len, 0, None, callback)
|
|
printf('Sha caled by dev: ', binascii.hexlify(read_data).decode('utf-8'))
|
|
if ret is True and read_data == fw_sha256:
|
|
printf('Verify success')
|
|
else:
|
|
printf('Verify fail')
|
|
self.flash_load_tips()
|
|
self.error_code_print('003E')
|
|
ret = False
|
|
self.bflb_serial_object.set_timeout(self._default_time_out)
|
|
return ret
|
|
|
|
def flash_load_specified(self, file, start_addr, callback=None):
|
|
ret = False
|
|
run_state = 0
|
|
temp_start_addr = start_addr
|
|
if len(self._skip_len) > 0 and self._skip_len[0] > 0:
|
|
fp = open_file(file, 'rb')
|
|
flash_data = fp.read()
|
|
fp.close()
|
|
flash_data_len = len(flash_data)
|
|
if self._skip_addr[0] > temp_start_addr + flash_data_len or self._skip_addr[-1] + self._skip_len[-1] < temp_start_addr:
|
|
ret = self.flash_load_opt(file, start_addr, callback)
|
|
return ret
|
|
for i in range(len(self._skip_addr)):
|
|
if self._skip_addr[i] <= start_addr:
|
|
if self._skip_addr[i] + self._skip_len[i] > start_addr and self._skip_addr[i] + self._skip_len[i] < temp_start_addr + flash_data_len:
|
|
addr = self._skip_addr[i] + self._skip_len[i]
|
|
start_addr = self._skip_addr[i] + self._skip_len[i]
|
|
else:
|
|
if self._skip_addr[i] > start_addr and self._skip_addr[i] + self._skip_len[i] < temp_start_addr + flash_data_len:
|
|
printf('skip flash file, skip addr 0x%08X, skip len 0x%08X' % (
|
|
self._skip_addr[i], self._skip_len[i]))
|
|
addr = start_addr
|
|
data = flash_data[start_addr - temp_start_addr:self._skip_addr[i] - temp_start_addr]
|
|
filename, ext = os.path.splitext(file)
|
|
file_temp = os.path.join(app_path, filename + '_skip' + str(i) + ext)
|
|
fp = open(file_temp, 'wb')
|
|
fp.write(data)
|
|
fp.close()
|
|
ret = self.flash_load_opt(file_temp, addr, callback)
|
|
start_addr = self._skip_addr[i] + self._skip_len[i]
|
|
else:
|
|
if self._skip_addr[i] > start_addr:
|
|
if self._skip_addr[i] < temp_start_addr + flash_data_len and self._skip_addr[i] + self._skip_len[i] >= temp_start_addr + flash_data_len:
|
|
printf('skip flash file, skip addr 0x%08X, skip len 0x%08X' % (
|
|
self._skip_addr[i], self._skip_len[i]))
|
|
addr = start_addr
|
|
data = flash_data[start_addr - temp_start_addr:self._skip_addr[i] - temp_start_addr]
|
|
filename, ext = os.path.splitext(file)
|
|
file_temp = os.path.join(app_path, filename + '_skip' + str(i) + ext)
|
|
fp = open(file_temp, 'wb')
|
|
fp.write(data)
|
|
fp.close()
|
|
ret = self.flash_load_opt(file_temp, addr, callback)
|
|
start_addr = temp_start_addr + flash_data_len
|
|
else:
|
|
if self._skip_addr[i] <= start_addr:
|
|
if self._skip_addr[i] + self._skip_len[i] >= temp_start_addr + flash_data_len:
|
|
printf('skip flash file, skip addr 0x%08X, skip len 0x%08X' % (
|
|
self._skip_addr[i], self._skip_len[i]))
|
|
start_addr = temp_start_addr + flash_data_len
|
|
return True
|
|
|
|
if start_addr < temp_start_addr + flash_data_len:
|
|
addr = start_addr
|
|
data = flash_data[start_addr - temp_start_addr:]
|
|
filename, ext = os.path.splitext(file)
|
|
file_temp = os.path.join(app_path, filename + '_skip' + str(i + 1) + ext)
|
|
fp = open(file_temp, 'wb')
|
|
fp.write(data)
|
|
fp.close()
|
|
ret = self.flash_load_opt(file_temp, addr, callback)
|
|
else:
|
|
ret = self.flash_load_opt(file, start_addr, callback)
|
|
return ret
|
|
|
|
def flash_load_main_process(self, file, start_addr, callback=None):
|
|
fp = open_file(file, 'rb')
|
|
flash_data = bytearray(fp.read())
|
|
fp.close()
|
|
flash_data_len = len(flash_data)
|
|
i = 0
|
|
cur_len = 0
|
|
if self.erase == 1:
|
|
ret = self.flash_erase_main_process(start_addr, start_addr + flash_data_len - 1)
|
|
if ret is False:
|
|
return False
|
|
start_time = time.time() * 1000
|
|
log = ''
|
|
if self.decompress_write and flash_data_len > 4096:
|
|
self.bflb_serial_object.set_timeout(30.0)
|
|
start_addr |= 2147483648
|
|
cmd_name = 'flash_decompress_write'
|
|
ret, flash_data, flash_data_len = self.flash_load_xz_compress(file)
|
|
if ret is False:
|
|
printf('Flash write data xz fail')
|
|
self.bflb_serial_object.set_timeout(self._default_time_out)
|
|
return False
|
|
if time.time() * 1000 - start_time > 2200:
|
|
printf(FLASH_LOAD_SHAKE_HAND)
|
|
if self.img_load_shake_hand() is False:
|
|
return False
|
|
else:
|
|
if time.time() * 1000 - start_time > 1800:
|
|
time.sleep(0.5)
|
|
printf(FLASH_LOAD_SHAKE_HAND)
|
|
if self.img_load_shake_hand() is False:
|
|
return False
|
|
printf('decompress flash load ', flash_data_len)
|
|
else:
|
|
cmd_name = 'flash_write'
|
|
while i < flash_data_len:
|
|
cur_len = flash_data_len - i
|
|
if cur_len > self._bflb_com_tx_size - 8:
|
|
cur_len = self._bflb_com_tx_size - 8
|
|
cmd_id = hexstr_to_bytearray(self._com_cmds.get(cmd_name)['cmd_id'])
|
|
data_send = int_to_4bytearray_l(i + start_addr) + flash_data[i:i + cur_len]
|
|
start_addr &= 2147483647
|
|
try_cnt = 0
|
|
while True:
|
|
ret, dmy = self.com_process_one_cmd(cmd_name, cmd_id, data_send)
|
|
if ret.startswith('OK'):
|
|
break
|
|
if try_cnt < self._checksum_err_retry_limit:
|
|
printf('Retry')
|
|
try_cnt += 1
|
|
else:
|
|
self.error_code_print('0036')
|
|
self.bflb_serial_object.set_timeout(self._default_time_out)
|
|
return False
|
|
|
|
i += cur_len
|
|
log = 'Load ' + str(i) + '/' + str(flash_data_len) + ' {"progress":' + str(i * 100 // flash_data_len) + '}'
|
|
printf(log)
|
|
if callback is not None:
|
|
if flash_data_len > 200:
|
|
callback(i, flash_data_len, 'APP_WR')
|
|
|
|
printf(log)
|
|
if self.flash_write_check_main_process() is False:
|
|
printf('Flash write check fail')
|
|
self.bflb_serial_object.set_timeout(self._default_time_out)
|
|
return False
|
|
self.bflb_serial_object.set_timeout(self._default_time_out)
|
|
printf('Flash load time cost(ms): ', time.time() * 1000 - start_time)
|
|
printf('Finished')
|
|
return True
|
|
|
|
def flash_xip_read_sha_main_process(self, start_addr, flash_data_len, shakehand=0, file=None, callback=None):
|
|
readdata = bytearray(0)
|
|
if shakehand != 0:
|
|
printf(FLASH_LOAD_SHAKE_HAND)
|
|
if self.img_load_shake_hand() is False:
|
|
return (False, None)
|
|
cmd_id = hexstr_to_bytearray(self._com_cmds.get('flash_xip_read_start')['cmd_id'])
|
|
ret, dmy = self.com_process_one_cmd('flash_xip_read_start', cmd_id, bytearray(0))
|
|
if ret.startswith('OK') is False:
|
|
self.error_code_print('0039')
|
|
return (False, None)
|
|
start_time = time.time() * 1000
|
|
log = ''
|
|
cmd_id = hexstr_to_bytearray(self._com_cmds.get('flash_xip_readSha')['cmd_id'])
|
|
data_send = int_to_4bytearray_l(start_addr) + int_to_4bytearray_l(flash_data_len)
|
|
try_cnt = 0
|
|
while True:
|
|
ret, data_read = self.com_process_one_cmd('flash_xip_readSha', cmd_id, data_send)
|
|
if ret.startswith('OK'):
|
|
break
|
|
if try_cnt < self._checksum_err_retry_limit:
|
|
printf('Retry')
|
|
try_cnt += 1
|
|
else:
|
|
printf('Read Fail')
|
|
cmd_id = hexstr_to_bytearray(self._com_cmds.get('flash_xip_read_finish')['cmd_id'])
|
|
ret, dmy = self.com_process_one_cmd('flash_xip_read_finish', cmd_id, bytearray(0))
|
|
if ret.startswith('OK') is False:
|
|
self.error_code_print('0039')
|
|
return (False, None)
|
|
return (False, None)
|
|
|
|
log += 'Read Sha256/' + str(flash_data_len)
|
|
if callback is not None:
|
|
callback(flash_data_len, flash_data_len, 'APP_VR')
|
|
readdata += data_read
|
|
printf(log)
|
|
printf('Flash xip readsha time cost(ms): ', time.time() * 1000 - start_time)
|
|
printf('Finished')
|
|
if file is not None:
|
|
fp = open_file(file, 'wb+')
|
|
fp.write(readdata)
|
|
fp.close()
|
|
cmd_id = hexstr_to_bytearray(self._com_cmds.get('flash_xip_read_finish')['cmd_id'])
|
|
ret, dmy = self.com_process_one_cmd('flash_xip_read_finish', cmd_id, bytearray(0))
|
|
if ret.startswith('OK') is False:
|
|
self.error_code_print('0039')
|
|
return (False, None)
|
|
return (True, readdata)
|
|
|
|
def flash_read_sha_main_process(self, start_addr, flash_data_len, shakehand=0, file=None, callback=None):
|
|
readdata = bytearray(0)
|
|
if shakehand != 0:
|
|
printf(FLASH_LOAD_SHAKE_HAND)
|
|
if self.img_load_shake_hand() is False:
|
|
return (False, None)
|
|
start_time = time.time() * 1000
|
|
log = ''
|
|
cmd_id = hexstr_to_bytearray(self._com_cmds.get('flash_readSha')['cmd_id'])
|
|
data_send = int_to_4bytearray_l(start_addr) + int_to_4bytearray_l(flash_data_len)
|
|
try_cnt = 0
|
|
while True:
|
|
ret, data_read = self.com_process_one_cmd('flash_readSha', cmd_id, data_send)
|
|
if ret.startswith('OK'):
|
|
break
|
|
if try_cnt < self._checksum_err_retry_limit:
|
|
printf('Retry')
|
|
try_cnt += 1
|
|
else:
|
|
self.error_code_print('0038')
|
|
return (False, None)
|
|
|
|
log += 'Read Sha256/' + str(flash_data_len)
|
|
if callback is not None:
|
|
callback(flash_data_len, flash_data_len, 'APP_VR')
|
|
readdata += data_read
|
|
printf(log)
|
|
printf('Flash readsha time cost(ms): ', time.time() * 1000 - start_time)
|
|
printf('Finished')
|
|
if file is not None:
|
|
fp = open_file(file, 'wb+')
|
|
fp.write(readdata)
|
|
fp.close()
|
|
return (True, readdata)
|
|
|
|
def flash_load_xz_compress(self, file):
|
|
try:
|
|
xz_filters = [
|
|
{'id':lzma.FILTER_LZMA2,
|
|
'dict_size':32768}]
|
|
fp = open_file(file, 'rb')
|
|
data = bytearray(fp.read())
|
|
fp.close()
|
|
flash_data = lzma.compress(data, check=(lzma.CHECK_CRC32), filters=xz_filters)
|
|
flash_data_len = len(flash_data)
|
|
except Exception as e:
|
|
try:
|
|
printf(e)
|
|
return (False, None, None)
|
|
finally:
|
|
e = None
|
|
del e
|
|
|
|
return (
|
|
True, flash_data, flash_data_len)
|
|
|
|
def flash_load_tips(self):
|
|
printf('########################################################################')
|
|
printf('请按照以下描述排查问题：')
|
|
printf('是否降低烧录波特率到500K测试过')
|
|
printf('烧写文件的大小是否超过Flash所能存储的最大空间')
|
|
printf('Flash是否被写保护')
|
|
printf('########################################################################')
|
|
|
|
def flash_erase_main_process(self, start_addr, end_addr, shakehand=0):
|
|
printf('========= flash erase =========')
|
|
printf('Erase flash from ', hex(start_addr), ' to ', hex(end_addr))
|
|
if shakehand != 0:
|
|
printf(FLASH_ERASE_SHAKE_HAND)
|
|
if self.img_load_shake_hand() is False:
|
|
printf('Shake hand fail')
|
|
return False
|
|
start_time = time.time() * 1000
|
|
self.set_temp_timeout()
|
|
cmd_id = hexstr_to_bytearray(self._com_cmds.get('flash_erase')['cmd_id'])
|
|
data_send = int_to_4bytearray_l(start_addr) + int_to_4bytearray_l(end_addr)
|
|
try_cnt = 0
|
|
while True:
|
|
ret, dmy = self.com_process_one_cmd('flash_erase', cmd_id, data_send)
|
|
if ret.startswith('OK'):
|
|
break
|
|
else:
|
|
if ret.startswith('PD'):
|
|
printf('erase pending')
|
|
while True:
|
|
ret = self.bflb_serial_object.deal_ack()
|
|
if ret.startswith('PD'):
|
|
printf('erase pending')
|
|
else:
|
|
self.bflb_serial_object.set_timeout(0.02)
|
|
self.bflb_serial_object.read(1000)
|
|
break
|
|
if time.time() * 1000 - start_time > self._erase_time_out:
|
|
printf('erase timeout')
|
|
break
|
|
|
|
if ret.startswith('OK'):
|
|
break
|
|
if try_cnt < self._checksum_err_retry_limit:
|
|
printf('Retry')
|
|
try_cnt += 1
|
|
else:
|
|
printf('Erase Fail')
|
|
self.bflb_serial_object.set_timeout(self._default_time_out)
|
|
self.error_code_print('0034')
|
|
return False
|
|
|
|
printf('Erase time cost(ms): ', time.time() * 1000 - start_time)
|
|
self.bflb_serial_object.set_timeout(self._default_time_out)
|
|
return True
|
|
|
|
def flash_read_main_process(self, start_addr, flash_data_len, shakehand=0, file=None, callback=None):
|
|
printf('========= flash read =========')
|
|
i = 0
|
|
cur_len = 0
|
|
readdata = bytearray(0)
|
|
if shakehand != 0:
|
|
printf(FLASH_LOAD_SHAKE_HAND)
|
|
if self.img_load_shake_hand() is False:
|
|
return (False, None)
|
|
start_time = time.time() * 1000
|
|
log = ''
|
|
while i < flash_data_len:
|
|
cur_len = flash_data_len - i
|
|
if cur_len > self._bflb_com_tx_size - 8:
|
|
cur_len = self._bflb_com_tx_size - 8
|
|
else:
|
|
cmd_id = hexstr_to_bytearray(self._com_cmds.get('flash_read')['cmd_id'])
|
|
data_send = int_to_4bytearray_l(i + start_addr) + int_to_4bytearray_l(cur_len)
|
|
try_cnt = 0
|
|
while True:
|
|
ret, data_read = self.com_process_one_cmd('flash_read', cmd_id, data_send)
|
|
if ret.startswith('OK'):
|
|
break
|
|
if try_cnt < self._checksum_err_retry_limit:
|
|
printf('Retry')
|
|
try_cnt += 1
|
|
else:
|
|
self.error_code_print('0035')
|
|
return (False, None)
|
|
|
|
i += cur_len
|
|
log += 'Read ' + str(i) + '/' + str(flash_data_len)
|
|
if len(log) > 50:
|
|
printf(log)
|
|
log = ''
|
|
else:
|
|
log += '\n'
|
|
if callback is not None:
|
|
callback(i, flash_data_len, 'APP_VR')
|
|
readdata += data_read
|
|
|
|
printf(log)
|
|
printf('Flash read time cost(ms): ', time.time() * 1000 - start_time)
|
|
printf('Finished')
|
|
if file is not None:
|
|
fp = open_file(file, 'wb+')
|
|
fp.write(readdata)
|
|
fp.close()
|
|
return (True, readdata)
|
|
|
|
def flash_write_check_main_process(self, shakehand=0):
|
|
printf('Write check')
|
|
if shakehand != 0:
|
|
printf(FLASH_LOAD_SHAKE_HAND)
|
|
if self.img_load_shake_hand() is False:
|
|
return False
|
|
cmd_id = hexstr_to_bytearray(self._com_cmds.get('flash_write_check')['cmd_id'])
|
|
try_cnt = 0
|
|
while True:
|
|
retry = 0
|
|
if self.decompress_write:
|
|
retry = 10
|
|
ret, dmy = self.com_process_one_cmd('flash_write_check', cmd_id, bytearray(0))
|
|
if ret.startswith('OK'):
|
|
break
|
|
if try_cnt < self._checksum_err_retry_limit + retry:
|
|
printf('Retry')
|
|
try_cnt += 1
|
|
else:
|
|
self.error_code_print('0037')
|
|
return False
|
|
|
|
return True
|
|
|
|
def efuse_load_specified(self, file, maskfile, efusedata, efusedatamask, verify=0, security_write=False):
|
|
printf('========= efuse load =========')
|
|
if self._need_shake_hand != 0:
|
|
printf('Efuse load shake hand')
|
|
ret = self.img_load_shake_hand()
|
|
if ret is False:
|
|
return False
|
|
ret = self.efuse_load_main_process(file, maskfile, efusedata, efusedatamask, verify, security_write)
|
|
return ret
|
|
|
|
def efuse_load_main_process(self, file, maskfile, efusedata, efusedatamask, verify=0, security_write=False):
|
|
if efusedata != bytearray(0):
|
|
printf('Load data')
|
|
efuse_data = efusedata
|
|
mask_data = efusedatamask
|
|
else:
|
|
if file is not None:
|
|
printf('Load file: ', file)
|
|
fp = open_file(file, 'rb')
|
|
efuse_data = bytearray(fp.read()) + bytearray(0)
|
|
fp.close()
|
|
if verify:
|
|
fp = open_file(maskfile, 'rb')
|
|
mask_data = bytearray(fp.read()) + bytearray(0)
|
|
fp.close()
|
|
else:
|
|
mask_data = bytearray(0)
|
|
if len(efuse_data) > 4096:
|
|
printf('Decrypt efuse data')
|
|
efuse_data = efuse_data[4096:]
|
|
security_key, security_iv = get_security_key()
|
|
efuse_data = aes_decrypt_data(efuse_data, security_key, security_iv, 0)
|
|
else:
|
|
efuse_data = self._efuse_data
|
|
if verify:
|
|
mask_data = self._efuse_mask_data
|
|
else:
|
|
mask_data = bytearray(0)
|
|
if security_write:
|
|
if self.get_ecdh_shared_key() is not True:
|
|
return False
|
|
printf('Load efuse 0')
|
|
if security_write:
|
|
cmd_name = 'efuse_security_write'
|
|
else:
|
|
cmd_name = 'efuse_write'
|
|
cmd_id = hexstr_to_bytearray(self._com_cmds.get(cmd_name)['cmd_id'])
|
|
data_send = efuse_data[0:124] + bytearray(4)
|
|
if security_write:
|
|
data_send = self.ecdh_encrypt_data(data_send)
|
|
data_send = int_to_4bytearray_l(0) + data_send
|
|
ret, dmy = self.com_process_one_cmd(cmd_name, cmd_id, data_send)
|
|
if ret.startswith('OK') is False:
|
|
printf('Write Fail')
|
|
self.error_code_print('0021')
|
|
return False
|
|
if verify >= 1:
|
|
ret, read_data = self.efuse_read_main_process(0, 128, shakehand=0, file=None, security_read=security_write)
|
|
if ret is True and self.efuse_compare(read_data, mask_data[0:124] + bytearray(4), efuse_data[0:124] + bytearray(4)):
|
|
printf('Verify success')
|
|
else:
|
|
printf('Read: ')
|
|
printf(binascii.hexlify(read_data[0:124]).decode('utf-8'))
|
|
printf('Expected: ')
|
|
printf(binascii.hexlify(efuse_data[0:124]).decode('utf-8'))
|
|
printf('Verify fail')
|
|
self.error_code_print('0022')
|
|
return False
|
|
data_send = bytearray(12) + efuse_data[124:128]
|
|
if security_write:
|
|
data_send = self.ecdh_encrypt_data(data_send)
|
|
data_send = int_to_4bytearray_l(112) + data_send
|
|
ret, dmy = self.com_process_one_cmd(cmd_name, cmd_id, data_send)
|
|
if ret.startswith('OK') is False:
|
|
printf('Write Fail')
|
|
self.error_code_print('0021')
|
|
return False
|
|
if verify >= 1:
|
|
ret, read_data = self.efuse_read_main_process(112, 16, shakehand=0, file=None, security_read=security_write)
|
|
if ret is True and self.efuse_compare(read_data, bytearray(12) + mask_data[124:128], bytearray(12) + efuse_data[124:128]):
|
|
printf('Verify success')
|
|
else:
|
|
printf('Read: ')
|
|
printf(binascii.hexlify(read_data[12:16]))
|
|
printf('Expected: ')
|
|
printf(binascii.hexlify(efuse_data[124:128]))
|
|
printf('Verify fail')
|
|
self.error_code_print('0022')
|
|
return False
|
|
if len(efuse_data) > 128:
|
|
printf('Load efuse 1')
|
|
if security_write:
|
|
cmd_name = 'efuse_security_write'
|
|
else:
|
|
cmd_name = 'efuse_write'
|
|
cmd_id = hexstr_to_bytearray(self._com_cmds.get(cmd_name)['cmd_id'])
|
|
data_send = efuse_data[128:252] + bytearray(4)
|
|
if security_write:
|
|
data_send = self.ecdh_encrypt_data(data_send)
|
|
data_send = int_to_4bytearray_l(128) + data_send
|
|
ret, dmy = self.com_process_one_cmd(cmd_name, cmd_id, data_send)
|
|
if ret.startswith('OK') is False:
|
|
printf('Write Fail')
|
|
self.error_code_print('0021')
|
|
return False
|
|
if verify >= 1:
|
|
ret, read_data = self.efuse_read_main_process(128, 128, shakehand=0, file=None, security_read=security_write)
|
|
if ret is True and self.efuse_compare(read_data, mask_data[128:252] + bytearray(4), efuse_data[128:252] + bytearray(4)):
|
|
printf('Verify success')
|
|
else:
|
|
printf('Verify fail')
|
|
self.error_code_print('0022')
|
|
return False
|
|
data_send = bytearray(12) + efuse_data[252:256]
|
|
if security_write:
|
|
data_send = self.ecdh_encrypt_data(data_send)
|
|
data_send = int_to_4bytearray_l(240) + data_send
|
|
ret, dmy = self.com_process_one_cmd(cmd_name, cmd_id, data_send)
|
|
if ret.startswith('OK') is False:
|
|
printf('Write Fail')
|
|
self.error_code_print('0021')
|
|
return False
|
|
if verify >= 1:
|
|
ret, read_data = self.efuse_read_main_process(240, 16, shakehand=0, file=None, security_read=security_write)
|
|
if ret is True and self.efuse_compare(read_data, bytearray(12) + mask_data[252:256], bytearray(12) + efuse_data[252:256]):
|
|
printf('Verify success')
|
|
else:
|
|
printf('Verify fail')
|
|
self.error_code_print('0022')
|
|
printf('Finished')
|
|
return True
|
|
|
|
def efuse_read_main_process(self, start_addr, data_len, shakehand=0, file=None, security_read=False):
|
|
readdata = bytearray(0)
|
|
if shakehand != 0:
|
|
printf(FLASH_LOAD_SHAKE_HAND)
|
|
if self.img_load_shake_hand() is False:
|
|
return (False, None)
|
|
if security_read:
|
|
cmd_name = 'efuse_security_read'
|
|
else:
|
|
cmd_name = 'efuse_read'
|
|
cmd_id = hexstr_to_bytearray(self._com_cmds.get(cmd_name)['cmd_id'])
|
|
data_send = int_to_4bytearray_l(start_addr) + int_to_4bytearray_l(data_len)
|
|
ret, data_read = self.com_process_one_cmd(cmd_name, cmd_id, data_send)
|
|
printf('Read efuse ')
|
|
if ret.startswith('OK') is False:
|
|
self.error_code_print('0020')
|
|
return (False, None)
|
|
readdata += data_read
|
|
if security_read:
|
|
readdata = self.ecdh_decrypt_data(readdata)
|
|
printf('Finished')
|
|
if file is not None:
|
|
fp = open_file(file, 'wb+')
|
|
fp.write(readdata)
|
|
fp.close()
|
|
return (True, readdata)
|
|
|
|
def efuse_compare(self, read_data, maskdata, write_data):
|
|
i = 0
|
|
for i in range(len(read_data)):
|
|
compare_data = read_data[i] & maskdata[i]
|
|
if compare_data & write_data[i] != write_data[i]:
|
|
printf('compare fail: ', i)
|
|
printf(read_data[i], write_data[i])
|
|
return False
|
|
|
|
return True
|
|
|
|
def img_load_shake_hand(self):
|
|
isp_sh_time = self.get_isp_sh_time()
|
|
if self.interface.lower() == 'uart':
|
|
self.bflb_serial_object.repeat_init(self.device, self.speed, self.chip_type, self.chip_name)
|
|
if self._bflb_com_img_loader.toggle_boot_or_shake_hand(2, do_reset=False,
|
|
reset_hold_time=100,
|
|
shake_hand_delay=100,
|
|
reset_revert=True,
|
|
cutoff_time=0,
|
|
isp_mode_sign=(self.isp_mode_sign),
|
|
isp_timeout=isp_sh_time,
|
|
boot_load=False,
|
|
shake_hand_retry=2) != 'OK':
|
|
self.error_code_print('0001')
|
|
return False
|
|
else:
|
|
self.bflb_serial_object.if_init(self.device, self.speed, self.chip_type, self.chip_name)
|
|
if self.bflb_serial_object.if_shakehand(do_reset=False, reset_hold_time=100,
|
|
shake_hand_delay=100,
|
|
reset_revert=True,
|
|
cutoff_time=0,
|
|
shake_hand_retry=2,
|
|
isp_timeout=isp_sh_time,
|
|
boot_load=False) != 'OK':
|
|
self.error_code_print('0001')
|
|
return False
|
|
self._need_shake_hand = False
|
|
return True
|
|
|
|
def com_process_one_cmd(self, section, cmd_id, data_send):
|
|
data_read = bytearray(0)
|
|
data_len = int_to_2bytearray_l(len(data_send))
|
|
checksum = 0
|
|
checksum += bytearray_to_int(data_len[0:1]) + bytearray_to_int(data_len[1:2])
|
|
for char in data_send:
|
|
checksum += char
|
|
|
|
data = cmd_id + int_to_2bytearray_l(checksum & 255)[0:1] + data_len + data_send
|
|
if self.interface.lower() == 'uart':
|
|
self.bflb_serial_object.write(data)
|
|
if section in self._resp_cmds:
|
|
res, data_read = self.bflb_serial_object.deal_response()
|
|
else:
|
|
res = self.bflb_serial_object.deal_ack()
|
|
else:
|
|
self.bflb_serial_object.if_write(data)
|
|
if section in self._resp_cmds:
|
|
res, data_read = self.bflb_serial_object.if_deal_response()
|
|
else:
|
|
res = self.bflb_serial_object.if_deal_ack()
|
|
return (
|
|
res, data_read)
|
|
|
|
def get_suitable_conf_name(self, cfg_dir, flash_id):
|
|
conf_files = []
|
|
for home, dirs, files in os.walk(cfg_dir):
|
|
for filename in files:
|
|
if filename.split('_')[-1] == flash_id + '.conf':
|
|
conf_files.append(filename)
|
|
|
|
if len(conf_files) > 1:
|
|
printf('Flash id duplicate and alternative is:')
|
|
for i in range(len(conf_files)):
|
|
tmp = conf_files[i].split('.')[0]
|
|
printf('%d:%s' % (i + 1, tmp))
|
|
|
|
return conf_files[i]
|
|
if len(conf_files) == 1:
|
|
return conf_files[0]
|
|
return ''
|
|
|
|
def get_ecdh_shared_key(self, shakehand=0):
|
|
printf('========= get ecdh shared key =========')
|
|
publickey_file = 'utils/pem/publickey_uecc.pem'
|
|
if shakehand != 0:
|
|
printf('Shake hand')
|
|
ret = self.img_load_shake_hand()
|
|
if ret is False:
|
|
return
|
|
tmp_ecdh = bflb_ecdh.BflbEcdh()
|
|
self._ecdh_public_key = tmp_ecdh.create_public_key()
|
|
self._ecdh_private_key = binascii.hexlify(tmp_ecdh.ecdh.private_key.to_string()).decode('utf-8')
|
|
printf('ecdh public key')
|
|
printf(self._ecdh_public_key)
|
|
printf('ecdh private key')
|
|
printf(self._ecdh_private_key)
|
|
cmd_id = hexstr_to_bytearray(self._com_cmds.get('ecdh_get_pk')['cmd_id'])
|
|
data_send = bytearray.fromhex(self._ecdh_public_key)
|
|
ret, data_read = self.com_process_one_cmd('ecdh_get_pk', cmd_id, data_send)
|
|
if ret.startswith('OK') is True:
|
|
self._ecdh_peer_public_key = binascii.hexlify(data_read).decode('utf-8')
|
|
printf('ecdh peer key')
|
|
printf(self._ecdh_peer_public_key)
|
|
self._ecdh_shared_key = tmp_ecdh.create_shared_key(self._ecdh_peer_public_key[0:128])
|
|
printf('ecdh shared key')
|
|
printf(self._ecdh_shared_key)
|
|
cmd_id = hexstr_to_bytearray(self._com_cmds.get('ecdh_chanllenge')['cmd_id'])
|
|
data_send = bytearray(0)
|
|
ret, data_read = self.com_process_one_cmd('ecdh_chanllenge', cmd_id, data_send)
|
|
if ret.startswith('OK') is True:
|
|
printf('challenge data')
|
|
printf(binascii.hexlify(data_read).decode('utf-8'))
|
|
encrypted_data = data_read[0:32]
|
|
signature = data_read[32:96]
|
|
signature_r = data_read[32:64]
|
|
signature_s = data_read[64:96]
|
|
vk = ecdsa.VerifyingKey.from_pem(open_file('utils\\pem\\room_root_publickey_ecc.pem').read())
|
|
try:
|
|
ret = vk.verify(signature, (self.ecdh_decrypt_data(encrypted_data)),
|
|
hashfunc=(hashlib.sha256),
|
|
sigdecode=(ecdsa.util.sigdecode_string))
|
|
except Exception as err:
|
|
try:
|
|
printf(err)
|
|
finally:
|
|
err = None
|
|
del err
|
|
|
|
if ret is True:
|
|
return True
|
|
printf('Challenge verify fail')
|
|
return False
|
|
else:
|
|
printf('Challenge ack fail')
|
|
return False
|
|
else:
|
|
printf('Get shared key fail')
|
|
return False
|
|
|
|
def ecdh_encrypt_data(self, data):
|
|
cryptor = AES.new(bytearray.fromhex(self._ecdh_shared_key[0:32]), AES.MODE_CBC, bytearray(16))
|
|
ciphertext = cryptor.encrypt(data)
|
|
return ciphertext
|
|
|
|
def ecdh_decrypt_data(self, data):
|
|
cryptor = AES.new(bytearray.fromhex(self._ecdh_shared_key[0:32]), AES.MODE_CBC, bytearray(16))
|
|
plaintext = cryptor.decrypt(data)
|
|
return plaintext
|
|
|
|
def close_serial(self):
|
|
if self.bflb_serial_object:
|
|
try:
|
|
self.bflb_serial_object.close()
|
|
except Exception as e:
|
|
try:
|
|
printf(e)
|
|
finally:
|
|
e = None
|
|
del e
|
|
|
|
def clear_all_data(self):
|
|
if self.bflb_serial_object:
|
|
try:
|
|
self.bflb_serial_object.clear_buf()
|
|
except Exception as e:
|
|
try:
|
|
printf(e)
|
|
finally:
|
|
e = None
|
|
del e
|
|
|
|
def base_reset_cpu(self):
|
|
if self.bflb_serial_object:
|
|
self.bflb_serial_object.reset_cpu()
|
|
|
|
def object_status_clear(self):
|
|
self.bootinfo = None
|
|
self._macaddr_check = bytearray(0)
|
|
self._macaddr_check_status = False
|
|
|
|
def flash_read_status_reg_process(self, cmd, len, callback=None):
|
|
printf('========= flash read status register =========')
|
|
readdata = bytearray(0)
|
|
if self._need_shake_hand is not False:
|
|
printf(FLASH_LOAD_SHAKE_HAND)
|
|
if self.img_load_shake_hand() is False:
|
|
return (False, None)
|
|
cmd_id = hexstr_to_bytearray(self._com_cmds.get('flash_read_status_reg')['cmd_id'])
|
|
data_send = int_to_4bytearray_l(int(cmd, 16)) + int_to_4bytearray_l(len)
|
|
ret, data_read = self.com_process_one_cmd('flash_read_status_reg', cmd_id, data_send)
|
|
printf('Read flash status register ')
|
|
if ret.startswith('OK') is False:
|
|
self.error_code_print('0031')
|
|
return (False, None)
|
|
readdata += data_read
|
|
printf('readdata: ')
|
|
printf(binascii.hexlify(readdata))
|
|
printf('Finished')
|
|
return (
|
|
True, readdata)
|
|
|
|
def flash_write_status_reg_process(self, cmd, len, write_data, callback=None):
|
|
printf('========= flash write status register =========')
|
|
if self._need_shake_hand is not False:
|
|
printf(FLASH_LOAD_SHAKE_HAND)
|
|
if self.img_load_shake_hand() is False:
|
|
return (False, 'Flash load shake hand fail')
|
|
printf('write_data ', write_data)
|
|
cmd_id = hexstr_to_bytearray(self._com_cmds.get('flash_write_status_reg')['cmd_id'])
|
|
data_send = int_to_4bytearray_l(int(cmd, 16)) + int_to_4bytearray_l(len) + int_to_4bytearray_l(int(write_data, 16))
|
|
ret, data_read = self.com_process_one_cmd('flash_write_status_reg', cmd_id, data_send)
|
|
printf('Write flash status register ')
|
|
if ret.startswith('OK') is False:
|
|
self.error_code_print('0032')
|
|
return (False, 'Write fail')
|
|
printf('Finished')
|
|
return (True, None)
|
|
|
|
def bl_get_largest_addr(self, addrs, files):
|
|
maxlen = 0
|
|
datalen = 0
|
|
for i in range(len(addrs)):
|
|
if int(addrs[i], 16) > maxlen:
|
|
maxlen = int(addrs[i], 16)
|
|
if os.path.exists(files[i]):
|
|
datalen = os.path.getsize(files[i])
|
|
else:
|
|
datalen = os.path.getsize(os.path.join(app_path, files[i]))
|
|
|
|
return maxlen + datalen
|
|
|
|
def bl_get_file_data(self, files):
|
|
datas = []
|
|
for file in files:
|
|
if os.path.exists(file):
|
|
temp_path = file
|
|
else:
|
|
temp_path = os.path.join(app_path, file)
|
|
with open(temp_path, 'rb') as fp:
|
|
data = fp.read()
|
|
datas.append(data)
|
|
|
|
return datas
|
|
|
|
def bl_create_flash_default_data(self, length):
|
|
datas = bytearray(length)
|
|
for i in range(length):
|
|
datas[i] = 255
|
|
|
|
return datas
|
|
|
|
def bl_write_flash_img(self, d_addrs, d_files, flash_size='1M'):
|
|
whole_img_len = self.bl_get_largest_addr(d_addrs, d_files)
|
|
whole_img_data = self.bl_create_flash_default_data(whole_img_len)
|
|
whole_img_file = os.path.join(chip_path, self.chip_name, 'img_create', 'whole_flash_data.bin')
|
|
if os.path.exists(whole_img_file):
|
|
os.remove(whole_img_file)
|
|
filedatas = self.bl_get_file_data(d_files)
|
|
for i in range(len(d_addrs)):
|
|
start_addr = int(d_addrs[i], 16)
|
|
whole_img_data[start_addr:start_addr + len(filedatas[i])] = filedatas[i]
|
|
|
|
if not os.path.exists(os.path.dirname(whole_img_file)):
|
|
os.makedirs(os.path.dirname(whole_img_file))
|
|
fp = open(whole_img_file, 'wb+')
|
|
fp.write(whole_img_data)
|
|
fp.close()
|
|
|
|
def get_flash_pin_from_bootinfo(self, chiptype, bootinfo):
|
|
return 128
|
|
# okay decompiling ./libs/base/bflb_base_eflash_loader.pyc
|