Return number of bytes in OTA receive buffer when calling 'onOTADataR…

…eceived'
arduino-libraries · Jul 27, 2020 · 8b64fea · 8b64fea
1 parent 336f33a
commit 8b64fea
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/src/utility/ota/OTALogic.cpp b/src/utility/ota/OTALogic.cpp
@@ -103,18 +103,20 @@ OTAError OTALogic::update()
   return _ota_error;
 }
 
-void OTALogic::onOTADataReceived(uint8_t const * const data, size_t const length)
+uint32_t OTALogic::onOTADataReceived(uint8_t const * const data, size_t const length)
 {
   size_t const bytes_available = (MQTT_OTA_BUF_SIZE - _mqtt_ota_buf.num_bytes);
   if(length <= bytes_available)
   {
     memcpy(_mqtt_ota_buf.buf + _mqtt_ota_buf.num_bytes, data, length);
     _mqtt_ota_buf.num_bytes += length;
+    return _mqtt_ota_buf.num_bytes;
   }
   else
   {
     _ota_state = OTAState::Error;
     _ota_error = OTAError::ReceivedDataOverrun;
+    return 0;
   }
 }
 

diff --git a/src/utility/ota/OTALogic.h b/src/utility/ota/OTALogic.h
@@ -76,7 +76,7 @@ class OTALogic
 
 
   OTAError update();
-  void onOTADataReceived(uint8_t const * const data, size_t const length);
+  uint32_t onOTADataReceived(uint8_t const * const data, size_t const length);
 
 #ifdef HOST
   inline OTAState state() const { return _ota_state; }