From b2d5f155713d1d55d125153b45fb059b4429f6b3 Mon Sep 17 00:00:00 2001
From: "Ira W. Snyder" <devel@irasnyder.com>
Date: Sun, 21 Oct 2007 16:33:44 -0700
Subject: [PATCH] Add new testing code

Add some of the examples from the FIPS document so that we can do
step-by-step testing.

Signed-off-by: Ira W. Snyder <devel@irasnyder.com>
---
 aes_test.cpp | 71 ++++++++++++++++++++++++++++++++++++++++++++++++++--
 1 file changed, 69 insertions(+), 2 deletions(-)

diff --git a/aes_test.cpp b/aes_test.cpp
index c3a0543..1c6559d 100644
--- a/aes_test.cpp
+++ b/aes_test.cpp
@@ -6,7 +6,7 @@ static void print_by_16s (const byteArray b, std::string name)
 {
 	int i, j;
 
-	std::cout << "Printing: " << name << "\n" << std::endl;
+	std::cout << "\nPrinting: " << name << std::endl;
 
 	for (i=0; i<b.size()/16; ++i)
 	{
@@ -21,10 +21,12 @@ static void print_by_16s (const byteArray b, std::string name)
 int main (int argc, char *argv[])
 {
 
-//#define DEBUG 1
+#define DEBUG 1
 //#define MY_TEST2 1
 //#define PROBLEM2 1
 //#define PROBLEM3 1
+//#define AES_128_EXAMPLE 1
+#define AES_128_EXAMPLE2 1
 
 #if PROBLEM2
 	int i;
@@ -139,6 +141,71 @@ int main (int argc, char *argv[])
 	print_by_16s (ciphertext, "Ciphertext");
 #endif // MY_TEST2
 
+#if AES_128_EXAMPLE
+	int i;
+	byteArray key (16, 0);
+	byteArray plaintext (16, 0);
+
+	for (i=0; i<16; ++i)
+		key.at (i) = i;
+
+	for (i=0; i<16; ++i)
+		plaintext.at (i) = (i) | (i<<4);
+
+	AES aes (key);
+	byteArray ciphertext = aes.encrypt (plaintext);
+
+	print_by_16s (key, "Key");
+	print_by_16s (plaintext, "Plaintext");
+	print_by_16s (ciphertext, "Ciphertext");
+#endif // AES_128_EXAMPLE
+
+#if AES_128_EXAMPLE2
+	int i;
+	byteArray key (16, 0);
+	byteArray plaintext (16, 0);
+
+	key[0] = 0x2b;
+	key[1] = 0x7e;
+	key[2] = 0x15;
+	key[3] = 0x16;
+	key[4] = 0x28;
+	key[5] = 0xae;
+	key[6] = 0xd2;
+	key[7] = 0xa6;
+	key[8] = 0xab;
+	key[9] = 0xf7;
+	key[10] = 0x15;
+	key[11] = 0x88;
+	key[12] = 0x09;
+	key[13] = 0xcf;
+	key[14] = 0x4f;
+	key[15] = 0x3c;
+
+	plaintext[0] = 0x32;
+	plaintext[1] = 0x43;
+	plaintext[2] = 0xf6;
+	plaintext[3] = 0xa8;
+	plaintext[4] = 0x88;
+	plaintext[5] = 0x5a;
+	plaintext[6] = 0x30;
+	plaintext[7] = 0x8d;
+	plaintext[8] = 0x31;
+	plaintext[9] = 0x31;
+	plaintext[10] = 0x98;
+	plaintext[11] = 0xa2;
+	plaintext[12] = 0xe0;
+	plaintext[13] = 0x37;
+	plaintext[14] = 0x07;
+	plaintext[15] = 0x34;
+
+	AES aes (key);
+	byteArray ciphertext = aes.encrypt (plaintext);
+
+	print_by_16s (key, "Key");
+	print_by_16s (plaintext, "Plaintext");
+	print_by_16s (ciphertext, "Ciphertext");
+#endif // AES_128_EXAMPLE
 	return 0;
 
 }
-- 
2.34.1