From 0cfe6c3513e8111ec3085eba1b8708947f3da9ab Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Love=20H=C3=B6rnquist=20=C3=85strand?= <lha@kth.se>
Date: Sat, 19 Jul 2003 07:40:02 +0000
Subject: [PATCH] add test for krb5_c_encrypt_length and krb5_c_decrypt

git-svn-id: svn://svn.h5l.se/heimdal/trunk/heimdal@12452 ec53bebd-3082-4978-b11e-865c3cabbd6b
---
 lib/krb5/aes-test.c | 58 ++++++++++++++++++++++++++++++++++++++++++---
 1 file changed, 55 insertions(+), 3 deletions(-)

diff --git a/lib/krb5/aes-test.c b/lib/krb5/aes-test.c
index 869f24fcc..f7cb6bb5e 100644
--- a/lib/krb5/aes-test.c
+++ b/lib/krb5/aes-test.c
@@ -46,12 +46,13 @@ hex_dump_data(krb5_data *data)
     unsigned char *p = data->data;
     int i, j;
 
-    for (i = j = 0; i < data->length; i++, j++) {
+    for (i = j = 0; i < data->length; i++) {
 	printf("%02x ", p[i]);
 	if (j > 15) {
 	    printf("\n");
 	    j = 0;
-	}
+	} else
+	    j++;
     }
     if (j != 0)
 	printf("\n");
@@ -612,6 +613,49 @@ krb_enc(krb5_context context,
     return 0;
 }
 
+static int
+krb_enc_mit(krb5_context context,
+	    krb5_enctype enctype,
+	    krb5_keyblock *key,
+	    unsigned usage,
+	    krb5_data *cipher, 
+	    krb5_data *clear)
+{
+    krb5_error_code ret;
+    krb5_enc_data e;
+    krb5_data decrypt;
+    size_t len;
+
+    e.kvno = 0;
+    e.enctype = enctype;
+    e.ciphertext = *cipher;
+
+    ret = krb5_c_decrypt(context, *key, usage, NULL, &e, &decrypt);
+    if (ret)
+	return ret;
+
+    if (decrypt.length != clear->length ||
+	memcmp(decrypt.data, clear->data, decrypt.length) != 0) {
+	krb5_warnx(context, "clear text not same");
+	return EINVAL;
+    }
+
+    krb5_data_free(&decrypt);
+
+    ret = krb5_c_encrypt_length(context, enctype, clear->length, &len);
+    if (ret)
+	return ret;
+
+    if (len != cipher->length) {
+	krb5_warnx(context, "c_encrypt_length wrong %d != %d",
+		   len, cipher->length);
+	return EINVAL;
+    }
+
+    return 0;
+}
+
+
 struct {
     krb5_enctype enctype;
     unsigned usage;
@@ -671,9 +715,17 @@ krb_enc_test(krb5_context context)
 		       
 	if (ret) {
 	    failed = 1;
-	    printf("failed with %d\n", ret);
+	    printf("krb_enc failed with %d\n", ret);
 	}
 	krb5_crypto_destroy(context, crypto);
+
+	ret = krb_enc_mit(context, krbencs[i].enctype, &kb, 
+			  krbencs[i].usage, &cipher, &plain);
+	if (ret) {
+	    failed = 1;
+	    printf("krb_enc_mit failed with %d\n", ret);
+	}
+
     }
 
     return failed;