/ lib / krb5 / test_enctypes.c
test_enctypes.c
  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  #include <assert.h>
 37  
 38  #ifdef HAVE_OPENSSL
 39  #include <openssl/evp.h>
 40  #endif
 41  
 42  static int verbose = 0;
 43  
 44  static void
 45  hex_dump_data(const void *data, size_t length)
 46  {
 47      char *p;
 48  
 49      hex_encode(data, length, &p);
 50      printf("%s\n", p);
 51      free(p);
 52  }
 53  
 54  struct {
 55      char *password;
 56      char *salt;
 57      int saltlen;
 58      int iterations;
 59      krb5_enctype enctype;
 60      size_t keylen;
 61      char *pbkdf2;
 62      char *key;
 63  } keys[] = {
 64      {
 65  	"password", "ATHENA.MIT.EDUraeburn", -1,
 66  	1,
 67  	ETYPE_AES128_CTS_HMAC_SHA1_96, 16,
 68  	"\xcd\xed\xb5\x28\x1b\xb2\xf8\x01\x56\x5a\x11\x22\xb2\x56\x35\x15",
 69  	"\x42\x26\x3c\x6e\x89\xf4\xfc\x28\xb8\xdf\x68\xee\x09\x79\x9f\x15"
 70      },
 71      {
 72  	"password", "ATHENA.MIT.EDUraeburn", -1,
 73  	1,
 74  	ETYPE_AES256_CTS_HMAC_SHA1_96, 32,
 75  	"\xcd\xed\xb5\x28\x1b\xb2\xf8\x01\x56\x5a\x11\x22\xb2\x56\x35\x15"
 76  	"\x0a\xd1\xf7\xa0\x4b\xb9\xf3\xa3\x33\xec\xc0\xe2\xe1\xf7\x08\x37",
 77  	"\xfe\x69\x7b\x52\xbc\x0d\x3c\xe1\x44\x32\xba\x03\x6a\x92\xe6\x5b"
 78  	"\xbb\x52\x28\x09\x90\xa2\xfa\x27\x88\x39\x98\xd7\x2a\xf3\x01\x61"
 79      },
 80      {
 81  	"password", "ATHENA.MIT.EDUraeburn", -1,
 82  	2,
 83  	ETYPE_AES128_CTS_HMAC_SHA1_96, 16,
 84  	"\x01\xdb\xee\x7f\x4a\x9e\x24\x3e\x98\x8b\x62\xc7\x3c\xda\x93\x5d",
 85  	"\xc6\x51\xbf\x29\xe2\x30\x0a\xc2\x7f\xa4\x69\xd6\x93\xbd\xda\x13"
 86      },
 87      {
 88  	"password", "ATHENA.MIT.EDUraeburn", -1,
 89  	2,
 90  	ETYPE_AES256_CTS_HMAC_SHA1_96, 32,
 91  	"\x01\xdb\xee\x7f\x4a\x9e\x24\x3e\x98\x8b\x62\xc7\x3c\xda\x93\x5d"
 92  	"\xa0\x53\x78\xb9\x32\x44\xec\x8f\x48\xa9\x9e\x61\xad\x79\x9d\x86",
 93  	"\xa2\xe1\x6d\x16\xb3\x60\x69\xc1\x35\xd5\xe9\xd2\xe2\x5f\x89\x61"
 94  	"\x02\x68\x56\x18\xb9\x59\x14\xb4\x67\xc6\x76\x22\x22\x58\x24\xff"
 95      },
 96      {
 97  	"password", "ATHENA.MIT.EDUraeburn", -1,
 98  	1200,
 99  	ETYPE_AES128_CTS_HMAC_SHA1_96, 16,
100  	"\x5c\x08\xeb\x61\xfd\xf7\x1e\x4e\x4e\xc3\xcf\x6b\xa1\xf5\x51\x2b",
101  	"\x4c\x01\xcd\x46\xd6\x32\xd0\x1e\x6d\xbe\x23\x0a\x01\xed\x64\x2a"
102      },
103      {
104  	"password", "ATHENA.MIT.EDUraeburn", -1,
105  	1200,
106  	ETYPE_AES256_CTS_HMAC_SHA1_96, 32,
107  	"\x5c\x08\xeb\x61\xfd\xf7\x1e\x4e\x4e\xc3\xcf\x6b\xa1\xf5\x51\x2b"
108  	"\xa7\xe5\x2d\xdb\xc5\xe5\x14\x2f\x70\x8a\x31\xe2\xe6\x2b\x1e\x13",
109  	"\x55\xa6\xac\x74\x0a\xd1\x7b\x48\x46\x94\x10\x51\xe1\xe8\xb0\xa7"
110  	"\x54\x8d\x93\xb0\xab\x30\xa8\xbc\x3f\xf1\x62\x80\x38\x2b\x8c\x2a"
111      },
112      {
113  	"password", "\x12\x34\x56\x78\x78\x56\x34\x12", 8,
114  	5,
115  	ETYPE_AES128_CTS_HMAC_SHA1_96, 16,
116  	"\xd1\xda\xa7\x86\x15\xf2\x87\xe6\xa1\xc8\xb1\x20\xd7\x06\x2a\x49",
117  	"\xe9\xb2\x3d\x52\x27\x37\x47\xdd\x5c\x35\xcb\x55\xbe\x61\x9d\x8e"
118      },
119      {
120  	"password", "\x12\x34\x56\x78\x78\x56\x34\x12", 8,
121  	5,
122  	ETYPE_AES256_CTS_HMAC_SHA1_96, 32,
123  	"\xd1\xda\xa7\x86\x15\xf2\x87\xe6\xa1\xc8\xb1\x20\xd7\x06\x2a\x49"
124  	"\x3f\x98\xd2\x03\xe6\xbe\x49\xa6\xad\xf4\xfa\x57\x4b\x6e\x64\xee",
125  	"\x97\xa4\xe7\x86\xbe\x20\xd8\x1a\x38\x2d\x5e\xbc\x96\xd5\x90\x9c"
126  	"\xab\xcd\xad\xc8\x7c\xa4\x8f\x57\x45\x04\x15\x9f\x16\xc3\x6e\x31"
127      },
128      {
129  	"XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX",
130  	"pass phrase equals block size", -1,
131  	1200,
132  	ETYPE_AES128_CTS_HMAC_SHA1_96, 16,
133  	"\x13\x9c\x30\xc0\x96\x6b\xc3\x2b\xa5\x5f\xdb\xf2\x12\x53\x0a\xc9",
134  	"\x59\xd1\xbb\x78\x9a\x82\x8b\x1a\xa5\x4e\xf9\xc2\x88\x3f\x69\xed"
135      },
136      {
137  	"XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX",
138  	"pass phrase equals block size", -1,
139  	1200,
140  	ETYPE_AES256_CTS_HMAC_SHA1_96, 32,
141  	"\x13\x9c\x30\xc0\x96\x6b\xc3\x2b\xa5\x5f\xdb\xf2\x12\x53\x0a\xc9"
142  	"\xc5\xec\x59\xf1\xa4\x52\xf5\xcc\x9a\xd9\x40\xfe\xa0\x59\x8e\xd1",
143  	"\x89\xad\xee\x36\x08\xdb\x8b\xc7\x1f\x1b\xfb\xfe\x45\x94\x86\xb0"
144  	"\x56\x18\xb7\x0c\xba\xe2\x20\x92\x53\x4e\x56\xc5\x53\xba\x4b\x34"
145      },
146      {
147  	"XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX",
148  	"pass phrase exceeds block size", -1,
149  	1200,
150  	ETYPE_AES128_CTS_HMAC_SHA1_96, 16,
151  	"\x9c\xca\xd6\xd4\x68\x77\x0c\xd5\x1b\x10\xe6\xa6\x87\x21\xbe\x61",
152  	"\xcb\x80\x05\xdc\x5f\x90\x17\x9a\x7f\x02\x10\x4c\x00\x18\x75\x1d"
153      },
154      {
155  	"XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX",
156  	"pass phrase exceeds block size", -1,
157  	1200,
158  	ETYPE_AES256_CTS_HMAC_SHA1_96, 32,
159  	"\x9c\xca\xd6\xd4\x68\x77\x0c\xd5\x1b\x10\xe6\xa6\x87\x21\xbe\x61"
160  	"\x1a\x8b\x4d\x28\x26\x01\xdb\x3b\x36\xbe\x92\x46\x91\x5e\xc8\x2a",
161  	"\xd7\x8c\x5c\x9c\xb8\x72\xa8\xc9\xda\xd4\x69\x7f\x0b\xb5\xb2\xd2"
162  	"\x14\x96\xc8\x2b\xeb\x2c\xae\xda\x21\x12\xfc\xee\xa0\x57\x40\x1b"
163  
164      },
165      {
166  	"\xf0\x9d\x84\x9e" /* g-clef */, "EXAMPLE.COMpianist", -1,
167  	50,
168  	ETYPE_AES128_CTS_HMAC_SHA1_96, 16,
169  	"\x6b\x9c\xf2\x6d\x45\x45\x5a\x43\xa5\xb8\xbb\x27\x6a\x40\x3b\x39",
170  	"\xf1\x49\xc1\xf2\xe1\x54\xa7\x34\x52\xd4\x3e\x7f\xe6\x2a\x56\xe5"
171      },
172      {
173  	"\xf0\x9d\x84\x9e" /* g-clef */, "EXAMPLE.COMpianist", -1,
174  	50,
175  	ETYPE_AES256_CTS_HMAC_SHA1_96, 32,
176  	"\x6b\x9c\xf2\x6d\x45\x45\x5a\x43\xa5\xb8\xbb\x27\x6a\x40\x3b\x39"
177  	"\xe7\xfe\x37\xa0\xc4\x1e\x02\xc2\x81\xff\x30\x69\xe1\xe9\x4f\x52",
178  	"\x4b\x6d\x98\x39\xf8\x44\x06\xdf\x1f\x09\xcc\x16\x6d\xb4\xb8\x3c"
179  	"\x57\x18\x48\xb7\x84\xa3\xd6\xbd\xc3\x46\x58\x9a\x3e\x39\x3f\x9e"
180      },
181      {
182  	"foo", "", -1,
183  	0,
184  	ETYPE_ARCFOUR_HMAC_MD5, 16,
185  	NULL,
186  	"\xac\x8e\x65\x7f\x83\xdf\x82\xbe\xea\x5d\x43\xbd\xaf\x78\x00\xcc"
187      },
188      {
189  	"test", "", -1,
190  	0,
191  	ETYPE_ARCFOUR_HMAC_MD5, 16,
192  	NULL,
193  	"\x0c\xb6\x94\x88\x05\xf7\x97\xbf\x2a\x82\x80\x79\x73\xb8\x95\x37"
194      }
195  };
196  
197  static int
198  string_to_key_test(krb5_context context)
199  {
200      krb5_data password, opaque;
201      krb5_error_code ret;
202      krb5_salt salt;
203      int val = 0;
204      char iter[4];
205      size_t i;
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", (int)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  			  (int)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  			   (int)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", (int)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  
286  static int
287  krb_enc(krb5_context context,
288  	krb5_crypto crypto,
289  	unsigned usage,
290  	krb5_data *cipher,
291  	krb5_data *clear)
292  {
293      krb5_data decrypt;
294      krb5_error_code ret;
295  
296      krb5_data_zero(&decrypt);
297  
298      ret = krb5_decrypt(context,
299  		       crypto,
300  		       usage,
301  		       cipher->data,
302  		       cipher->length,
303  		       &decrypt);
304  
305      if (ret) {
306  	krb5_warn(context, ret, "krb5_decrypt");
307  	return ret;
308      }
309  
310      if (decrypt.length != clear->length ||
311  	memcmp(decrypt.data, clear->data, decrypt.length) != 0) {
312  	krb5_warnx(context, "clear text not same");
313  	return EINVAL;
314      }
315  
316      krb5_data_free(&decrypt);
317  
318      return 0;
319  }
320  
321  static int
322  krb_enc_iov2(krb5_context context,
323  	     krb5_crypto crypto,
324  	     unsigned usage,
325  	     size_t cipher_len,
326  	     krb5_data *clear)
327  {
328      krb5_crypto_iov iov[4];
329      krb5_data decrypt;
330      int ret;
331      char *p, *q;
332      size_t len, i;
333  
334      p = clear->data;
335      len = clear->length;
336  
337      iov[0].flags = KRB5_CRYPTO_TYPE_HEADER;
338      krb5_crypto_length(context, crypto, iov[0].flags, &iov[0].data.length);
339      iov[0].data.data = emalloc(iov[0].data.length);
340  
341      iov[1].flags = KRB5_CRYPTO_TYPE_DATA;
342      iov[1].data.length = len;
343      iov[1].data.data = emalloc(iov[1].data.length);
344      memcpy(iov[1].data.data, p, iov[1].data.length);
345  
346      /* padding buffer */
347      iov[2].flags = KRB5_CRYPTO_TYPE_PADDING;
348      krb5_crypto_length(context, crypto, KRB5_CRYPTO_TYPE_PADDING, &iov[2].data.length);
349      iov[2].data.data = emalloc(iov[2].data.length);
350  
351      iov[3].flags = KRB5_CRYPTO_TYPE_TRAILER;
352      krb5_crypto_length(context, crypto, iov[3].flags, &iov[3].data.length);
353      iov[3].data.data = emalloc(iov[3].data.length);
354  
355      ret = krb5_encrypt_iov_ivec(context, crypto, usage,
356  				iov, sizeof(iov)/sizeof(iov[0]), NULL);
357      if (ret)
358  	errx(1, "encrypt iov failed: %d", ret);
359  
360      /* check len */
361      for (i = 0, len = 0; i < sizeof(iov)/sizeof(iov[0]); i++)
362  	len += iov[i].data.length;
363      if (len != cipher_len)
364  	errx(1, "cipher len wrong");
365  
366      /*
367       * Plain decrypt
368       */
369  
370      p = q = emalloc(len);
371      for (i = 0; i < sizeof(iov)/sizeof(iov[0]); i++) {
372  	memcpy(q, iov[i].data.data, iov[i].data.length);
373  	q += iov[i].data.length;
374      }
375  
376      ret = krb5_decrypt(context, crypto, usage, p, len, &decrypt);
377      if (ret)
378  	krb5_err(context, 1, ret, "krb5_decrypt");
379      else
380  	krb5_data_free(&decrypt);
381  
382      free(p);
383  
384      /*
385       * Now decrypt use iov
386       */
387  
388      /* padding turn into data */
389      p = q = emalloc(iov[1].data.length + iov[2].data.length);
390  
391      memcpy(q, iov[1].data.data, iov[1].data.length);
392      q += iov[1].data.length;
393      memcpy(q, iov[2].data.data, iov[2].data.length);
394  
395      free(iov[1].data.data);
396      free(iov[2].data.data);
397  
398      iov[1].data.data = p;
399      iov[1].data.length += iov[2].data.length;
400  
401      iov[2].flags = KRB5_CRYPTO_TYPE_EMPTY;
402      iov[2].data.length = 0;
403  
404      ret = krb5_decrypt_iov_ivec(context, crypto, usage,
405  				iov, sizeof(iov)/sizeof(iov[0]), NULL);
406      free(iov[0].data.data);
407      free(iov[3].data.data);
408  
409      if (ret)
410  	krb5_err(context, 1, ret, "decrypt iov failed: %d", ret);
411  
412      if (clear->length != iov[1].data.length)
413  	errx(1, "length incorrect");
414  
415      p = clear->data;
416      if (memcmp(iov[1].data.data, p, iov[1].data.length) != 0)
417  	errx(1, "iov[1] incorrect");
418  
419      free(iov[1].data.data);
420  
421      return 0;
422  }
423  
424  
425  static int
426  krb_enc_iov(krb5_context context,
427  	    krb5_crypto crypto,
428  	    unsigned usage,
429  	    krb5_data *cipher,
430  	    krb5_data *clear)
431  {
432      krb5_crypto_iov iov[3];
433      int ret;
434      char *p;
435      size_t len;
436  
437      p = cipher->data;
438      len = cipher->length;
439  
440      iov[0].flags = KRB5_CRYPTO_TYPE_HEADER;
441      krb5_crypto_length(context, crypto, iov[0].flags, &iov[0].data.length);
442      iov[0].data.data = emalloc(iov[0].data.length);
443      memcpy(iov[0].data.data, p, iov[0].data.length);
444      p += iov[0].data.length;
445      len -= iov[0].data.length;
446  
447      iov[1].flags = KRB5_CRYPTO_TYPE_TRAILER;
448      krb5_crypto_length(context, crypto, iov[1].flags, &iov[1].data.length);
449      iov[1].data.data = emalloc(iov[1].data.length);
450      memcpy(iov[1].data.data, p + len - iov[1].data.length, iov[1].data.length);
451      len -= iov[1].data.length;
452  
453      iov[2].flags = KRB5_CRYPTO_TYPE_DATA;
454      iov[2].data.length = len;
455      iov[2].data.data = emalloc(len);
456      memcpy(iov[2].data.data, p, len);
457  
458      ret = krb5_decrypt_iov_ivec(context, crypto, usage,
459  				iov, sizeof(iov)/sizeof(iov[0]), NULL);
460      if (ret)
461  	krb5_err(context, 1, ret, "krb_enc_iov decrypt iov failed: %d", ret);
462  
463      if (clear->length != iov[2].data.length)
464  	errx(1, "length incorrect");
465  
466      p = clear->data;
467      if (memcmp(iov[2].data.data, p, iov[2].data.length) != 0)
468  	errx(1, "iov[2] incorrect");
469  
470      free(iov[0].data.data);
471      free(iov[1].data.data);
472      free(iov[2].data.data);
473  
474  
475      return 0;
476  }
477  
478  static int
479  krb_checksum_iov(krb5_context context,
480  		 krb5_crypto crypto,
481  		 unsigned usage,
482  		 krb5_data *plain)
483  {
484      krb5_crypto_iov iov[4];
485      int ret;
486      char *p;
487      size_t len;
488  
489      p = plain->data;
490      len = plain->length;
491  
492      iov[0].flags = KRB5_CRYPTO_TYPE_CHECKSUM;
493      krb5_crypto_length(context, crypto, iov[0].flags, &iov[0].data.length);
494      iov[0].data.data = emalloc(iov[0].data.length);
495  
496      iov[1].flags = KRB5_CRYPTO_TYPE_DATA;
497      iov[1].data.length = len;
498      iov[1].data.data = p;
499  
500      iov[2].flags = KRB5_CRYPTO_TYPE_TRAILER;
501      krb5_crypto_length(context, crypto, iov[0].flags, &iov[2].data.length);
502      iov[2].data.data = malloc(iov[2].data.length);
503  
504      ret = krb5_create_checksum_iov(context, crypto, usage,
505  				   iov, sizeof(iov)/sizeof(iov[0]), NULL);
506      if (ret)
507  	krb5_err(context, 1, ret, "krb5_create_checksum_iov failed");
508  
509      ret = krb5_verify_checksum_iov(context, crypto, usage, iov, sizeof(iov)/sizeof(iov[0]), NULL);
510      if (ret)
511  	krb5_err(context, 1, ret, "krb5_verify_checksum_iov");
512  
513      free(iov[0].data.data);
514      free(iov[2].data.data);
515  
516      return 0;
517  }
518  
519  
520  static int
521  krb_enc_mit(krb5_context context,
522  	    krb5_enctype enctype,
523  	    krb5_keyblock *key,
524  	    unsigned usage,
525  	    krb5_data *cipher,
526  	    krb5_data *clear)
527  {
528  #ifndef HEIMDAL_SMALLER
529      krb5_error_code ret;
530      krb5_enc_data e;
531      krb5_data decrypt;
532      size_t len;
533  
534      e.kvno = 0;
535      e.enctype = enctype;
536      e.ciphertext = *cipher;
537  
538      ret = krb5_c_decrypt(context, *key, usage, NULL, &e, &decrypt);
539      if (ret)
540  	return ret;
541  
542      if (decrypt.length != clear->length ||
543  	memcmp(decrypt.data, clear->data, decrypt.length) != 0) {
544  	krb5_warnx(context, "clear text not same");
545  	return EINVAL;
546      }
547  
548      krb5_data_free(&decrypt);
549  
550      ret = krb5_c_encrypt_length(context, enctype, clear->length, &len);
551      if (ret)
552  	return ret;
553  
554      if (len != cipher->length) {
555  	krb5_warnx(context, "c_encrypt_length wrong %lu != %lu",
556  		   (unsigned long)len, (unsigned long)cipher->length);
557  	return EINVAL;
558      }
559  #endif /* HEIMDAL_SMALLER */
560      return 0;
561  }
562  
563  
564  struct {
565      krb5_enctype enctype;
566      unsigned usage;
567      size_t keylen;
568      void *key;
569      size_t elen;
570      void* edata;
571      size_t plen;
572      void *pdata;
573  } krbencs[] =  {
574      {
575  	ETYPE_AES256_CTS_HMAC_SHA1_96,
576  	7,
577  	32,
578  	"\x47\x75\x69\x64\x65\x6c\x69\x6e\x65\x73\x20\x74\x6f\x20\x41\x75"
579  	"\x74\x68\x6f\x72\x73\x20\x6f\x66\x20\x49\x6e\x74\x65\x72\x6e\x65",
580  	44,
581  	"\xcf\x79\x8f\x0d\x76\xf3\xe0\xbe\x8e\x66\x94\x70\xfa\xcc\x9e\x91"
582  	"\xa9\xec\x1c\x5c\x21\xfb\x6e\xef\x1a\x7a\xc8\xc1\xcc\x5a\x95\x24"
583  	"\x6f\x9f\xf4\xd5\xbe\x5d\x59\x97\x44\xd8\x47\xcd",
584  	16,
585  	"\x54\x68\x69\x73\x20\x69\x73\x20\x61\x20\x74\x65\x73\x74\x2e\x0a"
586      }
587  };
588  
589  
590  static int
591  krb_enc_test(krb5_context context)
592  {
593      krb5_error_code ret;
594      krb5_crypto crypto;
595      krb5_keyblock kb;
596      krb5_data cipher, plain;
597      size_t i;
598  
599      for (i = 0; i < sizeof(krbencs)/sizeof(krbencs[0]); i++) {
600  
601  	kb.keytype = krbencs[i].enctype;
602  	kb.keyvalue.length = krbencs[i].keylen;
603  	kb.keyvalue.data = krbencs[i].key;
604  
605  	ret = krb5_crypto_init(context, &kb, krbencs[i].enctype, &crypto);
606  	if (ret)
607  	    errx(1, "krb5_crypto_init failed with %d for test %d", ret, (int)i);
608  
609  	cipher.length = krbencs[i].elen;
610  	cipher.data = krbencs[i].edata;
611  	plain.length = krbencs[i].plen;
612  	plain.data = krbencs[i].pdata;
613  
614  	ret = krb_enc(context, crypto, krbencs[i].usage, &cipher, &plain);
615  
616  	if (ret)
617  	    errx(1, "krb_enc failed with %d for test %d", ret, (int)i);
618  
619  	ret = krb_enc_iov(context, crypto, krbencs[i].usage, &cipher, &plain);
620  	if (ret)
621  	    errx(1, "krb_enc_iov failed with %d for test %d", ret, (int)i);
622  
623  	ret = krb_enc_iov2(context, crypto, krbencs[i].usage,
624  			   cipher.length, &plain);
625  	if (ret)
626  	    errx(1, "krb_enc_iov2 failed with %d for test %d", ret, (int)i);
627  
628  	ret = krb_checksum_iov(context, crypto, krbencs[i].usage, &plain);
629  	if (ret)
630  	    errx(1, "krb_checksum_iov failed with %d for test %d", ret, (int)i);
631  
632  	krb5_crypto_destroy(context, crypto);
633  
634  	ret = krb_enc_mit(context, krbencs[i].enctype, &kb,
635  			  krbencs[i].usage, &cipher, &plain);
636  	if (ret)
637  	    errx(1, "krb_enc_mit failed with %d for test %d", ret, (int)i);
638      }
639  
640      return 0;
641  }
642  
643  static int
644  iov_test(krb5_context context)
645  {
646      krb5_enctype enctype = KRB5_ENCTYPE_AES256_CTS_HMAC_SHA1_96;
647      krb5_error_code ret;
648      krb5_crypto crypto;
649      krb5_keyblock key;
650      krb5_data signonly, in, in2;
651      krb5_crypto_iov iov[6];
652      size_t len, i;
653      unsigned char *base, *p;
654  
655      ret = krb5_generate_random_keyblock(context, enctype, &key);
656      if (ret)
657  	krb5_err(context, 1, ret, "krb5_generate_random_keyblock");
658  
659      ret = krb5_crypto_init(context, &key, 0, &crypto);
660      if (ret)
661  	krb5_err(context, 1, ret, "krb5_crypto_init");
662  
663  
664      ret = krb5_crypto_length(context, crypto, KRB5_CRYPTO_TYPE_HEADER, &len);
665      if (ret)
666  	krb5_err(context, 1, ret, "krb5_crypto_length");
667  
668      signonly.data = "This should be signed";
669      signonly.length = strlen(signonly.data);
670      in.data = "inputdata";
671      in.length = strlen(in.data);
672  
673      in2.data = "INPUTDATA";
674      in2.length = strlen(in2.data);
675  
676  
677      memset(iov, 0, sizeof(iov));
678  
679      iov[0].flags = KRB5_CRYPTO_TYPE_HEADER;
680      iov[1].flags = KRB5_CRYPTO_TYPE_DATA;
681      iov[1].data = in;
682      iov[2].flags = KRB5_CRYPTO_TYPE_SIGN_ONLY;
683      iov[2].data = signonly;
684      iov[3].flags = KRB5_CRYPTO_TYPE_EMPTY;
685      iov[4].flags = KRB5_CRYPTO_TYPE_PADDING;
686      iov[5].flags = KRB5_CRYPTO_TYPE_TRAILER;
687  
688      ret = krb5_crypto_length_iov(context, crypto, iov,
689  				 sizeof(iov)/sizeof(iov[0]));
690      if (ret)
691  	krb5_err(context, 1, ret, "krb5_crypto_length_iov");
692  
693      for (len = 0, i = 0; i < sizeof(iov)/sizeof(iov[0]); i++) {
694  	if (iov[i].flags == KRB5_CRYPTO_TYPE_SIGN_ONLY)
695  	    continue;
696  	len += iov[i].data.length;
697      }
698  
699      base = emalloc(len);
700  
701      /*
702       * Allocate data for the fields
703       */
704  
705      for (p = base, i = 0; i < sizeof(iov)/sizeof(iov[0]); i++) {
706  	if (iov[i].flags == KRB5_CRYPTO_TYPE_SIGN_ONLY)
707  	    continue;;
708  	iov[i].data.data = p;
709  	p += iov[i].data.length;
710      }
711      assert(iov[1].data.length == in.length);
712      memcpy(iov[1].data.data, in.data, iov[1].data.length);
713  
714      /*
715       * Encrypt
716       */
717  
718      ret = krb5_encrypt_iov_ivec(context, crypto, 7, iov,
719  				sizeof(iov)/sizeof(iov[0]), NULL);
720      if (ret)
721  	krb5_err(context, 1, ret, "krb5_encrypt_iov_ivec");
722  
723      /*
724       * Decrypt
725       */
726  
727      ret = krb5_decrypt_iov_ivec(context, crypto, 7,
728  				iov, sizeof(iov)/sizeof(iov[0]), NULL);
729      if (ret)
730  	krb5_err(context, 1, ret, "krb5_decrypt_iov_ivec");
731  
732      /*
733       * Verify data
734       */
735  
736      if (krb5_data_cmp(&iov[1].data, &in) != 0)
737  	krb5_errx(context, 1, "decrypted data not same");
738  
739      /*
740       * Free memory
741       */
742  
743      free(base);
744  
745      /* Set up for second try */
746  
747      iov[3].flags = KRB5_CRYPTO_TYPE_DATA;
748      iov[3].data = in;
749  
750      ret = krb5_crypto_length_iov(context, crypto,
751  				 iov, sizeof(iov)/sizeof(iov[0]));
752      if (ret)
753  	krb5_err(context, 1, ret, "krb5_crypto_length_iov");
754  
755      for (len = 0, i = 0; i < sizeof(iov)/sizeof(iov[0]); i++) {
756  	if (iov[i].flags == KRB5_CRYPTO_TYPE_SIGN_ONLY)
757  	    continue;
758  	len += iov[i].data.length;
759      }
760  
761      base = emalloc(len);
762  
763      /*
764       * Allocate data for the fields
765       */
766  
767      for (p = base, i = 0; i < sizeof(iov)/sizeof(iov[0]); i++) {
768  	if (iov[i].flags == KRB5_CRYPTO_TYPE_SIGN_ONLY)
769  	    continue;;
770  	iov[i].data.data = p;
771  	p += iov[i].data.length;
772      }
773      assert(iov[1].data.length == in.length);
774      memcpy(iov[1].data.data, in.data, iov[1].data.length);
775  
776      assert(iov[3].data.length == in2.length);
777      memcpy(iov[3].data.data, in2.data, iov[3].data.length);
778  
779  
780  
781      /*
782       * Encrypt
783       */
784  
785      ret = krb5_encrypt_iov_ivec(context, crypto, 7,
786  				iov, sizeof(iov)/sizeof(iov[0]), NULL);
787      if (ret)
788  	krb5_err(context, 1, ret, "krb5_encrypt_iov_ivec");
789  
790      /*
791       * Decrypt
792       */
793  
794      ret = krb5_decrypt_iov_ivec(context, crypto, 7,
795  				iov, sizeof(iov)/sizeof(iov[0]), NULL);
796      if (ret)
797  	krb5_err(context, 1, ret, "krb5_decrypt_iov_ivec");
798  
799      /*
800       * Verify data
801       */
802  
803      if (krb5_data_cmp(&iov[1].data, &in) != 0)
804  	krb5_errx(context, 1, "decrypted data 2.1 not same");
805  
806      if (krb5_data_cmp(&iov[3].data, &in2) != 0)
807  	krb5_errx(context, 1, "decrypted data 2.2 not same");
808  
809      /*
810       * Free memory
811       */
812  
813      free(base);
814  
815      krb5_crypto_destroy(context, crypto);
816  
817      krb5_free_keyblock_contents(context, &key);
818  
819      return 0;
820  }
821  
822  
823  
824  static int
825  random_to_key(krb5_context context)
826  {
827      krb5_error_code ret;
828      krb5_keyblock key;
829  
830      ret = krb5_random_to_key(context,
831  			     ETYPE_DES3_CBC_SHA1,
832  			     "\x21\x39\x04\x58\x6A\xBD\x7F"
833  			     "\x21\x39\x04\x58\x6A\xBD\x7F"
834  			     "\x21\x39\x04\x58\x6A\xBD\x7F",
835  			     21,
836  			     &key);
837      if (ret){
838  	krb5_warn(context, ret, "random_to_key");
839  	return 1;
840      }
841      if (key.keyvalue.length != 24)
842  	return 1;
843  
844      if (memcmp(key.keyvalue.data,
845  	       "\x20\x38\x04\x58\x6b\xbc\x7f\xc7"
846  	       "\x20\x38\x04\x58\x6b\xbc\x7f\xc7"
847  	       "\x20\x38\x04\x58\x6b\xbc\x7f\xc7",
848  	       24) != 0)
849  	return 1;
850  
851      krb5_free_keyblock_contents(context, &key);
852  
853      return 0;
854  }
855  
856  int
857  main(int argc, char **argv)
858  {
859      krb5_error_code ret;
860      krb5_context context;
861      int ival, val = 0;
862  
863      ret = krb5_init_context (&context);
864      if (ret)
865  	errx (1, "krb5_init_context failed: %d", ret);
866  
867      val |= ival = string_to_key_test(context);
868      if (ival && verbose)
869  	printf("string to key failed\n");
870  
871      val |= ival = krb_enc_test(context);
872      if (ival && verbose)
873  	printf("enc tests failed\n");
874  
875      val |= ival = random_to_key(context);
876      if (ival && verbose)
877  	printf("random to key failed\n");
878  
879      val |= ival = iov_test(context);
880      if (ival && verbose)
881  	printf("iov test failed\n");
882  
883      if (verbose && val == 0)
884  	printf("all ok\n");
885      if (val)
886  	printf("tests failed\n");
887  
888      krb5_free_context(context);
889  
890      return val;
891  }