Added ability to erase wifi sta/ap credentials from pycom config

Author: iwahdan88

esp32/mods/modpycom.c

@@ -395,9 +395,16 @@ STATIC MP_DEFINE_CONST_FUN_OBJ_VAR_BETWEEN(mod_pycom_wifi_mode_obj, 0, 1, mod_py
 
 STATIC mp_obj_t mod_pycom_wifi_ssid_sta (mp_uint_t n_args, const mp_obj_t *args) {
     if (n_args) {
-        if (MP_OBJ_IS_STR(args[0])) {
+        if(args[0] == mp_const_none)
+        {
+            config_set_sta_wifi_ssid (NULL, true);
+        }
+        else if (MP_OBJ_IS_STR(args[0]))
+        {
             config_set_sta_wifi_ssid ((uint8_t *)(mp_obj_str_get_str(args[0])), true);
         }
+        else{/*Nothing*/}
+
     } else {
         uint8_t * ssid = (uint8_t *)m_malloc(33);
         mp_obj_t ssid_obj;
@@ -418,9 +425,15 @@ STATIC MP_DEFINE_CONST_FUN_OBJ_VAR_BETWEEN(mod_pycom_wifi_ssid_sta_obj, 0, 1, mo
 
 STATIC mp_obj_t mod_pycom_wifi_pwd_sta (mp_uint_t n_args, const mp_obj_t *args) {
     if (n_args) {
-        if (MP_OBJ_IS_STR(args[0])) {
+        if(args[0] == mp_const_none)
+        {
+            config_set_wifi_sta_pwd (NULL, true);
+        }
+        else if (MP_OBJ_IS_STR(args[0]))
+        {
             config_set_wifi_sta_pwd ((uint8_t *)(mp_obj_str_get_str(args[0])), true);
         }
+        else{/*Nothing*/}
     } else {
         uint8_t * pwd = (uint8_t *)m_malloc(65);
         mp_obj_t pwd_obj;
@@ -441,9 +454,15 @@ STATIC MP_DEFINE_CONST_FUN_OBJ_VAR_BETWEEN(mod_pycom_wifi_pwd_sta_obj, 0, 1, mod
 
 STATIC mp_obj_t mod_pycom_wifi_ssid_ap (mp_uint_t n_args, const mp_obj_t *args) {
     if (n_args) {
-        if (MP_OBJ_IS_STR(args[0])) {
+        if(args[0] == mp_const_none)
+        {
+            config_set_wifi_ap_ssid (NULL);
+        }
+        else if (MP_OBJ_IS_STR(args[0]))
+        {
             config_set_wifi_ap_ssid ((uint8_t *)(mp_obj_str_get_str(args[0])));
         }
+        else{/*Nothing*/}
     } else {
         uint8_t * ssid = (uint8_t *)m_malloc(33);
         mp_obj_t ssid_obj;
@@ -464,9 +483,15 @@ STATIC MP_DEFINE_CONST_FUN_OBJ_VAR_BETWEEN(mod_pycom_wifi_ssid_ap_obj, 0, 1, mod
 
 STATIC mp_obj_t mod_pycom_wifi_pwd_ap (mp_uint_t n_args, const mp_obj_t *args) {
     if (n_args) {
-        if (MP_OBJ_IS_STR(args[0])) {
+        if(args[0] == mp_const_none)
+        {
+            config_set_wifi_ap_pwd (NULL);
+        }
+        else if (MP_OBJ_IS_STR(args[0]))
+        {
             config_set_wifi_ap_pwd ((uint8_t *)(mp_obj_str_get_str(args[0])));
         }
+        else{/*Nothing*/}
     } else {
         uint8_t * pwd = (uint8_t *)m_malloc(65);
         mp_obj_t pwd_obj;

esp32/pycom_config.c

@@ -180,7 +180,15 @@ bool config_get_wifi_antenna(void)
 
 bool config_set_wifi_ap_ssid(const uint8_t *wifi_ssid)
 {
-    memcpy(pycom_config_block.wifi_ap_config.wifi_ssid, wifi_ssid, sizeof(pycom_config_block.wifi_ap_config.wifi_ssid));
+    if(wifi_ssid != NULL)
+    {
+        memcpy(pycom_config_block.wifi_ap_config.wifi_ssid, wifi_ssid, sizeof(pycom_config_block.wifi_ap_config.wifi_ssid));
+    }
+    else
+    {
+        memset(pycom_config_block.wifi_ap_config.wifi_ssid, (uint8_t)0xFF, sizeof(pycom_config_block.wifi_ap_config.wifi_ssid));
+    }
+
     return config_write();
 }
 
@@ -195,7 +203,15 @@ bool config_get_wifi_ap_ssid(uint8_t *wifi_ssid)
 
 bool config_set_wifi_ap_pwd(const uint8_t *wifi_pwd)
 {
-    memcpy(pycom_config_block.wifi_ap_config.wifi_pwd, wifi_pwd, sizeof(pycom_config_block.wifi_ap_config.wifi_pwd));
+    if(wifi_pwd != NULL)
+    {
+        memcpy(pycom_config_block.wifi_ap_config.wifi_pwd, wifi_pwd, sizeof(pycom_config_block.wifi_ap_config.wifi_pwd));
+    }
+    else
+    {
+        memset(pycom_config_block.wifi_ap_config.wifi_pwd, (uint8_t)0xFF, sizeof(pycom_config_block.wifi_ap_config.wifi_pwd));
+    }
+
     return config_write();
 }
 
@@ -261,7 +277,13 @@ bool config_get_heartbeat_on_boot (void) {
 }
 
 bool config_set_sta_wifi_ssid (const uint8_t *wifi_ssid, bool update_flash) {
-    memcpy(pycom_config_block.wifi_sta_config.wifi_ssid, wifi_ssid, sizeof(pycom_config_block.wifi_sta_config.wifi_ssid));
+    if (wifi_ssid != NULL) {
+        memcpy(pycom_config_block.wifi_sta_config.wifi_ssid, wifi_ssid, sizeof(pycom_config_block.wifi_sta_config.wifi_ssid));
+    }
+    else
+    {
+        memset(pycom_config_block.wifi_sta_config.wifi_ssid, (uint8_t)0xFF, sizeof(pycom_config_block.wifi_sta_config.wifi_ssid));
+    }
     if (update_flash) {
         return config_write();
     }
@@ -280,7 +302,13 @@ bool config_get_wifi_sta_ssid (uint8_t *wifi_ssid) {
 }
 
 bool config_set_wifi_sta_pwd (const uint8_t *wifi_pwd, bool update_flash) {
-    memcpy(pycom_config_block.wifi_sta_config.wifi_pwd, wifi_pwd, sizeof(pycom_config_block.wifi_sta_config.wifi_pwd));
+    if (wifi_pwd != NULL) {
+        memcpy(pycom_config_block.wifi_sta_config.wifi_pwd, wifi_pwd, sizeof(pycom_config_block.wifi_sta_config.wifi_pwd));
+    }
+    else
+    {
+        memset(pycom_config_block.wifi_sta_config.wifi_pwd, (uint8_t)0xFF, sizeof(pycom_config_block.wifi_sta_config.wifi_pwd));
+    }
     if (update_flash) {
         return config_write();
     }