esp-idf/examples/performance/tcp_perf/main/main.c

/*
tcp_perf example

Using this example to test tcp throughput performance.
esp<->esp or esp<->ap

step1:
	init wifi as AP/STA using config SSID/PASSWORD.

step2:
	creat a tcp sever/client socket using config PORT/(IP).
	if sever: wating for connect.
	if client connect to sever.

step3:
	send/receive data to/from each other.
	if the tcp connect established. esp will send or receive data.
	you can see the info in com port output.

*/

#include "tcp_perf.h"

int connectedflag = 0;
int totle_data=0;

#if ESP_TCP_PERF_TX && ESP_TCP_DELAY_INFO

int totle_pack=0;
int send_success=0;
int send_fail=0;
int delay_classify[5]={0};

#endif

//this task establish a TCP connection and receive data from TCP
static void tcp_conn(void *pvParameters)
{
	ESP_LOGI(TAG, "task tcp_conn start.");
	//wating for connecting to AP
  	do
  	{
		vTaskDelay(100);
  	}
  	while(!connectedflag);
  	ESP_LOGI(TAG, "sta has connected to ap.");

  	//create tcp socket
  	int socret;
#if ESP_TCP_MODE_SEVER
  	ESP_LOGI(TAG, "creat_tcp_sever.");
  	socret=creat_tcp_sever();
#else
  	ESP_LOGI(TAG, "creat_tcp_client.");
  	socret=creat_tcp_client();
#endif
  	if(-1==socret)
  	{
  		ESP_LOGI(TAG, "creat tcp socket error,stop.");
  		vTaskDelete(NULL);
  	}


  	//create a task to tx/rx data
  	TaskHandle_t tx_rx_task;
#if ESP_TCP_PERF_TX
  	xTaskCreate(&send_data,"send_data",4096,NULL,4,&tx_rx_task);
#else
  	xTaskCreate(&recv_data,"recv_data",4096,NULL,4,&tx_rx_task);
#endif
  	int pps;
  	while(1)
  	{
		totle_data=0;
		//calc every 3s
		vTaskDelay(3000/ portTICK_RATE_MS);
		pps=totle_data/3;
#if ESP_TCP_PERF_TX
		ESP_LOGI(TAG, "tcp send %d byte per sec!",pps);
#if ESP_TCP_DELAY_INFO
		ESP_LOGI(TAG, "tcp send packet totle:%d  succeed:%d  failed:%d\n0-30:%d 30-100:%d 100-300:%d 300-1000:%d 1000+:%d\n",
				totle_pack,send_success,send_fail,delay_classify[0],delay_classify[1],delay_classify[2],delay_classify[3],delay_classify[
4]);
#endif
#else
		ESP_LOGI(TAG, "tcp recv %d byte per sec!\n",pps);
#endif
  	}
  	close_socket();
	vTaskDelete(tx_rx_task);
	vTaskDelete(NULL);
}



void app_main(void)
{
    nvs_flash_init();
#if ESP_WIFI_MODE_AP
    ESP_LOGI(TAG, "ESP_WIFI_MODE_AP\n");
    wifi_init_softap();
#else
    ESP_LOGI(TAG, "ESP_WIFI_MODE_STA\n");
    wifi_init_sta();
#endif
    xTaskCreate(&tcp_conn,"tcp_conn",4096,NULL,5,NULL);
}