PurpleI2P/i2pd
1
0
Fork 0
mirror of https://github.com/PurpleI2P/i2pd.git synced 2025-01-22 12:24:19 +00:00
Code Issues Releases Activity

invert expanded key schedule for decryption

Browse Source
This commit is contained in:
orignal 2014-05-08 15:11:38 -04:00
parent 0408371a84
commit 4ef971f63c
2 changed files with 54 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

42
aes.cpp
View File

@ -33,7 +33,7 @@ namespace crypto
"add $32, %%rcx \n" "add $32, %%rcx \n"
void ECBCryptoAESNI::SetKey (const uint8_t * key) void ECBCryptoAESNI::ExpandKey (const uint8_t * key)
{ {
__asm__ __asm__
( (
@ -68,11 +68,11 @@ namespace crypto
"movups %%xmm1, (%%rcx) \n" "movups %%xmm1, (%%rcx) \n"
: // output : // output
: "S" (key), "D" (m_KeySchedule) // input : "S" (key), "D" (m_KeySchedule) // input
: "%rcx" // modified : "%rcx" // clogged
); );
} }
void ECBCryptoAESNI::Encrypt (const ChipherBlock * in, ChipherBlock * out) void ECBEncryptionAESNI::Encrypt (const ChipherBlock * in, ChipherBlock * out)
{ {
__asm__ __asm__
( (
@ -92,11 +92,11 @@ namespace crypto
"aesenc 208(%%rdx), %%xmm0 \n" "aesenc 208(%%rdx), %%xmm0 \n"
"aesenclast 224(%%rdx), %%xmm0 \n" "aesenclast 224(%%rdx), %%xmm0 \n"
"movups %%xmm0, (%%rdi) \n" "movups %%xmm0, (%%rdi) \n"
: : "d" ((uint64_t)m_KeySchedule), "S" (in), "D" (out) : : "d" (m_KeySchedule), "S" (in), "D" (out)
); );
} }
void ECBCryptoAESNI::Decrypt (const ChipherBlock * in, ChipherBlock * out) void ECBDecryptionAESNI::Decrypt (const ChipherBlock * in, ChipherBlock * out)
{ {
__asm__ __asm__
( (
@ -116,10 +116,38 @@ namespace crypto
"aesdec 16(%%rdx), %%xmm0 \n" "aesdec 16(%%rdx), %%xmm0 \n"
"aesdeclast (%%rdx), %%xmm0 \n" "aesdeclast (%%rdx), %%xmm0 \n"
"movups %%xmm0, (%%rdi) \n" "movups %%xmm0, (%%rdi) \n"
: : "d" ((uint64_t)m_KeySchedule), "S" (in), "D" (out) : : "d" (m_KeySchedule), "S" (in), "D" (out)
); );
} }
#define CallAESIMC(offset) \
"movups "#offset"(%%rdx), %%xmm0 \n" \
"aesimc %%xmm0, %%xmm0 \n" \
"movups %%xmm0, "#offset"(%%rdx) \n"
void ECBDecryptionAESNI::SetKey (const uint8_t * key)
{
ExpandKey (key); // expand encryption key first
// then invert it using aesimc
__asm__
(
CallAESIMC(16)
CallAESIMC(32)
CallAESIMC(48)
CallAESIMC(64)
CallAESIMC(80)
CallAESIMC(96)
CallAESIMC(112)
CallAESIMC(128)
CallAESIMC(144)
CallAESIMC(160)
CallAESIMC(176)
CallAESIMC(192)
CallAESIMC(208)
: : "d" (m_KeySchedule)
);
}
#endif #endif

24
aes.h
View File

@ -18,16 +18,30 @@ namespace crypto
#ifdef __x86_64__ #ifdef __x86_64__
// AES-NI assumed // AES-NI assumed
class ECBCryptoAESNI class ECBCryptoAESNI
{
protected:
void ExpandKey (const uint8_t * key);
protected:
uint32_t m_KeySchedule[4*(14+1)]; // 14 rounds for AES-256
};
class ECBEncryptionAESNI: public ECBCryptoAESNI
{
public:
void SetKey (const uint8_t * key) { ExpandKey (key); };
void Encrypt (const ChipherBlock * in, ChipherBlock * out);
};
class ECBDecryptionAESNI: public ECBCryptoAESNI
{ {
public: public:
void SetKey (const uint8_t * key); void SetKey (const uint8_t * key);
void Encrypt (const ChipherBlock * in, ChipherBlock * out);
void Decrypt (const ChipherBlock * in, ChipherBlock * out); void Decrypt (const ChipherBlock * in, ChipherBlock * out);
private:
uint32_t m_KeySchedule[4*(14+1)]; // 14 rounds for AES-256
}; };
#endif #endif