Archives/i2pd
2
0
Fork 0
mirror of https://github.com/PurpleI2P/i2pd.git synced 2024-11-16 00:12:43 +00:00
Code Issues Projects Releases Wiki Activity

CBC encryption through AES-NI

Browse Source
This commit is contained in:
orignal 2014-05-12 22:51:59 -04:00
parent 3eb4cc9eed
commit e9fc2492dd
2 changed files with 67 additions and 37 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

100
aes.cpp
View File

@ -47,9 +47,9 @@ namespace crypto
( (
"movups (%[key]), %%xmm1 \n" "movups (%[key]), %%xmm1 \n"
"movups 16(%[key]), %%xmm3 \n" "movups 16(%[key]), %%xmm3 \n"
"movaps %%xmm1, (%[shed]) \n" "movaps %%xmm1, (%[sched]) \n"
"movaps %%xmm3, 16(%[shed]) \n" "movaps %%xmm3, 16(%[sched]) \n"
"lea 32(%[shed]), %%rcx \n" "lea 32(%[sched]), %%rcx \n"
"aeskeygenassist $1, %%xmm3, %%xmm2 \n" "aeskeygenassist $1, %%xmm3, %%xmm2 \n"
KeyExpansion256 KeyExpansion256
"aeskeygenassist $2, %%xmm3, %%xmm2 \n" "aeskeygenassist $2, %%xmm3, %%xmm2 \n"
@ -75,58 +75,64 @@ namespace crypto
"pxor %%xmm2, %%xmm1 \n" "pxor %%xmm2, %%xmm1 \n"
"movups %%xmm1, (%%rcx) \n" "movups %%xmm1, (%%rcx) \n"
: // output : // output
: [key]"r"(key), [shed]"r"(m_KeySchedule) // input : [key]"r"(key), [sched]"r"(m_KeySchedule) // input
: "%rcx", "%xmm1", "%xmm2", "%xmm3", "%xmm4" // clogged : "%rcx", "%xmm1", "%xmm2", "%xmm3", "%xmm4" // clogged
); );
} }
#define EncryptAES256 \
"pxor (%[sched]), %%xmm0 \n" \
"aesenc 16(%[sched]), %%xmm0 \n" \
"aesenc 32(%[sched]), %%xmm0 \n" \
"aesenc 48(%[sched]), %%xmm0 \n" \
"aesenc 64(%[sched]), %%xmm0 \n" \
"aesenc 80(%[sched]), %%xmm0 \n" \
"aesenc 96(%[sched]), %%xmm0 \n" \
"aesenc 112(%[sched]), %%xmm0 \n" \
"aesenc 128(%[sched]), %%xmm0 \n" \
"aesenc 144(%[sched]), %%xmm0 \n" \
"aesenc 160(%[sched]), %%xmm0 \n" \
"aesenc 176(%[sched]), %%xmm0 \n" \
"aesenc 192(%[sched]), %%xmm0 \n" \
"aesenc 208(%[sched]), %%xmm0 \n" \
"aesenclast 224(%[sched]), %%xmm0 \n"
void ECBEncryptionAESNI::Encrypt (const ChipherBlock * in, ChipherBlock * out) void ECBEncryptionAESNI::Encrypt (const ChipherBlock * in, ChipherBlock * out)
{ {
__asm__ __asm__
( (
"movups (%[in]), %%xmm0 \n" "movups (%[in]), %%xmm0 \n"
"pxor (%[shed]), %%xmm0 \n" EncryptAES256
"aesenc 16(%[shed]), %%xmm0 \n"
"aesenc 32(%[shed]), %%xmm0 \n"
"aesenc 48(%[shed]), %%xmm0 \n"
"aesenc 64(%[shed]), %%xmm0 \n"
"aesenc 80(%[shed]), %%xmm0 \n"
"aesenc 96(%[shed]), %%xmm0 \n"
"aesenc 112(%[shed]), %%xmm0 \n"
"aesenc 128(%[shed]), %%xmm0 \n"
"aesenc 144(%[shed]), %%xmm0 \n"
"aesenc 160(%[shed]), %%xmm0 \n"
"aesenc 176(%[shed]), %%xmm0 \n"
"aesenc 192(%[shed]), %%xmm0 \n"
"aesenc 208(%[shed]), %%xmm0 \n"
"aesenclast 224(%[shed]), %%xmm0 \n"
"movups %%xmm0, (%[out]) \n" "movups %%xmm0, (%[out]) \n"
: : [shed]"r"(m_KeySchedule), [in]"r"(in), [out]"r"(out) : "%xmm0" : : [sched]"r"(m_KeySchedule), [in]"r"(in), [out]"r"(out) : "%xmm0"
); );
} }
#define DecryptAES256 \
"pxor 224(%[sched]), %%xmm0 \n" \
"aesdec 208(%[sched]), %%xmm0 \n" \
"aesdec 192(%[sched]), %%xmm0 \n" \
"aesdec 176(%[sched]), %%xmm0 \n" \
"aesdec 160(%[sched]), %%xmm0 \n" \
"aesdec 144(%[sched]), %%xmm0 \n" \
"aesdec 128(%[sched]), %%xmm0 \n" \
"aesdec 112(%[sched]), %%xmm0 \n" \
"aesdec 96(%[sched]), %%xmm0 \n" \
"aesdec 80(%[sched]), %%xmm0 \n" \
"aesdec 64(%[sched]), %%xmm0 \n" \
"aesdec 48(%[sched]), %%xmm0 \n" \
"aesdec 32(%[sched]), %%xmm0 \n" \
"aesdec 16(%[sched]), %%xmm0 \n" \
"aesdeclast (%[sched]), %%xmm0 \n"
void ECBDecryptionAESNI::Decrypt (const ChipherBlock * in, ChipherBlock * out) void ECBDecryptionAESNI::Decrypt (const ChipherBlock * in, ChipherBlock * out)
{ {
__asm__ __asm__
( (
"movups (%[in]), %%xmm0 \n" "movups (%[in]), %%xmm0 \n"
"pxor 224(%[shed]), %%xmm0 \n" DecryptAES256
"aesdec 208(%[shed]), %%xmm0 \n"
"aesdec 192(%[shed]), %%xmm0 \n"
"aesdec 176(%[shed]), %%xmm0 \n"
"aesdec 160(%[shed]), %%xmm0 \n"
"aesdec 144(%[shed]), %%xmm0 \n"
"aesdec 128(%[shed]), %%xmm0 \n"
"aesdec 112(%[shed]), %%xmm0 \n"
"aesdec 96(%[shed]), %%xmm0 \n"
"aesdec 80(%[shed]), %%xmm0 \n"
"aesdec 64(%[shed]), %%xmm0 \n"
"aesdec 48(%[shed]), %%xmm0 \n"
"aesdec 32(%[shed]), %%xmm0 \n"
"aesdec 16(%[shed]), %%xmm0 \n"
"aesdeclast (%[shed]), %%xmm0 \n"
"movups %%xmm0, (%[out]) \n" "movups %%xmm0, (%[out]) \n"
: : [shed]"r"(m_KeySchedule), [in]"r"(in), [out]"r"(out) : "%xmm0" : : [sched]"r"(m_KeySchedule), [in]"r"(in), [out]"r"(out) : "%xmm0"
); );
} }
@ -163,12 +169,34 @@ namespace crypto
void CBCEncryption::Encrypt (int numBlocks, const ChipherBlock * in, ChipherBlock * out) void CBCEncryption::Encrypt (int numBlocks, const ChipherBlock * in, ChipherBlock * out)
{ {
#ifdef __x86_64__
__asm__
(
"movups (%[iv]), %%xmm1 \n"
"block: \n"
"movups (%[in]), %%xmm0 \n"
"pxor %%xmm1, %%xmm0 \n"
EncryptAES256
"movaps %%xmm0, %%xmm1 \n"
"movups %%xmm0, (%[out]) \n"
"add $16, %[in] \n"
"add $16, %[out] \n"
"dec %[num] \n"
"jnz block; \n"
"movups %%xmm1, (%[iv]) \n"
:
: [iv]"r"(&m_LastBlock), [sched]"r"(m_ECBEncryption.GetKeySchedule ()),
[in]"r"(in), [out]"r"(out), [num]"r"(numBlocks)
: "%xmm0", "%xmm1", "cc", "memory"
);
#else
for (int i = 0; i < numBlocks; i++) for (int i = 0; i < numBlocks; i++)
{ {
m_LastBlock ^= in[i]; m_LastBlock ^= in[i];
m_ECBEncryption.Encrypt (&m_LastBlock, &m_LastBlock); m_ECBEncryption.Encrypt (&m_LastBlock, &m_LastBlock);
out[i] = m_LastBlock; out[i] = m_LastBlock;
} }
#endif
} }
bool CBCEncryption::Encrypt (const uint8_t * in, std::size_t len, uint8_t * out) bool CBCEncryption::Encrypt (const uint8_t * in, std::size_t len, uint8_t * out)

2
aes.h
View File

@ -28,6 +28,7 @@ namespace crypto
public: public:
ECBCryptoAESNI (); ECBCryptoAESNI ();
uint8_t * GetKeySchedule () { return m_KeySchedule; };
protected: protected:
@ -114,6 +115,7 @@ namespace crypto
private: private:
ChipherBlock m_LastBlock; ChipherBlock m_LastBlock;
ECBEncryption m_ECBEncryption; ECBEncryption m_ECBEncryption;
}; };