3 * The Rijndael block cipher
5 * (c) 2000 Straylight/Edgeware
8 /*----- Licensing notice --------------------------------------------------*
10 * This file is part of Catacomb.
12 * Catacomb is free software; you can redistribute it and/or modify
13 * it under the terms of the GNU Library General Public License as
14 * published by the Free Software Foundation; either version 2 of the
15 * License, or (at your option) any later version.
17 * Catacomb is distributed in the hope that it will be useful,
18 * but WITHOUT ANY WARRANTY; without even the implied warranty of
19 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
20 * GNU Library General Public License for more details.
22 * You should have received a copy of the GNU Library General Public
23 * License along with Catacomb; if not, write to the Free
24 * Software Foundation, Inc., 59 Temple Place - Suite 330, Boston,
28 /*----- Header files ------------------------------------------------------*/
35 #include <mLib/bits.h>
41 #include "rijndael-base.h"
43 /*----- Main code ---------------------------------------------------------*/
45 /* --- @rijndael_init@ --- *
47 * Arguments: @rijndael_ctx *k@ = pointer to context to initialize
48 * @const void *buf@ = pointer to buffer of key material
49 * @size_t sz@ = size of the key material
53 * Use: Initializes a Rijndael context with a particular key. This
54 * implementation of Rijndael doesn't impose any particular
55 * limits on the key size except that it must be multiple of 4
56 * bytes long. 256 bits seems sensible, though.
59 void rijndael_init(rijndael_ctx *k, const void *buf, size_t sz)
61 rijndael_setup(k, RIJNDAEL_BLKSZ / 4, buf, sz);
64 /* --- @rijndael_eblk@, @rijndael_dblk@ --- *
66 * Arguments: @const rijndael_ctx *k@ = pointer to Rijndael context
67 * @const uint32 s[4]@ = pointer to source block
68 * @uint32 d[4]@ = pointer to destination block
72 * Use: Low-level block encryption and decryption.
75 CPU_DISPATCH(EMPTY, EMPTY, void, rijndael_eblk,
76 (const rijndael_ctx *k, const uint32 s[4], uint32 d[4]),
77 (k, s, d), pick_eblk, simple_eblk)
79 CPU_DISPATCH(EMPTY, EMPTY, void, rijndael_dblk,
80 (const rijndael_ctx *k, const uint32 s[4], uint32 d[4]),
81 (k, s, d), pick_dblk, simple_dblk)
83 #if CPUFAM_X86 || CPUFAM_AMD64
84 extern rijndael_eblk__functype rijndael_eblk_x86ish_aesni;
85 extern rijndael_dblk__functype rijndael_dblk_x86ish_aesni;
86 extern rijndael_eblk__functype rijndael_eblk_x86ish_aesni_avx;
87 extern rijndael_dblk__functype rijndael_dblk_x86ish_aesni_avx;
89 #if CPUFAM_ARMEL && HAVE_AS_ARMV8_CRYPTO
90 extern rijndael_eblk__functype rijndael_eblk_arm_crypto;
91 extern rijndael_dblk__functype rijndael_dblk_arm_crypto;
94 extern rijndael_eblk__functype rijndael_eblk_arm64_crypto;
95 extern rijndael_dblk__functype rijndael_dblk_arm64_crypto;
98 static rijndael_eblk__functype *pick_eblk(void)
100 #if CPUFAM_X86 || CPUFAM_AMD64
101 DISPATCH_PICK_COND(rijndael_eblk, rijndael_eblk_x86ish_aesni_avx,
102 cpu_feature_p(CPUFEAT_X86_AVX) &&
103 cpu_feature_p(CPUFEAT_X86_AESNI));
104 DISPATCH_PICK_COND(rijndael_eblk, rijndael_eblk_x86ish_aesni,
105 cpu_feature_p(CPUFEAT_X86_AESNI));
107 #if CPUFAM_ARMEL && HAVE_AS_ARMV8_CRYPTO
108 DISPATCH_PICK_COND(rijndael_eblk, rijndael_eblk_arm_crypto,
109 cpu_feature_p(CPUFEAT_ARM_AES));
112 DISPATCH_PICK_COND(rijndael_eblk, rijndael_eblk_arm64_crypto,
113 cpu_feature_p(CPUFEAT_ARM_AES));
115 DISPATCH_PICK_FALLBACK(rijndael_eblk, simple_eblk);
118 static rijndael_dblk__functype *pick_dblk(void)
120 #if CPUFAM_X86 || CPUFAM_AMD64
121 DISPATCH_PICK_COND(rijndael_dblk, rijndael_dblk_x86ish_aesni_avx,
122 cpu_feature_p(CPUFEAT_X86_AVX) &&
123 cpu_feature_p(CPUFEAT_X86_AESNI));
124 DISPATCH_PICK_COND(rijndael_dblk, rijndael_dblk_x86ish_aesni,
125 cpu_feature_p(CPUFEAT_X86_AESNI));
127 #if CPUFAM_ARMEL && HAVE_AS_ARMV8_CRYPTO
128 DISPATCH_PICK_COND(rijndael_dblk, rijndael_dblk_arm_crypto,
129 cpu_feature_p(CPUFEAT_ARM_AES));
132 DISPATCH_PICK_COND(rijndael_dblk, rijndael_dblk_arm64_crypto,
133 cpu_feature_p(CPUFEAT_ARM_AES));
135 DISPATCH_PICK_FALLBACK(rijndael_dblk, simple_dblk);
138 #define DO(what, t, aa, bb, cc, dd, a, b, c, d, w) do { \
139 aa = what(t, a, b, c, d) ^ *w++; \
140 bb = what(t, b, c, d, a) ^ *w++; \
141 cc = what(t, c, d, a, b) ^ *w++; \
142 dd = what(t, d, a, b, c) ^ *w++; \
145 #define UNDO(what, t, aa, bb, cc, dd, a, b, c, d, w) do { \
146 aa = what(t, a, d, c, b) ^ *w++; \
147 bb = what(t, b, a, d, c) ^ *w++; \
148 cc = what(t, c, b, a, d) ^ *w++; \
149 dd = what(t, d, c, b, a) ^ *w++; \
152 static void simple_eblk(const rijndael_ctx *k, const uint32 *s, uint32 *dst)
154 uint32 a = s[0], b = s[1], c = s[2], d = s[3];
155 uint32 aa, bb, cc, dd;
156 const uint32 *w = k->w;
158 a ^= *w++; b ^= *w++; c ^= *w++; d ^= *w++;
159 aa = a; bb = b; cc = c; dd = d;
163 DO(MIX, T, aa, bb, cc, dd, a, b, c, d, w);
165 DO(MIX, T, a, b, c, d, aa, bb, cc, dd, w);
167 DO(MIX, T, aa, bb, cc, dd, a, b, c, d, w);
169 DO(MIX, T, a, b, c, d, aa, bb, cc, dd, w);
172 DO(MIX, T, aa, bb, cc, dd, a, b, c, d, w);
173 DO(MIX, T, a, b, c, d, aa, bb, cc, dd, w);
174 DO(MIX, T, aa, bb, cc, dd, a, b, c, d, w);
175 DO(MIX, T, a, b, c, d, aa, bb, cc, dd, w);
176 DO(MIX, T, aa, bb, cc, dd, a, b, c, d, w);
177 DO(MIX, T, a, b, c, d, aa, bb, cc, dd, w);
178 DO(MIX, T, aa, bb, cc, dd, a, b, c, d, w);
179 DO(MIX, T, a, b, c, d, aa, bb, cc, dd, w);
180 DO(MIX, T, aa, bb, cc, dd, a, b, c, d, w);
182 DO(SUB, S, a, b, c, d, aa, bb, cc, dd, w);
184 dst[0] = a; dst[1] = b; dst[2] = c; dst[3] = d;
187 static void simple_dblk(const rijndael_ctx *k, const uint32 *s, uint32 *dst)
189 uint32 a = s[0], b = s[1], c = s[2], d = s[3];
190 uint32 aa, bb, cc, dd;
191 const uint32 *w = k->wi;
193 a ^= *w++; b ^= *w++; c ^= *w++; d ^= *w++;
194 aa = a; bb = b; cc = c; dd = d;
198 UNDO(MIX, TI, aa, bb, cc, dd, a, b, c, d, w);
200 UNDO(MIX, TI, a, b, c, d, aa, bb, cc, dd, w);
202 UNDO(MIX, TI, aa, bb, cc, dd, a, b, c, d, w);
204 UNDO(MIX, TI, a, b, c, d, aa, bb, cc, dd, w);
207 UNDO(MIX, TI, aa, bb, cc, dd, a, b, c, d, w);
208 UNDO(MIX, TI, a, b, c, d, aa, bb, cc, dd, w);
209 UNDO(MIX, TI, aa, bb, cc, dd, a, b, c, d, w);
210 UNDO(MIX, TI, a, b, c, d, aa, bb, cc, dd, w);
211 UNDO(MIX, TI, aa, bb, cc, dd, a, b, c, d, w);
212 UNDO(MIX, TI, a, b, c, d, aa, bb, cc, dd, w);
213 UNDO(MIX, TI, aa, bb, cc, dd, a, b, c, d, w);
214 UNDO(MIX, TI, a, b, c, d, aa, bb, cc, dd, w);
215 UNDO(MIX, TI, aa, bb, cc, dd, a, b, c, d, w);
217 UNDO(SUB, SI, a, b, c, d, aa, bb, cc, dd, w);
219 dst[0] = a; dst[1] = b; dst[2] = c; dst[3] = d;
222 BLKC_TEST(RIJNDAEL, rijndael)
224 /*----- That's all, folks -------------------------------------------------*/