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