i2pd/aes.h

189 lines
3.8 KiB
C
Raw Normal View History

2014-05-06 20:22:22 +04:00
#ifndef AES_H__
#define AES_H__
#include <inttypes.h>
#include <cryptopp/modes.h>
#include <cryptopp/aes.h>
namespace i2p
{
namespace crypto
{
union ChipherBlock
{
uint8_t buf[16];
uint64_t ll[2];
2014-05-09 19:44:39 +04:00
2014-05-09 20:05:04 +04:00
void operator^=(const ChipherBlock& other) // XOR
2014-05-09 19:44:39 +04:00
{
ll[0] ^= other.ll[0];
ll[1] ^= other.ll[1];
}
2014-05-06 20:22:22 +04:00
};
2014-06-02 18:05:04 +04:00
#ifdef AESNI
2014-05-07 23:40:24 +04:00
class ECBCryptoAESNI
{
2014-05-09 05:43:08 +04:00
public:
ECBCryptoAESNI ();
2014-05-13 06:51:59 +04:00
uint8_t * GetKeySchedule () { return m_KeySchedule; };
protected:
void ExpandKey (const uint8_t * key);
protected:
2014-05-09 05:43:08 +04:00
uint8_t * m_KeySchedule; // start of 16 bytes boundary of m_UnalignedBuffer
uint8_t m_UnalignedBuffer[256]; // 14 rounds for AES-256, 240 + 16 bytes
};
class ECBEncryptionAESNI: public ECBCryptoAESNI
2014-05-07 22:48:37 +04:00
{
public:
void SetKey (const uint8_t * key) { ExpandKey (key); };
2014-05-07 23:39:30 +04:00
void Encrypt (const ChipherBlock * in, ChipherBlock * out);
};
2014-05-07 22:48:37 +04:00
class ECBDecryptionAESNI: public ECBCryptoAESNI
{
public:
void SetKey (const uint8_t * key);
void Decrypt (const ChipherBlock * in, ChipherBlock * out);
2014-05-07 22:48:37 +04:00
};
2014-05-09 00:49:00 +04:00
typedef ECBEncryptionAESNI ECBEncryption;
typedef ECBDecryptionAESNI ECBDecryption;
#else // use crypto++
class ECBEncryption
{
public:
void SetKey (const uint8_t * key)
{
m_Encryption.SetKey (key, 32);
}
void Encrypt (const ChipherBlock * in, ChipherBlock * out)
{
m_Encryption.ProcessData (out->buf, in->buf, 16);
}
private:
CryptoPP::ECB_Mode<CryptoPP::AES>::Encryption m_Encryption;
};
class ECBDecryption
{
public:
void SetKey (const uint8_t * key)
{
m_Decryption.SetKey (key, 32);
}
void Decrypt (const ChipherBlock * in, ChipherBlock * out)
{
m_Decryption.ProcessData (out->buf, in->buf, 16);
}
private:
CryptoPP::ECB_Mode<CryptoPP::AES>::Decryption m_Decryption;
};
2014-05-07 22:48:37 +04:00
#endif
2014-05-06 20:22:22 +04:00
class CBCEncryption
{
public:
CBCEncryption () { memset (m_LastBlock.buf, 0, 16); };
2014-05-09 05:43:08 +04:00
void SetKey (const uint8_t * key) { m_ECBEncryption.SetKey (key); }; // 32 bytes
2014-05-07 06:30:09 +04:00
void SetIV (const uint8_t * iv) { memcpy (m_LastBlock.buf, iv, 16); }; // 16 bytes
2014-05-06 20:22:22 +04:00
void Encrypt (int numBlocks, const ChipherBlock * in, ChipherBlock * out);
void Encrypt (const uint8_t * in, std::size_t len, uint8_t * out);
2014-05-14 22:54:01 +04:00
void Encrypt (const uint8_t * in, uint8_t * out); // one block
2014-05-06 20:22:22 +04:00
private:
ChipherBlock m_LastBlock;
2014-05-13 06:51:59 +04:00
2014-05-09 05:43:08 +04:00
ECBEncryption m_ECBEncryption;
2014-05-06 20:22:22 +04:00
};
class CBCDecryption
{
public:
CBCDecryption () { memset (m_IV.buf, 0, 16); };
2014-05-09 05:43:08 +04:00
void SetKey (const uint8_t * key) { m_ECBDecryption.SetKey (key); }; // 32 bytes
2014-05-07 06:30:09 +04:00
void SetIV (const uint8_t * iv) { memcpy (m_IV.buf, iv, 16); }; // 16 bytes
2014-05-06 20:22:22 +04:00
void Decrypt (int numBlocks, const ChipherBlock * in, ChipherBlock * out);
void Decrypt (const uint8_t * in, std::size_t len, uint8_t * out);
2014-05-14 22:54:01 +04:00
void Decrypt (const uint8_t * in, uint8_t * out); // one block
2014-05-06 20:22:22 +04:00
private:
ChipherBlock m_IV;
2014-05-09 05:43:08 +04:00
ECBDecryption m_ECBDecryption;
2014-05-06 20:22:22 +04:00
};
2014-05-15 19:21:41 +04:00
class TunnelEncryption // with double IV encryption
{
public:
2014-05-16 01:55:09 +04:00
void SetKeys (const uint8_t * layerKey, const uint8_t * ivKey)
2014-05-15 19:21:41 +04:00
{
m_LayerEncryption.SetKey (layerKey);
m_IVEncryption.SetKey (ivKey);
}
void Encrypt (uint8_t * payload); // 1024 bytes (16 IV + 1008 data)
private:
ECBEncryption m_IVEncryption;
2014-06-02 18:05:04 +04:00
#ifdef AESNI
2014-05-15 20:59:07 +04:00
ECBEncryption m_LayerEncryption;
#else
2014-05-15 19:21:41 +04:00
CBCEncryption m_LayerEncryption;
2014-05-15 20:59:07 +04:00
#endif
2014-05-15 19:21:41 +04:00
};
class TunnelDecryption // with double IV encryption
{
public:
2014-05-16 01:55:09 +04:00
void SetKeys (const uint8_t * layerKey, const uint8_t * ivKey)
2014-05-15 19:21:41 +04:00
{
m_LayerDecryption.SetKey (layerKey);
m_IVDecryption.SetKey (ivKey);
}
void Decrypt (uint8_t * payload); // 1024 bytes (16 IV + 1008 data)
private:
ECBDecryption m_IVDecryption;
2014-06-02 18:05:04 +04:00
#ifdef AESNI
2014-05-15 21:10:07 +04:00
ECBDecryption m_LayerDecryption;
#else
2014-05-15 19:21:41 +04:00
CBCDecryption m_LayerDecryption;
2014-05-15 21:10:07 +04:00
#endif
2014-05-15 19:21:41 +04:00
};
2014-05-06 20:22:22 +04:00
}
}
#endif