From 3f16eaf8b33dc687c226dc4d8253afa75f4a4f26 Mon Sep 17 00:00:00 2001
From: Juraj Andrassy <juraj.andrassy@gmail.com>
Date: Sat, 9 Sep 2023 08:11:43 +0200
Subject: [PATCH] DNS server IP getter dnsIP()

---
 src/WiFi.cpp  | 16 ++++++++++++++++
 src/WiFi101.h |  2 ++
 2 files changed, 18 insertions(+)

diff --git a/src/WiFi.cpp b/src/WiFi.cpp
index aee2374e..dce7b202 100644
--- a/src/WiFi.cpp
+++ b/src/WiFi.cpp
@@ -85,6 +85,7 @@ void WiFiClass::handleEvent(uint8_t u8MsgType, void *pvMsg)
 						_localip = 0;
 						_submask = 0;
 						_gateway = 0;
+						_dnsip = 0;
 					}
 					// Close sockets to clean state
 					// Clients will need to reconnect once the physical link will be re-established
@@ -110,6 +111,7 @@ void WiFiClass::handleEvent(uint8_t u8MsgType, void *pvMsg)
 				_localip = pstrIPCfg->u32StaticIP;
 				_submask = pstrIPCfg->u32SubnetMask;
 				_gateway = pstrIPCfg->u32Gateway;
+				_dnsip = pstrIPCfg->u32DNS;
 				
 				_status = WL_CONNECTED;
 
@@ -150,6 +152,7 @@ void WiFiClass::handleEvent(uint8_t u8MsgType, void *pvMsg)
 				_localip = 0;
 				_submask = 0;
 				_gateway = 0;
+				_dnsip = 0;
 				m2m_wifi_connect((char *)pstrProvInfo->au8SSID, strlen((char *)pstrProvInfo->au8SSID),
 						pstrProvInfo->u8SecType, pstrProvInfo->au8Password, M2M_WIFI_CH_ALL);
 			} else {
@@ -306,6 +309,7 @@ int WiFiClass::init()
 	_localip = 0;
 	_submask = 0;
 	_gateway = 0;
+	_dnsip = 0;
 	_dhcp = 1;
 	_resolve = 0;
 	_remoteMacAddress = 0;
@@ -369,6 +373,7 @@ uint8_t WiFiClass::begin()
 		_localip = 0;
 		_submask = 0;
 		_gateway = 0;
+		_dnsip = 0;
 	}
 	if (m2m_wifi_default_connect() < 0) {
 		_status = WL_CONNECT_FAILED;
@@ -431,6 +436,7 @@ uint8_t WiFiClass::startConnect(const char *ssid, uint8_t u8SecType, const void
 		_localip = 0;
 		_submask = 0;
 		_gateway = 0;
+		_dnsip = 0;
 	}
 	if (m2m_wifi_connect((char*)ssid, strlen(ssid), u8SecType, (void*)pvAuthInfo, M2M_WIFI_CH_ALL) < 0) {
 		_status = WL_CONNECT_FAILED;
@@ -553,6 +559,7 @@ uint8_t WiFiClass::startAP(const char *ssid, uint8_t u8SecType, const void *pvAu
 	m2m_memcpy((uint8 *)&_localip, (uint8 *)&strM2MAPConfig.au8DHCPServerIP[0], 4);
 	_submask = 0x00FFFFFF;
 	_gateway = _localip;
+	_dnsip = _localip;
 
 #ifdef CONF_PERIPH
 	// WiFi led ON (rev A then rev B).
@@ -619,6 +626,7 @@ uint8_t WiFiClass::startProvision(const char *ssid, const char *url, uint8_t cha
 	m2m_memcpy((uint8 *)&_localip, (uint8 *)&strM2MAPConfig.au8DHCPServerIP[0], 4);
 	_submask = 0x00FFFFFF;
 	_gateway = _localip;
+	_dnsip = _localip;
 
 #ifdef CONF_PERIPH
 	// WiFi led ON (rev A then rev B).
@@ -674,6 +682,7 @@ void WiFiClass::config(IPAddress local_ip, IPAddress dns_server, IPAddress gatew
 	_localip = conf.u32StaticIP;
 	_submask = conf.u32SubnetMask;
 	_gateway = conf.u32Gateway;
+	_dnsip = conf.u32DNS;
 }
 
 void WiFiClass::hostname(const char* name)
@@ -763,6 +772,13 @@ uint32_t WiFiClass::gatewayIP()
 	return _gateway;
 }
 
+uint32_t WiFiClass::dnsIP(int n)
+{
+  if (n > 0)
+    return IPAddress(0, 0, 0, 0);
+	return _dnsip;
+}
+
 char* WiFiClass::SSID()
 {
 	if (_status == WL_CONNECTED || _status == WL_AP_LISTENING || _status == WL_AP_CONNECTED) {
diff --git a/src/WiFi101.h b/src/WiFi101.h
index 00b38308..a27355da 100644
--- a/src/WiFi101.h
+++ b/src/WiFi101.h
@@ -133,6 +133,7 @@ class WiFiClass
 	uint32_t localIP();
 	uint32_t subnetMask();
 	uint32_t gatewayIP();
+	uint32_t dnsIP(int n = 0);
 	char* SSID();
 	int32_t RSSI();
 	uint8_t encryptionType();
@@ -174,6 +175,7 @@ class WiFiClass
 	uint32_t _localip;
 	uint32_t _submask;
 	uint32_t _gateway;
+	uint32_t _dnsip;
 	int _dhcp;
 	uint32_t _resolve;
 	byte *_remoteMacAddress;