xref: /linux/crypto/ahash.c (revision a3f8e00991f96ee7c3c77834258baa06a39fbbff)
1 // SPDX-License-Identifier: GPL-2.0-or-later
2 /*
3  * Asynchronous Cryptographic Hash operations.
4  *
5  * This is the implementation of the ahash (asynchronous hash) API.  It differs
6  * from shash (synchronous hash) in that ahash supports asynchronous operations,
7  * and it hashes data from scatterlists instead of virtually addressed buffers.
8  *
9  * The ahash API provides access to both ahash and shash algorithms.  The shash
10  * API only provides access to shash algorithms.
11  *
12  * Copyright (c) 2008 Loc Ho <lho@amcc.com>
13  */
14 
15 #include <crypto/scatterwalk.h>
16 #include <linux/cryptouser.h>
17 #include <linux/err.h>
18 #include <linux/kernel.h>
19 #include <linux/mm.h>
20 #include <linux/module.h>
21 #include <linux/scatterlist.h>
22 #include <linux/slab.h>
23 #include <linux/seq_file.h>
24 #include <linux/string.h>
25 #include <linux/string_choices.h>
26 #include <net/netlink.h>
27 
28 #include "hash.h"
29 
30 #define CRYPTO_ALG_TYPE_AHASH_MASK	0x0000000e
31 
32 static int ahash_def_finup(struct ahash_request *req);
33 
34 static inline bool crypto_ahash_block_only(struct crypto_ahash *tfm)
35 {
36 	return crypto_ahash_alg(tfm)->halg.base.cra_flags &
37 	       CRYPTO_AHASH_ALG_BLOCK_ONLY;
38 }
39 
40 static inline bool crypto_ahash_final_nonzero(struct crypto_ahash *tfm)
41 {
42 	return crypto_ahash_alg(tfm)->halg.base.cra_flags &
43 	       CRYPTO_AHASH_ALG_FINAL_NONZERO;
44 }
45 
46 static inline bool crypto_ahash_need_fallback(struct crypto_ahash *tfm)
47 {
48 	return crypto_ahash_alg(tfm)->halg.base.cra_flags &
49 	       CRYPTO_ALG_NEED_FALLBACK;
50 }
51 
52 static inline void ahash_op_done(void *data, int err,
53 				 int (*finish)(struct ahash_request *, int))
54 {
55 	struct ahash_request *areq = data;
56 	crypto_completion_t compl;
57 
58 	compl = areq->saved_complete;
59 	data = areq->saved_data;
60 	if (err == -EINPROGRESS)
61 		goto out;
62 
63 	areq->base.flags &= ~CRYPTO_TFM_REQ_MAY_SLEEP;
64 
65 	err = finish(areq, err);
66 	if (err == -EINPROGRESS || err == -EBUSY)
67 		return;
68 
69 out:
70 	compl(data, err);
71 }
72 
73 static int hash_walk_next(struct crypto_hash_walk *walk)
74 {
75 	unsigned int offset = walk->offset;
76 	unsigned int nbytes = min(walk->entrylen,
77 				  ((unsigned int)(PAGE_SIZE)) - offset);
78 
79 	walk->data = kmap_local_page(walk->pg);
80 	walk->data += offset;
81 	walk->entrylen -= nbytes;
82 	return nbytes;
83 }
84 
85 static int hash_walk_new_entry(struct crypto_hash_walk *walk)
86 {
87 	struct scatterlist *sg;
88 
89 	sg = walk->sg;
90 	walk->offset = sg->offset;
91 	walk->pg = sg_page(walk->sg) + (walk->offset >> PAGE_SHIFT);
92 	walk->offset = offset_in_page(walk->offset);
93 	walk->entrylen = sg->length;
94 
95 	if (walk->entrylen > walk->total)
96 		walk->entrylen = walk->total;
97 	walk->total -= walk->entrylen;
98 
99 	return hash_walk_next(walk);
100 }
101 
102 int crypto_hash_walk_first(struct ahash_request *req,
103 			   struct crypto_hash_walk *walk)
104 {
105 	walk->total = req->nbytes;
106 	walk->entrylen = 0;
107 
108 	if (!walk->total)
109 		return 0;
110 
111 	walk->flags = req->base.flags;
112 
113 	if (ahash_request_isvirt(req)) {
114 		walk->data = req->svirt;
115 		walk->total = 0;
116 		return req->nbytes;
117 	}
118 
119 	walk->sg = req->src;
120 
121 	return hash_walk_new_entry(walk);
122 }
123 EXPORT_SYMBOL_GPL(crypto_hash_walk_first);
124 
125 int crypto_hash_walk_done(struct crypto_hash_walk *walk, int err)
126 {
127 	if ((walk->flags & CRYPTO_AHASH_REQ_VIRT))
128 		return err;
129 
130 	walk->data -= walk->offset;
131 
132 	kunmap_local(walk->data);
133 	crypto_yield(walk->flags);
134 
135 	if (err)
136 		return err;
137 
138 	if (walk->entrylen) {
139 		walk->offset = 0;
140 		walk->pg++;
141 		return hash_walk_next(walk);
142 	}
143 
144 	if (!walk->total)
145 		return 0;
146 
147 	walk->sg = sg_next(walk->sg);
148 
149 	return hash_walk_new_entry(walk);
150 }
151 EXPORT_SYMBOL_GPL(crypto_hash_walk_done);
152 
153 /*
154  * For an ahash tfm that is using an shash algorithm (instead of an ahash
155  * algorithm), this returns the underlying shash tfm.
156  */
157 static inline struct crypto_shash *ahash_to_shash(struct crypto_ahash *tfm)
158 {
159 	return *(struct crypto_shash **)crypto_ahash_ctx(tfm);
160 }
161 
162 static inline struct shash_desc *prepare_shash_desc(struct ahash_request *req,
163 						    struct crypto_ahash *tfm)
164 {
165 	struct shash_desc *desc = ahash_request_ctx(req);
166 
167 	desc->tfm = ahash_to_shash(tfm);
168 	return desc;
169 }
170 
171 int shash_ahash_update(struct ahash_request *req, struct shash_desc *desc)
172 {
173 	struct crypto_hash_walk walk;
174 	int nbytes;
175 
176 	for (nbytes = crypto_hash_walk_first(req, &walk); nbytes > 0;
177 	     nbytes = crypto_hash_walk_done(&walk, nbytes))
178 		nbytes = crypto_shash_update(desc, walk.data, nbytes);
179 
180 	return nbytes;
181 }
182 EXPORT_SYMBOL_GPL(shash_ahash_update);
183 
184 int shash_ahash_finup(struct ahash_request *req, struct shash_desc *desc)
185 {
186 	struct crypto_hash_walk walk;
187 	int nbytes;
188 
189 	nbytes = crypto_hash_walk_first(req, &walk);
190 	if (!nbytes)
191 		return crypto_shash_final(desc, req->result);
192 
193 	do {
194 		nbytes = crypto_hash_walk_last(&walk) ?
195 			 crypto_shash_finup(desc, walk.data, nbytes,
196 					    req->result) :
197 			 crypto_shash_update(desc, walk.data, nbytes);
198 		nbytes = crypto_hash_walk_done(&walk, nbytes);
199 	} while (nbytes > 0);
200 
201 	return nbytes;
202 }
203 EXPORT_SYMBOL_GPL(shash_ahash_finup);
204 
205 int shash_ahash_digest(struct ahash_request *req, struct shash_desc *desc)
206 {
207 	unsigned int nbytes = req->nbytes;
208 	struct scatterlist *sg;
209 	unsigned int offset;
210 	struct page *page;
211 	const u8 *data;
212 	int err;
213 
214 	data = req->svirt;
215 	if (!nbytes || ahash_request_isvirt(req))
216 		return crypto_shash_digest(desc, data, nbytes, req->result);
217 
218 	sg = req->src;
219 	if (nbytes > sg->length)
220 		return crypto_shash_init(desc) ?:
221 		       shash_ahash_finup(req, desc);
222 
223 	page = sg_page(sg);
224 	offset = sg->offset;
225 	data = lowmem_page_address(page) + offset;
226 	if (!IS_ENABLED(CONFIG_HIGHMEM))
227 		return crypto_shash_digest(desc, data, nbytes, req->result);
228 
229 	page += offset >> PAGE_SHIFT;
230 	offset = offset_in_page(offset);
231 
232 	if (nbytes > (unsigned int)PAGE_SIZE - offset)
233 		return crypto_shash_init(desc) ?:
234 		       shash_ahash_finup(req, desc);
235 
236 	data = kmap_local_page(page);
237 	err = crypto_shash_digest(desc, data + offset, nbytes,
238 				  req->result);
239 	kunmap_local(data);
240 	return err;
241 }
242 EXPORT_SYMBOL_GPL(shash_ahash_digest);
243 
244 static void crypto_exit_ahash_using_shash(struct crypto_tfm *tfm)
245 {
246 	struct crypto_shash **ctx = crypto_tfm_ctx(tfm);
247 
248 	crypto_free_shash(*ctx);
249 }
250 
251 static int crypto_init_ahash_using_shash(struct crypto_tfm *tfm)
252 {
253 	struct crypto_alg *calg = tfm->__crt_alg;
254 	struct crypto_ahash *crt = __crypto_ahash_cast(tfm);
255 	struct crypto_shash **ctx = crypto_tfm_ctx(tfm);
256 	struct crypto_shash *shash;
257 
258 	if (!crypto_mod_get(calg))
259 		return -EAGAIN;
260 
261 	shash = crypto_create_tfm(calg, &crypto_shash_type);
262 	if (IS_ERR(shash)) {
263 		crypto_mod_put(calg);
264 		return PTR_ERR(shash);
265 	}
266 
267 	crt->using_shash = true;
268 	*ctx = shash;
269 	tfm->exit = crypto_exit_ahash_using_shash;
270 
271 	crypto_ahash_set_flags(crt, crypto_shash_get_flags(shash) &
272 				    CRYPTO_TFM_NEED_KEY);
273 
274 	return 0;
275 }
276 
277 static int ahash_nosetkey(struct crypto_ahash *tfm, const u8 *key,
278 			  unsigned int keylen)
279 {
280 	return -ENOSYS;
281 }
282 
283 static void ahash_set_needkey(struct crypto_ahash *tfm, struct ahash_alg *alg)
284 {
285 	if (alg->setkey != ahash_nosetkey &&
286 	    !(alg->halg.base.cra_flags & CRYPTO_ALG_OPTIONAL_KEY))
287 		crypto_ahash_set_flags(tfm, CRYPTO_TFM_NEED_KEY);
288 }
289 
290 int crypto_ahash_setkey(struct crypto_ahash *tfm, const u8 *key,
291 			unsigned int keylen)
292 {
293 	if (likely(tfm->using_shash)) {
294 		struct crypto_shash *shash = ahash_to_shash(tfm);
295 		int err;
296 
297 		err = crypto_shash_setkey(shash, key, keylen);
298 		if (unlikely(err)) {
299 			crypto_ahash_set_flags(tfm,
300 					       crypto_shash_get_flags(shash) &
301 					       CRYPTO_TFM_NEED_KEY);
302 			return err;
303 		}
304 	} else {
305 		struct ahash_alg *alg = crypto_ahash_alg(tfm);
306 		int err;
307 
308 		err = alg->setkey(tfm, key, keylen);
309 		if (!err && crypto_ahash_need_fallback(tfm))
310 			err = crypto_ahash_setkey(crypto_ahash_fb(tfm),
311 						  key, keylen);
312 		if (unlikely(err)) {
313 			ahash_set_needkey(tfm, alg);
314 			return err;
315 		}
316 	}
317 	crypto_ahash_clear_flags(tfm, CRYPTO_TFM_NEED_KEY);
318 	return 0;
319 }
320 EXPORT_SYMBOL_GPL(crypto_ahash_setkey);
321 
322 static int ahash_do_req_chain(struct ahash_request *req,
323 			      int (*const *op)(struct ahash_request *req))
324 {
325 	struct crypto_ahash *tfm = crypto_ahash_reqtfm(req);
326 	int err;
327 
328 	if (crypto_ahash_req_virt(tfm) || !ahash_request_isvirt(req))
329 		return (*op)(req);
330 
331 	if (crypto_ahash_statesize(tfm) > HASH_MAX_STATESIZE)
332 		return -ENOSYS;
333 
334 	if (!crypto_ahash_need_fallback(tfm))
335 		return -ENOSYS;
336 
337 	if (crypto_hash_no_export_core(tfm))
338 		return -ENOSYS;
339 
340 	{
341 		u8 state[HASH_MAX_STATESIZE];
342 
343 		if (op == &crypto_ahash_alg(tfm)->digest) {
344 			ahash_request_set_tfm(req, crypto_ahash_fb(tfm));
345 			err = crypto_ahash_digest(req);
346 			goto out_no_state;
347 		}
348 
349 		err = crypto_ahash_export(req, state);
350 		ahash_request_set_tfm(req, crypto_ahash_fb(tfm));
351 		err = err ?: crypto_ahash_import(req, state);
352 
353 		if (op == &crypto_ahash_alg(tfm)->finup) {
354 			err = err ?: crypto_ahash_finup(req);
355 			goto out_no_state;
356 		}
357 
358 		err = err ?:
359 		      crypto_ahash_update(req) ?:
360 		      crypto_ahash_export(req, state);
361 
362 		ahash_request_set_tfm(req, tfm);
363 		return err ?: crypto_ahash_import(req, state);
364 
365 out_no_state:
366 		ahash_request_set_tfm(req, tfm);
367 		return err;
368 	}
369 }
370 
371 int crypto_ahash_init(struct ahash_request *req)
372 {
373 	struct crypto_ahash *tfm = crypto_ahash_reqtfm(req);
374 
375 	if (likely(tfm->using_shash))
376 		return crypto_shash_init(prepare_shash_desc(req, tfm));
377 	if (crypto_ahash_get_flags(tfm) & CRYPTO_TFM_NEED_KEY)
378 		return -ENOKEY;
379 	if (ahash_req_on_stack(req) && ahash_is_async(tfm))
380 		return -EAGAIN;
381 	if (crypto_ahash_block_only(tfm)) {
382 		u8 *buf = ahash_request_ctx(req);
383 
384 		buf += crypto_ahash_reqsize(tfm) - 1;
385 		*buf = 0;
386 	}
387 	return crypto_ahash_alg(tfm)->init(req);
388 }
389 EXPORT_SYMBOL_GPL(crypto_ahash_init);
390 
391 static void ahash_save_req(struct ahash_request *req, crypto_completion_t cplt)
392 {
393 	req->saved_complete = req->base.complete;
394 	req->saved_data = req->base.data;
395 	req->base.complete = cplt;
396 	req->base.data = req;
397 }
398 
399 static void ahash_restore_req(struct ahash_request *req)
400 {
401 	req->base.complete = req->saved_complete;
402 	req->base.data = req->saved_data;
403 }
404 
405 static int ahash_update_finish(struct ahash_request *req, int err)
406 {
407 	struct crypto_ahash *tfm = crypto_ahash_reqtfm(req);
408 	bool nonzero = crypto_ahash_final_nonzero(tfm);
409 	int bs = crypto_ahash_blocksize(tfm);
410 	u8 *blenp = ahash_request_ctx(req);
411 	int blen;
412 	u8 *buf;
413 
414 	blenp += crypto_ahash_reqsize(tfm) - 1;
415 	blen = *blenp;
416 	buf = blenp - bs;
417 
418 	if (blen) {
419 		req->src = req->sg_head + 1;
420 		if (sg_is_chain(req->src))
421 			req->src = sg_chain_ptr(req->src);
422 	}
423 
424 	req->nbytes += nonzero - blen;
425 
426 	blen = 0;
427 	if (err >= 0) {
428 		blen = err + nonzero;
429 		err = 0;
430 	}
431 	if (ahash_request_isvirt(req))
432 		memcpy(buf, req->svirt + req->nbytes - blen, blen);
433 	else
434 		memcpy_from_sglist(buf, req->src, req->nbytes - blen, blen);
435 	*blenp = blen;
436 
437 	ahash_restore_req(req);
438 
439 	return err;
440 }
441 
442 static void ahash_update_done(void *data, int err)
443 {
444 	ahash_op_done(data, err, ahash_update_finish);
445 }
446 
447 int crypto_ahash_update(struct ahash_request *req)
448 {
449 	struct crypto_ahash *tfm = crypto_ahash_reqtfm(req);
450 	bool nonzero = crypto_ahash_final_nonzero(tfm);
451 	int bs = crypto_ahash_blocksize(tfm);
452 	u8 *blenp = ahash_request_ctx(req);
453 	int blen, err;
454 	u8 *buf;
455 
456 	if (likely(tfm->using_shash))
457 		return shash_ahash_update(req, ahash_request_ctx(req));
458 	if (ahash_req_on_stack(req) && ahash_is_async(tfm))
459 		return -EAGAIN;
460 	if (!crypto_ahash_block_only(tfm))
461 		return ahash_do_req_chain(req, &crypto_ahash_alg(tfm)->update);
462 
463 	blenp += crypto_ahash_reqsize(tfm) - 1;
464 	blen = *blenp;
465 	buf = blenp - bs;
466 
467 	if (blen + req->nbytes < bs + nonzero) {
468 		if (ahash_request_isvirt(req))
469 			memcpy(buf + blen, req->svirt, req->nbytes);
470 		else
471 			memcpy_from_sglist(buf + blen, req->src, 0,
472 					   req->nbytes);
473 
474 		*blenp += req->nbytes;
475 		return 0;
476 	}
477 
478 	if (blen) {
479 		memset(req->sg_head, 0, sizeof(req->sg_head[0]));
480 		sg_set_buf(req->sg_head, buf, blen);
481 		if (req->src != req->sg_head + 1)
482 			sg_chain(req->sg_head, 2, req->src);
483 		req->src = req->sg_head;
484 		req->nbytes += blen;
485 	}
486 	req->nbytes -= nonzero;
487 
488 	ahash_save_req(req, ahash_update_done);
489 
490 	err = ahash_do_req_chain(req, &crypto_ahash_alg(tfm)->update);
491 	if (err == -EINPROGRESS || err == -EBUSY)
492 		return err;
493 
494 	return ahash_update_finish(req, err);
495 }
496 EXPORT_SYMBOL_GPL(crypto_ahash_update);
497 
498 static int ahash_finup_finish(struct ahash_request *req, int err)
499 {
500 	struct crypto_ahash *tfm = crypto_ahash_reqtfm(req);
501 	u8 *blenp = ahash_request_ctx(req);
502 	int blen;
503 
504 	blenp += crypto_ahash_reqsize(tfm) - 1;
505 	blen = *blenp;
506 
507 	if (blen) {
508 		if (sg_is_last(req->src))
509 			req->src = NULL;
510 		else {
511 			req->src = req->sg_head + 1;
512 			if (sg_is_chain(req->src))
513 				req->src = sg_chain_ptr(req->src);
514 		}
515 		req->nbytes -= blen;
516 	}
517 
518 	ahash_restore_req(req);
519 
520 	return err;
521 }
522 
523 static void ahash_finup_done(void *data, int err)
524 {
525 	ahash_op_done(data, err, ahash_finup_finish);
526 }
527 
528 int crypto_ahash_finup(struct ahash_request *req)
529 {
530 	struct crypto_ahash *tfm = crypto_ahash_reqtfm(req);
531 	int bs = crypto_ahash_blocksize(tfm);
532 	u8 *blenp = ahash_request_ctx(req);
533 	int blen, err;
534 	u8 *buf;
535 
536 	if (likely(tfm->using_shash))
537 		return shash_ahash_finup(req, ahash_request_ctx(req));
538 	if (ahash_req_on_stack(req) && ahash_is_async(tfm))
539 		return -EAGAIN;
540 	if (!crypto_ahash_alg(tfm)->finup)
541 		return ahash_def_finup(req);
542 	if (!crypto_ahash_block_only(tfm))
543 		return ahash_do_req_chain(req, &crypto_ahash_alg(tfm)->finup);
544 
545 	blenp += crypto_ahash_reqsize(tfm) - 1;
546 	blen = *blenp;
547 	buf = blenp - bs;
548 
549 	if (blen) {
550 		memset(req->sg_head, 0, sizeof(req->sg_head[0]));
551 		sg_set_buf(req->sg_head, buf, blen);
552 		if (!req->src)
553 			sg_mark_end(req->sg_head);
554 		else if (req->src != req->sg_head + 1)
555 			sg_chain(req->sg_head, 2, req->src);
556 		req->src = req->sg_head;
557 		req->nbytes += blen;
558 	}
559 
560 	ahash_save_req(req, ahash_finup_done);
561 
562 	err = ahash_do_req_chain(req, &crypto_ahash_alg(tfm)->finup);
563 	if (err == -EINPROGRESS || err == -EBUSY)
564 		return err;
565 
566 	return ahash_finup_finish(req, err);
567 }
568 EXPORT_SYMBOL_GPL(crypto_ahash_finup);
569 
570 int crypto_ahash_digest(struct ahash_request *req)
571 {
572 	struct crypto_ahash *tfm = crypto_ahash_reqtfm(req);
573 
574 	if (likely(tfm->using_shash))
575 		return shash_ahash_digest(req, prepare_shash_desc(req, tfm));
576 	if (ahash_req_on_stack(req) && ahash_is_async(tfm))
577 		return -EAGAIN;
578 	if (crypto_ahash_get_flags(tfm) & CRYPTO_TFM_NEED_KEY)
579 		return -ENOKEY;
580 	return ahash_do_req_chain(req, &crypto_ahash_alg(tfm)->digest);
581 }
582 EXPORT_SYMBOL_GPL(crypto_ahash_digest);
583 
584 static void ahash_def_finup_done2(void *data, int err)
585 {
586 	struct ahash_request *areq = data;
587 
588 	if (err == -EINPROGRESS)
589 		return;
590 
591 	ahash_restore_req(areq);
592 	ahash_request_complete(areq, err);
593 }
594 
595 static int ahash_def_finup_finish1(struct ahash_request *req, int err)
596 {
597 	struct crypto_ahash *tfm = crypto_ahash_reqtfm(req);
598 
599 	if (err)
600 		goto out;
601 
602 	req->base.complete = ahash_def_finup_done2;
603 
604 	err = crypto_ahash_alg(tfm)->final(req);
605 	if (err == -EINPROGRESS || err == -EBUSY)
606 		return err;
607 
608 out:
609 	ahash_restore_req(req);
610 	return err;
611 }
612 
613 static void ahash_def_finup_done1(void *data, int err)
614 {
615 	ahash_op_done(data, err, ahash_def_finup_finish1);
616 }
617 
618 static int ahash_def_finup(struct ahash_request *req)
619 {
620 	int err;
621 
622 	ahash_save_req(req, ahash_def_finup_done1);
623 
624 	err = crypto_ahash_update(req);
625 	if (err == -EINPROGRESS || err == -EBUSY)
626 		return err;
627 
628 	return ahash_def_finup_finish1(req, err);
629 }
630 
631 int crypto_ahash_export_core(struct ahash_request *req, void *out)
632 {
633 	struct crypto_ahash *tfm = crypto_ahash_reqtfm(req);
634 
635 	if (likely(tfm->using_shash))
636 		return crypto_shash_export_core(ahash_request_ctx(req), out);
637 	return crypto_ahash_alg(tfm)->export_core(req, out);
638 }
639 EXPORT_SYMBOL_GPL(crypto_ahash_export_core);
640 
641 int crypto_ahash_export(struct ahash_request *req, void *out)
642 {
643 	struct crypto_ahash *tfm = crypto_ahash_reqtfm(req);
644 
645 	if (likely(tfm->using_shash))
646 		return crypto_shash_export(ahash_request_ctx(req), out);
647 	if (crypto_ahash_block_only(tfm)) {
648 		unsigned int plen = crypto_ahash_blocksize(tfm) + 1;
649 		unsigned int reqsize = crypto_ahash_reqsize(tfm);
650 		unsigned int ss = crypto_ahash_statesize(tfm);
651 		u8 *buf = ahash_request_ctx(req);
652 
653 		memcpy(out + ss - plen, buf + reqsize - plen, plen);
654 	}
655 	return crypto_ahash_alg(tfm)->export(req, out);
656 }
657 EXPORT_SYMBOL_GPL(crypto_ahash_export);
658 
659 int crypto_ahash_import_core(struct ahash_request *req, const void *in)
660 {
661 	struct crypto_ahash *tfm = crypto_ahash_reqtfm(req);
662 
663 	if (likely(tfm->using_shash))
664 		return crypto_shash_import_core(prepare_shash_desc(req, tfm),
665 						in);
666 	if (crypto_ahash_get_flags(tfm) & CRYPTO_TFM_NEED_KEY)
667 		return -ENOKEY;
668 	if (crypto_ahash_block_only(tfm)) {
669 		unsigned int reqsize = crypto_ahash_reqsize(tfm);
670 		u8 *buf = ahash_request_ctx(req);
671 
672 		buf[reqsize - 1] = 0;
673 	}
674 	return crypto_ahash_alg(tfm)->import_core(req, in);
675 }
676 EXPORT_SYMBOL_GPL(crypto_ahash_import_core);
677 
678 int crypto_ahash_import(struct ahash_request *req, const void *in)
679 {
680 	struct crypto_ahash *tfm = crypto_ahash_reqtfm(req);
681 
682 	if (likely(tfm->using_shash))
683 		return crypto_shash_import(prepare_shash_desc(req, tfm), in);
684 	if (crypto_ahash_get_flags(tfm) & CRYPTO_TFM_NEED_KEY)
685 		return -ENOKEY;
686 	if (crypto_ahash_block_only(tfm)) {
687 		unsigned int plen = crypto_ahash_blocksize(tfm) + 1;
688 		unsigned int reqsize = crypto_ahash_reqsize(tfm);
689 		unsigned int ss = crypto_ahash_statesize(tfm);
690 		u8 *buf = ahash_request_ctx(req);
691 
692 		memcpy(buf + reqsize - plen, in + ss - plen, plen);
693 		if (buf[reqsize - 1] >= plen)
694 			return -EOVERFLOW;
695 	}
696 	return crypto_ahash_alg(tfm)->import(req, in);
697 }
698 EXPORT_SYMBOL_GPL(crypto_ahash_import);
699 
700 static void crypto_ahash_exit_tfm(struct crypto_tfm *tfm)
701 {
702 	struct crypto_ahash *hash = __crypto_ahash_cast(tfm);
703 	struct ahash_alg *alg = crypto_ahash_alg(hash);
704 
705 	if (alg->exit_tfm)
706 		alg->exit_tfm(hash);
707 	else if (tfm->__crt_alg->cra_exit)
708 		tfm->__crt_alg->cra_exit(tfm);
709 
710 	if (crypto_ahash_need_fallback(hash))
711 		crypto_free_ahash(crypto_ahash_fb(hash));
712 }
713 
714 static int crypto_ahash_init_tfm(struct crypto_tfm *tfm)
715 {
716 	struct crypto_ahash *hash = __crypto_ahash_cast(tfm);
717 	struct ahash_alg *alg = crypto_ahash_alg(hash);
718 	struct crypto_ahash *fb = NULL;
719 	int err;
720 
721 	crypto_ahash_set_statesize(hash, alg->halg.statesize);
722 	crypto_ahash_set_reqsize(hash, crypto_tfm_alg_reqsize(tfm));
723 
724 	if (tfm->__crt_alg->cra_type == &crypto_shash_type)
725 		return crypto_init_ahash_using_shash(tfm);
726 
727 	if (crypto_ahash_need_fallback(hash)) {
728 		fb = crypto_alloc_ahash(crypto_ahash_alg_name(hash),
729 					CRYPTO_ALG_REQ_VIRT,
730 					CRYPTO_ALG_ASYNC |
731 					CRYPTO_ALG_REQ_VIRT |
732 					CRYPTO_AHASH_ALG_NO_EXPORT_CORE);
733 		if (IS_ERR(fb))
734 			return PTR_ERR(fb);
735 
736 		tfm->fb = crypto_ahash_tfm(fb);
737 	}
738 
739 	ahash_set_needkey(hash, alg);
740 
741 	tfm->exit = crypto_ahash_exit_tfm;
742 
743 	if (alg->init_tfm)
744 		err = alg->init_tfm(hash);
745 	else if (tfm->__crt_alg->cra_init)
746 		err = tfm->__crt_alg->cra_init(tfm);
747 	else
748 		return 0;
749 
750 	if (err)
751 		goto out_free_sync_hash;
752 
753 	if (!ahash_is_async(hash) && crypto_ahash_reqsize(hash) >
754 				     MAX_SYNC_HASH_REQSIZE)
755 		goto out_exit_tfm;
756 
757 	BUILD_BUG_ON(HASH_MAX_DESCSIZE > MAX_SYNC_HASH_REQSIZE);
758 	if (crypto_ahash_reqsize(hash) < HASH_MAX_DESCSIZE)
759 		crypto_ahash_set_reqsize(hash, HASH_MAX_DESCSIZE);
760 
761 	return 0;
762 
763 out_exit_tfm:
764 	if (alg->exit_tfm)
765 		alg->exit_tfm(hash);
766 	else if (tfm->__crt_alg->cra_exit)
767 		tfm->__crt_alg->cra_exit(tfm);
768 	err = -EINVAL;
769 out_free_sync_hash:
770 	crypto_free_ahash(fb);
771 	return err;
772 }
773 
774 static unsigned int crypto_ahash_extsize(struct crypto_alg *alg)
775 {
776 	if (alg->cra_type == &crypto_shash_type)
777 		return sizeof(struct crypto_shash *);
778 
779 	return crypto_alg_extsize(alg);
780 }
781 
782 static void crypto_ahash_free_instance(struct crypto_instance *inst)
783 {
784 	struct ahash_instance *ahash = ahash_instance(inst);
785 
786 	ahash->free(ahash);
787 }
788 
789 static int __maybe_unused crypto_ahash_report(
790 	struct sk_buff *skb, struct crypto_alg *alg)
791 {
792 	struct crypto_report_hash rhash;
793 
794 	memset(&rhash, 0, sizeof(rhash));
795 
796 	strscpy(rhash.type, "ahash", sizeof(rhash.type));
797 
798 	rhash.blocksize = alg->cra_blocksize;
799 	rhash.digestsize = __crypto_hash_alg_common(alg)->digestsize;
800 
801 	return nla_put(skb, CRYPTOCFGA_REPORT_HASH, sizeof(rhash), &rhash);
802 }
803 
804 static void __maybe_unused crypto_ahash_show(struct seq_file *m,
805 					     struct crypto_alg *alg)
806 {
807 	seq_printf(m, "type         : ahash\n");
808 	seq_printf(m, "async        : %s\n",
809 		   str_yes_no(alg->cra_flags & CRYPTO_ALG_ASYNC));
810 	seq_printf(m, "blocksize    : %u\n", alg->cra_blocksize);
811 	seq_printf(m, "digestsize   : %u\n",
812 		   __crypto_hash_alg_common(alg)->digestsize);
813 }
814 
815 static const struct crypto_type crypto_ahash_type = {
816 	.extsize = crypto_ahash_extsize,
817 	.init_tfm = crypto_ahash_init_tfm,
818 	.free = crypto_ahash_free_instance,
819 #ifdef CONFIG_PROC_FS
820 	.show = crypto_ahash_show,
821 #endif
822 #if IS_ENABLED(CONFIG_CRYPTO_USER)
823 	.report = crypto_ahash_report,
824 #endif
825 	.maskclear = ~CRYPTO_ALG_TYPE_MASK,
826 	.maskset = CRYPTO_ALG_TYPE_AHASH_MASK,
827 	.type = CRYPTO_ALG_TYPE_AHASH,
828 	.tfmsize = offsetof(struct crypto_ahash, base),
829 	.algsize = offsetof(struct ahash_alg, halg.base),
830 };
831 
832 int crypto_grab_ahash(struct crypto_ahash_spawn *spawn,
833 		      struct crypto_instance *inst,
834 		      const char *name, u32 type, u32 mask)
835 {
836 	spawn->base.frontend = &crypto_ahash_type;
837 	return crypto_grab_spawn(&spawn->base, inst, name, type, mask);
838 }
839 EXPORT_SYMBOL_GPL(crypto_grab_ahash);
840 
841 struct crypto_ahash *crypto_alloc_ahash(const char *alg_name, u32 type,
842 					u32 mask)
843 {
844 	return crypto_alloc_tfm(alg_name, &crypto_ahash_type, type, mask);
845 }
846 EXPORT_SYMBOL_GPL(crypto_alloc_ahash);
847 
848 int crypto_has_ahash(const char *alg_name, u32 type, u32 mask)
849 {
850 	return crypto_type_has_alg(alg_name, &crypto_ahash_type, type, mask);
851 }
852 EXPORT_SYMBOL_GPL(crypto_has_ahash);
853 
854 bool crypto_hash_alg_has_setkey(struct hash_alg_common *halg)
855 {
856 	struct crypto_alg *alg = &halg->base;
857 
858 	if (alg->cra_type == &crypto_shash_type)
859 		return crypto_shash_alg_has_setkey(__crypto_shash_alg(alg));
860 
861 	return __crypto_ahash_alg(alg)->setkey != ahash_nosetkey;
862 }
863 EXPORT_SYMBOL_GPL(crypto_hash_alg_has_setkey);
864 
865 struct crypto_ahash *crypto_clone_ahash(struct crypto_ahash *hash)
866 {
867 	struct hash_alg_common *halg = crypto_hash_alg_common(hash);
868 	struct crypto_tfm *tfm = crypto_ahash_tfm(hash);
869 	struct crypto_ahash *fb = NULL;
870 	struct crypto_ahash *nhash;
871 	struct ahash_alg *alg;
872 	int err;
873 
874 	if (!crypto_hash_alg_has_setkey(halg)) {
875 		tfm = crypto_tfm_get(tfm);
876 		if (IS_ERR(tfm))
877 			return ERR_CAST(tfm);
878 
879 		return hash;
880 	}
881 
882 	nhash = crypto_clone_tfm(&crypto_ahash_type, tfm);
883 
884 	if (IS_ERR(nhash))
885 		return nhash;
886 
887 	nhash->reqsize = hash->reqsize;
888 	nhash->statesize = hash->statesize;
889 
890 	if (likely(hash->using_shash)) {
891 		struct crypto_shash **nctx = crypto_ahash_ctx(nhash);
892 		struct crypto_shash *shash;
893 
894 		shash = crypto_clone_shash(ahash_to_shash(hash));
895 		if (IS_ERR(shash)) {
896 			err = PTR_ERR(shash);
897 			goto out_free_nhash;
898 		}
899 		crypto_ahash_tfm(nhash)->exit = crypto_exit_ahash_using_shash;
900 		nhash->using_shash = true;
901 		*nctx = shash;
902 		return nhash;
903 	}
904 
905 	if (crypto_ahash_need_fallback(hash)) {
906 		fb = crypto_clone_ahash(crypto_ahash_fb(hash));
907 		err = PTR_ERR(fb);
908 		if (IS_ERR(fb))
909 			goto out_free_nhash;
910 
911 		crypto_ahash_tfm(nhash)->fb = crypto_ahash_tfm(fb);
912 	}
913 
914 	err = -ENOSYS;
915 	alg = crypto_ahash_alg(hash);
916 	if (!alg->clone_tfm)
917 		goto out_free_fb;
918 
919 	err = alg->clone_tfm(nhash, hash);
920 	if (err)
921 		goto out_free_fb;
922 
923 	crypto_ahash_tfm(nhash)->exit = crypto_ahash_exit_tfm;
924 
925 	return nhash;
926 
927 out_free_fb:
928 	crypto_free_ahash(fb);
929 out_free_nhash:
930 	crypto_free_ahash(nhash);
931 	return ERR_PTR(err);
932 }
933 EXPORT_SYMBOL_GPL(crypto_clone_ahash);
934 
935 static int ahash_default_export_core(struct ahash_request *req, void *out)
936 {
937 	return -ENOSYS;
938 }
939 
940 static int ahash_default_import_core(struct ahash_request *req, const void *in)
941 {
942 	return -ENOSYS;
943 }
944 
945 static int ahash_prepare_alg(struct ahash_alg *alg)
946 {
947 	struct crypto_alg *base = &alg->halg.base;
948 	int err;
949 
950 	if (alg->halg.statesize == 0)
951 		return -EINVAL;
952 
953 	if (base->cra_reqsize && base->cra_reqsize < alg->halg.statesize)
954 		return -EINVAL;
955 
956 	if (!(base->cra_flags & CRYPTO_ALG_ASYNC) &&
957 	    base->cra_reqsize > MAX_SYNC_HASH_REQSIZE)
958 		return -EINVAL;
959 
960 	if (base->cra_flags & CRYPTO_ALG_NEED_FALLBACK &&
961 	    base->cra_flags & CRYPTO_ALG_NO_FALLBACK)
962 		return -EINVAL;
963 
964 	err = hash_prepare_alg(&alg->halg);
965 	if (err)
966 		return err;
967 
968 	base->cra_type = &crypto_ahash_type;
969 	base->cra_flags |= CRYPTO_ALG_TYPE_AHASH;
970 
971 	if ((base->cra_flags ^ CRYPTO_ALG_REQ_VIRT) &
972 	    (CRYPTO_ALG_ASYNC | CRYPTO_ALG_REQ_VIRT) &&
973 	    !(base->cra_flags & CRYPTO_ALG_NO_FALLBACK))
974 		base->cra_flags |= CRYPTO_ALG_NEED_FALLBACK;
975 
976 	if (!alg->setkey)
977 		alg->setkey = ahash_nosetkey;
978 
979 	if (base->cra_flags & CRYPTO_AHASH_ALG_BLOCK_ONLY) {
980 		BUILD_BUG_ON(MAX_ALGAPI_BLOCKSIZE >= 256);
981 		if (!alg->finup)
982 			return -EINVAL;
983 
984 		base->cra_reqsize += base->cra_blocksize + 1;
985 		alg->halg.statesize += base->cra_blocksize + 1;
986 		alg->export_core = alg->export;
987 		alg->import_core = alg->import;
988 	} else if (!alg->export_core || !alg->import_core) {
989 		alg->export_core = ahash_default_export_core;
990 		alg->import_core = ahash_default_import_core;
991 		base->cra_flags |= CRYPTO_AHASH_ALG_NO_EXPORT_CORE;
992 	}
993 
994 	return 0;
995 }
996 
997 int crypto_register_ahash(struct ahash_alg *alg)
998 {
999 	struct crypto_alg *base = &alg->halg.base;
1000 	int err;
1001 
1002 	err = ahash_prepare_alg(alg);
1003 	if (err)
1004 		return err;
1005 
1006 	return crypto_register_alg(base);
1007 }
1008 EXPORT_SYMBOL_GPL(crypto_register_ahash);
1009 
1010 void crypto_unregister_ahash(struct ahash_alg *alg)
1011 {
1012 	crypto_unregister_alg(&alg->halg.base);
1013 }
1014 EXPORT_SYMBOL_GPL(crypto_unregister_ahash);
1015 
1016 int crypto_register_ahashes(struct ahash_alg *algs, int count)
1017 {
1018 	int i, ret;
1019 
1020 	for (i = 0; i < count; i++) {
1021 		ret = crypto_register_ahash(&algs[i]);
1022 		if (ret) {
1023 			crypto_unregister_ahashes(algs, i);
1024 			return ret;
1025 		}
1026 	}
1027 
1028 	return 0;
1029 }
1030 EXPORT_SYMBOL_GPL(crypto_register_ahashes);
1031 
1032 void crypto_unregister_ahashes(struct ahash_alg *algs, int count)
1033 {
1034 	int i;
1035 
1036 	for (i = count - 1; i >= 0; --i)
1037 		crypto_unregister_ahash(&algs[i]);
1038 }
1039 EXPORT_SYMBOL_GPL(crypto_unregister_ahashes);
1040 
1041 int ahash_register_instance(struct crypto_template *tmpl,
1042 			    struct ahash_instance *inst)
1043 {
1044 	int err;
1045 
1046 	if (WARN_ON(!inst->free))
1047 		return -EINVAL;
1048 
1049 	err = ahash_prepare_alg(&inst->alg);
1050 	if (err)
1051 		return err;
1052 
1053 	return crypto_register_instance(tmpl, ahash_crypto_instance(inst));
1054 }
1055 EXPORT_SYMBOL_GPL(ahash_register_instance);
1056 
1057 void ahash_request_free(struct ahash_request *req)
1058 {
1059 	if (unlikely(!req))
1060 		return;
1061 
1062 	if (!ahash_req_on_stack(req)) {
1063 		kfree(req);
1064 		return;
1065 	}
1066 
1067 	ahash_request_zero(req);
1068 }
1069 EXPORT_SYMBOL_GPL(ahash_request_free);
1070 
1071 int crypto_hash_digest(struct crypto_ahash *tfm, const u8 *data,
1072 		       unsigned int len, u8 *out)
1073 {
1074 	HASH_REQUEST_ON_STACK(req, crypto_ahash_fb(tfm));
1075 	int err;
1076 
1077 	ahash_request_set_callback(req, 0, NULL, NULL);
1078 	ahash_request_set_virt(req, data, out, len);
1079 	err = crypto_ahash_digest(req);
1080 
1081 	ahash_request_zero(req);
1082 
1083 	return err;
1084 }
1085 EXPORT_SYMBOL_GPL(crypto_hash_digest);
1086 
1087 void ahash_free_singlespawn_instance(struct ahash_instance *inst)
1088 {
1089 	crypto_drop_spawn(ahash_instance_ctx(inst));
1090 	kfree(inst);
1091 }
1092 EXPORT_SYMBOL_GPL(ahash_free_singlespawn_instance);
1093 
1094 MODULE_LICENSE("GPL");
1095 MODULE_DESCRIPTION("Asynchronous cryptographic hash type");
1096