1#include <stdio.h>
2#include <stdlib.h>
3#include <windows.h>
4
5#include "zlib.h"
6
7
8void MyDoMinus64(LARGE_INTEGER *R,LARGE_INTEGER A,LARGE_INTEGER B)
9{
10    R->HighPart = A.HighPart - B.HighPart;
11    if (A.LowPart >= B.LowPart)
12        R->LowPart = A.LowPart - B.LowPart;
13    else
14    {
15        R->LowPart = A.LowPart - B.LowPart;
16        R->HighPart --;
17    }
18}
19
20#ifdef _M_X64
21// see http://msdn2.microsoft.com/library/twchhe95(en-us,vs.80).aspx for __rdtsc
22unsigned __int64 __rdtsc(void);
23void BeginCountRdtsc(LARGE_INTEGER * pbeginTime64)
24{
25 //   printf("rdtsc = %I64x\n",__rdtsc());
26   pbeginTime64->QuadPart=__rdtsc();
27}
28
29LARGE_INTEGER GetResRdtsc(LARGE_INTEGER beginTime64,BOOL fComputeTimeQueryPerf)
30{
31    LARGE_INTEGER LIres;
32    unsigned _int64 res=__rdtsc()-((unsigned _int64)(beginTime64.QuadPart));
33    LIres.QuadPart=res;
34   // printf("rdtsc = %I64x\n",__rdtsc());
35    return LIres;
36}
37#else
38#ifdef _M_IX86
39void myGetRDTSC32(LARGE_INTEGER * pbeginTime64)
40{
41    DWORD dwEdx,dwEax;
42    _asm
43    {
44        rdtsc
45        mov dwEax,eax
46        mov dwEdx,edx
47    }
48    pbeginTime64->LowPart=dwEax;
49    pbeginTime64->HighPart=dwEdx;
50}
51
52void BeginCountRdtsc(LARGE_INTEGER * pbeginTime64)
53{
54    myGetRDTSC32(pbeginTime64);
55}
56
57LARGE_INTEGER GetResRdtsc(LARGE_INTEGER beginTime64,BOOL fComputeTimeQueryPerf)
58{
59    LARGE_INTEGER LIres,endTime64;
60    myGetRDTSC32(&endTime64);
61
62    LIres.LowPart=LIres.HighPart=0;
63    MyDoMinus64(&LIres,endTime64,beginTime64);
64    return LIres;
65}
66#else
67void myGetRDTSC32(LARGE_INTEGER * pbeginTime64)
68{
69}
70
71void BeginCountRdtsc(LARGE_INTEGER * pbeginTime64)
72{
73}
74
75LARGE_INTEGER GetResRdtsc(LARGE_INTEGER beginTime64,BOOL fComputeTimeQueryPerf)
76{
77    LARGE_INTEGER lr;
78    lr.QuadPart=0;
79    return lr;
80}
81#endif
82#endif
83
84void BeginCountPerfCounter(LARGE_INTEGER * pbeginTime64,BOOL fComputeTimeQueryPerf)
85{
86    if ((!fComputeTimeQueryPerf) || (!QueryPerformanceCounter(pbeginTime64)))
87    {
88        pbeginTime64->LowPart = GetTickCount();
89        pbeginTime64->HighPart = 0;
90    }
91}
92
93DWORD GetMsecSincePerfCounter(LARGE_INTEGER beginTime64,BOOL fComputeTimeQueryPerf)
94{
95    LARGE_INTEGER endTime64,ticksPerSecond,ticks;
96    DWORDLONG ticksShifted,tickSecShifted;
97    DWORD dwLog=16+0;
98    DWORD dwRet;
99    if ((!fComputeTimeQueryPerf) || (!QueryPerformanceCounter(&endTime64)))
100        dwRet = (GetTickCount() - beginTime64.LowPart)*1;
101    else
102    {
103        MyDoMinus64(&ticks,endTime64,beginTime64);
104        QueryPerformanceFrequency(&ticksPerSecond);
105
106
107        {
108            ticksShifted = Int64ShrlMod32(*(DWORDLONG*)&ticks,dwLog);
109            tickSecShifted = Int64ShrlMod32(*(DWORDLONG*)&ticksPerSecond,dwLog);
110
111        }
112
113        dwRet = (DWORD)((((DWORD)ticksShifted)*1000)/(DWORD)(tickSecShifted));
114        dwRet *=1;
115    }
116    return dwRet;
117}
118
119int ReadFileMemory(const char* filename,long* plFileSize,unsigned char** pFilePtr)
120{
121    FILE* stream;
122    unsigned char* ptr;
123    int retVal=1;
124    stream=fopen(filename, "rb");
125    if (stream==NULL)
126        return 0;
127
128    fseek(stream,0,SEEK_END);
129
130    *plFileSize=ftell(stream);
131    fseek(stream,0,SEEK_SET);
132    ptr=malloc((*plFileSize)+1);
133    if (ptr==NULL)
134        retVal=0;
135    else
136    {
137        if (fread(ptr, 1, *plFileSize,stream) != (*plFileSize))
138            retVal=0;
139    }
140    fclose(stream);
141    *pFilePtr=ptr;
142    return retVal;
143}
144
145int main(int argc, char *argv[])
146{
147    int BlockSizeCompress=0x8000;
148    int BlockSizeUncompress=0x8000;
149    int cprLevel=Z_DEFAULT_COMPRESSION ;
150    long lFileSize;
151    unsigned char* FilePtr;
152    long lBufferSizeCpr;
153    long lBufferSizeUncpr;
154    long lCompressedSize=0;
155    unsigned char* CprPtr;
156    unsigned char* UncprPtr;
157    long lSizeCpr,lSizeUncpr;
158    DWORD dwGetTick,dwMsecQP;
159    LARGE_INTEGER li_qp,li_rdtsc,dwResRdtsc;
160
161    if (argc<=1)
162    {
163        printf("run TestZlib <File> [BlockSizeCompress] [BlockSizeUncompress] [compres. level]\n");
164        return 0;
165    }
166
167    if (ReadFileMemory(argv[1],&lFileSize,&FilePtr)==0)
168    {
169        printf("error reading %s\n",argv[1]);
170        return 1;
171    }
172    else printf("file %s read, %u bytes\n",argv[1],lFileSize);
173
174    if (argc>=3)
175        BlockSizeCompress=atol(argv[2]);
176
177    if (argc>=4)
178        BlockSizeUncompress=atol(argv[3]);
179
180    if (argc>=5)
181        cprLevel=(int)atol(argv[4]);
182
183    lBufferSizeCpr = lFileSize + (lFileSize/0x10) + 0x200;
184    lBufferSizeUncpr = lBufferSizeCpr;
185
186    CprPtr=(unsigned char*)malloc(lBufferSizeCpr + BlockSizeCompress);
187
188    BeginCountPerfCounter(&li_qp,TRUE);
189    dwGetTick=GetTickCount();
190    BeginCountRdtsc(&li_rdtsc);
191    {
192        z_stream zcpr;
193        int ret=Z_OK;
194        long lOrigToDo = lFileSize;
195        long lOrigDone = 0;
196        int step=0;
197        memset(&zcpr,0,sizeof(z_stream));
198        deflateInit(&zcpr,cprLevel);
199
200        zcpr.next_in = FilePtr;
201        zcpr.next_out = CprPtr;
202
203
204        do
205        {
206            long all_read_before = zcpr.total_in;
207            zcpr.avail_in = min(lOrigToDo,BlockSizeCompress);
208            zcpr.avail_out = BlockSizeCompress;
209            ret=deflate(&zcpr,(zcpr.avail_in==lOrigToDo) ? Z_FINISH : Z_SYNC_FLUSH);
210            lOrigDone += (zcpr.total_in-all_read_before);
211            lOrigToDo -= (zcpr.total_in-all_read_before);
212            step++;
213        } while (ret==Z_OK);
214
215        lSizeCpr=zcpr.total_out;
216        deflateEnd(&zcpr);
217        dwGetTick=GetTickCount()-dwGetTick;
218        dwMsecQP=GetMsecSincePerfCounter(li_qp,TRUE);
219        dwResRdtsc=GetResRdtsc(li_rdtsc,TRUE);
220        printf("total compress size = %u, in %u step\n",lSizeCpr,step);
221        printf("time = %u msec = %f sec\n",dwGetTick,dwGetTick/(double)1000.);
222        printf("defcpr time QP = %u msec = %f sec\n",dwMsecQP,dwMsecQP/(double)1000.);
223        printf("defcpr result rdtsc = %I64x\n\n",dwResRdtsc.QuadPart);
224    }
225
226    CprPtr=(unsigned char*)realloc(CprPtr,lSizeCpr);
227    UncprPtr=(unsigned char*)malloc(lBufferSizeUncpr + BlockSizeUncompress);
228
229    BeginCountPerfCounter(&li_qp,TRUE);
230    dwGetTick=GetTickCount();
231    BeginCountRdtsc(&li_rdtsc);
232    {
233        z_stream zcpr;
234        int ret=Z_OK;
235        long lOrigToDo = lSizeCpr;
236        long lOrigDone = 0;
237        int step=0;
238        memset(&zcpr,0,sizeof(z_stream));
239        inflateInit(&zcpr);
240
241        zcpr.next_in = CprPtr;
242        zcpr.next_out = UncprPtr;
243
244
245        do
246        {
247            long all_read_before = zcpr.total_in;
248            zcpr.avail_in = min(lOrigToDo,BlockSizeUncompress);
249            zcpr.avail_out = BlockSizeUncompress;
250            ret=inflate(&zcpr,Z_SYNC_FLUSH);
251            lOrigDone += (zcpr.total_in-all_read_before);
252            lOrigToDo -= (zcpr.total_in-all_read_before);
253            step++;
254        } while (ret==Z_OK);
255
256        lSizeUncpr=zcpr.total_out;
257        inflateEnd(&zcpr);
258        dwGetTick=GetTickCount()-dwGetTick;
259        dwMsecQP=GetMsecSincePerfCounter(li_qp,TRUE);
260        dwResRdtsc=GetResRdtsc(li_rdtsc,TRUE);
261        printf("total uncompress size = %u, in %u step\n",lSizeUncpr,step);
262        printf("time = %u msec = %f sec\n",dwGetTick,dwGetTick/(double)1000.);
263        printf("uncpr  time QP = %u msec = %f sec\n",dwMsecQP,dwMsecQP/(double)1000.);
264        printf("uncpr  result rdtsc = %I64x\n\n",dwResRdtsc.QuadPart);
265    }
266
267    if (lSizeUncpr==lFileSize)
268    {
269        if (memcmp(FilePtr,UncprPtr,lFileSize)==0)
270            printf("compare ok\n");
271
272    }
273
274    return 0;
275}
276