hashlag/chaos
1
0 Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Arc4Gen: add IsInitialized() method.

Browse Source
This commit is contained in:
hashlag committed 2026-09-20 21:13:30 +03:00
1 parent 58718b3b0a
commit 11834e9e91
2 files changed
+15
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

Chaos/Cipher/Stream/Arc4/Arc4Gen.hpp
+5
View File
@@ -48,6 +48,11 @@ public:
Step(bytesCount);
}
bool IsInitialized() const noexcept
{
return IsInitialized_;
}
private:
std::array<uint8_t, 256> Lookup_;
uint8_t I_;
ChaosTests/Cipher/Stream/Arc4/Arc4GenTests.cpp
+10
View File
@@ -329,6 +329,7 @@ TEST(Arc4GenTests, TooSmallKeyTest)
{
ASSERT_EQ("Arc4Gen: key is too small", ex.GetMessage());
});
ASSERT_FALSE(gen.IsInitialized());
ASSERT_THROW_EX(Arc4Gen(key, key + strlen(key)),
Chaos::Service::ChaosException,
@@ -344,6 +345,7 @@ TEST(Arc4GenTests, UninitializedGenTest)
{
Arc4Gen gen;
ASSERT_FALSE(gen.IsInitialized());
ASSERT_THROW_EX(gen.Generate(out.begin(), out.size()),
Chaos::Service::ChaosException,
@@ -380,12 +382,14 @@ TEST(Arc4GenTests, UninitializedGenTest)
const char * key = "smal";
Arc4Gen gen;
ASSERT_FALSE(gen.IsInitialized());
ASSERT_THROW_EX(gen.Rekey(key, key + strlen(key)),
Chaos::Service::ChaosException,
{
ASSERT_EQ("Arc4Gen: key is too small", ex.GetMessage());
});
ASSERT_FALSE(gen.IsInitialized());
ASSERT_THROW_EX(gen.Generate(out.begin(), out.size()),
Chaos::Service::ChaosException,
@@ -423,6 +427,7 @@ TEST(Arc4GenTests, RekeyFailSafetyTest)
{
uint8_t key[] = { 0x01, 0x02, 0x03, 0x04, 0x05 };
Arc4Gen gen(key, key + std::size(key));
ASSERT_TRUE(gen.IsInitialized());
{
std::array<uint8_t, 2> out = {};
@@ -436,12 +441,15 @@ TEST(Arc4GenTests, RekeyFailSafetyTest)
{
std::array<uint8_t, 2> out = {};
ASSERT_TRUE(gen.IsInitialized());
uint8_t smallKey[] = { 0x01, 0x02, 0x03 };
ASSERT_THROW_EX(gen.Rekey(smallKey, smallKey + std::size(smallKey)),
Chaos::Service::ChaosException,
{
ASSERT_EQ("Arc4Gen: key is too small", ex.GetMessage());
});
ASSERT_FALSE(gen.IsInitialized());
ASSERT_THROW_EX(gen.Generate(out.begin(), out.size()),
Chaos::Service::ChaosException,
@@ -457,7 +465,9 @@ TEST(Arc4GenTests, RekeyFailSafetyTest)
}
{
ASSERT_FALSE(gen.IsInitialized());
gen.Rekey(key, key + std::size(key));
ASSERT_TRUE(gen.IsInitialized());
std::array<uint8_t, 5> out = {};
std::array<uint8_t, 5> expected = { 0xb2, 0x39, 0x63, 0x05, 0xf0 };