prevent zero size data

Author: rageworx

src/aes256/aes.c

@@ -461,6 +461,7 @@ void AES_CBC_encrypt_buffer(struct AES_ctx *ctx, uint8_t* buf, uint32_t length)
 {
   uintptr_t i;
   uint8_t *Iv = ctx->Iv;
+  
   for (i = 0; i < length; i += AES_BLOCKLEN)
   {
     XorWithIv(buf, Iv);
@@ -477,6 +478,7 @@ void AES_CBC_decrypt_buffer(struct AES_ctx* ctx, uint8_t* buf,  uint32_t length)
 {
   uintptr_t i;
   uint8_t storeNextIv[AES_BLOCKLEN];
+  
   for (i = 0; i < length; i += AES_BLOCKLEN)
   {
     memcpy(storeNextIv, buf, AES_BLOCKLEN);
@@ -497,9 +499,9 @@ void AES_CBC_decrypt_buffer(struct AES_ctx* ctx, uint8_t* buf,  uint32_t length)
 void AES_CTR_xcrypt_buffer(struct AES_ctx* ctx, uint8_t* buf, uint32_t length)
 {
   uint8_t buffer[AES_BLOCKLEN];
-  
   unsigned i;
   int bi;
+  
   for (i = 0, bi = AES_BLOCKLEN; i < length; ++i, ++bi)
   {
     if (bi == AES_BLOCKLEN) /* we need to regen xor compliment in buffer */

src/aes256/aes.h

@@ -2,7 +2,7 @@
 #define _AES_H_
 
 /*
-    -- variable key length each 16, 24, 32 AES variant source,
+    -- AES for variable key length each 16, 24, 32 ( 128, 192, 256 bit )
     -- (C)2021 Raphael K,
 */
 

src/endetool.cpp

@@ -75,7 +75,11 @@ void EnDeTool::reset()
         origintext = NULL;
     }
 
-    return;
+    AES_ctx* aesctx = (AES_ctx*)cryptcontext;
+    if ( aesctx != NULL )
+    {
+        memset( aesctx, 0, sizeof( AES_ctx ) );
+    }
 }
 
 void EnDeTool::compress( bool enabled )
@@ -85,7 +89,7 @@ void EnDeTool::compress( bool enabled )
 
 long long EnDeTool::encodebinary( const char* src, unsigned srcsize, char* &out )
 {
-    if ( src == NULL )
+    if ( ( src == NULL ) || ( srcsize == 0 ) )
         return -1;
 
     generateiv();