djsundog/circuitpython
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Implement async wifi connection on picow

Browse Source
This commit is contained in:
Bill Sideris 2022-10-17 18:17:14 +03:00
parent 5192082e64
commit bce024f59e
No known key found for this signature in database
GPG Key ID: 1BEF1BCEBA58EA33
1 changed files with 41 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

56
ports/raspberrypi/common-hal/wifi/Radio.c
View File

@ -173,24 +173,50 @@ wifi_radio_error_t common_hal_wifi_radio_connect(wifi_radio_obj_t *self, uint8_t
if (!common_hal_wifi_radio_get_enabled(self)) {
mp_raise_RuntimeError(translate("wifi is not enabled"));
}
unsigned timeout_ms = timeout <= 0 ? 8000 : (unsigned)MAX(0, MICROPY_FLOAT_C_FUN(ceil)(timeout * 1000));
// TODO use connect_async so we can service bg tasks & check for ctrl-c during
size_t timeout_ms = timeout <= 0 ? 8000 : (size_t)MICROPY_FLOAT_C_FUN(ceil)(timeout * 1000);
uint64_t start = port_get_raw_ticks(NULL);
uint64_t deadline = start + timeout_ms;
// connect
int result = cyw43_arch_wifi_connect_timeout_ms((const char *)ssid, (const char *)password, CYW43_AUTH_WPA2_AES_PSK, timeout_ms);
bindings_cyw43_wifi_enforce_pm();
cyw43_arch_wifi_connect_async((const char *)ssid, (const char *)password, CYW43_AUTH_WPA2_AES_PSK);
while (port_get_raw_ticks(NULL) < deadline) {
RUN_BACKGROUND_TASKS;
if (mp_hal_is_interrupted()) {
break;
}
int result = cyw43_wifi_link_status(&cyw43_state, CYW43_ITF_STA);
// While the async connection is running, it will return CYW43_LINK_JOIN.
int done = 0;
switch (result) {
case CYW43_LINK_UP:
done = 1;
break;
case (unsigned int)CYW43_LINK_FAIL:
return WIFI_RADIO_ERROR_CONNECTION_FAIL;
case (unsigned int)CYW43_LINK_NONET:
return WIFI_RADIO_ERROR_NO_AP_FOUND;
case (unsigned int)CYW43_LINK_BADAUTH:
return WIFI_RADIO_ERROR_AUTH_FAIL;
}
if (done == 1) {
break;
}
}
// Being here means we either timed out or got interrupted.
int result = cyw43_wifi_link_status(&cyw43_state, CYW43_ITF_STA);
switch (result) {
case 0:
case CYW43_LINK_UP:
case CYW43_LINK_JOIN:
// If CYW43_LINK_JOIN wasn't here, it wouldn't work even though we have ip.
bindings_cyw43_wifi_enforce_pm();
return WIFI_RADIO_ERROR_NONE;
// case CYW43_LINK_DOWN:
// case CYW43_LINK_JOIN:
// case CYW43_LINK_NOIP:
// case CYW43_LINK_UP:
case CYW43_LINK_FAIL:
return WIFI_RADIO_ERROR_CONNECTION_FAIL;
case CYW43_LINK_NONET:
return WIFI_RADIO_ERROR_NO_AP_FOUND;
case CYW43_LINK_BADAUTH:
return WIFI_RADIO_ERROR_AUTH_FAIL;
default:
return WIFI_RADIO_ERROR_UNSPECIFIED;