1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
|
////////////////////////////////////////////////////////////////////////////////
// This source file is part of the ZipArchive library source distribution and
// is Copyrighted 2000 - 2007 by Artpol Software - Tadeusz Dracz
//
// This program is free software; you can redistribute it and/or
// modify it under the terms of the GNU General Public License
// as published by the Free Software Foundation; either version 2
// of the License, or (at your option) any later version.
//
// For the licensing details refer to the License.txt file.
//
// Web Site: http://www.artpol-software.com
////////////////////////////////////////////////////////////////////////////////
/**
* \file DeflateCompressor.h
* Includes the ZipArchiveLib::CDeflateCompressor class.
*
*/
#if !defined(ZIPARCHIVE_DEFLATECOMPRESSOR_DOT_H)
#define ZIPARCHIVE_DEFLATECOMPRESSOR_DOT_H
#if _MSC_VER > 1000
#pragma once
#endif
#include "ZipExport.h"
#include "BaseLibCompressor.h"
#include "ZipException.h"
#include "zlib/zlib.h"
namespace ZipArchiveLib
{
/**
Compresses and decompresses data using the Zlib library.
*/
class ZIP_API CDeflateCompressor : public CBaseLibCompressor
{
public:
/**
Represents options of the CDeflateCompressor.
\see
<a href="kb">0610231446|options</a>
\see
CZipArchive::SetCompressionOptions
*/
struct ZIP_API COptions : CBaseLibCompressor::COptions
{
COptions()
{
m_bCheckLastBlock = true;
}
int GetType() const
{
return typeDeflate;
}
CZipCompressor::COptions* Clone() const
{
return new COptions(*this);
}
/**
Enables or disables checking, if the compressed data ends with an end-of-stream block.
This should be enabled to protect against malformed data.
\c true, if the checking of the last block should be enabled; \c false otherwise.
*/
bool m_bCheckLastBlock;
};
/**
Initializes a new instance of the CDeflateCompressor class.
\param pStorage
The current storage object.
*/
CDeflateCompressor(CZipStorage* pStorage);
bool CanProcess(WORD uMethod) {return uMethod == methodStore || uMethod == methodDeflate;}
void InitCompression(int iLevel, CZipFileHeader* pFile, CZipCryptograph* pCryptograph);
void InitDecompression(CZipFileHeader* pFile, CZipCryptograph* pCryptograph);
DWORD Decompress(void *pBuffer, DWORD uSize);
void Compress(const void *pBuffer, DWORD uSize);
void FinishCompression(bool bAfterException);
void FinishDecompression(bool bAfterException);
const CZipCompressor::COptions* GetOptions() const
{
return &m_options;
}
~CDeflateCompressor()
{
}
protected:
void UpdateOptions(const CZipCompressor::COptions* pOptions)
{
m_options = *(COptions*)pOptions;
}
int ConvertInternalError(int iErr) const
{
switch (iErr)
{
case Z_NEED_DICT:
return CZipException::needDict;
case Z_STREAM_END:
return CZipException::streamEnd;
case Z_ERRNO:
return CZipException::errNo;
case Z_STREAM_ERROR:
return CZipException::streamError;
case Z_DATA_ERROR:
return CZipException::dataError;
case Z_MEM_ERROR:
return CZipException::memError;
case Z_BUF_ERROR:
return CZipException::bufError;
case Z_VERSION_ERROR:
return CZipException::versionError;
default:
return CZipException::genericError;
}
}
bool IsCodeErrorOK(int iErr) const
{
return iErr == Z_OK || iErr == Z_NEED_DICT;
}
private:
COptions m_options;
zarch_z_stream m_stream;
};
} // namespace
#endif
|
