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 "fpdfsdk/formfiller/cba_fontmap.h"
8
9 #include <utility>
10
11 #include "core/fpdfapi/font/cpdf_font.h"
12 #include "core/fpdfapi/page/cpdf_page.h"
13 #include "core/fpdfapi/parser/cpdf_document.h"
14 #include "core/fpdfapi/parser/cpdf_reference.h"
15 #include "core/fpdfapi/parser/cpdf_simple_parser.h"
16 #include "core/fpdfapi/parser/cpdf_stream.h"
17 #include "core/fpdfapi/parser/fpdf_parser_decode.h"
18 #include "core/fpdfdoc/cpdf_formfield.h"
19 #include "core/fxge/cfx_substfont.h"
20 #include "fpdfsdk/cpdfsdk_annot.h"
21
CBA_FontMap(CPDFSDK_Annot * pAnnot,CFX_SystemHandler * pSystemHandler)22 CBA_FontMap::CBA_FontMap(CPDFSDK_Annot* pAnnot,
23 CFX_SystemHandler* pSystemHandler)
24 : CPWL_FontMap(pSystemHandler),
25 m_pDocument(nullptr),
26 m_pAnnotDict(nullptr),
27 m_pDefaultFont(nullptr),
28 m_sAPType("N") {
29 CPDF_Page* pPage = pAnnot->GetPDFPage();
30
31 m_pDocument = pPage->m_pDocument.Get();
32 m_pAnnotDict = pAnnot->GetPDFAnnot()->GetAnnotDict();
33 Initialize();
34 }
35
~CBA_FontMap()36 CBA_FontMap::~CBA_FontMap() {}
37
Reset()38 void CBA_FontMap::Reset() {
39 Empty();
40 m_pDefaultFont = nullptr;
41 m_sDefaultFontName.clear();
42 }
43
Initialize()44 void CBA_FontMap::Initialize() {
45 int32_t nCharset = FX_CHARSET_Default;
46
47 if (!m_pDefaultFont) {
48 m_pDefaultFont = GetAnnotDefaultFont(&m_sDefaultFontName);
49 if (m_pDefaultFont) {
50 if (const CFX_SubstFont* pSubstFont = m_pDefaultFont->GetSubstFont()) {
51 nCharset = pSubstFont->m_Charset;
52 } else {
53 if (m_sDefaultFontName == "Wingdings" ||
54 m_sDefaultFontName == "Wingdings2" ||
55 m_sDefaultFontName == "Wingdings3" ||
56 m_sDefaultFontName == "Webdings")
57 nCharset = FX_CHARSET_Symbol;
58 else
59 nCharset = FX_CHARSET_ANSI;
60 }
61 AddFontData(m_pDefaultFont.Get(), m_sDefaultFontName, nCharset);
62 AddFontToAnnotDict(m_pDefaultFont.Get(), m_sDefaultFontName);
63 }
64 }
65
66 if (nCharset != FX_CHARSET_ANSI)
67 CPWL_FontMap::Initialize();
68 }
69
SetDefaultFont(CPDF_Font * pFont,const ByteString & sFontName)70 void CBA_FontMap::SetDefaultFont(CPDF_Font* pFont,
71 const ByteString& sFontName) {
72 ASSERT(pFont);
73
74 if (m_pDefaultFont)
75 return;
76
77 m_pDefaultFont = pFont;
78 m_sDefaultFontName = sFontName;
79
80 int32_t nCharset = FX_CHARSET_Default;
81 if (const CFX_SubstFont* pSubstFont = m_pDefaultFont->GetSubstFont())
82 nCharset = pSubstFont->m_Charset;
83 AddFontData(m_pDefaultFont.Get(), m_sDefaultFontName, nCharset);
84 }
85
FindFontSameCharset(ByteString * sFontAlias,int32_t nCharset)86 CPDF_Font* CBA_FontMap::FindFontSameCharset(ByteString* sFontAlias,
87 int32_t nCharset) {
88 if (m_pAnnotDict->GetStringFor("Subtype") != "Widget")
89 return nullptr;
90
91 CPDF_Document* pDocument = GetDocument();
92 const CPDF_Dictionary* pRootDict = pDocument->GetRoot();
93 if (!pRootDict)
94 return nullptr;
95
96 CPDF_Dictionary* pAcroFormDict = pRootDict->GetDictFor("AcroForm");
97 if (!pAcroFormDict)
98 return nullptr;
99
100 CPDF_Dictionary* pDRDict = pAcroFormDict->GetDictFor("DR");
101 if (!pDRDict)
102 return nullptr;
103
104 return FindResFontSameCharset(pDRDict, sFontAlias, nCharset);
105 }
106
GetDocument()107 CPDF_Document* CBA_FontMap::GetDocument() {
108 return m_pDocument.Get();
109 }
110
FindResFontSameCharset(CPDF_Dictionary * pResDict,ByteString * sFontAlias,int32_t nCharset)111 CPDF_Font* CBA_FontMap::FindResFontSameCharset(CPDF_Dictionary* pResDict,
112 ByteString* sFontAlias,
113 int32_t nCharset) {
114 if (!pResDict)
115 return nullptr;
116
117 CPDF_Dictionary* pFonts = pResDict->GetDictFor("Font");
118 if (!pFonts)
119 return nullptr;
120
121 CPDF_Document* pDocument = GetDocument();
122 CPDF_Font* pFind = nullptr;
123 for (const auto& it : *pFonts) {
124 const ByteString& csKey = it.first;
125 if (!it.second)
126 continue;
127
128 CPDF_Dictionary* pElement = ToDictionary(it.second->GetDirect());
129 if (!pElement)
130 continue;
131 if (pElement->GetStringFor("Type") != "Font")
132 continue;
133
134 CPDF_Font* pFont = pDocument->LoadFont(pElement);
135 if (!pFont)
136 continue;
137 const CFX_SubstFont* pSubst = pFont->GetSubstFont();
138 if (!pSubst)
139 continue;
140 if (pSubst->m_Charset == nCharset) {
141 *sFontAlias = csKey;
142 pFind = pFont;
143 }
144 }
145 return pFind;
146 }
147
AddedFont(CPDF_Font * pFont,const ByteString & sFontAlias)148 void CBA_FontMap::AddedFont(CPDF_Font* pFont, const ByteString& sFontAlias) {
149 AddFontToAnnotDict(pFont, sFontAlias);
150 }
151
AddFontToAnnotDict(CPDF_Font * pFont,const ByteString & sAlias)152 void CBA_FontMap::AddFontToAnnotDict(CPDF_Font* pFont,
153 const ByteString& sAlias) {
154 if (!pFont)
155 return;
156
157 CPDF_Dictionary* pAPDict = m_pAnnotDict->GetDictFor("AP");
158 if (!pAPDict)
159 pAPDict = m_pAnnotDict->SetNewFor<CPDF_Dictionary>("AP");
160
161 // to avoid checkbox and radiobutton
162 CPDF_Object* pObject = pAPDict->GetObjectFor(m_sAPType);
163 if (ToDictionary(pObject))
164 return;
165
166 CPDF_Stream* pStream = pAPDict->GetStreamFor(m_sAPType);
167 if (!pStream) {
168 pStream = m_pDocument->NewIndirect<CPDF_Stream>();
169 pAPDict->SetNewFor<CPDF_Reference>(m_sAPType, m_pDocument.Get(),
170 pStream->GetObjNum());
171 }
172
173 CPDF_Dictionary* pStreamDict = pStream->GetDict();
174 if (!pStreamDict) {
175 auto pOwnedDict =
176 pdfium::MakeUnique<CPDF_Dictionary>(m_pDocument->GetByteStringPool());
177 pStreamDict = pOwnedDict.get();
178 pStream->InitStream(nullptr, 0, std::move(pOwnedDict));
179 }
180
181 CPDF_Dictionary* pStreamResList = pStreamDict->GetDictFor("Resources");
182 if (!pStreamResList)
183 pStreamResList = pStreamDict->SetNewFor<CPDF_Dictionary>("Resources");
184 CPDF_Dictionary* pStreamResFontList = pStreamResList->GetDictFor("Font");
185 if (!pStreamResFontList) {
186 pStreamResFontList = m_pDocument->NewIndirect<CPDF_Dictionary>();
187 pStreamResList->SetNewFor<CPDF_Reference>("Font", m_pDocument.Get(),
188 pStreamResFontList->GetObjNum());
189 }
190 if (!pStreamResFontList->KeyExist(sAlias)) {
191 pStreamResFontList->SetNewFor<CPDF_Reference>(
192 sAlias, m_pDocument.Get(), pFont->GetFontDict()->GetObjNum());
193 }
194 }
195
GetAnnotDefaultFont(ByteString * sAlias)196 CPDF_Font* CBA_FontMap::GetAnnotDefaultFont(ByteString* sAlias) {
197 CPDF_Dictionary* pAcroFormDict = nullptr;
198 const bool bWidget = (m_pAnnotDict->GetStringFor("Subtype") == "Widget");
199 if (bWidget) {
200 const CPDF_Dictionary* pRootDict = m_pDocument->GetRoot();
201 if (pRootDict)
202 pAcroFormDict = pRootDict->GetDictFor("AcroForm");
203 }
204
205 ByteString sDA;
206 CPDF_Object* pObj = FPDF_GetFieldAttr(m_pAnnotDict.Get(), "DA");
207 if (pObj)
208 sDA = pObj->GetString();
209
210 if (bWidget) {
211 if (sDA.IsEmpty()) {
212 pObj = FPDF_GetFieldAttr(pAcroFormDict, "DA");
213 sDA = pObj ? pObj->GetString() : ByteString();
214 }
215 }
216 if (sDA.IsEmpty())
217 return nullptr;
218
219 CPDF_SimpleParser syntax(sDA.AsStringView());
220 syntax.FindTagParamFromStart("Tf", 2);
221
222 ByteString sFontName(syntax.GetWord());
223 ByteString sDecodedFontName = PDF_NameDecode(sFontName);
224 *sAlias = sDecodedFontName.Right(sDecodedFontName.GetLength() - 1);
225
226 CPDF_Dictionary* pFontDict = nullptr;
227 if (CPDF_Dictionary* pAPDict = m_pAnnotDict->GetDictFor("AP")) {
228 if (CPDF_Dictionary* pNormalDict = pAPDict->GetDictFor("N")) {
229 if (CPDF_Dictionary* pNormalResDict =
230 pNormalDict->GetDictFor("Resources")) {
231 if (CPDF_Dictionary* pResFontDict = pNormalResDict->GetDictFor("Font"))
232 pFontDict = pResFontDict->GetDictFor(*sAlias);
233 }
234 }
235 }
236 if (bWidget && !pFontDict && pAcroFormDict) {
237 if (CPDF_Dictionary* pDRDict = pAcroFormDict->GetDictFor("DR")) {
238 if (CPDF_Dictionary* pDRFontDict = pDRDict->GetDictFor("Font"))
239 pFontDict = pDRFontDict->GetDictFor(*sAlias);
240 }
241 }
242 return pFontDict ? m_pDocument->LoadFont(pFontDict) : nullptr;
243 }
244
SetAPType(const ByteString & sAPType)245 void CBA_FontMap::SetAPType(const ByteString& sAPType) {
246 m_sAPType = sAPType;
247
248 Reset();
249 Initialize();
250 }
251