From db323584230f6d5b4a6e16574fd0baadc9d465e5 Mon Sep 17 00:00:00 2001 From: Maxim Mamontov Date: Sun, 22 Jun 2014 14:41:06 +0300 Subject: [PATCH] Added stream encryption/decryption tests. --- tests/Makefile | 1 + tests/test_bfstream.cpp | 140 ++++++++++++++++++++++++++++++++++++++++ 2 files changed, 141 insertions(+) create mode 100644 tests/test_bfstream.cpp diff --git a/tests/Makefile b/tests/Makefile index 0fcc6b3b..ca61b482 100644 --- a/tests/Makefile +++ b/tests/Makefile @@ -21,6 +21,7 @@ SOURCES = main.cpp \ test_fee_charge_rules.cpp \ test_reconnect_on_tariff_change.cpp \ test_crypto.cpp \ + test_bfstream.cpp \ ../projects/stargazer/tariff_impl.cpp \ ../projects/stargazer/user_impl.cpp \ ../projects/stargazer/user_property.cpp \ 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 +#include +#include + +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(data); +tracker.Call(block, size); +} + +} + +namespace tut +{ + 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]; + + 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); + } + +} -- 2.44.2