crypto: caam - remove unneeded edesc zeroization
[linux-2.6-microblaze.git] / drivers / crypto / caam / caamhash.c
1 /*
2  * caam - Freescale FSL CAAM support for ahash functions of crypto API
3  *
4  * Copyright 2011 Freescale Semiconductor, Inc.
5  *
6  * Based on caamalg.c crypto API driver.
7  *
8  * relationship of digest job descriptor or first job descriptor after init to
9  * shared descriptors:
10  *
11  * ---------------                     ---------------
12  * | JobDesc #1  |-------------------->|  ShareDesc  |
13  * | *(packet 1) |                     |  (hashKey)  |
14  * ---------------                     | (operation) |
15  *                                     ---------------
16  *
17  * relationship of subsequent job descriptors to shared descriptors:
18  *
19  * ---------------                     ---------------
20  * | JobDesc #2  |-------------------->|  ShareDesc  |
21  * | *(packet 2) |      |------------->|  (hashKey)  |
22  * ---------------      |    |-------->| (operation) |
23  *       .              |    |         | (load ctx2) |
24  *       .              |    |         ---------------
25  * ---------------      |    |
26  * | JobDesc #3  |------|    |
27  * | *(packet 3) |           |
28  * ---------------           |
29  *       .                   |
30  *       .                   |
31  * ---------------           |
32  * | JobDesc #4  |------------
33  * | *(packet 4) |
34  * ---------------
35  *
36  * The SharedDesc never changes for a connection unless rekeyed, but
37  * each packet will likely be in a different place. So all we need
38  * to know to process the packet is where the input is, where the
39  * output goes, and what context we want to process with. Context is
40  * in the SharedDesc, packet references in the JobDesc.
41  *
42  * So, a job desc looks like:
43  *
44  * ---------------------
45  * | Header            |
46  * | ShareDesc Pointer |
47  * | SEQ_OUT_PTR       |
48  * | (output buffer)   |
49  * | (output length)   |
50  * | SEQ_IN_PTR        |
51  * | (input buffer)    |
52  * | (input length)    |
53  * ---------------------
54  */
55
56 #include "compat.h"
57
58 #include "regs.h"
59 #include "intern.h"
60 #include "desc_constr.h"
61 #include "jr.h"
62 #include "error.h"
63 #include "sg_sw_sec4.h"
64 #include "key_gen.h"
65
66 #define CAAM_CRA_PRIORITY               3000
67
68 /* max hash key is max split key size */
69 #define CAAM_MAX_HASH_KEY_SIZE          (SHA512_DIGEST_SIZE * 2)
70
71 #define CAAM_MAX_HASH_BLOCK_SIZE        SHA512_BLOCK_SIZE
72 #define CAAM_MAX_HASH_DIGEST_SIZE       SHA512_DIGEST_SIZE
73
74 /* length of descriptors text */
75 #define DESC_AHASH_BASE                 (3 * CAAM_CMD_SZ)
76 #define DESC_AHASH_UPDATE_LEN           (6 * CAAM_CMD_SZ)
77 #define DESC_AHASH_UPDATE_FIRST_LEN     (DESC_AHASH_BASE + 4 * CAAM_CMD_SZ)
78 #define DESC_AHASH_FINAL_LEN            (DESC_AHASH_BASE + 5 * CAAM_CMD_SZ)
79 #define DESC_AHASH_FINUP_LEN            (DESC_AHASH_BASE + 5 * CAAM_CMD_SZ)
80 #define DESC_AHASH_DIGEST_LEN           (DESC_AHASH_BASE + 4 * CAAM_CMD_SZ)
81
82 #define DESC_HASH_MAX_USED_BYTES        (DESC_AHASH_FINAL_LEN + \
83                                          CAAM_MAX_HASH_KEY_SIZE)
84 #define DESC_HASH_MAX_USED_LEN          (DESC_HASH_MAX_USED_BYTES / CAAM_CMD_SZ)
85
86 /* caam context sizes for hashes: running digest + 8 */
87 #define HASH_MSG_LEN                    8
88 #define MAX_CTX_LEN                     (HASH_MSG_LEN + SHA512_DIGEST_SIZE)
89
90 #ifdef DEBUG
91 /* for print_hex_dumps with line references */
92 #define debug(format, arg...) printk(format, arg)
93 #else
94 #define debug(format, arg...)
95 #endif
96
97
98 static struct list_head hash_list;
99
100 /* ahash per-session context */
101 struct caam_hash_ctx {
102         u32 sh_desc_update[DESC_HASH_MAX_USED_LEN] ____cacheline_aligned;
103         u32 sh_desc_update_first[DESC_HASH_MAX_USED_LEN] ____cacheline_aligned;
104         u32 sh_desc_fin[DESC_HASH_MAX_USED_LEN] ____cacheline_aligned;
105         u32 sh_desc_digest[DESC_HASH_MAX_USED_LEN] ____cacheline_aligned;
106         dma_addr_t sh_desc_update_dma ____cacheline_aligned;
107         dma_addr_t sh_desc_update_first_dma;
108         dma_addr_t sh_desc_fin_dma;
109         dma_addr_t sh_desc_digest_dma;
110         struct device *jrdev;
111         u8 key[CAAM_MAX_HASH_KEY_SIZE];
112         int ctx_len;
113         struct alginfo adata;
114 };
115
116 /* ahash state */
117 struct caam_hash_state {
118         dma_addr_t buf_dma;
119         dma_addr_t ctx_dma;
120         u8 buf_0[CAAM_MAX_HASH_BLOCK_SIZE] ____cacheline_aligned;
121         int buflen_0;
122         u8 buf_1[CAAM_MAX_HASH_BLOCK_SIZE] ____cacheline_aligned;
123         int buflen_1;
124         u8 caam_ctx[MAX_CTX_LEN] ____cacheline_aligned;
125         int (*update)(struct ahash_request *req);
126         int (*final)(struct ahash_request *req);
127         int (*finup)(struct ahash_request *req);
128         int current_buf;
129 };
130
131 struct caam_export_state {
132         u8 buf[CAAM_MAX_HASH_BLOCK_SIZE];
133         u8 caam_ctx[MAX_CTX_LEN];
134         int buflen;
135         int (*update)(struct ahash_request *req);
136         int (*final)(struct ahash_request *req);
137         int (*finup)(struct ahash_request *req);
138 };
139
140 static inline void switch_buf(struct caam_hash_state *state)
141 {
142         state->current_buf ^= 1;
143 }
144
145 static inline u8 *current_buf(struct caam_hash_state *state)
146 {
147         return state->current_buf ? state->buf_1 : state->buf_0;
148 }
149
150 static inline u8 *alt_buf(struct caam_hash_state *state)
151 {
152         return state->current_buf ? state->buf_0 : state->buf_1;
153 }
154
155 static inline int *current_buflen(struct caam_hash_state *state)
156 {
157         return state->current_buf ? &state->buflen_1 : &state->buflen_0;
158 }
159
160 static inline int *alt_buflen(struct caam_hash_state *state)
161 {
162         return state->current_buf ? &state->buflen_0 : &state->buflen_1;
163 }
164
165 /* Common job descriptor seq in/out ptr routines */
166
167 /* Map state->caam_ctx, and append seq_out_ptr command that points to it */
168 static inline int map_seq_out_ptr_ctx(u32 *desc, struct device *jrdev,
169                                       struct caam_hash_state *state,
170                                       int ctx_len)
171 {
172         state->ctx_dma = dma_map_single(jrdev, state->caam_ctx,
173                                         ctx_len, DMA_FROM_DEVICE);
174         if (dma_mapping_error(jrdev, state->ctx_dma)) {
175                 dev_err(jrdev, "unable to map ctx\n");
176                 state->ctx_dma = 0;
177                 return -ENOMEM;
178         }
179
180         append_seq_out_ptr(desc, state->ctx_dma, ctx_len, 0);
181
182         return 0;
183 }
184
185 /* Map req->result, and append seq_out_ptr command that points to it */
186 static inline dma_addr_t map_seq_out_ptr_result(u32 *desc, struct device *jrdev,
187                                                 u8 *result, int digestsize)
188 {
189         dma_addr_t dst_dma;
190
191         dst_dma = dma_map_single(jrdev, result, digestsize, DMA_FROM_DEVICE);
192         append_seq_out_ptr(desc, dst_dma, digestsize, 0);
193
194         return dst_dma;
195 }
196
197 /* Map current buffer in state (if length > 0) and put it in link table */
198 static inline int buf_map_to_sec4_sg(struct device *jrdev,
199                                      struct sec4_sg_entry *sec4_sg,
200                                      struct caam_hash_state *state)
201 {
202         int buflen = *current_buflen(state);
203
204         if (!buflen)
205                 return 0;
206
207         state->buf_dma = dma_map_single(jrdev, current_buf(state), buflen,
208                                         DMA_TO_DEVICE);
209         if (dma_mapping_error(jrdev, state->buf_dma)) {
210                 dev_err(jrdev, "unable to map buf\n");
211                 state->buf_dma = 0;
212                 return -ENOMEM;
213         }
214
215         dma_to_sec4_sg_one(sec4_sg, state->buf_dma, buflen, 0);
216
217         return 0;
218 }
219
220 /* Map state->caam_ctx, and add it to link table */
221 static inline int ctx_map_to_sec4_sg(u32 *desc, struct device *jrdev,
222                                      struct caam_hash_state *state, int ctx_len,
223                                      struct sec4_sg_entry *sec4_sg, u32 flag)
224 {
225         state->ctx_dma = dma_map_single(jrdev, state->caam_ctx, ctx_len, flag);
226         if (dma_mapping_error(jrdev, state->ctx_dma)) {
227                 dev_err(jrdev, "unable to map ctx\n");
228                 state->ctx_dma = 0;
229                 return -ENOMEM;
230         }
231
232         dma_to_sec4_sg_one(sec4_sg, state->ctx_dma, ctx_len, 0);
233
234         return 0;
235 }
236
237 /*
238  * For ahash update, final and finup (import_ctx = true)
239  *     import context, read and write to seqout
240  * For ahash firsts and digest (import_ctx = false)
241  *     read and write to seqout
242  */
243 static inline void ahash_gen_sh_desc(u32 *desc, u32 state, int digestsize,
244                                      struct caam_hash_ctx *ctx, bool import_ctx)
245 {
246         u32 op = ctx->adata.algtype;
247         u32 *skip_key_load;
248
249         init_sh_desc(desc, HDR_SHARE_SERIAL);
250
251         /* Append key if it has been set; ahash update excluded */
252         if ((state != OP_ALG_AS_UPDATE) && (ctx->adata.keylen)) {
253                 /* Skip key loading if already shared */
254                 skip_key_load = append_jump(desc, JUMP_JSL | JUMP_TEST_ALL |
255                                             JUMP_COND_SHRD);
256
257                 append_key_as_imm(desc, ctx->key, ctx->adata.keylen_pad,
258                                   ctx->adata.keylen, CLASS_2 |
259                                   KEY_DEST_MDHA_SPLIT | KEY_ENC);
260
261                 set_jump_tgt_here(desc, skip_key_load);
262
263                 op |= OP_ALG_AAI_HMAC_PRECOMP;
264         }
265
266         /* If needed, import context from software */
267         if (import_ctx)
268                 append_seq_load(desc, ctx->ctx_len, LDST_CLASS_2_CCB |
269                                 LDST_SRCDST_BYTE_CONTEXT);
270
271         /* Class 2 operation */
272         append_operation(desc, op | state | OP_ALG_ENCRYPT);
273
274         /*
275          * Load from buf and/or src and write to req->result or state->context
276          * Calculate remaining bytes to read
277          */
278         append_math_add(desc, VARSEQINLEN, SEQINLEN, REG0, CAAM_CMD_SZ);
279         /* Read remaining bytes */
280         append_seq_fifo_load(desc, 0, FIFOLD_CLASS_CLASS2 | FIFOLD_TYPE_LAST2 |
281                              FIFOLD_TYPE_MSG | KEY_VLF);
282         /* Store class2 context bytes */
283         append_seq_store(desc, digestsize, LDST_CLASS_2_CCB |
284                          LDST_SRCDST_BYTE_CONTEXT);
285 }
286
287 static int ahash_set_sh_desc(struct crypto_ahash *ahash)
288 {
289         struct caam_hash_ctx *ctx = crypto_ahash_ctx(ahash);
290         int digestsize = crypto_ahash_digestsize(ahash);
291         struct device *jrdev = ctx->jrdev;
292         u32 *desc;
293
294         /* ahash_update shared descriptor */
295         desc = ctx->sh_desc_update;
296         ahash_gen_sh_desc(desc, OP_ALG_AS_UPDATE, ctx->ctx_len, ctx, true);
297         dma_sync_single_for_device(jrdev, ctx->sh_desc_update_dma,
298                                    desc_bytes(desc), DMA_TO_DEVICE);
299 #ifdef DEBUG
300         print_hex_dump(KERN_ERR,
301                        "ahash update shdesc@"__stringify(__LINE__)": ",
302                        DUMP_PREFIX_ADDRESS, 16, 4, desc, desc_bytes(desc), 1);
303 #endif
304
305         /* ahash_update_first shared descriptor */
306         desc = ctx->sh_desc_update_first;
307         ahash_gen_sh_desc(desc, OP_ALG_AS_INIT, ctx->ctx_len, ctx, false);
308         dma_sync_single_for_device(jrdev, ctx->sh_desc_update_first_dma,
309                                    desc_bytes(desc), DMA_TO_DEVICE);
310 #ifdef DEBUG
311         print_hex_dump(KERN_ERR,
312                        "ahash update first shdesc@"__stringify(__LINE__)": ",
313                        DUMP_PREFIX_ADDRESS, 16, 4, desc, desc_bytes(desc), 1);
314 #endif
315
316         /* ahash_final shared descriptor */
317         desc = ctx->sh_desc_fin;
318         ahash_gen_sh_desc(desc, OP_ALG_AS_FINALIZE, digestsize, ctx, true);
319         dma_sync_single_for_device(jrdev, ctx->sh_desc_fin_dma,
320                                    desc_bytes(desc), DMA_TO_DEVICE);
321 #ifdef DEBUG
322         print_hex_dump(KERN_ERR, "ahash final shdesc@"__stringify(__LINE__)": ",
323                        DUMP_PREFIX_ADDRESS, 16, 4, desc,
324                        desc_bytes(desc), 1);
325 #endif
326
327         /* ahash_digest shared descriptor */
328         desc = ctx->sh_desc_digest;
329         ahash_gen_sh_desc(desc, OP_ALG_AS_INITFINAL, digestsize, ctx, false);
330         dma_sync_single_for_device(jrdev, ctx->sh_desc_digest_dma,
331                                    desc_bytes(desc), DMA_TO_DEVICE);
332 #ifdef DEBUG
333         print_hex_dump(KERN_ERR,
334                        "ahash digest shdesc@"__stringify(__LINE__)": ",
335                        DUMP_PREFIX_ADDRESS, 16, 4, desc,
336                        desc_bytes(desc), 1);
337 #endif
338
339         return 0;
340 }
341
342 /* Digest hash size if it is too large */
343 static int hash_digest_key(struct caam_hash_ctx *ctx, const u8 *key_in,
344                            u32 *keylen, u8 *key_out, u32 digestsize)
345 {
346         struct device *jrdev = ctx->jrdev;
347         u32 *desc;
348         struct split_key_result result;
349         dma_addr_t src_dma, dst_dma;
350         int ret;
351
352         desc = kmalloc(CAAM_CMD_SZ * 8 + CAAM_PTR_SZ * 2, GFP_KERNEL | GFP_DMA);
353         if (!desc) {
354                 dev_err(jrdev, "unable to allocate key input memory\n");
355                 return -ENOMEM;
356         }
357
358         init_job_desc(desc, 0);
359
360         src_dma = dma_map_single(jrdev, (void *)key_in, *keylen,
361                                  DMA_TO_DEVICE);
362         if (dma_mapping_error(jrdev, src_dma)) {
363                 dev_err(jrdev, "unable to map key input memory\n");
364                 kfree(desc);
365                 return -ENOMEM;
366         }
367         dst_dma = dma_map_single(jrdev, (void *)key_out, digestsize,
368                                  DMA_FROM_DEVICE);
369         if (dma_mapping_error(jrdev, dst_dma)) {
370                 dev_err(jrdev, "unable to map key output memory\n");
371                 dma_unmap_single(jrdev, src_dma, *keylen, DMA_TO_DEVICE);
372                 kfree(desc);
373                 return -ENOMEM;
374         }
375
376         /* Job descriptor to perform unkeyed hash on key_in */
377         append_operation(desc, ctx->adata.algtype | OP_ALG_ENCRYPT |
378                          OP_ALG_AS_INITFINAL);
379         append_seq_in_ptr(desc, src_dma, *keylen, 0);
380         append_seq_fifo_load(desc, *keylen, FIFOLD_CLASS_CLASS2 |
381                              FIFOLD_TYPE_LAST2 | FIFOLD_TYPE_MSG);
382         append_seq_out_ptr(desc, dst_dma, digestsize, 0);
383         append_seq_store(desc, digestsize, LDST_CLASS_2_CCB |
384                          LDST_SRCDST_BYTE_CONTEXT);
385
386 #ifdef DEBUG
387         print_hex_dump(KERN_ERR, "key_in@"__stringify(__LINE__)": ",
388                        DUMP_PREFIX_ADDRESS, 16, 4, key_in, *keylen, 1);
389         print_hex_dump(KERN_ERR, "jobdesc@"__stringify(__LINE__)": ",
390                        DUMP_PREFIX_ADDRESS, 16, 4, desc, desc_bytes(desc), 1);
391 #endif
392
393         result.err = 0;
394         init_completion(&result.completion);
395
396         ret = caam_jr_enqueue(jrdev, desc, split_key_done, &result);
397         if (!ret) {
398                 /* in progress */
399                 wait_for_completion(&result.completion);
400                 ret = result.err;
401 #ifdef DEBUG
402                 print_hex_dump(KERN_ERR,
403                                "digested key@"__stringify(__LINE__)": ",
404                                DUMP_PREFIX_ADDRESS, 16, 4, key_in,
405                                digestsize, 1);
406 #endif
407         }
408         dma_unmap_single(jrdev, src_dma, *keylen, DMA_TO_DEVICE);
409         dma_unmap_single(jrdev, dst_dma, digestsize, DMA_FROM_DEVICE);
410
411         *keylen = digestsize;
412
413         kfree(desc);
414
415         return ret;
416 }
417
418 static int ahash_setkey(struct crypto_ahash *ahash,
419                         const u8 *key, unsigned int keylen)
420 {
421         struct caam_hash_ctx *ctx = crypto_ahash_ctx(ahash);
422         int blocksize = crypto_tfm_alg_blocksize(&ahash->base);
423         int digestsize = crypto_ahash_digestsize(ahash);
424         int ret;
425         u8 *hashed_key = NULL;
426
427 #ifdef DEBUG
428         printk(KERN_ERR "keylen %d\n", keylen);
429 #endif
430
431         if (keylen > blocksize) {
432                 hashed_key = kmalloc_array(digestsize,
433                                            sizeof(*hashed_key),
434                                            GFP_KERNEL | GFP_DMA);
435                 if (!hashed_key)
436                         return -ENOMEM;
437                 ret = hash_digest_key(ctx, key, &keylen, hashed_key,
438                                       digestsize);
439                 if (ret)
440                         goto bad_free_key;
441                 key = hashed_key;
442         }
443
444         ret = gen_split_key(ctx->jrdev, ctx->key, &ctx->adata, key, keylen,
445                             CAAM_MAX_HASH_KEY_SIZE);
446         if (ret)
447                 goto bad_free_key;
448
449 #ifdef DEBUG
450         print_hex_dump(KERN_ERR, "ctx.key@"__stringify(__LINE__)": ",
451                        DUMP_PREFIX_ADDRESS, 16, 4, ctx->key,
452                        ctx->adata.keylen_pad, 1);
453 #endif
454
455         kfree(hashed_key);
456         return ahash_set_sh_desc(ahash);
457  bad_free_key:
458         kfree(hashed_key);
459         crypto_ahash_set_flags(ahash, CRYPTO_TFM_RES_BAD_KEY_LEN);
460         return -EINVAL;
461 }
462
463 /*
464  * ahash_edesc - s/w-extended ahash descriptor
465  * @dst_dma: physical mapped address of req->result
466  * @sec4_sg_dma: physical mapped address of h/w link table
467  * @src_nents: number of segments in input scatterlist
468  * @sec4_sg_bytes: length of dma mapped sec4_sg space
469  * @hw_desc: the h/w job descriptor followed by any referenced link tables
470  * @sec4_sg: h/w link table
471  */
472 struct ahash_edesc {
473         dma_addr_t dst_dma;
474         dma_addr_t sec4_sg_dma;
475         int src_nents;
476         int sec4_sg_bytes;
477         u32 hw_desc[DESC_JOB_IO_LEN / sizeof(u32)] ____cacheline_aligned;
478         struct sec4_sg_entry sec4_sg[0];
479 };
480
481 static inline void ahash_unmap(struct device *dev,
482                         struct ahash_edesc *edesc,
483                         struct ahash_request *req, int dst_len)
484 {
485         struct caam_hash_state *state = ahash_request_ctx(req);
486
487         if (edesc->src_nents)
488                 dma_unmap_sg(dev, req->src, edesc->src_nents, DMA_TO_DEVICE);
489         if (edesc->dst_dma)
490                 dma_unmap_single(dev, edesc->dst_dma, dst_len, DMA_FROM_DEVICE);
491
492         if (edesc->sec4_sg_bytes)
493                 dma_unmap_single(dev, edesc->sec4_sg_dma,
494                                  edesc->sec4_sg_bytes, DMA_TO_DEVICE);
495
496         if (state->buf_dma) {
497                 dma_unmap_single(dev, state->buf_dma, *current_buflen(state),
498                                  DMA_TO_DEVICE);
499                 state->buf_dma = 0;
500         }
501 }
502
503 static inline void ahash_unmap_ctx(struct device *dev,
504                         struct ahash_edesc *edesc,
505                         struct ahash_request *req, int dst_len, u32 flag)
506 {
507         struct crypto_ahash *ahash = crypto_ahash_reqtfm(req);
508         struct caam_hash_ctx *ctx = crypto_ahash_ctx(ahash);
509         struct caam_hash_state *state = ahash_request_ctx(req);
510
511         if (state->ctx_dma) {
512                 dma_unmap_single(dev, state->ctx_dma, ctx->ctx_len, flag);
513                 state->ctx_dma = 0;
514         }
515         ahash_unmap(dev, edesc, req, dst_len);
516 }
517
518 static void ahash_done(struct device *jrdev, u32 *desc, u32 err,
519                        void *context)
520 {
521         struct ahash_request *req = context;
522         struct ahash_edesc *edesc;
523         struct crypto_ahash *ahash = crypto_ahash_reqtfm(req);
524         int digestsize = crypto_ahash_digestsize(ahash);
525 #ifdef DEBUG
526         struct caam_hash_ctx *ctx = crypto_ahash_ctx(ahash);
527         struct caam_hash_state *state = ahash_request_ctx(req);
528
529         dev_err(jrdev, "%s %d: err 0x%x\n", __func__, __LINE__, err);
530 #endif
531
532         edesc = container_of(desc, struct ahash_edesc, hw_desc[0]);
533         if (err)
534                 caam_jr_strstatus(jrdev, err);
535
536         ahash_unmap(jrdev, edesc, req, digestsize);
537         kfree(edesc);
538
539 #ifdef DEBUG
540         print_hex_dump(KERN_ERR, "ctx@"__stringify(__LINE__)": ",
541                        DUMP_PREFIX_ADDRESS, 16, 4, state->caam_ctx,
542                        ctx->ctx_len, 1);
543         if (req->result)
544                 print_hex_dump(KERN_ERR, "result@"__stringify(__LINE__)": ",
545                                DUMP_PREFIX_ADDRESS, 16, 4, req->result,
546                                digestsize, 1);
547 #endif
548
549         req->base.complete(&req->base, err);
550 }
551
552 static void ahash_done_bi(struct device *jrdev, u32 *desc, u32 err,
553                             void *context)
554 {
555         struct ahash_request *req = context;
556         struct ahash_edesc *edesc;
557         struct crypto_ahash *ahash = crypto_ahash_reqtfm(req);
558         struct caam_hash_ctx *ctx = crypto_ahash_ctx(ahash);
559         struct caam_hash_state *state = ahash_request_ctx(req);
560 #ifdef DEBUG
561         int digestsize = crypto_ahash_digestsize(ahash);
562
563         dev_err(jrdev, "%s %d: err 0x%x\n", __func__, __LINE__, err);
564 #endif
565
566         edesc = container_of(desc, struct ahash_edesc, hw_desc[0]);
567         if (err)
568                 caam_jr_strstatus(jrdev, err);
569
570         ahash_unmap_ctx(jrdev, edesc, req, ctx->ctx_len, DMA_BIDIRECTIONAL);
571         switch_buf(state);
572         kfree(edesc);
573
574 #ifdef DEBUG
575         print_hex_dump(KERN_ERR, "ctx@"__stringify(__LINE__)": ",
576                        DUMP_PREFIX_ADDRESS, 16, 4, state->caam_ctx,
577                        ctx->ctx_len, 1);
578         if (req->result)
579                 print_hex_dump(KERN_ERR, "result@"__stringify(__LINE__)": ",
580                                DUMP_PREFIX_ADDRESS, 16, 4, req->result,
581                                digestsize, 1);
582 #endif
583
584         req->base.complete(&req->base, err);
585 }
586
587 static void ahash_done_ctx_src(struct device *jrdev, u32 *desc, u32 err,
588                                void *context)
589 {
590         struct ahash_request *req = context;
591         struct ahash_edesc *edesc;
592         struct crypto_ahash *ahash = crypto_ahash_reqtfm(req);
593         int digestsize = crypto_ahash_digestsize(ahash);
594 #ifdef DEBUG
595         struct caam_hash_ctx *ctx = crypto_ahash_ctx(ahash);
596         struct caam_hash_state *state = ahash_request_ctx(req);
597
598         dev_err(jrdev, "%s %d: err 0x%x\n", __func__, __LINE__, err);
599 #endif
600
601         edesc = container_of(desc, struct ahash_edesc, hw_desc[0]);
602         if (err)
603                 caam_jr_strstatus(jrdev, err);
604
605         ahash_unmap_ctx(jrdev, edesc, req, digestsize, DMA_TO_DEVICE);
606         kfree(edesc);
607
608 #ifdef DEBUG
609         print_hex_dump(KERN_ERR, "ctx@"__stringify(__LINE__)": ",
610                        DUMP_PREFIX_ADDRESS, 16, 4, state->caam_ctx,
611                        ctx->ctx_len, 1);
612         if (req->result)
613                 print_hex_dump(KERN_ERR, "result@"__stringify(__LINE__)": ",
614                                DUMP_PREFIX_ADDRESS, 16, 4, req->result,
615                                digestsize, 1);
616 #endif
617
618         req->base.complete(&req->base, err);
619 }
620
621 static void ahash_done_ctx_dst(struct device *jrdev, u32 *desc, u32 err,
622                                void *context)
623 {
624         struct ahash_request *req = context;
625         struct ahash_edesc *edesc;
626         struct crypto_ahash *ahash = crypto_ahash_reqtfm(req);
627         struct caam_hash_ctx *ctx = crypto_ahash_ctx(ahash);
628         struct caam_hash_state *state = ahash_request_ctx(req);
629 #ifdef DEBUG
630         int digestsize = crypto_ahash_digestsize(ahash);
631
632         dev_err(jrdev, "%s %d: err 0x%x\n", __func__, __LINE__, err);
633 #endif
634
635         edesc = container_of(desc, struct ahash_edesc, hw_desc[0]);
636         if (err)
637                 caam_jr_strstatus(jrdev, err);
638
639         ahash_unmap_ctx(jrdev, edesc, req, ctx->ctx_len, DMA_FROM_DEVICE);
640         switch_buf(state);
641         kfree(edesc);
642
643 #ifdef DEBUG
644         print_hex_dump(KERN_ERR, "ctx@"__stringify(__LINE__)": ",
645                        DUMP_PREFIX_ADDRESS, 16, 4, state->caam_ctx,
646                        ctx->ctx_len, 1);
647         if (req->result)
648                 print_hex_dump(KERN_ERR, "result@"__stringify(__LINE__)": ",
649                                DUMP_PREFIX_ADDRESS, 16, 4, req->result,
650                                digestsize, 1);
651 #endif
652
653         req->base.complete(&req->base, err);
654 }
655
656 /*
657  * Allocate an enhanced descriptor, which contains the hardware descriptor
658  * and space for hardware scatter table containing sg_num entries.
659  */
660 static struct ahash_edesc *ahash_edesc_alloc(struct caam_hash_ctx *ctx,
661                                              int sg_num, u32 *sh_desc,
662                                              dma_addr_t sh_desc_dma,
663                                              gfp_t flags)
664 {
665         struct ahash_edesc *edesc;
666         unsigned int sg_size = sg_num * sizeof(struct sec4_sg_entry);
667
668         edesc = kzalloc(sizeof(*edesc) + sg_size, GFP_DMA | flags);
669         if (!edesc) {
670                 dev_err(ctx->jrdev, "could not allocate extended descriptor\n");
671                 return NULL;
672         }
673
674         init_job_desc_shared(edesc->hw_desc, sh_desc_dma, desc_len(sh_desc),
675                              HDR_SHARE_DEFER | HDR_REVERSE);
676
677         return edesc;
678 }
679
680 static int ahash_edesc_add_src(struct caam_hash_ctx *ctx,
681                                struct ahash_edesc *edesc,
682                                struct ahash_request *req, int nents,
683                                unsigned int first_sg,
684                                unsigned int first_bytes, size_t to_hash)
685 {
686         dma_addr_t src_dma;
687         u32 options;
688
689         if (nents > 1 || first_sg) {
690                 struct sec4_sg_entry *sg = edesc->sec4_sg;
691                 unsigned int sgsize = sizeof(*sg) * (first_sg + nents);
692
693                 sg_to_sec4_sg_last(req->src, nents, sg + first_sg, 0);
694
695                 src_dma = dma_map_single(ctx->jrdev, sg, sgsize, DMA_TO_DEVICE);
696                 if (dma_mapping_error(ctx->jrdev, src_dma)) {
697                         dev_err(ctx->jrdev, "unable to map S/G table\n");
698                         return -ENOMEM;
699                 }
700
701                 edesc->sec4_sg_bytes = sgsize;
702                 edesc->sec4_sg_dma = src_dma;
703                 options = LDST_SGF;
704         } else {
705                 src_dma = sg_dma_address(req->src);
706                 options = 0;
707         }
708
709         append_seq_in_ptr(edesc->hw_desc, src_dma, first_bytes + to_hash,
710                           options);
711
712         return 0;
713 }
714
715 /* submit update job descriptor */
716 static int ahash_update_ctx(struct ahash_request *req)
717 {
718         struct crypto_ahash *ahash = crypto_ahash_reqtfm(req);
719         struct caam_hash_ctx *ctx = crypto_ahash_ctx(ahash);
720         struct caam_hash_state *state = ahash_request_ctx(req);
721         struct device *jrdev = ctx->jrdev;
722         gfp_t flags = (req->base.flags & CRYPTO_TFM_REQ_MAY_SLEEP) ?
723                        GFP_KERNEL : GFP_ATOMIC;
724         u8 *buf = current_buf(state);
725         int *buflen = current_buflen(state);
726         u8 *next_buf = alt_buf(state);
727         int *next_buflen = alt_buflen(state), last_buflen;
728         int in_len = *buflen + req->nbytes, to_hash;
729         u32 *desc;
730         int src_nents, mapped_nents, sec4_sg_bytes, sec4_sg_src_index;
731         struct ahash_edesc *edesc;
732         int ret = 0;
733
734         last_buflen = *next_buflen;
735         *next_buflen = in_len & (crypto_tfm_alg_blocksize(&ahash->base) - 1);
736         to_hash = in_len - *next_buflen;
737
738         if (to_hash) {
739                 src_nents = sg_nents_for_len(req->src,
740                                              req->nbytes - (*next_buflen));
741                 if (src_nents < 0) {
742                         dev_err(jrdev, "Invalid number of src SG.\n");
743                         return src_nents;
744                 }
745
746                 if (src_nents) {
747                         mapped_nents = dma_map_sg(jrdev, req->src, src_nents,
748                                                   DMA_TO_DEVICE);
749                         if (!mapped_nents) {
750                                 dev_err(jrdev, "unable to DMA map source\n");
751                                 return -ENOMEM;
752                         }
753                 } else {
754                         mapped_nents = 0;
755                 }
756
757                 sec4_sg_src_index = 1 + (*buflen ? 1 : 0);
758                 sec4_sg_bytes = (sec4_sg_src_index + mapped_nents) *
759                                  sizeof(struct sec4_sg_entry);
760
761                 /*
762                  * allocate space for base edesc and hw desc commands,
763                  * link tables
764                  */
765                 edesc = ahash_edesc_alloc(ctx, sec4_sg_src_index + mapped_nents,
766                                           ctx->sh_desc_update,
767                                           ctx->sh_desc_update_dma, flags);
768                 if (!edesc) {
769                         dma_unmap_sg(jrdev, req->src, src_nents, DMA_TO_DEVICE);
770                         return -ENOMEM;
771                 }
772
773                 edesc->src_nents = src_nents;
774                 edesc->sec4_sg_bytes = sec4_sg_bytes;
775
776                 ret = ctx_map_to_sec4_sg(desc, jrdev, state, ctx->ctx_len,
777                                          edesc->sec4_sg, DMA_BIDIRECTIONAL);
778                 if (ret)
779                         goto unmap_ctx;
780
781                 ret = buf_map_to_sec4_sg(jrdev, edesc->sec4_sg + 1, state);
782                 if (ret)
783                         goto unmap_ctx;
784
785                 if (mapped_nents) {
786                         sg_to_sec4_sg_last(req->src, mapped_nents,
787                                            edesc->sec4_sg + sec4_sg_src_index,
788                                            0);
789                         if (*next_buflen)
790                                 scatterwalk_map_and_copy(next_buf, req->src,
791                                                          to_hash - *buflen,
792                                                          *next_buflen, 0);
793                 } else {
794                         sg_to_sec4_set_last(edesc->sec4_sg + sec4_sg_src_index -
795                                             1);
796                 }
797
798                 desc = edesc->hw_desc;
799
800                 edesc->sec4_sg_dma = dma_map_single(jrdev, edesc->sec4_sg,
801                                                      sec4_sg_bytes,
802                                                      DMA_TO_DEVICE);
803                 if (dma_mapping_error(jrdev, edesc->sec4_sg_dma)) {
804                         dev_err(jrdev, "unable to map S/G table\n");
805                         ret = -ENOMEM;
806                         goto unmap_ctx;
807                 }
808
809                 append_seq_in_ptr(desc, edesc->sec4_sg_dma, ctx->ctx_len +
810                                        to_hash, LDST_SGF);
811
812                 append_seq_out_ptr(desc, state->ctx_dma, ctx->ctx_len, 0);
813
814 #ifdef DEBUG
815                 print_hex_dump(KERN_ERR, "jobdesc@"__stringify(__LINE__)": ",
816                                DUMP_PREFIX_ADDRESS, 16, 4, desc,
817                                desc_bytes(desc), 1);
818 #endif
819
820                 ret = caam_jr_enqueue(jrdev, desc, ahash_done_bi, req);
821                 if (ret)
822                         goto unmap_ctx;
823
824                 ret = -EINPROGRESS;
825         } else if (*next_buflen) {
826                 scatterwalk_map_and_copy(buf + *buflen, req->src, 0,
827                                          req->nbytes, 0);
828                 *buflen = *next_buflen;
829                 *next_buflen = last_buflen;
830         }
831 #ifdef DEBUG
832         print_hex_dump(KERN_ERR, "buf@"__stringify(__LINE__)": ",
833                        DUMP_PREFIX_ADDRESS, 16, 4, buf, *buflen, 1);
834         print_hex_dump(KERN_ERR, "next buf@"__stringify(__LINE__)": ",
835                        DUMP_PREFIX_ADDRESS, 16, 4, next_buf,
836                        *next_buflen, 1);
837 #endif
838
839         return ret;
840  unmap_ctx:
841         ahash_unmap_ctx(jrdev, edesc, req, ctx->ctx_len, DMA_BIDIRECTIONAL);
842         kfree(edesc);
843         return ret;
844 }
845
846 static int ahash_final_ctx(struct ahash_request *req)
847 {
848         struct crypto_ahash *ahash = crypto_ahash_reqtfm(req);
849         struct caam_hash_ctx *ctx = crypto_ahash_ctx(ahash);
850         struct caam_hash_state *state = ahash_request_ctx(req);
851         struct device *jrdev = ctx->jrdev;
852         gfp_t flags = (req->base.flags & CRYPTO_TFM_REQ_MAY_SLEEP) ?
853                        GFP_KERNEL : GFP_ATOMIC;
854         int buflen = *current_buflen(state);
855         u32 *desc;
856         int sec4_sg_bytes, sec4_sg_src_index;
857         int digestsize = crypto_ahash_digestsize(ahash);
858         struct ahash_edesc *edesc;
859         int ret;
860
861         sec4_sg_src_index = 1 + (buflen ? 1 : 0);
862         sec4_sg_bytes = sec4_sg_src_index * sizeof(struct sec4_sg_entry);
863
864         /* allocate space for base edesc and hw desc commands, link tables */
865         edesc = ahash_edesc_alloc(ctx, sec4_sg_src_index,
866                                   ctx->sh_desc_fin, ctx->sh_desc_fin_dma,
867                                   flags);
868         if (!edesc)
869                 return -ENOMEM;
870
871         desc = edesc->hw_desc;
872
873         edesc->sec4_sg_bytes = sec4_sg_bytes;
874
875         ret = ctx_map_to_sec4_sg(desc, jrdev, state, ctx->ctx_len,
876                                  edesc->sec4_sg, DMA_TO_DEVICE);
877         if (ret)
878                 goto unmap_ctx;
879
880         ret = buf_map_to_sec4_sg(jrdev, edesc->sec4_sg + 1, state);
881         if (ret)
882                 goto unmap_ctx;
883
884         sg_to_sec4_set_last(edesc->sec4_sg + sec4_sg_src_index - 1);
885
886         edesc->sec4_sg_dma = dma_map_single(jrdev, edesc->sec4_sg,
887                                             sec4_sg_bytes, DMA_TO_DEVICE);
888         if (dma_mapping_error(jrdev, edesc->sec4_sg_dma)) {
889                 dev_err(jrdev, "unable to map S/G table\n");
890                 ret = -ENOMEM;
891                 goto unmap_ctx;
892         }
893
894         append_seq_in_ptr(desc, edesc->sec4_sg_dma, ctx->ctx_len + buflen,
895                           LDST_SGF);
896
897         edesc->dst_dma = map_seq_out_ptr_result(desc, jrdev, req->result,
898                                                 digestsize);
899         if (dma_mapping_error(jrdev, edesc->dst_dma)) {
900                 dev_err(jrdev, "unable to map dst\n");
901                 ret = -ENOMEM;
902                 goto unmap_ctx;
903         }
904
905 #ifdef DEBUG
906         print_hex_dump(KERN_ERR, "jobdesc@"__stringify(__LINE__)": ",
907                        DUMP_PREFIX_ADDRESS, 16, 4, desc, desc_bytes(desc), 1);
908 #endif
909
910         ret = caam_jr_enqueue(jrdev, desc, ahash_done_ctx_src, req);
911         if (ret)
912                 goto unmap_ctx;
913
914         return -EINPROGRESS;
915  unmap_ctx:
916         ahash_unmap_ctx(jrdev, edesc, req, digestsize, DMA_FROM_DEVICE);
917         kfree(edesc);
918         return ret;
919 }
920
921 static int ahash_finup_ctx(struct ahash_request *req)
922 {
923         struct crypto_ahash *ahash = crypto_ahash_reqtfm(req);
924         struct caam_hash_ctx *ctx = crypto_ahash_ctx(ahash);
925         struct caam_hash_state *state = ahash_request_ctx(req);
926         struct device *jrdev = ctx->jrdev;
927         gfp_t flags = (req->base.flags & CRYPTO_TFM_REQ_MAY_SLEEP) ?
928                        GFP_KERNEL : GFP_ATOMIC;
929         int buflen = *current_buflen(state);
930         u32 *desc;
931         int sec4_sg_src_index;
932         int src_nents, mapped_nents;
933         int digestsize = crypto_ahash_digestsize(ahash);
934         struct ahash_edesc *edesc;
935         int ret;
936
937         src_nents = sg_nents_for_len(req->src, req->nbytes);
938         if (src_nents < 0) {
939                 dev_err(jrdev, "Invalid number of src SG.\n");
940                 return src_nents;
941         }
942
943         if (src_nents) {
944                 mapped_nents = dma_map_sg(jrdev, req->src, src_nents,
945                                           DMA_TO_DEVICE);
946                 if (!mapped_nents) {
947                         dev_err(jrdev, "unable to DMA map source\n");
948                         return -ENOMEM;
949                 }
950         } else {
951                 mapped_nents = 0;
952         }
953
954         sec4_sg_src_index = 1 + (buflen ? 1 : 0);
955
956         /* allocate space for base edesc and hw desc commands, link tables */
957         edesc = ahash_edesc_alloc(ctx, sec4_sg_src_index + mapped_nents,
958                                   ctx->sh_desc_fin, ctx->sh_desc_fin_dma,
959                                   flags);
960         if (!edesc) {
961                 dma_unmap_sg(jrdev, req->src, src_nents, DMA_TO_DEVICE);
962                 return -ENOMEM;
963         }
964
965         desc = edesc->hw_desc;
966
967         edesc->src_nents = src_nents;
968
969         ret = ctx_map_to_sec4_sg(desc, jrdev, state, ctx->ctx_len,
970                                  edesc->sec4_sg, DMA_TO_DEVICE);
971         if (ret)
972                 goto unmap_ctx;
973
974         ret = buf_map_to_sec4_sg(jrdev, edesc->sec4_sg + 1, state);
975         if (ret)
976                 goto unmap_ctx;
977
978         ret = ahash_edesc_add_src(ctx, edesc, req, mapped_nents,
979                                   sec4_sg_src_index, ctx->ctx_len + buflen,
980                                   req->nbytes);
981         if (ret)
982                 goto unmap_ctx;
983
984         edesc->dst_dma = map_seq_out_ptr_result(desc, jrdev, req->result,
985                                                 digestsize);
986         if (dma_mapping_error(jrdev, edesc->dst_dma)) {
987                 dev_err(jrdev, "unable to map dst\n");
988                 ret = -ENOMEM;
989                 goto unmap_ctx;
990         }
991
992 #ifdef DEBUG
993         print_hex_dump(KERN_ERR, "jobdesc@"__stringify(__LINE__)": ",
994                        DUMP_PREFIX_ADDRESS, 16, 4, desc, desc_bytes(desc), 1);
995 #endif
996
997         ret = caam_jr_enqueue(jrdev, desc, ahash_done_ctx_src, req);
998         if (ret)
999                 goto unmap_ctx;
1000
1001         return -EINPROGRESS;
1002  unmap_ctx:
1003         ahash_unmap_ctx(jrdev, edesc, req, digestsize, DMA_FROM_DEVICE);
1004         kfree(edesc);
1005         return ret;
1006 }
1007
1008 static int ahash_digest(struct ahash_request *req)
1009 {
1010         struct crypto_ahash *ahash = crypto_ahash_reqtfm(req);
1011         struct caam_hash_ctx *ctx = crypto_ahash_ctx(ahash);
1012         struct caam_hash_state *state = ahash_request_ctx(req);
1013         struct device *jrdev = ctx->jrdev;
1014         gfp_t flags = (req->base.flags & CRYPTO_TFM_REQ_MAY_SLEEP) ?
1015                        GFP_KERNEL : GFP_ATOMIC;
1016         u32 *desc;
1017         int digestsize = crypto_ahash_digestsize(ahash);
1018         int src_nents, mapped_nents;
1019         struct ahash_edesc *edesc;
1020         int ret;
1021
1022         state->buf_dma = 0;
1023
1024         src_nents = sg_nents_for_len(req->src, req->nbytes);
1025         if (src_nents < 0) {
1026                 dev_err(jrdev, "Invalid number of src SG.\n");
1027                 return src_nents;
1028         }
1029
1030         if (src_nents) {
1031                 mapped_nents = dma_map_sg(jrdev, req->src, src_nents,
1032                                           DMA_TO_DEVICE);
1033                 if (!mapped_nents) {
1034                         dev_err(jrdev, "unable to map source for DMA\n");
1035                         return -ENOMEM;
1036                 }
1037         } else {
1038                 mapped_nents = 0;
1039         }
1040
1041         /* allocate space for base edesc and hw desc commands, link tables */
1042         edesc = ahash_edesc_alloc(ctx, mapped_nents > 1 ? mapped_nents : 0,
1043                                   ctx->sh_desc_digest, ctx->sh_desc_digest_dma,
1044                                   flags);
1045         if (!edesc) {
1046                 dma_unmap_sg(jrdev, req->src, src_nents, DMA_TO_DEVICE);
1047                 return -ENOMEM;
1048         }
1049
1050         edesc->src_nents = src_nents;
1051
1052         ret = ahash_edesc_add_src(ctx, edesc, req, mapped_nents, 0, 0,
1053                                   req->nbytes);
1054         if (ret) {
1055                 ahash_unmap(jrdev, edesc, req, digestsize);
1056                 kfree(edesc);
1057                 return ret;
1058         }
1059
1060         desc = edesc->hw_desc;
1061
1062         edesc->dst_dma = map_seq_out_ptr_result(desc, jrdev, req->result,
1063                                                 digestsize);
1064         if (dma_mapping_error(jrdev, edesc->dst_dma)) {
1065                 dev_err(jrdev, "unable to map dst\n");
1066                 ahash_unmap(jrdev, edesc, req, digestsize);
1067                 kfree(edesc);
1068                 return -ENOMEM;
1069         }
1070
1071 #ifdef DEBUG
1072         print_hex_dump(KERN_ERR, "jobdesc@"__stringify(__LINE__)": ",
1073                        DUMP_PREFIX_ADDRESS, 16, 4, desc, desc_bytes(desc), 1);
1074 #endif
1075
1076         ret = caam_jr_enqueue(jrdev, desc, ahash_done, req);
1077         if (!ret) {
1078                 ret = -EINPROGRESS;
1079         } else {
1080                 ahash_unmap(jrdev, edesc, req, digestsize);
1081                 kfree(edesc);
1082         }
1083
1084         return ret;
1085 }
1086
1087 /* submit ahash final if it the first job descriptor */
1088 static int ahash_final_no_ctx(struct ahash_request *req)
1089 {
1090         struct crypto_ahash *ahash = crypto_ahash_reqtfm(req);
1091         struct caam_hash_ctx *ctx = crypto_ahash_ctx(ahash);
1092         struct caam_hash_state *state = ahash_request_ctx(req);
1093         struct device *jrdev = ctx->jrdev;
1094         gfp_t flags = (req->base.flags & CRYPTO_TFM_REQ_MAY_SLEEP) ?
1095                        GFP_KERNEL : GFP_ATOMIC;
1096         u8 *buf = current_buf(state);
1097         int buflen = *current_buflen(state);
1098         u32 *desc;
1099         int digestsize = crypto_ahash_digestsize(ahash);
1100         struct ahash_edesc *edesc;
1101         int ret;
1102
1103         /* allocate space for base edesc and hw desc commands, link tables */
1104         edesc = ahash_edesc_alloc(ctx, 0, ctx->sh_desc_digest,
1105                                   ctx->sh_desc_digest_dma, flags);
1106         if (!edesc)
1107                 return -ENOMEM;
1108
1109         desc = edesc->hw_desc;
1110
1111         state->buf_dma = dma_map_single(jrdev, buf, buflen, DMA_TO_DEVICE);
1112         if (dma_mapping_error(jrdev, state->buf_dma)) {
1113                 dev_err(jrdev, "unable to map src\n");
1114                 goto unmap;
1115         }
1116
1117         append_seq_in_ptr(desc, state->buf_dma, buflen, 0);
1118
1119         edesc->dst_dma = map_seq_out_ptr_result(desc, jrdev, req->result,
1120                                                 digestsize);
1121         if (dma_mapping_error(jrdev, edesc->dst_dma)) {
1122                 dev_err(jrdev, "unable to map dst\n");
1123                 goto unmap;
1124         }
1125
1126 #ifdef DEBUG
1127         print_hex_dump(KERN_ERR, "jobdesc@"__stringify(__LINE__)": ",
1128                        DUMP_PREFIX_ADDRESS, 16, 4, desc, desc_bytes(desc), 1);
1129 #endif
1130
1131         ret = caam_jr_enqueue(jrdev, desc, ahash_done, req);
1132         if (!ret) {
1133                 ret = -EINPROGRESS;
1134         } else {
1135                 ahash_unmap(jrdev, edesc, req, digestsize);
1136                 kfree(edesc);
1137         }
1138
1139         return ret;
1140  unmap:
1141         ahash_unmap(jrdev, edesc, req, digestsize);
1142         kfree(edesc);
1143         return -ENOMEM;
1144
1145 }
1146
1147 /* submit ahash update if it the first job descriptor after update */
1148 static int ahash_update_no_ctx(struct ahash_request *req)
1149 {
1150         struct crypto_ahash *ahash = crypto_ahash_reqtfm(req);
1151         struct caam_hash_ctx *ctx = crypto_ahash_ctx(ahash);
1152         struct caam_hash_state *state = ahash_request_ctx(req);
1153         struct device *jrdev = ctx->jrdev;
1154         gfp_t flags = (req->base.flags & CRYPTO_TFM_REQ_MAY_SLEEP) ?
1155                        GFP_KERNEL : GFP_ATOMIC;
1156         u8 *buf = current_buf(state);
1157         int *buflen = current_buflen(state);
1158         u8 *next_buf = alt_buf(state);
1159         int *next_buflen = alt_buflen(state);
1160         int in_len = *buflen + req->nbytes, to_hash;
1161         int sec4_sg_bytes, src_nents, mapped_nents;
1162         struct ahash_edesc *edesc;
1163         u32 *desc;
1164         int ret = 0;
1165
1166         *next_buflen = in_len & (crypto_tfm_alg_blocksize(&ahash->base) - 1);
1167         to_hash = in_len - *next_buflen;
1168
1169         if (to_hash) {
1170                 src_nents = sg_nents_for_len(req->src,
1171                                              req->nbytes - *next_buflen);
1172                 if (src_nents < 0) {
1173                         dev_err(jrdev, "Invalid number of src SG.\n");
1174                         return src_nents;
1175                 }
1176
1177                 if (src_nents) {
1178                         mapped_nents = dma_map_sg(jrdev, req->src, src_nents,
1179                                                   DMA_TO_DEVICE);
1180                         if (!mapped_nents) {
1181                                 dev_err(jrdev, "unable to DMA map source\n");
1182                                 return -ENOMEM;
1183                         }
1184                 } else {
1185                         mapped_nents = 0;
1186                 }
1187
1188                 sec4_sg_bytes = (1 + mapped_nents) *
1189                                 sizeof(struct sec4_sg_entry);
1190
1191                 /*
1192                  * allocate space for base edesc and hw desc commands,
1193                  * link tables
1194                  */
1195                 edesc = ahash_edesc_alloc(ctx, 1 + mapped_nents,
1196                                           ctx->sh_desc_update_first,
1197                                           ctx->sh_desc_update_first_dma,
1198                                           flags);
1199                 if (!edesc) {
1200                         dma_unmap_sg(jrdev, req->src, src_nents, DMA_TO_DEVICE);
1201                         return -ENOMEM;
1202                 }
1203
1204                 edesc->src_nents = src_nents;
1205                 edesc->sec4_sg_bytes = sec4_sg_bytes;
1206
1207                 ret = buf_map_to_sec4_sg(jrdev, edesc->sec4_sg, state);
1208                 if (ret)
1209                         goto unmap_ctx;
1210
1211                 sg_to_sec4_sg_last(req->src, mapped_nents,
1212                                    edesc->sec4_sg + 1, 0);
1213
1214                 if (*next_buflen) {
1215                         scatterwalk_map_and_copy(next_buf, req->src,
1216                                                  to_hash - *buflen,
1217                                                  *next_buflen, 0);
1218                 }
1219
1220                 desc = edesc->hw_desc;
1221
1222                 edesc->sec4_sg_dma = dma_map_single(jrdev, edesc->sec4_sg,
1223                                                     sec4_sg_bytes,
1224                                                     DMA_TO_DEVICE);
1225                 if (dma_mapping_error(jrdev, edesc->sec4_sg_dma)) {
1226                         dev_err(jrdev, "unable to map S/G table\n");
1227                         ret = -ENOMEM;
1228                         goto unmap_ctx;
1229                 }
1230
1231                 append_seq_in_ptr(desc, edesc->sec4_sg_dma, to_hash, LDST_SGF);
1232
1233                 ret = map_seq_out_ptr_ctx(desc, jrdev, state, ctx->ctx_len);
1234                 if (ret)
1235                         goto unmap_ctx;
1236
1237 #ifdef DEBUG
1238                 print_hex_dump(KERN_ERR, "jobdesc@"__stringify(__LINE__)": ",
1239                                DUMP_PREFIX_ADDRESS, 16, 4, desc,
1240                                desc_bytes(desc), 1);
1241 #endif
1242
1243                 ret = caam_jr_enqueue(jrdev, desc, ahash_done_ctx_dst, req);
1244                 if (ret)
1245                         goto unmap_ctx;
1246
1247                 ret = -EINPROGRESS;
1248                 state->update = ahash_update_ctx;
1249                 state->finup = ahash_finup_ctx;
1250                 state->final = ahash_final_ctx;
1251         } else if (*next_buflen) {
1252                 scatterwalk_map_and_copy(buf + *buflen, req->src, 0,
1253                                          req->nbytes, 0);
1254                 *buflen = *next_buflen;
1255                 *next_buflen = 0;
1256         }
1257 #ifdef DEBUG
1258         print_hex_dump(KERN_ERR, "buf@"__stringify(__LINE__)": ",
1259                        DUMP_PREFIX_ADDRESS, 16, 4, buf, *buflen, 1);
1260         print_hex_dump(KERN_ERR, "next buf@"__stringify(__LINE__)": ",
1261                        DUMP_PREFIX_ADDRESS, 16, 4, next_buf,
1262                        *next_buflen, 1);
1263 #endif
1264
1265         return ret;
1266  unmap_ctx:
1267         ahash_unmap_ctx(jrdev, edesc, req, ctx->ctx_len, DMA_TO_DEVICE);
1268         kfree(edesc);
1269         return ret;
1270 }
1271
1272 /* submit ahash finup if it the first job descriptor after update */
1273 static int ahash_finup_no_ctx(struct ahash_request *req)
1274 {
1275         struct crypto_ahash *ahash = crypto_ahash_reqtfm(req);
1276         struct caam_hash_ctx *ctx = crypto_ahash_ctx(ahash);
1277         struct caam_hash_state *state = ahash_request_ctx(req);
1278         struct device *jrdev = ctx->jrdev;
1279         gfp_t flags = (req->base.flags & CRYPTO_TFM_REQ_MAY_SLEEP) ?
1280                        GFP_KERNEL : GFP_ATOMIC;
1281         int buflen = *current_buflen(state);
1282         u32 *desc;
1283         int sec4_sg_bytes, sec4_sg_src_index, src_nents, mapped_nents;
1284         int digestsize = crypto_ahash_digestsize(ahash);
1285         struct ahash_edesc *edesc;
1286         int ret;
1287
1288         src_nents = sg_nents_for_len(req->src, req->nbytes);
1289         if (src_nents < 0) {
1290                 dev_err(jrdev, "Invalid number of src SG.\n");
1291                 return src_nents;
1292         }
1293
1294         if (src_nents) {
1295                 mapped_nents = dma_map_sg(jrdev, req->src, src_nents,
1296                                           DMA_TO_DEVICE);
1297                 if (!mapped_nents) {
1298                         dev_err(jrdev, "unable to DMA map source\n");
1299                         return -ENOMEM;
1300                 }
1301         } else {
1302                 mapped_nents = 0;
1303         }
1304
1305         sec4_sg_src_index = 2;
1306         sec4_sg_bytes = (sec4_sg_src_index + mapped_nents) *
1307                          sizeof(struct sec4_sg_entry);
1308
1309         /* allocate space for base edesc and hw desc commands, link tables */
1310         edesc = ahash_edesc_alloc(ctx, sec4_sg_src_index + mapped_nents,
1311                                   ctx->sh_desc_digest, ctx->sh_desc_digest_dma,
1312                                   flags);
1313         if (!edesc) {
1314                 dma_unmap_sg(jrdev, req->src, src_nents, DMA_TO_DEVICE);
1315                 return -ENOMEM;
1316         }
1317
1318         desc = edesc->hw_desc;
1319
1320         edesc->src_nents = src_nents;
1321         edesc->sec4_sg_bytes = sec4_sg_bytes;
1322
1323         ret = buf_map_to_sec4_sg(jrdev, edesc->sec4_sg, state);
1324         if (ret)
1325                 goto unmap;
1326
1327         ret = ahash_edesc_add_src(ctx, edesc, req, mapped_nents, 1, buflen,
1328                                   req->nbytes);
1329         if (ret) {
1330                 dev_err(jrdev, "unable to map S/G table\n");
1331                 goto unmap;
1332         }
1333
1334         edesc->dst_dma = map_seq_out_ptr_result(desc, jrdev, req->result,
1335                                                 digestsize);
1336         if (dma_mapping_error(jrdev, edesc->dst_dma)) {
1337                 dev_err(jrdev, "unable to map dst\n");
1338                 goto unmap;
1339         }
1340
1341 #ifdef DEBUG
1342         print_hex_dump(KERN_ERR, "jobdesc@"__stringify(__LINE__)": ",
1343                        DUMP_PREFIX_ADDRESS, 16, 4, desc, desc_bytes(desc), 1);
1344 #endif
1345
1346         ret = caam_jr_enqueue(jrdev, desc, ahash_done, req);
1347         if (!ret) {
1348                 ret = -EINPROGRESS;
1349         } else {
1350                 ahash_unmap(jrdev, edesc, req, digestsize);
1351                 kfree(edesc);
1352         }
1353
1354         return ret;
1355  unmap:
1356         ahash_unmap(jrdev, edesc, req, digestsize);
1357         kfree(edesc);
1358         return -ENOMEM;
1359
1360 }
1361
1362 /* submit first update job descriptor after init */
1363 static int ahash_update_first(struct ahash_request *req)
1364 {
1365         struct crypto_ahash *ahash = crypto_ahash_reqtfm(req);
1366         struct caam_hash_ctx *ctx = crypto_ahash_ctx(ahash);
1367         struct caam_hash_state *state = ahash_request_ctx(req);
1368         struct device *jrdev = ctx->jrdev;
1369         gfp_t flags = (req->base.flags & CRYPTO_TFM_REQ_MAY_SLEEP) ?
1370                        GFP_KERNEL : GFP_ATOMIC;
1371         u8 *next_buf = alt_buf(state);
1372         int *next_buflen = alt_buflen(state);
1373         int to_hash;
1374         u32 *desc;
1375         int src_nents, mapped_nents;
1376         struct ahash_edesc *edesc;
1377         int ret = 0;
1378
1379         *next_buflen = req->nbytes & (crypto_tfm_alg_blocksize(&ahash->base) -
1380                                       1);
1381         to_hash = req->nbytes - *next_buflen;
1382
1383         if (to_hash) {
1384                 src_nents = sg_nents_for_len(req->src,
1385                                              req->nbytes - *next_buflen);
1386                 if (src_nents < 0) {
1387                         dev_err(jrdev, "Invalid number of src SG.\n");
1388                         return src_nents;
1389                 }
1390
1391                 if (src_nents) {
1392                         mapped_nents = dma_map_sg(jrdev, req->src, src_nents,
1393                                                   DMA_TO_DEVICE);
1394                         if (!mapped_nents) {
1395                                 dev_err(jrdev, "unable to map source for DMA\n");
1396                                 return -ENOMEM;
1397                         }
1398                 } else {
1399                         mapped_nents = 0;
1400                 }
1401
1402                 /*
1403                  * allocate space for base edesc and hw desc commands,
1404                  * link tables
1405                  */
1406                 edesc = ahash_edesc_alloc(ctx, mapped_nents > 1 ?
1407                                           mapped_nents : 0,
1408                                           ctx->sh_desc_update_first,
1409                                           ctx->sh_desc_update_first_dma,
1410                                           flags);
1411                 if (!edesc) {
1412                         dma_unmap_sg(jrdev, req->src, src_nents, DMA_TO_DEVICE);
1413                         return -ENOMEM;
1414                 }
1415
1416                 edesc->src_nents = src_nents;
1417
1418                 ret = ahash_edesc_add_src(ctx, edesc, req, mapped_nents, 0, 0,
1419                                           to_hash);
1420                 if (ret)
1421                         goto unmap_ctx;
1422
1423                 if (*next_buflen)
1424                         scatterwalk_map_and_copy(next_buf, req->src, to_hash,
1425                                                  *next_buflen, 0);
1426
1427                 desc = edesc->hw_desc;
1428
1429                 ret = map_seq_out_ptr_ctx(desc, jrdev, state, ctx->ctx_len);
1430                 if (ret)
1431                         goto unmap_ctx;
1432
1433 #ifdef DEBUG
1434                 print_hex_dump(KERN_ERR, "jobdesc@"__stringify(__LINE__)": ",
1435                                DUMP_PREFIX_ADDRESS, 16, 4, desc,
1436                                desc_bytes(desc), 1);
1437 #endif
1438
1439                 ret = caam_jr_enqueue(jrdev, desc, ahash_done_ctx_dst, req);
1440                 if (ret)
1441                         goto unmap_ctx;
1442
1443                 ret = -EINPROGRESS;
1444                 state->update = ahash_update_ctx;
1445                 state->finup = ahash_finup_ctx;
1446                 state->final = ahash_final_ctx;
1447         } else if (*next_buflen) {
1448                 state->update = ahash_update_no_ctx;
1449                 state->finup = ahash_finup_no_ctx;
1450                 state->final = ahash_final_no_ctx;
1451                 scatterwalk_map_and_copy(next_buf, req->src, 0,
1452                                          req->nbytes, 0);
1453                 switch_buf(state);
1454         }
1455 #ifdef DEBUG
1456         print_hex_dump(KERN_ERR, "next buf@"__stringify(__LINE__)": ",
1457                        DUMP_PREFIX_ADDRESS, 16, 4, next_buf,
1458                        *next_buflen, 1);
1459 #endif
1460
1461         return ret;
1462  unmap_ctx:
1463         ahash_unmap_ctx(jrdev, edesc, req, ctx->ctx_len, DMA_TO_DEVICE);
1464         kfree(edesc);
1465         return ret;
1466 }
1467
1468 static int ahash_finup_first(struct ahash_request *req)
1469 {
1470         return ahash_digest(req);
1471 }
1472
1473 static int ahash_init(struct ahash_request *req)
1474 {
1475         struct caam_hash_state *state = ahash_request_ctx(req);
1476
1477         state->update = ahash_update_first;
1478         state->finup = ahash_finup_first;
1479         state->final = ahash_final_no_ctx;
1480
1481         state->ctx_dma = 0;
1482         state->current_buf = 0;
1483         state->buf_dma = 0;
1484         state->buflen_0 = 0;
1485         state->buflen_1 = 0;
1486
1487         return 0;
1488 }
1489
1490 static int ahash_update(struct ahash_request *req)
1491 {
1492         struct caam_hash_state *state = ahash_request_ctx(req);
1493
1494         return state->update(req);
1495 }
1496
1497 static int ahash_finup(struct ahash_request *req)
1498 {
1499         struct caam_hash_state *state = ahash_request_ctx(req);
1500
1501         return state->finup(req);
1502 }
1503
1504 static int ahash_final(struct ahash_request *req)
1505 {
1506         struct caam_hash_state *state = ahash_request_ctx(req);
1507
1508         return state->final(req);
1509 }
1510
1511 static int ahash_export(struct ahash_request *req, void *out)
1512 {
1513         struct caam_hash_state *state = ahash_request_ctx(req);
1514         struct caam_export_state *export = out;
1515         int len;
1516         u8 *buf;
1517
1518         if (state->current_buf) {
1519                 buf = state->buf_1;
1520                 len = state->buflen_1;
1521         } else {
1522                 buf = state->buf_0;
1523                 len = state->buflen_0;
1524         }
1525
1526         memcpy(export->buf, buf, len);
1527         memcpy(export->caam_ctx, state->caam_ctx, sizeof(export->caam_ctx));
1528         export->buflen = len;
1529         export->update = state->update;
1530         export->final = state->final;
1531         export->finup = state->finup;
1532
1533         return 0;
1534 }
1535
1536 static int ahash_import(struct ahash_request *req, const void *in)
1537 {
1538         struct caam_hash_state *state = ahash_request_ctx(req);
1539         const struct caam_export_state *export = in;
1540
1541         memset(state, 0, sizeof(*state));
1542         memcpy(state->buf_0, export->buf, export->buflen);
1543         memcpy(state->caam_ctx, export->caam_ctx, sizeof(state->caam_ctx));
1544         state->buflen_0 = export->buflen;
1545         state->update = export->update;
1546         state->final = export->final;
1547         state->finup = export->finup;
1548
1549         return 0;
1550 }
1551
1552 struct caam_hash_template {
1553         char name[CRYPTO_MAX_ALG_NAME];
1554         char driver_name[CRYPTO_MAX_ALG_NAME];
1555         char hmac_name[CRYPTO_MAX_ALG_NAME];
1556         char hmac_driver_name[CRYPTO_MAX_ALG_NAME];
1557         unsigned int blocksize;
1558         struct ahash_alg template_ahash;
1559         u32 alg_type;
1560 };
1561
1562 /* ahash descriptors */
1563 static struct caam_hash_template driver_hash[] = {
1564         {
1565                 .name = "sha1",
1566                 .driver_name = "sha1-caam",
1567                 .hmac_name = "hmac(sha1)",
1568                 .hmac_driver_name = "hmac-sha1-caam",
1569                 .blocksize = SHA1_BLOCK_SIZE,
1570                 .template_ahash = {
1571                         .init = ahash_init,
1572                         .update = ahash_update,
1573                         .final = ahash_final,
1574                         .finup = ahash_finup,
1575                         .digest = ahash_digest,
1576                         .export = ahash_export,
1577                         .import = ahash_import,
1578                         .setkey = ahash_setkey,
1579                         .halg = {
1580                                 .digestsize = SHA1_DIGEST_SIZE,
1581                                 .statesize = sizeof(struct caam_export_state),
1582                         },
1583                 },
1584                 .alg_type = OP_ALG_ALGSEL_SHA1,
1585         }, {
1586                 .name = "sha224",
1587                 .driver_name = "sha224-caam",
1588                 .hmac_name = "hmac(sha224)",
1589                 .hmac_driver_name = "hmac-sha224-caam",
1590                 .blocksize = SHA224_BLOCK_SIZE,
1591                 .template_ahash = {
1592                         .init = ahash_init,
1593                         .update = ahash_update,
1594                         .final = ahash_final,
1595                         .finup = ahash_finup,
1596                         .digest = ahash_digest,
1597                         .export = ahash_export,
1598                         .import = ahash_import,
1599                         .setkey = ahash_setkey,
1600                         .halg = {
1601                                 .digestsize = SHA224_DIGEST_SIZE,
1602                                 .statesize = sizeof(struct caam_export_state),
1603                         },
1604                 },
1605                 .alg_type = OP_ALG_ALGSEL_SHA224,
1606         }, {
1607                 .name = "sha256",
1608                 .driver_name = "sha256-caam",
1609                 .hmac_name = "hmac(sha256)",
1610                 .hmac_driver_name = "hmac-sha256-caam",
1611                 .blocksize = SHA256_BLOCK_SIZE,
1612                 .template_ahash = {
1613                         .init = ahash_init,
1614                         .update = ahash_update,
1615                         .final = ahash_final,
1616                         .finup = ahash_finup,
1617                         .digest = ahash_digest,
1618                         .export = ahash_export,
1619                         .import = ahash_import,
1620                         .setkey = ahash_setkey,
1621                         .halg = {
1622                                 .digestsize = SHA256_DIGEST_SIZE,
1623                                 .statesize = sizeof(struct caam_export_state),
1624                         },
1625                 },
1626                 .alg_type = OP_ALG_ALGSEL_SHA256,
1627         }, {
1628                 .name = "sha384",
1629                 .driver_name = "sha384-caam",
1630                 .hmac_name = "hmac(sha384)",
1631                 .hmac_driver_name = "hmac-sha384-caam",
1632                 .blocksize = SHA384_BLOCK_SIZE,
1633                 .template_ahash = {
1634                         .init = ahash_init,
1635                         .update = ahash_update,
1636                         .final = ahash_final,
1637                         .finup = ahash_finup,
1638                         .digest = ahash_digest,
1639                         .export = ahash_export,
1640                         .import = ahash_import,
1641                         .setkey = ahash_setkey,
1642                         .halg = {
1643                                 .digestsize = SHA384_DIGEST_SIZE,
1644                                 .statesize = sizeof(struct caam_export_state),
1645                         },
1646                 },
1647                 .alg_type = OP_ALG_ALGSEL_SHA384,
1648         }, {
1649                 .name = "sha512",
1650                 .driver_name = "sha512-caam",
1651                 .hmac_name = "hmac(sha512)",
1652                 .hmac_driver_name = "hmac-sha512-caam",
1653                 .blocksize = SHA512_BLOCK_SIZE,
1654                 .template_ahash = {
1655                         .init = ahash_init,
1656                         .update = ahash_update,
1657                         .final = ahash_final,
1658                         .finup = ahash_finup,
1659                         .digest = ahash_digest,
1660                         .export = ahash_export,
1661                         .import = ahash_import,
1662                         .setkey = ahash_setkey,
1663                         .halg = {
1664                                 .digestsize = SHA512_DIGEST_SIZE,
1665                                 .statesize = sizeof(struct caam_export_state),
1666                         },
1667                 },
1668                 .alg_type = OP_ALG_ALGSEL_SHA512,
1669         }, {
1670                 .name = "md5",
1671                 .driver_name = "md5-caam",
1672                 .hmac_name = "hmac(md5)",
1673                 .hmac_driver_name = "hmac-md5-caam",
1674                 .blocksize = MD5_BLOCK_WORDS * 4,
1675                 .template_ahash = {
1676                         .init = ahash_init,
1677                         .update = ahash_update,
1678                         .final = ahash_final,
1679                         .finup = ahash_finup,
1680                         .digest = ahash_digest,
1681                         .export = ahash_export,
1682                         .import = ahash_import,
1683                         .setkey = ahash_setkey,
1684                         .halg = {
1685                                 .digestsize = MD5_DIGEST_SIZE,
1686                                 .statesize = sizeof(struct caam_export_state),
1687                         },
1688                 },
1689                 .alg_type = OP_ALG_ALGSEL_MD5,
1690         },
1691 };
1692
1693 struct caam_hash_alg {
1694         struct list_head entry;
1695         int alg_type;
1696         struct ahash_alg ahash_alg;
1697 };
1698
1699 static int caam_hash_cra_init(struct crypto_tfm *tfm)
1700 {
1701         struct crypto_ahash *ahash = __crypto_ahash_cast(tfm);
1702         struct crypto_alg *base = tfm->__crt_alg;
1703         struct hash_alg_common *halg =
1704                  container_of(base, struct hash_alg_common, base);
1705         struct ahash_alg *alg =
1706                  container_of(halg, struct ahash_alg, halg);
1707         struct caam_hash_alg *caam_hash =
1708                  container_of(alg, struct caam_hash_alg, ahash_alg);
1709         struct caam_hash_ctx *ctx = crypto_tfm_ctx(tfm);
1710         /* Sizes for MDHA running digests: MD5, SHA1, 224, 256, 384, 512 */
1711         static const u8 runninglen[] = { HASH_MSG_LEN + MD5_DIGEST_SIZE,
1712                                          HASH_MSG_LEN + SHA1_DIGEST_SIZE,
1713                                          HASH_MSG_LEN + 32,
1714                                          HASH_MSG_LEN + SHA256_DIGEST_SIZE,
1715                                          HASH_MSG_LEN + 64,
1716                                          HASH_MSG_LEN + SHA512_DIGEST_SIZE };
1717         dma_addr_t dma_addr;
1718
1719         /*
1720          * Get a Job ring from Job Ring driver to ensure in-order
1721          * crypto request processing per tfm
1722          */
1723         ctx->jrdev = caam_jr_alloc();
1724         if (IS_ERR(ctx->jrdev)) {
1725                 pr_err("Job Ring Device allocation for transform failed\n");
1726                 return PTR_ERR(ctx->jrdev);
1727         }
1728
1729         dma_addr = dma_map_single_attrs(ctx->jrdev, ctx->sh_desc_update,
1730                                         offsetof(struct caam_hash_ctx,
1731                                                  sh_desc_update_dma),
1732                                         DMA_TO_DEVICE, DMA_ATTR_SKIP_CPU_SYNC);
1733         if (dma_mapping_error(ctx->jrdev, dma_addr)) {
1734                 dev_err(ctx->jrdev, "unable to map shared descriptors\n");
1735                 caam_jr_free(ctx->jrdev);
1736                 return -ENOMEM;
1737         }
1738
1739         ctx->sh_desc_update_dma = dma_addr;
1740         ctx->sh_desc_update_first_dma = dma_addr +
1741                                         offsetof(struct caam_hash_ctx,
1742                                                  sh_desc_update_first);
1743         ctx->sh_desc_fin_dma = dma_addr + offsetof(struct caam_hash_ctx,
1744                                                    sh_desc_fin);
1745         ctx->sh_desc_digest_dma = dma_addr + offsetof(struct caam_hash_ctx,
1746                                                       sh_desc_digest);
1747
1748         /* copy descriptor header template value */
1749         ctx->adata.algtype = OP_TYPE_CLASS2_ALG | caam_hash->alg_type;
1750
1751         ctx->ctx_len = runninglen[(ctx->adata.algtype &
1752                                    OP_ALG_ALGSEL_SUBMASK) >>
1753                                   OP_ALG_ALGSEL_SHIFT];
1754
1755         crypto_ahash_set_reqsize(__crypto_ahash_cast(tfm),
1756                                  sizeof(struct caam_hash_state));
1757         return ahash_set_sh_desc(ahash);
1758 }
1759
1760 static void caam_hash_cra_exit(struct crypto_tfm *tfm)
1761 {
1762         struct caam_hash_ctx *ctx = crypto_tfm_ctx(tfm);
1763
1764         dma_unmap_single_attrs(ctx->jrdev, ctx->sh_desc_update_dma,
1765                                offsetof(struct caam_hash_ctx,
1766                                         sh_desc_update_dma),
1767                                DMA_TO_DEVICE, DMA_ATTR_SKIP_CPU_SYNC);
1768         caam_jr_free(ctx->jrdev);
1769 }
1770
1771 static void __exit caam_algapi_hash_exit(void)
1772 {
1773         struct caam_hash_alg *t_alg, *n;
1774
1775         if (!hash_list.next)
1776                 return;
1777
1778         list_for_each_entry_safe(t_alg, n, &hash_list, entry) {
1779                 crypto_unregister_ahash(&t_alg->ahash_alg);
1780                 list_del(&t_alg->entry);
1781                 kfree(t_alg);
1782         }
1783 }
1784
1785 static struct caam_hash_alg *
1786 caam_hash_alloc(struct caam_hash_template *template,
1787                 bool keyed)
1788 {
1789         struct caam_hash_alg *t_alg;
1790         struct ahash_alg *halg;
1791         struct crypto_alg *alg;
1792
1793         t_alg = kzalloc(sizeof(*t_alg), GFP_KERNEL);
1794         if (!t_alg) {
1795                 pr_err("failed to allocate t_alg\n");
1796                 return ERR_PTR(-ENOMEM);
1797         }
1798
1799         t_alg->ahash_alg = template->template_ahash;
1800         halg = &t_alg->ahash_alg;
1801         alg = &halg->halg.base;
1802
1803         if (keyed) {
1804                 snprintf(alg->cra_name, CRYPTO_MAX_ALG_NAME, "%s",
1805                          template->hmac_name);
1806                 snprintf(alg->cra_driver_name, CRYPTO_MAX_ALG_NAME, "%s",
1807                          template->hmac_driver_name);
1808         } else {
1809                 snprintf(alg->cra_name, CRYPTO_MAX_ALG_NAME, "%s",
1810                          template->name);
1811                 snprintf(alg->cra_driver_name, CRYPTO_MAX_ALG_NAME, "%s",
1812                          template->driver_name);
1813                 t_alg->ahash_alg.setkey = NULL;
1814         }
1815         alg->cra_module = THIS_MODULE;
1816         alg->cra_init = caam_hash_cra_init;
1817         alg->cra_exit = caam_hash_cra_exit;
1818         alg->cra_ctxsize = sizeof(struct caam_hash_ctx);
1819         alg->cra_priority = CAAM_CRA_PRIORITY;
1820         alg->cra_blocksize = template->blocksize;
1821         alg->cra_alignmask = 0;
1822         alg->cra_flags = CRYPTO_ALG_ASYNC | CRYPTO_ALG_TYPE_AHASH;
1823         alg->cra_type = &crypto_ahash_type;
1824
1825         t_alg->alg_type = template->alg_type;
1826
1827         return t_alg;
1828 }
1829
1830 static int __init caam_algapi_hash_init(void)
1831 {
1832         struct device_node *dev_node;
1833         struct platform_device *pdev;
1834         struct device *ctrldev;
1835         int i = 0, err = 0;
1836         struct caam_drv_private *priv;
1837         unsigned int md_limit = SHA512_DIGEST_SIZE;
1838         u32 cha_inst, cha_vid;
1839
1840         dev_node = of_find_compatible_node(NULL, NULL, "fsl,sec-v4.0");
1841         if (!dev_node) {
1842                 dev_node = of_find_compatible_node(NULL, NULL, "fsl,sec4.0");
1843                 if (!dev_node)
1844                         return -ENODEV;
1845         }
1846
1847         pdev = of_find_device_by_node(dev_node);
1848         if (!pdev) {
1849                 of_node_put(dev_node);
1850                 return -ENODEV;
1851         }
1852
1853         ctrldev = &pdev->dev;
1854         priv = dev_get_drvdata(ctrldev);
1855         of_node_put(dev_node);
1856
1857         /*
1858          * If priv is NULL, it's probably because the caam driver wasn't
1859          * properly initialized (e.g. RNG4 init failed). Thus, bail out here.
1860          */
1861         if (!priv)
1862                 return -ENODEV;
1863
1864         /*
1865          * Register crypto algorithms the device supports.  First, identify
1866          * presence and attributes of MD block.
1867          */
1868         cha_vid = rd_reg32(&priv->ctrl->perfmon.cha_id_ls);
1869         cha_inst = rd_reg32(&priv->ctrl->perfmon.cha_num_ls);
1870
1871         /*
1872          * Skip registration of any hashing algorithms if MD block
1873          * is not present.
1874          */
1875         if (!((cha_inst & CHA_ID_LS_MD_MASK) >> CHA_ID_LS_MD_SHIFT))
1876                 return -ENODEV;
1877
1878         /* Limit digest size based on LP256 */
1879         if ((cha_vid & CHA_ID_LS_MD_MASK) == CHA_ID_LS_MD_LP256)
1880                 md_limit = SHA256_DIGEST_SIZE;
1881
1882         INIT_LIST_HEAD(&hash_list);
1883
1884         /* register crypto algorithms the device supports */
1885         for (i = 0; i < ARRAY_SIZE(driver_hash); i++) {
1886                 struct caam_hash_alg *t_alg;
1887                 struct caam_hash_template *alg = driver_hash + i;
1888
1889                 /* If MD size is not supported by device, skip registration */
1890                 if (alg->template_ahash.halg.digestsize > md_limit)
1891                         continue;
1892
1893                 /* register hmac version */
1894                 t_alg = caam_hash_alloc(alg, true);
1895                 if (IS_ERR(t_alg)) {
1896                         err = PTR_ERR(t_alg);
1897                         pr_warn("%s alg allocation failed\n", alg->driver_name);
1898                         continue;
1899                 }
1900
1901                 err = crypto_register_ahash(&t_alg->ahash_alg);
1902                 if (err) {
1903                         pr_warn("%s alg registration failed: %d\n",
1904                                 t_alg->ahash_alg.halg.base.cra_driver_name,
1905                                 err);
1906                         kfree(t_alg);
1907                 } else
1908                         list_add_tail(&t_alg->entry, &hash_list);
1909
1910                 /* register unkeyed version */
1911                 t_alg = caam_hash_alloc(alg, false);
1912                 if (IS_ERR(t_alg)) {
1913                         err = PTR_ERR(t_alg);
1914                         pr_warn("%s alg allocation failed\n", alg->driver_name);
1915                         continue;
1916                 }
1917
1918                 err = crypto_register_ahash(&t_alg->ahash_alg);
1919                 if (err) {
1920                         pr_warn("%s alg registration failed: %d\n",
1921                                 t_alg->ahash_alg.halg.base.cra_driver_name,
1922                                 err);
1923                         kfree(t_alg);
1924                 } else
1925                         list_add_tail(&t_alg->entry, &hash_list);
1926         }
1927
1928         return err;
1929 }
1930
1931 module_init(caam_algapi_hash_init);
1932 module_exit(caam_algapi_hash_exit);
1933
1934 MODULE_LICENSE("GPL");
1935 MODULE_DESCRIPTION("FSL CAAM support for ahash functions of crypto API");
1936 MODULE_AUTHOR("Freescale Semiconductor - NMG");