Steffen Klassert | 5068c7a | 2010-01-07 15:57:19 +1100 | [diff] [blame] | 1 | /* |
| 2 | * pcrypt - Parallel crypto wrapper. |
| 3 | * |
| 4 | * Copyright (C) 2009 secunet Security Networks AG |
| 5 | * Copyright (C) 2009 Steffen Klassert <steffen.klassert@secunet.com> |
| 6 | * |
| 7 | * This program is free software; you can redistribute it and/or modify it |
| 8 | * under the terms and conditions of the GNU General Public License, |
| 9 | * version 2, as published by the Free Software Foundation. |
| 10 | * |
| 11 | * This program is distributed in the hope it will be useful, but WITHOUT |
| 12 | * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or |
| 13 | * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for |
| 14 | * more details. |
| 15 | * |
| 16 | * You should have received a copy of the GNU General Public License along with |
| 17 | * this program; if not, write to the Free Software Foundation, Inc., |
| 18 | * 51 Franklin St - Fifth Floor, Boston, MA 02110-1301 USA. |
| 19 | */ |
| 20 | |
| 21 | #include <crypto/algapi.h> |
| 22 | #include <crypto/internal/aead.h> |
Herbert Xu | a5a22e5 | 2015-05-22 20:34:22 +0800 | [diff] [blame] | 23 | #include <linux/atomic.h> |
Steffen Klassert | 5068c7a | 2010-01-07 15:57:19 +1100 | [diff] [blame] | 24 | #include <linux/err.h> |
| 25 | #include <linux/init.h> |
| 26 | #include <linux/module.h> |
| 27 | #include <linux/slab.h> |
Dan Kruchinin | e15bacb | 2010-07-14 14:31:57 +0400 | [diff] [blame] | 28 | #include <linux/notifier.h> |
Dan Kruchinin | a3fb1e3 | 2010-07-14 14:34:15 +0400 | [diff] [blame] | 29 | #include <linux/kobject.h> |
Steffen Klassert | d3f64e4 | 2010-07-27 07:18:46 +0200 | [diff] [blame] | 30 | #include <linux/cpu.h> |
Steffen Klassert | 5068c7a | 2010-01-07 15:57:19 +1100 | [diff] [blame] | 31 | #include <crypto/pcrypt.h> |
| 32 | |
Steffen Klassert | c57e842 | 2010-07-27 07:16:33 +0200 | [diff] [blame] | 33 | struct padata_pcrypt { |
Dan Kruchinin | e15bacb | 2010-07-14 14:31:57 +0400 | [diff] [blame] | 34 | struct padata_instance *pinst; |
| 35 | struct workqueue_struct *wq; |
| 36 | |
| 37 | /* |
| 38 | * Cpumask for callback CPUs. It should be |
| 39 | * equal to serial cpumask of corresponding padata instance, |
| 40 | * so it is updated when padata notifies us about serial |
| 41 | * cpumask change. |
| 42 | * |
| 43 | * cb_cpumask is protected by RCU. This fact prevents us from |
| 44 | * using cpumask_var_t directly because the actual type of |
| 45 | * cpumsak_var_t depends on kernel configuration(particularly on |
| 46 | * CONFIG_CPUMASK_OFFSTACK macro). Depending on the configuration |
| 47 | * cpumask_var_t may be either a pointer to the struct cpumask |
| 48 | * or a variable allocated on the stack. Thus we can not safely use |
| 49 | * cpumask_var_t with RCU operations such as rcu_assign_pointer or |
| 50 | * rcu_dereference. So cpumask_var_t is wrapped with struct |
| 51 | * pcrypt_cpumask which makes possible to use it with RCU. |
| 52 | */ |
| 53 | struct pcrypt_cpumask { |
| 54 | cpumask_var_t mask; |
| 55 | } *cb_cpumask; |
| 56 | struct notifier_block nblock; |
| 57 | }; |
| 58 | |
Steffen Klassert | c57e842 | 2010-07-27 07:16:33 +0200 | [diff] [blame] | 59 | static struct padata_pcrypt pencrypt; |
| 60 | static struct padata_pcrypt pdecrypt; |
Dan Kruchinin | a3fb1e3 | 2010-07-14 14:34:15 +0400 | [diff] [blame] | 61 | static struct kset *pcrypt_kset; |
Steffen Klassert | 5068c7a | 2010-01-07 15:57:19 +1100 | [diff] [blame] | 62 | |
| 63 | struct pcrypt_instance_ctx { |
Herbert Xu | 66d948e | 2015-05-21 15:10:58 +0800 | [diff] [blame] | 64 | struct crypto_aead_spawn spawn; |
Herbert Xu | a5a22e5 | 2015-05-22 20:34:22 +0800 | [diff] [blame] | 65 | atomic_t tfm_count; |
Steffen Klassert | 5068c7a | 2010-01-07 15:57:19 +1100 | [diff] [blame] | 66 | }; |
| 67 | |
| 68 | struct pcrypt_aead_ctx { |
| 69 | struct crypto_aead *child; |
| 70 | unsigned int cb_cpu; |
| 71 | }; |
| 72 | |
| 73 | static int pcrypt_do_parallel(struct padata_priv *padata, unsigned int *cb_cpu, |
Steffen Klassert | c57e842 | 2010-07-27 07:16:33 +0200 | [diff] [blame] | 74 | struct padata_pcrypt *pcrypt) |
Steffen Klassert | 5068c7a | 2010-01-07 15:57:19 +1100 | [diff] [blame] | 75 | { |
| 76 | unsigned int cpu_index, cpu, i; |
Dan Kruchinin | e15bacb | 2010-07-14 14:31:57 +0400 | [diff] [blame] | 77 | struct pcrypt_cpumask *cpumask; |
Steffen Klassert | 5068c7a | 2010-01-07 15:57:19 +1100 | [diff] [blame] | 78 | |
| 79 | cpu = *cb_cpu; |
| 80 | |
Dan Kruchinin | e15bacb | 2010-07-14 14:31:57 +0400 | [diff] [blame] | 81 | rcu_read_lock_bh(); |
Mathias Krause | 3110e40 | 2013-11-28 19:20:04 +0100 | [diff] [blame] | 82 | cpumask = rcu_dereference_bh(pcrypt->cb_cpumask); |
Dan Kruchinin | e15bacb | 2010-07-14 14:31:57 +0400 | [diff] [blame] | 83 | if (cpumask_test_cpu(cpu, cpumask->mask)) |
Steffen Klassert | 5068c7a | 2010-01-07 15:57:19 +1100 | [diff] [blame] | 84 | goto out; |
| 85 | |
Steffen Klassert | cc74f4b | 2010-07-20 08:52:20 +0200 | [diff] [blame] | 86 | if (!cpumask_weight(cpumask->mask)) |
| 87 | goto out; |
| 88 | |
Dan Kruchinin | e15bacb | 2010-07-14 14:31:57 +0400 | [diff] [blame] | 89 | cpu_index = cpu % cpumask_weight(cpumask->mask); |
Steffen Klassert | 5068c7a | 2010-01-07 15:57:19 +1100 | [diff] [blame] | 90 | |
Dan Kruchinin | e15bacb | 2010-07-14 14:31:57 +0400 | [diff] [blame] | 91 | cpu = cpumask_first(cpumask->mask); |
Steffen Klassert | 5068c7a | 2010-01-07 15:57:19 +1100 | [diff] [blame] | 92 | for (i = 0; i < cpu_index; i++) |
Dan Kruchinin | e15bacb | 2010-07-14 14:31:57 +0400 | [diff] [blame] | 93 | cpu = cpumask_next(cpu, cpumask->mask); |
Steffen Klassert | 5068c7a | 2010-01-07 15:57:19 +1100 | [diff] [blame] | 94 | |
| 95 | *cb_cpu = cpu; |
| 96 | |
| 97 | out: |
Dan Kruchinin | e15bacb | 2010-07-14 14:31:57 +0400 | [diff] [blame] | 98 | rcu_read_unlock_bh(); |
| 99 | return padata_do_parallel(pcrypt->pinst, padata, cpu); |
Steffen Klassert | 5068c7a | 2010-01-07 15:57:19 +1100 | [diff] [blame] | 100 | } |
| 101 | |
| 102 | static int pcrypt_aead_setkey(struct crypto_aead *parent, |
| 103 | const u8 *key, unsigned int keylen) |
| 104 | { |
| 105 | struct pcrypt_aead_ctx *ctx = crypto_aead_ctx(parent); |
| 106 | |
| 107 | return crypto_aead_setkey(ctx->child, key, keylen); |
| 108 | } |
| 109 | |
| 110 | static int pcrypt_aead_setauthsize(struct crypto_aead *parent, |
| 111 | unsigned int authsize) |
| 112 | { |
| 113 | struct pcrypt_aead_ctx *ctx = crypto_aead_ctx(parent); |
| 114 | |
| 115 | return crypto_aead_setauthsize(ctx->child, authsize); |
| 116 | } |
| 117 | |
| 118 | static void pcrypt_aead_serial(struct padata_priv *padata) |
| 119 | { |
| 120 | struct pcrypt_request *preq = pcrypt_padata_request(padata); |
| 121 | struct aead_request *req = pcrypt_request_ctx(preq); |
| 122 | |
| 123 | aead_request_complete(req->base.data, padata->info); |
| 124 | } |
| 125 | |
Steffen Klassert | 5068c7a | 2010-01-07 15:57:19 +1100 | [diff] [blame] | 126 | static void pcrypt_aead_done(struct crypto_async_request *areq, int err) |
| 127 | { |
| 128 | struct aead_request *req = areq->data; |
| 129 | struct pcrypt_request *preq = aead_request_ctx(req); |
| 130 | struct padata_priv *padata = pcrypt_request_padata(preq); |
| 131 | |
| 132 | padata->info = err; |
| 133 | req->base.flags &= ~CRYPTO_TFM_REQ_MAY_SLEEP; |
| 134 | |
| 135 | padata_do_serial(padata); |
| 136 | } |
| 137 | |
| 138 | static void pcrypt_aead_enc(struct padata_priv *padata) |
| 139 | { |
| 140 | struct pcrypt_request *preq = pcrypt_padata_request(padata); |
| 141 | struct aead_request *req = pcrypt_request_ctx(preq); |
| 142 | |
| 143 | padata->info = crypto_aead_encrypt(req); |
| 144 | |
Steffen Klassert | 5a1436b | 2010-02-04 11:40:17 +1100 | [diff] [blame] | 145 | if (padata->info == -EINPROGRESS) |
Steffen Klassert | 5068c7a | 2010-01-07 15:57:19 +1100 | [diff] [blame] | 146 | return; |
| 147 | |
| 148 | padata_do_serial(padata); |
| 149 | } |
| 150 | |
| 151 | static int pcrypt_aead_encrypt(struct aead_request *req) |
| 152 | { |
| 153 | int err; |
| 154 | struct pcrypt_request *preq = aead_request_ctx(req); |
| 155 | struct aead_request *creq = pcrypt_request_ctx(preq); |
| 156 | struct padata_priv *padata = pcrypt_request_padata(preq); |
| 157 | struct crypto_aead *aead = crypto_aead_reqtfm(req); |
| 158 | struct pcrypt_aead_ctx *ctx = crypto_aead_ctx(aead); |
| 159 | u32 flags = aead_request_flags(req); |
| 160 | |
| 161 | memset(padata, 0, sizeof(struct padata_priv)); |
| 162 | |
| 163 | padata->parallel = pcrypt_aead_enc; |
| 164 | padata->serial = pcrypt_aead_serial; |
| 165 | |
| 166 | aead_request_set_tfm(creq, ctx->child); |
| 167 | aead_request_set_callback(creq, flags & ~CRYPTO_TFM_REQ_MAY_SLEEP, |
| 168 | pcrypt_aead_done, req); |
| 169 | aead_request_set_crypt(creq, req->src, req->dst, |
| 170 | req->cryptlen, req->iv); |
Herbert Xu | 0496f56 | 2015-05-28 22:08:00 +0800 | [diff] [blame] | 171 | aead_request_set_ad(creq, req->assoclen); |
Steffen Klassert | 5068c7a | 2010-01-07 15:57:19 +1100 | [diff] [blame] | 172 | |
Dan Kruchinin | e15bacb | 2010-07-14 14:31:57 +0400 | [diff] [blame] | 173 | err = pcrypt_do_parallel(padata, &ctx->cb_cpu, &pencrypt); |
Steffen Klassert | 83f619f | 2010-07-07 15:32:02 +0200 | [diff] [blame] | 174 | if (!err) |
| 175 | return -EINPROGRESS; |
Steffen Klassert | 5068c7a | 2010-01-07 15:57:19 +1100 | [diff] [blame] | 176 | |
| 177 | return err; |
| 178 | } |
| 179 | |
| 180 | static void pcrypt_aead_dec(struct padata_priv *padata) |
| 181 | { |
| 182 | struct pcrypt_request *preq = pcrypt_padata_request(padata); |
| 183 | struct aead_request *req = pcrypt_request_ctx(preq); |
| 184 | |
| 185 | padata->info = crypto_aead_decrypt(req); |
| 186 | |
Steffen Klassert | 5a1436b | 2010-02-04 11:40:17 +1100 | [diff] [blame] | 187 | if (padata->info == -EINPROGRESS) |
Steffen Klassert | 5068c7a | 2010-01-07 15:57:19 +1100 | [diff] [blame] | 188 | return; |
| 189 | |
| 190 | padata_do_serial(padata); |
| 191 | } |
| 192 | |
| 193 | static int pcrypt_aead_decrypt(struct aead_request *req) |
| 194 | { |
| 195 | int err; |
| 196 | struct pcrypt_request *preq = aead_request_ctx(req); |
| 197 | struct aead_request *creq = pcrypt_request_ctx(preq); |
| 198 | struct padata_priv *padata = pcrypt_request_padata(preq); |
| 199 | struct crypto_aead *aead = crypto_aead_reqtfm(req); |
| 200 | struct pcrypt_aead_ctx *ctx = crypto_aead_ctx(aead); |
| 201 | u32 flags = aead_request_flags(req); |
| 202 | |
| 203 | memset(padata, 0, sizeof(struct padata_priv)); |
| 204 | |
| 205 | padata->parallel = pcrypt_aead_dec; |
| 206 | padata->serial = pcrypt_aead_serial; |
| 207 | |
| 208 | aead_request_set_tfm(creq, ctx->child); |
| 209 | aead_request_set_callback(creq, flags & ~CRYPTO_TFM_REQ_MAY_SLEEP, |
| 210 | pcrypt_aead_done, req); |
| 211 | aead_request_set_crypt(creq, req->src, req->dst, |
| 212 | req->cryptlen, req->iv); |
Herbert Xu | 0496f56 | 2015-05-28 22:08:00 +0800 | [diff] [blame] | 213 | aead_request_set_ad(creq, req->assoclen); |
Steffen Klassert | 5068c7a | 2010-01-07 15:57:19 +1100 | [diff] [blame] | 214 | |
Dan Kruchinin | e15bacb | 2010-07-14 14:31:57 +0400 | [diff] [blame] | 215 | err = pcrypt_do_parallel(padata, &ctx->cb_cpu, &pdecrypt); |
Steffen Klassert | 83f619f | 2010-07-07 15:32:02 +0200 | [diff] [blame] | 216 | if (!err) |
| 217 | return -EINPROGRESS; |
Steffen Klassert | 5068c7a | 2010-01-07 15:57:19 +1100 | [diff] [blame] | 218 | |
| 219 | return err; |
| 220 | } |
| 221 | |
Herbert Xu | 0496f56 | 2015-05-28 22:08:00 +0800 | [diff] [blame] | 222 | static int pcrypt_aead_init_tfm(struct crypto_aead *tfm) |
Steffen Klassert | 5068c7a | 2010-01-07 15:57:19 +1100 | [diff] [blame] | 223 | { |
| 224 | int cpu, cpu_index; |
Herbert Xu | 0496f56 | 2015-05-28 22:08:00 +0800 | [diff] [blame] | 225 | struct aead_instance *inst = aead_alg_instance(tfm); |
| 226 | struct pcrypt_instance_ctx *ictx = aead_instance_ctx(inst); |
| 227 | struct pcrypt_aead_ctx *ctx = crypto_aead_ctx(tfm); |
Steffen Klassert | 5068c7a | 2010-01-07 15:57:19 +1100 | [diff] [blame] | 228 | struct crypto_aead *cipher; |
| 229 | |
Herbert Xu | a5a22e5 | 2015-05-22 20:34:22 +0800 | [diff] [blame] | 230 | cpu_index = (unsigned int)atomic_inc_return(&ictx->tfm_count) % |
| 231 | cpumask_weight(cpu_online_mask); |
Steffen Klassert | 5068c7a | 2010-01-07 15:57:19 +1100 | [diff] [blame] | 232 | |
Steffen Klassert | fbf0ca1 | 2012-03-28 08:51:03 +0200 | [diff] [blame] | 233 | ctx->cb_cpu = cpumask_first(cpu_online_mask); |
Steffen Klassert | 5068c7a | 2010-01-07 15:57:19 +1100 | [diff] [blame] | 234 | for (cpu = 0; cpu < cpu_index; cpu++) |
Steffen Klassert | fbf0ca1 | 2012-03-28 08:51:03 +0200 | [diff] [blame] | 235 | ctx->cb_cpu = cpumask_next(ctx->cb_cpu, cpu_online_mask); |
Steffen Klassert | 5068c7a | 2010-01-07 15:57:19 +1100 | [diff] [blame] | 236 | |
Herbert Xu | 0496f56 | 2015-05-28 22:08:00 +0800 | [diff] [blame] | 237 | cipher = crypto_spawn_aead(&ictx->spawn); |
Steffen Klassert | 5068c7a | 2010-01-07 15:57:19 +1100 | [diff] [blame] | 238 | |
| 239 | if (IS_ERR(cipher)) |
| 240 | return PTR_ERR(cipher); |
| 241 | |
| 242 | ctx->child = cipher; |
Herbert Xu | 0496f56 | 2015-05-28 22:08:00 +0800 | [diff] [blame] | 243 | crypto_aead_set_reqsize(tfm, sizeof(struct pcrypt_request) + |
| 244 | sizeof(struct aead_request) + |
| 245 | crypto_aead_reqsize(cipher)); |
Steffen Klassert | 5068c7a | 2010-01-07 15:57:19 +1100 | [diff] [blame] | 246 | |
| 247 | return 0; |
| 248 | } |
| 249 | |
Herbert Xu | 0496f56 | 2015-05-28 22:08:00 +0800 | [diff] [blame] | 250 | static void pcrypt_aead_exit_tfm(struct crypto_aead *tfm) |
Steffen Klassert | 5068c7a | 2010-01-07 15:57:19 +1100 | [diff] [blame] | 251 | { |
Herbert Xu | 0496f56 | 2015-05-28 22:08:00 +0800 | [diff] [blame] | 252 | struct pcrypt_aead_ctx *ctx = crypto_aead_ctx(tfm); |
Steffen Klassert | 5068c7a | 2010-01-07 15:57:19 +1100 | [diff] [blame] | 253 | |
| 254 | crypto_free_aead(ctx->child); |
| 255 | } |
| 256 | |
Eric Biggers | c195a4c | 2017-12-20 14:28:25 -0800 | [diff] [blame] | 257 | static void pcrypt_free(struct aead_instance *inst) |
| 258 | { |
| 259 | struct pcrypt_instance_ctx *ctx = aead_instance_ctx(inst); |
| 260 | |
| 261 | crypto_drop_aead(&ctx->spawn); |
| 262 | kfree(inst); |
| 263 | } |
| 264 | |
Herbert Xu | 66d948e | 2015-05-21 15:10:58 +0800 | [diff] [blame] | 265 | static int pcrypt_init_instance(struct crypto_instance *inst, |
| 266 | struct crypto_alg *alg) |
Steffen Klassert | 5068c7a | 2010-01-07 15:57:19 +1100 | [diff] [blame] | 267 | { |
Steffen Klassert | 5068c7a | 2010-01-07 15:57:19 +1100 | [diff] [blame] | 268 | if (snprintf(inst->alg.cra_driver_name, CRYPTO_MAX_ALG_NAME, |
| 269 | "pcrypt(%s)", alg->cra_driver_name) >= CRYPTO_MAX_ALG_NAME) |
Herbert Xu | 66d948e | 2015-05-21 15:10:58 +0800 | [diff] [blame] | 270 | return -ENAMETOOLONG; |
Steffen Klassert | 5068c7a | 2010-01-07 15:57:19 +1100 | [diff] [blame] | 271 | |
| 272 | memcpy(inst->alg.cra_name, alg->cra_name, CRYPTO_MAX_ALG_NAME); |
| 273 | |
Steffen Klassert | 5068c7a | 2010-01-07 15:57:19 +1100 | [diff] [blame] | 274 | inst->alg.cra_priority = alg->cra_priority + 100; |
| 275 | inst->alg.cra_blocksize = alg->cra_blocksize; |
| 276 | inst->alg.cra_alignmask = alg->cra_alignmask; |
| 277 | |
Herbert Xu | 66d948e | 2015-05-21 15:10:58 +0800 | [diff] [blame] | 278 | return 0; |
Steffen Klassert | 5068c7a | 2010-01-07 15:57:19 +1100 | [diff] [blame] | 279 | } |
| 280 | |
Herbert Xu | 0496f56 | 2015-05-28 22:08:00 +0800 | [diff] [blame] | 281 | static int pcrypt_create_aead(struct crypto_template *tmpl, struct rtattr **tb, |
| 282 | u32 type, u32 mask) |
Steffen Klassert | 5068c7a | 2010-01-07 15:57:19 +1100 | [diff] [blame] | 283 | { |
Herbert Xu | 66d948e | 2015-05-21 15:10:58 +0800 | [diff] [blame] | 284 | struct pcrypt_instance_ctx *ctx; |
Herbert Xu | 846f97d | 2015-07-09 07:17:18 +0800 | [diff] [blame] | 285 | struct crypto_attr_type *algt; |
Herbert Xu | 0496f56 | 2015-05-28 22:08:00 +0800 | [diff] [blame] | 286 | struct aead_instance *inst; |
| 287 | struct aead_alg *alg; |
Herbert Xu | 66d948e | 2015-05-21 15:10:58 +0800 | [diff] [blame] | 288 | const char *name; |
| 289 | int err; |
Steffen Klassert | 5068c7a | 2010-01-07 15:57:19 +1100 | [diff] [blame] | 290 | |
Herbert Xu | 846f97d | 2015-07-09 07:17:18 +0800 | [diff] [blame] | 291 | algt = crypto_get_attr_type(tb); |
| 292 | if (IS_ERR(algt)) |
| 293 | return PTR_ERR(algt); |
| 294 | |
Herbert Xu | 66d948e | 2015-05-21 15:10:58 +0800 | [diff] [blame] | 295 | name = crypto_attr_alg_name(tb[1]); |
| 296 | if (IS_ERR(name)) |
Herbert Xu | 0496f56 | 2015-05-28 22:08:00 +0800 | [diff] [blame] | 297 | return PTR_ERR(name); |
Steffen Klassert | 5068c7a | 2010-01-07 15:57:19 +1100 | [diff] [blame] | 298 | |
Herbert Xu | 66d948e | 2015-05-21 15:10:58 +0800 | [diff] [blame] | 299 | inst = kzalloc(sizeof(*inst) + sizeof(*ctx), GFP_KERNEL); |
| 300 | if (!inst) |
Herbert Xu | 0496f56 | 2015-05-28 22:08:00 +0800 | [diff] [blame] | 301 | return -ENOMEM; |
Herbert Xu | 66d948e | 2015-05-21 15:10:58 +0800 | [diff] [blame] | 302 | |
Herbert Xu | 0496f56 | 2015-05-28 22:08:00 +0800 | [diff] [blame] | 303 | ctx = aead_instance_ctx(inst); |
| 304 | crypto_set_aead_spawn(&ctx->spawn, aead_crypto_instance(inst)); |
Herbert Xu | 66d948e | 2015-05-21 15:10:58 +0800 | [diff] [blame] | 305 | |
Herbert Xu | 5e4b8c1 | 2015-08-13 17:29:06 +0800 | [diff] [blame] | 306 | err = crypto_grab_aead(&ctx->spawn, name, 0, 0); |
Herbert Xu | 66d948e | 2015-05-21 15:10:58 +0800 | [diff] [blame] | 307 | if (err) |
| 308 | goto out_free_inst; |
| 309 | |
Herbert Xu | 0496f56 | 2015-05-28 22:08:00 +0800 | [diff] [blame] | 310 | alg = crypto_spawn_aead_alg(&ctx->spawn); |
| 311 | err = pcrypt_init_instance(aead_crypto_instance(inst), &alg->base); |
Herbert Xu | 66d948e | 2015-05-21 15:10:58 +0800 | [diff] [blame] | 312 | if (err) |
| 313 | goto out_drop_aead; |
Steffen Klassert | 5068c7a | 2010-01-07 15:57:19 +1100 | [diff] [blame] | 314 | |
Herbert Xu | 846f97d | 2015-07-09 07:17:18 +0800 | [diff] [blame] | 315 | inst->alg.base.cra_flags = CRYPTO_ALG_ASYNC; |
Herbert Xu | 846f97d | 2015-07-09 07:17:18 +0800 | [diff] [blame] | 316 | |
Herbert Xu | 0496f56 | 2015-05-28 22:08:00 +0800 | [diff] [blame] | 317 | inst->alg.ivsize = crypto_aead_alg_ivsize(alg); |
| 318 | inst->alg.maxauthsize = crypto_aead_alg_maxauthsize(alg); |
Steffen Klassert | 5068c7a | 2010-01-07 15:57:19 +1100 | [diff] [blame] | 319 | |
Herbert Xu | 0496f56 | 2015-05-28 22:08:00 +0800 | [diff] [blame] | 320 | inst->alg.base.cra_ctxsize = sizeof(struct pcrypt_aead_ctx); |
Steffen Klassert | 5068c7a | 2010-01-07 15:57:19 +1100 | [diff] [blame] | 321 | |
Herbert Xu | 0496f56 | 2015-05-28 22:08:00 +0800 | [diff] [blame] | 322 | inst->alg.init = pcrypt_aead_init_tfm; |
| 323 | inst->alg.exit = pcrypt_aead_exit_tfm; |
Steffen Klassert | 5068c7a | 2010-01-07 15:57:19 +1100 | [diff] [blame] | 324 | |
Herbert Xu | 0496f56 | 2015-05-28 22:08:00 +0800 | [diff] [blame] | 325 | inst->alg.setkey = pcrypt_aead_setkey; |
| 326 | inst->alg.setauthsize = pcrypt_aead_setauthsize; |
| 327 | inst->alg.encrypt = pcrypt_aead_encrypt; |
| 328 | inst->alg.decrypt = pcrypt_aead_decrypt; |
Steffen Klassert | 5068c7a | 2010-01-07 15:57:19 +1100 | [diff] [blame] | 329 | |
Eric Biggers | c195a4c | 2017-12-20 14:28:25 -0800 | [diff] [blame] | 330 | inst->free = pcrypt_free; |
| 331 | |
Herbert Xu | 0496f56 | 2015-05-28 22:08:00 +0800 | [diff] [blame] | 332 | err = aead_register_instance(tmpl, inst); |
| 333 | if (err) |
| 334 | goto out_drop_aead; |
Steffen Klassert | 5068c7a | 2010-01-07 15:57:19 +1100 | [diff] [blame] | 335 | |
Herbert Xu | 66d948e | 2015-05-21 15:10:58 +0800 | [diff] [blame] | 336 | out: |
Herbert Xu | 0496f56 | 2015-05-28 22:08:00 +0800 | [diff] [blame] | 337 | return err; |
Herbert Xu | 66d948e | 2015-05-21 15:10:58 +0800 | [diff] [blame] | 338 | |
| 339 | out_drop_aead: |
| 340 | crypto_drop_aead(&ctx->spawn); |
| 341 | out_free_inst: |
| 342 | kfree(inst); |
Herbert Xu | 66d948e | 2015-05-21 15:10:58 +0800 | [diff] [blame] | 343 | goto out; |
Steffen Klassert | 5068c7a | 2010-01-07 15:57:19 +1100 | [diff] [blame] | 344 | } |
| 345 | |
Herbert Xu | 0496f56 | 2015-05-28 22:08:00 +0800 | [diff] [blame] | 346 | static int pcrypt_create(struct crypto_template *tmpl, struct rtattr **tb) |
Steffen Klassert | 5068c7a | 2010-01-07 15:57:19 +1100 | [diff] [blame] | 347 | { |
| 348 | struct crypto_attr_type *algt; |
| 349 | |
| 350 | algt = crypto_get_attr_type(tb); |
| 351 | if (IS_ERR(algt)) |
Herbert Xu | 0496f56 | 2015-05-28 22:08:00 +0800 | [diff] [blame] | 352 | return PTR_ERR(algt); |
Steffen Klassert | 5068c7a | 2010-01-07 15:57:19 +1100 | [diff] [blame] | 353 | |
| 354 | switch (algt->type & algt->mask & CRYPTO_ALG_TYPE_MASK) { |
| 355 | case CRYPTO_ALG_TYPE_AEAD: |
Herbert Xu | 0496f56 | 2015-05-28 22:08:00 +0800 | [diff] [blame] | 356 | return pcrypt_create_aead(tmpl, tb, algt->type, algt->mask); |
Steffen Klassert | 5068c7a | 2010-01-07 15:57:19 +1100 | [diff] [blame] | 357 | } |
| 358 | |
Herbert Xu | 0496f56 | 2015-05-28 22:08:00 +0800 | [diff] [blame] | 359 | return -EINVAL; |
Steffen Klassert | 5068c7a | 2010-01-07 15:57:19 +1100 | [diff] [blame] | 360 | } |
| 361 | |
Dan Kruchinin | e15bacb | 2010-07-14 14:31:57 +0400 | [diff] [blame] | 362 | static int pcrypt_cpumask_change_notify(struct notifier_block *self, |
| 363 | unsigned long val, void *data) |
| 364 | { |
Steffen Klassert | c57e842 | 2010-07-27 07:16:33 +0200 | [diff] [blame] | 365 | struct padata_pcrypt *pcrypt; |
Dan Kruchinin | e15bacb | 2010-07-14 14:31:57 +0400 | [diff] [blame] | 366 | struct pcrypt_cpumask *new_mask, *old_mask; |
Steffen Klassert | d3f64e4 | 2010-07-27 07:18:46 +0200 | [diff] [blame] | 367 | struct padata_cpumask *cpumask = (struct padata_cpumask *)data; |
Dan Kruchinin | e15bacb | 2010-07-14 14:31:57 +0400 | [diff] [blame] | 368 | |
| 369 | if (!(val & PADATA_CPU_SERIAL)) |
| 370 | return 0; |
| 371 | |
Steffen Klassert | c57e842 | 2010-07-27 07:16:33 +0200 | [diff] [blame] | 372 | pcrypt = container_of(self, struct padata_pcrypt, nblock); |
Dan Kruchinin | e15bacb | 2010-07-14 14:31:57 +0400 | [diff] [blame] | 373 | new_mask = kmalloc(sizeof(*new_mask), GFP_KERNEL); |
| 374 | if (!new_mask) |
| 375 | return -ENOMEM; |
| 376 | if (!alloc_cpumask_var(&new_mask->mask, GFP_KERNEL)) { |
| 377 | kfree(new_mask); |
| 378 | return -ENOMEM; |
| 379 | } |
| 380 | |
| 381 | old_mask = pcrypt->cb_cpumask; |
| 382 | |
Steffen Klassert | d3f64e4 | 2010-07-27 07:18:46 +0200 | [diff] [blame] | 383 | cpumask_copy(new_mask->mask, cpumask->cbcpu); |
Dan Kruchinin | e15bacb | 2010-07-14 14:31:57 +0400 | [diff] [blame] | 384 | rcu_assign_pointer(pcrypt->cb_cpumask, new_mask); |
| 385 | synchronize_rcu_bh(); |
| 386 | |
| 387 | free_cpumask_var(old_mask->mask); |
| 388 | kfree(old_mask); |
| 389 | return 0; |
| 390 | } |
| 391 | |
Dan Kruchinin | a3fb1e3 | 2010-07-14 14:34:15 +0400 | [diff] [blame] | 392 | static int pcrypt_sysfs_add(struct padata_instance *pinst, const char *name) |
| 393 | { |
| 394 | int ret; |
| 395 | |
| 396 | pinst->kobj.kset = pcrypt_kset; |
Colin Ian King | a3f01dc | 2018-10-27 15:49:26 +0100 | [diff] [blame] | 397 | ret = kobject_add(&pinst->kobj, NULL, "%s", name); |
Dan Kruchinin | a3fb1e3 | 2010-07-14 14:34:15 +0400 | [diff] [blame] | 398 | if (!ret) |
| 399 | kobject_uevent(&pinst->kobj, KOBJ_ADD); |
| 400 | |
| 401 | return ret; |
| 402 | } |
| 403 | |
Steffen Klassert | c57e842 | 2010-07-27 07:16:33 +0200 | [diff] [blame] | 404 | static int pcrypt_init_padata(struct padata_pcrypt *pcrypt, |
| 405 | const char *name) |
Dan Kruchinin | e15bacb | 2010-07-14 14:31:57 +0400 | [diff] [blame] | 406 | { |
| 407 | int ret = -ENOMEM; |
| 408 | struct pcrypt_cpumask *mask; |
| 409 | |
Steffen Klassert | d3f64e4 | 2010-07-27 07:18:46 +0200 | [diff] [blame] | 410 | get_online_cpus(); |
| 411 | |
Kees Cook | d853754 | 2013-07-03 15:04:57 -0700 | [diff] [blame] | 412 | pcrypt->wq = alloc_workqueue("%s", WQ_MEM_RECLAIM | WQ_CPU_INTENSIVE, |
| 413 | 1, name); |
Dan Kruchinin | e15bacb | 2010-07-14 14:31:57 +0400 | [diff] [blame] | 414 | if (!pcrypt->wq) |
| 415 | goto err; |
| 416 | |
Steffen Klassert | e6cc117 | 2010-07-27 07:14:28 +0200 | [diff] [blame] | 417 | pcrypt->pinst = padata_alloc_possible(pcrypt->wq); |
Dan Kruchinin | e15bacb | 2010-07-14 14:31:57 +0400 | [diff] [blame] | 418 | if (!pcrypt->pinst) |
| 419 | goto err_destroy_workqueue; |
| 420 | |
| 421 | mask = kmalloc(sizeof(*mask), GFP_KERNEL); |
| 422 | if (!mask) |
| 423 | goto err_free_padata; |
| 424 | if (!alloc_cpumask_var(&mask->mask, GFP_KERNEL)) { |
| 425 | kfree(mask); |
| 426 | goto err_free_padata; |
| 427 | } |
| 428 | |
Steffen Klassert | fbf0ca1 | 2012-03-28 08:51:03 +0200 | [diff] [blame] | 429 | cpumask_and(mask->mask, cpu_possible_mask, cpu_online_mask); |
Dan Kruchinin | e15bacb | 2010-07-14 14:31:57 +0400 | [diff] [blame] | 430 | rcu_assign_pointer(pcrypt->cb_cpumask, mask); |
| 431 | |
| 432 | pcrypt->nblock.notifier_call = pcrypt_cpumask_change_notify; |
| 433 | ret = padata_register_cpumask_notifier(pcrypt->pinst, &pcrypt->nblock); |
| 434 | if (ret) |
| 435 | goto err_free_cpumask; |
| 436 | |
Dan Kruchinin | a3fb1e3 | 2010-07-14 14:34:15 +0400 | [diff] [blame] | 437 | ret = pcrypt_sysfs_add(pcrypt->pinst, name); |
| 438 | if (ret) |
| 439 | goto err_unregister_notifier; |
| 440 | |
Steffen Klassert | d3f64e4 | 2010-07-27 07:18:46 +0200 | [diff] [blame] | 441 | put_online_cpus(); |
| 442 | |
Dan Kruchinin | e15bacb | 2010-07-14 14:31:57 +0400 | [diff] [blame] | 443 | return ret; |
Steffen Klassert | d3f64e4 | 2010-07-27 07:18:46 +0200 | [diff] [blame] | 444 | |
Dan Kruchinin | a3fb1e3 | 2010-07-14 14:34:15 +0400 | [diff] [blame] | 445 | err_unregister_notifier: |
| 446 | padata_unregister_cpumask_notifier(pcrypt->pinst, &pcrypt->nblock); |
Dan Kruchinin | e15bacb | 2010-07-14 14:31:57 +0400 | [diff] [blame] | 447 | err_free_cpumask: |
| 448 | free_cpumask_var(mask->mask); |
| 449 | kfree(mask); |
| 450 | err_free_padata: |
| 451 | padata_free(pcrypt->pinst); |
| 452 | err_destroy_workqueue: |
| 453 | destroy_workqueue(pcrypt->wq); |
| 454 | err: |
Steffen Klassert | d3f64e4 | 2010-07-27 07:18:46 +0200 | [diff] [blame] | 455 | put_online_cpus(); |
| 456 | |
Dan Kruchinin | e15bacb | 2010-07-14 14:31:57 +0400 | [diff] [blame] | 457 | return ret; |
| 458 | } |
| 459 | |
Steffen Klassert | c57e842 | 2010-07-27 07:16:33 +0200 | [diff] [blame] | 460 | static void pcrypt_fini_padata(struct padata_pcrypt *pcrypt) |
Dan Kruchinin | e15bacb | 2010-07-14 14:31:57 +0400 | [diff] [blame] | 461 | { |
| 462 | free_cpumask_var(pcrypt->cb_cpumask->mask); |
| 463 | kfree(pcrypt->cb_cpumask); |
| 464 | |
| 465 | padata_stop(pcrypt->pinst); |
| 466 | padata_unregister_cpumask_notifier(pcrypt->pinst, &pcrypt->nblock); |
| 467 | destroy_workqueue(pcrypt->wq); |
| 468 | padata_free(pcrypt->pinst); |
| 469 | } |
| 470 | |
Steffen Klassert | 5068c7a | 2010-01-07 15:57:19 +1100 | [diff] [blame] | 471 | static struct crypto_template pcrypt_tmpl = { |
| 472 | .name = "pcrypt", |
Herbert Xu | 0496f56 | 2015-05-28 22:08:00 +0800 | [diff] [blame] | 473 | .create = pcrypt_create, |
Steffen Klassert | 5068c7a | 2010-01-07 15:57:19 +1100 | [diff] [blame] | 474 | .module = THIS_MODULE, |
| 475 | }; |
| 476 | |
| 477 | static int __init pcrypt_init(void) |
| 478 | { |
Dan Kruchinin | a3fb1e3 | 2010-07-14 14:34:15 +0400 | [diff] [blame] | 479 | int err = -ENOMEM; |
| 480 | |
| 481 | pcrypt_kset = kset_create_and_add("pcrypt", NULL, kernel_kobj); |
| 482 | if (!pcrypt_kset) |
| 483 | goto err; |
Dan Kruchinin | e15bacb | 2010-07-14 14:31:57 +0400 | [diff] [blame] | 484 | |
Steffen Klassert | c57e842 | 2010-07-27 07:16:33 +0200 | [diff] [blame] | 485 | err = pcrypt_init_padata(&pencrypt, "pencrypt"); |
Dan Kruchinin | e15bacb | 2010-07-14 14:31:57 +0400 | [diff] [blame] | 486 | if (err) |
Dan Kruchinin | a3fb1e3 | 2010-07-14 14:34:15 +0400 | [diff] [blame] | 487 | goto err_unreg_kset; |
Steffen Klassert | 5068c7a | 2010-01-07 15:57:19 +1100 | [diff] [blame] | 488 | |
Steffen Klassert | c57e842 | 2010-07-27 07:16:33 +0200 | [diff] [blame] | 489 | err = pcrypt_init_padata(&pdecrypt, "pdecrypt"); |
Steffen Klassert | 4c87917 | 2010-07-07 15:30:10 +0200 | [diff] [blame] | 490 | if (err) |
Dan Kruchinin | e15bacb | 2010-07-14 14:31:57 +0400 | [diff] [blame] | 491 | goto err_deinit_pencrypt; |
Steffen Klassert | 4c87917 | 2010-07-07 15:30:10 +0200 | [diff] [blame] | 492 | |
Dan Kruchinin | e15bacb | 2010-07-14 14:31:57 +0400 | [diff] [blame] | 493 | padata_start(pencrypt.pinst); |
| 494 | padata_start(pdecrypt.pinst); |
Steffen Klassert | 5068c7a | 2010-01-07 15:57:19 +1100 | [diff] [blame] | 495 | |
| 496 | return crypto_register_template(&pcrypt_tmpl); |
| 497 | |
Dan Kruchinin | e15bacb | 2010-07-14 14:31:57 +0400 | [diff] [blame] | 498 | err_deinit_pencrypt: |
Steffen Klassert | c57e842 | 2010-07-27 07:16:33 +0200 | [diff] [blame] | 499 | pcrypt_fini_padata(&pencrypt); |
Dan Kruchinin | a3fb1e3 | 2010-07-14 14:34:15 +0400 | [diff] [blame] | 500 | err_unreg_kset: |
| 501 | kset_unregister(pcrypt_kset); |
Steffen Klassert | 5068c7a | 2010-01-07 15:57:19 +1100 | [diff] [blame] | 502 | err: |
Steffen Klassert | 4c87917 | 2010-07-07 15:30:10 +0200 | [diff] [blame] | 503 | return err; |
Steffen Klassert | 5068c7a | 2010-01-07 15:57:19 +1100 | [diff] [blame] | 504 | } |
| 505 | |
| 506 | static void __exit pcrypt_exit(void) |
| 507 | { |
Steffen Klassert | c57e842 | 2010-07-27 07:16:33 +0200 | [diff] [blame] | 508 | pcrypt_fini_padata(&pencrypt); |
| 509 | pcrypt_fini_padata(&pdecrypt); |
Steffen Klassert | 5068c7a | 2010-01-07 15:57:19 +1100 | [diff] [blame] | 510 | |
Dan Kruchinin | a3fb1e3 | 2010-07-14 14:34:15 +0400 | [diff] [blame] | 511 | kset_unregister(pcrypt_kset); |
Steffen Klassert | 5068c7a | 2010-01-07 15:57:19 +1100 | [diff] [blame] | 512 | crypto_unregister_template(&pcrypt_tmpl); |
| 513 | } |
| 514 | |
| 515 | module_init(pcrypt_init); |
| 516 | module_exit(pcrypt_exit); |
| 517 | |
| 518 | MODULE_LICENSE("GPL"); |
| 519 | MODULE_AUTHOR("Steffen Klassert <steffen.klassert@secunet.com>"); |
| 520 | MODULE_DESCRIPTION("Parallel crypto wrapper"); |
Kees Cook | 4943ba1 | 2014-11-24 16:32:38 -0800 | [diff] [blame] | 521 | MODULE_ALIAS_CRYPTO("pcrypt"); |