1 files changed, 11 insertions, 3 deletions
diff --git a/examples/aes128ofb_example.c b/examples/aes128ofb_example.c
index 07137f6..7685008 100644
--- a/examples/aes128ofb_example.c
+++ b/examples/aes128ofb_example.c
@@ -12,7 +12,7 @@
 
 int main()
 {
-    __declspec(align(16)) AesBlock128 plain, key, cypher, decrypted, iv;
+    __declspec(align(16)) AesBlock128 plain, key, cypher, decrypted, iv, next_iv;
     __declspec(align(16)) Aes128KeySchedule key_schedule;
 
     plain = make_aes_block128(0xffeeddcc, 0xbbaa9988, 0x77665544, 0x33221100);
@@ -37,15 +37,23 @@ int main()
     for (int i = 0; i < 11; ++i)
         printf("\t[%d]: %s\n", i, format_aes_block128(&key_schedule.keys[i]).str);
 
-    cypher = aes128ofb_encrypt(plain, &key_schedule, iv);
+    cypher = aes128ofb_encrypt(plain, &key_schedule, iv, &next_iv);
     printf("\n");
     printf("Cypher: %s\n", format_aes_block128(&cypher).str);
     print_aes_block128_fips_matrix_style(&cypher);
 
-    decrypted = aes128ofb_decrypt(cypher, &key_schedule, iv);
+    printf("\n");
+    printf("Next initialization vector: %s\n", format_aes_block128(&next_iv).str);
+    print_aes_block128_fips_matrix_style(&next_iv);
+
+    decrypted = aes128ofb_decrypt(cypher, &key_schedule, iv, &next_iv);
     printf("\n");
     printf("Decrypted: %s\n", format_aes_block128(&decrypted).str);
     print_aes_block128_fips_matrix_style(&decrypted);
 
+    printf("\n");
+    printf("Next initialization vector: %s\n", format_aes_block128(&next_iv).str);
+    print_aes_block128_fips_matrix_style(&next_iv);
+
     return 0;
 }