1 files changed, 23 insertions, 2 deletions
diff --git a/examples/aes128ecb_example.c b/examples/aes128ecb_example.c
index afb8dd9..aff4182 100644
--- a/examples/aes128ecb_example.c
+++ b/examples/aes128ecb_example.c
@@ -13,6 +13,7 @@
 int main()
 {
     __declspec(align(16)) AesBlock128 plain, key, cypher, decrypted;
+    __declspec(align(16)) Aes128KeySchedule key_schedule, inverted_schedule;
 
     plain = make_aes_block128(0xffeeddcc, 0xbbaa9988, 0x77665544, 0x33221100);
     key   = make_aes_block128(0x0f0e0d0c, 0x0b0a0908, 0x07060504, 0x03020100);
@@ -26,13 +27,33 @@ int main()
     printf("     %s\n", format_aes_block128_fips_style(&key).str);
     print_aes_block128_fips_matrix_style(&key);
 
-    cypher = aes128ecb_encrypt(plain, key);
+    aes128_expand_key_schedule(key, &key_schedule);
+
+    printf("\n");
+    printf("Key schedule:\n");
+    for (int i = 0; i < 11; ++i)
+    {
+        printf("\t[%d]: %s\n", i, format_aes_block128(&key_schedule.keys[i]).str);
+        printf("\t[%d]: %s\n", i, format_aes_block128_fips_style(&key_schedule.keys[i]).str);
+    }
+
+    cypher = aes128ecb_encrypt(plain, &key_schedule);
     printf("\n");
     printf("Cypher: %s\n", format_aes_block128(&cypher).str);
     printf("        %s\n", format_aes_block128_fips_style(&cypher).str);
     print_aes_block128_fips_matrix_style(&cypher);
 
-    decrypted = aes128ecb_decrypt(cypher, key);
+    aes128_invert_key_schedule(&key_schedule, &inverted_schedule);
+
+    printf("\n");
+    printf("Inverted key schedule:\n");
+    for (int i = 0; i < 11; ++i)
+    {
+        printf("\t[%d]: %s\n", i, format_aes_block128(&inverted_schedule.keys[i]).str);
+        printf("\t[%d]: %s\n", i, format_aes_block128_fips_style(&inverted_schedule.keys[i]).str);
+    }
+
+    decrypted = aes128ecb_decrypt(cypher, &inverted_schedule);
     printf("\n");
     printf("Decrypted: %s\n", format_aes_block128(&decrypted).str);
     printf("           %s\n", format_aes_block128_fips_style(&decrypted).str);