mirror of https://github.com/qpdf/qpdf.git
58 lines
1.0 KiB
C++
58 lines
1.0 KiB
C++
|
|
#include <qpdf/Pl_RC4.hh>
|
|
|
|
#include <qpdf/QUtil.hh>
|
|
|
|
Pl_RC4::Pl_RC4(char const* identifier, Pipeline* next,
|
|
unsigned char const* key_data, int key_len,
|
|
int out_bufsize) :
|
|
Pipeline(identifier, next),
|
|
out_bufsize(out_bufsize),
|
|
rc4(key_data, key_len)
|
|
{
|
|
this->outbuf = new unsigned char[out_bufsize];
|
|
}
|
|
|
|
Pl_RC4::~Pl_RC4()
|
|
{
|
|
if (this->outbuf)
|
|
{
|
|
delete [] this->outbuf;
|
|
this->outbuf = 0;
|
|
}
|
|
}
|
|
|
|
void
|
|
Pl_RC4::write(unsigned char* data, int len)
|
|
{
|
|
if (this->outbuf == 0)
|
|
{
|
|
throw Exception(
|
|
this->identifier +
|
|
": Pl_RC4: write() called after finish() called");
|
|
}
|
|
|
|
int bytes_left = len;
|
|
unsigned char* p = data;
|
|
|
|
while (bytes_left > 0)
|
|
{
|
|
int bytes = (bytes_left < this->out_bufsize ? bytes_left : out_bufsize);
|
|
bytes_left -= bytes;
|
|
rc4.process(p, bytes, outbuf);
|
|
p += bytes;
|
|
getNext()->write(outbuf, bytes);
|
|
}
|
|
}
|
|
|
|
void
|
|
Pl_RC4::finish()
|
|
{
|
|
if (this->outbuf)
|
|
{
|
|
delete [] this->outbuf;
|
|
this->outbuf = 0;
|
|
}
|
|
this->getNext()->finish();
|
|
}
|