X-Git-Url: https://git.stg.codes/stg.git/blobdiff_plain/27fd26be047af5ccd9d023cf0134fcbf272f18cd..db323584230f6d5b4a6e16574fd0baadc9d465e5:/tests/test_bfstream.cpp?ds=sidebyside

diff --git a/tests/test_bfstream.cpp b/tests/test_bfstream.cpp
new file mode 100644
index 00000000..18884d6d
--- /dev/null
+++ b/tests/test_bfstream.cpp
@@ -0,0 +1,140 @@
+#include "tut/tut.hpp"
+
+#include "stg/bfstream.h"
+#include "stg/os_int.h"
+
+#include <algorithm>
+#include <string>
+#include <cstring>
+
+namespace
+{
+
+class TRACKER
+{
+    public:
+        TRACKER() : m_lastSize(0), m_callCount(0), m_lastBlock(NULL) {}
+        ~TRACKER() { delete[] m_lastBlock; }
+        void Call(const void * block, size_t size)
+        {
+        delete[] m_lastBlock;
+        if (size > 0)
+            {
+            m_lastBlock = new char[size];
+            memcpy(m_lastBlock,  block, size);
+            }
+        else
+            m_lastBlock = NULL;
+        m_lastSize = size;
+        ++m_callCount;
+        }
+        size_t LastSize() const { return m_lastSize; }
+        size_t CallCount() const { return m_callCount; }
+        const void * LastBlock() const { return m_lastBlock; }
+
+    private:
+        size_t m_lastSize;
+        size_t m_callCount;
+        char * m_lastBlock;
+};
+
+void Callback(const void * block, size_t size, void * data)
+{
+TRACKER & tracker = *static_cast<TRACKER *>(data);
+tracker.Call(block, size);
+}
+
+}
+
+namespace tut
+{
+    struct bfstream_data {
+    };
+
+    typedef test_group<bfstream_data> tg;
+    tg bfstream_test_group("BFStream tests group");
+
+    typedef tg::object testobject;
+
+    template<>
+    template<>
+    void testobject::test<1>()
+    {
+        set_test_name("Check bfstream mechanics");
+
+        TRACKER tracker;
+        STG::ENCRYPT_STREAM stream("pr7Hhen", Callback, &tracker);
+        ensure_equals("CallCount() == 0 after construction", tracker.CallCount(), 0);
+
+        uint32_t block[2] = {0x12345678, 0x87654321};
+        stream.Put(&block[0], sizeof(block[0]));
+        ensure_equals("CallCount() == 0 after first put", tracker.CallCount(), 0);
+        stream.Put(&block[1], sizeof(block[1]));
+        ensure_equals("CallCount() == 1 after second put", tracker.CallCount(), 1);
+
+        uint32_t block2[4] = {0x12345678, 0x87654321, 0x12345678, 0x87654321};
+        stream.Put(&block2[0], sizeof(block2[0]) * 3);
+        ensure_equals("CallCount() == 2 after third put", tracker.CallCount(), 2);
+        stream.Put(&block2[3], sizeof(block2[3]));
+        ensure_equals("CallCount() == 3 after fourth put", tracker.CallCount(), 3);
+    }
+
+    template<>
+    template<>
+    void testobject::test<2>()
+    {
+        set_test_name("Check bfstream encryption");
+
+        TRACKER tracker;
+        STG::ENCRYPT_STREAM stream("pr7Hhen", Callback, &tracker);
+
+        uint32_t block[2] = {0x12345678, 0x87654321};
+        stream.Put(&block[0], sizeof(block[0]));
+        ensure_equals("LastSize() == 0 after first put", tracker.LastSize(), 0);
+        ensure_equals("LastBlock() == NULL after first put", tracker.LastBlock(), static_cast<const void *>(NULL));
+        stream.Put(&block[1], sizeof(block[1]));
+        ensure_equals("LastSize() == 8 after second put", tracker.LastSize(), 8);
+        const uint32_t * ptr = static_cast<const uint32_t *>(tracker.LastBlock());
+        ensure_equals("ptr[0] == 0xd3988cd after second put", ptr[0], 0xd3988cd);
+        ensure_equals("ptr[1] == 0x7996c6d6 after second put", ptr[1], 0x7996c6d6);
+
+        uint32_t block2[4] = {0x12345678, 0x87654321, 0x12345678, 0x87654321};
+        stream.Put(&block2[0], sizeof(block2[0]) * 3);
+        ensure_equals("LastSize() == 8 after third put", tracker.LastSize(), 8);
+        ptr = static_cast<const uint32_t *>(tracker.LastBlock());
+        ensure_equals("ptr[0] == 0xd3988cd after third put", ptr[0], 0xd3988cd);
+        ensure_equals("ptr[1] == 0x7996c6d6 after third put", ptr[1], 0x7996c6d6);
+
+        stream.Put(&block2[3], sizeof(block2[3]));
+        ensure_equals("LastSize() == 8 after fourth put", tracker.LastSize(), 8);
+        ptr = static_cast<const uint32_t *>(tracker.LastBlock());
+        ensure_equals("ptr[0] == 0xd3988cd after fourth put", ptr[0], 0xd3988cd);
+        ensure_equals("ptr[1] == 0x7996c6d6 after fourth put", ptr[1], 0x7996c6d6);
+    }
+
+    template<>
+    template<>
+    void testobject::test<3>()
+    {
+        set_test_name("Check bfstream long string processing");
+
+        TRACKER tracker;
+        STG::ENCRYPT_STREAM estream("pr7Hhen", Callback, &tracker);
+        std::string source = "This is a test long string for checking stream encryption/decryption. \"abcdefghijklmnopqrstuvwxyz 0123456789 ABCDEFGHIJKLMNOPQRSTUVWXYZ\"";
+        char buffer[source.length() + 9];
+
+        estream.Put(source.c_str(), source.length() + 1, true);
+        ensure("Encryption long string LastSize()", tracker.LastSize() >= source.length() + 1);
+        ensure("Encryption long string LastBlock() != NULL", tracker.LastBlock() != NULL);
+        memcpy(buffer, tracker.LastBlock(), std::min(tracker.LastSize(), sizeof(buffer)));
+
+        STG::DECRYPT_STREAM dstream("pr7Hhen", Callback, &tracker);
+        dstream.Put(buffer, sizeof(buffer), true);
+        ensure("Decryption long string LastSize() decryption", tracker.LastSize() >= sizeof(buffer));
+        ensure("Decryption long string LastBlock() != NULL", tracker.LastBlock() != NULL);
+        memcpy(buffer, tracker.LastBlock(), std::min(tracker.LastSize(), sizeof(buffer)));
+
+        ensure_equals("Decrypt(Encrypt(source)) == source", std::string(buffer), source);
+    }
+
+}