SEGV in CFX_BaseSegmentedArray::Iterate() when CS has malformed dictionary.
[pdfium.git] / core / src / fpdfdoc / doc_bookmark.cpp
1 // Copyright 2014 PDFium Authors. All rights reserved.
2 // Use of this source code is governed by a BSD-style license that can be
3 // found in the LICENSE file.
4  
5 // Original code copyright 2014 Foxit Software Inc. http://www.foxitsoftware.com
6
7 #include "../../include/fpdfdoc/fpdf_doc.h"
8 CPDF_Bookmark CPDF_BookmarkTree::GetFirstChild(const CPDF_Bookmark& parent) const
9 {
10     if (!parent.m_pDict) {
11         CPDF_Dictionary* pRoot = m_pDocument->GetRoot()->GetDict("Outlines");
12         if (!pRoot) {
13             return CPDF_Bookmark();
14         }
15         return CPDF_Bookmark(pRoot->GetDict("First"));
16     }
17     return CPDF_Bookmark(parent.m_pDict->GetDict("First"));
18 }
19 CPDF_Bookmark CPDF_BookmarkTree::GetNextSibling(const CPDF_Bookmark& bookmark) const
20 {
21     if (!bookmark.m_pDict) {
22         return CPDF_Bookmark();
23     }
24     CPDF_Dictionary *pNext = bookmark.m_pDict->GetDict("Next");
25     return pNext == bookmark.m_pDict ? CPDF_Bookmark() : CPDF_Bookmark(pNext);
26 }
27 FX_DWORD CPDF_Bookmark::GetColorRef() const
28 {
29     if (!m_pDict) {
30         return 0;
31     }
32     CPDF_Array* pColor = m_pDict->GetArray("C");
33     if (!pColor) {
34         return FXSYS_RGB(0, 0, 0);
35     }
36     int r = FXSYS_round(pColor->GetNumber(0) * 255);
37     int g = FXSYS_round(pColor->GetNumber(1) * 255);
38     int b = FXSYS_round(pColor->GetNumber(2) * 255);
39     return FXSYS_RGB(r, g, b);
40 }
41 FX_DWORD CPDF_Bookmark::GetFontStyle() const
42 {
43     if (!m_pDict) {
44         return 0;
45     }
46     return m_pDict->GetInteger("F");
47 }
48 CFX_WideString CPDF_Bookmark::GetTitle() const
49 {
50     if (!m_pDict) {
51         return CFX_WideString();
52     }
53     CPDF_String* pString = (CPDF_String*)m_pDict->GetElementValue("Title");
54     if (!pString || pString->GetType() != PDFOBJ_STRING) {
55         return CFX_WideString();
56     }
57     CFX_WideString title = pString->GetUnicodeText();
58     FX_LPWSTR buf = title.LockBuffer();
59     int len = title.GetLength();
60     for (int i = 0; i < len; i++) {
61         if (buf[i] < 0x20) {
62             buf[i] = 0x20;
63         }
64     }
65     title.ReleaseBuffer(len);
66     return title;
67 }
68 CPDF_Dest CPDF_Bookmark::GetDest(CPDF_Document* pDocument) const
69 {
70     if (!m_pDict) {
71         return CPDF_Dest();
72     }
73     CPDF_Object* pDest = m_pDict->GetElementValue("Dest");
74     if (!pDest) {
75         return CPDF_Dest();
76     }
77     if (pDest->GetType() == PDFOBJ_STRING || pDest->GetType() == PDFOBJ_NAME) {
78         CPDF_NameTree name_tree(pDocument, FX_BSTRC("Dests"));
79         CFX_ByteStringC name = pDest->GetString();
80         return CPDF_Dest(name_tree.LookupNamedDest(pDocument, name));
81     }
82     if (pDest->GetType() == PDFOBJ_ARRAY) {
83         return CPDF_Dest((CPDF_Array*)pDest);
84     }
85     return CPDF_Dest();
86 }
87 CPDF_Action CPDF_Bookmark::GetAction() const
88 {
89     if (!m_pDict) {
90         return CPDF_Action();
91     }
92     return CPDF_Action(m_pDict->GetDict("A"));
93 }