aes.c 72 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667166816691670167116721673167416751676167716781679168016811682168316841685168616871688168916901691169216931694169516961697169816991700170117021703170417051706170717081709171017111712171317141715171617171718171917201721172217231724172517261727172817291730173117321733173417351736173717381739174017411742174317441745174617471748174917501751175217531754175517561757175817591760176117621763176417651766176717681769177017711772177317741775177617771778177917801781178217831784178517861787178817891790179117921793179417951796179717981799180018011802180318041805180618071808180918101811181218131814181518161817181818191820182118221823182418251826182718281829183018311832183318341835183618371838183918401841184218431844184518461847184818491850185118521853185418551856185718581859186018611862186318641865186618671868186918701871187218731874187518761877187818791880188118821883188418851886188718881889189018911892189318941895189618971898189919001901190219031904190519061907190819091910191119121913191419151916191719181919192019211922192319241925192619271928192919301931193219331934193519361937193819391940194119421943194419451946194719481949195019511952195319541955195619571958195919601961196219631964196519661967196819691970197119721973197419751976197719781979198019811982198319841985198619871988198919901991199219931994199519961997199819992000200120022003200420052006200720082009201020112012201320142015201620172018201920202021202220232024202520262027202820292030203120322033203420352036203720382039204020412042204320442045204620472048204920502051205220532054205520562057205820592060206120622063206420652066206720682069207020712072207320742075207620772078207920802081208220832084208520862087208820892090209120922093209420952096209720982099210021012102210321042105210621072108210921102111211221132114211521162117211821192120212121222123212421252126212721282129213021312132213321342135213621372138213921402141214221432144214521462147214821492150215121522153215421552156215721582159216021612162216321642165216621672168216921702171217221732174217521762177217821792180218121822183218421852186218721882189219021912192219321942195219621972198219922002201220222032204220522062207220822092210221122122213
  1. /*
  2. * FIPS-197 compliant AES implementation
  3. *
  4. * Copyright The Mbed TLS Contributors
  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. /*
  20. * The AES block cipher was designed by Vincent Rijmen and Joan Daemen.
  21. *
  22. * http://csrc.nist.gov/encryption/aes/rijndael/Rijndael.pdf
  23. * http://csrc.nist.gov/publications/fips/fips197/fips-197.pdf
  24. */
  25. #include "common.h"
  26. #if defined(MBEDTLS_AES_C)
  27. #include <string.h>
  28. #include "mbedtls/aes.h"
  29. #include "mbedtls/platform.h"
  30. #include "mbedtls/platform_util.h"
  31. #include "mbedtls/error.h"
  32. #if defined(MBEDTLS_PADLOCK_C)
  33. #include "mbedtls/padlock.h"
  34. #endif
  35. #if defined(MBEDTLS_AESNI_C)
  36. #include "mbedtls/aesni.h"
  37. #endif
  38. #if defined(MBEDTLS_SELF_TEST)
  39. #if defined(MBEDTLS_PLATFORM_C)
  40. #include "mbedtls/platform.h"
  41. #else
  42. #include <stdio.h>
  43. #define mbedtls_printf printf
  44. #endif /* MBEDTLS_PLATFORM_C */
  45. #endif /* MBEDTLS_SELF_TEST */
  46. #if !defined(MBEDTLS_AES_ALT)
  47. /* Parameter validation macros based on platform_util.h */
  48. #define AES_VALIDATE_RET( cond ) \
  49. MBEDTLS_INTERNAL_VALIDATE_RET( cond, MBEDTLS_ERR_AES_BAD_INPUT_DATA )
  50. #define AES_VALIDATE( cond ) \
  51. MBEDTLS_INTERNAL_VALIDATE( cond )
  52. /*
  53. * 32-bit integer manipulation macros (little endian)
  54. */
  55. #ifndef GET_UINT32_LE
  56. #define GET_UINT32_LE(n,b,i) \
  57. { \
  58. (n) = ( (uint32_t) (b)[(i) ] ) \
  59. | ( (uint32_t) (b)[(i) + 1] << 8 ) \
  60. | ( (uint32_t) (b)[(i) + 2] << 16 ) \
  61. | ( (uint32_t) (b)[(i) + 3] << 24 ); \
  62. }
  63. #endif
  64. #ifndef PUT_UINT32_LE
  65. #define PUT_UINT32_LE(n,b,i) \
  66. { \
  67. (b)[(i) ] = (unsigned char) ( ( (n) ) & 0xFF ); \
  68. (b)[(i) + 1] = (unsigned char) ( ( (n) >> 8 ) & 0xFF ); \
  69. (b)[(i) + 2] = (unsigned char) ( ( (n) >> 16 ) & 0xFF ); \
  70. (b)[(i) + 3] = (unsigned char) ( ( (n) >> 24 ) & 0xFF ); \
  71. }
  72. #endif
  73. #if defined(MBEDTLS_PADLOCK_C) && \
  74. ( defined(MBEDTLS_HAVE_X86) || defined(MBEDTLS_PADLOCK_ALIGN16) )
  75. static int aes_padlock_ace = -1;
  76. #endif
  77. #if defined(MBEDTLS_AES_ROM_TABLES)
  78. /*
  79. * Forward S-box
  80. */
  81. static const unsigned char FSb[256] =
  82. {
  83. 0x63, 0x7C, 0x77, 0x7B, 0xF2, 0x6B, 0x6F, 0xC5,
  84. 0x30, 0x01, 0x67, 0x2B, 0xFE, 0xD7, 0xAB, 0x76,
  85. 0xCA, 0x82, 0xC9, 0x7D, 0xFA, 0x59, 0x47, 0xF0,
  86. 0xAD, 0xD4, 0xA2, 0xAF, 0x9C, 0xA4, 0x72, 0xC0,
  87. 0xB7, 0xFD, 0x93, 0x26, 0x36, 0x3F, 0xF7, 0xCC,
  88. 0x34, 0xA5, 0xE5, 0xF1, 0x71, 0xD8, 0x31, 0x15,
  89. 0x04, 0xC7, 0x23, 0xC3, 0x18, 0x96, 0x05, 0x9A,
  90. 0x07, 0x12, 0x80, 0xE2, 0xEB, 0x27, 0xB2, 0x75,
  91. 0x09, 0x83, 0x2C, 0x1A, 0x1B, 0x6E, 0x5A, 0xA0,
  92. 0x52, 0x3B, 0xD6, 0xB3, 0x29, 0xE3, 0x2F, 0x84,
  93. 0x53, 0xD1, 0x00, 0xED, 0x20, 0xFC, 0xB1, 0x5B,
  94. 0x6A, 0xCB, 0xBE, 0x39, 0x4A, 0x4C, 0x58, 0xCF,
  95. 0xD0, 0xEF, 0xAA, 0xFB, 0x43, 0x4D, 0x33, 0x85,
  96. 0x45, 0xF9, 0x02, 0x7F, 0x50, 0x3C, 0x9F, 0xA8,
  97. 0x51, 0xA3, 0x40, 0x8F, 0x92, 0x9D, 0x38, 0xF5,
  98. 0xBC, 0xB6, 0xDA, 0x21, 0x10, 0xFF, 0xF3, 0xD2,
  99. 0xCD, 0x0C, 0x13, 0xEC, 0x5F, 0x97, 0x44, 0x17,
  100. 0xC4, 0xA7, 0x7E, 0x3D, 0x64, 0x5D, 0x19, 0x73,
  101. 0x60, 0x81, 0x4F, 0xDC, 0x22, 0x2A, 0x90, 0x88,
  102. 0x46, 0xEE, 0xB8, 0x14, 0xDE, 0x5E, 0x0B, 0xDB,
  103. 0xE0, 0x32, 0x3A, 0x0A, 0x49, 0x06, 0x24, 0x5C,
  104. 0xC2, 0xD3, 0xAC, 0x62, 0x91, 0x95, 0xE4, 0x79,
  105. 0xE7, 0xC8, 0x37, 0x6D, 0x8D, 0xD5, 0x4E, 0xA9,
  106. 0x6C, 0x56, 0xF4, 0xEA, 0x65, 0x7A, 0xAE, 0x08,
  107. 0xBA, 0x78, 0x25, 0x2E, 0x1C, 0xA6, 0xB4, 0xC6,
  108. 0xE8, 0xDD, 0x74, 0x1F, 0x4B, 0xBD, 0x8B, 0x8A,
  109. 0x70, 0x3E, 0xB5, 0x66, 0x48, 0x03, 0xF6, 0x0E,
  110. 0x61, 0x35, 0x57, 0xB9, 0x86, 0xC1, 0x1D, 0x9E,
  111. 0xE1, 0xF8, 0x98, 0x11, 0x69, 0xD9, 0x8E, 0x94,
  112. 0x9B, 0x1E, 0x87, 0xE9, 0xCE, 0x55, 0x28, 0xDF,
  113. 0x8C, 0xA1, 0x89, 0x0D, 0xBF, 0xE6, 0x42, 0x68,
  114. 0x41, 0x99, 0x2D, 0x0F, 0xB0, 0x54, 0xBB, 0x16
  115. };
  116. /*
  117. * Forward tables
  118. */
  119. #define FT \
  120. \
  121. V(A5,63,63,C6), V(84,7C,7C,F8), V(99,77,77,EE), V(8D,7B,7B,F6), \
  122. V(0D,F2,F2,FF), V(BD,6B,6B,D6), V(B1,6F,6F,DE), V(54,C5,C5,91), \
  123. V(50,30,30,60), V(03,01,01,02), V(A9,67,67,CE), V(7D,2B,2B,56), \
  124. V(19,FE,FE,E7), V(62,D7,D7,B5), V(E6,AB,AB,4D), V(9A,76,76,EC), \
  125. V(45,CA,CA,8F), V(9D,82,82,1F), V(40,C9,C9,89), V(87,7D,7D,FA), \
  126. V(15,FA,FA,EF), V(EB,59,59,B2), V(C9,47,47,8E), V(0B,F0,F0,FB), \
  127. V(EC,AD,AD,41), V(67,D4,D4,B3), V(FD,A2,A2,5F), V(EA,AF,AF,45), \
  128. V(BF,9C,9C,23), V(F7,A4,A4,53), V(96,72,72,E4), V(5B,C0,C0,9B), \
  129. V(C2,B7,B7,75), V(1C,FD,FD,E1), V(AE,93,93,3D), V(6A,26,26,4C), \
  130. V(5A,36,36,6C), V(41,3F,3F,7E), V(02,F7,F7,F5), V(4F,CC,CC,83), \
  131. V(5C,34,34,68), V(F4,A5,A5,51), V(34,E5,E5,D1), V(08,F1,F1,F9), \
  132. V(93,71,71,E2), V(73,D8,D8,AB), V(53,31,31,62), V(3F,15,15,2A), \
  133. V(0C,04,04,08), V(52,C7,C7,95), V(65,23,23,46), V(5E,C3,C3,9D), \
  134. V(28,18,18,30), V(A1,96,96,37), V(0F,05,05,0A), V(B5,9A,9A,2F), \
  135. V(09,07,07,0E), V(36,12,12,24), V(9B,80,80,1B), V(3D,E2,E2,DF), \
  136. V(26,EB,EB,CD), V(69,27,27,4E), V(CD,B2,B2,7F), V(9F,75,75,EA), \
  137. V(1B,09,09,12), V(9E,83,83,1D), V(74,2C,2C,58), V(2E,1A,1A,34), \
  138. V(2D,1B,1B,36), V(B2,6E,6E,DC), V(EE,5A,5A,B4), V(FB,A0,A0,5B), \
  139. V(F6,52,52,A4), V(4D,3B,3B,76), V(61,D6,D6,B7), V(CE,B3,B3,7D), \
  140. V(7B,29,29,52), V(3E,E3,E3,DD), V(71,2F,2F,5E), V(97,84,84,13), \
  141. V(F5,53,53,A6), V(68,D1,D1,B9), V(00,00,00,00), V(2C,ED,ED,C1), \
  142. V(60,20,20,40), V(1F,FC,FC,E3), V(C8,B1,B1,79), V(ED,5B,5B,B6), \
  143. V(BE,6A,6A,D4), V(46,CB,CB,8D), V(D9,BE,BE,67), V(4B,39,39,72), \
  144. V(DE,4A,4A,94), V(D4,4C,4C,98), V(E8,58,58,B0), V(4A,CF,CF,85), \
  145. V(6B,D0,D0,BB), V(2A,EF,EF,C5), V(E5,AA,AA,4F), V(16,FB,FB,ED), \
  146. V(C5,43,43,86), V(D7,4D,4D,9A), V(55,33,33,66), V(94,85,85,11), \
  147. V(CF,45,45,8A), V(10,F9,F9,E9), V(06,02,02,04), V(81,7F,7F,FE), \
  148. V(F0,50,50,A0), V(44,3C,3C,78), V(BA,9F,9F,25), V(E3,A8,A8,4B), \
  149. V(F3,51,51,A2), V(FE,A3,A3,5D), V(C0,40,40,80), V(8A,8F,8F,05), \
  150. V(AD,92,92,3F), V(BC,9D,9D,21), V(48,38,38,70), V(04,F5,F5,F1), \
  151. V(DF,BC,BC,63), V(C1,B6,B6,77), V(75,DA,DA,AF), V(63,21,21,42), \
  152. V(30,10,10,20), V(1A,FF,FF,E5), V(0E,F3,F3,FD), V(6D,D2,D2,BF), \
  153. V(4C,CD,CD,81), V(14,0C,0C,18), V(35,13,13,26), V(2F,EC,EC,C3), \
  154. V(E1,5F,5F,BE), V(A2,97,97,35), V(CC,44,44,88), V(39,17,17,2E), \
  155. V(57,C4,C4,93), V(F2,A7,A7,55), V(82,7E,7E,FC), V(47,3D,3D,7A), \
  156. V(AC,64,64,C8), V(E7,5D,5D,BA), V(2B,19,19,32), V(95,73,73,E6), \
  157. V(A0,60,60,C0), V(98,81,81,19), V(D1,4F,4F,9E), V(7F,DC,DC,A3), \
  158. V(66,22,22,44), V(7E,2A,2A,54), V(AB,90,90,3B), V(83,88,88,0B), \
  159. V(CA,46,46,8C), V(29,EE,EE,C7), V(D3,B8,B8,6B), V(3C,14,14,28), \
  160. V(79,DE,DE,A7), V(E2,5E,5E,BC), V(1D,0B,0B,16), V(76,DB,DB,AD), \
  161. V(3B,E0,E0,DB), V(56,32,32,64), V(4E,3A,3A,74), V(1E,0A,0A,14), \
  162. V(DB,49,49,92), V(0A,06,06,0C), V(6C,24,24,48), V(E4,5C,5C,B8), \
  163. V(5D,C2,C2,9F), V(6E,D3,D3,BD), V(EF,AC,AC,43), V(A6,62,62,C4), \
  164. V(A8,91,91,39), V(A4,95,95,31), V(37,E4,E4,D3), V(8B,79,79,F2), \
  165. V(32,E7,E7,D5), V(43,C8,C8,8B), V(59,37,37,6E), V(B7,6D,6D,DA), \
  166. V(8C,8D,8D,01), V(64,D5,D5,B1), V(D2,4E,4E,9C), V(E0,A9,A9,49), \
  167. V(B4,6C,6C,D8), V(FA,56,56,AC), V(07,F4,F4,F3), V(25,EA,EA,CF), \
  168. V(AF,65,65,CA), V(8E,7A,7A,F4), V(E9,AE,AE,47), V(18,08,08,10), \
  169. V(D5,BA,BA,6F), V(88,78,78,F0), V(6F,25,25,4A), V(72,2E,2E,5C), \
  170. V(24,1C,1C,38), V(F1,A6,A6,57), V(C7,B4,B4,73), V(51,C6,C6,97), \
  171. V(23,E8,E8,CB), V(7C,DD,DD,A1), V(9C,74,74,E8), V(21,1F,1F,3E), \
  172. V(DD,4B,4B,96), V(DC,BD,BD,61), V(86,8B,8B,0D), V(85,8A,8A,0F), \
  173. V(90,70,70,E0), V(42,3E,3E,7C), V(C4,B5,B5,71), V(AA,66,66,CC), \
  174. V(D8,48,48,90), V(05,03,03,06), V(01,F6,F6,F7), V(12,0E,0E,1C), \
  175. V(A3,61,61,C2), V(5F,35,35,6A), V(F9,57,57,AE), V(D0,B9,B9,69), \
  176. V(91,86,86,17), V(58,C1,C1,99), V(27,1D,1D,3A), V(B9,9E,9E,27), \
  177. V(38,E1,E1,D9), V(13,F8,F8,EB), V(B3,98,98,2B), V(33,11,11,22), \
  178. V(BB,69,69,D2), V(70,D9,D9,A9), V(89,8E,8E,07), V(A7,94,94,33), \
  179. V(B6,9B,9B,2D), V(22,1E,1E,3C), V(92,87,87,15), V(20,E9,E9,C9), \
  180. V(49,CE,CE,87), V(FF,55,55,AA), V(78,28,28,50), V(7A,DF,DF,A5), \
  181. V(8F,8C,8C,03), V(F8,A1,A1,59), V(80,89,89,09), V(17,0D,0D,1A), \
  182. V(DA,BF,BF,65), V(31,E6,E6,D7), V(C6,42,42,84), V(B8,68,68,D0), \
  183. V(C3,41,41,82), V(B0,99,99,29), V(77,2D,2D,5A), V(11,0F,0F,1E), \
  184. V(CB,B0,B0,7B), V(FC,54,54,A8), V(D6,BB,BB,6D), V(3A,16,16,2C)
  185. #define V(a,b,c,d) 0x##a##b##c##d
  186. static const uint32_t FT0[256] = { FT };
  187. #undef V
  188. #if !defined(MBEDTLS_AES_FEWER_TABLES)
  189. #define V(a,b,c,d) 0x##b##c##d##a
  190. static const uint32_t FT1[256] = { FT };
  191. #undef V
  192. #define V(a,b,c,d) 0x##c##d##a##b
  193. static const uint32_t FT2[256] = { FT };
  194. #undef V
  195. #define V(a,b,c,d) 0x##d##a##b##c
  196. static const uint32_t FT3[256] = { FT };
  197. #undef V
  198. #endif /* !MBEDTLS_AES_FEWER_TABLES */
  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. #if !defined(MBEDTLS_AES_FEWER_TABLES)
  311. #define V(a,b,c,d) 0x##b##c##d##a
  312. static const uint32_t RT1[256] = { RT };
  313. #undef V
  314. #define V(a,b,c,d) 0x##c##d##a##b
  315. static const uint32_t RT2[256] = { RT };
  316. #undef V
  317. #define V(a,b,c,d) 0x##d##a##b##c
  318. static const uint32_t RT3[256] = { RT };
  319. #undef V
  320. #endif /* !MBEDTLS_AES_FEWER_TABLES */
  321. #undef RT
  322. /*
  323. * Round constants
  324. */
  325. static const uint32_t RCON[10] =
  326. {
  327. 0x00000001, 0x00000002, 0x00000004, 0x00000008,
  328. 0x00000010, 0x00000020, 0x00000040, 0x00000080,
  329. 0x0000001B, 0x00000036
  330. };
  331. #else /* MBEDTLS_AES_ROM_TABLES */
  332. /*
  333. * Forward S-box & tables
  334. */
  335. static unsigned char FSb[256];
  336. static uint32_t FT0[256];
  337. #if !defined(MBEDTLS_AES_FEWER_TABLES)
  338. static uint32_t FT1[256];
  339. static uint32_t FT2[256];
  340. static uint32_t FT3[256];
  341. #endif /* !MBEDTLS_AES_FEWER_TABLES */
  342. /*
  343. * Reverse S-box & tables
  344. */
  345. static unsigned char RSb[256];
  346. static uint32_t RT0[256];
  347. #if !defined(MBEDTLS_AES_FEWER_TABLES)
  348. static uint32_t RT1[256];
  349. static uint32_t RT2[256];
  350. static uint32_t RT3[256];
  351. #endif /* !MBEDTLS_AES_FEWER_TABLES */
  352. /*
  353. * Round constants
  354. */
  355. static uint32_t RCON[10];
  356. /*
  357. * Tables generation code
  358. */
  359. #define ROTL8(x) ( ( (x) << 8 ) & 0xFFFFFFFF ) | ( (x) >> 24 )
  360. #define XTIME(x) ( ( (x) << 1 ) ^ ( ( (x) & 0x80 ) ? 0x1B : 0x00 ) )
  361. #define MUL(x,y) ( ( (x) && (y) ) ? pow[(log[(x)]+log[(y)]) % 255] : 0 )
  362. static int aes_init_done = 0;
  363. static void aes_gen_tables( void )
  364. {
  365. int i, x, y, z;
  366. int pow[256];
  367. int log[256];
  368. /*
  369. * compute pow and log tables over GF(2^8)
  370. */
  371. for( i = 0, x = 1; i < 256; i++ )
  372. {
  373. pow[i] = x;
  374. log[x] = i;
  375. x = ( x ^ XTIME( x ) ) & 0xFF;
  376. }
  377. /*
  378. * calculate the round constants
  379. */
  380. for( i = 0, x = 1; i < 10; i++ )
  381. {
  382. RCON[i] = (uint32_t) x;
  383. x = XTIME( x ) & 0xFF;
  384. }
  385. /*
  386. * generate the forward and reverse S-boxes
  387. */
  388. FSb[0x00] = 0x63;
  389. RSb[0x63] = 0x00;
  390. for( i = 1; i < 256; i++ )
  391. {
  392. x = pow[255 - log[i]];
  393. y = x; y = ( ( y << 1 ) | ( y >> 7 ) ) & 0xFF;
  394. x ^= y; y = ( ( y << 1 ) | ( y >> 7 ) ) & 0xFF;
  395. x ^= y; y = ( ( y << 1 ) | ( y >> 7 ) ) & 0xFF;
  396. x ^= y; y = ( ( y << 1 ) | ( y >> 7 ) ) & 0xFF;
  397. x ^= y ^ 0x63;
  398. FSb[i] = (unsigned char) x;
  399. RSb[x] = (unsigned char) i;
  400. }
  401. /*
  402. * generate the forward and reverse tables
  403. */
  404. for( i = 0; i < 256; i++ )
  405. {
  406. x = FSb[i];
  407. y = XTIME( x ) & 0xFF;
  408. z = ( y ^ x ) & 0xFF;
  409. FT0[i] = ( (uint32_t) y ) ^
  410. ( (uint32_t) x << 8 ) ^
  411. ( (uint32_t) x << 16 ) ^
  412. ( (uint32_t) z << 24 );
  413. #if !defined(MBEDTLS_AES_FEWER_TABLES)
  414. FT1[i] = ROTL8( FT0[i] );
  415. FT2[i] = ROTL8( FT1[i] );
  416. FT3[i] = ROTL8( FT2[i] );
  417. #endif /* !MBEDTLS_AES_FEWER_TABLES */
  418. x = RSb[i];
  419. RT0[i] = ( (uint32_t) MUL( 0x0E, x ) ) ^
  420. ( (uint32_t) MUL( 0x09, x ) << 8 ) ^
  421. ( (uint32_t) MUL( 0x0D, x ) << 16 ) ^
  422. ( (uint32_t) MUL( 0x0B, x ) << 24 );
  423. #if !defined(MBEDTLS_AES_FEWER_TABLES)
  424. RT1[i] = ROTL8( RT0[i] );
  425. RT2[i] = ROTL8( RT1[i] );
  426. RT3[i] = ROTL8( RT2[i] );
  427. #endif /* !MBEDTLS_AES_FEWER_TABLES */
  428. }
  429. }
  430. #undef ROTL8
  431. #endif /* MBEDTLS_AES_ROM_TABLES */
  432. #if defined(MBEDTLS_AES_FEWER_TABLES)
  433. #define ROTL8(x) ( (uint32_t)( ( x ) << 8 ) + (uint32_t)( ( x ) >> 24 ) )
  434. #define ROTL16(x) ( (uint32_t)( ( x ) << 16 ) + (uint32_t)( ( x ) >> 16 ) )
  435. #define ROTL24(x) ( (uint32_t)( ( x ) << 24 ) + (uint32_t)( ( x ) >> 8 ) )
  436. #define AES_RT0(idx) RT0[idx]
  437. #define AES_RT1(idx) ROTL8( RT0[idx] )
  438. #define AES_RT2(idx) ROTL16( RT0[idx] )
  439. #define AES_RT3(idx) ROTL24( RT0[idx] )
  440. #define AES_FT0(idx) FT0[idx]
  441. #define AES_FT1(idx) ROTL8( FT0[idx] )
  442. #define AES_FT2(idx) ROTL16( FT0[idx] )
  443. #define AES_FT3(idx) ROTL24( FT0[idx] )
  444. #else /* MBEDTLS_AES_FEWER_TABLES */
  445. #define AES_RT0(idx) RT0[idx]
  446. #define AES_RT1(idx) RT1[idx]
  447. #define AES_RT2(idx) RT2[idx]
  448. #define AES_RT3(idx) RT3[idx]
  449. #define AES_FT0(idx) FT0[idx]
  450. #define AES_FT1(idx) FT1[idx]
  451. #define AES_FT2(idx) FT2[idx]
  452. #define AES_FT3(idx) FT3[idx]
  453. #endif /* MBEDTLS_AES_FEWER_TABLES */
  454. void mbedtls_aes_init( mbedtls_aes_context *ctx )
  455. {
  456. AES_VALIDATE( ctx != NULL );
  457. memset( ctx, 0, sizeof( mbedtls_aes_context ) );
  458. }
  459. void mbedtls_aes_free( mbedtls_aes_context *ctx )
  460. {
  461. if( ctx == NULL )
  462. return;
  463. mbedtls_platform_zeroize( ctx, sizeof( mbedtls_aes_context ) );
  464. }
  465. #if defined(MBEDTLS_CIPHER_MODE_XTS)
  466. void mbedtls_aes_xts_init( mbedtls_aes_xts_context *ctx )
  467. {
  468. AES_VALIDATE( ctx != NULL );
  469. mbedtls_aes_init( &ctx->crypt );
  470. mbedtls_aes_init( &ctx->tweak );
  471. }
  472. void mbedtls_aes_xts_free( mbedtls_aes_xts_context *ctx )
  473. {
  474. if( ctx == NULL )
  475. return;
  476. mbedtls_aes_free( &ctx->crypt );
  477. mbedtls_aes_free( &ctx->tweak );
  478. }
  479. #endif /* MBEDTLS_CIPHER_MODE_XTS */
  480. /*
  481. * AES key schedule (encryption)
  482. */
  483. #if !defined(MBEDTLS_AES_SETKEY_ENC_ALT)
  484. int mbedtls_aes_setkey_enc( mbedtls_aes_context *ctx, const unsigned char *key,
  485. unsigned int keybits )
  486. {
  487. unsigned int i;
  488. uint32_t *RK;
  489. AES_VALIDATE_RET( ctx != NULL );
  490. AES_VALIDATE_RET( key != NULL );
  491. switch( keybits )
  492. {
  493. case 128: ctx->nr = 10; break;
  494. case 192: ctx->nr = 12; break;
  495. case 256: ctx->nr = 14; break;
  496. default : return( MBEDTLS_ERR_AES_INVALID_KEY_LENGTH );
  497. }
  498. #if !defined(MBEDTLS_AES_ROM_TABLES)
  499. if( aes_init_done == 0 )
  500. {
  501. aes_gen_tables();
  502. aes_init_done = 1;
  503. }
  504. #endif
  505. #if defined(MBEDTLS_PADLOCK_C) && defined(MBEDTLS_PADLOCK_ALIGN16)
  506. if( aes_padlock_ace == -1 )
  507. aes_padlock_ace = mbedtls_padlock_has_support( MBEDTLS_PADLOCK_ACE );
  508. if( aes_padlock_ace )
  509. ctx->rk = RK = MBEDTLS_PADLOCK_ALIGN16( ctx->buf );
  510. else
  511. #endif
  512. ctx->rk = RK = ctx->buf;
  513. #if defined(MBEDTLS_AESNI_C) && defined(MBEDTLS_HAVE_X86_64)
  514. if( mbedtls_aesni_has_support( MBEDTLS_AESNI_AES ) )
  515. return( mbedtls_aesni_setkey_enc( (unsigned char *) ctx->rk, key, keybits ) );
  516. #endif
  517. for( i = 0; i < ( keybits >> 5 ); i++ )
  518. {
  519. GET_UINT32_LE( RK[i], key, i << 2 );
  520. }
  521. switch( ctx->nr )
  522. {
  523. case 10:
  524. for( i = 0; i < 10; i++, RK += 4 )
  525. {
  526. RK[4] = RK[0] ^ RCON[i] ^
  527. ( (uint32_t) FSb[ ( RK[3] >> 8 ) & 0xFF ] ) ^
  528. ( (uint32_t) FSb[ ( RK[3] >> 16 ) & 0xFF ] << 8 ) ^
  529. ( (uint32_t) FSb[ ( RK[3] >> 24 ) & 0xFF ] << 16 ) ^
  530. ( (uint32_t) FSb[ ( RK[3] ) & 0xFF ] << 24 );
  531. RK[5] = RK[1] ^ RK[4];
  532. RK[6] = RK[2] ^ RK[5];
  533. RK[7] = RK[3] ^ RK[6];
  534. }
  535. break;
  536. case 12:
  537. for( i = 0; i < 8; i++, RK += 6 )
  538. {
  539. RK[6] = RK[0] ^ RCON[i] ^
  540. ( (uint32_t) FSb[ ( RK[5] >> 8 ) & 0xFF ] ) ^
  541. ( (uint32_t) FSb[ ( RK[5] >> 16 ) & 0xFF ] << 8 ) ^
  542. ( (uint32_t) FSb[ ( RK[5] >> 24 ) & 0xFF ] << 16 ) ^
  543. ( (uint32_t) FSb[ ( RK[5] ) & 0xFF ] << 24 );
  544. RK[7] = RK[1] ^ RK[6];
  545. RK[8] = RK[2] ^ RK[7];
  546. RK[9] = RK[3] ^ RK[8];
  547. RK[10] = RK[4] ^ RK[9];
  548. RK[11] = RK[5] ^ RK[10];
  549. }
  550. break;
  551. case 14:
  552. for( i = 0; i < 7; i++, RK += 8 )
  553. {
  554. RK[8] = RK[0] ^ RCON[i] ^
  555. ( (uint32_t) FSb[ ( RK[7] >> 8 ) & 0xFF ] ) ^
  556. ( (uint32_t) FSb[ ( RK[7] >> 16 ) & 0xFF ] << 8 ) ^
  557. ( (uint32_t) FSb[ ( RK[7] >> 24 ) & 0xFF ] << 16 ) ^
  558. ( (uint32_t) FSb[ ( RK[7] ) & 0xFF ] << 24 );
  559. RK[9] = RK[1] ^ RK[8];
  560. RK[10] = RK[2] ^ RK[9];
  561. RK[11] = RK[3] ^ RK[10];
  562. RK[12] = RK[4] ^
  563. ( (uint32_t) FSb[ ( RK[11] ) & 0xFF ] ) ^
  564. ( (uint32_t) FSb[ ( RK[11] >> 8 ) & 0xFF ] << 8 ) ^
  565. ( (uint32_t) FSb[ ( RK[11] >> 16 ) & 0xFF ] << 16 ) ^
  566. ( (uint32_t) FSb[ ( RK[11] >> 24 ) & 0xFF ] << 24 );
  567. RK[13] = RK[5] ^ RK[12];
  568. RK[14] = RK[6] ^ RK[13];
  569. RK[15] = RK[7] ^ RK[14];
  570. }
  571. break;
  572. }
  573. return( 0 );
  574. }
  575. #endif /* !MBEDTLS_AES_SETKEY_ENC_ALT */
  576. /*
  577. * AES key schedule (decryption)
  578. */
  579. #if !defined(MBEDTLS_AES_SETKEY_DEC_ALT)
  580. int mbedtls_aes_setkey_dec( mbedtls_aes_context *ctx, const unsigned char *key,
  581. unsigned int keybits )
  582. {
  583. int i, j, ret;
  584. mbedtls_aes_context cty;
  585. uint32_t *RK;
  586. uint32_t *SK;
  587. AES_VALIDATE_RET( ctx != NULL );
  588. AES_VALIDATE_RET( key != NULL );
  589. mbedtls_aes_init( &cty );
  590. #if defined(MBEDTLS_PADLOCK_C) && defined(MBEDTLS_PADLOCK_ALIGN16)
  591. if( aes_padlock_ace == -1 )
  592. aes_padlock_ace = mbedtls_padlock_has_support( MBEDTLS_PADLOCK_ACE );
  593. if( aes_padlock_ace )
  594. ctx->rk = RK = MBEDTLS_PADLOCK_ALIGN16( ctx->buf );
  595. else
  596. #endif
  597. ctx->rk = RK = ctx->buf;
  598. /* Also checks keybits */
  599. if( ( ret = mbedtls_aes_setkey_enc( &cty, key, keybits ) ) != 0 )
  600. goto exit;
  601. ctx->nr = cty.nr;
  602. #if defined(MBEDTLS_AESNI_C) && defined(MBEDTLS_HAVE_X86_64)
  603. if( mbedtls_aesni_has_support( MBEDTLS_AESNI_AES ) )
  604. {
  605. mbedtls_aesni_inverse_key( (unsigned char *) ctx->rk,
  606. (const unsigned char *) cty.rk, ctx->nr );
  607. goto exit;
  608. }
  609. #endif
  610. SK = cty.rk + cty.nr * 4;
  611. *RK++ = *SK++;
  612. *RK++ = *SK++;
  613. *RK++ = *SK++;
  614. *RK++ = *SK++;
  615. for( i = ctx->nr - 1, SK -= 8; i > 0; i--, SK -= 8 )
  616. {
  617. for( j = 0; j < 4; j++, SK++ )
  618. {
  619. *RK++ = AES_RT0( FSb[ ( *SK ) & 0xFF ] ) ^
  620. AES_RT1( FSb[ ( *SK >> 8 ) & 0xFF ] ) ^
  621. AES_RT2( FSb[ ( *SK >> 16 ) & 0xFF ] ) ^
  622. AES_RT3( FSb[ ( *SK >> 24 ) & 0xFF ] );
  623. }
  624. }
  625. *RK++ = *SK++;
  626. *RK++ = *SK++;
  627. *RK++ = *SK++;
  628. *RK++ = *SK++;
  629. exit:
  630. mbedtls_aes_free( &cty );
  631. return( ret );
  632. }
  633. #endif /* !MBEDTLS_AES_SETKEY_DEC_ALT */
  634. #if defined(MBEDTLS_CIPHER_MODE_XTS)
  635. static int mbedtls_aes_xts_decode_keys( const unsigned char *key,
  636. unsigned int keybits,
  637. const unsigned char **key1,
  638. unsigned int *key1bits,
  639. const unsigned char **key2,
  640. unsigned int *key2bits )
  641. {
  642. const unsigned int half_keybits = keybits / 2;
  643. const unsigned int half_keybytes = half_keybits / 8;
  644. switch( keybits )
  645. {
  646. case 256: break;
  647. case 512: break;
  648. default : return( MBEDTLS_ERR_AES_INVALID_KEY_LENGTH );
  649. }
  650. *key1bits = half_keybits;
  651. *key2bits = half_keybits;
  652. *key1 = &key[0];
  653. *key2 = &key[half_keybytes];
  654. return 0;
  655. }
  656. int mbedtls_aes_xts_setkey_enc( mbedtls_aes_xts_context *ctx,
  657. const unsigned char *key,
  658. unsigned int keybits)
  659. {
  660. int ret = MBEDTLS_ERR_ERROR_CORRUPTION_DETECTED;
  661. const unsigned char *key1, *key2;
  662. unsigned int key1bits, key2bits;
  663. AES_VALIDATE_RET( ctx != NULL );
  664. AES_VALIDATE_RET( key != NULL );
  665. ret = mbedtls_aes_xts_decode_keys( key, keybits, &key1, &key1bits,
  666. &key2, &key2bits );
  667. if( ret != 0 )
  668. return( ret );
  669. /* Set the tweak key. Always set tweak key for the encryption mode. */
  670. ret = mbedtls_aes_setkey_enc( &ctx->tweak, key2, key2bits );
  671. if( ret != 0 )
  672. return( ret );
  673. /* Set crypt key for encryption. */
  674. return mbedtls_aes_setkey_enc( &ctx->crypt, key1, key1bits );
  675. }
  676. int mbedtls_aes_xts_setkey_dec( mbedtls_aes_xts_context *ctx,
  677. const unsigned char *key,
  678. unsigned int keybits)
  679. {
  680. int ret = MBEDTLS_ERR_ERROR_CORRUPTION_DETECTED;
  681. const unsigned char *key1, *key2;
  682. unsigned int key1bits, key2bits;
  683. AES_VALIDATE_RET( ctx != NULL );
  684. AES_VALIDATE_RET( key != NULL );
  685. ret = mbedtls_aes_xts_decode_keys( key, keybits, &key1, &key1bits,
  686. &key2, &key2bits );
  687. if( ret != 0 )
  688. return( ret );
  689. /* Set the tweak key. Always set tweak key for encryption. */
  690. ret = mbedtls_aes_setkey_enc( &ctx->tweak, key2, key2bits );
  691. if( ret != 0 )
  692. return( ret );
  693. /* Set crypt key for decryption. */
  694. return mbedtls_aes_setkey_dec( &ctx->crypt, key1, key1bits );
  695. }
  696. #endif /* MBEDTLS_CIPHER_MODE_XTS */
  697. #define AES_FROUND(X0,X1,X2,X3,Y0,Y1,Y2,Y3) \
  698. do \
  699. { \
  700. (X0) = *RK++ ^ AES_FT0( ( (Y0) ) & 0xFF ) ^ \
  701. AES_FT1( ( (Y1) >> 8 ) & 0xFF ) ^ \
  702. AES_FT2( ( (Y2) >> 16 ) & 0xFF ) ^ \
  703. AES_FT3( ( (Y3) >> 24 ) & 0xFF ); \
  704. \
  705. (X1) = *RK++ ^ AES_FT0( ( (Y1) ) & 0xFF ) ^ \
  706. AES_FT1( ( (Y2) >> 8 ) & 0xFF ) ^ \
  707. AES_FT2( ( (Y3) >> 16 ) & 0xFF ) ^ \
  708. AES_FT3( ( (Y0) >> 24 ) & 0xFF ); \
  709. \
  710. (X2) = *RK++ ^ AES_FT0( ( (Y2) ) & 0xFF ) ^ \
  711. AES_FT1( ( (Y3) >> 8 ) & 0xFF ) ^ \
  712. AES_FT2( ( (Y0) >> 16 ) & 0xFF ) ^ \
  713. AES_FT3( ( (Y1) >> 24 ) & 0xFF ); \
  714. \
  715. (X3) = *RK++ ^ AES_FT0( ( (Y3) ) & 0xFF ) ^ \
  716. AES_FT1( ( (Y0) >> 8 ) & 0xFF ) ^ \
  717. AES_FT2( ( (Y1) >> 16 ) & 0xFF ) ^ \
  718. AES_FT3( ( (Y2) >> 24 ) & 0xFF ); \
  719. } while( 0 )
  720. #define AES_RROUND(X0,X1,X2,X3,Y0,Y1,Y2,Y3) \
  721. do \
  722. { \
  723. (X0) = *RK++ ^ AES_RT0( ( (Y0) ) & 0xFF ) ^ \
  724. AES_RT1( ( (Y3) >> 8 ) & 0xFF ) ^ \
  725. AES_RT2( ( (Y2) >> 16 ) & 0xFF ) ^ \
  726. AES_RT3( ( (Y1) >> 24 ) & 0xFF ); \
  727. \
  728. (X1) = *RK++ ^ AES_RT0( ( (Y1) ) & 0xFF ) ^ \
  729. AES_RT1( ( (Y0) >> 8 ) & 0xFF ) ^ \
  730. AES_RT2( ( (Y3) >> 16 ) & 0xFF ) ^ \
  731. AES_RT3( ( (Y2) >> 24 ) & 0xFF ); \
  732. \
  733. (X2) = *RK++ ^ AES_RT0( ( (Y2) ) & 0xFF ) ^ \
  734. AES_RT1( ( (Y1) >> 8 ) & 0xFF ) ^ \
  735. AES_RT2( ( (Y0) >> 16 ) & 0xFF ) ^ \
  736. AES_RT3( ( (Y3) >> 24 ) & 0xFF ); \
  737. \
  738. (X3) = *RK++ ^ AES_RT0( ( (Y3) ) & 0xFF ) ^ \
  739. AES_RT1( ( (Y2) >> 8 ) & 0xFF ) ^ \
  740. AES_RT2( ( (Y1) >> 16 ) & 0xFF ) ^ \
  741. AES_RT3( ( (Y0) >> 24 ) & 0xFF ); \
  742. } while( 0 )
  743. /*
  744. * AES-ECB block encryption
  745. */
  746. #if !defined(MBEDTLS_AES_ENCRYPT_ALT)
  747. int mbedtls_internal_aes_encrypt( mbedtls_aes_context *ctx,
  748. const unsigned char input[16],
  749. unsigned char output[16] )
  750. {
  751. int i;
  752. uint32_t *RK = ctx->rk;
  753. struct
  754. {
  755. uint32_t X[4];
  756. uint32_t Y[4];
  757. } t;
  758. GET_UINT32_LE( t.X[0], input, 0 ); t.X[0] ^= *RK++;
  759. GET_UINT32_LE( t.X[1], input, 4 ); t.X[1] ^= *RK++;
  760. GET_UINT32_LE( t.X[2], input, 8 ); t.X[2] ^= *RK++;
  761. GET_UINT32_LE( t.X[3], input, 12 ); t.X[3] ^= *RK++;
  762. for( i = ( ctx->nr >> 1 ) - 1; i > 0; i-- )
  763. {
  764. AES_FROUND( t.Y[0], t.Y[1], t.Y[2], t.Y[3], t.X[0], t.X[1], t.X[2], t.X[3] );
  765. AES_FROUND( t.X[0], t.X[1], t.X[2], t.X[3], t.Y[0], t.Y[1], t.Y[2], t.Y[3] );
  766. }
  767. AES_FROUND( t.Y[0], t.Y[1], t.Y[2], t.Y[3], t.X[0], t.X[1], t.X[2], t.X[3] );
  768. t.X[0] = *RK++ ^ \
  769. ( (uint32_t) FSb[ ( t.Y[0] ) & 0xFF ] ) ^
  770. ( (uint32_t) FSb[ ( t.Y[1] >> 8 ) & 0xFF ] << 8 ) ^
  771. ( (uint32_t) FSb[ ( t.Y[2] >> 16 ) & 0xFF ] << 16 ) ^
  772. ( (uint32_t) FSb[ ( t.Y[3] >> 24 ) & 0xFF ] << 24 );
  773. t.X[1] = *RK++ ^ \
  774. ( (uint32_t) FSb[ ( t.Y[1] ) & 0xFF ] ) ^
  775. ( (uint32_t) FSb[ ( t.Y[2] >> 8 ) & 0xFF ] << 8 ) ^
  776. ( (uint32_t) FSb[ ( t.Y[3] >> 16 ) & 0xFF ] << 16 ) ^
  777. ( (uint32_t) FSb[ ( t.Y[0] >> 24 ) & 0xFF ] << 24 );
  778. t.X[2] = *RK++ ^ \
  779. ( (uint32_t) FSb[ ( t.Y[2] ) & 0xFF ] ) ^
  780. ( (uint32_t) FSb[ ( t.Y[3] >> 8 ) & 0xFF ] << 8 ) ^
  781. ( (uint32_t) FSb[ ( t.Y[0] >> 16 ) & 0xFF ] << 16 ) ^
  782. ( (uint32_t) FSb[ ( t.Y[1] >> 24 ) & 0xFF ] << 24 );
  783. t.X[3] = *RK++ ^ \
  784. ( (uint32_t) FSb[ ( t.Y[3] ) & 0xFF ] ) ^
  785. ( (uint32_t) FSb[ ( t.Y[0] >> 8 ) & 0xFF ] << 8 ) ^
  786. ( (uint32_t) FSb[ ( t.Y[1] >> 16 ) & 0xFF ] << 16 ) ^
  787. ( (uint32_t) FSb[ ( t.Y[2] >> 24 ) & 0xFF ] << 24 );
  788. PUT_UINT32_LE( t.X[0], output, 0 );
  789. PUT_UINT32_LE( t.X[1], output, 4 );
  790. PUT_UINT32_LE( t.X[2], output, 8 );
  791. PUT_UINT32_LE( t.X[3], output, 12 );
  792. mbedtls_platform_zeroize( &t, sizeof( t ) );
  793. return( 0 );
  794. }
  795. #endif /* !MBEDTLS_AES_ENCRYPT_ALT */
  796. #if !defined(MBEDTLS_DEPRECATED_REMOVED)
  797. void mbedtls_aes_encrypt( mbedtls_aes_context *ctx,
  798. const unsigned char input[16],
  799. unsigned char output[16] )
  800. {
  801. mbedtls_internal_aes_encrypt( ctx, input, output );
  802. }
  803. #endif /* !MBEDTLS_DEPRECATED_REMOVED */
  804. /*
  805. * AES-ECB block decryption
  806. */
  807. #if !defined(MBEDTLS_AES_DECRYPT_ALT)
  808. int mbedtls_internal_aes_decrypt( mbedtls_aes_context *ctx,
  809. const unsigned char input[16],
  810. unsigned char output[16] )
  811. {
  812. int i;
  813. uint32_t *RK = ctx->rk;
  814. struct
  815. {
  816. uint32_t X[4];
  817. uint32_t Y[4];
  818. } t;
  819. GET_UINT32_LE( t.X[0], input, 0 ); t.X[0] ^= *RK++;
  820. GET_UINT32_LE( t.X[1], input, 4 ); t.X[1] ^= *RK++;
  821. GET_UINT32_LE( t.X[2], input, 8 ); t.X[2] ^= *RK++;
  822. GET_UINT32_LE( t.X[3], input, 12 ); t.X[3] ^= *RK++;
  823. for( i = ( ctx->nr >> 1 ) - 1; i > 0; i-- )
  824. {
  825. AES_RROUND( t.Y[0], t.Y[1], t.Y[2], t.Y[3], t.X[0], t.X[1], t.X[2], t.X[3] );
  826. AES_RROUND( t.X[0], t.X[1], t.X[2], t.X[3], t.Y[0], t.Y[1], t.Y[2], t.Y[3] );
  827. }
  828. AES_RROUND( t.Y[0], t.Y[1], t.Y[2], t.Y[3], t.X[0], t.X[1], t.X[2], t.X[3] );
  829. t.X[0] = *RK++ ^ \
  830. ( (uint32_t) RSb[ ( t.Y[0] ) & 0xFF ] ) ^
  831. ( (uint32_t) RSb[ ( t.Y[3] >> 8 ) & 0xFF ] << 8 ) ^
  832. ( (uint32_t) RSb[ ( t.Y[2] >> 16 ) & 0xFF ] << 16 ) ^
  833. ( (uint32_t) RSb[ ( t.Y[1] >> 24 ) & 0xFF ] << 24 );
  834. t.X[1] = *RK++ ^ \
  835. ( (uint32_t) RSb[ ( t.Y[1] ) & 0xFF ] ) ^
  836. ( (uint32_t) RSb[ ( t.Y[0] >> 8 ) & 0xFF ] << 8 ) ^
  837. ( (uint32_t) RSb[ ( t.Y[3] >> 16 ) & 0xFF ] << 16 ) ^
  838. ( (uint32_t) RSb[ ( t.Y[2] >> 24 ) & 0xFF ] << 24 );
  839. t.X[2] = *RK++ ^ \
  840. ( (uint32_t) RSb[ ( t.Y[2] ) & 0xFF ] ) ^
  841. ( (uint32_t) RSb[ ( t.Y[1] >> 8 ) & 0xFF ] << 8 ) ^
  842. ( (uint32_t) RSb[ ( t.Y[0] >> 16 ) & 0xFF ] << 16 ) ^
  843. ( (uint32_t) RSb[ ( t.Y[3] >> 24 ) & 0xFF ] << 24 );
  844. t.X[3] = *RK++ ^ \
  845. ( (uint32_t) RSb[ ( t.Y[3] ) & 0xFF ] ) ^
  846. ( (uint32_t) RSb[ ( t.Y[2] >> 8 ) & 0xFF ] << 8 ) ^
  847. ( (uint32_t) RSb[ ( t.Y[1] >> 16 ) & 0xFF ] << 16 ) ^
  848. ( (uint32_t) RSb[ ( t.Y[0] >> 24 ) & 0xFF ] << 24 );
  849. PUT_UINT32_LE( t.X[0], output, 0 );
  850. PUT_UINT32_LE( t.X[1], output, 4 );
  851. PUT_UINT32_LE( t.X[2], output, 8 );
  852. PUT_UINT32_LE( t.X[3], output, 12 );
  853. mbedtls_platform_zeroize( &t, sizeof( t ) );
  854. return( 0 );
  855. }
  856. #endif /* !MBEDTLS_AES_DECRYPT_ALT */
  857. #if !defined(MBEDTLS_DEPRECATED_REMOVED)
  858. void mbedtls_aes_decrypt( mbedtls_aes_context *ctx,
  859. const unsigned char input[16],
  860. unsigned char output[16] )
  861. {
  862. mbedtls_internal_aes_decrypt( ctx, input, output );
  863. }
  864. #endif /* !MBEDTLS_DEPRECATED_REMOVED */
  865. /*
  866. * AES-ECB block encryption/decryption
  867. */
  868. int mbedtls_aes_crypt_ecb( mbedtls_aes_context *ctx,
  869. int mode,
  870. const unsigned char input[16],
  871. unsigned char output[16] )
  872. {
  873. AES_VALIDATE_RET( ctx != NULL );
  874. AES_VALIDATE_RET( input != NULL );
  875. AES_VALIDATE_RET( output != NULL );
  876. AES_VALIDATE_RET( mode == MBEDTLS_AES_ENCRYPT ||
  877. mode == MBEDTLS_AES_DECRYPT );
  878. #if defined(MBEDTLS_AESNI_C) && defined(MBEDTLS_HAVE_X86_64)
  879. if( mbedtls_aesni_has_support( MBEDTLS_AESNI_AES ) )
  880. return( mbedtls_aesni_crypt_ecb( ctx, mode, input, output ) );
  881. #endif
  882. #if defined(MBEDTLS_PADLOCK_C) && defined(MBEDTLS_HAVE_X86)
  883. if( aes_padlock_ace )
  884. {
  885. if( mbedtls_padlock_xcryptecb( ctx, mode, input, output ) == 0 )
  886. return( 0 );
  887. // If padlock data misaligned, we just fall back to
  888. // unaccelerated mode
  889. //
  890. }
  891. #endif
  892. if( mode == MBEDTLS_AES_ENCRYPT )
  893. return( mbedtls_internal_aes_encrypt( ctx, input, output ) );
  894. else
  895. return( mbedtls_internal_aes_decrypt( ctx, input, output ) );
  896. }
  897. #if defined(MBEDTLS_CIPHER_MODE_CBC)
  898. /*
  899. * AES-CBC buffer encryption/decryption
  900. */
  901. int mbedtls_aes_crypt_cbc( mbedtls_aes_context *ctx,
  902. int mode,
  903. size_t length,
  904. unsigned char iv[16],
  905. const unsigned char *input,
  906. unsigned char *output )
  907. {
  908. int i;
  909. unsigned char temp[16];
  910. AES_VALIDATE_RET( ctx != NULL );
  911. AES_VALIDATE_RET( mode == MBEDTLS_AES_ENCRYPT ||
  912. mode == MBEDTLS_AES_DECRYPT );
  913. AES_VALIDATE_RET( iv != NULL );
  914. AES_VALIDATE_RET( input != NULL );
  915. AES_VALIDATE_RET( output != NULL );
  916. if( length % 16 )
  917. return( MBEDTLS_ERR_AES_INVALID_INPUT_LENGTH );
  918. #if defined(MBEDTLS_PADLOCK_C) && defined(MBEDTLS_HAVE_X86)
  919. if( aes_padlock_ace )
  920. {
  921. if( mbedtls_padlock_xcryptcbc( ctx, mode, length, iv, input, output ) == 0 )
  922. return( 0 );
  923. // If padlock data misaligned, we just fall back to
  924. // unaccelerated mode
  925. //
  926. }
  927. #endif
  928. if( mode == MBEDTLS_AES_DECRYPT )
  929. {
  930. while( length > 0 )
  931. {
  932. memcpy( temp, input, 16 );
  933. mbedtls_aes_crypt_ecb( ctx, mode, input, output );
  934. for( i = 0; i < 16; i++ )
  935. output[i] = (unsigned char)( output[i] ^ iv[i] );
  936. memcpy( iv, temp, 16 );
  937. input += 16;
  938. output += 16;
  939. length -= 16;
  940. }
  941. }
  942. else
  943. {
  944. while( length > 0 )
  945. {
  946. for( i = 0; i < 16; i++ )
  947. output[i] = (unsigned char)( input[i] ^ iv[i] );
  948. mbedtls_aes_crypt_ecb( ctx, mode, output, output );
  949. memcpy( iv, output, 16 );
  950. input += 16;
  951. output += 16;
  952. length -= 16;
  953. }
  954. }
  955. return( 0 );
  956. }
  957. #endif /* MBEDTLS_CIPHER_MODE_CBC */
  958. #if defined(MBEDTLS_CIPHER_MODE_XTS)
  959. /* Endianess with 64 bits values */
  960. #ifndef GET_UINT64_LE
  961. #define GET_UINT64_LE(n,b,i) \
  962. { \
  963. (n) = ( (uint64_t) (b)[(i) + 7] << 56 ) \
  964. | ( (uint64_t) (b)[(i) + 6] << 48 ) \
  965. | ( (uint64_t) (b)[(i) + 5] << 40 ) \
  966. | ( (uint64_t) (b)[(i) + 4] << 32 ) \
  967. | ( (uint64_t) (b)[(i) + 3] << 24 ) \
  968. | ( (uint64_t) (b)[(i) + 2] << 16 ) \
  969. | ( (uint64_t) (b)[(i) + 1] << 8 ) \
  970. | ( (uint64_t) (b)[(i) ] ); \
  971. }
  972. #endif
  973. #ifndef PUT_UINT64_LE
  974. #define PUT_UINT64_LE(n,b,i) \
  975. { \
  976. (b)[(i) + 7] = (unsigned char) ( (n) >> 56 ); \
  977. (b)[(i) + 6] = (unsigned char) ( (n) >> 48 ); \
  978. (b)[(i) + 5] = (unsigned char) ( (n) >> 40 ); \
  979. (b)[(i) + 4] = (unsigned char) ( (n) >> 32 ); \
  980. (b)[(i) + 3] = (unsigned char) ( (n) >> 24 ); \
  981. (b)[(i) + 2] = (unsigned char) ( (n) >> 16 ); \
  982. (b)[(i) + 1] = (unsigned char) ( (n) >> 8 ); \
  983. (b)[(i) ] = (unsigned char) ( (n) ); \
  984. }
  985. #endif
  986. typedef unsigned char mbedtls_be128[16];
  987. /*
  988. * GF(2^128) multiplication function
  989. *
  990. * This function multiplies a field element by x in the polynomial field
  991. * representation. It uses 64-bit word operations to gain speed but compensates
  992. * for machine endianess and hence works correctly on both big and little
  993. * endian machines.
  994. */
  995. static void mbedtls_gf128mul_x_ble( unsigned char r[16],
  996. const unsigned char x[16] )
  997. {
  998. uint64_t a, b, ra, rb;
  999. GET_UINT64_LE( a, x, 0 );
  1000. GET_UINT64_LE( b, x, 8 );
  1001. ra = ( a << 1 ) ^ 0x0087 >> ( 8 - ( ( b >> 63 ) << 3 ) );
  1002. rb = ( a >> 63 ) | ( b << 1 );
  1003. PUT_UINT64_LE( ra, r, 0 );
  1004. PUT_UINT64_LE( rb, r, 8 );
  1005. }
  1006. /*
  1007. * AES-XTS buffer encryption/decryption
  1008. */
  1009. int mbedtls_aes_crypt_xts( mbedtls_aes_xts_context *ctx,
  1010. int mode,
  1011. size_t length,
  1012. const unsigned char data_unit[16],
  1013. const unsigned char *input,
  1014. unsigned char *output )
  1015. {
  1016. int ret = MBEDTLS_ERR_ERROR_CORRUPTION_DETECTED;
  1017. size_t blocks = length / 16;
  1018. size_t leftover = length % 16;
  1019. unsigned char tweak[16];
  1020. unsigned char prev_tweak[16];
  1021. unsigned char tmp[16];
  1022. AES_VALIDATE_RET( ctx != NULL );
  1023. AES_VALIDATE_RET( mode == MBEDTLS_AES_ENCRYPT ||
  1024. mode == MBEDTLS_AES_DECRYPT );
  1025. AES_VALIDATE_RET( data_unit != NULL );
  1026. AES_VALIDATE_RET( input != NULL );
  1027. AES_VALIDATE_RET( output != NULL );
  1028. /* Data units must be at least 16 bytes long. */
  1029. if( length < 16 )
  1030. return MBEDTLS_ERR_AES_INVALID_INPUT_LENGTH;
  1031. /* NIST SP 800-38E disallows data units larger than 2**20 blocks. */
  1032. if( length > ( 1 << 20 ) * 16 )
  1033. return MBEDTLS_ERR_AES_INVALID_INPUT_LENGTH;
  1034. /* Compute the tweak. */
  1035. ret = mbedtls_aes_crypt_ecb( &ctx->tweak, MBEDTLS_AES_ENCRYPT,
  1036. data_unit, tweak );
  1037. if( ret != 0 )
  1038. return( ret );
  1039. while( blocks-- )
  1040. {
  1041. size_t i;
  1042. if( leftover && ( mode == MBEDTLS_AES_DECRYPT ) && blocks == 0 )
  1043. {
  1044. /* We are on the last block in a decrypt operation that has
  1045. * leftover bytes, so we need to use the next tweak for this block,
  1046. * and this tweak for the lefover bytes. Save the current tweak for
  1047. * the leftovers and then update the current tweak for use on this,
  1048. * the last full block. */
  1049. memcpy( prev_tweak, tweak, sizeof( tweak ) );
  1050. mbedtls_gf128mul_x_ble( tweak, tweak );
  1051. }
  1052. for( i = 0; i < 16; i++ )
  1053. tmp[i] = input[i] ^ tweak[i];
  1054. ret = mbedtls_aes_crypt_ecb( &ctx->crypt, mode, tmp, tmp );
  1055. if( ret != 0 )
  1056. return( ret );
  1057. for( i = 0; i < 16; i++ )
  1058. output[i] = tmp[i] ^ tweak[i];
  1059. /* Update the tweak for the next block. */
  1060. mbedtls_gf128mul_x_ble( tweak, tweak );
  1061. output += 16;
  1062. input += 16;
  1063. }
  1064. if( leftover )
  1065. {
  1066. /* If we are on the leftover bytes in a decrypt operation, we need to
  1067. * use the previous tweak for these bytes (as saved in prev_tweak). */
  1068. unsigned char *t = mode == MBEDTLS_AES_DECRYPT ? prev_tweak : tweak;
  1069. /* We are now on the final part of the data unit, which doesn't divide
  1070. * evenly by 16. It's time for ciphertext stealing. */
  1071. size_t i;
  1072. unsigned char *prev_output = output - 16;
  1073. /* Copy ciphertext bytes from the previous block to our output for each
  1074. * byte of cyphertext we won't steal. At the same time, copy the
  1075. * remainder of the input for this final round (since the loop bounds
  1076. * are the same). */
  1077. for( i = 0; i < leftover; i++ )
  1078. {
  1079. output[i] = prev_output[i];
  1080. tmp[i] = input[i] ^ t[i];
  1081. }
  1082. /* Copy ciphertext bytes from the previous block for input in this
  1083. * round. */
  1084. for( ; i < 16; i++ )
  1085. tmp[i] = prev_output[i] ^ t[i];
  1086. ret = mbedtls_aes_crypt_ecb( &ctx->crypt, mode, tmp, tmp );
  1087. if( ret != 0 )
  1088. return ret;
  1089. /* Write the result back to the previous block, overriding the previous
  1090. * output we copied. */
  1091. for( i = 0; i < 16; i++ )
  1092. prev_output[i] = tmp[i] ^ t[i];
  1093. }
  1094. return( 0 );
  1095. }
  1096. #endif /* MBEDTLS_CIPHER_MODE_XTS */
  1097. #if defined(MBEDTLS_CIPHER_MODE_CFB)
  1098. /*
  1099. * AES-CFB128 buffer encryption/decryption
  1100. */
  1101. int mbedtls_aes_crypt_cfb128( mbedtls_aes_context *ctx,
  1102. int mode,
  1103. size_t length,
  1104. size_t *iv_off,
  1105. unsigned char iv[16],
  1106. const unsigned char *input,
  1107. unsigned char *output )
  1108. {
  1109. int c;
  1110. size_t n;
  1111. AES_VALIDATE_RET( ctx != NULL );
  1112. AES_VALIDATE_RET( mode == MBEDTLS_AES_ENCRYPT ||
  1113. mode == MBEDTLS_AES_DECRYPT );
  1114. AES_VALIDATE_RET( iv_off != NULL );
  1115. AES_VALIDATE_RET( iv != NULL );
  1116. AES_VALIDATE_RET( input != NULL );
  1117. AES_VALIDATE_RET( output != NULL );
  1118. n = *iv_off;
  1119. if( n > 15 )
  1120. return( MBEDTLS_ERR_AES_BAD_INPUT_DATA );
  1121. if( mode == MBEDTLS_AES_DECRYPT )
  1122. {
  1123. while( length-- )
  1124. {
  1125. if( n == 0 )
  1126. mbedtls_aes_crypt_ecb( ctx, MBEDTLS_AES_ENCRYPT, iv, iv );
  1127. c = *input++;
  1128. *output++ = (unsigned char)( c ^ iv[n] );
  1129. iv[n] = (unsigned char) c;
  1130. n = ( n + 1 ) & 0x0F;
  1131. }
  1132. }
  1133. else
  1134. {
  1135. while( length-- )
  1136. {
  1137. if( n == 0 )
  1138. mbedtls_aes_crypt_ecb( ctx, MBEDTLS_AES_ENCRYPT, iv, iv );
  1139. iv[n] = *output++ = (unsigned char)( iv[n] ^ *input++ );
  1140. n = ( n + 1 ) & 0x0F;
  1141. }
  1142. }
  1143. *iv_off = n;
  1144. return( 0 );
  1145. }
  1146. /*
  1147. * AES-CFB8 buffer encryption/decryption
  1148. */
  1149. int mbedtls_aes_crypt_cfb8( mbedtls_aes_context *ctx,
  1150. int mode,
  1151. size_t length,
  1152. unsigned char iv[16],
  1153. const unsigned char *input,
  1154. unsigned char *output )
  1155. {
  1156. unsigned char c;
  1157. unsigned char ov[17];
  1158. AES_VALIDATE_RET( ctx != NULL );
  1159. AES_VALIDATE_RET( mode == MBEDTLS_AES_ENCRYPT ||
  1160. mode == MBEDTLS_AES_DECRYPT );
  1161. AES_VALIDATE_RET( iv != NULL );
  1162. AES_VALIDATE_RET( input != NULL );
  1163. AES_VALIDATE_RET( output != NULL );
  1164. while( length-- )
  1165. {
  1166. memcpy( ov, iv, 16 );
  1167. mbedtls_aes_crypt_ecb( ctx, MBEDTLS_AES_ENCRYPT, iv, iv );
  1168. if( mode == MBEDTLS_AES_DECRYPT )
  1169. ov[16] = *input;
  1170. c = *output++ = (unsigned char)( iv[0] ^ *input++ );
  1171. if( mode == MBEDTLS_AES_ENCRYPT )
  1172. ov[16] = c;
  1173. memcpy( iv, ov + 1, 16 );
  1174. }
  1175. return( 0 );
  1176. }
  1177. #endif /* MBEDTLS_CIPHER_MODE_CFB */
  1178. #if defined(MBEDTLS_CIPHER_MODE_OFB)
  1179. /*
  1180. * AES-OFB (Output Feedback Mode) buffer encryption/decryption
  1181. */
  1182. int mbedtls_aes_crypt_ofb( mbedtls_aes_context *ctx,
  1183. size_t length,
  1184. size_t *iv_off,
  1185. unsigned char iv[16],
  1186. const unsigned char *input,
  1187. unsigned char *output )
  1188. {
  1189. int ret = 0;
  1190. size_t n;
  1191. AES_VALIDATE_RET( ctx != NULL );
  1192. AES_VALIDATE_RET( iv_off != NULL );
  1193. AES_VALIDATE_RET( iv != NULL );
  1194. AES_VALIDATE_RET( input != NULL );
  1195. AES_VALIDATE_RET( output != NULL );
  1196. n = *iv_off;
  1197. if( n > 15 )
  1198. return( MBEDTLS_ERR_AES_BAD_INPUT_DATA );
  1199. while( length-- )
  1200. {
  1201. if( n == 0 )
  1202. {
  1203. ret = mbedtls_aes_crypt_ecb( ctx, MBEDTLS_AES_ENCRYPT, iv, iv );
  1204. if( ret != 0 )
  1205. goto exit;
  1206. }
  1207. *output++ = *input++ ^ iv[n];
  1208. n = ( n + 1 ) & 0x0F;
  1209. }
  1210. *iv_off = n;
  1211. exit:
  1212. return( ret );
  1213. }
  1214. #endif /* MBEDTLS_CIPHER_MODE_OFB */
  1215. #if defined(MBEDTLS_CIPHER_MODE_CTR)
  1216. /*
  1217. * AES-CTR buffer encryption/decryption
  1218. */
  1219. int mbedtls_aes_crypt_ctr( mbedtls_aes_context *ctx,
  1220. size_t length,
  1221. size_t *nc_off,
  1222. unsigned char nonce_counter[16],
  1223. unsigned char stream_block[16],
  1224. const unsigned char *input,
  1225. unsigned char *output )
  1226. {
  1227. int c, i;
  1228. size_t n;
  1229. AES_VALIDATE_RET( ctx != NULL );
  1230. AES_VALIDATE_RET( nc_off != NULL );
  1231. AES_VALIDATE_RET( nonce_counter != NULL );
  1232. AES_VALIDATE_RET( stream_block != NULL );
  1233. AES_VALIDATE_RET( input != NULL );
  1234. AES_VALIDATE_RET( output != NULL );
  1235. n = *nc_off;
  1236. if ( n > 0x0F )
  1237. return( MBEDTLS_ERR_AES_BAD_INPUT_DATA );
  1238. while( length-- )
  1239. {
  1240. if( n == 0 ) {
  1241. mbedtls_aes_crypt_ecb( ctx, MBEDTLS_AES_ENCRYPT, nonce_counter, stream_block );
  1242. for( i = 16; i > 0; i-- )
  1243. if( ++nonce_counter[i - 1] != 0 )
  1244. break;
  1245. }
  1246. c = *input++;
  1247. *output++ = (unsigned char)( c ^ stream_block[n] );
  1248. n = ( n + 1 ) & 0x0F;
  1249. }
  1250. *nc_off = n;
  1251. return( 0 );
  1252. }
  1253. #endif /* MBEDTLS_CIPHER_MODE_CTR */
  1254. #endif /* !MBEDTLS_AES_ALT */
  1255. #if defined(MBEDTLS_SELF_TEST)
  1256. /*
  1257. * AES test vectors from:
  1258. *
  1259. * http://csrc.nist.gov/archive/aes/rijndael/rijndael-vals.zip
  1260. */
  1261. static const unsigned char aes_test_ecb_dec[3][16] =
  1262. {
  1263. { 0x44, 0x41, 0x6A, 0xC2, 0xD1, 0xF5, 0x3C, 0x58,
  1264. 0x33, 0x03, 0x91, 0x7E, 0x6B, 0xE9, 0xEB, 0xE0 },
  1265. { 0x48, 0xE3, 0x1E, 0x9E, 0x25, 0x67, 0x18, 0xF2,
  1266. 0x92, 0x29, 0x31, 0x9C, 0x19, 0xF1, 0x5B, 0xA4 },
  1267. { 0x05, 0x8C, 0xCF, 0xFD, 0xBB, 0xCB, 0x38, 0x2D,
  1268. 0x1F, 0x6F, 0x56, 0x58, 0x5D, 0x8A, 0x4A, 0xDE }
  1269. };
  1270. static const unsigned char aes_test_ecb_enc[3][16] =
  1271. {
  1272. { 0xC3, 0x4C, 0x05, 0x2C, 0xC0, 0xDA, 0x8D, 0x73,
  1273. 0x45, 0x1A, 0xFE, 0x5F, 0x03, 0xBE, 0x29, 0x7F },
  1274. { 0xF3, 0xF6, 0x75, 0x2A, 0xE8, 0xD7, 0x83, 0x11,
  1275. 0x38, 0xF0, 0x41, 0x56, 0x06, 0x31, 0xB1, 0x14 },
  1276. { 0x8B, 0x79, 0xEE, 0xCC, 0x93, 0xA0, 0xEE, 0x5D,
  1277. 0xFF, 0x30, 0xB4, 0xEA, 0x21, 0x63, 0x6D, 0xA4 }
  1278. };
  1279. #if defined(MBEDTLS_CIPHER_MODE_CBC)
  1280. static const unsigned char aes_test_cbc_dec[3][16] =
  1281. {
  1282. { 0xFA, 0xCA, 0x37, 0xE0, 0xB0, 0xC8, 0x53, 0x73,
  1283. 0xDF, 0x70, 0x6E, 0x73, 0xF7, 0xC9, 0xAF, 0x86 },
  1284. { 0x5D, 0xF6, 0x78, 0xDD, 0x17, 0xBA, 0x4E, 0x75,
  1285. 0xB6, 0x17, 0x68, 0xC6, 0xAD, 0xEF, 0x7C, 0x7B },
  1286. { 0x48, 0x04, 0xE1, 0x81, 0x8F, 0xE6, 0x29, 0x75,
  1287. 0x19, 0xA3, 0xE8, 0x8C, 0x57, 0x31, 0x04, 0x13 }
  1288. };
  1289. static const unsigned char aes_test_cbc_enc[3][16] =
  1290. {
  1291. { 0x8A, 0x05, 0xFC, 0x5E, 0x09, 0x5A, 0xF4, 0x84,
  1292. 0x8A, 0x08, 0xD3, 0x28, 0xD3, 0x68, 0x8E, 0x3D },
  1293. { 0x7B, 0xD9, 0x66, 0xD5, 0x3A, 0xD8, 0xC1, 0xBB,
  1294. 0x85, 0xD2, 0xAD, 0xFA, 0xE8, 0x7B, 0xB1, 0x04 },
  1295. { 0xFE, 0x3C, 0x53, 0x65, 0x3E, 0x2F, 0x45, 0xB5,
  1296. 0x6F, 0xCD, 0x88, 0xB2, 0xCC, 0x89, 0x8F, 0xF0 }
  1297. };
  1298. #endif /* MBEDTLS_CIPHER_MODE_CBC */
  1299. #if defined(MBEDTLS_CIPHER_MODE_CFB)
  1300. /*
  1301. * AES-CFB128 test vectors from:
  1302. *
  1303. * http://csrc.nist.gov/publications/nistpubs/800-38a/sp800-38a.pdf
  1304. */
  1305. static const unsigned char aes_test_cfb128_key[3][32] =
  1306. {
  1307. { 0x2B, 0x7E, 0x15, 0x16, 0x28, 0xAE, 0xD2, 0xA6,
  1308. 0xAB, 0xF7, 0x15, 0x88, 0x09, 0xCF, 0x4F, 0x3C },
  1309. { 0x8E, 0x73, 0xB0, 0xF7, 0xDA, 0x0E, 0x64, 0x52,
  1310. 0xC8, 0x10, 0xF3, 0x2B, 0x80, 0x90, 0x79, 0xE5,
  1311. 0x62, 0xF8, 0xEA, 0xD2, 0x52, 0x2C, 0x6B, 0x7B },
  1312. { 0x60, 0x3D, 0xEB, 0x10, 0x15, 0xCA, 0x71, 0xBE,
  1313. 0x2B, 0x73, 0xAE, 0xF0, 0x85, 0x7D, 0x77, 0x81,
  1314. 0x1F, 0x35, 0x2C, 0x07, 0x3B, 0x61, 0x08, 0xD7,
  1315. 0x2D, 0x98, 0x10, 0xA3, 0x09, 0x14, 0xDF, 0xF4 }
  1316. };
  1317. static const unsigned char aes_test_cfb128_iv[16] =
  1318. {
  1319. 0x00, 0x01, 0x02, 0x03, 0x04, 0x05, 0x06, 0x07,
  1320. 0x08, 0x09, 0x0A, 0x0B, 0x0C, 0x0D, 0x0E, 0x0F
  1321. };
  1322. static const unsigned char aes_test_cfb128_pt[64] =
  1323. {
  1324. 0x6B, 0xC1, 0xBE, 0xE2, 0x2E, 0x40, 0x9F, 0x96,
  1325. 0xE9, 0x3D, 0x7E, 0x11, 0x73, 0x93, 0x17, 0x2A,
  1326. 0xAE, 0x2D, 0x8A, 0x57, 0x1E, 0x03, 0xAC, 0x9C,
  1327. 0x9E, 0xB7, 0x6F, 0xAC, 0x45, 0xAF, 0x8E, 0x51,
  1328. 0x30, 0xC8, 0x1C, 0x46, 0xA3, 0x5C, 0xE4, 0x11,
  1329. 0xE5, 0xFB, 0xC1, 0x19, 0x1A, 0x0A, 0x52, 0xEF,
  1330. 0xF6, 0x9F, 0x24, 0x45, 0xDF, 0x4F, 0x9B, 0x17,
  1331. 0xAD, 0x2B, 0x41, 0x7B, 0xE6, 0x6C, 0x37, 0x10
  1332. };
  1333. static const unsigned char aes_test_cfb128_ct[3][64] =
  1334. {
  1335. { 0x3B, 0x3F, 0xD9, 0x2E, 0xB7, 0x2D, 0xAD, 0x20,
  1336. 0x33, 0x34, 0x49, 0xF8, 0xE8, 0x3C, 0xFB, 0x4A,
  1337. 0xC8, 0xA6, 0x45, 0x37, 0xA0, 0xB3, 0xA9, 0x3F,
  1338. 0xCD, 0xE3, 0xCD, 0xAD, 0x9F, 0x1C, 0xE5, 0x8B,
  1339. 0x26, 0x75, 0x1F, 0x67, 0xA3, 0xCB, 0xB1, 0x40,
  1340. 0xB1, 0x80, 0x8C, 0xF1, 0x87, 0xA4, 0xF4, 0xDF,
  1341. 0xC0, 0x4B, 0x05, 0x35, 0x7C, 0x5D, 0x1C, 0x0E,
  1342. 0xEA, 0xC4, 0xC6, 0x6F, 0x9F, 0xF7, 0xF2, 0xE6 },
  1343. { 0xCD, 0xC8, 0x0D, 0x6F, 0xDD, 0xF1, 0x8C, 0xAB,
  1344. 0x34, 0xC2, 0x59, 0x09, 0xC9, 0x9A, 0x41, 0x74,
  1345. 0x67, 0xCE, 0x7F, 0x7F, 0x81, 0x17, 0x36, 0x21,
  1346. 0x96, 0x1A, 0x2B, 0x70, 0x17, 0x1D, 0x3D, 0x7A,
  1347. 0x2E, 0x1E, 0x8A, 0x1D, 0xD5, 0x9B, 0x88, 0xB1,
  1348. 0xC8, 0xE6, 0x0F, 0xED, 0x1E, 0xFA, 0xC4, 0xC9,
  1349. 0xC0, 0x5F, 0x9F, 0x9C, 0xA9, 0x83, 0x4F, 0xA0,
  1350. 0x42, 0xAE, 0x8F, 0xBA, 0x58, 0x4B, 0x09, 0xFF },
  1351. { 0xDC, 0x7E, 0x84, 0xBF, 0xDA, 0x79, 0x16, 0x4B,
  1352. 0x7E, 0xCD, 0x84, 0x86, 0x98, 0x5D, 0x38, 0x60,
  1353. 0x39, 0xFF, 0xED, 0x14, 0x3B, 0x28, 0xB1, 0xC8,
  1354. 0x32, 0x11, 0x3C, 0x63, 0x31, 0xE5, 0x40, 0x7B,
  1355. 0xDF, 0x10, 0x13, 0x24, 0x15, 0xE5, 0x4B, 0x92,
  1356. 0xA1, 0x3E, 0xD0, 0xA8, 0x26, 0x7A, 0xE2, 0xF9,
  1357. 0x75, 0xA3, 0x85, 0x74, 0x1A, 0xB9, 0xCE, 0xF8,
  1358. 0x20, 0x31, 0x62, 0x3D, 0x55, 0xB1, 0xE4, 0x71 }
  1359. };
  1360. #endif /* MBEDTLS_CIPHER_MODE_CFB */
  1361. #if defined(MBEDTLS_CIPHER_MODE_OFB)
  1362. /*
  1363. * AES-OFB test vectors from:
  1364. *
  1365. * https://csrc.nist.gov/publications/detail/sp/800-38a/final
  1366. */
  1367. static const unsigned char aes_test_ofb_key[3][32] =
  1368. {
  1369. { 0x2B, 0x7E, 0x15, 0x16, 0x28, 0xAE, 0xD2, 0xA6,
  1370. 0xAB, 0xF7, 0x15, 0x88, 0x09, 0xCF, 0x4F, 0x3C },
  1371. { 0x8E, 0x73, 0xB0, 0xF7, 0xDA, 0x0E, 0x64, 0x52,
  1372. 0xC8, 0x10, 0xF3, 0x2B, 0x80, 0x90, 0x79, 0xE5,
  1373. 0x62, 0xF8, 0xEA, 0xD2, 0x52, 0x2C, 0x6B, 0x7B },
  1374. { 0x60, 0x3D, 0xEB, 0x10, 0x15, 0xCA, 0x71, 0xBE,
  1375. 0x2B, 0x73, 0xAE, 0xF0, 0x85, 0x7D, 0x77, 0x81,
  1376. 0x1F, 0x35, 0x2C, 0x07, 0x3B, 0x61, 0x08, 0xD7,
  1377. 0x2D, 0x98, 0x10, 0xA3, 0x09, 0x14, 0xDF, 0xF4 }
  1378. };
  1379. static const unsigned char aes_test_ofb_iv[16] =
  1380. {
  1381. 0x00, 0x01, 0x02, 0x03, 0x04, 0x05, 0x06, 0x07,
  1382. 0x08, 0x09, 0x0A, 0x0B, 0x0C, 0x0D, 0x0E, 0x0F
  1383. };
  1384. static const unsigned char aes_test_ofb_pt[64] =
  1385. {
  1386. 0x6B, 0xC1, 0xBE, 0xE2, 0x2E, 0x40, 0x9F, 0x96,
  1387. 0xE9, 0x3D, 0x7E, 0x11, 0x73, 0x93, 0x17, 0x2A,
  1388. 0xAE, 0x2D, 0x8A, 0x57, 0x1E, 0x03, 0xAC, 0x9C,
  1389. 0x9E, 0xB7, 0x6F, 0xAC, 0x45, 0xAF, 0x8E, 0x51,
  1390. 0x30, 0xC8, 0x1C, 0x46, 0xA3, 0x5C, 0xE4, 0x11,
  1391. 0xE5, 0xFB, 0xC1, 0x19, 0x1A, 0x0A, 0x52, 0xEF,
  1392. 0xF6, 0x9F, 0x24, 0x45, 0xDF, 0x4F, 0x9B, 0x17,
  1393. 0xAD, 0x2B, 0x41, 0x7B, 0xE6, 0x6C, 0x37, 0x10
  1394. };
  1395. static const unsigned char aes_test_ofb_ct[3][64] =
  1396. {
  1397. { 0x3B, 0x3F, 0xD9, 0x2E, 0xB7, 0x2D, 0xAD, 0x20,
  1398. 0x33, 0x34, 0x49, 0xF8, 0xE8, 0x3C, 0xFB, 0x4A,
  1399. 0x77, 0x89, 0x50, 0x8d, 0x16, 0x91, 0x8f, 0x03,
  1400. 0xf5, 0x3c, 0x52, 0xda, 0xc5, 0x4e, 0xd8, 0x25,
  1401. 0x97, 0x40, 0x05, 0x1e, 0x9c, 0x5f, 0xec, 0xf6,
  1402. 0x43, 0x44, 0xf7, 0xa8, 0x22, 0x60, 0xed, 0xcc,
  1403. 0x30, 0x4c, 0x65, 0x28, 0xf6, 0x59, 0xc7, 0x78,
  1404. 0x66, 0xa5, 0x10, 0xd9, 0xc1, 0xd6, 0xae, 0x5e },
  1405. { 0xCD, 0xC8, 0x0D, 0x6F, 0xDD, 0xF1, 0x8C, 0xAB,
  1406. 0x34, 0xC2, 0x59, 0x09, 0xC9, 0x9A, 0x41, 0x74,
  1407. 0xfc, 0xc2, 0x8b, 0x8d, 0x4c, 0x63, 0x83, 0x7c,
  1408. 0x09, 0xe8, 0x17, 0x00, 0xc1, 0x10, 0x04, 0x01,
  1409. 0x8d, 0x9a, 0x9a, 0xea, 0xc0, 0xf6, 0x59, 0x6f,
  1410. 0x55, 0x9c, 0x6d, 0x4d, 0xaf, 0x59, 0xa5, 0xf2,
  1411. 0x6d, 0x9f, 0x20, 0x08, 0x57, 0xca, 0x6c, 0x3e,
  1412. 0x9c, 0xac, 0x52, 0x4b, 0xd9, 0xac, 0xc9, 0x2a },
  1413. { 0xDC, 0x7E, 0x84, 0xBF, 0xDA, 0x79, 0x16, 0x4B,
  1414. 0x7E, 0xCD, 0x84, 0x86, 0x98, 0x5D, 0x38, 0x60,
  1415. 0x4f, 0xeb, 0xdc, 0x67, 0x40, 0xd2, 0x0b, 0x3a,
  1416. 0xc8, 0x8f, 0x6a, 0xd8, 0x2a, 0x4f, 0xb0, 0x8d,
  1417. 0x71, 0xab, 0x47, 0xa0, 0x86, 0xe8, 0x6e, 0xed,
  1418. 0xf3, 0x9d, 0x1c, 0x5b, 0xba, 0x97, 0xc4, 0x08,
  1419. 0x01, 0x26, 0x14, 0x1d, 0x67, 0xf3, 0x7b, 0xe8,
  1420. 0x53, 0x8f, 0x5a, 0x8b, 0xe7, 0x40, 0xe4, 0x84 }
  1421. };
  1422. #endif /* MBEDTLS_CIPHER_MODE_OFB */
  1423. #if defined(MBEDTLS_CIPHER_MODE_CTR)
  1424. /*
  1425. * AES-CTR test vectors from:
  1426. *
  1427. * http://www.faqs.org/rfcs/rfc3686.html
  1428. */
  1429. static const unsigned char aes_test_ctr_key[3][16] =
  1430. {
  1431. { 0xAE, 0x68, 0x52, 0xF8, 0x12, 0x10, 0x67, 0xCC,
  1432. 0x4B, 0xF7, 0xA5, 0x76, 0x55, 0x77, 0xF3, 0x9E },
  1433. { 0x7E, 0x24, 0x06, 0x78, 0x17, 0xFA, 0xE0, 0xD7,
  1434. 0x43, 0xD6, 0xCE, 0x1F, 0x32, 0x53, 0x91, 0x63 },
  1435. { 0x76, 0x91, 0xBE, 0x03, 0x5E, 0x50, 0x20, 0xA8,
  1436. 0xAC, 0x6E, 0x61, 0x85, 0x29, 0xF9, 0xA0, 0xDC }
  1437. };
  1438. static const unsigned char aes_test_ctr_nonce_counter[3][16] =
  1439. {
  1440. { 0x00, 0x00, 0x00, 0x30, 0x00, 0x00, 0x00, 0x00,
  1441. 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x01 },
  1442. { 0x00, 0x6C, 0xB6, 0xDB, 0xC0, 0x54, 0x3B, 0x59,
  1443. 0xDA, 0x48, 0xD9, 0x0B, 0x00, 0x00, 0x00, 0x01 },
  1444. { 0x00, 0xE0, 0x01, 0x7B, 0x27, 0x77, 0x7F, 0x3F,
  1445. 0x4A, 0x17, 0x86, 0xF0, 0x00, 0x00, 0x00, 0x01 }
  1446. };
  1447. static const unsigned char aes_test_ctr_pt[3][48] =
  1448. {
  1449. { 0x53, 0x69, 0x6E, 0x67, 0x6C, 0x65, 0x20, 0x62,
  1450. 0x6C, 0x6F, 0x63, 0x6B, 0x20, 0x6D, 0x73, 0x67 },
  1451. { 0x00, 0x01, 0x02, 0x03, 0x04, 0x05, 0x06, 0x07,
  1452. 0x08, 0x09, 0x0A, 0x0B, 0x0C, 0x0D, 0x0E, 0x0F,
  1453. 0x10, 0x11, 0x12, 0x13, 0x14, 0x15, 0x16, 0x17,
  1454. 0x18, 0x19, 0x1A, 0x1B, 0x1C, 0x1D, 0x1E, 0x1F },
  1455. { 0x00, 0x01, 0x02, 0x03, 0x04, 0x05, 0x06, 0x07,
  1456. 0x08, 0x09, 0x0A, 0x0B, 0x0C, 0x0D, 0x0E, 0x0F,
  1457. 0x10, 0x11, 0x12, 0x13, 0x14, 0x15, 0x16, 0x17,
  1458. 0x18, 0x19, 0x1A, 0x1B, 0x1C, 0x1D, 0x1E, 0x1F,
  1459. 0x20, 0x21, 0x22, 0x23 }
  1460. };
  1461. static const unsigned char aes_test_ctr_ct[3][48] =
  1462. {
  1463. { 0xE4, 0x09, 0x5D, 0x4F, 0xB7, 0xA7, 0xB3, 0x79,
  1464. 0x2D, 0x61, 0x75, 0xA3, 0x26, 0x13, 0x11, 0xB8 },
  1465. { 0x51, 0x04, 0xA1, 0x06, 0x16, 0x8A, 0x72, 0xD9,
  1466. 0x79, 0x0D, 0x41, 0xEE, 0x8E, 0xDA, 0xD3, 0x88,
  1467. 0xEB, 0x2E, 0x1E, 0xFC, 0x46, 0xDA, 0x57, 0xC8,
  1468. 0xFC, 0xE6, 0x30, 0xDF, 0x91, 0x41, 0xBE, 0x28 },
  1469. { 0xC1, 0xCF, 0x48, 0xA8, 0x9F, 0x2F, 0xFD, 0xD9,
  1470. 0xCF, 0x46, 0x52, 0xE9, 0xEF, 0xDB, 0x72, 0xD7,
  1471. 0x45, 0x40, 0xA4, 0x2B, 0xDE, 0x6D, 0x78, 0x36,
  1472. 0xD5, 0x9A, 0x5C, 0xEA, 0xAE, 0xF3, 0x10, 0x53,
  1473. 0x25, 0xB2, 0x07, 0x2F }
  1474. };
  1475. static const int aes_test_ctr_len[3] =
  1476. { 16, 32, 36 };
  1477. #endif /* MBEDTLS_CIPHER_MODE_CTR */
  1478. #if defined(MBEDTLS_CIPHER_MODE_XTS)
  1479. /*
  1480. * AES-XTS test vectors from:
  1481. *
  1482. * IEEE P1619/D16 Annex B
  1483. * https://web.archive.org/web/20150629024421/http://grouper.ieee.org/groups/1619/email/pdf00086.pdf
  1484. * (Archived from original at http://grouper.ieee.org/groups/1619/email/pdf00086.pdf)
  1485. */
  1486. static const unsigned char aes_test_xts_key[][32] =
  1487. {
  1488. { 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
  1489. 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
  1490. 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
  1491. 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00 },
  1492. { 0x11, 0x11, 0x11, 0x11, 0x11, 0x11, 0x11, 0x11,
  1493. 0x11, 0x11, 0x11, 0x11, 0x11, 0x11, 0x11, 0x11,
  1494. 0x22, 0x22, 0x22, 0x22, 0x22, 0x22, 0x22, 0x22,
  1495. 0x22, 0x22, 0x22, 0x22, 0x22, 0x22, 0x22, 0x22 },
  1496. { 0xff, 0xfe, 0xfd, 0xfc, 0xfb, 0xfa, 0xf9, 0xf8,
  1497. 0xf7, 0xf6, 0xf5, 0xf4, 0xf3, 0xf2, 0xf1, 0xf0,
  1498. 0x22, 0x22, 0x22, 0x22, 0x22, 0x22, 0x22, 0x22,
  1499. 0x22, 0x22, 0x22, 0x22, 0x22, 0x22, 0x22, 0x22 },
  1500. };
  1501. static const unsigned char aes_test_xts_pt32[][32] =
  1502. {
  1503. { 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
  1504. 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
  1505. 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
  1506. 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00 },
  1507. { 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44,
  1508. 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44,
  1509. 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44,
  1510. 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44 },
  1511. { 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44,
  1512. 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44,
  1513. 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44,
  1514. 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44, 0x44 },
  1515. };
  1516. static const unsigned char aes_test_xts_ct32[][32] =
  1517. {
  1518. { 0x91, 0x7c, 0xf6, 0x9e, 0xbd, 0x68, 0xb2, 0xec,
  1519. 0x9b, 0x9f, 0xe9, 0xa3, 0xea, 0xdd, 0xa6, 0x92,
  1520. 0xcd, 0x43, 0xd2, 0xf5, 0x95, 0x98, 0xed, 0x85,
  1521. 0x8c, 0x02, 0xc2, 0x65, 0x2f, 0xbf, 0x92, 0x2e },
  1522. { 0xc4, 0x54, 0x18, 0x5e, 0x6a, 0x16, 0x93, 0x6e,
  1523. 0x39, 0x33, 0x40, 0x38, 0xac, 0xef, 0x83, 0x8b,
  1524. 0xfb, 0x18, 0x6f, 0xff, 0x74, 0x80, 0xad, 0xc4,
  1525. 0x28, 0x93, 0x82, 0xec, 0xd6, 0xd3, 0x94, 0xf0 },
  1526. { 0xaf, 0x85, 0x33, 0x6b, 0x59, 0x7a, 0xfc, 0x1a,
  1527. 0x90, 0x0b, 0x2e, 0xb2, 0x1e, 0xc9, 0x49, 0xd2,
  1528. 0x92, 0xdf, 0x4c, 0x04, 0x7e, 0x0b, 0x21, 0x53,
  1529. 0x21, 0x86, 0xa5, 0x97, 0x1a, 0x22, 0x7a, 0x89 },
  1530. };
  1531. static const unsigned char aes_test_xts_data_unit[][16] =
  1532. {
  1533. { 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
  1534. 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00 },
  1535. { 0x33, 0x33, 0x33, 0x33, 0x33, 0x00, 0x00, 0x00,
  1536. 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00 },
  1537. { 0x33, 0x33, 0x33, 0x33, 0x33, 0x00, 0x00, 0x00,
  1538. 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00 },
  1539. };
  1540. #endif /* MBEDTLS_CIPHER_MODE_XTS */
  1541. /*
  1542. * Checkup routine
  1543. */
  1544. int mbedtls_aes_self_test( int verbose )
  1545. {
  1546. int ret = 0, i, j, u, mode;
  1547. unsigned int keybits;
  1548. unsigned char key[32];
  1549. unsigned char buf[64];
  1550. const unsigned char *aes_tests;
  1551. #if defined(MBEDTLS_CIPHER_MODE_CBC) || defined(MBEDTLS_CIPHER_MODE_CFB)
  1552. unsigned char iv[16];
  1553. #endif
  1554. #if defined(MBEDTLS_CIPHER_MODE_CBC)
  1555. unsigned char prv[16];
  1556. #endif
  1557. #if defined(MBEDTLS_CIPHER_MODE_CTR) || defined(MBEDTLS_CIPHER_MODE_CFB) || \
  1558. defined(MBEDTLS_CIPHER_MODE_OFB)
  1559. size_t offset;
  1560. #endif
  1561. #if defined(MBEDTLS_CIPHER_MODE_CTR) || defined(MBEDTLS_CIPHER_MODE_XTS)
  1562. int len;
  1563. #endif
  1564. #if defined(MBEDTLS_CIPHER_MODE_CTR)
  1565. unsigned char nonce_counter[16];
  1566. unsigned char stream_block[16];
  1567. #endif
  1568. mbedtls_aes_context ctx;
  1569. memset( key, 0, 32 );
  1570. mbedtls_aes_init( &ctx );
  1571. /*
  1572. * ECB mode
  1573. */
  1574. for( i = 0; i < 6; i++ )
  1575. {
  1576. u = i >> 1;
  1577. keybits = 128 + u * 64;
  1578. mode = i & 1;
  1579. if( verbose != 0 )
  1580. mbedtls_printf( " AES-ECB-%3u (%s): ", keybits,
  1581. ( mode == MBEDTLS_AES_DECRYPT ) ? "dec" : "enc" );
  1582. memset( buf, 0, 16 );
  1583. if( mode == MBEDTLS_AES_DECRYPT )
  1584. {
  1585. ret = mbedtls_aes_setkey_dec( &ctx, key, keybits );
  1586. aes_tests = aes_test_ecb_dec[u];
  1587. }
  1588. else
  1589. {
  1590. ret = mbedtls_aes_setkey_enc( &ctx, key, keybits );
  1591. aes_tests = aes_test_ecb_enc[u];
  1592. }
  1593. /*
  1594. * AES-192 is an optional feature that may be unavailable when
  1595. * there is an alternative underlying implementation i.e. when
  1596. * MBEDTLS_AES_ALT is defined.
  1597. */
  1598. if( ret == MBEDTLS_ERR_PLATFORM_FEATURE_UNSUPPORTED && keybits == 192 )
  1599. {
  1600. mbedtls_printf( "skipped\n" );
  1601. continue;
  1602. }
  1603. else if( ret != 0 )
  1604. {
  1605. goto exit;
  1606. }
  1607. for( j = 0; j < 10000; j++ )
  1608. {
  1609. ret = mbedtls_aes_crypt_ecb( &ctx, mode, buf, buf );
  1610. if( ret != 0 )
  1611. goto exit;
  1612. }
  1613. if( memcmp( buf, aes_tests, 16 ) != 0 )
  1614. {
  1615. ret = 1;
  1616. goto exit;
  1617. }
  1618. if( verbose != 0 )
  1619. mbedtls_printf( "passed\n" );
  1620. }
  1621. if( verbose != 0 )
  1622. mbedtls_printf( "\n" );
  1623. #if defined(MBEDTLS_CIPHER_MODE_CBC)
  1624. /*
  1625. * CBC mode
  1626. */
  1627. for( i = 0; i < 6; i++ )
  1628. {
  1629. u = i >> 1;
  1630. keybits = 128 + u * 64;
  1631. mode = i & 1;
  1632. if( verbose != 0 )
  1633. mbedtls_printf( " AES-CBC-%3u (%s): ", keybits,
  1634. ( mode == MBEDTLS_AES_DECRYPT ) ? "dec" : "enc" );
  1635. memset( iv , 0, 16 );
  1636. memset( prv, 0, 16 );
  1637. memset( buf, 0, 16 );
  1638. if( mode == MBEDTLS_AES_DECRYPT )
  1639. {
  1640. ret = mbedtls_aes_setkey_dec( &ctx, key, keybits );
  1641. aes_tests = aes_test_cbc_dec[u];
  1642. }
  1643. else
  1644. {
  1645. ret = mbedtls_aes_setkey_enc( &ctx, key, keybits );
  1646. aes_tests = aes_test_cbc_enc[u];
  1647. }
  1648. /*
  1649. * AES-192 is an optional feature that may be unavailable when
  1650. * there is an alternative underlying implementation i.e. when
  1651. * MBEDTLS_AES_ALT is defined.
  1652. */
  1653. if( ret == MBEDTLS_ERR_PLATFORM_FEATURE_UNSUPPORTED && keybits == 192 )
  1654. {
  1655. mbedtls_printf( "skipped\n" );
  1656. continue;
  1657. }
  1658. else if( ret != 0 )
  1659. {
  1660. goto exit;
  1661. }
  1662. for( j = 0; j < 10000; j++ )
  1663. {
  1664. if( mode == MBEDTLS_AES_ENCRYPT )
  1665. {
  1666. unsigned char tmp[16];
  1667. memcpy( tmp, prv, 16 );
  1668. memcpy( prv, buf, 16 );
  1669. memcpy( buf, tmp, 16 );
  1670. }
  1671. ret = mbedtls_aes_crypt_cbc( &ctx, mode, 16, iv, buf, buf );
  1672. if( ret != 0 )
  1673. goto exit;
  1674. }
  1675. if( memcmp( buf, aes_tests, 16 ) != 0 )
  1676. {
  1677. ret = 1;
  1678. goto exit;
  1679. }
  1680. if( verbose != 0 )
  1681. mbedtls_printf( "passed\n" );
  1682. }
  1683. if( verbose != 0 )
  1684. mbedtls_printf( "\n" );
  1685. #endif /* MBEDTLS_CIPHER_MODE_CBC */
  1686. #if defined(MBEDTLS_CIPHER_MODE_CFB)
  1687. /*
  1688. * CFB128 mode
  1689. */
  1690. for( i = 0; i < 6; i++ )
  1691. {
  1692. u = i >> 1;
  1693. keybits = 128 + u * 64;
  1694. mode = i & 1;
  1695. if( verbose != 0 )
  1696. mbedtls_printf( " AES-CFB128-%3u (%s): ", keybits,
  1697. ( mode == MBEDTLS_AES_DECRYPT ) ? "dec" : "enc" );
  1698. memcpy( iv, aes_test_cfb128_iv, 16 );
  1699. memcpy( key, aes_test_cfb128_key[u], keybits / 8 );
  1700. offset = 0;
  1701. ret = mbedtls_aes_setkey_enc( &ctx, key, keybits );
  1702. /*
  1703. * AES-192 is an optional feature that may be unavailable when
  1704. * there is an alternative underlying implementation i.e. when
  1705. * MBEDTLS_AES_ALT is defined.
  1706. */
  1707. if( ret == MBEDTLS_ERR_PLATFORM_FEATURE_UNSUPPORTED && keybits == 192 )
  1708. {
  1709. mbedtls_printf( "skipped\n" );
  1710. continue;
  1711. }
  1712. else if( ret != 0 )
  1713. {
  1714. goto exit;
  1715. }
  1716. if( mode == MBEDTLS_AES_DECRYPT )
  1717. {
  1718. memcpy( buf, aes_test_cfb128_ct[u], 64 );
  1719. aes_tests = aes_test_cfb128_pt;
  1720. }
  1721. else
  1722. {
  1723. memcpy( buf, aes_test_cfb128_pt, 64 );
  1724. aes_tests = aes_test_cfb128_ct[u];
  1725. }
  1726. ret = mbedtls_aes_crypt_cfb128( &ctx, mode, 64, &offset, iv, buf, buf );
  1727. if( ret != 0 )
  1728. goto exit;
  1729. if( memcmp( buf, aes_tests, 64 ) != 0 )
  1730. {
  1731. ret = 1;
  1732. goto exit;
  1733. }
  1734. if( verbose != 0 )
  1735. mbedtls_printf( "passed\n" );
  1736. }
  1737. if( verbose != 0 )
  1738. mbedtls_printf( "\n" );
  1739. #endif /* MBEDTLS_CIPHER_MODE_CFB */
  1740. #if defined(MBEDTLS_CIPHER_MODE_OFB)
  1741. /*
  1742. * OFB mode
  1743. */
  1744. for( i = 0; i < 6; i++ )
  1745. {
  1746. u = i >> 1;
  1747. keybits = 128 + u * 64;
  1748. mode = i & 1;
  1749. if( verbose != 0 )
  1750. mbedtls_printf( " AES-OFB-%3u (%s): ", keybits,
  1751. ( mode == MBEDTLS_AES_DECRYPT ) ? "dec" : "enc" );
  1752. memcpy( iv, aes_test_ofb_iv, 16 );
  1753. memcpy( key, aes_test_ofb_key[u], keybits / 8 );
  1754. offset = 0;
  1755. ret = mbedtls_aes_setkey_enc( &ctx, key, keybits );
  1756. /*
  1757. * AES-192 is an optional feature that may be unavailable when
  1758. * there is an alternative underlying implementation i.e. when
  1759. * MBEDTLS_AES_ALT is defined.
  1760. */
  1761. if( ret == MBEDTLS_ERR_PLATFORM_FEATURE_UNSUPPORTED && keybits == 192 )
  1762. {
  1763. mbedtls_printf( "skipped\n" );
  1764. continue;
  1765. }
  1766. else if( ret != 0 )
  1767. {
  1768. goto exit;
  1769. }
  1770. if( mode == MBEDTLS_AES_DECRYPT )
  1771. {
  1772. memcpy( buf, aes_test_ofb_ct[u], 64 );
  1773. aes_tests = aes_test_ofb_pt;
  1774. }
  1775. else
  1776. {
  1777. memcpy( buf, aes_test_ofb_pt, 64 );
  1778. aes_tests = aes_test_ofb_ct[u];
  1779. }
  1780. ret = mbedtls_aes_crypt_ofb( &ctx, 64, &offset, iv, buf, buf );
  1781. if( ret != 0 )
  1782. goto exit;
  1783. if( memcmp( buf, aes_tests, 64 ) != 0 )
  1784. {
  1785. ret = 1;
  1786. goto exit;
  1787. }
  1788. if( verbose != 0 )
  1789. mbedtls_printf( "passed\n" );
  1790. }
  1791. if( verbose != 0 )
  1792. mbedtls_printf( "\n" );
  1793. #endif /* MBEDTLS_CIPHER_MODE_OFB */
  1794. #if defined(MBEDTLS_CIPHER_MODE_CTR)
  1795. /*
  1796. * CTR mode
  1797. */
  1798. for( i = 0; i < 6; i++ )
  1799. {
  1800. u = i >> 1;
  1801. mode = i & 1;
  1802. if( verbose != 0 )
  1803. mbedtls_printf( " AES-CTR-128 (%s): ",
  1804. ( mode == MBEDTLS_AES_DECRYPT ) ? "dec" : "enc" );
  1805. memcpy( nonce_counter, aes_test_ctr_nonce_counter[u], 16 );
  1806. memcpy( key, aes_test_ctr_key[u], 16 );
  1807. offset = 0;
  1808. if( ( ret = mbedtls_aes_setkey_enc( &ctx, key, 128 ) ) != 0 )
  1809. goto exit;
  1810. len = aes_test_ctr_len[u];
  1811. if( mode == MBEDTLS_AES_DECRYPT )
  1812. {
  1813. memcpy( buf, aes_test_ctr_ct[u], len );
  1814. aes_tests = aes_test_ctr_pt[u];
  1815. }
  1816. else
  1817. {
  1818. memcpy( buf, aes_test_ctr_pt[u], len );
  1819. aes_tests = aes_test_ctr_ct[u];
  1820. }
  1821. ret = mbedtls_aes_crypt_ctr( &ctx, len, &offset, nonce_counter,
  1822. stream_block, buf, buf );
  1823. if( ret != 0 )
  1824. goto exit;
  1825. if( memcmp( buf, aes_tests, len ) != 0 )
  1826. {
  1827. ret = 1;
  1828. goto exit;
  1829. }
  1830. if( verbose != 0 )
  1831. mbedtls_printf( "passed\n" );
  1832. }
  1833. if( verbose != 0 )
  1834. mbedtls_printf( "\n" );
  1835. #endif /* MBEDTLS_CIPHER_MODE_CTR */
  1836. #if defined(MBEDTLS_CIPHER_MODE_XTS)
  1837. {
  1838. static const int num_tests =
  1839. sizeof(aes_test_xts_key) / sizeof(*aes_test_xts_key);
  1840. mbedtls_aes_xts_context ctx_xts;
  1841. /*
  1842. * XTS mode
  1843. */
  1844. mbedtls_aes_xts_init( &ctx_xts );
  1845. for( i = 0; i < num_tests << 1; i++ )
  1846. {
  1847. const unsigned char *data_unit;
  1848. u = i >> 1;
  1849. mode = i & 1;
  1850. if( verbose != 0 )
  1851. mbedtls_printf( " AES-XTS-128 (%s): ",
  1852. ( mode == MBEDTLS_AES_DECRYPT ) ? "dec" : "enc" );
  1853. memset( key, 0, sizeof( key ) );
  1854. memcpy( key, aes_test_xts_key[u], 32 );
  1855. data_unit = aes_test_xts_data_unit[u];
  1856. len = sizeof( *aes_test_xts_ct32 );
  1857. if( mode == MBEDTLS_AES_DECRYPT )
  1858. {
  1859. ret = mbedtls_aes_xts_setkey_dec( &ctx_xts, key, 256 );
  1860. if( ret != 0)
  1861. goto exit;
  1862. memcpy( buf, aes_test_xts_ct32[u], len );
  1863. aes_tests = aes_test_xts_pt32[u];
  1864. }
  1865. else
  1866. {
  1867. ret = mbedtls_aes_xts_setkey_enc( &ctx_xts, key, 256 );
  1868. if( ret != 0)
  1869. goto exit;
  1870. memcpy( buf, aes_test_xts_pt32[u], len );
  1871. aes_tests = aes_test_xts_ct32[u];
  1872. }
  1873. ret = mbedtls_aes_crypt_xts( &ctx_xts, mode, len, data_unit,
  1874. buf, buf );
  1875. if( ret != 0 )
  1876. goto exit;
  1877. if( memcmp( buf, aes_tests, len ) != 0 )
  1878. {
  1879. ret = 1;
  1880. goto exit;
  1881. }
  1882. if( verbose != 0 )
  1883. mbedtls_printf( "passed\n" );
  1884. }
  1885. if( verbose != 0 )
  1886. mbedtls_printf( "\n" );
  1887. mbedtls_aes_xts_free( &ctx_xts );
  1888. }
  1889. #endif /* MBEDTLS_CIPHER_MODE_XTS */
  1890. ret = 0;
  1891. exit:
  1892. if( ret != 0 && verbose != 0 )
  1893. mbedtls_printf( "failed\n" );
  1894. mbedtls_aes_free( &ctx );
  1895. return( ret );
  1896. }
  1897. #endif /* MBEDTLS_SELF_TEST */
  1898. #endif /* MBEDTLS_AES_C */