GNU Linux-libre 4.14.266-gnu1
[releases.git] / arch / x86 / crypto / aes-i586-asm_32.S
1 // -------------------------------------------------------------------------
2 // Copyright (c) 2001, Dr Brian Gladman <                 >, Worcester, UK.
3 // All rights reserved.
4 //
5 // LICENSE TERMS
6 //
7 // The free distribution and use of this software in both source and binary 
8 // form is allowed (with or without changes) provided that:
9 //
10 //   1. distributions of this source code include the above copyright 
11 //      notice, this list of conditions and the following disclaimer//
12 //
13 //   2. distributions in binary form include the above copyright
14 //      notice, this list of conditions and the following disclaimer
15 //      in the documentation and/or other associated materials//
16 //
17 //   3. the copyright holder's name is not used to endorse products 
18 //      built using this software without specific written permission.
19 //
20 //
21 // ALTERNATIVELY, provided that this notice is retained in full, this product
22 // may be distributed under the terms of the GNU General Public License (GPL),
23 // in which case the provisions of the GPL apply INSTEAD OF those given above.
24 //
25 // Copyright (c) 2004 Linus Torvalds <torvalds@osdl.org>
26 // Copyright (c) 2004 Red Hat, Inc., James Morris <jmorris@redhat.com>
27
28 // DISCLAIMER
29 //
30 // This software is provided 'as is' with no explicit or implied warranties
31 // in respect of its properties including, but not limited to, correctness 
32 // and fitness for purpose.
33 // -------------------------------------------------------------------------
34 // Issue Date: 29/07/2002
35
36 .file "aes-i586-asm.S"
37 .text
38
39 #include <linux/linkage.h>
40 #include <asm/asm-offsets.h>
41
42 #define tlen 1024   // length of each of 4 'xor' arrays (256 32-bit words)
43
44 /* offsets to parameters with one register pushed onto stack */
45 #define ctx 8
46 #define out_blk 12
47 #define in_blk 16
48
49 /* offsets in crypto_aes_ctx structure */
50 #define klen (480)
51 #define ekey (0)
52 #define dkey (240)
53
54 // register mapping for encrypt and decrypt subroutines
55
56 #define r0  eax
57 #define r1  ebx
58 #define r2  ecx
59 #define r3  edx
60 #define r4  esi
61 #define r5  edi
62
63 #define eaxl  al
64 #define eaxh  ah
65 #define ebxl  bl
66 #define ebxh  bh
67 #define ecxl  cl
68 #define ecxh  ch
69 #define edxl  dl
70 #define edxh  dh
71
72 #define _h(reg) reg##h
73 #define h(reg) _h(reg)
74
75 #define _l(reg) reg##l
76 #define l(reg) _l(reg)
77
78 // This macro takes a 32-bit word representing a column and uses
79 // each of its four bytes to index into four tables of 256 32-bit
80 // words to obtain values that are then xored into the appropriate
81 // output registers r0, r1, r4 or r5.  
82
83 // Parameters:
84 // table table base address
85 //   %1  out_state[0]
86 //   %2  out_state[1]
87 //   %3  out_state[2]
88 //   %4  out_state[3]
89 //   idx input register for the round (destroyed)
90 //   tmp scratch register for the round
91 // sched key schedule
92
93 #define do_col(table, a1,a2,a3,a4, idx, tmp)    \
94         movzx   %l(idx),%tmp;                   \
95         xor     table(,%tmp,4),%a1;             \
96         movzx   %h(idx),%tmp;                   \
97         shr     $16,%idx;                       \
98         xor     table+tlen(,%tmp,4),%a2;        \
99         movzx   %l(idx),%tmp;                   \
100         movzx   %h(idx),%idx;                   \
101         xor     table+2*tlen(,%tmp,4),%a3;      \
102         xor     table+3*tlen(,%idx,4),%a4;
103
104 // initialise output registers from the key schedule
105 // NB1: original value of a3 is in idx on exit
106 // NB2: original values of a1,a2,a4 aren't used
107 #define do_fcol(table, a1,a2,a3,a4, idx, tmp, sched) \
108         mov     0 sched,%a1;                    \
109         movzx   %l(idx),%tmp;                   \
110         mov     12 sched,%a2;                   \
111         xor     table(,%tmp,4),%a1;             \
112         mov     4 sched,%a4;                    \
113         movzx   %h(idx),%tmp;                   \
114         shr     $16,%idx;                       \
115         xor     table+tlen(,%tmp,4),%a2;        \
116         movzx   %l(idx),%tmp;                   \
117         movzx   %h(idx),%idx;                   \
118         xor     table+3*tlen(,%idx,4),%a4;      \
119         mov     %a3,%idx;                       \
120         mov     8 sched,%a3;                    \
121         xor     table+2*tlen(,%tmp,4),%a3;
122
123 // initialise output registers from the key schedule
124 // NB1: original value of a3 is in idx on exit
125 // NB2: original values of a1,a2,a4 aren't used
126 #define do_icol(table, a1,a2,a3,a4, idx, tmp, sched) \
127         mov     0 sched,%a1;                    \
128         movzx   %l(idx),%tmp;                   \
129         mov     4 sched,%a2;                    \
130         xor     table(,%tmp,4),%a1;             \
131         mov     12 sched,%a4;                   \
132         movzx   %h(idx),%tmp;                   \
133         shr     $16,%idx;                       \
134         xor     table+tlen(,%tmp,4),%a2;        \
135         movzx   %l(idx),%tmp;                   \
136         movzx   %h(idx),%idx;                   \
137         xor     table+3*tlen(,%idx,4),%a4;      \
138         mov     %a3,%idx;                       \
139         mov     8 sched,%a3;                    \
140         xor     table+2*tlen(,%tmp,4),%a3;
141
142
143 // original Gladman had conditional saves to MMX regs.
144 #define save(a1, a2)            \
145         mov     %a2,4*a1(%esp)
146
147 #define restore(a1, a2)         \
148         mov     4*a2(%esp),%a1
149
150 // These macros perform a forward encryption cycle. They are entered with
151 // the first previous round column values in r0,r1,r4,r5 and
152 // exit with the final values in the same registers, using stack
153 // for temporary storage.
154
155 // round column values
156 // on entry: r0,r1,r4,r5
157 // on exit:  r2,r1,r4,r5
158 #define fwd_rnd1(arg, table)                                            \
159         save   (0,r1);                                                  \
160         save   (1,r5);                                                  \
161                                                                         \
162         /* compute new column values */                                 \
163         do_fcol(table, r2,r5,r4,r1, r0,r3, arg);        /* idx=r0 */    \
164         do_col (table, r4,r1,r2,r5, r0,r3);             /* idx=r4 */    \
165         restore(r0,0);                                                  \
166         do_col (table, r1,r2,r5,r4, r0,r3);             /* idx=r1 */    \
167         restore(r0,1);                                                  \
168         do_col (table, r5,r4,r1,r2, r0,r3);             /* idx=r5 */
169
170 // round column values
171 // on entry: r2,r1,r4,r5
172 // on exit:  r0,r1,r4,r5
173 #define fwd_rnd2(arg, table)                                            \
174         save   (0,r1);                                                  \
175         save   (1,r5);                                                  \
176                                                                         \
177         /* compute new column values */                                 \
178         do_fcol(table, r0,r5,r4,r1, r2,r3, arg);        /* idx=r2 */    \
179         do_col (table, r4,r1,r0,r5, r2,r3);             /* idx=r4 */    \
180         restore(r2,0);                                                  \
181         do_col (table, r1,r0,r5,r4, r2,r3);             /* idx=r1 */    \
182         restore(r2,1);                                                  \
183         do_col (table, r5,r4,r1,r0, r2,r3);             /* idx=r5 */
184
185 // These macros performs an inverse encryption cycle. They are entered with
186 // the first previous round column values in r0,r1,r4,r5 and
187 // exit with the final values in the same registers, using stack
188 // for temporary storage
189
190 // round column values
191 // on entry: r0,r1,r4,r5
192 // on exit:  r2,r1,r4,r5
193 #define inv_rnd1(arg, table)                                            \
194         save    (0,r1);                                                 \
195         save    (1,r5);                                                 \
196                                                                         \
197         /* compute new column values */                                 \
198         do_icol(table, r2,r1,r4,r5, r0,r3, arg);        /* idx=r0 */    \
199         do_col (table, r4,r5,r2,r1, r0,r3);             /* idx=r4 */    \
200         restore(r0,0);                                                  \
201         do_col (table, r1,r4,r5,r2, r0,r3);             /* idx=r1 */    \
202         restore(r0,1);                                                  \
203         do_col (table, r5,r2,r1,r4, r0,r3);             /* idx=r5 */
204
205 // round column values
206 // on entry: r2,r1,r4,r5
207 // on exit:  r0,r1,r4,r5
208 #define inv_rnd2(arg, table)                                            \
209         save    (0,r1);                                                 \
210         save    (1,r5);                                                 \
211                                                                         \
212         /* compute new column values */                                 \
213         do_icol(table, r0,r1,r4,r5, r2,r3, arg);        /* idx=r2 */    \
214         do_col (table, r4,r5,r0,r1, r2,r3);             /* idx=r4 */    \
215         restore(r2,0);                                                  \
216         do_col (table, r1,r4,r5,r0, r2,r3);             /* idx=r1 */    \
217         restore(r2,1);                                                  \
218         do_col (table, r5,r0,r1,r4, r2,r3);             /* idx=r5 */
219
220 // AES (Rijndael) Encryption Subroutine
221 /* void aes_enc_blk(struct crypto_aes_ctx *ctx, u8 *out_blk, const u8 *in_blk) */
222
223 .extern  crypto_ft_tab
224 .extern  crypto_fl_tab
225
226 ENTRY(aes_enc_blk)
227         push    %ebp
228         mov     ctx(%esp),%ebp
229
230 // CAUTION: the order and the values used in these assigns 
231 // rely on the register mappings
232
233 1:      push    %ebx
234         mov     in_blk+4(%esp),%r2
235         push    %esi
236         mov     klen(%ebp),%r3   // key size
237         push    %edi
238 #if ekey != 0
239         lea     ekey(%ebp),%ebp  // key pointer
240 #endif
241
242 // input four columns and xor in first round key
243
244         mov     (%r2),%r0
245         mov     4(%r2),%r1
246         mov     8(%r2),%r4
247         mov     12(%r2),%r5
248         xor     (%ebp),%r0
249         xor     4(%ebp),%r1
250         xor     8(%ebp),%r4
251         xor     12(%ebp),%r5
252
253         sub     $8,%esp         // space for register saves on stack
254         add     $16,%ebp        // increment to next round key
255         cmp     $24,%r3
256         jb      4f              // 10 rounds for 128-bit key
257         lea     32(%ebp),%ebp
258         je      3f              // 12 rounds for 192-bit key
259         lea     32(%ebp),%ebp
260
261 2:      fwd_rnd1( -64(%ebp), crypto_ft_tab)     // 14 rounds for 256-bit key
262         fwd_rnd2( -48(%ebp), crypto_ft_tab)
263 3:      fwd_rnd1( -32(%ebp), crypto_ft_tab)     // 12 rounds for 192-bit key
264         fwd_rnd2( -16(%ebp), crypto_ft_tab)
265 4:      fwd_rnd1(    (%ebp), crypto_ft_tab)     // 10 rounds for 128-bit key
266         fwd_rnd2( +16(%ebp), crypto_ft_tab)
267         fwd_rnd1( +32(%ebp), crypto_ft_tab)
268         fwd_rnd2( +48(%ebp), crypto_ft_tab)
269         fwd_rnd1( +64(%ebp), crypto_ft_tab)
270         fwd_rnd2( +80(%ebp), crypto_ft_tab)
271         fwd_rnd1( +96(%ebp), crypto_ft_tab)
272         fwd_rnd2(+112(%ebp), crypto_ft_tab)
273         fwd_rnd1(+128(%ebp), crypto_ft_tab)
274         fwd_rnd2(+144(%ebp), crypto_fl_tab)     // last round uses a different table
275
276 // move final values to the output array.  CAUTION: the 
277 // order of these assigns rely on the register mappings
278
279         add     $8,%esp
280         mov     out_blk+12(%esp),%ebp
281         mov     %r5,12(%ebp)
282         pop     %edi
283         mov     %r4,8(%ebp)
284         pop     %esi
285         mov     %r1,4(%ebp)
286         pop     %ebx
287         mov     %r0,(%ebp)
288         pop     %ebp
289         ret
290 ENDPROC(aes_enc_blk)
291
292 // AES (Rijndael) Decryption Subroutine
293 /* void aes_dec_blk(struct crypto_aes_ctx *ctx, u8 *out_blk, const u8 *in_blk) */
294
295 .extern  crypto_it_tab
296 .extern  crypto_il_tab
297
298 ENTRY(aes_dec_blk)
299         push    %ebp
300         mov     ctx(%esp),%ebp
301
302 // CAUTION: the order and the values used in these assigns 
303 // rely on the register mappings
304
305 1:      push    %ebx
306         mov     in_blk+4(%esp),%r2
307         push    %esi
308         mov     klen(%ebp),%r3   // key size
309         push    %edi
310 #if dkey != 0
311         lea     dkey(%ebp),%ebp  // key pointer
312 #endif
313         
314 // input four columns and xor in first round key
315
316         mov     (%r2),%r0
317         mov     4(%r2),%r1
318         mov     8(%r2),%r4
319         mov     12(%r2),%r5
320         xor     (%ebp),%r0
321         xor     4(%ebp),%r1
322         xor     8(%ebp),%r4
323         xor     12(%ebp),%r5
324
325         sub     $8,%esp         // space for register saves on stack
326         add     $16,%ebp        // increment to next round key
327         cmp     $24,%r3
328         jb      4f              // 10 rounds for 128-bit key
329         lea     32(%ebp),%ebp
330         je      3f              // 12 rounds for 192-bit key
331         lea     32(%ebp),%ebp
332
333 2:      inv_rnd1( -64(%ebp), crypto_it_tab)     // 14 rounds for 256-bit key
334         inv_rnd2( -48(%ebp), crypto_it_tab)
335 3:      inv_rnd1( -32(%ebp), crypto_it_tab)     // 12 rounds for 192-bit key
336         inv_rnd2( -16(%ebp), crypto_it_tab)
337 4:      inv_rnd1(    (%ebp), crypto_it_tab)     // 10 rounds for 128-bit key
338         inv_rnd2( +16(%ebp), crypto_it_tab)
339         inv_rnd1( +32(%ebp), crypto_it_tab)
340         inv_rnd2( +48(%ebp), crypto_it_tab)
341         inv_rnd1( +64(%ebp), crypto_it_tab)
342         inv_rnd2( +80(%ebp), crypto_it_tab)
343         inv_rnd1( +96(%ebp), crypto_it_tab)
344         inv_rnd2(+112(%ebp), crypto_it_tab)
345         inv_rnd1(+128(%ebp), crypto_it_tab)
346         inv_rnd2(+144(%ebp), crypto_il_tab)     // last round uses a different table
347
348 // move final values to the output array.  CAUTION: the 
349 // order of these assigns rely on the register mappings
350
351         add     $8,%esp
352         mov     out_blk+12(%esp),%ebp
353         mov     %r5,12(%ebp)
354         pop     %edi
355         mov     %r4,8(%ebp)
356         pop     %esi
357         mov     %r1,4(%ebp)
358         pop     %ebx
359         mov     %r0,(%ebp)
360         pop     %ebp
361         ret
362 ENDPROC(aes_dec_blk)