]> git.saurik.com Git - wxWidgets.git/blame - tests/base64/base64.cpp
A couple of fixes for wxChm
[wxWidgets.git] / tests / base64 / base64.cpp
CommitLineData
4db03d26
VZ
1///////////////////////////////////////////////////////////////////////////////
2// Name: tests/base64/base64.cpp
3// Purpose: wxBase64Encode/Decode unit test
4// Author: Charles Reimers
5// Created: 2007-06-22
6// RCS-ID: $Id$
7///////////////////////////////////////////////////////////////////////////////
8
9// ----------------------------------------------------------------------------
10// headers
11// ----------------------------------------------------------------------------
12
13#include "testprec.h"
14
15#ifdef __BORLANDC__
16 #pragma hdrstop
17#endif
18
19#ifndef WX_PRECOMP
20 #include "wx/wx.h"
21#endif // WX_PRECOMP
22
23#if wxUSE_BASE64
24
25#include "wx/base64.h"
26
27static const char encoded0to255[] =
28 "AAECAwQFBgcICQoLDA0ODxAREhMUFRYXGBkaGxwdHh8gISIj"
29 "JCUmJygpKissLS4vMDEyMzQ1Njc4OTo7PD0+P0BBQkNERUZH"
30 "SElKS0xNTk9QUVJTVFVWV1hZWltcXV5fYGFiY2RlZmdoaWpr"
31 "bG1ub3BxcnN0dXZ3eHl6e3x9fn+AgYKDhIWGh4iJiouMjY6P"
32 "kJGSk5SVlpeYmZqbnJ2en6ChoqOkpaanqKmqq6ytrq+wsbKz"
33 "tLW2t7i5uru8vb6/wMHCw8TFxsfIycrLzM3Oz9DR0tPU1dbX"
34 "2Nna29zd3t/g4eLj5OXm5+jp6uvs7e7v8PHy8/T19vf4+fr7"
35 "/P3+/w==";
36
37static void
38generatePatternedData(void* buff, size_t len, unsigned char startVal,
39 unsigned char addVal, unsigned char multVal = 1,
40 unsigned char xorMask = 0, unsigned char andMask = 255)
41{
42 unsigned char *cbuff = (unsigned char *)buff;
43 unsigned char curval = startVal;
44 while(len--)
45 {
46 *(cbuff++) = curval;
47 curval = (((curval + addVal) * multVal) ^ xorMask) & andMask;
48 }
49}
50
51static void generateRandomData(void* buff, size_t len)
52{
53 unsigned char *cbuff = (unsigned char *)buff;
54 while(len--)
55 {
56 *(cbuff++) = (unsigned char)(((rand() * 255) / RAND_MAX));
57 }
58}
59
60static void generateGibberish(void* buff, size_t len)
61{
d2001a56 62 static const unsigned char cb64[] =
4db03d26
VZ
63 "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+/";
64
65 unsigned char *cbuff = (unsigned char *)buff;
66 while(len--)
67 {
68 *(cbuff++) = cb64[((rand() * 64) / RAND_MAX)];
69 }
70}
71
72// --------------------------------------------------------------------------
73// test class
74// --------------------------------------------------------------------------
75
76class Base64TestCase : public CppUnit::TestCase
77{
78public:
79 Base64TestCase() { }
80
81private:
82 CPPUNIT_TEST_SUITE( Base64TestCase );
83 CPPUNIT_TEST( EncodeDecodeEmpty );
84 CPPUNIT_TEST( EncodeDecodeA );
85 CPPUNIT_TEST( EncodeDecodeAB );
86 CPPUNIT_TEST( EncodeDecodeABC );
87 CPPUNIT_TEST( EncodeDecodeABCD );
88 CPPUNIT_TEST( EncodeDecode0to255 );
89 CPPUNIT_TEST( EncodeDecodePatternA );
90 CPPUNIT_TEST( EncodeDecodePatternB );
91 CPPUNIT_TEST( EncodeDecodePatternC );
92 CPPUNIT_TEST( EncodeDecodeRandom );
7bac0684 93 CPPUNIT_TEST( DecodeInvalid );
4db03d26
VZ
94 CPPUNIT_TEST_SUITE_END();
95
96 void EncodeDecodeEmpty();
97 void EncodeDecodeA();
98 void EncodeDecodeAB();
99 void EncodeDecodeABC();
100 void EncodeDecodeABCD();
101 void EncodeDecode0to255();
102 void EncodeDecodePatternA();
103 void EncodeDecodePatternB();
104 void EncodeDecodePatternC();
105 void EncodeDecodeRandom();
7bac0684 106 void DecodeInvalid();
4db03d26
VZ
107
108 DECLARE_NO_COPY_CLASS(Base64TestCase)
109};
110
111// register in the unnamed registry so that these tests are run by default
112CPPUNIT_TEST_SUITE_REGISTRATION( Base64TestCase );
113
114// also include in it's own registry so that these tests can be run alone
115CPPUNIT_TEST_SUITE_NAMED_REGISTRATION( Base64TestCase, "Base64TestCase" );
116
117void Base64TestCase::EncodeDecodeEmpty()
118{
119 char shouldBeEmpty[10];
120 shouldBeEmpty[0] = '\0';
121 size_t len = 10;
122
123 CPPUNIT_ASSERT(wxBase64Encode(shouldBeEmpty, len, "", 0) != wxCONV_FAILED);
124 CPPUNIT_ASSERT_EQUAL('\0', shouldBeEmpty[0]);
125
126 CPPUNIT_ASSERT(wxBase64Decode(shouldBeEmpty, len, "") != wxCONV_FAILED);
127 CPPUNIT_ASSERT_EQUAL('\0', shouldBeEmpty[0]);
128
129 wxMemoryBuffer bufmt;
ee8c79c1
VZ
130 wxString resultEmpty = wxBase64Encode(bufmt);
131 CPPUNIT_ASSERT(resultEmpty.empty());
4db03d26 132
ee8c79c1 133 bufmt = wxBase64Decode(resultEmpty);
1de532f5 134 CPPUNIT_ASSERT_EQUAL(0, bufmt.GetDataLen());
4db03d26
VZ
135}
136
137void Base64TestCase::EncodeDecodeA()
138{
139 const wxString str = wxBase64Encode("A", 1);
140 CPPUNIT_ASSERT_EQUAL(wxString("QQ=="), str);
141
142 wxMemoryBuffer buf = wxBase64Decode(str);
1de532f5 143 CPPUNIT_ASSERT_EQUAL(1, buf.GetDataLen());
4db03d26 144 CPPUNIT_ASSERT_EQUAL('A', *(char *)buf.GetData());
07890fbe
VZ
145
146 char cbuf[10];
147 memset(cbuf, (char)-1, sizeof(cbuf));
148 CPPUNIT_ASSERT_EQUAL( 1, wxBase64Decode(cbuf, 1, str) );
149 CPPUNIT_ASSERT_EQUAL( 'A', cbuf[0] );
150 CPPUNIT_ASSERT_EQUAL( (char)-1, cbuf[1] );
151 CPPUNIT_ASSERT_EQUAL( (char)-1, cbuf[2] );
4db03d26
VZ
152}
153
154void Base64TestCase::EncodeDecodeAB()
155{
156 const wxString str = wxBase64Encode("AB", 2);
157 CPPUNIT_ASSERT_EQUAL(wxString("QUI="), str);
158
159 wxMemoryBuffer buf = wxBase64Decode(str);
1de532f5 160 CPPUNIT_ASSERT_EQUAL(2, buf.GetDataLen());
4db03d26
VZ
161 CPPUNIT_ASSERT_EQUAL('A', buf[0]);
162 CPPUNIT_ASSERT_EQUAL('B', buf[1]);
07890fbe
VZ
163
164 char cbuf[10];
165 memset(cbuf, (char)-1, sizeof(cbuf));
166 CPPUNIT_ASSERT_EQUAL( wxCONV_FAILED, wxBase64Decode(cbuf, 1, str) );
167 CPPUNIT_ASSERT_EQUAL( 2, wxBase64Decode(cbuf, 2, str) );
168 CPPUNIT_ASSERT_EQUAL( 'A', cbuf[0] );
169 CPPUNIT_ASSERT_EQUAL( 'B', cbuf[1] );
170 CPPUNIT_ASSERT_EQUAL( (char)-1, cbuf[2] );
171 CPPUNIT_ASSERT_EQUAL( (char)-1, cbuf[3] );
4db03d26
VZ
172}
173
174void Base64TestCase::EncodeDecodeABC()
175{
176 const wxString str = wxBase64Encode("ABC", 3);
177 CPPUNIT_ASSERT_EQUAL(wxString("QUJD"), str);
178
179 wxMemoryBuffer buf = wxBase64Decode(str);
1de532f5 180 CPPUNIT_ASSERT_EQUAL(3, buf.GetDataLen());
4db03d26
VZ
181 CPPUNIT_ASSERT_EQUAL('A', buf[0]);
182 CPPUNIT_ASSERT_EQUAL('B', buf[1]);
183 CPPUNIT_ASSERT_EQUAL('C', buf[2]);
07890fbe
VZ
184
185 char cbuf[10];
186 memset(cbuf, (char)-1, sizeof(cbuf));
187 CPPUNIT_ASSERT_EQUAL( wxCONV_FAILED, wxBase64Decode(cbuf, 2, str) );
188 CPPUNIT_ASSERT_EQUAL( 3, wxBase64Decode(cbuf, 3, str) );
189 CPPUNIT_ASSERT_EQUAL( 'A', cbuf[0] );
190 CPPUNIT_ASSERT_EQUAL( 'B', cbuf[1] );
191 CPPUNIT_ASSERT_EQUAL( 'C', cbuf[2] );
192 CPPUNIT_ASSERT_EQUAL( (char)-1, cbuf[3] );
193 CPPUNIT_ASSERT_EQUAL( (char)-1, cbuf[4] );
4db03d26
VZ
194}
195
196void Base64TestCase::EncodeDecodeABCD()
197{
198 const wxString str = wxBase64Encode("ABCD", 4);
199 CPPUNIT_ASSERT_EQUAL(wxString("QUJDRA=="), str);
200
201 wxMemoryBuffer buf = wxBase64Decode(str);
1de532f5 202 CPPUNIT_ASSERT_EQUAL(4, buf.GetDataLen());
4db03d26
VZ
203 CPPUNIT_ASSERT_EQUAL('A', buf[0]);
204 CPPUNIT_ASSERT_EQUAL('B', buf[1]);
205 CPPUNIT_ASSERT_EQUAL('C', buf[2]);
206 CPPUNIT_ASSERT_EQUAL('D', buf[3]);
07890fbe
VZ
207
208 char cbuf[10];
209 memset(cbuf, (char)-1, sizeof(cbuf));
210 CPPUNIT_ASSERT_EQUAL( wxCONV_FAILED, wxBase64Decode(cbuf, 3, str) );
211 CPPUNIT_ASSERT_EQUAL( 4, wxBase64Decode(cbuf, 4, str) );
212 CPPUNIT_ASSERT_EQUAL( 'A', cbuf[0] );
213 CPPUNIT_ASSERT_EQUAL( 'B', cbuf[1] );
214 CPPUNIT_ASSERT_EQUAL( 'C', cbuf[2] );
215 CPPUNIT_ASSERT_EQUAL( 'D', cbuf[3] );
216 CPPUNIT_ASSERT_EQUAL( (char)-1, cbuf[4] );
217 CPPUNIT_ASSERT_EQUAL( (char)-1, cbuf[5] );
4db03d26
VZ
218}
219
220void Base64TestCase::EncodeDecode0to255()
221{
222 unsigned char buff[256];
223 generatePatternedData(buff, 256, 0, 1);
224 wxString str = wxBase64Encode(buff, 256);
225 wxMemoryBuffer mbuff = wxBase64Decode(str);
226 CPPUNIT_ASSERT(memcmp(mbuff.GetData(), buff, mbuff.GetDataLen()) == 0);
227
228 mbuff = wxBase64Decode(encoded0to255);
229 CPPUNIT_ASSERT(memcmp(mbuff.GetData(), buff, mbuff.GetDataLen()) == 0);
230}
231
232void Base64TestCase::EncodeDecodePatternA()
233{
234 unsigned char buff[350];
235 generatePatternedData(buff, 350, 24, 5, 3);
236 wxString str = wxBase64Encode(buff, 350);
237 wxMemoryBuffer mbuff = wxBase64Decode(str);
238 CPPUNIT_ASSERT(memcmp(mbuff.GetData(), buff, mbuff.GetDataLen()) == 0);
239}
240
241void Base64TestCase::EncodeDecodePatternB()
242{
243 unsigned char buff[350];
244 generatePatternedData(buff, 350, 0, 1, 1, 0xAA);
245 wxString str = wxBase64Encode(buff, 350);
246 wxMemoryBuffer mbuff = wxBase64Decode(str);
247 CPPUNIT_ASSERT(memcmp(mbuff.GetData(), buff, mbuff.GetDataLen()) == 0);
248}
249
250void Base64TestCase::EncodeDecodePatternC()
251{
252 unsigned char buff[11];
253 generatePatternedData(buff, 11, 1, 0, 2);
254 wxString str = wxBase64Encode(buff, 11);
255 wxMemoryBuffer mbuff = wxBase64Decode(str);
256 CPPUNIT_ASSERT(memcmp(mbuff.GetData(), buff, mbuff.GetDataLen()) == 0);
257}
258
259void Base64TestCase::EncodeDecodeRandom()
260{
261 size_t size = rand() * 3000 / RAND_MAX + 11;
262 unsigned char *buff = new unsigned char[size];
263 generateRandomData(buff, size);
264 wxString str = wxBase64Encode(buff, size);
265 wxMemoryBuffer mbuff = wxBase64Decode(str);
266 CPPUNIT_ASSERT(memcmp(mbuff.GetData(), buff, mbuff.GetDataLen()) == 0);
267
268 generateGibberish(buff, size);
269 char *buff2 = new char[size];
270 size_t realsize = size;
271 CPPUNIT_ASSERT(wxBase64Decode(buff2, realsize, (char *)buff, size));
272 CPPUNIT_ASSERT(wxBase64Encode(buff2, size, buff2, realsize));
273}
274
7bac0684
VZ
275void Base64TestCase::DecodeInvalid()
276{
277 size_t rc, posErr;
278 rc = wxBase64Decode(NULL, 0, "one two!", wxNO_LEN,
279 wxBase64DecodeMode_Strict, &posErr);
280 CPPUNIT_ASSERT_EQUAL( wxCONV_FAILED, rc);
1de532f5 281 CPPUNIT_ASSERT_EQUAL( 3, posErr );
7bac0684
VZ
282
283 rc = wxBase64Decode(NULL, 0, "one two!", wxNO_LEN,
284 wxBase64DecodeMode_SkipWS, &posErr);
285 CPPUNIT_ASSERT_EQUAL( wxCONV_FAILED, rc);
1de532f5 286 CPPUNIT_ASSERT_EQUAL( 7, posErr );
7bac0684
VZ
287
288 rc = wxBase64Decode(NULL, 0, "? QQ==", wxNO_LEN,
289 wxBase64DecodeMode_SkipWS, &posErr);
290 CPPUNIT_ASSERT_EQUAL( wxCONV_FAILED, rc);
1de532f5 291 CPPUNIT_ASSERT_EQUAL( 0, posErr );
7bac0684
VZ
292
293 posErr = (size_t)-1;
294 rc = wxBase64Decode(NULL, 0, " QQ==", wxNO_LEN,
295 wxBase64DecodeMode_SkipWS, &posErr);
1de532f5
VZ
296 CPPUNIT_ASSERT_EQUAL( 1, rc );
297 CPPUNIT_ASSERT_EQUAL( -1, posErr );
7bac0684
VZ
298
299 rc = wxBase64Decode(NULL, 0, "? QQ==", wxNO_LEN,
300 wxBase64DecodeMode_Relaxed, &posErr);
1de532f5
VZ
301 CPPUNIT_ASSERT_EQUAL( 1, rc );
302 CPPUNIT_ASSERT_EQUAL( -1, posErr );
7bac0684
VZ
303
304 CPPUNIT_ASSERT( !wxBase64Decode("wxGetApp()").GetDataLen() );
305}
306
4db03d26 307#endif // wxUSE_BASE64