change disable logic

Author: Jason2866

libraries/Update/src/Updater.cpp

@@ -9,9 +9,9 @@
 #include "spi_flash_mmap.h"
 #include "esp_ota_ops.h"
 #include "esp_image_format.h"
-#ifdef UPDATE_CRYPT
+#ifndef UPDATE_NOCRYPT
 #include "mbedtls/aes.h"
-#endif /* UPDATE_CRYPT */
+#endif /* UPDATE_NOCRYPT */
 
 static const char *_err2str(uint8_t _error) {
   if (_error == UPDATE_ERROR_OK) {
@@ -40,10 +40,10 @@ static const char *_err2str(uint8_t _error) {
     return ("Bad Argument");
   } else if (_error == UPDATE_ERROR_ABORT) {
     return ("Aborted");
-#ifdef UPDATE_CRYPT
+#ifndef UPDATE_NOCRYPT
   } else if (_error == UPDATE_ERROR_DECRYPT) {
     return ("Decryption error");
-#endif /* UPDATE_CRYPT */
+#endif /* UPDATE_NOCRYPT */
   }
   return ("UNKNOWN");
 }
@@ -72,13 +72,13 @@ bool UpdateClass::_enablePartition(const esp_partition_t *partition) {
 
 UpdateClass::UpdateClass()
   : _error(0),
-#ifdef UPDATE_CRYPT
+#ifndef UPDATE_NOCRYPT
 _cryptKey(0), _cryptBuffer(0),
-#endif /* UPDATE_CRYPT */
+#endif /* UPDATE_NOCRYPT */
 _buffer(0), _skipBuffer(0), _bufferLen(0), _size(0), _progress_callback(NULL), _progress(0), _paroffset(0), _command(U_FLASH), _partition(NULL)
-#ifdef UPDATE_CRYPT
+#ifndef UPDATE_NOCRYPT
 , _cryptMode(U_AES_DECRYPT_AUTO), _cryptAddress(0), _cryptCfg(0xf)
-#endif /* UPDATE_CRYPT */
+#endif /* UPDATE_NOCRYPT */
 {}
 
 UpdateClass &UpdateClass::onProgress(THandlerFunction_Progress fn) {
@@ -94,9 +94,9 @@ void UpdateClass::_reset() {
     delete[] _skipBuffer;
   }
 
-#ifdef UPDATE_CRYPT
+#ifndef UPDATE_NOCRYPT
   _cryptBuffer = nullptr;
-#endif /* UPDATE_CRYPT */
+#endif /* UPDATE_NOCRYPT */
   _buffer = nullptr;
   _skipBuffer = nullptr;
   _bufferLen = 0;
@@ -188,7 +188,7 @@ bool UpdateClass::begin(size_t size, int command, int ledPin, uint8_t ledOn, con
   return true;
 }
 
-#ifdef UPDATE_CRYPT
+#ifndef UPDATE_NOCRYPT
 bool UpdateClass::setupCrypt(const uint8_t *cryptKey, size_t cryptAddress, uint8_t cryptConfig, int cryptMode) {
   if (setCryptKey(cryptKey)) {
     if (setCryptMode(cryptMode)) {
@@ -230,7 +230,7 @@ bool UpdateClass::setCryptMode(const int cryptMode) {
   }
   return true;
 }
-#endif /* UPDATE_CRYPT */
+#endif /* UPDATE_NOCRYPT */
 
 void UpdateClass::_abort(uint8_t err) {
   _reset();
@@ -241,7 +241,7 @@ void UpdateClass::abort() {
   _abort(UPDATE_ERROR_ABORT);
 }
 
-#ifdef UPDATE_CRYPT
+#ifndef UPDATE_NOCRYPT
 void UpdateClass::_cryptKeyTweak(size_t cryptAddress, uint8_t *tweaked_key) {
   memcpy(tweaked_key, _cryptKey, ENCRYPTED_KEY_SIZE);
   if (_cryptCfg == 0) {
@@ -354,10 +354,10 @@ bool UpdateClass::_decryptBuffer() {
   }
   return true;
 }
-#endif /* UPDATE_CRYPT */
+#endif /* UPDATE_NOCRYPT */
 
 bool UpdateClass::_writeBuffer() {
-#ifdef UPDATE_CRYPT
+#ifndef UPDATE_NOCRYPT
   //first bytes of loading image, check to see if loading image needs decrypting
   if (!_progress) {
     _cryptMode &= U_AES_DECRYPT_MODE_MASK;
@@ -378,7 +378,7 @@ bool UpdateClass::_writeBuffer() {
       return false;
     }
   }
-  #endif /* UPDATE_CRYPT */
+  #endif /* UPDATE_NOCRYPT */
   //first bytes of new firmware
   uint8_t skip = 0;
   if (!_progress && _command == U_FLASH) {
@@ -428,13 +428,13 @@ bool UpdateClass::_writeBuffer() {
   if (!_progress && _command == U_FLASH) {
     _buffer[0] = ESP_IMAGE_HEADER_MAGIC;
   }
-#ifdef UPDATE_CRYPT
+#ifndef UPDATE_NOCRYPT
   if (_target_md5_decrypted) {
-#endif /* UPDATE_CRYPT */
+#endif /* UPDATE_NOCRYPT */
     _md5.add(_buffer, _bufferLen);
-#ifdef UPDATE_CRYPT
+#ifndef UPDATE_NOCRYPT
   }
-#endif /* UPDATE_CRYPT */
+#endif /* UPDATE_NOCRYPT */
   _progress += _bufferLen;
   _bufferLen = 0;
   if (_progress_callback) {
@@ -477,18 +477,18 @@ bool UpdateClass::_verifyEnd() {
 }
 
 bool UpdateClass::setMD5(const char *expected_md5
-#ifdef UPDATE_CRYPT
+#ifndef UPDATE_NOCRYPT
 ,bool calc_post_decryption
-#endif /* UPDATE_CRYPT */
+#endif /* UPDATE_NOCRYPT */
 ) {
   if (strlen(expected_md5) != 32) {
     return false;
   }
   _target_md5 = expected_md5;
   _target_md5.toLowerCase();
-#ifdef UPDATE_CRYPT
+#ifndef UPDATE_NOCRYPT
   _target_md5_decrypted = calc_post_decryption;
-#endif /* UPDATE_CRYPT */
+#endif /* UPDATE_NOCRYPT */
   return true;
 }
 
@@ -561,16 +561,16 @@ size_t UpdateClass::writeStream(Stream &data) {
     return 0;
   }
 
-#ifdef UPDATE_CRYPT
+#ifndef UPDATE_NOCRYPT
   if (_command == U_FLASH && !_cryptMode) {
-#endif /* UPDATE_CRYPT */
+#endif /* UPDATE_NOCRYPT */
     if (!_verifyHeader(data.peek())) {
       _reset();
       return 0;
     }
-#ifdef UPDATE_CRYPT
+#ifndef UPDATE_NOCRYPT
   }
-#endif /* UPDATE_CRYPT */
+#endif /* UPDATE_NOCRYPT */
 
   if (_ledPin != -1) {
     pinMode(_ledPin, OUTPUT);