From e625f0aa58f341981ea8cba99660ae7c1eac1b05 Mon Sep 17 00:00:00 2001
From: Egor Tensin <Egor.Tensin@gmail.com>
Date: Tue, 16 Jun 2015 13:57:44 +0300
Subject: handle block formatting/printing functions errors

---
 examples/aes256ctr.c | 52 +++++++++++++++++++++++++++++++++++++++++-----------
 1 file changed, 41 insertions(+), 11 deletions(-)

(limited to 'examples/aes256ctr.c')

diff --git a/examples/aes256ctr.c b/examples/aes256ctr.c
index 05c5090..efb36e4 100644
--- a/examples/aes256ctr.c
+++ b/examples/aes256ctr.c
@@ -20,33 +20,63 @@ int main()
     key = aesni_make_block256(0x1f1e1d1c, 0x1b1a1918, 0x17161514, 0x13121110, 0x0f0e0d0c, 0x0b0a0908, 0x07060504, 0x03020100);
     iv = aesni_make_block128(0xfedcba98, 0x76543210, 0xfedcba98, 0x76543210);
 
-    printf("Plain: %s\n", aesni_format_block128(&plain).str);
-    aesni_print_block128_as_matrix(&plain);
+    {
+        AesNI_BlockString128 str;
+        aesni_format_block128(&str, &plain, NULL);
+        printf("Plain: %s\n", str.str);
+        aesni_print_block128_as_matrix(&plain, NULL);
+    }
 
     printf("\n");
-    printf("Key: %s\n", aesni_format_block256(&key).str);
-    aesni_print_block256_as_matrix(&key);
+
+    {
+        AesNI_BlockString256 str;
+        aesni_format_block256(&str, &key, NULL);
+        printf("Key: %s\n", str.str);
+        aesni_print_block256_as_matrix(&key, NULL);
+    }
 
     printf("\n");
-    printf("Initialization vector: %s\n", aesni_format_block128(&iv).str);
-    aesni_print_block128_as_matrix(&iv);
+
+    {
+        AesNI_BlockString128 str;
+        aesni_format_block128(&str, &iv, NULL);
+        printf("Initialization vector: %s\n", str.str);
+        aesni_print_block128_as_matrix(&iv, NULL);
+    }
 
     aesni_expand_key_schedule256(&key, &key_schedule);
 
     printf("\n");
     printf("Key schedule:\n");
     for (int i = 0; i < 15; ++i)
-        printf("\t[%d]: %s\n", i, aesni_format_block128(&key_schedule.keys[i]).str);
+    {
+        AesNI_BlockString128 str;
+        aesni_format_block128(&str, &key_schedule.keys[i], NULL);
+        printf("\t[%d]: %s\n", i, str.str);
+    }
 
     cipher = aesni_encrypt_block_ctr256(plain, &key_schedule, iv, 0);
+
     printf("\n");
-    printf("Cipher: %s\n", aesni_format_block128(&cipher).str);
-    aesni_print_block128_as_matrix(&cipher);
+
+    {
+        AesNI_BlockString128 str;
+        aesni_format_block128(&str, &cipher, NULL);
+        printf("Cipher: %s\n", str.str);
+        aesni_print_block128_as_matrix(&cipher, NULL);
+    }
 
     decrypted = aesni_decrypt_block_ctr256(cipher, &key_schedule, iv, 0);
+
     printf("\n");
-    printf("Decrypted: %s\n", aesni_format_block128(&decrypted).str);
-    aesni_print_block128_as_matrix(&decrypted);
+
+    {
+        AesNI_BlockString128 str;
+        aesni_format_block128(&str, &decrypted, NULL);
+        printf("Decrypted: %s\n", str.str);
+        aesni_print_block128_as_matrix(&decrypted, NULL);
+    }
 
     return 0;
 }
-- 
cgit v1.2.3