mirror of
https://github.com/espressif/esp-idf
synced 2025-03-09 17:19:09 -04:00
123 lines
4.4 KiB
Python
123 lines
4.4 KiB
Python
# SPDX-FileCopyrightText: 2022-2024 Espressif Systems (Shanghai) CO LTD
|
|
# SPDX-License-Identifier: Apache-2.0
|
|
import json
|
|
import logging
|
|
import os
|
|
import tempfile
|
|
from typing import Any
|
|
|
|
import espefuse
|
|
import pytest
|
|
from _pytest.fixtures import FixtureRequest
|
|
from _pytest.monkeypatch import MonkeyPatch
|
|
from pytest_embedded_idf.app import FlashFile
|
|
from pytest_embedded_idf.serial import IdfSerial
|
|
|
|
|
|
# This is a custom IdfSerial class to support custom functionality
|
|
# which is required only for this test
|
|
class EfuseFlashEncSerial(IdfSerial):
|
|
|
|
@IdfSerial.use_esptool()
|
|
def write_flash_no_enc(self) -> None:
|
|
self.app.flash_settings['encrypt'] = False
|
|
flash_files = []
|
|
for file in self.app.flash_files:
|
|
# Set encrypted flag to false for each file.
|
|
flash_files.append(file._replace(encrypted=False))
|
|
# Replace the original tuple with modified tuple with all the files marked as unencrypted.
|
|
self.app.flash_files = tuple(flash_files)
|
|
# Now flash the files
|
|
self.flash()
|
|
|
|
def bootloader_flash(self) -> None:
|
|
"""
|
|
Flash bootloader.
|
|
|
|
:return: None
|
|
"""
|
|
logging.info('Flashing bootloader')
|
|
bootloader_path = os.path.join(self.app.binary_path, 'bootloader', 'bootloader.bin')
|
|
offs = int(self.app.sdkconfig.get('BOOTLOADER_OFFSET_IN_FLASH', 0))
|
|
logging.info('bootloader offset is {0}'.format(hex(offs)))
|
|
prev_flash_files = self.app.flash_files
|
|
flash_files = []
|
|
flash_files.append(
|
|
FlashFile(
|
|
offs,
|
|
bootloader_path,
|
|
False,
|
|
)
|
|
)
|
|
self.app.flash_files = flash_files
|
|
self.app.flash_settings['encrypt'] = False
|
|
self.flash()
|
|
# Restore self.app.flash files to original value
|
|
self.app.flash_files = prev_flash_files
|
|
|
|
def erase_field_on_emul_efuse_by_name(self, efuse_names: list) -> None:
|
|
pos_of_bits = []
|
|
for name in efuse_names:
|
|
pos_of_bits.append(self.get_efuse_offset(name))
|
|
self.erase_field_on_emul_efuse(pos_of_bits)
|
|
|
|
def erase_field_on_emul_efuse(self, pos_of_bits: list) -> None:
|
|
emul_efuse_bin_path = os.path.join(self.app.binary_path, 'emul_efuse.bin')
|
|
self.dump_flash(output=emul_efuse_bin_path, partition='emul_efuse')
|
|
logging.info('Erasing field on emulated efuse')
|
|
|
|
def erase_bit(pos_of_bit: int) -> None:
|
|
nbytes, nbits = divmod(pos_of_bit, 8)
|
|
with open(emul_efuse_bin_path, 'r+b') as f:
|
|
f.seek(nbytes)
|
|
data = ord(f.read(1))
|
|
data &= ~(1 << nbits)
|
|
f.seek(-1, os.SEEK_CUR)
|
|
f.write(bytes([data]))
|
|
|
|
for pos_of_bit in sorted(pos_of_bits):
|
|
erase_bit(pos_of_bit)
|
|
|
|
offs = self.app.partition_table['emul_efuse']['offset']
|
|
logging.info('emul efuse offset is {0}'.format(hex(offs)))
|
|
prev_flash_files = self.app.flash_files
|
|
|
|
flash_files = []
|
|
flash_files.append(
|
|
FlashFile(
|
|
offs,
|
|
emul_efuse_bin_path,
|
|
False,
|
|
)
|
|
)
|
|
self.app.flash_files = flash_files
|
|
self.app.flash_settings['encrypt'] = False
|
|
self.flash()
|
|
self.app.flash_files = prev_flash_files
|
|
|
|
def get_efuse_offset(self, efuse_name: str) -> Any:
|
|
with tempfile.NamedTemporaryFile(suffix='.json') as temp_file:
|
|
temp_file_path = temp_file.name
|
|
espefuse.main(f'--virt -c {self.target} summary --format json --file {temp_file_path}'.split())
|
|
with open(temp_file_path, 'r', encoding='utf-8') as file:
|
|
efuse_summary = json.load(file)
|
|
if efuse_name in efuse_summary:
|
|
data = efuse_summary[efuse_name]
|
|
offset = int(data['word'] * 32) + data['pos']
|
|
print(f'{efuse_name} offset = {offset}')
|
|
return offset
|
|
else:
|
|
raise ValueError(f"eFuse '{efuse_name}' not found in the summary.")
|
|
|
|
|
|
@pytest.fixture(scope='module')
|
|
def monkeypatch_module(request: FixtureRequest) -> MonkeyPatch:
|
|
mp = MonkeyPatch()
|
|
request.addfinalizer(mp.undo)
|
|
return mp
|
|
|
|
|
|
@pytest.fixture(scope='module', autouse=True)
|
|
def replace_dut_class(monkeypatch_module: MonkeyPatch) -> None:
|
|
monkeypatch_module.setattr('pytest_embedded_idf.IdfSerial', EfuseFlashEncSerial)
|