mirror of
https://github.com/PurpleI2P/i2pd.git
synced 2025-02-08 22:13:48 +01:00
Split tunnel-specific crypto from aes.h/cpp.
This commit is contained in:
parent
d9bb09780f
commit
9597917183
9 changed files with 410 additions and 385 deletions
66
AESNIMacros.h
Normal file
66
AESNIMacros.h
Normal file
|
@ -0,0 +1,66 @@
|
||||||
|
#ifndef AESNIMACROS_H__
|
||||||
|
#define AESNIMACROS_H__
|
||||||
|
|
||||||
|
#define KeyExpansion256(round0,round1) \
|
||||||
|
"pshufd $0xff, %%xmm2, %%xmm2 \n" \
|
||||||
|
"movaps %%xmm1, %%xmm4 \n" \
|
||||||
|
"pslldq $4, %%xmm4 \n" \
|
||||||
|
"pxor %%xmm4, %%xmm1 \n" \
|
||||||
|
"pslldq $4, %%xmm4 \n" \
|
||||||
|
"pxor %%xmm4, %%xmm1 \n" \
|
||||||
|
"pslldq $4, %%xmm4 \n" \
|
||||||
|
"pxor %%xmm4, %%xmm1 \n" \
|
||||||
|
"pxor %%xmm2, %%xmm1 \n" \
|
||||||
|
"movaps %%xmm1, "#round0"(%[sched]) \n" \
|
||||||
|
"aeskeygenassist $0, %%xmm1, %%xmm4 \n" \
|
||||||
|
"pshufd $0xaa, %%xmm4, %%xmm2 \n" \
|
||||||
|
"movaps %%xmm3, %%xmm4 \n" \
|
||||||
|
"pslldq $4, %%xmm4 \n" \
|
||||||
|
"pxor %%xmm4, %%xmm3 \n" \
|
||||||
|
"pslldq $4, %%xmm4 \n" \
|
||||||
|
"pxor %%xmm4, %%xmm3 \n" \
|
||||||
|
"pslldq $4, %%xmm4 \n" \
|
||||||
|
"pxor %%xmm4, %%xmm3 \n" \
|
||||||
|
"pxor %%xmm2, %%xmm3 \n" \
|
||||||
|
"movaps %%xmm3, "#round1"(%[sched]) \n"
|
||||||
|
|
||||||
|
#define EncryptAES256(sched) \
|
||||||
|
"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"
|
||||||
|
|
||||||
|
#define DecryptAES256(sched) \
|
||||||
|
"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"
|
||||||
|
|
||||||
|
#define CallAESIMC(offset) \
|
||||||
|
"movaps "#offset"(%[shed]), %%xmm0 \n" \
|
||||||
|
"aesimc %%xmm0, %%xmm0 \n" \
|
||||||
|
"movaps %%xmm0, "#offset"(%[shed]) \n"
|
||||||
|
|
||||||
|
#endif
|
|
@ -31,6 +31,7 @@ set (COMMON_SRC
|
||||||
"${CMAKE_SOURCE_DIR}/tunnel/TunnelGateway.cpp"
|
"${CMAKE_SOURCE_DIR}/tunnel/TunnelGateway.cpp"
|
||||||
"${CMAKE_SOURCE_DIR}/tunnel/TunnelEndpoint.cpp"
|
"${CMAKE_SOURCE_DIR}/tunnel/TunnelEndpoint.cpp"
|
||||||
"${CMAKE_SOURCE_DIR}/tunnel/TunnelPool.cpp"
|
"${CMAKE_SOURCE_DIR}/tunnel/TunnelPool.cpp"
|
||||||
|
"${CMAKE_SOURCE_DIR}/tunnel/TunnelCrypto.cpp"
|
||||||
"${CMAKE_SOURCE_DIR}/AddressBook.cpp"
|
"${CMAKE_SOURCE_DIR}/AddressBook.cpp"
|
||||||
"${CMAKE_SOURCE_DIR}/Garlic.cpp"
|
"${CMAKE_SOURCE_DIR}/Garlic.cpp"
|
||||||
"${CMAKE_SOURCE_DIR}/I2NPProtocol.cpp"
|
"${CMAKE_SOURCE_DIR}/I2NPProtocol.cpp"
|
||||||
|
|
567
crypto/aes.cpp
567
crypto/aes.cpp
|
@ -1,357 +1,222 @@
|
||||||
#include <stdlib.h>
|
#include <stdlib.h>
|
||||||
#include "tunnel/TunnelBase.h"
|
|
||||||
#include "aes.h"
|
#include "aes.h"
|
||||||
|
|
||||||
namespace i2p
|
namespace i2p {
|
||||||
|
namespace crypto {
|
||||||
|
|
||||||
|
#ifdef AESNI
|
||||||
|
#include "AESNIMacros.h"
|
||||||
|
|
||||||
|
void ECBCryptoAESNI::ExpandKey (const AESKey& key)
|
||||||
{
|
{
|
||||||
namespace crypto
|
__asm__
|
||||||
{
|
(
|
||||||
|
"movups (%[key]), %%xmm1 \n"
|
||||||
#ifdef AESNI
|
"movups 16(%[key]), %%xmm3 \n"
|
||||||
|
"movaps %%xmm1, (%[sched]) \n"
|
||||||
#define KeyExpansion256(round0,round1) \
|
"movaps %%xmm3, 16(%[sched]) \n"
|
||||||
"pshufd $0xff, %%xmm2, %%xmm2 \n" \
|
"aeskeygenassist $1, %%xmm3, %%xmm2 \n"
|
||||||
"movaps %%xmm1, %%xmm4 \n" \
|
KeyExpansion256(32,48)
|
||||||
"pslldq $4, %%xmm4 \n" \
|
"aeskeygenassist $2, %%xmm3, %%xmm2 \n"
|
||||||
"pxor %%xmm4, %%xmm1 \n" \
|
KeyExpansion256(64,80)
|
||||||
"pslldq $4, %%xmm4 \n" \
|
"aeskeygenassist $4, %%xmm3, %%xmm2 \n"
|
||||||
"pxor %%xmm4, %%xmm1 \n" \
|
KeyExpansion256(96,112)
|
||||||
"pslldq $4, %%xmm4 \n" \
|
"aeskeygenassist $8, %%xmm3, %%xmm2 \n"
|
||||||
"pxor %%xmm4, %%xmm1 \n" \
|
KeyExpansion256(128,144)
|
||||||
"pxor %%xmm2, %%xmm1 \n" \
|
"aeskeygenassist $16, %%xmm3, %%xmm2 \n"
|
||||||
"movaps %%xmm1, "#round0"(%[sched]) \n" \
|
KeyExpansion256(160,176)
|
||||||
"aeskeygenassist $0, %%xmm1, %%xmm4 \n" \
|
"aeskeygenassist $32, %%xmm3, %%xmm2 \n"
|
||||||
"pshufd $0xaa, %%xmm4, %%xmm2 \n" \
|
KeyExpansion256(192,208)
|
||||||
"movaps %%xmm3, %%xmm4 \n" \
|
"aeskeygenassist $64, %%xmm3, %%xmm2 \n"
|
||||||
"pslldq $4, %%xmm4 \n" \
|
// key expansion final
|
||||||
"pxor %%xmm4, %%xmm3 \n" \
|
"pshufd $0xff, %%xmm2, %%xmm2 \n"
|
||||||
"pslldq $4, %%xmm4 \n" \
|
"movaps %%xmm1, %%xmm4 \n"
|
||||||
"pxor %%xmm4, %%xmm3 \n" \
|
"pslldq $4, %%xmm4 \n"
|
||||||
"pslldq $4, %%xmm4 \n" \
|
"pxor %%xmm4, %%xmm1 \n"
|
||||||
"pxor %%xmm4, %%xmm3 \n" \
|
"pslldq $4, %%xmm4 \n"
|
||||||
"pxor %%xmm2, %%xmm3 \n" \
|
"pxor %%xmm4, %%xmm1 \n"
|
||||||
"movaps %%xmm3, "#round1"(%[sched]) \n"
|
"pslldq $4, %%xmm4 \n"
|
||||||
|
"pxor %%xmm4, %%xmm1 \n"
|
||||||
void ECBCryptoAESNI::ExpandKey (const AESKey& key)
|
"pxor %%xmm2, %%xmm1 \n"
|
||||||
{
|
"movups %%xmm1, 224(%[sched]) \n"
|
||||||
__asm__
|
: // output
|
||||||
(
|
: [key]"r"((const uint8_t *)key), [sched]"r"(GetKeySchedule ()) // input
|
||||||
"movups (%[key]), %%xmm1 \n"
|
: "%xmm1", "%xmm2", "%xmm3", "%xmm4", "memory" // clogged
|
||||||
"movups 16(%[key]), %%xmm3 \n"
|
);
|
||||||
"movaps %%xmm1, (%[sched]) \n"
|
|
||||||
"movaps %%xmm3, 16(%[sched]) \n"
|
|
||||||
"aeskeygenassist $1, %%xmm3, %%xmm2 \n"
|
|
||||||
KeyExpansion256(32,48)
|
|
||||||
"aeskeygenassist $2, %%xmm3, %%xmm2 \n"
|
|
||||||
KeyExpansion256(64,80)
|
|
||||||
"aeskeygenassist $4, %%xmm3, %%xmm2 \n"
|
|
||||||
KeyExpansion256(96,112)
|
|
||||||
"aeskeygenassist $8, %%xmm3, %%xmm2 \n"
|
|
||||||
KeyExpansion256(128,144)
|
|
||||||
"aeskeygenassist $16, %%xmm3, %%xmm2 \n"
|
|
||||||
KeyExpansion256(160,176)
|
|
||||||
"aeskeygenassist $32, %%xmm3, %%xmm2 \n"
|
|
||||||
KeyExpansion256(192,208)
|
|
||||||
"aeskeygenassist $64, %%xmm3, %%xmm2 \n"
|
|
||||||
// key expansion final
|
|
||||||
"pshufd $0xff, %%xmm2, %%xmm2 \n"
|
|
||||||
"movaps %%xmm1, %%xmm4 \n"
|
|
||||||
"pslldq $4, %%xmm4 \n"
|
|
||||||
"pxor %%xmm4, %%xmm1 \n"
|
|
||||||
"pslldq $4, %%xmm4 \n"
|
|
||||||
"pxor %%xmm4, %%xmm1 \n"
|
|
||||||
"pslldq $4, %%xmm4 \n"
|
|
||||||
"pxor %%xmm4, %%xmm1 \n"
|
|
||||||
"pxor %%xmm2, %%xmm1 \n"
|
|
||||||
"movups %%xmm1, 224(%[sched]) \n"
|
|
||||||
: // output
|
|
||||||
: [key]"r"((const uint8_t *)key), [sched]"r"(GetKeySchedule ()) // input
|
|
||||||
: "%xmm1", "%xmm2", "%xmm3", "%xmm4", "memory" // clogged
|
|
||||||
);
|
|
||||||
}
|
|
||||||
|
|
||||||
#define EncryptAES256(sched) \
|
|
||||||
"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)
|
|
||||||
{
|
|
||||||
__asm__
|
|
||||||
(
|
|
||||||
"movups (%[in]), %%xmm0 \n"
|
|
||||||
EncryptAES256(sched)
|
|
||||||
"movups %%xmm0, (%[out]) \n"
|
|
||||||
: : [sched]"r"(GetKeySchedule ()), [in]"r"(in), [out]"r"(out) : "%xmm0", "memory"
|
|
||||||
);
|
|
||||||
}
|
|
||||||
|
|
||||||
#define DecryptAES256(sched) \
|
|
||||||
"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)
|
|
||||||
{
|
|
||||||
__asm__
|
|
||||||
(
|
|
||||||
"movups (%[in]), %%xmm0 \n"
|
|
||||||
DecryptAES256(sched)
|
|
||||||
"movups %%xmm0, (%[out]) \n"
|
|
||||||
: : [sched]"r"(GetKeySchedule ()), [in]"r"(in), [out]"r"(out) : "%xmm0", "memory"
|
|
||||||
);
|
|
||||||
}
|
|
||||||
|
|
||||||
#define CallAESIMC(offset) \
|
|
||||||
"movaps "#offset"(%[shed]), %%xmm0 \n" \
|
|
||||||
"aesimc %%xmm0, %%xmm0 \n" \
|
|
||||||
"movaps %%xmm0, "#offset"(%[shed]) \n"
|
|
||||||
|
|
||||||
void ECBDecryptionAESNI::SetKey (const AESKey& 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)
|
|
||||||
: : [shed]"r"(GetKeySchedule ()) : "%xmm0", "memory"
|
|
||||||
);
|
|
||||||
}
|
|
||||||
|
|
||||||
#endif
|
|
||||||
|
|
||||||
|
|
||||||
void CBCEncryption::Encrypt (int numBlocks, const ChipherBlock * in, ChipherBlock * out)
|
|
||||||
{
|
|
||||||
#ifdef AESNI
|
|
||||||
__asm__
|
|
||||||
(
|
|
||||||
"movups (%[iv]), %%xmm1 \n"
|
|
||||||
"1: \n"
|
|
||||||
"movups (%[in]), %%xmm0 \n"
|
|
||||||
"pxor %%xmm1, %%xmm0 \n"
|
|
||||||
EncryptAES256(sched)
|
|
||||||
"movaps %%xmm0, %%xmm1 \n"
|
|
||||||
"movups %%xmm0, (%[out]) \n"
|
|
||||||
"add $16, %[in] \n"
|
|
||||||
"add $16, %[out] \n"
|
|
||||||
"dec %[num] \n"
|
|
||||||
"jnz 1b \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++)
|
|
||||||
{
|
|
||||||
m_LastBlock ^= in[i];
|
|
||||||
m_ECBEncryption.Encrypt (&m_LastBlock, &m_LastBlock);
|
|
||||||
out[i] = m_LastBlock;
|
|
||||||
}
|
|
||||||
#endif
|
|
||||||
}
|
|
||||||
|
|
||||||
void CBCEncryption::Encrypt (const uint8_t * in, std::size_t len, uint8_t * out)
|
|
||||||
{
|
|
||||||
// len/16
|
|
||||||
int numBlocks = len >> 4;
|
|
||||||
if (numBlocks > 0)
|
|
||||||
Encrypt (numBlocks, (const ChipherBlock *)in, (ChipherBlock *)out);
|
|
||||||
}
|
|
||||||
|
|
||||||
void CBCEncryption::Encrypt (const uint8_t * in, uint8_t * out)
|
|
||||||
{
|
|
||||||
#ifdef AESNI
|
|
||||||
__asm__
|
|
||||||
(
|
|
||||||
"movups (%[iv]), %%xmm1 \n"
|
|
||||||
"movups (%[in]), %%xmm0 \n"
|
|
||||||
"pxor %%xmm1, %%xmm0 \n"
|
|
||||||
EncryptAES256(sched)
|
|
||||||
"movups %%xmm0, (%[out]) \n"
|
|
||||||
"movups %%xmm0, (%[iv]) \n"
|
|
||||||
:
|
|
||||||
: [iv]"r"(&m_LastBlock), [sched]"r"(m_ECBEncryption.GetKeySchedule ()),
|
|
||||||
[in]"r"(in), [out]"r"(out)
|
|
||||||
: "%xmm0", "%xmm1", "memory"
|
|
||||||
);
|
|
||||||
#else
|
|
||||||
Encrypt (1, (const ChipherBlock *)in, (ChipherBlock *)out);
|
|
||||||
#endif
|
|
||||||
}
|
|
||||||
|
|
||||||
void CBCDecryption::Decrypt (int numBlocks, const ChipherBlock * in, ChipherBlock * out)
|
|
||||||
{
|
|
||||||
#ifdef AESNI
|
|
||||||
__asm__
|
|
||||||
(
|
|
||||||
"movups (%[iv]), %%xmm1 \n"
|
|
||||||
"1: \n"
|
|
||||||
"movups (%[in]), %%xmm0 \n"
|
|
||||||
"movaps %%xmm0, %%xmm2 \n"
|
|
||||||
DecryptAES256(sched)
|
|
||||||
"pxor %%xmm1, %%xmm0 \n"
|
|
||||||
"movups %%xmm0, (%[out]) \n"
|
|
||||||
"movaps %%xmm2, %%xmm1 \n"
|
|
||||||
"add $16, %[in] \n"
|
|
||||||
"add $16, %[out] \n"
|
|
||||||
"dec %[num] \n"
|
|
||||||
"jnz 1b \n"
|
|
||||||
"movups %%xmm1, (%[iv]) \n"
|
|
||||||
:
|
|
||||||
: [iv]"r"(&m_IV), [sched]"r"(m_ECBDecryption.GetKeySchedule ()),
|
|
||||||
[in]"r"(in), [out]"r"(out), [num]"r"(numBlocks)
|
|
||||||
: "%xmm0", "%xmm1", "%xmm2", "cc", "memory"
|
|
||||||
);
|
|
||||||
#else
|
|
||||||
for (int i = 0; i < numBlocks; i++)
|
|
||||||
{
|
|
||||||
ChipherBlock tmp = in[i];
|
|
||||||
m_ECBDecryption.Decrypt (in + i, out + i);
|
|
||||||
out[i] ^= m_IV;
|
|
||||||
m_IV = tmp;
|
|
||||||
}
|
|
||||||
#endif
|
|
||||||
}
|
|
||||||
|
|
||||||
void CBCDecryption::Decrypt (const uint8_t * in, std::size_t len, uint8_t * out)
|
|
||||||
{
|
|
||||||
int numBlocks = len >> 4;
|
|
||||||
if (numBlocks > 0)
|
|
||||||
Decrypt (numBlocks, (const ChipherBlock *)in, (ChipherBlock *)out);
|
|
||||||
}
|
|
||||||
|
|
||||||
void CBCDecryption::Decrypt (const uint8_t * in, uint8_t * out)
|
|
||||||
{
|
|
||||||
#ifdef AESNI
|
|
||||||
__asm__
|
|
||||||
(
|
|
||||||
"movups (%[iv]), %%xmm1 \n"
|
|
||||||
"movups (%[in]), %%xmm0 \n"
|
|
||||||
"movups %%xmm0, (%[iv]) \n"
|
|
||||||
DecryptAES256(sched)
|
|
||||||
"pxor %%xmm1, %%xmm0 \n"
|
|
||||||
"movups %%xmm0, (%[out]) \n"
|
|
||||||
:
|
|
||||||
: [iv]"r"(&m_IV), [sched]"r"(m_ECBDecryption.GetKeySchedule ()),
|
|
||||||
[in]"r"(in), [out]"r"(out)
|
|
||||||
: "%xmm0", "%xmm1", "memory"
|
|
||||||
);
|
|
||||||
#else
|
|
||||||
Decrypt (1, (const ChipherBlock *)in, (ChipherBlock *)out);
|
|
||||||
#endif
|
|
||||||
}
|
|
||||||
|
|
||||||
void TunnelEncryption::Encrypt (const uint8_t * in, uint8_t * out)
|
|
||||||
{
|
|
||||||
#ifdef AESNI
|
|
||||||
__asm__
|
|
||||||
(
|
|
||||||
// encrypt IV
|
|
||||||
"movups (%[in]), %%xmm0 \n"
|
|
||||||
EncryptAES256(sched_iv)
|
|
||||||
"movaps %%xmm0, %%xmm1 \n"
|
|
||||||
// double IV encryption
|
|
||||||
EncryptAES256(sched_iv)
|
|
||||||
"movups %%xmm0, (%[out]) \n"
|
|
||||||
// encrypt data, IV is xmm1
|
|
||||||
"1: \n"
|
|
||||||
"add $16, %[in] \n"
|
|
||||||
"add $16, %[out] \n"
|
|
||||||
"movups (%[in]), %%xmm0 \n"
|
|
||||||
"pxor %%xmm1, %%xmm0 \n"
|
|
||||||
EncryptAES256(sched_l)
|
|
||||||
"movaps %%xmm0, %%xmm1 \n"
|
|
||||||
"movups %%xmm0, (%[out]) \n"
|
|
||||||
"dec %[num] \n"
|
|
||||||
"jnz 1b \n"
|
|
||||||
:
|
|
||||||
: [sched_iv]"r"(m_IVEncryption.GetKeySchedule ()), [sched_l]"r"(m_LayerEncryption.GetKeySchedule ()),
|
|
||||||
[in]"r"(in), [out]"r"(out), [num]"r"(63) // 63 blocks = 1008 bytes
|
|
||||||
: "%xmm0", "%xmm1", "cc", "memory"
|
|
||||||
);
|
|
||||||
#else
|
|
||||||
m_IVEncryption.Encrypt ((const ChipherBlock *)in, (ChipherBlock *)out); // iv
|
|
||||||
m_LayerEncryption.SetIV (out);
|
|
||||||
m_LayerEncryption.Encrypt (in + 16, i2p::tunnel::TUNNEL_DATA_ENCRYPTED_SIZE, out + 16); // data
|
|
||||||
m_IVEncryption.Encrypt ((ChipherBlock *)out, (ChipherBlock *)out); // double iv
|
|
||||||
#endif
|
|
||||||
}
|
|
||||||
|
|
||||||
void TunnelDecryption::Decrypt (const uint8_t * in, uint8_t * out)
|
|
||||||
{
|
|
||||||
#ifdef AESNI
|
|
||||||
__asm__
|
|
||||||
(
|
|
||||||
// decrypt IV
|
|
||||||
"movups (%[in]), %%xmm0 \n"
|
|
||||||
DecryptAES256(sched_iv)
|
|
||||||
"movaps %%xmm0, %%xmm1 \n"
|
|
||||||
// double IV encryption
|
|
||||||
DecryptAES256(sched_iv)
|
|
||||||
"movups %%xmm0, (%[out]) \n"
|
|
||||||
// decrypt data, IV is xmm1
|
|
||||||
"1: \n"
|
|
||||||
"add $16, %[in] \n"
|
|
||||||
"add $16, %[out] \n"
|
|
||||||
"movups (%[in]), %%xmm0 \n"
|
|
||||||
"movaps %%xmm0, %%xmm2 \n"
|
|
||||||
DecryptAES256(sched_l)
|
|
||||||
"pxor %%xmm1, %%xmm0 \n"
|
|
||||||
"movups %%xmm0, (%[out]) \n"
|
|
||||||
"movaps %%xmm2, %%xmm1 \n"
|
|
||||||
"dec %[num] \n"
|
|
||||||
"jnz 1b \n"
|
|
||||||
:
|
|
||||||
: [sched_iv]"r"(m_IVDecryption.GetKeySchedule ()), [sched_l]"r"(m_LayerDecryption.GetKeySchedule ()),
|
|
||||||
[in]"r"(in), [out]"r"(out), [num]"r"(63) // 63 blocks = 1008 bytes
|
|
||||||
: "%xmm0", "%xmm1", "%xmm2", "cc", "memory"
|
|
||||||
);
|
|
||||||
#else
|
|
||||||
m_IVDecryption.Decrypt ((const ChipherBlock *)in, (ChipherBlock *)out); // iv
|
|
||||||
m_LayerDecryption.SetIV (out);
|
|
||||||
m_LayerDecryption.Decrypt (in + 16, i2p::tunnel::TUNNEL_DATA_ENCRYPTED_SIZE, out + 16); // data
|
|
||||||
m_IVDecryption.Decrypt ((ChipherBlock *)out, (ChipherBlock *)out); // double iv
|
|
||||||
#endif
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
void ECBEncryptionAESNI::Encrypt (const ChipherBlock * in, ChipherBlock * out)
|
||||||
|
{
|
||||||
|
__asm__
|
||||||
|
(
|
||||||
|
"movups (%[in]), %%xmm0 \n"
|
||||||
|
EncryptAES256(sched)
|
||||||
|
"movups %%xmm0, (%[out]) \n"
|
||||||
|
: : [sched]"r"(GetKeySchedule ()), [in]"r"(in), [out]"r"(out) : "%xmm0", "memory"
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
void ECBDecryptionAESNI::Decrypt (const ChipherBlock * in, ChipherBlock * out)
|
||||||
|
{
|
||||||
|
__asm__
|
||||||
|
(
|
||||||
|
"movups (%[in]), %%xmm0 \n"
|
||||||
|
DecryptAES256(sched)
|
||||||
|
"movups %%xmm0, (%[out]) \n"
|
||||||
|
: : [sched]"r"(GetKeySchedule ()), [in]"r"(in), [out]"r"(out) : "%xmm0", "memory"
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
void ECBDecryptionAESNI::SetKey (const AESKey& 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)
|
||||||
|
: : [shed]"r"(GetKeySchedule ()) : "%xmm0", "memory"
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
#endif
|
||||||
|
|
||||||
|
|
||||||
|
void CBCEncryption::Encrypt (int numBlocks, const ChipherBlock * in, ChipherBlock * out)
|
||||||
|
{
|
||||||
|
#ifdef AESNI
|
||||||
|
__asm__
|
||||||
|
(
|
||||||
|
"movups (%[iv]), %%xmm1 \n"
|
||||||
|
"1: \n"
|
||||||
|
"movups (%[in]), %%xmm0 \n"
|
||||||
|
"pxor %%xmm1, %%xmm0 \n"
|
||||||
|
EncryptAES256(sched)
|
||||||
|
"movaps %%xmm0, %%xmm1 \n"
|
||||||
|
"movups %%xmm0, (%[out]) \n"
|
||||||
|
"add $16, %[in] \n"
|
||||||
|
"add $16, %[out] \n"
|
||||||
|
"dec %[num] \n"
|
||||||
|
"jnz 1b \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++)
|
||||||
|
{
|
||||||
|
m_LastBlock ^= in[i];
|
||||||
|
m_ECBEncryption.Encrypt (&m_LastBlock, &m_LastBlock);
|
||||||
|
out[i] = m_LastBlock;
|
||||||
|
}
|
||||||
|
#endif
|
||||||
|
}
|
||||||
|
|
||||||
|
void CBCEncryption::Encrypt (const uint8_t * in, std::size_t len, uint8_t * out)
|
||||||
|
{
|
||||||
|
// len/16
|
||||||
|
int numBlocks = len >> 4;
|
||||||
|
if (numBlocks > 0)
|
||||||
|
Encrypt (numBlocks, (const ChipherBlock *)in, (ChipherBlock *)out);
|
||||||
|
}
|
||||||
|
|
||||||
|
void CBCEncryption::Encrypt (const uint8_t * in, uint8_t * out)
|
||||||
|
{
|
||||||
|
#ifdef AESNI
|
||||||
|
__asm__
|
||||||
|
(
|
||||||
|
"movups (%[iv]), %%xmm1 \n"
|
||||||
|
"movups (%[in]), %%xmm0 \n"
|
||||||
|
"pxor %%xmm1, %%xmm0 \n"
|
||||||
|
EncryptAES256(sched)
|
||||||
|
"movups %%xmm0, (%[out]) \n"
|
||||||
|
"movups %%xmm0, (%[iv]) \n"
|
||||||
|
:
|
||||||
|
: [iv]"r"(&m_LastBlock), [sched]"r"(m_ECBEncryption.GetKeySchedule ()),
|
||||||
|
[in]"r"(in), [out]"r"(out)
|
||||||
|
: "%xmm0", "%xmm1", "memory"
|
||||||
|
);
|
||||||
|
#else
|
||||||
|
Encrypt (1, (const ChipherBlock *)in, (ChipherBlock *)out);
|
||||||
|
#endif
|
||||||
|
}
|
||||||
|
|
||||||
|
void CBCDecryption::Decrypt (int numBlocks, const ChipherBlock * in, ChipherBlock * out)
|
||||||
|
{
|
||||||
|
#ifdef AESNI
|
||||||
|
__asm__
|
||||||
|
(
|
||||||
|
"movups (%[iv]), %%xmm1 \n"
|
||||||
|
"1: \n"
|
||||||
|
"movups (%[in]), %%xmm0 \n"
|
||||||
|
"movaps %%xmm0, %%xmm2 \n"
|
||||||
|
DecryptAES256(sched)
|
||||||
|
"pxor %%xmm1, %%xmm0 \n"
|
||||||
|
"movups %%xmm0, (%[out]) \n"
|
||||||
|
"movaps %%xmm2, %%xmm1 \n"
|
||||||
|
"add $16, %[in] \n"
|
||||||
|
"add $16, %[out] \n"
|
||||||
|
"dec %[num] \n"
|
||||||
|
"jnz 1b \n"
|
||||||
|
"movups %%xmm1, (%[iv]) \n"
|
||||||
|
:
|
||||||
|
: [iv]"r"(&m_IV), [sched]"r"(m_ECBDecryption.GetKeySchedule ()),
|
||||||
|
[in]"r"(in), [out]"r"(out), [num]"r"(numBlocks)
|
||||||
|
: "%xmm0", "%xmm1", "%xmm2", "cc", "memory"
|
||||||
|
);
|
||||||
|
#else
|
||||||
|
for (int i = 0; i < numBlocks; i++)
|
||||||
|
{
|
||||||
|
ChipherBlock tmp = in[i];
|
||||||
|
m_ECBDecryption.Decrypt (in + i, out + i);
|
||||||
|
out[i] ^= m_IV;
|
||||||
|
m_IV = tmp;
|
||||||
|
}
|
||||||
|
#endif
|
||||||
|
}
|
||||||
|
|
||||||
|
void CBCDecryption::Decrypt (const uint8_t * in, std::size_t len, uint8_t * out)
|
||||||
|
{
|
||||||
|
int numBlocks = len >> 4;
|
||||||
|
if (numBlocks > 0)
|
||||||
|
Decrypt (numBlocks, (const ChipherBlock *)in, (ChipherBlock *)out);
|
||||||
|
}
|
||||||
|
|
||||||
|
void CBCDecryption::Decrypt (const uint8_t * in, uint8_t * out)
|
||||||
|
{
|
||||||
|
#ifdef AESNI
|
||||||
|
__asm__
|
||||||
|
(
|
||||||
|
"movups (%[iv]), %%xmm1 \n"
|
||||||
|
"movups (%[in]), %%xmm0 \n"
|
||||||
|
"movups %%xmm0, (%[iv]) \n"
|
||||||
|
DecryptAES256(sched)
|
||||||
|
"pxor %%xmm1, %%xmm0 \n"
|
||||||
|
"movups %%xmm0, (%[out]) \n"
|
||||||
|
:
|
||||||
|
: [iv]"r"(&m_IV), [sched]"r"(m_ECBDecryption.GetKeySchedule ()),
|
||||||
|
[in]"r"(in), [out]"r"(out)
|
||||||
|
: "%xmm0", "%xmm1", "memory"
|
||||||
|
);
|
||||||
|
#else
|
||||||
|
Decrypt (1, (const ChipherBlock *)in, (ChipherBlock *)out);
|
||||||
|
#endif
|
||||||
|
}
|
||||||
|
|
||||||
|
} // crypto
|
||||||
|
} // i2p
|
||||||
|
|
||||||
|
|
44
crypto/aes.h
44
crypto/aes.h
|
@ -181,51 +181,7 @@ namespace crypto
|
||||||
ECBDecryption m_ECBDecryption;
|
ECBDecryption m_ECBDecryption;
|
||||||
};
|
};
|
||||||
|
|
||||||
class TunnelEncryption // with double IV encryption
|
|
||||||
{
|
|
||||||
public:
|
|
||||||
|
|
||||||
void SetKeys (const AESKey& layerKey, const AESKey& ivKey)
|
|
||||||
{
|
|
||||||
m_LayerEncryption.SetKey (layerKey);
|
|
||||||
m_IVEncryption.SetKey (ivKey);
|
|
||||||
}
|
|
||||||
|
|
||||||
void Encrypt (const uint8_t * in, uint8_t * out); // 1024 bytes (16 IV + 1008 data)
|
|
||||||
|
|
||||||
private:
|
|
||||||
|
|
||||||
ECBEncryption m_IVEncryption;
|
|
||||||
#ifdef AESNI
|
|
||||||
ECBEncryption m_LayerEncryption;
|
|
||||||
#else
|
|
||||||
CBCEncryption m_LayerEncryption;
|
|
||||||
#endif
|
|
||||||
};
|
|
||||||
|
|
||||||
class TunnelDecryption // with double IV encryption
|
|
||||||
{
|
|
||||||
public:
|
|
||||||
|
|
||||||
void SetKeys (const AESKey& layerKey, const AESKey& ivKey)
|
|
||||||
{
|
|
||||||
m_LayerDecryption.SetKey (layerKey);
|
|
||||||
m_IVDecryption.SetKey (ivKey);
|
|
||||||
}
|
|
||||||
|
|
||||||
void Decrypt (const uint8_t * in, uint8_t * out); // 1024 bytes (16 IV + 1008 data)
|
|
||||||
|
|
||||||
private:
|
|
||||||
|
|
||||||
ECBDecryption m_IVDecryption;
|
|
||||||
#ifdef AESNI
|
|
||||||
ECBDecryption m_LayerDecryption;
|
|
||||||
#else
|
|
||||||
CBCDecryption m_LayerDecryption;
|
|
||||||
#endif
|
|
||||||
};
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
|
|
|
@ -5,7 +5,7 @@ COMMON_SRC = \
|
||||||
crypto/CryptoConst.cpp crypto/aes.cpp crypto/Signature.cpp \
|
crypto/CryptoConst.cpp crypto/aes.cpp crypto/Signature.cpp \
|
||||||
tunnel/Tunnel.cpp tunnel/TransitTunnel.cpp tunnel/TunnelEndpoint.cpp \
|
tunnel/Tunnel.cpp tunnel/TransitTunnel.cpp tunnel/TunnelEndpoint.cpp \
|
||||||
tunnel/TunnelEndpoint.cpp tunnel/TunnelPool.cpp tunnel/TunnelGateway.cpp \
|
tunnel/TunnelEndpoint.cpp tunnel/TunnelPool.cpp tunnel/TunnelGateway.cpp \
|
||||||
Datagram.cpp Garlic.cpp I2NPProtocol.cpp LeaseSet.cpp \
|
tunnel/TunnelCrypto.cpp Datagram.cpp Garlic.cpp I2NPProtocol.cpp LeaseSet.cpp \
|
||||||
NetDb.cpp NetDbRequests.cpp Profiling.cpp Reseed.cpp \
|
NetDb.cpp NetDbRequests.cpp Profiling.cpp Reseed.cpp \
|
||||||
RouterContext.cpp RouterInfo.cpp Streaming.cpp Identity.cpp \
|
RouterContext.cpp RouterInfo.cpp Streaming.cpp Identity.cpp \
|
||||||
Destination.cpp UPnP.cpp
|
Destination.cpp UPnP.cpp
|
||||||
|
|
|
@ -5,7 +5,7 @@
|
||||||
#include <vector>
|
#include <vector>
|
||||||
#include <mutex>
|
#include <mutex>
|
||||||
#include <memory>
|
#include <memory>
|
||||||
#include "crypto/aes.h"
|
#include "TunnelCrypto.h"
|
||||||
#include "I2NPProtocol.h"
|
#include "I2NPProtocol.h"
|
||||||
#include "TunnelEndpoint.h"
|
#include "TunnelEndpoint.h"
|
||||||
#include "TunnelGateway.h"
|
#include "TunnelGateway.h"
|
||||||
|
|
|
@ -5,7 +5,7 @@
|
||||||
#include <sstream>
|
#include <sstream>
|
||||||
#include <vector>
|
#include <vector>
|
||||||
#include <memory>
|
#include <memory>
|
||||||
#include "crypto/aes.h"
|
#include "TunnelCrypto.h"
|
||||||
#include "RouterInfo.h"
|
#include "RouterInfo.h"
|
||||||
#include "RouterContext.h"
|
#include "RouterContext.h"
|
||||||
#include "util/Timestamp.h"
|
#include "util/Timestamp.h"
|
||||||
|
|
88
tunnel/TunnelCrypto.cpp
Normal file
88
tunnel/TunnelCrypto.cpp
Normal file
|
@ -0,0 +1,88 @@
|
||||||
|
#include "TunnelCrypto.h"
|
||||||
|
#include "TunnelBase.h"
|
||||||
|
#include "AESNIMacros.h"
|
||||||
|
|
||||||
|
namespace i2p {
|
||||||
|
namespace crypto {
|
||||||
|
|
||||||
|
void TunnelEncryption::SetKeys (const AESKey& layerKey, const AESKey& ivKey)
|
||||||
|
{
|
||||||
|
m_LayerEncryption.SetKey (layerKey);
|
||||||
|
m_IVEncryption.SetKey (ivKey);
|
||||||
|
}
|
||||||
|
|
||||||
|
void TunnelEncryption::Encrypt (const uint8_t * in, uint8_t * out)
|
||||||
|
{
|
||||||
|
#ifdef AESNI
|
||||||
|
__asm__
|
||||||
|
(
|
||||||
|
// encrypt IV
|
||||||
|
"movups (%[in]), %%xmm0 \n"
|
||||||
|
EncryptAES256(sched_iv)
|
||||||
|
"movaps %%xmm0, %%xmm1 \n"
|
||||||
|
// double IV encryption
|
||||||
|
EncryptAES256(sched_iv)
|
||||||
|
"movups %%xmm0, (%[out]) \n"
|
||||||
|
// encrypt data, IV is xmm1
|
||||||
|
"1: \n"
|
||||||
|
"add $16, %[in] \n"
|
||||||
|
"add $16, %[out] \n"
|
||||||
|
"movups (%[in]), %%xmm0 \n"
|
||||||
|
"pxor %%xmm1, %%xmm0 \n"
|
||||||
|
EncryptAES256(sched_l)
|
||||||
|
"movaps %%xmm0, %%xmm1 \n"
|
||||||
|
"movups %%xmm0, (%[out]) \n"
|
||||||
|
"dec %[num] \n"
|
||||||
|
"jnz 1b \n"
|
||||||
|
:
|
||||||
|
: [sched_iv]"r"(m_IVEncryption.GetKeySchedule ()), [sched_l]"r"(m_LayerEncryption.GetKeySchedule ()),
|
||||||
|
[in]"r"(in), [out]"r"(out), [num]"r"(63) // 63 blocks = 1008 bytes
|
||||||
|
: "%xmm0", "%xmm1", "cc", "memory"
|
||||||
|
);
|
||||||
|
#else
|
||||||
|
m_IVEncryption.Encrypt ((const ChipherBlock *)in, (ChipherBlock *)out); // iv
|
||||||
|
m_LayerEncryption.SetIV (out);
|
||||||
|
m_LayerEncryption.Encrypt (in + 16, i2p::tunnel::TUNNEL_DATA_ENCRYPTED_SIZE, out + 16); // data
|
||||||
|
m_IVEncryption.Encrypt ((ChipherBlock *)out, (ChipherBlock *)out); // double iv
|
||||||
|
#endif
|
||||||
|
}
|
||||||
|
|
||||||
|
void TunnelDecryption::Decrypt (const uint8_t * in, uint8_t * out)
|
||||||
|
{
|
||||||
|
#ifdef AESNI
|
||||||
|
__asm__
|
||||||
|
(
|
||||||
|
// decrypt IV
|
||||||
|
"movups (%[in]), %%xmm0 \n"
|
||||||
|
DecryptAES256(sched_iv)
|
||||||
|
"movaps %%xmm0, %%xmm1 \n"
|
||||||
|
// double IV encryption
|
||||||
|
DecryptAES256(sched_iv)
|
||||||
|
"movups %%xmm0, (%[out]) \n"
|
||||||
|
// decrypt data, IV is xmm1
|
||||||
|
"1: \n"
|
||||||
|
"add $16, %[in] \n"
|
||||||
|
"add $16, %[out] \n"
|
||||||
|
"movups (%[in]), %%xmm0 \n"
|
||||||
|
"movaps %%xmm0, %%xmm2 \n"
|
||||||
|
DecryptAES256(sched_l)
|
||||||
|
"pxor %%xmm1, %%xmm0 \n"
|
||||||
|
"movups %%xmm0, (%[out]) \n"
|
||||||
|
"movaps %%xmm2, %%xmm1 \n"
|
||||||
|
"dec %[num] \n"
|
||||||
|
"jnz 1b \n"
|
||||||
|
:
|
||||||
|
: [sched_iv]"r"(m_IVDecryption.GetKeySchedule ()), [sched_l]"r"(m_LayerDecryption.GetKeySchedule ()),
|
||||||
|
[in]"r"(in), [out]"r"(out), [num]"r"(63) // 63 blocks = 1008 bytes
|
||||||
|
: "%xmm0", "%xmm1", "%xmm2", "cc", "memory"
|
||||||
|
);
|
||||||
|
#else
|
||||||
|
m_IVDecryption.Decrypt ((const ChipherBlock *)in, (ChipherBlock *)out); // iv
|
||||||
|
m_LayerDecryption.SetIV (out);
|
||||||
|
m_LayerDecryption.Decrypt (in + 16, i2p::tunnel::TUNNEL_DATA_ENCRYPTED_SIZE, out + 16); // data
|
||||||
|
m_IVDecryption.Decrypt ((ChipherBlock *)out, (ChipherBlock *)out); // double iv
|
||||||
|
#endif
|
||||||
|
}
|
||||||
|
|
||||||
|
} // crypto
|
||||||
|
} // i2p
|
49
tunnel/TunnelCrypto.h
Normal file
49
tunnel/TunnelCrypto.h
Normal file
|
@ -0,0 +1,49 @@
|
||||||
|
#ifndef TUNNEL_CRYPTO_H__
|
||||||
|
#define TUNNEL_CRYPTO_H__
|
||||||
|
|
||||||
|
#include "crypto/aes.h"
|
||||||
|
|
||||||
|
namespace i2p {
|
||||||
|
namespace crypto {
|
||||||
|
|
||||||
|
class TunnelEncryption { // with double IV encryption
|
||||||
|
public:
|
||||||
|
void SetKeys (const AESKey& layerKey, const AESKey& ivKey);
|
||||||
|
|
||||||
|
void Encrypt (const uint8_t * in, uint8_t * out); // 1024 bytes (16 IV + 1008 data)
|
||||||
|
|
||||||
|
private:
|
||||||
|
|
||||||
|
ECBEncryption m_IVEncryption;
|
||||||
|
#ifdef AESNI
|
||||||
|
ECBEncryption m_LayerEncryption;
|
||||||
|
#else
|
||||||
|
CBCEncryption m_LayerEncryption;
|
||||||
|
#endif
|
||||||
|
};
|
||||||
|
|
||||||
|
class TunnelDecryption { // with double IV encryption
|
||||||
|
public:
|
||||||
|
|
||||||
|
void SetKeys (const AESKey& layerKey, const AESKey& ivKey)
|
||||||
|
{
|
||||||
|
m_LayerDecryption.SetKey (layerKey);
|
||||||
|
m_IVDecryption.SetKey (ivKey);
|
||||||
|
}
|
||||||
|
|
||||||
|
void Decrypt (const uint8_t * in, uint8_t * out); // 1024 bytes (16 IV + 1008 data)
|
||||||
|
|
||||||
|
private:
|
||||||
|
|
||||||
|
ECBDecryption m_IVDecryption;
|
||||||
|
#ifdef AESNI
|
||||||
|
ECBDecryption m_LayerDecryption;
|
||||||
|
#else
|
||||||
|
CBCDecryption m_LayerDecryption;
|
||||||
|
#endif
|
||||||
|
};
|
||||||
|
|
||||||
|
} // crypto
|
||||||
|
} // i2p
|
||||||
|
|
||||||
|
#endif
|
Loading…
Add table
Reference in a new issue