diff options
Diffstat (limited to 'fpdfsdk/fpdfdoc.cpp')
| -rw-r--r-- | fpdfsdk/fpdfdoc.cpp | 372 |
1 files changed, 372 insertions, 0 deletions
diff --git a/fpdfsdk/fpdfdoc.cpp b/fpdfsdk/fpdfdoc.cpp new file mode 100644 index 0000000000..9fa9f4fcab --- /dev/null +++ b/fpdfsdk/fpdfdoc.cpp @@ -0,0 +1,372 @@ +// Copyright 2014 PDFium Authors. All rights reserved. +// Use of this source code is governed by a BSD-style license that can be +// found in the LICENSE file. + +// Original code copyright 2014 Foxit Software Inc. http://www.foxitsoftware.com + +#include "public/fpdf_doc.h" + +#include <set> + +#include "core/include/fpdfapi/cpdf_array.h" +#include "core/include/fpdfapi/cpdf_document.h" +#include "fpdfsdk/include/fsdk_define.h" +#include "third_party/base/stl_util.h" + +namespace { + +int THISMODULE = 0; + +CPDF_Bookmark FindBookmark(const CPDF_BookmarkTree& tree, + CPDF_Bookmark bookmark, + const CFX_WideString& title, + std::set<CPDF_Dictionary*>* visited) { + // Return if already checked to avoid circular calling. + if (pdfium::ContainsKey(*visited, bookmark.GetDict())) + return CPDF_Bookmark(); + visited->insert(bookmark.GetDict()); + + if (bookmark.GetDict() && + bookmark.GetTitle().CompareNoCase(title.c_str()) == 0) { + // First check this item. + return bookmark; + } + + // Go into children items. + CPDF_Bookmark child = tree.GetFirstChild(bookmark); + while (child.GetDict() && !pdfium::ContainsKey(*visited, child.GetDict())) { + // Check this item and its children. + CPDF_Bookmark found = FindBookmark(tree, child, title, visited); + if (found.GetDict()) + return found; + child = tree.GetNextSibling(child); + } + return CPDF_Bookmark(); +} + +void ReleaseLinkList(void* data) { + delete (CPDF_LinkList*)data; +} + +CPDF_LinkList* GetLinkList(CPDF_Page* page) { + if (!page) + return nullptr; + + // Link list is stored with the document + CPDF_Document* pDoc = page->m_pDocument; + CPDF_LinkList* pLinkList = (CPDF_LinkList*)pDoc->GetPrivateData(&THISMODULE); + if (!pLinkList) { + pLinkList = new CPDF_LinkList; + pDoc->SetPrivateData(&THISMODULE, pLinkList, ReleaseLinkList); + } + return pLinkList; +} + +} // namespace + +DLLEXPORT FPDF_BOOKMARK STDCALL +FPDFBookmark_GetFirstChild(FPDF_DOCUMENT document, FPDF_BOOKMARK pDict) { + CPDF_Document* pDoc = CPDFDocumentFromFPDFDocument(document); + if (!pDoc) + return nullptr; + CPDF_BookmarkTree tree(pDoc); + CPDF_Bookmark bookmark = + CPDF_Bookmark(ToDictionary(static_cast<CPDF_Object*>(pDict))); + return tree.GetFirstChild(bookmark).GetDict(); +} + +DLLEXPORT FPDF_BOOKMARK STDCALL +FPDFBookmark_GetNextSibling(FPDF_DOCUMENT document, FPDF_BOOKMARK pDict) { + if (!pDict) + return nullptr; + CPDF_Document* pDoc = CPDFDocumentFromFPDFDocument(document); + if (!pDoc) + return nullptr; + CPDF_BookmarkTree tree(pDoc); + CPDF_Bookmark bookmark = + CPDF_Bookmark(ToDictionary(static_cast<CPDF_Object*>(pDict))); + return tree.GetNextSibling(bookmark).GetDict(); +} + +DLLEXPORT unsigned long STDCALL FPDFBookmark_GetTitle(FPDF_BOOKMARK pDict, + void* buffer, + unsigned long buflen) { + if (!pDict) + return 0; + CPDF_Bookmark bookmark(ToDictionary(static_cast<CPDF_Object*>(pDict))); + CFX_WideString title = bookmark.GetTitle(); + CFX_ByteString encodedTitle = title.UTF16LE_Encode(); + unsigned long len = encodedTitle.GetLength(); + if (buffer && buflen >= len) { + FXSYS_memcpy(buffer, encodedTitle.c_str(), len); + } + return len; +} + +DLLEXPORT FPDF_BOOKMARK STDCALL FPDFBookmark_Find(FPDF_DOCUMENT document, + FPDF_WIDESTRING title) { + if (!title || title[0] == 0) + return nullptr; + CPDF_Document* pDoc = CPDFDocumentFromFPDFDocument(document); + if (!pDoc) + return nullptr; + CPDF_BookmarkTree tree(pDoc); + FX_STRSIZE len = CFX_WideString::WStringLength(title); + CFX_WideString encodedTitle = CFX_WideString::FromUTF16LE(title, len); + std::set<CPDF_Dictionary*> visited; + return FindBookmark(tree, CPDF_Bookmark(), encodedTitle, &visited).GetDict(); +} + +DLLEXPORT FPDF_DEST STDCALL FPDFBookmark_GetDest(FPDF_DOCUMENT document, + FPDF_BOOKMARK pDict) { + if (!pDict) + return nullptr; + CPDF_Document* pDoc = CPDFDocumentFromFPDFDocument(document); + if (!pDoc) + return nullptr; + CPDF_Bookmark bookmark(ToDictionary(static_cast<CPDF_Object*>(pDict))); + CPDF_Dest dest = bookmark.GetDest(pDoc); + if (dest.GetObject()) + return dest.GetObject(); + // If this bookmark is not directly associated with a dest, we try to get + // action + CPDF_Action action = bookmark.GetAction(); + if (!action.GetDict()) + return nullptr; + return action.GetDest(pDoc).GetObject(); +} + +DLLEXPORT FPDF_ACTION STDCALL FPDFBookmark_GetAction(FPDF_BOOKMARK pDict) { + if (!pDict) + return NULL; + CPDF_Bookmark bookmark(ToDictionary(static_cast<CPDF_Object*>(pDict))); + return bookmark.GetAction().GetDict(); +} + +DLLEXPORT unsigned long STDCALL FPDFAction_GetType(FPDF_ACTION pDict) { + if (!pDict) + return PDFACTION_UNSUPPORTED; + + CPDF_Action action(ToDictionary(static_cast<CPDF_Object*>(pDict))); + CPDF_Action::ActionType type = action.GetType(); + switch (type) { + case CPDF_Action::GoTo: + return PDFACTION_GOTO; + case CPDF_Action::GoToR: + return PDFACTION_REMOTEGOTO; + case CPDF_Action::URI: + return PDFACTION_URI; + case CPDF_Action::Launch: + return PDFACTION_LAUNCH; + default: + return PDFACTION_UNSUPPORTED; + } +} + +DLLEXPORT FPDF_DEST STDCALL FPDFAction_GetDest(FPDF_DOCUMENT document, + FPDF_ACTION pDict) { + if (!pDict) + return nullptr; + CPDF_Document* pDoc = CPDFDocumentFromFPDFDocument(document); + if (!pDoc) + return nullptr; + CPDF_Action action(ToDictionary(static_cast<CPDF_Object*>(pDict))); + return action.GetDest(pDoc).GetObject(); +} + +DLLEXPORT unsigned long STDCALL FPDFAction_GetFilePath(FPDF_ACTION pDict, + void* buffer, + unsigned long buflen) { + unsigned long type = FPDFAction_GetType(pDict); + if (type != PDFACTION_REMOTEGOTO && type != PDFACTION_LAUNCH) + return 0; + + CPDF_Action action(ToDictionary(static_cast<CPDF_Object*>(pDict))); + CFX_ByteString path = action.GetFilePath().UTF8Encode(); + unsigned long len = path.GetLength() + 1; + if (buffer && buflen >= len) + FXSYS_memcpy(buffer, path.c_str(), len); + return len; +} + +DLLEXPORT unsigned long STDCALL FPDFAction_GetURIPath(FPDF_DOCUMENT document, + FPDF_ACTION pDict, + void* buffer, + unsigned long buflen) { + if (!pDict) + return 0; + CPDF_Document* pDoc = CPDFDocumentFromFPDFDocument(document); + if (!pDoc) + return 0; + CPDF_Action action(ToDictionary(static_cast<CPDF_Object*>(pDict))); + CFX_ByteString path = action.GetURI(pDoc); + unsigned long len = path.GetLength() + 1; + if (buffer && buflen >= len) + FXSYS_memcpy(buffer, path.c_str(), len); + return len; +} + +DLLEXPORT unsigned long STDCALL FPDFDest_GetPageIndex(FPDF_DOCUMENT document, + FPDF_DEST pDict) { + if (!pDict) + return 0; + CPDF_Document* pDoc = CPDFDocumentFromFPDFDocument(document); + if (!pDoc) + return 0; + CPDF_Dest dest(static_cast<CPDF_Array*>(pDict)); + return dest.GetPageIndex(pDoc); +} + +DLLEXPORT FPDF_LINK STDCALL FPDFLink_GetLinkAtPoint(FPDF_PAGE page, + double x, + double y) { + CPDF_Page* pPage = CPDFPageFromFPDFPage(page); + if (!pPage) + return nullptr; + + CPDF_LinkList* pLinkList = GetLinkList(pPage); + if (!pLinkList) + return nullptr; + + return pLinkList->GetLinkAtPoint(pPage, (FX_FLOAT)x, (FX_FLOAT)y, nullptr) + .GetDict(); +} + +DLLEXPORT int STDCALL FPDFLink_GetLinkZOrderAtPoint(FPDF_PAGE page, + double x, + double y) { + CPDF_Page* pPage = CPDFPageFromFPDFPage(page); + if (!pPage) + return -1; + + CPDF_LinkList* pLinkList = GetLinkList(pPage); + if (!pLinkList) + return -1; + + int z_order = -1; + pLinkList->GetLinkAtPoint(pPage, (FX_FLOAT)x, (FX_FLOAT)y, &z_order); + return z_order; +} + +DLLEXPORT FPDF_DEST STDCALL FPDFLink_GetDest(FPDF_DOCUMENT document, + FPDF_LINK pDict) { + if (!pDict) + return nullptr; + CPDF_Document* pDoc = CPDFDocumentFromFPDFDocument(document); + if (!pDoc) + return nullptr; + CPDF_Link link(ToDictionary(static_cast<CPDF_Object*>(pDict))); + FPDF_DEST dest = link.GetDest(pDoc).GetObject(); + if (dest) + return dest; + // If this link is not directly associated with a dest, we try to get action + CPDF_Action action = link.GetAction(); + if (!action.GetDict()) + return nullptr; + return action.GetDest(pDoc).GetObject(); +} + +DLLEXPORT FPDF_ACTION STDCALL FPDFLink_GetAction(FPDF_LINK pDict) { + if (!pDict) + return nullptr; + + CPDF_Link link(ToDictionary(static_cast<CPDF_Object*>(pDict))); + return link.GetAction().GetDict(); +} + +DLLEXPORT FPDF_BOOL STDCALL FPDFLink_Enumerate(FPDF_PAGE page, + int* startPos, + FPDF_LINK* linkAnnot) { + if (!startPos || !linkAnnot) + return FALSE; + CPDF_Page* pPage = CPDFPageFromFPDFPage(page); + if (!pPage || !pPage->m_pFormDict) + return FALSE; + CPDF_Array* pAnnots = pPage->m_pFormDict->GetArrayBy("Annots"); + if (!pAnnots) + return FALSE; + for (int i = *startPos; i < (int)pAnnots->GetCount(); i++) { + CPDF_Dictionary* pDict = + ToDictionary(static_cast<CPDF_Object*>(pAnnots->GetElementValue(i))); + if (!pDict) + continue; + if (pDict->GetStringBy("Subtype").Equal("Link")) { + *startPos = i + 1; + *linkAnnot = (FPDF_LINK)pDict; + return TRUE; + } + } + return FALSE; +} + +DLLEXPORT FPDF_BOOL STDCALL FPDFLink_GetAnnotRect(FPDF_LINK linkAnnot, + FS_RECTF* rect) { + if (!linkAnnot || !rect) + return FALSE; + CPDF_Dictionary* pAnnotDict = + ToDictionary(static_cast<CPDF_Object*>(linkAnnot)); + CFX_FloatRect rt = pAnnotDict->GetRectBy("Rect"); + rect->left = rt.left; + rect->bottom = rt.bottom; + rect->right = rt.right; + rect->top = rt.top; + return TRUE; +} + +DLLEXPORT int STDCALL FPDFLink_CountQuadPoints(FPDF_LINK linkAnnot) { + if (!linkAnnot) + return 0; + CPDF_Dictionary* pAnnotDict = + ToDictionary(static_cast<CPDF_Object*>(linkAnnot)); + CPDF_Array* pArray = pAnnotDict->GetArrayBy("QuadPoints"); + if (!pArray) + return 0; + return pArray->GetCount() / 8; +} + +DLLEXPORT FPDF_BOOL STDCALL FPDFLink_GetQuadPoints(FPDF_LINK linkAnnot, + int quadIndex, + FS_QUADPOINTSF* quadPoints) { + if (!linkAnnot || !quadPoints) + return FALSE; + CPDF_Dictionary* pAnnotDict = + ToDictionary(static_cast<CPDF_Object*>(linkAnnot)); + CPDF_Array* pArray = pAnnotDict->GetArrayBy("QuadPoints"); + if (pArray) { + if (quadIndex < 0 || quadIndex >= (int)pArray->GetCount() / 8 || + ((quadIndex * 8 + 7) >= (int)pArray->GetCount())) + return FALSE; + quadPoints->x1 = pArray->GetNumberAt(quadIndex * 8); + quadPoints->y1 = pArray->GetNumberAt(quadIndex * 8 + 1); + quadPoints->x2 = pArray->GetNumberAt(quadIndex * 8 + 2); + quadPoints->y2 = pArray->GetNumberAt(quadIndex * 8 + 3); + quadPoints->x3 = pArray->GetNumberAt(quadIndex * 8 + 4); + quadPoints->y3 = pArray->GetNumberAt(quadIndex * 8 + 5); + quadPoints->x4 = pArray->GetNumberAt(quadIndex * 8 + 6); + quadPoints->y4 = pArray->GetNumberAt(quadIndex * 8 + 7); + return TRUE; + } + return FALSE; +} + +DLLEXPORT unsigned long STDCALL FPDF_GetMetaText(FPDF_DOCUMENT doc, + FPDF_BYTESTRING tag, + void* buffer, + unsigned long buflen) { + if (!tag) + return 0; + CPDF_Document* pDoc = CPDFDocumentFromFPDFDocument(doc); + if (!pDoc) + return 0; + CPDF_Dictionary* pInfo = pDoc->GetInfo(); + if (!pInfo) + return 0; + CFX_WideString text = pInfo->GetUnicodeTextBy(tag); + // Use UTF-16LE encoding + CFX_ByteString encodedText = text.UTF16LE_Encode(); + unsigned long len = encodedText.GetLength(); + if (buffer && buflen >= len) { + FXSYS_memcpy(buffer, encodedText.c_str(), len); + } + return len; +} |