sha3_generic.c 8.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305
  1. // SPDX-License-Identifier: GPL-2.0-or-later
  2. /*
  3. * Cryptographic API.
  4. *
  5. * SHA-3, as specified in
  6. * https://nvlpubs.nist.gov/nistpubs/FIPS/NIST.FIPS.202.pdf
  7. *
  8. * SHA-3 code by Jeff Garzik <jeff@garzik.org>
  9. * Ard Biesheuvel <ard.biesheuvel@linaro.org>
  10. */
  11. #include <crypto/internal/hash.h>
  12. #include <linux/init.h>
  13. #include <linux/module.h>
  14. #include <linux/types.h>
  15. #include <crypto/sha3.h>
  16. #include <asm/unaligned.h>
  17. /*
  18. * On some 32-bit architectures (h8300), GCC ends up using
  19. * over 1 KB of stack if we inline the round calculation into the loop
  20. * in keccakf(). On the other hand, on 64-bit architectures with plenty
  21. * of [64-bit wide] general purpose registers, not inlining it severely
  22. * hurts performance. So let's use 64-bitness as a heuristic to decide
  23. * whether to inline or not.
  24. */
  25. #ifdef CONFIG_64BIT
  26. #define SHA3_INLINE inline
  27. #else
  28. #define SHA3_INLINE noinline
  29. #endif
  30. #define KECCAK_ROUNDS 24
  31. static const u64 keccakf_rndc[24] = {
  32. 0x0000000000000001ULL, 0x0000000000008082ULL, 0x800000000000808aULL,
  33. 0x8000000080008000ULL, 0x000000000000808bULL, 0x0000000080000001ULL,
  34. 0x8000000080008081ULL, 0x8000000000008009ULL, 0x000000000000008aULL,
  35. 0x0000000000000088ULL, 0x0000000080008009ULL, 0x000000008000000aULL,
  36. 0x000000008000808bULL, 0x800000000000008bULL, 0x8000000000008089ULL,
  37. 0x8000000000008003ULL, 0x8000000000008002ULL, 0x8000000000000080ULL,
  38. 0x000000000000800aULL, 0x800000008000000aULL, 0x8000000080008081ULL,
  39. 0x8000000000008080ULL, 0x0000000080000001ULL, 0x8000000080008008ULL
  40. };
  41. /* update the state with given number of rounds */
  42. static SHA3_INLINE void keccakf_round(u64 st[25])
  43. {
  44. u64 t[5], tt, bc[5];
  45. /* Theta */
  46. bc[0] = st[0] ^ st[5] ^ st[10] ^ st[15] ^ st[20];
  47. bc[1] = st[1] ^ st[6] ^ st[11] ^ st[16] ^ st[21];
  48. bc[2] = st[2] ^ st[7] ^ st[12] ^ st[17] ^ st[22];
  49. bc[3] = st[3] ^ st[8] ^ st[13] ^ st[18] ^ st[23];
  50. bc[4] = st[4] ^ st[9] ^ st[14] ^ st[19] ^ st[24];
  51. t[0] = bc[4] ^ rol64(bc[1], 1);
  52. t[1] = bc[0] ^ rol64(bc[2], 1);
  53. t[2] = bc[1] ^ rol64(bc[3], 1);
  54. t[3] = bc[2] ^ rol64(bc[4], 1);
  55. t[4] = bc[3] ^ rol64(bc[0], 1);
  56. st[0] ^= t[0];
  57. /* Rho Pi */
  58. tt = st[1];
  59. st[ 1] = rol64(st[ 6] ^ t[1], 44);
  60. st[ 6] = rol64(st[ 9] ^ t[4], 20);
  61. st[ 9] = rol64(st[22] ^ t[2], 61);
  62. st[22] = rol64(st[14] ^ t[4], 39);
  63. st[14] = rol64(st[20] ^ t[0], 18);
  64. st[20] = rol64(st[ 2] ^ t[2], 62);
  65. st[ 2] = rol64(st[12] ^ t[2], 43);
  66. st[12] = rol64(st[13] ^ t[3], 25);
  67. st[13] = rol64(st[19] ^ t[4], 8);
  68. st[19] = rol64(st[23] ^ t[3], 56);
  69. st[23] = rol64(st[15] ^ t[0], 41);
  70. st[15] = rol64(st[ 4] ^ t[4], 27);
  71. st[ 4] = rol64(st[24] ^ t[4], 14);
  72. st[24] = rol64(st[21] ^ t[1], 2);
  73. st[21] = rol64(st[ 8] ^ t[3], 55);
  74. st[ 8] = rol64(st[16] ^ t[1], 45);
  75. st[16] = rol64(st[ 5] ^ t[0], 36);
  76. st[ 5] = rol64(st[ 3] ^ t[3], 28);
  77. st[ 3] = rol64(st[18] ^ t[3], 21);
  78. st[18] = rol64(st[17] ^ t[2], 15);
  79. st[17] = rol64(st[11] ^ t[1], 10);
  80. st[11] = rol64(st[ 7] ^ t[2], 6);
  81. st[ 7] = rol64(st[10] ^ t[0], 3);
  82. st[10] = rol64( tt ^ t[1], 1);
  83. /* Chi */
  84. bc[ 0] = ~st[ 1] & st[ 2];
  85. bc[ 1] = ~st[ 2] & st[ 3];
  86. bc[ 2] = ~st[ 3] & st[ 4];
  87. bc[ 3] = ~st[ 4] & st[ 0];
  88. bc[ 4] = ~st[ 0] & st[ 1];
  89. st[ 0] ^= bc[ 0];
  90. st[ 1] ^= bc[ 1];
  91. st[ 2] ^= bc[ 2];
  92. st[ 3] ^= bc[ 3];
  93. st[ 4] ^= bc[ 4];
  94. bc[ 0] = ~st[ 6] & st[ 7];
  95. bc[ 1] = ~st[ 7] & st[ 8];
  96. bc[ 2] = ~st[ 8] & st[ 9];
  97. bc[ 3] = ~st[ 9] & st[ 5];
  98. bc[ 4] = ~st[ 5] & st[ 6];
  99. st[ 5] ^= bc[ 0];
  100. st[ 6] ^= bc[ 1];
  101. st[ 7] ^= bc[ 2];
  102. st[ 8] ^= bc[ 3];
  103. st[ 9] ^= bc[ 4];
  104. bc[ 0] = ~st[11] & st[12];
  105. bc[ 1] = ~st[12] & st[13];
  106. bc[ 2] = ~st[13] & st[14];
  107. bc[ 3] = ~st[14] & st[10];
  108. bc[ 4] = ~st[10] & st[11];
  109. st[10] ^= bc[ 0];
  110. st[11] ^= bc[ 1];
  111. st[12] ^= bc[ 2];
  112. st[13] ^= bc[ 3];
  113. st[14] ^= bc[ 4];
  114. bc[ 0] = ~st[16] & st[17];
  115. bc[ 1] = ~st[17] & st[18];
  116. bc[ 2] = ~st[18] & st[19];
  117. bc[ 3] = ~st[19] & st[15];
  118. bc[ 4] = ~st[15] & st[16];
  119. st[15] ^= bc[ 0];
  120. st[16] ^= bc[ 1];
  121. st[17] ^= bc[ 2];
  122. st[18] ^= bc[ 3];
  123. st[19] ^= bc[ 4];
  124. bc[ 0] = ~st[21] & st[22];
  125. bc[ 1] = ~st[22] & st[23];
  126. bc[ 2] = ~st[23] & st[24];
  127. bc[ 3] = ~st[24] & st[20];
  128. bc[ 4] = ~st[20] & st[21];
  129. st[20] ^= bc[ 0];
  130. st[21] ^= bc[ 1];
  131. st[22] ^= bc[ 2];
  132. st[23] ^= bc[ 3];
  133. st[24] ^= bc[ 4];
  134. }
  135. static void keccakf(u64 st[25])
  136. {
  137. int round;
  138. for (round = 0; round < KECCAK_ROUNDS; round++) {
  139. keccakf_round(st);
  140. /* Iota */
  141. st[0] ^= keccakf_rndc[round];
  142. }
  143. }
  144. int crypto_sha3_init(struct shash_desc *desc)
  145. {
  146. struct sha3_state *sctx = shash_desc_ctx(desc);
  147. unsigned int digest_size = crypto_shash_digestsize(desc->tfm);
  148. sctx->rsiz = 200 - 2 * digest_size;
  149. sctx->rsizw = sctx->rsiz / 8;
  150. sctx->partial = 0;
  151. memset(sctx->st, 0, sizeof(sctx->st));
  152. return 0;
  153. }
  154. EXPORT_SYMBOL(crypto_sha3_init);
  155. int crypto_sha3_update(struct shash_desc *desc, const u8 *data,
  156. unsigned int len)
  157. {
  158. struct sha3_state *sctx = shash_desc_ctx(desc);
  159. unsigned int done;
  160. const u8 *src;
  161. done = 0;
  162. src = data;
  163. if ((sctx->partial + len) > (sctx->rsiz - 1)) {
  164. if (sctx->partial) {
  165. done = -sctx->partial;
  166. memcpy(sctx->buf + sctx->partial, data,
  167. done + sctx->rsiz);
  168. src = sctx->buf;
  169. }
  170. do {
  171. unsigned int i;
  172. for (i = 0; i < sctx->rsizw; i++)
  173. sctx->st[i] ^= get_unaligned_le64(src + 8 * i);
  174. keccakf(sctx->st);
  175. done += sctx->rsiz;
  176. src = data + done;
  177. } while (done + (sctx->rsiz - 1) < len);
  178. sctx->partial = 0;
  179. }
  180. memcpy(sctx->buf + sctx->partial, src, len - done);
  181. sctx->partial += (len - done);
  182. return 0;
  183. }
  184. EXPORT_SYMBOL(crypto_sha3_update);
  185. int crypto_sha3_final(struct shash_desc *desc, u8 *out)
  186. {
  187. struct sha3_state *sctx = shash_desc_ctx(desc);
  188. unsigned int i, inlen = sctx->partial;
  189. unsigned int digest_size = crypto_shash_digestsize(desc->tfm);
  190. __le64 *digest = (__le64 *)out;
  191. sctx->buf[inlen++] = 0x06;
  192. memset(sctx->buf + inlen, 0, sctx->rsiz - inlen);
  193. sctx->buf[sctx->rsiz - 1] |= 0x80;
  194. for (i = 0; i < sctx->rsizw; i++)
  195. sctx->st[i] ^= get_unaligned_le64(sctx->buf + 8 * i);
  196. keccakf(sctx->st);
  197. for (i = 0; i < digest_size / 8; i++)
  198. put_unaligned_le64(sctx->st[i], digest++);
  199. if (digest_size & 4)
  200. put_unaligned_le32(sctx->st[i], (__le32 *)digest);
  201. memset(sctx, 0, sizeof(*sctx));
  202. return 0;
  203. }
  204. EXPORT_SYMBOL(crypto_sha3_final);
  205. static struct shash_alg algs[] = { {
  206. .digestsize = SHA3_224_DIGEST_SIZE,
  207. .init = crypto_sha3_init,
  208. .update = crypto_sha3_update,
  209. .final = crypto_sha3_final,
  210. .descsize = sizeof(struct sha3_state),
  211. .base.cra_name = "sha3-224",
  212. .base.cra_driver_name = "sha3-224-generic",
  213. .base.cra_blocksize = SHA3_224_BLOCK_SIZE,
  214. .base.cra_module = THIS_MODULE,
  215. }, {
  216. .digestsize = SHA3_256_DIGEST_SIZE,
  217. .init = crypto_sha3_init,
  218. .update = crypto_sha3_update,
  219. .final = crypto_sha3_final,
  220. .descsize = sizeof(struct sha3_state),
  221. .base.cra_name = "sha3-256",
  222. .base.cra_driver_name = "sha3-256-generic",
  223. .base.cra_blocksize = SHA3_256_BLOCK_SIZE,
  224. .base.cra_module = THIS_MODULE,
  225. }, {
  226. .digestsize = SHA3_384_DIGEST_SIZE,
  227. .init = crypto_sha3_init,
  228. .update = crypto_sha3_update,
  229. .final = crypto_sha3_final,
  230. .descsize = sizeof(struct sha3_state),
  231. .base.cra_name = "sha3-384",
  232. .base.cra_driver_name = "sha3-384-generic",
  233. .base.cra_blocksize = SHA3_384_BLOCK_SIZE,
  234. .base.cra_module = THIS_MODULE,
  235. }, {
  236. .digestsize = SHA3_512_DIGEST_SIZE,
  237. .init = crypto_sha3_init,
  238. .update = crypto_sha3_update,
  239. .final = crypto_sha3_final,
  240. .descsize = sizeof(struct sha3_state),
  241. .base.cra_name = "sha3-512",
  242. .base.cra_driver_name = "sha3-512-generic",
  243. .base.cra_blocksize = SHA3_512_BLOCK_SIZE,
  244. .base.cra_module = THIS_MODULE,
  245. } };
  246. static int __init sha3_generic_mod_init(void)
  247. {
  248. return crypto_register_shashes(algs, ARRAY_SIZE(algs));
  249. }
  250. static void __exit sha3_generic_mod_fini(void)
  251. {
  252. crypto_unregister_shashes(algs, ARRAY_SIZE(algs));
  253. }
  254. subsys_initcall(sha3_generic_mod_init);
  255. module_exit(sha3_generic_mod_fini);
  256. MODULE_LICENSE("GPL");
  257. MODULE_DESCRIPTION("SHA-3 Secure Hash Algorithm");
  258. MODULE_ALIAS_CRYPTO("sha3-224");
  259. MODULE_ALIAS_CRYPTO("sha3-224-generic");
  260. MODULE_ALIAS_CRYPTO("sha3-256");
  261. MODULE_ALIAS_CRYPTO("sha3-256-generic");
  262. MODULE_ALIAS_CRYPTO("sha3-384");
  263. MODULE_ALIAS_CRYPTO("sha3-384-generic");
  264. MODULE_ALIAS_CRYPTO("sha3-512");
  265. MODULE_ALIAS_CRYPTO("sha3-512-generic");