commit 9c56858302d19894ffec33089251cc9363944fb0
author Shubham Patil <shubham.patil@espressif.com> Fri Aug 25 16:00:56 2023 +0530
committer GitHub <noreply@github.com> Fri Aug 25 10:30:56 2023 +0000
tree a526a6f0f9928fbb3b7f75c157310471c6135c36
parent 6f71e28cd80f703c0e90c33ba7f8f35fe21deac3

[ESP32] encrypted_ota: Call abort API when aborting OTA and some refactoring (#28778)

* [ESP32] encrypted_ota: Call abort API when aborting and some refactoring

* Make DecryptEnd return an error and honour it

* Restyled by clang-format

---------

Co-authored-by: Restyled.io <commits@restyled.io>

src/platform/ESP32/OTAImageProcessorImpl.h

diff --git a/src/platform/ESP32/OTAImageProcessorImpl.h b/src/platform/ESP32/OTAImageProcessorImpl.h
index 03e1367..4162711 100644
--- a/src/platform/ESP32/OTAImageProcessorImpl.h
+++ b/src/platform/ESP32/OTAImageProcessorImpl.h
@@ -67,7 +67,14 @@
     OTAImageHeaderParser mHeaderParser;
 
 #if CONFIG_ENABLE_ENCRYPTED_OTA
-    void EndDecryption();
+    CHIP_ERROR DecryptStart();
+    CHIP_ERROR DecryptEnd();
+    void DecryptAbort();
+
+    // This API decrypts the blockToDecrypt, dynamically allocates the memory for storing the
+    // plain text, and return that in decryptedBlock.
+    // Caller shall free the memory after use by calling free() on decryptedBlock.data()
+    CHIP_ERROR DecryptBlock(const ByteSpan & blockToDecrypt, ByteSpan & decryptedBlock);
 
     CharSpan mKey;
     bool mEncryptedOTAEnabled                 = false;