blob: 365affa9113bc901b1b21dff4268ecd5dd062c4a (
plain)
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
|
/*
* Lzma Compressor
* (C) 2001 Peter J Jones
* 2001-2007,2014 Jack Lloyd
* 2006 Matt Johnston
* 2012 Vojtech Kral
*
* Botan is released under the Simplified BSD License (see license.txt)
*/
#include <botan/lzma.h>
#include <botan/internal/compress_utils.h>
#include <botan/exceptn.h>
#include <lzma.h>
namespace Botan {
namespace {
class LZMA_Stream : public Zlib_Style_Stream<lzma_stream, uint8_t>
{
public:
LZMA_Stream()
{
auto a = new ::lzma_allocator;
a->opaque = alloc();
a->alloc = Compression_Alloc_Info::malloc<size_t>;
a->free = Compression_Alloc_Info::free;
streamp()->allocator = a;
}
~LZMA_Stream()
{
::lzma_end(streamp());
delete streamp()->allocator;
}
bool run(uint32_t flags) override
{
lzma_ret rc = ::lzma_code(streamp(), static_cast<lzma_action>(flags));
if(rc == LZMA_MEM_ERROR)
throw Exception("lzma memory allocation failed");
else if (rc != LZMA_OK && rc != LZMA_STREAM_END)
throw Exception("Lzma error");
return (rc == LZMA_STREAM_END);
}
uint32_t run_flag() const override { return LZMA_RUN; }
uint32_t flush_flag() const override { return LZMA_FULL_FLUSH; }
uint32_t finish_flag() const override { return LZMA_FINISH; }
};
class LZMA_Compression_Stream final : public LZMA_Stream
{
public:
explicit LZMA_Compression_Stream(size_t level)
{
if(level == 0)
level = 6; // default
else if(level > 9)
level = 9; // clamp to maximum allowed value
lzma_ret rc = ::lzma_easy_encoder(streamp(), level, LZMA_CHECK_CRC64);
if(rc == LZMA_MEM_ERROR)
throw Exception("lzma memory allocation failed");
else if(rc != LZMA_OK)
throw Exception("lzma compress initialization failed");
}
};
class LZMA_Decompression_Stream final : public LZMA_Stream
{
public:
LZMA_Decompression_Stream()
{
lzma_ret rc = ::lzma_stream_decoder(streamp(), UINT64_MAX,
LZMA_TELL_UNSUPPORTED_CHECK);
if(rc == LZMA_MEM_ERROR)
throw Exception("lzma memory allocation failed");
else if(rc != LZMA_OK)
throw Exception("Bad setting in lzma_stream_decoder");
}
};
}
Compression_Stream* LZMA_Compression::make_stream(size_t level) const
{
return new LZMA_Compression_Stream(level);
}
Compression_Stream* LZMA_Decompression::make_stream() const
{
return new LZMA_Decompression_Stream;
}
}
|
