Merge branch 'refactor/remove_spi_flash_include' into 'master'

refactor: remove dependency on spi_flash include for sector size

See merge request espressif/esp-idf!31392

Author: RathiSonika

components/app_update/esp_ota_ops.c

@@ -1,5 +1,5 @@
 /*
- * SPDX-FileCopyrightText: 2015-2023 Espressif Systems (Shanghai) CO LTD
+ * SPDX-FileCopyrightText: 2015-2024 Espressif Systems (Shanghai) CO LTD
  *
  * SPDX-License-Identifier: Apache-2.0
  */
@@ -80,7 +80,7 @@ static const esp_partition_t *read_otadata(esp_ota_select_entry_t *two_otadata)
         return NULL;
     } else {
         memcpy(&two_otadata[0], result, sizeof(esp_ota_select_entry_t));
-        memcpy(&two_otadata[1], result + SPI_FLASH_SEC_SIZE, sizeof(esp_ota_select_entry_t));
+        memcpy(&two_otadata[1], result + otadata_partition->erase_size, sizeof(esp_ota_select_entry_t));
         esp_partition_munmap(ota_data_map);
     }
     return otadata_partition;
@@ -149,7 +149,7 @@ esp_err_t esp_ota_begin(const esp_partition_t *partition, size_t image_size, esp
         if ((image_size == 0) || (image_size == OTA_SIZE_UNKNOWN)) {
             ret = esp_partition_erase_range(partition, 0, partition->size);
         } else {
-            const int aligned_erase_size = (image_size + SPI_FLASH_SEC_SIZE - 1) & ~(SPI_FLASH_SEC_SIZE - 1);
+            const int aligned_erase_size = (image_size + partition->erase_size - 1) & ~(partition->erase_size - 1);
             ret = esp_partition_erase_range(partition, 0, aligned_erase_size);
         }
         if (ret != ESP_OK) {
@@ -192,14 +192,14 @@ esp_err_t esp_ota_write(esp_ota_handle_t handle, const void *data, size_t size)
         if (it->handle == handle) {
             if (it->need_erase) {
                 // must erase the partition before writing to it
-                uint32_t first_sector = it->wrote_size / SPI_FLASH_SEC_SIZE; // first affected sector
-                uint32_t last_sector = (it->wrote_size + size - 1) / SPI_FLASH_SEC_SIZE; // last affected sector
+                uint32_t first_sector = it->wrote_size / it->part->erase_size; // first affected sector
+                uint32_t last_sector = (it->wrote_size + size - 1) / it->part->erase_size; // last affected sector
 
                 ret = ESP_OK;
-                if ((it->wrote_size % SPI_FLASH_SEC_SIZE) == 0) {
-                    ret = esp_partition_erase_range(it->part, it->wrote_size, ((last_sector - first_sector) + 1) * SPI_FLASH_SEC_SIZE);
+                if ((it->wrote_size % it->part->erase_size) == 0) {
+                    ret = esp_partition_erase_range(it->part, it->wrote_size, ((last_sector - first_sector) + 1) * it->part->erase_size);
                 } else if (first_sector != last_sector) {
-                    ret = esp_partition_erase_range(it->part, (first_sector + 1) * SPI_FLASH_SEC_SIZE, (last_sector - first_sector) * SPI_FLASH_SEC_SIZE);
+                    ret = esp_partition_erase_range(it->part, (first_sector + 1) * it->part->erase_size, (last_sector - first_sector) * it->part->erase_size);
                 }
                 if (ret != ESP_OK) {
                     return ret;
@@ -369,11 +369,11 @@ static esp_err_t rewrite_ota_seq(esp_ota_select_entry_t *two_otadata, uint32_t s
 
     two_otadata[sec_id].ota_seq = seq;
     two_otadata[sec_id].crc = bootloader_common_ota_select_crc(&two_otadata[sec_id]);
-    esp_err_t ret = esp_partition_erase_range(ota_data_partition, sec_id * SPI_FLASH_SEC_SIZE, SPI_FLASH_SEC_SIZE);
+    esp_err_t ret = esp_partition_erase_range(ota_data_partition, sec_id * ota_data_partition->erase_size, ota_data_partition->erase_size);
     if (ret != ESP_OK) {
         return ret;
     } else {
-        return esp_partition_write(ota_data_partition, SPI_FLASH_SEC_SIZE * sec_id, &two_otadata[sec_id], sizeof(esp_ota_select_entry_t));
+        return esp_partition_write(ota_data_partition, ota_data_partition->erase_size * sec_id, &two_otadata[sec_id], sizeof(esp_ota_select_entry_t));
     }
 }
 
@@ -906,7 +906,7 @@ esp_err_t esp_ota_erase_last_boot_app_partition(void)
     }
 
     int sec_id = inactive_otadata;
-    err = esp_partition_erase_range(ota_data_partition, sec_id * SPI_FLASH_SEC_SIZE, SPI_FLASH_SEC_SIZE);
+    err = esp_partition_erase_range(ota_data_partition, sec_id * ota_data_partition->erase_size, ota_data_partition->erase_size);
     if (err != ESP_OK) {
         return err;
     }

components/app_update/test_apps/test_app_update/main/test_switch_ota.c

@@ -1,5 +1,5 @@
 /*
- * SPDX-FileCopyrightText: 2021-2023 Espressif Systems (Shanghai) CO LTD
+ * SPDX-FileCopyrightText: 2021-2024 Espressif Systems (Shanghai) CO LTD
  *
  * SPDX-License-Identifier: Apache-2.0
  */
@@ -177,7 +177,7 @@ static void erase_ota_data(void)
 {
     const esp_partition_t *data_partition = esp_partition_find_first(ESP_PARTITION_TYPE_DATA, ESP_PARTITION_SUBTYPE_DATA_OTA, NULL);
     TEST_ASSERT_NOT_EQUAL(NULL, data_partition);
-    TEST_ESP_OK(esp_partition_erase_range(data_partition, 0, 2 * SPI_FLASH_SEC_SIZE));
+    TEST_ESP_OK(esp_partition_erase_range(data_partition, 0, 2 * data_partition->erase_size));
 }
 
 /* @brief Reboots ESP using mode deep sleep. This mode guaranty that RTC_DATA_ATTR variables is not reset.
@@ -251,7 +251,7 @@ static void get_ota_data(const esp_partition_t *otadata_partition, esp_ota_selec
         TEST_ASSERT_NOT_EQUAL(NULL, ota_select_map);
 
         memcpy(ota_data_0, ota_select_map, sizeof(esp_ota_select_entry_t));
-        memcpy(ota_data_1, (uint8_t *)ota_select_map + SPI_FLASH_SEC_SIZE, sizeof(esp_ota_select_entry_t));
+        memcpy(ota_data_1, (uint8_t *)ota_select_map + otadata_partition->erase_size, sizeof(esp_ota_select_entry_t));
         bootloader_munmap(ota_select_map);
     }
 }
@@ -264,7 +264,7 @@ static void get_ota_data(const esp_partition_t *otadata_partition, esp_ota_selec
  */
 static void write_ota_data(const esp_partition_t *otadata_partition, esp_ota_select_entry_t *ota_data, int sec_id)
 {
-    esp_partition_write(otadata_partition, SPI_FLASH_SEC_SIZE * sec_id, &ota_data[sec_id], sizeof(esp_ota_select_entry_t));
+    esp_partition_write(otadata_partition, otadata_partition->erase_size * sec_id, &ota_data[sec_id], sizeof(esp_ota_select_entry_t));
 }
 
 /* @brief Makes a corrupt of ota_data.

components/esp_driver_spi/test_apps/master/main/test_spi_bus_lock.c

@@ -1,5 +1,5 @@
 /*
- * SPDX-FileCopyrightText: 2021-2022 Espressif Systems (Shanghai) CO LTD
+ * SPDX-FileCopyrightText: 2021-2024 Espressif Systems (Shanghai) CO LTD
  *
  * SPDX-License-Identifier: Apache-2.0
  */
@@ -10,7 +10,6 @@
 #include "esp_flash_spi_init.h"
 #include "test_utils.h"
 #include "test_spi_utils.h"
-#include "spi_flash_mmap.h"
 #include "unity.h"
 
 #if CONFIG_IDF_TARGET_ESP32
@@ -146,7 +145,7 @@ static void write_large_buffer(esp_flash_t *chip, const esp_partition_t *part, c
 {
     printf("Erasing chip %p, %d bytes\n", chip, length);
 
-    TEST_ESP_OK(esp_flash_erase_region(chip, part->address, (length + SPI_FLASH_SEC_SIZE) & ~(SPI_FLASH_SEC_SIZE - 1)));
+    TEST_ESP_OK(esp_flash_erase_region(chip, part->address, (length + part->erase_size) & ~(part->erase_size - 1)));
 
     printf("Writing chip %p, %d bytes from source %p\n", chip, length, source);
     // note writing to unaligned address
@@ -195,7 +194,7 @@ void spi_task4(void* arg)
 
     ESP_LOGI(TAG, "Testing chip %p...", chip);
     const esp_partition_t *part = get_test_data_partition();
-    TEST_ASSERT(part->size > test_len + 2 + SPI_FLASH_SEC_SIZE);
+    TEST_ASSERT(part->size > test_len + 2 + part->erase_size);
 
     write_large_buffer(chip, part, source_buf, test_len);
     read_and_check(chip, part, source_buf, test_len);

components/esp_partition/include/esp_partition.h

@@ -1,5 +1,5 @@
 /*
- * SPDX-FileCopyrightText: 2015-2023 Espressif Systems (Shanghai) CO LTD
+ * SPDX-FileCopyrightText: 2015-2024 Espressif Systems (Shanghai) CO LTD
  *
  * SPDX-License-Identifier: Apache-2.0
  */
@@ -464,6 +464,14 @@ esp_err_t esp_partition_deregister_external(const esp_partition_t* partition);
  */
 void esp_partition_unload_all(void);
 
+/**
+ * @brief Get the main flash sector size
+ * @return
+ *      - SPI_FLASH_SEC_SIZE - For esp32xx target
+ *      - ESP_PARTITION_EMULATED_SECTOR_SIZE - For linux target
+ */
+uint32_t esp_partition_get_main_flash_sector_size(void);
+
 #ifdef __cplusplus
 }
 #endif

components/esp_partition/partition_linux.c

@@ -547,6 +547,11 @@ esp_partition_file_mmap_ctrl_t *esp_partition_get_file_mmap_ctrl_act(void)
     return &s_esp_partition_file_mmap_ctrl_act;
 }
 
+uint32_t esp_partition_get_main_flash_sector_size(void)
+{
+    return ESP_PARTITION_EMULATED_SECTOR_SIZE;
+}
+
 #ifdef CONFIG_ESP_PARTITION_ENABLE_STATS
 // timing data for ESP8266, 160MHz CPU frequency, 80MHz flash frequency
 // all values in microseconds

components/esp_partition/partition_target.c

@@ -233,3 +233,8 @@ bool esp_partition_main_flash_region_safe(size_t addr, size_t size)
     }
     return true;
 }
+
+uint32_t esp_partition_get_main_flash_sector_size(void)
+{
+    return SPI_FLASH_SEC_SIZE;
+}

components/nvs_flash/CMakeLists.txt

@@ -25,7 +25,6 @@ idf_component_register(SRCS "${srcs}"
                     REQUIRES "${requires}"
                     PRIV_REQUIRES "${priv_requires}"
                     INCLUDE_DIRS "include"
-                    "../spi_flash/include"
                     PRIV_INCLUDE_DIRS "private_include")
 
 # If we use the linux target, we need to redirect the crc functions to the linux

components/nvs_flash/host_test/nvs_host_test/main/CMakeLists.txt

@@ -9,7 +9,8 @@ idf_component_register(SRCS "test_nvs.cpp"
                             "../../../private_include"
                             "../../../../mbedtls/mbedtls/include"
                        WHOLE_ARCHIVE
-                       REQUIRES nvs_flash)
+                       REQUIRES nvs_flash
+                       PRIV_REQUIRES spi_flash)
 
 if(CMAKE_C_COMPILER_ID MATCHES "Clang")
     target_compile_options(${COMPONENT_LIB} PRIVATE -std=gnu++20)

components/nvs_flash/host_test/nvs_host_test/main/test_fixtures.hpp

@@ -9,6 +9,7 @@
 #include <random>
 #include <fcntl.h>
 #include <unistd.h>
+#include "esp_partition.h"
 
 class PartitionEmulationFixture {
 public:
@@ -22,8 +23,9 @@ class PartitionEmulationFixture {
             FAIL("Failed to initialize esp_partition_file_mmap");
         }
 
-        esp_partition.address = start_sector * SPI_FLASH_SEC_SIZE;
-        esp_partition.size = (start_sector + sector_size) * SPI_FLASH_SEC_SIZE;
+        const uint32_t sec_size = esp_partition_get_main_flash_sector_size();
+        esp_partition.address = start_sector * sec_size;
+        esp_partition.size = (start_sector + sector_size) * sec_size;
         esp_partition.erase_size = ESP_PARTITION_EMULATED_SECTOR_SIZE;
         esp_partition.type = ESP_PARTITION_TYPE_DATA;
         esp_partition.subtype = ESP_PARTITION_SUBTYPE_DATA_NVS;
@@ -42,6 +44,7 @@ class PartitionEmulationFixture {
         off_t size = -1;
         void *p_buff = nullptr;
         char const *fail_msg = nullptr;
+        const uint32_t sec_size = esp_partition_get_main_flash_sector_size();
 
         do {
             // get file size
@@ -57,7 +60,7 @@ class PartitionEmulationFixture {
             }
 
             // check if file fits into the partitiion
-            if (size > sector_size * SPI_FLASH_SEC_SIZE) {
+            if (size > sector_size * sec_size) {
                 fail_msg = "file with partition content doesn't fit into the partition";
                 break;
             }
@@ -82,7 +85,7 @@ class PartitionEmulationFixture {
             }
 
             // erase whole partition
-            if (ESP_OK != esp_partition_erase_range(&esp_partition, 0, sector_size * SPI_FLASH_SEC_SIZE)) {
+            if (ESP_OK != esp_partition_erase_range(&esp_partition, 0, sector_size * sec_size)) {
                 fail_msg = "cannot erase partition prior to write partition binary from file";
                 break;
             }
@@ -123,7 +126,8 @@ class PartitionEmulationFixture {
     // absolute sectorNumber is used here
     bool erase(size_t sectorNumber)
     {
-        size_t offset = sectorNumber * SPI_FLASH_SEC_SIZE;
+        const uint32_t sec_size = esp_partition_get_main_flash_sector_size();
+        size_t offset = sectorNumber * sec_size;
 
         // check the upper bound
         esp_partition_file_mmap_ctrl_t *p_ctrl = esp_partition_get_file_mmap_ctrl_act();
@@ -135,7 +139,7 @@ class PartitionEmulationFixture {
         // esp_partition_erase_range uses offset relative to the beginning of partition
         return (esp_partition_erase_range(&esp_partition,
                                           offset - esp_partition.address,
-                                          SPI_FLASH_SEC_SIZE) == ESP_OK);
+                                          sec_size) == ESP_OK);
     }
 
     ~PartitionEmulationFixture()
@@ -175,9 +179,10 @@ class PartitionEmulationFixture2 : public PartitionEmulationFixture {
                               ) :
         PartitionEmulationFixture(start_sector1, sector_size1, partition_name1), esp_partition2()
     {
+        const uint32_t sec_size = esp_partition_get_main_flash_sector_size();
         // for 2nd partition
-        esp_partition2.address = start_sector2 * SPI_FLASH_SEC_SIZE;
-        esp_partition2.size = (start_sector2 + sector_size2) * SPI_FLASH_SEC_SIZE;
+        esp_partition2.address = start_sector2 * sec_size;
+        esp_partition2.size = (start_sector2 + sector_size2) * sec_size;
         esp_partition2.erase_size = ESP_PARTITION_EMULATED_SECTOR_SIZE;
         esp_partition2.type = ESP_PARTITION_TYPE_DATA;
         esp_partition2.subtype = ESP_PARTITION_SUBTYPE_DATA_NVS;

components/nvs_flash/host_test/nvs_host_test/main/test_nvs.cpp

@@ -18,6 +18,7 @@
 #include <string>
 #include <random>
 #include "test_fixtures.hpp"
+#include "spi_flash_mmap.h"
 
 #define TEST_ESP_ERR(rc, res) CHECK((rc) == (res))
 #define TEST_ESP_OK(rc) CHECK((rc) == ESP_OK)