zlib.cpp
2.35 KB
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
// zlib.cpp - written and placed in the public domain by Wei Dai
// "zlib" is the name of a well known C language compression library
// (http://www.zlib.org) and also the name of a compression format
// (RFC 1950) that the library implements. This file is part of a
// complete reimplementation of the zlib compression format.
#include "pch.h"
#include "zlib.h"
#include "zdeflate.h"
#include "zinflate.h"
NAMESPACE_BEGIN(CryptoPP)
static const byte DEFLATE_METHOD = 8;
static const byte FDICT_FLAG = 1 << 5;
// *************************************************************
void ZlibCompressor::WritePrestreamHeader()
{
m_adler32.Restart();
byte cmf = DEFLATE_METHOD | ((GetLog2WindowSize()-8) << 4);
byte flags = GetCompressionLevel() << 6;
AttachedTransformation()->PutWord16(RoundUpToMultipleOf(cmf*256+flags, 31));
}
void ZlibCompressor::ProcessUncompressedData(const byte *inString, size_t length)
{
m_adler32.Update(inString, length);
}
void ZlibCompressor::WritePoststreamTail()
{
FixedSizeSecBlock<byte, 4> adler32;
m_adler32.Final(adler32);
AttachedTransformation()->Put(adler32, 4);
}
unsigned int ZlibCompressor::GetCompressionLevel() const
{
static const unsigned int deflateToCompressionLevel[] = {0, 1, 1, 1, 2, 2, 2, 2, 2, 3};
return deflateToCompressionLevel[GetDeflateLevel()];
}
// *************************************************************
ZlibDecompressor::ZlibDecompressor(BufferedTransformation *attachment, bool repeat, int propagation)
: Inflator(attachment, repeat, propagation)
{
}
void ZlibDecompressor::ProcessPrestreamHeader()
{
m_adler32.Restart();
byte cmf;
byte flags;
if (!m_inQueue.Get(cmf) || !m_inQueue.Get(flags))
throw HeaderErr();
if ((cmf*256+flags) % 31 != 0)
throw HeaderErr(); // if you hit this exception, you're probably trying to decompress invalid data
if ((cmf & 0xf) != DEFLATE_METHOD)
throw UnsupportedAlgorithm();
if (flags & FDICT_FLAG)
throw UnsupportedPresetDictionary();
m_log2WindowSize = 8 + (cmf >> 4);
}
void ZlibDecompressor::ProcessDecompressedData(const byte *inString, size_t length)
{
AttachedTransformation()->Put(inString, length);
m_adler32.Update(inString, length);
}
void ZlibDecompressor::ProcessPoststreamTail()
{
FixedSizeSecBlock<byte, 4> adler32;
if (m_inQueue.Get(adler32, 4) != 4)
throw Adler32Err();
if (!m_adler32.Verify(adler32))
throw Adler32Err();
}
NAMESPACE_END