mirror of
https://github.com/espressif/esp-idf.git
synced 2024-10-05 20:47:46 -04:00
252 lines
6.5 KiB
C
252 lines
6.5 KiB
C
|
|
|||
|
|
#include "tcp_perf.h"
|
|||
|
|
|
|||
|
|
extern int connectedflag;
|
|||
|
|
extern int totle_data;
|
|||
|
|
|
|||
|
|
#if ESP_TCP_PERF_TX && ESP_TCP_DELAY_INFO
|
|||
|
|
|
|||
|
|
extern int totle_pack;
|
|||
|
|
extern int send_success;
|
|||
|
|
extern int send_fail;
|
|||
|
|
extern int delay_classify[5];
|
|||
|
|
|
|||
|
|
#endif
|
|||
|
|
|
|||
|
|
/* FreeRTOS event group to signal when we are connected & ready to make a request */
|
|||
|
|
static EventGroupHandle_t wifi_event_group;
|
|||
|
|
/*socket*/
|
|||
|
|
static int sever_socket;
|
|||
|
|
static struct sockaddr_in sever_addr;
|
|||
|
|
static struct sockaddr_in client_addr;
|
|||
|
|
static unsigned int socklen = sizeof(client_addr);
|
|||
|
|
static int connect_soc;
|
|||
|
|
|
|||
|
|
static esp_err_t event_handler(void *ctx, system_event_t *event)
|
|||
|
|
{
|
|||
|
|
switch(event->event_id) {
|
|||
|
|
case SYSTEM_EVENT_STA_START:
|
|||
|
|
esp_wifi_connect();
|
|||
|
|
break;
|
|||
|
|
case SYSTEM_EVENT_STA_DISCONNECTED:
|
|||
|
|
esp_wifi_connect();
|
|||
|
|
break;
|
|||
|
|
case SYSTEM_EVENT_STA_CONNECTED:
|
|||
|
|
break;
|
|||
|
|
case SYSTEM_EVENT_STA_GOT_IP:
|
|||
|
|
ESP_LOGI(TAG, "event_handler:SYSTEM_EVENT_STA_GOT_IP!\n");
|
|||
|
|
ESP_LOGI(TAG, "ip:%s\n",ip4addr_ntoa(&event->event_info.got_ip.ip_info.ip));
|
|||
|
|
connectedflag=1;
|
|||
|
|
break;
|
|||
|
|
case SYSTEM_EVENT_AP_STACONNECTED:
|
|||
|
|
ESP_LOGI(TAG, "station:"MACSTR" join,AID=%d\n",
|
|||
|
|
MAC2STR(event->event_info.sta_connected.mac),
|
|||
|
|
event->event_info.sta_connected.aid);
|
|||
|
|
connectedflag=1;
|
|||
|
|
break;
|
|||
|
|
case SYSTEM_EVENT_AP_STADISCONNECTED:
|
|||
|
|
ESP_LOGI(TAG, "station:"MACSTR"leave,AID=%d\n",
|
|||
|
|
MAC2STR(event->event_info.sta_disconnected.mac),
|
|||
|
|
event->event_info.sta_disconnected.aid);
|
|||
|
|
break;
|
|||
|
|
default:
|
|||
|
|
break;
|
|||
|
|
}
|
|||
|
|
return ESP_OK;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
//send data
|
|||
|
|
void send_data(void *pvParameters)
|
|||
|
|
{
|
|||
|
|
int len=0;
|
|||
|
|
char databuff[DEFAULT_PKTSIZE];
|
|||
|
|
memset(databuff,97,DEFAULT_PKTSIZE);
|
|||
|
|
vTaskDelay(100/portTICK_RATE_MS);
|
|||
|
|
ESP_LOGI(TAG,"start sending...");
|
|||
|
|
while(1)
|
|||
|
|
{
|
|||
|
|
#if ESP_TCP_PERF_TX && ESP_TCP_DELAY_INFO
|
|||
|
|
//delaytime
|
|||
|
|
struct timeval tv_start;
|
|||
|
|
struct timeval tv_finish;
|
|||
|
|
unsigned int send_delay_ms;
|
|||
|
|
|
|||
|
|
totle_pack++;
|
|||
|
|
gettimeofday(&tv_start,NULL);
|
|||
|
|
#endif
|
|||
|
|
|
|||
|
|
len=send(connect_soc, databuff, DEFAULT_PKTSIZE, 0);
|
|||
|
|
|
|||
|
|
#if ESP_TCP_PERF_TX && ESP_TCP_DELAY_INFO
|
|||
|
|
gettimeofday(&tv_finish,NULL);
|
|||
|
|
#endif
|
|||
|
|
if(len>0)
|
|||
|
|
{
|
|||
|
|
totle_data+=len;
|
|||
|
|
#if ESP_TCP_PERF_TX && ESP_TCP_DELAY_INFO
|
|||
|
|
send_success++;
|
|||
|
|
send_delay_ms=(tv_finish.tv_sec-tv_start.tv_sec)*1000
|
|||
|
|
+(tv_finish.tv_usec-tv_start.tv_usec)/1000;
|
|||
|
|
//ESP_LOGI(TAG, "send_delay_ms=%d",send_delay_ms);
|
|||
|
|
if(send_delay_ms<30)
|
|||
|
|
delay_classify[0]++;
|
|||
|
|
else if(send_delay_ms<100)
|
|||
|
|
delay_classify[1]++;
|
|||
|
|
else if(send_delay_ms<300)
|
|||
|
|
delay_classify[2]++;
|
|||
|
|
else if(send_delay_ms<1000)
|
|||
|
|
delay_classify[3]++;
|
|||
|
|
else
|
|||
|
|
delay_classify[4]++;
|
|||
|
|
#endif
|
|||
|
|
}
|
|||
|
|
else
|
|||
|
|
{
|
|||
|
|
#if ESP_TCP_PERF_TX && ESP_TCP_DELAY_INFO
|
|||
|
|
send_fail++;
|
|||
|
|
#endif
|
|||
|
|
/*for faster send&receive,don't show error code.
|
|||
|
|
*if it can't work as expectations,unnote the two lines here
|
|||
|
|
**/
|
|||
|
|
//perror("data_count error");
|
|||
|
|
//vTaskDelay(500/portTICK_RATE_MS);
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
//send data
|
|||
|
|
void recv_data(void *pvParameters)
|
|||
|
|
{
|
|||
|
|
int len=0;
|
|||
|
|
char databuff[DEFAULT_PKTSIZE];
|
|||
|
|
while(1)
|
|||
|
|
{
|
|||
|
|
len=recv(connect_soc, databuff, DEFAULT_PKTSIZE, 0);
|
|||
|
|
if(len>0)
|
|||
|
|
{
|
|||
|
|
totle_data+=len;
|
|||
|
|
}
|
|||
|
|
else
|
|||
|
|
{
|
|||
|
|
perror("data_count error");
|
|||
|
|
vTaskDelay(500/portTICK_RATE_MS);
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
//use this esp32 as a tcp sever. return 0:success -1:error
|
|||
|
|
int creat_tcp_sever()
|
|||
|
|
{
|
|||
|
|
ESP_LOGI(TAG, "sever socket....port=%d\n",DEFAULTPORT);
|
|||
|
|
sever_socket = socket(AF_INET, SOCK_STREAM, 0);
|
|||
|
|
if(sever_socket < 0)
|
|||
|
|
{
|
|||
|
|
perror("socket() error:");
|
|||
|
|
return -1;
|
|||
|
|
}
|
|||
|
|
sever_addr.sin_family = AF_INET;
|
|||
|
|
sever_addr.sin_port = htons(DEFAULTPORT);
|
|||
|
|
sever_addr.sin_addr.s_addr = htonl(INADDR_ANY);
|
|||
|
|
if(bind(sever_socket,(struct sockaddr*)&sever_addr,sizeof(sever_addr))<0)
|
|||
|
|
{
|
|||
|
|
perror("bind() error");
|
|||
|
|
close(sever_socket);
|
|||
|
|
return -1;
|
|||
|
|
}
|
|||
|
|
if(listen(sever_socket,5)<0)
|
|||
|
|
{
|
|||
|
|
perror("listen() error");
|
|||
|
|
close(sever_socket);
|
|||
|
|
return -1;
|
|||
|
|
}
|
|||
|
|
connect_soc = accept(sever_socket,(struct sockaddr*)&client_addr,&socklen);
|
|||
|
|
if(connect_soc<0)
|
|||
|
|
{
|
|||
|
|
perror("accept() error");
|
|||
|
|
close(sever_socket);
|
|||
|
|
return -1;
|
|||
|
|
}
|
|||
|
|
/*connection established，now can send/recv*/
|
|||
|
|
ESP_LOGI(TAG, "tcp connection established!");
|
|||
|
|
return 0;
|
|||
|
|
}
|
|||
|
|
//use this esp32 as a tcp client. return 0:success -1:error
|
|||
|
|
int creat_tcp_client()
|
|||
|
|
{
|
|||
|
|
ESP_LOGI(TAG, "client socket....severip:port=%s:%d\n",DEFAULTSEVERIP,DEFAULTPORT);
|
|||
|
|
connect_soc = socket(AF_INET, SOCK_STREAM, 0);
|
|||
|
|
if(connect_soc < 0)
|
|||
|
|
{
|
|||
|
|
perror("socket failed!");
|
|||
|
|
return -1;
|
|||
|
|
}
|
|||
|
|
sever_addr.sin_family = AF_INET;
|
|||
|
|
sever_addr.sin_port = htons(DEFAULTPORT);
|
|||
|
|
sever_addr.sin_addr.s_addr = inet_addr(DEFAULTSEVERIP);
|
|||
|
|
printf("connecting to sever...");
|
|||
|
|
if(connect(connect_soc, (struct sockaddr *)&sever_addr, sizeof(sever_addr)) < 0)
|
|||
|
|
{
|
|||
|
|
perror("connect to sever error!");
|
|||
|
|
return -1;
|
|||
|
|
}
|
|||
|
|
ESP_LOGI(TAG,"connect to sever success!");
|
|||
|
|
return 0;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
//wifi_init_sta
|
|||
|
|
void wifi_init_sta()
|
|||
|
|
{
|
|||
|
|
tcpip_adapter_init();
|
|||
|
|
wifi_event_group = xEventGroupCreate();
|
|||
|
|
ESP_ERROR_CHECK( esp_event_loop_init(event_handler, NULL) );
|
|||
|
|
|
|||
|
|
wifi_init_config_t cfg = WIFI_INIT_CONFIG_DEFAULT();
|
|||
|
|
ESP_ERROR_CHECK(esp_wifi_init(&cfg));
|
|||
|
|
wifi_config_t wifi_config = {
|
|||
|
|
.sta = {
|
|||
|
|
.ssid = DEFAULTSSID,
|
|||
|
|
.password = DEFAULTPWD
|
|||
|
|
},
|
|||
|
|
};
|
|||
|
|
|
|||
|
|
ESP_ERROR_CHECK( esp_wifi_set_mode(WIFI_MODE_STA) );
|
|||
|
|
ESP_ERROR_CHECK( esp_wifi_set_config(ESP_IF_WIFI_STA, &wifi_config) );
|
|||
|
|
ESP_ERROR_CHECK( esp_wifi_start() );
|
|||
|
|
|
|||
|
|
ESP_LOGI(TAG, "wifi_init_sta finished.");
|
|||
|
|
ESP_LOGI(TAG, "connect to ap SSID:%s password:%s \n",DEFAULTSSID,DEFAULTPWD);
|
|||
|
|
}
|
|||
|
|
//wifi_init_softap
|
|||
|
|
void wifi_init_softap()
|
|||
|
|
{
|
|||
|
|
tcpip_adapter_init();
|
|||
|
|
wifi_event_group = xEventGroupCreate();
|
|||
|
|
ESP_ERROR_CHECK( esp_event_loop_init(event_handler, NULL) );
|
|||
|
|
|
|||
|
|
wifi_init_config_t cfg = WIFI_INIT_CONFIG_DEFAULT();
|
|||
|
|
ESP_ERROR_CHECK(esp_wifi_init(&cfg));
|
|||
|
|
wifi_config_t wifi_config = {
|
|||
|
|
.ap = {
|
|||
|
|
.ssid = DEFAULTSSID,
|
|||
|
|
.ssid_len=0,
|
|||
|
|
.max_connection=MAXSTACONN,
|
|||
|
|
.password = DEFAULTPWD,
|
|||
|
|
.authmode=WIFI_AUTH_WPA_WPA2_PSK
|
|||
|
|
},
|
|||
|
|
};
|
|||
|
|
|
|||
|
|
ESP_ERROR_CHECK( esp_wifi_set_mode(WIFI_MODE_AP) );
|
|||
|
|
ESP_ERROR_CHECK( esp_wifi_set_config(ESP_IF_WIFI_AP, &wifi_config) );
|
|||
|
|
ESP_ERROR_CHECK( esp_wifi_start() );
|
|||
|
|
|
|||
|
|
ESP_LOGI(TAG, "wifi_init_softap finished.SSID:%s password:%s \n",DEFAULTSSID,DEFAULTPWD);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
void close_socket()
|
|||
|
|
{
|
|||
|
|
close(connect_soc);
|
|||
|
|
close(sever_socket);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
|
|||
|
|
|