Add macros for AES block size

common/aes.h

@@ -8,6 +8,7 @@
 #define AES192_KEYBYTES 24
 #define AES256_KEYBYTES 32
 #define AESCTR_NONCEBYTES 12
+#define AES_BLOCKBYTES 16
 
 typedef struct {
     uint64_t sk_exp[88];

test/common/aes.c

@@ -78,19 +78,19 @@ int main(void)
   }
 
 
-  aes128_ecb(ct, msg, 3, &ctx128);
+  aes128_ecb(ct, msg, sizeof(msg) / AES_BLOCKBYTES, &ctx128);
   if(memcmp(ct, ct128, 48)) {
     printf("ERROR AES128ECB output does not match test vector.\n");
     r = 1;
   }
 
-  aes192_ecb(ct, msg, 3, &ctx192);
+  aes192_ecb(ct, msg, sizeof(msg) / AES_BLOCKBYTES, &ctx192);
   if(memcmp(ct, ct192, 48)) {
     printf("ERROR AES192ECB output does not match test vector.\n");
     r = 1;
   }
 
-  aes256_ecb(ct, msg, 3, &ctx256);
+  aes256_ecb(ct, msg, sizeof(msg) / AES_BLOCKBYTES, &ctx256);
   if(memcmp(ct, ct256, 48)) {
     printf("ERROR AES256ECB output does not match test vector.\n");
     r = 1;