3 #include "stg/bfstream.h"
4 #include "stg/os_int.h"
16 TRACKER() : m_lastSize(0), m_callCount(0), m_lastBlock(NULL) {}
17 ~TRACKER() { delete[] m_lastBlock; }
18 void Call(const void * block, size_t size)
23 m_lastBlock = new char[size];
24 memcpy(m_lastBlock, block, size);
31 size_t LastSize() const { return m_lastSize; }
32 size_t CallCount() const { return m_callCount; }
33 const void * LastBlock() const { return m_lastBlock; }
41 void Callback(const void * block, size_t size, void * data)
43 TRACKER & tracker = *static_cast<TRACKER *>(data);
44 tracker.Call(block, size);
51 struct bfstream_data {
54 typedef test_group<bfstream_data> tg;
55 tg bfstream_test_group("BFStream tests group");
57 typedef tg::object testobject;
61 void testobject::test<1>()
63 set_test_name("Check bfstream mechanics");
66 STG::ENCRYPT_STREAM stream("pr7Hhen", Callback, &tracker);
67 ensure_equals("CallCount() == 0 after construction", tracker.CallCount(), 0);
69 uint32_t block[2] = {0x12345678, 0x87654321};
70 stream.Put(&block[0], sizeof(block[0]));
71 ensure_equals("CallCount() == 0 after first put", tracker.CallCount(), 0);
72 stream.Put(&block[1], sizeof(block[1]));
73 ensure_equals("CallCount() == 1 after second put", tracker.CallCount(), 1);
75 uint32_t block2[4] = {0x12345678, 0x87654321, 0x12345678, 0x87654321};
76 stream.Put(&block2[0], sizeof(block2[0]) * 3);
77 ensure_equals("CallCount() == 2 after third put", tracker.CallCount(), 2);
78 stream.Put(&block2[3], sizeof(block2[3]));
79 ensure_equals("CallCount() == 3 after fourth put", tracker.CallCount(), 3);
84 void testobject::test<2>()
86 set_test_name("Check bfstream encryption");
89 STG::ENCRYPT_STREAM stream("pr7Hhen", Callback, &tracker);
91 uint32_t block[2] = {0x12345678, 0x87654321};
92 stream.Put(&block[0], sizeof(block[0]));
93 ensure_equals("LastSize() == 0 after first put", tracker.LastSize(), 0);
94 ensure_equals("LastBlock() == NULL after first put", tracker.LastBlock(), static_cast<const void *>(NULL));
95 stream.Put(&block[1], sizeof(block[1]));
96 ensure_equals("LastSize() == 8 after second put", tracker.LastSize(), 8);
97 const uint32_t * ptr = static_cast<const uint32_t *>(tracker.LastBlock());
98 ensure_equals("ptr[0] == 0xd3988cd after second put", ptr[0], 0xd3988cd);
99 ensure_equals("ptr[1] == 0x7996c6d6 after second put", ptr[1], 0x7996c6d6);
101 uint32_t block2[4] = {0x12345678, 0x87654321, 0x12345678, 0x87654321};
102 stream.Put(&block2[0], sizeof(block2[0]) * 3);
103 ensure_equals("LastSize() == 8 after third put", tracker.LastSize(), 8);
104 ptr = static_cast<const uint32_t *>(tracker.LastBlock());
105 ensure_equals("ptr[0] == 0xd3988cd after third put", ptr[0], 0xd3988cd);
106 ensure_equals("ptr[1] == 0x7996c6d6 after third put", ptr[1], 0x7996c6d6);
108 stream.Put(&block2[3], sizeof(block2[3]));
109 ensure_equals("LastSize() == 8 after fourth put", tracker.LastSize(), 8);
110 ptr = static_cast<const uint32_t *>(tracker.LastBlock());
111 ensure_equals("ptr[0] == 0xd3988cd after fourth put", ptr[0], 0xd3988cd);
112 ensure_equals("ptr[1] == 0x7996c6d6 after fourth put", ptr[1], 0x7996c6d6);
117 void testobject::test<3>()
119 set_test_name("Check bfstream long string processing");
122 STG::ENCRYPT_STREAM estream("pr7Hhen", Callback, &tracker);
123 std::string source = "This is a test long string for checking stream encryption/decryption. \"abcdefghijklmnopqrstuvwxyz 0123456789 ABCDEFGHIJKLMNOPQRSTUVWXYZ\"";
124 char buffer[source.length() + 9];
125 memset(buffer, 0, sizeof(buffer));
127 estream.Put(source.c_str(), source.length() + 1, true);
128 ensure("Encryption long string LastSize()", tracker.LastSize() >= source.length() + 1);
129 ensure("Encryption long string LastBlock() != NULL", tracker.LastBlock() != NULL);
130 memcpy(buffer, tracker.LastBlock(), std::min(tracker.LastSize(), sizeof(buffer)));
132 STG::DECRYPT_STREAM dstream("pr7Hhen", Callback, &tracker);
133 dstream.Put(buffer, sizeof(buffer), true);
134 ensure("Decryption long string LastSize() decryption", tracker.LastSize() >= sizeof(buffer));
135 ensure("Decryption long string LastBlock() != NULL", tracker.LastBlock() != NULL);
136 memcpy(buffer, tracker.LastBlock(), std::min(tracker.LastSize(), sizeof(buffer)));
138 ensure_equals("Decrypt(Encrypt(source)) == source", std::string(buffer), source);