Overriding Default Chip Drivers

[中文]

Warning

Customizing SPI Flash Chip Drivers is considered an "expert" feature. Users should only do so at their own risk. (See the notes below)

During the SPI Flash driver's initialization (i.e., esp_flash_init()), there is a chip detection step during which the driver iterates through a Default Chip Driver List and determine which chip driver can properly support the currently connected flash chip. The Default Chip Drivers are provided by the ESP-IDF, thus are updated in together with each ESP-IDF version. However ESP-IDF also allows users to customize their own chip drivers.

Users should note the following when customizing chip drivers:

  1. You may need to rely on some non-public ESP-IDF functions, which have slight possibility to change between ESP-IDF versions. On the one hand, these changes may be useful bug fixes for your driver, on the other hand, they may also be breaking changes (i.e., breaks your code).

  2. Some ESP-IDF bug fixes to other chip drivers are not automatically applied to your own custom chip drivers.

  3. If the protection of flash is not handled properly, there may be some random reliability issues.

  4. If you update to a newer ESP-IDF version that has support for more chips, you will have to manually add those new chip drivers into your custom chip driver list. Otherwise the driver will only search for the drivers in custom list you provided.

Steps For Creating Custom Chip Drivers and Overriding the ESP-IDF Default Driver List

Bootloader Flash Driver

To implement bootloader driver in order that the ESP chip can boot successfully, please read this part carefully. Wrong code will make ESP chip fail to boot or introduce random issues. And issues in bootloader is not fixable via OTA. Please fully comprehend below parts before making any changes.

Currently there are two parts to override in bootloader: bootloader_flash_unlock and bootloader_flash_qe_support_list (and its size). See below sections.

Ensure that all the commands should be sent out and recognized successfully. Otherwise if the flash receives an unknown command, it may ignore silently and continue working while the protection is not re-enabled. This might lead to mis-writing/erasing, mis-locked issues.

All bootloader flash driver functions reside in a component for the bootloader, which should be placed under bootloader_components. For example, storage/custom_flash_driver/bootloader_components/. Only the components under bootloader_components will be added into the component list of bootloader automatically.

There is slight difference in the CMakeLists.txt of the component. The customized flash functions take effect by overriding existing weak ones. In order to link them into the bootloader, linker argument -u should be used against the overriding functions in the CMakeLists.txt. See storage/custom_flash_driver/bootloader_components/bootloader_flash/CMakeLists.txt.

Bootloader flash unlock

bootloader_flash_unlock function is used to unlock flash write protection, which means once the flash is locked by accident, IDF is able to unlock it. By default, the unlock function will clear all bits in Status Register and Configuration Register except QE bit.

Please don't touch QE bits, otherwise, your chip will not be able to run under QUAD modes.

Please check the default case below. If your flash have different behavior from that, some modifications are required.

You can start from copying bootloader_flash_unlock_default from components/bootloader_support/bootloader_flash/src/bootloader_flash.c into a file in your customized component folder before modifying it. The function should be renamed to bootloader_flash_unlock to override the IDF bootloader function. For example, storage/custom_flash_driver/bootloader_components/bootloader_flash/bootloader_flash_unlock_custom.c.

After that, your implementation of bootloader_flash_unlock will be linked instead of IDF default one, which has been declared as a weak symbol. So when IDF 2nd bootloader boots, your implementation of bootloader_flash_unlock will be executed. If you want to make sure your function is truly linked, you can enable CONFIG_BOOTLOADER_LOG_LEVEL_DEBUG and you will see a Using overridden bootloader_flash_unlock log.

In the default function, we have provided 3 common behaviors, please check first:

  • case 1 (Default): Please check whether QE bit in your flash chip is placed at bit 1 in status register-2 written by the second byte after command 01H (WRSR). If so, this matches the default behavior and you don't need to anything else.

  • case 2: Please check whether QE bit in your flash chip is placed at bit 1 in status register-2 to be written by command 31H. (And your flash doesn't support case 1, using 01H+2bytes to write it). If so , you need to add your chip ID in function is_qe_bit_in_bit1_at_reg2(), for example:

IRAM_ATTR bool is_qe_bit_in_bit1_at_reg2(const esp_rom_spiflash_chip_t* chip)
{
   bool ret = true;
   switch (chip->device_id) {
   /****GD series***/
   case 0xC84016:
   case 0xC84017:
   case 0xC84018:
      break;
   /**** your flash series ****/
   case /*your flash ID*/:
      break;
   default:
      ret = false;
   }
   return ret;
}
  • case 3: Please check whether QE bit in your flash chip is placed at bit 6 in status register-1 to be written by command 01H. If so, you need to add your chip ID in function is_qe_bit_in_bit6_at_reg1(), for example:

IRAM_ATTR bool is_qe_bit_in_bit6_at_reg1(const esp_rom_spiflash_chip_t* chip)
{
   bool ret = true;
   switch (chip->device_id) {
   /***ISSI series***/
   case 0x9D4016:
   case 0x9D4017:
      break;

   /***MXIC series***/
   case 0xC22016:
   case 0xC22017:
      break;

   /****your clash series***/
   case /*your flash ID*/:
      break;
   default:
      ret = false;
   }
   return ret;
}

Bootloader flash quad mode support

Pointer bootloader_flash_qe_support_list is used for iteration in bootloader for selecting the correct behavior to enable flash chip work under QUAD mode. In order to make flash works under QUAD mode, enable QE bit in flash status register is necessary. If you want to use your flash chip under QUAD mode, please read this part and make implementation for yourself.

  • case 1: If QE bit is placed at bit1 in status register-2 to be written by command 31H, nothing needs to be done because this is the default behavior.

  • case 2: If QE bit on your chip is placed at different places, or need to use different command. Please add your own support.

To add your own support, you can start from copying the bootloader_flash_qe_support_list_user function from flash_qio_mode.c , into your file, renaming to bootloader_flash_qe_support_list. Please also define a corresponding bootloader_flash_qe_list_count.

Add your flash into it, with the chip's name, ID, and the functions to write registers etc. into bootloader_flash_qio_support_list. You can also reuse the existing functions like bootloader_read_status_8b_rdsr.

If functions there cannot fully cover your usage, you can define your own function with bootloader_execute_flash_command, like bootloader_read_status_otp_mode_8b, and bootloader_write_status_otp_mode_8b. For example, bootloader_flash_custom.c .

Put everything together:

const DRAM_ATTR bootloader_qio_info_t bootloader_flash_qe_support_list_user[] = {
   /*   Manufacturer,   mfg_id, flash_id, id mask, Read Status,                Write Status,               QIE Bit */
   { "MXIC",        0xC2,   0x2000, 0xFF00,    bootloader_read_status_8b_rdsr,        bootloader_write_status_8b_wrsr,       6 },
   { "ISSI",        0x9D,   0x4000, 0xCF00,    bootloader_read_status_8b_rdsr,        bootloader_write_status_8b_wrsr,       6 },
   { "WinBond",     0xEF,   0x4000, 0xFF00,    bootloader_read_status_16b_rdsr_rdsr2, bootloader_write_status_16b_wrsr,      9 },
   { "GD",          0xC8,   0x4000, 0xFFFF,    bootloader_read_status_16b_rdsr_rdsr2, bootloader_write_status_16b_wrsr,      9 },
   { "XM25QU64A",   0x20,   0x3817, 0xFFFF,    bootloader_read_status_8b_xmc25qu64a,  bootloader_write_status_8b_xmc25qu64a, 6 },
   { "TH",          0xCD,   0x6000, 0xFF00,    bootloader_read_status_16b_rdsr_rdsr2, bootloader_write_status_16b_wrsr,      9 },
   { "EON",         0x1C,   0x7000, 0xFF00,    bootloader_read_status_otp_mode_8b,    bootloader_write_status_otp_mode_8b,   6 },

   /* Final entry is default entry, if no other IDs have matched.

      This approach works for chips including:
      GigaDevice (mfg ID 0xC8, flash IDs including 4016),
      FM25Q32 (QOUT mode only, mfg ID 0xA1, flash IDs including 4016)
      BY25Q32 (mfg ID 0x68, flash IDs including 4016)
   */
   { NULL,          0xFF,    0xFFFF, 0xFFFF,   bootloader_read_status_8b_rdsr2,       bootloader_write_status_8b_wrsr2,      1 },
};
const DRAM_ATTR bootloader_qio_info_t* bootloader_flash_qe_support_list = bootloader_flash_qe_support_list_user;
uint8_t DRAM_ATTR bootloader_flash_qe_list_count = (sizeof(bootloader_flash_qe_support_list_user) / sizeof(bootloader_qio_info_t));

App flash driver

Generic Flash driver

The flash driver in the application is used to read, write, erase, etc. to save data. And some advanced features like the OTA. Below is the guide on how to customize the driver for your own flash model.

  • STEP 1: The last item of default_registered_chips should be the generic chip driver . When your flash chip cannot match any one chip driver above, it will use the generic driver. Please check what behavior of your flash is different from the generic driver, including but not limited to different commands, different dummys, different data bytes, different status registers.

  • STEP 2: If you have found something different from the generic driver, you need to implement your own chip driver. Create a new file naming spi_flash_chip_<vendor>.c implementing specific behavior there, and copy the esp_flash_chip_generic structure into it as a start. Don't forget to include "spi_flash_chip_generic.h". Here is an example esp_flash_nor .

  • STEP 3: Implement the functions with difference and point to them from the spi_flash_chip_t. Note: if some behavior of your flash is same as the generic one, just keep the function of generic driver and no need to customize them. Only the different part should be implemented. Here is an example:

Important

Flash work for suspend (means you want to enable CONFIG_SPI_FLASH_AUTO_SUSPEND) should be tested carefully and systematically due to different flash hardware design. If you want to use suspend feature for massive production, please contact espressif business support team.

const DRAM_ATTR spi_flash_chip_t esp_flash_chip_eon = {
   .name = chip_name,
   .timeout = &spi_flash_chip_generic_timeout,  /*<! default behavior*/
   .probe = spi_flash_chip_eon_probe,   /*<! EON specific */
   .reset = spi_flash_chip_generic_reset,
   .detect_size = spi_flash_chip_generic_detect_size,
   .erase_chip = spi_flash_chip_generic_erase_chip,
   .erase_sector = spi_flash_chip_generic_erase_sector,
   .erase_block = spi_flash_chip_generic_erase_block,
   .sector_size = 4 * 1024,
   .block_erase_size = 64 * 1024,

   .get_chip_write_protect = spi_flash_chip_generic_get_write_protect,
   .set_chip_write_protect = spi_flash_chip_generic_set_write_protect,

   .num_protectable_regions = 0,
   .protectable_regions = NULL,
   .get_protected_regions = NULL,
   .set_protected_regions = NULL,

   .read = spi_flash_chip_generic_read,
   .write = spi_flash_chip_generic_write,
   .program_page = spi_flash_chip_generic_page_program,
   .page_size = 256,
   .write_encrypted = spi_flash_chip_generic_write_encrypted,

   .wait_idle = spi_flash_chip_generic_wait_idle,
   .set_io_mode = spi_flash_chip_eon_set_io_mode,
   .get_io_mode = spi_flash_chip_eon_get_io_mode,

   .read_reg = spi_flash_chip_generic_read_reg,
   .yield = spi_flash_chip_generic_yield,
   .sus_setup = spi_flash_chip_eon_suspend_cmd_conf,
   .get_chip_caps = spi_flash_chip_eon_get_caps,
};

Note

  • When writing your own flash chip driver, you can set your flash chip capabilities through spi_flash_chip_***(vendor)_get_caps and points the function pointer get_chip_caps for protection to the spi_flash_chip_***_get_caps function. The steps are as follows.

    1. Please check whether your flash chip have the capabilities listed in spi_flash_caps_t by checking the flash datasheet.

    2. Write a function named spi_flash_chip_***(vendor)_get_caps. Take the example below as a reference (if the flash support suspend and read unique id).

    3. Points the pointer get_chip_caps (in spi_flash_chip_t) to the function mentioned above.

    spi_flash_caps_t spi_flash_chip_***(vendor)_get_caps(esp_flash_t *chip)
    {
       spi_flash_caps_t caps_flags = 0;
       // 32-bit-address flash is not supported
       caps_flags |= SPI_FLAHS_CHIP_CAP_SUSPEND;
       // flash read unique id.
       caps_flags |= SPI_FLASH_CHIP_CAP_UNIQUE_ID;
       return caps_flags;
    }
    
    const spi_flash_chip_t esp_flash_chip_eon = {
       // Other function pointers
       .get_chip_caps = spi_flash_chip_eon_get_caps,
    };
    
  • You also can see how to implement this in the example storage/custom_flash_driver. This example demonstrates how to override the default chip driver list.

  • STEP 4: Add a header (e.g. spi_flash_chip_<vendor>.h) and extern declare this structure, so that other component or source code can reuse this structure. Wrap all your chip drivers (source files as well as their headers) with a chip-driver component. Add the include path and the source file to component CMakeLists.txt.

  • STEP 5: The linker.lf is used to put every chip driver that you are going to use whilst cache is disabled into internal RAM. See Linker Script Generation for more details. Make sure this file covers all the source files that you add.

  • STEP 6: Add a new component in your project, e.g., custom_chip_driver. List your chip object under default_registered_chips in custom_chip_driver/chip_drivers.c. Then enable the CONFIG_SPI_FLASH_OVERRIDE_CHIP_DRIVER_LIST config option. This prevents compilation and linking of the Default Chip Driver List (default_registered_chips) provided by ESP-IDF. Instead, the linker searches for the structure of the same name (default_registered_chips) that must be provided by the user. You can refer to storage/custom_flash_driver/components/custom_chip_driver/chip_drivers.c.

  • STEP 7: Build your project, and you will see the new flash driver is used.

Example

See storage/custom_flash_driver, which make use of an external component to add support for some customized flash. This example mainly shows how to override IDF driver, and the external component shows how to implement customized drivers.


Was this page helpful?