115 lines
4.4 KiB
C++
115 lines
4.4 KiB
C++
#include <gtest/gtest.h>
|
|
#include "TestHelpers/AssertThrowEx.hpp"
|
|
|
|
#include <cstring>
|
|
#include <cstdint>
|
|
#include <vector>
|
|
|
|
#include "Cipher/Block/Des/DesCrypt.hpp"
|
|
#include "Cipher/Block/Mode/Ecb.hpp"
|
|
#include "Cipher/Block/BlockAdapter.hpp"
|
|
#include "Padding/PadderPkcs7.hpp"
|
|
#include "Service/ChaosException.hpp"
|
|
|
|
using namespace Chaos::Cipher::Block;
|
|
using namespace Chaos::Cipher::Block::Mode;
|
|
using namespace Chaos::Padding;
|
|
|
|
TEST(BlockAdapterTests, EncryptTest)
|
|
{
|
|
using Adapter = BlockAdapter<EcbMode<Des::DesCrypt, PadderPkcs7>>;
|
|
|
|
{
|
|
const char * str = "smoke and mirrors";
|
|
std::vector<uint8_t> data(str, str + strlen(str));
|
|
std::vector<uint8_t> key = { 0x0c, 0xfd, 0x01, 0xa3, 0x60, 0xd0, 0x15, 0xa6 };
|
|
|
|
std::vector<uint8_t> expected = { 0x84, 0x39, 0xcd, 0xe5, 0x1f, 0x3d,
|
|
0x2a, 0x11, 0x63, 0xc3, 0x56, 0x28,
|
|
0xf5, 0x89, 0xdb, 0xc4, 0xa6, 0xca,
|
|
0x52, 0xb3, 0xa8, 0xce, 0x64, 0x2d };
|
|
|
|
ASSERT_EQ(expected, Adapter::Encrypt(key, data));
|
|
}
|
|
|
|
{
|
|
std::vector<uint8_t> data = {};
|
|
std::vector<uint8_t> key = { 0xaa, 0xbb, 0xaa, 0xbb, 0xaa, 0xbb, 0xaa, 0xbb };
|
|
|
|
std::vector<uint8_t> expected = { 0xef, 0xe2, 0x4d, 0xe9, 0x9d, 0xe7, 0x9b, 0xbf };
|
|
|
|
ASSERT_EQ(expected, Adapter::Encrypt(key, data));
|
|
}
|
|
|
|
{
|
|
std::vector<uint8_t> data = { 0x96, 0x9f, 0x42, 0xd3, 0x47, 0x10, 0x4d, 0xb4, 0x89, 0x54,
|
|
0x5c, 0xad, 0x0a, 0xb9, 0xac, 0x4e, 0xbf, 0x00, 0x1a, 0x47,
|
|
0x3d, 0xd1, 0xe7, 0xb3, 0x70, 0xb7, 0x4e, 0xd9, 0x73, 0x7d,
|
|
0xb3, 0x24 };
|
|
std::vector<uint8_t> key = { 0xe9, 0xeb, 0x1f, 0xfa, 0x39, 0x95, 0xf0, 0xb9 };
|
|
|
|
std::vector<uint8_t> expected = { 0x53, 0x06, 0x73, 0x79, 0xb8, 0xd7, 0xbe, 0x5d, 0x6e, 0xa0,
|
|
0x1f, 0x03, 0x46, 0x07, 0xd6, 0x48, 0x65, 0x4f, 0x9a, 0x8f,
|
|
0x1a, 0xe4, 0xa7, 0x7f, 0x13, 0xc9, 0x06, 0x8c, 0x57, 0xbd,
|
|
0x79, 0x5e, 0x3d, 0x21, 0x02, 0x13, 0x5f, 0xae, 0xc9, 0x45 };
|
|
|
|
ASSERT_EQ(expected, Adapter::Encrypt(key, data));
|
|
}
|
|
|
|
{
|
|
std::vector<uint8_t> data = { 0x51, 0x89, 0x9f, 0x0c, 0x32, 0x9b, 0x1d };
|
|
uint8_t key[] = { 0x28, 0x1c, 0xf3, 0x11, 0xce, 0xc6, 0xc2, 0x38 };
|
|
|
|
std::vector<uint8_t> expected = { 0x64, 0x32, 0x13, 0xe7, 0x31, 0x06, 0xc6, 0x6f };
|
|
|
|
ASSERT_EQ(expected, Adapter::Encrypt({ key, key + std::size(key) }, data));
|
|
}
|
|
}
|
|
|
|
TEST(BlockAdapterTests, InvalidKeyTest)
|
|
{
|
|
using Adapter = BlockAdapter<EcbMode<Des::DesCrypt, PadderPkcs7>>;
|
|
|
|
{
|
|
std::vector<uint8_t> key = { 0x01 };
|
|
|
|
ASSERT_THROW_EX(Adapter::Key{ key },
|
|
Chaos::Service::ChaosException,
|
|
{
|
|
ASSERT_EQ("DesCrypt::Key: invalid key length (8 bytes required)",
|
|
ex.GetMessage());
|
|
});
|
|
|
|
ASSERT_THROW_EX((Adapter::Key{ key.begin(), key.end() }),
|
|
Chaos::Service::ChaosException,
|
|
{
|
|
ASSERT_EQ("DesCrypt::Key: invalid key length (8 bytes required)",
|
|
ex.GetMessage());
|
|
});
|
|
}
|
|
|
|
{
|
|
std::vector<uint8_t> data = { 0x51, 0x89, 0x9f, 0x0c, 0x32, 0x9b, 0x1d };
|
|
std::vector<uint8_t> key = { 0xe9, 0xeb, 0x11, 0x1f, 0xfa, 0xb3, 0x39, 0x95, 0xf0, 0xb9 };
|
|
|
|
ASSERT_THROW_EX(Adapter::Encrypt(key, data),
|
|
Chaos::Service::ChaosException,
|
|
{
|
|
ASSERT_EQ("DesCrypt::Key: invalid key length (8 bytes required)",
|
|
ex.GetMessage());
|
|
});
|
|
}
|
|
|
|
{
|
|
std::vector<uint8_t> data = { 0x51, 0x89, 0x9f, 0x0c, 0x32, 0x9b, 0x1d };
|
|
uint8_t key[] = { 0x28, 0x1c, 0xf3, 0x11, 0xce, 0xc6, 0xc2 };
|
|
|
|
ASSERT_THROW_EX(Adapter::Encrypt({ key, key + std::size(key) }, data),
|
|
Chaos::Service::ChaosException,
|
|
{
|
|
ASSERT_EQ("DesCrypt::Key: invalid key length (8 bytes required)",
|
|
ex.GetMessage());
|
|
});
|
|
}
|
|
}
|