2
1
mirror of https://github.com/qpdf/qpdf.git synced 2024-12-22 19:08:59 +00:00
qpdf/libqpdf/Pl_Buffer.cc

86 lines
1.7 KiB
C++
Raw Normal View History

#include <qpdf/Pl_Buffer.hh>
2018-08-16 15:52:59 +00:00
#include <algorithm>
2023-05-20 11:22:32 +00:00
#include <cstdlib>
#include <cstring>
#include <stdexcept>
Pl_Buffer::Pl_Buffer(char const* identifier, Pipeline* next) :
Pipeline(identifier, next),
m(new Members())
{
}
Pl_Buffer::~Pl_Buffer() // NOLINT (modernize-use-equals-default)
{
// Must be explicit and not inline -- see QPDF_DLL_CLASS in README-maintainer
}
void
Pl_Buffer::write(unsigned char const* buf, size_t len)
{
if (!len) {
return;
}
m->data.append(reinterpret_cast<char const*>(buf), len);
m->ready = false;
if (getNext(true)) {
getNext()->write(buf, len);
}
}
void
Pl_Buffer::finish()
{
m->ready = true;
if (getNext(true)) {
getNext()->finish();
}
}
Buffer*
Pl_Buffer::getBuffer()
{
if (!m->ready) {
throw std::logic_error("Pl_Buffer::getBuffer() called when not ready");
}
auto* b = new Buffer(std::move(m->data));
m->data.clear();
return b;
}
2021-12-17 12:33:42 +00:00
2023-11-16 16:26:04 +00:00
std::string
Pl_Buffer::getString()
{
if (!m->ready) {
throw std::logic_error("Pl_Buffer::getString() called when not ready");
}
auto s = std::move(m->data);
m->data.clear();
return s;
}
std::shared_ptr<Buffer>
Pl_Buffer::getBufferSharedPointer()
{
return std::shared_ptr<Buffer>(getBuffer());
}
2021-12-17 12:33:42 +00:00
void
Pl_Buffer::getMallocBuffer(unsigned char** buf, size_t* len)
2021-12-17 12:33:42 +00:00
{
if (!m->ready) {
2023-05-21 17:35:09 +00:00
throw std::logic_error("Pl_Buffer::getMallocBuffer() called when not ready");
2021-12-17 12:33:42 +00:00
}
auto size = m->data.size();
*len = size;
if (size > 0) {
*buf = reinterpret_cast<unsigned char*>(malloc(size));
memcpy(*buf, m->data.data(), size);
} else {
2021-12-17 12:33:42 +00:00
*buf = nullptr;
}
m->data.clear();
2021-12-17 12:33:42 +00:00
}