1/*
2 * Copyright (c) 2003 Kungliga Tekniska H�gskolan
3 * (Royal Institute of Technology, Stockholm, Sweden).
4 * All rights reserved.
5 *
6 * Redistribution and use in source and binary forms, with or without
7 * modification, are permitted provided that the following conditions
8 * are met:
9 *
10 * 1. Redistributions of source code must retain the above copyright
11 *    notice, this list of conditions and the following disclaimer.
12 *
13 * 2. Redistributions in binary form must reproduce the above copyright
14 *    notice, this list of conditions and the following disclaimer in the
15 *    documentation and/or other materials provided with the distribution.
16 *
17 * 3. Neither the name of KTH nor the names of its contributors may be
18 *    used to endorse or promote products derived from this software without
19 *    specific prior written permission.
20 *
21 * THIS SOFTWARE IS PROVIDED BY KTH AND ITS CONTRIBUTORS ``AS IS'' AND ANY
22 * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
23 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
24 * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL KTH OR ITS CONTRIBUTORS BE
25 * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
26 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
27 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR
28 * BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY,
29 * WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR
30 * OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF
31 * ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. */
32
33#include "krb5_locl.h"
34#include <hex.h>
35#include <err.h>
36
37#ifdef HAVE_OPENSSL
38#include <openssl/evp.h>
39#endif
40
41RCSID("$Id: aes-test.c 18301 2006-10-07 13:50:34Z lha $");
42
43static int verbose = 0;
44
45static void
46hex_dump_data(const void *data, size_t length)
47{
48    char *p;
49
50    hex_encode(data, length, &p);
51    printf("%s\n", p);
52    free(p);
53}
54
55struct {
56    char *password;
57    char *salt;
58    int saltlen;
59    int iterations;
60    krb5_enctype enctype;
61    size_t keylen;
62    char *pbkdf2;
63    char *key;
64} keys[] = {
65    {
66	"password", "ATHENA.MIT.EDUraeburn", -1,
67	1,
68	ETYPE_AES128_CTS_HMAC_SHA1_96, 16,
69	"\xcd\xed\xb5\x28\x1b\xb2\xf8\x01\x56\x5a\x11\x22\xb2\x56\x35\x15",
70	"\x42\x26\x3c\x6e\x89\xf4\xfc\x28\xb8\xdf\x68\xee\x09\x79\x9f\x15"
71    },
72    {
73	"password", "ATHENA.MIT.EDUraeburn", -1,
74	1,
75	ETYPE_AES256_CTS_HMAC_SHA1_96, 32,
76	"\xcd\xed\xb5\x28\x1b\xb2\xf8\x01\x56\x5a\x11\x22\xb2\x56\x35\x15"
77	"\x0a\xd1\xf7\xa0\x4b\xb9\xf3\xa3\x33\xec\xc0\xe2\xe1\xf7\x08\x37",
78	"\xfe\x69\x7b\x52\xbc\x0d\x3c\xe1\x44\x32\xba\x03\x6a\x92\xe6\x5b"
79	"\xbb\x52\x28\x09\x90\xa2\xfa\x27\x88\x39\x98\xd7\x2a\xf3\x01\x61"
80    },
81    {
82	"password", "ATHENA.MIT.EDUraeburn", -1,
83	2,
84	ETYPE_AES128_CTS_HMAC_SHA1_96, 16,
85	"\x01\xdb\xee\x7f\x4a\x9e\x24\x3e\x98\x8b\x62\xc7\x3c\xda\x93\x5d",
86	"\xc6\x51\xbf\x29\xe2\x30\x0a\xc2\x7f\xa4\x69\xd6\x93\xbd\xda\x13"
87    },
88    {
89	"password", "ATHENA.MIT.EDUraeburn", -1,
90	2,
91	ETYPE_AES256_CTS_HMAC_SHA1_96, 32,
92	"\x01\xdb\xee\x7f\x4a\x9e\x24\x3e\x98\x8b\x62\xc7\x3c\xda\x93\x5d"
93	"\xa0\x53\x78\xb9\x32\x44\xec\x8f\x48\xa9\x9e\x61\xad\x79\x9d\x86",
94	"\xa2\xe1\x6d\x16\xb3\x60\x69\xc1\x35\xd5\xe9\xd2\xe2\x5f\x89\x61"
95	"\x02\x68\x56\x18\xb9\x59\x14\xb4\x67\xc6\x76\x22\x22\x58\x24\xff"
96    },
97    {
98	"password", "ATHENA.MIT.EDUraeburn", -1,
99	1200,
100	ETYPE_AES128_CTS_HMAC_SHA1_96, 16,
101	"\x5c\x08\xeb\x61\xfd\xf7\x1e\x4e\x4e\xc3\xcf\x6b\xa1\xf5\x51\x2b",
102	"\x4c\x01\xcd\x46\xd6\x32\xd0\x1e\x6d\xbe\x23\x0a\x01\xed\x64\x2a"
103    },
104    {
105	"password", "ATHENA.MIT.EDUraeburn", -1,
106	1200,
107	ETYPE_AES256_CTS_HMAC_SHA1_96, 32,
108	"\x5c\x08\xeb\x61\xfd\xf7\x1e\x4e\x4e\xc3\xcf\x6b\xa1\xf5\x51\x2b"
109	"\xa7\xe5\x2d\xdb\xc5\xe5\x14\x2f\x70\x8a\x31\xe2\xe6\x2b\x1e\x13",
110	"\x55\xa6\xac\x74\x0a\xd1\x7b\x48\x46\x94\x10\x51\xe1\xe8\xb0\xa7"
111	"\x54\x8d\x93\xb0\xab\x30\xa8\xbc\x3f\xf1\x62\x80\x38\x2b\x8c\x2a"
112    },
113    {
114	"password", "\x12\x34\x56\x78\x78\x56\x34\x12", 8,
115	5,
116	ETYPE_AES128_CTS_HMAC_SHA1_96, 16,
117	"\xd1\xda\xa7\x86\x15\xf2\x87\xe6\xa1\xc8\xb1\x20\xd7\x06\x2a\x49",
118	"\xe9\xb2\x3d\x52\x27\x37\x47\xdd\x5c\x35\xcb\x55\xbe\x61\x9d\x8e"
119    },
120    {
121	"password", "\x12\x34\x56\x78\x78\x56\x34\x12", 8,
122	5,
123	ETYPE_AES256_CTS_HMAC_SHA1_96, 32,
124	"\xd1\xda\xa7\x86\x15\xf2\x87\xe6\xa1\xc8\xb1\x20\xd7\x06\x2a\x49"
125	"\x3f\x98\xd2\x03\xe6\xbe\x49\xa6\xad\xf4\xfa\x57\x4b\x6e\x64\xee",
126	"\x97\xa4\xe7\x86\xbe\x20\xd8\x1a\x38\x2d\x5e\xbc\x96\xd5\x90\x9c"
127	"\xab\xcd\xad\xc8\x7c\xa4\x8f\x57\x45\x04\x15\x9f\x16\xc3\x6e\x31"
128    },
129    {
130	"XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX",
131	"pass phrase equals block size", -1,
132	1200,
133	ETYPE_AES128_CTS_HMAC_SHA1_96, 16,
134	"\x13\x9c\x30\xc0\x96\x6b\xc3\x2b\xa5\x5f\xdb\xf2\x12\x53\x0a\xc9",
135	"\x59\xd1\xbb\x78\x9a\x82\x8b\x1a\xa5\x4e\xf9\xc2\x88\x3f\x69\xed"
136    },
137    {
138	"XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX",
139	"pass phrase equals block size", -1,
140	1200,
141	ETYPE_AES256_CTS_HMAC_SHA1_96, 32,
142	"\x13\x9c\x30\xc0\x96\x6b\xc3\x2b\xa5\x5f\xdb\xf2\x12\x53\x0a\xc9"
143	"\xc5\xec\x59\xf1\xa4\x52\xf5\xcc\x9a\xd9\x40\xfe\xa0\x59\x8e\xd1",
144	"\x89\xad\xee\x36\x08\xdb\x8b\xc7\x1f\x1b\xfb\xfe\x45\x94\x86\xb0"
145	"\x56\x18\xb7\x0c\xba\xe2\x20\x92\x53\x4e\x56\xc5\x53\xba\x4b\x34"
146    },
147    {
148	"XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX",
149	"pass phrase exceeds block size", -1,
150	1200,
151	ETYPE_AES128_CTS_HMAC_SHA1_96, 16,
152	"\x9c\xca\xd6\xd4\x68\x77\x0c\xd5\x1b\x10\xe6\xa6\x87\x21\xbe\x61",
153	"\xcb\x80\x05\xdc\x5f\x90\x17\x9a\x7f\x02\x10\x4c\x00\x18\x75\x1d"
154    },
155    {
156	"XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX",
157	"pass phrase exceeds block size", -1,
158	1200,
159	ETYPE_AES256_CTS_HMAC_SHA1_96, 32,
160	"\x9c\xca\xd6\xd4\x68\x77\x0c\xd5\x1b\x10\xe6\xa6\x87\x21\xbe\x61"
161	"\x1a\x8b\x4d\x28\x26\x01\xdb\x3b\x36\xbe\x92\x46\x91\x5e\xc8\x2a",
162	"\xd7\x8c\x5c\x9c\xb8\x72\xa8\xc9\xda\xd4\x69\x7f\x0b\xb5\xb2\xd2"
163	"\x14\x96\xc8\x2b\xeb\x2c\xae\xda\x21\x12\xfc\xee\xa0\x57\x40\x1b"
164
165    },
166    {
167	"\xf0\x9d\x84\x9e" /* g-clef */, "EXAMPLE.COMpianist", -1,
168	50,
169	ETYPE_AES128_CTS_HMAC_SHA1_96, 16,
170	"\x6b\x9c\xf2\x6d\x45\x45\x5a\x43\xa5\xb8\xbb\x27\x6a\x40\x3b\x39",
171	"\xf1\x49\xc1\xf2\xe1\x54\xa7\x34\x52\xd4\x3e\x7f\xe6\x2a\x56\xe5"
172    },
173    {
174	"\xf0\x9d\x84\x9e" /* g-clef */, "EXAMPLE.COMpianist", -1,
175	50,
176	ETYPE_AES256_CTS_HMAC_SHA1_96, 32,
177	"\x6b\x9c\xf2\x6d\x45\x45\x5a\x43\xa5\xb8\xbb\x27\x6a\x40\x3b\x39"
178	"\xe7\xfe\x37\xa0\xc4\x1e\x02\xc2\x81\xff\x30\x69\xe1\xe9\x4f\x52",
179	"\x4b\x6d\x98\x39\xf8\x44\x06\xdf\x1f\x09\xcc\x16\x6d\xb4\xb8\x3c"
180	"\x57\x18\x48\xb7\x84\xa3\xd6\xbd\xc3\x46\x58\x9a\x3e\x39\x3f\x9e"
181    },
182    {
183	"foo", "", -1,
184	0,
185	ETYPE_ARCFOUR_HMAC_MD5, 16,
186	NULL,
187	"\xac\x8e\x65\x7f\x83\xdf\x82\xbe\xea\x5d\x43\xbd\xaf\x78\x00\xcc"
188    },
189    {
190	"test", "", -1,
191	0,
192	ETYPE_ARCFOUR_HMAC_MD5, 16,
193	NULL,
194	"\x0c\xb6\x94\x88\x05\xf7\x97\xbf\x2a\x82\x80\x79\x73\xb8\x95\x37"
195    }
196};
197
198static int
199string_to_key_test(krb5_context context)
200{
201    krb5_data password, opaque;
202    krb5_error_code ret;
203    krb5_salt salt;
204    int i, val = 0;
205    char iter[4];
206
207    for (i = 0; i < sizeof(keys)/sizeof(keys[0]); i++) {
208
209	password.data = keys[i].password;
210	password.length = strlen(password.data);
211
212	salt.salttype = KRB5_PW_SALT;
213	salt.saltvalue.data = keys[i].salt;
214	if (keys[i].saltlen == -1)
215	    salt.saltvalue.length = strlen(salt.saltvalue.data);
216	else
217	    salt.saltvalue.length = keys[i].saltlen;
218
219	opaque.data = iter;
220	opaque.length = sizeof(iter);
221	_krb5_put_int(iter, keys[i].iterations, 4);
222
223	if (keys[i].pbkdf2) {
224	    unsigned char keyout[32];
225
226	    if (keys[i].keylen > sizeof(keyout))
227		abort();
228
229	    PKCS5_PBKDF2_HMAC_SHA1(password.data, password.length,
230				   salt.saltvalue.data, salt.saltvalue.length,
231				   keys[i].iterations,
232				   keys[i].keylen, keyout);
233
234	    if (memcmp(keyout, keys[i].pbkdf2, keys[i].keylen) != 0) {
235		krb5_warnx(context, "%d: pbkdf2", i);
236		val = 1;
237		continue;
238	    }
239
240	    if (verbose) {
241		printf("PBKDF2:\n");
242		hex_dump_data(keyout, keys[i].keylen);
243	    }
244	}
245
246	{
247	    krb5_keyblock key;
248
249	    ret = krb5_string_to_key_data_salt_opaque (context,
250						       keys[i].enctype,
251						       password,
252						       salt,
253						       opaque,
254						       &key);
255	    if (ret) {
256		krb5_warn(context, ret, "%d: string_to_key_data_salt_opaque",
257			  i);
258		val = 1;
259		continue;
260	    }
261
262	    if (key.keyvalue.length != keys[i].keylen) {
263		krb5_warnx(context, "%d: key wrong length (%lu/%lu)",
264			   i, (unsigned long)key.keyvalue.length,
265			   (unsigned long)keys[i].keylen);
266		val = 1;
267		continue;
268	    }
269
270	    if (memcmp(key.keyvalue.data, keys[i].key, keys[i].keylen) != 0) {
271		krb5_warnx(context, "%d: key wrong", i);
272		val = 1;
273		continue;
274	    }
275
276	    if (verbose) {
277		printf("key:\n");
278		hex_dump_data(key.keyvalue.data, key.keyvalue.length);
279	    }
280	    krb5_free_keyblock_contents(context, &key);
281	}
282    }
283    return val;
284}
285
286struct enc_test {
287    size_t len;
288    char *input;
289    char *output;
290    char *nextiv;
291};
292
293struct enc_test encs1[] = {
294    {
295	17,
296	"\x49\x20\x77\x6f\x75\x6c\x64\x20\x6c\x69\x6b\x65\x20\x74\x68\x65"
297	"\x20",
298	"\xc6\x35\x35\x68\xf2\xbf\x8c\xb4\xd8\xa5\x80\x36\x2d\xa7\xff\x7f"
299	"\x97",
300	"\xc6\x35\x35\x68\xf2\xbf\x8c\xb4\xd8\xa5\x80\x36\x2d\xa7\xff\x7f"
301    },
302    {
303	31,
304	"\x49\x20\x77\x6f\x75\x6c\x64\x20\x6c\x69\x6b\x65\x20\x74\x68\x65"
305	"\x20\x47\x65\x6e\x65\x72\x61\x6c\x20\x47\x61\x75\x27\x73\x20",
306	"\xfc\x00\x78\x3e\x0e\xfd\xb2\xc1\xd4\x45\xd4\xc8\xef\xf7\xed\x22"
307	"\x97\x68\x72\x68\xd6\xec\xcc\xc0\xc0\x7b\x25\xe2\x5e\xcf\xe5",
308	"\xfc\x00\x78\x3e\x0e\xfd\xb2\xc1\xd4\x45\xd4\xc8\xef\xf7\xed\x22"
309    },
310    {
311	32,
312	"\x49\x20\x77\x6f\x75\x6c\x64\x20\x6c\x69\x6b\x65\x20\x74\x68\x65"
313	"\x20\x47\x65\x6e\x65\x72\x61\x6c\x20\x47\x61\x75\x27\x73\x20\x43",
314	"\x39\x31\x25\x23\xa7\x86\x62\xd5\xbe\x7f\xcb\xcc\x98\xeb\xf5\xa8"
315	"\x97\x68\x72\x68\xd6\xec\xcc\xc0\xc0\x7b\x25\xe2\x5e\xcf\xe5\x84",
316	"\x39\x31\x25\x23\xa7\x86\x62\xd5\xbe\x7f\xcb\xcc\x98\xeb\xf5\xa8"
317    },
318    {
319	47,
320	"\x49\x20\x77\x6f\x75\x6c\x64\x20\x6c\x69\x6b\x65\x20\x74\x68\x65"
321	"\x20\x47\x65\x6e\x65\x72\x61\x6c\x20\x47\x61\x75\x27\x73\x20\x43"
322	"\x68\x69\x63\x6b\x65\x6e\x2c\x20\x70\x6c\x65\x61\x73\x65\x2c",
323	"\x97\x68\x72\x68\xd6\xec\xcc\xc0\xc0\x7b\x25\xe2\x5e\xcf\xe5\x84"
324	"\xb3\xff\xfd\x94\x0c\x16\xa1\x8c\x1b\x55\x49\xd2\xf8\x38\x02\x9e"
325	"\x39\x31\x25\x23\xa7\x86\x62\xd5\xbe\x7f\xcb\xcc\x98\xeb\xf5",
326	"\xb3\xff\xfd\x94\x0c\x16\xa1\x8c\x1b\x55\x49\xd2\xf8\x38\x02\x9e"
327    },
328    {
329	48,
330	"\x49\x20\x77\x6f\x75\x6c\x64\x20\x6c\x69\x6b\x65\x20\x74\x68\x65"
331	"\x20\x47\x65\x6e\x65\x72\x61\x6c\x20\x47\x61\x75\x27\x73\x20\x43"
332	"\x68\x69\x63\x6b\x65\x6e\x2c\x20\x70\x6c\x65\x61\x73\x65\x2c\x20",
333	"\x97\x68\x72\x68\xd6\xec\xcc\xc0\xc0\x7b\x25\xe2\x5e\xcf\xe5\x84"
334	"\x9d\xad\x8b\xbb\x96\xc4\xcd\xc0\x3b\xc1\x03\xe1\xa1\x94\xbb\xd8"
335	"\x39\x31\x25\x23\xa7\x86\x62\xd5\xbe\x7f\xcb\xcc\x98\xeb\xf5\xa8",
336	"\x9d\xad\x8b\xbb\x96\xc4\xcd\xc0\x3b\xc1\x03\xe1\xa1\x94\xbb\xd8"
337    },
338    {
339	64,
340	"\x49\x20\x77\x6f\x75\x6c\x64\x20\x6c\x69\x6b\x65\x20\x74\x68\x65"
341	"\x20\x47\x65\x6e\x65\x72\x61\x6c\x20\x47\x61\x75\x27\x73\x20\x43"
342	"\x68\x69\x63\x6b\x65\x6e\x2c\x20\x70\x6c\x65\x61\x73\x65\x2c\x20"
343	"\x61\x6e\x64\x20\x77\x6f\x6e\x74\x6f\x6e\x20\x73\x6f\x75\x70\x2e",
344	"\x97\x68\x72\x68\xd6\xec\xcc\xc0\xc0\x7b\x25\xe2\x5e\xcf\xe5\x84"
345	"\x39\x31\x25\x23\xa7\x86\x62\xd5\xbe\x7f\xcb\xcc\x98\xeb\xf5\xa8"
346	"\x48\x07\xef\xe8\x36\xee\x89\xa5\x26\x73\x0d\xbc\x2f\x7b\xc8\x40"
347	"\x9d\xad\x8b\xbb\x96\xc4\xcd\xc0\x3b\xc1\x03\xe1\xa1\x94\xbb\xd8",
348	"\x48\x07\xef\xe8\x36\xee\x89\xa5\x26\x73\x0d\xbc\x2f\x7b\xc8\x40"
349    }
350};
351
352
353struct enc_test encs2[] = {
354    {
355	17,
356	"\x49\x20\x77\x6f\x75\x6c\x64\x20\x6c\x69\x6b\x65\x20\x74\x68\x65"
357	"\x20",
358	"\x5c\x13\x26\x27\xc4\xcb\xca\x04\x14\x43\x8a\xb5\x97\x97\x7c\x10"
359	"\x16"
360    },
361    {
362	31,
363	"\x49\x20\x77\x6f\x75\x6c\x64\x20\x6c\x69\x6b\x65\x20\x74\x68\x65"
364	"\x20\x47\x65\x6e\x65\x72\x61\x6c\x20\x47\x61\x75\x27\x73\x20",
365	"\x16\xb3\xd8\xe5\xcd\x93\xe6\x2c\x28\x70\xa0\x36\x6e\x9a\xb9\x74"
366	"\x16\xc1\xee\xdf\x39\xc8\x3f\xfb\xc5\xf6\x72\xe9\xc1\x6e\x53"
367    },
368    {
369	32,
370	"\x49\x20\x77\x6f\x75\x6c\x64\x20\x6c\x69\x6b\x65\x20\x74\x68\x65"
371	"\x20\x47\x65\x6e\x65\x72\x61\x6c\x20\x47\x61\x75\x27\x73\x20\x43",
372	"\x69\xde\xce\x59\x83\x6a\x82\xe1\xcd\x21\x93\xd0\x9e\x2a\xff\xc8"
373	"\x16\xc1\xee\xdf\x39\xc8\x3f\xfb\xc5\xf6\x72\xe9\xc1\x6e\x53\x0c"
374    },
375    {
376	47,
377	"\x49\x20\x77\x6f\x75\x6c\x64\x20\x6c\x69\x6b\x65\x20\x74\x68\x65"
378	"\x20\x47\x65\x6e\x65\x72\x61\x6c\x20\x47\x61\x75\x27\x73\x20\x43"
379	"\x68\x69\x63\x6b\x65\x6e\x2c\x20\x70\x6c\x65\x61\x73\x65\x2c",
380	"\x16\xc1\xee\xdf\x39\xc8\x3f\xfb\xc5\xf6\x72\xe9\xc1\x6e\x53\x0c"
381	"\xe5\x56\xb4\x88\x41\xb9\xde\x27\xf0\x07\xa1\x6e\x89\x94\x47\xf1"
382	"\x69\xde\xce\x59\x83\x6a\x82\xe1\xcd\x21\x93\xd0\x9e\x2a\xff"
383    },
384    {
385	48,
386	"\x49\x20\x77\x6f\x75\x6c\x64\x20\x6c\x69\x6b\x65\x20\x74\x68\x65"
387	"\x20\x47\x65\x6e\x65\x72\x61\x6c\x20\x47\x61\x75\x27\x73\x20\x43"
388	"\x68\x69\x63\x6b\x65\x6e\x2c\x20\x70\x6c\x65\x61\x73\x65\x2c\x20",
389	"\x16\xc1\xee\xdf\x39\xc8\x3f\xfb\xc5\xf6\x72\xe9\xc1\x6e\x53\x0c"
390	"\xfd\x68\xd1\x56\x32\x23\x7b\xfa\xb0\x09\x86\x3b\x17\x53\xfa\x30"
391	"\x69\xde\xce\x59\x83\x6a\x82\xe1\xcd\x21\x93\xd0\x9e\x2a\xff\xc8"
392    },
393    {
394	64,
395	"\x49\x20\x77\x6f\x75\x6c\x64\x20\x6c\x69\x6b\x65\x20\x74\x68\x65"
396	"\x20\x47\x65\x6e\x65\x72\x61\x6c\x20\x47\x61\x75\x27\x73\x20\x43"
397	"\x68\x69\x63\x6b\x65\x6e\x2c\x20\x70\x6c\x65\x61\x73\x65\x2c\x20"
398	"\x61\x6e\x64\x20\x77\x6f\x6e\x74\x6f\x6e\x20\x73\x6f\x75\x70\x2e",
399	"\x16\xc1\xee\xdf\x39\xc8\x3f\xfb\xc5\xf6\x72\xe9\xc1\x6e\x53\x0c"
400	"\x69\xde\xce\x59\x83\x6a\x82\xe1\xcd\x21\x93\xd0\x9e\x2a\xff\xc8"
401	"\x70\x29\xf2\x6f\x7c\x79\xc1\x77\x91\xad\x94\xb0\x78\x62\x27\x67"
402	"\xfd\x68\xd1\x56\x32\x23\x7b\xfa\xb0\x09\x86\x3b\x17\x53\xfa\x30"
403    },
404    {
405	78,
406	"\x49\x20\x77\x6f\x75\x6c\x64\x20\x6c\x69\x6b\x65\x20\x74\x68\x65"
407	"\x20\x47\x65\x6e\x65\x72\x61\x6c\x20\x47\x61\x75\x27\x73\x20\x43"
408	"\x68\x69\x63\x6b\x65\x6e\x2c\x20\x70\x6c\x65\x61\x73\x65\x2c\x20"
409	"\x61\x6e\x64\x20\x77\x6f\x6e\x74\x6f\x6e\x20\x73\x6f\x75\x70\x2e"
410	"\x41\x41\x41\x41\x41\x41\x41\x41\x41\x41\x41\x41\x41\x41",
411	"\x16\xc1\xee\xdf\x39\xc8\x3f\xfb\xc5\xf6\x72\xe9\xc1\x6e\x53\x0c"
412	"\x69\xde\xce\x59\x83\x6a\x82\xe1\xcd\x21\x93\xd0\x9e\x2a\xff\xc8"
413	"\xfd\x68\xd1\x56\x32\x23\x7b\xfa\xb0\x09\x86\x3b\x17\x53\xfa\x30"
414	"\x73\xfb\x2c\x36\x76\xaf\xcf\x31\xff\xe3\x8a\x89\x0c\x7e\x99\x3f"
415	"\x70\x29\xf2\x6f\x7c\x79\xc1\x77\x91\xad\x94\xb0\x78\x62"
416    },
417    {
418	83,
419	"\x49\x20\x77\x6f\x75\x6c\x64\x20\x6c\x69\x6b\x65\x20\x74\x68\x65"
420	"\x20\x47\x65\x6e\x65\x72\x61\x6c\x20\x47\x61\x75\x27\x73\x20\x43"
421	"\x68\x69\x63\x6b\x65\x6e\x2c\x20\x70\x6c\x65\x61\x73\x65\x2c\x20"
422	"\x61\x6e\x64\x20\x77\x6f\x6e\x74\x6f\x6e\x20\x73\x6f\x75\x70\x2e"
423	"\x41\x41\x41\x41\x41\x41\x41\x41\x41\x41\x41\x41\x41\x41\x41\x41"
424	"\x41\x41\x41",
425	"\x16\xc1\xee\xdf\x39\xc8\x3f\xfb\xc5\xf6\x72\xe9\xc1\x6e\x53\x0c"
426	"\x69\xde\xce\x59\x83\x6a\x82\xe1\xcd\x21\x93\xd0\x9e\x2a\xff\xc8"
427	"\xfd\x68\xd1\x56\x32\x23\x7b\xfa\xb0\x09\x86\x3b\x17\x53\xfa\x30"
428	"\x70\x29\xf2\x6f\x7c\x79\xc1\x77\x91\xad\x94\xb0\x78\x62\x27\x67"
429	"\x65\x39\x3a\xdb\x92\x05\x4d\x4f\x08\xa1\xfa\x59\xda\x56\x58\x0e"
430	"\x3b\xac\x12"
431    },
432    {
433	92,
434	"\x49\x20\x77\x6f\x75\x6c\x64\x20\x6c\x69\x6b\x65\x20\x74\x68\x65"
435	"\x20\x47\x65\x6e\x65\x72\x61\x6c\x20\x47\x61\x75\x27\x73\x20\x43"
436	"\x68\x69\x63\x6b\x65\x6e\x2c\x20\x70\x6c\x65\x61\x73\x65\x2c\x20"
437	"\x61\x6e\x64\x20\x77\x6f\x6e\x74\x6f\x6e\x20\x73\x6f\x75\x70\x2e"
438	"\x41\x41\x41\x41\x41\x41\x41\x41\x41\x41\x41\x41\x41\x41\x41\x41"
439	"\x41\x41\x41\x41\x41\x41\x41\x41\x41\x41\x41\x41",
440	"\x16\xc1\xee\xdf\x39\xc8\x3f\xfb\xc5\xf6\x72\xe9\xc1\x6e\x53\x0c"
441	"\x69\xde\xce\x59\x83\x6a\x82\xe1\xcd\x21\x93\xd0\x9e\x2a\xff\xc8"
442	"\xfd\x68\xd1\x56\x32\x23\x7b\xfa\xb0\x09\x86\x3b\x17\x53\xfa\x30"
443	"\x70\x29\xf2\x6f\x7c\x79\xc1\x77\x91\xad\x94\xb0\x78\x62\x27\x67"
444	"\x0c\xff\xd7\x63\x50\xf8\x4e\xf9\xec\x56\x1c\x79\xc5\xc8\xfe\x50"
445	"\x3b\xac\x12\x6e\xd3\x2d\x02\xc4\xe5\x06\x43\x5f"
446    },
447    {
448	96,
449	"\x49\x20\x77\x6f\x75\x6c\x64\x20\x6c\x69\x6b\x65\x20\x74\x68\x65"
450	"\x20\x47\x65\x6e\x65\x72\x61\x6c\x20\x47\x61\x75\x27\x73\x20\x43"
451	"\x68\x69\x63\x6b\x65\x6e\x2c\x20\x70\x6c\x65\x61\x73\x65\x2c\x20"
452	"\x61\x6e\x64\x20\x77\x6f\x6e\x74\x6f\x6e\x20\x73\x6f\x75\x70\x2e"
453	"\x41\x41\x41\x41\x41\x41\x41\x41\x41\x41\x41\x41\x41\x41\x41\x41"
454	"\x41\x41\x41\x41\x41\x41\x41\x41\x41\x41\x41\x41\x41\x41\x41\x41",
455	"\x16\xc1\xee\xdf\x39\xc8\x3f\xfb\xc5\xf6\x72\xe9\xc1\x6e\x53\x0c"
456	"\x69\xde\xce\x59\x83\x6a\x82\xe1\xcd\x21\x93\xd0\x9e\x2a\xff\xc8"
457	"\xfd\x68\xd1\x56\x32\x23\x7b\xfa\xb0\x09\x86\x3b\x17\x53\xfa\x30"
458	"\x70\x29\xf2\x6f\x7c\x79\xc1\x77\x91\xad\x94\xb0\x78\x62\x27\x67"
459	"\x08\x28\x49\xad\xfc\x2d\x8e\x86\xae\x69\xa5\xa8\xd9\x29\x9e\xe4"
460	"\x3b\xac\x12\x6e\xd3\x2d\x02\xc4\xe5\x06\x43\x5f\x4c\x41\xd1\xb8"
461    }
462};
463
464
465
466char *aes_key1 =
467	"\x63\x68\x69\x63\x6b\x65\x6e\x20\x74\x65\x72\x69\x79\x61\x6b\x69";
468
469char *aes_key2 =
470	"\x63\x68\x69\x63\x6b\x65\x6e\x20\x74\x65\x72\x69\x79\x61\x6b\x69"
471	"\x2c\x20\x79\x75\x6d\x6d\x79\x20\x79\x75\x6d\x6d\x79\x21\x21\x21";
472
473
474static int
475samep(int testn, char *type, const void *pp1, const void *pp2, size_t len)
476{
477    const unsigned char *p1 = pp1, *p2 = pp2;
478    size_t i;
479    int val = 1;
480
481    for (i = 0; i < len; i++) {
482	if (p1[i] != p2[i]) {
483	    if (verbose)
484		printf("M");
485	    val = 0;
486	} else {
487	    if (verbose)
488		printf(".");
489	}
490    }
491    if (verbose)
492	printf("\n");
493    return val;
494}
495
496static int
497encryption_test(krb5_context context, const void *key, size_t keylen,
498		struct enc_test *enc, int numenc)
499{
500    unsigned char iv[AES_BLOCK_SIZE];
501    int i, val, failed = 0;
502    AES_KEY ekey, dkey;
503    unsigned char *p;
504
505    AES_set_encrypt_key(key, keylen, &ekey);
506    AES_set_decrypt_key(key, keylen, &dkey);
507
508    for (i = 0; i < numenc; i++) {
509	val = 0;
510
511	if (verbose)
512	    printf("test: %d\n", i);
513	memset(iv, 0, sizeof(iv));
514
515	p = malloc(enc[i].len + 1);
516	if (p == NULL)
517	    krb5_errx(context, 1, "malloc");
518
519	p[enc[i].len] = '\0';
520
521	memcpy(p, enc[i].input, enc[i].len);
522
523	_krb5_aes_cts_encrypt(p, p, enc[i].len,
524			      &ekey, iv, AES_ENCRYPT);
525
526	if (p[enc[i].len] != '\0') {
527	    krb5_warnx(context, "%d: encrypt modified off end", i);
528	    val = 1;
529	}
530
531	if (!samep(i, "cipher", p, enc[i].output, enc[i].len)) {
532	    krb5_warnx(context, "%d: cipher", i);
533	    val = 1;
534	}
535
536	if (enc[i].nextiv && !samep(i, "iv", iv, enc[i].nextiv, 16)){ /*XXX*/
537	    krb5_warnx(context, "%d: iv", i);
538	    val = 1;
539	}
540
541	memset(iv, 0, sizeof(iv));
542
543	_krb5_aes_cts_encrypt(p, p, enc[i].len,
544			      &dkey, iv, AES_DECRYPT);
545
546	if (p[enc[i].len] != '\0') {
547	    krb5_warnx(context, "%d: decrypt modified off end", i);
548	    val = 1;
549	}
550
551	if (!samep(i, "clear", p, enc[i].input, enc[i].len))
552	    val = 1;
553
554	if (enc[i].nextiv && !samep(i, "iv", iv, enc[i].nextiv, 16)){ /*XXX*/
555	    krb5_warnx(context, "%d: iv", i);
556	    val = 1;
557	}
558
559	free(p);
560
561	if (val) {
562	    printf("test %d failed\n", i);
563	    failed = 1;
564	}
565	val = 0;
566    }
567    return failed;
568}
569
570static int
571krb_enc(krb5_context context,
572	krb5_crypto crypto,
573	unsigned usage,
574	krb5_data *cipher,
575	krb5_data *clear)
576{
577    krb5_data decrypt;
578    krb5_error_code ret;
579
580    krb5_data_zero(&decrypt);
581
582    ret = krb5_decrypt(context,
583		       crypto,
584		       usage,
585		       cipher->data,
586		       cipher->length,
587		       &decrypt);
588
589    if (ret) {
590	krb5_warn(context, ret, "krb5_decrypt");
591	return ret;
592    }
593
594    if (decrypt.length != clear->length ||
595	memcmp(decrypt.data, clear->data, decrypt.length) != 0) {
596	krb5_warnx(context, "clear text not same");
597	return EINVAL;
598    }
599
600    krb5_data_free(&decrypt);
601
602    return 0;
603}
604
605static int
606krb_enc_mit(krb5_context context,
607	    krb5_enctype enctype,
608	    krb5_keyblock *key,
609	    unsigned usage,
610	    krb5_data *cipher,
611	    krb5_data *clear)
612{
613    krb5_error_code ret;
614    krb5_enc_data e;
615    krb5_data decrypt;
616    size_t len;
617
618    e.kvno = 0;
619    e.enctype = enctype;
620    e.ciphertext = *cipher;
621
622    ret = krb5_c_decrypt(context, *key, usage, NULL, &e, &decrypt);
623    if (ret)
624	return ret;
625
626    if (decrypt.length != clear->length ||
627	memcmp(decrypt.data, clear->data, decrypt.length) != 0) {
628	krb5_warnx(context, "clear text not same");
629	return EINVAL;
630    }
631
632    krb5_data_free(&decrypt);
633
634    ret = krb5_c_encrypt_length(context, enctype, clear->length, &len);
635    if (ret)
636	return ret;
637
638    if (len != cipher->length) {
639	krb5_warnx(context, "c_encrypt_length wrong %lu != %lu",
640		   (unsigned long)len, (unsigned long)cipher->length);
641	return EINVAL;
642    }
643
644    return 0;
645}
646
647
648struct {
649    krb5_enctype enctype;
650    unsigned usage;
651    size_t keylen;
652    void *key;
653    size_t elen;
654    void* edata;
655    size_t plen;
656    void *pdata;
657} krbencs[] =  {
658    {
659	ETYPE_AES256_CTS_HMAC_SHA1_96,
660	7,
661	32,
662	"\x47\x75\x69\x64\x65\x6c\x69\x6e\x65\x73\x20\x74\x6f\x20\x41\x75"
663	"\x74\x68\x6f\x72\x73\x20\x6f\x66\x20\x49\x6e\x74\x65\x72\x6e\x65",
664	44,
665	"\xcf\x79\x8f\x0d\x76\xf3\xe0\xbe\x8e\x66\x94\x70\xfa\xcc\x9e\x91"
666	"\xa9\xec\x1c\x5c\x21\xfb\x6e\xef\x1a\x7a\xc8\xc1\xcc\x5a\x95\x24"
667	"\x6f\x9f\xf4\xd5\xbe\x5d\x59\x97\x44\xd8\x47\xcd",
668	16,
669	"\x54\x68\x69\x73\x20\x69\x73\x20\x61\x20\x74\x65\x73\x74\x2e\x0a"
670    }
671};
672
673
674static int
675krb_enc_test(krb5_context context)
676{
677    krb5_error_code ret;
678    krb5_crypto crypto;
679    krb5_keyblock kb;
680    krb5_data cipher, plain;
681    int i, failed = 0;
682
683    for (i = 0; i < sizeof(krbencs)/sizeof(krbencs[0]); i++) {
684
685	kb.keytype = krbencs[i].enctype;
686	kb.keyvalue.length = krbencs[i].keylen;
687	kb.keyvalue.data = krbencs[i].key;
688
689	ret = krb5_crypto_init(context, &kb, krbencs[i].enctype, &crypto);
690
691	cipher.length = krbencs[i].elen;
692	cipher.data = krbencs[i].edata;
693	plain.length = krbencs[i].plen;
694	plain.data = krbencs[i].pdata;
695
696	ret = krb_enc(context, crypto, krbencs[i].usage, &cipher, &plain);
697
698	if (ret) {
699	    failed = 1;
700	    printf("krb_enc failed with %d\n", ret);
701	}
702	krb5_crypto_destroy(context, crypto);
703
704	ret = krb_enc_mit(context, krbencs[i].enctype, &kb,
705			  krbencs[i].usage, &cipher, &plain);
706	if (ret) {
707	    failed = 1;
708	    printf("krb_enc_mit failed with %d\n", ret);
709	}
710
711    }
712
713    return failed;
714}
715
716
717static int
718random_to_key(krb5_context context)
719{
720    krb5_error_code ret;
721    krb5_keyblock key;
722
723    ret = krb5_random_to_key(context,
724			     ETYPE_DES3_CBC_SHA1,
725			     "\x21\x39\x04\x58\x6A\xBD\x7F"
726			     "\x21\x39\x04\x58\x6A\xBD\x7F"
727			     "\x21\x39\x04\x58\x6A\xBD\x7F",
728			     21,
729			     &key);
730    if (ret){
731	krb5_warn(context, ret, "random_to_key");
732	return 1;
733    }
734    if (key.keyvalue.length != 24)
735	return 1;
736
737    if (memcmp(key.keyvalue.data,
738	       "\x20\x38\x04\x58\x6b\xbc\x7f\xc7"
739	       "\x20\x38\x04\x58\x6b\xbc\x7f\xc7"
740	       "\x20\x38\x04\x58\x6b\xbc\x7f\xc7",
741	       24) != 0)
742	return 1;
743
744    krb5_free_keyblock_contents(context, &key);
745
746    return 0;
747}
748
749
750int
751main(int argc, char **argv)
752{
753    krb5_error_code ret;
754    krb5_context context;
755    int val = 0;
756
757    ret = krb5_init_context (&context);
758    if (ret)
759	errx (1, "krb5_init_context failed: %d", ret);
760
761    val |= string_to_key_test(context);
762
763    val |= encryption_test(context, aes_key1, 128,
764			   encs1, sizeof(encs1)/sizeof(encs1[0]));
765    val |= encryption_test(context, aes_key2, 256,
766			   encs2, sizeof(encs2)/sizeof(encs2[0]));
767    val |= krb_enc_test(context);
768    val |= random_to_key(context);
769
770    if (verbose && val == 0)
771	printf("all ok\n");
772    if (val)
773	printf("tests failed\n");
774
775    krb5_free_context(context);
776
777    return val;
778}
779