18d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt/* 28d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt * AES (Rijndael) cipher - encrypt 38d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt * 48d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt * Modifications to public domain implementation: 58d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt * - support only 128-bit keys 68d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt * - cleanup 78d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt * - use C pre-processor to make it easier to change S table access 88d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt * - added option (AES_SMALL_TABLES) for reducing code size by about 8 kB at 98d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt * cost of reduced throughput (quite small difference on Pentium 4, 108d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt * 10-25% when using -O1 or -O2 optimization) 118d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt * 128d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt * Copyright (c) 2003-2005, Jouni Malinen <j@w1.fi> 138d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt * 148d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt * This program is free software; you can redistribute it and/or modify 158d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt * it under the terms of the GNU General Public License version 2 as 168d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt * published by the Free Software Foundation. 178d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt * 188d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt * Alternatively, this software may be distributed under the terms of BSD 198d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt * license. 208d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt * 218d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt * See README and COPYING for more details. 228d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt */ 238d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt 248d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt#include "includes.h" 258d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt 268d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt#include "common.h" 278d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt#include "crypto.h" 288d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt#include "aes_i.h" 298d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt 308d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidtvoid rijndaelEncrypt(const u32 rk[/*44*/], const u8 pt[16], u8 ct[16]) 318d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt{ 328d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt u32 s0, s1, s2, s3, t0, t1, t2, t3; 338d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt const int Nr = 10; 348d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt#ifndef FULL_UNROLL 358d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt int r; 368d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt#endif /* ?FULL_UNROLL */ 378d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt 388d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt /* 398d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt * map byte array block to cipher state 408d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt * and add initial round key: 418d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt */ 428d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt s0 = GETU32(pt ) ^ rk[0]; 438d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt s1 = GETU32(pt + 4) ^ rk[1]; 448d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt s2 = GETU32(pt + 8) ^ rk[2]; 458d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt s3 = GETU32(pt + 12) ^ rk[3]; 468d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt 478d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt#define ROUND(i,d,s) \ 488d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidtd##0 = TE0(s##0) ^ TE1(s##1) ^ TE2(s##2) ^ TE3(s##3) ^ rk[4 * i]; \ 498d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidtd##1 = TE0(s##1) ^ TE1(s##2) ^ TE2(s##3) ^ TE3(s##0) ^ rk[4 * i + 1]; \ 508d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidtd##2 = TE0(s##2) ^ TE1(s##3) ^ TE2(s##0) ^ TE3(s##1) ^ rk[4 * i + 2]; \ 518d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidtd##3 = TE0(s##3) ^ TE1(s##0) ^ TE2(s##1) ^ TE3(s##2) ^ rk[4 * i + 3] 528d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt 538d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt#ifdef FULL_UNROLL 548d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt 558d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt ROUND(1,t,s); 568d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt ROUND(2,s,t); 578d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt ROUND(3,t,s); 588d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt ROUND(4,s,t); 598d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt ROUND(5,t,s); 608d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt ROUND(6,s,t); 618d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt ROUND(7,t,s); 628d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt ROUND(8,s,t); 638d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt ROUND(9,t,s); 648d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt 658d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt rk += Nr << 2; 668d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt 678d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt#else /* !FULL_UNROLL */ 688d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt 698d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt /* Nr - 1 full rounds: */ 708d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt r = Nr >> 1; 718d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt for (;;) { 728d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt ROUND(1,t,s); 738d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt rk += 8; 748d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt if (--r == 0) 758d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt break; 768d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt ROUND(0,s,t); 778d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt } 788d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt 798d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt#endif /* ?FULL_UNROLL */ 808d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt 818d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt#undef ROUND 828d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt 838d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt /* 848d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt * apply last round and 858d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt * map cipher state to byte array block: 868d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt */ 878d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt s0 = TE41(t0) ^ TE42(t1) ^ TE43(t2) ^ TE44(t3) ^ rk[0]; 888d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt PUTU32(ct , s0); 898d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt s1 = TE41(t1) ^ TE42(t2) ^ TE43(t3) ^ TE44(t0) ^ rk[1]; 908d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt PUTU32(ct + 4, s1); 918d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt s2 = TE41(t2) ^ TE42(t3) ^ TE43(t0) ^ TE44(t1) ^ rk[2]; 928d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt PUTU32(ct + 8, s2); 938d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt s3 = TE41(t3) ^ TE42(t0) ^ TE43(t1) ^ TE44(t2) ^ rk[3]; 948d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt PUTU32(ct + 12, s3); 958d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt} 968d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt 978d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt 988d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidtvoid * aes_encrypt_init(const u8 *key, size_t len) 998d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt{ 1008d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt u32 *rk; 1018d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt if (len != 16) 1028d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt return NULL; 1038d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt rk = os_malloc(AES_PRIV_SIZE); 1048d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt if (rk == NULL) 1058d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt return NULL; 1068d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt rijndaelKeySetupEnc(rk, key); 1078d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt return rk; 1088d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt} 1098d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt 1108d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt 1118d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidtvoid aes_encrypt(void *ctx, const u8 *plain, u8 *crypt) 1128d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt{ 1138d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt rijndaelEncrypt(ctx, plain, crypt); 1148d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt} 1158d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt 1168d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt 1178d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidtvoid aes_encrypt_deinit(void *ctx) 1188d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt{ 1198d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt os_memset(ctx, 0, AES_PRIV_SIZE); 1208d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt os_free(ctx); 1218d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt} 122