Move Arc4 to the new Cipher/Stream subdirectory.

Chaos/Cipher/Stream/Arc4/Arc4Crypt.hpp

@@ -0,0 +1,88 @@
+#ifndef CHAOS_CIPHER_STREAM_ARC4_ARC4CRYPT_HPP
+#define CHAOS_CIPHER_STREAM_ARC4_ARC4CRYPT_HPP
+
+#include <cstdint>
+#include <algorithm>
+
+#include "Arc4Gen.hpp"
+#include "Service/SeArray.hpp"
+#include "Service/ChaosException.hpp"
+
+namespace Chaos::Cipher::Stream::Arc4
+{
+
+class Arc4Crypt
+{
+public:
+    Arc4Crypt()
+        : IsInitialized_(false)
+    { }
+
+    template<typename InputIt>
+    Arc4Crypt(InputIt keyBegin, InputIt keyEnd)
+    {
+        RekeyImpl(keyBegin, keyEnd);
+    }
+
+    template<typename InputIt>
+    void Rekey(InputIt keyBegin, InputIt keyEnd)
+    {
+        RekeyImpl(keyBegin, keyEnd);
+    }
+
+    template<typename OutputIt, typename InputIt>
+    void Encrypt(OutputIt out, InputIt in, uint64_t count)
+    {
+        EnsureInitialized();
+        EncryptDecryptImpl(out, in, count);
+    }
+
+    template<typename OutputIt, typename InputIt>
+    void Decrypt(OutputIt out, InputIt in, uint64_t count)
+    {
+        EnsureInitialized();
+        EncryptDecryptImpl(out, in, count);
+    }
+
+private:
+    Arc4Gen Gen_;
+    bool IsInitialized_;
+
+    void EnsureInitialized() const
+    {
+        if (!IsInitialized_)
+        {
+            throw Service::ChaosException("Arc4Crypt: not initialized");
+        }
+    }
+
+    template<typename InputIt>
+    void RekeyImpl(InputIt keyBegin, InputIt keyEnd)
+    {
+        Gen_.Rekey(keyBegin, keyEnd);
+        IsInitialized_ = true;
+    }
+
+    template<typename OutputIt, typename InputIt>
+    void EncryptDecryptImpl(OutputIt out, InputIt in, uint64_t count)
+    {
+        Service::SeArray<uint8_t, 512> keyBuf;
+
+        while (count > 0)
+        {
+            uint64_t keyMaterialBytes = std::min<uint64_t>(keyBuf.Size(), count);
+            Gen_.Generate(keyBuf.Begin(), keyMaterialBytes);
+
+            for (auto keyBufIt = keyBuf.Begin();
+                 keyMaterialBytes > 0 && count > 0;
+                 ++keyBufIt, --keyMaterialBytes, --count)
+            {
+                *out++ = *in++ ^ *keyBufIt;
+            }
+        }
+    }
+};
+
+} // namespace Chaos::Cipher::Stream::Arc4
+
+#endif // CHAOS_CIPHER_STREAM_ARC4_ARC4CRYPT_HPP