8#include <botan/internal/compress_utils.h>
10#include <botan/exceptn.h>
11#include <botan/mem_ops.h>
12#include <botan/internal/fmt.h>
13#include <botan/internal/safeint.h>
19 Exception(
fmt(
"Compression API {} failed with return code {}", func_name, rc)), m_type(type), m_rc(rc) {}
21void* Compression_Alloc_Info::do_malloc(
size_t n,
size_t size) {
26 void* ptr = std::calloc(n, size);
37 m_current_allocs[ptr] = n * size;
43void Compression_Alloc_Info::do_free(
void* ptr) {
45 auto i = m_current_allocs.find(ptr);
47 if(i == m_current_allocs.end()) {
48 throw Internal_Error(
"Compression_Alloc_Info::free got pointer not allocated by us");
53 m_current_allocs.erase(i);
61void Stream_Compression::start(
size_t level) {
62 m_stream = make_stream(level);
65void Stream_Compression::process(secure_vector<uint8_t>& buf,
size_t offset, uint32_t flags) {
70 if(buf.size() == offset && flags == m_stream->run_flag()) {
74 if(m_buffer.size() < buf.size() + offset) {
75 m_buffer.resize(buf.size() + offset);
83 if(m_buffer.empty()) {
87 m_stream->next_in(buf.data() + offset, buf.size() - offset);
88 m_stream->next_out(m_buffer.data() + offset, m_buffer.size() - offset);
91 const bool stream_end = m_stream->run(flags);
94 BOTAN_ASSERT(m_stream->avail_in() == 0,
"After stream is done, no input remains to be processed");
95 m_buffer.resize(m_buffer.size() - m_stream->avail_out());
97 }
else if(m_stream->avail_out() == 0) {
98 const size_t added = 8 + m_buffer.size();
99 m_buffer.resize(m_buffer.size() + added);
100 m_stream->next_out(m_buffer.data() + m_buffer.size() - added, added);
101 }
else if(m_stream->avail_in() == 0) {
102 m_buffer.resize(m_buffer.size() - m_stream->avail_out());
107 copy_mem(m_buffer.data(), buf.data(), offset);
113 process(buf, offset, flush ? m_stream->flush_flag() : m_stream->run_flag());
118 process(buf, offset, m_stream->finish_flag());
126void Stream_Decompression::start() {
127 m_stream = make_stream();
130void Stream_Decompression::process(secure_vector<uint8_t>& buf,
size_t offset, uint32_t flags) {
134 if(m_buffer.size() < buf.size() + offset) {
135 m_buffer.resize(buf.size() + offset);
138 m_stream->next_in(buf.data() + offset, buf.size() - offset);
139 m_stream->next_out(m_buffer.data() + offset, m_buffer.size() - offset);
142 const bool stream_end = m_stream->run(flags);
145 if(m_stream->avail_in() == 0)
147 m_buffer.resize(m_buffer.size() - m_stream->avail_out());
153 const size_t read = (buf.size() - offset) - m_stream->avail_in();
155 m_stream->next_in(buf.data() + offset + read, buf.size() - offset - read);
158 if(m_stream->avail_out() == 0) {
159 const size_t added = 8 + m_buffer.size();
160 m_buffer.resize(m_buffer.size() + added);
161 m_stream->next_out(m_buffer.data() + m_buffer.size() - added, added);
162 }
else if(m_stream->avail_in() == 0) {
163 m_buffer.resize(m_buffer.size() - m_stream->avail_out());
168 copy_mem(m_buffer.data(), buf.data(), offset);
173 process(buf, offset, m_stream->run_flag());
177 if(buf.size() != offset || m_stream.get()) {
178 process(buf, offset, m_stream->finish_flag());
#define BOTAN_ASSERT(expr, assertion_made)
Compression_Error(const char *func_name, ErrorType type, int rc)
virtual std::string name() const =0
void update(secure_vector< uint8_t > &buf, size_t offset, bool flush) final
void finish(secure_vector< uint8_t > &buf, size_t offset) final
void update(secure_vector< uint8_t > &buf, size_t offset) final
void finish(secure_vector< uint8_t > &buf, size_t offset) final
std::string fmt(std::string_view format, const T &... args)
void secure_scrub_memory(void *ptr, size_t n)
std::vector< T, secure_allocator< T > > secure_vector
constexpr void copy_mem(T *out, const T *in, size_t n)
#define BOTAN_CHECKED_MUL(x, y)