2 files changed, 49 insertions, 83 deletions
diff --git a/examples/aes256ecb.c b/examples/aes256ecb.c
deleted file mode 100644
index ddb90ca..0000000
--- a/examples/aes256ecb.c
+++ /dev/null
@@ -1,83 +0,0 @@
-/**
- * \file
- * \author Egor Tensin <Egor.Tensin@gmail.com>
- * \date 2015
- * \copyright This file is licensed under the terms of the MIT License.
- *            See LICENSE.txt for details.
- */
-
-#include <aesni/all.h>
-
-#include <stdio.h>
-
-int main()
-{
-    AesNI_Block128 plain, cipher, decrypted;
-    AesNI_Block256 key;
-    AesNI_KeySchedule256 key_schedule, inverted_schedule;
-
-    plain = aesni_make_block128(0xffeeddcc, 0xbbaa9988, 0x77665544, 0x33221100);
-    key = aesni_make_block256(0x1f1e1d1c, 0x1b1a1918, 0x17161514, 0x13121110, 0x0f0e0d0c, 0x0b0a0908, 0x07060504, 0x03020100);
-
-    {
-        AesNI_BlockString128 str;
-        aesni_format_block128(&str, &plain, NULL);
-        printf("Plain: %s\n", str.str);
-        aesni_print_block128_as_matrix(&plain, NULL);
-    }
-
-    printf("\n");
-
-    {
-        AesNI_BlockString256 str;
-        aesni_format_block256(&str, &key, NULL);
-        printf("Key: %s\n", str.str);
-        aesni_print_block256_as_matrix(&key, NULL);
-    }
-
-    aesni_expand_key_schedule256(&key, &key_schedule);
-
-    printf("\n");
-    printf("Key schedule:\n");
-    for (int i = 0; i < 15; ++i)
-    {
-        AesNI_BlockString128 str;
-        aesni_format_block128(&str, &key_schedule.keys[i], NULL);
-        printf("\t[%d]: %s\n", i, str.str);
-    }
-
-    cipher = aesni_encrypt_block_ecb256(plain, &key_schedule);
-
-    printf("\n");
-
-    {
-        AesNI_BlockString128 str;
-        aesni_format_block128(&str, &cipher, NULL);
-        printf("Cipher: %s\n", str.str);
-        aesni_print_block128_as_matrix(&cipher, NULL);
-    }
-
-    aesni_invert_key_schedule256(&key_schedule, &inverted_schedule);
-
-    printf("\n");
-    printf("Inverted key schedule:\n");
-    for (int i = 0; i < 15; ++i)
-    {
-        AesNI_BlockString128 str;
-        aesni_format_block128(&str, &inverted_schedule.keys[i], NULL);
-        printf("\t[%d]: %s\n", i, str.str);
-    }
-
-    decrypted = aesni_decrypt_block_ecb256(cipher, &inverted_schedule);
-
-    printf("\n");
-
-    {
-        AesNI_BlockString128 str;
-        aesni_format_block128(&str, &decrypted, NULL);
-        printf("Decrypted: %s\n", str.str);
-        aesni_print_block128_as_matrix(&decrypted, NULL);
-    }
-
-    return 0;
-}
diff --git a/examples/aes256ecb.cpp b/examples/aes256ecb.cpp
new file mode 100644
index 0000000..72f132e
--- /dev/null
+++ b/examples/aes256ecb.cpp
@@ -0,0 +1,49 @@
+/**
+ * \file
+ * \author Egor Tensin <Egor.Tensin@gmail.com>
+ * \date 2015
+ * \copyright This file is licensed under the terms of the MIT License.
+ *            See LICENSE.txt for details.
+ */
+
+#include "common.hpp"
+
+#include <aesni/all.h>
+
+#include <aesnixx/all.hpp>
+
+#include <exception>
+#include <iostream>
+
+int main()
+{
+    try
+    {
+        aesni::Block128 plaintext;
+        make_default_plaintext(plaintext);
+
+        aesni::Block256 key;
+        make_default_key(key);
+
+        aesni::KeySchedule256 encryption_schedule;
+        aesni_expand_key_schedule256(&key, &encryption_schedule);
+        dump_encryption_schedule(encryption_schedule);
+
+        const auto ciphertext = aesni_encrypt_block_ecb256(plaintext, &encryption_schedule);
+        dump_ciphertext(ciphertext);
+
+        aesni::KeySchedule256 decryption_schedule;
+        aesni_invert_key_schedule256(&encryption_schedule, &decryption_schedule);
+        dump_decryption_schedule(decryption_schedule);
+
+        const auto decrypted = aesni_decrypt_block_ecb256(ciphertext, &decryption_schedule);
+        dump_decrypted(decrypted);
+
+        return 0;
+    }
+    catch (const std::exception& e)
+    {
+        std::cerr << e.what() << "\n";
+        return 1;
+    }
+}