GNU Linux-libre 5.10.153-gnu1
[releases.git] / drivers / crypto / allwinner / sun8i-ss / sun8i-ss-hash.c
1 // SPDX-License-Identifier: GPL-2.0
2 /*
3  * sun8i-ss-hash.c - hardware cryptographic offloader for
4  * Allwinner A80/A83T SoC
5  *
6  * Copyright (C) 2015-2020 Corentin Labbe <clabbe@baylibre.com>
7  *
8  * This file add support for MD5 and SHA1/SHA224/SHA256.
9  *
10  * You could find the datasheet in Documentation/arm/sunxi.rst
11  */
12 #include <linux/bottom_half.h>
13 #include <linux/dma-mapping.h>
14 #include <linux/pm_runtime.h>
15 #include <linux/scatterlist.h>
16 #include <crypto/internal/hash.h>
17 #include <crypto/sha.h>
18 #include <crypto/md5.h>
19 #include "sun8i-ss.h"
20
21 int sun8i_ss_hash_crainit(struct crypto_tfm *tfm)
22 {
23         struct sun8i_ss_hash_tfm_ctx *op = crypto_tfm_ctx(tfm);
24         struct ahash_alg *alg = __crypto_ahash_alg(tfm->__crt_alg);
25         struct sun8i_ss_alg_template *algt;
26         int err;
27
28         memset(op, 0, sizeof(struct sun8i_ss_hash_tfm_ctx));
29
30         algt = container_of(alg, struct sun8i_ss_alg_template, alg.hash);
31         op->ss = algt->ss;
32
33         op->enginectx.op.do_one_request = sun8i_ss_hash_run;
34         op->enginectx.op.prepare_request = NULL;
35         op->enginectx.op.unprepare_request = NULL;
36
37         /* FALLBACK */
38         op->fallback_tfm = crypto_alloc_ahash(crypto_tfm_alg_name(tfm), 0,
39                                               CRYPTO_ALG_NEED_FALLBACK);
40         if (IS_ERR(op->fallback_tfm)) {
41                 dev_err(algt->ss->dev, "Fallback driver could no be loaded\n");
42                 return PTR_ERR(op->fallback_tfm);
43         }
44
45         if (algt->alg.hash.halg.statesize < crypto_ahash_statesize(op->fallback_tfm))
46                 algt->alg.hash.halg.statesize = crypto_ahash_statesize(op->fallback_tfm);
47
48         crypto_ahash_set_reqsize(__crypto_ahash_cast(tfm),
49                                  sizeof(struct sun8i_ss_hash_reqctx) +
50                                  crypto_ahash_reqsize(op->fallback_tfm));
51
52         dev_info(op->ss->dev, "Fallback for %s is %s\n",
53                  crypto_tfm_alg_driver_name(tfm),
54                  crypto_tfm_alg_driver_name(&op->fallback_tfm->base));
55         err = pm_runtime_get_sync(op->ss->dev);
56         if (err < 0)
57                 goto error_pm;
58         return 0;
59 error_pm:
60         pm_runtime_put_noidle(op->ss->dev);
61         crypto_free_ahash(op->fallback_tfm);
62         return err;
63 }
64
65 void sun8i_ss_hash_craexit(struct crypto_tfm *tfm)
66 {
67         struct sun8i_ss_hash_tfm_ctx *tfmctx = crypto_tfm_ctx(tfm);
68
69         crypto_free_ahash(tfmctx->fallback_tfm);
70         pm_runtime_put_sync_suspend(tfmctx->ss->dev);
71 }
72
73 int sun8i_ss_hash_init(struct ahash_request *areq)
74 {
75         struct sun8i_ss_hash_reqctx *rctx = ahash_request_ctx(areq);
76         struct crypto_ahash *tfm = crypto_ahash_reqtfm(areq);
77         struct sun8i_ss_hash_tfm_ctx *tfmctx = crypto_ahash_ctx(tfm);
78
79         memset(rctx, 0, sizeof(struct sun8i_ss_hash_reqctx));
80
81         ahash_request_set_tfm(&rctx->fallback_req, tfmctx->fallback_tfm);
82         rctx->fallback_req.base.flags = areq->base.flags & CRYPTO_TFM_REQ_MAY_SLEEP;
83
84         return crypto_ahash_init(&rctx->fallback_req);
85 }
86
87 int sun8i_ss_hash_export(struct ahash_request *areq, void *out)
88 {
89         struct sun8i_ss_hash_reqctx *rctx = ahash_request_ctx(areq);
90         struct crypto_ahash *tfm = crypto_ahash_reqtfm(areq);
91         struct sun8i_ss_hash_tfm_ctx *tfmctx = crypto_ahash_ctx(tfm);
92
93         ahash_request_set_tfm(&rctx->fallback_req, tfmctx->fallback_tfm);
94         rctx->fallback_req.base.flags = areq->base.flags & CRYPTO_TFM_REQ_MAY_SLEEP;
95
96         return crypto_ahash_export(&rctx->fallback_req, out);
97 }
98
99 int sun8i_ss_hash_import(struct ahash_request *areq, const void *in)
100 {
101         struct sun8i_ss_hash_reqctx *rctx = ahash_request_ctx(areq);
102         struct crypto_ahash *tfm = crypto_ahash_reqtfm(areq);
103         struct sun8i_ss_hash_tfm_ctx *tfmctx = crypto_ahash_ctx(tfm);
104
105         ahash_request_set_tfm(&rctx->fallback_req, tfmctx->fallback_tfm);
106         rctx->fallback_req.base.flags = areq->base.flags & CRYPTO_TFM_REQ_MAY_SLEEP;
107
108         return crypto_ahash_import(&rctx->fallback_req, in);
109 }
110
111 int sun8i_ss_hash_final(struct ahash_request *areq)
112 {
113         struct sun8i_ss_hash_reqctx *rctx = ahash_request_ctx(areq);
114         struct crypto_ahash *tfm = crypto_ahash_reqtfm(areq);
115         struct sun8i_ss_hash_tfm_ctx *tfmctx = crypto_ahash_ctx(tfm);
116 #ifdef CONFIG_CRYPTO_DEV_SUN8I_SS_DEBUG
117         struct ahash_alg *alg = __crypto_ahash_alg(tfm->base.__crt_alg);
118         struct sun8i_ss_alg_template *algt;
119 #endif
120
121         ahash_request_set_tfm(&rctx->fallback_req, tfmctx->fallback_tfm);
122         rctx->fallback_req.base.flags = areq->base.flags &
123                                         CRYPTO_TFM_REQ_MAY_SLEEP;
124         rctx->fallback_req.result = areq->result;
125
126 #ifdef CONFIG_CRYPTO_DEV_SUN8I_SS_DEBUG
127         algt = container_of(alg, struct sun8i_ss_alg_template, alg.hash);
128         algt->stat_fb++;
129 #endif
130
131         return crypto_ahash_final(&rctx->fallback_req);
132 }
133
134 int sun8i_ss_hash_update(struct ahash_request *areq)
135 {
136         struct sun8i_ss_hash_reqctx *rctx = ahash_request_ctx(areq);
137         struct crypto_ahash *tfm = crypto_ahash_reqtfm(areq);
138         struct sun8i_ss_hash_tfm_ctx *tfmctx = crypto_ahash_ctx(tfm);
139
140         ahash_request_set_tfm(&rctx->fallback_req, tfmctx->fallback_tfm);
141         rctx->fallback_req.base.flags = areq->base.flags &
142                                         CRYPTO_TFM_REQ_MAY_SLEEP;
143         rctx->fallback_req.nbytes = areq->nbytes;
144         rctx->fallback_req.src = areq->src;
145
146         return crypto_ahash_update(&rctx->fallback_req);
147 }
148
149 int sun8i_ss_hash_finup(struct ahash_request *areq)
150 {
151         struct sun8i_ss_hash_reqctx *rctx = ahash_request_ctx(areq);
152         struct crypto_ahash *tfm = crypto_ahash_reqtfm(areq);
153         struct sun8i_ss_hash_tfm_ctx *tfmctx = crypto_ahash_ctx(tfm);
154 #ifdef CONFIG_CRYPTO_DEV_SUN8I_SS_DEBUG
155         struct ahash_alg *alg = __crypto_ahash_alg(tfm->base.__crt_alg);
156         struct sun8i_ss_alg_template *algt;
157 #endif
158
159         ahash_request_set_tfm(&rctx->fallback_req, tfmctx->fallback_tfm);
160         rctx->fallback_req.base.flags = areq->base.flags &
161                                         CRYPTO_TFM_REQ_MAY_SLEEP;
162
163         rctx->fallback_req.nbytes = areq->nbytes;
164         rctx->fallback_req.src = areq->src;
165         rctx->fallback_req.result = areq->result;
166 #ifdef CONFIG_CRYPTO_DEV_SUN8I_SS_DEBUG
167         algt = container_of(alg, struct sun8i_ss_alg_template, alg.hash);
168         algt->stat_fb++;
169 #endif
170
171         return crypto_ahash_finup(&rctx->fallback_req);
172 }
173
174 static int sun8i_ss_hash_digest_fb(struct ahash_request *areq)
175 {
176         struct sun8i_ss_hash_reqctx *rctx = ahash_request_ctx(areq);
177         struct crypto_ahash *tfm = crypto_ahash_reqtfm(areq);
178         struct sun8i_ss_hash_tfm_ctx *tfmctx = crypto_ahash_ctx(tfm);
179 #ifdef CONFIG_CRYPTO_DEV_SUN8I_SS_DEBUG
180         struct ahash_alg *alg = __crypto_ahash_alg(tfm->base.__crt_alg);
181         struct sun8i_ss_alg_template *algt;
182 #endif
183
184         ahash_request_set_tfm(&rctx->fallback_req, tfmctx->fallback_tfm);
185         rctx->fallback_req.base.flags = areq->base.flags &
186                                         CRYPTO_TFM_REQ_MAY_SLEEP;
187
188         rctx->fallback_req.nbytes = areq->nbytes;
189         rctx->fallback_req.src = areq->src;
190         rctx->fallback_req.result = areq->result;
191 #ifdef CONFIG_CRYPTO_DEV_SUN8I_SS_DEBUG
192         algt = container_of(alg, struct sun8i_ss_alg_template, alg.hash);
193         algt->stat_fb++;
194 #endif
195
196         return crypto_ahash_digest(&rctx->fallback_req);
197 }
198
199 static int sun8i_ss_run_hash_task(struct sun8i_ss_dev *ss,
200                                   struct sun8i_ss_hash_reqctx *rctx,
201                                   const char *name)
202 {
203         int flow = rctx->flow;
204         u32 v = SS_START;
205         int i;
206
207 #ifdef CONFIG_CRYPTO_DEV_SUN8I_SS_DEBUG
208         ss->flows[flow].stat_req++;
209 #endif
210
211         /* choose between stream0/stream1 */
212         if (flow)
213                 v |= SS_FLOW1;
214         else
215                 v |= SS_FLOW0;
216
217         v |= rctx->method;
218
219         for (i = 0; i < MAX_SG; i++) {
220                 if (!rctx->t_dst[i].addr)
221                         break;
222
223                 mutex_lock(&ss->mlock);
224                 if (i > 0) {
225                         v |= BIT(17);
226                         writel(rctx->t_dst[i - 1].addr, ss->base + SS_KEY_ADR_REG);
227                         writel(rctx->t_dst[i - 1].addr, ss->base + SS_IV_ADR_REG);
228                 }
229
230                 dev_dbg(ss->dev,
231                         "Processing SG %d on flow %d %s ctl=%x %d to %d method=%x src=%x dst=%x\n",
232                         i, flow, name, v,
233                         rctx->t_src[i].len, rctx->t_dst[i].len,
234                         rctx->method, rctx->t_src[i].addr, rctx->t_dst[i].addr);
235
236                 writel(rctx->t_src[i].addr, ss->base + SS_SRC_ADR_REG);
237                 writel(rctx->t_dst[i].addr, ss->base + SS_DST_ADR_REG);
238                 writel(rctx->t_src[i].len, ss->base + SS_LEN_ADR_REG);
239                 writel(BIT(0) | BIT(1), ss->base + SS_INT_CTL_REG);
240
241                 reinit_completion(&ss->flows[flow].complete);
242                 ss->flows[flow].status = 0;
243                 wmb();
244
245                 writel(v, ss->base + SS_CTL_REG);
246                 mutex_unlock(&ss->mlock);
247                 wait_for_completion_interruptible_timeout(&ss->flows[flow].complete,
248                                                           msecs_to_jiffies(2000));
249                 if (ss->flows[flow].status == 0) {
250                         dev_err(ss->dev, "DMA timeout for %s\n", name);
251                         return -EFAULT;
252                 }
253         }
254
255         return 0;
256 }
257
258 static bool sun8i_ss_hash_need_fallback(struct ahash_request *areq)
259 {
260         struct scatterlist *sg;
261
262         if (areq->nbytes == 0)
263                 return true;
264         /* we need to reserve one SG for the padding one */
265         if (sg_nents(areq->src) > MAX_SG - 1)
266                 return true;
267         sg = areq->src;
268         while (sg) {
269                 /* SS can operate hash only on full block size
270                  * since SS support only MD5,sha1,sha224 and sha256, blocksize
271                  * is always 64
272                  * TODO: handle request if last SG is not len%64
273                  * but this will need to copy data on a new SG of size=64
274                  */
275                 if (sg->length % 64 || !IS_ALIGNED(sg->offset, sizeof(u32)))
276                         return true;
277                 sg = sg_next(sg);
278         }
279         return false;
280 }
281
282 int sun8i_ss_hash_digest(struct ahash_request *areq)
283 {
284         struct crypto_ahash *tfm = crypto_ahash_reqtfm(areq);
285         struct ahash_alg *alg = __crypto_ahash_alg(tfm->base.__crt_alg);
286         struct sun8i_ss_hash_reqctx *rctx = ahash_request_ctx(areq);
287         struct sun8i_ss_alg_template *algt;
288         struct sun8i_ss_dev *ss;
289         struct crypto_engine *engine;
290         struct scatterlist *sg;
291         int nr_sgs, e, i;
292
293         if (sun8i_ss_hash_need_fallback(areq))
294                 return sun8i_ss_hash_digest_fb(areq);
295
296         nr_sgs = sg_nents(areq->src);
297         if (nr_sgs > MAX_SG - 1)
298                 return sun8i_ss_hash_digest_fb(areq);
299
300         for_each_sg(areq->src, sg, nr_sgs, i) {
301                 if (sg->length % 4 || !IS_ALIGNED(sg->offset, sizeof(u32)))
302                         return sun8i_ss_hash_digest_fb(areq);
303         }
304
305         algt = container_of(alg, struct sun8i_ss_alg_template, alg.hash);
306         ss = algt->ss;
307
308         e = sun8i_ss_get_engine_number(ss);
309         rctx->flow = e;
310         engine = ss->flows[e].engine;
311
312         return crypto_transfer_hash_request_to_engine(engine, areq);
313 }
314
315 /* sun8i_ss_hash_run - run an ahash request
316  * Send the data of the request to the SS along with an extra SG with padding
317  */
318 int sun8i_ss_hash_run(struct crypto_engine *engine, void *breq)
319 {
320         struct ahash_request *areq = container_of(breq, struct ahash_request, base);
321         struct crypto_ahash *tfm = crypto_ahash_reqtfm(areq);
322         struct ahash_alg *alg = __crypto_ahash_alg(tfm->base.__crt_alg);
323         struct sun8i_ss_hash_reqctx *rctx = ahash_request_ctx(areq);
324         struct sun8i_ss_alg_template *algt;
325         struct sun8i_ss_dev *ss;
326         struct scatterlist *sg;
327         int nr_sgs, err, digestsize;
328         unsigned int len;
329         u64 fill, min_fill, byte_count;
330         void *pad, *result;
331         int j, i, todo;
332         __be64 *bebits;
333         __le64 *lebits;
334         dma_addr_t addr_res, addr_pad;
335         __le32 *bf;
336
337         algt = container_of(alg, struct sun8i_ss_alg_template, alg.hash);
338         ss = algt->ss;
339
340         digestsize = algt->alg.hash.halg.digestsize;
341         if (digestsize == SHA224_DIGEST_SIZE)
342                 digestsize = SHA256_DIGEST_SIZE;
343
344         result = ss->flows[rctx->flow].result;
345         pad = ss->flows[rctx->flow].pad;
346         memset(pad, 0, algt->alg.hash.halg.base.cra_blocksize * 2);
347         bf = (__le32 *)pad;
348
349         for (i = 0; i < MAX_SG; i++) {
350                 rctx->t_dst[i].addr = 0;
351                 rctx->t_dst[i].len = 0;
352         }
353
354 #ifdef CONFIG_CRYPTO_DEV_SUN8I_SS_DEBUG
355         algt->stat_req++;
356 #endif
357
358         rctx->method = ss->variant->alg_hash[algt->ss_algo_id];
359
360         nr_sgs = dma_map_sg(ss->dev, areq->src, sg_nents(areq->src), DMA_TO_DEVICE);
361         if (nr_sgs <= 0 || nr_sgs > MAX_SG) {
362                 dev_err(ss->dev, "Invalid sg number %d\n", nr_sgs);
363                 err = -EINVAL;
364                 goto theend;
365         }
366
367         addr_res = dma_map_single(ss->dev, result, digestsize, DMA_FROM_DEVICE);
368         if (dma_mapping_error(ss->dev, addr_res)) {
369                 dev_err(ss->dev, "DMA map dest\n");
370                 err = -EINVAL;
371                 goto theend;
372         }
373
374         len = areq->nbytes;
375         sg = areq->src;
376         i = 0;
377         while (len > 0 && sg) {
378                 if (sg_dma_len(sg) == 0) {
379                         sg = sg_next(sg);
380                         continue;
381                 }
382                 rctx->t_src[i].addr = sg_dma_address(sg);
383                 todo = min(len, sg_dma_len(sg));
384                 rctx->t_src[i].len = todo / 4;
385                 len -= todo;
386                 rctx->t_dst[i].addr = addr_res;
387                 rctx->t_dst[i].len = digestsize / 4;
388                 sg = sg_next(sg);
389                 i++;
390         }
391         if (len > 0) {
392                 dev_err(ss->dev, "remaining len %d\n", len);
393                 err = -EINVAL;
394                 goto theend;
395         }
396
397         byte_count = areq->nbytes;
398         j = 0;
399         bf[j++] = cpu_to_le32(0x80);
400
401         fill = 64 - (byte_count % 64);
402         min_fill = 3 * sizeof(u32);
403
404         if (fill < min_fill)
405                 fill += 64;
406
407         j += (fill - min_fill) / sizeof(u32);
408
409         switch (algt->ss_algo_id) {
410         case SS_ID_HASH_MD5:
411                 lebits = (__le64 *)&bf[j];
412                 *lebits = cpu_to_le64(byte_count << 3);
413                 j += 2;
414                 break;
415         case SS_ID_HASH_SHA1:
416         case SS_ID_HASH_SHA224:
417         case SS_ID_HASH_SHA256:
418                 bebits = (__be64 *)&bf[j];
419                 *bebits = cpu_to_be64(byte_count << 3);
420                 j += 2;
421                 break;
422         }
423
424         addr_pad = dma_map_single(ss->dev, pad, j * 4, DMA_TO_DEVICE);
425         rctx->t_src[i].addr = addr_pad;
426         rctx->t_src[i].len = j;
427         rctx->t_dst[i].addr = addr_res;
428         rctx->t_dst[i].len = digestsize / 4;
429         if (dma_mapping_error(ss->dev, addr_pad)) {
430                 dev_err(ss->dev, "DMA error on padding SG\n");
431                 err = -EINVAL;
432                 goto theend;
433         }
434
435         err = sun8i_ss_run_hash_task(ss, rctx, crypto_tfm_alg_name(areq->base.tfm));
436
437         dma_unmap_single(ss->dev, addr_pad, j * 4, DMA_TO_DEVICE);
438         dma_unmap_sg(ss->dev, areq->src, nr_sgs, DMA_TO_DEVICE);
439         dma_unmap_single(ss->dev, addr_res, digestsize, DMA_FROM_DEVICE);
440
441         memcpy(areq->result, result, algt->alg.hash.halg.digestsize);
442 theend:
443         local_bh_disable();
444         crypto_finalize_hash_request(engine, breq, err);
445         local_bh_enable();
446         return 0;
447 }