1// SPDX-License-Identifier: GPL-2.0-only
2/*
3 * AMD Cryptographic Coprocessor (CCP) AES XTS crypto API support
4 *
5 * Copyright (C) 2013,2017 Advanced Micro Devices, Inc.
6 *
7 * Author: Gary R Hook <gary.hook@amd.com>
8 * Author: Tom Lendacky <thomas.lendacky@amd.com>
9 */
10
11#include <linux/module.h>
12#include <linux/sched.h>
13#include <linux/delay.h>
14#include <linux/scatterlist.h>
15#include <crypto/aes.h>
16#include <crypto/xts.h>
17#include <crypto/internal/skcipher.h>
18#include <crypto/scatterwalk.h>
19
20#include "ccp-crypto.h"
21
22struct ccp_aes_xts_def {
23 const char *name;
24 const char *drv_name;
25};
26
27static const struct ccp_aes_xts_def aes_xts_algs[] = {
28 {
29 .name = "xts(aes)",
30 .drv_name = "xts-aes-ccp",
31 },
32};
33
34struct ccp_unit_size_map {
35 unsigned int size;
36 u32 value;
37};
38
39static struct ccp_unit_size_map xts_unit_sizes[] = {
40 {
41 .size = 16,
42 .value = CCP_XTS_AES_UNIT_SIZE_16,
43 },
44 {
45 .size = 512,
46 .value = CCP_XTS_AES_UNIT_SIZE_512,
47 },
48 {
49 .size = 1024,
50 .value = CCP_XTS_AES_UNIT_SIZE_1024,
51 },
52 {
53 .size = 2048,
54 .value = CCP_XTS_AES_UNIT_SIZE_2048,
55 },
56 {
57 .size = 4096,
58 .value = CCP_XTS_AES_UNIT_SIZE_4096,
59 },
60};
61
62static int ccp_aes_xts_complete(struct crypto_async_request *async_req, int ret)
63{
64 struct skcipher_request *req = skcipher_request_cast(req: async_req);
65 struct ccp_aes_req_ctx *rctx = skcipher_request_ctx_dma(req);
66
67 if (ret)
68 return ret;
69
70 memcpy(req->iv, rctx->iv, AES_BLOCK_SIZE);
71
72 return 0;
73}
74
75static int ccp_aes_xts_setkey(struct crypto_skcipher *tfm, const u8 *key,
76 unsigned int key_len)
77{
78 struct ccp_ctx *ctx = crypto_skcipher_ctx_dma(tfm);
79 unsigned int ccpversion = ccp_version();
80 int ret;
81
82 ret = xts_verify_key(tfm, key, keylen: key_len);
83 if (ret)
84 return ret;
85
86 /* Version 3 devices support 128-bit keys; version 5 devices can
87 * accommodate 128- and 256-bit keys.
88 */
89 switch (key_len) {
90 case AES_KEYSIZE_128 * 2:
91 memcpy(ctx->u.aes.key, key, key_len);
92 break;
93 case AES_KEYSIZE_256 * 2:
94 if (ccpversion > CCP_VERSION(3, 0))
95 memcpy(ctx->u.aes.key, key, key_len);
96 break;
97 }
98 ctx->u.aes.key_len = key_len / 2;
99 sg_init_one(&ctx->u.aes.key_sg, ctx->u.aes.key, key_len);
100
101 return crypto_skcipher_setkey(tfm: ctx->u.aes.tfm_skcipher, key, keylen: key_len);
102}
103
104static int ccp_aes_xts_crypt(struct skcipher_request *req,
105 unsigned int encrypt)
106{
107 struct crypto_skcipher *tfm = crypto_skcipher_reqtfm(req);
108 struct ccp_ctx *ctx = crypto_skcipher_ctx_dma(tfm);
109 struct ccp_aes_req_ctx *rctx = skcipher_request_ctx_dma(req);
110 unsigned int ccpversion = ccp_version();
111 unsigned int fallback = 0;
112 unsigned int unit;
113 u32 unit_size;
114 int ret;
115
116 if (!ctx->u.aes.key_len)
117 return -EINVAL;
118
119 if (!req->iv)
120 return -EINVAL;
121
122 /* Check conditions under which the CCP can fulfill a request. The
123 * device can handle input plaintext of a length that is a multiple
124 * of the unit_size, bug the crypto implementation only supports
125 * the unit_size being equal to the input length. This limits the
126 * number of scenarios we can handle.
127 */
128 unit_size = CCP_XTS_AES_UNIT_SIZE__LAST;
129 for (unit = 0; unit < ARRAY_SIZE(xts_unit_sizes); unit++) {
130 if (req->cryptlen == xts_unit_sizes[unit].size) {
131 unit_size = unit;
132 break;
133 }
134 }
135 /* The CCP has restrictions on block sizes. Also, a version 3 device
136 * only supports AES-128 operations; version 5 CCPs support both
137 * AES-128 and -256 operations.
138 */
139 if (unit_size == CCP_XTS_AES_UNIT_SIZE__LAST)
140 fallback = 1;
141 if ((ccpversion < CCP_VERSION(5, 0)) &&
142 (ctx->u.aes.key_len != AES_KEYSIZE_128))
143 fallback = 1;
144 if ((ctx->u.aes.key_len != AES_KEYSIZE_128) &&
145 (ctx->u.aes.key_len != AES_KEYSIZE_256))
146 fallback = 1;
147 if (fallback) {
148 /* Use the fallback to process the request for any
149 * unsupported unit sizes or key sizes
150 */
151 skcipher_request_set_tfm(req: &rctx->fallback_req,
152 tfm: ctx->u.aes.tfm_skcipher);
153 skcipher_request_set_callback(req: &rctx->fallback_req,
154 flags: req->base.flags,
155 compl: req->base.complete,
156 data: req->base.data);
157 skcipher_request_set_crypt(req: &rctx->fallback_req, src: req->src,
158 dst: req->dst, cryptlen: req->cryptlen, iv: req->iv);
159 ret = encrypt ? crypto_skcipher_encrypt(req: &rctx->fallback_req) :
160 crypto_skcipher_decrypt(req: &rctx->fallback_req);
161 return ret;
162 }
163
164 memcpy(rctx->iv, req->iv, AES_BLOCK_SIZE);
165 sg_init_one(&rctx->iv_sg, rctx->iv, AES_BLOCK_SIZE);
166
167 memset(&rctx->cmd, 0, sizeof(rctx->cmd));
168 INIT_LIST_HEAD(list: &rctx->cmd.entry);
169 rctx->cmd.engine = CCP_ENGINE_XTS_AES_128;
170 rctx->cmd.u.xts.type = CCP_AES_TYPE_128;
171 rctx->cmd.u.xts.action = (encrypt) ? CCP_AES_ACTION_ENCRYPT
172 : CCP_AES_ACTION_DECRYPT;
173 rctx->cmd.u.xts.unit_size = unit_size;
174 rctx->cmd.u.xts.key = &ctx->u.aes.key_sg;
175 rctx->cmd.u.xts.key_len = ctx->u.aes.key_len;
176 rctx->cmd.u.xts.iv = &rctx->iv_sg;
177 rctx->cmd.u.xts.iv_len = AES_BLOCK_SIZE;
178 rctx->cmd.u.xts.src = req->src;
179 rctx->cmd.u.xts.src_len = req->cryptlen;
180 rctx->cmd.u.xts.dst = req->dst;
181
182 ret = ccp_crypto_enqueue_request(req: &req->base, cmd: &rctx->cmd);
183
184 return ret;
185}
186
187static int ccp_aes_xts_encrypt(struct skcipher_request *req)
188{
189 return ccp_aes_xts_crypt(req, encrypt: 1);
190}
191
192static int ccp_aes_xts_decrypt(struct skcipher_request *req)
193{
194 return ccp_aes_xts_crypt(req, encrypt: 0);
195}
196
197static int ccp_aes_xts_init_tfm(struct crypto_skcipher *tfm)
198{
199 struct ccp_ctx *ctx = crypto_skcipher_ctx_dma(tfm);
200 struct crypto_skcipher *fallback_tfm;
201
202 ctx->complete = ccp_aes_xts_complete;
203 ctx->u.aes.key_len = 0;
204
205 fallback_tfm = crypto_alloc_skcipher(alg_name: "xts(aes)", type: 0,
206 CRYPTO_ALG_NEED_FALLBACK);
207 if (IS_ERR(ptr: fallback_tfm)) {
208 pr_warn("could not load fallback driver xts(aes)\n");
209 return PTR_ERR(ptr: fallback_tfm);
210 }
211 ctx->u.aes.tfm_skcipher = fallback_tfm;
212
213 crypto_skcipher_set_reqsize_dma(skcipher: tfm,
214 reqsize: sizeof(struct ccp_aes_req_ctx) +
215 crypto_skcipher_reqsize(tfm: fallback_tfm));
216
217 return 0;
218}
219
220static void ccp_aes_xts_exit_tfm(struct crypto_skcipher *tfm)
221{
222 struct ccp_ctx *ctx = crypto_skcipher_ctx_dma(tfm);
223
224 crypto_free_skcipher(tfm: ctx->u.aes.tfm_skcipher);
225}
226
227static int ccp_register_aes_xts_alg(struct list_head *head,
228 const struct ccp_aes_xts_def *def)
229{
230 struct ccp_crypto_skcipher_alg *ccp_alg;
231 struct skcipher_alg *alg;
232 int ret;
233
234 ccp_alg = kzalloc(size: sizeof(*ccp_alg), GFP_KERNEL);
235 if (!ccp_alg)
236 return -ENOMEM;
237
238 INIT_LIST_HEAD(list: &ccp_alg->entry);
239
240 alg = &ccp_alg->alg;
241
242 snprintf(buf: alg->base.cra_name, CRYPTO_MAX_ALG_NAME, fmt: "%s", def->name);
243 snprintf(buf: alg->base.cra_driver_name, CRYPTO_MAX_ALG_NAME, fmt: "%s",
244 def->drv_name);
245 alg->base.cra_flags = CRYPTO_ALG_ASYNC |
246 CRYPTO_ALG_ALLOCATES_MEMORY |
247 CRYPTO_ALG_KERN_DRIVER_ONLY |
248 CRYPTO_ALG_NEED_FALLBACK;
249 alg->base.cra_blocksize = AES_BLOCK_SIZE;
250 alg->base.cra_ctxsize = sizeof(struct ccp_ctx) +
251 crypto_dma_padding();
252 alg->base.cra_priority = CCP_CRA_PRIORITY;
253 alg->base.cra_module = THIS_MODULE;
254
255 alg->setkey = ccp_aes_xts_setkey;
256 alg->encrypt = ccp_aes_xts_encrypt;
257 alg->decrypt = ccp_aes_xts_decrypt;
258 alg->min_keysize = AES_MIN_KEY_SIZE * 2;
259 alg->max_keysize = AES_MAX_KEY_SIZE * 2;
260 alg->ivsize = AES_BLOCK_SIZE;
261 alg->init = ccp_aes_xts_init_tfm;
262 alg->exit = ccp_aes_xts_exit_tfm;
263
264 ret = crypto_register_skcipher(alg);
265 if (ret) {
266 pr_err("%s skcipher algorithm registration error (%d)\n",
267 alg->base.cra_name, ret);
268 kfree(objp: ccp_alg);
269 return ret;
270 }
271
272 list_add(new: &ccp_alg->entry, head);
273
274 return 0;
275}
276
277int ccp_register_aes_xts_algs(struct list_head *head)
278{
279 int i, ret;
280
281 for (i = 0; i < ARRAY_SIZE(aes_xts_algs); i++) {
282 ret = ccp_register_aes_xts_alg(head, def: &aes_xts_algs[i]);
283 if (ret)
284 return ret;
285 }
286
287 return 0;
288}
289

source code of linux/drivers/crypto/ccp/ccp-crypto-aes-xts.c