X-Git-Url: https://git.cryptolib.org/?a=blobdiff_plain;ds=sidebyside;f=nessie_bc_test.c;h=11deea73aadf0d18da56a5f811c0fe76190e6c13;hb=2aee09dcaf731b4086e7cdb865ee77b2715ce751;hp=b6256d4e31e6e77960f3477d8015bb51285d67a2;hpb=d4b9cfc34ce9f08df95df2933627e4e04337d36d;p=avr-crypto-lib.git

diff --git a/nessie_bc_test.c b/nessie_bc_test.c
index b6256d4..11deea7 100644
--- a/nessie_bc_test.c
+++ b/nessie_bc_test.c
@@ -15,6 +15,15 @@
 
 nessie_bc_ctx_t nessie_bc_ctx;
 
+void nessie_bc_init(void){
+	memset(&nessie_bc_ctx, 0, sizeof(nessie_bc_ctx_t));	
+}
+static
+void nessie_bc_free(void* ctx){
+	if(nessie_bc_ctx.cipher_free)
+		nessie_bc_ctx.cipher_free(ctx);
+}
+
 void nessie_bc_enc(uint8_t* key, uint8_t* pt){
 	uint8_t ctx[nessie_bc_ctx.ctx_size_B];
 	uint8_t buffer[nessie_bc_ctx.blocksize_B];
@@ -43,6 +52,7 @@ void nessie_bc_enc(uint8_t* key, uint8_t* pt){
 	}
 	nessie_print_item("Iterated 1000 times", buffer, nessie_bc_ctx.blocksize_B);
 #endif
+	nessie_bc_free(ctx);
 }
 
 void nessie_bc_dec(uint8_t* key, uint8_t* ct){
@@ -58,7 +68,7 @@ void nessie_bc_dec(uint8_t* key, uint8_t* ct){
 	nessie_print_item("plain", buffer, nessie_bc_ctx.blocksize_B);
 	nessie_bc_ctx.cipher_enc(buffer, ctx);
 	nessie_print_item("encrypted", buffer, nessie_bc_ctx.blocksize_B);
-	
+	nessie_bc_free(ctx);
 }
 
 void nessie_bc_run(void){