| // Copyright 2016 The PDFium Authors |
| // 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 "core/fpdfdoc/cpdf_interactiveform.h" |
| |
| #include <optional> |
| #include <type_traits> |
| #include <utility> |
| #include <vector> |
| |
| #include "build/build_config.h" |
| #include "constants/form_fields.h" |
| #include "constants/form_flags.h" |
| #include "constants/stream_dict_common.h" |
| #include "core/fpdfapi/font/cpdf_font.h" |
| #include "core/fpdfapi/font/cpdf_fontencoding.h" |
| #include "core/fpdfapi/page/cpdf_docpagedata.h" |
| #include "core/fpdfapi/page/cpdf_page.h" |
| #include "core/fpdfapi/parser/cfdf_document.h" |
| #include "core/fpdfapi/parser/cpdf_array.h" |
| #include "core/fpdfapi/parser/cpdf_dictionary.h" |
| #include "core/fpdfapi/parser/cpdf_document.h" |
| #include "core/fpdfapi/parser/cpdf_name.h" |
| #include "core/fpdfapi/parser/cpdf_reference.h" |
| #include "core/fpdfapi/parser/cpdf_stream.h" |
| #include "core/fpdfapi/parser/cpdf_string.h" |
| #include "core/fpdfapi/parser/fpdf_parser_utility.h" |
| #include "core/fpdfdoc/cpdf_filespec.h" |
| #include "core/fpdfdoc/cpdf_formcontrol.h" |
| #include "core/fxcrt/check.h" |
| #include "core/fxcrt/compiler_specific.h" |
| #include "core/fxcrt/containers/contains.h" |
| #include "core/fxcrt/fx_codepage.h" |
| #include "core/fxcrt/fx_memcpy_wrappers.h" |
| #include "core/fxcrt/numerics/safe_conversions.h" |
| #include "core/fxcrt/stl_util.h" |
| #include "core/fxge/fx_font.h" |
| |
| #if BUILDFLAG(IS_WIN) |
| #include "core/fxcrt/win/win_util.h" |
| #endif |
| |
| namespace { |
| |
| constexpr int kMaxRecursion = 32; |
| |
| #if BUILDFLAG(IS_WIN) |
| struct PDF_FONTDATA { |
| bool bFind; |
| LOGFONTA lf; |
| }; |
| |
| int CALLBACK EnumFontFamExProc(ENUMLOGFONTEXA* lpelfe, |
| NEWTEXTMETRICEX* lpntme, |
| DWORD FontType, |
| LPARAM lParam) { |
| if (FontType != 0x004 || |
| UNSAFE_TODO(strchr(lpelfe->elfLogFont.lfFaceName, '@'))) { |
| return 1; |
| } |
| |
| PDF_FONTDATA* pData = (PDF_FONTDATA*)lParam; |
| pData->lf = lpelfe->elfLogFont; |
| pData->bFind = true; |
| return 0; |
| } |
| |
| bool RetrieveSpecificFont(FX_Charset charset, |
| LPCSTR pcsFontName, |
| LOGFONTA& lf) { |
| lf = {}; // Aggregate initialization, not construction. |
| static_assert(std::is_aggregate_v<std::remove_reference_t<decltype(lf)>>); |
| lf.lfCharSet = static_cast<int>(charset); |
| lf.lfPitchAndFamily = DEFAULT_PITCH | FF_DONTCARE; |
| if (pcsFontName) { |
| // TODO(dsinclair): Should this be strncpy? |
| // NOLINTNEXTLINE(runtime/printf) |
| UNSAFE_TODO(strcpy(lf.lfFaceName, pcsFontName)); |
| } |
| |
| PDF_FONTDATA fd = {}; // Aggregate initialization, not construction. |
| static_assert(std::is_aggregate_v<decltype(fd)>); |
| HDC hDC = ::GetDC(nullptr); |
| EnumFontFamiliesExA(hDC, &lf, (FONTENUMPROCA)EnumFontFamExProc, (LPARAM)&fd, |
| 0); |
| ::ReleaseDC(nullptr, hDC); |
| if (fd.bFind) { |
| UNSAFE_TODO(FXSYS_memcpy(&lf, &fd.lf, sizeof(LOGFONTA))); |
| } |
| return fd.bFind; |
| } |
| #endif // BUILDFLAG(IS_WIN) |
| |
| ByteString GetNativeFontName(FX_Charset charset, void* log_font) { |
| ByteString font_name; |
| #if BUILDFLAG(IS_WIN) |
| LOGFONTA lf = {}; |
| if (charset == FX_Charset::kANSI) { |
| return CFX_Font::kDefaultAnsiFontName; |
| } |
| |
| if (!pdfium::IsUser32AndGdi32Available()) { |
| // Without GDI32 and User32, GetDC / EnumFontFamiliesExW / ReleaseDC all |
| // fail, which is called by RetrieveSpecificFont. We won't be able to look |
| // up native fonts without GDI. |
| return ByteString(); |
| } |
| |
| bool result = false; |
| const ByteString default_font_name = |
| CFX_Font::GetDefaultFontNameByCharset(charset); |
| if (!default_font_name.IsEmpty()) { |
| result = RetrieveSpecificFont(charset, default_font_name.c_str(), lf); |
| } |
| if (!result) { |
| result = |
| RetrieveSpecificFont(charset, CFX_Font::kUniversalDefaultFontName, lf); |
| } |
| if (!result) { |
| result = RetrieveSpecificFont(charset, "Microsoft Sans Serif", lf); |
| } |
| if (!result) { |
| result = RetrieveSpecificFont(charset, nullptr, lf); |
| } |
| if (result) { |
| if (log_font) { |
| UNSAFE_TODO(FXSYS_memcpy(log_font, &lf, sizeof(LOGFONTA))); |
| } |
| font_name = lf.lfFaceName; |
| } |
| #endif |
| return font_name; |
| } |
| |
| ByteString GenerateNewFontResourceName(const CPDF_Dictionary* resource_dict, |
| ByteString prefix) { |
| static constexpr auto kDummyFontName = pdfium::span_from_cstring("ZiTi"); |
| if (prefix.IsEmpty()) { |
| prefix = ByteStringView(kDummyFontName); |
| } |
| |
| const size_t prefix_length = prefix.GetLength(); |
| size_t m = 0; |
| ByteString actual_prefix; |
| while (m < kDummyFontName.size() && m < prefix_length) { |
| actual_prefix += prefix[m++]; |
| } |
| while (m < kDummyFontName.size()) { |
| actual_prefix += '0' + m % 10; |
| m++; |
| } |
| |
| RetainPtr<const CPDF_Dictionary> pDict = resource_dict->GetDictFor("Font"); |
| DCHECK(pDict); |
| |
| int num = 0; |
| ByteString key_number; |
| while (true) { |
| ByteString key = actual_prefix + key_number; |
| if (!pDict->KeyExist(key.AsStringView())) { |
| return key; |
| } |
| |
| if (m < prefix_length) { |
| actual_prefix += prefix[m++]; |
| } else { |
| key_number = ByteString::FormatInteger(num++); |
| } |
| m++; |
| } |
| } |
| |
| RetainPtr<CPDF_Font> AddStandardFont(CPDF_Document* document) { |
| auto* page_data = CPDF_DocPageData::FromDocument(document); |
| static const CPDF_FontEncoding encoding(FontEncoding::kWinAnsi); |
| return page_data->AddStandardFont(CFX_Font::kDefaultAnsiFontName, &encoding); |
| } |
| |
| RetainPtr<CPDF_Font> AddNativeFont(FX_Charset charset, |
| CPDF_Document* document) { |
| DCHECK(document); |
| |
| #if BUILDFLAG(IS_WIN) |
| LOGFONTA lf; |
| ByteString font_name = GetNativeFontName(charset, &lf); |
| if (!font_name.IsEmpty()) { |
| if (font_name == CFX_Font::kDefaultAnsiFontName) { |
| return AddStandardFont(document); |
| } |
| return CPDF_DocPageData::FromDocument(document)->AddWindowsFont(&lf); |
| } |
| #endif |
| return nullptr; |
| } |
| |
| bool FindFont(const CPDF_Dictionary* form_dict, |
| const CPDF_Font* font, |
| ByteString* name_tag) { |
| RetainPtr<const CPDF_Dictionary> pDR = form_dict->GetDictFor("DR"); |
| if (!pDR) { |
| return false; |
| } |
| |
| RetainPtr<const CPDF_Dictionary> font_dict = pDR->GetDictFor("Font"); |
| // TODO(tsepez): this eventually locks the dict, pass locker instead. |
| if (!ValidateFontResourceDict(font_dict.Get())) { |
| return false; |
| } |
| |
| CPDF_DictionaryLocker locker(std::move(font_dict)); |
| for (const auto& it : locker) { |
| const ByteString& key = it.first; |
| RetainPtr<const CPDF_Dictionary> element = |
| ToDictionary(it.second->GetDirect()); |
| if (!ValidateDictType(element.Get(), "Font")) { |
| continue; |
| } |
| if (font->FontDictIs(element)) { |
| *name_tag = key; |
| return true; |
| } |
| } |
| return false; |
| } |
| |
| bool FindFontFromDoc(const CPDF_Dictionary* form_dict, |
| CPDF_Document* document, |
| ByteString font_name, |
| RetainPtr<CPDF_Font>& font, |
| ByteString* name_tag) { |
| if (font_name.IsEmpty()) { |
| return false; |
| } |
| |
| RetainPtr<const CPDF_Dictionary> pDR = form_dict->GetDictFor("DR"); |
| if (!pDR) { |
| return false; |
| } |
| |
| RetainPtr<const CPDF_Dictionary> font_dict = pDR->GetDictFor("Font"); |
| if (!ValidateFontResourceDict(font_dict.Get())) { |
| return false; |
| } |
| |
| font_name.Remove(' '); |
| CPDF_DictionaryLocker locker(font_dict); |
| for (const auto& it : locker) { |
| const ByteString& key = it.first; |
| RetainPtr<CPDF_Dictionary> element = |
| ToDictionary(it.second->GetMutableDirect()); |
| if (!ValidateDictType(element.Get(), "Font")) { |
| continue; |
| } |
| |
| auto* pData = CPDF_DocPageData::FromDocument(document); |
| font = pData->GetFont(std::move(element)); |
| if (!font) { |
| continue; |
| } |
| |
| ByteString base_font = font->GetBaseFontName(); |
| base_font.Remove(' '); |
| if (base_font == font_name) { |
| *name_tag = key; |
| return true; |
| } |
| } |
| return false; |
| } |
| |
| void AddFont(CPDF_Dictionary* form_dict, |
| CPDF_Document* document, |
| const RetainPtr<CPDF_Font>& font, |
| ByteString* name_tag) { |
| DCHECK(form_dict); |
| DCHECK(font); |
| |
| ByteString tag; |
| if (FindFont(form_dict, font.Get(), &tag)) { |
| *name_tag = std::move(tag); |
| return; |
| } |
| |
| RetainPtr<CPDF_Dictionary> pDR = form_dict->GetOrCreateDictFor("DR"); |
| RetainPtr<CPDF_Dictionary> font_dict = pDR->GetOrCreateDictFor("Font"); |
| |
| if (name_tag->IsEmpty()) { |
| *name_tag = font->GetBaseFontName(); |
| } |
| |
| name_tag->Remove(' '); |
| *name_tag = GenerateNewFontResourceName(pDR.Get(), *name_tag); |
| font_dict->SetNewFor<CPDF_Reference>(*name_tag, document, |
| font->GetFontDictObjNum()); |
| } |
| |
| FX_Charset GetNativeCharSet() { |
| return FX_GetCharsetFromCodePage(FX_GetACP()); |
| } |
| |
| RetainPtr<CPDF_Dictionary> InitDict(CPDF_Document* document) { |
| auto form_dict = document->NewIndirect<CPDF_Dictionary>(); |
| document->GetMutableRoot()->SetNewFor<CPDF_Reference>("AcroForm", document, |
| form_dict->GetObjNum()); |
| |
| ByteString base_name; |
| FX_Charset charset = GetNativeCharSet(); |
| RetainPtr<CPDF_Font> font = AddStandardFont(document); |
| if (font) { |
| AddFont(form_dict.Get(), document, font, &base_name); |
| } |
| if (charset != FX_Charset::kANSI) { |
| ByteString font_name = GetNativeFontName(charset, nullptr); |
| if (!font || font_name != CFX_Font::kDefaultAnsiFontName) { |
| RetainPtr<CPDF_Font> native_font = AddNativeFont(charset, document); |
| if (native_font) { |
| base_name.clear(); |
| AddFont(form_dict.Get(), document, native_font, &base_name); |
| font = std::move(native_font); |
| } |
| } |
| } |
| ByteString default_appearance; |
| if (font) { |
| default_appearance = "/" + PDF_NameEncode(base_name) + " 12 Tf "; |
| } |
| default_appearance += "0 g"; |
| form_dict->SetNewFor<CPDF_String>("DA", std::move(default_appearance)); |
| return form_dict; |
| } |
| |
| RetainPtr<CPDF_Font> GetNativeFont(const CPDF_Dictionary* form_dict, |
| CPDF_Document* document, |
| FX_Charset charset, |
| ByteString* name_tag) { |
| RetainPtr<const CPDF_Dictionary> pDR = form_dict->GetDictFor("DR"); |
| if (!pDR) { |
| return nullptr; |
| } |
| |
| RetainPtr<const CPDF_Dictionary> font_dict = pDR->GetDictFor("Font"); |
| if (!ValidateFontResourceDict(font_dict.Get())) { |
| return nullptr; |
| } |
| |
| CPDF_DictionaryLocker locker(font_dict); |
| for (const auto& it : locker) { |
| const ByteString& key = it.first; |
| RetainPtr<CPDF_Dictionary> element = |
| ToDictionary(it.second->GetMutableDirect()); |
| if (!ValidateDictType(element.Get(), "Font")) { |
| continue; |
| } |
| |
| auto* pData = CPDF_DocPageData::FromDocument(document); |
| RetainPtr<CPDF_Font> pFind = pData->GetFont(std::move(element)); |
| if (!pFind) { |
| continue; |
| } |
| |
| auto maybe_charset = pFind->GetSubstFontCharset(); |
| if (maybe_charset.has_value() && maybe_charset.value() == charset) { |
| *name_tag = key; |
| return pFind; |
| } |
| } |
| return nullptr; |
| } |
| |
| class CFieldNameExtractor { |
| public: |
| explicit CFieldNameExtractor(const WideString& full_name) |
| : full_name_(full_name) {} |
| |
| WideStringView GetNext() { |
| size_t start_pos = cur_; |
| while (cur_ < full_name_.GetLength() && full_name_[cur_] != L'.') { |
| ++cur_; |
| } |
| |
| size_t length = cur_ - start_pos; |
| if (cur_ < full_name_.GetLength() && full_name_[cur_] == L'.') { |
| ++cur_; |
| } |
| |
| return full_name_.AsStringView().Substr(start_pos, length); |
| } |
| |
| protected: |
| const WideString full_name_; |
| size_t cur_ = 0; |
| }; |
| |
| } // namespace |
| |
| class CFieldTree { |
| public: |
| class Node { |
| public: |
| Node() : level_(0) {} |
| Node(const WideString& short_name, int level) |
| : short_name_(short_name), level_(level) {} |
| ~Node() = default; |
| |
| void AddChildNode(std::unique_ptr<Node> node) { |
| children_.push_back(std::move(node)); |
| } |
| |
| size_t GetChildrenCount() const { return children_.size(); } |
| |
| Node* GetChildAt(size_t i) { return children_[i].get(); } |
| const Node* GetChildAt(size_t i) const { return children_[i].get(); } |
| |
| CPDF_FormField* GetFieldAtIndex(size_t index) { |
| size_t nFieldsToGo = index; |
| return GetFieldInternal(&nFieldsToGo); |
| } |
| |
| size_t CountFields() const { return CountFieldsInternal(); } |
| |
| void SetField(std::unique_ptr<CPDF_FormField> field) { |
| field_ = std::move(field); |
| } |
| |
| CPDF_FormField* GetField() const { return field_.get(); } |
| WideString GetShortName() const { return short_name_; } |
| int GetLevel() const { return level_; } |
| |
| private: |
| CPDF_FormField* GetFieldInternal(size_t* pFieldsToGo) { |
| if (field_) { |
| if (*pFieldsToGo == 0) { |
| return field_.get(); |
| } |
| |
| --*pFieldsToGo; |
| } |
| for (size_t i = 0; i < GetChildrenCount(); ++i) { |
| CPDF_FormField* field = GetChildAt(i)->GetFieldInternal(pFieldsToGo); |
| if (field) { |
| return field; |
| } |
| } |
| return nullptr; |
| } |
| |
| size_t CountFieldsInternal() const { |
| size_t count = 0; |
| if (field_) { |
| ++count; |
| } |
| |
| for (size_t i = 0; i < GetChildrenCount(); ++i) { |
| count += GetChildAt(i)->CountFieldsInternal(); |
| } |
| return count; |
| } |
| |
| std::vector<std::unique_ptr<Node>> children_; |
| WideString short_name_; |
| std::unique_ptr<CPDF_FormField> field_; |
| const int level_; |
| }; |
| |
| CFieldTree(); |
| ~CFieldTree(); |
| |
| bool SetField(const WideString& full_name, |
| std::unique_ptr<CPDF_FormField> field); |
| CPDF_FormField* GetField(const WideString& full_name); |
| |
| Node* GetRoot() { return root_.get(); } |
| Node* FindNode(const WideString& full_name); |
| Node* AddChild(Node* pParent, const WideString& short_name); |
| Node* Lookup(Node* pParent, WideStringView short_name); |
| |
| private: |
| std::unique_ptr<Node> root_; |
| }; |
| |
| CFieldTree::CFieldTree() : root_(std::make_unique<Node>()) {} |
| |
| CFieldTree::~CFieldTree() = default; |
| |
| CFieldTree::Node* CFieldTree::AddChild(Node* pParent, |
| const WideString& short_name) { |
| if (!pParent) { |
| return nullptr; |
| } |
| |
| int level = pParent->GetLevel() + 1; |
| if (level > kMaxRecursion) { |
| return nullptr; |
| } |
| |
| auto new_node = std::make_unique<Node>(short_name, pParent->GetLevel() + 1); |
| Node* pChild = new_node.get(); |
| pParent->AddChildNode(std::move(new_node)); |
| return pChild; |
| } |
| |
| CFieldTree::Node* CFieldTree::Lookup(Node* pParent, WideStringView short_name) { |
| if (!pParent) { |
| return nullptr; |
| } |
| |
| for (size_t i = 0; i < pParent->GetChildrenCount(); ++i) { |
| Node* node = pParent->GetChildAt(i); |
| if (node->GetShortName() == short_name) { |
| return node; |
| } |
| } |
| return nullptr; |
| } |
| |
| bool CFieldTree::SetField(const WideString& full_name, |
| std::unique_ptr<CPDF_FormField> field) { |
| if (full_name.IsEmpty()) { |
| return false; |
| } |
| |
| Node* node = GetRoot(); |
| Node* last_node = nullptr; |
| CFieldNameExtractor name_extractor(full_name); |
| while (true) { |
| WideStringView name_view = name_extractor.GetNext(); |
| if (name_view.IsEmpty()) { |
| break; |
| } |
| last_node = node; |
| node = Lookup(last_node, name_view); |
| if (node) { |
| continue; |
| } |
| node = AddChild(last_node, WideString(name_view)); |
| if (!node) { |
| return false; |
| } |
| } |
| if (node == GetRoot()) { |
| return false; |
| } |
| |
| node->SetField(std::move(field)); |
| return true; |
| } |
| |
| CPDF_FormField* CFieldTree::GetField(const WideString& full_name) { |
| if (full_name.IsEmpty()) { |
| return nullptr; |
| } |
| |
| Node* node = GetRoot(); |
| Node* last_node = nullptr; |
| CFieldNameExtractor name_extractor(full_name); |
| while (node) { |
| WideStringView name_view = name_extractor.GetNext(); |
| if (name_view.IsEmpty()) { |
| break; |
| } |
| last_node = node; |
| node = Lookup(last_node, name_view); |
| } |
| return node ? node->GetField() : nullptr; |
| } |
| |
| CFieldTree::Node* CFieldTree::FindNode(const WideString& full_name) { |
| if (full_name.IsEmpty()) { |
| return nullptr; |
| } |
| |
| Node* node = GetRoot(); |
| Node* last_node = nullptr; |
| CFieldNameExtractor name_extractor(full_name); |
| while (node) { |
| WideStringView name_view = name_extractor.GetNext(); |
| if (name_view.IsEmpty()) { |
| break; |
| } |
| last_node = node; |
| node = Lookup(last_node, name_view); |
| } |
| return node; |
| } |
| |
| CPDF_InteractiveForm::CPDF_InteractiveForm(CPDF_Document* document) |
| : document_(document), field_tree_(std::make_unique<CFieldTree>()) { |
| RetainPtr<CPDF_Dictionary> pRoot = document_->GetMutableRoot(); |
| if (!pRoot) { |
| return; |
| } |
| |
| form_dict_ = pRoot->GetMutableDictFor("AcroForm"); |
| if (!form_dict_) { |
| return; |
| } |
| |
| RetainPtr<CPDF_Array> fields = form_dict_->GetMutableArrayFor("Fields"); |
| if (!fields) { |
| return; |
| } |
| |
| for (size_t i = 0; i < fields->size(); ++i) { |
| LoadField(fields->GetMutableDictAt(i), 0); |
| } |
| } |
| |
| CPDF_InteractiveForm::~CPDF_InteractiveForm() = default; |
| |
| bool CPDF_InteractiveForm::s_bUpdateAP = true; |
| |
| // static |
| bool CPDF_InteractiveForm::IsUpdateAPEnabled() { |
| return s_bUpdateAP; |
| } |
| |
| // static |
| void CPDF_InteractiveForm::SetUpdateAP(bool bUpdateAP) { |
| s_bUpdateAP = bUpdateAP; |
| } |
| |
| // static |
| RetainPtr<CPDF_Font> CPDF_InteractiveForm::AddNativeInteractiveFormFont( |
| CPDF_Document* document, |
| ByteString* name_tag) { |
| DCHECK(document); |
| DCHECK(name_tag); |
| |
| RetainPtr<CPDF_Dictionary> form_dict = |
| document->GetMutableRoot()->GetMutableDictFor("AcroForm"); |
| if (!form_dict) { |
| form_dict = InitDict(document); |
| } |
| |
| FX_Charset charset = GetNativeCharSet(); |
| ByteString tag; |
| RetainPtr<CPDF_Font> font = |
| GetNativeFont(form_dict.Get(), document, charset, &tag); |
| if (font) { |
| *name_tag = std::move(tag); |
| return font; |
| } |
| ByteString font_name = GetNativeFontName(charset, nullptr); |
| if (FindFontFromDoc(form_dict.Get(), document, font_name, font, name_tag)) { |
| return font; |
| } |
| |
| font = AddNativeFont(charset, document); |
| if (!font) { |
| return nullptr; |
| } |
| |
| AddFont(form_dict.Get(), document, font, name_tag); |
| return font; |
| } |
| |
| // static |
| RetainPtr<CPDF_Dictionary> CPDF_InteractiveForm::InitAcroFormDict( |
| CPDF_Document* document) { |
| return InitDict(document); |
| } |
| |
| size_t CPDF_InteractiveForm::CountFields(const WideString& field_name) const { |
| if (field_name.IsEmpty()) { |
| return field_tree_->GetRoot()->CountFields(); |
| } |
| |
| CFieldTree::Node* node = field_tree_->FindNode(field_name); |
| return node ? node->CountFields() : 0; |
| } |
| |
| CPDF_FormField* CPDF_InteractiveForm::GetField( |
| size_t index, |
| const WideString& field_name) const { |
| if (field_name.IsEmpty()) { |
| return field_tree_->GetRoot()->GetFieldAtIndex(index); |
| } |
| |
| CFieldTree::Node* node = field_tree_->FindNode(field_name); |
| return node ? node->GetFieldAtIndex(index) : nullptr; |
| } |
| |
| CPDF_FormField* CPDF_InteractiveForm::GetFieldByDict( |
| const CPDF_Dictionary* field_dict) const { |
| if (!field_dict) { |
| return nullptr; |
| } |
| |
| return field_tree_->GetField(CPDF_FormField::GetFullNameForDict(field_dict)); |
| } |
| |
| const CPDF_FormControl* CPDF_InteractiveForm::GetControlAtPoint( |
| const CPDF_Page* page, |
| const CFX_PointF& point, |
| int* z_order) const { |
| RetainPtr<const CPDF_Array> annots = page->GetAnnotsArray(); |
| if (!annots) { |
| return nullptr; |
| } |
| |
| for (size_t i = annots->size(); i > 0; --i) { |
| size_t annot_index = i - 1; |
| RetainPtr<const CPDF_Dictionary> annot = annots->GetDictAt(annot_index); |
| if (!annot) { |
| continue; |
| } |
| |
| const auto it = control_map_.find(annot.Get()); |
| if (it == control_map_.end()) { |
| continue; |
| } |
| |
| const CPDF_FormControl* control = it->second.get(); |
| if (!control->GetRect().Contains(point)) { |
| continue; |
| } |
| |
| if (z_order) { |
| *z_order = static_cast<int>(annot_index); |
| } |
| return control; |
| } |
| return nullptr; |
| } |
| |
| CPDF_FormControl* CPDF_InteractiveForm::GetControlByDict( |
| const CPDF_Dictionary* widget_dict) const { |
| const auto it = control_map_.find(widget_dict); |
| return it != control_map_.end() ? it->second.get() : nullptr; |
| } |
| |
| bool CPDF_InteractiveForm::NeedConstructAP() const { |
| return form_dict_ && form_dict_->GetBooleanFor("NeedAppearances", false); |
| } |
| |
| int CPDF_InteractiveForm::CountFieldsInCalculationOrder() { |
| if (!form_dict_) { |
| return 0; |
| } |
| |
| RetainPtr<const CPDF_Array> pArray = form_dict_->GetArrayFor("CO"); |
| return pArray ? fxcrt::CollectionSize<int>(*pArray) : 0; |
| } |
| |
| CPDF_FormField* CPDF_InteractiveForm::GetFieldInCalculationOrder(int index) { |
| if (!form_dict_ || index < 0) { |
| return nullptr; |
| } |
| |
| RetainPtr<const CPDF_Array> pArray = form_dict_->GetArrayFor("CO"); |
| if (!pArray) { |
| return nullptr; |
| } |
| |
| RetainPtr<const CPDF_Dictionary> element = |
| ToDictionary(pArray->GetDirectObjectAt(index)); |
| return element ? GetFieldByDict(element.Get()) : nullptr; |
| } |
| |
| int CPDF_InteractiveForm::FindFieldInCalculationOrder( |
| const CPDF_FormField* field) { |
| if (!form_dict_) { |
| return -1; |
| } |
| |
| RetainPtr<const CPDF_Array> pArray = form_dict_->GetArrayFor("CO"); |
| if (!pArray) { |
| return -1; |
| } |
| |
| std::optional<size_t> maybe_found = pArray->Find(field->GetFieldDict()); |
| if (!maybe_found.has_value()) { |
| return -1; |
| } |
| |
| return pdfium::checked_cast<int>(maybe_found.value()); |
| } |
| |
| RetainPtr<CPDF_Font> CPDF_InteractiveForm::GetFormFont( |
| ByteString name_tag) const { |
| if (!form_dict_) { |
| return nullptr; |
| } |
| ByteString alias = PDF_NameDecode(name_tag.AsStringView()); |
| if (alias.IsEmpty()) { |
| return nullptr; |
| } |
| |
| RetainPtr<CPDF_Dictionary> pDR = form_dict_->GetMutableDictFor("DR"); |
| if (!pDR) { |
| return nullptr; |
| } |
| |
| RetainPtr<CPDF_Dictionary> font_dict = pDR->GetMutableDictFor("Font"); |
| if (!ValidateFontResourceDict(font_dict.Get())) { |
| return nullptr; |
| } |
| |
| RetainPtr<CPDF_Dictionary> element = font_dict->GetMutableDictFor(alias); |
| if (!ValidateDictType(element.Get(), "Font")) { |
| return nullptr; |
| } |
| |
| return GetFontForElement(std::move(element)); |
| } |
| |
| RetainPtr<CPDF_Font> CPDF_InteractiveForm::GetFontForElement( |
| RetainPtr<CPDF_Dictionary> element) const { |
| auto* pData = CPDF_DocPageData::FromDocument(document_); |
| return pData->GetFont(std::move(element)); |
| } |
| |
| CPDF_DefaultAppearance CPDF_InteractiveForm::GetDefaultAppearance() const { |
| return CPDF_DefaultAppearance(form_dict_ ? form_dict_->GetByteStringFor("DA") |
| : ""); |
| } |
| |
| int CPDF_InteractiveForm::GetFormAlignment() const { |
| return form_dict_ ? form_dict_->GetIntegerFor("Q", 0) : 0; |
| } |
| |
| void CPDF_InteractiveForm::ResetForm(pdfium::span<CPDF_FormField*> fields, |
| bool bIncludeOrExclude) { |
| CFieldTree::Node* pRoot = field_tree_->GetRoot(); |
| const size_t nCount = pRoot->CountFields(); |
| for (size_t i = 0; i < nCount; ++i) { |
| CPDF_FormField* field = pRoot->GetFieldAtIndex(i); |
| if (!field) { |
| continue; |
| } |
| |
| if (bIncludeOrExclude == pdfium::Contains(fields, field)) { |
| field->ResetField(); |
| } |
| } |
| if (form_notify_) { |
| form_notify_->AfterFormReset(this); |
| } |
| } |
| |
| void CPDF_InteractiveForm::ResetForm() { |
| ResetForm(/*fields=*/{}, /*bIncludeOrExclude=*/false); |
| } |
| |
| const std::vector<UnownedPtr<CPDF_FormControl>>& |
| CPDF_InteractiveForm::GetControlsForField(const CPDF_FormField* field) { |
| return control_lists_[pdfium::WrapUnowned(field)]; |
| } |
| |
| void CPDF_InteractiveForm::LoadField(RetainPtr<CPDF_Dictionary> field_dict, |
| int nLevel) { |
| if (nLevel > kMaxRecursion) { |
| return; |
| } |
| if (!field_dict) { |
| return; |
| } |
| |
| uint32_t dwParentObjNum = field_dict->GetObjNum(); |
| RetainPtr<CPDF_Array> kids = |
| field_dict->GetMutableArrayFor(pdfium::form_fields::kKids); |
| if (!kids) { |
| AddTerminalField(std::move(field_dict)); |
| return; |
| } |
| |
| RetainPtr<const CPDF_Dictionary> pFirstKid = kids->GetDictAt(0); |
| if (!pFirstKid) { |
| return; |
| } |
| |
| if (!pFirstKid->KeyExist(pdfium::form_fields::kT) && |
| !pFirstKid->KeyExist(pdfium::form_fields::kKids)) { |
| AddTerminalField(std::move(field_dict)); |
| return; |
| } |
| for (size_t i = 0; i < kids->size(); i++) { |
| RetainPtr<CPDF_Dictionary> pChildDict = kids->GetMutableDictAt(i); |
| if (pChildDict && pChildDict->GetObjNum() != dwParentObjNum) { |
| LoadField(std::move(pChildDict), nLevel + 1); |
| } |
| } |
| } |
| |
| void CPDF_InteractiveForm::FixPageFields(CPDF_Page* page) { |
| RetainPtr<CPDF_Array> annots = page->GetMutableAnnotsArray(); |
| if (!annots) { |
| return; |
| } |
| |
| for (size_t i = 0; i < annots->size(); i++) { |
| RetainPtr<CPDF_Dictionary> annot = annots->GetMutableDictAt(i); |
| if (annot && annot->GetNameFor("Subtype") == "Widget") { |
| LoadField(std::move(annot), 0); |
| } |
| } |
| } |
| |
| void CPDF_InteractiveForm::AddTerminalField( |
| RetainPtr<CPDF_Dictionary> field_dict) { |
| if (!field_dict->KeyExist(pdfium::form_fields::kFT)) { |
| // Key "FT" is required for terminal fields, it is also inheritable. |
| RetainPtr<const CPDF_Dictionary> pParentDict = |
| field_dict->GetDictFor(pdfium::form_fields::kParent); |
| if (!pParentDict || !pParentDict->KeyExist(pdfium::form_fields::kFT)) { |
| return; |
| } |
| } |
| |
| WideString field_name = CPDF_FormField::GetFullNameForDict(field_dict.Get()); |
| if (field_name.IsEmpty()) { |
| return; |
| } |
| |
| CPDF_FormField* field = field_tree_->GetField(field_name); |
| if (!field) { |
| RetainPtr<CPDF_Dictionary> pParent(field_dict); |
| if (!field_dict->KeyExist(pdfium::form_fields::kT) && |
| field_dict->GetNameFor("Subtype") == "Widget") { |
| pParent = field_dict->GetMutableDictFor(pdfium::form_fields::kParent); |
| if (!pParent) { |
| pParent = field_dict; |
| } |
| } |
| |
| if (pParent && pParent != field_dict && |
| !pParent->KeyExist(pdfium::form_fields::kFT)) { |
| if (field_dict->KeyExist(pdfium::form_fields::kFT)) { |
| RetainPtr<const CPDF_Object> pFTValue = |
| field_dict->GetDirectObjectFor(pdfium::form_fields::kFT); |
| if (pFTValue) { |
| pParent->SetFor(pdfium::form_fields::kFT, pFTValue->Clone()); |
| } |
| } |
| |
| if (field_dict->KeyExist(pdfium::form_fields::kFf)) { |
| RetainPtr<const CPDF_Object> pFfValue = |
| field_dict->GetDirectObjectFor(pdfium::form_fields::kFf); |
| if (pFfValue) { |
| pParent->SetFor(pdfium::form_fields::kFf, pFfValue->Clone()); |
| } |
| } |
| } |
| |
| auto new_field = std::make_unique<CPDF_FormField>(this, std::move(pParent)); |
| field = new_field.get(); |
| RetainPtr<const CPDF_Object> t_obj = |
| field_dict->GetObjectFor(pdfium::form_fields::kT); |
| if (ToReference(t_obj)) { |
| RetainPtr<CPDF_Object> t_obj_clone = t_obj->CloneDirectObject(); |
| if (t_obj_clone && t_obj_clone->IsString()) { |
| field_dict->SetFor(pdfium::form_fields::kT, std::move(t_obj_clone)); |
| } else { |
| field_dict->SetNewFor<CPDF_String>(pdfium::form_fields::kT, |
| ByteString()); |
| } |
| } |
| if (!field_tree_->SetField(field_name, std::move(new_field))) { |
| return; |
| } |
| } |
| |
| RetainPtr<CPDF_Array> kids = |
| field_dict->GetMutableArrayFor(pdfium::form_fields::kKids); |
| if (!kids) { |
| if (field_dict->GetNameFor("Subtype") == "Widget") { |
| AddControl(field, std::move(field_dict)); |
| } |
| return; |
| } |
| for (size_t i = 0; i < kids->size(); i++) { |
| RetainPtr<CPDF_Dictionary> kid = kids->GetMutableDictAt(i); |
| if (kid && kid->GetNameFor("Subtype") == "Widget") { |
| AddControl(field, std::move(kid)); |
| } |
| } |
| } |
| |
| CPDF_FormControl* CPDF_InteractiveForm::AddControl( |
| CPDF_FormField* field, |
| RetainPtr<CPDF_Dictionary> widget_dict) { |
| DCHECK(widget_dict); |
| const auto it = control_map_.find(widget_dict.Get()); |
| if (it != control_map_.end()) { |
| return it->second.get(); |
| } |
| |
| auto new_control = |
| std::make_unique<CPDF_FormControl>(field, widget_dict, this); |
| CPDF_FormControl* control = new_control.get(); |
| control_map_[widget_dict] = std::move(new_control); |
| control_lists_[pdfium::WrapUnowned(field)].emplace_back(control); |
| return control; |
| } |
| |
| bool CPDF_InteractiveForm::CheckRequiredFields( |
| const std::vector<CPDF_FormField*>* fields, |
| bool bIncludeOrExclude) const { |
| CFieldTree::Node* pRoot = field_tree_->GetRoot(); |
| const size_t nCount = pRoot->CountFields(); |
| for (size_t i = 0; i < nCount; ++i) { |
| CPDF_FormField* field = pRoot->GetFieldAtIndex(i); |
| if (!field) { |
| continue; |
| } |
| |
| int32_t iType = field->GetType(); |
| if (iType == CPDF_FormField::kPushButton || |
| iType == CPDF_FormField::kCheckBox || |
| iType == CPDF_FormField::kListBox) { |
| continue; |
| } |
| if (field->IsNoExport()) { |
| continue; |
| } |
| |
| bool bFind = true; |
| if (fields) { |
| bFind = pdfium::Contains(*fields, field); |
| } |
| if (bIncludeOrExclude == bFind) { |
| RetainPtr<const CPDF_Dictionary> field_dict = field->GetFieldDict(); |
| if (field->IsRequired() && |
| field_dict->GetByteStringFor(pdfium::form_fields::kV).IsEmpty()) { |
| return false; |
| } |
| } |
| } |
| return true; |
| } |
| |
| std::unique_ptr<CFDF_Document> CPDF_InteractiveForm::ExportToFDF( |
| const WideString& pdf_path) const { |
| std::vector<CPDF_FormField*> fields; |
| CFieldTree::Node* pRoot = field_tree_->GetRoot(); |
| const size_t nCount = pRoot->CountFields(); |
| for (size_t i = 0; i < nCount; ++i) { |
| fields.push_back(pRoot->GetFieldAtIndex(i)); |
| } |
| return ExportToFDF(pdf_path, fields, true); |
| } |
| |
| std::unique_ptr<CFDF_Document> CPDF_InteractiveForm::ExportToFDF( |
| const WideString& pdf_path, |
| const std::vector<CPDF_FormField*>& fields, |
| bool bIncludeOrExclude) const { |
| std::unique_ptr<CFDF_Document> pDoc = CFDF_Document::CreateNewDoc(); |
| if (!pDoc) { |
| return nullptr; |
| } |
| |
| RetainPtr<CPDF_Dictionary> pMainDict = |
| pDoc->GetMutableRoot()->GetMutableDictFor("FDF"); |
| if (!pdf_path.IsEmpty()) { |
| auto new_dict = pDoc->New<CPDF_Dictionary>(); |
| new_dict->SetNewFor<CPDF_Name>("Type", "Filespec"); |
| WideString wsStr = CPDF_FileSpec::EncodeFileName(pdf_path); |
| new_dict->SetNewFor<CPDF_String>(pdfium::stream::kF, wsStr.ToDefANSI()); |
| new_dict->SetNewFor<CPDF_String>("UF", wsStr.AsStringView()); |
| pMainDict->SetFor("F", new_dict); |
| } |
| |
| auto fields_array = pMainDict->SetNewFor<CPDF_Array>("Fields"); |
| CFieldTree::Node* pRoot = field_tree_->GetRoot(); |
| const size_t nCount = pRoot->CountFields(); |
| for (size_t i = 0; i < nCount; ++i) { |
| CPDF_FormField* field = pRoot->GetFieldAtIndex(i); |
| if (!field || field->GetType() == CPDF_FormField::kPushButton) { |
| continue; |
| } |
| |
| uint32_t dwFlags = field->GetFieldFlags(); |
| if (dwFlags & pdfium::form_flags::kNoExport) { |
| continue; |
| } |
| |
| if (bIncludeOrExclude != pdfium::Contains(fields, field)) { |
| continue; |
| } |
| |
| if ((dwFlags & pdfium::form_flags::kRequired) != 0 && |
| field->GetFieldDict() |
| ->GetByteStringFor(pdfium::form_fields::kV) |
| .IsEmpty()) { |
| continue; |
| } |
| |
| WideString fullname = |
| CPDF_FormField::GetFullNameForDict(field->GetFieldDict()); |
| auto field_dict = pDoc->New<CPDF_Dictionary>(); |
| field_dict->SetNewFor<CPDF_String>(pdfium::form_fields::kT, |
| fullname.AsStringView()); |
| if (field->GetType() == CPDF_FormField::kCheckBox || |
| field->GetType() == CPDF_FormField::kRadioButton) { |
| ByteString export_value = |
| PDF_EncodeText(field->GetCheckValue(false).AsStringView()); |
| RetainPtr<const CPDF_Object> opt = field->GetFieldAttr("Opt"); |
| if (opt) { |
| field_dict->SetNewFor<CPDF_String>(pdfium::form_fields::kV, |
| export_value); |
| } else { |
| field_dict->SetNewFor<CPDF_Name>(pdfium::form_fields::kV, export_value); |
| } |
| } else { |
| RetainPtr<const CPDF_Object> value = |
| field->GetFieldAttr(pdfium::form_fields::kV); |
| if (value) { |
| field_dict->SetFor(pdfium::form_fields::kV, value->CloneDirectObject()); |
| } |
| } |
| fields_array->Append(field_dict); |
| } |
| return pDoc; |
| } |
| |
| void CPDF_InteractiveForm::SetNotifierIface(NotifierIface* notify) { |
| form_notify_ = notify; |
| } |
| |
| bool CPDF_InteractiveForm::NotifyBeforeValueChange(CPDF_FormField* field, |
| const WideString& value) { |
| return !form_notify_ || form_notify_->BeforeValueChange(field, value); |
| } |
| |
| void CPDF_InteractiveForm::NotifyAfterValueChange(CPDF_FormField* field) { |
| if (form_notify_) { |
| form_notify_->AfterValueChange(field); |
| } |
| } |
| |
| bool CPDF_InteractiveForm::NotifyBeforeSelectionChange( |
| CPDF_FormField* field, |
| const WideString& value) { |
| return !form_notify_ || form_notify_->BeforeSelectionChange(field, value); |
| } |
| |
| void CPDF_InteractiveForm::NotifyAfterSelectionChange(CPDF_FormField* field) { |
| if (form_notify_) { |
| form_notify_->AfterSelectionChange(field); |
| } |
| } |
| |
| void CPDF_InteractiveForm::NotifyAfterCheckedStatusChange( |
| CPDF_FormField* field) { |
| if (form_notify_) { |
| form_notify_->AfterCheckedStatusChange(field); |
| } |
| } |