From a48e682e38d8ceb06319a428acb2b80024548228 Mon Sep 17 00:00:00 2001
From: xiehang <xiehang@espressif.com>
Date: Tue, 16 Apr 2019 12:09:33 +0800
Subject: [PATCH] fix iperf exit error

---
 examples/wifi/iperf/components/iperf.c | 12 +++++++++---
 1 file changed, 9 insertions(+), 3 deletions(-)

diff --git a/examples/wifi/iperf/components/iperf.c b/examples/wifi/iperf/components/iperf.c
index 80cadb400..0dfa044d6 100644
--- a/examples/wifi/iperf/components/iperf.c
+++ b/examples/wifi/iperf/components/iperf.c
@@ -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;
 }