1/*
2 * CDDL HEADER START
3 *
4 * This file and its contents are supplied under the terms of the
5 * Common Development and Distribution License ("CDDL"), version 1.0.
6 * You may only use this file in accordance with the terms of version
7 * 1.0 of the CDDL.
8 *
9 * A full copy of the text of the CDDL should have accompanied this
10 * source.  A copy of the CDDL is also available via the Internet at
11 * http://www.illumos.org/license/CDDL.
12 *
13 * CDDL HEADER END
14 */
15
16/*
17 * Copyright (c) 2017, Datto, Inc. All rights reserved.
18 */
19
20#include <stdio.h>
21#include <string.h>
22#include <sys/crypto/icp.h>
23#include <sys/sha2.h>
24#include <sys/hkdf.h>
25
26/*
27 * Byte arrays are given as char pointers so that they
28 * can be specified as strings.
29 */
30typedef struct hkdf_tv {
31	/* test vector input values */
32	const char		*ikm;
33	uint_t		ikm_len;
34	const char		*salt;
35	uint_t		salt_len;
36	const char		*info;
37	uint_t		info_len;
38	uint_t		okm_len;
39
40	/* expected output */
41	const char		*okm;
42} hkdf_tv_t;
43
44/*
45 * XXX Neither NIST nor IETF has published official test
46 * vectors for testing HKDF with SHA512. The following
47 * vectors should be updated if these are ever published.
48 * The current vectors were taken from:
49 * https://www.kullo.net/blog/hkdf-sha-512-test-vectors/
50 */
51static const hkdf_tv_t test_vectors[] = {
52	{
53		.ikm =	"\x0b\x0b\x0b\x0b\x0b\x0b\x0b\x0b"
54			"\x0b\x0b\x0b\x0b\x0b\x0b\x0b\x0b"
55			"\x0b\x0b\x0b\x0b\x0b\x0b",
56		.ikm_len = 22,
57		.salt =	"\x00\x01\x02\x03\x04\x05\x06\x07"
58			"\x08\x09\x0a\x0b\x0c",
59		.salt_len = 13,
60		.info =	"\xf0\xf1\xf2\xf3\xf4\xf5\xf6\xf7"
61			"\xf8\xf9",
62		.info_len = 10,
63		.okm_len = 42,
64		.okm =	"\x83\x23\x90\x08\x6c\xda\x71\xfb"
65			"\x47\x62\x5b\xb5\xce\xb1\x68\xe4"
66			"\xc8\xe2\x6a\x1a\x16\xed\x34\xd9"
67			"\xfc\x7f\xe9\x2c\x14\x81\x57\x93"
68			"\x38\xda\x36\x2c\xb8\xd9\xf9\x25"
69			"\xd7\xcb",
70	},
71	{
72		.ikm = 	"\x00\x01\x02\x03\x04\x05\x06\x07"
73			"\x08\x09\x0a\x0b\x0c\x0d\x0e\x0f"
74			"\x10\x11\x12\x13\x14\x15\x16\x17"
75			"\x18\x19\x1a\x1b\x1c\x1d\x1e\x1f"
76			"\x20\x21\x22\x23\x24\x25\x26\x27"
77			"\x28\x29\x2a\x2b\x2c\x2d\x2e\x2f"
78			"\x30\x31\x32\x33\x34\x35\x36\x37"
79			"\x38\x39\x3a\x3b\x3c\x3d\x3e\x3f"
80			"\x40\x41\x42\x43\x44\x45\x46\x47"
81			"\x48\x49\x4a\x4b\x4c\x4d\x4e\x4f",
82		.ikm_len = 80,
83		.salt =	"\x60\x61\x62\x63\x64\x65\x66\x67"
84			"\x68\x69\x6a\x6b\x6c\x6d\x6e\x6f"
85			"\x70\x71\x72\x73\x74\x75\x76\x77"
86			"\x78\x79\x7a\x7b\x7c\x7d\x7e\x7f"
87			"\x80\x81\x82\x83\x84\x85\x86\x87"
88			"\x88\x89\x8a\x8b\x8c\x8d\x8e\x8f"
89			"\x90\x91\x92\x93\x94\x95\x96\x97"
90			"\x98\x99\x9a\x9b\x9c\x9d\x9e\x9f"
91			"\xa0\xa1\xa2\xa3\xa4\xa5\xa6\xa7"
92			"\xa8\xa9\xaa\xab\xac\xad\xae\xaf",
93		.salt_len = 80,
94		.info =	"\xb0\xb1\xb2\xb3\xb4\xb5\xb6\xb7"
95			"\xb8\xb9\xba\xbb\xbc\xbd\xbe\xbf"
96			"\xc0\xc1\xc2\xc3\xc4\xc5\xc6\xc7"
97			"\xc8\xc9\xca\xcb\xcc\xcd\xce\xcf"
98			"\xd0\xd1\xd2\xd3\xd4\xd5\xd6\xd7"
99			"\xd8\xd9\xda\xdb\xdc\xdd\xde\xdf"
100			"\xe0\xe1\xe2\xe3\xe4\xe5\xe6\xe7"
101			"\xe8\xe9\xea\xeb\xec\xed\xee\xef"
102			"\xf0\xf1\xf2\xf3\xf4\xf5\xf6\xf7"
103			"\xf8\xf9\xfa\xfb\xfc\xfd\xfe\xff",
104		.info_len = 80,
105		.okm_len = 42,
106		.okm =	"\xce\x6c\x97\x19\x28\x05\xb3\x46"
107			"\xe6\x16\x1e\x82\x1e\xd1\x65\x67"
108			"\x3b\x84\xf4\x00\xa2\xb5\x14\xb2"
109			"\xfe\x23\xd8\x4c\xd1\x89\xdd\xf1"
110			"\xb6\x95\xb4\x8c\xbd\x1c\x83\x88"
111			"\x44\x11\x37\xb3\xce\x28\xf1\x6a"
112			"\xa6\x4b\xa3\x3b\xa4\x66\xb2\x4d"
113			"\xf6\xcf\xcb\x02\x1e\xcf\xf2\x35"
114			"\xf6\xa2\x05\x6c\xe3\xaf\x1d\xe4"
115			"\x4d\x57\x20\x97\xa8\x50\x5d\x9e"
116			"\x7a\x93",
117	},
118	{
119		.ikm =	"\x0b\x0b\x0b\x0b\x0b\x0b\x0b\x0b"
120			"\x0b\x0b\x0b\x0b\x0b\x0b\x0b\x0b"
121			"\x0b\x0b\x0b\x0b\x0b\x0b",
122		.ikm_len = 22,
123		.salt =	NULL,
124		.salt_len = 0,
125		.info =	NULL,
126		.info_len = 0,
127		.okm_len = 42,
128		.okm =	"\xf5\xfa\x02\xb1\x82\x98\xa7\x2a"
129			"\x8c\x23\x89\x8a\x87\x03\x47\x2c"
130			"\x6e\xb1\x79\xdc\x20\x4c\x03\x42"
131			"\x5c\x97\x0e\x3b\x16\x4b\xf9\x0f"
132			"\xff\x22\xd0\x48\x36\xd0\xe2\x34"
133			"\x3b\xac",
134	},
135	{
136		.ikm =	"\x0b\x0b\x0b\x0b\x0b\x0b\x0b\x0b"
137			"\x0b\x0b\x0b",
138		.ikm_len = 11,
139		.salt =	"\x00\x01\x02\x03\x04\x05\x06\x07"
140			"\x08\x09\x0a\x0b\x0c",
141		.salt_len = 13,
142		.info =	"\xf0\xf1\xf2\xf3\xf4\xf5\xf6\xf7"
143			"\xf8\xf9",
144		.info_len = 10,
145		.okm_len = 42,
146		.okm =	"\x74\x13\xe8\x99\x7e\x02\x06\x10"
147			"\xfb\xf6\x82\x3f\x2c\xe1\x4b\xff"
148			"\x01\x87\x5d\xb1\xca\x55\xf6\x8c"
149			"\xfc\xf3\x95\x4d\xc8\xaf\xf5\x35"
150			"\x59\xbd\x5e\x30\x28\xb0\x80\xf7"
151			"\xc0\x68",
152	},
153	{
154		.ikm =	"\x0c\x0c\x0c\x0c\x0c\x0c\x0c\x0c"
155			"\x0c\x0c\x0c\x0c\x0c\x0c\x0c\x0c"
156			"\x0c\x0c\x0c\x0c\x0c\x0c",
157		.ikm_len = 22,
158		.salt =	NULL,
159		.salt_len = 0,
160		.info = NULL,
161		.info_len = 0,
162		.okm_len = 42,
163		.okm =	"\x14\x07\xd4\x60\x13\xd9\x8b\xc6"
164			"\xde\xce\xfc\xfe\xe5\x5f\x0f\x90"
165			"\xb0\xc7\xf6\x3d\x68\xeb\x1a\x80"
166			"\xea\xf0\x7e\x95\x3c\xfc\x0a\x3a"
167			"\x52\x40\xa1\x55\xd6\xe4\xda\xa9"
168			"\x65\xbb",
169	},
170};
171
172static void
173hexdump(const char *str, uint8_t *src, uint_t len)
174{
175	printf("\t%s\t", str);
176	for (int i = 0; i < len; i++)
177		printf("%02hhx", src[i]);
178	printf("\n");
179}
180
181static int
182run_test(int i, const hkdf_tv_t *tv)
183{
184	int ret;
185	uint8_t good[SHA512_DIGEST_LENGTH];
186
187	printf("TEST %d:\t", i);
188
189	ret = hkdf_sha512((uint8_t *)tv->ikm, tv->ikm_len, (uint8_t *)tv->salt,
190	    tv->salt_len, (uint8_t *)tv->info, tv->info_len, good, tv->okm_len);
191	if (ret != 0) {
192		printf("HKDF failed with error code %d\n", ret);
193		return (ret);
194	}
195
196	if (memcmp(good, tv->okm, tv->okm_len) != 0) {
197		printf("Output Mismatch\n");
198		hexdump("Expected:", (uint8_t *)tv->okm, tv->okm_len);
199		hexdump("Actual:  ", good, tv->okm_len);
200		return (1);
201	}
202
203	printf("Passed\n");
204
205	return (0);
206}
207
208int
209main(void)
210{
211	int ret, i;
212
213	icp_init();
214
215	for (i = 0; i < ARRAY_SIZE(test_vectors); i++) {
216		ret = run_test(i, &test_vectors[i]);
217		if (ret != 0)
218			break;
219	}
220
221	icp_fini();
222
223	if (ret == 0) {
224		printf("All tests passed successfully.\n");
225		return (0);
226	} else {
227		printf("Test failed.\n");
228		return (1);
229	}
230}
231