c3a05430cfbc94ba5eaf0befceebd71989a1d985
[aes.git] / aes_test.cpp
1 #include "aes.hpp"
2 #include <iostream>
3 #include <fstream>
4
5 static void print_by_16s (const byteArray b, std::string name)
6 {
7         int i, j;
8
9         std::cout << "Printing: " << name << "\n" << std::endl;
10
11         for (i=0; i<b.size()/16; ++i)
12         {
13                 for (j=0; j<16; ++j)
14                         std::printf ("%.2x ", b.at(i*16+j));
15
16                 std::printf ("\n");
17         }
18 }
19
20
21 int main (int argc, char *argv[])
22 {
23
24 //#define DEBUG 1
25 //#define MY_TEST2 1
26 //#define PROBLEM2 1
27 //#define PROBLEM3 1
28
29 #if PROBLEM2
30         int i;
31
32         /* Read the Key */
33         byteArray key;
34
35         for (i=0; i<16; ++i)
36                 key.push_back (getchar());
37
38         /* Create the AES object */
39         AES aes (key);
40
41         /* Read all characters of input */
42         byteArray input;
43
44         for (char c=getchar(); c != EOF; c=getchar())
45         {
46                 input.push_back (c);
47
48                 if (input.size() == 16)
49                 {
50                         byteArray ciphertext = aes.encrypt (input);
51                         byteArray plaintext  = aes.decrypt (ciphertext);
52
53                         /* Print it */
54                         for (i=0; i<plaintext.size(); ++i)
55                                 std::printf ("%c", plaintext.at(i));
56
57                         input.clear();
58                 }
59
60         }
61 #endif // PROBLEM2
62
63 #if PROBLEM3
64         std::ifstream fin ("Problem3.out", ios::binary);
65         int i;
66
67         byteArray key;
68
69         for (i=0; i<24; ++i)
70                 key.push_back (fin.get());
71
72         /* Create the AES Object */
73         AES aes (key);
74
75         byteArray input;
76
77         /* Read the input and decrypt */
78         for (char c=fin.get(); !fin.eof(); c=fin.get())
79         {
80                 input.push_back (c);
81
82                 if (input.size() == 16)
83                 {
84                         byteArray plaintext = aes.decrypt (input);
85
86                         /* Print it */
87                         for (i=0; i<plaintext.size(); ++i)
88                                 std::printf ("%c", plaintext.at(i));
89
90                 input.clear();
91                 }
92         }
93 #endif //PROBLEM3
94
95 #if MY_TEST
96         const int inputSize = 4096;
97
98         byteArray key (16, 2);
99         byteArray input (inputSize, 0);
100         int i, j;
101
102         for (i=0; i<inputSize; ++i)
103                 input.at (i) = i;
104
105         AES aes (key);
106
107         print_by_16s (input, "Input");
108
109         byteArray ciphertext = aes.encrypt (input);
110
111         print_by_16s (ciphertext, "Ciphertext");
112
113         byteArray plaintext  = aes.decrypt (ciphertext);
114
115         print_by_16s (plaintext, "Plaintext");
116 #endif
117
118 #if MY_TEST2
119         int i;
120
121         byteArray key (16, 0);
122
123         for (i=0; i<16; ++i)
124                 key.at (i) = i;
125
126         print_by_16s (key, "Key");
127
128         byteArray plaintext (16, 0);
129
130         for (i=0; i<16; ++i)
131                 plaintext.at (i) = (i << 4) | i;
132
133         print_by_16s (plaintext, "Plaintext");
134
135         AES aes (key);
136
137         byteArray ciphertext = aes.encrypt (plaintext);
138
139         print_by_16s (ciphertext, "Ciphertext");
140 #endif // MY_TEST2
141
142         return 0;
143
144 }
145
146 /* vim: set ts=4 sts=4 sw=4 noet tw=120 nowrap: */