at v4.18 3.4 kB view raw
1/* 2 * Cryptographic API. 3 * 4 * Cipher operations. 5 * 6 * Copyright (c) 2002 James Morris <jmorris@intercode.com.au> 7 * Copyright (c) 2005 Herbert Xu <herbert@gondor.apana.org.au> 8 * 9 * This program is free software; you can redistribute it and/or modify it 10 * under the terms of the GNU General Public License as published by the Free 11 * Software Foundation; either version 2 of the License, or (at your option) 12 * any later version. 13 * 14 */ 15 16#include <crypto/algapi.h> 17#include <linux/kernel.h> 18#include <linux/crypto.h> 19#include <linux/errno.h> 20#include <linux/slab.h> 21#include <linux/string.h> 22#include "internal.h" 23 24static int setkey_unaligned(struct crypto_tfm *tfm, const u8 *key, 25 unsigned int keylen) 26{ 27 struct cipher_alg *cia = &tfm->__crt_alg->cra_cipher; 28 unsigned long alignmask = crypto_tfm_alg_alignmask(tfm); 29 int ret; 30 u8 *buffer, *alignbuffer; 31 unsigned long absize; 32 33 absize = keylen + alignmask; 34 buffer = kmalloc(absize, GFP_ATOMIC); 35 if (!buffer) 36 return -ENOMEM; 37 38 alignbuffer = (u8 *)ALIGN((unsigned long)buffer, alignmask + 1); 39 memcpy(alignbuffer, key, keylen); 40 ret = cia->cia_setkey(tfm, alignbuffer, keylen); 41 memset(alignbuffer, 0, keylen); 42 kfree(buffer); 43 return ret; 44 45} 46 47static int setkey(struct crypto_tfm *tfm, const u8 *key, unsigned int keylen) 48{ 49 struct cipher_alg *cia = &tfm->__crt_alg->cra_cipher; 50 unsigned long alignmask = crypto_tfm_alg_alignmask(tfm); 51 52 tfm->crt_flags &= ~CRYPTO_TFM_RES_MASK; 53 if (keylen < cia->cia_min_keysize || keylen > cia->cia_max_keysize) { 54 tfm->crt_flags |= CRYPTO_TFM_RES_BAD_KEY_LEN; 55 return -EINVAL; 56 } 57 58 if ((unsigned long)key & alignmask) 59 return setkey_unaligned(tfm, key, keylen); 60 61 return cia->cia_setkey(tfm, key, keylen); 62} 63 64static void cipher_crypt_unaligned(void (*fn)(struct crypto_tfm *, u8 *, 65 const u8 *), 66 struct crypto_tfm *tfm, 67 u8 *dst, const u8 *src) 68{ 69 unsigned long alignmask = crypto_tfm_alg_alignmask(tfm); 70 unsigned int size = crypto_tfm_alg_blocksize(tfm); 71 u8 buffer[MAX_CIPHER_BLOCKSIZE + MAX_CIPHER_ALIGNMASK]; 72 u8 *tmp = (u8 *)ALIGN((unsigned long)buffer, alignmask + 1); 73 74 memcpy(tmp, src, size); 75 fn(tfm, tmp, tmp); 76 memcpy(dst, tmp, size); 77} 78 79static void cipher_encrypt_unaligned(struct crypto_tfm *tfm, 80 u8 *dst, const u8 *src) 81{ 82 unsigned long alignmask = crypto_tfm_alg_alignmask(tfm); 83 struct cipher_alg *cipher = &tfm->__crt_alg->cra_cipher; 84 85 if (unlikely(((unsigned long)dst | (unsigned long)src) & alignmask)) { 86 cipher_crypt_unaligned(cipher->cia_encrypt, tfm, dst, src); 87 return; 88 } 89 90 cipher->cia_encrypt(tfm, dst, src); 91} 92 93static void cipher_decrypt_unaligned(struct crypto_tfm *tfm, 94 u8 *dst, const u8 *src) 95{ 96 unsigned long alignmask = crypto_tfm_alg_alignmask(tfm); 97 struct cipher_alg *cipher = &tfm->__crt_alg->cra_cipher; 98 99 if (unlikely(((unsigned long)dst | (unsigned long)src) & alignmask)) { 100 cipher_crypt_unaligned(cipher->cia_decrypt, tfm, dst, src); 101 return; 102 } 103 104 cipher->cia_decrypt(tfm, dst, src); 105} 106 107int crypto_init_cipher_ops(struct crypto_tfm *tfm) 108{ 109 struct cipher_tfm *ops = &tfm->crt_cipher; 110 struct cipher_alg *cipher = &tfm->__crt_alg->cra_cipher; 111 112 ops->cit_setkey = setkey; 113 ops->cit_encrypt_one = crypto_tfm_alg_alignmask(tfm) ? 114 cipher_encrypt_unaligned : cipher->cia_encrypt; 115 ops->cit_decrypt_one = crypto_tfm_alg_alignmask(tfm) ? 116 cipher_decrypt_unaligned : cipher->cia_decrypt; 117 118 return 0; 119}