mirror of
				https://github.com/espressif/esp-idf.git
				synced 2025-11-03 08:31:44 +01:00 
			
		
		
		
	Partially reverted 08f1bbe0c7.
The host should have this flexibility, which is consistent to the cs_hold argument.
However, the user should know as less as possible about the host.
So the wrapper layer (esp_flash_spi_init.c) should cover this, helping to set cs_setup to 1, to meet the common requirements.
		
	
		
			
				
	
	
		
			356 lines
		
	
	
		
			11 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
			
		
		
	
	
			356 lines
		
	
	
		
			11 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
/*
 | 
						|
 * SPDX-FileCopyrightText: 2021 Espressif Systems (Shanghai) CO LTD
 | 
						|
 *
 | 
						|
 * SPDX-License-Identifier: Apache-2.0
 | 
						|
 */
 | 
						|
#include "sdkconfig.h"
 | 
						|
#include "esp_log.h"
 | 
						|
#include "driver/spi_master.h"
 | 
						|
#include "driver/gpio.h"
 | 
						|
#include "esp_flash_spi_init.h"
 | 
						|
 | 
						|
#include "test/test_common_spi.h"
 | 
						|
#include "unity.h"
 | 
						|
 | 
						|
#if !TEMPORARY_DISABLED_FOR_TARGETS(ESP32S3, ESP32C3)
 | 
						|
 | 
						|
#if CONFIG_IDF_TARGET_ESP32
 | 
						|
// The VSPI pins on UT_T1_ESP_FLASH are connected to a external flash
 | 
						|
#define TEST_BUS_PIN_NUM_MISO   VSPI_IOMUX_PIN_NUM_MISO
 | 
						|
#define TEST_BUS_PIN_NUM_MOSI   VSPI_IOMUX_PIN_NUM_MOSI
 | 
						|
#define TEST_BUS_PIN_NUM_CLK    VSPI_IOMUX_PIN_NUM_CLK
 | 
						|
#define TEST_BUS_PIN_NUM_CS     VSPI_IOMUX_PIN_NUM_CS
 | 
						|
#define TEST_BUS_PIN_NUM_WP     VSPI_IOMUX_PIN_NUM_WP
 | 
						|
#define TEST_BUS_PIN_NUM_HD     VSPI_IOMUX_PIN_NUM_HD
 | 
						|
 | 
						|
#elif CONFIG_IDF_TARGET_ESP32S2 || CONFIG_IDF_TARGET_ESP32S3
 | 
						|
#define TEST_BUS_PIN_NUM_MISO   FSPI_IOMUX_PIN_NUM_MISO
 | 
						|
#define TEST_BUS_PIN_NUM_MOSI   FSPI_IOMUX_PIN_NUM_MOSI
 | 
						|
#define TEST_BUS_PIN_NUM_CLK    FSPI_IOMUX_PIN_NUM_CLK
 | 
						|
#define TEST_BUS_PIN_NUM_CS     FSPI_IOMUX_PIN_NUM_CS
 | 
						|
#define TEST_BUS_PIN_NUM_WP     FSPI_IOMUX_PIN_NUM_WP
 | 
						|
#define TEST_BUS_PIN_NUM_HD     FSPI_IOMUX_PIN_NUM_HD
 | 
						|
 | 
						|
#endif
 | 
						|
 | 
						|
 | 
						|
typedef struct {
 | 
						|
    union {
 | 
						|
        spi_device_handle_t handle;
 | 
						|
        esp_flash_t* chip;
 | 
						|
    };
 | 
						|
    bool finished;
 | 
						|
} task_context_t;
 | 
						|
 | 
						|
#ifndef CONFIG_ESP32_SPIRAM_SUPPORT
 | 
						|
 | 
						|
const static char TAG[] = "test_spi";
 | 
						|
 | 
						|
void spi_task1(void* arg)
 | 
						|
{
 | 
						|
    //task1 send 50 polling transactions, acquire the bus and send another 50
 | 
						|
    int count=0;
 | 
						|
    spi_transaction_t t = {
 | 
						|
        .flags = SPI_TRANS_USE_TXDATA,
 | 
						|
        .tx_data = { 0x80, 0x12, 0x34, 0x56 },
 | 
						|
        .length = 4*8,
 | 
						|
    };
 | 
						|
    spi_device_handle_t handle = ((task_context_t*)arg)->handle;
 | 
						|
    for( int j = 0; j < 50; j ++ ) {
 | 
						|
        TEST_ESP_OK(spi_device_polling_transmit( handle, &t ));
 | 
						|
        ESP_LOGI(TAG, "task1:%d", count++ );
 | 
						|
    }
 | 
						|
    TEST_ESP_OK(spi_device_acquire_bus( handle, portMAX_DELAY ));
 | 
						|
    for( int j = 0; j < 50; j ++ ) {
 | 
						|
        TEST_ESP_OK(spi_device_polling_transmit( handle, &t ));
 | 
						|
        ESP_LOGI(TAG, "task1:%d", count++ );
 | 
						|
    }
 | 
						|
    spi_device_release_bus(handle);
 | 
						|
    ESP_LOGI(TAG, "task1 terminates");
 | 
						|
    ((task_context_t*)arg)->finished = true;
 | 
						|
    vTaskDelete(NULL);
 | 
						|
}
 | 
						|
 | 
						|
void spi_task2(void* arg)
 | 
						|
{
 | 
						|
    int count=0;
 | 
						|
    //task2 acquire the bus, send 50 polling transactions and then 50 non-polling
 | 
						|
    spi_transaction_t t = {
 | 
						|
        .flags = SPI_TRANS_USE_TXDATA,
 | 
						|
        .tx_data = { 0x80, 0x12, 0x34, 0x56 },
 | 
						|
        .length = 4*8,
 | 
						|
    };
 | 
						|
    spi_transaction_t *ret_t;
 | 
						|
    spi_device_handle_t handle = ((task_context_t*)arg)->handle;
 | 
						|
    TEST_ESP_OK(spi_device_acquire_bus( handle, portMAX_DELAY ));
 | 
						|
 | 
						|
    for (int i = 0; i < 50; i ++) {
 | 
						|
        TEST_ESP_OK(spi_device_polling_transmit(handle, &t));
 | 
						|
        ESP_LOGI( TAG, "task2: %d", count++ );
 | 
						|
    }
 | 
						|
 | 
						|
    for( int j = 0; j < 50; j ++ ) {
 | 
						|
        TEST_ESP_OK(spi_device_queue_trans(handle, &t, portMAX_DELAY));
 | 
						|
    }
 | 
						|
    for( int j = 0; j < 50; j ++ ) {
 | 
						|
        TEST_ESP_OK(spi_device_get_trans_result(handle, &ret_t, portMAX_DELAY));
 | 
						|
        assert(ret_t == &t);
 | 
						|
        ESP_LOGI( TAG, "task2: %d", count++ );
 | 
						|
    }
 | 
						|
    spi_device_release_bus(handle);
 | 
						|
    vTaskDelay(1);
 | 
						|
    ESP_LOGI(TAG, "task2 terminates");
 | 
						|
    ((task_context_t*)arg)->finished = true;
 | 
						|
    vTaskDelete(NULL);
 | 
						|
}
 | 
						|
 | 
						|
void spi_task3(void* arg)
 | 
						|
{
 | 
						|
    //task3 send 30 polling transactions, acquire the bus, send 20 polling transactions and then 50 non-polling
 | 
						|
    int count=0;
 | 
						|
    spi_transaction_t t = {
 | 
						|
        .flags = SPI_TRANS_USE_TXDATA,
 | 
						|
        .tx_data = { 0x80, 0x12, 0x34, 0x56 },
 | 
						|
        .length = 4*8,
 | 
						|
    };
 | 
						|
    spi_transaction_t *ret_t;
 | 
						|
    spi_device_handle_t handle = ((task_context_t*)arg)->handle;
 | 
						|
 | 
						|
    for (int i = 0; i < 30; i ++) {
 | 
						|
        TEST_ESP_OK(spi_device_polling_transmit(handle, &t));
 | 
						|
        ESP_LOGI( TAG, "task3: %d", count++ );
 | 
						|
    }
 | 
						|
 | 
						|
    TEST_ESP_OK(spi_device_acquire_bus( handle, portMAX_DELAY ));
 | 
						|
    for (int i = 0; i < 20; i ++) {
 | 
						|
        TEST_ESP_OK(spi_device_polling_transmit(handle, &t));
 | 
						|
        ESP_LOGI( TAG, "task3: %d", count++ );
 | 
						|
    }
 | 
						|
 | 
						|
    for (int j = 0; j < 50; j++) {
 | 
						|
        TEST_ESP_OK(spi_device_queue_trans(handle, &t, portMAX_DELAY));
 | 
						|
    }
 | 
						|
    for (int j = 0; j < 50; j++) {
 | 
						|
        TEST_ESP_OK(spi_device_get_trans_result(handle, &ret_t, portMAX_DELAY));
 | 
						|
        assert(ret_t == &t);
 | 
						|
        ESP_LOGI(TAG, "task3: %d", count++);
 | 
						|
    }
 | 
						|
    spi_device_release_bus(handle);
 | 
						|
 | 
						|
    ESP_LOGI(TAG, "task3 terminates");
 | 
						|
    ((task_context_t*)arg)->finished = true;
 | 
						|
    vTaskDelete(NULL);
 | 
						|
}
 | 
						|
 | 
						|
static void write_large_buffer(esp_flash_t *chip, const esp_partition_t *part, const uint8_t *source, size_t length)
 | 
						|
{
 | 
						|
    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)) );
 | 
						|
 | 
						|
    printf("Writing chip %p, %d bytes from source %p\n", chip, length, source);
 | 
						|
    // note writing to unaligned address
 | 
						|
    TEST_ESP_OK(esp_flash_write(chip, source, part->address + 1, length) );
 | 
						|
 | 
						|
    printf("Write done.\n");
 | 
						|
}
 | 
						|
 | 
						|
static void read_and_check(esp_flash_t *chip, const esp_partition_t *part, const uint8_t *source, size_t length)
 | 
						|
{
 | 
						|
    printf("Checking chip %p, %d bytes\n", chip, length);
 | 
						|
    uint8_t *buf = malloc(length);
 | 
						|
    TEST_ASSERT_NOT_NULL(buf);
 | 
						|
    TEST_ESP_OK(esp_flash_read(chip, buf, part->address + 1, length) );
 | 
						|
    TEST_ASSERT_EQUAL_HEX8_ARRAY(source, buf, length);
 | 
						|
    free(buf);
 | 
						|
 | 
						|
    // check nothing was written at beginning or end
 | 
						|
    uint8_t ends[8];
 | 
						|
 | 
						|
    TEST_ESP_OK(esp_flash_read(chip, ends, part->address, sizeof(ends)) );
 | 
						|
    TEST_ASSERT_EQUAL_HEX8(0xFF, ends[0]);
 | 
						|
    TEST_ASSERT_EQUAL_HEX8(source[0], ends[1]);
 | 
						|
 | 
						|
    TEST_ESP_OK(esp_flash_read(chip, ends, part->address + length, sizeof(ends)) );
 | 
						|
 | 
						|
    TEST_ASSERT_EQUAL_HEX8(source[length - 1], ends[0]);
 | 
						|
    TEST_ASSERT_EQUAL_HEX8(0xFF, ends[1]);
 | 
						|
    TEST_ASSERT_EQUAL_HEX8(0xFF, ends[2]);
 | 
						|
    TEST_ASSERT_EQUAL_HEX8(0xFF, ends[3]);
 | 
						|
}
 | 
						|
 | 
						|
void spi_task4(void* arg)
 | 
						|
{
 | 
						|
    esp_flash_t *chip = ((task_context_t*)arg)->chip;
 | 
						|
 | 
						|
    // buffer in RAM
 | 
						|
    const int test_len = 16400;
 | 
						|
    uint8_t *source_buf = heap_caps_malloc(test_len, MALLOC_CAP_INTERNAL | MALLOC_CAP_8BIT);
 | 
						|
    TEST_ASSERT_NOT_NULL(source_buf);
 | 
						|
 | 
						|
    srand(676);
 | 
						|
    for (int i = 0; i < test_len; i++) {
 | 
						|
        source_buf[i] = rand();
 | 
						|
    }
 | 
						|
 | 
						|
    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);
 | 
						|
 | 
						|
    write_large_buffer(chip, part, source_buf, test_len);
 | 
						|
    read_and_check(chip, part, source_buf, test_len);
 | 
						|
 | 
						|
    free(source_buf);
 | 
						|
 | 
						|
    ESP_LOGI(TAG, "task4 terminates");
 | 
						|
    ((task_context_t*)arg)->finished = true;
 | 
						|
    vTaskDelete(NULL);
 | 
						|
}
 | 
						|
 | 
						|
static void test_bus_lock(bool test_flash)
 | 
						|
{
 | 
						|
    task_context_t context1={};
 | 
						|
    task_context_t context2={};
 | 
						|
    task_context_t context3={};
 | 
						|
    task_context_t context4={};
 | 
						|
    TaskHandle_t task1, task2, task3, task4;
 | 
						|
    esp_err_t ret;
 | 
						|
    spi_bus_config_t buscfg=SPI_BUS_TEST_DEFAULT_CONFIG();
 | 
						|
    buscfg.miso_io_num = TEST_BUS_PIN_NUM_MISO;
 | 
						|
    buscfg.mosi_io_num = TEST_BUS_PIN_NUM_MOSI;
 | 
						|
    buscfg.sclk_io_num = TEST_BUS_PIN_NUM_CLK;
 | 
						|
 | 
						|
    spi_device_interface_config_t devcfg=SPI_DEVICE_TEST_DEFAULT_CONFIG();
 | 
						|
    devcfg.queue_size = 100;
 | 
						|
 | 
						|
    //Initialize the SPI bus and 3 devices
 | 
						|
    ret=spi_bus_initialize(TEST_SPI_HOST, &buscfg, 1);
 | 
						|
    TEST_ESP_OK(ret);
 | 
						|
    ret=spi_bus_add_device(TEST_SPI_HOST, &devcfg, &context1.handle);
 | 
						|
    TEST_ESP_OK(ret);
 | 
						|
    ret=spi_bus_add_device(TEST_SPI_HOST, &devcfg, &context2.handle);
 | 
						|
    TEST_ESP_OK(ret);
 | 
						|
 | 
						|
    //only have 3 cs pins, leave one for the flash
 | 
						|
    devcfg.spics_io_num = -1;
 | 
						|
    ret=spi_bus_add_device(TEST_SPI_HOST, &devcfg, &context3.handle);
 | 
						|
    TEST_ESP_OK(ret);
 | 
						|
    esp_flash_spi_device_config_t flash_cfg = {
 | 
						|
        .host_id = TEST_SPI_HOST,
 | 
						|
        .cs_id = 2,
 | 
						|
        .cs_io_num = TEST_BUS_PIN_NUM_CS,
 | 
						|
        .io_mode = SPI_FLASH_DIO,
 | 
						|
        .speed = ESP_FLASH_5MHZ,
 | 
						|
        .input_delay_ns = 0,
 | 
						|
    };
 | 
						|
 | 
						|
    //Clamp the WP and HD pins to VDD to make it work in DIO mode
 | 
						|
    gpio_set_direction(TEST_BUS_PIN_NUM_HD, GPIO_MODE_OUTPUT);
 | 
						|
    gpio_set_direction(TEST_BUS_PIN_NUM_WP, GPIO_MODE_OUTPUT);
 | 
						|
    gpio_set_level(TEST_BUS_PIN_NUM_HD, 1);
 | 
						|
    gpio_set_level(TEST_BUS_PIN_NUM_WP, 1);
 | 
						|
 | 
						|
    esp_flash_t *chip;
 | 
						|
    (void) chip;
 | 
						|
    if (test_flash) {
 | 
						|
        ret = spi_bus_add_flash_device(&chip, &flash_cfg);
 | 
						|
        TEST_ESP_OK(ret);
 | 
						|
        ret = esp_flash_init(chip);
 | 
						|
        TEST_ESP_OK(ret);
 | 
						|
        context4.chip = chip;
 | 
						|
    }
 | 
						|
    ESP_LOGI(TAG, "Start testing...");
 | 
						|
 | 
						|
    xTaskCreate( spi_task1, "task1", 4096, &context1, 0, &task1 );
 | 
						|
    xTaskCreate( spi_task2, "task2", 4096, &context2, 0, &task2 );
 | 
						|
    xTaskCreate( spi_task3, "task3", 4096, &context3, 0, &task3 );
 | 
						|
    if (test_flash) {
 | 
						|
        xTaskCreate( spi_task4, "task4", 2048, &context4, 0, &task4 );
 | 
						|
    } else {
 | 
						|
        context4.finished = true;
 | 
						|
    }
 | 
						|
 | 
						|
    for(;;){
 | 
						|
        vTaskDelay(10);
 | 
						|
        if (context1.finished && context2.finished && context3.finished && context4.finished) break;
 | 
						|
    }
 | 
						|
 | 
						|
    TEST_ESP_OK(spi_bus_remove_device(context1.handle));
 | 
						|
    TEST_ESP_OK(spi_bus_remove_device(context2.handle));
 | 
						|
    TEST_ESP_OK(spi_bus_remove_device(context3.handle));
 | 
						|
    if (test_flash) {
 | 
						|
        TEST_ESP_OK(spi_bus_remove_flash_device(chip));
 | 
						|
    }
 | 
						|
    TEST_ESP_OK(spi_bus_free(TEST_SPI_HOST) );
 | 
						|
}
 | 
						|
 | 
						|
#if !TEMPORARY_DISABLED_FOR_TARGETS(ESP32S2)
 | 
						|
//no runners
 | 
						|
TEST_CASE("spi bus lock, with flash","[spi][test_env=UT_T1_ESP_FLASH]")
 | 
						|
{
 | 
						|
    test_bus_lock(true);
 | 
						|
}
 | 
						|
#endif //!TEMPORARY_DISABLED_FOR_TARGETS(ESP32S2)
 | 
						|
 | 
						|
 | 
						|
TEST_CASE("spi bus lock","[spi]")
 | 
						|
{
 | 
						|
    test_bus_lock(false);
 | 
						|
}
 | 
						|
 | 
						|
#if !TEMPORARY_DISABLED_FOR_TARGETS(ESP32S2)
 | 
						|
//SPI1 not supported by driver
 | 
						|
static IRAM_ATTR esp_err_t test_polling_send(spi_device_handle_t handle)
 | 
						|
{
 | 
						|
    for (int i = 0; i < 10; i++) {
 | 
						|
        spi_transaction_t trans = {
 | 
						|
            .length = 16,
 | 
						|
            .flags = SPI_TRANS_USE_TXDATA | SPI_TRANS_USE_RXDATA,
 | 
						|
        };
 | 
						|
        esp_err_t err = spi_device_polling_transmit(handle, &trans);
 | 
						|
        if (err != ESP_OK) {
 | 
						|
            return err;
 | 
						|
        }
 | 
						|
    }
 | 
						|
    return ESP_OK;
 | 
						|
}
 | 
						|
 | 
						|
static IRAM_ATTR NOINLINE_ATTR void test_acquire(spi_device_handle_t handle)
 | 
						|
{
 | 
						|
    esp_err_t err = spi_device_acquire_bus(handle, portMAX_DELAY);
 | 
						|
    if (err == ESP_OK) {
 | 
						|
        err = test_polling_send(handle);
 | 
						|
        spi_device_release_bus(handle);
 | 
						|
    }
 | 
						|
 | 
						|
    TEST_ESP_OK(err);
 | 
						|
}
 | 
						|
 | 
						|
TEST_CASE("spi master can be used on SPI1", "[spi]")
 | 
						|
{
 | 
						|
    spi_device_interface_config_t dev_cfg = {
 | 
						|
        .mode = 1,
 | 
						|
        .clock_speed_hz = 1*1000*1000,
 | 
						|
        .spics_io_num = -1,
 | 
						|
        .queue_size = 1,
 | 
						|
    };
 | 
						|
    spi_device_handle_t handle;
 | 
						|
    esp_err_t err;
 | 
						|
    err = spi_bus_add_device(SPI1_HOST, &dev_cfg, &handle);
 | 
						|
    TEST_ESP_OK(err);
 | 
						|
 | 
						|
    err = test_polling_send(handle);
 | 
						|
    TEST_ESP_OK(err);
 | 
						|
    test_acquire(handle);
 | 
						|
 | 
						|
    err = spi_bus_remove_device(handle);
 | 
						|
    TEST_ESP_OK(err);
 | 
						|
}
 | 
						|
#endif //!TEMPORARY_DISABLED_FOR_TARGETS(ESP32S2)
 | 
						|
 | 
						|
//TODO: add a case when a non-polling transaction happened in the bus-acquiring time and then release the bus then queue a new trans
 | 
						|
 | 
						|
#endif //!CONFIG_ESP32_SPIRAM_SUPPORT
 | 
						|
 | 
						|
#endif //!TEMPORARY_DISABLED_FOR_TARGETS(ESP32S3, ESP32C3)
 |