| 1 | 
#include <iostream> | 
| 2 | 
#include <string> | 
| 3 | 
 | 
| 4 | 
#include "io/basic_ifstrstream.hpp" | 
| 5 | 
#include "ZipstreamTestCase.hpp" | 
| 6 | 
 | 
| 7 | 
// Registers the fixture into the 'registry' | 
| 8 | 
 | 
| 9 | 
using namespace std; | 
| 10 | 
using namespace oopse; | 
| 11 | 
CPPUNIT_TEST_SUITE_REGISTRATION( ZipstreamTeseCase ); | 
| 12 | 
 | 
| 13 | 
#include "zipstream.hpp" | 
| 14 | 
#include <iostream> | 
| 15 | 
#include <string> | 
| 16 | 
#include <sstream> | 
| 17 | 
#include <fstream> | 
| 18 | 
#include <cmath> | 
| 19 | 
#include <cstdlib> | 
| 20 | 
 | 
| 21 | 
using namespace std; | 
| 22 | 
using namespace zlib_stream; | 
| 23 | 
 | 
| 24 | 
// a dummy class | 
| 25 | 
class dummy | 
| 26 | 
{ | 
| 27 | 
public: | 
| 28 | 
        dummy(float f_ = 3.14f, int i_= 1) : f(f_), i(i_) | 
| 29 | 
        {}; | 
| 30 | 
        void reset(){f=0.0;i=0;}; | 
| 31 | 
 | 
| 32 | 
        friend ostream& operator << ( ostream& out, dummy const& d) | 
| 33 | 
        { | 
| 34 | 
                out<<" "<<d.f<<" "<<d.i; | 
| 35 | 
                return out; | 
| 36 | 
        } | 
| 37 | 
        friend istream& operator >> ( istream& in, dummy& d) | 
| 38 | 
        { | 
| 39 | 
                in>>d.f>>d.i; | 
| 40 | 
                return in; | 
| 41 | 
        } | 
| 42 | 
 | 
| 43 | 
        friend wostream& operator << ( wostream& out, dummy const& d) | 
| 44 | 
        { | 
| 45 | 
                out<<L" "<<d.f<<L" "<<d.i; | 
| 46 | 
                return out; | 
| 47 | 
        } | 
| 48 | 
        friend wistream& operator >> ( wistream& in, dummy& d) | 
| 49 | 
        { | 
| 50 | 
                in>>d.f>>d.i; | 
| 51 | 
                return in; | 
| 52 | 
        } | 
| 53 | 
protected: | 
| 54 | 
        float f; | 
| 55 | 
        int i; | 
| 56 | 
}; | 
| 57 | 
 | 
| 58 | 
void ZipstreamTeseCase::test_buffer_to_buffer() | 
| 59 | 
{ | 
| 60 | 
        const size_t n= 1024; | 
| 61 | 
        char in_buffer[n]={'\0'}; | 
| 62 | 
        char zip_buffer[n]={'\0'}; | 
| 63 | 
 | 
| 64 | 
        for (size_t i=0;i<n-1;++i) | 
| 65 | 
                in_buffer[i]=static_cast<char>(48+i%48); | 
| 66 | 
 | 
| 67 | 
 | 
| 68 | 
        ostringstream out; | 
| 69 | 
        zip_ostream zipper(out); | 
| 70 | 
 | 
| 71 | 
        zipper.write(in_buffer, n); | 
| 72 | 
        if (zipper.fail()) | 
| 73 | 
                cerr<<"failed to write stream"<<endl; | 
| 74 | 
 | 
| 75 | 
        zipper.zflush(); | 
| 76 | 
 | 
| 77 | 
 | 
| 78 | 
        istringstream in( out.str() ); | 
| 79 | 
        zip_istream unzipper(in); | 
| 80 | 
        unzipper.read(zip_buffer, n); | 
| 81 | 
 | 
| 82 | 
        cerr<<"buffers equals: "<<endl | 
| 83 | 
                <<"-----------------"<<endl | 
| 84 | 
                 <<"\t crc source:"<<crc32(0L,(unsigned char*)in_buffer,n)<<endl | 
| 85 | 
                 <<"\t crc result:"<<crc32(0L,(unsigned char*)zip_buffer,n)<<endl; | 
| 86 | 
 | 
| 87 | 
} | 
| 88 | 
 | 
| 89 | 
void ZipstreamTeseCase::test_wbuffer_to_wbuffer() | 
| 90 | 
{ | 
| 91 | 
        const size_t n= 1024; | 
| 92 | 
        wchar_t in_buffer[n]={'\0'}; | 
| 93 | 
        wchar_t zip_buffer[n]={'\0'}; | 
| 94 | 
 | 
| 95 | 
        for (size_t i=0;i<n-1;++i) | 
| 96 | 
                in_buffer[i]=static_cast<wchar_t>(48+i%48); | 
| 97 | 
 | 
| 98 | 
 | 
| 99 | 
        wostringstream out; | 
| 100 | 
        zip_wostream zipper(out); | 
| 101 | 
 | 
| 102 | 
        zipper.write(in_buffer, n); | 
| 103 | 
        if (zipper.fail()) | 
| 104 | 
                cerr<<"failed to write stream"<<endl; | 
| 105 | 
 | 
| 106 | 
        zipper.zflush(); | 
| 107 | 
 | 
| 108 | 
 | 
| 109 | 
        wistringstream in( out.str() ); | 
| 110 | 
        zip_wistream unzipper(in); | 
| 111 | 
        unzipper.read(zip_buffer, n); | 
| 112 | 
 | 
| 113 | 
        wcerr<<L"buffers equals: "<<endl | 
| 114 | 
                 <<L"-----------------"<<endl | 
| 115 | 
                 <<L"\t crc source:"<<crc32(0L,(unsigned char*)in_buffer,n*sizeof(wchar_t))<<endl | 
| 116 | 
                 <<L"\t crc result:"<<crc32(0L,(unsigned char*)zip_buffer,n*sizeof(wchar_t))<<endl; | 
| 117 | 
} | 
| 118 | 
 | 
| 119 | 
void ZipstreamTeseCase::test_string_string() | 
| 120 | 
{ | 
| 121 | 
        // create some test values | 
| 122 | 
        char c_r='-',c= 'a'; | 
| 123 | 
        string s_r="",s = "std::string"; | 
| 124 | 
        double d_r=0,d = asin( static_cast<double>(1.0) ) *2.0; | 
| 125 | 
        float f_r=0, f = static_cast<float>(asin( 1.0 ) *2.0); | 
| 126 | 
        unsigned int ui_r=0,ui = rand(); | 
| 127 | 
        unsigned long ul_r=0,ul = rand(); | 
| 128 | 
        unsigned short us_r=0, us = rand(); | 
| 129 | 
        dummy dum,dum_r(0,0); | 
| 130 | 
 | 
| 131 | 
        /*---------------------------------------------------------------------- | 
| 132 | 
         | 
| 133 | 
        Testing string to string zipping/unzipping | 
| 134 | 
         | 
| 135 | 
        -------------------------------------------------------------------------*/ | 
| 136 | 
        // creating the target zip string, could be a fstream | 
| 137 | 
        ostringstream ostringstream_(ios::out ); | 
| 138 | 
        // creating the zip layer | 
| 139 | 
        zip_ostream zipper(ostringstream_); | 
| 140 | 
 | 
| 141 | 
        // writing data | 
| 142 | 
        zipper<<f<<" "<<d<<" "<<ui<<" "<<ul<<" "<<us<<" "<<c<<" "<<dum; | 
| 143 | 
        // zip ostream needs special flushing... | 
| 144 | 
        zipper.zflush(); | 
| 145 | 
 | 
| 146 | 
        // create a stream on zip string | 
| 147 | 
        istringstream istringstream_( ostringstream_.str(), ios::in); | 
| 148 | 
        // create unzipper istream | 
| 149 | 
        zip_istream unzipper( istringstream_); | 
| 150 | 
 | 
| 151 | 
        // unzipping | 
| 152 | 
        unzipper>>f_r>>d_r>>ui_r>>ul_r>>us_r>>c_r>>dum_r; | 
| 153 | 
 | 
| 154 | 
        // ouputing results | 
| 155 | 
        cerr<<"tests string-string,  char:"<<endl | 
| 156 | 
                <<"----------------------------"<<endl | 
| 157 | 
                <<"double : "<<d<<" "<<d_r<<endl | 
| 158 | 
                <<"char : "<<c<<" "<<c_r<<endl | 
| 159 | 
                <<"float : "<<f<<" "<<f_r<<endl | 
| 160 | 
                <<"unsigned int : "<<ui<<" "<<ui_r<<endl | 
| 161 | 
                <<"unsigned long : "<<ul<<" "<<ul_r<<endl | 
| 162 | 
                <<"unsigned short : "<<us<<" "<<us_r<<endl | 
| 163 | 
                <<"dummy class: "<<dum<<" "<<dum_r<<endl | 
| 164 | 
                <<endl | 
| 165 | 
                ; | 
| 166 | 
} | 
| 167 | 
 | 
| 168 | 
void ZipstreamTeseCase::test_wstring_wstring() | 
| 169 | 
{ | 
| 170 | 
        // create some test values | 
| 171 | 
        char c_r='-',c= 'a'; | 
| 172 | 
        double d_r=0,d = asin( 1.0 ) *2.0; | 
| 173 | 
        float f_r=0, f = static_cast<float>(asin( 1.0 ) *2.0); | 
| 174 | 
        unsigned int ui_r=0,ui = rand(); | 
| 175 | 
        unsigned long ul_r=0,ul = rand(); | 
| 176 | 
        unsigned short us_r=0, us = rand(); | 
| 177 | 
        dummy dum,dum_r(0,0); | 
| 178 | 
 | 
| 179 | 
        /* | 
| 180 | 
 | 
| 181 | 
        Testing wide characters | 
| 182 | 
 | 
| 183 | 
        */ | 
| 184 | 
        f_r=0.0f; | 
| 185 | 
        d_r=0.0; | 
| 186 | 
        ui_r=ul_r=us_r=0; | 
| 187 | 
        dum_r.reset(); | 
| 188 | 
        // creating the target zip string, could be a fstream | 
| 189 | 
        wostringstream wostringstream_; | 
| 190 | 
        // creating the zip layer | 
| 191 | 
        zip_wostream wzipper(wostringstream_); | 
| 192 | 
 | 
| 193 | 
        // writing data | 
| 194 | 
        wzipper<<f<<L" "<<d<<L" "<<ui<<L" "<<ul<<L" "<<us<<L" "<<dum; | 
| 195 | 
        // zip ostream needs special flushing... | 
| 196 | 
        wzipper.zflush(); | 
| 197 | 
 | 
| 198 | 
 | 
| 199 | 
        // create a stream on zip string | 
| 200 | 
        wistringstream wistringstream_( wostringstream_.str()); | 
| 201 | 
        // create unzipper istream | 
| 202 | 
        zip_wistream wunzipper( wistringstream_ ); | 
| 203 | 
 | 
| 204 | 
        // unzipping | 
| 205 | 
        wunzipper>>f_r>>d_r>>ui_r>>ul_r>>us_r>>dum_r; | 
| 206 | 
 | 
| 207 | 
        // ouputing results | 
| 208 | 
        cerr<<"tests string-string (wchar_t):"<<endl | 
| 209 | 
                <<"------------------------------"<<endl | 
| 210 | 
                <<"double : "<<d<<" "<<d_r<<endl | 
| 211 | 
                <<"float : "<<f<<" "<<f_r<<endl | 
| 212 | 
                <<"unsigned int : "<<ui<<" "<<ui_r<<endl | 
| 213 | 
                <<"unsigned long : "<<ul<<" "<<ul_r<<endl | 
| 214 | 
                <<"unsigned short : "<<us<<" "<<us_r<<endl | 
| 215 | 
                <<"dummy class: "<<dum<<" "<<dum_r<<endl | 
| 216 | 
                <<endl | 
| 217 | 
                ; | 
| 218 | 
} | 
| 219 | 
 | 
| 220 | 
void ZipstreamTeseCase::test_file_file() | 
| 221 | 
{ | 
| 222 | 
      bool add_gzip_header = true; | 
| 223 | 
        // create some test values | 
| 224 | 
        char c_r='-',c= 'a'; | 
| 225 | 
        const char* sz = "const char*"; | 
| 226 | 
        string s_r="",s = "std::string"; | 
| 227 | 
        double d_r=0,d = asin( 1.0 ) *2.0; | 
| 228 | 
        float f_r=0, f = static_cast<float>(asin( 1.0 ) *2.0); | 
| 229 | 
        unsigned int ui_r=0,ui = rand(); | 
| 230 | 
        unsigned long ul_r=0,ul = rand(); | 
| 231 | 
        unsigned short us_r=0, us = rand(); | 
| 232 | 
        dummy dum,dum_r(0,0); | 
| 233 | 
        char sbuf[256]={'-'}; | 
| 234 | 
        long crc_z(0), crc_uz(0); | 
| 235 | 
        long in_size_z(0), in_size_uz(0), out_size_z(0), out_size_uz(0); | 
| 236 | 
 | 
| 237 | 
        /*---------------------------------------------------------------------------- | 
| 238 | 
         | 
| 239 | 
        Testing file to file unzipping | 
| 240 | 
 | 
| 241 | 
        ------------------------------------------------------------------------------*/ | 
| 242 | 
        f_r=0.0f; | 
| 243 | 
        d_r=0.0; | 
| 244 | 
        ui_r=ul_r=us_r=0; dum_r.reset(); | 
| 245 | 
 | 
| 246 | 
        { | 
| 247 | 
                // creating the target zip string, could be a fstream | 
| 248 | 
        ofstream ofstream_("test.zip",ios::out | ios::binary ); | 
| 249 | 
                // creating the zip layer | 
| 250 | 
        zip_ostream fzipper(ofstream_, add_gzip_header); | 
| 251 | 
                // writing data | 
| 252 | 
                fzipper<<f<<" "<<d<<" "<<ui<<" "<<ul<<" "<<us<<" "<<c<<" "<<dum; | 
| 253 | 
                // zip ostream needs special flushing... | 
| 254 | 
                fzipper.zflush(); | 
| 255 | 
                crc_z = fzipper.rdbuf()->get_crc(); | 
| 256 | 
                in_size_z = fzipper.rdbuf()->get_in_size(); | 
| 257 | 
                out_size_z = fzipper.rdbuf()->get_out_size(); | 
| 258 | 
        } | 
| 259 | 
 | 
| 260 | 
        // create a stream on zip string | 
| 261 | 
        ifstream ifstream_; | 
| 262 | 
        ifstream_.open("test.zip", ios::in | ios::binary); | 
| 263 | 
        if (!ifstream_.is_open()) | 
| 264 | 
        { | 
| 265 | 
                cerr<<"Could not open file test.zip"<<endl; | 
| 266 | 
        } | 
| 267 | 
        // create unzipper istream | 
| 268 | 
        zip_istream funzipper( ifstream_); | 
| 269 | 
 | 
| 270 | 
        // unzipping | 
| 271 | 
        funzipper>>f_r>>d_r>>ui_r>>ul_r>>us_r>>c_r>>dum_r; | 
| 272 | 
        funzipper.read_footer(); | 
| 273 | 
 | 
| 274 | 
        crc_uz = funzipper.rdbuf()->get_crc(); | 
| 275 | 
        in_size_uz = funzipper.rdbuf()->get_in_size(); | 
| 276 | 
        out_size_uz = funzipper.rdbuf()->get_out_size(); | 
| 277 | 
 | 
| 278 | 
 | 
| 279 | 
        // ouputing results | 
| 280 | 
        cerr<<"tests file-to-file (char, "<<(funzipper.is_gzip() ? "gzip" : "no gzip")<<"):"<<endl | 
| 281 | 
                <<"------------------------------"<<endl | 
| 282 | 
                <<"double : "<<d<<" "<<d_r<<endl | 
| 283 | 
                <<"char : "<<c<<" "<<c_r<<endl | 
| 284 | 
                <<"float : "<<f<<" "<<f_r<<endl | 
| 285 | 
                <<"unsigned int : "<<ui<<" "<<ui_r<<endl | 
| 286 | 
                <<"unsigned long : "<<ul<<" "<<ul_r<<endl | 
| 287 | 
                <<"unsigned short : "<<us<<" "<<us_r<<endl | 
| 288 | 
                <<"dummy class: "<<dum<<" "<<dum_r<<endl | 
| 289 | 
                <<"crc (zipped, unzipped, from file): "<<crc_z<<" "<<crc_uz<<"  | 
| 290 | 
"<<funzipper.get_gzip_crc()<<endl | 
| 291 | 
                <<"uncompressed data size: "<<in_size_z<<" "<<out_size_uz<<"  | 
| 292 | 
"<<funzipper.get_gzip_data_size()<<endl | 
| 293 | 
                <<"compressed data size: "<<out_size_z<<" "<<in_size_uz<<endl | 
| 294 | 
                <<"check_crc: "<<( funzipper.check_crc() ? "ok" : "failed")<<endl | 
| 295 | 
                <<"check_data_size: "<<( funzipper.check_data_size() ? "ok" : "failed")<<endl | 
| 296 | 
                ; | 
| 297 | 
} | 
| 298 | 
 | 
| 299 | 
void ZipstreamTeseCase::test_crc() | 
| 300 | 
{ | 
| 301 | 
int i; | 
| 302 | 
 | 
| 303 | 
{ | 
| 304 | 
    ofstream outFile("test.gz", ios::out | ios::binary); | 
| 305 | 
    zip_ostream zipOut(outFile,true); | 
| 306 | 
    char buff[102400]; | 
| 307 | 
    for (i = 0; i < 102400; i++) | 
| 308 | 
    { | 
| 309 | 
        buff[i] = (char)48+i%48; | 
| 310 | 
    } | 
| 311 | 
    zipOut.write(buff, sizeof(buff)); | 
| 312 | 
} | 
| 313 | 
 | 
| 314 | 
ifstream inFile("test.gz", ios::in | ios::binary); | 
| 315 | 
 | 
| 316 | 
zip_istream unzipper(inFile); | 
| 317 | 
 | 
| 318 | 
char c; | 
| 319 | 
 | 
| 320 | 
for (i = 0; i < 102400; i++) | 
| 321 | 
{ | 
| 322 | 
    unzipper >> c; | 
| 323 | 
} | 
| 324 | 
 | 
| 325 | 
unzipper.read_footer(); | 
| 326 | 
 | 
| 327 | 
std::cout << "DoMyTest() "  | 
| 328 | 
<< ", outsize " << unzipper.get_out_size() | 
| 329 | 
<< ", insize " << unzipper.get_in_size()  | 
| 330 | 
<< ", z_err " << unzipper.get_zerr()  | 
| 331 | 
<< ", crc " << unzipper.get_crc()  | 
| 332 | 
<< ", gzip data size " << unzipper.get_gzip_data_size() << std::endl; | 
| 333 | 
} |