95e74db6ec
All checks were successful
Chaos Ci / test-and-benchmark (push) Successful in 1m50s
Simpler and more consistent code.
555 lines
18 KiB
C++
555 lines
18 KiB
C++
#include <gtest/gtest.h>
|
|
|
|
#include "Cipher/Block/Des/DesCrypt.hpp"
|
|
#include "Cipher/Block/Encryptor.hpp"
|
|
|
|
using namespace Chaos::Cipher::Block::Des;
|
|
using namespace Chaos::Cipher::Block;
|
|
|
|
TEST(DesCryptTests, KeyScheduleTest)
|
|
{
|
|
Inner_::RawKey key;
|
|
|
|
key[0] = 0b00010011;
|
|
key[1] = 0b00110100;
|
|
key[2] = 0b01010111;
|
|
key[3] = 0b01111001;
|
|
key[4] = 0b10011011;
|
|
key[5] = 0b10111100;
|
|
key[6] = 0b11011111;
|
|
key[7] = 0b11110001;
|
|
|
|
Inner_::KeySchedule schedule(Inner_::KeySchedule::Direction::Encrypt, key);
|
|
|
|
ASSERT_EQ(0b000110110000001011101111111111000111000001110010ULL, schedule[0]);
|
|
ASSERT_EQ(0b011110011010111011011001110110111100100111100101ULL, schedule[1]);
|
|
ASSERT_EQ(0b010101011111110010001010010000101100111110011001ULL, schedule[2]);
|
|
ASSERT_EQ(0b011100101010110111010110110110110011010100011101ULL, schedule[3]);
|
|
ASSERT_EQ(0b011111001110110000000111111010110101001110101000ULL, schedule[4]);
|
|
ASSERT_EQ(0b011000111010010100111110010100000111101100101111ULL, schedule[5]);
|
|
ASSERT_EQ(0b111011001000010010110111111101100001100010111100ULL, schedule[6]);
|
|
ASSERT_EQ(0b111101111000101000111010110000010011101111111011ULL, schedule[7]);
|
|
ASSERT_EQ(0b111000001101101111101011111011011110011110000001ULL, schedule[8]);
|
|
ASSERT_EQ(0b101100011111001101000111101110100100011001001111ULL, schedule[9]);
|
|
ASSERT_EQ(0b001000010101111111010011110111101101001110000110ULL, schedule[10]);
|
|
ASSERT_EQ(0b011101010111000111110101100101000110011111101001ULL, schedule[11]);
|
|
ASSERT_EQ(0b100101111100010111010001111110101011101001000001ULL, schedule[12]);
|
|
ASSERT_EQ(0b010111110100001110110111111100101110011100111010ULL, schedule[13]);
|
|
ASSERT_EQ(0b101111111001000110001101001111010011111100001010ULL, schedule[14]);
|
|
ASSERT_EQ(0b110010110011110110001011000011100001011111110101ULL, schedule[15]);
|
|
}
|
|
|
|
TEST(DesCryptTests, EncryptTest)
|
|
{
|
|
struct Helper
|
|
{
|
|
std::array<uint8_t, DesCrypt::BlockSize> operator()(const std::array<uint8_t, DesCrypt::BlockSize> & data,
|
|
const std::array<uint8_t, 8> & key) const
|
|
{
|
|
std::array<uint8_t, DesCrypt::BlockSize> result = {};
|
|
|
|
DesCrypt::Key desKey(key.begin(), key.end());
|
|
DesCrypt::DesEncryptor enc(desKey);
|
|
enc.EncryptBlock(result.begin(), result.end(), data.begin(), data.end());
|
|
|
|
return result;
|
|
}
|
|
};
|
|
|
|
Helper des;
|
|
|
|
{
|
|
std::array<uint8_t, DesCrypt::BlockSize> data = { 0x01, 0x23, 0x45, 0x67, 0x89, 0xab, 0xcd, 0xef };
|
|
std::array<uint8_t, 8> key = { 0x13, 0x34, 0x57, 0x79, 0x9b, 0xbc, 0xdf, 0xf1 };
|
|
|
|
std::array<uint8_t, DesCrypt::BlockSize> expected = { 0x85, 0xe8, 0x13, 0x54, 0x0f, 0x0a, 0xb4, 0x05 };
|
|
|
|
ASSERT_EQ(expected, des(data, key));
|
|
}
|
|
|
|
{
|
|
std::array<uint8_t, DesCrypt::BlockSize> data = { 0xaa, 0xf3, 0x83, 0x16, 0x2d, 0x2e, 0x6b, 0xcb };
|
|
std::array<uint8_t, 8> key = { 0x44, 0xbf, 0x32, 0x19, 0x99, 0x25, 0x81, 0x51 };
|
|
|
|
std::array<uint8_t, DesCrypt::BlockSize> expected = { 0x07, 0xe8, 0x7f, 0xaa, 0xb3, 0x17, 0x13, 0x18 };
|
|
|
|
ASSERT_EQ(expected, des(data, key));
|
|
}
|
|
|
|
{
|
|
std::array<uint8_t, DesCrypt::BlockSize> data = { 0xe5, 0x1a, 0x9f, 0xd4, 0x19, 0xa7, 0x93, 0x44 };
|
|
std::array<uint8_t, 8> key = { 0xda, 0xec, 0x68, 0xae, 0x83, 0xe0, 0x1e, 0xab };
|
|
|
|
std::array<uint8_t, DesCrypt::BlockSize> expected = { 0x42, 0x27, 0x88, 0xa6, 0x7b, 0x6c, 0x18, 0xed };
|
|
|
|
ASSERT_EQ(expected, des(data, key));
|
|
}
|
|
}
|
|
|
|
TEST(DesCryptTests, EncryptUInt64BlockTest)
|
|
{
|
|
struct Helper
|
|
{
|
|
uint64_t operator()(uint64_t data,
|
|
const std::array<uint8_t, 8> & key) const
|
|
{
|
|
DesCrypt::Key desKey(key.begin(), key.end());
|
|
DesCrypt::DesEncryptor enc(desKey);
|
|
|
|
return enc.EncryptBlock(data);
|
|
}
|
|
};
|
|
|
|
Helper des;
|
|
|
|
{
|
|
uint64_t data = 0x0123456789abcdef;
|
|
std::array<uint8_t, 8> key = { 0x13, 0x34, 0x57, 0x79, 0x9b, 0xbc, 0xdf, 0xf1 };
|
|
|
|
uint64_t expected = 0x85e813540f0ab405;
|
|
|
|
ASSERT_EQ(expected, des(data, key));
|
|
}
|
|
|
|
{
|
|
uint64_t data = 0xaaf383162d2e6bcb;
|
|
std::array<uint8_t, 8> key = { 0x44, 0xbf, 0x32, 0x19, 0x99, 0x25, 0x81, 0x51 };
|
|
|
|
uint64_t expected = 0x07e87faab3171318;
|
|
|
|
ASSERT_EQ(expected, des(data, key));
|
|
}
|
|
|
|
{
|
|
uint64_t data = 0xe51a9fd419a79344;
|
|
std::array<uint8_t, 8> key = { 0xda, 0xec, 0x68, 0xae, 0x83, 0xe0, 0x1e, 0xab };
|
|
|
|
uint64_t expected = 0x422788a67b6c18ed;
|
|
|
|
ASSERT_EQ(expected, des(data, key));
|
|
}
|
|
}
|
|
|
|
TEST(DesCryptTests, EncryptShortDataTest)
|
|
{
|
|
struct Helper
|
|
{
|
|
std::vector<uint8_t> operator()(const std::vector<uint8_t> & data,
|
|
const std::vector<uint8_t> & key)
|
|
{
|
|
std::vector<uint8_t> result;
|
|
result.resize(DesCrypt::BlockSize, 0);
|
|
|
|
DesCrypt::Key desKey(key.begin(), key.end());
|
|
DesCrypt::DesEncryptor enc(desKey);
|
|
enc.EncryptBlock(result.begin(), result.end(), data.begin(), data.end());
|
|
|
|
return result;
|
|
}
|
|
};
|
|
|
|
Helper des;
|
|
|
|
{
|
|
// treated as { 0xe5, 0x1a, 0x9f, 0xd4, 0x19, 0x00, 0x00, 0x00 }
|
|
std::vector<uint8_t> dataShort = { 0xe5, 0x1a, 0x9f, 0xd4, 0x19 };
|
|
|
|
std::vector<uint8_t> data = { 0xe5, 0x1a, 0x9f, 0xd4, 0x19, 0x00, 0x00, 0x00 };
|
|
std::vector<uint8_t> key = { 0xda, 0xec, 0x68, 0xae, 0x83, 0xe0, 0x1e, 0xab };
|
|
|
|
std::vector<uint8_t> expected = { 0xd8, 0xa8, 0xb8, 0xb4, 0xc0, 0x9b, 0x04, 0x09 };
|
|
|
|
ASSERT_EQ(expected, des(data, key));
|
|
ASSERT_EQ(expected, des(dataShort, key));
|
|
}
|
|
}
|
|
|
|
TEST(DesCryptTests, EncryptLongDataTest)
|
|
{
|
|
struct Helper
|
|
{
|
|
std::vector<uint8_t> operator()(const std::vector<uint8_t> & data,
|
|
const std::vector<uint8_t> & key)
|
|
{
|
|
std::vector<uint8_t> result;
|
|
result.resize(DesCrypt::BlockSize, 0);
|
|
|
|
DesCrypt::Key desKey(key.begin(), key.end());
|
|
DesCrypt::DesEncryptor enc(desKey);
|
|
enc.EncryptBlock(result.begin(), result.end(), data.begin(), data.end());
|
|
|
|
return result;
|
|
}
|
|
};
|
|
|
|
Helper des;
|
|
|
|
{
|
|
// treated as { 0xe5, 0x1a, 0x9f, 0xd4, 0x19, 0xa7, 0x93, 0x44 }
|
|
std::vector<uint8_t> dataLong = { 0xe5, 0x1a, 0x9f, 0xd4, 0x19, 0xa7, 0x93, 0x44, 0xaa, 0xbb };
|
|
|
|
std::vector<uint8_t> data = { 0xe5, 0x1a, 0x9f, 0xd4, 0x19, 0xa7, 0x93, 0x44 };
|
|
std::vector<uint8_t> key = { 0xda, 0xec, 0x68, 0xae, 0x83, 0xe0, 0x1e, 0xab };
|
|
|
|
std::vector<uint8_t> expected = { 0x42, 0x27, 0x88, 0xa6, 0x7b, 0x6c, 0x18, 0xed };
|
|
|
|
ASSERT_EQ(expected, des(data, key));
|
|
ASSERT_EQ(expected, des(dataLong, key));
|
|
}
|
|
}
|
|
|
|
TEST(DesCryptTests, DecryptTest)
|
|
{
|
|
struct Helper
|
|
{
|
|
std::array<uint8_t, DesCrypt::BlockSize> operator()(const std::array<uint8_t, DesCrypt::BlockSize> & data,
|
|
const std::array<uint8_t, 8> & key) const
|
|
{
|
|
std::array<uint8_t, DesCrypt::BlockSize> result = {};
|
|
|
|
DesCrypt::Key desKey(key.begin(), key.end());
|
|
DesCrypt::DesDecryptor dec(desKey);
|
|
dec.DecryptBlock(result.begin(), result.end(), data.begin(), data.end());
|
|
|
|
return result;
|
|
}
|
|
};
|
|
|
|
Helper des;
|
|
|
|
{
|
|
std::array<uint8_t, DesCrypt::BlockSize> data = { 0x85, 0xe8, 0x13, 0x54, 0x0f, 0x0a, 0xb4, 0x05 };
|
|
std::array<uint8_t, 8> key = { 0x13, 0x34, 0x57, 0x79, 0x9b, 0xbc, 0xdf, 0xf1 };
|
|
|
|
std::array<uint8_t, DesCrypt::BlockSize> expected = { 0x01, 0x23, 0x45, 0x67, 0x89, 0xab, 0xcd, 0xef };
|
|
|
|
ASSERT_EQ(expected, des(data, key));
|
|
}
|
|
|
|
{
|
|
std::array<uint8_t, DesCrypt::BlockSize> data = { 0x07, 0xe8, 0x7f, 0xaa, 0xb3, 0x17, 0x13, 0x18 };
|
|
std::array<uint8_t, 8> key = { 0x44, 0xbf, 0x32, 0x19, 0x99, 0x25, 0x81, 0x51 };
|
|
|
|
std::array<uint8_t, DesCrypt::BlockSize> expected = { 0xaa, 0xf3, 0x83, 0x16, 0x2d, 0x2e, 0x6b, 0xcb };
|
|
|
|
ASSERT_EQ(expected, des(data, key));
|
|
}
|
|
|
|
{
|
|
std::array<uint8_t, DesCrypt::BlockSize> data = { 0x42, 0x27, 0x88, 0xa6, 0x7b, 0x6c, 0x18, 0xed };
|
|
std::array<uint8_t, 8> key = { 0xda, 0xec, 0x68, 0xae, 0x83, 0xe0, 0x1e, 0xab };
|
|
|
|
std::array<uint8_t, DesCrypt::BlockSize> expected = { 0xe5, 0x1a, 0x9f, 0xd4, 0x19, 0xa7, 0x93, 0x44 };
|
|
|
|
ASSERT_EQ(expected, des(data, key));
|
|
}
|
|
}
|
|
|
|
TEST(DesCryptTests, DecryptUInt64BlockTest)
|
|
{
|
|
struct Helper
|
|
{
|
|
uint64_t operator()(uint64_t data,
|
|
const std::array<uint8_t, 8> & key) const
|
|
{
|
|
DesCrypt::Key desKey(key.begin(), key.end());
|
|
DesCrypt::DesDecryptor dec(desKey);
|
|
|
|
return dec.DecryptBlock(data);
|
|
}
|
|
};
|
|
|
|
Helper des;
|
|
|
|
{
|
|
uint64_t data = 0x85e813540f0ab405;
|
|
std::array<uint8_t, 8> key = { 0x13, 0x34, 0x57, 0x79, 0x9b, 0xbc, 0xdf, 0xf1 };
|
|
|
|
uint64_t expected = 0x0123456789abcdef;
|
|
|
|
ASSERT_EQ(expected, des(data, key));
|
|
}
|
|
|
|
{
|
|
uint64_t data = 0x07e87faab3171318;
|
|
std::array<uint8_t, 8> key = { 0x44, 0xbf, 0x32, 0x19, 0x99, 0x25, 0x81, 0x51 };
|
|
|
|
uint64_t expected = 0xaaf383162d2e6bcb;
|
|
|
|
ASSERT_EQ(expected, des(data, key));
|
|
}
|
|
|
|
{
|
|
uint64_t data = 0x422788a67b6c18ed;
|
|
std::array<uint8_t, 8> key = { 0xda, 0xec, 0x68, 0xae, 0x83, 0xe0, 0x1e, 0xab };
|
|
|
|
uint64_t expected = 0xe51a9fd419a79344;
|
|
|
|
ASSERT_EQ(expected, des(data, key));
|
|
}
|
|
}
|
|
|
|
TEST(DesCryptTests, DecryptShortDataTest)
|
|
{
|
|
struct Helper
|
|
{
|
|
std::vector<uint8_t> operator()(const std::vector<uint8_t> & data,
|
|
const std::vector<uint8_t> & key)
|
|
{
|
|
std::vector<uint8_t> result;
|
|
result.resize(DesCrypt::BlockSize, 0);
|
|
|
|
DesCrypt::Key desKey(key.begin(), key.end());
|
|
DesCrypt::DesDecryptor dec(desKey);
|
|
dec.DecryptBlock(result.begin(), result.end(), data.begin(), data.end());
|
|
|
|
return result;
|
|
}
|
|
};
|
|
|
|
Helper des;
|
|
|
|
{
|
|
// treated as { 0xe5, 0x1a, 0x9f, 0xd4, 0x19, 0x00, 0x00, 0x00 }
|
|
std::vector<uint8_t> dataShort = { 0xe5, 0x1a, 0x9f, 0xd4, 0x19 };
|
|
|
|
std::vector<uint8_t> data = { 0xe5, 0x1a, 0x9f, 0xd4, 0x19, 0x00, 0x00, 0x00 };
|
|
std::vector<uint8_t> key = { 0xda, 0xec, 0x68, 0xae, 0x83, 0xe0, 0x1e, 0xab };
|
|
|
|
std::vector<uint8_t> expected = { 0x90, 0xe5, 0xf6, 0x5b, 0xe1, 0xd3, 0x8a, 0x64 };
|
|
|
|
ASSERT_EQ(expected, des(data, key));
|
|
ASSERT_EQ(expected, des(dataShort, key));
|
|
}
|
|
}
|
|
|
|
TEST(DesCryptTests, DecryptLongDataTest)
|
|
{
|
|
struct Helper
|
|
{
|
|
std::vector<uint8_t> operator()(const std::vector<uint8_t> & data,
|
|
const std::vector<uint8_t> & key)
|
|
{
|
|
std::vector<uint8_t> result;
|
|
result.resize(DesCrypt::BlockSize, 0);
|
|
|
|
DesCrypt::Key desKey(key.begin(), key.end());
|
|
DesCrypt::DesDecryptor dec(desKey);
|
|
dec.DecryptBlock(result.begin(), result.end(), data.begin(), data.end());
|
|
|
|
return result;
|
|
}
|
|
};
|
|
|
|
Helper des;
|
|
|
|
{
|
|
// treated as { 0xe5, 0x1a, 0x9f, 0xd4, 0x19, 0xa7, 0x93, 0x44 }
|
|
std::vector<uint8_t> dataLong = { 0xe5, 0x1a, 0x9f, 0xd4, 0x19, 0xa7, 0x93, 0x44, 0xaa, 0xbb };
|
|
|
|
std::vector<uint8_t> data = { 0xe5, 0x1a, 0x9f, 0xd4, 0x19, 0xa7, 0x93, 0x44 };
|
|
std::vector<uint8_t> key = { 0xda, 0xec, 0x68, 0xae, 0x83, 0xe0, 0x1e, 0xab };
|
|
|
|
std::vector<uint8_t> expected = { 0x45, 0x69, 0x71, 0x17, 0x13, 0xfb, 0x3e, 0xee };
|
|
|
|
ASSERT_EQ(expected, des(data, key));
|
|
ASSERT_EQ(expected, des(dataLong, key));
|
|
}
|
|
}
|
|
|
|
TEST(DesCryptTests, ShortKeyTest)
|
|
{
|
|
{
|
|
std::array<uint8_t, 7> key = {};
|
|
ASSERT_THROW(DesCrypt::Key(key.begin(), key.end()), Chaos::Service::ChaosException);
|
|
}
|
|
}
|
|
|
|
TEST(DesCryptTests, LongKeyTest)
|
|
{
|
|
{
|
|
std::array<uint8_t, 9> key = {};
|
|
ASSERT_THROW(DesCrypt::Key(key.begin(), key.end()), Chaos::Service::ChaosException);
|
|
}
|
|
}
|
|
|
|
TEST(DesCryptTests, OutIteratorUsageEncryptTest)
|
|
{
|
|
{
|
|
std::array<uint8_t, DesCrypt::BlockSize> data = { 0xe5, 0x1a, 0x9f, 0xd4, 0x19, 0xa7, 0x93, 0x44 };
|
|
std::array<uint8_t, 8> key = { 0xda, 0xec, 0x68, 0xae, 0x83, 0xe0, 0x1e, 0xab };
|
|
|
|
std::array<uint8_t, 11> fact = {};
|
|
// First and last 3 bytes should be untouched.
|
|
std::array<uint8_t, 11> expected = { 0x00, 0x00, 0x00, 0x42, 0x27, 0x88, 0xa6, 0x7b, 0x00, 0x00, 0x00 };
|
|
|
|
DesCrypt::Key desKey(key.begin(), key.end());
|
|
DesCrypt::DesEncryptor enc(desKey);
|
|
enc.EncryptBlock(fact.begin() + 3, fact.end() - 3, data.begin(), data.end());
|
|
|
|
ASSERT_EQ(expected, fact);
|
|
}
|
|
|
|
{
|
|
std::array<uint8_t, DesCrypt::BlockSize + 2> data = { 0xe5, 0x1a, 0x9f, 0xd4, 0x19, 0xa7, 0x93, 0x44, 0x44, 0x44 };
|
|
std::array<uint8_t, 8> key = { 0xda, 0xec, 0x68, 0xae, 0x83, 0xe0, 0x1e, 0xab };
|
|
|
|
std::array<uint8_t, 12> fact = {};
|
|
// First and last 4 bytes should be untouched.
|
|
std::array<uint8_t, 12> expected = { 0x00, 0x00, 0x00, 0x00, 0x42, 0x27, 0x88, 0xa6, 0x00, 0x00, 0x00, 0x00 };
|
|
|
|
DesCrypt::Key desKey(key.begin(), key.end());
|
|
DesCrypt::DesEncryptor enc(desKey);
|
|
enc.EncryptBlock(fact.begin() + 4, fact.end() - 4, data.begin(), data.end());
|
|
|
|
ASSERT_EQ(expected, fact);
|
|
}
|
|
|
|
{
|
|
std::array<uint8_t, DesCrypt::BlockSize + 2> data = { 0xe5, 0x1a, 0x9f, 0xd4, 0x19, 0xa7, 0x93, 0x44, 0x44, 0x44 };
|
|
std::array<uint8_t, 8> key = { 0xda, 0xec, 0x68, 0xae, 0x83, 0xe0, 0x1e, 0xab };
|
|
|
|
std::array<uint8_t, 12> fact = {};
|
|
std::array<uint8_t, 12> expected = {};
|
|
|
|
DesCrypt::Key desKey(key.begin(), key.end());
|
|
DesCrypt::DesEncryptor enc(desKey);
|
|
enc.EncryptBlock(fact.begin() + 3, fact.begin() + 3, data.begin(), data.end());
|
|
|
|
ASSERT_EQ(expected, fact);
|
|
}
|
|
}
|
|
|
|
TEST(DesCryptTests, OutIteratorUsageDecryptTest)
|
|
{
|
|
{
|
|
std::array<uint8_t, DesCrypt::BlockSize> data = { 0xe5, 0x1a, 0x9f, 0xd4, 0x19, 0xa7, 0x93, 0x44 };
|
|
std::array<uint8_t, 8> key = { 0xda, 0xec, 0x68, 0xae, 0x83, 0xe0, 0x1e, 0xab };
|
|
|
|
std::array<uint8_t, 11> fact = {};
|
|
// First and last 3 bytes should be untouched.
|
|
std::array<uint8_t, 11> expected = { 0x00, 0x00, 0x00, 0x45, 0x69, 0x71, 0x17, 0x13, 0x00, 0x00, 0x00 };
|
|
|
|
DesCrypt::Key desKey(key.begin(), key.end());
|
|
DesCrypt::DesDecryptor dec(desKey);
|
|
dec.DecryptBlock(fact.begin() + 3, fact.end() - 3, data.begin(), data.end());
|
|
|
|
ASSERT_EQ(expected, fact);
|
|
}
|
|
|
|
{
|
|
std::array<uint8_t, DesCrypt::BlockSize + 2> data = { 0xe5, 0x1a, 0x9f, 0xd4, 0x19, 0xa7, 0x93, 0x44, 0x44, 0x44 };
|
|
std::array<uint8_t, 8> key = { 0xda, 0xec, 0x68, 0xae, 0x83, 0xe0, 0x1e, 0xab };
|
|
|
|
std::array<uint8_t, 12> fact = {};
|
|
// First and last 4 bytes should be untouched.
|
|
std::array<uint8_t, 12> expected = { 0x00, 0x00, 0x00, 0x00, 0x45, 0x69, 0x71, 0x17, 0x00, 0x00, 0x00, 0x00 };
|
|
|
|
DesCrypt::Key desKey(key.begin(), key.end());
|
|
DesCrypt::DesDecryptor dec(desKey);
|
|
dec.DecryptBlock(fact.begin() + 4, fact.end() - 4, data.begin(), data.end());
|
|
|
|
ASSERT_EQ(expected, fact);
|
|
}
|
|
|
|
{
|
|
std::array<uint8_t, DesCrypt::BlockSize + 2> data = { 0xe5, 0x1a, 0x9f, 0xd4, 0x19, 0xa7, 0x93, 0x44, 0x44, 0x44 };
|
|
std::array<uint8_t, 8> key = { 0xda, 0xec, 0x68, 0xae, 0x83, 0xe0, 0x1e, 0xab };
|
|
|
|
std::array<uint8_t, 12> fact = {};
|
|
std::array<uint8_t, 12> expected = {};
|
|
|
|
DesCrypt::Key desKey(key.begin(), key.end());
|
|
DesCrypt::DesDecryptor dec(desKey);
|
|
dec.DecryptBlock(fact.begin() + 3, fact.begin() + 3, data.begin(), data.end());
|
|
|
|
ASSERT_EQ(expected, fact);
|
|
}
|
|
}
|
|
|
|
template<typename Impl, typename InputIt>
|
|
static std::vector<uint8_t> EncryptThroughBase(const Encryptor<Impl> & enc,
|
|
InputIt begin, InputIt end)
|
|
{
|
|
std::vector<uint8_t> result;
|
|
result.resize(enc.GetBlockSize(), 0);
|
|
|
|
enc.EncryptBlock(result.begin(), result.end(), begin, end);
|
|
return result;
|
|
}
|
|
|
|
TEST(DesCryptTests, EncryptThroughBaseTest)
|
|
{
|
|
std::array<uint8_t, 8> key = { 0x13, 0x34, 0x57, 0x79, 0x9b, 0xbc, 0xdf, 0xf1 };
|
|
|
|
std::vector<uint8_t> data = { 0x01, 0x23, 0x45, 0x67, 0x89, 0xab, 0xcd, 0xef };
|
|
std::vector<uint8_t> expected = { 0x85, 0xe8, 0x13, 0x54, 0x0f, 0x0a, 0xb4, 0x05 };
|
|
|
|
DesCrypt::Key desKey(key.begin(), key.end());
|
|
DesCrypt::DesEncryptor enc(desKey);
|
|
|
|
ASSERT_EQ(expected, EncryptThroughBase(enc, data.begin(), data.end()));
|
|
}
|
|
|
|
template<typename Impl>
|
|
static uint64_t EncryptUInt64BlockThroughBase(const Encryptor<Impl> & enc, uint64_t block)
|
|
{
|
|
return enc.EncryptBlock(block);
|
|
}
|
|
|
|
TEST(DesCryptTests, EncryptUInt64BlockThroughBaseTest)
|
|
{
|
|
std::array<uint8_t, 8> key = { 0x13, 0x34, 0x57, 0x79, 0x9b, 0xbc, 0xdf, 0xf1 };
|
|
|
|
uint64_t data = 0x0123456789abcdef;
|
|
uint64_t expected = 0x85e813540f0ab405;
|
|
|
|
DesCrypt::Key desKey(key.begin(), key.end());
|
|
DesCrypt::DesEncryptor enc(desKey);
|
|
|
|
ASSERT_EQ(expected, EncryptUInt64BlockThroughBase(enc, data));
|
|
}
|
|
|
|
template<typename Impl, typename InputIt>
|
|
static std::vector<uint8_t> DecryptThroughBase(const Decryptor<Impl> & dec,
|
|
InputIt begin, InputIt end)
|
|
{
|
|
std::vector<uint8_t> result;
|
|
result.resize(dec.GetBlockSize(), 0);
|
|
|
|
dec.DecryptBlock(result.begin(), result.end(), begin, end);
|
|
return result;
|
|
}
|
|
|
|
TEST(DesCryptTests, DecryptThroughBaseTest)
|
|
{
|
|
std::array<uint8_t, 8> key = { 0x13, 0x34, 0x57, 0x79, 0x9b, 0xbc, 0xdf, 0xf1 };
|
|
|
|
std::vector<uint8_t> data = { 0x85, 0xe8, 0x13, 0x54, 0x0f, 0x0a, 0xb4, 0x05 };
|
|
std::vector<uint8_t> expected = { 0x01, 0x23, 0x45, 0x67, 0x89, 0xab, 0xcd, 0xef };
|
|
|
|
DesCrypt::Key desKey(key.begin(), key.end());
|
|
DesCrypt::DesDecryptor dec(desKey);
|
|
|
|
ASSERT_EQ(expected, DecryptThroughBase(dec, data.begin(), data.end()));
|
|
}
|
|
|
|
template<typename Impl>
|
|
static uint64_t DecryptUInt64BlockThroughBase(const Decryptor<Impl> & dec, uint64_t block)
|
|
{
|
|
return dec.DecryptBlock(block);
|
|
}
|
|
|
|
TEST(DesCryptTests, DecryptUInt64BlockThroughBaseTest)
|
|
{
|
|
std::array<uint8_t, 8> key = { 0x13, 0x34, 0x57, 0x79, 0x9b, 0xbc, 0xdf, 0xf1 };
|
|
|
|
uint64_t data = 0x85e813540f0ab405;
|
|
uint64_t expected = 0x0123456789abcdef;
|
|
|
|
DesCrypt::Key desKey(key.begin(), key.end());
|
|
DesCrypt::DesDecryptor dec(desKey);
|
|
|
|
ASSERT_EQ(expected, DecryptUInt64BlockThroughBase(dec, data));
|
|
}
|