1 #include "test/test_common_spi.h"
2 #include "driver/spi_slave.h"
3 #include "esp_log.h"
4 #include "driver/gpio.h"
5 #include "hal/gpio_hal.h"
6 
7 int test_freq_default[]=TEST_FREQ_DEFAULT();
8 
9 const char MASTER_TAG[] = "test_master";
10 const char SLAVE_TAG[] = "test_slave";
11 
12 DRAM_ATTR uint8_t spitest_master_send[] = {
13     0x93, 0x34, 0x56, 0x78, 0x9a, 0xbc, 0xde, 0xf0, 0xaa, 0xcc, 0xff, 0xee, 0x55, 0x77, 0x88, 0x43,
14     0x74,
15     0x93, 0x34, 0x56, 0x78, 0x9a, 0xbc, 0xde, 0xf0, 0xaa, 0xcc, 0xff, 0xee, 0x55, 0x77, 0x88, 0x43,
16     0x74,
17     0x93, 0x34, 0x56, 0x78, 0x9a, 0xbc, 0xde, 0xf0, 0xaa, 0xcc, 0xff, 0xee, 0x55, 0x77, 0x88, 0x43,
18     0x74,
19     };
20 DRAM_ATTR uint8_t spitest_slave_send[] = {
21     0xaa, 0xdc, 0xba, 0x98, 0x76, 0x54, 0x32, 0x10, 0x13, 0x57, 0x9b, 0xdf, 0x24, 0x68, 0xac, 0xe0,
22     0xda,
23     0xaa, 0xdc, 0xba, 0x98, 0x76, 0x54, 0x32, 0x10, 0x13, 0x57, 0x9b, 0xdf, 0x24, 0x68, 0xac, 0xe0,
24     0xda,
25     0xaa, 0xdc, 0xba, 0x98, 0x76, 0x54, 0x32, 0x10, 0x13, 0x57, 0x9b, 0xdf, 0x24, 0x68, 0xac, 0xe0,
26     0xda,
27     };
28 
spitest_def_param(void * arg)29 void spitest_def_param(void* arg)
30 {
31     spitest_param_set_t *param_set=(spitest_param_set_t*)arg;
32     param_set->test_size = 8;
33     if (param_set->freq_list==NULL) param_set->freq_list = test_freq_default;
34 }
35 
36 /**********************************************************************************
37  * functions for slave task
38  *********************************************************************************/
init_slave_context(spi_slave_task_context_t * context)39 esp_err_t init_slave_context(spi_slave_task_context_t *context)
40 {
41     context->data_to_send = xQueueCreate( 16, sizeof( slave_txdata_t ));
42     if ( context->data_to_send == NULL ) {
43         return ESP_ERR_NO_MEM;
44     }
45     context->data_received = xRingbufferCreate( 1024, RINGBUF_TYPE_NOSPLIT );
46     if ( context->data_received == NULL ) {
47         return ESP_ERR_NO_MEM;
48     }
49     context->spi=TEST_SLAVE_HOST;
50     return ESP_OK;
51 }
52 
deinit_slave_context(spi_slave_task_context_t * context)53 void deinit_slave_context(spi_slave_task_context_t *context)
54 {
55     TEST_ASSERT( context->data_to_send != NULL );
56     vQueueDelete( context->data_to_send );
57     context->data_to_send = NULL;
58     TEST_ASSERT( context->data_received != NULL );
59     vRingbufferDelete( context->data_received );
60     context->data_received = NULL;
61 }
62 
63 /* The task requires a queue and a ringbuf, which should be initialized before task starts.
64    Send ``slave_txdata_t`` to the queue to make the task send data;
65    the task returns data got to the ringbuf, which should have sufficient size.
66 */
spitest_slave_task(void * arg)67 void spitest_slave_task(void* arg)
68 {
69     spi_slave_task_context_t* context = (spi_slave_task_context_t*) arg;
70     QueueHandle_t queue = context->data_to_send;
71     RingbufHandle_t ringbuf = context->data_received;
72     uint8_t recvbuf[320+8];
73     slave_txdata_t txdata;
74 
75     ESP_LOGI( SLAVE_TAG, "slave up" );
76     //never quit, but blocked by the queue, waiting to be killed, when no more send from main task.
77     while( 1 ) {
78         BaseType_t ret = xQueueReceive( queue, &txdata, portMAX_DELAY );
79         assert(ret);
80 
81         spi_slave_transaction_t t = {};
82         t.length = txdata.len;
83         t.tx_buffer = txdata.start;
84         t.rx_buffer = recvbuf+8;
85         //loop until trans_len != 0 to skip glitches
86         do {
87             TEST_ESP_OK( spi_slave_transmit( context->spi, &t, portMAX_DELAY ) );
88         } while ( t.trans_len <= 2 );
89         memcpy(recvbuf, &t.trans_len, sizeof(uint32_t));
90         *(uint8_t**)(recvbuf+4) = (uint8_t*)txdata.start;
91         ESP_LOGD( SLAVE_TAG, "received: %d", t.trans_len );
92         xRingbufferSend( ringbuf, recvbuf, 8+(t.trans_len+7)/8, portMAX_DELAY );
93     }
94 }
95 
slave_pull_up(const spi_bus_config_t * cfg,int spics_io_num)96 void slave_pull_up(const spi_bus_config_t* cfg, int spics_io_num)
97 {
98     gpio_set_pull_mode(cfg->mosi_io_num, GPIO_PULLUP_ONLY);
99     gpio_set_pull_mode(cfg->sclk_io_num, GPIO_PULLUP_ONLY);
100     gpio_set_pull_mode(spics_io_num, GPIO_PULLUP_ONLY);
101 }
102 
103 /**********************************************************************************
104  * functions for slave task
105  *********************************************************************************/
106 
107 static int test_len[] = {1, 3, 5, 7, 9, 11, 33, 64};
108 
spitest_init_transactions(const spitest_param_set_t * cfg,spitest_context_t * context)109 void spitest_init_transactions(const spitest_param_set_t *cfg, spitest_context_t* context)
110 {
111     spi_transaction_t* trans = context->master_trans;
112     uint8_t *rx_buf_ptr = context->master_rxbuf;
113     const spi_dup_t dup = cfg->dup;
114 
115     for (int i = 0; i < cfg->test_size; i++) {
116         const void* tx_buffer = spitest_master_send + i%8;
117         int length = 8*test_len[i];
118         if (cfg->length_aligned) length = (length+31)&(~31);
119 
120         if (dup == HALF_DUPLEX_MISO) {
121             trans[i] = (spi_transaction_t) {
122                 .rx_buffer = rx_buf_ptr,
123                 .rxlength = length,
124             };
125         } else if (dup == HALF_DUPLEX_MOSI) {
126             trans[i] = (spi_transaction_t) {
127                 .tx_buffer = tx_buffer,
128                 .length = length,
129             };
130         } else {
131             trans[i] = (spi_transaction_t) {
132                 .tx_buffer = tx_buffer,
133                 .length = length,
134                 .rx_buffer = rx_buf_ptr,
135             };
136         }
137         rx_buf_ptr = (uint8_t*)( (uint32_t)(rx_buf_ptr + (length+7)/8 + 3) & (~3));
138 
139         const void* slave_tx = spitest_slave_send + (cfg->slave_unaligned_addr? i%3: (i%3)*4);
140         //prepare slave tx data
141         context->slave_trans[i] = (slave_txdata_t) {
142             .start = slave_tx,
143             .len = 512,
144         };
145         if (cfg->slave_dma_chan != 0) context->slave_trans[i].len = 1024;
146     }
147 }
148 
spitest_master_print_data(spi_transaction_t * t,int rxlength)149 void spitest_master_print_data(spi_transaction_t *t, int rxlength)
150 {
151     if (t->tx_buffer) ESP_LOG_BUFFER_HEX( "master tx", t->tx_buffer, t->length/8 );
152     if (t->rx_buffer) ESP_LOG_BUFFER_HEX( "master rx", t->rx_buffer, rxlength/8 );
153 }
154 
spitest_slave_print_data(slave_rxdata_t * t,bool print_rxdata)155 void spitest_slave_print_data(slave_rxdata_t *t, bool print_rxdata)
156 {
157     int rcv_len = (t->len+7)/8;
158     ESP_LOGI(SLAVE_TAG, "trans_len: %d", t->len);
159     ESP_LOG_BUFFER_HEX("slave tx", t->tx_start, rcv_len);
160     if (print_rxdata) ESP_LOG_BUFFER_HEX("slave rx", t->data, rcv_len);
161 }
162 
spitest_check_data(int len,spi_transaction_t * master_t,slave_rxdata_t * slave_t,bool check_master_data,bool check_slave_len,bool check_slave_data)163 esp_err_t spitest_check_data(int len, spi_transaction_t *master_t, slave_rxdata_t *slave_t, bool check_master_data, bool check_slave_len, bool check_slave_data)
164 {
165     esp_err_t ret = ESP_OK;
166     uint32_t rcv_len = slave_t->len;
167     //currently the rcv_len can be in range of [t->length-1, t->length+3]
168     if (check_slave_len &&
169         (rcv_len < len - 1 || rcv_len > len + 4)) {
170             ret = ESP_FAIL;
171     }
172 
173     if (check_master_data &&
174         memcmp(slave_t->tx_start, master_t->rx_buffer, (len + 7) / 8) != 0 ) {
175             ret = ESP_FAIL;
176     }
177 
178     if (check_slave_data &&
179         memcmp(master_t->tx_buffer, slave_t->data, (len + 7) / 8) != 0 ) {
180             ret = ESP_FAIL;
181     }
182     if (ret != ESP_OK) {
183         ESP_LOGI(SLAVE_TAG, "slave_recv_len: %d", rcv_len);
184         spitest_master_print_data(master_t, len);
185         spitest_slave_print_data(slave_t, true);
186         //already failed, try to use the TEST_ASSERT to output the reason...
187         if (check_slave_len) {
188             TEST_ASSERT(rcv_len >= len - 1 && rcv_len <= len + 4);
189         }
190         TEST_ASSERT_EQUAL_HEX8_ARRAY(slave_t->tx_start, master_t->rx_buffer, (len + 7) / 8);
191         TEST_ASSERT_EQUAL_HEX8_ARRAY(master_t->tx_buffer, slave_t->data, (len + 7) / 8);
192     }
193 
194     return ESP_OK;
195 }
196 
197 
master_free_device_bus(spi_device_handle_t spi)198 void master_free_device_bus(spi_device_handle_t spi)
199 {
200     TEST_ESP_OK( spi_bus_remove_device(spi) );
201     TEST_ESP_OK( spi_bus_free(TEST_SPI_HOST) );
202 }
203 
spitest_gpio_output_sel(uint32_t gpio_num,int func,uint32_t signal_idx)204 void spitest_gpio_output_sel(uint32_t gpio_num, int func, uint32_t signal_idx)
205 {
206     gpio_hal_iomux_func_sel(GPIO_PIN_MUX_REG[gpio_num], func);
207     GPIO.func_out_sel_cfg[gpio_num].func_sel = signal_idx;
208 }
209 
spitest_gpio_input_sel(uint32_t gpio_num,int func,uint32_t signal_idx)210 void spitest_gpio_input_sel(uint32_t gpio_num, int func, uint32_t signal_idx)
211 {
212     gpio_hal_iomux_func_sel(GPIO_PIN_MUX_REG[gpio_num], func);
213     GPIO.func_in_sel_cfg[signal_idx].func_sel = gpio_num;
214 }
215 
216 //Note this cs_num is the ID of the connected devices' ID, e.g. if 2 devices are connected to the bus,
217 //then the cs_num of the 1st and 2nd devices are 0 and 1 respectively.
same_pin_func_sel(spi_bus_config_t bus,spi_device_interface_config_t dev,uint8_t cs_num)218 void same_pin_func_sel(spi_bus_config_t bus, spi_device_interface_config_t dev, uint8_t cs_num)
219 {
220     spitest_gpio_output_sel(bus.mosi_io_num, FUNC_GPIO, spi_periph_signal[TEST_SPI_HOST].spid_out);
221     spitest_gpio_input_sel(bus.mosi_io_num, FUNC_GPIO, spi_periph_signal[TEST_SLAVE_HOST].spid_in);
222 
223     spitest_gpio_output_sel(bus.miso_io_num, FUNC_GPIO, spi_periph_signal[TEST_SLAVE_HOST].spiq_out);
224     spitest_gpio_input_sel(bus.miso_io_num, FUNC_GPIO, spi_periph_signal[TEST_SPI_HOST].spiq_in);
225 
226     spitest_gpio_output_sel(dev.spics_io_num, FUNC_GPIO, spi_periph_signal[TEST_SPI_HOST].spics_out[cs_num]);
227     spitest_gpio_input_sel(dev.spics_io_num, FUNC_GPIO, spi_periph_signal[TEST_SLAVE_HOST].spics_in);
228 
229     spitest_gpio_output_sel(bus.sclk_io_num, FUNC_GPIO, spi_periph_signal[TEST_SPI_HOST].spiclk_out);
230     spitest_gpio_input_sel(bus.sclk_io_num, FUNC_GPIO, spi_periph_signal[TEST_SLAVE_HOST].spiclk_in);
231 
232 #if CONFIG_IDF_TARGET_ESP32S2 || CONFIG_IDF_TARGET_ESP32S3
233     GPIO.func_in_sel_cfg[FSPIQ_IN_IDX].sig_in_sel = 1;
234 #endif
235 }
236