completed set 2 challenge 11

1 files changed, 3 insertions, 1 deletions

diff --git a/lib/lib2.h b/lib/lib2.h
index 7036388..a4b0174 100644
--- a/lib/lib2.h
+++ b/lib/lib2.h
@@ -7,7 +7,9 @@
 #include <openssl/aes.h>
 
 char *pkcs7_padding(char *string, int length_string, int blocksize);
-int aes_cbc(char *in, int length_in, char *out, unsigned char *string_key, char *init_vector);
+int aes_cbc(char *in, int length_in, char *out, unsigned char *string_key, char *init_vector, int encrypt);
 int valid_pkcs7_padding(const char *in, int length_in, char *unpadded, int blocksize);
+int aes_ecb(char *in, int length_in, char *out, unsigned char *string_key,
+		int blocksize, int encrypt);
 
 #endif