crypto/algboss.c

   1 // SPDX-License-Identifier: GPL-2.0-or-later
   2 /*
   3  * Create default crypto algorithm instances.
   4  *
   5  * Copyright (c) 2006 Herbert Xu <[email protected]>
   6  */
   7
   8 #include <crypto/internal/aead.h>
   9 #include <linux/completion.h>
  10 #include <linux/ctype.h>
  11 #include <linux/err.h>
  12 #include <linux/init.h>
  13 #include <linux/kthread.h>
  14 #include <linux/module.h>
  15 #include <linux/notifier.h>
  16 #include <linux/rtnetlink.h>
  17 #include <linux/sched/signal.h>
  18 #include <linux/slab.h>
  19 #include <linux/string.h>
  20
  21 #include "internal.h"
  22
  23 struct cryptomgr_param {
  24         struct rtattr *tb[CRYPTO_MAX_ATTRS + 2];
  25
  26         struct {
  27                 struct rtattr attr;
  28                 struct crypto_attr_type data;
  29         } type;
  30
  31         struct {
  32                 struct rtattr attr;
  33                 struct crypto_attr_alg data;
  34         } attrs[CRYPTO_MAX_ATTRS];
  35
  36         char template[CRYPTO_MAX_ALG_NAME];
  37
  38         struct crypto_larval *larval;
  39
  40         u32 otype;
  41         u32 omask;
  42 };
  43
  44 struct crypto_test_param {
  45         char driver[CRYPTO_MAX_ALG_NAME];
  46         char alg[CRYPTO_MAX_ALG_NAME];
  47         u32 type;
  48 };
  49
  50 static int cryptomgr_probe(void *data)
  51 {
  52         struct cryptomgr_param *param = data;
  53         struct crypto_template *tmpl;
  54         int err;
  55
  56         tmpl = crypto_lookup_template(param->template);
  57         if (!tmpl)
  58                 goto out;
  59
  60         do {
  61                 err = tmpl->create(tmpl, param->tb);
  62         } while (err == -EAGAIN && !signal_pending(current));
  63
  64         crypto_tmpl_put(tmpl);
  65
  66 out:
  67         complete_all(&param->larval->completion);
  68         crypto_alg_put(&param->larval->alg);
  69         kfree(param);
  70         module_put_and_kthread_exit(0);
  71 }
  72
  73 static int cryptomgr_schedule_probe(struct crypto_larval *larval)
  74 {
  75         struct task_struct *thread;
  76         struct cryptomgr_param *param;
  77         const char *name = larval->alg.cra_name;
  78         const char *p;
  79         unsigned int len;
  80         int i;
  81
  82         if (!try_module_get(THIS_MODULE))
  83                 goto err;
  84
  85         param = kzalloc(sizeof(*param), GFP_KERNEL);
  86         if (!param)
  87                 goto err_put_module;
  88
  89         for (p = name; isalnum(*p) || *p == '-' || *p == '_'; p++)
  90                 ;
  91
  92         len = p - name;
  93         if (!len || *p != '(')
  94                 goto err_free_param;
  95
  96         memcpy(param->template, name, len);
  97
  98         i = 0;
  99         for (;;) {
 100                 name = ++p;
 101
 102                 for (; isalnum(*p) || *p == '-' || *p == '_'; p++)
 103                         ;
 104
 105                 if (*p == '(') {
 106                         int recursion = 0;
 107
 108                         for (;;) {
 109                                 if (!*++p)
 110                                         goto err_free_param;
 111                                 if (*p == '(')
 112                                         recursion++;
 113                                 else if (*p == ')' && !recursion--)
 114                                         break;
 115                         }
 116
 117                         p++;
 118                 }
 119
 120                 len = p - name;
 121                 if (!len)
 122                         goto err_free_param;
 123
 124                 param->attrs[i].attr.rta_len = sizeof(param->attrs[i]);
 125                 param->attrs[i].attr.rta_type = CRYPTOA_ALG;
 126                 memcpy(param->attrs[i].data.name, name, len);
 127
 128                 param->tb[i + 1] = &param->attrs[i].attr;
 129                 i++;
 130
 131                 if (i >= CRYPTO_MAX_ATTRS)
 132                         goto err_free_param;
 133
 134                 if (*p == ')')
 135                         break;
 136
 137                 if (*p != ',')
 138                         goto err_free_param;
 139         }
 140
 141         param->tb[i + 1] = NULL;
 142
 143         param->type.attr.rta_len = sizeof(param->type);
 144         param->type.attr.rta_type = CRYPTOA_TYPE;
 145         param->type.data.type = larval->alg.cra_flags & ~CRYPTO_ALG_TESTED;
 146         param->type.data.mask = larval->mask & ~CRYPTO_ALG_TESTED;
 147         param->tb[0] = &param->type.attr;
 148
 149         param->otype = larval->alg.cra_flags;
 150         param->omask = larval->mask;
 151
 152         crypto_alg_get(&larval->alg);
 153         param->larval = larval;
 154
 155         thread = kthread_run(cryptomgr_probe, param, "cryptomgr_probe");
 156         if (IS_ERR(thread))
 157                 goto err_put_larval;
 158
 159         return NOTIFY_STOP;
 160
 161 err_put_larval:
 162         crypto_alg_put(&larval->alg);
 163 err_free_param:
 164         kfree(param);
 165 err_put_module:
 166         module_put(THIS_MODULE);
 167 err:
 168         return NOTIFY_OK;
 169 }
 170
 171 static int cryptomgr_test(void *data)
 172 {
 173         struct crypto_test_param *param = data;
 174         u32 type = param->type;
 175         int err;
 176
 177         err = alg_test(param->driver, param->alg, type, CRYPTO_ALG_TESTED);
 178
 179         crypto_alg_tested(param->driver, err);
 180
 181         kfree(param);
 182         module_put_and_kthread_exit(0);
 183 }
 184
 185 static int cryptomgr_schedule_test(struct crypto_alg *alg)
 186 {
 187         struct task_struct *thread;
 188         struct crypto_test_param *param;
 189
 190         if (IS_ENABLED(CONFIG_CRYPTO_MANAGER_DISABLE_TESTS))
 191                 return NOTIFY_DONE;
 192
 193         if (!try_module_get(THIS_MODULE))
 194                 goto err;
 195
 196         param = kzalloc(sizeof(*param), GFP_KERNEL);
 197         if (!param)
 198                 goto err_put_module;
 199
 200         memcpy(param->driver, alg->cra_driver_name, sizeof(param->driver));
 201         memcpy(param->alg, alg->cra_name, sizeof(param->alg));
 202         param->type = alg->cra_flags;
 203
 204         thread = kthread_run(cryptomgr_test, param, "cryptomgr_test");
 205         if (IS_ERR(thread))
 206                 goto err_free_param;
 207
 208         return NOTIFY_STOP;
 209
 210 err_free_param:
 211         kfree(param);
 212 err_put_module:
 213         module_put(THIS_MODULE);
 214 err:
 215         return NOTIFY_OK;
 216 }
 217
 218 static int cryptomgr_notify(struct notifier_block *this, unsigned long msg,
 219                             void *data)
 220 {
 221         switch (msg) {
 222         case CRYPTO_MSG_ALG_REQUEST:
 223                 return cryptomgr_schedule_probe(data);
 224         case CRYPTO_MSG_ALG_REGISTER:
 225                 return cryptomgr_schedule_test(data);
 226         case CRYPTO_MSG_ALG_LOADED:
 227                 break;
 228         }
 229
 230         return NOTIFY_DONE;
 231 }
 232
 233 static struct notifier_block cryptomgr_notifier = {
 234         .notifier_call = cryptomgr_notify,
 235 };
 236
 237 static int __init cryptomgr_init(void)
 238 {
 239         return crypto_register_notifier(&cryptomgr_notifier);
 240 }
 241
 242 static void __exit cryptomgr_exit(void)
 243 {
 244         int err = crypto_unregister_notifier(&cryptomgr_notifier);
 245         BUG_ON(err);
 246 }
 247
 248 /*
 249  * This is arch_initcall() so that the crypto self-tests are run on algorithms
 250  * registered early by subsys_initcall().  subsys_initcall() is needed for
 251  * generic implementations so that they're available for comparison tests when
 252  * other implementations are registered later by module_init().
 253  */
 254 arch_initcall(cryptomgr_init);
 255 module_exit(cryptomgr_exit);
 256
 257 MODULE_LICENSE("GPL");
 258 MODULE_DESCRIPTION("Crypto Algorithm Manager");