c906e6ff8773c2ad8d500df92aeeedcbc91e49a8
[openssl.git] / crypto / evp / e_aes.c
1 /* ====================================================================
2  * Copyright (c) 2001-2014 The OpenSSL Project.  All rights reserved.
3  *
4  * Redistribution and use in source and binary forms, with or without
5  * modification, are permitted provided that the following conditions
6  * are met:
7  *
8  * 1. Redistributions of source code must retain the above copyright
9  *    notice, this list of conditions and the following disclaimer.
10  *
11  * 2. Redistributions in binary form must reproduce the above copyright
12  *    notice, this list of conditions and the following disclaimer in
13  *    the documentation and/or other materials provided with the
14  *    distribution.
15  *
16  * 3. All advertising materials mentioning features or use of this
17  *    software must display the following acknowledgment:
18  *    "This product includes software developed by the OpenSSL Project
19  *    for use in the OpenSSL Toolkit. (http://www.openssl.org/)"
20  *
21  * 4. The names "OpenSSL Toolkit" and "OpenSSL Project" must not be used to
22  *    endorse or promote products derived from this software without
23  *    prior written permission. For written permission, please contact
24  *    openssl-core@openssl.org.
25  *
26  * 5. Products derived from this software may not be called "OpenSSL"
27  *    nor may "OpenSSL" appear in their names without prior written
28  *    permission of the OpenSSL Project.
29  *
30  * 6. Redistributions of any form whatsoever must retain the following
31  *    acknowledgment:
32  *    "This product includes software developed by the OpenSSL Project
33  *    for use in the OpenSSL Toolkit (http://www.openssl.org/)"
34  *
35  * THIS SOFTWARE IS PROVIDED BY THE OpenSSL PROJECT ``AS IS'' AND ANY
36  * EXPRESSED OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
37  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
38  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE OpenSSL PROJECT OR
39  * ITS CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
40  * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
41  * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
42  * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
43  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT,
44  * STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
45  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED
46  * OF THE POSSIBILITY OF SUCH DAMAGE.
47  * ====================================================================
48  *
49  */
50
51 #include <openssl/opensslconf.h>
52 #ifndef OPENSSL_NO_AES
53 # include <openssl/crypto.h>
54 # include <openssl/evp.h>
55 # include <openssl/err.h>
56 # include <string.h>
57 # include <assert.h>
58 # include <openssl/aes.h>
59 # include "internal/evp_int.h"
60 # include "modes_lcl.h"
61 # include <openssl/rand.h>
62
63 typedef struct {
64     union {
65         double align;
66         AES_KEY ks;
67     } ks;
68     block128_f block;
69     union {
70         cbc128_f cbc;
71         ctr128_f ctr;
72     } stream;
73 } EVP_AES_KEY;
74
75 typedef struct {
76     union {
77         double align;
78         AES_KEY ks;
79     } ks;                       /* AES key schedule to use */
80     int key_set;                /* Set if key initialised */
81     int iv_set;                 /* Set if an iv is set */
82     GCM128_CONTEXT gcm;
83     unsigned char *iv;          /* Temporary IV store */
84     int ivlen;                  /* IV length */
85     int taglen;
86     int iv_gen;                 /* It is OK to generate IVs */
87     int tls_aad_len;            /* TLS AAD length */
88     ctr128_f ctr;
89 } EVP_AES_GCM_CTX;
90
91 typedef struct {
92     union {
93         double align;
94         AES_KEY ks;
95     } ks1, ks2;                 /* AES key schedules to use */
96     XTS128_CONTEXT xts;
97     void (*stream) (const unsigned char *in,
98                     unsigned char *out, size_t length,
99                     const AES_KEY *key1, const AES_KEY *key2,
100                     const unsigned char iv[16]);
101 } EVP_AES_XTS_CTX;
102
103 typedef struct {
104     union {
105         double align;
106         AES_KEY ks;
107     } ks;                       /* AES key schedule to use */
108     int key_set;                /* Set if key initialised */
109     int iv_set;                 /* Set if an iv is set */
110     int tag_set;                /* Set if tag is valid */
111     int len_set;                /* Set if message length set */
112     int L, M;                   /* L and M parameters from RFC3610 */
113     int tls_aad_len;            /* TLS AAD length */
114     CCM128_CONTEXT ccm;
115     ccm128_f str;
116 } EVP_AES_CCM_CTX;
117
118 # ifndef OPENSSL_NO_OCB
119 typedef struct {
120     union {
121         double align;
122         AES_KEY ks;
123     } ksenc;                    /* AES key schedule to use for encryption */
124     union {
125         double align;
126         AES_KEY ks;
127     } ksdec;                    /* AES key schedule to use for decryption */
128     int key_set;                /* Set if key initialised */
129     int iv_set;                 /* Set if an iv is set */
130     OCB128_CONTEXT ocb;
131     unsigned char *iv;          /* Temporary IV store */
132     unsigned char tag[16];
133     unsigned char data_buf[16]; /* Store partial data blocks */
134     unsigned char aad_buf[16];  /* Store partial AAD blocks */
135     int data_buf_len;
136     int aad_buf_len;
137     int ivlen;                  /* IV length */
138     int taglen;
139 } EVP_AES_OCB_CTX;
140 # endif
141
142 # define MAXBITCHUNK     ((size_t)1<<(sizeof(size_t)*8-4))
143
144 # ifdef VPAES_ASM
145 int vpaes_set_encrypt_key(const unsigned char *userKey, int bits,
146                           AES_KEY *key);
147 int vpaes_set_decrypt_key(const unsigned char *userKey, int bits,
148                           AES_KEY *key);
149
150 void vpaes_encrypt(const unsigned char *in, unsigned char *out,
151                    const AES_KEY *key);
152 void vpaes_decrypt(const unsigned char *in, unsigned char *out,
153                    const AES_KEY *key);
154
155 void vpaes_cbc_encrypt(const unsigned char *in,
156                        unsigned char *out,
157                        size_t length,
158                        const AES_KEY *key, unsigned char *ivec, int enc);
159 # endif
160 # ifdef BSAES_ASM
161 void bsaes_cbc_encrypt(const unsigned char *in, unsigned char *out,
162                        size_t length, const AES_KEY *key,
163                        unsigned char ivec[16], int enc);
164 void bsaes_ctr32_encrypt_blocks(const unsigned char *in, unsigned char *out,
165                                 size_t len, const AES_KEY *key,
166                                 const unsigned char ivec[16]);
167 void bsaes_xts_encrypt(const unsigned char *inp, unsigned char *out,
168                        size_t len, const AES_KEY *key1,
169                        const AES_KEY *key2, const unsigned char iv[16]);
170 void bsaes_xts_decrypt(const unsigned char *inp, unsigned char *out,
171                        size_t len, const AES_KEY *key1,
172                        const AES_KEY *key2, const unsigned char iv[16]);
173 # endif
174 # ifdef AES_CTR_ASM
175 void AES_ctr32_encrypt(const unsigned char *in, unsigned char *out,
176                        size_t blocks, const AES_KEY *key,
177                        const unsigned char ivec[AES_BLOCK_SIZE]);
178 # endif
179 # ifdef AES_XTS_ASM
180 void AES_xts_encrypt(const char *inp, char *out, size_t len,
181                      const AES_KEY *key1, const AES_KEY *key2,
182                      const unsigned char iv[16]);
183 void AES_xts_decrypt(const char *inp, char *out, size_t len,
184                      const AES_KEY *key1, const AES_KEY *key2,
185                      const unsigned char iv[16]);
186 # endif
187
188 # if     defined(OPENSSL_CPUID_OBJ) && (defined(__powerpc__) || defined(__ppc__) || defined(_ARCH_PPC))
189 #  include "ppc_arch.h"
190 #  ifdef VPAES_ASM
191 #   define VPAES_CAPABLE (OPENSSL_ppccap_P & PPC_ALTIVEC)
192 #  endif
193 #  define HWAES_CAPABLE  (OPENSSL_ppccap_P & PPC_CRYPTO207)
194 #  define HWAES_set_encrypt_key aes_p8_set_encrypt_key
195 #  define HWAES_set_decrypt_key aes_p8_set_decrypt_key
196 #  define HWAES_encrypt aes_p8_encrypt
197 #  define HWAES_decrypt aes_p8_decrypt
198 #  define HWAES_cbc_encrypt aes_p8_cbc_encrypt
199 #  define HWAES_ctr32_encrypt_blocks aes_p8_ctr32_encrypt_blocks
200 # endif
201
202 # if     defined(AES_ASM) && !defined(I386_ONLY) &&      (  \
203         ((defined(__i386)       || defined(__i386__)    || \
204           defined(_M_IX86)) && defined(OPENSSL_IA32_SSE2))|| \
205         defined(__x86_64)       || defined(__x86_64__)  || \
206         defined(_M_AMD64)       || defined(_M_X64)      || \
207         defined(__INTEL__)                              )
208
209 extern unsigned int OPENSSL_ia32cap_P[];
210
211 #  ifdef VPAES_ASM
212 #   define VPAES_CAPABLE   (OPENSSL_ia32cap_P[1]&(1<<(41-32)))
213 #  endif
214 #  ifdef BSAES_ASM
215 #   define BSAES_CAPABLE   (OPENSSL_ia32cap_P[1]&(1<<(41-32)))
216 #  endif
217 /*
218  * AES-NI section
219  */
220 #  define AESNI_CAPABLE   (OPENSSL_ia32cap_P[1]&(1<<(57-32)))
221
222 int aesni_set_encrypt_key(const unsigned char *userKey, int bits,
223                           AES_KEY *key);
224 int aesni_set_decrypt_key(const unsigned char *userKey, int bits,
225                           AES_KEY *key);
226
227 void aesni_encrypt(const unsigned char *in, unsigned char *out,
228                    const AES_KEY *key);
229 void aesni_decrypt(const unsigned char *in, unsigned char *out,
230                    const AES_KEY *key);
231
232 void aesni_ecb_encrypt(const unsigned char *in,
233                        unsigned char *out,
234                        size_t length, const AES_KEY *key, int enc);
235 void aesni_cbc_encrypt(const unsigned char *in,
236                        unsigned char *out,
237                        size_t length,
238                        const AES_KEY *key, unsigned char *ivec, int enc);
239
240 void aesni_ctr32_encrypt_blocks(const unsigned char *in,
241                                 unsigned char *out,
242                                 size_t blocks,
243                                 const void *key, const unsigned char *ivec);
244
245 void aesni_xts_encrypt(const unsigned char *in,
246                        unsigned char *out,
247                        size_t length,
248                        const AES_KEY *key1, const AES_KEY *key2,
249                        const unsigned char iv[16]);
250
251 void aesni_xts_decrypt(const unsigned char *in,
252                        unsigned char *out,
253                        size_t length,
254                        const AES_KEY *key1, const AES_KEY *key2,
255                        const unsigned char iv[16]);
256
257 void aesni_ccm64_encrypt_blocks(const unsigned char *in,
258                                 unsigned char *out,
259                                 size_t blocks,
260                                 const void *key,
261                                 const unsigned char ivec[16],
262                                 unsigned char cmac[16]);
263
264 void aesni_ccm64_decrypt_blocks(const unsigned char *in,
265                                 unsigned char *out,
266                                 size_t blocks,
267                                 const void *key,
268                                 const unsigned char ivec[16],
269                                 unsigned char cmac[16]);
270
271 #  if defined(__x86_64) || defined(__x86_64__) || defined(_M_AMD64) || defined(_M_X64)
272 size_t aesni_gcm_encrypt(const unsigned char *in,
273                          unsigned char *out,
274                          size_t len,
275                          const void *key, unsigned char ivec[16], u64 *Xi);
276 #   define AES_gcm_encrypt aesni_gcm_encrypt
277 size_t aesni_gcm_decrypt(const unsigned char *in,
278                          unsigned char *out,
279                          size_t len,
280                          const void *key, unsigned char ivec[16], u64 *Xi);
281 #   define AES_gcm_decrypt aesni_gcm_decrypt
282 void gcm_ghash_avx(u64 Xi[2], const u128 Htable[16], const u8 *in,
283                    size_t len);
284 #   define AES_GCM_ASM(gctx)       (gctx->ctr==aesni_ctr32_encrypt_blocks && \
285                                  gctx->gcm.ghash==gcm_ghash_avx)
286 #   define AES_GCM_ASM2(gctx)      (gctx->gcm.block==(block128_f)aesni_encrypt && \
287                                  gctx->gcm.ghash==gcm_ghash_avx)
288 #   undef AES_GCM_ASM2          /* minor size optimization */
289 #  endif
290
291 static int aesni_init_key(EVP_CIPHER_CTX *ctx, const unsigned char *key,
292                           const unsigned char *iv, int enc)
293 {
294     int ret, mode;
295     EVP_AES_KEY *dat = EVP_C_DATA(EVP_AES_KEY,ctx);
296
297     mode = EVP_CIPHER_CTX_mode(ctx);
298     if ((mode == EVP_CIPH_ECB_MODE || mode == EVP_CIPH_CBC_MODE)
299         && !enc) {
300         ret = aesni_set_decrypt_key(key, EVP_CIPHER_CTX_key_length(ctx) * 8,
301                                     &dat->ks.ks);
302         dat->block = (block128_f) aesni_decrypt;
303         dat->stream.cbc = mode == EVP_CIPH_CBC_MODE ?
304             (cbc128_f) aesni_cbc_encrypt : NULL;
305     } else {
306         ret = aesni_set_encrypt_key(key, EVP_CIPHER_CTX_key_length(ctx) * 8,
307                                     &dat->ks.ks);
308         dat->block = (block128_f) aesni_encrypt;
309         if (mode == EVP_CIPH_CBC_MODE)
310             dat->stream.cbc = (cbc128_f) aesni_cbc_encrypt;
311         else if (mode == EVP_CIPH_CTR_MODE)
312             dat->stream.ctr = (ctr128_f) aesni_ctr32_encrypt_blocks;
313         else
314             dat->stream.cbc = NULL;
315     }
316
317     if (ret < 0) {
318         EVPerr(EVP_F_AESNI_INIT_KEY, EVP_R_AES_KEY_SETUP_FAILED);
319         return 0;
320     }
321
322     return 1;
323 }
324
325 static int aesni_cbc_cipher(EVP_CIPHER_CTX *ctx, unsigned char *out,
326                             const unsigned char *in, size_t len)
327 {
328     aesni_cbc_encrypt(in, out, len, &EVP_C_DATA(EVP_AES_KEY,ctx)->ks.ks,
329                       EVP_CIPHER_CTX_iv_noconst(ctx),
330                       EVP_CIPHER_CTX_encrypting(ctx));
331
332     return 1;
333 }
334
335 static int aesni_ecb_cipher(EVP_CIPHER_CTX *ctx, unsigned char *out,
336                             const unsigned char *in, size_t len)
337 {
338     size_t bl = EVP_CIPHER_CTX_block_size(ctx);
339
340     if (len < bl)
341         return 1;
342
343     aesni_ecb_encrypt(in, out, len, &EVP_C_DATA(EVP_AES_KEY,ctx)->ks.ks,
344                       EVP_CIPHER_CTX_encrypting(ctx));
345
346     return 1;
347 }
348
349 #  define aesni_ofb_cipher aes_ofb_cipher
350 static int aesni_ofb_cipher(EVP_CIPHER_CTX *ctx, unsigned char *out,
351                             const unsigned char *in, size_t len);
352
353 #  define aesni_cfb_cipher aes_cfb_cipher
354 static int aesni_cfb_cipher(EVP_CIPHER_CTX *ctx, unsigned char *out,
355                             const unsigned char *in, size_t len);
356
357 #  define aesni_cfb8_cipher aes_cfb8_cipher
358 static int aesni_cfb8_cipher(EVP_CIPHER_CTX *ctx, unsigned char *out,
359                              const unsigned char *in, size_t len);
360
361 #  define aesni_cfb1_cipher aes_cfb1_cipher
362 static int aesni_cfb1_cipher(EVP_CIPHER_CTX *ctx, unsigned char *out,
363                              const unsigned char *in, size_t len);
364
365 #  define aesni_ctr_cipher aes_ctr_cipher
366 static int aesni_ctr_cipher(EVP_CIPHER_CTX *ctx, unsigned char *out,
367                             const unsigned char *in, size_t len);
368
369 static int aesni_gcm_init_key(EVP_CIPHER_CTX *ctx, const unsigned char *key,
370                               const unsigned char *iv, int enc)
371 {
372     EVP_AES_GCM_CTX *gctx = EVP_C_DATA(EVP_AES_GCM_CTX,ctx);
373     if (!iv && !key)
374         return 1;
375     if (key) {
376         aesni_set_encrypt_key(key, EVP_CIPHER_CTX_key_length(ctx) * 8,
377                               &gctx->ks.ks);
378         CRYPTO_gcm128_init(&gctx->gcm, &gctx->ks, (block128_f) aesni_encrypt);
379         gctx->ctr = (ctr128_f) aesni_ctr32_encrypt_blocks;
380         /*
381          * If we have an iv can set it directly, otherwise use saved IV.
382          */
383         if (iv == NULL && gctx->iv_set)
384             iv = gctx->iv;
385         if (iv) {
386             CRYPTO_gcm128_setiv(&gctx->gcm, iv, gctx->ivlen);
387             gctx->iv_set = 1;
388         }
389         gctx->key_set = 1;
390     } else {
391         /* If key set use IV, otherwise copy */
392         if (gctx->key_set)
393             CRYPTO_gcm128_setiv(&gctx->gcm, iv, gctx->ivlen);
394         else
395             memcpy(gctx->iv, iv, gctx->ivlen);
396         gctx->iv_set = 1;
397         gctx->iv_gen = 0;
398     }
399     return 1;
400 }
401
402 #  define aesni_gcm_cipher aes_gcm_cipher
403 static int aesni_gcm_cipher(EVP_CIPHER_CTX *ctx, unsigned char *out,
404                             const unsigned char *in, size_t len);
405
406 static int aesni_xts_init_key(EVP_CIPHER_CTX *ctx, const unsigned char *key,
407                               const unsigned char *iv, int enc)
408 {
409     EVP_AES_XTS_CTX *xctx = EVP_C_DATA(EVP_AES_XTS_CTX,ctx);
410     if (!iv && !key)
411         return 1;
412
413     if (key) {
414         /* key_len is two AES keys */
415         if (enc) {
416             aesni_set_encrypt_key(key, EVP_CIPHER_CTX_key_length(ctx) * 4,
417                                   &xctx->ks1.ks);
418             xctx->xts.block1 = (block128_f) aesni_encrypt;
419             xctx->stream = aesni_xts_encrypt;
420         } else {
421             aesni_set_decrypt_key(key, EVP_CIPHER_CTX_key_length(ctx) * 4,
422                                   &xctx->ks1.ks);
423             xctx->xts.block1 = (block128_f) aesni_decrypt;
424             xctx->stream = aesni_xts_decrypt;
425         }
426
427         aesni_set_encrypt_key(key + EVP_CIPHER_CTX_key_length(ctx) / 2,
428                               EVP_CIPHER_CTX_key_length(ctx) * 4,
429                               &xctx->ks2.ks);
430         xctx->xts.block2 = (block128_f) aesni_encrypt;
431
432         xctx->xts.key1 = &xctx->ks1;
433     }
434
435     if (iv) {
436         xctx->xts.key2 = &xctx->ks2;
437         memcpy(EVP_CIPHER_CTX_iv_noconst(ctx), iv, 16);
438     }
439
440     return 1;
441 }
442
443 #  define aesni_xts_cipher aes_xts_cipher
444 static int aesni_xts_cipher(EVP_CIPHER_CTX *ctx, unsigned char *out,
445                             const unsigned char *in, size_t len);
446
447 static int aesni_ccm_init_key(EVP_CIPHER_CTX *ctx, const unsigned char *key,
448                               const unsigned char *iv, int enc)
449 {
450     EVP_AES_CCM_CTX *cctx = EVP_C_DATA(EVP_AES_CCM_CTX,ctx);
451     if (!iv && !key)
452         return 1;
453     if (key) {
454         aesni_set_encrypt_key(key, EVP_CIPHER_CTX_key_length(ctx) * 8,
455                               &cctx->ks.ks);
456         CRYPTO_ccm128_init(&cctx->ccm, cctx->M, cctx->L,
457                            &cctx->ks, (block128_f) aesni_encrypt);
458         cctx->str = enc ? (ccm128_f) aesni_ccm64_encrypt_blocks :
459             (ccm128_f) aesni_ccm64_decrypt_blocks;
460         cctx->key_set = 1;
461     }
462     if (iv) {
463         memcpy(EVP_CIPHER_CTX_iv_noconst(ctx), iv, 15 - cctx->L);
464         cctx->iv_set = 1;
465     }
466     return 1;
467 }
468
469 #  define aesni_ccm_cipher aes_ccm_cipher
470 static int aesni_ccm_cipher(EVP_CIPHER_CTX *ctx, unsigned char *out,
471                             const unsigned char *in, size_t len);
472
473 #  ifndef OPENSSL_NO_OCB
474 void aesni_ocb_encrypt(const unsigned char *in, unsigned char *out,
475                        size_t blocks, const void *key,
476                        size_t start_block_num,
477                        unsigned char offset_i[16],
478                        const unsigned char L_[][16],
479                        unsigned char checksum[16]);
480 void aesni_ocb_decrypt(const unsigned char *in, unsigned char *out,
481                        size_t blocks, const void *key,
482                        size_t start_block_num,
483                        unsigned char offset_i[16],
484                        const unsigned char L_[][16],
485                        unsigned char checksum[16]);
486
487 static int aesni_ocb_init_key(EVP_CIPHER_CTX *ctx, const unsigned char *key,
488                               const unsigned char *iv, int enc)
489 {
490     EVP_AES_OCB_CTX *octx = EVP_C_DATA(EVP_AES_OCB_CTX,ctx);
491     if (!iv && !key)
492         return 1;
493     if (key) {
494         do {
495             /*
496              * We set both the encrypt and decrypt key here because decrypt
497              * needs both. We could possibly optimise to remove setting the
498              * decrypt for an encryption operation.
499              */
500             aesni_set_encrypt_key(key, EVP_CIPHER_CTX_key_length(ctx) * 8,
501                                   &octx->ksenc.ks);
502             aesni_set_decrypt_key(key, EVP_CIPHER_CTX_key_length(ctx) * 8,
503                                   &octx->ksdec.ks);
504             if (!CRYPTO_ocb128_init(&octx->ocb,
505                                     &octx->ksenc.ks, &octx->ksdec.ks,
506                                     (block128_f) aesni_encrypt,
507                                     (block128_f) aesni_decrypt,
508                                     enc ? aesni_ocb_encrypt
509                                         : aesni_ocb_decrypt))
510                 return 0;
511         }
512         while (0);
513
514         /*
515          * If we have an iv we can set it directly, otherwise use saved IV.
516          */
517         if (iv == NULL && octx->iv_set)
518             iv = octx->iv;
519         if (iv) {
520             if (CRYPTO_ocb128_setiv(&octx->ocb, iv, octx->ivlen, octx->taglen)
521                 != 1)
522                 return 0;
523             octx->iv_set = 1;
524         }
525         octx->key_set = 1;
526     } else {
527         /* If key set use IV, otherwise copy */
528         if (octx->key_set)
529             CRYPTO_ocb128_setiv(&octx->ocb, iv, octx->ivlen, octx->taglen);
530         else
531             memcpy(octx->iv, iv, octx->ivlen);
532         octx->iv_set = 1;
533     }
534     return 1;
535 }
536
537 #   define aesni_ocb_cipher aes_ocb_cipher
538 static int aesni_ocb_cipher(EVP_CIPHER_CTX *ctx, unsigned char *out,
539                             const unsigned char *in, size_t len);
540 #  endif                        /* OPENSSL_NO_OCB */
541
542 #  define BLOCK_CIPHER_generic(nid,keylen,blocksize,ivlen,nmode,mode,MODE,flags) \
543 static const EVP_CIPHER aesni_##keylen##_##mode = { \
544         nid##_##keylen##_##nmode,blocksize,keylen/8,ivlen, \
545         flags|EVP_CIPH_##MODE##_MODE,   \
546         aesni_init_key,                 \
547         aesni_##mode##_cipher,          \
548         NULL,                           \
549         sizeof(EVP_AES_KEY),            \
550         NULL,NULL,NULL,NULL }; \
551 static const EVP_CIPHER aes_##keylen##_##mode = { \
552         nid##_##keylen##_##nmode,blocksize,     \
553         keylen/8,ivlen, \
554         flags|EVP_CIPH_##MODE##_MODE,   \
555         aes_init_key,                   \
556         aes_##mode##_cipher,            \
557         NULL,                           \
558         sizeof(EVP_AES_KEY),            \
559         NULL,NULL,NULL,NULL }; \
560 const EVP_CIPHER *EVP_aes_##keylen##_##mode(void) \
561 { return AESNI_CAPABLE?&aesni_##keylen##_##mode:&aes_##keylen##_##mode; }
562
563 #  define BLOCK_CIPHER_custom(nid,keylen,blocksize,ivlen,mode,MODE,flags) \
564 static const EVP_CIPHER aesni_##keylen##_##mode = { \
565         nid##_##keylen##_##mode,blocksize, \
566         (EVP_CIPH_##MODE##_MODE==EVP_CIPH_XTS_MODE?2:1)*keylen/8, ivlen, \
567         flags|EVP_CIPH_##MODE##_MODE,   \
568         aesni_##mode##_init_key,        \
569         aesni_##mode##_cipher,          \
570         aes_##mode##_cleanup,           \
571         sizeof(EVP_AES_##MODE##_CTX),   \
572         NULL,NULL,aes_##mode##_ctrl,NULL }; \
573 static const EVP_CIPHER aes_##keylen##_##mode = { \
574         nid##_##keylen##_##mode,blocksize, \
575         (EVP_CIPH_##MODE##_MODE==EVP_CIPH_XTS_MODE?2:1)*keylen/8, ivlen, \
576         flags|EVP_CIPH_##MODE##_MODE,   \
577         aes_##mode##_init_key,          \
578         aes_##mode##_cipher,            \
579         aes_##mode##_cleanup,           \
580         sizeof(EVP_AES_##MODE##_CTX),   \
581         NULL,NULL,aes_##mode##_ctrl,NULL }; \
582 const EVP_CIPHER *EVP_aes_##keylen##_##mode(void) \
583 { return AESNI_CAPABLE?&aesni_##keylen##_##mode:&aes_##keylen##_##mode; }
584
585 # elif   defined(AES_ASM) && (defined(__sparc) || defined(__sparc__))
586
587 #  include "sparc_arch.h"
588
589 extern unsigned int OPENSSL_sparcv9cap_P[];
590
591 #  define SPARC_AES_CAPABLE       (OPENSSL_sparcv9cap_P[1] & CFR_AES)
592
593 void aes_t4_set_encrypt_key(const unsigned char *key, int bits, AES_KEY *ks);
594 void aes_t4_set_decrypt_key(const unsigned char *key, int bits, AES_KEY *ks);
595 void aes_t4_encrypt(const unsigned char *in, unsigned char *out,
596                     const AES_KEY *key);
597 void aes_t4_decrypt(const unsigned char *in, unsigned char *out,
598                     const AES_KEY *key);
599 /*
600  * Key-length specific subroutines were chosen for following reason.
601  * Each SPARC T4 core can execute up to 8 threads which share core's
602  * resources. Loading as much key material to registers allows to
603  * minimize references to shared memory interface, as well as amount
604  * of instructions in inner loops [much needed on T4]. But then having
605  * non-key-length specific routines would require conditional branches
606  * either in inner loops or on subroutines' entries. Former is hardly
607  * acceptable, while latter means code size increase to size occupied
608  * by multiple key-length specific subroutines, so why fight?
609  */
610 void aes128_t4_cbc_encrypt(const unsigned char *in, unsigned char *out,
611                            size_t len, const AES_KEY *key,
612                            unsigned char *ivec);
613 void aes128_t4_cbc_decrypt(const unsigned char *in, unsigned char *out,
614                            size_t len, const AES_KEY *key,
615                            unsigned char *ivec);
616 void aes192_t4_cbc_encrypt(const unsigned char *in, unsigned char *out,
617                            size_t len, const AES_KEY *key,
618                            unsigned char *ivec);
619 void aes192_t4_cbc_decrypt(const unsigned char *in, unsigned char *out,
620                            size_t len, const AES_KEY *key,
621                            unsigned char *ivec);
622 void aes256_t4_cbc_encrypt(const unsigned char *in, unsigned char *out,
623                            size_t len, const AES_KEY *key,
624                            unsigned char *ivec);
625 void aes256_t4_cbc_decrypt(const unsigned char *in, unsigned char *out,
626                            size_t len, const AES_KEY *key,
627                            unsigned char *ivec);
628 void aes128_t4_ctr32_encrypt(const unsigned char *in, unsigned char *out,
629                              size_t blocks, const AES_KEY *key,
630                              unsigned char *ivec);
631 void aes192_t4_ctr32_encrypt(const unsigned char *in, unsigned char *out,
632                              size_t blocks, const AES_KEY *key,
633                              unsigned char *ivec);
634 void aes256_t4_ctr32_encrypt(const unsigned char *in, unsigned char *out,
635                              size_t blocks, const AES_KEY *key,
636                              unsigned char *ivec);
637 void aes128_t4_xts_encrypt(const unsigned char *in, unsigned char *out,
638                            size_t blocks, const AES_KEY *key1,
639                            const AES_KEY *key2, const unsigned char *ivec);
640 void aes128_t4_xts_decrypt(const unsigned char *in, unsigned char *out,
641                            size_t blocks, const AES_KEY *key1,
642                            const AES_KEY *key2, const unsigned char *ivec);
643 void aes256_t4_xts_encrypt(const unsigned char *in, unsigned char *out,
644                            size_t blocks, const AES_KEY *key1,
645                            const AES_KEY *key2, const unsigned char *ivec);
646 void aes256_t4_xts_decrypt(const unsigned char *in, unsigned char *out,
647                            size_t blocks, const AES_KEY *key1,
648                            const AES_KEY *key2, const unsigned char *ivec);
649
650 static int aes_t4_init_key(EVP_CIPHER_CTX *ctx, const unsigned char *key,
651                            const unsigned char *iv, int enc)
652 {
653     int ret, mode, bits;
654     EVP_AES_KEY *dat = EVP_C_DATA(EVP_AES_KEY,ctx);
655
656     mode = EVP_CIPHER_CTX_mode(ctx);
657     bits = EVP_CIPHER_CTX_key_length(ctx) * 8;
658     if ((mode == EVP_CIPH_ECB_MODE || mode == EVP_CIPH_CBC_MODE)
659         && !enc) {
660         ret = 0;
661         aes_t4_set_decrypt_key(key, bits, &dat->ks.ks);
662         dat->block = (block128_f) aes_t4_decrypt;
663         switch (bits) {
664         case 128:
665             dat->stream.cbc = mode == EVP_CIPH_CBC_MODE ?
666                 (cbc128_f) aes128_t4_cbc_decrypt : NULL;
667             break;
668         case 192:
669             dat->stream.cbc = mode == EVP_CIPH_CBC_MODE ?
670                 (cbc128_f) aes192_t4_cbc_decrypt : NULL;
671             break;
672         case 256:
673             dat->stream.cbc = mode == EVP_CIPH_CBC_MODE ?
674                 (cbc128_f) aes256_t4_cbc_decrypt : NULL;
675             break;
676         default:
677             ret = -1;
678         }
679     } else {
680         ret = 0;
681         aes_t4_set_encrypt_key(key, bits, &dat->ks.ks);
682         dat->block = (block128_f) aes_t4_encrypt;
683         switch (bits) {
684         case 128:
685             if (mode == EVP_CIPH_CBC_MODE)
686                 dat->stream.cbc = (cbc128_f) aes128_t4_cbc_encrypt;
687             else if (mode == EVP_CIPH_CTR_MODE)
688                 dat->stream.ctr = (ctr128_f) aes128_t4_ctr32_encrypt;
689             else
690                 dat->stream.cbc = NULL;
691             break;
692         case 192:
693             if (mode == EVP_CIPH_CBC_MODE)
694                 dat->stream.cbc = (cbc128_f) aes192_t4_cbc_encrypt;
695             else if (mode == EVP_CIPH_CTR_MODE)
696                 dat->stream.ctr = (ctr128_f) aes192_t4_ctr32_encrypt;
697             else
698                 dat->stream.cbc = NULL;
699             break;
700         case 256:
701             if (mode == EVP_CIPH_CBC_MODE)
702                 dat->stream.cbc = (cbc128_f) aes256_t4_cbc_encrypt;
703             else if (mode == EVP_CIPH_CTR_MODE)
704                 dat->stream.ctr = (ctr128_f) aes256_t4_ctr32_encrypt;
705             else
706                 dat->stream.cbc = NULL;
707             break;
708         default:
709             ret = -1;
710         }
711     }
712
713     if (ret < 0) {
714         EVPerr(EVP_F_AES_T4_INIT_KEY, EVP_R_AES_KEY_SETUP_FAILED);
715         return 0;
716     }
717
718     return 1;
719 }
720
721 #  define aes_t4_cbc_cipher aes_cbc_cipher
722 static int aes_t4_cbc_cipher(EVP_CIPHER_CTX *ctx, unsigned char *out,
723                              const unsigned char *in, size_t len);
724
725 #  define aes_t4_ecb_cipher aes_ecb_cipher
726 static int aes_t4_ecb_cipher(EVP_CIPHER_CTX *ctx, unsigned char *out,
727                              const unsigned char *in, size_t len);
728
729 #  define aes_t4_ofb_cipher aes_ofb_cipher
730 static int aes_t4_ofb_cipher(EVP_CIPHER_CTX *ctx, unsigned char *out,
731                              const unsigned char *in, size_t len);
732
733 #  define aes_t4_cfb_cipher aes_cfb_cipher
734 static int aes_t4_cfb_cipher(EVP_CIPHER_CTX *ctx, unsigned char *out,
735                              const unsigned char *in, size_t len);
736
737 #  define aes_t4_cfb8_cipher aes_cfb8_cipher
738 static int aes_t4_cfb8_cipher(EVP_CIPHER_CTX *ctx, unsigned char *out,
739                               const unsigned char *in, size_t len);
740
741 #  define aes_t4_cfb1_cipher aes_cfb1_cipher
742 static int aes_t4_cfb1_cipher(EVP_CIPHER_CTX *ctx, unsigned char *out,
743                               const unsigned char *in, size_t len);
744
745 #  define aes_t4_ctr_cipher aes_ctr_cipher
746 static int aes_t4_ctr_cipher(EVP_CIPHER_CTX *ctx, unsigned char *out,
747                              const unsigned char *in, size_t len);
748
749 static int aes_t4_gcm_init_key(EVP_CIPHER_CTX *ctx, const unsigned char *key,
750                                const unsigned char *iv, int enc)
751 {
752     EVP_AES_GCM_CTX *gctx = EVP_C_DATA(EVP_AES_GCM_CTX,ctx);
753     if (!iv && !key)
754         return 1;
755     if (key) {
756         int bits = EVP_CIPHER_CTX_key_length(ctx) * 8;
757         aes_t4_set_encrypt_key(key, bits, &gctx->ks.ks);
758         CRYPTO_gcm128_init(&gctx->gcm, &gctx->ks,
759                            (block128_f) aes_t4_encrypt);
760         switch (bits) {
761         case 128:
762             gctx->ctr = (ctr128_f) aes128_t4_ctr32_encrypt;
763             break;
764         case 192:
765             gctx->ctr = (ctr128_f) aes192_t4_ctr32_encrypt;
766             break;
767         case 256:
768             gctx->ctr = (ctr128_f) aes256_t4_ctr32_encrypt;
769             break;
770         default:
771             return 0;
772         }
773         /*
774          * If we have an iv can set it directly, otherwise use saved IV.
775          */
776         if (iv == NULL && gctx->iv_set)
777             iv = gctx->iv;
778         if (iv) {
779             CRYPTO_gcm128_setiv(&gctx->gcm, iv, gctx->ivlen);
780             gctx->iv_set = 1;
781         }
782         gctx->key_set = 1;
783     } else {
784         /* If key set use IV, otherwise copy */
785         if (gctx->key_set)
786             CRYPTO_gcm128_setiv(&gctx->gcm, iv, gctx->ivlen);
787         else
788             memcpy(gctx->iv, iv, gctx->ivlen);
789         gctx->iv_set = 1;
790         gctx->iv_gen = 0;
791     }
792     return 1;
793 }
794
795 #  define aes_t4_gcm_cipher aes_gcm_cipher
796 static int aes_t4_gcm_cipher(EVP_CIPHER_CTX *ctx, unsigned char *out,
797                              const unsigned char *in, size_t len);
798
799 static int aes_t4_xts_init_key(EVP_CIPHER_CTX *ctx, const unsigned char *key,
800                                const unsigned char *iv, int enc)
801 {
802     EVP_AES_XTS_CTX *xctx = EVP_C_DATA(EVP_AES_XTS_CTX,ctx);
803     if (!iv && !key)
804         return 1;
805
806     if (key) {
807         int bits = EVP_CIPHER_CTX_key_length(ctx) * 4;
808         xctx->stream = NULL;
809         /* key_len is two AES keys */
810         if (enc) {
811             aes_t4_set_encrypt_key(key, bits, &xctx->ks1.ks);
812             xctx->xts.block1 = (block128_f) aes_t4_encrypt;
813             switch (bits) {
814             case 128:
815                 xctx->stream = aes128_t4_xts_encrypt;
816                 break;
817             case 256:
818                 xctx->stream = aes256_t4_xts_encrypt;
819                 break;
820             default:
821                 return 0;
822             }
823         } else {
824             aes_t4_set_decrypt_key(key, EVP_CIPHER_CTX_key_length(ctx) * 4,
825                                    &xctx->ks1.ks);
826             xctx->xts.block1 = (block128_f) aes_t4_decrypt;
827             switch (bits) {
828             case 128:
829                 xctx->stream = aes128_t4_xts_decrypt;
830                 break;
831             case 256:
832                 xctx->stream = aes256_t4_xts_decrypt;
833                 break;
834             default:
835                 return 0;
836             }
837         }
838
839         aes_t4_set_encrypt_key(key + EVP_CIPHER_CTX_key_length(ctx) / 2,
840                                EVP_CIPHER_CTX_key_length(ctx) * 4,
841                                &xctx->ks2.ks);
842         xctx->xts.block2 = (block128_f) aes_t4_encrypt;
843
844         xctx->xts.key1 = &xctx->ks1;
845     }
846
847     if (iv) {
848         xctx->xts.key2 = &xctx->ks2;
849         memcpy(EVP_CIPHER_CTX_iv_noconst(ctx), iv, 16);
850     }
851
852     return 1;
853 }
854
855 #  define aes_t4_xts_cipher aes_xts_cipher
856 static int aes_t4_xts_cipher(EVP_CIPHER_CTX *ctx, unsigned char *out,
857                              const unsigned char *in, size_t len);
858
859 static int aes_t4_ccm_init_key(EVP_CIPHER_CTX *ctx, const unsigned char *key,
860                                const unsigned char *iv, int enc)
861 {
862     EVP_AES_CCM_CTX *cctx = EVP_C_DATA(EVP_AES_CCM_CTX,ctx);
863     if (!iv && !key)
864         return 1;
865     if (key) {
866         int bits = EVP_CIPHER_CTX_key_length(ctx) * 8;
867         aes_t4_set_encrypt_key(key, bits, &cctx->ks.ks);
868         CRYPTO_ccm128_init(&cctx->ccm, cctx->M, cctx->L,
869                            &cctx->ks, (block128_f) aes_t4_encrypt);
870         cctx->str = NULL;
871         cctx->key_set = 1;
872     }
873     if (iv) {
874         memcpy(EVP_CIPHER_CTX_iv_noconst(ctx), iv, 15 - cctx->L);
875         cctx->iv_set = 1;
876     }
877     return 1;
878 }
879
880 #  define aes_t4_ccm_cipher aes_ccm_cipher
881 static int aes_t4_ccm_cipher(EVP_CIPHER_CTX *ctx, unsigned char *out,
882                              const unsigned char *in, size_t len);
883
884 #  ifndef OPENSSL_NO_OCB
885 static int aes_t4_ocb_init_key(EVP_CIPHER_CTX *ctx, const unsigned char *key,
886                                const unsigned char *iv, int enc)
887 {
888     EVP_AES_OCB_CTX *octx = EVP_C_DATA(EVP_AES_OCB_CTX,ctx);
889     if (!iv && !key)
890         return 1;
891     if (key) {
892         do {
893             /*
894              * We set both the encrypt and decrypt key here because decrypt
895              * needs both. We could possibly optimise to remove setting the
896              * decrypt for an encryption operation.
897              */
898             aes_t4_set_encrypt_key(key, EVP_CIPHER_CTX_key_length(ctx) * 8,
899                                    &octx->ksenc.ks);
900             aes_t4_set_decrypt_key(key, EVP_CIPHER_CTX_key_length(ctx) * 8,
901                                    &octx->ksdec.ks);
902             if (!CRYPTO_ocb128_init(&octx->ocb,
903                                     &octx->ksenc.ks, &octx->ksdec.ks,
904                                     (block128_f) aes_t4_encrypt,
905                                     (block128_f) aes_t4_decrypt,
906                                     NULL))
907                 return 0;
908         }
909         while (0);
910
911         /*
912          * If we have an iv we can set it directly, otherwise use saved IV.
913          */
914         if (iv == NULL && octx->iv_set)
915             iv = octx->iv;
916         if (iv) {
917             if (CRYPTO_ocb128_setiv(&octx->ocb, iv, octx->ivlen, octx->taglen)
918                 != 1)
919                 return 0;
920             octx->iv_set = 1;
921         }
922         octx->key_set = 1;
923     } else {
924         /* If key set use IV, otherwise copy */
925         if (octx->key_set)
926             CRYPTO_ocb128_setiv(&octx->ocb, iv, octx->ivlen, octx->taglen);
927         else
928             memcpy(octx->iv, iv, octx->ivlen);
929         octx->iv_set = 1;
930     }
931     return 1;
932 }
933
934 #   define aes_t4_ocb_cipher aes_ocb_cipher
935 static int aes_t4_ocb_cipher(EVP_CIPHER_CTX *ctx, unsigned char *out,
936                              const unsigned char *in, size_t len);
937 #  endif                        /* OPENSSL_NO_OCB */
938
939 #  define BLOCK_CIPHER_generic(nid,keylen,blocksize,ivlen,nmode,mode,MODE,flags) \
940 static const EVP_CIPHER aes_t4_##keylen##_##mode = { \
941         nid##_##keylen##_##nmode,blocksize,keylen/8,ivlen, \
942         flags|EVP_CIPH_##MODE##_MODE,   \
943         aes_t4_init_key,                \
944         aes_t4_##mode##_cipher,         \
945         NULL,                           \
946         sizeof(EVP_AES_KEY),            \
947         NULL,NULL,NULL,NULL }; \
948 static const EVP_CIPHER aes_##keylen##_##mode = { \
949         nid##_##keylen##_##nmode,blocksize,     \
950         keylen/8,ivlen, \
951         flags|EVP_CIPH_##MODE##_MODE,   \
952         aes_init_key,                   \
953         aes_##mode##_cipher,            \
954         NULL,                           \
955         sizeof(EVP_AES_KEY),            \
956         NULL,NULL,NULL,NULL }; \
957 const EVP_CIPHER *EVP_aes_##keylen##_##mode(void) \
958 { return SPARC_AES_CAPABLE?&aes_t4_##keylen##_##mode:&aes_##keylen##_##mode; }
959
960 #  define BLOCK_CIPHER_custom(nid,keylen,blocksize,ivlen,mode,MODE,flags) \
961 static const EVP_CIPHER aes_t4_##keylen##_##mode = { \
962         nid##_##keylen##_##mode,blocksize, \
963         (EVP_CIPH_##MODE##_MODE==EVP_CIPH_XTS_MODE?2:1)*keylen/8, ivlen, \
964         flags|EVP_CIPH_##MODE##_MODE,   \
965         aes_t4_##mode##_init_key,       \
966         aes_t4_##mode##_cipher,         \
967         aes_##mode##_cleanup,           \
968         sizeof(EVP_AES_##MODE##_CTX),   \
969         NULL,NULL,aes_##mode##_ctrl,NULL }; \
970 static const EVP_CIPHER aes_##keylen##_##mode = { \
971         nid##_##keylen##_##mode,blocksize, \
972         (EVP_CIPH_##MODE##_MODE==EVP_CIPH_XTS_MODE?2:1)*keylen/8, ivlen, \
973         flags|EVP_CIPH_##MODE##_MODE,   \
974         aes_##mode##_init_key,          \
975         aes_##mode##_cipher,            \
976         aes_##mode##_cleanup,           \
977         sizeof(EVP_AES_##MODE##_CTX),   \
978         NULL,NULL,aes_##mode##_ctrl,NULL }; \
979 const EVP_CIPHER *EVP_aes_##keylen##_##mode(void) \
980 { return SPARC_AES_CAPABLE?&aes_t4_##keylen##_##mode:&aes_##keylen##_##mode; }
981
982 # else
983
984 #  define BLOCK_CIPHER_generic(nid,keylen,blocksize,ivlen,nmode,mode,MODE,flags) \
985 static const EVP_CIPHER aes_##keylen##_##mode = { \
986         nid##_##keylen##_##nmode,blocksize,keylen/8,ivlen, \
987         flags|EVP_CIPH_##MODE##_MODE,   \
988         aes_init_key,                   \
989         aes_##mode##_cipher,            \
990         NULL,                           \
991         sizeof(EVP_AES_KEY),            \
992         NULL,NULL,NULL,NULL }; \
993 const EVP_CIPHER *EVP_aes_##keylen##_##mode(void) \
994 { return &aes_##keylen##_##mode; }
995
996 #  define BLOCK_CIPHER_custom(nid,keylen,blocksize,ivlen,mode,MODE,flags) \
997 static const EVP_CIPHER aes_##keylen##_##mode = { \
998         nid##_##keylen##_##mode,blocksize, \
999         (EVP_CIPH_##MODE##_MODE==EVP_CIPH_XTS_MODE?2:1)*keylen/8, ivlen, \
1000         flags|EVP_CIPH_##MODE##_MODE,   \
1001         aes_##mode##_init_key,          \
1002         aes_##mode##_cipher,            \
1003         aes_##mode##_cleanup,           \
1004         sizeof(EVP_AES_##MODE##_CTX),   \
1005         NULL,NULL,aes_##mode##_ctrl,NULL }; \
1006 const EVP_CIPHER *EVP_aes_##keylen##_##mode(void) \
1007 { return &aes_##keylen##_##mode; }
1008
1009 # endif
1010
1011 # if defined(OPENSSL_CPUID_OBJ) && (defined(__arm__) || defined(__arm) || defined(__aarch64__))
1012 #  include "arm_arch.h"
1013 #  if __ARM_MAX_ARCH__>=7
1014 #   if defined(BSAES_ASM)
1015 #    define BSAES_CAPABLE (OPENSSL_armcap_P & ARMV7_NEON)
1016 #   endif
1017 #   if defined(VPAES_ASM)
1018 #    define VPAES_CAPABLE (OPENSSL_armcap_P & ARMV7_NEON)
1019 #   endif
1020 #   define HWAES_CAPABLE (OPENSSL_armcap_P & ARMV8_AES)
1021 #   define HWAES_set_encrypt_key aes_v8_set_encrypt_key
1022 #   define HWAES_set_decrypt_key aes_v8_set_decrypt_key
1023 #   define HWAES_encrypt aes_v8_encrypt
1024 #   define HWAES_decrypt aes_v8_decrypt
1025 #   define HWAES_cbc_encrypt aes_v8_cbc_encrypt
1026 #   define HWAES_ctr32_encrypt_blocks aes_v8_ctr32_encrypt_blocks
1027 #  endif
1028 # endif
1029
1030 # if defined(HWAES_CAPABLE)
1031 int HWAES_set_encrypt_key(const unsigned char *userKey, const int bits,
1032                           AES_KEY *key);
1033 int HWAES_set_decrypt_key(const unsigned char *userKey, const int bits,
1034                           AES_KEY *key);
1035 void HWAES_encrypt(const unsigned char *in, unsigned char *out,
1036                    const AES_KEY *key);
1037 void HWAES_decrypt(const unsigned char *in, unsigned char *out,
1038                    const AES_KEY *key);
1039 void HWAES_cbc_encrypt(const unsigned char *in, unsigned char *out,
1040                        size_t length, const AES_KEY *key,
1041                        unsigned char *ivec, const int enc);
1042 void HWAES_ctr32_encrypt_blocks(const unsigned char *in, unsigned char *out,
1043                                 size_t len, const AES_KEY *key,
1044                                 const unsigned char ivec[16]);
1045 # endif
1046
1047 # define BLOCK_CIPHER_generic_pack(nid,keylen,flags)             \
1048         BLOCK_CIPHER_generic(nid,keylen,16,16,cbc,cbc,CBC,flags|EVP_CIPH_FLAG_DEFAULT_ASN1)     \
1049         BLOCK_CIPHER_generic(nid,keylen,16,0,ecb,ecb,ECB,flags|EVP_CIPH_FLAG_DEFAULT_ASN1)      \
1050         BLOCK_CIPHER_generic(nid,keylen,1,16,ofb128,ofb,OFB,flags|EVP_CIPH_FLAG_DEFAULT_ASN1)   \
1051         BLOCK_CIPHER_generic(nid,keylen,1,16,cfb128,cfb,CFB,flags|EVP_CIPH_FLAG_DEFAULT_ASN1)   \
1052         BLOCK_CIPHER_generic(nid,keylen,1,16,cfb1,cfb1,CFB,flags)       \
1053         BLOCK_CIPHER_generic(nid,keylen,1,16,cfb8,cfb8,CFB,flags)       \
1054         BLOCK_CIPHER_generic(nid,keylen,1,16,ctr,ctr,CTR,flags)
1055
1056 static int aes_init_key(EVP_CIPHER_CTX *ctx, const unsigned char *key,
1057                         const unsigned char *iv, int enc)
1058 {
1059     int ret, mode;
1060     EVP_AES_KEY *dat = EVP_C_DATA(EVP_AES_KEY,ctx);
1061
1062     mode = EVP_CIPHER_CTX_mode(ctx);
1063     if ((mode == EVP_CIPH_ECB_MODE || mode == EVP_CIPH_CBC_MODE)
1064         && !enc)
1065 # ifdef HWAES_CAPABLE
1066         if (HWAES_CAPABLE) {
1067             ret = HWAES_set_decrypt_key(key,
1068                                         EVP_CIPHER_CTX_key_length(ctx) * 8,
1069                                         &dat->ks.ks);
1070             dat->block = (block128_f) HWAES_decrypt;
1071             dat->stream.cbc = NULL;
1072 #  ifdef HWAES_cbc_encrypt
1073             if (mode == EVP_CIPH_CBC_MODE)
1074                 dat->stream.cbc = (cbc128_f) HWAES_cbc_encrypt;
1075 #  endif
1076         } else
1077 # endif
1078 # ifdef BSAES_CAPABLE
1079         if (BSAES_CAPABLE && mode == EVP_CIPH_CBC_MODE) {
1080             ret = AES_set_decrypt_key(key, EVP_CIPHER_CTX_key_length(ctx) * 8,
1081                                       &dat->ks.ks);
1082             dat->block = (block128_f) AES_decrypt;
1083             dat->stream.cbc = (cbc128_f) bsaes_cbc_encrypt;
1084         } else
1085 # endif
1086 # ifdef VPAES_CAPABLE
1087         if (VPAES_CAPABLE) {
1088             ret = vpaes_set_decrypt_key(key,
1089                                         EVP_CIPHER_CTX_key_length(ctx) * 8,
1090                                         &dat->ks.ks);
1091             dat->block = (block128_f) vpaes_decrypt;
1092             dat->stream.cbc = mode == EVP_CIPH_CBC_MODE ?
1093                 (cbc128_f) vpaes_cbc_encrypt : NULL;
1094         } else
1095 # endif
1096         {
1097             ret = AES_set_decrypt_key(key,
1098                                       EVP_CIPHER_CTX_key_length(ctx) * 8,
1099                                       &dat->ks.ks);
1100             dat->block = (block128_f) AES_decrypt;
1101             dat->stream.cbc = mode == EVP_CIPH_CBC_MODE ?
1102                 (cbc128_f) AES_cbc_encrypt : NULL;
1103     } else
1104 # ifdef HWAES_CAPABLE
1105     if (HWAES_CAPABLE) {
1106         ret = HWAES_set_encrypt_key(key, EVP_CIPHER_CTX_key_length(ctx) * 8,
1107                                     &dat->ks.ks);
1108         dat->block = (block128_f) HWAES_encrypt;
1109         dat->stream.cbc = NULL;
1110 #  ifdef HWAES_cbc_encrypt
1111         if (mode == EVP_CIPH_CBC_MODE)
1112             dat->stream.cbc = (cbc128_f) HWAES_cbc_encrypt;
1113         else
1114 #  endif
1115 #  ifdef HWAES_ctr32_encrypt_blocks
1116         if (mode == EVP_CIPH_CTR_MODE)
1117             dat->stream.ctr = (ctr128_f) HWAES_ctr32_encrypt_blocks;
1118         else
1119 #  endif
1120             (void)0;            /* terminate potentially open 'else' */
1121     } else
1122 # endif
1123 # ifdef BSAES_CAPABLE
1124     if (BSAES_CAPABLE && mode == EVP_CIPH_CTR_MODE) {
1125         ret = AES_set_encrypt_key(key, EVP_CIPHER_CTX_key_length(ctx) * 8,
1126                                   &dat->ks.ks);
1127         dat->block = (block128_f) AES_encrypt;
1128         dat->stream.ctr = (ctr128_f) bsaes_ctr32_encrypt_blocks;
1129     } else
1130 # endif
1131 # ifdef VPAES_CAPABLE
1132     if (VPAES_CAPABLE) {
1133         ret = vpaes_set_encrypt_key(key, EVP_CIPHER_CTX_key_length(ctx) * 8,
1134                                     &dat->ks.ks);
1135         dat->block = (block128_f) vpaes_encrypt;
1136         dat->stream.cbc = mode == EVP_CIPH_CBC_MODE ?
1137             (cbc128_f) vpaes_cbc_encrypt : NULL;
1138     } else
1139 # endif
1140     {
1141         ret = AES_set_encrypt_key(key, EVP_CIPHER_CTX_key_length(ctx) * 8,
1142                                   &dat->ks.ks);
1143         dat->block = (block128_f) AES_encrypt;
1144         dat->stream.cbc = mode == EVP_CIPH_CBC_MODE ?
1145             (cbc128_f) AES_cbc_encrypt : NULL;
1146 # ifdef AES_CTR_ASM
1147         if (mode == EVP_CIPH_CTR_MODE)
1148             dat->stream.ctr = (ctr128_f) AES_ctr32_encrypt;
1149 # endif
1150     }
1151
1152     if (ret < 0) {
1153         EVPerr(EVP_F_AES_INIT_KEY, EVP_R_AES_KEY_SETUP_FAILED);
1154         return 0;
1155     }
1156
1157     return 1;
1158 }
1159
1160 static int aes_cbc_cipher(EVP_CIPHER_CTX *ctx, unsigned char *out,
1161                           const unsigned char *in, size_t len)
1162 {
1163     EVP_AES_KEY *dat = EVP_C_DATA(EVP_AES_KEY,ctx);
1164
1165     if (dat->stream.cbc)
1166         (*dat->stream.cbc) (in, out, len, &dat->ks,
1167                             EVP_CIPHER_CTX_iv_noconst(ctx),
1168                             EVP_CIPHER_CTX_encrypting(ctx));
1169     else if (EVP_CIPHER_CTX_encrypting(ctx))
1170         CRYPTO_cbc128_encrypt(in, out, len, &dat->ks,
1171                               EVP_CIPHER_CTX_iv_noconst(ctx), dat->block);
1172     else
1173         CRYPTO_cbc128_decrypt(in, out, len, &dat->ks,
1174                               EVP_CIPHER_CTX_iv_noconst(ctx), dat->block);
1175
1176     return 1;
1177 }
1178
1179 static int aes_ecb_cipher(EVP_CIPHER_CTX *ctx, unsigned char *out,
1180                           const unsigned char *in, size_t len)
1181 {
1182     size_t bl = EVP_CIPHER_CTX_block_size(ctx);
1183     size_t i;
1184     EVP_AES_KEY *dat = EVP_C_DATA(EVP_AES_KEY,ctx);
1185
1186     if (len < bl)
1187         return 1;
1188
1189     for (i = 0, len -= bl; i <= len; i += bl)
1190         (*dat->block) (in + i, out + i, &dat->ks);
1191
1192     return 1;
1193 }
1194
1195 static int aes_ofb_cipher(EVP_CIPHER_CTX *ctx, unsigned char *out,
1196                           const unsigned char *in, size_t len)
1197 {
1198     EVP_AES_KEY *dat = EVP_C_DATA(EVP_AES_KEY,ctx);
1199
1200     int num = EVP_CIPHER_CTX_num(ctx);
1201     CRYPTO_ofb128_encrypt(in, out, len, &dat->ks,
1202                           EVP_CIPHER_CTX_iv_noconst(ctx), &num, dat->block);
1203     EVP_CIPHER_CTX_set_num(ctx, num);
1204     return 1;
1205 }
1206
1207 static int aes_cfb_cipher(EVP_CIPHER_CTX *ctx, unsigned char *out,
1208                           const unsigned char *in, size_t len)
1209 {
1210     EVP_AES_KEY *dat = EVP_C_DATA(EVP_AES_KEY,ctx);
1211
1212     int num = EVP_CIPHER_CTX_num(ctx);
1213     CRYPTO_cfb128_encrypt(in, out, len, &dat->ks,
1214                           EVP_CIPHER_CTX_iv_noconst(ctx), &num,
1215                           EVP_CIPHER_CTX_encrypting(ctx), dat->block);
1216     EVP_CIPHER_CTX_set_num(ctx, num);
1217     return 1;
1218 }
1219
1220 static int aes_cfb8_cipher(EVP_CIPHER_CTX *ctx, unsigned char *out,
1221                            const unsigned char *in, size_t len)
1222 {
1223     EVP_AES_KEY *dat = EVP_C_DATA(EVP_AES_KEY,ctx);
1224
1225     int num = EVP_CIPHER_CTX_num(ctx);
1226     CRYPTO_cfb128_8_encrypt(in, out, len, &dat->ks,
1227                             EVP_CIPHER_CTX_iv_noconst(ctx), &num,
1228                             EVP_CIPHER_CTX_encrypting(ctx), dat->block);
1229     EVP_CIPHER_CTX_set_num(ctx, num);
1230     return 1;
1231 }
1232
1233 static int aes_cfb1_cipher(EVP_CIPHER_CTX *ctx, unsigned char *out,
1234                            const unsigned char *in, size_t len)
1235 {
1236     EVP_AES_KEY *dat = EVP_C_DATA(EVP_AES_KEY,ctx);
1237
1238     if (EVP_CIPHER_CTX_test_flags(ctx, EVP_CIPH_FLAG_LENGTH_BITS)) {
1239         int num = EVP_CIPHER_CTX_num(ctx);
1240         CRYPTO_cfb128_1_encrypt(in, out, len, &dat->ks,
1241                                 EVP_CIPHER_CTX_iv_noconst(ctx), &num,
1242                                 EVP_CIPHER_CTX_encrypting(ctx), dat->block);
1243         EVP_CIPHER_CTX_set_num(ctx, num);
1244         return 1;
1245     }
1246
1247     while (len >= MAXBITCHUNK) {
1248         int num = EVP_CIPHER_CTX_num(ctx);
1249         CRYPTO_cfb128_1_encrypt(in, out, MAXBITCHUNK * 8, &dat->ks,
1250                                 EVP_CIPHER_CTX_iv_noconst(ctx), &num,
1251                                 EVP_CIPHER_CTX_encrypting(ctx), dat->block);
1252         EVP_CIPHER_CTX_set_num(ctx, num);
1253         len -= MAXBITCHUNK;
1254     }
1255     if (len) {
1256         int num = EVP_CIPHER_CTX_num(ctx);
1257         CRYPTO_cfb128_1_encrypt(in, out, len * 8, &dat->ks,
1258                                 EVP_CIPHER_CTX_iv_noconst(ctx), &num,
1259                                 EVP_CIPHER_CTX_encrypting(ctx), dat->block);
1260         EVP_CIPHER_CTX_set_num(ctx, num);
1261     }
1262
1263     return 1;
1264 }
1265
1266 static int aes_ctr_cipher(EVP_CIPHER_CTX *ctx, unsigned char *out,
1267                           const unsigned char *in, size_t len)
1268 {
1269     unsigned int num = EVP_CIPHER_CTX_num(ctx);
1270     EVP_AES_KEY *dat = EVP_C_DATA(EVP_AES_KEY,ctx);
1271
1272     if (dat->stream.ctr)
1273         CRYPTO_ctr128_encrypt_ctr32(in, out, len, &dat->ks,
1274                                     EVP_CIPHER_CTX_iv_noconst(ctx),
1275                                     EVP_CIPHER_CTX_buf_noconst(ctx),
1276                                     &num, dat->stream.ctr);
1277     else
1278         CRYPTO_ctr128_encrypt(in, out, len, &dat->ks,
1279                               EVP_CIPHER_CTX_iv_noconst(ctx),
1280                               EVP_CIPHER_CTX_buf_noconst(ctx), &num,
1281                               dat->block);
1282     EVP_CIPHER_CTX_set_num(ctx, num);
1283     return 1;
1284 }
1285
1286 BLOCK_CIPHER_generic_pack(NID_aes, 128, 0)
1287     BLOCK_CIPHER_generic_pack(NID_aes, 192, 0)
1288     BLOCK_CIPHER_generic_pack(NID_aes, 256, 0)
1289
1290 static int aes_gcm_cleanup(EVP_CIPHER_CTX *c)
1291 {
1292     EVP_AES_GCM_CTX *gctx = EVP_C_DATA(EVP_AES_GCM_CTX,c);
1293     OPENSSL_cleanse(&gctx->gcm, sizeof(gctx->gcm));
1294     if (gctx->iv != EVP_CIPHER_CTX_iv_noconst(c))
1295         OPENSSL_free(gctx->iv);
1296     return 1;
1297 }
1298
1299 /* increment counter (64-bit int) by 1 */
1300 static void ctr64_inc(unsigned char *counter)
1301 {
1302     int n = 8;
1303     unsigned char c;
1304
1305     do {
1306         --n;
1307         c = counter[n];
1308         ++c;
1309         counter[n] = c;
1310         if (c)
1311             return;
1312     } while (n);
1313 }
1314
1315 static int aes_gcm_ctrl(EVP_CIPHER_CTX *c, int type, int arg, void *ptr)
1316 {
1317     EVP_AES_GCM_CTX *gctx = EVP_C_DATA(EVP_AES_GCM_CTX,c);
1318     switch (type) {
1319     case EVP_CTRL_INIT:
1320         gctx->key_set = 0;
1321         gctx->iv_set = 0;
1322         gctx->ivlen = EVP_CIPHER_CTX_iv_length(c);
1323         gctx->iv = EVP_CIPHER_CTX_iv_noconst(c);
1324         gctx->taglen = -1;
1325         gctx->iv_gen = 0;
1326         gctx->tls_aad_len = -1;
1327         return 1;
1328
1329     case EVP_CTRL_AEAD_SET_IVLEN:
1330         if (arg <= 0)
1331             return 0;
1332         /* Allocate memory for IV if needed */
1333         if ((arg > EVP_MAX_IV_LENGTH) && (arg > gctx->ivlen)) {
1334             if (gctx->iv != EVP_CIPHER_CTX_iv_noconst(c))
1335                 OPENSSL_free(gctx->iv);
1336             gctx->iv = OPENSSL_malloc(arg);
1337             if (gctx->iv == NULL)
1338                 return 0;
1339         }
1340         gctx->ivlen = arg;
1341         return 1;
1342
1343     case EVP_CTRL_AEAD_SET_TAG:
1344         if (arg <= 0 || arg > 16 || EVP_CIPHER_CTX_encrypting(c))
1345             return 0;
1346         memcpy(EVP_CIPHER_CTX_buf_noconst(c), ptr, arg);
1347         gctx->taglen = arg;
1348         return 1;
1349
1350     case EVP_CTRL_AEAD_GET_TAG:
1351         if (arg <= 0 || arg > 16 || !EVP_CIPHER_CTX_encrypting(c)
1352             || gctx->taglen < 0)
1353             return 0;
1354         memcpy(ptr, EVP_CIPHER_CTX_buf_noconst(c), arg);
1355         return 1;
1356
1357     case EVP_CTRL_GCM_SET_IV_FIXED:
1358         /* Special case: -1 length restores whole IV */
1359         if (arg == -1) {
1360             memcpy(gctx->iv, ptr, gctx->ivlen);
1361             gctx->iv_gen = 1;
1362             return 1;
1363         }
1364         /*
1365          * Fixed field must be at least 4 bytes and invocation field at least
1366          * 8.
1367          */
1368         if ((arg < 4) || (gctx->ivlen - arg) < 8)
1369             return 0;
1370         if (arg)
1371             memcpy(gctx->iv, ptr, arg);
1372         if (EVP_CIPHER_CTX_encrypting(c)
1373             && RAND_bytes(gctx->iv + arg, gctx->ivlen - arg) <= 0)
1374             return 0;
1375         gctx->iv_gen = 1;
1376         return 1;
1377
1378     case EVP_CTRL_GCM_IV_GEN:
1379         if (gctx->iv_gen == 0 || gctx->key_set == 0)
1380             return 0;
1381         CRYPTO_gcm128_setiv(&gctx->gcm, gctx->iv, gctx->ivlen);
1382         if (arg <= 0 || arg > gctx->ivlen)
1383             arg = gctx->ivlen;
1384         memcpy(ptr, gctx->iv + gctx->ivlen - arg, arg);
1385         /*
1386          * Invocation field will be at least 8 bytes in size and so no need
1387          * to check wrap around or increment more than last 8 bytes.
1388          */
1389         ctr64_inc(gctx->iv + gctx->ivlen - 8);
1390         gctx->iv_set = 1;
1391         return 1;
1392
1393     case EVP_CTRL_GCM_SET_IV_INV:
1394         if (gctx->iv_gen == 0 || gctx->key_set == 0
1395             || EVP_CIPHER_CTX_encrypting(c))
1396             return 0;
1397         memcpy(gctx->iv + gctx->ivlen - arg, ptr, arg);
1398         CRYPTO_gcm128_setiv(&gctx->gcm, gctx->iv, gctx->ivlen);
1399         gctx->iv_set = 1;
1400         return 1;
1401
1402     case EVP_CTRL_AEAD_TLS1_AAD:
1403         /* Save the AAD for later use */
1404         if (arg != EVP_AEAD_TLS1_AAD_LEN)
1405             return 0;
1406         memcpy(EVP_CIPHER_CTX_buf_noconst(c), ptr, arg);
1407         gctx->tls_aad_len = arg;
1408         {
1409             unsigned int len =
1410                 EVP_CIPHER_CTX_buf_noconst(c)[arg - 2] << 8
1411                 | EVP_CIPHER_CTX_buf_noconst(c)[arg - 1];
1412             /* Correct length for explicit IV */
1413             len -= EVP_GCM_TLS_EXPLICIT_IV_LEN;
1414             /* If decrypting correct for tag too */
1415             if (!EVP_CIPHER_CTX_encrypting(c))
1416                 len -= EVP_GCM_TLS_TAG_LEN;
1417             EVP_CIPHER_CTX_buf_noconst(c)[arg - 2] = len >> 8;
1418             EVP_CIPHER_CTX_buf_noconst(c)[arg - 1] = len & 0xff;
1419         }
1420         /* Extra padding: tag appended to record */
1421         return EVP_GCM_TLS_TAG_LEN;
1422
1423     case EVP_CTRL_COPY:
1424         {
1425             EVP_CIPHER_CTX *out = ptr;
1426             EVP_AES_GCM_CTX *gctx_out = EVP_C_DATA(EVP_AES_GCM_CTX,out);
1427             if (gctx->gcm.key) {
1428                 if (gctx->gcm.key != &gctx->ks)
1429                     return 0;
1430                 gctx_out->gcm.key = &gctx_out->ks;
1431             }
1432             if (gctx->iv == EVP_CIPHER_CTX_iv_noconst(c))
1433                 gctx_out->iv = EVP_CIPHER_CTX_iv_noconst(out);
1434             else {
1435                 gctx_out->iv = OPENSSL_malloc(gctx->ivlen);
1436                 if (gctx_out->iv == NULL)
1437                     return 0;
1438                 memcpy(gctx_out->iv, gctx->iv, gctx->ivlen);
1439             }
1440             return 1;
1441         }
1442
1443     default:
1444         return -1;
1445
1446     }
1447 }
1448
1449 static int aes_gcm_init_key(EVP_CIPHER_CTX *ctx, const unsigned char *key,
1450                             const unsigned char *iv, int enc)
1451 {
1452     EVP_AES_GCM_CTX *gctx = EVP_C_DATA(EVP_AES_GCM_CTX,ctx);
1453     if (!iv && !key)
1454         return 1;
1455     if (key) {
1456         do {
1457 # ifdef HWAES_CAPABLE
1458             if (HWAES_CAPABLE) {
1459                 HWAES_set_encrypt_key(key, EVP_CIPHER_CTX_key_length(ctx) * 8,
1460                                       &gctx->ks.ks);
1461                 CRYPTO_gcm128_init(&gctx->gcm, &gctx->ks,
1462                                    (block128_f) HWAES_encrypt);
1463 #  ifdef HWAES_ctr32_encrypt_blocks
1464                 gctx->ctr = (ctr128_f) HWAES_ctr32_encrypt_blocks;
1465 #  else
1466                 gctx->ctr = NULL;
1467 #  endif
1468                 break;
1469             } else
1470 # endif
1471 # ifdef BSAES_CAPABLE
1472             if (BSAES_CAPABLE) {
1473                 AES_set_encrypt_key(key, EVP_CIPHER_CTX_key_length(ctx) * 8,
1474                                     &gctx->ks.ks);
1475                 CRYPTO_gcm128_init(&gctx->gcm, &gctx->ks,
1476                                    (block128_f) AES_encrypt);
1477                 gctx->ctr = (ctr128_f) bsaes_ctr32_encrypt_blocks;
1478                 break;
1479             } else
1480 # endif
1481 # ifdef VPAES_CAPABLE
1482             if (VPAES_CAPABLE) {
1483                 vpaes_set_encrypt_key(key, EVP_CIPHER_CTX_key_length(ctx) * 8,
1484                                       &gctx->ks.ks);
1485                 CRYPTO_gcm128_init(&gctx->gcm, &gctx->ks,
1486                                    (block128_f) vpaes_encrypt);
1487                 gctx->ctr = NULL;
1488                 break;
1489             } else
1490 # endif
1491                 (void)0;        /* terminate potentially open 'else' */
1492
1493             AES_set_encrypt_key(key, EVP_CIPHER_CTX_key_length(ctx) * 8,
1494                                 &gctx->ks.ks);
1495             CRYPTO_gcm128_init(&gctx->gcm, &gctx->ks,
1496                                (block128_f) AES_encrypt);
1497 # ifdef AES_CTR_ASM
1498             gctx->ctr = (ctr128_f) AES_ctr32_encrypt;
1499 # else
1500             gctx->ctr = NULL;
1501 # endif
1502         } while (0);
1503
1504         /*
1505          * If we have an iv can set it directly, otherwise use saved IV.
1506          */
1507         if (iv == NULL && gctx->iv_set)
1508             iv = gctx->iv;
1509         if (iv) {
1510             CRYPTO_gcm128_setiv(&gctx->gcm, iv, gctx->ivlen);
1511             gctx->iv_set = 1;
1512         }
1513         gctx->key_set = 1;
1514     } else {
1515         /* If key set use IV, otherwise copy */
1516         if (gctx->key_set)
1517             CRYPTO_gcm128_setiv(&gctx->gcm, iv, gctx->ivlen);
1518         else
1519             memcpy(gctx->iv, iv, gctx->ivlen);
1520         gctx->iv_set = 1;
1521         gctx->iv_gen = 0;
1522     }
1523     return 1;
1524 }
1525
1526 /*
1527  * Handle TLS GCM packet format. This consists of the last portion of the IV
1528  * followed by the payload and finally the tag. On encrypt generate IV,
1529  * encrypt payload and write the tag. On verify retrieve IV, decrypt payload
1530  * and verify tag.
1531  */
1532
1533 static int aes_gcm_tls_cipher(EVP_CIPHER_CTX *ctx, unsigned char *out,
1534                               const unsigned char *in, size_t len)
1535 {
1536     EVP_AES_GCM_CTX *gctx = EVP_C_DATA(EVP_AES_GCM_CTX,ctx);
1537     int rv = -1;
1538     /* Encrypt/decrypt must be performed in place */
1539     if (out != in
1540         || len < (EVP_GCM_TLS_EXPLICIT_IV_LEN + EVP_GCM_TLS_TAG_LEN))
1541         return -1;
1542     /*
1543      * Set IV from start of buffer or generate IV and write to start of
1544      * buffer.
1545      */
1546     if (EVP_CIPHER_CTX_ctrl(ctx, EVP_CIPHER_CTX_encrypting(ctx) ?
1547                             EVP_CTRL_GCM_IV_GEN : EVP_CTRL_GCM_SET_IV_INV,
1548                             EVP_GCM_TLS_EXPLICIT_IV_LEN, out) <= 0)
1549         goto err;
1550     /* Use saved AAD */
1551     if (CRYPTO_gcm128_aad(&gctx->gcm, EVP_CIPHER_CTX_buf_noconst(ctx),
1552                           gctx->tls_aad_len))
1553         goto err;
1554     /* Fix buffer and length to point to payload */
1555     in += EVP_GCM_TLS_EXPLICIT_IV_LEN;
1556     out += EVP_GCM_TLS_EXPLICIT_IV_LEN;
1557     len -= EVP_GCM_TLS_EXPLICIT_IV_LEN + EVP_GCM_TLS_TAG_LEN;
1558     if (EVP_CIPHER_CTX_encrypting(ctx)) {
1559         /* Encrypt payload */
1560         if (gctx->ctr) {
1561             size_t bulk = 0;
1562 # if defined(AES_GCM_ASM)
1563             if (len >= 32 && AES_GCM_ASM(gctx)) {
1564                 if (CRYPTO_gcm128_encrypt(&gctx->gcm, NULL, NULL, 0))
1565                     return -1;
1566
1567                 bulk = AES_gcm_encrypt(in, out, len,
1568                                        gctx->gcm.key,
1569                                        gctx->gcm.Yi.c, gctx->gcm.Xi.u);
1570                 gctx->gcm.len.u[1] += bulk;
1571             }
1572 # endif
1573             if (CRYPTO_gcm128_encrypt_ctr32(&gctx->gcm,
1574                                             in + bulk,
1575                                             out + bulk,
1576                                             len - bulk, gctx->ctr))
1577                 goto err;
1578         } else {
1579             size_t bulk = 0;
1580 # if defined(AES_GCM_ASM2)
1581             if (len >= 32 && AES_GCM_ASM2(gctx)) {
1582                 if (CRYPTO_gcm128_encrypt(&gctx->gcm, NULL, NULL, 0))
1583                     return -1;
1584
1585                 bulk = AES_gcm_encrypt(in, out, len,
1586                                        gctx->gcm.key,
1587                                        gctx->gcm.Yi.c, gctx->gcm.Xi.u);
1588                 gctx->gcm.len.u[1] += bulk;
1589             }
1590 # endif
1591             if (CRYPTO_gcm128_encrypt(&gctx->gcm,
1592                                       in + bulk, out + bulk, len - bulk))
1593                 goto err;
1594         }
1595         out += len;
1596         /* Finally write tag */
1597         CRYPTO_gcm128_tag(&gctx->gcm, out, EVP_GCM_TLS_TAG_LEN);
1598         rv = len + EVP_GCM_TLS_EXPLICIT_IV_LEN + EVP_GCM_TLS_TAG_LEN;
1599     } else {
1600         /* Decrypt */
1601         if (gctx->ctr) {
1602             size_t bulk = 0;
1603 # if defined(AES_GCM_ASM)
1604             if (len >= 16 && AES_GCM_ASM(gctx)) {
1605                 if (CRYPTO_gcm128_decrypt(&gctx->gcm, NULL, NULL, 0))
1606                     return -1;
1607
1608                 bulk = AES_gcm_decrypt(in, out, len,
1609                                        gctx->gcm.key,
1610                                        gctx->gcm.Yi.c, gctx->gcm.Xi.u);
1611                 gctx->gcm.len.u[1] += bulk;
1612             }
1613 # endif
1614             if (CRYPTO_gcm128_decrypt_ctr32(&gctx->gcm,
1615                                             in + bulk,
1616                                             out + bulk,
1617                                             len - bulk, gctx->ctr))
1618                 goto err;
1619         } else {
1620             size_t bulk = 0;
1621 # if defined(AES_GCM_ASM2)
1622             if (len >= 16 && AES_GCM_ASM2(gctx)) {
1623                 if (CRYPTO_gcm128_decrypt(&gctx->gcm, NULL, NULL, 0))
1624                     return -1;
1625
1626                 bulk = AES_gcm_decrypt(in, out, len,
1627                                        gctx->gcm.key,
1628                                        gctx->gcm.Yi.c, gctx->gcm.Xi.u);
1629                 gctx->gcm.len.u[1] += bulk;
1630             }
1631 # endif
1632             if (CRYPTO_gcm128_decrypt(&gctx->gcm,
1633                                       in + bulk, out + bulk, len - bulk))
1634                 goto err;
1635         }
1636         /* Retrieve tag */
1637         CRYPTO_gcm128_tag(&gctx->gcm, EVP_CIPHER_CTX_buf_noconst(ctx),
1638                           EVP_GCM_TLS_TAG_LEN);
1639         /* If tag mismatch wipe buffer */
1640         if (CRYPTO_memcmp(EVP_CIPHER_CTX_buf_noconst(ctx), in + len,
1641                           EVP_GCM_TLS_TAG_LEN)) {
1642             OPENSSL_cleanse(out, len);
1643             goto err;
1644         }
1645         rv = len;
1646     }
1647
1648  err:
1649     gctx->iv_set = 0;
1650     gctx->tls_aad_len = -1;
1651     return rv;
1652 }
1653
1654 static int aes_gcm_cipher(EVP_CIPHER_CTX *ctx, unsigned char *out,
1655                           const unsigned char *in, size_t len)
1656 {
1657     EVP_AES_GCM_CTX *gctx = EVP_C_DATA(EVP_AES_GCM_CTX,ctx);
1658     /* If not set up, return error */
1659     if (!gctx->key_set)
1660         return -1;
1661
1662     if (gctx->tls_aad_len >= 0)
1663         return aes_gcm_tls_cipher(ctx, out, in, len);
1664
1665     if (!gctx->iv_set)
1666         return -1;
1667     if (in) {
1668         if (out == NULL) {
1669             if (CRYPTO_gcm128_aad(&gctx->gcm, in, len))
1670                 return -1;
1671         } else if (EVP_CIPHER_CTX_encrypting(ctx)) {
1672             if (gctx->ctr) {
1673                 size_t bulk = 0;
1674 # if defined(AES_GCM_ASM)
1675                 if (len >= 32 && AES_GCM_ASM(gctx)) {
1676                     size_t res = (16 - gctx->gcm.mres) % 16;
1677
1678                     if (CRYPTO_gcm128_encrypt(&gctx->gcm, in, out, res))
1679                         return -1;
1680
1681                     bulk = AES_gcm_encrypt(in + res,
1682                                            out + res, len - res,
1683                                            gctx->gcm.key, gctx->gcm.Yi.c,
1684                                            gctx->gcm.Xi.u);
1685                     gctx->gcm.len.u[1] += bulk;
1686                     bulk += res;
1687                 }
1688 # endif
1689                 if (CRYPTO_gcm128_encrypt_ctr32(&gctx->gcm,
1690                                                 in + bulk,
1691                                                 out + bulk,
1692                                                 len - bulk, gctx->ctr))
1693                     return -1;
1694             } else {
1695                 size_t bulk = 0;
1696 # if defined(AES_GCM_ASM2)
1697                 if (len >= 32 && AES_GCM_ASM2(gctx)) {
1698                     size_t res = (16 - gctx->gcm.mres) % 16;
1699
1700                     if (CRYPTO_gcm128_encrypt(&gctx->gcm, in, out, res))
1701                         return -1;
1702
1703                     bulk = AES_gcm_encrypt(in + res,
1704                                            out + res, len - res,
1705                                            gctx->gcm.key, gctx->gcm.Yi.c,
1706                                            gctx->gcm.Xi.u);
1707                     gctx->gcm.len.u[1] += bulk;
1708                     bulk += res;
1709                 }
1710 # endif
1711                 if (CRYPTO_gcm128_encrypt(&gctx->gcm,
1712                                           in + bulk, out + bulk, len - bulk))
1713                     return -1;
1714             }
1715         } else {
1716             if (gctx->ctr) {
1717                 size_t bulk = 0;
1718 # if defined(AES_GCM_ASM)
1719                 if (len >= 16 && AES_GCM_ASM(gctx)) {
1720                     size_t res = (16 - gctx->gcm.mres) % 16;
1721
1722                     if (CRYPTO_gcm128_decrypt(&gctx->gcm, in, out, res))
1723                         return -1;
1724
1725                     bulk = AES_gcm_decrypt(in + res,
1726                                            out + res, len - res,
1727                                            gctx->gcm.key,
1728                                            gctx->gcm.Yi.c, gctx->gcm.Xi.u);
1729                     gctx->gcm.len.u[1] += bulk;
1730                     bulk += res;
1731                 }
1732 # endif
1733                 if (CRYPTO_gcm128_decrypt_ctr32(&gctx->gcm,
1734                                                 in + bulk,
1735                                                 out + bulk,
1736                                                 len - bulk, gctx->ctr))
1737                     return -1;
1738             } else {
1739                 size_t bulk = 0;
1740 # if defined(AES_GCM_ASM2)
1741                 if (len >= 16 && AES_GCM_ASM2(gctx)) {
1742                     size_t res = (16 - gctx->gcm.mres) % 16;
1743
1744                     if (CRYPTO_gcm128_decrypt(&gctx->gcm, in, out, res))
1745                         return -1;
1746
1747                     bulk = AES_gcm_decrypt(in + res,
1748                                            out + res, len - res,
1749                                            gctx->gcm.key,
1750                                            gctx->gcm.Yi.c, gctx->gcm.Xi.u);
1751                     gctx->gcm.len.u[1] += bulk;
1752                     bulk += res;
1753                 }
1754 # endif
1755                 if (CRYPTO_gcm128_decrypt(&gctx->gcm,
1756                                           in + bulk, out + bulk, len - bulk))
1757                     return -1;
1758             }
1759         }
1760         return len;
1761     } else {
1762         if (!EVP_CIPHER_CTX_encrypting(ctx)) {
1763             if (gctx->taglen < 0)
1764                 return -1;
1765             if (CRYPTO_gcm128_finish(&gctx->gcm,
1766                                      EVP_CIPHER_CTX_buf_noconst(ctx),
1767                                      gctx->taglen) != 0)
1768                 return -1;
1769             gctx->iv_set = 0;
1770             return 0;
1771         }
1772         CRYPTO_gcm128_tag(&gctx->gcm, EVP_CIPHER_CTX_buf_noconst(ctx), 16);
1773         gctx->taglen = 16;
1774         /* Don't reuse the IV */
1775         gctx->iv_set = 0;
1776         return 0;
1777     }
1778
1779 }
1780
1781 # define CUSTOM_FLAGS    (EVP_CIPH_FLAG_DEFAULT_ASN1 \
1782                 | EVP_CIPH_CUSTOM_IV | EVP_CIPH_FLAG_CUSTOM_CIPHER \
1783                 | EVP_CIPH_ALWAYS_CALL_INIT | EVP_CIPH_CTRL_INIT \
1784                 | EVP_CIPH_CUSTOM_COPY)
1785
1786 BLOCK_CIPHER_custom(NID_aes, 128, 1, 12, gcm, GCM,
1787                     EVP_CIPH_FLAG_AEAD_CIPHER | CUSTOM_FLAGS)
1788     BLOCK_CIPHER_custom(NID_aes, 192, 1, 12, gcm, GCM,
1789                     EVP_CIPH_FLAG_AEAD_CIPHER | CUSTOM_FLAGS)
1790     BLOCK_CIPHER_custom(NID_aes, 256, 1, 12, gcm, GCM,
1791                     EVP_CIPH_FLAG_AEAD_CIPHER | CUSTOM_FLAGS)
1792
1793 static int aes_xts_ctrl(EVP_CIPHER_CTX *c, int type, int arg, void *ptr)
1794 {
1795     EVP_AES_XTS_CTX *xctx = EVP_C_DATA(EVP_AES_XTS_CTX,c);
1796     if (type == EVP_CTRL_COPY) {
1797         EVP_CIPHER_CTX *out = ptr;
1798         EVP_AES_XTS_CTX *xctx_out = EVP_C_DATA(EVP_AES_XTS_CTX,out);
1799         if (xctx->xts.key1) {
1800             if (xctx->xts.key1 != &xctx->ks1)
1801                 return 0;
1802             xctx_out->xts.key1 = &xctx_out->ks1;
1803         }
1804         if (xctx->xts.key2) {
1805             if (xctx->xts.key2 != &xctx->ks2)
1806                 return 0;
1807             xctx_out->xts.key2 = &xctx_out->ks2;
1808         }
1809         return 1;
1810     } else if (type != EVP_CTRL_INIT)
1811         return -1;
1812     /* key1 and key2 are used as an indicator both key and IV are set */
1813     xctx->xts.key1 = NULL;
1814     xctx->xts.key2 = NULL;
1815     return 1;
1816 }
1817
1818 static int aes_xts_init_key(EVP_CIPHER_CTX *ctx, const unsigned char *key,
1819                             const unsigned char *iv, int enc)
1820 {
1821     EVP_AES_XTS_CTX *xctx = EVP_C_DATA(EVP_AES_XTS_CTX,ctx);
1822     if (!iv && !key)
1823         return 1;
1824
1825     if (key)
1826         do {
1827 # ifdef AES_XTS_ASM
1828             xctx->stream = enc ? AES_xts_encrypt : AES_xts_decrypt;
1829 # else
1830             xctx->stream = NULL;
1831 # endif
1832             /* key_len is two AES keys */
1833 # ifdef HWAES_CAPABLE
1834             if (HWAES_CAPABLE) {
1835                 if (enc) {
1836                     HWAES_set_encrypt_key(key,
1837                                           EVP_CIPHER_CTX_key_length(ctx) * 4,
1838                                           &xctx->ks1.ks);
1839                     xctx->xts.block1 = (block128_f) HWAES_encrypt;
1840                 } else {
1841                     HWAES_set_decrypt_key(key,
1842                                           EVP_CIPHER_CTX_key_length(ctx) * 4,
1843                                           &xctx->ks1.ks);
1844                     xctx->xts.block1 = (block128_f) HWAES_decrypt;
1845                 }
1846
1847                 HWAES_set_encrypt_key(key + EVP_CIPHER_CTX_key_length(ctx) / 2,
1848                                       EVP_CIPHER_CTX_key_length(ctx) * 4,
1849                                       &xctx->ks2.ks);
1850                 xctx->xts.block2 = (block128_f) HWAES_encrypt;
1851
1852                 xctx->xts.key1 = &xctx->ks1;
1853                 break;
1854             } else
1855 # endif
1856 # ifdef BSAES_CAPABLE
1857             if (BSAES_CAPABLE)
1858                 xctx->stream = enc ? bsaes_xts_encrypt : bsaes_xts_decrypt;
1859             else
1860 # endif
1861 # ifdef VPAES_CAPABLE
1862             if (VPAES_CAPABLE) {
1863                 if (enc) {
1864                     vpaes_set_encrypt_key(key,
1865                                           EVP_CIPHER_CTX_key_length(ctx) * 4,
1866                                           &xctx->ks1.ks);
1867                     xctx->xts.block1 = (block128_f) vpaes_encrypt;
1868                 } else {
1869                     vpaes_set_decrypt_key(key,
1870                                           EVP_CIPHER_CTX_key_length(ctx) * 4,
1871                                           &xctx->ks1.ks);
1872                     xctx->xts.block1 = (block128_f) vpaes_decrypt;
1873                 }
1874
1875                 vpaes_set_encrypt_key(key + EVP_CIPHER_CTX_key_length(ctx) / 2,
1876                                       EVP_CIPHER_CTX_key_length(ctx) * 4,
1877                                       &xctx->ks2.ks);
1878                 xctx->xts.block2 = (block128_f) vpaes_encrypt;
1879
1880                 xctx->xts.key1 = &xctx->ks1;
1881                 break;
1882             } else
1883 # endif
1884                 (void)0;        /* terminate potentially open 'else' */
1885
1886             if (enc) {
1887                 AES_set_encrypt_key(key, EVP_CIPHER_CTX_key_length(ctx) * 4,
1888                                     &xctx->ks1.ks);
1889                 xctx->xts.block1 = (block128_f) AES_encrypt;
1890             } else {
1891                 AES_set_decrypt_key(key, EVP_CIPHER_CTX_key_length(ctx) * 4,
1892                                     &xctx->ks1.ks);
1893                 xctx->xts.block1 = (block128_f) AES_decrypt;
1894             }
1895
1896             AES_set_encrypt_key(key + EVP_CIPHER_CTX_key_length(ctx) / 2,
1897                                 EVP_CIPHER_CTX_key_length(ctx) * 4,
1898                                 &xctx->ks2.ks);
1899             xctx->xts.block2 = (block128_f) AES_encrypt;
1900
1901             xctx->xts.key1 = &xctx->ks1;
1902         } while (0);
1903
1904     if (iv) {
1905         xctx->xts.key2 = &xctx->ks2;
1906         memcpy(EVP_CIPHER_CTX_iv_noconst(ctx), iv, 16);
1907     }
1908
1909     return 1;
1910 }
1911
1912 static int aes_xts_cipher(EVP_CIPHER_CTX *ctx, unsigned char *out,
1913                           const unsigned char *in, size_t len)
1914 {
1915     EVP_AES_XTS_CTX *xctx = EVP_C_DATA(EVP_AES_XTS_CTX,ctx);
1916     if (!xctx->xts.key1 || !xctx->xts.key2)
1917         return 0;
1918     if (!out || !in || len < AES_BLOCK_SIZE)
1919         return 0;
1920     if (xctx->stream)
1921         (*xctx->stream) (in, out, len,
1922                          xctx->xts.key1, xctx->xts.key2,
1923                          EVP_CIPHER_CTX_iv_noconst(ctx));
1924     else if (CRYPTO_xts128_encrypt(&xctx->xts, EVP_CIPHER_CTX_iv_noconst(ctx),
1925                                    in, out, len,
1926                                    EVP_CIPHER_CTX_encrypting(ctx)))
1927         return 0;
1928     return 1;
1929 }
1930
1931 # define aes_xts_cleanup NULL
1932
1933 # define XTS_FLAGS       (EVP_CIPH_FLAG_DEFAULT_ASN1 | EVP_CIPH_CUSTOM_IV \
1934                          | EVP_CIPH_ALWAYS_CALL_INIT | EVP_CIPH_CTRL_INIT \
1935                          | EVP_CIPH_CUSTOM_COPY)
1936
1937 BLOCK_CIPHER_custom(NID_aes, 128, 1, 16, xts, XTS, XTS_FLAGS)
1938     BLOCK_CIPHER_custom(NID_aes, 256, 1, 16, xts, XTS, XTS_FLAGS)
1939
1940 static int aes_ccm_ctrl(EVP_CIPHER_CTX *c, int type, int arg, void *ptr)
1941 {
1942     EVP_AES_CCM_CTX *cctx = EVP_C_DATA(EVP_AES_CCM_CTX,c);
1943     switch (type) {
1944     case EVP_CTRL_INIT:
1945         cctx->key_set = 0;
1946         cctx->iv_set = 0;
1947         cctx->L = 8;
1948         cctx->M = 12;
1949         cctx->tag_set = 0;
1950         cctx->len_set = 0;
1951         cctx->tls_aad_len = -1;
1952         return 1;
1953
1954     case EVP_CTRL_AEAD_TLS1_AAD:
1955         /* Save the AAD for later use */
1956         if (arg != EVP_AEAD_TLS1_AAD_LEN)
1957             return 0;
1958         memcpy(EVP_CIPHER_CTX_buf_noconst(c), ptr, arg);
1959         cctx->tls_aad_len = arg;
1960         {
1961             uint16_t len =
1962                 EVP_CIPHER_CTX_buf_noconst(c)[arg - 2] << 8
1963                 | EVP_CIPHER_CTX_buf_noconst(c)[arg - 1];
1964             /* Correct length for explicit IV */
1965             len -= EVP_CCM_TLS_EXPLICIT_IV_LEN;
1966             /* If decrypting correct for tag too */
1967             if (!EVP_CIPHER_CTX_encrypting(c))
1968                 len -= cctx->M;
1969             EVP_CIPHER_CTX_buf_noconst(c)[arg - 2] = len >> 8;
1970             EVP_CIPHER_CTX_buf_noconst(c)[arg - 1] = len & 0xff;
1971         }
1972         /* Extra padding: tag appended to record */
1973         return cctx->M;
1974
1975     case EVP_CTRL_CCM_SET_IV_FIXED:
1976         /* Sanity check length */
1977         if (arg != EVP_CCM_TLS_FIXED_IV_LEN)
1978             return 0;
1979         /* Just copy to first part of IV */
1980         memcpy(EVP_CIPHER_CTX_iv_noconst(c), ptr, arg);
1981         return 1;
1982
1983     case EVP_CTRL_AEAD_SET_IVLEN:
1984         arg = 15 - arg;
1985     case EVP_CTRL_CCM_SET_L:
1986         if (arg < 2 || arg > 8)
1987             return 0;
1988         cctx->L = arg;
1989         return 1;
1990
1991     case EVP_CTRL_AEAD_SET_TAG:
1992         if ((arg & 1) || arg < 4 || arg > 16)
1993             return 0;
1994         if (EVP_CIPHER_CTX_encrypting(c) && ptr)
1995             return 0;
1996         if (ptr) {
1997             cctx->tag_set = 1;
1998             memcpy(EVP_CIPHER_CTX_buf_noconst(c), ptr, arg);
1999         }
2000         cctx->M = arg;
2001         return 1;
2002
2003     case EVP_CTRL_AEAD_GET_TAG:
2004         if (!EVP_CIPHER_CTX_encrypting(c) || !cctx->tag_set)
2005             return 0;
2006         if (!CRYPTO_ccm128_tag(&cctx->ccm, ptr, (size_t)arg))
2007             return 0;
2008         cctx->tag_set = 0;
2009         cctx->iv_set = 0;
2010         cctx->len_set = 0;
2011         return 1;
2012
2013     case EVP_CTRL_COPY:
2014         {
2015             EVP_CIPHER_CTX *out = ptr;
2016             EVP_AES_CCM_CTX *cctx_out = EVP_C_DATA(EVP_AES_CCM_CTX,out);
2017             if (cctx->ccm.key) {
2018                 if (cctx->ccm.key != &cctx->ks)
2019                     return 0;
2020                 cctx_out->ccm.key = &cctx_out->ks;
2021             }
2022             return 1;
2023         }
2024
2025     default:
2026         return -1;
2027
2028     }
2029 }
2030
2031 static int aes_ccm_init_key(EVP_CIPHER_CTX *ctx, const unsigned char *key,
2032                             const unsigned char *iv, int enc)
2033 {
2034     EVP_AES_CCM_CTX *cctx = EVP_C_DATA(EVP_AES_CCM_CTX,ctx);
2035     if (!iv && !key)
2036         return 1;
2037     if (key)
2038         do {
2039 # ifdef HWAES_CAPABLE
2040             if (HWAES_CAPABLE) {
2041                 HWAES_set_encrypt_key(key, EVP_CIPHER_CTX_key_length(ctx) * 8,
2042                                       &cctx->ks.ks);
2043
2044                 CRYPTO_ccm128_init(&cctx->ccm, cctx->M, cctx->L,
2045                                    &cctx->ks, (block128_f) HWAES_encrypt);
2046                 cctx->str = NULL;
2047                 cctx->key_set = 1;
2048                 break;
2049             } else
2050 # endif
2051 # ifdef VPAES_CAPABLE
2052             if (VPAES_CAPABLE) {
2053                 vpaes_set_encrypt_key(key, EVP_CIPHER_CTX_key_length(ctx) * 8,
2054                                       &cctx->ks.ks);
2055                 CRYPTO_ccm128_init(&cctx->ccm, cctx->M, cctx->L,
2056                                    &cctx->ks, (block128_f) vpaes_encrypt);
2057                 cctx->str = NULL;
2058                 cctx->key_set = 1;
2059                 break;
2060             }
2061 # endif
2062             AES_set_encrypt_key(key, EVP_CIPHER_CTX_key_length(ctx) * 8,
2063                                 &cctx->ks.ks);
2064             CRYPTO_ccm128_init(&cctx->ccm, cctx->M, cctx->L,
2065                                &cctx->ks, (block128_f) AES_encrypt);
2066             cctx->str = NULL;
2067             cctx->key_set = 1;
2068         } while (0);
2069     if (iv) {
2070         memcpy(EVP_CIPHER_CTX_iv_noconst(ctx), iv, 15 - cctx->L);
2071         cctx->iv_set = 1;
2072     }
2073     return 1;
2074 }
2075
2076 static int aes_ccm_tls_cipher(EVP_CIPHER_CTX *ctx, unsigned char *out,
2077                               const unsigned char *in, size_t len)
2078 {
2079     EVP_AES_CCM_CTX *cctx = EVP_C_DATA(EVP_AES_CCM_CTX,ctx);
2080     CCM128_CONTEXT *ccm = &cctx->ccm;
2081     /* Encrypt/decrypt must be performed in place */
2082     if (out != in || len < (EVP_CCM_TLS_EXPLICIT_IV_LEN + (size_t)cctx->M))
2083         return -1;
2084     /* If encrypting set explicit IV from sequence number (start of AAD) */
2085     if (EVP_CIPHER_CTX_encrypting(ctx))
2086         memcpy(out, EVP_CIPHER_CTX_buf_noconst(ctx),
2087                EVP_CCM_TLS_EXPLICIT_IV_LEN);
2088     /* Get rest of IV from explicit IV */
2089     memcpy(EVP_CIPHER_CTX_iv_noconst(ctx) + EVP_CCM_TLS_FIXED_IV_LEN, in,
2090            EVP_CCM_TLS_EXPLICIT_IV_LEN);
2091     /* Correct length value */
2092     len -= EVP_CCM_TLS_EXPLICIT_IV_LEN + cctx->M;
2093     if (CRYPTO_ccm128_setiv(ccm, EVP_CIPHER_CTX_iv_noconst(ctx), 15 - cctx->L,
2094                             len))
2095             return -1;
2096     /* Use saved AAD */
2097     CRYPTO_ccm128_aad(ccm, EVP_CIPHER_CTX_buf_noconst(ctx), cctx->tls_aad_len);
2098     /* Fix buffer to point to payload */
2099     in += EVP_CCM_TLS_EXPLICIT_IV_LEN;
2100     out += EVP_CCM_TLS_EXPLICIT_IV_LEN;
2101     if (EVP_CIPHER_CTX_encrypting(ctx)) {
2102         if (cctx->str ? CRYPTO_ccm128_encrypt_ccm64(ccm, in, out, len,
2103                                                     cctx->str) :
2104             CRYPTO_ccm128_encrypt(ccm, in, out, len))
2105             return -1;
2106         if (!CRYPTO_ccm128_tag(ccm, out + len, cctx->M))
2107             return -1;
2108         return len + EVP_CCM_TLS_EXPLICIT_IV_LEN + cctx->M;
2109     } else {
2110         if (cctx->str ? !CRYPTO_ccm128_decrypt_ccm64(ccm, in, out, len,
2111                                                      cctx->str) :
2112             !CRYPTO_ccm128_decrypt(ccm, in, out, len)) {
2113             unsigned char tag[16];
2114             if (CRYPTO_ccm128_tag(ccm, tag, cctx->M)) {
2115                 if (!CRYPTO_memcmp(tag, in + len, cctx->M))
2116                     return len;
2117             }
2118         }
2119         OPENSSL_cleanse(out, len);
2120         return -1;
2121     }
2122 }
2123
2124 static int aes_ccm_cipher(EVP_CIPHER_CTX *ctx, unsigned char *out,
2125                           const unsigned char *in, size_t len)
2126 {
2127     EVP_AES_CCM_CTX *cctx = EVP_C_DATA(EVP_AES_CCM_CTX,ctx);
2128     CCM128_CONTEXT *ccm = &cctx->ccm;
2129     /* If not set up, return error */
2130     if (!cctx->key_set)
2131         return -1;
2132
2133     if (cctx->tls_aad_len >= 0)
2134         return aes_ccm_tls_cipher(ctx, out, in, len);
2135
2136     if (!cctx->iv_set)
2137         return -1;
2138
2139     if (!EVP_CIPHER_CTX_encrypting(ctx) && !cctx->tag_set)
2140         return -1;
2141     if (!out) {
2142         if (!in) {
2143             if (CRYPTO_ccm128_setiv(ccm, EVP_CIPHER_CTX_iv_noconst(ctx),
2144                                     15 - cctx->L, len))
2145                 return -1;
2146             cctx->len_set = 1;
2147             return len;
2148         }
2149         /* If have AAD need message length */
2150         if (!cctx->len_set && len)
2151             return -1;
2152         CRYPTO_ccm128_aad(ccm, in, len);
2153         return len;
2154     }
2155     /* EVP_*Final() doesn't return any data */
2156     if (!in)
2157         return 0;
2158     /* If not set length yet do it */
2159     if (!cctx->len_set) {
2160         if (CRYPTO_ccm128_setiv(ccm, EVP_CIPHER_CTX_iv_noconst(ctx),
2161                                 15 - cctx->L, len))
2162             return -1;
2163         cctx->len_set = 1;
2164     }
2165     if (EVP_CIPHER_CTX_encrypting(ctx)) {
2166         if (cctx->str ? CRYPTO_ccm128_encrypt_ccm64(ccm, in, out, len,
2167                                                     cctx->str) :
2168             CRYPTO_ccm128_encrypt(ccm, in, out, len))
2169             return -1;
2170         cctx->tag_set = 1;
2171         return len;
2172     } else {
2173         int rv = -1;
2174         if (cctx->str ? !CRYPTO_ccm128_decrypt_ccm64(ccm, in, out, len,
2175                                                      cctx->str) :
2176             !CRYPTO_ccm128_decrypt(ccm, in, out, len)) {
2177             unsigned char tag[16];
2178             if (CRYPTO_ccm128_tag(ccm, tag, cctx->M)) {
2179                 if (!CRYPTO_memcmp(tag, EVP_CIPHER_CTX_buf_noconst(ctx),
2180                                    cctx->M))
2181                     rv = len;
2182             }
2183         }
2184         if (rv == -1)
2185             OPENSSL_cleanse(out, len);
2186         cctx->iv_set = 0;
2187         cctx->tag_set = 0;
2188         cctx->len_set = 0;
2189         return rv;
2190     }
2191 }
2192
2193 # define aes_ccm_cleanup NULL
2194
2195 BLOCK_CIPHER_custom(NID_aes, 128, 1, 12, ccm, CCM,
2196                     EVP_CIPH_FLAG_AEAD_CIPHER | CUSTOM_FLAGS)
2197     BLOCK_CIPHER_custom(NID_aes, 192, 1, 12, ccm, CCM,
2198                         EVP_CIPH_FLAG_AEAD_CIPHER | CUSTOM_FLAGS)
2199     BLOCK_CIPHER_custom(NID_aes, 256, 1, 12, ccm, CCM,
2200                         EVP_CIPH_FLAG_AEAD_CIPHER | CUSTOM_FLAGS)
2201
2202 typedef struct {
2203     union {
2204         double align;
2205         AES_KEY ks;
2206     } ks;
2207     /* Indicates if IV has been set */
2208     unsigned char *iv;
2209 } EVP_AES_WRAP_CTX;
2210
2211 static int aes_wrap_init_key(EVP_CIPHER_CTX *ctx, const unsigned char *key,
2212                              const unsigned char *iv, int enc)
2213 {
2214     EVP_AES_WRAP_CTX *wctx = EVP_C_DATA(EVP_AES_WRAP_CTX,ctx);
2215     if (!iv && !key)
2216         return 1;
2217     if (key) {
2218         if (EVP_CIPHER_CTX_encrypting(ctx))
2219             AES_set_encrypt_key(key, EVP_CIPHER_CTX_key_length(ctx) * 8,
2220                                 &wctx->ks.ks);
2221         else
2222             AES_set_decrypt_key(key, EVP_CIPHER_CTX_key_length(ctx) * 8,
2223                                 &wctx->ks.ks);
2224         if (!iv)
2225             wctx->iv = NULL;
2226     }
2227     if (iv) {
2228         memcpy(EVP_CIPHER_CTX_iv_noconst(ctx), iv, EVP_CIPHER_CTX_iv_length(ctx));
2229         wctx->iv = EVP_CIPHER_CTX_iv_noconst(ctx);
2230     }
2231     return 1;
2232 }
2233
2234 static int aes_wrap_cipher(EVP_CIPHER_CTX *ctx, unsigned char *out,
2235                            const unsigned char *in, size_t inlen)
2236 {
2237     EVP_AES_WRAP_CTX *wctx = EVP_C_DATA(EVP_AES_WRAP_CTX,ctx);
2238     size_t rv;
2239     /* AES wrap with padding has IV length of 4, without padding 8 */
2240     int pad = EVP_CIPHER_CTX_iv_length(ctx) == 4;
2241     /* No final operation so always return zero length */
2242     if (!in)
2243         return 0;
2244     /* Input length must always be non-zero */
2245     if (!inlen)
2246         return -1;
2247     /* If decrypting need at least 16 bytes and multiple of 8 */
2248     if (!EVP_CIPHER_CTX_encrypting(ctx) && (inlen < 16 || inlen & 0x7))
2249         return -1;
2250     /* If not padding input must be multiple of 8 */
2251     if (!pad && inlen & 0x7)
2252         return -1;
2253     if (!out) {
2254         if (EVP_CIPHER_CTX_encrypting(ctx)) {
2255             /* If padding round up to multiple of 8 */
2256             if (pad)
2257                 inlen = (inlen + 7) / 8 * 8;
2258             /* 8 byte prefix */
2259             return inlen + 8;
2260         } else {
2261             /*
2262              * If not padding output will be exactly 8 bytes smaller than
2263              * input. If padding it will be at least 8 bytes smaller but we
2264              * don't know how much.
2265              */
2266             return inlen - 8;
2267         }
2268     }
2269     if (pad) {
2270         if (EVP_CIPHER_CTX_encrypting(ctx))
2271             rv = CRYPTO_128_wrap_pad(&wctx->ks.ks, wctx->iv,
2272                                      out, in, inlen,
2273                                      (block128_f) AES_encrypt);
2274         else
2275             rv = CRYPTO_128_unwrap_pad(&wctx->ks.ks, wctx->iv,
2276                                        out, in, inlen,
2277                                        (block128_f) AES_decrypt);
2278     } else {
2279         if (EVP_CIPHER_CTX_encrypting(ctx))
2280             rv = CRYPTO_128_wrap(&wctx->ks.ks, wctx->iv,
2281                                  out, in, inlen, (block128_f) AES_encrypt);
2282         else
2283             rv = CRYPTO_128_unwrap(&wctx->ks.ks, wctx->iv,
2284                                    out, in, inlen, (block128_f) AES_decrypt);
2285     }
2286     return rv ? (int)rv : -1;
2287 }
2288
2289 # define WRAP_FLAGS      (EVP_CIPH_WRAP_MODE \
2290                 | EVP_CIPH_CUSTOM_IV | EVP_CIPH_FLAG_CUSTOM_CIPHER \
2291                 | EVP_CIPH_ALWAYS_CALL_INIT | EVP_CIPH_FLAG_DEFAULT_ASN1)
2292
2293 static const EVP_CIPHER aes_128_wrap = {
2294     NID_id_aes128_wrap,
2295     8, 16, 8, WRAP_FLAGS,
2296     aes_wrap_init_key, aes_wrap_cipher,
2297     NULL,
2298     sizeof(EVP_AES_WRAP_CTX),
2299     NULL, NULL, NULL, NULL
2300 };
2301
2302 const EVP_CIPHER *EVP_aes_128_wrap(void)
2303 {
2304     return &aes_128_wrap;
2305 }
2306
2307 static const EVP_CIPHER aes_192_wrap = {
2308     NID_id_aes192_wrap,
2309     8, 24, 8, WRAP_FLAGS,
2310     aes_wrap_init_key, aes_wrap_cipher,
2311     NULL,
2312     sizeof(EVP_AES_WRAP_CTX),
2313     NULL, NULL, NULL, NULL
2314 };
2315
2316 const EVP_CIPHER *EVP_aes_192_wrap(void)
2317 {
2318     return &aes_192_wrap;
2319 }
2320
2321 static const EVP_CIPHER aes_256_wrap = {
2322     NID_id_aes256_wrap,
2323     8, 32, 8, WRAP_FLAGS,
2324     aes_wrap_init_key, aes_wrap_cipher,
2325     NULL,
2326     sizeof(EVP_AES_WRAP_CTX),
2327     NULL, NULL, NULL, NULL
2328 };
2329
2330 const EVP_CIPHER *EVP_aes_256_wrap(void)
2331 {
2332     return &aes_256_wrap;
2333 }
2334
2335 static const EVP_CIPHER aes_128_wrap_pad = {
2336     NID_id_aes128_wrap_pad,
2337     8, 16, 4, WRAP_FLAGS,
2338     aes_wrap_init_key, aes_wrap_cipher,
2339     NULL,
2340     sizeof(EVP_AES_WRAP_CTX),
2341     NULL, NULL, NULL, NULL
2342 };
2343
2344 const EVP_CIPHER *EVP_aes_128_wrap_pad(void)
2345 {
2346     return &aes_128_wrap_pad;
2347 }
2348
2349 static const EVP_CIPHER aes_192_wrap_pad = {
2350     NID_id_aes192_wrap_pad,
2351     8, 24, 4, WRAP_FLAGS,
2352     aes_wrap_init_key, aes_wrap_cipher,
2353     NULL,
2354     sizeof(EVP_AES_WRAP_CTX),
2355     NULL, NULL, NULL, NULL
2356 };
2357
2358 const EVP_CIPHER *EVP_aes_192_wrap_pad(void)
2359 {
2360     return &aes_192_wrap_pad;
2361 }
2362
2363 static const EVP_CIPHER aes_256_wrap_pad = {
2364     NID_id_aes256_wrap_pad,
2365     8, 32, 4, WRAP_FLAGS,
2366     aes_wrap_init_key, aes_wrap_cipher,
2367     NULL,
2368     sizeof(EVP_AES_WRAP_CTX),
2369     NULL, NULL, NULL, NULL
2370 };
2371
2372 const EVP_CIPHER *EVP_aes_256_wrap_pad(void)
2373 {
2374     return &aes_256_wrap_pad;
2375 }
2376
2377 # ifndef OPENSSL_NO_OCB
2378 static int aes_ocb_ctrl(EVP_CIPHER_CTX *c, int type, int arg, void *ptr)
2379 {
2380     EVP_AES_OCB_CTX *octx = EVP_C_DATA(EVP_AES_OCB_CTX,c);
2381     EVP_CIPHER_CTX *newc;
2382     EVP_AES_OCB_CTX *new_octx;
2383
2384     switch (type) {
2385     case EVP_CTRL_INIT:
2386         octx->key_set = 0;
2387         octx->iv_set = 0;
2388         octx->ivlen = EVP_CIPHER_CTX_iv_length(c);
2389         octx->iv = EVP_CIPHER_CTX_iv_noconst(c);
2390         octx->taglen = 16;
2391         octx->data_buf_len = 0;
2392         octx->aad_buf_len = 0;
2393         return 1;
2394
2395     case EVP_CTRL_AEAD_SET_IVLEN:
2396         /* IV len must be 1 to 15 */
2397         if (arg <= 0 || arg > 15)
2398             return 0;
2399
2400         octx->ivlen = arg;
2401         return 1;
2402
2403     case EVP_CTRL_AEAD_SET_TAG:
2404         if (!ptr) {
2405             /* Tag len must be 0 to 16 */
2406             if (arg < 0 || arg > 16)
2407                 return 0;
2408
2409             octx->taglen = arg;
2410             return 1;
2411         }
2412         if (arg != octx->taglen || EVP_CIPHER_CTX_encrypting(c))
2413             return 0;
2414         memcpy(octx->tag, ptr, arg);
2415         return 1;
2416
2417     case EVP_CTRL_AEAD_GET_TAG:
2418         if (arg != octx->taglen || !EVP_CIPHER_CTX_encrypting(c))
2419             return 0;
2420
2421         memcpy(ptr, octx->tag, arg);
2422         return 1;
2423
2424     case EVP_CTRL_COPY:
2425         newc = (EVP_CIPHER_CTX *)ptr;
2426         new_octx = EVP_C_DATA(EVP_AES_OCB_CTX,newc);
2427         return CRYPTO_ocb128_copy_ctx(&new_octx->ocb, &octx->ocb,
2428                                       &new_octx->ksenc.ks,
2429                                       &new_octx->ksdec.ks);
2430
2431     default:
2432         return -1;
2433
2434     }
2435 }
2436
2437 #  ifdef HWAES_CAPABLE
2438 #   ifdef HWAES_ocb_encrypt
2439 void HWAES_ocb_encrypt(const unsigned char *in, unsigned char *out,
2440                        size_t blocks, const void *key,
2441                        size_t start_block_num,
2442                        unsigned char offset_i[16],
2443                        const unsigned char L_[][16],
2444                        unsigned char checksum[16]);
2445 #   else
2446 #     define HWAES_ocb_encrypt NULL
2447 #   endif
2448 #   ifdef HWAES_ocb_decrypt
2449 void HWAES_ocb_decrypt(const unsigned char *in, unsigned char *out,
2450                        size_t blocks, const void *key,
2451                        size_t start_block_num,
2452                        unsigned char offset_i[16],
2453                        const unsigned char L_[][16],
2454                        unsigned char checksum[16]);
2455 #   else
2456 #     define HWAES_ocb_decrypt NULL
2457 #   endif
2458 #  endif
2459
2460 static int aes_ocb_init_key(EVP_CIPHER_CTX *ctx, const unsigned char *key,
2461                             const unsigned char *iv, int enc)
2462 {
2463     EVP_AES_OCB_CTX *octx = EVP_C_DATA(EVP_AES_OCB_CTX,ctx);
2464     if (!iv && !key)
2465         return 1;
2466     if (key) {
2467         do {
2468             /*
2469              * We set both the encrypt and decrypt key here because decrypt
2470              * needs both. We could possibly optimise to remove setting the
2471              * decrypt for an encryption operation.
2472              */
2473 #  ifdef HWAES_CAPABLE
2474             if (HWAES_CAPABLE) {
2475                 HWAES_set_encrypt_key(key, EVP_CIPHER_CTX_key_length(ctx) * 8,
2476                                       &octx->ksenc.ks);
2477                 HWAES_set_decrypt_key(key, EVP_CIPHER_CTX_key_length(ctx) * 8,
2478                                       &octx->ksdec.ks);
2479                 if (!CRYPTO_ocb128_init(&octx->ocb,
2480                                         &octx->ksenc.ks, &octx->ksdec.ks,
2481                                         (block128_f) HWAES_encrypt,
2482                                         (block128_f) HWAES_decrypt,
2483                                         enc ? HWAES_ocb_encrypt
2484                                             : HWAES_ocb_decrypt))
2485                     return 0;
2486                 break;
2487             }
2488 #  endif
2489 #  ifdef VPAES_CAPABLE
2490             if (VPAES_CAPABLE) {
2491                 vpaes_set_encrypt_key(key, EVP_CIPHER_CTX_key_length(ctx) * 8,
2492                                       &octx->ksenc.ks);
2493                 vpaes_set_decrypt_key(key, EVP_CIPHER_CTX_key_length(ctx) * 8,
2494                                       &octx->ksdec.ks);
2495                 if (!CRYPTO_ocb128_init(&octx->ocb,
2496                                         &octx->ksenc.ks, &octx->ksdec.ks,
2497                                         (block128_f) vpaes_encrypt,
2498                                         (block128_f) vpaes_decrypt,
2499                                         NULL))
2500                     return 0;
2501                 break;
2502             }
2503 #  endif
2504             AES_set_encrypt_key(key, EVP_CIPHER_CTX_key_length(ctx) * 8,
2505                                 &octx->ksenc.ks);
2506             AES_set_decrypt_key(key, EVP_CIPHER_CTX_key_length(ctx) * 8,
2507                                 &octx->ksdec.ks);
2508             if (!CRYPTO_ocb128_init(&octx->ocb,
2509                                     &octx->ksenc.ks, &octx->ksdec.ks,
2510                                     (block128_f) AES_encrypt,
2511                                     (block128_f) AES_decrypt,
2512                                     NULL))
2513                 return 0;
2514         }
2515         while (0);
2516
2517         /*
2518          * If we have an iv we can set it directly, otherwise use saved IV.
2519          */
2520         if (iv == NULL && octx->iv_set)
2521             iv = octx->iv;
2522         if (iv) {
2523             if (CRYPTO_ocb128_setiv(&octx->ocb, iv, octx->ivlen, octx->taglen)
2524                 != 1)
2525                 return 0;
2526             octx->iv_set = 1;
2527         }
2528         octx->key_set = 1;
2529     } else {
2530         /* If key set use IV, otherwise copy */
2531         if (octx->key_set)
2532             CRYPTO_ocb128_setiv(&octx->ocb, iv, octx->ivlen, octx->taglen);
2533         else
2534             memcpy(octx->iv, iv, octx->ivlen);
2535         octx->iv_set = 1;
2536     }
2537     return 1;
2538 }
2539
2540 static int aes_ocb_cipher(EVP_CIPHER_CTX *ctx, unsigned char *out,
2541                           const unsigned char *in, size_t len)
2542 {
2543     unsigned char *buf;
2544     int *buf_len;
2545     int written_len = 0;
2546     size_t trailing_len;
2547     EVP_AES_OCB_CTX *octx = EVP_C_DATA(EVP_AES_OCB_CTX,ctx);
2548
2549     /* If IV or Key not set then return error */
2550     if (!octx->iv_set)
2551         return -1;
2552
2553     if (!octx->key_set)
2554         return -1;
2555
2556     if (in) {
2557         /*
2558          * Need to ensure we are only passing full blocks to low level OCB
2559          * routines. We do it here rather than in EVP_EncryptUpdate/
2560          * EVP_DecryptUpdate because we need to pass full blocks of AAD too
2561          * and those routines don't support that
2562          */
2563
2564         /* Are we dealing with AAD or normal data here? */
2565         if (out == NULL) {
2566             buf = octx->aad_buf;
2567             buf_len = &(octx->aad_buf_len);
2568         } else {
2569             buf = octx->data_buf;
2570             buf_len = &(octx->data_buf_len);
2571         }
2572
2573         /*
2574          * If we've got a partially filled buffer from a previous call then
2575          * use that data first
2576          */
2577         if (*buf_len) {
2578             unsigned int remaining;
2579
2580             remaining = 16 - (*buf_len);
2581             if (remaining > len) {
2582                 memcpy(buf + (*buf_len), in, len);
2583                 *(buf_len) += len;
2584                 return 0;
2585             }
2586             memcpy(buf + (*buf_len), in, remaining);
2587
2588             /*
2589              * If we get here we've filled the buffer, so process it
2590              */
2591             len -= remaining;
2592             in += remaining;
2593             if (out == NULL) {
2594                 if (!CRYPTO_ocb128_aad(&octx->ocb, buf, 16))
2595                     return -1;
2596             } else if (EVP_CIPHER_CTX_encrypting(ctx)) {
2597                 if (!CRYPTO_ocb128_encrypt(&octx->ocb, buf, out, 16))
2598                     return -1;
2599             } else {
2600                 if (!CRYPTO_ocb128_decrypt(&octx->ocb, buf, out, 16))
2601                     return -1;
2602             }
2603             written_len = 16;
2604             *buf_len = 0;
2605         }
2606
2607         /* Do we have a partial block to handle at the end? */
2608         trailing_len = len % 16;
2609
2610         /*
2611          * If we've got some full blocks to handle, then process these first
2612          */
2613         if (len != trailing_len) {
2614             if (out == NULL) {
2615                 if (!CRYPTO_ocb128_aad(&octx->ocb, in, len - trailing_len))
2616                     return -1;
2617             } else if (EVP_CIPHER_CTX_encrypting(ctx)) {
2618                 if (!CRYPTO_ocb128_encrypt
2619                     (&octx->ocb, in, out, len - trailing_len))
2620                     return -1;
2621             } else {
2622                 if (!CRYPTO_ocb128_decrypt
2623                     (&octx->ocb, in, out, len - trailing_len))
2624                     return -1;
2625             }
2626             written_len += len - trailing_len;
2627             in += len - trailing_len;
2628         }
2629
2630         /* Handle any trailing partial block */
2631         if (trailing_len) {
2632             memcpy(buf, in, trailing_len);
2633             *buf_len = trailing_len;
2634         }
2635
2636         return written_len;
2637     } else {
2638         /*
2639          * First of all empty the buffer of any partial block that we might
2640          * have been provided - both for data and AAD
2641          */
2642         if (octx->data_buf_len) {
2643             if (EVP_CIPHER_CTX_encrypting(ctx)) {
2644                 if (!CRYPTO_ocb128_encrypt(&octx->ocb, octx->data_buf, out,
2645                                            octx->data_buf_len))
2646                     return -1;
2647             } else {
2648                 if (!CRYPTO_ocb128_decrypt(&octx->ocb, octx->data_buf, out,
2649                                            octx->data_buf_len))
2650                     return -1;
2651             }
2652             written_len = octx->data_buf_len;
2653             octx->data_buf_len = 0;
2654         }
2655         if (octx->aad_buf_len) {
2656             if (!CRYPTO_ocb128_aad
2657                 (&octx->ocb, octx->aad_buf, octx->aad_buf_len))
2658                 return -1;
2659             octx->aad_buf_len = 0;
2660         }
2661         /* If decrypting then verify */
2662         if (!EVP_CIPHER_CTX_encrypting(ctx)) {
2663             if (octx->taglen < 0)
2664                 return -1;
2665             if (CRYPTO_ocb128_finish(&octx->ocb,
2666                                      octx->tag, octx->taglen) != 0)
2667                 return -1;
2668             octx->iv_set = 0;
2669             return written_len;
2670         }
2671         /* If encrypting then just get the tag */
2672         if (CRYPTO_ocb128_tag(&octx->ocb, octx->tag, 16) != 1)
2673             return -1;
2674         /* Don't reuse the IV */
2675         octx->iv_set = 0;
2676         return written_len;
2677     }
2678 }
2679
2680 static int aes_ocb_cleanup(EVP_CIPHER_CTX *c)
2681 {
2682     EVP_AES_OCB_CTX *octx = EVP_C_DATA(EVP_AES_OCB_CTX,c);
2683     CRYPTO_ocb128_cleanup(&octx->ocb);
2684     return 1;
2685 }
2686
2687 BLOCK_CIPHER_custom(NID_aes, 128, 16, 12, ocb, OCB,
2688                     EVP_CIPH_FLAG_AEAD_CIPHER | CUSTOM_FLAGS)
2689 BLOCK_CIPHER_custom(NID_aes, 192, 16, 12, ocb, OCB,
2690                     EVP_CIPH_FLAG_AEAD_CIPHER | CUSTOM_FLAGS)
2691 BLOCK_CIPHER_custom(NID_aes, 256, 16, 12, ocb, OCB,
2692                     EVP_CIPH_FLAG_AEAD_CIPHER | CUSTOM_FLAGS)
2693 # endif                         /* OPENSSL_NO_OCB */
2694 #endif