]> git.zerfleddert.de Git - proxmark3-svn/blob - client/aes.h
HF Search - refactoring cmds to work with it
[proxmark3-svn] / client / aes.h
1 /**
2 * \file aes.h
3 *
4 * \brief AES block cipher
5 *
6 * Copyright (C) 2006-2014, Brainspark B.V.
7 *
8 * This file is part of PolarSSL (http://www.polarssl.org)
9 * Lead Maintainer: Paul Bakker <polarssl_maintainer at polarssl.org>
10 *
11 * All rights reserved.
12 *
13 * This program is free software; you can redistribute it and/or modify
14 * it under the terms of the GNU General Public License as published by
15 * the Free Software Foundation; either version 2 of the License, or
16 * (at your option) any later version.
17 *
18 * This program is distributed in the hope that it will be useful,
19 * but WITHOUT ANY WARRANTY; without even the implied warranty of
20 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
21 * GNU General Public License for more details.
22 *
23 * You should have received a copy of the GNU General Public License along
24 * with this program; if not, write to the Free Software Foundation, Inc.,
25 * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
26 */
27 #ifndef POLARSSL_AES_H
28 #define POLARSSL_AES_H
29
30 #if !defined(POLARSSL_CONFIG_FILE)
31 #include "polarssl_config.h"
32 #else
33 #include POLARSSL_CONFIG_FILE
34 #endif
35
36 #include <string.h>
37
38 #if defined(_MSC_VER) && !defined(EFIX64) && !defined(EFI32)
39 #include <basetsd.h>
40 typedef UINT32 uint32_t;
41 #else
42 #include <inttypes.h>
43 #endif
44
45 /* padlock.c and aesni.c rely on these values! */
46 #define AES_ENCRYPT 1
47 #define AES_DECRYPT 0
48
49 #define POLARSSL_ERR_AES_INVALID_KEY_LENGTH -0x0020 /**< Invalid key length. */
50 #define POLARSSL_ERR_AES_INVALID_INPUT_LENGTH -0x0022 /**< Invalid data input length. */
51
52 #if !defined(POLARSSL_AES_ALT)
53 // Regular implementation
54 //
55
56 #ifdef __cplusplus
57 extern "C" {
58 #endif
59
60 /**
61 * \brief AES context structure
62 *
63 * \note buf is able to hold 32 extra bytes, which can be used:
64 * - for alignment purposes if VIA padlock is used, and/or
65 * - to simplify key expansion in the 256-bit case by
66 * generating an extra round key
67 */
68 typedef struct
69 {
70 int nr; /*!< number of rounds */
71 uint32_t *rk; /*!< AES round keys */
72 uint32_t buf[68]; /*!< unaligned data */
73 }
74 aes_context;
75
76 /**
77 * \brief Initialize AES context
78 *
79 * \param ctx AES context to be initialized
80 */
81 void aes_init( aes_context *ctx );
82
83 /**
84 * \brief Clear AES context
85 *
86 * \param ctx AES context to be cleared
87 */
88 void aes_free( aes_context *ctx );
89
90 /**
91 * \brief AES key schedule (encryption)
92 *
93 * \param ctx AES context to be initialized
94 * \param key encryption key
95 * \param keysize must be 128, 192 or 256
96 *
97 * \return 0 if successful, or POLARSSL_ERR_AES_INVALID_KEY_LENGTH
98 */
99 int aes_setkey_enc( aes_context *ctx, const unsigned char *key,
100 unsigned int keysize );
101
102 /**
103 * \brief AES key schedule (decryption)
104 *
105 * \param ctx AES context to be initialized
106 * \param key decryption key
107 * \param keysize must be 128, 192 or 256
108 *
109 * \return 0 if successful, or POLARSSL_ERR_AES_INVALID_KEY_LENGTH
110 */
111 int aes_setkey_dec( aes_context *ctx, const unsigned char *key,
112 unsigned int keysize );
113
114 /**
115 * \brief AES-ECB block encryption/decryption
116 *
117 * \param ctx AES context
118 * \param mode AES_ENCRYPT or AES_DECRYPT
119 * \param input 16-byte input block
120 * \param output 16-byte output block
121 *
122 * \return 0 if successful
123 */
124 int aes_crypt_ecb( aes_context *ctx,
125 int mode,
126 const unsigned char input[16],
127 unsigned char output[16] );
128
129 #if defined(POLARSSL_CIPHER_MODE_CBC)
130 /**
131 * \brief AES-CBC buffer encryption/decryption
132 * Length should be a multiple of the block
133 * size (16 bytes)
134 *
135 * \param ctx AES context
136 * \param mode AES_ENCRYPT or AES_DECRYPT
137 * \param length length of the input data
138 * \param iv initialization vector (updated after use)
139 * \param input buffer holding the input data
140 * \param output buffer holding the output data
141 *
142 * \return 0 if successful, or POLARSSL_ERR_AES_INVALID_INPUT_LENGTH
143 */
144 int aes_crypt_cbc( aes_context *ctx,
145 int mode,
146 size_t length,
147 unsigned char iv[16],
148 const unsigned char *input,
149 unsigned char *output );
150 #endif /* POLARSSL_CIPHER_MODE_CBC */
151
152 #if defined(POLARSSL_CIPHER_MODE_CFB)
153 /**
154 * \brief AES-CFB128 buffer encryption/decryption.
155 *
156 * Note: Due to the nature of CFB you should use the same key schedule for
157 * both encryption and decryption. So a context initialized with
158 * aes_setkey_enc() for both AES_ENCRYPT and AES_DECRYPT.
159 *
160 * \param ctx AES context
161 * \param mode AES_ENCRYPT or AES_DECRYPT
162 * \param length length of the input data
163 * \param iv_off offset in IV (updated after use)
164 * \param iv initialization vector (updated after use)
165 * \param input buffer holding the input data
166 * \param output buffer holding the output data
167 *
168 * \return 0 if successful
169 */
170 int aes_crypt_cfb128( aes_context *ctx,
171 int mode,
172 size_t length,
173 size_t *iv_off,
174 unsigned char iv[16],
175 const unsigned char *input,
176 unsigned char *output );
177
178 /**
179 * \brief AES-CFB8 buffer encryption/decryption.
180 *
181 * Note: Due to the nature of CFB you should use the same key schedule for
182 * both encryption and decryption. So a context initialized with
183 * aes_setkey_enc() for both AES_ENCRYPT and AES_DECRYPT.
184 *
185 * \param ctx AES context
186 * \param mode AES_ENCRYPT or AES_DECRYPT
187 * \param length length of the input data
188 * \param iv initialization vector (updated after use)
189 * \param input buffer holding the input data
190 * \param output buffer holding the output data
191 *
192 * \return 0 if successful
193 */
194 int aes_crypt_cfb8( aes_context *ctx,
195 int mode,
196 size_t length,
197 unsigned char iv[16],
198 const unsigned char *input,
199 unsigned char *output );
200 #endif /*POLARSSL_CIPHER_MODE_CFB */
201
202 #if defined(POLARSSL_CIPHER_MODE_CTR)
203 /**
204 * \brief AES-CTR buffer encryption/decryption
205 *
206 * Warning: You have to keep the maximum use of your counter in mind!
207 *
208 * Note: Due to the nature of CTR you should use the same key schedule for
209 * both encryption and decryption. So a context initialized with
210 * aes_setkey_enc() for both AES_ENCRYPT and AES_DECRYPT.
211 *
212 * \param ctx AES context
213 * \param length The length of the data
214 * \param nc_off The offset in the current stream_block (for resuming
215 * within current cipher stream). The offset pointer to
216 * should be 0 at the start of a stream.
217 * \param nonce_counter The 128-bit nonce and counter.
218 * \param stream_block The saved stream-block for resuming. Is overwritten
219 * by the function.
220 * \param input The input data stream
221 * \param output The output data stream
222 *
223 * \return 0 if successful
224 */
225 int aes_crypt_ctr( aes_context *ctx,
226 size_t length,
227 size_t *nc_off,
228 unsigned char nonce_counter[16],
229 unsigned char stream_block[16],
230 const unsigned char *input,
231 unsigned char *output );
232 #endif /* POLARSSL_CIPHER_MODE_CTR */
233
234 #ifdef __cplusplus
235 }
236 #endif
237
238 #else /* POLARSSL_AES_ALT */
239 #include "aes_alt.h"
240 #endif /* POLARSSL_AES_ALT */
241
242 #ifdef __cplusplus
243 extern "C" {
244 #endif
245
246 /**
247 * \brief Checkup routine
248 *
249 * \return 0 if successful, or 1 if the test failed
250 */
251 int aes_self_test( int verbose );
252
253 #ifdef __cplusplus
254 }
255 #endif
256
257 #endif /* aes.h */
Impressum, Datenschutz