Wait for Wi-Fi management to start up (#8834)

project-chip · Aug 30, 2021 · 9f1db8a · 9f1db8a
1 parent f07ff95
commit 9f1db8a
Show file tree

Hide file tree

Showing 3 changed files with 20 additions and 0 deletions.
diff --git a/examples/platform/linux/AppMain.cpp b/examples/platform/linux/AppMain.cpp
@@ -65,6 +65,9 @@ using namespace chip::Transport;
 using chip::Shell::Engine;
 #endif
 
+static constexpr useconds_t kWifiStartCheckTime = 100*1000; // 100 ms
+static constexpr uint8_t kWifiStartCheckAttempts = 5;
+
 namespace {
 void EventHandler(const chip::DeviceLayer::ChipDeviceEvent * event, intptr_t arg)
 {
@@ -80,6 +83,7 @@ int ChipLinuxAppInit(int argc, char ** argv)
 {
     CHIP_ERROR err                                   = CHIP_NO_ERROR;
     chip::RendezvousInformationFlags rendezvousFlags = chip::RendezvousInformationFlag::kBLE;
+    int counter                                      = 0;
 
 #ifdef CONFIG_RENDEZVOUS_MODE
     rendezvousFlags = static_cast<chip::RendezvousInformationFlags>(CONFIG_RENDEZVOUS_MODE);
@@ -120,6 +124,16 @@ int ChipLinuxAppInit(int argc, char ** argv)
     if (LinuxDeviceOptions::GetInstance().mWiFi)
     {
         chip::DeviceLayer::ConnectivityMgrImpl().StartWiFiManagement();
+        while (chip::DeviceLayer::ConnectivityMgrImpl().IsWiFiManagementStarted() == false)
+        {
+            usleep(kWifiStartCheckTime);
+            counter++;
+            if (counter == kWifiStartCheckAttempts)
+            {
+                ChipLogError(NotSpecified, "Wi-Fi Management taking too long to start.");
+                break;
+            }
+        }
     }
 #endif // CHIP_DEVICE_CONFIG_ENABLE_WPA
 

diff --git a/src/platform/Linux/ConnectivityManagerImpl.cpp b/src/platform/Linux/ConnectivityManagerImpl.cpp
@@ -633,6 +633,11 @@ void ConnectivityManagerImpl::StartWiFiManagement()
                                                 kWpaSupplicantObjectPath, nullptr, _OnWpaProxyReady, nullptr);
 }
 
+bool ConnectivityManagerImpl::IsWiFiManagementStarted()
+{
+    return mWpaSupplicant.state == GDBusWpaSupplicant::WPA_INTERFACE_CONNECTED;
+}
+
 void ConnectivityManagerImpl::DriveAPState()
 {
     CHIP_ERROR err = CHIP_NO_ERROR;

diff --git a/src/platform/Linux/ConnectivityManagerImpl.h b/src/platform/Linux/ConnectivityManagerImpl.h
@@ -108,6 +108,7 @@ class ConnectivityManagerImpl final : public ConnectivityManager,
 
 #if CHIP_DEVICE_CONFIG_ENABLE_WPA
     void StartWiFiManagement();
+    bool IsWiFiManagementStarted();
 #endif
 
 private: