xref: /linux/crypto/drbg.c (revision 74ce1896c6c65b2f8cccbf59162d542988835835)
1 /*
2  * DRBG: Deterministic Random Bits Generator
3  *       Based on NIST Recommended DRBG from NIST SP800-90A with the following
4  *       properties:
5  *		* CTR DRBG with DF with AES-128, AES-192, AES-256 cores
6  *		* Hash DRBG with DF with SHA-1, SHA-256, SHA-384, SHA-512 cores
7  *		* HMAC DRBG with DF with SHA-1, SHA-256, SHA-384, SHA-512 cores
8  *		* with and without prediction resistance
9  *
10  * Copyright Stephan Mueller <smueller@chronox.de>, 2014
11  *
12  * Redistribution and use in source and binary forms, with or without
13  * modification, are permitted provided that the following conditions
14  * are met:
15  * 1. Redistributions of source code must retain the above copyright
16  *    notice, and the entire permission notice in its entirety,
17  *    including the disclaimer of warranties.
18  * 2. Redistributions in binary form must reproduce the above copyright
19  *    notice, this list of conditions and the following disclaimer in the
20  *    documentation and/or other materials provided with the distribution.
21  * 3. The name of the author may not be used to endorse or promote
22  *    products derived from this software without specific prior
23  *    written permission.
24  *
25  * ALTERNATIVELY, this product may be distributed under the terms of
26  * the GNU General Public License, in which case the provisions of the GPL are
27  * required INSTEAD OF the above restrictions.  (This clause is
28  * necessary due to a potential bad interaction between the GPL and
29  * the restrictions contained in a BSD-style copyright.)
30  *
31  * THIS SOFTWARE IS PROVIDED ``AS IS'' AND ANY EXPRESS OR IMPLIED
32  * WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
33  * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE, ALL OF
34  * WHICH ARE HEREBY DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR BE
35  * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
36  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT
37  * OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR
38  * BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
39  * LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
40  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE
41  * USE OF THIS SOFTWARE, EVEN IF NOT ADVISED OF THE POSSIBILITY OF SUCH
42  * DAMAGE.
43  *
44  * DRBG Usage
45  * ==========
46  * The SP 800-90A DRBG allows the user to specify a personalization string
47  * for initialization as well as an additional information string for each
48  * random number request. The following code fragments show how a caller
49  * uses the kernel crypto API to use the full functionality of the DRBG.
50  *
51  * Usage without any additional data
52  * ---------------------------------
53  * struct crypto_rng *drng;
54  * int err;
55  * char data[DATALEN];
56  *
57  * drng = crypto_alloc_rng(drng_name, 0, 0);
58  * err = crypto_rng_get_bytes(drng, &data, DATALEN);
59  * crypto_free_rng(drng);
60  *
61  *
62  * Usage with personalization string during initialization
63  * -------------------------------------------------------
64  * struct crypto_rng *drng;
65  * int err;
66  * char data[DATALEN];
67  * struct drbg_string pers;
68  * char personalization[11] = "some-string";
69  *
70  * drbg_string_fill(&pers, personalization, strlen(personalization));
71  * drng = crypto_alloc_rng(drng_name, 0, 0);
72  * // The reset completely re-initializes the DRBG with the provided
73  * // personalization string
74  * err = crypto_rng_reset(drng, &personalization, strlen(personalization));
75  * err = crypto_rng_get_bytes(drng, &data, DATALEN);
76  * crypto_free_rng(drng);
77  *
78  *
79  * Usage with additional information string during random number request
80  * ---------------------------------------------------------------------
81  * struct crypto_rng *drng;
82  * int err;
83  * char data[DATALEN];
84  * char addtl_string[11] = "some-string";
85  * string drbg_string addtl;
86  *
87  * drbg_string_fill(&addtl, addtl_string, strlen(addtl_string));
88  * drng = crypto_alloc_rng(drng_name, 0, 0);
89  * // The following call is a wrapper to crypto_rng_get_bytes() and returns
90  * // the same error codes.
91  * err = crypto_drbg_get_bytes_addtl(drng, &data, DATALEN, &addtl);
92  * crypto_free_rng(drng);
93  *
94  *
95  * Usage with personalization and additional information strings
96  * -------------------------------------------------------------
97  * Just mix both scenarios above.
98  */
99 
100 #include <crypto/drbg.h>
101 #include <linux/kernel.h>
102 
103 /***************************************************************
104  * Backend cipher definitions available to DRBG
105  ***************************************************************/
106 
107 /*
108  * The order of the DRBG definitions here matter: every DRBG is registered
109  * as stdrng. Each DRBG receives an increasing cra_priority values the later
110  * they are defined in this array (see drbg_fill_array).
111  *
112  * HMAC DRBGs are favored over Hash DRBGs over CTR DRBGs, and
113  * the SHA256 / AES 256 over other ciphers. Thus, the favored
114  * DRBGs are the latest entries in this array.
115  */
116 static const struct drbg_core drbg_cores[] = {
117 #ifdef CONFIG_CRYPTO_DRBG_CTR
118 	{
119 		.flags = DRBG_CTR | DRBG_STRENGTH128,
120 		.statelen = 32, /* 256 bits as defined in 10.2.1 */
121 		.blocklen_bytes = 16,
122 		.cra_name = "ctr_aes128",
123 		.backend_cra_name = "aes",
124 	}, {
125 		.flags = DRBG_CTR | DRBG_STRENGTH192,
126 		.statelen = 40, /* 320 bits as defined in 10.2.1 */
127 		.blocklen_bytes = 16,
128 		.cra_name = "ctr_aes192",
129 		.backend_cra_name = "aes",
130 	}, {
131 		.flags = DRBG_CTR | DRBG_STRENGTH256,
132 		.statelen = 48, /* 384 bits as defined in 10.2.1 */
133 		.blocklen_bytes = 16,
134 		.cra_name = "ctr_aes256",
135 		.backend_cra_name = "aes",
136 	},
137 #endif /* CONFIG_CRYPTO_DRBG_CTR */
138 #ifdef CONFIG_CRYPTO_DRBG_HASH
139 	{
140 		.flags = DRBG_HASH | DRBG_STRENGTH128,
141 		.statelen = 55, /* 440 bits */
142 		.blocklen_bytes = 20,
143 		.cra_name = "sha1",
144 		.backend_cra_name = "sha1",
145 	}, {
146 		.flags = DRBG_HASH | DRBG_STRENGTH256,
147 		.statelen = 111, /* 888 bits */
148 		.blocklen_bytes = 48,
149 		.cra_name = "sha384",
150 		.backend_cra_name = "sha384",
151 	}, {
152 		.flags = DRBG_HASH | DRBG_STRENGTH256,
153 		.statelen = 111, /* 888 bits */
154 		.blocklen_bytes = 64,
155 		.cra_name = "sha512",
156 		.backend_cra_name = "sha512",
157 	}, {
158 		.flags = DRBG_HASH | DRBG_STRENGTH256,
159 		.statelen = 55, /* 440 bits */
160 		.blocklen_bytes = 32,
161 		.cra_name = "sha256",
162 		.backend_cra_name = "sha256",
163 	},
164 #endif /* CONFIG_CRYPTO_DRBG_HASH */
165 #ifdef CONFIG_CRYPTO_DRBG_HMAC
166 	{
167 		.flags = DRBG_HMAC | DRBG_STRENGTH128,
168 		.statelen = 20, /* block length of cipher */
169 		.blocklen_bytes = 20,
170 		.cra_name = "hmac_sha1",
171 		.backend_cra_name = "hmac(sha1)",
172 	}, {
173 		.flags = DRBG_HMAC | DRBG_STRENGTH256,
174 		.statelen = 48, /* block length of cipher */
175 		.blocklen_bytes = 48,
176 		.cra_name = "hmac_sha384",
177 		.backend_cra_name = "hmac(sha384)",
178 	}, {
179 		.flags = DRBG_HMAC | DRBG_STRENGTH256,
180 		.statelen = 64, /* block length of cipher */
181 		.blocklen_bytes = 64,
182 		.cra_name = "hmac_sha512",
183 		.backend_cra_name = "hmac(sha512)",
184 	}, {
185 		.flags = DRBG_HMAC | DRBG_STRENGTH256,
186 		.statelen = 32, /* block length of cipher */
187 		.blocklen_bytes = 32,
188 		.cra_name = "hmac_sha256",
189 		.backend_cra_name = "hmac(sha256)",
190 	},
191 #endif /* CONFIG_CRYPTO_DRBG_HMAC */
192 };
193 
194 static int drbg_uninstantiate(struct drbg_state *drbg);
195 
196 /******************************************************************
197  * Generic helper functions
198  ******************************************************************/
199 
200 /*
201  * Return strength of DRBG according to SP800-90A section 8.4
202  *
203  * @flags DRBG flags reference
204  *
205  * Return: normalized strength in *bytes* value or 32 as default
206  *	   to counter programming errors
207  */
208 static inline unsigned short drbg_sec_strength(drbg_flag_t flags)
209 {
210 	switch (flags & DRBG_STRENGTH_MASK) {
211 	case DRBG_STRENGTH128:
212 		return 16;
213 	case DRBG_STRENGTH192:
214 		return 24;
215 	case DRBG_STRENGTH256:
216 		return 32;
217 	default:
218 		return 32;
219 	}
220 }
221 
222 /*
223  * Convert an integer into a byte representation of this integer.
224  * The byte representation is big-endian
225  *
226  * @val value to be converted
227  * @buf buffer holding the converted integer -- caller must ensure that
228  *      buffer size is at least 32 bit
229  */
230 #if (defined(CONFIG_CRYPTO_DRBG_HASH) || defined(CONFIG_CRYPTO_DRBG_CTR))
231 static inline void drbg_cpu_to_be32(__u32 val, unsigned char *buf)
232 {
233 	struct s {
234 		__be32 conv;
235 	};
236 	struct s *conversion = (struct s *) buf;
237 
238 	conversion->conv = cpu_to_be32(val);
239 }
240 #endif /* defined(CONFIG_CRYPTO_DRBG_HASH) || defined(CONFIG_CRYPTO_DRBG_CTR) */
241 
242 /******************************************************************
243  * CTR DRBG callback functions
244  ******************************************************************/
245 
246 #ifdef CONFIG_CRYPTO_DRBG_CTR
247 #define CRYPTO_DRBG_CTR_STRING "CTR "
248 MODULE_ALIAS_CRYPTO("drbg_pr_ctr_aes256");
249 MODULE_ALIAS_CRYPTO("drbg_nopr_ctr_aes256");
250 MODULE_ALIAS_CRYPTO("drbg_pr_ctr_aes192");
251 MODULE_ALIAS_CRYPTO("drbg_nopr_ctr_aes192");
252 MODULE_ALIAS_CRYPTO("drbg_pr_ctr_aes128");
253 MODULE_ALIAS_CRYPTO("drbg_nopr_ctr_aes128");
254 
255 static void drbg_kcapi_symsetkey(struct drbg_state *drbg,
256 				 const unsigned char *key);
257 static int drbg_kcapi_sym(struct drbg_state *drbg, unsigned char *outval,
258 			  const struct drbg_string *in);
259 static int drbg_init_sym_kernel(struct drbg_state *drbg);
260 static int drbg_fini_sym_kernel(struct drbg_state *drbg);
261 static int drbg_kcapi_sym_ctr(struct drbg_state *drbg,
262 			      u8 *inbuf, u32 inbuflen,
263 			      u8 *outbuf, u32 outlen);
264 #define DRBG_CTR_NULL_LEN 128
265 #define DRBG_OUTSCRATCHLEN DRBG_CTR_NULL_LEN
266 
267 /* BCC function for CTR DRBG as defined in 10.4.3 */
268 static int drbg_ctr_bcc(struct drbg_state *drbg,
269 			unsigned char *out, const unsigned char *key,
270 			struct list_head *in)
271 {
272 	int ret = 0;
273 	struct drbg_string *curr = NULL;
274 	struct drbg_string data;
275 	short cnt = 0;
276 
277 	drbg_string_fill(&data, out, drbg_blocklen(drbg));
278 
279 	/* 10.4.3 step 2 / 4 */
280 	drbg_kcapi_symsetkey(drbg, key);
281 	list_for_each_entry(curr, in, list) {
282 		const unsigned char *pos = curr->buf;
283 		size_t len = curr->len;
284 		/* 10.4.3 step 4.1 */
285 		while (len) {
286 			/* 10.4.3 step 4.2 */
287 			if (drbg_blocklen(drbg) == cnt) {
288 				cnt = 0;
289 				ret = drbg_kcapi_sym(drbg, out, &data);
290 				if (ret)
291 					return ret;
292 			}
293 			out[cnt] ^= *pos;
294 			pos++;
295 			cnt++;
296 			len--;
297 		}
298 	}
299 	/* 10.4.3 step 4.2 for last block */
300 	if (cnt)
301 		ret = drbg_kcapi_sym(drbg, out, &data);
302 
303 	return ret;
304 }
305 
306 /*
307  * scratchpad usage: drbg_ctr_update is interlinked with drbg_ctr_df
308  * (and drbg_ctr_bcc, but this function does not need any temporary buffers),
309  * the scratchpad is used as follows:
310  * drbg_ctr_update:
311  *	temp
312  *		start: drbg->scratchpad
313  *		length: drbg_statelen(drbg) + drbg_blocklen(drbg)
314  *			note: the cipher writing into this variable works
315  *			blocklen-wise. Now, when the statelen is not a multiple
316  *			of blocklen, the generateion loop below "spills over"
317  *			by at most blocklen. Thus, we need to give sufficient
318  *			memory.
319  *	df_data
320  *		start: drbg->scratchpad +
321  *				drbg_statelen(drbg) + drbg_blocklen(drbg)
322  *		length: drbg_statelen(drbg)
323  *
324  * drbg_ctr_df:
325  *	pad
326  *		start: df_data + drbg_statelen(drbg)
327  *		length: drbg_blocklen(drbg)
328  *	iv
329  *		start: pad + drbg_blocklen(drbg)
330  *		length: drbg_blocklen(drbg)
331  *	temp
332  *		start: iv + drbg_blocklen(drbg)
333  *		length: drbg_satelen(drbg) + drbg_blocklen(drbg)
334  *			note: temp is the buffer that the BCC function operates
335  *			on. BCC operates blockwise. drbg_statelen(drbg)
336  *			is sufficient when the DRBG state length is a multiple
337  *			of the block size. For AES192 (and maybe other ciphers)
338  *			this is not correct and the length for temp is
339  *			insufficient (yes, that also means for such ciphers,
340  *			the final output of all BCC rounds are truncated).
341  *			Therefore, add drbg_blocklen(drbg) to cover all
342  *			possibilities.
343  */
344 
345 /* Derivation Function for CTR DRBG as defined in 10.4.2 */
346 static int drbg_ctr_df(struct drbg_state *drbg,
347 		       unsigned char *df_data, size_t bytes_to_return,
348 		       struct list_head *seedlist)
349 {
350 	int ret = -EFAULT;
351 	unsigned char L_N[8];
352 	/* S3 is input */
353 	struct drbg_string S1, S2, S4, cipherin;
354 	LIST_HEAD(bcc_list);
355 	unsigned char *pad = df_data + drbg_statelen(drbg);
356 	unsigned char *iv = pad + drbg_blocklen(drbg);
357 	unsigned char *temp = iv + drbg_blocklen(drbg);
358 	size_t padlen = 0;
359 	unsigned int templen = 0;
360 	/* 10.4.2 step 7 */
361 	unsigned int i = 0;
362 	/* 10.4.2 step 8 */
363 	const unsigned char *K = (unsigned char *)
364 			   "\x00\x01\x02\x03\x04\x05\x06\x07"
365 			   "\x08\x09\x0a\x0b\x0c\x0d\x0e\x0f"
366 			   "\x10\x11\x12\x13\x14\x15\x16\x17"
367 			   "\x18\x19\x1a\x1b\x1c\x1d\x1e\x1f";
368 	unsigned char *X;
369 	size_t generated_len = 0;
370 	size_t inputlen = 0;
371 	struct drbg_string *seed = NULL;
372 
373 	memset(pad, 0, drbg_blocklen(drbg));
374 	memset(iv, 0, drbg_blocklen(drbg));
375 
376 	/* 10.4.2 step 1 is implicit as we work byte-wise */
377 
378 	/* 10.4.2 step 2 */
379 	if ((512/8) < bytes_to_return)
380 		return -EINVAL;
381 
382 	/* 10.4.2 step 2 -- calculate the entire length of all input data */
383 	list_for_each_entry(seed, seedlist, list)
384 		inputlen += seed->len;
385 	drbg_cpu_to_be32(inputlen, &L_N[0]);
386 
387 	/* 10.4.2 step 3 */
388 	drbg_cpu_to_be32(bytes_to_return, &L_N[4]);
389 
390 	/* 10.4.2 step 5: length is L_N, input_string, one byte, padding */
391 	padlen = (inputlen + sizeof(L_N) + 1) % (drbg_blocklen(drbg));
392 	/* wrap the padlen appropriately */
393 	if (padlen)
394 		padlen = drbg_blocklen(drbg) - padlen;
395 	/*
396 	 * pad / padlen contains the 0x80 byte and the following zero bytes.
397 	 * As the calculated padlen value only covers the number of zero
398 	 * bytes, this value has to be incremented by one for the 0x80 byte.
399 	 */
400 	padlen++;
401 	pad[0] = 0x80;
402 
403 	/* 10.4.2 step 4 -- first fill the linked list and then order it */
404 	drbg_string_fill(&S1, iv, drbg_blocklen(drbg));
405 	list_add_tail(&S1.list, &bcc_list);
406 	drbg_string_fill(&S2, L_N, sizeof(L_N));
407 	list_add_tail(&S2.list, &bcc_list);
408 	list_splice_tail(seedlist, &bcc_list);
409 	drbg_string_fill(&S4, pad, padlen);
410 	list_add_tail(&S4.list, &bcc_list);
411 
412 	/* 10.4.2 step 9 */
413 	while (templen < (drbg_keylen(drbg) + (drbg_blocklen(drbg)))) {
414 		/*
415 		 * 10.4.2 step 9.1 - the padding is implicit as the buffer
416 		 * holds zeros after allocation -- even the increment of i
417 		 * is irrelevant as the increment remains within length of i
418 		 */
419 		drbg_cpu_to_be32(i, iv);
420 		/* 10.4.2 step 9.2 -- BCC and concatenation with temp */
421 		ret = drbg_ctr_bcc(drbg, temp + templen, K, &bcc_list);
422 		if (ret)
423 			goto out;
424 		/* 10.4.2 step 9.3 */
425 		i++;
426 		templen += drbg_blocklen(drbg);
427 	}
428 
429 	/* 10.4.2 step 11 */
430 	X = temp + (drbg_keylen(drbg));
431 	drbg_string_fill(&cipherin, X, drbg_blocklen(drbg));
432 
433 	/* 10.4.2 step 12: overwriting of outval is implemented in next step */
434 
435 	/* 10.4.2 step 13 */
436 	drbg_kcapi_symsetkey(drbg, temp);
437 	while (generated_len < bytes_to_return) {
438 		short blocklen = 0;
439 		/*
440 		 * 10.4.2 step 13.1: the truncation of the key length is
441 		 * implicit as the key is only drbg_blocklen in size based on
442 		 * the implementation of the cipher function callback
443 		 */
444 		ret = drbg_kcapi_sym(drbg, X, &cipherin);
445 		if (ret)
446 			goto out;
447 		blocklen = (drbg_blocklen(drbg) <
448 				(bytes_to_return - generated_len)) ?
449 			    drbg_blocklen(drbg) :
450 				(bytes_to_return - generated_len);
451 		/* 10.4.2 step 13.2 and 14 */
452 		memcpy(df_data + generated_len, X, blocklen);
453 		generated_len += blocklen;
454 	}
455 
456 	ret = 0;
457 
458 out:
459 	memset(iv, 0, drbg_blocklen(drbg));
460 	memset(temp, 0, drbg_statelen(drbg) + drbg_blocklen(drbg));
461 	memset(pad, 0, drbg_blocklen(drbg));
462 	return ret;
463 }
464 
465 /*
466  * update function of CTR DRBG as defined in 10.2.1.2
467  *
468  * The reseed variable has an enhanced meaning compared to the update
469  * functions of the other DRBGs as follows:
470  * 0 => initial seed from initialization
471  * 1 => reseed via drbg_seed
472  * 2 => first invocation from drbg_ctr_update when addtl is present. In
473  *      this case, the df_data scratchpad is not deleted so that it is
474  *      available for another calls to prevent calling the DF function
475  *      again.
476  * 3 => second invocation from drbg_ctr_update. When the update function
477  *      was called with addtl, the df_data memory already contains the
478  *      DFed addtl information and we do not need to call DF again.
479  */
480 static int drbg_ctr_update(struct drbg_state *drbg, struct list_head *seed,
481 			   int reseed)
482 {
483 	int ret = -EFAULT;
484 	/* 10.2.1.2 step 1 */
485 	unsigned char *temp = drbg->scratchpad;
486 	unsigned char *df_data = drbg->scratchpad + drbg_statelen(drbg) +
487 				 drbg_blocklen(drbg);
488 
489 	if (3 > reseed)
490 		memset(df_data, 0, drbg_statelen(drbg));
491 
492 	if (!reseed) {
493 		/*
494 		 * The DRBG uses the CTR mode of the underlying AES cipher. The
495 		 * CTR mode increments the counter value after the AES operation
496 		 * but SP800-90A requires that the counter is incremented before
497 		 * the AES operation. Hence, we increment it at the time we set
498 		 * it by one.
499 		 */
500 		crypto_inc(drbg->V, drbg_blocklen(drbg));
501 
502 		ret = crypto_skcipher_setkey(drbg->ctr_handle, drbg->C,
503 					     drbg_keylen(drbg));
504 		if (ret)
505 			goto out;
506 	}
507 
508 	/* 10.2.1.3.2 step 2 and 10.2.1.4.2 step 2 */
509 	if (seed) {
510 		ret = drbg_ctr_df(drbg, df_data, drbg_statelen(drbg), seed);
511 		if (ret)
512 			goto out;
513 	}
514 
515 	ret = drbg_kcapi_sym_ctr(drbg, df_data, drbg_statelen(drbg),
516 				 temp, drbg_statelen(drbg));
517 	if (ret)
518 		return ret;
519 
520 	/* 10.2.1.2 step 5 */
521 	ret = crypto_skcipher_setkey(drbg->ctr_handle, temp,
522 				     drbg_keylen(drbg));
523 	if (ret)
524 		goto out;
525 	/* 10.2.1.2 step 6 */
526 	memcpy(drbg->V, temp + drbg_keylen(drbg), drbg_blocklen(drbg));
527 	/* See above: increment counter by one to compensate timing of CTR op */
528 	crypto_inc(drbg->V, drbg_blocklen(drbg));
529 	ret = 0;
530 
531 out:
532 	memset(temp, 0, drbg_statelen(drbg) + drbg_blocklen(drbg));
533 	if (2 != reseed)
534 		memset(df_data, 0, drbg_statelen(drbg));
535 	return ret;
536 }
537 
538 /*
539  * scratchpad use: drbg_ctr_update is called independently from
540  * drbg_ctr_extract_bytes. Therefore, the scratchpad is reused
541  */
542 /* Generate function of CTR DRBG as defined in 10.2.1.5.2 */
543 static int drbg_ctr_generate(struct drbg_state *drbg,
544 			     unsigned char *buf, unsigned int buflen,
545 			     struct list_head *addtl)
546 {
547 	int ret;
548 	int len = min_t(int, buflen, INT_MAX);
549 
550 	/* 10.2.1.5.2 step 2 */
551 	if (addtl && !list_empty(addtl)) {
552 		ret = drbg_ctr_update(drbg, addtl, 2);
553 		if (ret)
554 			return 0;
555 	}
556 
557 	/* 10.2.1.5.2 step 4.1 */
558 	ret = drbg_kcapi_sym_ctr(drbg, drbg->ctr_null_value, DRBG_CTR_NULL_LEN,
559 				 buf, len);
560 	if (ret)
561 		return ret;
562 
563 	/* 10.2.1.5.2 step 6 */
564 	ret = drbg_ctr_update(drbg, NULL, 3);
565 	if (ret)
566 		len = ret;
567 
568 	return len;
569 }
570 
571 static const struct drbg_state_ops drbg_ctr_ops = {
572 	.update		= drbg_ctr_update,
573 	.generate	= drbg_ctr_generate,
574 	.crypto_init	= drbg_init_sym_kernel,
575 	.crypto_fini	= drbg_fini_sym_kernel,
576 };
577 #endif /* CONFIG_CRYPTO_DRBG_CTR */
578 
579 /******************************************************************
580  * HMAC DRBG callback functions
581  ******************************************************************/
582 
583 #if defined(CONFIG_CRYPTO_DRBG_HASH) || defined(CONFIG_CRYPTO_DRBG_HMAC)
584 static int drbg_kcapi_hash(struct drbg_state *drbg, unsigned char *outval,
585 			   const struct list_head *in);
586 static void drbg_kcapi_hmacsetkey(struct drbg_state *drbg,
587 				  const unsigned char *key);
588 static int drbg_init_hash_kernel(struct drbg_state *drbg);
589 static int drbg_fini_hash_kernel(struct drbg_state *drbg);
590 #endif /* (CONFIG_CRYPTO_DRBG_HASH || CONFIG_CRYPTO_DRBG_HMAC) */
591 
592 #ifdef CONFIG_CRYPTO_DRBG_HMAC
593 #define CRYPTO_DRBG_HMAC_STRING "HMAC "
594 MODULE_ALIAS_CRYPTO("drbg_pr_hmac_sha512");
595 MODULE_ALIAS_CRYPTO("drbg_nopr_hmac_sha512");
596 MODULE_ALIAS_CRYPTO("drbg_pr_hmac_sha384");
597 MODULE_ALIAS_CRYPTO("drbg_nopr_hmac_sha384");
598 MODULE_ALIAS_CRYPTO("drbg_pr_hmac_sha256");
599 MODULE_ALIAS_CRYPTO("drbg_nopr_hmac_sha256");
600 MODULE_ALIAS_CRYPTO("drbg_pr_hmac_sha1");
601 MODULE_ALIAS_CRYPTO("drbg_nopr_hmac_sha1");
602 
603 /* update function of HMAC DRBG as defined in 10.1.2.2 */
604 static int drbg_hmac_update(struct drbg_state *drbg, struct list_head *seed,
605 			    int reseed)
606 {
607 	int ret = -EFAULT;
608 	int i = 0;
609 	struct drbg_string seed1, seed2, vdata;
610 	LIST_HEAD(seedlist);
611 	LIST_HEAD(vdatalist);
612 
613 	if (!reseed) {
614 		/* 10.1.2.3 step 2 -- memset(0) of C is implicit with kzalloc */
615 		memset(drbg->V, 1, drbg_statelen(drbg));
616 		drbg_kcapi_hmacsetkey(drbg, drbg->C);
617 	}
618 
619 	drbg_string_fill(&seed1, drbg->V, drbg_statelen(drbg));
620 	list_add_tail(&seed1.list, &seedlist);
621 	/* buffer of seed2 will be filled in for loop below with one byte */
622 	drbg_string_fill(&seed2, NULL, 1);
623 	list_add_tail(&seed2.list, &seedlist);
624 	/* input data of seed is allowed to be NULL at this point */
625 	if (seed)
626 		list_splice_tail(seed, &seedlist);
627 
628 	drbg_string_fill(&vdata, drbg->V, drbg_statelen(drbg));
629 	list_add_tail(&vdata.list, &vdatalist);
630 	for (i = 2; 0 < i; i--) {
631 		/* first round uses 0x0, second 0x1 */
632 		unsigned char prefix = DRBG_PREFIX0;
633 		if (1 == i)
634 			prefix = DRBG_PREFIX1;
635 		/* 10.1.2.2 step 1 and 4 -- concatenation and HMAC for key */
636 		seed2.buf = &prefix;
637 		ret = drbg_kcapi_hash(drbg, drbg->C, &seedlist);
638 		if (ret)
639 			return ret;
640 		drbg_kcapi_hmacsetkey(drbg, drbg->C);
641 
642 		/* 10.1.2.2 step 2 and 5 -- HMAC for V */
643 		ret = drbg_kcapi_hash(drbg, drbg->V, &vdatalist);
644 		if (ret)
645 			return ret;
646 
647 		/* 10.1.2.2 step 3 */
648 		if (!seed)
649 			return ret;
650 	}
651 
652 	return 0;
653 }
654 
655 /* generate function of HMAC DRBG as defined in 10.1.2.5 */
656 static int drbg_hmac_generate(struct drbg_state *drbg,
657 			      unsigned char *buf,
658 			      unsigned int buflen,
659 			      struct list_head *addtl)
660 {
661 	int len = 0;
662 	int ret = 0;
663 	struct drbg_string data;
664 	LIST_HEAD(datalist);
665 
666 	/* 10.1.2.5 step 2 */
667 	if (addtl && !list_empty(addtl)) {
668 		ret = drbg_hmac_update(drbg, addtl, 1);
669 		if (ret)
670 			return ret;
671 	}
672 
673 	drbg_string_fill(&data, drbg->V, drbg_statelen(drbg));
674 	list_add_tail(&data.list, &datalist);
675 	while (len < buflen) {
676 		unsigned int outlen = 0;
677 		/* 10.1.2.5 step 4.1 */
678 		ret = drbg_kcapi_hash(drbg, drbg->V, &datalist);
679 		if (ret)
680 			return ret;
681 		outlen = (drbg_blocklen(drbg) < (buflen - len)) ?
682 			  drbg_blocklen(drbg) : (buflen - len);
683 
684 		/* 10.1.2.5 step 4.2 */
685 		memcpy(buf + len, drbg->V, outlen);
686 		len += outlen;
687 	}
688 
689 	/* 10.1.2.5 step 6 */
690 	if (addtl && !list_empty(addtl))
691 		ret = drbg_hmac_update(drbg, addtl, 1);
692 	else
693 		ret = drbg_hmac_update(drbg, NULL, 1);
694 	if (ret)
695 		return ret;
696 
697 	return len;
698 }
699 
700 static const struct drbg_state_ops drbg_hmac_ops = {
701 	.update		= drbg_hmac_update,
702 	.generate	= drbg_hmac_generate,
703 	.crypto_init	= drbg_init_hash_kernel,
704 	.crypto_fini	= drbg_fini_hash_kernel,
705 };
706 #endif /* CONFIG_CRYPTO_DRBG_HMAC */
707 
708 /******************************************************************
709  * Hash DRBG callback functions
710  ******************************************************************/
711 
712 #ifdef CONFIG_CRYPTO_DRBG_HASH
713 #define CRYPTO_DRBG_HASH_STRING "HASH "
714 MODULE_ALIAS_CRYPTO("drbg_pr_sha512");
715 MODULE_ALIAS_CRYPTO("drbg_nopr_sha512");
716 MODULE_ALIAS_CRYPTO("drbg_pr_sha384");
717 MODULE_ALIAS_CRYPTO("drbg_nopr_sha384");
718 MODULE_ALIAS_CRYPTO("drbg_pr_sha256");
719 MODULE_ALIAS_CRYPTO("drbg_nopr_sha256");
720 MODULE_ALIAS_CRYPTO("drbg_pr_sha1");
721 MODULE_ALIAS_CRYPTO("drbg_nopr_sha1");
722 
723 /*
724  * Increment buffer
725  *
726  * @dst buffer to increment
727  * @add value to add
728  */
729 static inline void drbg_add_buf(unsigned char *dst, size_t dstlen,
730 				const unsigned char *add, size_t addlen)
731 {
732 	/* implied: dstlen > addlen */
733 	unsigned char *dstptr;
734 	const unsigned char *addptr;
735 	unsigned int remainder = 0;
736 	size_t len = addlen;
737 
738 	dstptr = dst + (dstlen-1);
739 	addptr = add + (addlen-1);
740 	while (len) {
741 		remainder += *dstptr + *addptr;
742 		*dstptr = remainder & 0xff;
743 		remainder >>= 8;
744 		len--; dstptr--; addptr--;
745 	}
746 	len = dstlen - addlen;
747 	while (len && remainder > 0) {
748 		remainder = *dstptr + 1;
749 		*dstptr = remainder & 0xff;
750 		remainder >>= 8;
751 		len--; dstptr--;
752 	}
753 }
754 
755 /*
756  * scratchpad usage: as drbg_hash_update and drbg_hash_df are used
757  * interlinked, the scratchpad is used as follows:
758  * drbg_hash_update
759  *	start: drbg->scratchpad
760  *	length: drbg_statelen(drbg)
761  * drbg_hash_df:
762  *	start: drbg->scratchpad + drbg_statelen(drbg)
763  *	length: drbg_blocklen(drbg)
764  *
765  * drbg_hash_process_addtl uses the scratchpad, but fully completes
766  * before either of the functions mentioned before are invoked. Therefore,
767  * drbg_hash_process_addtl does not need to be specifically considered.
768  */
769 
770 /* Derivation Function for Hash DRBG as defined in 10.4.1 */
771 static int drbg_hash_df(struct drbg_state *drbg,
772 			unsigned char *outval, size_t outlen,
773 			struct list_head *entropylist)
774 {
775 	int ret = 0;
776 	size_t len = 0;
777 	unsigned char input[5];
778 	unsigned char *tmp = drbg->scratchpad + drbg_statelen(drbg);
779 	struct drbg_string data;
780 
781 	/* 10.4.1 step 3 */
782 	input[0] = 1;
783 	drbg_cpu_to_be32((outlen * 8), &input[1]);
784 
785 	/* 10.4.1 step 4.1 -- concatenation of data for input into hash */
786 	drbg_string_fill(&data, input, 5);
787 	list_add(&data.list, entropylist);
788 
789 	/* 10.4.1 step 4 */
790 	while (len < outlen) {
791 		short blocklen = 0;
792 		/* 10.4.1 step 4.1 */
793 		ret = drbg_kcapi_hash(drbg, tmp, entropylist);
794 		if (ret)
795 			goto out;
796 		/* 10.4.1 step 4.2 */
797 		input[0]++;
798 		blocklen = (drbg_blocklen(drbg) < (outlen - len)) ?
799 			    drbg_blocklen(drbg) : (outlen - len);
800 		memcpy(outval + len, tmp, blocklen);
801 		len += blocklen;
802 	}
803 
804 out:
805 	memset(tmp, 0, drbg_blocklen(drbg));
806 	return ret;
807 }
808 
809 /* update function for Hash DRBG as defined in 10.1.1.2 / 10.1.1.3 */
810 static int drbg_hash_update(struct drbg_state *drbg, struct list_head *seed,
811 			    int reseed)
812 {
813 	int ret = 0;
814 	struct drbg_string data1, data2;
815 	LIST_HEAD(datalist);
816 	LIST_HEAD(datalist2);
817 	unsigned char *V = drbg->scratchpad;
818 	unsigned char prefix = DRBG_PREFIX1;
819 
820 	if (!seed)
821 		return -EINVAL;
822 
823 	if (reseed) {
824 		/* 10.1.1.3 step 1 */
825 		memcpy(V, drbg->V, drbg_statelen(drbg));
826 		drbg_string_fill(&data1, &prefix, 1);
827 		list_add_tail(&data1.list, &datalist);
828 		drbg_string_fill(&data2, V, drbg_statelen(drbg));
829 		list_add_tail(&data2.list, &datalist);
830 	}
831 	list_splice_tail(seed, &datalist);
832 
833 	/* 10.1.1.2 / 10.1.1.3 step 2 and 3 */
834 	ret = drbg_hash_df(drbg, drbg->V, drbg_statelen(drbg), &datalist);
835 	if (ret)
836 		goto out;
837 
838 	/* 10.1.1.2 / 10.1.1.3 step 4  */
839 	prefix = DRBG_PREFIX0;
840 	drbg_string_fill(&data1, &prefix, 1);
841 	list_add_tail(&data1.list, &datalist2);
842 	drbg_string_fill(&data2, drbg->V, drbg_statelen(drbg));
843 	list_add_tail(&data2.list, &datalist2);
844 	/* 10.1.1.2 / 10.1.1.3 step 4 */
845 	ret = drbg_hash_df(drbg, drbg->C, drbg_statelen(drbg), &datalist2);
846 
847 out:
848 	memset(drbg->scratchpad, 0, drbg_statelen(drbg));
849 	return ret;
850 }
851 
852 /* processing of additional information string for Hash DRBG */
853 static int drbg_hash_process_addtl(struct drbg_state *drbg,
854 				   struct list_head *addtl)
855 {
856 	int ret = 0;
857 	struct drbg_string data1, data2;
858 	LIST_HEAD(datalist);
859 	unsigned char prefix = DRBG_PREFIX2;
860 
861 	/* 10.1.1.4 step 2 */
862 	if (!addtl || list_empty(addtl))
863 		return 0;
864 
865 	/* 10.1.1.4 step 2a */
866 	drbg_string_fill(&data1, &prefix, 1);
867 	drbg_string_fill(&data2, drbg->V, drbg_statelen(drbg));
868 	list_add_tail(&data1.list, &datalist);
869 	list_add_tail(&data2.list, &datalist);
870 	list_splice_tail(addtl, &datalist);
871 	ret = drbg_kcapi_hash(drbg, drbg->scratchpad, &datalist);
872 	if (ret)
873 		goto out;
874 
875 	/* 10.1.1.4 step 2b */
876 	drbg_add_buf(drbg->V, drbg_statelen(drbg),
877 		     drbg->scratchpad, drbg_blocklen(drbg));
878 
879 out:
880 	memset(drbg->scratchpad, 0, drbg_blocklen(drbg));
881 	return ret;
882 }
883 
884 /* Hashgen defined in 10.1.1.4 */
885 static int drbg_hash_hashgen(struct drbg_state *drbg,
886 			     unsigned char *buf,
887 			     unsigned int buflen)
888 {
889 	int len = 0;
890 	int ret = 0;
891 	unsigned char *src = drbg->scratchpad;
892 	unsigned char *dst = drbg->scratchpad + drbg_statelen(drbg);
893 	struct drbg_string data;
894 	LIST_HEAD(datalist);
895 
896 	/* 10.1.1.4 step hashgen 2 */
897 	memcpy(src, drbg->V, drbg_statelen(drbg));
898 
899 	drbg_string_fill(&data, src, drbg_statelen(drbg));
900 	list_add_tail(&data.list, &datalist);
901 	while (len < buflen) {
902 		unsigned int outlen = 0;
903 		/* 10.1.1.4 step hashgen 4.1 */
904 		ret = drbg_kcapi_hash(drbg, dst, &datalist);
905 		if (ret) {
906 			len = ret;
907 			goto out;
908 		}
909 		outlen = (drbg_blocklen(drbg) < (buflen - len)) ?
910 			  drbg_blocklen(drbg) : (buflen - len);
911 		/* 10.1.1.4 step hashgen 4.2 */
912 		memcpy(buf + len, dst, outlen);
913 		len += outlen;
914 		/* 10.1.1.4 hashgen step 4.3 */
915 		if (len < buflen)
916 			crypto_inc(src, drbg_statelen(drbg));
917 	}
918 
919 out:
920 	memset(drbg->scratchpad, 0,
921 	       (drbg_statelen(drbg) + drbg_blocklen(drbg)));
922 	return len;
923 }
924 
925 /* generate function for Hash DRBG as defined in  10.1.1.4 */
926 static int drbg_hash_generate(struct drbg_state *drbg,
927 			      unsigned char *buf, unsigned int buflen,
928 			      struct list_head *addtl)
929 {
930 	int len = 0;
931 	int ret = 0;
932 	union {
933 		unsigned char req[8];
934 		__be64 req_int;
935 	} u;
936 	unsigned char prefix = DRBG_PREFIX3;
937 	struct drbg_string data1, data2;
938 	LIST_HEAD(datalist);
939 
940 	/* 10.1.1.4 step 2 */
941 	ret = drbg_hash_process_addtl(drbg, addtl);
942 	if (ret)
943 		return ret;
944 	/* 10.1.1.4 step 3 */
945 	len = drbg_hash_hashgen(drbg, buf, buflen);
946 
947 	/* this is the value H as documented in 10.1.1.4 */
948 	/* 10.1.1.4 step 4 */
949 	drbg_string_fill(&data1, &prefix, 1);
950 	list_add_tail(&data1.list, &datalist);
951 	drbg_string_fill(&data2, drbg->V, drbg_statelen(drbg));
952 	list_add_tail(&data2.list, &datalist);
953 	ret = drbg_kcapi_hash(drbg, drbg->scratchpad, &datalist);
954 	if (ret) {
955 		len = ret;
956 		goto out;
957 	}
958 
959 	/* 10.1.1.4 step 5 */
960 	drbg_add_buf(drbg->V, drbg_statelen(drbg),
961 		     drbg->scratchpad, drbg_blocklen(drbg));
962 	drbg_add_buf(drbg->V, drbg_statelen(drbg),
963 		     drbg->C, drbg_statelen(drbg));
964 	u.req_int = cpu_to_be64(drbg->reseed_ctr);
965 	drbg_add_buf(drbg->V, drbg_statelen(drbg), u.req, 8);
966 
967 out:
968 	memset(drbg->scratchpad, 0, drbg_blocklen(drbg));
969 	return len;
970 }
971 
972 /*
973  * scratchpad usage: as update and generate are used isolated, both
974  * can use the scratchpad
975  */
976 static const struct drbg_state_ops drbg_hash_ops = {
977 	.update		= drbg_hash_update,
978 	.generate	= drbg_hash_generate,
979 	.crypto_init	= drbg_init_hash_kernel,
980 	.crypto_fini	= drbg_fini_hash_kernel,
981 };
982 #endif /* CONFIG_CRYPTO_DRBG_HASH */
983 
984 /******************************************************************
985  * Functions common for DRBG implementations
986  ******************************************************************/
987 
988 static inline int __drbg_seed(struct drbg_state *drbg, struct list_head *seed,
989 			      int reseed)
990 {
991 	int ret = drbg->d_ops->update(drbg, seed, reseed);
992 
993 	if (ret)
994 		return ret;
995 
996 	drbg->seeded = true;
997 	/* 10.1.1.2 / 10.1.1.3 step 5 */
998 	drbg->reseed_ctr = 1;
999 
1000 	return ret;
1001 }
1002 
1003 static void drbg_async_seed(struct work_struct *work)
1004 {
1005 	struct drbg_string data;
1006 	LIST_HEAD(seedlist);
1007 	struct drbg_state *drbg = container_of(work, struct drbg_state,
1008 					       seed_work);
1009 	unsigned int entropylen = drbg_sec_strength(drbg->core->flags);
1010 	unsigned char entropy[32];
1011 
1012 	BUG_ON(!entropylen);
1013 	BUG_ON(entropylen > sizeof(entropy));
1014 	get_random_bytes(entropy, entropylen);
1015 
1016 	drbg_string_fill(&data, entropy, entropylen);
1017 	list_add_tail(&data.list, &seedlist);
1018 
1019 	mutex_lock(&drbg->drbg_mutex);
1020 
1021 	/* If nonblocking pool is initialized, deactivate Jitter RNG */
1022 	crypto_free_rng(drbg->jent);
1023 	drbg->jent = NULL;
1024 
1025 	/* Set seeded to false so that if __drbg_seed fails the
1026 	 * next generate call will trigger a reseed.
1027 	 */
1028 	drbg->seeded = false;
1029 
1030 	__drbg_seed(drbg, &seedlist, true);
1031 
1032 	if (drbg->seeded)
1033 		drbg->reseed_threshold = drbg_max_requests(drbg);
1034 
1035 	mutex_unlock(&drbg->drbg_mutex);
1036 
1037 	memzero_explicit(entropy, entropylen);
1038 }
1039 
1040 /*
1041  * Seeding or reseeding of the DRBG
1042  *
1043  * @drbg: DRBG state struct
1044  * @pers: personalization / additional information buffer
1045  * @reseed: 0 for initial seed process, 1 for reseeding
1046  *
1047  * return:
1048  *	0 on success
1049  *	error value otherwise
1050  */
1051 static int drbg_seed(struct drbg_state *drbg, struct drbg_string *pers,
1052 		     bool reseed)
1053 {
1054 	int ret;
1055 	unsigned char entropy[((32 + 16) * 2)];
1056 	unsigned int entropylen = drbg_sec_strength(drbg->core->flags);
1057 	struct drbg_string data1;
1058 	LIST_HEAD(seedlist);
1059 
1060 	/* 9.1 / 9.2 / 9.3.1 step 3 */
1061 	if (pers && pers->len > (drbg_max_addtl(drbg))) {
1062 		pr_devel("DRBG: personalization string too long %zu\n",
1063 			 pers->len);
1064 		return -EINVAL;
1065 	}
1066 
1067 	if (list_empty(&drbg->test_data.list)) {
1068 		drbg_string_fill(&data1, drbg->test_data.buf,
1069 				 drbg->test_data.len);
1070 		pr_devel("DRBG: using test entropy\n");
1071 	} else {
1072 		/*
1073 		 * Gather entropy equal to the security strength of the DRBG.
1074 		 * With a derivation function, a nonce is required in addition
1075 		 * to the entropy. A nonce must be at least 1/2 of the security
1076 		 * strength of the DRBG in size. Thus, entropy + nonce is 3/2
1077 		 * of the strength. The consideration of a nonce is only
1078 		 * applicable during initial seeding.
1079 		 */
1080 		BUG_ON(!entropylen);
1081 		if (!reseed)
1082 			entropylen = ((entropylen + 1) / 2) * 3;
1083 		BUG_ON((entropylen * 2) > sizeof(entropy));
1084 
1085 		/* Get seed from in-kernel /dev/urandom */
1086 		get_random_bytes(entropy, entropylen);
1087 
1088 		if (!drbg->jent) {
1089 			drbg_string_fill(&data1, entropy, entropylen);
1090 			pr_devel("DRBG: (re)seeding with %u bytes of entropy\n",
1091 				 entropylen);
1092 		} else {
1093 			/* Get seed from Jitter RNG */
1094 			ret = crypto_rng_get_bytes(drbg->jent,
1095 						   entropy + entropylen,
1096 						   entropylen);
1097 			if (ret) {
1098 				pr_devel("DRBG: jent failed with %d\n", ret);
1099 				return ret;
1100 			}
1101 
1102 			drbg_string_fill(&data1, entropy, entropylen * 2);
1103 			pr_devel("DRBG: (re)seeding with %u bytes of entropy\n",
1104 				 entropylen * 2);
1105 		}
1106 	}
1107 	list_add_tail(&data1.list, &seedlist);
1108 
1109 	/*
1110 	 * concatenation of entropy with personalization str / addtl input)
1111 	 * the variable pers is directly handed in by the caller, so check its
1112 	 * contents whether it is appropriate
1113 	 */
1114 	if (pers && pers->buf && 0 < pers->len) {
1115 		list_add_tail(&pers->list, &seedlist);
1116 		pr_devel("DRBG: using personalization string\n");
1117 	}
1118 
1119 	if (!reseed) {
1120 		memset(drbg->V, 0, drbg_statelen(drbg));
1121 		memset(drbg->C, 0, drbg_statelen(drbg));
1122 	}
1123 
1124 	ret = __drbg_seed(drbg, &seedlist, reseed);
1125 
1126 	memzero_explicit(entropy, entropylen * 2);
1127 
1128 	return ret;
1129 }
1130 
1131 /* Free all substructures in a DRBG state without the DRBG state structure */
1132 static inline void drbg_dealloc_state(struct drbg_state *drbg)
1133 {
1134 	if (!drbg)
1135 		return;
1136 	kzfree(drbg->V);
1137 	drbg->Vbuf = NULL;
1138 	kzfree(drbg->C);
1139 	drbg->Cbuf = NULL;
1140 	kzfree(drbg->scratchpadbuf);
1141 	drbg->scratchpadbuf = NULL;
1142 	drbg->reseed_ctr = 0;
1143 	drbg->d_ops = NULL;
1144 	drbg->core = NULL;
1145 }
1146 
1147 /*
1148  * Allocate all sub-structures for a DRBG state.
1149  * The DRBG state structure must already be allocated.
1150  */
1151 static inline int drbg_alloc_state(struct drbg_state *drbg)
1152 {
1153 	int ret = -ENOMEM;
1154 	unsigned int sb_size = 0;
1155 
1156 	switch (drbg->core->flags & DRBG_TYPE_MASK) {
1157 #ifdef CONFIG_CRYPTO_DRBG_HMAC
1158 	case DRBG_HMAC:
1159 		drbg->d_ops = &drbg_hmac_ops;
1160 		break;
1161 #endif /* CONFIG_CRYPTO_DRBG_HMAC */
1162 #ifdef CONFIG_CRYPTO_DRBG_HASH
1163 	case DRBG_HASH:
1164 		drbg->d_ops = &drbg_hash_ops;
1165 		break;
1166 #endif /* CONFIG_CRYPTO_DRBG_HASH */
1167 #ifdef CONFIG_CRYPTO_DRBG_CTR
1168 	case DRBG_CTR:
1169 		drbg->d_ops = &drbg_ctr_ops;
1170 		break;
1171 #endif /* CONFIG_CRYPTO_DRBG_CTR */
1172 	default:
1173 		ret = -EOPNOTSUPP;
1174 		goto err;
1175 	}
1176 
1177 	ret = drbg->d_ops->crypto_init(drbg);
1178 	if (ret < 0)
1179 		goto err;
1180 
1181 	drbg->Vbuf = kmalloc(drbg_statelen(drbg) + ret, GFP_KERNEL);
1182 	if (!drbg->Vbuf) {
1183 		ret = -ENOMEM;
1184 		goto fini;
1185 	}
1186 	drbg->V = PTR_ALIGN(drbg->Vbuf, ret + 1);
1187 	drbg->Cbuf = kmalloc(drbg_statelen(drbg) + ret, GFP_KERNEL);
1188 	if (!drbg->Cbuf) {
1189 		ret = -ENOMEM;
1190 		goto fini;
1191 	}
1192 	drbg->C = PTR_ALIGN(drbg->Cbuf, ret + 1);
1193 	/* scratchpad is only generated for CTR and Hash */
1194 	if (drbg->core->flags & DRBG_HMAC)
1195 		sb_size = 0;
1196 	else if (drbg->core->flags & DRBG_CTR)
1197 		sb_size = drbg_statelen(drbg) + drbg_blocklen(drbg) + /* temp */
1198 			  drbg_statelen(drbg) +	/* df_data */
1199 			  drbg_blocklen(drbg) +	/* pad */
1200 			  drbg_blocklen(drbg) +	/* iv */
1201 			  drbg_statelen(drbg) + drbg_blocklen(drbg); /* temp */
1202 	else
1203 		sb_size = drbg_statelen(drbg) + drbg_blocklen(drbg);
1204 
1205 	if (0 < sb_size) {
1206 		drbg->scratchpadbuf = kzalloc(sb_size + ret, GFP_KERNEL);
1207 		if (!drbg->scratchpadbuf) {
1208 			ret = -ENOMEM;
1209 			goto fini;
1210 		}
1211 		drbg->scratchpad = PTR_ALIGN(drbg->scratchpadbuf, ret + 1);
1212 	}
1213 
1214 	return 0;
1215 
1216 fini:
1217 	drbg->d_ops->crypto_fini(drbg);
1218 err:
1219 	drbg_dealloc_state(drbg);
1220 	return ret;
1221 }
1222 
1223 /*************************************************************************
1224  * DRBG interface functions
1225  *************************************************************************/
1226 
1227 /*
1228  * DRBG generate function as required by SP800-90A - this function
1229  * generates random numbers
1230  *
1231  * @drbg DRBG state handle
1232  * @buf Buffer where to store the random numbers -- the buffer must already
1233  *      be pre-allocated by caller
1234  * @buflen Length of output buffer - this value defines the number of random
1235  *	   bytes pulled from DRBG
1236  * @addtl Additional input that is mixed into state, may be NULL -- note
1237  *	  the entropy is pulled by the DRBG internally unconditionally
1238  *	  as defined in SP800-90A. The additional input is mixed into
1239  *	  the state in addition to the pulled entropy.
1240  *
1241  * return: 0 when all bytes are generated; < 0 in case of an error
1242  */
1243 static int drbg_generate(struct drbg_state *drbg,
1244 			 unsigned char *buf, unsigned int buflen,
1245 			 struct drbg_string *addtl)
1246 {
1247 	int len = 0;
1248 	LIST_HEAD(addtllist);
1249 
1250 	if (!drbg->core) {
1251 		pr_devel("DRBG: not yet seeded\n");
1252 		return -EINVAL;
1253 	}
1254 	if (0 == buflen || !buf) {
1255 		pr_devel("DRBG: no output buffer provided\n");
1256 		return -EINVAL;
1257 	}
1258 	if (addtl && NULL == addtl->buf && 0 < addtl->len) {
1259 		pr_devel("DRBG: wrong format of additional information\n");
1260 		return -EINVAL;
1261 	}
1262 
1263 	/* 9.3.1 step 2 */
1264 	len = -EINVAL;
1265 	if (buflen > (drbg_max_request_bytes(drbg))) {
1266 		pr_devel("DRBG: requested random numbers too large %u\n",
1267 			 buflen);
1268 		goto err;
1269 	}
1270 
1271 	/* 9.3.1 step 3 is implicit with the chosen DRBG */
1272 
1273 	/* 9.3.1 step 4 */
1274 	if (addtl && addtl->len > (drbg_max_addtl(drbg))) {
1275 		pr_devel("DRBG: additional information string too long %zu\n",
1276 			 addtl->len);
1277 		goto err;
1278 	}
1279 	/* 9.3.1 step 5 is implicit with the chosen DRBG */
1280 
1281 	/*
1282 	 * 9.3.1 step 6 and 9 supplemented by 9.3.2 step c is implemented
1283 	 * here. The spec is a bit convoluted here, we make it simpler.
1284 	 */
1285 	if (drbg->reseed_threshold < drbg->reseed_ctr)
1286 		drbg->seeded = false;
1287 
1288 	if (drbg->pr || !drbg->seeded) {
1289 		pr_devel("DRBG: reseeding before generation (prediction "
1290 			 "resistance: %s, state %s)\n",
1291 			 drbg->pr ? "true" : "false",
1292 			 drbg->seeded ? "seeded" : "unseeded");
1293 		/* 9.3.1 steps 7.1 through 7.3 */
1294 		len = drbg_seed(drbg, addtl, true);
1295 		if (len)
1296 			goto err;
1297 		/* 9.3.1 step 7.4 */
1298 		addtl = NULL;
1299 	}
1300 
1301 	if (addtl && 0 < addtl->len)
1302 		list_add_tail(&addtl->list, &addtllist);
1303 	/* 9.3.1 step 8 and 10 */
1304 	len = drbg->d_ops->generate(drbg, buf, buflen, &addtllist);
1305 
1306 	/* 10.1.1.4 step 6, 10.1.2.5 step 7, 10.2.1.5.2 step 7 */
1307 	drbg->reseed_ctr++;
1308 	if (0 >= len)
1309 		goto err;
1310 
1311 	/*
1312 	 * Section 11.3.3 requires to re-perform self tests after some
1313 	 * generated random numbers. The chosen value after which self
1314 	 * test is performed is arbitrary, but it should be reasonable.
1315 	 * However, we do not perform the self tests because of the following
1316 	 * reasons: it is mathematically impossible that the initial self tests
1317 	 * were successfully and the following are not. If the initial would
1318 	 * pass and the following would not, the kernel integrity is violated.
1319 	 * In this case, the entire kernel operation is questionable and it
1320 	 * is unlikely that the integrity violation only affects the
1321 	 * correct operation of the DRBG.
1322 	 *
1323 	 * Albeit the following code is commented out, it is provided in
1324 	 * case somebody has a need to implement the test of 11.3.3.
1325 	 */
1326 #if 0
1327 	if (drbg->reseed_ctr && !(drbg->reseed_ctr % 4096)) {
1328 		int err = 0;
1329 		pr_devel("DRBG: start to perform self test\n");
1330 		if (drbg->core->flags & DRBG_HMAC)
1331 			err = alg_test("drbg_pr_hmac_sha256",
1332 				       "drbg_pr_hmac_sha256", 0, 0);
1333 		else if (drbg->core->flags & DRBG_CTR)
1334 			err = alg_test("drbg_pr_ctr_aes128",
1335 				       "drbg_pr_ctr_aes128", 0, 0);
1336 		else
1337 			err = alg_test("drbg_pr_sha256",
1338 				       "drbg_pr_sha256", 0, 0);
1339 		if (err) {
1340 			pr_err("DRBG: periodical self test failed\n");
1341 			/*
1342 			 * uninstantiate implies that from now on, only errors
1343 			 * are returned when reusing this DRBG cipher handle
1344 			 */
1345 			drbg_uninstantiate(drbg);
1346 			return 0;
1347 		} else {
1348 			pr_devel("DRBG: self test successful\n");
1349 		}
1350 	}
1351 #endif
1352 
1353 	/*
1354 	 * All operations were successful, return 0 as mandated by
1355 	 * the kernel crypto API interface.
1356 	 */
1357 	len = 0;
1358 err:
1359 	return len;
1360 }
1361 
1362 /*
1363  * Wrapper around drbg_generate which can pull arbitrary long strings
1364  * from the DRBG without hitting the maximum request limitation.
1365  *
1366  * Parameters: see drbg_generate
1367  * Return codes: see drbg_generate -- if one drbg_generate request fails,
1368  *		 the entire drbg_generate_long request fails
1369  */
1370 static int drbg_generate_long(struct drbg_state *drbg,
1371 			      unsigned char *buf, unsigned int buflen,
1372 			      struct drbg_string *addtl)
1373 {
1374 	unsigned int len = 0;
1375 	unsigned int slice = 0;
1376 	do {
1377 		int err = 0;
1378 		unsigned int chunk = 0;
1379 		slice = ((buflen - len) / drbg_max_request_bytes(drbg));
1380 		chunk = slice ? drbg_max_request_bytes(drbg) : (buflen - len);
1381 		mutex_lock(&drbg->drbg_mutex);
1382 		err = drbg_generate(drbg, buf + len, chunk, addtl);
1383 		mutex_unlock(&drbg->drbg_mutex);
1384 		if (0 > err)
1385 			return err;
1386 		len += chunk;
1387 	} while (slice > 0 && (len < buflen));
1388 	return 0;
1389 }
1390 
1391 static void drbg_schedule_async_seed(struct random_ready_callback *rdy)
1392 {
1393 	struct drbg_state *drbg = container_of(rdy, struct drbg_state,
1394 					       random_ready);
1395 
1396 	schedule_work(&drbg->seed_work);
1397 }
1398 
1399 static int drbg_prepare_hrng(struct drbg_state *drbg)
1400 {
1401 	int err;
1402 
1403 	/* We do not need an HRNG in test mode. */
1404 	if (list_empty(&drbg->test_data.list))
1405 		return 0;
1406 
1407 	INIT_WORK(&drbg->seed_work, drbg_async_seed);
1408 
1409 	drbg->random_ready.owner = THIS_MODULE;
1410 	drbg->random_ready.func = drbg_schedule_async_seed;
1411 
1412 	err = add_random_ready_callback(&drbg->random_ready);
1413 
1414 	switch (err) {
1415 	case 0:
1416 		break;
1417 
1418 	case -EALREADY:
1419 		err = 0;
1420 		/* fall through */
1421 
1422 	default:
1423 		drbg->random_ready.func = NULL;
1424 		return err;
1425 	}
1426 
1427 	drbg->jent = crypto_alloc_rng("jitterentropy_rng", 0, 0);
1428 
1429 	/*
1430 	 * Require frequent reseeds until the seed source is fully
1431 	 * initialized.
1432 	 */
1433 	drbg->reseed_threshold = 50;
1434 
1435 	return err;
1436 }
1437 
1438 /*
1439  * DRBG instantiation function as required by SP800-90A - this function
1440  * sets up the DRBG handle, performs the initial seeding and all sanity
1441  * checks required by SP800-90A
1442  *
1443  * @drbg memory of state -- if NULL, new memory is allocated
1444  * @pers Personalization string that is mixed into state, may be NULL -- note
1445  *	 the entropy is pulled by the DRBG internally unconditionally
1446  *	 as defined in SP800-90A. The additional input is mixed into
1447  *	 the state in addition to the pulled entropy.
1448  * @coreref reference to core
1449  * @pr prediction resistance enabled
1450  *
1451  * return
1452  *	0 on success
1453  *	error value otherwise
1454  */
1455 static int drbg_instantiate(struct drbg_state *drbg, struct drbg_string *pers,
1456 			    int coreref, bool pr)
1457 {
1458 	int ret;
1459 	bool reseed = true;
1460 
1461 	pr_devel("DRBG: Initializing DRBG core %d with prediction resistance "
1462 		 "%s\n", coreref, pr ? "enabled" : "disabled");
1463 	mutex_lock(&drbg->drbg_mutex);
1464 
1465 	/* 9.1 step 1 is implicit with the selected DRBG type */
1466 
1467 	/*
1468 	 * 9.1 step 2 is implicit as caller can select prediction resistance
1469 	 * and the flag is copied into drbg->flags --
1470 	 * all DRBG types support prediction resistance
1471 	 */
1472 
1473 	/* 9.1 step 4 is implicit in  drbg_sec_strength */
1474 
1475 	if (!drbg->core) {
1476 		drbg->core = &drbg_cores[coreref];
1477 		drbg->pr = pr;
1478 		drbg->seeded = false;
1479 		drbg->reseed_threshold = drbg_max_requests(drbg);
1480 
1481 		ret = drbg_alloc_state(drbg);
1482 		if (ret)
1483 			goto unlock;
1484 
1485 		ret = drbg_prepare_hrng(drbg);
1486 		if (ret)
1487 			goto free_everything;
1488 
1489 		if (IS_ERR(drbg->jent)) {
1490 			ret = PTR_ERR(drbg->jent);
1491 			drbg->jent = NULL;
1492 			if (fips_enabled || ret != -ENOENT)
1493 				goto free_everything;
1494 			pr_info("DRBG: Continuing without Jitter RNG\n");
1495 		}
1496 
1497 		reseed = false;
1498 	}
1499 
1500 	ret = drbg_seed(drbg, pers, reseed);
1501 
1502 	if (ret && !reseed)
1503 		goto free_everything;
1504 
1505 	mutex_unlock(&drbg->drbg_mutex);
1506 	return ret;
1507 
1508 unlock:
1509 	mutex_unlock(&drbg->drbg_mutex);
1510 	return ret;
1511 
1512 free_everything:
1513 	mutex_unlock(&drbg->drbg_mutex);
1514 	drbg_uninstantiate(drbg);
1515 	return ret;
1516 }
1517 
1518 /*
1519  * DRBG uninstantiate function as required by SP800-90A - this function
1520  * frees all buffers and the DRBG handle
1521  *
1522  * @drbg DRBG state handle
1523  *
1524  * return
1525  *	0 on success
1526  */
1527 static int drbg_uninstantiate(struct drbg_state *drbg)
1528 {
1529 	if (drbg->random_ready.func) {
1530 		del_random_ready_callback(&drbg->random_ready);
1531 		cancel_work_sync(&drbg->seed_work);
1532 		crypto_free_rng(drbg->jent);
1533 		drbg->jent = NULL;
1534 	}
1535 
1536 	if (drbg->d_ops)
1537 		drbg->d_ops->crypto_fini(drbg);
1538 	drbg_dealloc_state(drbg);
1539 	/* no scrubbing of test_data -- this shall survive an uninstantiate */
1540 	return 0;
1541 }
1542 
1543 /*
1544  * Helper function for setting the test data in the DRBG
1545  *
1546  * @drbg DRBG state handle
1547  * @data test data
1548  * @len test data length
1549  */
1550 static void drbg_kcapi_set_entropy(struct crypto_rng *tfm,
1551 				   const u8 *data, unsigned int len)
1552 {
1553 	struct drbg_state *drbg = crypto_rng_ctx(tfm);
1554 
1555 	mutex_lock(&drbg->drbg_mutex);
1556 	drbg_string_fill(&drbg->test_data, data, len);
1557 	mutex_unlock(&drbg->drbg_mutex);
1558 }
1559 
1560 /***************************************************************
1561  * Kernel crypto API cipher invocations requested by DRBG
1562  ***************************************************************/
1563 
1564 #if defined(CONFIG_CRYPTO_DRBG_HASH) || defined(CONFIG_CRYPTO_DRBG_HMAC)
1565 struct sdesc {
1566 	struct shash_desc shash;
1567 	char ctx[];
1568 };
1569 
1570 static int drbg_init_hash_kernel(struct drbg_state *drbg)
1571 {
1572 	struct sdesc *sdesc;
1573 	struct crypto_shash *tfm;
1574 
1575 	tfm = crypto_alloc_shash(drbg->core->backend_cra_name, 0, 0);
1576 	if (IS_ERR(tfm)) {
1577 		pr_info("DRBG: could not allocate digest TFM handle: %s\n",
1578 				drbg->core->backend_cra_name);
1579 		return PTR_ERR(tfm);
1580 	}
1581 	BUG_ON(drbg_blocklen(drbg) != crypto_shash_digestsize(tfm));
1582 	sdesc = kzalloc(sizeof(struct shash_desc) + crypto_shash_descsize(tfm),
1583 			GFP_KERNEL);
1584 	if (!sdesc) {
1585 		crypto_free_shash(tfm);
1586 		return -ENOMEM;
1587 	}
1588 
1589 	sdesc->shash.tfm = tfm;
1590 	sdesc->shash.flags = 0;
1591 	drbg->priv_data = sdesc;
1592 
1593 	return crypto_shash_alignmask(tfm);
1594 }
1595 
1596 static int drbg_fini_hash_kernel(struct drbg_state *drbg)
1597 {
1598 	struct sdesc *sdesc = (struct sdesc *)drbg->priv_data;
1599 	if (sdesc) {
1600 		crypto_free_shash(sdesc->shash.tfm);
1601 		kzfree(sdesc);
1602 	}
1603 	drbg->priv_data = NULL;
1604 	return 0;
1605 }
1606 
1607 static void drbg_kcapi_hmacsetkey(struct drbg_state *drbg,
1608 				  const unsigned char *key)
1609 {
1610 	struct sdesc *sdesc = (struct sdesc *)drbg->priv_data;
1611 
1612 	crypto_shash_setkey(sdesc->shash.tfm, key, drbg_statelen(drbg));
1613 }
1614 
1615 static int drbg_kcapi_hash(struct drbg_state *drbg, unsigned char *outval,
1616 			   const struct list_head *in)
1617 {
1618 	struct sdesc *sdesc = (struct sdesc *)drbg->priv_data;
1619 	struct drbg_string *input = NULL;
1620 
1621 	crypto_shash_init(&sdesc->shash);
1622 	list_for_each_entry(input, in, list)
1623 		crypto_shash_update(&sdesc->shash, input->buf, input->len);
1624 	return crypto_shash_final(&sdesc->shash, outval);
1625 }
1626 #endif /* (CONFIG_CRYPTO_DRBG_HASH || CONFIG_CRYPTO_DRBG_HMAC) */
1627 
1628 #ifdef CONFIG_CRYPTO_DRBG_CTR
1629 static int drbg_fini_sym_kernel(struct drbg_state *drbg)
1630 {
1631 	struct crypto_cipher *tfm =
1632 		(struct crypto_cipher *)drbg->priv_data;
1633 	if (tfm)
1634 		crypto_free_cipher(tfm);
1635 	drbg->priv_data = NULL;
1636 
1637 	if (drbg->ctr_handle)
1638 		crypto_free_skcipher(drbg->ctr_handle);
1639 	drbg->ctr_handle = NULL;
1640 
1641 	if (drbg->ctr_req)
1642 		skcipher_request_free(drbg->ctr_req);
1643 	drbg->ctr_req = NULL;
1644 
1645 	kfree(drbg->ctr_null_value_buf);
1646 	drbg->ctr_null_value = NULL;
1647 
1648 	kfree(drbg->outscratchpadbuf);
1649 	drbg->outscratchpadbuf = NULL;
1650 
1651 	return 0;
1652 }
1653 
1654 static void drbg_skcipher_cb(struct crypto_async_request *req, int error)
1655 {
1656 	struct drbg_state *drbg = req->data;
1657 
1658 	if (error == -EINPROGRESS)
1659 		return;
1660 	drbg->ctr_async_err = error;
1661 	complete(&drbg->ctr_completion);
1662 }
1663 
1664 static int drbg_init_sym_kernel(struct drbg_state *drbg)
1665 {
1666 	struct crypto_cipher *tfm;
1667 	struct crypto_skcipher *sk_tfm;
1668 	struct skcipher_request *req;
1669 	unsigned int alignmask;
1670 	char ctr_name[CRYPTO_MAX_ALG_NAME];
1671 
1672 	tfm = crypto_alloc_cipher(drbg->core->backend_cra_name, 0, 0);
1673 	if (IS_ERR(tfm)) {
1674 		pr_info("DRBG: could not allocate cipher TFM handle: %s\n",
1675 				drbg->core->backend_cra_name);
1676 		return PTR_ERR(tfm);
1677 	}
1678 	BUG_ON(drbg_blocklen(drbg) != crypto_cipher_blocksize(tfm));
1679 	drbg->priv_data = tfm;
1680 
1681 	if (snprintf(ctr_name, CRYPTO_MAX_ALG_NAME, "ctr(%s)",
1682 	    drbg->core->backend_cra_name) >= CRYPTO_MAX_ALG_NAME) {
1683 		drbg_fini_sym_kernel(drbg);
1684 		return -EINVAL;
1685 	}
1686 	sk_tfm = crypto_alloc_skcipher(ctr_name, 0, 0);
1687 	if (IS_ERR(sk_tfm)) {
1688 		pr_info("DRBG: could not allocate CTR cipher TFM handle: %s\n",
1689 				ctr_name);
1690 		drbg_fini_sym_kernel(drbg);
1691 		return PTR_ERR(sk_tfm);
1692 	}
1693 	drbg->ctr_handle = sk_tfm;
1694 	init_completion(&drbg->ctr_completion);
1695 
1696 	req = skcipher_request_alloc(sk_tfm, GFP_KERNEL);
1697 	if (!req) {
1698 		pr_info("DRBG: could not allocate request queue\n");
1699 		drbg_fini_sym_kernel(drbg);
1700 		return -ENOMEM;
1701 	}
1702 	drbg->ctr_req = req;
1703 	skcipher_request_set_callback(req, CRYPTO_TFM_REQ_MAY_BACKLOG,
1704 					drbg_skcipher_cb, drbg);
1705 
1706 	alignmask = crypto_skcipher_alignmask(sk_tfm);
1707 	drbg->ctr_null_value_buf = kzalloc(DRBG_CTR_NULL_LEN + alignmask,
1708 					   GFP_KERNEL);
1709 	if (!drbg->ctr_null_value_buf) {
1710 		drbg_fini_sym_kernel(drbg);
1711 		return -ENOMEM;
1712 	}
1713 	drbg->ctr_null_value = (u8 *)PTR_ALIGN(drbg->ctr_null_value_buf,
1714 					       alignmask + 1);
1715 
1716 	drbg->outscratchpadbuf = kmalloc(DRBG_OUTSCRATCHLEN + alignmask,
1717 					 GFP_KERNEL);
1718 	if (!drbg->outscratchpadbuf) {
1719 		drbg_fini_sym_kernel(drbg);
1720 		return -ENOMEM;
1721 	}
1722 	drbg->outscratchpad = (u8 *)PTR_ALIGN(drbg->outscratchpadbuf,
1723 					      alignmask + 1);
1724 
1725 	return alignmask;
1726 }
1727 
1728 static void drbg_kcapi_symsetkey(struct drbg_state *drbg,
1729 				 const unsigned char *key)
1730 {
1731 	struct crypto_cipher *tfm =
1732 		(struct crypto_cipher *)drbg->priv_data;
1733 
1734 	crypto_cipher_setkey(tfm, key, (drbg_keylen(drbg)));
1735 }
1736 
1737 static int drbg_kcapi_sym(struct drbg_state *drbg, unsigned char *outval,
1738 			  const struct drbg_string *in)
1739 {
1740 	struct crypto_cipher *tfm =
1741 		(struct crypto_cipher *)drbg->priv_data;
1742 
1743 	/* there is only component in *in */
1744 	BUG_ON(in->len < drbg_blocklen(drbg));
1745 	crypto_cipher_encrypt_one(tfm, outval, in->buf);
1746 	return 0;
1747 }
1748 
1749 static int drbg_kcapi_sym_ctr(struct drbg_state *drbg,
1750 			      u8 *inbuf, u32 inlen,
1751 			      u8 *outbuf, u32 outlen)
1752 {
1753 	struct scatterlist sg_in, sg_out;
1754 	int ret;
1755 
1756 	sg_init_one(&sg_in, inbuf, inlen);
1757 	sg_init_one(&sg_out, drbg->outscratchpad, DRBG_OUTSCRATCHLEN);
1758 
1759 	while (outlen) {
1760 		u32 cryptlen = min3(inlen, outlen, (u32)DRBG_OUTSCRATCHLEN);
1761 
1762 		/* Output buffer may not be valid for SGL, use scratchpad */
1763 		skcipher_request_set_crypt(drbg->ctr_req, &sg_in, &sg_out,
1764 					   cryptlen, drbg->V);
1765 		ret = crypto_skcipher_encrypt(drbg->ctr_req);
1766 		switch (ret) {
1767 		case 0:
1768 			break;
1769 		case -EINPROGRESS:
1770 		case -EBUSY:
1771 			wait_for_completion(&drbg->ctr_completion);
1772 			if (!drbg->ctr_async_err) {
1773 				reinit_completion(&drbg->ctr_completion);
1774 				break;
1775 			}
1776 		default:
1777 			goto out;
1778 		}
1779 		init_completion(&drbg->ctr_completion);
1780 
1781 		memcpy(outbuf, drbg->outscratchpad, cryptlen);
1782 
1783 		outlen -= cryptlen;
1784 		outbuf += cryptlen;
1785 	}
1786 	ret = 0;
1787 
1788 out:
1789 	memzero_explicit(drbg->outscratchpad, DRBG_OUTSCRATCHLEN);
1790 	return ret;
1791 }
1792 #endif /* CONFIG_CRYPTO_DRBG_CTR */
1793 
1794 /***************************************************************
1795  * Kernel crypto API interface to register DRBG
1796  ***************************************************************/
1797 
1798 /*
1799  * Look up the DRBG flags by given kernel crypto API cra_name
1800  * The code uses the drbg_cores definition to do this
1801  *
1802  * @cra_name kernel crypto API cra_name
1803  * @coreref reference to integer which is filled with the pointer to
1804  *  the applicable core
1805  * @pr reference for setting prediction resistance
1806  *
1807  * return: flags
1808  */
1809 static inline void drbg_convert_tfm_core(const char *cra_driver_name,
1810 					 int *coreref, bool *pr)
1811 {
1812 	int i = 0;
1813 	size_t start = 0;
1814 	int len = 0;
1815 
1816 	*pr = true;
1817 	/* disassemble the names */
1818 	if (!memcmp(cra_driver_name, "drbg_nopr_", 10)) {
1819 		start = 10;
1820 		*pr = false;
1821 	} else if (!memcmp(cra_driver_name, "drbg_pr_", 8)) {
1822 		start = 8;
1823 	} else {
1824 		return;
1825 	}
1826 
1827 	/* remove the first part */
1828 	len = strlen(cra_driver_name) - start;
1829 	for (i = 0; ARRAY_SIZE(drbg_cores) > i; i++) {
1830 		if (!memcmp(cra_driver_name + start, drbg_cores[i].cra_name,
1831 			    len)) {
1832 			*coreref = i;
1833 			return;
1834 		}
1835 	}
1836 }
1837 
1838 static int drbg_kcapi_init(struct crypto_tfm *tfm)
1839 {
1840 	struct drbg_state *drbg = crypto_tfm_ctx(tfm);
1841 
1842 	mutex_init(&drbg->drbg_mutex);
1843 
1844 	return 0;
1845 }
1846 
1847 static void drbg_kcapi_cleanup(struct crypto_tfm *tfm)
1848 {
1849 	drbg_uninstantiate(crypto_tfm_ctx(tfm));
1850 }
1851 
1852 /*
1853  * Generate random numbers invoked by the kernel crypto API:
1854  * The API of the kernel crypto API is extended as follows:
1855  *
1856  * src is additional input supplied to the RNG.
1857  * slen is the length of src.
1858  * dst is the output buffer where random data is to be stored.
1859  * dlen is the length of dst.
1860  */
1861 static int drbg_kcapi_random(struct crypto_rng *tfm,
1862 			     const u8 *src, unsigned int slen,
1863 			     u8 *dst, unsigned int dlen)
1864 {
1865 	struct drbg_state *drbg = crypto_rng_ctx(tfm);
1866 	struct drbg_string *addtl = NULL;
1867 	struct drbg_string string;
1868 
1869 	if (slen) {
1870 		/* linked list variable is now local to allow modification */
1871 		drbg_string_fill(&string, src, slen);
1872 		addtl = &string;
1873 	}
1874 
1875 	return drbg_generate_long(drbg, dst, dlen, addtl);
1876 }
1877 
1878 /*
1879  * Seed the DRBG invoked by the kernel crypto API
1880  */
1881 static int drbg_kcapi_seed(struct crypto_rng *tfm,
1882 			   const u8 *seed, unsigned int slen)
1883 {
1884 	struct drbg_state *drbg = crypto_rng_ctx(tfm);
1885 	struct crypto_tfm *tfm_base = crypto_rng_tfm(tfm);
1886 	bool pr = false;
1887 	struct drbg_string string;
1888 	struct drbg_string *seed_string = NULL;
1889 	int coreref = 0;
1890 
1891 	drbg_convert_tfm_core(crypto_tfm_alg_driver_name(tfm_base), &coreref,
1892 			      &pr);
1893 	if (0 < slen) {
1894 		drbg_string_fill(&string, seed, slen);
1895 		seed_string = &string;
1896 	}
1897 
1898 	return drbg_instantiate(drbg, seed_string, coreref, pr);
1899 }
1900 
1901 /***************************************************************
1902  * Kernel module: code to load the module
1903  ***************************************************************/
1904 
1905 /*
1906  * Tests as defined in 11.3.2 in addition to the cipher tests: testing
1907  * of the error handling.
1908  *
1909  * Note: testing of failing seed source as defined in 11.3.2 is not applicable
1910  * as seed source of get_random_bytes does not fail.
1911  *
1912  * Note 2: There is no sensible way of testing the reseed counter
1913  * enforcement, so skip it.
1914  */
1915 static inline int __init drbg_healthcheck_sanity(void)
1916 {
1917 	int len = 0;
1918 #define OUTBUFLEN 16
1919 	unsigned char buf[OUTBUFLEN];
1920 	struct drbg_state *drbg = NULL;
1921 	int ret = -EFAULT;
1922 	int rc = -EFAULT;
1923 	bool pr = false;
1924 	int coreref = 0;
1925 	struct drbg_string addtl;
1926 	size_t max_addtllen, max_request_bytes;
1927 
1928 	/* only perform test in FIPS mode */
1929 	if (!fips_enabled)
1930 		return 0;
1931 
1932 #ifdef CONFIG_CRYPTO_DRBG_CTR
1933 	drbg_convert_tfm_core("drbg_nopr_ctr_aes128", &coreref, &pr);
1934 #elif defined CONFIG_CRYPTO_DRBG_HASH
1935 	drbg_convert_tfm_core("drbg_nopr_sha256", &coreref, &pr);
1936 #else
1937 	drbg_convert_tfm_core("drbg_nopr_hmac_sha256", &coreref, &pr);
1938 #endif
1939 
1940 	drbg = kzalloc(sizeof(struct drbg_state), GFP_KERNEL);
1941 	if (!drbg)
1942 		return -ENOMEM;
1943 
1944 	mutex_init(&drbg->drbg_mutex);
1945 	drbg->core = &drbg_cores[coreref];
1946 	drbg->reseed_threshold = drbg_max_requests(drbg);
1947 
1948 	/*
1949 	 * if the following tests fail, it is likely that there is a buffer
1950 	 * overflow as buf is much smaller than the requested or provided
1951 	 * string lengths -- in case the error handling does not succeed
1952 	 * we may get an OOPS. And we want to get an OOPS as this is a
1953 	 * grave bug.
1954 	 */
1955 
1956 	max_addtllen = drbg_max_addtl(drbg);
1957 	max_request_bytes = drbg_max_request_bytes(drbg);
1958 	drbg_string_fill(&addtl, buf, max_addtllen + 1);
1959 	/* overflow addtllen with additonal info string */
1960 	len = drbg_generate(drbg, buf, OUTBUFLEN, &addtl);
1961 	BUG_ON(0 < len);
1962 	/* overflow max_bits */
1963 	len = drbg_generate(drbg, buf, (max_request_bytes + 1), NULL);
1964 	BUG_ON(0 < len);
1965 
1966 	/* overflow max addtllen with personalization string */
1967 	ret = drbg_seed(drbg, &addtl, false);
1968 	BUG_ON(0 == ret);
1969 	/* all tests passed */
1970 	rc = 0;
1971 
1972 	pr_devel("DRBG: Sanity tests for failure code paths successfully "
1973 		 "completed\n");
1974 
1975 	kfree(drbg);
1976 	return rc;
1977 }
1978 
1979 static struct rng_alg drbg_algs[22];
1980 
1981 /*
1982  * Fill the array drbg_algs used to register the different DRBGs
1983  * with the kernel crypto API. To fill the array, the information
1984  * from drbg_cores[] is used.
1985  */
1986 static inline void __init drbg_fill_array(struct rng_alg *alg,
1987 					  const struct drbg_core *core, int pr)
1988 {
1989 	int pos = 0;
1990 	static int priority = 200;
1991 
1992 	memcpy(alg->base.cra_name, "stdrng", 6);
1993 	if (pr) {
1994 		memcpy(alg->base.cra_driver_name, "drbg_pr_", 8);
1995 		pos = 8;
1996 	} else {
1997 		memcpy(alg->base.cra_driver_name, "drbg_nopr_", 10);
1998 		pos = 10;
1999 	}
2000 	memcpy(alg->base.cra_driver_name + pos, core->cra_name,
2001 	       strlen(core->cra_name));
2002 
2003 	alg->base.cra_priority = priority;
2004 	priority++;
2005 	/*
2006 	 * If FIPS mode enabled, the selected DRBG shall have the
2007 	 * highest cra_priority over other stdrng instances to ensure
2008 	 * it is selected.
2009 	 */
2010 	if (fips_enabled)
2011 		alg->base.cra_priority += 200;
2012 
2013 	alg->base.cra_ctxsize 	= sizeof(struct drbg_state);
2014 	alg->base.cra_module	= THIS_MODULE;
2015 	alg->base.cra_init	= drbg_kcapi_init;
2016 	alg->base.cra_exit	= drbg_kcapi_cleanup;
2017 	alg->generate		= drbg_kcapi_random;
2018 	alg->seed		= drbg_kcapi_seed;
2019 	alg->set_ent		= drbg_kcapi_set_entropy;
2020 	alg->seedsize		= 0;
2021 }
2022 
2023 static int __init drbg_init(void)
2024 {
2025 	unsigned int i = 0; /* pointer to drbg_algs */
2026 	unsigned int j = 0; /* pointer to drbg_cores */
2027 	int ret;
2028 
2029 	ret = drbg_healthcheck_sanity();
2030 	if (ret)
2031 		return ret;
2032 
2033 	if (ARRAY_SIZE(drbg_cores) * 2 > ARRAY_SIZE(drbg_algs)) {
2034 		pr_info("DRBG: Cannot register all DRBG types"
2035 			"(slots needed: %zu, slots available: %zu)\n",
2036 			ARRAY_SIZE(drbg_cores) * 2, ARRAY_SIZE(drbg_algs));
2037 		return -EFAULT;
2038 	}
2039 
2040 	/*
2041 	 * each DRBG definition can be used with PR and without PR, thus
2042 	 * we instantiate each DRBG in drbg_cores[] twice.
2043 	 *
2044 	 * As the order of placing them into the drbg_algs array matters
2045 	 * (the later DRBGs receive a higher cra_priority) we register the
2046 	 * prediction resistance DRBGs first as the should not be too
2047 	 * interesting.
2048 	 */
2049 	for (j = 0; ARRAY_SIZE(drbg_cores) > j; j++, i++)
2050 		drbg_fill_array(&drbg_algs[i], &drbg_cores[j], 1);
2051 	for (j = 0; ARRAY_SIZE(drbg_cores) > j; j++, i++)
2052 		drbg_fill_array(&drbg_algs[i], &drbg_cores[j], 0);
2053 	return crypto_register_rngs(drbg_algs, (ARRAY_SIZE(drbg_cores) * 2));
2054 }
2055 
2056 static void __exit drbg_exit(void)
2057 {
2058 	crypto_unregister_rngs(drbg_algs, (ARRAY_SIZE(drbg_cores) * 2));
2059 }
2060 
2061 module_init(drbg_init);
2062 module_exit(drbg_exit);
2063 #ifndef CRYPTO_DRBG_HASH_STRING
2064 #define CRYPTO_DRBG_HASH_STRING ""
2065 #endif
2066 #ifndef CRYPTO_DRBG_HMAC_STRING
2067 #define CRYPTO_DRBG_HMAC_STRING ""
2068 #endif
2069 #ifndef CRYPTO_DRBG_CTR_STRING
2070 #define CRYPTO_DRBG_CTR_STRING ""
2071 #endif
2072 MODULE_LICENSE("GPL");
2073 MODULE_AUTHOR("Stephan Mueller <smueller@chronox.de>");
2074 MODULE_DESCRIPTION("NIST SP800-90A Deterministic Random Bit Generator (DRBG) "
2075 		   "using following cores: "
2076 		   CRYPTO_DRBG_HASH_STRING
2077 		   CRYPTO_DRBG_HMAC_STRING
2078 		   CRYPTO_DRBG_CTR_STRING);
2079 MODULE_ALIAS_CRYPTO("stdrng");
2080