alex/esp-idf
alex/esp-idf
1
0
Fork 0
mirror of https://github.com/espressif/esp-idf.git synced 2024-10-05 20:47:46 -04:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

fix iperf exit error

Browse Source
This commit is contained in:
xiehang 2019-04-16 12:09:33 +08:00 committed by bot
parent 6250918dfa
commit a48e682e38
1 changed files with 9 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
examples/wifi/iperf/components/iperf.c
View File

@ -93,7 +93,7 @@ void iperf_report_task(void* arg)
printf("%4d-%4d sec %.2f Mbits/sec\n", cur, cur+interval, (double)((s_iperf_ctrl.total_len - last_len)*8)/interval/1e6);
cur += interval;
last_len = s_iperf_ctrl.total_len;
if (cur == time) {
if (cur >= time) {
break;
}
}
@ -156,7 +156,6 @@ esp_err_t iperf_run_tcp_server(void)
return ESP_FAIL;
}
iperf_start_report();
buffer = s_iperf_ctrl.buffer;
want_recv = s_iperf_ctrl.buffer_len;
while (!s_iperf_ctrl.finish) {
@ -169,6 +168,7 @@ esp_err_t iperf_run_tcp_server(void)
return ESP_FAIL;
} else {
printf("accept: %s,%d\n", inet_ntoa(remote_addr.sin_addr), htons(remote_addr.sin_port));
iperf_start_report();
t.tv_sec = IPERF_SOCKET_RX_TIMEOUT;
setsockopt(sockfd, SOL_SOCKET, SO_RCVTIMEO, &t, sizeof(t));
@ -178,6 +178,7 @@ esp_err_t iperf_run_tcp_server(void)
actual_recv = recv(sockfd, buffer, want_recv, 0);
if (actual_recv < 0) {
iperf_show_socket_error_reason("tcp server recv", listen_socket);
s_iperf_ctrl.finish = true;
break;
} else {
s_iperf_ctrl.total_len += actual_recv;
@ -202,6 +203,7 @@ esp_err_t IRAM_ATTR iperf_run_udp_server(void)
uint8_t *buffer;
int sockfd;
int opt;
bool udp_recv_start = true ;
sockfd = socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP);
if (sockfd < 0) {
@ -224,7 +226,6 @@ esp_err_t IRAM_ATTR iperf_run_udp_server(void)
addr.sin_port = htons(s_iperf_ctrl.cfg.sport);
addr.sin_addr.s_addr = s_iperf_ctrl.cfg.sip;
iperf_start_report();
buffer = s_iperf_ctrl.buffer;
want_recv = s_iperf_ctrl.buffer_len;
ESP_LOGI(TAG, "want recv=%d", want_recv);
@ -237,6 +238,10 @@ esp_err_t IRAM_ATTR iperf_run_udp_server(void)
if (actual_recv < 0) {
iperf_show_socket_error_reason("udp server recv", sockfd);
} else {
if(udp_recv_start){
iperf_start_report();
udp_recv_start = false;
}
s_iperf_ctrl.total_len += actual_recv;
}
}
@ -370,6 +375,7 @@ esp_err_t iperf_run_tcp_client(void)
}
}
s_iperf_ctrl.finish = true;
close(sockfd);
return ESP_OK;
}