From 5d83b9c2b01a204129a0069767e4f85bc4584441 Mon Sep 17 00:00:00 2001
From: dongyou <dongyou@espressif.com>
Date: Fri, 25 Dec 2020 11:45:12 +0800
Subject: [PATCH] add exception handling for iperf example's scan when no ap
 was found

---
 examples/wifi/iperf/main/cmd_wifi.c | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/examples/wifi/iperf/main/cmd_wifi.c b/examples/wifi/iperf/main/cmd_wifi.c
index a6be001c9b..d59b7c878a 100644
--- a/examples/wifi/iperf/main/cmd_wifi.c
+++ b/examples/wifi/iperf/main/cmd_wifi.c
@@ -61,6 +61,11 @@ static void scan_done_handler(void* arg, esp_event_base_t event_base,
     wifi_ap_record_t *ap_list_buffer;
 
     esp_wifi_scan_get_ap_num(&sta_number);
+    if (!sta_number) {
+        ESP_LOGE(TAG, "No AP found");
+        return;
+    }
+
     ap_list_buffer = malloc(sta_number * sizeof(wifi_ap_record_t));
     if (ap_list_buffer == NULL) {
         ESP_LOGE(TAG, "Failed to malloc buffer to print scan results");