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 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; 32 m_pAnnotDict = pAnnot->GetPDFAnnot()->GetAnnotDict(); 33 Initialize(); 34 } 35 36 CBA_FontMap::~CBA_FontMap() {} 37 38 void CBA_FontMap::Reset() { 39 Empty(); 40 m_pDefaultFont = nullptr; 41 m_sDefaultFontName = ""; 42 } 43 44 void CBA_FontMap::Initialize() { 45 int32_t nCharset = FXFONT_DEFAULT_CHARSET; 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 = FXFONT_SYMBOL_CHARSET; 58 else 59 nCharset = FXFONT_ANSI_CHARSET; 60 } 61 AddFontData(m_pDefaultFont, m_sDefaultFontName, nCharset); 62 AddFontToAnnotDict(m_pDefaultFont, m_sDefaultFontName); 63 } 64 } 65 66 if (nCharset != FXFONT_ANSI_CHARSET) 67 CPWL_FontMap::Initialize(); 68 } 69 70 void CBA_FontMap::SetDefaultFont(CPDF_Font* pFont, 71 const CFX_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 = FXFONT_DEFAULT_CHARSET; 81 if (const CFX_SubstFont* pSubstFont = m_pDefaultFont->GetSubstFont()) 82 nCharset = pSubstFont->m_Charset; 83 AddFontData(m_pDefaultFont, m_sDefaultFontName, nCharset); 84 } 85 86 CPDF_Font* CBA_FontMap::FindFontSameCharset(CFX_ByteString& sFontAlias, 87 int32_t nCharset) { 88 if (m_pAnnotDict->GetStringFor("Subtype") != "Widget") 89 return nullptr; 90 91 CPDF_Document* pDocument = GetDocument(); 92 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 107 CPDF_Document* CBA_FontMap::GetDocument() { 108 return m_pDocument; 109 } 110 111 CPDF_Font* CBA_FontMap::FindResFontSameCharset(CPDF_Dictionary* pResDict, 112 CFX_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 CFX_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 148 void CBA_FontMap::AddedFont(CPDF_Font* pFont, 149 const CFX_ByteString& sFontAlias) { 150 AddFontToAnnotDict(pFont, sFontAlias); 151 } 152 153 void CBA_FontMap::AddFontToAnnotDict(CPDF_Font* pFont, 154 const CFX_ByteString& sAlias) { 155 if (!pFont) 156 return; 157 158 CPDF_Dictionary* pAPDict = m_pAnnotDict->GetDictFor("AP"); 159 if (!pAPDict) 160 pAPDict = m_pAnnotDict->SetNewFor<CPDF_Dictionary>("AP"); 161 162 // to avoid checkbox and radiobutton 163 CPDF_Object* pObject = pAPDict->GetObjectFor(m_sAPType); 164 if (ToDictionary(pObject)) 165 return; 166 167 CPDF_Stream* pStream = pAPDict->GetStreamFor(m_sAPType); 168 if (!pStream) { 169 pStream = m_pDocument->NewIndirect<CPDF_Stream>(); 170 pAPDict->SetNewFor<CPDF_Reference>(m_sAPType, m_pDocument, 171 pStream->GetObjNum()); 172 } 173 174 CPDF_Dictionary* pStreamDict = pStream->GetDict(); 175 if (!pStreamDict) { 176 auto pOwnedDict = 177 pdfium::MakeUnique<CPDF_Dictionary>(m_pDocument->GetByteStringPool()); 178 pStreamDict = pOwnedDict.get(); 179 pStream->InitStream(nullptr, 0, std::move(pOwnedDict)); 180 } 181 182 CPDF_Dictionary* pStreamResList = pStreamDict->GetDictFor("Resources"); 183 if (!pStreamResList) 184 pStreamResList = pStreamDict->SetNewFor<CPDF_Dictionary>("Resources"); 185 CPDF_Dictionary* pStreamResFontList = pStreamResList->GetDictFor("Font"); 186 if (!pStreamResFontList) { 187 pStreamResFontList = m_pDocument->NewIndirect<CPDF_Dictionary>(); 188 pStreamResList->SetNewFor<CPDF_Reference>("Font", m_pDocument, 189 pStreamResFontList->GetObjNum()); 190 } 191 if (!pStreamResFontList->KeyExist(sAlias)) { 192 pStreamResFontList->SetNewFor<CPDF_Reference>( 193 sAlias, m_pDocument, pFont->GetFontDict()->GetObjNum()); 194 } 195 } 196 197 CPDF_Font* CBA_FontMap::GetAnnotDefaultFont(CFX_ByteString& sAlias) { 198 CPDF_Dictionary* pAcroFormDict = nullptr; 199 const bool bWidget = (m_pAnnotDict->GetStringFor("Subtype") == "Widget"); 200 if (bWidget) { 201 if (CPDF_Dictionary* pRootDict = m_pDocument->GetRoot()) 202 pAcroFormDict = pRootDict->GetDictFor("AcroForm"); 203 } 204 205 CFX_ByteString sDA; 206 CPDF_Object* pObj = FPDF_GetFieldAttr(m_pAnnotDict, "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() : CFX_ByteString(); 214 } 215 } 216 if (sDA.IsEmpty()) 217 return nullptr; 218 219 CPDF_SimpleParser syntax(sDA.AsStringC()); 220 syntax.FindTagParamFromStart("Tf", 2); 221 CFX_ByteString sFontName(syntax.GetWord()); 222 sAlias = PDF_NameDecode(sFontName).Mid(1); 223 CPDF_Dictionary* pFontDict = nullptr; 224 225 if (CPDF_Dictionary* pAPDict = m_pAnnotDict->GetDictFor("AP")) { 226 if (CPDF_Dictionary* pNormalDict = pAPDict->GetDictFor("N")) { 227 if (CPDF_Dictionary* pNormalResDict = 228 pNormalDict->GetDictFor("Resources")) { 229 if (CPDF_Dictionary* pResFontDict = pNormalResDict->GetDictFor("Font")) 230 pFontDict = pResFontDict->GetDictFor(sAlias); 231 } 232 } 233 } 234 235 if (bWidget && !pFontDict && pAcroFormDict) { 236 if (CPDF_Dictionary* pDRDict = pAcroFormDict->GetDictFor("DR")) { 237 if (CPDF_Dictionary* pDRFontDict = pDRDict->GetDictFor("Font")) 238 pFontDict = pDRFontDict->GetDictFor(sAlias); 239 } 240 } 241 242 return pFontDict ? m_pDocument->LoadFont(pFontDict) : nullptr; 243 } 244 245 void CBA_FontMap::SetAPType(const CFX_ByteString& sAPType) { 246 m_sAPType = sAPType; 247 248 Reset(); 249 Initialize(); 250 } 251