mirror of
https://github.com/qpdf/qpdf.git
synced 2024-09-22 10:09:06 +00:00
12f1eb15ca
Run this: for i in **/*.cc **/*.c **/*.h **/*.hh; do clang-format < $i >| $i.new && mv $i.new $i done
165 lines
4.5 KiB
C++
165 lines
4.5 KiB
C++
#include <qpdf/QPDF.hh>
|
|
#include <qpdf/QPDFObjectHandle.hh>
|
|
|
|
#include <algorithm>
|
|
#include <cstdlib>
|
|
#include <iostream>
|
|
#include <map>
|
|
#include <set>
|
|
#include <sstream>
|
|
#include <string>
|
|
#include <vector>
|
|
|
|
void
|
|
usage()
|
|
{
|
|
std::cerr << "Usage: test_parsedoffset INPUT.pdf" << std::endl;
|
|
}
|
|
|
|
std::string
|
|
make_objdesc(qpdf_offset_t offset, QPDFObjectHandle obj)
|
|
{
|
|
std::stringstream ss;
|
|
ss << "offset = " << offset << " (0x" << std::hex << offset << std::dec
|
|
<< "), ";
|
|
|
|
if (obj.isIndirect()) {
|
|
ss << "indirect " << obj.getObjectID() << "/" << obj.getGeneration()
|
|
<< ", ";
|
|
} else {
|
|
ss << "direct, ";
|
|
}
|
|
|
|
ss << obj.getTypeName();
|
|
|
|
return ss.str();
|
|
}
|
|
|
|
void
|
|
walk(
|
|
size_t stream_number,
|
|
QPDFObjectHandle obj,
|
|
std::vector<std::vector<std::pair<qpdf_offset_t, std::string>>>& result)
|
|
{
|
|
qpdf_offset_t offset = obj.getParsedOffset();
|
|
std::pair<qpdf_offset_t, std::string> p =
|
|
std::make_pair(offset, make_objdesc(offset, obj));
|
|
|
|
if (result.size() < stream_number + 1) {
|
|
result.resize(stream_number + 1);
|
|
}
|
|
result[stream_number].push_back(p);
|
|
|
|
if (obj.isArray()) {
|
|
std::vector<QPDFObjectHandle> array = obj.getArrayAsVector();
|
|
for (std::vector<QPDFObjectHandle>::iterator iter = array.begin();
|
|
iter != array.end();
|
|
++iter) {
|
|
if (!iter->isIndirect()) {
|
|
// QPDF::GetAllObjects() enumerates all indirect objects.
|
|
// So only the direct objects are recursed here.
|
|
walk(stream_number, *iter, result);
|
|
}
|
|
}
|
|
} else if (obj.isDictionary()) {
|
|
std::set<std::string> keys = obj.getKeys();
|
|
for (std::set<std::string>::iterator iter = keys.begin();
|
|
iter != keys.end();
|
|
++iter) {
|
|
QPDFObjectHandle item = obj.getKey(*iter);
|
|
if (!item.isIndirect()) {
|
|
// QPDF::GetAllObjects() enumerates all indirect objects.
|
|
// So only the direct objects are recursed here.
|
|
walk(stream_number, item, result);
|
|
}
|
|
}
|
|
} else if (obj.isStream()) {
|
|
walk(stream_number, obj.getDict(), result);
|
|
}
|
|
}
|
|
|
|
void
|
|
process(
|
|
std::string fn,
|
|
std::vector<std::vector<std::pair<qpdf_offset_t, std::string>>>& result)
|
|
{
|
|
QPDF qpdf;
|
|
qpdf.processFile(fn.c_str());
|
|
std::vector<QPDFObjectHandle> objs = qpdf.getAllObjects();
|
|
std::map<QPDFObjGen, QPDFXRefEntry> xrefs = qpdf.getXRefTable();
|
|
|
|
for (std::vector<QPDFObjectHandle>::iterator iter = objs.begin();
|
|
iter != objs.end();
|
|
++iter) {
|
|
if (xrefs.count(iter->getObjGen()) == 0) {
|
|
std::cerr << iter->getObjectID() << "/" << iter->getGeneration()
|
|
<< " is not found in xref table" << std::endl;
|
|
std::exit(2);
|
|
}
|
|
|
|
QPDFXRefEntry xref = xrefs[iter->getObjGen()];
|
|
size_t stream_number;
|
|
|
|
switch (xref.getType()) {
|
|
case 0:
|
|
std::cerr << iter->getObjectID() << "/" << iter->getGeneration()
|
|
<< " xref entry is free" << std::endl;
|
|
std::exit(2);
|
|
case 1:
|
|
stream_number = 0;
|
|
break;
|
|
case 2:
|
|
stream_number = static_cast<size_t>(xref.getObjStreamNumber());
|
|
break;
|
|
default:
|
|
std::cerr << "unknown xref entry type" << std::endl;
|
|
std::exit(2);
|
|
}
|
|
|
|
walk(stream_number, *iter, result);
|
|
}
|
|
}
|
|
|
|
int
|
|
main(int argc, char* argv[])
|
|
{
|
|
if (argc != 2) {
|
|
usage();
|
|
std::exit(2);
|
|
}
|
|
|
|
try {
|
|
std::vector<std::vector<std::pair<qpdf_offset_t, std::string>>> table;
|
|
|
|
process(argv[1], table);
|
|
|
|
for (size_t i = 0; i < table.size(); ++i) {
|
|
if (table[i].size() == 0) {
|
|
continue;
|
|
}
|
|
|
|
std::sort(table[i].begin(), table[i].end());
|
|
if (i == 0) {
|
|
std::cout << "--- objects not in streams ---" << std::endl;
|
|
} else {
|
|
std::cout << "--- objects in stream " << i << " ---"
|
|
<< std::endl;
|
|
}
|
|
|
|
for (std::vector<std::pair<qpdf_offset_t, std::string>>::iterator
|
|
iter = table[i].begin();
|
|
iter != table[i].end();
|
|
++iter) {
|
|
std::cout << iter->second << std::endl;
|
|
}
|
|
}
|
|
|
|
std::cout << "succeeded" << std::endl;
|
|
} catch (std::exception& e) {
|
|
std::cerr << e.what() << std::endl;
|
|
std::exit(2);
|
|
}
|
|
|
|
return 0;
|
|
}
|