Rework framework for assembler support for AES counter mode and add
[openssl.git] / crypto / evp / e_aes.c
1 /* ====================================================================
2  * Copyright (c) 2001 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/evp.h>
54 #include <openssl/err.h>
55 #include <string.h>
56 #include <assert.h>
57 #include <openssl/aes.h>
58 #include <openssl/modes.h>
59 #include "evp_locl.h"
60
61 static int aes_init_key(EVP_CIPHER_CTX *ctx, const unsigned char *key,
62                                         const unsigned char *iv, int enc);
63
64 typedef struct
65         {
66         AES_KEY ks;
67         } EVP_AES_KEY;
68
69 #define data(ctx)       EVP_C_DATA(EVP_AES_KEY,ctx)
70
71 IMPLEMENT_BLOCK_CIPHER(aes_128, ks, AES, EVP_AES_KEY,
72                        NID_aes_128, 16, 16, 16, 128,
73                        0, aes_init_key, NULL, 
74                        EVP_CIPHER_set_asn1_iv,
75                        EVP_CIPHER_get_asn1_iv,
76                        NULL)
77 IMPLEMENT_BLOCK_CIPHER(aes_192, ks, AES, EVP_AES_KEY,
78                        NID_aes_192, 16, 24, 16, 128,
79                        0, aes_init_key, NULL, 
80                        EVP_CIPHER_set_asn1_iv,
81                        EVP_CIPHER_get_asn1_iv,
82                        NULL)
83 IMPLEMENT_BLOCK_CIPHER(aes_256, ks, AES, EVP_AES_KEY,
84                        NID_aes_256, 16, 32, 16, 128,
85                        0, aes_init_key, NULL, 
86                        EVP_CIPHER_set_asn1_iv,
87                        EVP_CIPHER_get_asn1_iv,
88                        NULL)
89
90 #define IMPLEMENT_AES_CFBR(ksize,cbits) IMPLEMENT_CFBR(aes,AES,EVP_AES_KEY,ks,ksize,cbits,16)
91
92 IMPLEMENT_AES_CFBR(128,1)
93 IMPLEMENT_AES_CFBR(192,1)
94 IMPLEMENT_AES_CFBR(256,1)
95
96 IMPLEMENT_AES_CFBR(128,8)
97 IMPLEMENT_AES_CFBR(192,8)
98 IMPLEMENT_AES_CFBR(256,8)
99
100 static int aes_counter (EVP_CIPHER_CTX *ctx, unsigned char *out,
101                 const unsigned char *in, size_t len)
102 {
103         unsigned int num;
104         num = ctx->num;
105 #ifdef AES_CTR_ASM
106         void AES_ctr32_encrypt(const unsigned char *in, unsigned char *out,
107                         size_t blocks, const AES_KEY *key,
108                         const unsigned char ivec[AES_BLOCK_SIZE]);
109
110         CRYPTO_ctr128_encrypt_ctr32(in,out,len,
111                 &((EVP_AES_KEY *)ctx->cipher_data)->ks,
112                 ctx->iv,ctx->buf,&num,(ctr128_f)AES_ctr32_encrypt);
113 #else
114         CRYPTO_ctr128_encrypt(in,out,len,
115                 &((EVP_AES_KEY *)ctx->cipher_data)->ks,
116                 ctx->iv,ctx->buf,&num,(block128_f)AES_encrypt);
117 #endif
118         ctx->num = (size_t)num;
119         return 1;
120 }
121
122 static const EVP_CIPHER aes_128_ctr_cipher=
123         {
124         NID_aes_128_ctr,1,16,16,
125         EVP_CIPH_CUSTOM_IV,
126         aes_init_key,
127         aes_counter,
128         NULL,
129         sizeof(EVP_AES_KEY),
130         NULL,
131         NULL,
132         NULL,
133         NULL
134         };
135
136 const EVP_CIPHER *EVP_aes_128_ctr (void)
137 {       return &aes_128_ctr_cipher;     }
138
139 static const EVP_CIPHER aes_192_ctr_cipher=
140         {
141         NID_aes_192_ctr,1,24,16,
142         EVP_CIPH_CUSTOM_IV,
143         aes_init_key,
144         aes_counter,
145         NULL,
146         sizeof(EVP_AES_KEY),
147         NULL,
148         NULL,
149         NULL,
150         NULL
151         };
152
153 const EVP_CIPHER *EVP_aes_192_ctr (void)
154 {       return &aes_192_ctr_cipher;     }
155
156 static const EVP_CIPHER aes_256_ctr_cipher=
157         {
158         NID_aes_256_ctr,1,32,16,
159         EVP_CIPH_CUSTOM_IV,
160         aes_init_key,
161         aes_counter,
162         NULL,
163         sizeof(EVP_AES_KEY),
164         NULL,
165         NULL,
166         NULL,
167         NULL
168         };
169
170 const EVP_CIPHER *EVP_aes_256_ctr (void)
171 {       return &aes_256_ctr_cipher;     }
172
173 static int aes_init_key(EVP_CIPHER_CTX *ctx, const unsigned char *key,
174                    const unsigned char *iv, int enc)
175         {
176         int ret;
177
178         if (((ctx->cipher->flags & EVP_CIPH_MODE) == EVP_CIPH_ECB_MODE
179             || (ctx->cipher->flags & EVP_CIPH_MODE) == EVP_CIPH_CBC_MODE)
180             && !enc) 
181                 ret=AES_set_decrypt_key(key, ctx->key_len * 8, ctx->cipher_data);
182         else
183                 ret=AES_set_encrypt_key(key, ctx->key_len * 8, ctx->cipher_data);
184
185         if(ret < 0)
186                 {
187                 EVPerr(EVP_F_AES_INIT_KEY,EVP_R_AES_KEY_SETUP_FAILED);
188                 return 0;
189                 }
190
191         if (ctx->cipher->flags&EVP_CIPH_CUSTOM_IV)
192                 {
193                 if (iv!=NULL)
194                         memcpy (ctx->iv,iv,ctx->cipher->iv_len);
195                 else    {
196                         EVPerr(EVP_F_AES_INIT_KEY,EVP_R_AES_IV_SETUP_FAILED);
197                         return 0;
198                         }
199                 }
200
201         return 1;
202         }
203
204 #endif