src: Disable PM mode during wifi join.

Had a few reports that joining wifi can be very slow. Disabling power
management seems to fix the problem. The PM mode is cleared when a join
is started and restored when complete on success or failure.

Fixes #9

Author: peterharperuk

src/cyw43.h

@@ -124,6 +124,9 @@ typedef struct _cyw43_t {
 
     // mac from otp (or from cyw43_hal_generate_laa_mac if not set)
     uint8_t mac[6];
+
+    // Power management setting, restored after connect
+    uint32_t saved_pm;
 } cyw43_t;
 
 extern cyw43_t cyw43_state;

src/cyw43_ctrl.c

@@ -309,6 +309,7 @@ STATIC void cyw43_dump_async_event(const cyw43_async_event_t *ev) {
 
 void cyw43_cb_process_async_event(void *cb_data, const cyw43_async_event_t *ev) {
     cyw43_t *self = cb_data;
+    bool connect_done = false;
 
     if (self->trace_flags & CYW43_TRACE_ASYNC_EV) {
         cyw43_dump_async_event(ev);
@@ -352,10 +353,12 @@ void cyw43_cb_process_async_event(void *cb_data, const cyw43_async_event_t *ev)
             // Success setting SSID
         } else if (ev->status == 3 && ev->reason == 0) {
             self->wifi_join_state = WIFI_JOIN_STATE_NONET;
+            connect_done = true;
             // No matching SSID found (could be out of range, or down)
         } else {
             // Other failure setting SSID
             self->wifi_join_state = WIFI_JOIN_STATE_FAIL;
+            connect_done = true;
         }
     } else if (ev->event_type == CYW43_EV_AUTH) {
         if (ev->status == 0) {
@@ -369,6 +372,7 @@ void cyw43_cb_process_async_event(void *cb_data, const cyw43_async_event_t *ev)
         } else {
             // Cannot authenticate
             self->wifi_join_state = WIFI_JOIN_STATE_BADAUTH;
+            connect_done = true;
         }
     } else if (ev->event_type == CYW43_EV_DEAUTH_IND) {
         if (ev->status == 0 && ev->reason == 2) {
@@ -401,13 +405,22 @@ void cyw43_cb_process_async_event(void *cb_data, const cyw43_async_event_t *ev)
         } else {
             // PSK_SUP failure
             self->wifi_join_state = WIFI_JOIN_STATE_BADAUTH;
+            connect_done = true;
         }
     }
 
     if (self->wifi_join_state == WIFI_JOIN_STATE_ALL) {
+
         // STA connected
         self->wifi_join_state = WIFI_JOIN_STATE_ACTIVE;
         cyw43_cb_tcpip_set_link_up(self, CYW43_ITF_STA);
+        connect_done = true;
+    }
+
+    // Restore power management value
+    if (connect_done && self->saved_pm) {
+        cyw43_wifi_pm(self, self->saved_pm);
+        self->saved_pm = 0;
     }
 }
 
@@ -604,6 +617,11 @@ int cyw43_wifi_join(cyw43_t *self, size_t ssid_len, const uint8_t *ssid, size_t
         return ret;
     }
 
+    // Disable power management while connecting
+    if (cyw43_wifi_get_pm(self, &self->saved_pm) == 0) {
+        cyw43_wifi_pm(self, self->saved_pm & ~0xf);
+    }
+
     ret = cyw43_ll_wifi_join(&self->cyw43_ll, ssid_len, ssid, key_len, key, auth_type, bssid, channel);
 
     if (ret == 0) {