Add Arc4 cipher class draft implementation

This commit is contained in:
hashlag
2025-09-16 23:46:24 +03:00
parent 2c21ffea0d
commit e21e36db87
3 changed files with 90 additions and 2 deletions

View File

@@ -15,6 +15,7 @@ set(ChaosTests_SOURCE Hash/Md4HasherTests.cpp
Hash/Md5HasherTests.cpp
Mac/HmacTests.cpp
Cipher/Arc4GenTests.cpp
Cipher/Arc4CryptTests.cpp
Service/SeArrayTests.cpp)
add_executable(ChaosTests ${ChaosTests_SOURCE})

View File

@@ -0,0 +1,65 @@
#include <gtest/gtest.h>
#include <vector>
#include <string>
#include "Cipher/Arc4/Arc4Crypt.hpp"
using namespace Chaos::Cipher::Arc4;
static std::vector<uint8_t> StrToU8Vec(const char * str)
{
std::vector<uint8_t> result;
result.reserve(strlen(str));
for (; *str != '\0'; ++str)
{
result.push_back(static_cast<uint8_t>(*str));
}
return result;
}
static std::string U8VecToStr(const std::vector<uint8_t> & vec)
{
std::string result;
result.reserve(vec.size());
for (auto it = vec.begin(); it != vec.end(); ++it)
{
result.push_back(static_cast<char>(*it));
}
return result;
}
TEST(Arc4CryptTests, BasicTest)
{
const std::vector<uint8_t> key = StrToU8Vec("Secret");
const std::vector<uint8_t> data = StrToU8Vec("Attack at dawn");
std::vector<uint8_t> ciphertext;
ciphertext.resize(data.size());
{
Arc4Crypt arc4;
arc4.Rekey(key.begin(), key.end());
arc4.Encrypt(ciphertext.begin(), data.begin(), data.size());
}
ASSERT_EQ(std::vector<uint8_t>({ 0x45, 0xA0, 0x1F, 0x64, 0x5F, 0xC3, 0x5B,
0x38, 0x35, 0x52, 0x54, 0x4B, 0x9B, 0xF5 }),
ciphertext);
std::vector<uint8_t> recoveredData;
recoveredData.resize(data.size());
{
Arc4Crypt arc4;
arc4.Rekey(key.begin(), key.end());
arc4.Decrypt(recoveredData.begin(), ciphertext.begin(), ciphertext.size());
}
ASSERT_EQ(data, recoveredData);
}