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 
24 struct 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  */
34 struct 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 
111 static void
112 printhex(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 
123 int
124 main(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