Fix leak of encryption buffer.

2019-08-02 18:59:42 -04:00
parent 360c0ade66
commit 57f77e4f8e
1 changed files with 3 additions and 1 deletions
--- a/platform-independent/c/core/src/mpw-util.c
+++ b/platform-independent/c/core/src/mpw-util.c
@@ -310,8 +310,10 @@ static uint8_t const *mpw_aes(bool encrypt, const uint8_t *key, const size_t key
    memcpy( aesBuf, buf, *bufSize );
    memset( aesBuf + *bufSize, aesSize - *bufSize, aesSize - *bufSize );
    uint8_t *resultBuf = malloc( aesSize );
-    if (!resultBuf)
+    if (!resultBuf) {
+        mpw_free( &aesBuf, aesSize );
        return NULL;
+    }

    if (encrypt)
        AES_CBC_encrypt_buffer( resultBuf, aesBuf, aesSize, key, iv );