2
1
mirror of https://github.com/qpdf/qpdf.git synced 2024-12-22 19:08:59 +00:00
qpdf/libqpdf/QPDFOutlineObjectHelper.cc
m-holger 0a081e1f09 In QPDFOutlineObjectHelper detect loops in direct children
Also, add diagnostic messages in qpdf_fuzzer and additional fuzz test case.
2024-06-29 12:38:07 +01:00

103 lines
2.6 KiB
C++

#include <qpdf/QPDFOutlineObjectHelper.hh>
#include <qpdf/QPDFOutlineDocumentHelper.hh>
#include <qpdf/QTC.hh>
QPDFOutlineObjectHelper::Members::Members(QPDFOutlineDocumentHelper& dh) :
dh(dh)
{
}
QPDFOutlineObjectHelper::QPDFOutlineObjectHelper(
QPDFObjectHandle oh, QPDFOutlineDocumentHelper& dh, int depth) :
QPDFObjectHelper(oh),
m(new Members(dh))
{
if (depth > 50) {
// Not exercised in test suite, but was tested manually by temporarily changing max depth
// to 1.
return;
}
if (QPDFOutlineDocumentHelper::Accessor::checkSeen(m->dh, this->oh.getObjGen())) {
QTC::TC("qpdf", "QPDFOutlineObjectHelper loop");
return;
}
QPDFObjGen::set children;
QPDFObjectHandle cur = oh.getKey("/First");
while (!cur.isNull() && children.add(cur)) {
QPDFOutlineObjectHelper new_ooh(cur, dh, 1 + depth);
new_ooh.m->parent = std::make_shared<QPDFOutlineObjectHelper>(*this);
m->kids.push_back(new_ooh);
cur = cur.getKey("/Next");
}
}
std::shared_ptr<QPDFOutlineObjectHelper>
QPDFOutlineObjectHelper::getParent()
{
return m->parent;
}
std::vector<QPDFOutlineObjectHelper>
QPDFOutlineObjectHelper::getKids()
{
return m->kids;
}
QPDFObjectHandle
QPDFOutlineObjectHelper::getDest()
{
QPDFObjectHandle dest;
QPDFObjectHandle A;
if (this->oh.hasKey("/Dest")) {
QTC::TC("qpdf", "QPDFOutlineObjectHelper direct dest");
dest = this->oh.getKey("/Dest");
} else if (
(A = this->oh.getKey("/A")).isDictionary() && A.getKey("/S").isName() &&
(A.getKey("/S").getName() == "/GoTo") && A.hasKey("/D")) {
QTC::TC("qpdf", "QPDFOutlineObjectHelper action dest");
dest = A.getKey("/D");
}
if (!dest.isInitialized()) {
dest = QPDFObjectHandle::newNull();
}
if (dest.isName() || dest.isString()) {
QTC::TC("qpdf", "QPDFOutlineObjectHelper named dest");
dest = m->dh.resolveNamedDest(dest);
}
return dest;
}
QPDFObjectHandle
QPDFOutlineObjectHelper::getDestPage()
{
QPDFObjectHandle dest = getDest();
if ((dest.isArray()) && (dest.getArrayNItems() > 0)) {
return dest.getArrayItem(0);
}
return QPDFObjectHandle::newNull();
}
int
QPDFOutlineObjectHelper::getCount()
{
int count = 0;
if (this->oh.hasKey("/Count")) {
count = this->oh.getKey("/Count").getIntValueAsInt();
}
return count;
}
std::string
QPDFOutlineObjectHelper::getTitle()
{
std::string result;
if (this->oh.hasKey("/Title")) {
result = this->oh.getKey("/Title").getUTF8Value();
}
return result;
}