Update ArduinoOTA and examples with MDNS.update() calls

libraries/ArduinoOTA/ArduinoOTA.cpp

@@ -33,6 +33,7 @@ ArduinoOTAClass::ArduinoOTAClass()
 , _udp_ota(0)
 , _initialized(false)
 , _rebootOnSuccess(true)
+, _useMDNS(true)
 , _state(OTA_IDLE)
 , _size(0)
 , _cmd(0)
@@ -103,10 +104,12 @@ void ArduinoOTAClass::setRebootOnSuccess(bool reboot){
   _rebootOnSuccess = reboot;
 }
 
-void ArduinoOTAClass::begin() {
+void ArduinoOTAClass::begin(bool useMDNS) {
   if (_initialized)
     return;
 
+  _useMDNS = useMDNS;
+
   if (!_hostname.length()) {
     char tmp[15];
     sprintf(tmp, "esp8266-%06x", ESP.getChipId());
@@ -127,12 +130,15 @@ void ArduinoOTAClass::begin() {
   if(!_udp_ota->listen(IP_ADDR_ANY, _port))
     return;
   _udp_ota->onRx(std::bind(&ArduinoOTAClass::_onRx, this));
-  MDNS.begin(_hostname.c_str());
 
-  if (_password.length()) {
-    MDNS.enableArduino(_port, true);
-  } else {
-    MDNS.enableArduino(_port);
+  if(_useMDNS) {
+    MDNS.begin(_hostname.c_str());
+
+    if (_password.length()) {
+      MDNS.enableArduino(_port, true);
+    } else {
+      MDNS.enableArduino(_port);
+    }
   }
   _initialized = true;
   _state = OTA_IDLE;
@@ -348,11 +354,15 @@ void ArduinoOTAClass::_runUpdate() {
   }
 }
 
+//this needs to be called in the loop()
 void ArduinoOTAClass::handle() {
   if (_state == OTA_RUNUPDATE) {
     _runUpdate();
     _state = OTA_IDLE;
   }
+
+  if(_useMDNS)
+    MDNS.update(); //handle MDNS update as well, given that ArduinoOTA relies on it anyways
 }
 
 int ArduinoOTAClass::getCommand() {

libraries/ArduinoOTA/ArduinoOTA.h

@@ -60,9 +60,9 @@ class ArduinoOTAClass
     void onProgress(THandlerFunction_Progress fn);
 
     //Starts the ArduinoOTA service
-    void begin();
+    void begin(bool useMDNS = true);
 
-    //Call this in loop() to run the service
+    //Call this in loop() to run the service. Also calls MDNS.update() when begin() or begin(true) is used.
     void handle();
 
     //Gets update command type after OTA has started. Either U_FLASH or U_SPIFFS
@@ -76,6 +76,7 @@ class ArduinoOTAClass
     UdpContext *_udp_ota;
     bool _initialized;
     bool _rebootOnSuccess;
+    bool _useMDNS;
     ota_state_t _state;
     int _size;
     int _cmd;

libraries/DNSServer/examples/CaptivePortalAdvanced/CaptivePortalAdvanced.ino

@@ -130,6 +130,9 @@ void loop() {
         WiFi.disconnect();
       }
     }
+    if (s == WL_CONNECTED) {
+      MDNS.update();
+    }
   }
   // Do work:
   //DNS

libraries/ESP8266AVRISP/examples/Arduino_Wifi_AVRISP/Arduino_Wifi_AVRISP.ino

@@ -70,4 +70,8 @@ void loop() {
   if (last_state != AVRISP_STATE_IDLE) {
     avrprog.serve();
   }
+
+  if (WiFi.status() == WL_CONNECTED) {
+    MDNS.update();
+  }
 }

libraries/ESP8266HTTPUpdateServer/examples/SecureBearSSLUpdater/SecureBearSSLUpdater.ino

@@ -119,4 +119,5 @@ void setup()
 void loop()
 {
   httpServer.handleClient();
+  MDNS.update();
 }

libraries/ESP8266HTTPUpdateServer/examples/SecureHTTPSUpdater/SecureHTTPSUpdater.ino

@@ -188,4 +188,5 @@ void setup() {
 
 void loop() {
   httpServer.handleClient();
+  MDNS.update();
 }

libraries/ESP8266HTTPUpdateServer/examples/SecureWebUpdater/SecureWebUpdater.ino

@@ -47,4 +47,5 @@ void setup(void) {
 
 void loop(void) {
   httpServer.handleClient();
+  MDNS.update();
 }

libraries/ESP8266HTTPUpdateServer/examples/WebUpdater/WebUpdater.ino

@@ -44,4 +44,5 @@ void setup(void) {
 
 void loop(void) {
   httpServer.handleClient();
+  MDNS.update();
 }

libraries/ESP8266WebServer/examples/AdvancedWebServer/AdvancedWebServer.ino

@@ -130,6 +130,7 @@ void setup(void) {
 
 void loop(void) {
   server.handleClient();
+  MDNS.update();
 }
 
 void drawGraph() {

libraries/ESP8266WebServer/examples/FSBrowser/FSBrowser.ino

@@ -282,4 +282,5 @@ void setup(void) {
 
 void loop(void) {
   server.handleClient();
+  MDNS.update();
 }

libraries/ESP8266WebServer/examples/HelloServer/HelloServer.ino

@@ -75,4 +75,5 @@ void setup(void) {
 
 void loop(void) {
   server.handleClient();
+  MDNS.update();
 }

libraries/ESP8266WebServer/examples/HelloServerBearSSL/HelloServerBearSSL.ino

@@ -144,4 +144,5 @@ void setup(void){
 
 void loop(void){
   server.handleClient();
+  MDNS.update();
 }

libraries/ESP8266WebServer/examples/HelloServerSecure/HelloServerSecure.ino

@@ -181,4 +181,5 @@ void setup(void) {
 
 void loop(void) {
   server.handleClient();
+  MDNS.update();
 }

libraries/ESP8266WebServer/examples/SDWebServer/SDWebServer.ino

@@ -315,4 +315,5 @@ void setup(void) {
 
 void loop(void) {
   server.handleClient();
+  MDNS.update();
 }

libraries/ESP8266WebServer/examples/WebUpdate/WebUpdate.ino

@@ -70,5 +70,5 @@ void setup(void) {
 
 void loop(void) {
   server.handleClient();
-  delay(1);
+  MDNS.update();
 }

libraries/ESP8266mDNS/examples/OTA-mDNS-SPIFFS/OTA-mDNS-SPIFFS.ino

@@ -238,6 +238,5 @@ void setup() {
 void loop() {
   // Handle OTA server.
   ArduinoOTA.handle();
-  yield();
 }
 

libraries/ESP8266mDNS/examples/mDNS-SD_Extended/mDNS-SD_Extended.ino

@@ -76,5 +76,5 @@ void setup() {
 
 void loop() {
   // put your main code here, to run repeatedly:
-
+  MDNS.update();
 }

libraries/ESP8266mDNS/examples/mDNS_Web_Server/mDNS_Web_Server.ino

@@ -72,6 +72,9 @@ void setup(void) {
 }
 
 void loop(void) {
+
+  MDNS.update();
+
   // Check if a client has connected
   WiFiClient client = server.available();
   if (!client) {