Update WiFi timeout condition (fixes #122)

Author: maakbaas

src/WiFiManager.cpp

@@ -39,7 +39,7 @@ void WifiManager::begin(char const *apName, unsigned long newTimeout)
         WiFi.begin();
     }
 
-    if (WiFi.waitForConnectResult(timeout) == WL_CONNECTED)
+    if (waitForConnectResult(timeout) == WL_CONNECTED)
     {
         //connected
         Serial.println(PSTR("Connected to stored WiFi details"));
@@ -52,6 +52,23 @@ void WifiManager::begin(char const *apName, unsigned long newTimeout)
     }
 }
 
+//Upgraded default waitForConnectResult function to incorporate WL_NO_SSID_AVAIL, fixes issue #122
+int8_t WifiManager::waitForConnectResult(unsigned long timeoutLength) {
+    //1 and 3 have STA enabled
+    if((wifi_get_opmode() & 1) == 0) {
+        return WL_DISCONNECTED;
+    }
+    using esp8266::polledTimeout::oneShot;
+    oneShot timeout(timeoutLength); // number of milliseconds to wait before returning timeout error
+    while(!timeout) {
+        yield();
+        if(WiFi.status() != WL_DISCONNECTED && WiFi.status() != WL_NO_SSID_AVAIL) {
+            return WiFi.status();
+        }
+    }
+    return -1; // -1 indicates timeout
+}
+
 //function to forget current WiFi details and start a captive portal
 void WifiManager::forget()
 { 

src/WiFiManager.h

@@ -25,6 +25,7 @@ class WifiManager
     void stopCaptivePortal();
     void connectNewWifi(String newSSID, String newPass);    
     void storeToEEPROM();
+    int8_t waitForConnectResult(unsigned long timeoutLength);
 
 public : 
     void begin(char const *apName, unsigned long newTimeout = 60000);