1 /* SPDX-License-Identifier: GPL-2.0 */ 1 /* SPDX-License-Identifier: GPL-2.0 */
2 2
3 #ifndef _CRYPTO_INTERNAL_CHACHA_H 3 #ifndef _CRYPTO_INTERNAL_CHACHA_H
4 #define _CRYPTO_INTERNAL_CHACHA_H 4 #define _CRYPTO_INTERNAL_CHACHA_H
5 5
6 #include <crypto/chacha.h> 6 #include <crypto/chacha.h>
7 #include <crypto/internal/skcipher.h> 7 #include <crypto/internal/skcipher.h>
8 #include <linux/crypto.h> 8 #include <linux/crypto.h>
9 9
10 struct chacha_ctx { 10 struct chacha_ctx {
11 u32 key[8]; 11 u32 key[8];
12 int nrounds; 12 int nrounds;
13 }; 13 };
14 14
15 static inline int chacha_setkey(struct crypto_ 15 static inline int chacha_setkey(struct crypto_skcipher *tfm, const u8 *key,
16 unsigned int k 16 unsigned int keysize, int nrounds)
17 { 17 {
18 struct chacha_ctx *ctx = crypto_skciph 18 struct chacha_ctx *ctx = crypto_skcipher_ctx(tfm);
19 int i; 19 int i;
20 20
21 if (keysize != CHACHA_KEY_SIZE) 21 if (keysize != CHACHA_KEY_SIZE)
22 return -EINVAL; 22 return -EINVAL;
23 23
24 for (i = 0; i < ARRAY_SIZE(ctx->key); 24 for (i = 0; i < ARRAY_SIZE(ctx->key); i++)
25 ctx->key[i] = get_unaligned_le 25 ctx->key[i] = get_unaligned_le32(key + i * sizeof(u32));
26 26
27 ctx->nrounds = nrounds; 27 ctx->nrounds = nrounds;
28 return 0; 28 return 0;
29 } 29 }
30 30
31 static inline int chacha20_setkey(struct crypt 31 static inline int chacha20_setkey(struct crypto_skcipher *tfm, const u8 *key,
32 unsigned int 32 unsigned int keysize)
33 { 33 {
34 return chacha_setkey(tfm, key, keysize 34 return chacha_setkey(tfm, key, keysize, 20);
35 } 35 }
36 36
37 static inline int chacha12_setkey(struct crypt 37 static inline int chacha12_setkey(struct crypto_skcipher *tfm, const u8 *key,
38 unsigned int 38 unsigned int keysize)
39 { 39 {
40 return chacha_setkey(tfm, key, keysize 40 return chacha_setkey(tfm, key, keysize, 12);
41 } 41 }
42 42
43 #endif /* _CRYPTO_CHACHA_H */ 43 #endif /* _CRYPTO_CHACHA_H */
44 44
Linux® is a registered trademark of Linus Torvalds in the United States and other countries.
TOMOYO® is a registered trademark of NTT DATA CORPORATION.