diff options
Diffstat (limited to 'arch/arm64/crypto/sha2-ce-glue.c')
-rw-r--r-- | arch/arm64/crypto/sha2-ce-glue.c | 255 |
1 files changed, 255 insertions, 0 deletions
diff --git a/arch/arm64/crypto/sha2-ce-glue.c b/arch/arm64/crypto/sha2-ce-glue.c new file mode 100644 index 000000000000..c294e67d3925 --- /dev/null +++ b/arch/arm64/crypto/sha2-ce-glue.c | |||
@@ -0,0 +1,255 @@ | |||
1 | /* | ||
2 | * sha2-ce-glue.c - SHA-224/SHA-256 using ARMv8 Crypto Extensions | ||
3 | * | ||
4 | * Copyright (C) 2014 Linaro Ltd <ard.biesheuvel@linaro.org> | ||
5 | * | ||
6 | * This program is free software; you can redistribute it and/or modify | ||
7 | * it under the terms of the GNU General Public License version 2 as | ||
8 | * published by the Free Software Foundation. | ||
9 | */ | ||
10 | |||
11 | #include <asm/neon.h> | ||
12 | #include <asm/unaligned.h> | ||
13 | #include <crypto/internal/hash.h> | ||
14 | #include <crypto/sha.h> | ||
15 | #include <linux/cpufeature.h> | ||
16 | #include <linux/crypto.h> | ||
17 | #include <linux/module.h> | ||
18 | |||
19 | MODULE_DESCRIPTION("SHA-224/SHA-256 secure hash using ARMv8 Crypto Extensions"); | ||
20 | MODULE_AUTHOR("Ard Biesheuvel <ard.biesheuvel@linaro.org>"); | ||
21 | MODULE_LICENSE("GPL v2"); | ||
22 | |||
23 | asmlinkage int sha2_ce_transform(int blocks, u8 const *src, u32 *state, | ||
24 | u8 *head, long bytes); | ||
25 | |||
26 | static int sha224_init(struct shash_desc *desc) | ||
27 | { | ||
28 | struct sha256_state *sctx = shash_desc_ctx(desc); | ||
29 | |||
30 | *sctx = (struct sha256_state){ | ||
31 | .state = { | ||
32 | SHA224_H0, SHA224_H1, SHA224_H2, SHA224_H3, | ||
33 | SHA224_H4, SHA224_H5, SHA224_H6, SHA224_H7, | ||
34 | } | ||
35 | }; | ||
36 | return 0; | ||
37 | } | ||
38 | |||
39 | static int sha256_init(struct shash_desc *desc) | ||
40 | { | ||
41 | struct sha256_state *sctx = shash_desc_ctx(desc); | ||
42 | |||
43 | *sctx = (struct sha256_state){ | ||
44 | .state = { | ||
45 | SHA256_H0, SHA256_H1, SHA256_H2, SHA256_H3, | ||
46 | SHA256_H4, SHA256_H5, SHA256_H6, SHA256_H7, | ||
47 | } | ||
48 | }; | ||
49 | return 0; | ||
50 | } | ||
51 | |||
52 | static int sha2_update(struct shash_desc *desc, const u8 *data, | ||
53 | unsigned int len) | ||
54 | { | ||
55 | struct sha256_state *sctx = shash_desc_ctx(desc); | ||
56 | unsigned int partial = sctx->count % SHA256_BLOCK_SIZE; | ||
57 | |||
58 | sctx->count += len; | ||
59 | |||
60 | if ((partial + len) >= SHA256_BLOCK_SIZE) { | ||
61 | int blocks; | ||
62 | |||
63 | if (partial) { | ||
64 | int p = SHA256_BLOCK_SIZE - partial; | ||
65 | |||
66 | memcpy(sctx->buf + partial, data, p); | ||
67 | data += p; | ||
68 | len -= p; | ||
69 | } | ||
70 | |||
71 | blocks = len / SHA256_BLOCK_SIZE; | ||
72 | len %= SHA256_BLOCK_SIZE; | ||
73 | |||
74 | kernel_neon_begin_partial(28); | ||
75 | sha2_ce_transform(blocks, data, sctx->state, | ||
76 | partial ? sctx->buf : NULL, 0); | ||
77 | kernel_neon_end(); | ||
78 | |||
79 | data += blocks * SHA256_BLOCK_SIZE; | ||
80 | partial = 0; | ||
81 | } | ||
82 | if (len) | ||
83 | memcpy(sctx->buf + partial, data, len); | ||
84 | return 0; | ||
85 | } | ||
86 | |||
87 | static void sha2_final(struct shash_desc *desc) | ||
88 | { | ||
89 | static const u8 padding[SHA256_BLOCK_SIZE] = { 0x80, }; | ||
90 | |||
91 | struct sha256_state *sctx = shash_desc_ctx(desc); | ||
92 | __be64 bits = cpu_to_be64(sctx->count << 3); | ||
93 | u32 padlen = SHA256_BLOCK_SIZE | ||
94 | - ((sctx->count + sizeof(bits)) % SHA256_BLOCK_SIZE); | ||
95 | |||
96 | sha2_update(desc, padding, padlen); | ||
97 | sha2_update(desc, (const u8 *)&bits, sizeof(bits)); | ||
98 | } | ||
99 | |||
100 | static int sha224_final(struct shash_desc *desc, u8 *out) | ||
101 | { | ||
102 | struct sha256_state *sctx = shash_desc_ctx(desc); | ||
103 | __be32 *dst = (__be32 *)out; | ||
104 | int i; | ||
105 | |||
106 | sha2_final(desc); | ||
107 | |||
108 | for (i = 0; i < SHA224_DIGEST_SIZE / sizeof(__be32); i++) | ||
109 | put_unaligned_be32(sctx->state[i], dst++); | ||
110 | |||
111 | *sctx = (struct sha256_state){}; | ||
112 | return 0; | ||
113 | } | ||
114 | |||
115 | static int sha256_final(struct shash_desc *desc, u8 *out) | ||
116 | { | ||
117 | struct sha256_state *sctx = shash_desc_ctx(desc); | ||
118 | __be32 *dst = (__be32 *)out; | ||
119 | int i; | ||
120 | |||
121 | sha2_final(desc); | ||
122 | |||
123 | for (i = 0; i < SHA256_DIGEST_SIZE / sizeof(__be32); i++) | ||
124 | put_unaligned_be32(sctx->state[i], dst++); | ||
125 | |||
126 | *sctx = (struct sha256_state){}; | ||
127 | return 0; | ||
128 | } | ||
129 | |||
130 | static void sha2_finup(struct shash_desc *desc, const u8 *data, | ||
131 | unsigned int len) | ||
132 | { | ||
133 | struct sha256_state *sctx = shash_desc_ctx(desc); | ||
134 | int blocks; | ||
135 | |||
136 | if (sctx->count || !len || (len % SHA256_BLOCK_SIZE)) { | ||
137 | sha2_update(desc, data, len); | ||
138 | sha2_final(desc); | ||
139 | return; | ||
140 | } | ||
141 | |||
142 | /* | ||
143 | * Use a fast path if the input is a multiple of 64 bytes. In | ||
144 | * this case, there is no need to copy data around, and we can | ||
145 | * perform the entire digest calculation in a single invocation | ||
146 | * of sha2_ce_transform() | ||
147 | */ | ||
148 | blocks = len / SHA256_BLOCK_SIZE; | ||
149 | |||
150 | kernel_neon_begin_partial(28); | ||
151 | sha2_ce_transform(blocks, data, sctx->state, NULL, len); | ||
152 | kernel_neon_end(); | ||
153 | data += blocks * SHA256_BLOCK_SIZE; | ||
154 | } | ||
155 | |||
156 | static int sha224_finup(struct shash_desc *desc, const u8 *data, | ||
157 | unsigned int len, u8 *out) | ||
158 | { | ||
159 | struct sha256_state *sctx = shash_desc_ctx(desc); | ||
160 | __be32 *dst = (__be32 *)out; | ||
161 | int i; | ||
162 | |||
163 | sha2_finup(desc, data, len); | ||
164 | |||
165 | for (i = 0; i < SHA224_DIGEST_SIZE / sizeof(__be32); i++) | ||
166 | put_unaligned_be32(sctx->state[i], dst++); | ||
167 | |||
168 | *sctx = (struct sha256_state){}; | ||
169 | return 0; | ||
170 | } | ||
171 | |||
172 | static int sha256_finup(struct shash_desc *desc, const u8 *data, | ||
173 | unsigned int len, u8 *out) | ||
174 | { | ||
175 | struct sha256_state *sctx = shash_desc_ctx(desc); | ||
176 | __be32 *dst = (__be32 *)out; | ||
177 | int i; | ||
178 | |||
179 | sha2_finup(desc, data, len); | ||
180 | |||
181 | for (i = 0; i < SHA256_DIGEST_SIZE / sizeof(__be32); i++) | ||
182 | put_unaligned_be32(sctx->state[i], dst++); | ||
183 | |||
184 | *sctx = (struct sha256_state){}; | ||
185 | return 0; | ||
186 | } | ||
187 | |||
188 | static int sha2_export(struct shash_desc *desc, void *out) | ||
189 | { | ||
190 | struct sha256_state *sctx = shash_desc_ctx(desc); | ||
191 | struct sha256_state *dst = out; | ||
192 | |||
193 | *dst = *sctx; | ||
194 | return 0; | ||
195 | } | ||
196 | |||
197 | static int sha2_import(struct shash_desc *desc, const void *in) | ||
198 | { | ||
199 | struct sha256_state *sctx = shash_desc_ctx(desc); | ||
200 | struct sha256_state const *src = in; | ||
201 | |||
202 | *sctx = *src; | ||
203 | return 0; | ||
204 | } | ||
205 | |||
206 | static struct shash_alg algs[] = { { | ||
207 | .init = sha224_init, | ||
208 | .update = sha2_update, | ||
209 | .final = sha224_final, | ||
210 | .finup = sha224_finup, | ||
211 | .export = sha2_export, | ||
212 | .import = sha2_import, | ||
213 | .descsize = sizeof(struct sha256_state), | ||
214 | .digestsize = SHA224_DIGEST_SIZE, | ||
215 | .statesize = sizeof(struct sha256_state), | ||
216 | .base = { | ||
217 | .cra_name = "sha224", | ||
218 | .cra_driver_name = "sha224-ce", | ||
219 | .cra_priority = 200, | ||
220 | .cra_flags = CRYPTO_ALG_TYPE_SHASH, | ||
221 | .cra_blocksize = SHA256_BLOCK_SIZE, | ||
222 | .cra_module = THIS_MODULE, | ||
223 | } | ||
224 | }, { | ||
225 | .init = sha256_init, | ||
226 | .update = sha2_update, | ||
227 | .final = sha256_final, | ||
228 | .finup = sha256_finup, | ||
229 | .export = sha2_export, | ||
230 | .import = sha2_import, | ||
231 | .descsize = sizeof(struct sha256_state), | ||
232 | .digestsize = SHA256_DIGEST_SIZE, | ||
233 | .statesize = sizeof(struct sha256_state), | ||
234 | .base = { | ||
235 | .cra_name = "sha256", | ||
236 | .cra_driver_name = "sha256-ce", | ||
237 | .cra_priority = 200, | ||
238 | .cra_flags = CRYPTO_ALG_TYPE_SHASH, | ||
239 | .cra_blocksize = SHA256_BLOCK_SIZE, | ||
240 | .cra_module = THIS_MODULE, | ||
241 | } | ||
242 | } }; | ||
243 | |||
244 | static int __init sha2_ce_mod_init(void) | ||
245 | { | ||
246 | return crypto_register_shashes(algs, ARRAY_SIZE(algs)); | ||
247 | } | ||
248 | |||
249 | static void __exit sha2_ce_mod_fini(void) | ||
250 | { | ||
251 | crypto_unregister_shashes(algs, ARRAY_SIZE(algs)); | ||
252 | } | ||
253 | |||
254 | module_cpu_feature_match(SHA2, sha2_ce_mod_init); | ||
255 | module_exit(sha2_ce_mod_fini); | ||