1 /*
2 * Poly1305 authenticator algorithm, RFC7539
3 *
4 * Copyright (C) 2015 Martin Willi
5 *
6 * Based on public domain code by Andrew Moon and Daniel J. Bernstein.
7 *
8 * This program is free software; you can redistribute it and/or modify
9 * it under the terms of the GNU General Public License as published by
10 * the Free Software Foundation; either version 2 of the License, or
11 * (at your option) any later version.
12 */
13
14 #include <crypto/algapi.h>
15 #include <crypto/internal/hash.h>
16 #include <crypto/internal/poly1305.h>
17 #include <linux/crypto.h>
18 #include <linux/kernel.h>
19 #include <linux/module.h>
20 #include <asm/unaligned.h>
21
crypto_poly1305_init(struct shash_desc * desc)22 static int crypto_poly1305_init(struct shash_desc *desc)
23 {
24 struct poly1305_desc_ctx *dctx = shash_desc_ctx(desc);
25
26 poly1305_core_init(&dctx->h);
27 dctx->buflen = 0;
28 dctx->rset = 0;
29 dctx->sset = false;
30
31 return 0;
32 }
33
crypto_poly1305_setdesckey(struct poly1305_desc_ctx * dctx,const u8 * src,unsigned int srclen)34 static unsigned int crypto_poly1305_setdesckey(struct poly1305_desc_ctx *dctx,
35 const u8 *src, unsigned int srclen)
36 {
37 if (!dctx->sset) {
38 if (!dctx->rset && srclen >= POLY1305_BLOCK_SIZE) {
39 poly1305_core_setkey(&dctx->core_r, src);
40 src += POLY1305_BLOCK_SIZE;
41 srclen -= POLY1305_BLOCK_SIZE;
42 dctx->rset = 2;
43 }
44 if (srclen >= POLY1305_BLOCK_SIZE) {
45 dctx->s[0] = get_unaligned_le32(src + 0);
46 dctx->s[1] = get_unaligned_le32(src + 4);
47 dctx->s[2] = get_unaligned_le32(src + 8);
48 dctx->s[3] = get_unaligned_le32(src + 12);
49 src += POLY1305_BLOCK_SIZE;
50 srclen -= POLY1305_BLOCK_SIZE;
51 dctx->sset = true;
52 }
53 }
54 return srclen;
55 }
56
poly1305_blocks(struct poly1305_desc_ctx * dctx,const u8 * src,unsigned int srclen)57 static void poly1305_blocks(struct poly1305_desc_ctx *dctx, const u8 *src,
58 unsigned int srclen)
59 {
60 unsigned int datalen;
61
62 if (unlikely(!dctx->sset)) {
63 datalen = crypto_poly1305_setdesckey(dctx, src, srclen);
64 src += srclen - datalen;
65 srclen = datalen;
66 }
67
68 poly1305_core_blocks(&dctx->h, &dctx->core_r, src,
69 srclen / POLY1305_BLOCK_SIZE, 1);
70 }
71
crypto_poly1305_update(struct shash_desc * desc,const u8 * src,unsigned int srclen)72 static int crypto_poly1305_update(struct shash_desc *desc,
73 const u8 *src, unsigned int srclen)
74 {
75 struct poly1305_desc_ctx *dctx = shash_desc_ctx(desc);
76 unsigned int bytes;
77
78 if (unlikely(dctx->buflen)) {
79 bytes = min(srclen, POLY1305_BLOCK_SIZE - dctx->buflen);
80 memcpy(dctx->buf + dctx->buflen, src, bytes);
81 src += bytes;
82 srclen -= bytes;
83 dctx->buflen += bytes;
84
85 if (dctx->buflen == POLY1305_BLOCK_SIZE) {
86 poly1305_blocks(dctx, dctx->buf,
87 POLY1305_BLOCK_SIZE);
88 dctx->buflen = 0;
89 }
90 }
91
92 if (likely(srclen >= POLY1305_BLOCK_SIZE)) {
93 poly1305_blocks(dctx, src, srclen);
94 src += srclen - (srclen % POLY1305_BLOCK_SIZE);
95 srclen %= POLY1305_BLOCK_SIZE;
96 }
97
98 if (unlikely(srclen)) {
99 dctx->buflen = srclen;
100 memcpy(dctx->buf, src, srclen);
101 }
102
103 return 0;
104 }
105
crypto_poly1305_final(struct shash_desc * desc,u8 * dst)106 static int crypto_poly1305_final(struct shash_desc *desc, u8 *dst)
107 {
108 struct poly1305_desc_ctx *dctx = shash_desc_ctx(desc);
109
110 if (unlikely(!dctx->sset))
111 return -ENOKEY;
112
113 poly1305_final_generic(dctx, dst);
114 return 0;
115 }
116
117 static struct shash_alg poly1305_alg = {
118 .digestsize = POLY1305_DIGEST_SIZE,
119 .init = crypto_poly1305_init,
120 .update = crypto_poly1305_update,
121 .final = crypto_poly1305_final,
122 .descsize = sizeof(struct poly1305_desc_ctx),
123 .base = {
124 .cra_name = "poly1305",
125 .cra_driver_name = "poly1305-generic",
126 .cra_priority = 100,
127 .cra_blocksize = POLY1305_BLOCK_SIZE,
128 .cra_module = THIS_MODULE,
129 },
130 };
131
poly1305_mod_init(void)132 static int __init poly1305_mod_init(void)
133 {
134 return crypto_register_shash(&poly1305_alg);
135 }
136
poly1305_mod_exit(void)137 static void __exit poly1305_mod_exit(void)
138 {
139 crypto_unregister_shash(&poly1305_alg);
140 }
141
142 subsys_initcall(poly1305_mod_init);
143 module_exit(poly1305_mod_exit);
144
145 MODULE_LICENSE("GPL");
146 MODULE_AUTHOR("Martin Willi <martin@strongswan.org>");
147 MODULE_DESCRIPTION("Poly1305 authenticator");
148 MODULE_ALIAS_CRYPTO("poly1305");
149 MODULE_ALIAS_CRYPTO("poly1305-generic");
150