TCP: do not disconnect if ota client is connected

Author: pennam

src/ArduinoIoTCloudTCP.cpp

@@ -240,6 +240,18 @@ void ArduinoIoTCloudTCP::update()
 #if defined (ARDUINO_ARCH_SAMD) || defined (ARDUINO_ARCH_MBED)
   watchdog_reset();
 #endif
+
+#if OTA_ENABLED
+  /* OTA FSM needs to reach the Idle state before being able to run independently from
+   * the mqttClient. The state can be reached only after the mqttClient is connected to
+   * the broker.
+   */
+  if((_ota.getState() != OTACloudProcessInterface::Resume &&
+      _ota.getState() != OTACloudProcessInterface::OtaBegin) ||
+      _mqttClient.connected()) {
+    _ota.update();
+  }
+#endif // OTA_ENABLED
 }
 
 int ArduinoIoTCloudTCP::connected()
@@ -319,7 +331,15 @@ ArduinoIoTCloudTCP::State ArduinoIoTCloudTCP::handle_ConnectMqttBroker()
 
 ArduinoIoTCloudTCP::State ArduinoIoTCloudTCP::handle_Connected()
 {
-  if (!_mqttClient.connected() || !_thing.connected() || !_device.connected())
+#if  OTA_ENABLED
+  if(_get_ota_confirmation != nullptr &&
+      _ota.getState() == OTACloudProcessInterface::State::OtaAvailable &&
+      _get_ota_confirmation()) {
+    _ota.approveOta();
+  }
+#endif // OTA_ENABLED
+
+  if ((!_mqttClient.connected() || !_thing.connected() || !_device.connected()))
   {
     return State::Disconnect;
   }
@@ -338,16 +358,6 @@ ArduinoIoTCloudTCP::State ArduinoIoTCloudTCP::handle_Connected()
   /* Call CloudDevice process to get configuration */
   _device.update();
 
-#if  OTA_ENABLED
-  if(_get_ota_confirmation != nullptr &&
-      _ota.getState() == OTACloudProcessInterface::State::OtaAvailable &&
-      _get_ota_confirmation()) {
-    _ota.approveOta();
-  }
-
-  _ota.update();
-#endif // OTA_ENABLED
-
 
   if (_device.isAttached()) {
     /* Call CloudThing process to synchronize properties */