2
1
mirror of https://github.com/qpdf/qpdf.git synced 2024-12-23 03:18:59 +00:00
qpdf/zlib-flate/zlib-flate.cc
Jay Berkenbilt ba0ef7a124 Replace PointerHolder with std::shared_ptr in the rest of the code
Increase to POINTERHOLDER_TRANSITION=3

patrepl s/PointerHolder/std::shared_ptr/g **/*.cc **/*.hh
patrepl s/make_pointer_holder/std::make_shared/g **/*.cc
patrepl s/make_array_pointer_holder/QUtil::make_shared_array/g **/*.cc
patrepl s,qpdf/std::shared_ptr,qpdf/PointerHolder, **/*.cc **/*.hh
git restore include/qpdf/PointerHolder.hh
git restore libtests/pointer_holder.cc
cleanpatch
./format-code
2022-04-09 17:33:29 -04:00

93 lines
2.4 KiB
C++

#include <qpdf/Pl_Flate.hh>
#include <qpdf/Pl_StdioFile.hh>
#include <qpdf/QPDF.hh>
#include <qpdf/QUtil.hh>
#include <fcntl.h>
#include <iostream>
#include <stdio.h>
#include <stdlib.h>
#include <string.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;
}
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();
auto out = std::make_shared<Pl_StdioFile>("stdout", stdout);
auto flate = std::make_shared<Pl_Flate>("flate", out.get(), 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;
}