fix issue with STA emiting the wrong event

me-no-dev · me-no-dev · commit 8332a235d636 · 2018-01-18T13:54:26.000+02:00
diff --git a/libraries/WiFi/src/WiFiGeneric.cpp b/libraries/WiFi/src/WiFiGeneric.cpp
@@ -313,8 +313,12 @@ esp_err_t WiFiGenericClass::_eventCallback(void *arg, system_event_t *event)
         WiFiSTAClass::_setStatus(WL_DISCONNECTED);
     } else if(event->event_id == SYSTEM_EVENT_STA_STOP) {
         WiFiSTAClass::_setStatus(WL_NO_SHIELD);
+    } else if(event->event_id == SYSTEM_EVENT_STA_CONNECTED) {
+        WiFiSTAClass::_setStatus(WL_IDLE_STATUS);
     } else if(event->event_id == SYSTEM_EVENT_STA_GOT_IP) {
-        WiFiSTAClass::_setStatus(WL_CONNECTED);
+        if(WiFiSTAClass::status() == WL_IDLE_STATUS) {
+            WiFiSTAClass::_setStatus(WL_CONNECTED);
+        }
     }
 
     for(uint32_t i = 0; i < cbEventList.size(); i++) {
diff --git a/libraries/WiFi/src/WiFiSTA.cpp b/libraries/WiFi/src/WiFiSTA.cpp
@@ -358,7 +358,7 @@ uint8_t WiFiSTAClass::waitForConnectResult()
         return WL_DISCONNECTED;
     }
     int i = 0;
-    while(status() >= WL_DISCONNECTED && i++ < 100) {
+    while((!status() || status() >= WL_DISCONNECTED) && i++ < 100) {
         delay(100);
     }
     return status();
diff --git a/libraries/WiFi/src/WiFiSTA.h b/libraries/WiFi/src/WiFiSTA.h
@@ -72,7 +72,7 @@ class WiFiSTAClass
     bool setHostname(const char * hostname);
 
     // STA WiFi info
-    wl_status_t status();
+    static wl_status_t status();
     String SSID() const;
     String psk() const;
 

Original file line number	Diff line number	Diff line change
`@@ -358,7 +358,7 @@ uint8_t WiFiSTAClass::waitForConnectResult()`
`358`	`358`	`return WL_DISCONNECTED;`
`359`	`359`	`}`
`360`	`360`	`int i = 0;`
`361`		`- while(status() >= WL_DISCONNECTED && i++ < 100) {`
	`361`	`+ while((!status() \|\| status() >= WL_DISCONNECTED) && i++ < 100) {`
`362`	`362`	`delay(100);`
`363`	`363`	`}`
`364`	`364`	`return status();`