[bouffalo lab] fix bl616 wifi scan operation with/without specified ssid (project-chip#38877)

* [bouffalo lab] fix bl616 wifi scan operation with/without specified ssid

* Restyled by clang-format

* check ssid length before

* Verify SSID length prior to copying and correct pScanResult reference handling when SSID is specified.

* Restyled by clang-format

* remove +1 for mScanSSID

* Restyled by clang-format

---------

Co-authored-by: Restyled.io <[email protected]>

Author: wy-hh

src/platform/bouffalolab/BL616/NetworkCommissioningDriver.cpp

@@ -205,10 +205,14 @@ void BLWiFiDriver::ScanNetworks(ByteSpan ssid, WiFiDriver::ScanCallback * callba
 {
     if (callback != nullptr)
     {
-        mpScanCallback = nullptr;
-        if (0 == wifi_start_scan(ssid.data(), ssid.size()))
+        mpScanCallback  = nullptr;
+        mScanSSIDlength = 0;
+
+        if (ssid.size() <= sizeof(mScanSSID) && 0 == wifi_start_scan(ssid.data(), ssid.size()))
         {
-            mpScanCallback = callback;
+            memcpy(mScanSSID, ssid.data(), ssid.size());
+            mScanSSIDlength = ssid.size();
+            mpScanCallback  = callback;
         }
         else
         {
@@ -231,9 +235,30 @@ void BLWiFiDriver::OnScanWiFiNetworkDone()
         }
         else
         {
+            wifi_mgmr_scan_item_t * pScanResult = NULL;
+            uint32_t scanResultNum              = 0;
+
+            if (mScanSSIDlength)
+            {
+                for (uint32_t i = 0; i < nums; i++)
+                {
+                    if (mScanSSIDlength == pScanList[i].ssid_len && memcmp(pScanList[i].ssid, mScanSSID, mScanSSIDlength) == 0)
+                    {
+                        pScanResult   = &pScanList[i];
+                        scanResultNum = 1;
+                        break;
+                    }
+                }
+            }
+            else
+            {
+                pScanResult   = pScanList;
+                scanResultNum = nums;
+            }
+
+            if (CHIP_NO_ERROR != DeviceLayer::SystemLayer().ScheduleLambda([scanResultNum, pScanResult, pScanList]() {
+                    BLScanResponseIterator iter(scanResultNum, pScanResult);
 
-            if (CHIP_NO_ERROR == DeviceLayer::SystemLayer().ScheduleLambda([nums, pScanList]() {
-                    BLScanResponseIterator iter(nums, pScanList);
                     if (GetInstance().mpScanCallback)
                     {
                         GetInstance().mpScanCallback->OnFinished(Status::kSuccess, CharSpan(), &iter);
@@ -243,6 +268,8 @@ void BLWiFiDriver::OnScanWiFiNetworkDone()
                     {
                         ChipLogError(DeviceLayer, "can't find the ScanCallback function");
                     }
+
+                    MemoryFree(pScanList);
                 }))
             {
                 MemoryFree(pScanList);

src/platform/bouffalolab/BL616/NetworkCommissioningDriver.h

@@ -141,6 +141,9 @@ class BLWiFiDriver final : public WiFiDriver
     ConnectCallback * mpConnectCallback;
     NetworkStatusChangeCallback * mpStatusChangeCallback = nullptr;
     int32_t mLastDisconnectedReason;
+
+    char mScanSSID[DeviceLayer::Internal::kMaxWiFiSSIDLength];
+    int mScanSSIDlength;
 };
 
 } // namespace NetworkCommissioning