tests/test_bfstream.cpp

   1 #include "tut/tut.hpp"
   2
   3 #include "longstring.h"
   4
   5 #include "stg/bfstream.h"
   6 #include "stg/os_int.h"
   7
   8 #include <algorithm>
   9 #include <string>
  10 #include <cstring>
  11
  12 namespace
  13 {
  14
  15 class TRACKER
  16 {
  17     public:
  18         TRACKER() : m_lastSize(0), m_callCount(0), m_lastBlock(NULL) {}
  19         ~TRACKER() { delete[] m_lastBlock; }
  20         bool Call(const void * block, size_t size)
  21         {
  22         delete[] m_lastBlock;
  23         if (size > 0)
  24             {
  25             m_lastBlock = new char[size];
  26             memcpy(m_lastBlock,  block, size);
  27             m_result.append(m_lastBlock, size);
  28             }
  29         else
  30             m_lastBlock = NULL;
  31         m_lastSize = size;
  32         ++m_callCount;
  33         return true;
  34         }
  35         size_t LastSize() const { return m_lastSize; }
  36         size_t CallCount() const { return m_callCount; }
  37         const void * LastBlock() const { return m_lastBlock; }
  38
  39         const std::string& Result() const { return m_result; }
  40
  41     private:
  42         size_t m_lastSize;
  43         size_t m_callCount;
  44         char * m_lastBlock;
  45
  46         std::string m_result;
  47 };
  48
  49 bool DecryptCallback(const void * block, size_t size, void * data);
  50
  51 class Decryptor
  52 {
  53     public:
  54         Decryptor(const std::string & key)
  55             : m_stream(key, DecryptCallback, this)
  56         {}
  57
  58         bool Call(const void * block, size_t size)
  59         {
  60             m_stream.Put(block, size);
  61             return true;
  62         }
  63
  64         bool Put(const void * block, size_t size)
  65         {
  66             const char * data = static_cast<const char *>(block);
  67             size = strnlen(data, size);
  68             m_result.append(data, size);
  69             return true;
  70         }
  71
  72         void Flush()
  73         {
  74             m_stream.Put(NULL, 0);
  75         }
  76
  77         const std::string & Result() const { return m_result; }
  78
  79     private:
  80         STG::DECRYPT_STREAM m_stream;
  81         std::string m_result;
  82 };
  83
  84 bool EncryptCallback(const void * block, size_t size, void * data)
  85 {
  86 Decryptor & decryptor = *static_cast<Decryptor *>(data);
  87 return decryptor.Call(block, size);
  88 }
  89
  90 bool DecryptCallback(const void * block, size_t size, void * data)
  91 {
  92 Decryptor & decryptor = *static_cast<Decryptor *>(data);
  93 return decryptor.Put(block, size);
  94 }
  95
  96 bool Callback(const void * block, size_t size, void * data)
  97 {
  98 TRACKER & tracker = *static_cast<TRACKER *>(data);
  99 return tracker.Call(block, size);
 100 }
 101
 102 }
 103
 104 namespace tut
 105 {
 106     struct bfstream_data {
 107     };
 108
 109     typedef test_group<bfstream_data> tg;
 110     tg bfstream_test_group("BFStream tests group");
 111
 112     typedef tg::object testobject;
 113
 114     template<>
 115     template<>
 116     void testobject::test<1>()
 117     {
 118         set_test_name("Check bfstream mechanics");
 119
 120         TRACKER tracker;
 121         STG::ENCRYPT_STREAM stream("pr7Hhen", Callback, &tracker);
 122         ensure_equals("CallCount() == 0 after construction", tracker.CallCount(), 0);
 123
 124         uint32_t block[2] = {0x12345678, 0x87654321};
 125         stream.Put(&block[0], sizeof(block[0]));
 126         ensure_equals("CallCount() == 0 after first put", tracker.CallCount(), 0);
 127         stream.Put(&block[1], sizeof(block[1]));
 128         ensure_equals("CallCount() == 1 after second put", tracker.CallCount(), 1);
 129
 130         uint32_t block2[4] = {0x12345678, 0x87654321, 0x12345678, 0x87654321};
 131         stream.Put(&block2[0], sizeof(block2[0]) * 3);
 132         ensure_equals("CallCount() == 2 after third put", tracker.CallCount(), 2);
 133         stream.Put(&block2[3], sizeof(block2[3]));
 134         ensure_equals("CallCount() == 3 after fourth put", tracker.CallCount(), 3);
 135     }
 136
 137     template<>
 138     template<>
 139     void testobject::test<2>()
 140     {
 141         set_test_name("Check bfstream encryption");
 142
 143         TRACKER tracker;
 144         STG::ENCRYPT_STREAM stream("pr7Hhen", Callback, &tracker);
 145
 146         uint32_t block[2] = {0x12345678, 0x87654321};
 147         stream.Put(&block[0], sizeof(block[0]));
 148         ensure_equals("LastSize() == 0 after first put", tracker.LastSize(), 0);
 149         ensure_equals("LastBlock() == NULL after first put", tracker.LastBlock(), static_cast<const void *>(NULL));
 150         stream.Put(&block[1], sizeof(block[1]));
 151         ensure_equals("LastSize() == 8 after second put", tracker.LastSize(), 8);
 152         const uint32_t * ptr = static_cast<const uint32_t *>(tracker.LastBlock());
 153         ensure_equals("ptr[0] == 0xd3988cd after second put", ptr[0], 0xd3988cd);
 154         ensure_equals("ptr[1] == 0x7996c6d6 after second put", ptr[1], 0x7996c6d6);
 155
 156         uint32_t block2[4] = {0x12345678, 0x87654321, 0x12345678, 0x87654321};
 157         stream.Put(&block2[0], sizeof(block2[0]) * 3);
 158         ensure_equals("LastSize() == 8 after third put", tracker.LastSize(), 8);
 159         ptr = static_cast<const uint32_t *>(tracker.LastBlock());
 160         ensure_equals("ptr[0] == 0xd3988cd after third put", ptr[0], 0xd3988cd);
 161         ensure_equals("ptr[1] == 0x7996c6d6 after third put", ptr[1], 0x7996c6d6);
 162
 163         stream.Put(&block2[3], sizeof(block2[3]));
 164         ensure_equals("LastSize() == 8 after fourth put", tracker.LastSize(), 8);
 165         ptr = static_cast<const uint32_t *>(tracker.LastBlock());
 166         ensure_equals("ptr[0] == 0xd3988cd after fourth put", ptr[0], 0xd3988cd);
 167         ensure_equals("ptr[1] == 0x7996c6d6 after fourth put", ptr[1], 0x7996c6d6);
 168     }
 169
 170     template<>
 171     template<>
 172     void testobject::test<3>()
 173     {
 174         set_test_name("Check bfstream long string processing");
 175
 176         TRACKER tracker;
 177         STG::ENCRYPT_STREAM estream("pr7Hhen", Callback, &tracker);
 178         std::string source = "This is a test long string for checking stream encryption/decryption. \"abcdefghijklmnopqrstuvwxyz 0123456789 ABCDEFGHIJKLMNOPQRSTUVWXYZ\"";
 179         char buffer[source.length() + 9];
 180         memset(buffer, 0, sizeof(buffer));
 181
 182         estream.Put(source.c_str(), source.length() + 1, true);
 183         ensure("Encryption long string LastSize()", tracker.LastSize() >= source.length() + 1);
 184         ensure("Encryption long string LastBlock() != NULL", tracker.LastBlock() != NULL);
 185         memcpy(buffer, tracker.LastBlock(), std::min(tracker.LastSize(), sizeof(buffer)));
 186
 187         STG::DECRYPT_STREAM dstream("pr7Hhen", Callback, &tracker);
 188         dstream.Put(buffer, sizeof(buffer), true);
 189         ensure("Decryption long string LastSize() decryption", tracker.LastSize() >= sizeof(buffer));
 190         ensure("Decryption long string LastBlock() != NULL", tracker.LastBlock() != NULL);
 191         memcpy(buffer, tracker.LastBlock(), std::min(tracker.LastSize(), sizeof(buffer)));
 192
 193         ensure_equals("Decrypt(Encrypt(source)) == source", std::string(buffer), source);
 194     }
 195
 196     template<>
 197     template<>
 198     void testobject::test<4>()
 199     {
 200         set_test_name("Check bfstream very long string processing");
 201
 202         Decryptor decryptor("pr7Hhen");
 203         STG::ENCRYPT_STREAM estream("pr7Hhen", EncryptCallback, &decryptor);
 204         //char buffer[source.length() + 9];
 205         //memset(buffer, 0, sizeof(buffer));
 206
 207         estream.Put(longString.c_str(), longString.length() + 1, true);
 208         //ensure("Encryption long string LastSize()", tracker.LastSize() >= source.length() + 1);
 209         //ensure("Encryption long string LastBlock() != NULL", tracker.LastBlock() != NULL);
 210         //memcpy(buffer, tracker.LastBlock(), std::min(tracker.LastSize(), sizeof(buffer)));
 211
 212         //dstream.Put(buffer, sizeof(buffer), true);
 213         //ensure("Decryption long string LastSize() decryption", tracker.LastSize() >= sizeof(buffer));
 214         //ensure("Decryption long string LastBlock() != NULL", tracker.LastBlock() != NULL);
 215         //memcpy(buffer, tracker.LastBlock(), std::min(tracker.LastSize(), sizeof(buffer)));
 216
 217         ensure_equals("Decrypt(Encrypt(source)) == source", decryptor.Result(), longString);
 218     }
 219
 220     template<>
 221     template<>
 222     void testobject::test<5>()
 223     {
 224         set_test_name("Check bfstream mechanics");
 225
 226         TRACKER tracker;
 227         STG::ENCRYPT_STREAM stream("pr7Hhen", Callback, &tracker);
 228         ensure_equals("CallCount() == 0 after construction", tracker.CallCount(), 0);
 229
 230         uint32_t block[2] = {0x12345678, 0x87654321};
 231         stream.Put(&block[0], sizeof(block[0]));
 232         ensure_equals("CallCount() == 0 after first put", tracker.CallCount(), 0);
 233         stream.Put(&block[1], sizeof(block[1]));
 234         ensure_equals("CallCount() == 1 after second put", tracker.CallCount(), 1);
 235         stream.Put(&block[0], 0, true); // Check last callback
 236         ensure_equals("CallCount() == 2 after third (null) put", tracker.CallCount(), 2);
 237     }
 238
 239 }