1/*	$OpenBSD: pkcs5_pbkdf2_test.c,v 1.1 2012/09/06 20:49:59 matthew Exp $	*/
2/*-
3 * Copyright (c) 2008 Damien Bergamini <damien.bergamini@free.fr>
4 *
5 * Permission to use, copy, modify, and distribute this software for any
6 * purpose with or without fee is hereby granted, provided that the above
7 * copyright notice and this permission notice appear in all copies.
8 *
9 * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
10 * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
11 * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
12 * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
13 * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
14 * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
15 * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
16 */
17
18#include <sys/types.h>
19#include <err.h>
20#include <stdio.h>
21#include <string.h>
22#include <util.h>
23
24struct test_vector {
25	u_int rounds;
26	const char *pass;
27	const char *salt;
28	const char expected[32];
29};
30
31/*
32 * Test vectors from RFC 3962
33 */
34struct test_vector test_vectors[] = {
35	{
36		1,
37		"password",
38		"ATHENA.MIT.EDUraeburn",
39		{
40			0xcd, 0xed, 0xb5, 0x28, 0x1b, 0xb2, 0xf8, 0x01,
41			0x56, 0x5a, 0x11, 0x22, 0xb2, 0x56, 0x35, 0x15,
42			0x0a, 0xd1, 0xf7, 0xa0, 0x4b, 0xb9, 0xf3, 0xa3,
43			0x33, 0xec, 0xc0, 0xe2, 0xe1, 0xf7, 0x08, 0x37
44		},
45	}, {
46		2,
47		"password",
48		"ATHENA.MIT.EDUraeburn",
49		{
50			0x01, 0xdb, 0xee, 0x7f, 0x4a, 0x9e, 0x24, 0x3e,
51			0x98, 0x8b, 0x62, 0xc7, 0x3c, 0xda, 0x93, 0x5d,
52			0xa0, 0x53, 0x78, 0xb9, 0x32, 0x44, 0xec, 0x8f,
53			0x48, 0xa9, 0x9e, 0x61, 0xad, 0x79, 0x9d, 0x86
54		},
55	}, {
56		1200,
57		"password",
58		"ATHENA.MIT.EDUraeburn",
59		{
60			0x5c, 0x08, 0xeb, 0x61, 0xfd, 0xf7, 0x1e, 0x4e,
61			0x4e, 0xc3, 0xcf, 0x6b, 0xa1, 0xf5, 0x51, 0x2b,
62			0xa7, 0xe5, 0x2d, 0xdb, 0xc5, 0xe5, 0x14, 0x2f,
63			0x70, 0x8a, 0x31, 0xe2, 0xe6, 0x2b, 0x1e, 0x13
64		},
65	}, {
66		5,
67		"password",
68		"\0224VxxV4\022", /* 0x1234567878563412 */
69		{
70			0xd1, 0xda, 0xa7, 0x86, 0x15, 0xf2, 0x87, 0xe6,
71			0xa1, 0xc8, 0xb1, 0x20, 0xd7, 0x06, 0x2a, 0x49,
72			0x3f, 0x98, 0xd2, 0x03, 0xe6, 0xbe, 0x49, 0xa6,
73			0xad, 0xf4, 0xfa, 0x57, 0x4b, 0x6e, 0x64, 0xee
74		},
75	}, {
76		1200,
77		"XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX"
78		"XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX",
79		"pass phrase equals block size",
80		{
81			0x13, 0x9c, 0x30, 0xc0, 0x96, 0x6b, 0xc3, 0x2b,
82			0xa5, 0x5f, 0xdb, 0xf2, 0x12, 0x53, 0x0a, 0xc9,
83			0xc5, 0xec, 0x59, 0xf1, 0xa4, 0x52, 0xf5, 0xcc,
84			0x9a, 0xd9, 0x40, 0xfe, 0xa0, 0x59, 0x8e, 0xd1
85		},
86	}, {
87		1200,
88		"XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX"
89		"XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX",
90		"pass phrase exceeds block size",
91		{
92			0x9c, 0xca, 0xd6, 0xd4, 0x68, 0x77, 0x0c, 0xd5,
93			0x1b, 0x10, 0xe6, 0xa6, 0x87, 0x21, 0xbe, 0x61,
94			0x1a, 0x8b, 0x4d, 0x28, 0x26, 0x01, 0xdb, 0x3b,
95			0x36, 0xbe, 0x92, 0x46, 0x91, 0x5e, 0xc8, 0x2a
96		},
97	}, {
98		50,
99		"\360\235\204\236", /* g-clef (0xf09d849e) */
100		"EXAMPLE.COMpianist",
101		{
102			0x6b, 0x9c, 0xf2, 0x6d, 0x45, 0x45, 0x5a, 0x43,
103			0xa5, 0xb8, 0xbb, 0x27, 0x6a, 0x40, 0x3b, 0x39,
104			0xe7, 0xfe, 0x37, 0xa0, 0xc4, 0x1e, 0x02, 0xc2,
105			0x81, 0xff, 0x30, 0x69, 0xe1, 0xe9, 0x4f, 0x52
106		},
107	}
108};
109#define NVECS (sizeof(test_vectors) / sizeof(*test_vectors))
110
111static void
112printhex(const char *s, const u_int8_t *buf, size_t len)
113{
114	size_t i;
115
116	printf("%s: ", s);
117	for (i = 0; i < len; i++)
118		printf("%02x", buf[i]);
119	printf("\n");
120	fflush(stdout);
121}
122
123int
124main(int argc, char **argv)
125{
126	u_int i, j;
127	u_char result[32];
128	struct test_vector *vec;
129
130	for (i = 0; i < NVECS; i++) {
131		vec = &test_vectors[i];
132		for (j = 2; j <= sizeof(result); j += 3) {
133			if (pkcs5_pbkdf2(vec->pass, strlen(vec->pass),
134			    vec->salt, strlen(vec->salt),
135			    result, j, vec->rounds) != 0)
136				errx(1, "pbkdf2 vector %u failed", i);
137			if (memcmp(result, vec->expected, j) != 0) {
138				printhex(" got", result, j);
139				printhex("want", vec->expected, j);
140				return 1;
141			}
142		}
143	}
144	return 0;
145}
146