/src/openssl/crypto/evp/e_chacha20_poly1305.c
Line  | Count  | Source (jump to first uncovered line)  | 
1  |  | /*  | 
2  |  |  * Copyright 2015-2023 The OpenSSL Project Authors. All Rights Reserved.  | 
3  |  |  *  | 
4  |  |  * Licensed under the Apache License 2.0 (the "License").  You may not use  | 
5  |  |  * this file except in compliance with the License.  You can obtain a copy  | 
6  |  |  * in the file LICENSE in the source distribution or at  | 
7  |  |  * https://www.openssl.org/source/license.html  | 
8  |  |  */  | 
9  |  |  | 
10  |  | #include <stdio.h>  | 
11  |  | #include "internal/cryptlib.h"  | 
12  |  | #include "internal/endian.h"  | 
13  |  |  | 
14  |  | #ifndef OPENSSL_NO_CHACHA  | 
15  |  |  | 
16  |  | # include <openssl/evp.h>  | 
17  |  | # include <openssl/objects.h>  | 
18  |  | # include "crypto/evp.h"  | 
19  |  | # include "evp_local.h"  | 
20  |  | # include "crypto/chacha.h"  | 
21  |  |  | 
22  |  | typedef struct { | 
23  |  |     union { | 
24  |  |         OSSL_UNION_ALIGN;  /* this ensures even sizeof(EVP_CHACHA_KEY)%8==0 */  | 
25  |  |         unsigned int d[CHACHA_KEY_SIZE / 4];  | 
26  |  |     } key;  | 
27  |  |     unsigned int  counter[CHACHA_CTR_SIZE / 4];  | 
28  |  |     unsigned char buf[CHACHA_BLK_SIZE];  | 
29  |  |     unsigned int  partial_len;  | 
30  |  | } EVP_CHACHA_KEY;  | 
31  |  |  | 
32  | 0  | #define data(ctx)   ((EVP_CHACHA_KEY *)(ctx)->cipher_data)  | 
33  |  |  | 
34  | 0  | #define CHACHA20_POLY1305_MAX_IVLEN     12  | 
35  |  |  | 
36  |  | static int chacha_init_key(EVP_CIPHER_CTX *ctx,  | 
37  |  |                            const unsigned char user_key[CHACHA_KEY_SIZE],  | 
38  |  |                            const unsigned char iv[CHACHA_CTR_SIZE], int enc)  | 
39  | 0  | { | 
40  | 0  |     EVP_CHACHA_KEY *key = data(ctx);  | 
41  | 0  |     unsigned int i;  | 
42  |  | 
  | 
43  | 0  |     if (user_key)  | 
44  | 0  |         for (i = 0; i < CHACHA_KEY_SIZE; i+=4) { | 
45  | 0  |             key->key.d[i/4] = CHACHA_U8TOU32(user_key+i);  | 
46  | 0  |         }  | 
47  |  | 
  | 
48  | 0  |     if (iv)  | 
49  | 0  |         for (i = 0; i < CHACHA_CTR_SIZE; i+=4) { | 
50  | 0  |             key->counter[i/4] = CHACHA_U8TOU32(iv+i);  | 
51  | 0  |         }  | 
52  |  | 
  | 
53  | 0  |     key->partial_len = 0;  | 
54  |  | 
  | 
55  | 0  |     return 1;  | 
56  | 0  | }  | 
57  |  |  | 
58  |  | static int chacha_cipher(EVP_CIPHER_CTX *ctx, unsigned char *out,  | 
59  |  |                          const unsigned char *inp, size_t len)  | 
60  | 0  | { | 
61  | 0  |     EVP_CHACHA_KEY *key = data(ctx);  | 
62  | 0  |     unsigned int n, rem, ctr32;  | 
63  |  | 
  | 
64  | 0  |     if ((n = key->partial_len)) { | 
65  | 0  |         while (len && n < CHACHA_BLK_SIZE) { | 
66  | 0  |             *out++ = *inp++ ^ key->buf[n++];  | 
67  | 0  |             len--;  | 
68  | 0  |         }  | 
69  | 0  |         key->partial_len = n;  | 
70  |  | 
  | 
71  | 0  |         if (len == 0)  | 
72  | 0  |             return 1;  | 
73  |  |  | 
74  | 0  |         if (n == CHACHA_BLK_SIZE) { | 
75  | 0  |             key->partial_len = 0;  | 
76  | 0  |             key->counter[0]++;  | 
77  | 0  |             if (key->counter[0] == 0)  | 
78  | 0  |                 key->counter[1]++;  | 
79  | 0  |         }  | 
80  | 0  |     }  | 
81  |  |  | 
82  | 0  |     rem = (unsigned int)(len % CHACHA_BLK_SIZE);  | 
83  | 0  |     len -= rem;  | 
84  | 0  |     ctr32 = key->counter[0];  | 
85  | 0  |     while (len >= CHACHA_BLK_SIZE) { | 
86  | 0  |         size_t blocks = len / CHACHA_BLK_SIZE;  | 
87  |  |         /*  | 
88  |  |          * 1<<28 is just a not-so-small yet not-so-large number...  | 
89  |  |          * Below condition is practically never met, but it has to  | 
90  |  |          * be checked for code correctness.  | 
91  |  |          */  | 
92  | 0  |         if (sizeof(size_t)>sizeof(unsigned int) && blocks>(1U<<28))  | 
93  | 0  |             blocks = (1U<<28);  | 
94  |  |  | 
95  |  |         /*  | 
96  |  |          * As ChaCha20_ctr32 operates on 32-bit counter, caller  | 
97  |  |          * has to handle overflow. 'if' below detects the  | 
98  |  |          * overflow, which is then handled by limiting the  | 
99  |  |          * amount of blocks to the exact overflow point...  | 
100  |  |          */  | 
101  | 0  |         ctr32 += (unsigned int)blocks;  | 
102  | 0  |         if (ctr32 < blocks) { | 
103  | 0  |             blocks -= ctr32;  | 
104  | 0  |             ctr32 = 0;  | 
105  | 0  |         }  | 
106  | 0  |         blocks *= CHACHA_BLK_SIZE;  | 
107  | 0  |         ChaCha20_ctr32(out, inp, blocks, key->key.d, key->counter);  | 
108  | 0  |         len -= blocks;  | 
109  | 0  |         inp += blocks;  | 
110  | 0  |         out += blocks;  | 
111  |  | 
  | 
112  | 0  |         key->counter[0] = ctr32;  | 
113  | 0  |         if (ctr32 == 0) key->counter[1]++;  | 
114  | 0  |     }  | 
115  |  | 
  | 
116  | 0  |     if (rem) { | 
117  | 0  |         memset(key->buf, 0, sizeof(key->buf));  | 
118  | 0  |         ChaCha20_ctr32(key->buf, key->buf, CHACHA_BLK_SIZE,  | 
119  | 0  |                        key->key.d, key->counter);  | 
120  | 0  |         for (n = 0; n < rem; n++)  | 
121  | 0  |             out[n] = inp[n] ^ key->buf[n];  | 
122  | 0  |         key->partial_len = rem;  | 
123  | 0  |     }  | 
124  |  | 
  | 
125  | 0  |     return 1;  | 
126  | 0  | }  | 
127  |  |  | 
128  |  | static const EVP_CIPHER chacha20 = { | 
129  |  |     NID_chacha20,  | 
130  |  |     1,                      /* block_size */  | 
131  |  |     CHACHA_KEY_SIZE,        /* key_len */  | 
132  |  |     CHACHA_CTR_SIZE,        /* iv_len, 128-bit counter in the context */  | 
133  |  |     EVP_CIPH_CUSTOM_IV | EVP_CIPH_ALWAYS_CALL_INIT,  | 
134  |  |     EVP_ORIG_GLOBAL,  | 
135  |  |     chacha_init_key,  | 
136  |  |     chacha_cipher,  | 
137  |  |     NULL,  | 
138  |  |     sizeof(EVP_CHACHA_KEY),  | 
139  |  |     NULL,  | 
140  |  |     NULL,  | 
141  |  |     NULL,  | 
142  |  |     NULL  | 
143  |  | };  | 
144  |  |  | 
145  |  | const EVP_CIPHER *EVP_chacha20(void)  | 
146  | 3  | { | 
147  | 3  |     return &chacha20;  | 
148  | 3  | }  | 
149  |  |  | 
150  |  | # ifndef OPENSSL_NO_POLY1305  | 
151  |  | #  include "crypto/poly1305.h"  | 
152  |  |  | 
153  |  | typedef struct { | 
154  |  |     EVP_CHACHA_KEY key;  | 
155  |  |     unsigned int nonce[12/4];  | 
156  |  |     unsigned char tag[POLY1305_BLOCK_SIZE];  | 
157  |  |     unsigned char tls_aad[POLY1305_BLOCK_SIZE];  | 
158  |  |     struct { uint64_t aad, text; } len; | 
159  |  |     int aad, mac_inited, tag_len, nonce_len;  | 
160  |  |     size_t tls_payload_length;  | 
161  |  | } EVP_CHACHA_AEAD_CTX;  | 
162  |  |  | 
163  | 0  | #  define NO_TLS_PAYLOAD_LENGTH ((size_t)-1)  | 
164  | 0  | #  define aead_data(ctx)        ((EVP_CHACHA_AEAD_CTX *)(ctx)->cipher_data)  | 
165  | 0  | #  define POLY1305_ctx(actx)    ((POLY1305 *)(actx + 1))  | 
166  |  |  | 
167  |  | static int chacha20_poly1305_init_key(EVP_CIPHER_CTX *ctx,  | 
168  |  |                                       const unsigned char *inkey,  | 
169  |  |                                       const unsigned char *iv, int enc)  | 
170  | 0  | { | 
171  | 0  |     EVP_CHACHA_AEAD_CTX *actx = aead_data(ctx);  | 
172  |  | 
  | 
173  | 0  |     if (!inkey && !iv)  | 
174  | 0  |         return 1;  | 
175  |  |  | 
176  | 0  |     actx->len.aad = 0;  | 
177  | 0  |     actx->len.text = 0;  | 
178  | 0  |     actx->aad = 0;  | 
179  | 0  |     actx->mac_inited = 0;  | 
180  | 0  |     actx->tls_payload_length = NO_TLS_PAYLOAD_LENGTH;  | 
181  |  | 
  | 
182  | 0  |     if (iv != NULL) { | 
183  | 0  |         unsigned char temp[CHACHA_CTR_SIZE] = { 0 }; | 
184  |  |  | 
185  |  |         /* pad on the left */  | 
186  | 0  |         if (actx->nonce_len <= CHACHA_CTR_SIZE)  | 
187  | 0  |             memcpy(temp + CHACHA_CTR_SIZE - actx->nonce_len, iv,  | 
188  | 0  |                    actx->nonce_len);  | 
189  |  | 
  | 
190  | 0  |         chacha_init_key(ctx, inkey, temp, enc);  | 
191  |  | 
  | 
192  | 0  |         actx->nonce[0] = actx->key.counter[1];  | 
193  | 0  |         actx->nonce[1] = actx->key.counter[2];  | 
194  | 0  |         actx->nonce[2] = actx->key.counter[3];  | 
195  | 0  |     } else { | 
196  | 0  |         chacha_init_key(ctx, inkey, NULL, enc);  | 
197  | 0  |     }  | 
198  |  | 
  | 
199  | 0  |     return 1;  | 
200  | 0  | }  | 
201  |  |  | 
202  |  | #  if !defined(OPENSSL_SMALL_FOOTPRINT)  | 
203  |  |  | 
204  |  | #   if defined(POLY1305_ASM) && (defined(__x86_64) || defined(__x86_64__) || \  | 
205  |  |                                  defined(_M_AMD64) || defined(_M_X64))  | 
206  |  | #    define XOR128_HELPERS  | 
207  |  | void *xor128_encrypt_n_pad(void *out, const void *inp, void *otp, size_t len);  | 
208  |  | void *xor128_decrypt_n_pad(void *out, const void *inp, void *otp, size_t len);  | 
209  |  | static const unsigned char zero[4 * CHACHA_BLK_SIZE] = { 0 }; | 
210  |  | #   else  | 
211  |  | static const unsigned char zero[2 * CHACHA_BLK_SIZE] = { 0 }; | 
212  |  | #   endif  | 
213  |  |  | 
214  |  | static int chacha20_poly1305_tls_cipher(EVP_CIPHER_CTX *ctx, unsigned char *out,  | 
215  |  |                                         const unsigned char *in, size_t len)  | 
216  | 0  | { | 
217  | 0  |     EVP_CHACHA_AEAD_CTX *actx = aead_data(ctx);  | 
218  | 0  |     size_t tail, tohash_len, buf_len, plen = actx->tls_payload_length;  | 
219  | 0  |     unsigned char *buf, *tohash, *ctr, storage[sizeof(zero) + 32];  | 
220  |  | 
  | 
221  | 0  |     if (len != plen + POLY1305_BLOCK_SIZE)  | 
222  | 0  |         return -1;  | 
223  |  |  | 
224  | 0  |     buf = storage + ((0 - (size_t)storage) & 15);   /* align */  | 
225  | 0  |     ctr = buf + CHACHA_BLK_SIZE;  | 
226  | 0  |     tohash = buf + CHACHA_BLK_SIZE - POLY1305_BLOCK_SIZE;  | 
227  |  | 
  | 
228  |  | #   ifdef XOR128_HELPERS  | 
229  |  |     if (plen <= 3 * CHACHA_BLK_SIZE) { | 
230  |  |         actx->key.counter[0] = 0;  | 
231  |  |         buf_len = (plen + 2 * CHACHA_BLK_SIZE - 1) & (0 - CHACHA_BLK_SIZE);  | 
232  |  |         ChaCha20_ctr32(buf, zero, buf_len, actx->key.key.d,  | 
233  |  |                        actx->key.counter);  | 
234  |  |         Poly1305_Init(POLY1305_ctx(actx), buf);  | 
235  |  |         actx->key.partial_len = 0;  | 
236  |  |         memcpy(tohash, actx->tls_aad, POLY1305_BLOCK_SIZE);  | 
237  |  |         tohash_len = POLY1305_BLOCK_SIZE;  | 
238  |  |         actx->len.aad = EVP_AEAD_TLS1_AAD_LEN;  | 
239  |  |         actx->len.text = plen;  | 
240  |  |  | 
241  |  |         if (plen) { | 
242  |  |             if (EVP_CIPHER_CTX_is_encrypting(ctx))  | 
243  |  |                 ctr = xor128_encrypt_n_pad(out, in, ctr, plen);  | 
244  |  |             else  | 
245  |  |                 ctr = xor128_decrypt_n_pad(out, in, ctr, plen);  | 
246  |  |  | 
247  |  |             in += plen;  | 
248  |  |             out += plen;  | 
249  |  |             tohash_len = (size_t)(ctr - tohash);  | 
250  |  |         }  | 
251  |  |     }  | 
252  |  | #   else  | 
253  | 0  |     if (plen <= CHACHA_BLK_SIZE) { | 
254  | 0  |         size_t i;  | 
255  |  | 
  | 
256  | 0  |         actx->key.counter[0] = 0;  | 
257  | 0  |         ChaCha20_ctr32(buf, zero, (buf_len = 2 * CHACHA_BLK_SIZE),  | 
258  | 0  |                        actx->key.key.d, actx->key.counter);  | 
259  | 0  |         Poly1305_Init(POLY1305_ctx(actx), buf);  | 
260  | 0  |         actx->key.partial_len = 0;  | 
261  | 0  |         memcpy(tohash, actx->tls_aad, POLY1305_BLOCK_SIZE);  | 
262  | 0  |         tohash_len = POLY1305_BLOCK_SIZE;  | 
263  | 0  |         actx->len.aad = EVP_AEAD_TLS1_AAD_LEN;  | 
264  | 0  |         actx->len.text = plen;  | 
265  |  | 
  | 
266  | 0  |         if (EVP_CIPHER_CTX_is_encrypting(ctx)) { | 
267  | 0  |             for (i = 0; i < plen; i++) { | 
268  | 0  |                 out[i] = ctr[i] ^= in[i];  | 
269  | 0  |             }  | 
270  | 0  |         } else { | 
271  | 0  |             for (i = 0; i < plen; i++) { | 
272  | 0  |                 unsigned char c = in[i];  | 
273  | 0  |                 out[i] = ctr[i] ^ c;  | 
274  | 0  |                 ctr[i] = c;  | 
275  | 0  |             }  | 
276  | 0  |         }  | 
277  |  | 
  | 
278  | 0  |         in += i;  | 
279  | 0  |         out += i;  | 
280  |  | 
  | 
281  | 0  |         tail = (0 - i) & (POLY1305_BLOCK_SIZE - 1);  | 
282  | 0  |         memset(ctr + i, 0, tail);  | 
283  | 0  |         ctr += i + tail;  | 
284  | 0  |         tohash_len += i + tail;  | 
285  | 0  |     }  | 
286  | 0  | #   endif  | 
287  | 0  |     else { | 
288  | 0  |         actx->key.counter[0] = 0;  | 
289  | 0  |         ChaCha20_ctr32(buf, zero, (buf_len = CHACHA_BLK_SIZE),  | 
290  | 0  |                        actx->key.key.d, actx->key.counter);  | 
291  | 0  |         Poly1305_Init(POLY1305_ctx(actx), buf);  | 
292  | 0  |         actx->key.counter[0] = 1;  | 
293  | 0  |         actx->key.partial_len = 0;  | 
294  | 0  |         Poly1305_Update(POLY1305_ctx(actx), actx->tls_aad, POLY1305_BLOCK_SIZE);  | 
295  | 0  |         tohash = ctr;  | 
296  | 0  |         tohash_len = 0;  | 
297  | 0  |         actx->len.aad = EVP_AEAD_TLS1_AAD_LEN;  | 
298  | 0  |         actx->len.text = plen;  | 
299  |  | 
  | 
300  | 0  |         if (EVP_CIPHER_CTX_is_encrypting(ctx)) { | 
301  | 0  |             ChaCha20_ctr32(out, in, plen, actx->key.key.d, actx->key.counter);  | 
302  | 0  |             Poly1305_Update(POLY1305_ctx(actx), out, plen);  | 
303  | 0  |         } else { | 
304  | 0  |             Poly1305_Update(POLY1305_ctx(actx), in, plen);  | 
305  | 0  |             ChaCha20_ctr32(out, in, plen, actx->key.key.d, actx->key.counter);  | 
306  | 0  |         }  | 
307  |  | 
  | 
308  | 0  |         in += plen;  | 
309  | 0  |         out += plen;  | 
310  | 0  |         tail = (0 - plen) & (POLY1305_BLOCK_SIZE - 1);  | 
311  | 0  |         Poly1305_Update(POLY1305_ctx(actx), zero, tail);  | 
312  | 0  |     }  | 
313  |  | 
  | 
314  | 0  |     { | 
315  | 0  |         DECLARE_IS_ENDIAN;  | 
316  |  | 
  | 
317  | 0  |         if (IS_LITTLE_ENDIAN) { | 
318  | 0  |             memcpy(ctr, (unsigned char *)&actx->len, POLY1305_BLOCK_SIZE);  | 
319  | 0  |         } else { | 
320  | 0  |             ctr[0]  = (unsigned char)(actx->len.aad);  | 
321  | 0  |             ctr[1]  = (unsigned char)(actx->len.aad>>8);  | 
322  | 0  |             ctr[2]  = (unsigned char)(actx->len.aad>>16);  | 
323  | 0  |             ctr[3]  = (unsigned char)(actx->len.aad>>24);  | 
324  | 0  |             ctr[4]  = (unsigned char)(actx->len.aad>>32);  | 
325  | 0  |             ctr[5]  = (unsigned char)(actx->len.aad>>40);  | 
326  | 0  |             ctr[6]  = (unsigned char)(actx->len.aad>>48);  | 
327  | 0  |             ctr[7]  = (unsigned char)(actx->len.aad>>56);  | 
328  |  | 
  | 
329  | 0  |             ctr[8]  = (unsigned char)(actx->len.text);  | 
330  | 0  |             ctr[9]  = (unsigned char)(actx->len.text>>8);  | 
331  | 0  |             ctr[10] = (unsigned char)(actx->len.text>>16);  | 
332  | 0  |             ctr[11] = (unsigned char)(actx->len.text>>24);  | 
333  | 0  |             ctr[12] = (unsigned char)(actx->len.text>>32);  | 
334  | 0  |             ctr[13] = (unsigned char)(actx->len.text>>40);  | 
335  | 0  |             ctr[14] = (unsigned char)(actx->len.text>>48);  | 
336  | 0  |             ctr[15] = (unsigned char)(actx->len.text>>56);  | 
337  | 0  |         }  | 
338  | 0  |         tohash_len += POLY1305_BLOCK_SIZE;  | 
339  | 0  |     }  | 
340  |  | 
  | 
341  | 0  |     Poly1305_Update(POLY1305_ctx(actx), tohash, tohash_len);  | 
342  | 0  |     OPENSSL_cleanse(buf, buf_len);  | 
343  | 0  |     Poly1305_Final(POLY1305_ctx(actx),  | 
344  | 0  |                    EVP_CIPHER_CTX_is_encrypting(ctx) ? actx->tag : tohash);  | 
345  |  | 
  | 
346  | 0  |     actx->tls_payload_length = NO_TLS_PAYLOAD_LENGTH;  | 
347  |  | 
  | 
348  | 0  |     if (EVP_CIPHER_CTX_is_encrypting(ctx)) { | 
349  | 0  |         memcpy(out, actx->tag, POLY1305_BLOCK_SIZE);  | 
350  | 0  |     } else { | 
351  | 0  |         if (CRYPTO_memcmp(tohash, in, POLY1305_BLOCK_SIZE)) { | 
352  | 0  |             memset(out - (len - POLY1305_BLOCK_SIZE), 0,  | 
353  | 0  |                    len - POLY1305_BLOCK_SIZE);  | 
354  | 0  |             return -1;  | 
355  | 0  |         }  | 
356  | 0  |     }  | 
357  |  |  | 
358  | 0  |     return len;  | 
359  | 0  | }  | 
360  |  | #  else  | 
361  |  | static const unsigned char zero[CHACHA_BLK_SIZE] = { 0 }; | 
362  |  | #  endif  | 
363  |  |  | 
364  |  | static int chacha20_poly1305_cipher(EVP_CIPHER_CTX *ctx, unsigned char *out,  | 
365  |  |                                     const unsigned char *in, size_t len)  | 
366  | 0  | { | 
367  | 0  |     EVP_CHACHA_AEAD_CTX *actx = aead_data(ctx);  | 
368  | 0  |     size_t rem, plen = actx->tls_payload_length;  | 
369  |  | 
  | 
370  | 0  |     if (!actx->mac_inited) { | 
371  | 0  | #  if !defined(OPENSSL_SMALL_FOOTPRINT)  | 
372  | 0  |         if (plen != NO_TLS_PAYLOAD_LENGTH && out != NULL)  | 
373  | 0  |             return chacha20_poly1305_tls_cipher(ctx, out, in, len);  | 
374  | 0  | #  endif  | 
375  | 0  |         actx->key.counter[0] = 0;  | 
376  | 0  |         ChaCha20_ctr32(actx->key.buf, zero, CHACHA_BLK_SIZE,  | 
377  | 0  |                        actx->key.key.d, actx->key.counter);  | 
378  | 0  |         Poly1305_Init(POLY1305_ctx(actx), actx->key.buf);  | 
379  | 0  |         actx->key.counter[0] = 1;  | 
380  | 0  |         actx->key.partial_len = 0;  | 
381  | 0  |         actx->len.aad = actx->len.text = 0;  | 
382  | 0  |         actx->mac_inited = 1;  | 
383  | 0  |         if (plen != NO_TLS_PAYLOAD_LENGTH) { | 
384  | 0  |             Poly1305_Update(POLY1305_ctx(actx), actx->tls_aad,  | 
385  | 0  |                             EVP_AEAD_TLS1_AAD_LEN);  | 
386  | 0  |             actx->len.aad = EVP_AEAD_TLS1_AAD_LEN;  | 
387  | 0  |             actx->aad = 1;  | 
388  | 0  |         }  | 
389  | 0  |     }  | 
390  |  |  | 
391  | 0  |     if (in) {                                   /* aad or text */ | 
392  | 0  |         if (out == NULL) {                      /* aad */ | 
393  | 0  |             Poly1305_Update(POLY1305_ctx(actx), in, len);  | 
394  | 0  |             actx->len.aad += len;  | 
395  | 0  |             actx->aad = 1;  | 
396  | 0  |             return len;  | 
397  | 0  |         } else {                                /* plain- or ciphertext */ | 
398  | 0  |             if (actx->aad) {                    /* wrap up aad */ | 
399  | 0  |                 if ((rem = (size_t)actx->len.aad % POLY1305_BLOCK_SIZE))  | 
400  | 0  |                     Poly1305_Update(POLY1305_ctx(actx), zero,  | 
401  | 0  |                                     POLY1305_BLOCK_SIZE - rem);  | 
402  | 0  |                 actx->aad = 0;  | 
403  | 0  |             }  | 
404  |  | 
  | 
405  | 0  |             actx->tls_payload_length = NO_TLS_PAYLOAD_LENGTH;  | 
406  | 0  |             if (plen == NO_TLS_PAYLOAD_LENGTH)  | 
407  | 0  |                 plen = len;  | 
408  | 0  |             else if (len != plen + POLY1305_BLOCK_SIZE)  | 
409  | 0  |                 return -1;  | 
410  |  |  | 
411  | 0  |             if (EVP_CIPHER_CTX_is_encrypting(ctx)) {    /* plaintext */ | 
412  | 0  |                 chacha_cipher(ctx, out, in, plen);  | 
413  | 0  |                 Poly1305_Update(POLY1305_ctx(actx), out, plen);  | 
414  | 0  |                 in += plen;  | 
415  | 0  |                 out += plen;  | 
416  | 0  |                 actx->len.text += plen;  | 
417  | 0  |             } else {                            /* ciphertext */ | 
418  | 0  |                 Poly1305_Update(POLY1305_ctx(actx), in, plen);  | 
419  | 0  |                 chacha_cipher(ctx, out, in, plen);  | 
420  | 0  |                 in += plen;  | 
421  | 0  |                 out += plen;  | 
422  | 0  |                 actx->len.text += plen;  | 
423  | 0  |             }  | 
424  | 0  |         }  | 
425  | 0  |     }  | 
426  | 0  |     if (in == NULL                              /* explicit final */  | 
427  | 0  |         || plen != len) {                       /* or tls mode */ | 
428  | 0  |         DECLARE_IS_ENDIAN;  | 
429  | 0  |         unsigned char temp[POLY1305_BLOCK_SIZE];  | 
430  |  | 
  | 
431  | 0  |         if (actx->aad) {                        /* wrap up aad */ | 
432  | 0  |             if ((rem = (size_t)actx->len.aad % POLY1305_BLOCK_SIZE))  | 
433  | 0  |                 Poly1305_Update(POLY1305_ctx(actx), zero,  | 
434  | 0  |                                 POLY1305_BLOCK_SIZE - rem);  | 
435  | 0  |             actx->aad = 0;  | 
436  | 0  |         }  | 
437  |  | 
  | 
438  | 0  |         if ((rem = (size_t)actx->len.text % POLY1305_BLOCK_SIZE))  | 
439  | 0  |             Poly1305_Update(POLY1305_ctx(actx), zero,  | 
440  | 0  |                             POLY1305_BLOCK_SIZE - rem);  | 
441  |  | 
  | 
442  | 0  |         if (IS_LITTLE_ENDIAN) { | 
443  | 0  |             Poly1305_Update(POLY1305_ctx(actx),  | 
444  | 0  |                             (unsigned char *)&actx->len, POLY1305_BLOCK_SIZE);  | 
445  | 0  |         } else { | 
446  | 0  |             temp[0]  = (unsigned char)(actx->len.aad);  | 
447  | 0  |             temp[1]  = (unsigned char)(actx->len.aad>>8);  | 
448  | 0  |             temp[2]  = (unsigned char)(actx->len.aad>>16);  | 
449  | 0  |             temp[3]  = (unsigned char)(actx->len.aad>>24);  | 
450  | 0  |             temp[4]  = (unsigned char)(actx->len.aad>>32);  | 
451  | 0  |             temp[5]  = (unsigned char)(actx->len.aad>>40);  | 
452  | 0  |             temp[6]  = (unsigned char)(actx->len.aad>>48);  | 
453  | 0  |             temp[7]  = (unsigned char)(actx->len.aad>>56);  | 
454  |  | 
  | 
455  | 0  |             temp[8]  = (unsigned char)(actx->len.text);  | 
456  | 0  |             temp[9]  = (unsigned char)(actx->len.text>>8);  | 
457  | 0  |             temp[10] = (unsigned char)(actx->len.text>>16);  | 
458  | 0  |             temp[11] = (unsigned char)(actx->len.text>>24);  | 
459  | 0  |             temp[12] = (unsigned char)(actx->len.text>>32);  | 
460  | 0  |             temp[13] = (unsigned char)(actx->len.text>>40);  | 
461  | 0  |             temp[14] = (unsigned char)(actx->len.text>>48);  | 
462  | 0  |             temp[15] = (unsigned char)(actx->len.text>>56);  | 
463  |  | 
  | 
464  | 0  |             Poly1305_Update(POLY1305_ctx(actx), temp, POLY1305_BLOCK_SIZE);  | 
465  | 0  |         }  | 
466  | 0  |         Poly1305_Final(POLY1305_ctx(actx),  | 
467  | 0  |                        EVP_CIPHER_CTX_is_encrypting(ctx) ? actx->tag : temp);  | 
468  | 0  |         actx->mac_inited = 0;  | 
469  |  | 
  | 
470  | 0  |         if (in != NULL && len != plen) {        /* tls mode */ | 
471  | 0  |             if (EVP_CIPHER_CTX_is_encrypting(ctx)) { | 
472  | 0  |                 memcpy(out, actx->tag, POLY1305_BLOCK_SIZE);  | 
473  | 0  |             } else { | 
474  | 0  |                 if (CRYPTO_memcmp(temp, in, POLY1305_BLOCK_SIZE)) { | 
475  | 0  |                     memset(out - plen, 0, plen);  | 
476  | 0  |                     return -1;  | 
477  | 0  |                 }  | 
478  | 0  |             }  | 
479  | 0  |         }  | 
480  | 0  |         else if (!EVP_CIPHER_CTX_is_encrypting(ctx)) { | 
481  | 0  |             if (CRYPTO_memcmp(temp, actx->tag, actx->tag_len))  | 
482  | 0  |                 return -1;  | 
483  | 0  |         }  | 
484  | 0  |     }  | 
485  | 0  |     return len;  | 
486  | 0  | }  | 
487  |  |  | 
488  |  | static int chacha20_poly1305_cleanup(EVP_CIPHER_CTX *ctx)  | 
489  | 0  | { | 
490  | 0  |     EVP_CHACHA_AEAD_CTX *actx = aead_data(ctx);  | 
491  | 0  |     if (actx)  | 
492  | 0  |         OPENSSL_cleanse(ctx->cipher_data, sizeof(*actx) + Poly1305_ctx_size());  | 
493  | 0  |     return 1;  | 
494  | 0  | }  | 
495  |  |  | 
496  |  | static int chacha20_poly1305_ctrl(EVP_CIPHER_CTX *ctx, int type, int arg,  | 
497  |  |                                   void *ptr)  | 
498  | 0  | { | 
499  | 0  |     EVP_CHACHA_AEAD_CTX *actx = aead_data(ctx);  | 
500  |  | 
  | 
501  | 0  |     switch (type) { | 
502  | 0  |     case EVP_CTRL_INIT:  | 
503  | 0  |         if (actx == NULL)  | 
504  | 0  |             actx = ctx->cipher_data  | 
505  | 0  |                  = OPENSSL_zalloc(sizeof(*actx) + Poly1305_ctx_size());  | 
506  | 0  |         if (actx == NULL) { | 
507  | 0  |             ERR_raise(ERR_LIB_EVP, EVP_R_INITIALIZATION_ERROR);  | 
508  | 0  |             return 0;  | 
509  | 0  |         }  | 
510  | 0  |         actx->len.aad = 0;  | 
511  | 0  |         actx->len.text = 0;  | 
512  | 0  |         actx->aad = 0;  | 
513  | 0  |         actx->mac_inited = 0;  | 
514  | 0  |         actx->tag_len = 0;  | 
515  | 0  |         actx->nonce_len = 12;  | 
516  | 0  |         actx->tls_payload_length = NO_TLS_PAYLOAD_LENGTH;  | 
517  | 0  |         memset(actx->tls_aad, 0, POLY1305_BLOCK_SIZE);  | 
518  | 0  |         return 1;  | 
519  |  |  | 
520  | 0  |     case EVP_CTRL_COPY:  | 
521  | 0  |         if (actx) { | 
522  | 0  |             EVP_CIPHER_CTX *dst = (EVP_CIPHER_CTX *)ptr;  | 
523  |  | 
  | 
524  | 0  |             dst->cipher_data =  | 
525  | 0  |                    OPENSSL_memdup(actx, sizeof(*actx) + Poly1305_ctx_size());  | 
526  | 0  |             if (dst->cipher_data == NULL) { | 
527  | 0  |                 ERR_raise(ERR_LIB_EVP, EVP_R_COPY_ERROR);  | 
528  | 0  |                 return 0;  | 
529  | 0  |             }  | 
530  | 0  |         }  | 
531  | 0  |         return 1;  | 
532  |  |  | 
533  | 0  |     case EVP_CTRL_GET_IVLEN:  | 
534  | 0  |         *(int *)ptr = actx->nonce_len;  | 
535  | 0  |         return 1;  | 
536  |  |  | 
537  | 0  |     case EVP_CTRL_AEAD_SET_IVLEN:  | 
538  | 0  |         if (arg <= 0 || arg > CHACHA20_POLY1305_MAX_IVLEN)  | 
539  | 0  |             return 0;  | 
540  | 0  |         actx->nonce_len = arg;  | 
541  | 0  |         return 1;  | 
542  |  |  | 
543  | 0  |     case EVP_CTRL_AEAD_SET_IV_FIXED:  | 
544  | 0  |         if (arg != 12)  | 
545  | 0  |             return 0;  | 
546  | 0  |         actx->nonce[0] = actx->key.counter[1]  | 
547  | 0  |                        = CHACHA_U8TOU32((unsigned char *)ptr);  | 
548  | 0  |         actx->nonce[1] = actx->key.counter[2]  | 
549  | 0  |                        = CHACHA_U8TOU32((unsigned char *)ptr+4);  | 
550  | 0  |         actx->nonce[2] = actx->key.counter[3]  | 
551  | 0  |                        = CHACHA_U8TOU32((unsigned char *)ptr+8);  | 
552  | 0  |         return 1;  | 
553  |  |  | 
554  | 0  |     case EVP_CTRL_AEAD_SET_TAG:  | 
555  | 0  |         if (arg <= 0 || arg > POLY1305_BLOCK_SIZE)  | 
556  | 0  |             return 0;  | 
557  | 0  |         if (ptr != NULL) { | 
558  | 0  |             memcpy(actx->tag, ptr, arg);  | 
559  | 0  |             actx->tag_len = arg;  | 
560  | 0  |         }  | 
561  | 0  |         return 1;  | 
562  |  |  | 
563  | 0  |     case EVP_CTRL_AEAD_GET_TAG:  | 
564  | 0  |         if (arg <= 0 || arg > POLY1305_BLOCK_SIZE ||  | 
565  | 0  |                 !EVP_CIPHER_CTX_is_encrypting(ctx))  | 
566  | 0  |             return 0;  | 
567  | 0  |         memcpy(ptr, actx->tag, arg);  | 
568  | 0  |         return 1;  | 
569  |  |  | 
570  | 0  |     case EVP_CTRL_AEAD_TLS1_AAD:  | 
571  | 0  |         if (arg != EVP_AEAD_TLS1_AAD_LEN)  | 
572  | 0  |             return 0;  | 
573  | 0  |         { | 
574  | 0  |             unsigned int len;  | 
575  | 0  |             unsigned char *aad = ptr;  | 
576  |  | 
  | 
577  | 0  |             memcpy(actx->tls_aad, ptr, EVP_AEAD_TLS1_AAD_LEN);  | 
578  | 0  |             len = aad[EVP_AEAD_TLS1_AAD_LEN - 2] << 8 |  | 
579  | 0  |                   aad[EVP_AEAD_TLS1_AAD_LEN - 1];  | 
580  | 0  |             aad = actx->tls_aad;  | 
581  | 0  |             if (!EVP_CIPHER_CTX_is_encrypting(ctx)) { | 
582  | 0  |                 if (len < POLY1305_BLOCK_SIZE)  | 
583  | 0  |                     return 0;  | 
584  | 0  |                 len -= POLY1305_BLOCK_SIZE;     /* discount attached tag */  | 
585  | 0  |                 aad[EVP_AEAD_TLS1_AAD_LEN - 2] = (unsigned char)(len >> 8);  | 
586  | 0  |                 aad[EVP_AEAD_TLS1_AAD_LEN - 1] = (unsigned char)len;  | 
587  | 0  |             }  | 
588  | 0  |             actx->tls_payload_length = len;  | 
589  |  |  | 
590  |  |             /*  | 
591  |  |              * merge record sequence number as per RFC7905  | 
592  |  |              */  | 
593  | 0  |             actx->key.counter[1] = actx->nonce[0];  | 
594  | 0  |             actx->key.counter[2] = actx->nonce[1] ^ CHACHA_U8TOU32(aad);  | 
595  | 0  |             actx->key.counter[3] = actx->nonce[2] ^ CHACHA_U8TOU32(aad+4);  | 
596  | 0  |             actx->mac_inited = 0;  | 
597  |  | 
  | 
598  | 0  |             return POLY1305_BLOCK_SIZE;         /* tag length */  | 
599  | 0  |         }  | 
600  |  |  | 
601  | 0  |     case EVP_CTRL_AEAD_SET_MAC_KEY:  | 
602  |  |         /* no-op */  | 
603  | 0  |         return 1;  | 
604  |  |  | 
605  | 0  |     default:  | 
606  | 0  |         return -1;  | 
607  | 0  |     }  | 
608  | 0  | }  | 
609  |  |  | 
610  |  | static EVP_CIPHER chacha20_poly1305 = { | 
611  |  |     NID_chacha20_poly1305,  | 
612  |  |     1,                  /* block_size */  | 
613  |  |     CHACHA_KEY_SIZE,    /* key_len */  | 
614  |  |     12,                 /* iv_len, 96-bit nonce in the context */  | 
615  |  |     EVP_CIPH_FLAG_AEAD_CIPHER | EVP_CIPH_CUSTOM_IV |  | 
616  |  |     EVP_CIPH_ALWAYS_CALL_INIT | EVP_CIPH_CTRL_INIT |  | 
617  |  |     EVP_CIPH_CUSTOM_COPY | EVP_CIPH_FLAG_CUSTOM_CIPHER |  | 
618  |  |     EVP_CIPH_CUSTOM_IV_LENGTH,  | 
619  |  |     EVP_ORIG_GLOBAL,  | 
620  |  |     chacha20_poly1305_init_key,  | 
621  |  |     chacha20_poly1305_cipher,  | 
622  |  |     chacha20_poly1305_cleanup,  | 
623  |  |     0,          /* 0 moves context-specific structure allocation to ctrl */  | 
624  |  |     NULL,       /* set_asn1_parameters */  | 
625  |  |     NULL,       /* get_asn1_parameters */  | 
626  |  |     chacha20_poly1305_ctrl,  | 
627  |  |     NULL        /* app_data */  | 
628  |  | };  | 
629  |  |  | 
630  |  | const EVP_CIPHER *EVP_chacha20_poly1305(void)  | 
631  | 3  | { | 
632  | 3  |     return(&chacha20_poly1305);  | 
633  | 3  | }  | 
634  |  | # endif  | 
635  |  | #endif  |