X-Git-Url: https://git.stg.codes/stg.git/blobdiff_plain/46b0747592074017ff0ea4b33d4a7194235886e5..2574a28cbf000603bc31f61593dbf061ff56c1d5:/tests/test_bfstream.cpp diff --git a/tests/test_bfstream.cpp b/tests/test_bfstream.cpp index a3252c9e..f6c5bad7 100644 --- a/tests/test_bfstream.cpp +++ b/tests/test_bfstream.cpp @@ -1,9 +1,17 @@ -#include "tut/tut.hpp" +#define BOOST_TEST_MODULE STGBFStream #include "longstring.h" #include "stg/bfstream.h" +#pragma GCC diagnostic push +#pragma GCC diagnostic ignored "-Wold-style-cast" +#pragma GCC diagnostic ignored "-Wunused-parameter" +#pragma GCC diagnostic ignored "-Wsign-compare" +#pragma GCC diagnostic ignored "-Wparentheses" +#include +#pragma GCC diagnostic pop + #include #include #include @@ -12,61 +20,58 @@ namespace { -class TRACKER +class Tracker { public: - TRACKER() : m_lastSize(0), m_callCount(0), m_lastBlock(NULL) {} - ~TRACKER() { delete[] m_lastBlock; } - bool Call(const void * block, size_t size) + 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) + delete[] m_lastBlock; + if (size > 0) { - m_lastBlock = new char[size]; - memcpy(m_lastBlock, block, size); - m_result.append(m_lastBlock, size); + m_lastBlock = new char[size]; + memcpy(m_lastBlock, block, size); + m_result.append(m_lastBlock, size); } - else - m_lastBlock = NULL; - m_lastSize = size; - ++m_callCount; - return true; + 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; } + const void* LastBlock() const { return m_lastBlock; } const std::string& Result() const { return m_result; } private: size_t m_lastSize; size_t m_callCount; - char * m_lastBlock; + char* m_lastBlock; std::string m_result; }; -bool DecryptCallback(const void * block, size_t size, void * data); +bool DecryptCallback(const void* block, size_t size, void* data); class Decryptor { public: - Decryptor(const std::string & key) + Decryptor(const std::string& key) : m_stream(key, DecryptCallback, this) {} - bool Call(const void * block, size_t size) + void Call(const void* block, size_t size) { m_stream.Put(block, size); - return true; } - bool Put(const void * block, size_t size) + void Put(const void* block, size_t size) { - const char * data = static_cast(block); + const auto* data = static_cast(block); size = strnlen(data, size); m_result.append(data, size); - return true; } void Flush() @@ -81,159 +86,123 @@ class Decryptor std::string m_result; }; -bool EncryptCallback(const void * block, size_t size, void * data) +bool EncryptCallback(const void* block, size_t size, void* data) +{ + auto& decryptor = *static_cast(data); + decryptor.Call(block, size); + return true; +} + +bool DecryptCallback(const void* block, size_t size, void* data) { -Decryptor & decryptor = *static_cast(data); -return decryptor.Call(block, size); + auto& decryptor = *static_cast(data); + decryptor.Put(block, size); + return true; +} + +bool Callback(const void* block, size_t size, void* data) +{ + auto& tracker = *static_cast(data); + tracker.Call(block, size); + return true; +} + } -bool DecryptCallback(const void * block, size_t size, void * data) +BOOST_AUTO_TEST_SUITE(BFStream) + +BOOST_AUTO_TEST_CASE(Mechanics) { -Decryptor & decryptor = *static_cast(data); -return decryptor.Put(block, size); + Tracker tracker; + STG::ENCRYPT_STREAM stream("pr7Hhen", Callback, &tracker); + BOOST_CHECK_EQUAL(tracker.CallCount(), 0); + + uint32_t block[2] = {0x12345678, 0x87654321}; + stream.Put(&block[0], sizeof(block[0])); + BOOST_CHECK_EQUAL(tracker.CallCount(), 0); + stream.Put(&block[1], sizeof(block[1])); + BOOST_CHECK_EQUAL(tracker.CallCount(), 1); + + uint32_t block2[4] = {0x12345678, 0x87654321, 0x12345678, 0x87654321}; + stream.Put(&block2[0], sizeof(block2[0]) * 3); + BOOST_CHECK_EQUAL(tracker.CallCount(), 2); + stream.Put(&block2[3], sizeof(block2[3])); + BOOST_CHECK_EQUAL(tracker.CallCount(), 3); } -bool Callback(const void * block, size_t size, void * data) +BOOST_AUTO_TEST_CASE(Encryption) { -TRACKER & tracker = *static_cast(data); -return tracker.Call(block, size); + Tracker tracker; + STG::ENCRYPT_STREAM stream("pr7Hhen", Callback, &tracker); + + uint32_t block[2] = {0x12345678, 0x87654321}; + stream.Put(&block[0], sizeof(block[0])); + BOOST_CHECK_EQUAL(tracker.LastSize(), 0); + BOOST_CHECK_EQUAL(tracker.LastBlock(), static_cast(NULL)); + stream.Put(&block[1], sizeof(block[1])); + BOOST_CHECK_EQUAL(tracker.LastSize(), 8); + const uint32_t * ptr = static_cast(tracker.LastBlock()); + BOOST_CHECK_EQUAL(ptr[0], 0xd3988cd); + BOOST_CHECK_EQUAL(ptr[1], 0x7996c6d6); + + uint32_t block2[4] = {0x12345678, 0x87654321, 0x12345678, 0x87654321}; + stream.Put(&block2[0], sizeof(block2[0]) * 3); + BOOST_CHECK_EQUAL(tracker.LastSize(), 8); + ptr = static_cast(tracker.LastBlock()); + BOOST_CHECK_EQUAL(ptr[0], 0xd3988cd); + BOOST_CHECK_EQUAL(ptr[1], 0x7996c6d6); + + stream.Put(&block2[3], sizeof(block2[3])); + BOOST_CHECK_EQUAL(tracker.LastSize(), 8); + ptr = static_cast(tracker.LastBlock()); + BOOST_CHECK_EQUAL(ptr[0], 0xd3988cd); + BOOST_CHECK_EQUAL(ptr[1], 0x7996c6d6); } +BOOST_AUTO_TEST_CASE(LongStringProcessing) +{ + Tracker tracker; + STG::ENCRYPT_STREAM estream("pr7Hhen", Callback, &tracker); + const std::string source = "This is a test long string for checking stream encryption/decryption. \"abcdefghijklmnopqrstuvwxyz 0123456789 ABCDEFGHIJKLMNOPQRSTUVWXYZ\""; + std::vector buffer(source.length() + 9, 0); + + estream.Put(source.c_str(), source.length() + 1, true); + BOOST_CHECK(tracker.LastSize() >= source.length() + 1); + BOOST_CHECK(tracker.LastBlock() != NULL); + memcpy(buffer.data(), tracker.LastBlock(), std::min(tracker.LastSize(), buffer.size())); + + STG::DECRYPT_STREAM dstream("pr7Hhen", Callback, &tracker); + dstream.Put(buffer.data(), buffer.size(), true); + BOOST_CHECK(tracker.LastSize() >= buffer.size()); + BOOST_CHECK(tracker.LastBlock() != NULL); + memcpy(buffer.data(), tracker.LastBlock(), std::min(tracker.LastSize(), buffer.size())); + + BOOST_CHECK_EQUAL(std::string(buffer.data()), source); } -namespace tut +BOOST_AUTO_TEST_CASE(VeryLongStringProcessing) { - struct bfstream_data { - }; - - typedef test_group 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(NULL)); - stream.Put(&block[1], sizeof(block[1])); - ensure_equals("LastSize() == 8 after second put", tracker.LastSize(), 8); - const uint32_t * ptr = static_cast(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(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(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]; - memset(buffer, 0, sizeof(buffer)); - - 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); - } - - template<> - template<> - void testobject::test<4>() - { - set_test_name("Check bfstream very long string processing"); - - Decryptor decryptor("pr7Hhen"); - STG::ENCRYPT_STREAM estream("pr7Hhen", EncryptCallback, &decryptor); - //char buffer[source.length() + 9]; - //memset(buffer, 0, sizeof(buffer)); - - estream.Put(longString.c_str(), longString.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))); - - //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", decryptor.Result(), longString); - } - - template<> - template<> - void testobject::test<5>() - { - 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); - stream.Put(&block[0], 0, true); // Check last callback - ensure_equals("CallCount() == 2 after third (null) put", tracker.CallCount(), 2); - } + Decryptor decryptor("pr7Hhen"); + STG::ENCRYPT_STREAM estream("pr7Hhen", EncryptCallback, &decryptor); + + estream.Put(longString.c_str(), longString.length() + 1, true); + BOOST_CHECK_EQUAL(decryptor.Result(), longString); } + +BOOST_AUTO_TEST_CASE(Mechanics2) +{ + Tracker tracker; + STG::ENCRYPT_STREAM stream("pr7Hhen", Callback, &tracker); + BOOST_CHECK_EQUAL(tracker.CallCount(), 0); + + uint32_t block[2] = {0x12345678, 0x87654321}; + stream.Put(&block[0], sizeof(block[0])); + BOOST_CHECK_EQUAL(tracker.CallCount(), 0); + stream.Put(&block[1], sizeof(block[1])); + BOOST_CHECK_EQUAL(tracker.CallCount(), 1); + stream.Put(&block[0], 0, true); // Check last callback + BOOST_CHECK_EQUAL(tracker.CallCount(), 2); +} + +BOOST_AUTO_TEST_SUITE_END()