Home | History | Annotate | Download | only in formfiller 
      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