2
1
mirror of https://github.com/qpdf/qpdf.git synced 2024-11-01 03:12:29 +00:00
qpdf/zlib-flate/zlib-flate.cc
2021-12-20 17:25:18 -05:00

118 lines
2.4 KiB
C++

#include <qpdf/Pl_Flate.hh>
#include <qpdf/Pl_StdioFile.hh>
#include <qpdf/QUtil.hh>
#include <qpdf/QPDF.hh>
#include <stdio.h>
#include <string.h>
#include <iostream>
#include <stdlib.h>
#include <fcntl.h>
static char const* whoami = 0;
void usage()
{
std::cerr << "Usage: " << whoami << " { -uncompress | -compress[=n] }"
<< std::endl
<< "If n is specified with -compress, it is a"
<< " zlib compression level from" << std::endl
<< "1 to 9 where lower numbers are faster and"
<< " less compressed and higher" << std::endl
<< "numbers are slower and more compressed"
<< std::endl;
exit(2);
}
int main(int argc, char* argv[])
{
if ((whoami = strrchr(argv[0], '/')) == NULL)
{
whoami = argv[0];
}
else
{
++whoami;
}
// For libtool's sake....
if (strncmp(whoami, "lt-", 3) == 0)
{
whoami += 3;
}
if ((argc == 2) && (strcmp(argv[1], "--version") == 0))
{
std::cout << whoami << " from qpdf version "
<< QPDF::QPDFVersion() << std::endl;
exit(0);
}
if (argc != 2)
{
usage();
}
Pl_Flate::action_e action = Pl_Flate::a_inflate;
if ((strcmp(argv[1], "-uncompress") == 0))
{
// okay
}
else if ((strcmp(argv[1], "-compress") == 0))
{
action = Pl_Flate::a_deflate;
}
else if ((strncmp(argv[1], "-compress=", 10) == 0))
{
action = Pl_Flate::a_deflate;
int level = QUtil::string_to_int(argv[1] + 10);
Pl_Flate::setCompressionLevel(level);
}
else
{
usage();
}
QUtil::binary_stdout();
QUtil::binary_stdin();
PointerHolder<Pl_StdioFile> out = new Pl_StdioFile("stdout", stdout);
PointerHolder<Pl_Flate> flate =
new Pl_Flate("flate", out.getPointer(), action);
bool warn = false;
flate->setWarnCallback([&warn](char const* msg, int code) {
warn = true;
std::cerr << whoami << ": WARNING: zlib code " << code
<< ", msg = " << msg << std::endl;
});
try
{
unsigned char buf[10000];
bool done = false;
while (! done)
{
size_t len = fread(buf, 1, sizeof(buf), stdin);
if (len <= 0)
{
done = true;
}
else
{
flate->write(buf, len);
}
}
flate->finish();
}
catch (std::exception& e)
{
std::cerr << whoami << ": " << e.what() << std::endl;
exit(2);
}
if (warn)
{
exit(3);
}
return 0;
}