aes.c 49 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530
  1. /*
  2. * FIPS-197 compliant AES implementation
  3. *
  4. * Copyright (C) 2006-2015, ARM Limited, All Rights Reserved
  5. * SPDX-License-Identifier: Apache-2.0
  6. *
  7. * Licensed under the Apache License, Version 2.0 (the "License"); you may
  8. * not use this file except in compliance with the License.
  9. * You may obtain a copy of the License at
  10. *
  11. * http://www.apache.org/licenses/LICENSE-2.0
  12. *
  13. * Unless required by applicable law or agreed to in writing, software
  14. * distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
  15. * WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  16. * See the License for the specific language governing permissions and
  17. * limitations under the License.
  18. *
  19. * This file is part of mbed TLS (https://tls.mbed.org)
  20. */
  21. /*
  22. * The AES block cipher was designed by Vincent Rijmen and Joan Daemen.
  23. *
  24. * http://csrc.nist.gov/encryption/aes/rijndael/Rijndael.pdf
  25. * http://csrc.nist.gov/publications/fips/fips197/fips-197.pdf
  26. */
  27. #if !defined(MBEDTLS_CONFIG_FILE)
  28. #include "mbedtls/config.h"
  29. #else
  30. #include MBEDTLS_CONFIG_FILE
  31. #endif
  32. #if defined(MBEDTLS_AES_C)
  33. #include <string.h>
  34. #include "mbedtls/aes.h"
  35. #if defined(MBEDTLS_PADLOCK_C)
  36. #include "mbedtls/padlock.h"
  37. #endif
  38. #if defined(MBEDTLS_AESNI_C)
  39. #include "mbedtls/aesni.h"
  40. #endif
  41. #if defined(MBEDTLS_SELF_TEST)
  42. #if defined(MBEDTLS_PLATFORM_C)
  43. #include "mbedtls/platform.h"
  44. #else
  45. #include <stdio.h>
  46. #define mbedtls_printf printf
  47. #endif /* MBEDTLS_PLATFORM_C */
  48. #endif /* MBEDTLS_SELF_TEST */
  49. #if !defined(MBEDTLS_AES_ALT)
  50. /* Implementation that should never be optimized out by the compiler */
  51. static void mbedtls_zeroize( void *v, size_t n ) {
  52. volatile unsigned char *p = (unsigned char*)v; while( n-- ) *p++ = 0;
  53. }
  54. /*
  55. * 32-bit integer manipulation macros (little endian)
  56. */
  57. #ifndef GET_UINT32_LE
  58. #define GET_UINT32_LE(n,b,i) \
  59. { \
  60. (n) = ( (uint32_t) (b)[(i) ] ) \
  61. | ( (uint32_t) (b)[(i) + 1] << 8 ) \
  62. | ( (uint32_t) (b)[(i) + 2] << 16 ) \
  63. | ( (uint32_t) (b)[(i) + 3] << 24 ); \
  64. }
  65. #endif
  66. #ifndef PUT_UINT32_LE
  67. #define PUT_UINT32_LE(n,b,i) \
  68. { \
  69. (b)[(i) ] = (unsigned char) ( ( (n) ) & 0xFF ); \
  70. (b)[(i) + 1] = (unsigned char) ( ( (n) >> 8 ) & 0xFF ); \
  71. (b)[(i) + 2] = (unsigned char) ( ( (n) >> 16 ) & 0xFF ); \
  72. (b)[(i) + 3] = (unsigned char) ( ( (n) >> 24 ) & 0xFF ); \
  73. }
  74. #endif
  75. #if defined(MBEDTLS_PADLOCK_C) && \
  76. ( defined(MBEDTLS_HAVE_X86) || defined(MBEDTLS_PADLOCK_ALIGN16) )
  77. static int aes_padlock_ace = -1;
  78. #endif
  79. #if defined(MBEDTLS_AES_ROM_TABLES)
  80. /*
  81. * Forward S-box
  82. */
  83. static const unsigned char FSb[256] =
  84. {
  85. 0x63, 0x7C, 0x77, 0x7B, 0xF2, 0x6B, 0x6F, 0xC5,
  86. 0x30, 0x01, 0x67, 0x2B, 0xFE, 0xD7, 0xAB, 0x76,
  87. 0xCA, 0x82, 0xC9, 0x7D, 0xFA, 0x59, 0x47, 0xF0,
  88. 0xAD, 0xD4, 0xA2, 0xAF, 0x9C, 0xA4, 0x72, 0xC0,
  89. 0xB7, 0xFD, 0x93, 0x26, 0x36, 0x3F, 0xF7, 0xCC,
  90. 0x34, 0xA5, 0xE5, 0xF1, 0x71, 0xD8, 0x31, 0x15,
  91. 0x04, 0xC7, 0x23, 0xC3, 0x18, 0x96, 0x05, 0x9A,
  92. 0x07, 0x12, 0x80, 0xE2, 0xEB, 0x27, 0xB2, 0x75,
  93. 0x09, 0x83, 0x2C, 0x1A, 0x1B, 0x6E, 0x5A, 0xA0,
  94. 0x52, 0x3B, 0xD6, 0xB3, 0x29, 0xE3, 0x2F, 0x84,
  95. 0x53, 0xD1, 0x00, 0xED, 0x20, 0xFC, 0xB1, 0x5B,
  96. 0x6A, 0xCB, 0xBE, 0x39, 0x4A, 0x4C, 0x58, 0xCF,
  97. 0xD0, 0xEF, 0xAA, 0xFB, 0x43, 0x4D, 0x33, 0x85,
  98. 0x45, 0xF9, 0x02, 0x7F, 0x50, 0x3C, 0x9F, 0xA8,
  99. 0x51, 0xA3, 0x40, 0x8F, 0x92, 0x9D, 0x38, 0xF5,
  100. 0xBC, 0xB6, 0xDA, 0x21, 0x10, 0xFF, 0xF3, 0xD2,
  101. 0xCD, 0x0C, 0x13, 0xEC, 0x5F, 0x97, 0x44, 0x17,
  102. 0xC4, 0xA7, 0x7E, 0x3D, 0x64, 0x5D, 0x19, 0x73,
  103. 0x60, 0x81, 0x4F, 0xDC, 0x22, 0x2A, 0x90, 0x88,
  104. 0x46, 0xEE, 0xB8, 0x14, 0xDE, 0x5E, 0x0B, 0xDB,
  105. 0xE0, 0x32, 0x3A, 0x0A, 0x49, 0x06, 0x24, 0x5C,
  106. 0xC2, 0xD3, 0xAC, 0x62, 0x91, 0x95, 0xE4, 0x79,
  107. 0xE7, 0xC8, 0x37, 0x6D, 0x8D, 0xD5, 0x4E, 0xA9,
  108. 0x6C, 0x56, 0xF4, 0xEA, 0x65, 0x7A, 0xAE, 0x08,
  109. 0xBA, 0x78, 0x25, 0x2E, 0x1C, 0xA6, 0xB4, 0xC6,
  110. 0xE8, 0xDD, 0x74, 0x1F, 0x4B, 0xBD, 0x8B, 0x8A,
  111. 0x70, 0x3E, 0xB5, 0x66, 0x48, 0x03, 0xF6, 0x0E,
  112. 0x61, 0x35, 0x57, 0xB9, 0x86, 0xC1, 0x1D, 0x9E,
  113. 0xE1, 0xF8, 0x98, 0x11, 0x69, 0xD9, 0x8E, 0x94,
  114. 0x9B, 0x1E, 0x87, 0xE9, 0xCE, 0x55, 0x28, 0xDF,
  115. 0x8C, 0xA1, 0x89, 0x0D, 0xBF, 0xE6, 0x42, 0x68,
  116. 0x41, 0x99, 0x2D, 0x0F, 0xB0, 0x54, 0xBB, 0x16
  117. };
  118. /*
  119. * Forward tables
  120. */
  121. #define FT \
  122. \
  123. V(A5,63,63,C6), V(84,7C,7C,F8), V(99,77,77,EE), V(8D,7B,7B,F6), \
  124. V(0D,F2,F2,FF), V(BD,6B,6B,D6), V(B1,6F,6F,DE), V(54,C5,C5,91), \
  125. V(50,30,30,60), V(03,01,01,02), V(A9,67,67,CE), V(7D,2B,2B,56), \
  126. V(19,FE,FE,E7), V(62,D7,D7,B5), V(E6,AB,AB,4D), V(9A,76,76,EC), \
  127. V(45,CA,CA,8F), V(9D,82,82,1F), V(40,C9,C9,89), V(87,7D,7D,FA), \
  128. V(15,FA,FA,EF), V(EB,59,59,B2), V(C9,47,47,8E), V(0B,F0,F0,FB), \
  129. V(EC,AD,AD,41), V(67,D4,D4,B3), V(FD,A2,A2,5F), V(EA,AF,AF,45), \
  130. V(BF,9C,9C,23), V(F7,A4,A4,53), V(96,72,72,E4), V(5B,C0,C0,9B), \
  131. V(C2,B7,B7,75), V(1C,FD,FD,E1), V(AE,93,93,3D), V(6A,26,26,4C), \
  132. V(5A,36,36,6C), V(41,3F,3F,7E), V(02,F7,F7,F5), V(4F,CC,CC,83), \
  133. V(5C,34,34,68), V(F4,A5,A5,51), V(34,E5,E5,D1), V(08,F1,F1,F9), \
  134. V(93,71,71,E2), V(73,D8,D8,AB), V(53,31,31,62), V(3F,15,15,2A), \
  135. V(0C,04,04,08), V(52,C7,C7,95), V(65,23,23,46), V(5E,C3,C3,9D), \
  136. V(28,18,18,30), V(A1,96,96,37), V(0F,05,05,0A), V(B5,9A,9A,2F), \
  137. V(09,07,07,0E), V(36,12,12,24), V(9B,80,80,1B), V(3D,E2,E2,DF), \
  138. V(26,EB,EB,CD), V(69,27,27,4E), V(CD,B2,B2,7F), V(9F,75,75,EA), \
  139. V(1B,09,09,12), V(9E,83,83,1D), V(74,2C,2C,58), V(2E,1A,1A,34), \
  140. V(2D,1B,1B,36), V(B2,6E,6E,DC), V(EE,5A,5A,B4), V(FB,A0,A0,5B), \
  141. V(F6,52,52,A4), V(4D,3B,3B,76), V(61,D6,D6,B7), V(CE,B3,B3,7D), \
  142. V(7B,29,29,52), V(3E,E3,E3,DD), V(71,2F,2F,5E), V(97,84,84,13), \
  143. V(F5,53,53,A6), V(68,D1,D1,B9), V(00,00,00,00), V(2C,ED,ED,C1), \
  144. V(60,20,20,40), V(1F,FC,FC,E3), V(C8,B1,B1,79), V(ED,5B,5B,B6), \
  145. V(BE,6A,6A,D4), V(46,CB,CB,8D), V(D9,BE,BE,67), V(4B,39,39,72), \
  146. V(DE,4A,4A,94), V(D4,4C,4C,98), V(E8,58,58,B0), V(4A,CF,CF,85), \
  147. V(6B,D0,D0,BB), V(2A,EF,EF,C5), V(E5,AA,AA,4F), V(16,FB,FB,ED), \
  148. V(C5,43,43,86), V(D7,4D,4D,9A), V(55,33,33,66), V(94,85,85,11), \
  149. V(CF,45,45,8A), V(10,F9,F9,E9), V(06,02,02,04), V(81,7F,7F,FE), \
  150. V(F0,50,50,A0), V(44,3C,3C,78), V(BA,9F,9F,25), V(E3,A8,A8,4B), \
  151. V(F3,51,51,A2), V(FE,A3,A3,5D), V(C0,40,40,80), V(8A,8F,8F,05), \
  152. V(AD,92,92,3F), V(BC,9D,9D,21), V(48,38,38,70), V(04,F5,F5,F1), \
  153. V(DF,BC,BC,63), V(C1,B6,B6,77), V(75,DA,DA,AF), V(63,21,21,42), \
  154. V(30,10,10,20), V(1A,FF,FF,E5), V(0E,F3,F3,FD), V(6D,D2,D2,BF), \
  155. V(4C,CD,CD,81), V(14,0C,0C,18), V(35,13,13,26), V(2F,EC,EC,C3), \
  156. V(E1,5F,5F,BE), V(A2,97,97,35), V(CC,44,44,88), V(39,17,17,2E), \
  157. V(57,C4,C4,93), V(F2,A7,A7,55), V(82,7E,7E,FC), V(47,3D,3D,7A), \
  158. V(AC,64,64,C8), V(E7,5D,5D,BA), V(2B,19,19,32), V(95,73,73,E6), \
  159. V(A0,60,60,C0), V(98,81,81,19), V(D1,4F,4F,9E), V(7F,DC,DC,A3), \
  160. V(66,22,22,44), V(7E,2A,2A,54), V(AB,90,90,3B), V(83,88,88,0B), \
  161. V(CA,46,46,8C), V(29,EE,EE,C7), V(D3,B8,B8,6B), V(3C,14,14,28), \
  162. V(79,DE,DE,A7), V(E2,5E,5E,BC), V(1D,0B,0B,16), V(76,DB,DB,AD), \
  163. V(3B,E0,E0,DB), V(56,32,32,64), V(4E,3A,3A,74), V(1E,0A,0A,14), \
  164. V(DB,49,49,92), V(0A,06,06,0C), V(6C,24,24,48), V(E4,5C,5C,B8), \
  165. V(5D,C2,C2,9F), V(6E,D3,D3,BD), V(EF,AC,AC,43), V(A6,62,62,C4), \
  166. V(A8,91,91,39), V(A4,95,95,31), V(37,E4,E4,D3), V(8B,79,79,F2), \
  167. V(32,E7,E7,D5), V(43,C8,C8,8B), V(59,37,37,6E), V(B7,6D,6D,DA), \
  168. V(8C,8D,8D,01), V(64,D5,D5,B1), V(D2,4E,4E,9C), V(E0,A9,A9,49), \
  169. V(B4,6C,6C,D8), V(FA,56,56,AC), V(07,F4,F4,F3), V(25,EA,EA,CF), \
  170. V(AF,65,65,CA), V(8E,7A,7A,F4), V(E9,AE,AE,47), V(18,08,08,10), \
  171. V(D5,BA,BA,6F), V(88,78,78,F0), V(6F,25,25,4A), V(72,2E,2E,5C), \
  172. V(24,1C,1C,38), V(F1,A6,A6,57), V(C7,B4,B4,73), V(51,C6,C6,97), \
  173. V(23,E8,E8,CB), V(7C,DD,DD,A1), V(9C,74,74,E8), V(21,1F,1F,3E), \
  174. V(DD,4B,4B,96), V(DC,BD,BD,61), V(86,8B,8B,0D), V(85,8A,8A,0F), \
  175. V(90,70,70,E0), V(42,3E,3E,7C), V(C4,B5,B5,71), V(AA,66,66,CC), \
  176. V(D8,48,48,90), V(05,03,03,06), V(01,F6,F6,F7), V(12,0E,0E,1C), \
  177. V(A3,61,61,C2), V(5F,35,35,6A), V(F9,57,57,AE), V(D0,B9,B9,69), \
  178. V(91,86,86,17), V(58,C1,C1,99), V(27,1D,1D,3A), V(B9,9E,9E,27), \
  179. V(38,E1,E1,D9), V(13,F8,F8,EB), V(B3,98,98,2B), V(33,11,11,22), \
  180. V(BB,69,69,D2), V(70,D9,D9,A9), V(89,8E,8E,07), V(A7,94,94,33), \
  181. V(B6,9B,9B,2D), V(22,1E,1E,3C), V(92,87,87,15), V(20,E9,E9,C9), \
  182. V(49,CE,CE,87), V(FF,55,55,AA), V(78,28,28,50), V(7A,DF,DF,A5), \
  183. V(8F,8C,8C,03), V(F8,A1,A1,59), V(80,89,89,09), V(17,0D,0D,1A), \
  184. V(DA,BF,BF,65), V(31,E6,E6,D7), V(C6,42,42,84), V(B8,68,68,D0), \
  185. V(C3,41,41,82), V(B0,99,99,29), V(77,2D,2D,5A), V(11,0F,0F,1E), \
  186. V(CB,B0,B0,7B), V(FC,54,54,A8), V(D6,BB,BB,6D), V(3A,16,16,2C)
  187. #define V(a,b,c,d) 0x##a##b##c##d
  188. static const uint32_t FT0[256] = { FT };
  189. #undef V
  190. #define V(a,b,c,d) 0x##b##c##d##a
  191. static const uint32_t FT1[256] = { FT };
  192. #undef V
  193. #define V(a,b,c,d) 0x##c##d##a##b
  194. static const uint32_t FT2[256] = { FT };
  195. #undef V
  196. #define V(a,b,c,d) 0x##d##a##b##c
  197. static const uint32_t FT3[256] = { FT };
  198. #undef V
  199. #undef FT
  200. /*
  201. * Reverse S-box
  202. */
  203. static const unsigned char RSb[256] =
  204. {
  205. 0x52, 0x09, 0x6A, 0xD5, 0x30, 0x36, 0xA5, 0x38,
  206. 0xBF, 0x40, 0xA3, 0x9E, 0x81, 0xF3, 0xD7, 0xFB,
  207. 0x7C, 0xE3, 0x39, 0x82, 0x9B, 0x2F, 0xFF, 0x87,
  208. 0x34, 0x8E, 0x43, 0x44, 0xC4, 0xDE, 0xE9, 0xCB,
  209. 0x54, 0x7B, 0x94, 0x32, 0xA6, 0xC2, 0x23, 0x3D,
  210. 0xEE, 0x4C, 0x95, 0x0B, 0x42, 0xFA, 0xC3, 0x4E,
  211. 0x08, 0x2E, 0xA1, 0x66, 0x28, 0xD9, 0x24, 0xB2,
  212. 0x76, 0x5B, 0xA2, 0x49, 0x6D, 0x8B, 0xD1, 0x25,
  213. 0x72, 0xF8, 0xF6, 0x64, 0x86, 0x68, 0x98, 0x16,
  214. 0xD4, 0xA4, 0x5C, 0xCC, 0x5D, 0x65, 0xB6, 0x92,
  215. 0x6C, 0x70, 0x48, 0x50, 0xFD, 0xED, 0xB9, 0xDA,
  216. 0x5E, 0x15, 0x46, 0x57, 0xA7, 0x8D, 0x9D, 0x84,
  217. 0x90, 0xD8, 0xAB, 0x00, 0x8C, 0xBC, 0xD3, 0x0A,
  218. 0xF7, 0xE4, 0x58, 0x05, 0xB8, 0xB3, 0x45, 0x06,
  219. 0xD0, 0x2C, 0x1E, 0x8F, 0xCA, 0x3F, 0x0F, 0x02,
  220. 0xC1, 0xAF, 0xBD, 0x03, 0x01, 0x13, 0x8A, 0x6B,
  221. 0x3A, 0x91, 0x11, 0x41, 0x4F, 0x67, 0xDC, 0xEA,
  222. 0x97, 0xF2, 0xCF, 0xCE, 0xF0, 0xB4, 0xE6, 0x73,
  223. 0x96, 0xAC, 0x74, 0x22, 0xE7, 0xAD, 0x35, 0x85,
  224. 0xE2, 0xF9, 0x37, 0xE8, 0x1C, 0x75, 0xDF, 0x6E,
  225. 0x47, 0xF1, 0x1A, 0x71, 0x1D, 0x29, 0xC5, 0x89,
  226. 0x6F, 0xB7, 0x62, 0x0E, 0xAA, 0x18, 0xBE, 0x1B,
  227. 0xFC, 0x56, 0x3E, 0x4B, 0xC6, 0xD2, 0x79, 0x20,
  228. 0x9A, 0xDB, 0xC0, 0xFE, 0x78, 0xCD, 0x5A, 0xF4,
  229. 0x1F, 0xDD, 0xA8, 0x33, 0x88, 0x07, 0xC7, 0x31,
  230. 0xB1, 0x12, 0x10, 0x59, 0x27, 0x80, 0xEC, 0x5F,
  231. 0x60, 0x51, 0x7F, 0xA9, 0x19, 0xB5, 0x4A, 0x0D,
  232. 0x2D, 0xE5, 0x7A, 0x9F, 0x93, 0xC9, 0x9C, 0xEF,
  233. 0xA0, 0xE0, 0x3B, 0x4D, 0xAE, 0x2A, 0xF5, 0xB0,
  234. 0xC8, 0xEB, 0xBB, 0x3C, 0x83, 0x53, 0x99, 0x61,
  235. 0x17, 0x2B, 0x04, 0x7E, 0xBA, 0x77, 0xD6, 0x26,
  236. 0xE1, 0x69, 0x14, 0x63, 0x55, 0x21, 0x0C, 0x7D
  237. };
  238. /*
  239. * Reverse tables
  240. */
  241. #define RT \
  242. \
  243. V(50,A7,F4,51), V(53,65,41,7E), V(C3,A4,17,1A), V(96,5E,27,3A), \
  244. V(CB,6B,AB,3B), V(F1,45,9D,1F), V(AB,58,FA,AC), V(93,03,E3,4B), \
  245. V(55,FA,30,20), V(F6,6D,76,AD), V(91,76,CC,88), V(25,4C,02,F5), \
  246. V(FC,D7,E5,4F), V(D7,CB,2A,C5), V(80,44,35,26), V(8F,A3,62,B5), \
  247. V(49,5A,B1,DE), V(67,1B,BA,25), V(98,0E,EA,45), V(E1,C0,FE,5D), \
  248. V(02,75,2F,C3), V(12,F0,4C,81), V(A3,97,46,8D), V(C6,F9,D3,6B), \
  249. V(E7,5F,8F,03), V(95,9C,92,15), V(EB,7A,6D,BF), V(DA,59,52,95), \
  250. V(2D,83,BE,D4), V(D3,21,74,58), V(29,69,E0,49), V(44,C8,C9,8E), \
  251. V(6A,89,C2,75), V(78,79,8E,F4), V(6B,3E,58,99), V(DD,71,B9,27), \
  252. V(B6,4F,E1,BE), V(17,AD,88,F0), V(66,AC,20,C9), V(B4,3A,CE,7D), \
  253. V(18,4A,DF,63), V(82,31,1A,E5), V(60,33,51,97), V(45,7F,53,62), \
  254. V(E0,77,64,B1), V(84,AE,6B,BB), V(1C,A0,81,FE), V(94,2B,08,F9), \
  255. V(58,68,48,70), V(19,FD,45,8F), V(87,6C,DE,94), V(B7,F8,7B,52), \
  256. V(23,D3,73,AB), V(E2,02,4B,72), V(57,8F,1F,E3), V(2A,AB,55,66), \
  257. V(07,28,EB,B2), V(03,C2,B5,2F), V(9A,7B,C5,86), V(A5,08,37,D3), \
  258. V(F2,87,28,30), V(B2,A5,BF,23), V(BA,6A,03,02), V(5C,82,16,ED), \
  259. V(2B,1C,CF,8A), V(92,B4,79,A7), V(F0,F2,07,F3), V(A1,E2,69,4E), \
  260. V(CD,F4,DA,65), V(D5,BE,05,06), V(1F,62,34,D1), V(8A,FE,A6,C4), \
  261. V(9D,53,2E,34), V(A0,55,F3,A2), V(32,E1,8A,05), V(75,EB,F6,A4), \
  262. V(39,EC,83,0B), V(AA,EF,60,40), V(06,9F,71,5E), V(51,10,6E,BD), \
  263. V(F9,8A,21,3E), V(3D,06,DD,96), V(AE,05,3E,DD), V(46,BD,E6,4D), \
  264. V(B5,8D,54,91), V(05,5D,C4,71), V(6F,D4,06,04), V(FF,15,50,60), \
  265. V(24,FB,98,19), V(97,E9,BD,D6), V(CC,43,40,89), V(77,9E,D9,67), \
  266. V(BD,42,E8,B0), V(88,8B,89,07), V(38,5B,19,E7), V(DB,EE,C8,79), \
  267. V(47,0A,7C,A1), V(E9,0F,42,7C), V(C9,1E,84,F8), V(00,00,00,00), \
  268. V(83,86,80,09), V(48,ED,2B,32), V(AC,70,11,1E), V(4E,72,5A,6C), \
  269. V(FB,FF,0E,FD), V(56,38,85,0F), V(1E,D5,AE,3D), V(27,39,2D,36), \
  270. V(64,D9,0F,0A), V(21,A6,5C,68), V(D1,54,5B,9B), V(3A,2E,36,24), \
  271. V(B1,67,0A,0C), V(0F,E7,57,93), V(D2,96,EE,B4), V(9E,91,9B,1B), \
  272. V(4F,C5,C0,80), V(A2,20,DC,61), V(69,4B,77,5A), V(16,1A,12,1C), \
  273. V(0A,BA,93,E2), V(E5,2A,A0,C0), V(43,E0,22,3C), V(1D,17,1B,12), \
  274. V(0B,0D,09,0E), V(AD,C7,8B,F2), V(B9,A8,B6,2D), V(C8,A9,1E,14), \
  275. V(85,19,F1,57), V(4C,07,75,AF), V(BB,DD,99,EE), V(FD,60,7F,A3), \
  276. V(9F,26,01,F7), V(BC,F5,72,5C), V(C5,3B,66,44), V(34,7E,FB,5B), \
  277. V(76,29,43,8B), V(DC,C6,23,CB), V(68,FC,ED,B6), V(63,F1,E4,B8), \
  278. V(CA,DC,31,D7), V(10,85,63,42), V(40,22,97,13), V(20,11,C6,84), \
  279. V(7D,24,4A,85), V(F8,3D,BB,D2), V(11,32,F9,AE), V(6D,A1,29,C7), \
  280. V(4B,2F,9E,1D), V(F3,30,B2,DC), V(EC,52,86,0D), V(D0,E3,C1,77), \
  281. V(6C,16,B3,2B), V(99,B9,70,A9), V(FA,48,94,11), V(22,64,E9,47), \
  282. V(C4,8C,FC,A8), V(1A,3F,F0,A0), V(D8,2C,7D,56), V(EF,90,33,22), \
  283. V(C7,4E,49,87), V(C1,D1,38,D9), V(FE,A2,CA,8C), V(36,0B,D4,98), \
  284. V(CF,81,F5,A6), V(28,DE,7A,A5), V(26,8E,B7,DA), V(A4,BF,AD,3F), \
  285. V(E4,9D,3A,2C), V(0D,92,78,50), V(9B,CC,5F,6A), V(62,46,7E,54), \
  286. V(C2,13,8D,F6), V(E8,B8,D8,90), V(5E,F7,39,2E), V(F5,AF,C3,82), \
  287. V(BE,80,5D,9F), V(7C,93,D0,69), V(A9,2D,D5,6F), V(B3,12,25,CF), \
  288. V(3B,99,AC,C8), V(A7,7D,18,10), V(6E,63,9C,E8), V(7B,BB,3B,DB), \
  289. V(09,78,26,CD), V(F4,18,59,6E), V(01,B7,9A,EC), V(A8,9A,4F,83), \
  290. V(65,6E,95,E6), V(7E,E6,FF,AA), V(08,CF,BC,21), V(E6,E8,15,EF), \
  291. V(D9,9B,E7,BA), V(CE,36,6F,4A), V(D4,09,9F,EA), V(D6,7C,B0,29), \
  292. V(AF,B2,A4,31), V(31,23,3F,2A), V(30,94,A5,C6), V(C0,66,A2,35), \
  293. V(37,BC,4E,74), V(A6,CA,82,FC), V(B0,D0,90,E0), V(15,D8,A7,33), \
  294. V(4A,98,04,F1), V(F7,DA,EC,41), V(0E,50,CD,7F), V(2F,F6,91,17), \
  295. V(8D,D6,4D,76), V(4D,B0,EF,43), V(54,4D,AA,CC), V(DF,04,96,E4), \
  296. V(E3,B5,D1,9E), V(1B,88,6A,4C), V(B8,1F,2C,C1), V(7F,51,65,46), \
  297. V(04,EA,5E,9D), V(5D,35,8C,01), V(73,74,87,FA), V(2E,41,0B,FB), \
  298. V(5A,1D,67,B3), V(52,D2,DB,92), V(33,56,10,E9), V(13,47,D6,6D), \
  299. V(8C,61,D7,9A), V(7A,0C,A1,37), V(8E,14,F8,59), V(89,3C,13,EB), \
  300. V(EE,27,A9,CE), V(35,C9,61,B7), V(ED,E5,1C,E1), V(3C,B1,47,7A), \
  301. V(59,DF,D2,9C), V(3F,73,F2,55), V(79,CE,14,18), V(BF,37,C7,73), \
  302. V(EA,CD,F7,53), V(5B,AA,FD,5F), V(14,6F,3D,DF), V(86,DB,44,78), \
  303. V(81,F3,AF,CA), V(3E,C4,68,B9), V(2C,34,24,38), V(5F,40,A3,C2), \
  304. V(72,C3,1D,16), V(0C,25,E2,BC), V(8B,49,3C,28), V(41,95,0D,FF), \
  305. V(71,01,A8,39), V(DE,B3,0C,08), V(9C,E4,B4,D8), V(90,C1,56,64), \
  306. V(61,84,CB,7B), V(70,B6,32,D5), V(74,5C,6C,48), V(42,57,B8,D0)
  307. #define V(a,b,c,d) 0x##a##b##c##d
  308. static const uint32_t RT0[256] = { RT };
  309. #undef V
  310. #define V(a,b,c,d) 0x##b##c##d##a
  311. static const uint32_t RT1[256] = { RT };
  312. #undef V
  313. #define V(a,b,c,d) 0x##c##d##a##b
  314. static const uint32_t RT2[256] = { RT };
  315. #undef V
  316. #define V(a,b,c,d) 0x##d##a##b##c
  317. static const uint32_t RT3[256] = { RT };
  318. #undef V
  319. #undef RT
  320. /*
  321. * Round constants
  322. */
  323. static const uint32_t RCON[10] =
  324. {
  325. 0x00000001, 0x00000002, 0x00000004, 0x00000008,
  326. 0x00000010, 0x00000020, 0x00000040, 0x00000080,
  327. 0x0000001B, 0x00000036
  328. };
  329. #else /* MBEDTLS_AES_ROM_TABLES */
  330. /*
  331. * Forward S-box & tables
  332. */
  333. static unsigned char FSb[256];
  334. static uint32_t FT0[256];
  335. static uint32_t FT1[256];
  336. static uint32_t FT2[256];
  337. static uint32_t FT3[256];
  338. /*
  339. * Reverse S-box & tables
  340. */
  341. static unsigned char RSb[256];
  342. static uint32_t RT0[256];
  343. static uint32_t RT1[256];
  344. static uint32_t RT2[256];
  345. static uint32_t RT3[256];
  346. /*
  347. * Round constants
  348. */
  349. static uint32_t RCON[10];
  350. /*
  351. * Tables generation code
  352. */
  353. #define ROTL8(x) ( ( x << 8 ) & 0xFFFFFFFF ) | ( x >> 24 )
  354. #define XTIME(x) ( ( x << 1 ) ^ ( ( x & 0x80 ) ? 0x1B : 0x00 ) )
  355. #define MUL(x,y) ( ( x && y ) ? pow[(log[x]+log[y]) % 255] : 0 )
  356. static int aes_init_done = 0;
  357. static void aes_gen_tables( void )
  358. {
  359. int i, x, y, z;
  360. int pow[256];
  361. int log[256];
  362. /*
  363. * compute pow and log tables over GF(2^8)
  364. */
  365. for( i = 0, x = 1; i < 256; i++ )
  366. {
  367. pow[i] = x;
  368. log[x] = i;
  369. x = ( x ^ XTIME( x ) ) & 0xFF;
  370. }
  371. /*
  372. * calculate the round constants
  373. */
  374. for( i = 0, x = 1; i < 10; i++ )
  375. {
  376. RCON[i] = (uint32_t) x;
  377. x = XTIME( x ) & 0xFF;
  378. }
  379. /*
  380. * generate the forward and reverse S-boxes
  381. */
  382. FSb[0x00] = 0x63;
  383. RSb[0x63] = 0x00;
  384. for( i = 1; i < 256; i++ )
  385. {
  386. x = pow[255 - log[i]];
  387. y = x; y = ( ( y << 1 ) | ( y >> 7 ) ) & 0xFF;
  388. x ^= y; y = ( ( y << 1 ) | ( y >> 7 ) ) & 0xFF;
  389. x ^= y; y = ( ( y << 1 ) | ( y >> 7 ) ) & 0xFF;
  390. x ^= y; y = ( ( y << 1 ) | ( y >> 7 ) ) & 0xFF;
  391. x ^= y ^ 0x63;
  392. FSb[i] = (unsigned char) x;
  393. RSb[x] = (unsigned char) i;
  394. }
  395. /*
  396. * generate the forward and reverse tables
  397. */
  398. for( i = 0; i < 256; i++ )
  399. {
  400. x = FSb[i];
  401. y = XTIME( x ) & 0xFF;
  402. z = ( y ^ x ) & 0xFF;
  403. FT0[i] = ( (uint32_t) y ) ^
  404. ( (uint32_t) x << 8 ) ^
  405. ( (uint32_t) x << 16 ) ^
  406. ( (uint32_t) z << 24 );
  407. FT1[i] = ROTL8( FT0[i] );
  408. FT2[i] = ROTL8( FT1[i] );
  409. FT3[i] = ROTL8( FT2[i] );
  410. x = RSb[i];
  411. RT0[i] = ( (uint32_t) MUL( 0x0E, x ) ) ^
  412. ( (uint32_t) MUL( 0x09, x ) << 8 ) ^
  413. ( (uint32_t) MUL( 0x0D, x ) << 16 ) ^
  414. ( (uint32_t) MUL( 0x0B, x ) << 24 );
  415. RT1[i] = ROTL8( RT0[i] );
  416. RT2[i] = ROTL8( RT1[i] );
  417. RT3[i] = ROTL8( RT2[i] );
  418. }
  419. }
  420. #endif /* MBEDTLS_AES_ROM_TABLES */
  421. void mbedtls_aes_init( mbedtls_aes_context *ctx )
  422. {
  423. memset( ctx, 0, sizeof( mbedtls_aes_context ) );
  424. }
  425. void mbedtls_aes_free( mbedtls_aes_context *ctx )
  426. {
  427. if( ctx == NULL )
  428. return;
  429. mbedtls_zeroize( ctx, sizeof( mbedtls_aes_context ) );
  430. }
  431. /*
  432. * AES key schedule (encryption)
  433. */
  434. #if !defined(MBEDTLS_AES_SETKEY_ENC_ALT)
  435. int mbedtls_aes_setkey_enc( mbedtls_aes_context *ctx, const unsigned char *key,
  436. unsigned int keybits )
  437. {
  438. unsigned int i;
  439. uint32_t *RK;
  440. #if !defined(MBEDTLS_AES_ROM_TABLES)
  441. if( aes_init_done == 0 )
  442. {
  443. aes_gen_tables();
  444. aes_init_done = 1;
  445. }
  446. #endif
  447. switch( keybits )
  448. {
  449. case 128: ctx->nr = 10; break;
  450. case 192: ctx->nr = 12; break;
  451. case 256: ctx->nr = 14; break;
  452. default : return( MBEDTLS_ERR_AES_INVALID_KEY_LENGTH );
  453. }
  454. #if defined(MBEDTLS_PADLOCK_C) && defined(MBEDTLS_PADLOCK_ALIGN16)
  455. if( aes_padlock_ace == -1 )
  456. aes_padlock_ace = mbedtls_padlock_has_support( MBEDTLS_PADLOCK_ACE );
  457. if( aes_padlock_ace )
  458. ctx->rk = RK = MBEDTLS_PADLOCK_ALIGN16( ctx->buf );
  459. else
  460. #endif
  461. ctx->rk = RK = ctx->buf;
  462. #if defined(MBEDTLS_AESNI_C) && defined(MBEDTLS_HAVE_X86_64)
  463. if( mbedtls_aesni_has_support( MBEDTLS_AESNI_AES ) )
  464. return( mbedtls_aesni_setkey_enc( (unsigned char *) ctx->rk, key, keybits ) );
  465. #endif
  466. for( i = 0; i < ( keybits >> 5 ); i++ )
  467. {
  468. GET_UINT32_LE( RK[i], key, i << 2 );
  469. }
  470. switch( ctx->nr )
  471. {
  472. case 10:
  473. for( i = 0; i < 10; i++, RK += 4 )
  474. {
  475. RK[4] = RK[0] ^ RCON[i] ^
  476. ( (uint32_t) FSb[ ( RK[3] >> 8 ) & 0xFF ] ) ^
  477. ( (uint32_t) FSb[ ( RK[3] >> 16 ) & 0xFF ] << 8 ) ^
  478. ( (uint32_t) FSb[ ( RK[3] >> 24 ) & 0xFF ] << 16 ) ^
  479. ( (uint32_t) FSb[ ( RK[3] ) & 0xFF ] << 24 );
  480. RK[5] = RK[1] ^ RK[4];
  481. RK[6] = RK[2] ^ RK[5];
  482. RK[7] = RK[3] ^ RK[6];
  483. }
  484. break;
  485. case 12:
  486. for( i = 0; i < 8; i++, RK += 6 )
  487. {
  488. RK[6] = RK[0] ^ RCON[i] ^
  489. ( (uint32_t) FSb[ ( RK[5] >> 8 ) & 0xFF ] ) ^
  490. ( (uint32_t) FSb[ ( RK[5] >> 16 ) & 0xFF ] << 8 ) ^
  491. ( (uint32_t) FSb[ ( RK[5] >> 24 ) & 0xFF ] << 16 ) ^
  492. ( (uint32_t) FSb[ ( RK[5] ) & 0xFF ] << 24 );
  493. RK[7] = RK[1] ^ RK[6];
  494. RK[8] = RK[2] ^ RK[7];
  495. RK[9] = RK[3] ^ RK[8];
  496. RK[10] = RK[4] ^ RK[9];
  497. RK[11] = RK[5] ^ RK[10];
  498. }
  499. break;
  500. case 14:
  501. for( i = 0; i < 7; i++, RK += 8 )
  502. {
  503. RK[8] = RK[0] ^ RCON[i] ^
  504. ( (uint32_t) FSb[ ( RK[7] >> 8 ) & 0xFF ] ) ^
  505. ( (uint32_t) FSb[ ( RK[7] >> 16 ) & 0xFF ] << 8 ) ^
  506. ( (uint32_t) FSb[ ( RK[7] >> 24 ) & 0xFF ] << 16 ) ^
  507. ( (uint32_t) FSb[ ( RK[7] ) & 0xFF ] << 24 );
  508. RK[9] = RK[1] ^ RK[8];
  509. RK[10] = RK[2] ^ RK[9];
  510. RK[11] = RK[3] ^ RK[10];
  511. RK[12] = RK[4] ^
  512. ( (uint32_t) FSb[ ( RK[11] ) & 0xFF ] ) ^
  513. ( (uint32_t) FSb[ ( RK[11] >> 8 ) & 0xFF ] << 8 ) ^
  514. ( (uint32_t) FSb[ ( RK[11] >> 16 ) & 0xFF ] << 16 ) ^
  515. ( (uint32_t) FSb[ ( RK[11] >> 24 ) & 0xFF ] << 24 );
  516. RK[13] = RK[5] ^ RK[12];
  517. RK[14] = RK[6] ^ RK[13];
  518. RK[15] = RK[7] ^ RK[14];
  519. }
  520. break;
  521. }
  522. return( 0 );
  523. }
  524. #endif /* !MBEDTLS_AES_SETKEY_ENC_ALT */
  525. /*
  526. * AES key schedule (decryption)
  527. */
  528. #if !defined(MBEDTLS_AES_SETKEY_DEC_ALT)
  529. int mbedtls_aes_setkey_dec( mbedtls_aes_context *ctx, const unsigned char *key,
  530. unsigned int keybits )
  531. {
  532. int i, j, ret;
  533. mbedtls_aes_context cty;
  534. uint32_t *RK;
  535. uint32_t *SK;
  536. mbedtls_aes_init( &cty );
  537. #if defined(MBEDTLS_PADLOCK_C) && defined(MBEDTLS_PADLOCK_ALIGN16)
  538. if( aes_padlock_ace == -1 )
  539. aes_padlock_ace = mbedtls_padlock_has_support( MBEDTLS_PADLOCK_ACE );
  540. if( aes_padlock_ace )
  541. ctx->rk = RK = MBEDTLS_PADLOCK_ALIGN16( ctx->buf );
  542. else
  543. #endif
  544. ctx->rk = RK = ctx->buf;
  545. /* Also checks keybits */
  546. if( ( ret = mbedtls_aes_setkey_enc( &cty, key, keybits ) ) != 0 )
  547. goto exit;
  548. ctx->nr = cty.nr;
  549. #if defined(MBEDTLS_AESNI_C) && defined(MBEDTLS_HAVE_X86_64)
  550. if( mbedtls_aesni_has_support( MBEDTLS_AESNI_AES ) )
  551. {
  552. mbedtls_aesni_inverse_key( (unsigned char *) ctx->rk,
  553. (const unsigned char *) cty.rk, ctx->nr );
  554. goto exit;
  555. }
  556. #endif
  557. SK = cty.rk + cty.nr * 4;
  558. *RK++ = *SK++;
  559. *RK++ = *SK++;
  560. *RK++ = *SK++;
  561. *RK++ = *SK++;
  562. for( i = ctx->nr - 1, SK -= 8; i > 0; i--, SK -= 8 )
  563. {
  564. for( j = 0; j < 4; j++, SK++ )
  565. {
  566. *RK++ = RT0[ FSb[ ( *SK ) & 0xFF ] ] ^
  567. RT1[ FSb[ ( *SK >> 8 ) & 0xFF ] ] ^
  568. RT2[ FSb[ ( *SK >> 16 ) & 0xFF ] ] ^
  569. RT3[ FSb[ ( *SK >> 24 ) & 0xFF ] ];
  570. }
  571. }
  572. *RK++ = *SK++;
  573. *RK++ = *SK++;
  574. *RK++ = *SK++;
  575. *RK++ = *SK++;
  576. exit:
  577. mbedtls_aes_free( &cty );
  578. return( ret );
  579. }
  580. #endif /* !MBEDTLS_AES_SETKEY_DEC_ALT */
  581. #define AES_FROUND(X0,X1,X2,X3,Y0,Y1,Y2,Y3) \
  582. { \
  583. X0 = *RK++ ^ FT0[ ( Y0 ) & 0xFF ] ^ \
  584. FT1[ ( Y1 >> 8 ) & 0xFF ] ^ \
  585. FT2[ ( Y2 >> 16 ) & 0xFF ] ^ \
  586. FT3[ ( Y3 >> 24 ) & 0xFF ]; \
  587. \
  588. X1 = *RK++ ^ FT0[ ( Y1 ) & 0xFF ] ^ \
  589. FT1[ ( Y2 >> 8 ) & 0xFF ] ^ \
  590. FT2[ ( Y3 >> 16 ) & 0xFF ] ^ \
  591. FT3[ ( Y0 >> 24 ) & 0xFF ]; \
  592. \
  593. X2 = *RK++ ^ FT0[ ( Y2 ) & 0xFF ] ^ \
  594. FT1[ ( Y3 >> 8 ) & 0xFF ] ^ \
  595. FT2[ ( Y0 >> 16 ) & 0xFF ] ^ \
  596. FT3[ ( Y1 >> 24 ) & 0xFF ]; \
  597. \
  598. X3 = *RK++ ^ FT0[ ( Y3 ) & 0xFF ] ^ \
  599. FT1[ ( Y0 >> 8 ) & 0xFF ] ^ \
  600. FT2[ ( Y1 >> 16 ) & 0xFF ] ^ \
  601. FT3[ ( Y2 >> 24 ) & 0xFF ]; \
  602. }
  603. #define AES_RROUND(X0,X1,X2,X3,Y0,Y1,Y2,Y3) \
  604. { \
  605. X0 = *RK++ ^ RT0[ ( Y0 ) & 0xFF ] ^ \
  606. RT1[ ( Y3 >> 8 ) & 0xFF ] ^ \
  607. RT2[ ( Y2 >> 16 ) & 0xFF ] ^ \
  608. RT3[ ( Y1 >> 24 ) & 0xFF ]; \
  609. \
  610. X1 = *RK++ ^ RT0[ ( Y1 ) & 0xFF ] ^ \
  611. RT1[ ( Y0 >> 8 ) & 0xFF ] ^ \
  612. RT2[ ( Y3 >> 16 ) & 0xFF ] ^ \
  613. RT3[ ( Y2 >> 24 ) & 0xFF ]; \
  614. \
  615. X2 = *RK++ ^ RT0[ ( Y2 ) & 0xFF ] ^ \
  616. RT1[ ( Y1 >> 8 ) & 0xFF ] ^ \
  617. RT2[ ( Y0 >> 16 ) & 0xFF ] ^ \
  618. RT3[ ( Y3 >> 24 ) & 0xFF ]; \
  619. \
  620. X3 = *RK++ ^ RT0[ ( Y3 ) & 0xFF ] ^ \
  621. RT1[ ( Y2 >> 8 ) & 0xFF ] ^ \
  622. RT2[ ( Y1 >> 16 ) & 0xFF ] ^ \
  623. RT3[ ( Y0 >> 24 ) & 0xFF ]; \
  624. }
  625. /*
  626. * AES-ECB block encryption
  627. */
  628. #if !defined(MBEDTLS_AES_ENCRYPT_ALT)
  629. int mbedtls_internal_aes_encrypt( mbedtls_aes_context *ctx,
  630. const unsigned char input[16],
  631. unsigned char output[16] )
  632. {
  633. int i;
  634. uint32_t *RK, X0, X1, X2, X3, Y0, Y1, Y2, Y3;
  635. RK = ctx->rk;
  636. GET_UINT32_LE( X0, input, 0 ); X0 ^= *RK++;
  637. GET_UINT32_LE( X1, input, 4 ); X1 ^= *RK++;
  638. GET_UINT32_LE( X2, input, 8 ); X2 ^= *RK++;
  639. GET_UINT32_LE( X3, input, 12 ); X3 ^= *RK++;
  640. for( i = ( ctx->nr >> 1 ) - 1; i > 0; i-- )
  641. {
  642. AES_FROUND( Y0, Y1, Y2, Y3, X0, X1, X2, X3 );
  643. AES_FROUND( X0, X1, X2, X3, Y0, Y1, Y2, Y3 );
  644. }
  645. AES_FROUND( Y0, Y1, Y2, Y3, X0, X1, X2, X3 );
  646. X0 = *RK++ ^ \
  647. ( (uint32_t) FSb[ ( Y0 ) & 0xFF ] ) ^
  648. ( (uint32_t) FSb[ ( Y1 >> 8 ) & 0xFF ] << 8 ) ^
  649. ( (uint32_t) FSb[ ( Y2 >> 16 ) & 0xFF ] << 16 ) ^
  650. ( (uint32_t) FSb[ ( Y3 >> 24 ) & 0xFF ] << 24 );
  651. X1 = *RK++ ^ \
  652. ( (uint32_t) FSb[ ( Y1 ) & 0xFF ] ) ^
  653. ( (uint32_t) FSb[ ( Y2 >> 8 ) & 0xFF ] << 8 ) ^
  654. ( (uint32_t) FSb[ ( Y3 >> 16 ) & 0xFF ] << 16 ) ^
  655. ( (uint32_t) FSb[ ( Y0 >> 24 ) & 0xFF ] << 24 );
  656. X2 = *RK++ ^ \
  657. ( (uint32_t) FSb[ ( Y2 ) & 0xFF ] ) ^
  658. ( (uint32_t) FSb[ ( Y3 >> 8 ) & 0xFF ] << 8 ) ^
  659. ( (uint32_t) FSb[ ( Y0 >> 16 ) & 0xFF ] << 16 ) ^
  660. ( (uint32_t) FSb[ ( Y1 >> 24 ) & 0xFF ] << 24 );
  661. X3 = *RK++ ^ \
  662. ( (uint32_t) FSb[ ( Y3 ) & 0xFF ] ) ^
  663. ( (uint32_t) FSb[ ( Y0 >> 8 ) & 0xFF ] << 8 ) ^
  664. ( (uint32_t) FSb[ ( Y1 >> 16 ) & 0xFF ] << 16 ) ^
  665. ( (uint32_t) FSb[ ( Y2 >> 24 ) & 0xFF ] << 24 );
  666. PUT_UINT32_LE( X0, output, 0 );
  667. PUT_UINT32_LE( X1, output, 4 );
  668. PUT_UINT32_LE( X2, output, 8 );
  669. PUT_UINT32_LE( X3, output, 12 );
  670. return( 0 );
  671. }
  672. #endif /* !MBEDTLS_AES_ENCRYPT_ALT */
  673. #if !defined(MBEDTLS_DEPRECATED_REMOVED)
  674. void mbedtls_aes_encrypt( mbedtls_aes_context *ctx,
  675. const unsigned char input[16],
  676. unsigned char output[16] )
  677. {
  678. mbedtls_internal_aes_encrypt( ctx, input, output );
  679. }
  680. #endif /* !MBEDTLS_DEPRECATED_REMOVED */
  681. /*
  682. * AES-ECB block decryption
  683. */
  684. #if !defined(MBEDTLS_AES_DECRYPT_ALT)
  685. int mbedtls_internal_aes_decrypt( mbedtls_aes_context *ctx,
  686. const unsigned char input[16],
  687. unsigned char output[16] )
  688. {
  689. int i;
  690. uint32_t *RK, X0, X1, X2, X3, Y0, Y1, Y2, Y3;
  691. RK = ctx->rk;
  692. GET_UINT32_LE( X0, input, 0 ); X0 ^= *RK++;
  693. GET_UINT32_LE( X1, input, 4 ); X1 ^= *RK++;
  694. GET_UINT32_LE( X2, input, 8 ); X2 ^= *RK++;
  695. GET_UINT32_LE( X3, input, 12 ); X3 ^= *RK++;
  696. for( i = ( ctx->nr >> 1 ) - 1; i > 0; i-- )
  697. {
  698. AES_RROUND( Y0, Y1, Y2, Y3, X0, X1, X2, X3 );
  699. AES_RROUND( X0, X1, X2, X3, Y0, Y1, Y2, Y3 );
  700. }
  701. AES_RROUND( Y0, Y1, Y2, Y3, X0, X1, X2, X3 );
  702. X0 = *RK++ ^ \
  703. ( (uint32_t) RSb[ ( Y0 ) & 0xFF ] ) ^
  704. ( (uint32_t) RSb[ ( Y3 >> 8 ) & 0xFF ] << 8 ) ^
  705. ( (uint32_t) RSb[ ( Y2 >> 16 ) & 0xFF ] << 16 ) ^
  706. ( (uint32_t) RSb[ ( Y1 >> 24 ) & 0xFF ] << 24 );
  707. X1 = *RK++ ^ \
  708. ( (uint32_t) RSb[ ( Y1 ) & 0xFF ] ) ^
  709. ( (uint32_t) RSb[ ( Y0 >> 8 ) & 0xFF ] << 8 ) ^
  710. ( (uint32_t) RSb[ ( Y3 >> 16 ) & 0xFF ] << 16 ) ^
  711. ( (uint32_t) RSb[ ( Y2 >> 24 ) & 0xFF ] << 24 );
  712. X2 = *RK++ ^ \
  713. ( (uint32_t) RSb[ ( Y2 ) & 0xFF ] ) ^
  714. ( (uint32_t) RSb[ ( Y1 >> 8 ) & 0xFF ] << 8 ) ^
  715. ( (uint32_t) RSb[ ( Y0 >> 16 ) & 0xFF ] << 16 ) ^
  716. ( (uint32_t) RSb[ ( Y3 >> 24 ) & 0xFF ] << 24 );
  717. X3 = *RK++ ^ \
  718. ( (uint32_t) RSb[ ( Y3 ) & 0xFF ] ) ^
  719. ( (uint32_t) RSb[ ( Y2 >> 8 ) & 0xFF ] << 8 ) ^
  720. ( (uint32_t) RSb[ ( Y1 >> 16 ) & 0xFF ] << 16 ) ^
  721. ( (uint32_t) RSb[ ( Y0 >> 24 ) & 0xFF ] << 24 );
  722. PUT_UINT32_LE( X0, output, 0 );
  723. PUT_UINT32_LE( X1, output, 4 );
  724. PUT_UINT32_LE( X2, output, 8 );
  725. PUT_UINT32_LE( X3, output, 12 );
  726. return( 0 );
  727. }
  728. #endif /* !MBEDTLS_AES_DECRYPT_ALT */
  729. #if !defined(MBEDTLS_DEPRECATED_REMOVED)
  730. void mbedtls_aes_decrypt( mbedtls_aes_context *ctx,
  731. const unsigned char input[16],
  732. unsigned char output[16] )
  733. {
  734. mbedtls_internal_aes_decrypt( ctx, input, output );
  735. }
  736. #endif /* !MBEDTLS_DEPRECATED_REMOVED */
  737. /*
  738. * AES-ECB block encryption/decryption
  739. */
  740. int mbedtls_aes_crypt_ecb( mbedtls_aes_context *ctx,
  741. int mode,
  742. const unsigned char input[16],
  743. unsigned char output[16] )
  744. {
  745. #if defined(MBEDTLS_AESNI_C) && defined(MBEDTLS_HAVE_X86_64)
  746. if( mbedtls_aesni_has_support( MBEDTLS_AESNI_AES ) )
  747. return( mbedtls_aesni_crypt_ecb( ctx, mode, input, output ) );
  748. #endif
  749. #if defined(MBEDTLS_PADLOCK_C) && defined(MBEDTLS_HAVE_X86)
  750. if( aes_padlock_ace )
  751. {
  752. if( mbedtls_padlock_xcryptecb( ctx, mode, input, output ) == 0 )
  753. return( 0 );
  754. // If padlock data misaligned, we just fall back to
  755. // unaccelerated mode
  756. //
  757. }
  758. #endif
  759. if( mode == MBEDTLS_AES_ENCRYPT )
  760. return( mbedtls_internal_aes_encrypt( ctx, input, output ) );
  761. else
  762. return( mbedtls_internal_aes_decrypt( ctx, input, output ) );
  763. }
  764. #if defined(MBEDTLS_CIPHER_MODE_CBC)
  765. /*
  766. * AES-CBC buffer encryption/decryption
  767. */
  768. int mbedtls_aes_crypt_cbc( mbedtls_aes_context *ctx,
  769. int mode,
  770. size_t length,
  771. unsigned char iv[16],
  772. const unsigned char *input,
  773. unsigned char *output )
  774. {
  775. int i;
  776. unsigned char temp[16];
  777. if( length % 16 )
  778. return( MBEDTLS_ERR_AES_INVALID_INPUT_LENGTH );
  779. #if defined(MBEDTLS_PADLOCK_C) && defined(MBEDTLS_HAVE_X86)
  780. if( aes_padlock_ace )
  781. {
  782. if( mbedtls_padlock_xcryptcbc( ctx, mode, length, iv, input, output ) == 0 )
  783. return( 0 );
  784. // If padlock data misaligned, we just fall back to
  785. // unaccelerated mode
  786. //
  787. }
  788. #endif
  789. if( mode == MBEDTLS_AES_DECRYPT )
  790. {
  791. while( length > 0 )
  792. {
  793. memcpy( temp, input, 16 );
  794. mbedtls_aes_crypt_ecb( ctx, mode, input, output );
  795. for( i = 0; i < 16; i++ )
  796. output[i] = (unsigned char)( output[i] ^ iv[i] );
  797. memcpy( iv, temp, 16 );
  798. input += 16;
  799. output += 16;
  800. length -= 16;
  801. }
  802. }
  803. else
  804. {
  805. while( length > 0 )
  806. {
  807. for( i = 0; i < 16; i++ )
  808. output[i] = (unsigned char)( input[i] ^ iv[i] );
  809. mbedtls_aes_crypt_ecb( ctx, mode, output, output );
  810. memcpy( iv, output, 16 );
  811. input += 16;
  812. output += 16;
  813. length -= 16;
  814. }
  815. }
  816. return( 0 );
  817. }
  818. #endif /* MBEDTLS_CIPHER_MODE_CBC */
  819. #if defined(MBEDTLS_CIPHER_MODE_CFB)
  820. /*
  821. * AES-CFB128 buffer encryption/decryption
  822. */
  823. int mbedtls_aes_crypt_cfb128( mbedtls_aes_context *ctx,
  824. int mode,
  825. size_t length,
  826. size_t *iv_off,
  827. unsigned char iv[16],
  828. const unsigned char *input,
  829. unsigned char *output )
  830. {
  831. int c;
  832. size_t n = *iv_off;
  833. if( mode == MBEDTLS_AES_DECRYPT )
  834. {
  835. while( length-- )
  836. {
  837. if( n == 0 )
  838. mbedtls_aes_crypt_ecb( ctx, MBEDTLS_AES_ENCRYPT, iv, iv );
  839. c = *input++;
  840. *output++ = (unsigned char)( c ^ iv[n] );
  841. iv[n] = (unsigned char) c;
  842. n = ( n + 1 ) & 0x0F;
  843. }
  844. }
  845. else
  846. {
  847. while( length-- )
  848. {
  849. if( n == 0 )
  850. mbedtls_aes_crypt_ecb( ctx, MBEDTLS_AES_ENCRYPT, iv, iv );
  851. iv[n] = *output++ = (unsigned char)( iv[n] ^ *input++ );
  852. n = ( n + 1 ) & 0x0F;
  853. }
  854. }
  855. *iv_off = n;
  856. return( 0 );
  857. }
  858. /*
  859. * AES-CFB8 buffer encryption/decryption
  860. */
  861. int mbedtls_aes_crypt_cfb8( mbedtls_aes_context *ctx,
  862. int mode,
  863. size_t length,
  864. unsigned char iv[16],
  865. const unsigned char *input,
  866. unsigned char *output )
  867. {
  868. unsigned char c;
  869. unsigned char ov[17];
  870. while( length-- )
  871. {
  872. memcpy( ov, iv, 16 );
  873. mbedtls_aes_crypt_ecb( ctx, MBEDTLS_AES_ENCRYPT, iv, iv );
  874. if( mode == MBEDTLS_AES_DECRYPT )
  875. ov[16] = *input;
  876. c = *output++ = (unsigned char)( iv[0] ^ *input++ );
  877. if( mode == MBEDTLS_AES_ENCRYPT )
  878. ov[16] = c;
  879. memcpy( iv, ov + 1, 16 );
  880. }
  881. return( 0 );
  882. }
  883. #endif /*MBEDTLS_CIPHER_MODE_CFB */
  884. #if defined(MBEDTLS_CIPHER_MODE_CTR)
  885. /*
  886. * AES-CTR buffer encryption/decryption
  887. */
  888. int mbedtls_aes_crypt_ctr( mbedtls_aes_context *ctx,
  889. size_t length,
  890. size_t *nc_off,
  891. unsigned char nonce_counter[16],
  892. unsigned char stream_block[16],
  893. const unsigned char *input,
  894. unsigned char *output )
  895. {
  896. int c, i;
  897. size_t n = *nc_off;
  898. while( length-- )
  899. {
  900. if( n == 0 ) {
  901. mbedtls_aes_crypt_ecb( ctx, MBEDTLS_AES_ENCRYPT, nonce_counter, stream_block );
  902. for( i = 16; i > 0; i-- )
  903. if( ++nonce_counter[i - 1] != 0 )
  904. break;
  905. }
  906. c = *input++;
  907. *output++ = (unsigned char)( c ^ stream_block[n] );
  908. n = ( n + 1 ) & 0x0F;
  909. }
  910. *nc_off = n;
  911. return( 0 );
  912. }
  913. #endif /* MBEDTLS_CIPHER_MODE_CTR */
  914. #endif /* !MBEDTLS_AES_ALT */
  915. #if defined(MBEDTLS_SELF_TEST)
  916. /*
  917. * AES test vectors from:
  918. *
  919. * http://csrc.nist.gov/archive/aes/rijndael/rijndael-vals.zip
  920. */
  921. static const unsigned char aes_test_ecb_dec[3][16] =
  922. {
  923. { 0x44, 0x41, 0x6A, 0xC2, 0xD1, 0xF5, 0x3C, 0x58,
  924. 0x33, 0x03, 0x91, 0x7E, 0x6B, 0xE9, 0xEB, 0xE0 },
  925. { 0x48, 0xE3, 0x1E, 0x9E, 0x25, 0x67, 0x18, 0xF2,
  926. 0x92, 0x29, 0x31, 0x9C, 0x19, 0xF1, 0x5B, 0xA4 },
  927. { 0x05, 0x8C, 0xCF, 0xFD, 0xBB, 0xCB, 0x38, 0x2D,
  928. 0x1F, 0x6F, 0x56, 0x58, 0x5D, 0x8A, 0x4A, 0xDE }
  929. };
  930. static const unsigned char aes_test_ecb_enc[3][16] =
  931. {
  932. { 0xC3, 0x4C, 0x05, 0x2C, 0xC0, 0xDA, 0x8D, 0x73,
  933. 0x45, 0x1A, 0xFE, 0x5F, 0x03, 0xBE, 0x29, 0x7F },
  934. { 0xF3, 0xF6, 0x75, 0x2A, 0xE8, 0xD7, 0x83, 0x11,
  935. 0x38, 0xF0, 0x41, 0x56, 0x06, 0x31, 0xB1, 0x14 },
  936. { 0x8B, 0x79, 0xEE, 0xCC, 0x93, 0xA0, 0xEE, 0x5D,
  937. 0xFF, 0x30, 0xB4, 0xEA, 0x21, 0x63, 0x6D, 0xA4 }
  938. };
  939. #if defined(MBEDTLS_CIPHER_MODE_CBC)
  940. static const unsigned char aes_test_cbc_dec[3][16] =
  941. {
  942. { 0xFA, 0xCA, 0x37, 0xE0, 0xB0, 0xC8, 0x53, 0x73,
  943. 0xDF, 0x70, 0x6E, 0x73, 0xF7, 0xC9, 0xAF, 0x86 },
  944. { 0x5D, 0xF6, 0x78, 0xDD, 0x17, 0xBA, 0x4E, 0x75,
  945. 0xB6, 0x17, 0x68, 0xC6, 0xAD, 0xEF, 0x7C, 0x7B },
  946. { 0x48, 0x04, 0xE1, 0x81, 0x8F, 0xE6, 0x29, 0x75,
  947. 0x19, 0xA3, 0xE8, 0x8C, 0x57, 0x31, 0x04, 0x13 }
  948. };
  949. static const unsigned char aes_test_cbc_enc[3][16] =
  950. {
  951. { 0x8A, 0x05, 0xFC, 0x5E, 0x09, 0x5A, 0xF4, 0x84,
  952. 0x8A, 0x08, 0xD3, 0x28, 0xD3, 0x68, 0x8E, 0x3D },
  953. { 0x7B, 0xD9, 0x66, 0xD5, 0x3A, 0xD8, 0xC1, 0xBB,
  954. 0x85, 0xD2, 0xAD, 0xFA, 0xE8, 0x7B, 0xB1, 0x04 },
  955. { 0xFE, 0x3C, 0x53, 0x65, 0x3E, 0x2F, 0x45, 0xB5,
  956. 0x6F, 0xCD, 0x88, 0xB2, 0xCC, 0x89, 0x8F, 0xF0 }
  957. };
  958. #endif /* MBEDTLS_CIPHER_MODE_CBC */
  959. #if defined(MBEDTLS_CIPHER_MODE_CFB)
  960. /*
  961. * AES-CFB128 test vectors from:
  962. *
  963. * http://csrc.nist.gov/publications/nistpubs/800-38a/sp800-38a.pdf
  964. */
  965. static const unsigned char aes_test_cfb128_key[3][32] =
  966. {
  967. { 0x2B, 0x7E, 0x15, 0x16, 0x28, 0xAE, 0xD2, 0xA6,
  968. 0xAB, 0xF7, 0x15, 0x88, 0x09, 0xCF, 0x4F, 0x3C },
  969. { 0x8E, 0x73, 0xB0, 0xF7, 0xDA, 0x0E, 0x64, 0x52,
  970. 0xC8, 0x10, 0xF3, 0x2B, 0x80, 0x90, 0x79, 0xE5,
  971. 0x62, 0xF8, 0xEA, 0xD2, 0x52, 0x2C, 0x6B, 0x7B },
  972. { 0x60, 0x3D, 0xEB, 0x10, 0x15, 0xCA, 0x71, 0xBE,
  973. 0x2B, 0x73, 0xAE, 0xF0, 0x85, 0x7D, 0x77, 0x81,
  974. 0x1F, 0x35, 0x2C, 0x07, 0x3B, 0x61, 0x08, 0xD7,
  975. 0x2D, 0x98, 0x10, 0xA3, 0x09, 0x14, 0xDF, 0xF4 }
  976. };
  977. static const unsigned char aes_test_cfb128_iv[16] =
  978. {
  979. 0x00, 0x01, 0x02, 0x03, 0x04, 0x05, 0x06, 0x07,
  980. 0x08, 0x09, 0x0A, 0x0B, 0x0C, 0x0D, 0x0E, 0x0F
  981. };
  982. static const unsigned char aes_test_cfb128_pt[64] =
  983. {
  984. 0x6B, 0xC1, 0xBE, 0xE2, 0x2E, 0x40, 0x9F, 0x96,
  985. 0xE9, 0x3D, 0x7E, 0x11, 0x73, 0x93, 0x17, 0x2A,
  986. 0xAE, 0x2D, 0x8A, 0x57, 0x1E, 0x03, 0xAC, 0x9C,
  987. 0x9E, 0xB7, 0x6F, 0xAC, 0x45, 0xAF, 0x8E, 0x51,
  988. 0x30, 0xC8, 0x1C, 0x46, 0xA3, 0x5C, 0xE4, 0x11,
  989. 0xE5, 0xFB, 0xC1, 0x19, 0x1A, 0x0A, 0x52, 0xEF,
  990. 0xF6, 0x9F, 0x24, 0x45, 0xDF, 0x4F, 0x9B, 0x17,
  991. 0xAD, 0x2B, 0x41, 0x7B, 0xE6, 0x6C, 0x37, 0x10
  992. };
  993. static const unsigned char aes_test_cfb128_ct[3][64] =
  994. {
  995. { 0x3B, 0x3F, 0xD9, 0x2E, 0xB7, 0x2D, 0xAD, 0x20,
  996. 0x33, 0x34, 0x49, 0xF8, 0xE8, 0x3C, 0xFB, 0x4A,
  997. 0xC8, 0xA6, 0x45, 0x37, 0xA0, 0xB3, 0xA9, 0x3F,
  998. 0xCD, 0xE3, 0xCD, 0xAD, 0x9F, 0x1C, 0xE5, 0x8B,
  999. 0x26, 0x75, 0x1F, 0x67, 0xA3, 0xCB, 0xB1, 0x40,
  1000. 0xB1, 0x80, 0x8C, 0xF1, 0x87, 0xA4, 0xF4, 0xDF,
  1001. 0xC0, 0x4B, 0x05, 0x35, 0x7C, 0x5D, 0x1C, 0x0E,
  1002. 0xEA, 0xC4, 0xC6, 0x6F, 0x9F, 0xF7, 0xF2, 0xE6 },
  1003. { 0xCD, 0xC8, 0x0D, 0x6F, 0xDD, 0xF1, 0x8C, 0xAB,
  1004. 0x34, 0xC2, 0x59, 0x09, 0xC9, 0x9A, 0x41, 0x74,
  1005. 0x67, 0xCE, 0x7F, 0x7F, 0x81, 0x17, 0x36, 0x21,
  1006. 0x96, 0x1A, 0x2B, 0x70, 0x17, 0x1D, 0x3D, 0x7A,
  1007. 0x2E, 0x1E, 0x8A, 0x1D, 0xD5, 0x9B, 0x88, 0xB1,
  1008. 0xC8, 0xE6, 0x0F, 0xED, 0x1E, 0xFA, 0xC4, 0xC9,
  1009. 0xC0, 0x5F, 0x9F, 0x9C, 0xA9, 0x83, 0x4F, 0xA0,
  1010. 0x42, 0xAE, 0x8F, 0xBA, 0x58, 0x4B, 0x09, 0xFF },
  1011. { 0xDC, 0x7E, 0x84, 0xBF, 0xDA, 0x79, 0x16, 0x4B,
  1012. 0x7E, 0xCD, 0x84, 0x86, 0x98, 0x5D, 0x38, 0x60,
  1013. 0x39, 0xFF, 0xED, 0x14, 0x3B, 0x28, 0xB1, 0xC8,
  1014. 0x32, 0x11, 0x3C, 0x63, 0x31, 0xE5, 0x40, 0x7B,
  1015. 0xDF, 0x10, 0x13, 0x24, 0x15, 0xE5, 0x4B, 0x92,
  1016. 0xA1, 0x3E, 0xD0, 0xA8, 0x26, 0x7A, 0xE2, 0xF9,
  1017. 0x75, 0xA3, 0x85, 0x74, 0x1A, 0xB9, 0xCE, 0xF8,
  1018. 0x20, 0x31, 0x62, 0x3D, 0x55, 0xB1, 0xE4, 0x71 }
  1019. };
  1020. #endif /* MBEDTLS_CIPHER_MODE_CFB */
  1021. #if defined(MBEDTLS_CIPHER_MODE_CTR)
  1022. /*
  1023. * AES-CTR test vectors from:
  1024. *
  1025. * http://www.faqs.org/rfcs/rfc3686.html
  1026. */
  1027. static const unsigned char aes_test_ctr_key[3][16] =
  1028. {
  1029. { 0xAE, 0x68, 0x52, 0xF8, 0x12, 0x10, 0x67, 0xCC,
  1030. 0x4B, 0xF7, 0xA5, 0x76, 0x55, 0x77, 0xF3, 0x9E },
  1031. { 0x7E, 0x24, 0x06, 0x78, 0x17, 0xFA, 0xE0, 0xD7,
  1032. 0x43, 0xD6, 0xCE, 0x1F, 0x32, 0x53, 0x91, 0x63 },
  1033. { 0x76, 0x91, 0xBE, 0x03, 0x5E, 0x50, 0x20, 0xA8,
  1034. 0xAC, 0x6E, 0x61, 0x85, 0x29, 0xF9, 0xA0, 0xDC }
  1035. };
  1036. static const unsigned char aes_test_ctr_nonce_counter[3][16] =
  1037. {
  1038. { 0x00, 0x00, 0x00, 0x30, 0x00, 0x00, 0x00, 0x00,
  1039. 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x01 },
  1040. { 0x00, 0x6C, 0xB6, 0xDB, 0xC0, 0x54, 0x3B, 0x59,
  1041. 0xDA, 0x48, 0xD9, 0x0B, 0x00, 0x00, 0x00, 0x01 },
  1042. { 0x00, 0xE0, 0x01, 0x7B, 0x27, 0x77, 0x7F, 0x3F,
  1043. 0x4A, 0x17, 0x86, 0xF0, 0x00, 0x00, 0x00, 0x01 }
  1044. };
  1045. static const unsigned char aes_test_ctr_pt[3][48] =
  1046. {
  1047. { 0x53, 0x69, 0x6E, 0x67, 0x6C, 0x65, 0x20, 0x62,
  1048. 0x6C, 0x6F, 0x63, 0x6B, 0x20, 0x6D, 0x73, 0x67 },
  1049. { 0x00, 0x01, 0x02, 0x03, 0x04, 0x05, 0x06, 0x07,
  1050. 0x08, 0x09, 0x0A, 0x0B, 0x0C, 0x0D, 0x0E, 0x0F,
  1051. 0x10, 0x11, 0x12, 0x13, 0x14, 0x15, 0x16, 0x17,
  1052. 0x18, 0x19, 0x1A, 0x1B, 0x1C, 0x1D, 0x1E, 0x1F },
  1053. { 0x00, 0x01, 0x02, 0x03, 0x04, 0x05, 0x06, 0x07,
  1054. 0x08, 0x09, 0x0A, 0x0B, 0x0C, 0x0D, 0x0E, 0x0F,
  1055. 0x10, 0x11, 0x12, 0x13, 0x14, 0x15, 0x16, 0x17,
  1056. 0x18, 0x19, 0x1A, 0x1B, 0x1C, 0x1D, 0x1E, 0x1F,
  1057. 0x20, 0x21, 0x22, 0x23 }
  1058. };
  1059. static const unsigned char aes_test_ctr_ct[3][48] =
  1060. {
  1061. { 0xE4, 0x09, 0x5D, 0x4F, 0xB7, 0xA7, 0xB3, 0x79,
  1062. 0x2D, 0x61, 0x75, 0xA3, 0x26, 0x13, 0x11, 0xB8 },
  1063. { 0x51, 0x04, 0xA1, 0x06, 0x16, 0x8A, 0x72, 0xD9,
  1064. 0x79, 0x0D, 0x41, 0xEE, 0x8E, 0xDA, 0xD3, 0x88,
  1065. 0xEB, 0x2E, 0x1E, 0xFC, 0x46, 0xDA, 0x57, 0xC8,
  1066. 0xFC, 0xE6, 0x30, 0xDF, 0x91, 0x41, 0xBE, 0x28 },
  1067. { 0xC1, 0xCF, 0x48, 0xA8, 0x9F, 0x2F, 0xFD, 0xD9,
  1068. 0xCF, 0x46, 0x52, 0xE9, 0xEF, 0xDB, 0x72, 0xD7,
  1069. 0x45, 0x40, 0xA4, 0x2B, 0xDE, 0x6D, 0x78, 0x36,
  1070. 0xD5, 0x9A, 0x5C, 0xEA, 0xAE, 0xF3, 0x10, 0x53,
  1071. 0x25, 0xB2, 0x07, 0x2F }
  1072. };
  1073. static const int aes_test_ctr_len[3] =
  1074. { 16, 32, 36 };
  1075. #endif /* MBEDTLS_CIPHER_MODE_CTR */
  1076. /*
  1077. * Checkup routine
  1078. */
  1079. int mbedtls_aes_self_test( int verbose )
  1080. {
  1081. int ret = 0, i, j, u, mode;
  1082. unsigned int keybits;
  1083. unsigned char key[32];
  1084. unsigned char buf[64];
  1085. const unsigned char *aes_tests;
  1086. #if defined(MBEDTLS_CIPHER_MODE_CBC) || defined(MBEDTLS_CIPHER_MODE_CFB)
  1087. unsigned char iv[16];
  1088. #endif
  1089. #if defined(MBEDTLS_CIPHER_MODE_CBC)
  1090. unsigned char prv[16];
  1091. #endif
  1092. #if defined(MBEDTLS_CIPHER_MODE_CTR) || defined(MBEDTLS_CIPHER_MODE_CFB)
  1093. size_t offset;
  1094. #endif
  1095. #if defined(MBEDTLS_CIPHER_MODE_CTR)
  1096. int len;
  1097. unsigned char nonce_counter[16];
  1098. unsigned char stream_block[16];
  1099. #endif
  1100. mbedtls_aes_context ctx;
  1101. memset( key, 0, 32 );
  1102. mbedtls_aes_init( &ctx );
  1103. /*
  1104. * ECB mode
  1105. */
  1106. for( i = 0; i < 6; i++ )
  1107. {
  1108. u = i >> 1;
  1109. keybits = 128 + u * 64;
  1110. mode = i & 1;
  1111. if( verbose != 0 )
  1112. mbedtls_printf( " AES-ECB-%3d (%s): ", keybits,
  1113. ( mode == MBEDTLS_AES_DECRYPT ) ? "dec" : "enc" );
  1114. memset( buf, 0, 16 );
  1115. if( mode == MBEDTLS_AES_DECRYPT )
  1116. {
  1117. ret = mbedtls_aes_setkey_dec( &ctx, key, keybits );
  1118. aes_tests = aes_test_ecb_dec[u];
  1119. }
  1120. else
  1121. {
  1122. ret = mbedtls_aes_setkey_enc( &ctx, key, keybits );
  1123. aes_tests = aes_test_ecb_enc[u];
  1124. }
  1125. /*
  1126. * AES-192 is an optional feature that may be unavailable when
  1127. * there is an alternative underlying implementation i.e. when
  1128. * MBEDTLS_AES_ALT is defined.
  1129. */
  1130. if( ret == MBEDTLS_ERR_AES_FEATURE_UNAVAILABLE && keybits == 192 )
  1131. {
  1132. mbedtls_printf( "skipped\n" );
  1133. continue;
  1134. }
  1135. else if( ret != 0 )
  1136. {
  1137. goto exit;
  1138. }
  1139. for( j = 0; j < 10000; j++ )
  1140. {
  1141. ret = mbedtls_aes_crypt_ecb( &ctx, mode, buf, buf );
  1142. if( ret != 0 )
  1143. goto exit;
  1144. }
  1145. if( memcmp( buf, aes_tests, 16 ) != 0 )
  1146. {
  1147. ret = 1;
  1148. goto exit;
  1149. }
  1150. if( verbose != 0 )
  1151. mbedtls_printf( "passed\n" );
  1152. }
  1153. if( verbose != 0 )
  1154. mbedtls_printf( "\n" );
  1155. #if defined(MBEDTLS_CIPHER_MODE_CBC)
  1156. /*
  1157. * CBC mode
  1158. */
  1159. for( i = 0; i < 6; i++ )
  1160. {
  1161. u = i >> 1;
  1162. keybits = 128 + u * 64;
  1163. mode = i & 1;
  1164. if( verbose != 0 )
  1165. mbedtls_printf( " AES-CBC-%3d (%s): ", keybits,
  1166. ( mode == MBEDTLS_AES_DECRYPT ) ? "dec" : "enc" );
  1167. memset( iv , 0, 16 );
  1168. memset( prv, 0, 16 );
  1169. memset( buf, 0, 16 );
  1170. if( mode == MBEDTLS_AES_DECRYPT )
  1171. {
  1172. ret = mbedtls_aes_setkey_dec( &ctx, key, keybits );
  1173. aes_tests = aes_test_cbc_dec[u];
  1174. }
  1175. else
  1176. {
  1177. ret = mbedtls_aes_setkey_enc( &ctx, key, keybits );
  1178. aes_tests = aes_test_cbc_enc[u];
  1179. }
  1180. /*
  1181. * AES-192 is an optional feature that may be unavailable when
  1182. * there is an alternative underlying implementation i.e. when
  1183. * MBEDTLS_AES_ALT is defined.
  1184. */
  1185. if( ret == MBEDTLS_ERR_AES_FEATURE_UNAVAILABLE && keybits == 192 )
  1186. {
  1187. mbedtls_printf( "skipped\n" );
  1188. continue;
  1189. }
  1190. else if( ret != 0 )
  1191. {
  1192. goto exit;
  1193. }
  1194. for( j = 0; j < 10000; j++ )
  1195. {
  1196. if( mode == MBEDTLS_AES_ENCRYPT )
  1197. {
  1198. unsigned char tmp[16];
  1199. memcpy( tmp, prv, 16 );
  1200. memcpy( prv, buf, 16 );
  1201. memcpy( buf, tmp, 16 );
  1202. }
  1203. ret = mbedtls_aes_crypt_cbc( &ctx, mode, 16, iv, buf, buf );
  1204. if( ret != 0 )
  1205. goto exit;
  1206. }
  1207. if( memcmp( buf, aes_tests, 16 ) != 0 )
  1208. {
  1209. ret = 1;
  1210. goto exit;
  1211. }
  1212. if( verbose != 0 )
  1213. mbedtls_printf( "passed\n" );
  1214. }
  1215. if( verbose != 0 )
  1216. mbedtls_printf( "\n" );
  1217. #endif /* MBEDTLS_CIPHER_MODE_CBC */
  1218. #if defined(MBEDTLS_CIPHER_MODE_CFB)
  1219. /*
  1220. * CFB128 mode
  1221. */
  1222. for( i = 0; i < 6; i++ )
  1223. {
  1224. u = i >> 1;
  1225. keybits = 128 + u * 64;
  1226. mode = i & 1;
  1227. if( verbose != 0 )
  1228. mbedtls_printf( " AES-CFB128-%3d (%s): ", keybits,
  1229. ( mode == MBEDTLS_AES_DECRYPT ) ? "dec" : "enc" );
  1230. memcpy( iv, aes_test_cfb128_iv, 16 );
  1231. memcpy( key, aes_test_cfb128_key[u], keybits / 8 );
  1232. offset = 0;
  1233. ret = mbedtls_aes_setkey_enc( &ctx, key, keybits );
  1234. /*
  1235. * AES-192 is an optional feature that may be unavailable when
  1236. * there is an alternative underlying implementation i.e. when
  1237. * MBEDTLS_AES_ALT is defined.
  1238. */
  1239. if( ret == MBEDTLS_ERR_AES_FEATURE_UNAVAILABLE && keybits == 192 )
  1240. {
  1241. mbedtls_printf( "skipped\n" );
  1242. continue;
  1243. }
  1244. else if( ret != 0 )
  1245. {
  1246. goto exit;
  1247. }
  1248. if( mode == MBEDTLS_AES_DECRYPT )
  1249. {
  1250. memcpy( buf, aes_test_cfb128_ct[u], 64 );
  1251. aes_tests = aes_test_cfb128_pt;
  1252. }
  1253. else
  1254. {
  1255. memcpy( buf, aes_test_cfb128_pt, 64 );
  1256. aes_tests = aes_test_cfb128_ct[u];
  1257. }
  1258. ret = mbedtls_aes_crypt_cfb128( &ctx, mode, 64, &offset, iv, buf, buf );
  1259. if( ret != 0 )
  1260. goto exit;
  1261. if( memcmp( buf, aes_tests, 64 ) != 0 )
  1262. {
  1263. ret = 1;
  1264. goto exit;
  1265. }
  1266. if( verbose != 0 )
  1267. mbedtls_printf( "passed\n" );
  1268. }
  1269. if( verbose != 0 )
  1270. mbedtls_printf( "\n" );
  1271. #endif /* MBEDTLS_CIPHER_MODE_CFB */
  1272. #if defined(MBEDTLS_CIPHER_MODE_CTR)
  1273. /*
  1274. * CTR mode
  1275. */
  1276. for( i = 0; i < 6; i++ )
  1277. {
  1278. u = i >> 1;
  1279. mode = i & 1;
  1280. if( verbose != 0 )
  1281. mbedtls_printf( " AES-CTR-128 (%s): ",
  1282. ( mode == MBEDTLS_AES_DECRYPT ) ? "dec" : "enc" );
  1283. memcpy( nonce_counter, aes_test_ctr_nonce_counter[u], 16 );
  1284. memcpy( key, aes_test_ctr_key[u], 16 );
  1285. offset = 0;
  1286. if( ( ret = mbedtls_aes_setkey_enc( &ctx, key, 128 ) ) != 0 )
  1287. goto exit;
  1288. len = aes_test_ctr_len[u];
  1289. if( mode == MBEDTLS_AES_DECRYPT )
  1290. {
  1291. memcpy( buf, aes_test_ctr_ct[u], len );
  1292. aes_tests = aes_test_ctr_pt[u];
  1293. }
  1294. else
  1295. {
  1296. memcpy( buf, aes_test_ctr_pt[u], len );
  1297. aes_tests = aes_test_ctr_ct[u];
  1298. }
  1299. ret = mbedtls_aes_crypt_ctr( &ctx, len, &offset, nonce_counter,
  1300. stream_block, buf, buf );
  1301. if( ret != 0 )
  1302. goto exit;
  1303. if( memcmp( buf, aes_tests, len ) != 0 )
  1304. {
  1305. ret = 1;
  1306. goto exit;
  1307. }
  1308. if( verbose != 0 )
  1309. mbedtls_printf( "passed\n" );
  1310. }
  1311. if( verbose != 0 )
  1312. mbedtls_printf( "\n" );
  1313. #endif /* MBEDTLS_CIPHER_MODE_CTR */
  1314. ret = 0;
  1315. exit:
  1316. if( ret != 0 && verbose != 0 )
  1317. mbedtls_printf( "failed\n" );
  1318. mbedtls_aes_free( &ctx );
  1319. return( ret );
  1320. }
  1321. #endif /* MBEDTLS_SELF_TEST */
  1322. #endif /* MBEDTLS_AES_C */