From bac884b9e6c0a070068d143909959b9e2d829a7d Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Love=20H=C3=B6rnquist=20=C3=85strand?= <lha@kth.se>
Date: Fri, 13 Jan 2006 08:42:49 +0000
Subject: [PATCH] Test for PKCS12_key_gen.

git-svn-id: svn://svn.h5l.se/heimdal/trunk/heimdal@16543 ec53bebd-3082-4978-b11e-865c3cabbd6b
---
 lib/des/test_pkcs12.c | 128 ++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 128 insertions(+)
 create mode 100644 lib/des/test_pkcs12.c

diff --git a/lib/des/test_pkcs12.c b/lib/des/test_pkcs12.c
new file mode 100644
index 000000000..f434718e8
--- /dev/null
+++ b/lib/des/test_pkcs12.c
@@ -0,0 +1,128 @@
+/*
+ * Copyright (c) 2006 Kungliga Tekniska Högskolan
+ * (Royal Institute of Technology, Stockholm, Sweden). 
+ * All rights reserved. 
+ *
+ * Redistribution and use in source and binary forms, with or without 
+ * modification, are permitted provided that the following conditions 
+ * are met: 
+ *
+ * 1. Redistributions of source code must retain the above copyright 
+ *    notice, this list of conditions and the following disclaimer. 
+ *
+ * 2. Redistributions in binary form must reproduce the above copyright 
+ *    notice, this list of conditions and the following disclaimer in the 
+ *    documentation and/or other materials provided with the distribution. 
+ *
+ * 3. Neither the name of the Institute nor the names of its contributors 
+ *    may be used to endorse or promote products derived from this software 
+ *    without specific prior written permission. 
+ *
+ * THIS SOFTWARE IS PROVIDED BY THE INSTITUTE AND CONTRIBUTORS ``AS IS'' AND 
+ * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 
+ * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 
+ * ARE DISCLAIMED.  IN NO EVENT SHALL THE INSTITUTE OR CONTRIBUTORS BE LIABLE 
+ * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 
+ * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 
+ * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 
+ * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 
+ * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 
+ * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 
+ * SUCH DAMAGE. 
+ */
+
+#ifdef HAVE_CONFIG_H
+#include <config.h>
+#endif
+
+#ifdef RCSID
+RCSID("$Id$");
+#endif
+
+#include <sys/types.h>
+#include <limits.h>
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+
+#include <pkcs12.h>
+#include <evp.h>
+
+struct tests {
+    int id;
+    const char *password;
+    void *salt;
+    size_t saltsize;
+    int iterations;
+    size_t keylen;
+    const EVP_MD * (*md)(void);
+    void *key;
+};
+
+struct tests p12_pbe_tests[] = {
+    { PKCS12_KEY_ID,
+      "foobar",
+      "\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00",
+      16,
+      100,
+      16,
+      EVP_sha1,
+      "\x79\x95\xbf\x3f\x1c\x6d\xe\xe8\xd3\x71\xc4\x94\xd\xb\x18\xb5"
+    },
+    { PKCS12_KEY_ID,
+      "foobar",
+      "\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00",
+      16,
+      2048,
+      24,
+      EVP_sha1,
+      "\x0b\xb5\xe\xa6\x71\x0d\x0c\xf7\x44\xe\xe1\x9b\xb5\xdf\xf1\xdc\x4f\xb0\xca\xe\xee\x4f\xb9\xfd"
+    },
+    { PKCS12_IV_ID,
+      "foobar",
+      "\x3c\xdf\x84\x32\x59\xd3\xda\x69",
+      8,
+      2048,
+      8,
+      EVP_sha1,
+      "\xbf\x9a\x12\xb7\x26\x69\xfd\x05"
+    }
+
+};
+
+static int
+test_pkcs12_pbe(struct tests *t)
+{
+    void *key;
+
+    key = malloc(t->keylen);
+
+    if (!PKCS12_key_gen(t->password, strlen(t->password), 
+			t->salt, t->saltsize,
+			t->id, t->iterations, t->keylen, 
+			key, t->md()))
+    {
+	printf("key_gen failed\n");
+	return 1;
+    }
+
+    if (memcmp(t->key, key, t->keylen) != 0) {
+	printf("incorrect key\n");
+	free(key);
+	return 1;
+    }
+    free(key);
+    return 0;
+}
+
+int
+main(int argc, char **argv)
+{
+    int ret = 0;
+    int i;
+
+    for (i = 0; i < sizeof(p12_pbe_tests)/sizeof(p12_pbe_tests[0]); i++)
+	ret += test_pkcs12_pbe(&p12_pbe_tests[i]);
+
+    return ret;
+}