1 /* $NetBSD: aes.c,v 1.1.1.1 2011/04/13 18:14:49 elric Exp $ */
2
3 /*
4 * Copyright (c) 2003 Kungliga Tekniska Högskolan
5 * (Royal Institute of Technology, Stockholm, Sweden).
6 * All rights reserved.
7 *
8 * Redistribution and use in source and binary forms, with or without
9 * modification, are permitted provided that the following conditions
10 * are met:
11 *
12 * 1. Redistributions of source code must retain the above copyright
13 * notice, this list of conditions and the following disclaimer.
14 *
15 * 2. Redistributions in binary form must reproduce the above copyright
16 * notice, this list of conditions and the following disclaimer in the
17 * documentation and/or other materials provided with the distribution.
18 *
19 * 3. Neither the name of the Institute nor the names of its contributors
20 * may be used to endorse or promote products derived from this software
21 * without specific prior written permission.
22 *
23 * THIS SOFTWARE IS PROVIDED BY THE INSTITUTE AND CONTRIBUTORS ``AS IS'' AND
24 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
25 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
26 * ARE DISCLAIMED. IN NO EVENT SHALL THE INSTITUTE OR CONTRIBUTORS BE LIABLE
27 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
28 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
29 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
30 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
31 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
32 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
33 * SUCH DAMAGE.
34 */
35
36 #include "config.h"
37
38
39 #ifdef KRB5
40 #include <krb5/krb5-types.h>
41 #endif
42
43 #include <string.h>
44
45 #include "rijndael-alg-fst.h"
46 #include "aes.h"
47
48 int
AES_set_encrypt_key(const unsigned char * userkey,const int bits,AES_KEY * key)49 AES_set_encrypt_key(const unsigned char *userkey, const int bits, AES_KEY *key)
50 {
51 key->rounds = rijndaelKeySetupEnc(key->key, userkey, bits);
52 if (key->rounds == 0)
53 return -1;
54 return 0;
55 }
56
57 int
AES_set_decrypt_key(const unsigned char * userkey,const int bits,AES_KEY * key)58 AES_set_decrypt_key(const unsigned char *userkey, const int bits, AES_KEY *key)
59 {
60 key->rounds = rijndaelKeySetupDec(key->key, userkey, bits);
61 if (key->rounds == 0)
62 return -1;
63 return 0;
64 }
65
66 void
AES_encrypt(const unsigned char * in,unsigned char * out,const AES_KEY * key)67 AES_encrypt(const unsigned char *in, unsigned char *out, const AES_KEY *key)
68 {
69 rijndaelEncrypt(key->key, key->rounds, in, out);
70 }
71
72 void
AES_decrypt(const unsigned char * in,unsigned char * out,const AES_KEY * key)73 AES_decrypt(const unsigned char *in, unsigned char *out, const AES_KEY *key)
74 {
75 rijndaelDecrypt(key->key, key->rounds, in, out);
76 }
77
78 void
AES_cbc_encrypt(const unsigned char * in,unsigned char * out,unsigned long size,const AES_KEY * key,unsigned char * iv,int forward_encrypt)79 AES_cbc_encrypt(const unsigned char *in, unsigned char *out,
80 unsigned long size, const AES_KEY *key,
81 unsigned char *iv, int forward_encrypt)
82 {
83 unsigned char tmp[AES_BLOCK_SIZE];
84 int i;
85
86 if (forward_encrypt) {
87 while (size >= AES_BLOCK_SIZE) {
88 for (i = 0; i < AES_BLOCK_SIZE; i++)
89 tmp[i] = in[i] ^ iv[i];
90 AES_encrypt(tmp, out, key);
91 memcpy(iv, out, AES_BLOCK_SIZE);
92 size -= AES_BLOCK_SIZE;
93 in += AES_BLOCK_SIZE;
94 out += AES_BLOCK_SIZE;
95 }
96 if (size) {
97 for (i = 0; i < size; i++)
98 tmp[i] = in[i] ^ iv[i];
99 for (i = size; i < AES_BLOCK_SIZE; i++)
100 tmp[i] = iv[i];
101 AES_encrypt(tmp, out, key);
102 memcpy(iv, out, AES_BLOCK_SIZE);
103 }
104 } else {
105 while (size >= AES_BLOCK_SIZE) {
106 memcpy(tmp, in, AES_BLOCK_SIZE);
107 AES_decrypt(tmp, out, key);
108 for (i = 0; i < AES_BLOCK_SIZE; i++)
109 out[i] ^= iv[i];
110 memcpy(iv, tmp, AES_BLOCK_SIZE);
111 size -= AES_BLOCK_SIZE;
112 in += AES_BLOCK_SIZE;
113 out += AES_BLOCK_SIZE;
114 }
115 if (size) {
116 memcpy(tmp, in, AES_BLOCK_SIZE);
117 AES_decrypt(tmp, out, key);
118 for (i = 0; i < size; i++)
119 out[i] ^= iv[i];
120 memcpy(iv, tmp, AES_BLOCK_SIZE);
121 }
122 }
123 }
124
125 void
AES_cfb8_encrypt(const unsigned char * in,unsigned char * out,unsigned long size,const AES_KEY * key,unsigned char * iv,int forward_encrypt)126 AES_cfb8_encrypt(const unsigned char *in, unsigned char *out,
127 unsigned long size, const AES_KEY *key,
128 unsigned char *iv, int forward_encrypt)
129 {
130 int i;
131
132 for (i = 0; i < size; i++) {
133 unsigned char tmp[AES_BLOCK_SIZE + 1];
134
135 memcpy(tmp, iv, AES_BLOCK_SIZE);
136 AES_encrypt(iv, iv, key);
137 if (!forward_encrypt) {
138 tmp[AES_BLOCK_SIZE] = in[i];
139 }
140 out[i] = in[i] ^ iv[0];
141 if (forward_encrypt) {
142 tmp[AES_BLOCK_SIZE] = out[i];
143 }
144 memcpy(iv, &tmp[1], AES_BLOCK_SIZE);
145 }
146 }
147