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
|
/*
* Shared code for compression libraries
* (C) 2014 Jack Lloyd
*
* Botan is released under the Simplified BSD License (see license.txt)
*/
#ifndef BOTAN_COMPRESSION_UTILS_H__
#define BOTAN_COMPRESSION_UTILS_H__
#include <botan/compression.h>
#include <cstring>
#include <memory>
#include <unordered_map>
namespace Botan {
/*
* Allocation Size Tracking Helper for Zlib/Bzlib/LZMA
*/
class Compression_Alloc_Info
{
public:
template<typename T>
static void* malloc(void* self, T n, T size)
{
return static_cast<Compression_Alloc_Info*>(self)->do_malloc(n, size);
}
static void free(void* self, void* ptr)
{
static_cast<Compression_Alloc_Info*>(self)->do_free(ptr);
}
private:
void* do_malloc(size_t n, size_t size);
void do_free(void* ptr);
std::unordered_map<void*, size_t> m_current_allocs;
};
/**
* Wrapper for Zlib/Bzlib/LZMA stream types
*/
template<typename Stream, typename ByteType>
class Zlib_Style_Stream : public Compression_Stream
{
public:
void next_in(byte* b, size_t len) override
{
m_stream.next_in = reinterpret_cast<ByteType*>(b);
m_stream.avail_in = len;
}
void next_out(byte* b, size_t len) override
{
m_stream.next_out = reinterpret_cast<ByteType*>(b);
m_stream.avail_out = len;
}
size_t avail_in() const override { return m_stream.avail_in; }
size_t avail_out() const override { return m_stream.avail_out; }
Zlib_Style_Stream()
{
std::memset(&m_stream, 0, sizeof(stream_t));
m_allocs.reset(new Compression_Alloc_Info);
}
~Zlib_Style_Stream()
{
std::memset(&m_stream, 0, sizeof(stream_t));
m_allocs.reset();
}
protected:
typedef Stream stream_t;
stream_t* streamp() { return &m_stream; }
Compression_Alloc_Info* alloc() { return m_allocs.get(); }
private:
stream_t m_stream;
std::unique_ptr<Compression_Alloc_Info> m_allocs;
};
}
#endif
|
