2
1
mirror of https://github.com/qpdf/qpdf.git synced 2024-12-23 03:18:59 +00:00
qpdf/libqpdf/Pl_RunLength.cc

156 lines
4.4 KiB
C++

#include <qpdf/Pl_RunLength.hh>
#include <qpdf/QTC.hh>
#include <qpdf/QUtil.hh>
Pl_RunLength::Members::Members(action_e action) :
action(action),
state(st_top),
length(0)
{
}
Pl_RunLength::Pl_RunLength(
char const* identifier, Pipeline* next, action_e action) :
Pipeline(identifier, next),
m(new Members(action))
{
}
Pl_RunLength::~Pl_RunLength()
{
// Must be explicit and not inline -- see QPDF_DLL_CLASS in
// README-maintainer
}
void
Pl_RunLength::write(unsigned char const* data, size_t len)
{
if (this->m->action == a_encode) {
encode(data, len);
} else {
decode(data, len);
}
}
void
Pl_RunLength::encode(unsigned char const* data, size_t len)
{
for (size_t i = 0; i < len; ++i) {
if ((this->m->state == st_top) != (this->m->length <= 1)) {
throw std::logic_error(
"Pl_RunLength::encode: state/length inconsistency");
}
unsigned char ch = data[i];
if ((this->m->length > 0) &&
((this->m->state == st_copying) || (this->m->length < 128)) &&
(ch == this->m->buf[this->m->length - 1])) {
QTC::TC(
"libtests",
"Pl_RunLength: switch to run",
(this->m->length == 128) ? 0 : 1);
if (this->m->state == st_copying) {
--this->m->length;
flush_encode();
this->m->buf[0] = ch;
this->m->length = 1;
}
this->m->state = st_run;
this->m->buf[this->m->length] = ch;
++this->m->length;
} else {
if ((this->m->length == 128) || (this->m->state == st_run)) {
flush_encode();
} else if (this->m->length > 0) {
this->m->state = st_copying;
}
this->m->buf[this->m->length] = ch;
++this->m->length;
}
}
}
void
Pl_RunLength::decode(unsigned char const* data, size_t len)
{
for (size_t i = 0; i < len; ++i) {
unsigned char ch = data[i];
switch (this->m->state) {
case st_top:
if (ch < 128) {
// length represents remaining number of bytes to copy
this->m->length = 1U + ch;
this->m->state = st_copying;
} else if (ch > 128) {
// length represents number of copies of next byte
this->m->length = 257U - ch;
this->m->state = st_run;
} else // ch == 128
{
// EOD; stay in this state
}
break;
case st_copying:
this->getNext()->write(&ch, 1);
if (--this->m->length == 0) {
this->m->state = st_top;
}
break;
case st_run:
for (unsigned int j = 0; j < this->m->length; ++j) {
this->getNext()->write(&ch, 1);
}
this->m->state = st_top;
break;
}
}
}
void
Pl_RunLength::flush_encode()
{
if (this->m->length == 128) {
QTC::TC(
"libtests",
"Pl_RunLength flush full buffer",
(this->m->state == st_copying ? 0
: this->m->state == st_run ? 1
: -1));
}
if (this->m->length == 0) {
QTC::TC("libtests", "Pl_RunLength flush empty buffer");
}
if (this->m->state == st_run) {
if ((this->m->length < 2) || (this->m->length > 128)) {
throw std::logic_error(
"Pl_RunLength: invalid length in flush_encode for run");
}
unsigned char ch = static_cast<unsigned char>(257 - this->m->length);
this->getNext()->write(&ch, 1);
this->getNext()->write(&this->m->buf[0], 1);
} else if (this->m->length > 0) {
unsigned char ch = static_cast<unsigned char>(this->m->length - 1);
this->getNext()->write(&ch, 1);
this->getNext()->write(this->m->buf, this->m->length);
}
this->m->state = st_top;
this->m->length = 0;
}
void
Pl_RunLength::finish()
{
// When decoding, we might have read a length byte not followed by
// data, which means the stream was terminated early, but we will
// just ignore this case since this is the only sensible thing to
// do.
if (this->m->action == a_encode) {
flush_encode();
unsigned char ch = 128;
this->getNext()->write(&ch, 1);
}
this->getNext()->finish();
}