# 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_flash_select.py
import os, re
from libs import bflb_utils
from libs.bflb_utils import app_path, chip_path, conf_sign
from libs.bflb_configobj import BFConfigParser

def get_int_mask(pos, length):
    ones = '11111111111111111111111111111111'
    zeros = '00000000000000000000000000000000'
    mask = ones[0:32 - pos - length] + zeros[0:length] + ones[0:pos]
    return int(mask, 2)


def update_flash_para_from_cfg(config_keys, config_file):
    section = 'FLASH_CFG'
    cfg = BFConfigParser()
    cfg.read(config_file)
    filelen = 0
    offset = 0
    minOffset = 4294967295
    maxOffset = 0
    flashCrcOffset = 0
    crcOffset = 0
    if config_keys.get('crc32') != None:
        crcOffset = int(config_keys.get('crc32')['offset'], 10)
    if config_keys.get('flashcfg_crc32') != None:
        flashCrcOffset = int(config_keys.get('flashcfg_crc32')['offset'], 10)
    for key in cfg.options(section):
        if config_keys.get(key) == None:
            continue
        else:
            offset = int(config_keys.get(key)['offset'], 10)
            if offset < minOffset:
                minOffset = offset
        if offset > maxOffset:
            maxOffset = offset

    filelen = maxOffset - minOffset + 4
    data = bytearray(filelen)
    for key in cfg.options(section):
        if config_keys.get(key) == None:
            bflb_utils.printf(key + ' not exist')
            continue
        else:
            val = cfg.get(section, key)
            if val.startswith('0x'):
                val = int(val, 16)
            else:
                val = int(val, 10)
            offset = int(config_keys.get(key)['offset'], 10) - minOffset
            pos = int(config_keys.get(key)['pos'], 10)
            bitlen = int(config_keys.get(key)['bitlen'], 10)
            oldval = bflb_utils.bytearray_to_int(bflb_utils.bytearray_reverse(data[offset:offset + 4]))
            newval = (oldval & get_int_mask(pos, bitlen)) + (val << pos)
            data[offset:offset + 4] = bflb_utils.int_to_4bytearray_l(newval)

    return (minOffset, filelen, data, flashCrcOffset, crcOffset)


def update_flash_cfg_data_do(chiptype, flash_id):
    cfg_dir = app_path + '/utils/flash/' + chiptype + '/'
    sub_module = __import__(('libs.base.' + chiptype), fromlist=[chiptype])
    conf_name = sub_module.flash_select_do.get_suitable_file_name(cfg_dir, flash_id)
    if os.path.isfile(cfg_dir + conf_name) == False:
        return (None, None, None, None, None)
    return update_flash_para_from_cfg(sub_module.bootheader_cfg_keys.bootheader_cfg_keys, cfg_dir + conf_name)


def flash_bootheader_config_check(chiptype, flashid, file, parafile):
    magic_code = 1347307074
    flash_magic_code = 1195787078
    offset, flashCfgLen, data, flashCrcOffset, crcOffset = update_flash_cfg_data_do(chiptype, flashid)
    if data is None:
        offset = 12
        flashCfgLen = 84
    if parafile != '':
        if data != None:
            fp = open(os.path.join(app_path, parafile), 'wb')
            fp.write(data)
            fp.close()
    fp = open(os.path.join(app_path, file), 'rb')
    rdata = bytearray(fp.read())
    fp.close()
    i = 0
    length = 128
    flashCfg = bytearray(256)
    while i < length:
        if rdata[i:i + 4] == bflb_utils.int_to_4bytearray_l(magic_code):
            if rdata[i + 8:i + 12] == bflb_utils.int_to_4bytearray_l(flash_magic_code):
                if data != None:
                    data[2:4] = rdata[i + 14:i + 16]
                flashCfg = rdata[i + offset:i + offset + flashCfgLen]
                if data != None:
                    if not data != flashCfg and flashCfg[13:14] != b'\xff' or flashCfg[13:14] != b'\x00':
                        return False
                else:
                    if flashCfg[13:14] != b'\xff':
                        if flashCfg[13:14] != b'\x00':
                            return False
        i += 4

    return True


def update_flash_cfg_data(chiptype, flash_id, cfg, bh_cfg_file, cfg_key):
    cfg2 = BFConfigParser()
    cfg2.read(bh_cfg_file)
    magic_code = cfg2.get(cfg_key, 'magic_code')
    magic_code = int(magic_code, 16)
    flash_magic_code = cfg2.get(cfg_key, 'flashcfg_magic_code')
    flash_magic_code = int(flash_magic_code, 16)
    sub_module = __import__(('libs.' + chiptype), fromlist=[chiptype])
    offset, flashCfgLen, data, flashCrcOffset, crcOffset = update_flash_cfg_data_do(chiptype, flash_id)
    para_file = cfg.get('FLASH_CFG', 'flash_para')
    if para_file != '':
        fp = open(os.path.join(app_path, para_file), 'wb')
        fp.write(data)
        fp.close()
    flash_file = re.compile('\\s+').split(cfg.get('FLASH_CFG', 'file'))
    for f in flash_file:
        fp = open(os.path.join(app_path, f), 'rb')
        rdata = bytearray(fp.read())
        fp.close()
        i = 0
        length = len(rdata)
        while i < length:
            if rdata[i:i + 4] == bflb_utils.int_to_4bytearray_l(magic_code):
                if rdata[i + 8:i + 12] == bflb_utils.int_to_4bytearray_l(flash_magic_code):
                    data[2:4] = rdata[i + 14:i + 16]
                    flashCfg = rdata[i + offset:i + offset + flashCfgLen]
                    if data != flashCfg:
                        return False
            i += 4

    return True


def check_basic_flash_cfg(cfg_file, section):
    if os.path.isfile(cfg_file) is False:
        return False
    cfg = BFConfigParser()
    cfg.read(cfg_file)
    if cfg.has_option(section, 'mfg_id'):
        if cfg.get(section, 'mfg_id') == '0xff' or cfg.get(section, 'mfg_id') == '0xFF':
            cfg.set(section, 'io_mode', '0x10')
            cfg.set(section, 'cont_read_support', '0')
            cfg.write(cfg_file, 'w+')
            return True
        if cfg.get(section, 'mfg_id') == '0x00':
            return True
    return False


def update_flash_cfg(chipname, chiptype, flash_id, file=None, create=False, section=None):
    sub_module = __import__(('libs.' + chiptype), fromlist=[chiptype])
    if check_basic_flash_cfg(file, section):
        return True
    if sub_module.flash_select_do.update_flash_cfg_do(chipname, chiptype, flash_id, file, create, section) == False:
        return False
    return True


def get_supported_flash(chiptype):
    sub_module = __import__(('libs.' + chiptype), fromlist=[chiptype])
    return sub_module.flash_select_do.get_supported_flash_do()
# okay decompiling ./libs/base/bflb_flash_select.pyc