lib: crypto: Implement AES message free functions

Signed-off-by: Ian Moffett <ian@osmora.org>

1 files changed, 23 insertions, 0 deletions

diff --git a/lib/crypto/aes.c b/lib/crypto/aes.c
index ea9fe98..c51c594 100644
--- a/lib/crypto/aes.c
+++ b/lib/crypto/aes.c
@@ -33,6 +33,7 @@
 #include <openssl/rand.h>
 #include <crypto/aes.h>
 #include <stdio.h>
+#include <string.h>
 
 /*
  * Encrypt 'data' with AES-256-GCM.
@@ -232,3 +233,25 @@ aes256_decrypt(struct aes_message *amp, const unsigned char *key,
     *res = plaintext;
     return 0;
 }
+
+/*
+ * Free data from AES message structure.
+ */
+int
+aes256_free_msg(struct aes_message *amp)
+{
+    free(amp->ciphertext);
+    return 0;
+}
+
+/*
+ * Free memory used for plaintext.
+ */
+int
+aes256_free_plain(unsigned char *plain, size_t len)
+{
+    memset(plain, 0, len);
+    munlock(plain, len);
+    free(plain);
+    return 0;
+}