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

Disable the long way and return an ap_info object

Browse Source 
still needs work and cleanup
This commit is contained in:
Ryan T. Hamilton 2020-10-15 23:18:30 -07:00
parent 99f27bea61
commit b336039aab
4 changed files with 104 additions and 59 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

83
ports/esp32s2/common-hal/wifi/Radio.c
View File

@ -25,6 +25,7 @@
*/
#include "shared-bindings/wifi/Radio.h"
#include "shared-bindings/wifi/Network.h"
#include <string.h>
@ -149,6 +150,30 @@ wifi_radio_error_t common_hal_wifi_radio_connect(wifi_radio_obj_t *self, uint8_t
return WIFI_RADIO_ERROR_NONE;
}
mp_obj_t common_hal_wifi_radio_get_ap_info(wifi_radio_obj_t *self) {
if (!esp_netif_is_netif_up(self->netif)) {
return mp_const_none;
}
// Make sure the interface is in STA mode
if (self->sta_mode){
return mp_const_none;
}
wifi_network_obj_t *apnet = m_new_obj(wifi_network_obj_t);
apnet->base.type = &wifi_network_type;
// From esp_wifi.h, the possible return values (typos theirs):
// ESP_OK: succeed
// ESP_ERR_WIFI_CONN: The station interface don't initialized
// ESP_ERR_WIFI_NOT_CONNECT: The station is in disconnect status
if (esp_wifi_sta_get_ap_info(&self->apnet.record) != ESP_OK){
return mp_const_none;
} else {
memcpy(&apnet->record, &self->apnet.record, sizeof(wifi_ap_record_t));
return MP_OBJ_FROM_PTR(apnet);
}
}
mp_obj_t common_hal_wifi_radio_get_ap_rssi(wifi_radio_obj_t *self) {
if (!esp_netif_is_netif_up(self->netif)) {
return mp_const_none;
@ -170,40 +195,40 @@ mp_obj_t common_hal_wifi_radio_get_ap_rssi(wifi_radio_obj_t *self) {
}
}
mp_obj_t common_hal_wifi_radio_get_ap_ssid(wifi_radio_obj_t *self) {
if (!esp_netif_is_netif_up(self->netif)) {
return mp_const_none;
}
// mp_obj_t common_hal_wifi_radio_get_ap_ssid(wifi_radio_obj_t *self) {
// if (!esp_netif_is_netif_up(self->netif)) {
// return mp_const_none;
// }
// Make sure the interface is in STA mode
if (self->sta_mode){
return mp_const_none;
}
// // Make sure the interface is in STA mode
// if (self->sta_mode){
// return mp_const_none;
// }
if (esp_wifi_sta_get_ap_info(&self->ap_info) != ESP_OK){
return mp_const_none;
} else {
const char* cstr = (const char*) self->ap_info.ssid;
return mp_obj_new_str(cstr, strlen(cstr));
}
}
// if (esp_wifi_sta_get_ap_info(&self->ap_info) != ESP_OK){
// return mp_const_none;
// } else {
// const char* cstr = (const char*) self->ap_info.ssid;
// return mp_obj_new_str(cstr, strlen(cstr));
// }
// }
mp_obj_t common_hal_wifi_radio_get_ap_bssid(wifi_radio_obj_t *self) {
if (!esp_netif_is_netif_up(self->netif)) {
return mp_const_none;
}
// mp_obj_t common_hal_wifi_radio_get_ap_bssid(wifi_radio_obj_t *self) {
// if (!esp_netif_is_netif_up(self->netif)) {
// return mp_const_none;
// }
// Make sure the interface is in STA mode
if (self->sta_mode){
return mp_const_none;
}
// // Make sure the interface is in STA mode
// if (self->sta_mode){
// return mp_const_none;
// }
if (esp_wifi_sta_get_ap_info(&self->ap_info) != ESP_OK){
return mp_const_none;
} else {
return mp_obj_new_bytes(self->ap_info.bssid, MAC_ADDRESS_LENGTH);
}
}
// if (esp_wifi_sta_get_ap_info(&self->ap_info) != ESP_OK){
// return mp_const_none;
// } else {
// return mp_obj_new_bytes(self->ap_info.bssid, MAC_ADDRESS_LENGTH);
// }
// }
mp_obj_t common_hal_wifi_radio_get_ipv4_gateway(wifi_radio_obj_t *self) {
if (!esp_netif_is_netif_up(self->netif)) {

2
ports/esp32s2/common-hal/wifi/Radio.h
View File

@ -32,6 +32,7 @@
#include "components/esp_event/include/esp_event.h"
#include "shared-bindings/wifi/ScannedNetworks.h"
#include "shared-bindings/wifi/Network.h"
// Event bits for the Radio event group.
#define WIFI_SCAN_DONE_BIT BIT0
@ -47,6 +48,7 @@ typedef struct {
EventGroupHandle_t event_group_handle;
wifi_config_t sta_config;
wifi_ap_record_t ap_info;
wifi_network_obj_t apnet;
esp_netif_ip_info_t ip_info;
esp_netif_dns_info_t dns_info;
esp_netif_t *netif;

73
shared-bindings/wifi/Radio.c
View File

@ -179,37 +179,37 @@ const mp_obj_property_t wifi_radio_ap_rssi_obj = {
(mp_obj_t)&mp_const_none_obj },
};
//| ap_ssid: int
//| """SSID of the currently connected AP. Returns none if not connected"""
//|
STATIC mp_obj_t wifi_radio_get_ap_ssid(mp_obj_t self) {
return common_hal_wifi_radio_get_ap_ssid(self);
// //| ap_ssid: int
// //| """SSID of the currently connected AP. Returns none if not connected"""
// //|
// STATIC mp_obj_t wifi_radio_get_ap_ssid(mp_obj_t self) {
// return common_hal_wifi_radio_get_ap_ssid(self);
}
MP_DEFINE_CONST_FUN_OBJ_1(wifi_radio_get_ap_ssid_obj, wifi_radio_get_ap_ssid);
// }
// MP_DEFINE_CONST_FUN_OBJ_1(wifi_radio_get_ap_ssid_obj, wifi_radio_get_ap_ssid);
const mp_obj_property_t wifi_radio_ap_ssid_obj = {
.base.type = &mp_type_property,
.proxy = { (mp_obj_t)&wifi_radio_get_ap_ssid_obj,
(mp_obj_t)&mp_const_none_obj,
(mp_obj_t)&mp_const_none_obj },
};
// const mp_obj_property_t wifi_radio_ap_ssid_obj = {
// .base.type = &mp_type_property,
// .proxy = { (mp_obj_t)&wifi_radio_get_ap_ssid_obj,
// (mp_obj_t)&mp_const_none_obj,
// (mp_obj_t)&mp_const_none_obj },
// };
//| ap_bssid: int
//| """BSSID (usually MAC) of the currently connected AP. Returns none if not connected"""
//|
STATIC mp_obj_t wifi_radio_get_ap_bssid(mp_obj_t self) {
return common_hal_wifi_radio_get_ap_bssid(self);
// //| ap_bssid: int
// //| """BSSID (usually MAC) of the currently connected AP. Returns none if not connected"""
// //|
// STATIC mp_obj_t wifi_radio_get_ap_bssid(mp_obj_t self) {
// return common_hal_wifi_radio_get_ap_bssid(self);
}
MP_DEFINE_CONST_FUN_OBJ_1(wifi_radio_get_ap_bssid_obj, wifi_radio_get_ap_bssid);
// }
// MP_DEFINE_CONST_FUN_OBJ_1(wifi_radio_get_ap_bssid_obj, wifi_radio_get_ap_bssid);
const mp_obj_property_t wifi_radio_ap_bssid_obj = {
.base.type = &mp_type_property,
.proxy = { (mp_obj_t)&wifi_radio_get_ap_bssid_obj,
(mp_obj_t)&mp_const_none_obj,
(mp_obj_t)&mp_const_none_obj },
};
// const mp_obj_property_t wifi_radio_ap_bssid_obj = {
// .base.type = &mp_type_property,
// .proxy = { (mp_obj_t)&wifi_radio_get_ap_bssid_obj,
// (mp_obj_t)&mp_const_none_obj,
// (mp_obj_t)&mp_const_none_obj },
// };
//| ipv4_gateway: Optional[ipaddress.IPv4Address]
//| """IP v4 Address of the gateway when connected to an access point. None otherwise."""
@ -275,6 +275,22 @@ const mp_obj_property_t wifi_radio_ipv4_dns_obj = {
(mp_obj_t)&mp_const_none_obj },
};
//| ap_info: Optional[Network]
//| """None otherwise."""
//|
STATIC mp_obj_t wifi_radio_get_ap_info(mp_obj_t self) {
return common_hal_wifi_radio_get_ap_info(self);
}
MP_DEFINE_CONST_FUN_OBJ_1(wifi_radio_get_ap_info_obj, wifi_radio_get_ap_info);
const mp_obj_property_t wifi_radio_ap_info_obj = {
.base.type = &mp_type_property,
.proxy = { (mp_obj_t)&wifi_radio_get_ap_info_obj,
(mp_obj_t)&mp_const_none_obj,
(mp_obj_t)&mp_const_none_obj },
};
//| def ping(self, ip, *, timeout: float = 0.5) -> float:
//| """Ping an IP to test connectivity. Returns echo time in seconds.
//| Returns None when it times out."""
@ -315,9 +331,10 @@ STATIC const mp_rom_map_elem_t wifi_radio_locals_dict_table[] = {
{ MP_ROM_QSTR(MP_QSTR_connect), MP_ROM_PTR(&wifi_radio_connect_obj) },
// { MP_ROM_QSTR(MP_QSTR_connect_to_enterprise), MP_ROM_PTR(&wifi_radio_connect_to_enterprise_obj) },
{ MP_ROM_QSTR(MP_QSTR_ap_info), MP_ROM_PTR(&wifi_radio_ap_info_obj) },
{ MP_ROM_QSTR(MP_QSTR_ap_rssi), MP_ROM_PTR(&wifi_radio_ap_rssi_obj) },
{ MP_ROM_QSTR(MP_QSTR_ap_ssid), MP_ROM_PTR(&wifi_radio_ap_ssid_obj) },
{ MP_ROM_QSTR(MP_QSTR_ap_bssid), MP_ROM_PTR(&wifi_radio_ap_bssid_obj) },
// { MP_ROM_QSTR(MP_QSTR_ap_ssid), MP_ROM_PTR(&wifi_radio_ap_ssid_obj) },
// { MP_ROM_QSTR(MP_QSTR_ap_bssid), MP_ROM_PTR(&wifi_radio_ap_bssid_obj) },
{ MP_ROM_QSTR(MP_QSTR_ipv4_dns), MP_ROM_PTR(&wifi_radio_ipv4_dns_obj) },
{ MP_ROM_QSTR(MP_QSTR_ipv4_gateway), MP_ROM_PTR(&wifi_radio_ipv4_gateway_obj) },
{ MP_ROM_QSTR(MP_QSTR_ipv4_subnet), MP_ROM_PTR(&wifi_radio_ipv4_subnet_obj) },

5
shared-bindings/wifi/Radio.h
View File

@ -53,9 +53,10 @@ extern void common_hal_wifi_radio_stop_scanning_networks(wifi_radio_obj_t *self)
extern wifi_radio_error_t common_hal_wifi_radio_connect(wifi_radio_obj_t *self, uint8_t* ssid, size_t ssid_len, uint8_t* password, size_t password_len, uint8_t channel, mp_float_t timeout, uint8_t* bssid, size_t bssid_len);
extern mp_obj_t common_hal_wifi_radio_get_ap_info(wifi_radio_obj_t *self);
extern mp_obj_t common_hal_wifi_radio_get_ap_rssi(wifi_radio_obj_t *self);
extern mp_obj_t common_hal_wifi_radio_get_ap_ssid(wifi_radio_obj_t *self);
extern mp_obj_t common_hal_wifi_radio_get_ap_bssid(wifi_radio_obj_t *self);
// extern mp_obj_t common_hal_wifi_radio_get_ap_ssid(wifi_radio_obj_t *self);
// extern mp_obj_t common_hal_wifi_radio_get_ap_bssid(wifi_radio_obj_t *self);
extern mp_obj_t common_hal_wifi_radio_get_ipv4_dns(wifi_radio_obj_t *self);
extern mp_obj_t common_hal_wifi_radio_get_ipv4_gateway(wifi_radio_obj_t *self);
extern mp_obj_t common_hal_wifi_radio_get_ipv4_subnet(wifi_radio_obj_t *self);