Path: blob/a-new-beginning/SharedDependencies/Sources/cryptopp/cham_simd.cpp
2 views
// cham_simd.cpp - written and placed in the public domain by Jeffrey Walton1//2// This source file uses intrinsics and built-ins to gain access to3// SSSE3, ARM NEON and ARMv8a, and Power7 Altivec instructions. A separate4// source file is needed because additional CXXFLAGS are required to enable5// the appropriate instructions sets in some build configurations.67#include "pch.h"8#include "config.h"910#include "cham.h"11#include "misc.h"1213// Uncomment for benchmarking C++ against SSE or NEON.14// Do so in both simon.cpp and simon_simd.cpp.15// #undef CRYPTOPP_SSSE3_AVAILABLE16// #undef CRYPTOPP_ARM_NEON_AVAILABLE1718#if (CRYPTOPP_SSSE3_AVAILABLE)19#include "adv_simd.h"20# include <pmmintrin.h>21# include <tmmintrin.h>22#endif2324#if defined(__XOP__)25# if defined(CRYPTOPP_GCC_COMPATIBLE)26# include <x86intrin.h>27# endif28# include <ammintrin.h>29#endif // XOP3031// Clang intrinsic casts, http://bugs.llvm.org/show_bug.cgi?id=2067032#define DOUBLE_CAST(x) ((double*)(void*)(x))33#define CONST_DOUBLE_CAST(x) ((const double*)(const void*)(x))3435// Squash MS LNK4221 and libtool warnings36extern const char CHAM_SIMD_FNAME[] = __FILE__;3738ANONYMOUS_NAMESPACE_BEGIN3940using CryptoPP::word16;41using CryptoPP::word32;4243#if (CRYPTOPP_SSSE3_AVAILABLE)4445//////////////////////////////////////////////////////////////////////////4647NAMESPACE_BEGIN(W32) // CHAM128, 32-bit word size4849template <unsigned int R>50inline __m128i RotateLeft32(const __m128i& val)51{52#if defined(__XOP__)53return _mm_roti_epi32(val, R);54#else55return _mm_or_si128(56_mm_slli_epi32(val, R), _mm_srli_epi32(val, 32-R));57#endif58}5960template <unsigned int R>61inline __m128i RotateRight32(const __m128i& val)62{63#if defined(__XOP__)64return _mm_roti_epi32(val, 32-R);65#else66return _mm_or_si128(67_mm_slli_epi32(val, 32-R), _mm_srli_epi32(val, R));68#endif69}7071// Faster than two Shifts and an Or. Thanks to Louis Wingers and Bryan Weeks.72template <>73inline __m128i RotateLeft32<8>(const __m128i& val)74{75#if defined(__XOP__)76return _mm_roti_epi32(val, 8);77#else78const __m128i mask = _mm_set_epi8(14,13,12,15, 10,9,8,11, 6,5,4,7, 2,1,0,3);79return _mm_shuffle_epi8(val, mask);80#endif81}8283// Faster than two Shifts and an Or. Thanks to Louis Wingers and Bryan Weeks.84template <>85inline __m128i RotateRight32<8>(const __m128i& val)86{87#if defined(__XOP__)88return _mm_roti_epi32(val, 32-8);89#else90const __m128i mask = _mm_set_epi8(12,15,14,13, 8,11,10,9, 4,7,6,5, 0,3,2,1);91return _mm_shuffle_epi8(val, mask);92#endif93}9495template <unsigned int IDX>96inline __m128i UnpackXMM(const __m128i& a, const __m128i& b, const __m128i& c, const __m128i& d)97{98// Should not be instantiated99CRYPTOPP_UNUSED(a); CRYPTOPP_UNUSED(b);100CRYPTOPP_UNUSED(c); CRYPTOPP_UNUSED(d);101CRYPTOPP_ASSERT(0);102return _mm_setzero_si128();103}104105template <>106inline __m128i UnpackXMM<0>(const __m128i& a, const __m128i& b, const __m128i& c, const __m128i& d)107{108// The shuffle converts to and from little-endian for SSE. A specialized109// CHAM implementation can avoid the shuffle by framing the data for110// encryption, decryption and benchmarks. The library cannot take the111// speed-up because of the byte oriented API.112const __m128i r1 = _mm_unpacklo_epi32(a, b);113const __m128i r2 = _mm_unpacklo_epi32(c, d);114return _mm_shuffle_epi8(_mm_unpacklo_epi64(r1, r2),115_mm_set_epi8(12,13,14,15, 8,9,10,11, 4,5,6,7, 0,1,2,3));116}117118template <>119inline __m128i UnpackXMM<1>(const __m128i& a, const __m128i& b, const __m128i& c, const __m128i& d)120{121// The shuffle converts to and from little-endian for SSE. A specialized122// CHAM implementation can avoid the shuffle by framing the data for123// encryption, decryption and benchmarks. The library cannot take the124// speed-up because of the byte oriented API.125const __m128i r1 = _mm_unpacklo_epi32(a, b);126const __m128i r2 = _mm_unpacklo_epi32(c, d);127return _mm_shuffle_epi8(_mm_unpackhi_epi64(r1, r2),128_mm_set_epi8(12,13,14,15, 8,9,10,11, 4,5,6,7, 0,1,2,3));129}130131template <>132inline __m128i UnpackXMM<2>(const __m128i& a, const __m128i& b, const __m128i& c, const __m128i& d)133{134// The shuffle converts to and from little-endian for SSE. A specialized135// CHAM implementation can avoid the shuffle by framing the data for136// encryption, decryption and benchmarks. The library cannot take the137// speed-up because of the byte oriented API.138const __m128i r1 = _mm_unpackhi_epi32(a, b);139const __m128i r2 = _mm_unpackhi_epi32(c, d);140return _mm_shuffle_epi8(_mm_unpacklo_epi64(r1, r2),141_mm_set_epi8(12,13,14,15, 8,9,10,11, 4,5,6,7, 0,1,2,3));142}143144template <>145inline __m128i UnpackXMM<3>(const __m128i& a, const __m128i& b, const __m128i& c, const __m128i& d)146{147// The shuffle converts to and from little-endian for SSE. A specialized148// CHAM implementation can avoid the shuffle by framing the data for149// encryption, decryption and benchmarks. The library cannot take the150// speed-up because of the byte oriented API.151const __m128i r1 = _mm_unpackhi_epi32(a, b);152const __m128i r2 = _mm_unpackhi_epi32(c, d);153return _mm_shuffle_epi8(_mm_unpackhi_epi64(r1, r2),154_mm_set_epi8(12,13,14,15, 8,9,10,11, 4,5,6,7, 0,1,2,3));155}156157template <unsigned int IDX>158inline __m128i UnpackXMM(const __m128i& v)159{160// Should not be instantiated161CRYPTOPP_UNUSED(v); CRYPTOPP_ASSERT(0);162return _mm_setzero_si128();163}164165template <>166inline __m128i UnpackXMM<0>(const __m128i& v)167{168return _mm_shuffle_epi8(v, _mm_set_epi8(0,1,2,3, 0,1,2,3, 0,1,2,3, 0,1,2,3));169}170171template <>172inline __m128i UnpackXMM<1>(const __m128i& v)173{174return _mm_shuffle_epi8(v, _mm_set_epi8(4,5,6,7, 4,5,6,7, 4,5,6,7, 4,5,6,7));175}176177template <>178inline __m128i UnpackXMM<2>(const __m128i& v)179{180return _mm_shuffle_epi8(v, _mm_set_epi8(8,9,10,11, 8,9,10,11, 8,9,10,11, 8,9,10,11));181}182183template <>184inline __m128i UnpackXMM<3>(const __m128i& v)185{186return _mm_shuffle_epi8(v, _mm_set_epi8(12,13,14,15, 12,13,14,15, 12,13,14,15, 12,13,14,15));187}188189template <unsigned int IDX>190inline __m128i RepackXMM(const __m128i& a, const __m128i& b, const __m128i& c, const __m128i& d)191{192return UnpackXMM<IDX>(a, b, c, d);193}194195template <unsigned int IDX>196inline __m128i RepackXMM(const __m128i& v)197{198return UnpackXMM<IDX>(v);199}200201inline void CHAM128_Enc_Block(__m128i &block0,202const word32 *subkeys, unsigned int rounds)203{204// Rearrange the data for vectorization. UnpackXMM includes a205// little-endian swap for SSE. Thanks to Peter Cordes for help206// with packing and unpacking.207// [A1 A2 A3 A4][B1 B2 B3 B4] ... => [A1 B1 C1 D1][A2 B2 C2 D2] ...208__m128i a = UnpackXMM<0>(block0);209__m128i b = UnpackXMM<1>(block0);210__m128i c = UnpackXMM<2>(block0);211__m128i d = UnpackXMM<3>(block0);212213__m128i counter = _mm_set_epi32(0,0,0,0);214__m128i increment = _mm_set_epi32(1,1,1,1);215216const unsigned int MASK = (rounds == 80 ? 7 : 15);217for (int i=0; i<static_cast<int>(rounds); i+=4)218{219__m128i k, k1, k2, t1, t2;220k = _mm_castpd_si128(_mm_load_sd(CONST_DOUBLE_CAST(&subkeys[(i+0) & MASK])));221222// Shuffle out two subkeys223k1 = _mm_shuffle_epi8(k, _mm_set_epi8(3,2,1,0, 3,2,1,0, 3,2,1,0, 3,2,1,0));224k2 = _mm_shuffle_epi8(k, _mm_set_epi8(7,6,5,4, 7,6,5,4, 7,6,5,4, 7,6,5,4));225226t1 = _mm_xor_si128(a, counter);227t2 = _mm_xor_si128(RotateLeft32<1>(b), k1);228a = RotateLeft32<8>(_mm_add_epi32(t1, t2));229230counter = _mm_add_epi32(counter, increment);231232t1 = _mm_xor_si128(b, counter);233t2 = _mm_xor_si128(RotateLeft32<8>(c), k2);234b = RotateLeft32<1>(_mm_add_epi32(t1, t2));235236counter = _mm_add_epi32(counter, increment);237k = _mm_castpd_si128(_mm_load_sd(CONST_DOUBLE_CAST(&subkeys[(i+2) & MASK])));238239// Shuffle out two subkeys240k1 = _mm_shuffle_epi8(k, _mm_set_epi8(3,2,1,0, 3,2,1,0, 3,2,1,0, 3,2,1,0));241k2 = _mm_shuffle_epi8(k, _mm_set_epi8(7,6,5,4, 7,6,5,4, 7,6,5,4, 7,6,5,4));242243t1 = _mm_xor_si128(c, counter);244t2 = _mm_xor_si128(RotateLeft32<1>(d), k1);245c = RotateLeft32<8>(_mm_add_epi32(t1, t2));246247counter = _mm_add_epi32(counter, increment);248249t1 = _mm_xor_si128(d, counter);250t2 = _mm_xor_si128(RotateLeft32<8>(a), k2);251d = RotateLeft32<1>(_mm_add_epi32(t1, t2));252253counter = _mm_add_epi32(counter, increment);254}255256// [A1 B1 C1 D1][A2 B2 C2 D2] ... => [A1 A2 A3 A4][B1 B2 B3 B4] ...257block0 = RepackXMM<0>(a,b,c,d);258}259260inline void CHAM128_Dec_Block(__m128i &block0,261const word32 *subkeys, unsigned int rounds)262{263// Rearrange the data for vectorization. UnpackXMM includes a264// little-endian swap for SSE. Thanks to Peter Cordes for help265// with packing and unpacking.266// [A1 A2 A3 A4][B1 B2 B3 B4] ... => [A1 B1 C1 D1][A2 B2 C2 D2] ...267__m128i a = UnpackXMM<0>(block0);268__m128i b = UnpackXMM<1>(block0);269__m128i c = UnpackXMM<2>(block0);270__m128i d = UnpackXMM<3>(block0);271272__m128i counter = _mm_set_epi32(rounds-1,rounds-1,rounds-1,rounds-1);273__m128i decrement = _mm_set_epi32(1,1,1,1);274275const unsigned int MASK = (rounds == 80 ? 7 : 15);276for (int i = static_cast<int>(rounds)-1; i >= 0; i-=4)277{278__m128i k, k1, k2, t1, t2;279k = _mm_castpd_si128(_mm_load_sd(CONST_DOUBLE_CAST(&subkeys[(i-1) & MASK])));280281// Shuffle out two subkeys282k1 = _mm_shuffle_epi8(k, _mm_set_epi8(7,6,5,4, 7,6,5,4, 7,6,5,4, 7,6,5,4));283k2 = _mm_shuffle_epi8(k, _mm_set_epi8(3,2,1,0, 3,2,1,0, 3,2,1,0, 3,2,1,0));284285// Odd round286t1 = RotateRight32<1>(d);287t2 = _mm_xor_si128(RotateLeft32<8>(a), k1);288d = _mm_xor_si128(_mm_sub_epi32(t1, t2), counter);289290counter = _mm_sub_epi32(counter, decrement);291292// Even round293t1 = RotateRight32<8>(c);294t2 = _mm_xor_si128(RotateLeft32<1>(d), k2);295c = _mm_xor_si128(_mm_sub_epi32(t1, t2), counter);296297counter = _mm_sub_epi32(counter, decrement);298k = _mm_castpd_si128(_mm_load_sd(CONST_DOUBLE_CAST(&subkeys[(i-3) & MASK])));299300// Shuffle out two subkeys301k1 = _mm_shuffle_epi8(k, _mm_set_epi8(7,6,5,4, 7,6,5,4, 7,6,5,4, 7,6,5,4));302k2 = _mm_shuffle_epi8(k, _mm_set_epi8(3,2,1,0, 3,2,1,0, 3,2,1,0, 3,2,1,0));303304// Odd round305t1 = RotateRight32<1>(b);306t2 = _mm_xor_si128(RotateLeft32<8>(c), k1);307b = _mm_xor_si128(_mm_sub_epi32(t1, t2), counter);308309counter = _mm_sub_epi32(counter, decrement);310311// Even round312t1 = RotateRight32<8>(a);313t2 = _mm_xor_si128(RotateLeft32<1>(b), k2);314a = _mm_xor_si128(_mm_sub_epi32(t1, t2), counter);315316counter = _mm_sub_epi32(counter, decrement);317}318319// [A1 B1 C1 D1][A2 B2 C2 D2] ... => [A1 A2 A3 A4][B1 B2 B3 B4] ...320block0 = RepackXMM<0>(a,b,c,d);321}322323inline void CHAM128_Enc_4_Blocks(__m128i &block0, __m128i &block1,324__m128i &block2, __m128i &block3, const word32 *subkeys, unsigned int rounds)325{326// Rearrange the data for vectorization. UnpackXMM includes a327// little-endian swap for SSE. Thanks to Peter Cordes for help328// with packing and unpacking.329// [A1 A2 A3 A4][B1 B2 B3 B4] ... => [A1 B1 C1 D1][A2 B2 C2 D2] ...330__m128i a = UnpackXMM<0>(block0, block1, block2, block3);331__m128i b = UnpackXMM<1>(block0, block1, block2, block3);332__m128i c = UnpackXMM<2>(block0, block1, block2, block3);333__m128i d = UnpackXMM<3>(block0, block1, block2, block3);334335__m128i counter = _mm_set_epi32(0,0,0,0);336__m128i increment = _mm_set_epi32(1,1,1,1);337338const unsigned int MASK = (rounds == 80 ? 7 : 15);339for (int i=0; i<static_cast<int>(rounds); i+=4)340{341__m128i k, k1, k2, t1, t2;342k = _mm_castpd_si128(_mm_load_sd(CONST_DOUBLE_CAST(&subkeys[(i+0) & MASK])));343344// Shuffle out two subkeys345k1 = _mm_shuffle_epi8(k, _mm_set_epi8(3,2,1,0, 3,2,1,0, 3,2,1,0, 3,2,1,0));346k2 = _mm_shuffle_epi8(k, _mm_set_epi8(7,6,5,4, 7,6,5,4, 7,6,5,4, 7,6,5,4));347348t1 = _mm_xor_si128(a, counter);349t2 = _mm_xor_si128(RotateLeft32<1>(b), k1);350a = RotateLeft32<8>(_mm_add_epi32(t1, t2));351352counter = _mm_add_epi32(counter, increment);353354t1 = _mm_xor_si128(b, counter);355t2 = _mm_xor_si128(RotateLeft32<8>(c), k2);356b = RotateLeft32<1>(_mm_add_epi32(t1, t2));357358counter = _mm_add_epi32(counter, increment);359k = _mm_castpd_si128(_mm_load_sd(CONST_DOUBLE_CAST(&subkeys[(i+2) & MASK])));360361// Shuffle out two subkeys362k1 = _mm_shuffle_epi8(k, _mm_set_epi8(3,2,1,0, 3,2,1,0, 3,2,1,0, 3,2,1,0));363k2 = _mm_shuffle_epi8(k, _mm_set_epi8(7,6,5,4, 7,6,5,4, 7,6,5,4, 7,6,5,4));364365t1 = _mm_xor_si128(c, counter);366t2 = _mm_xor_si128(RotateLeft32<1>(d), k1);367c = RotateLeft32<8>(_mm_add_epi32(t1, t2));368369counter = _mm_add_epi32(counter, increment);370371t1 = _mm_xor_si128(d, counter);372t2 = _mm_xor_si128(RotateLeft32<8>(a), k2);373d = RotateLeft32<1>(_mm_add_epi32(t1, t2));374375counter = _mm_add_epi32(counter, increment);376}377378// [A1 B1 C1 D1][A2 B2 C2 D2] ... => [A1 A2 A3 A4][B1 B2 B3 B4] ...379block0 = RepackXMM<0>(a,b,c,d);380block1 = RepackXMM<1>(a,b,c,d);381block2 = RepackXMM<2>(a,b,c,d);382block3 = RepackXMM<3>(a,b,c,d);383}384385inline void CHAM128_Dec_4_Blocks(__m128i &block0, __m128i &block1,386__m128i &block2, __m128i &block3, const word32 *subkeys, unsigned int rounds)387{388// Rearrange the data for vectorization. UnpackXMM includes a389// little-endian swap for SSE. Thanks to Peter Cordes for help390// with packing and unpacking.391// [A1 A2 A3 A4][B1 B2 B3 B4] ... => [A1 B1 C1 D1][A2 B2 C2 D2] ...392__m128i a = UnpackXMM<0>(block0, block1, block2, block3);393__m128i b = UnpackXMM<1>(block0, block1, block2, block3);394__m128i c = UnpackXMM<2>(block0, block1, block2, block3);395__m128i d = UnpackXMM<3>(block0, block1, block2, block3);396397__m128i counter = _mm_set_epi32(rounds-1,rounds-1,rounds-1,rounds-1);398__m128i decrement = _mm_set_epi32(1,1,1,1);399400const unsigned int MASK = (rounds == 80 ? 7 : 15);401for (int i = static_cast<int>(rounds)-1; i >= 0; i-=4)402{403__m128i k, k1, k2, t1, t2;404k = _mm_castpd_si128(_mm_load_sd(CONST_DOUBLE_CAST(&subkeys[(i-1) & MASK])));405406// Shuffle out two subkeys407k1 = _mm_shuffle_epi8(k, _mm_set_epi8(7,6,5,4, 7,6,5,4, 7,6,5,4, 7,6,5,4));408k2 = _mm_shuffle_epi8(k, _mm_set_epi8(3,2,1,0, 3,2,1,0, 3,2,1,0, 3,2,1,0));409410// Odd round411t1 = RotateRight32<1>(d);412t2 = _mm_xor_si128(RotateLeft32<8>(a), k1);413d = _mm_xor_si128(_mm_sub_epi32(t1, t2), counter);414415counter = _mm_sub_epi32(counter, decrement);416417// Even round418t1 = RotateRight32<8>(c);419t2 = _mm_xor_si128(RotateLeft32<1>(d), k2);420c = _mm_xor_si128(_mm_sub_epi32(t1, t2), counter);421422counter = _mm_sub_epi32(counter, decrement);423k = _mm_castpd_si128(_mm_load_sd(CONST_DOUBLE_CAST(&subkeys[(i-3) & MASK])));424425// Shuffle out two subkeys426k1 = _mm_shuffle_epi8(k, _mm_set_epi8(7,6,5,4, 7,6,5,4, 7,6,5,4, 7,6,5,4));427k2 = _mm_shuffle_epi8(k, _mm_set_epi8(3,2,1,0, 3,2,1,0, 3,2,1,0, 3,2,1,0));428429// Odd round430t1 = RotateRight32<1>(b);431t2 = _mm_xor_si128(RotateLeft32<8>(c), k1);432b = _mm_xor_si128(_mm_sub_epi32(t1, t2), counter);433434counter = _mm_sub_epi32(counter, decrement);435436// Even round437t1 = RotateRight32<8>(a);438t2 = _mm_xor_si128(RotateLeft32<1>(b), k2);439a = _mm_xor_si128(_mm_sub_epi32(t1, t2), counter);440441counter = _mm_sub_epi32(counter, decrement);442}443444// [A1 B1 C1 D1][A2 B2 C2 D2] ... => [A1 A2 A3 A4][B1 B2 B3 B4] ...445block0 = RepackXMM<0>(a,b,c,d);446block1 = RepackXMM<1>(a,b,c,d);447block2 = RepackXMM<2>(a,b,c,d);448block3 = RepackXMM<3>(a,b,c,d);449}450451//////////////////////////////////////////////////////////////////////////452453NAMESPACE_END // W32454455#endif // CRYPTOPP_SSSE3_AVAILABLE456457ANONYMOUS_NAMESPACE_END458459NAMESPACE_BEGIN(CryptoPP)460461#if defined(CRYPTOPP_SSSE3_AVAILABLE)462size_t CHAM128_Enc_AdvancedProcessBlocks_SSSE3(const word32* subKeys, size_t rounds,463const byte *inBlocks, const byte *xorBlocks, byte *outBlocks, size_t length, word32 flags)464{465return AdvancedProcessBlocks128_4x1_SSE(W32::CHAM128_Enc_Block, W32::CHAM128_Enc_4_Blocks,466subKeys, rounds, inBlocks, xorBlocks, outBlocks, length, flags);467}468469size_t CHAM128_Dec_AdvancedProcessBlocks_SSSE3(const word32* subKeys, size_t rounds,470const byte *inBlocks, const byte *xorBlocks, byte *outBlocks, size_t length, word32 flags)471{472return AdvancedProcessBlocks128_4x1_SSE(W32::CHAM128_Dec_Block, W32::CHAM128_Dec_4_Blocks,473subKeys, rounds, inBlocks, xorBlocks, outBlocks, length, flags);474}475#endif // CRYPTOPP_SSSE3_AVAILABLE476477NAMESPACE_END478479480