| Index: fpdfsdk/src/fpdfppo.cpp
|
| diff --git a/fpdfsdk/src/fpdfppo.cpp b/fpdfsdk/src/fpdfppo.cpp
|
| index ed2446471ee713b49525bfced37bdd3f7c1b52ed..6521258485b5ec3b9b6b66fc1bcf0709879b2102 100644
|
| --- a/fpdfsdk/src/fpdfppo.cpp
|
| +++ b/fpdfsdk/src/fpdfppo.cpp
|
| @@ -59,10 +59,10 @@ FX_BOOL CPDF_PageOrganizer::PDFDocInit(CPDF_Document* pDestPDFDoc,
|
| pNewRoot->SetAt("Type", new CPDF_Name("Catalog"));
|
| }
|
|
|
| - CPDF_Dictionary* pNewPages =
|
| - (CPDF_Dictionary*)(pNewRoot->GetElement("Pages")
|
| - ? pNewRoot->GetElement("Pages")->GetDirect()
|
| - : NULL);
|
| + CPDF_Dictionary* pNewPages = ToDictionary(
|
| + pNewRoot->GetElement("Pages") ? pNewRoot->GetElement("Pages")->GetDirect()
|
| + : nullptr);
|
| +
|
| if (!pNewPages) {
|
| pNewPages = new CPDF_Dictionary;
|
| FX_DWORD NewPagesON = pDestPDFDoc->AddIndirectObject(pNewPages);
|
| @@ -191,29 +191,25 @@ CPDF_Object* CPDF_PageOrganizer::PageDictGetInheritableTag(
|
|
|
| if (pType->GetString().Compare("Page"))
|
| return NULL;
|
| -
|
| if (!pDict->KeyExist("Parent"))
|
| return NULL;
|
| - CPDF_Object* pParent = pDict->GetElement("Parent")->GetDirect();
|
| - if (!pParent || pParent->GetType() != PDFOBJ_DICTIONARY)
|
| - return NULL;
|
|
|
| - CPDF_Dictionary* pp = (CPDF_Dictionary*)pParent;
|
| + CPDF_Dictionary* pp = ToDictionary(pDict->GetElement("Parent")->GetDirect());
|
| + if (!pp)
|
| + return nullptr;
|
|
|
| if (pDict->KeyExist((const char*)nSrctag))
|
| return pDict->GetElement((const char*)nSrctag);
|
| +
|
| while (pp) {
|
| if (pp->KeyExist((const char*)nSrctag))
|
| return pp->GetElement((const char*)nSrctag);
|
| - if (pp->KeyExist("Parent")) {
|
| - pp = (CPDF_Dictionary*)pp->GetElement("Parent")->GetDirect();
|
| - if (pp->GetType() == PDFOBJ_NULL)
|
| - break;
|
| - } else
|
| + if (!pp->KeyExist("Parent")) {
|
| break;
|
| + }
|
| + pp = ToDictionary(pp->GetElement("Parent")->GetDirect());
|
| }
|
| -
|
| - return NULL;
|
| + return nullptr;
|
| }
|
|
|
| FX_BOOL CPDF_PageOrganizer::UpdateReference(CPDF_Object* pObj,
|
| @@ -229,7 +225,7 @@ FX_BOOL CPDF_PageOrganizer::UpdateReference(CPDF_Object* pObj,
|
| break;
|
| }
|
| case PDFOBJ_DICTIONARY: {
|
| - CPDF_Dictionary* pDict = (CPDF_Dictionary*)pObj;
|
| + CPDF_Dictionary* pDict = pObj->AsDictionary();
|
|
|
| FX_POSITION pos = pDict->GetStartPos();
|
| while (pos) {
|
| @@ -300,8 +296,7 @@ int CPDF_PageOrganizer::GetNewObjId(CPDF_Document* pDoc,
|
| return 0;
|
| }
|
|
|
| - if (pClone->GetType() == PDFOBJ_DICTIONARY) {
|
| - CPDF_Dictionary* pDictClone = (CPDF_Dictionary*)pClone;
|
| + if (CPDF_Dictionary* pDictClone = pClone->AsDictionary()) {
|
| if (pDictClone->KeyExist("Type")) {
|
| CFX_ByteString strType = pDictClone->GetString("Type");
|
| if (!FXSYS_stricmp(strType, "Pages")) {
|
|
|