esp32/modusocket:Added mpy API to get/se/clear DNS Servers Ipv4

Author: iwahdan88

esp32/mods/modusocket.c

@@ -72,6 +72,7 @@
  ******************************************************************************/
 #define MODUSOCKET_MAX_SOCKETS                      15
 #define MODUSOCKET_CONN_TIMEOUT                     -2
+#define MODUSOCKET_MAX_DNS_SERV                     2
 /******************************************************************************
  DEFINE PRIVATE TYPES
  ******************************************************************************/
@@ -891,11 +892,115 @@ STATIC mp_obj_t mod_usocket_getaddrinfo(size_t n_args, const mp_obj_t *args) {
 }
 STATIC MP_DEFINE_CONST_FUN_OBJ_VAR_BETWEEN(mod_usocket_getaddrinfo_obj, 2, 6, mod_usocket_getaddrinfo);
 
+STATIC mp_obj_t mod_usocket_dnsserver(size_t n_args, const mp_obj_t *args)
+{
+    if(n_args == 1)
+    {
+        mp_obj_t tuple[2];
+        ip_addr_t ipaddr;
+        uint8_t numdns;
+        if(MP_OBJ_IS_INT(args[0]))
+        {
+            numdns = mp_obj_get_int(args[0]);
+            if (numdns > MODUSOCKET_MAX_DNS_SERV)
+            {
+                nlr_raise(mp_obj_new_exception_msg(&mp_type_ValueError, "DNS index Greater than MAX DNS Servers Possible to Config!\n"));
+            }
+        }
+        else
+        {
+            nlr_raise(mp_obj_new_exception_msg(&mp_type_OSError, "Please Specify Valid DNS index\n"));
+        }
+
+        ipaddr = dns_getserver(numdns);
+        if(ipaddr.type == 0)
+        {
+            tuple[0] = mp_obj_new_int(numdns);
+            tuple[1] = netutils_format_ipv4_addr((uint8_t *)&ipaddr.u_addr.ip4.addr, NETUTILS_BIG);
+        }
+        else
+        {
+            nlr_raise(mp_obj_new_exception_msg(&mp_type_OSError, "Only IPv4 addresses are currently supported\n"));
+        }
+        return mp_obj_new_tuple(2, tuple);
+    }
+    else if(n_args > 1)
+    {
+        ip_addr_t dnsserver;
+        uint8_t numdns;
+        //get DNS Server index
+        if(mp_obj_is_integer(args[0]))
+        {
+            numdns = mp_obj_get_int(args[0]);
+            if (numdns > MODUSOCKET_MAX_DNS_SERV)
+            {
+                nlr_raise(mp_obj_new_exception_msg(&mp_type_ValueError, "DNS index Greater than MAX DNS Servers Possible to Config!\n"));
+            }
+        }
+        else
+        {
+            nlr_raise(mp_obj_new_exception_msg(&mp_type_OSError, "Please Specify Valid DNS index\n"));
+        }
+        //parse dns Server IP
+        netutils_parse_ipv4_addr(args[1], (uint8_t *)&dnsserver.u_addr.ip4.addr, NETUTILS_BIG);
+        //IPv4
+        dnsserver.type = 0;
+
+        //set DNS Server
+        dns_setserver(numdns, &dnsserver);
+
+        return mp_const_none;
+
+    }
+    else
+    {
+        nlr_raise(mp_obj_new_exception_msg(&mp_type_ValueError, mpexception_value_invalid_arguments));
+    }
+    return mp_const_none;
+}
+STATIC MP_DEFINE_CONST_FUN_OBJ_VAR_BETWEEN(mod_usocket_dnsserver_obj, 1, 2, mod_usocket_dnsserver);
+
+STATIC mp_obj_t mod_usocket_dnsclear(size_t n_args, const mp_obj_t *args)
+{
+    uint8_t numdns;
+    if(n_args > 0)
+    {
+        if(MP_OBJ_IS_INT(args[0]))
+        {
+            numdns = mp_obj_get_int(args[0]);
+            if (numdns > MODUSOCKET_MAX_DNS_SERV)
+            {
+                nlr_raise(mp_obj_new_exception_msg(&mp_type_ValueError, "DNS index Greater than MAX DNS Servers Possible to Config!\n"));
+            }
+        }
+        else
+        {
+            nlr_raise(mp_obj_new_exception_msg(&mp_type_ValueError, "Please Specify Valid DNS index\n"));
+        }
+
+        //clear DNS server specified
+        dns_setserver(numdns, NULL);
+    }
+    else
+    {
+        uint8_t index;
+        // clear all DNS servers
+        for(index = 0; index < MODUSOCKET_MAX_DNS_SERV; index++)
+        {
+            dns_setserver(index, NULL);
+        }
+    }
+    return mp_const_none;
+}
+STATIC MP_DEFINE_CONST_FUN_OBJ_VAR_BETWEEN(mod_usocket_dnsclear_obj, 0, 1, mod_usocket_dnsclear);
+
 STATIC const mp_map_elem_t mp_module_usocket_globals_table[] = {
     { MP_OBJ_NEW_QSTR(MP_QSTR___name__),        MP_OBJ_NEW_QSTR(MP_QSTR_usocket) },
 
     { MP_OBJ_NEW_QSTR(MP_QSTR_socket),          (mp_obj_t)&socket_type },
     { MP_OBJ_NEW_QSTR(MP_QSTR_getaddrinfo),     (mp_obj_t)&mod_usocket_getaddrinfo_obj },
+    { MP_OBJ_NEW_QSTR(MP_QSTR_dnsserver),       (mp_obj_t)&mod_usocket_dnsserver_obj },
+    { MP_OBJ_NEW_QSTR(MP_QSTR_dns_clear),       (mp_obj_t)&mod_usocket_dnsclear_obj },
 
     // class exceptions
     { MP_OBJ_NEW_QSTR(MP_QSTR_error),           (mp_obj_t)&mp_type_OSError },