From d19aa7c572c252dd0398f277d13ec1733f1c7098 Mon Sep 17 00:00:00 2001 From: Dan Sinclair Date: Mon, 6 Nov 2017 17:10:21 +0000 Subject: Move CXFA_ResolveProcessor to CFXJSE_ResolveProcessor. The resolve processor is only used by cfxjse classes, so move to live in fxjs/. Change-Id: I41b8c8e0a491a2b4ee04c1410cbddef63d843105 Reviewed-on: https://pdfium-review.googlesource.com/17850 Reviewed-by: Tom Sepez Commit-Queue: dsinclair --- fxjs/cfxjse_engine.cpp | 10 +- fxjs/cfxjse_engine.h | 4 +- fxjs/cfxjse_resolveprocessor.cpp | 799 +++++++++++++++++++++++++++++++++++++++ fxjs/cfxjse_resolveprocessor.h | 80 ++++ 4 files changed, 886 insertions(+), 7 deletions(-) create mode 100644 fxjs/cfxjse_resolveprocessor.cpp create mode 100644 fxjs/cfxjse_resolveprocessor.h (limited to 'fxjs') diff --git a/fxjs/cfxjse_engine.cpp b/fxjs/cfxjse_engine.cpp index fb36c6dba0..6d990f66fb 100644 --- a/fxjs/cfxjse_engine.cpp +++ b/fxjs/cfxjse_engine.cpp @@ -13,6 +13,7 @@ #include "core/fxcrt/fx_extension.h" #include "fxjs/cfxjse_arguments.h" #include "fxjs/cfxjse_class.h" +#include "fxjs/cfxjse_resolveprocessor.h" #include "fxjs/cfxjse_value.h" #include "third_party/base/ptr_util.h" #include "third_party/base/stl_util.h" @@ -24,7 +25,6 @@ #include "xfa/fxfa/parser/cxfa_nodehelper.h" #include "xfa/fxfa/parser/cxfa_nodelist.h" #include "xfa/fxfa/parser/cxfa_object.h" -#include "xfa/fxfa/parser/cxfa_resolveprocessor.h" #include "xfa/fxfa/parser/cxfa_thisproxy.h" #include "xfa/fxfa/parser/xfa_basic_data.h" #include "xfa/fxfa/parser/xfa_resolvenode_rs.h" @@ -142,7 +142,7 @@ void CFXJSE_Engine::Initialize(v8::Isolate* pIsolate) { m_pIsolate = pIsolate; DefineJsContext(); DefineJsClass(); - m_ResolveProcessor = pdfium::MakeUnique(); + m_ResolveProcessor = pdfium::MakeUnique(); } bool CFXJSE_Engine::RunScript(XFA_SCRIPTLANGTYPE eScriptType, @@ -574,11 +574,11 @@ int32_t CFXJSE_Engine::ResolveObjects(CXFA_Object* refObject, m_ResolveProcessor->GetNodeHelper()->m_pCreateParent = nullptr; m_ResolveProcessor->GetNodeHelper()->m_iCurAllStart = -1; - CXFA_ResolveNodesData rndFind; + CFXJSE_ResolveNodeData rndFind(this); int32_t nStart = 0; int32_t nLevel = 0; int32_t nRet = -1; - rndFind.m_pSC = this; + std::vector findObjects; findObjects.push_back(refObject ? refObject : m_pDocument->GetRoot()); int32_t nNodes = 0; @@ -633,7 +633,7 @@ int32_t CFXJSE_Engine::ResolveObjects(CXFA_Object* refObject, if (((dwStyles & XFA_RESOLVENODE_Bind) || (dwStyles & XFA_RESOLVENODE_CreateNode)) && nNodes > 1) { - CXFA_ResolveNodesData rndBind; + CFXJSE_ResolveNodeData rndBind(nullptr); m_ResolveProcessor->GetFilter(wsExpression, nStart, rndBind); m_ResolveProcessor->SetIndexDataBind(rndBind.m_wsCondition, i, nNodes); bDataBind = true; diff --git a/fxjs/cfxjse_engine.h b/fxjs/cfxjse_engine.h index 34383cac76..f901850203 100644 --- a/fxjs/cfxjse_engine.h +++ b/fxjs/cfxjse_engine.h @@ -19,7 +19,7 @@ #define XFA_RESOLVENODE_TagName 0x0002 -class CXFA_ResolveProcessor; +class CFXJSE_ResolveProcessor; class CFXJSE_Engine { public: @@ -116,7 +116,7 @@ class CFXJSE_Engine { // CacheList holds the NodeList items so we can clean them up when we're done. std::vector> m_CacheList; std::vector* m_pScriptNodeArray; - std::unique_ptr m_ResolveProcessor; + std::unique_ptr m_ResolveProcessor; std::unique_ptr m_FM2JSContext; CXFA_Object* m_pThisObject; uint32_t m_dwBuiltInInFlags; diff --git a/fxjs/cfxjse_resolveprocessor.cpp b/fxjs/cfxjse_resolveprocessor.cpp new file mode 100644 index 0000000000..de4800119b --- /dev/null +++ b/fxjs/cfxjse_resolveprocessor.cpp @@ -0,0 +1,799 @@ +// Copyright 2014 PDFium Authors. All rights reserved. +// 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 "fxjs/cfxjse_resolveprocessor.h" + +#include +#include +#include + +#include "core/fxcrt/fx_extension.h" +#include "fxjs/cfxjse_engine.h" +#include "third_party/base/ptr_util.h" +#include "third_party/base/stl_util.h" +#include "xfa/fxfa/parser/cxfa_document.h" +#include "xfa/fxfa/parser/cxfa_localemgr.h" +#include "xfa/fxfa/parser/cxfa_node.h" +#include "xfa/fxfa/parser/cxfa_nodehelper.h" +#include "xfa/fxfa/parser/cxfa_object.h" +#include "xfa/fxfa/parser/xfa_resolvenode_rs.h" +#include "xfa/fxfa/parser/xfa_utils.h" + +CFXJSE_ResolveProcessor::CFXJSE_ResolveProcessor() + : m_iCurStart(0), m_pNodeHelper(pdfium::MakeUnique()) {} + +CFXJSE_ResolveProcessor::~CFXJSE_ResolveProcessor() {} + +int32_t CFXJSE_ResolveProcessor::Resolve(CFXJSE_ResolveNodeData& rnd) { + if (!rnd.m_CurObject) + return -1; + + if (!rnd.m_CurObject->IsNode()) { + if (rnd.m_dwStyles & XFA_RESOLVENODE_Attributes) { + return ResolveForAttributeRs(rnd.m_CurObject, rnd, + rnd.m_wsName.AsStringView()); + } + return 0; + } + if (rnd.m_dwStyles & XFA_RESOLVENODE_AnyChild) { + return ResolveAnyChild(rnd); + } + if (rnd.m_wsName.GetLength()) { + wchar_t wch = rnd.m_wsName[0]; + switch (wch) { + case '$': + return ResolveDollar(rnd); + case '!': + return ResolveExcalmatory(rnd); + case '#': + return ResolveNumberSign(rnd); + case '*': + return ResolveAsterisk(rnd); + // TODO(dsinclair): We could probably remove this. + case '.': + return ResolveAnyChild(rnd); + default: + break; + } + } + if (rnd.m_uHashName == XFA_HASHCODE_This && rnd.m_nLevel == 0) { + rnd.m_Objects.push_back(rnd.m_pSC->GetThisObject()); + return 1; + } + if (rnd.m_CurObject->GetElementType() == XFA_Element::Xfa) { + CXFA_Object* pObjNode = + rnd.m_pSC->GetDocument()->GetXFAObject(rnd.m_uHashName); + if (pObjNode) { + rnd.m_Objects.push_back(pObjNode); + } else if (rnd.m_uHashName == XFA_HASHCODE_Xfa) { + rnd.m_Objects.push_back(rnd.m_CurObject); + } else if ((rnd.m_dwStyles & XFA_RESOLVENODE_Attributes) && + ResolveForAttributeRs(rnd.m_CurObject, rnd, + rnd.m_wsName.AsStringView())) { + return 1; + } + if (!rnd.m_Objects.empty()) + FilterCondition(rnd, rnd.m_wsCondition); + + return pdfium::CollectionSize(rnd.m_Objects); + } + if (ResolveNormal(rnd) < 1 && rnd.m_uHashName == XFA_HASHCODE_Xfa) + rnd.m_Objects.push_back(rnd.m_pSC->GetDocument()->GetRoot()); + + return pdfium::CollectionSize(rnd.m_Objects); +} + +int32_t CFXJSE_ResolveProcessor::ResolveAnyChild(CFXJSE_ResolveNodeData& rnd) { + WideString wsName = rnd.m_wsName; + WideString wsCondition = rnd.m_wsCondition; + CXFA_Node* findNode = nullptr; + bool bClassName = false; + if (wsName.GetLength() && wsName[0] == '#') { + bClassName = true; + wsName = wsName.Right(wsName.GetLength() - 1); + } + findNode = m_pNodeHelper->ResolveNodes_GetOneChild( + ToNode(rnd.m_CurObject), wsName.c_str(), bClassName); + if (!findNode) { + return 0; + } + if (wsCondition.IsEmpty()) { + rnd.m_Objects.push_back(findNode); + return pdfium::CollectionSize(rnd.m_Objects); + } + std::vector tempNodes; + for (auto* pObject : rnd.m_Objects) + tempNodes.push_back(pObject->AsNode()); + m_pNodeHelper->CountSiblings(findNode, XFA_LOGIC_Transparent, &tempNodes, + bClassName); + rnd.m_Objects = std::vector(tempNodes.begin(), tempNodes.end()); + FilterCondition(rnd, wsCondition); + return pdfium::CollectionSize(rnd.m_Objects); +} + +int32_t CFXJSE_ResolveProcessor::ResolveDollar(CFXJSE_ResolveNodeData& rnd) { + WideString wsName = rnd.m_wsName; + WideString wsCondition = rnd.m_wsCondition; + int32_t iNameLen = wsName.GetLength(); + if (iNameLen == 1) { + rnd.m_Objects.push_back(rnd.m_CurObject); + return 1; + } + if (rnd.m_nLevel > 0) { + return -1; + } + XFA_HashCode dwNameHash = static_cast(FX_HashCode_GetW( + WideStringView(wsName.c_str() + 1, iNameLen - 1), false)); + if (dwNameHash == XFA_HASHCODE_Xfa) { + rnd.m_Objects.push_back(rnd.m_pSC->GetDocument()->GetRoot()); + } else { + CXFA_Object* pObjNode = rnd.m_pSC->GetDocument()->GetXFAObject(dwNameHash); + if (pObjNode) + rnd.m_Objects.push_back(pObjNode); + } + if (!rnd.m_Objects.empty()) + FilterCondition(rnd, wsCondition); + + return pdfium::CollectionSize(rnd.m_Objects); +} + +int32_t CFXJSE_ResolveProcessor::ResolveExcalmatory( + CFXJSE_ResolveNodeData& rnd) { + if (rnd.m_nLevel > 0) + return 0; + + CXFA_Node* datasets = + ToNode(rnd.m_pSC->GetDocument()->GetXFAObject(XFA_HASHCODE_Datasets)); + if (!datasets) + return 0; + + CFXJSE_ResolveNodeData rndFind(rnd.m_pSC); + rndFind.m_CurObject = datasets; + rndFind.m_wsName = rnd.m_wsName.Right(rnd.m_wsName.GetLength() - 1); + rndFind.m_uHashName = static_cast( + FX_HashCode_GetW(rndFind.m_wsName.AsStringView(), false)); + rndFind.m_nLevel = rnd.m_nLevel + 1; + rndFind.m_dwStyles = XFA_RESOLVENODE_Children; + rndFind.m_wsCondition = rnd.m_wsCondition; + Resolve(rndFind); + rnd.m_Objects.insert(rnd.m_Objects.end(), rndFind.m_Objects.begin(), + rndFind.m_Objects.end()); + return pdfium::CollectionSize(rnd.m_Objects); +} + +int32_t CFXJSE_ResolveProcessor::ResolveNumberSign( + CFXJSE_ResolveNodeData& rnd) { + WideString wsName = rnd.m_wsName.Right(rnd.m_wsName.GetLength() - 1); + WideString wsCondition = rnd.m_wsCondition; + CXFA_Node* curNode = ToNode(rnd.m_CurObject); + if (ResolveForAttributeRs(curNode, rnd, wsName.AsStringView())) + return 1; + + CFXJSE_ResolveNodeData rndFind(rnd.m_pSC); + rndFind.m_nLevel = rnd.m_nLevel + 1; + rndFind.m_dwStyles = rnd.m_dwStyles; + rndFind.m_dwStyles |= XFA_RESOLVENODE_TagName; + rndFind.m_dwStyles &= ~XFA_RESOLVENODE_Attributes; + rndFind.m_wsName = wsName; + rndFind.m_uHashName = static_cast( + FX_HashCode_GetW(rndFind.m_wsName.AsStringView(), false)); + rndFind.m_wsCondition = wsCondition; + rndFind.m_CurObject = curNode; + ResolveNormal(rndFind); + if (rndFind.m_Objects.empty()) + return 0; + + if (wsCondition.GetLength() == 0 && + pdfium::ContainsValue(rndFind.m_Objects, curNode)) { + rnd.m_Objects.push_back(curNode); + } else { + rnd.m_Objects.insert(rnd.m_Objects.end(), rndFind.m_Objects.begin(), + rndFind.m_Objects.end()); + } + return pdfium::CollectionSize(rnd.m_Objects); +} + +int32_t CFXJSE_ResolveProcessor::ResolveForAttributeRs( + CXFA_Object* curNode, + CFXJSE_ResolveNodeData& rnd, + const WideStringView& strAttr) { + const XFA_SCRIPTATTRIBUTEINFO* lpScriptAttribute = + XFA_GetScriptAttributeByName(curNode->GetElementType(), strAttr); + if (!lpScriptAttribute) + return 0; + + rnd.m_pScriptAttribute = lpScriptAttribute; + rnd.m_Objects.push_back(curNode); + rnd.m_dwFlag = XFA_RESOVENODE_RSTYPE_Attribute; + return 1; +} + +int32_t CFXJSE_ResolveProcessor::ResolveNormal(CFXJSE_ResolveNodeData& rnd) { + if (rnd.m_nLevel > 32 || !rnd.m_CurObject->IsNode()) + return 0; + + CXFA_Node* curNode = rnd.m_CurObject->AsNode(); + size_t nNum = rnd.m_Objects.size(); + uint32_t dwStyles = rnd.m_dwStyles; + WideString& wsName = rnd.m_wsName; + XFA_HashCode uNameHash = rnd.m_uHashName; + WideString& wsCondition = rnd.m_wsCondition; + + CFXJSE_ResolveNodeData rndFind(rnd.m_pSC); + rndFind.m_wsName = rnd.m_wsName; + rndFind.m_wsCondition = rnd.m_wsCondition; + rndFind.m_nLevel = rnd.m_nLevel + 1; + rndFind.m_uHashName = uNameHash; + + std::vector children; + std::vector properties; + CXFA_Node* pVariablesNode = nullptr; + CXFA_Node* pPageSetNode = nullptr; + for (CXFA_Node* pChild = curNode->GetNodeItem(XFA_NODEITEM_FirstChild); + pChild; pChild = pChild->GetNodeItem(XFA_NODEITEM_NextSibling)) { + if (pChild->GetElementType() == XFA_Element::Variables) { + pVariablesNode = pChild; + continue; + } + if (pChild->GetElementType() == XFA_Element::PageSet) { + pPageSetNode = pChild; + continue; + } + const XFA_PROPERTY* pProperty = XFA_GetPropertyOfElement( + curNode->GetElementType(), pChild->GetElementType(), + XFA_XDPPACKET_UNKNOWN); + if (pProperty) + properties.push_back(pChild); + else + children.push_back(pChild); + } + if ((dwStyles & XFA_RESOLVENODE_Properties) && pVariablesNode) { + uint32_t uPropHash = pVariablesNode->GetClassHashCode(); + if (uPropHash == uNameHash) { + rnd.m_Objects.push_back(pVariablesNode); + } else { + rndFind.m_CurObject = pVariablesNode; + SetStylesForChild(dwStyles, rndFind); + WideString wsSaveCondition = rndFind.m_wsCondition; + rndFind.m_wsCondition.clear(); + ResolveNormal(rndFind); + rndFind.m_wsCondition = wsSaveCondition; + rnd.m_Objects.insert(rnd.m_Objects.end(), rndFind.m_Objects.begin(), + rndFind.m_Objects.end()); + rndFind.m_Objects.clear(); + } + if (rnd.m_Objects.size() > nNum) { + FilterCondition(rnd, wsCondition); + return !rnd.m_Objects.empty() ? 1 : 0; + } + } + if (dwStyles & XFA_RESOLVENODE_Children) { + bool bSetFlag = false; + if (pPageSetNode && (dwStyles & XFA_RESOLVENODE_Properties)) + children.push_back(pPageSetNode); + + for (CXFA_Node* child : children) { + if (dwStyles & XFA_RESOLVENODE_TagName) { + if (child->GetClassHashCode() == uNameHash) + rnd.m_Objects.push_back(child); + } else if (child->GetNameHash() == uNameHash) { + rnd.m_Objects.push_back(child); + } + if (m_pNodeHelper->NodeIsTransparent(child) && + child->GetElementType() != XFA_Element::PageSet) { + if (!bSetFlag) { + SetStylesForChild(dwStyles, rndFind); + bSetFlag = true; + } + rndFind.m_CurObject = child; + WideString wsSaveCondition = rndFind.m_wsCondition; + rndFind.m_wsCondition.clear(); + ResolveNormal(rndFind); + rndFind.m_wsCondition = wsSaveCondition; + rnd.m_Objects.insert(rnd.m_Objects.end(), rndFind.m_Objects.begin(), + rndFind.m_Objects.end()); + rndFind.m_Objects.clear(); + } + } + if (rnd.m_Objects.size() > nNum) { + if (!(dwStyles & XFA_RESOLVENODE_ALL)) { + std::vector upArrayNodes; + if (m_pNodeHelper->NodeIsTransparent(ToNode(curNode))) { + m_pNodeHelper->CountSiblings(ToNode(rnd.m_Objects[0]), + XFA_LOGIC_Transparent, &upArrayNodes, + !!(dwStyles & XFA_RESOLVENODE_TagName)); + } + if (upArrayNodes.size() > rnd.m_Objects.size()) { + CXFA_Object* pSaveObject = rnd.m_Objects.front(); + rnd.m_Objects = std::vector(upArrayNodes.begin(), + upArrayNodes.end()); + rnd.m_Objects.front() = pSaveObject; + } + } + FilterCondition(rnd, wsCondition); + return !rnd.m_Objects.empty() ? 1 : 0; + } + } + if (dwStyles & XFA_RESOLVENODE_Attributes) { + if (ResolveForAttributeRs(curNode, rnd, wsName.AsStringView())) + return 1; + } + if (dwStyles & XFA_RESOLVENODE_Properties) { + for (CXFA_Node* pChildProperty : properties) { + if (pChildProperty->IsUnnamed()) { + if (pChildProperty->GetClassHashCode() == uNameHash) + rnd.m_Objects.push_back(pChildProperty); + continue; + } + if (pChildProperty->GetNameHash() == uNameHash && + pChildProperty->GetElementType() != XFA_Element::Extras && + pChildProperty->GetElementType() != XFA_Element::Items) { + rnd.m_Objects.push_back(pChildProperty); + } + } + if (rnd.m_Objects.size() > nNum) { + FilterCondition(rnd, wsCondition); + return !rnd.m_Objects.empty() ? 1 : 0; + } + CXFA_Node* pProp = nullptr; + if (XFA_Element::Subform == curNode->GetElementType() && + XFA_HASHCODE_Occur == uNameHash) { + CXFA_Node* pInstanceManager = + curNode->AsNode()->GetInstanceMgrOfSubform(); + if (pInstanceManager) { + pProp = pInstanceManager->JSNode()->GetProperty(0, XFA_Element::Occur, + true); + } + } else { + XFA_Element eType = XFA_GetElementTypeForName(wsName.AsStringView()); + if (eType != XFA_Element::Unknown) { + pProp = curNode->AsNode()->JSNode()->GetProperty( + 0, eType, eType != XFA_Element::PageSet); + } + } + if (pProp) { + rnd.m_Objects.push_back(pProp); + return pdfium::CollectionSize(rnd.m_Objects); + } + } + CXFA_Node* parentNode = m_pNodeHelper->ResolveNodes_GetParent( + curNode->AsNode(), XFA_LOGIC_NoTransparent); + uint32_t uCurClassHash = curNode->GetClassHashCode(); + if (!parentNode) { + if (uCurClassHash == uNameHash) { + rnd.m_Objects.push_back(curNode->AsNode()); + FilterCondition(rnd, wsCondition); + if (!rnd.m_Objects.empty()) + return 1; + } + return 0; + } + if (dwStyles & XFA_RESOLVENODE_Siblings) { + CXFA_Node* child = parentNode->GetNodeItem(XFA_NODEITEM_FirstChild); + uint32_t dwSubStyles = + XFA_RESOLVENODE_Children | XFA_RESOLVENODE_Properties; + if (dwStyles & XFA_RESOLVENODE_TagName) + dwSubStyles |= XFA_RESOLVENODE_TagName; + if (dwStyles & XFA_RESOLVENODE_ALL) + dwSubStyles |= XFA_RESOLVENODE_ALL; + rndFind.m_dwStyles = dwSubStyles; + while (child) { + if (child == curNode) { + if (dwStyles & XFA_RESOLVENODE_TagName) { + if (uCurClassHash == uNameHash) { + rnd.m_Objects.push_back(curNode); + } + } else { + if (child->GetNameHash() == uNameHash) { + rnd.m_Objects.push_back(curNode); + if (rnd.m_nLevel == 0 && wsCondition.GetLength() == 0) { + rnd.m_Objects.clear(); + rnd.m_Objects.push_back(curNode); + return 1; + } + } + } + child = child->GetNodeItem(XFA_NODEITEM_NextSibling); + continue; + } + if (dwStyles & XFA_RESOLVENODE_TagName) { + if (child->GetClassHashCode() == uNameHash) + rnd.m_Objects.push_back(child); + } else if (child->GetNameHash() == uNameHash) { + rnd.m_Objects.push_back(child); + } + const XFA_PROPERTY* pPropert = XFA_GetPropertyOfElement( + parentNode->GetElementType(), child->GetElementType(), + XFA_XDPPACKET_UNKNOWN); + bool bInnerSearch = false; + if (pPropert) { + if ((child->GetElementType() == XFA_Element::Variables || + child->GetElementType() == XFA_Element::PageSet)) { + bInnerSearch = true; + } + } else { + if (m_pNodeHelper->NodeIsTransparent(child)) { + bInnerSearch = true; + } + } + if (bInnerSearch) { + rndFind.m_CurObject = child; + WideString wsOriginCondition = rndFind.m_wsCondition; + rndFind.m_wsCondition.clear(); + uint32_t dwOriginStyle = rndFind.m_dwStyles; + rndFind.m_dwStyles = dwOriginStyle | XFA_RESOLVENODE_ALL; + ResolveNormal(rndFind); + rndFind.m_dwStyles = dwOriginStyle; + rndFind.m_wsCondition = wsOriginCondition; + rnd.m_Objects.insert(rnd.m_Objects.end(), rndFind.m_Objects.begin(), + rndFind.m_Objects.end()); + rndFind.m_Objects.clear(); + } + child = child->GetNodeItem(XFA_NODEITEM_NextSibling); + } + if (rnd.m_Objects.size() > nNum) { + if (m_pNodeHelper->NodeIsTransparent(parentNode)) { + std::vector upArrayNodes; + m_pNodeHelper->CountSiblings(ToNode(rnd.m_Objects.front()), + XFA_LOGIC_Transparent, &upArrayNodes, + !!(dwStyles & XFA_RESOLVENODE_TagName)); + if (upArrayNodes.size() > rnd.m_Objects.size()) { + CXFA_Object* pSaveObject = rnd.m_Objects.front(); + rnd.m_Objects = std::vector(upArrayNodes.begin(), + upArrayNodes.end()); + rnd.m_Objects.front() = pSaveObject; + } + } + FilterCondition(rnd, wsCondition); + return !rnd.m_Objects.empty() ? 1 : 0; + } + } + if (dwStyles & XFA_RESOLVENODE_Parent) { + uint32_t dwSubStyles = XFA_RESOLVENODE_Siblings | XFA_RESOLVENODE_Parent | + XFA_RESOLVENODE_Properties; + if (dwStyles & XFA_RESOLVENODE_TagName) { + dwSubStyles |= XFA_RESOLVENODE_TagName; + } + if (dwStyles & XFA_RESOLVENODE_ALL) { + dwSubStyles |= XFA_RESOLVENODE_ALL; + } + rndFind.m_dwStyles = dwSubStyles; + rndFind.m_CurObject = parentNode; + rnd.m_pSC->GetUpObjectArray()->push_back(parentNode); + ResolveNormal(rndFind); + rnd.m_Objects.insert(rnd.m_Objects.end(), rndFind.m_Objects.begin(), + rndFind.m_Objects.end()); + rndFind.m_Objects.clear(); + if (rnd.m_Objects.size() > nNum) + return 1; + } + return 0; +} + +int32_t CFXJSE_ResolveProcessor::ResolveAsterisk(CFXJSE_ResolveNodeData& rnd) { + CXFA_Node* curNode = ToNode(rnd.m_CurObject); + std::vector array = + curNode->GetNodeList(XFA_NODEFILTER_Children | XFA_NODEFILTER_Properties, + XFA_Element::Unknown); + rnd.m_Objects.insert(rnd.m_Objects.end(), array.begin(), array.end()); + return pdfium::CollectionSize(rnd.m_Objects); +} + +int32_t CFXJSE_ResolveProcessor::ResolvePopStack(std::vector* stack) { + if (stack->empty()) + return -1; + + int32_t nType = stack->back(); + stack->pop_back(); + return nType; +} + +int32_t CFXJSE_ResolveProcessor::GetFilter(const WideStringView& wsExpression, + int32_t nStart, + CFXJSE_ResolveNodeData& rnd) { + ASSERT(nStart > -1); + int32_t iLength = wsExpression.GetLength(); + if (nStart >= iLength) { + return 0; + } + WideString& wsName = rnd.m_wsName; + WideString& wsCondition = rnd.m_wsCondition; + wchar_t* pNameBuf = wsName.GetBuffer(iLength - nStart); + wchar_t* pConditionBuf = wsCondition.GetBuffer(iLength - nStart); + int32_t nNameCount = 0; + int32_t nConditionCount = 0; + std::vector stack; + int32_t nType = -1; + const wchar_t* pSrc = wsExpression.unterminated_c_str(); + wchar_t wPrev = 0, wCur; + bool bIsCondition = false; + while (nStart < iLength) { + wCur = pSrc[nStart++]; + if (wCur == '.') { + if (wPrev == '\\') { + pNameBuf[nNameCount - 1] = wPrev = '.'; + continue; + } + if (nNameCount == 0) { + rnd.m_dwStyles |= XFA_RESOLVENODE_AnyChild; + continue; + } + wchar_t wLookahead = nStart < iLength ? pSrc[nStart] : 0; + if (wLookahead != '[' && wLookahead != '(') { + if (nType < 0) { + break; + } + } + } + if (wCur == '[' || wCur == '(') { + bIsCondition = true; + } else if (wCur == '.' && nStart < iLength && + (pSrc[nStart] == '[' || pSrc[nStart] == '(')) { + bIsCondition = true; + } + if (bIsCondition) { + pConditionBuf[nConditionCount++] = wCur; + } else { + pNameBuf[nNameCount++] = wCur; + } + bool bRecursive = true; + switch (nType) { + case 0: + if (wCur == ']') { + nType = ResolvePopStack(&stack); + bRecursive = false; + } + break; + case 1: + if (wCur == ')') { + nType = ResolvePopStack(&stack); + bRecursive = false; + } + break; + case 2: + if (wCur == '"') { + nType = ResolvePopStack(&stack); + bRecursive = false; + } + break; + } + if (bRecursive) { + switch (wCur) { + case '[': + stack.push_back(nType); + nType = 0; + break; + case '(': + stack.push_back(nType); + nType = 1; + break; + case '"': + stack.push_back(nType); + nType = 2; + break; + } + } + wPrev = wCur; + } + if (!stack.empty()) + return -1; + + wsName.ReleaseBuffer(nNameCount); + wsName.TrimLeft(); + wsName.TrimRight(); + wsCondition.ReleaseBuffer(nConditionCount); + wsCondition.TrimLeft(); + wsCondition.TrimRight(); + rnd.m_uHashName = + static_cast(FX_HashCode_GetW(wsName.AsStringView(), false)); + return nStart; +} +void CFXJSE_ResolveProcessor::ConditionArray(int32_t iCurIndex, + WideString wsCondition, + int32_t iFoundCount, + CFXJSE_ResolveNodeData& rnd) { + int32_t iLen = wsCondition.GetLength(); + bool bRelative = false; + bool bAll = false; + int32_t i = 1; + for (; i < iLen; ++i) { + wchar_t ch = wsCondition[i]; + if (ch == ' ') { + continue; + } + if (ch == '+' || ch == '-') { + bRelative = true; + break; + } else if (ch == '*') { + bAll = true; + break; + } else { + break; + } + } + if (bAll) { + if (rnd.m_dwStyles & XFA_RESOLVENODE_CreateNode) { + if (rnd.m_dwStyles & XFA_RESOLVENODE_Bind) { + m_pNodeHelper->m_pCreateParent = ToNode(rnd.m_CurObject); + m_pNodeHelper->m_iCreateCount = 1; + rnd.m_Objects.clear(); + m_pNodeHelper->m_iCurAllStart = -1; + m_pNodeHelper->m_pAllStartParent = nullptr; + } else { + if (m_pNodeHelper->m_iCurAllStart == -1) { + m_pNodeHelper->m_iCurAllStart = m_iCurStart; + m_pNodeHelper->m_pAllStartParent = ToNode(rnd.m_CurObject); + } + } + } else if (rnd.m_dwStyles & XFA_RESOLVENODE_BindNew) { + if (m_pNodeHelper->m_iCurAllStart == -1) { + m_pNodeHelper->m_iCurAllStart = m_iCurStart; + } + } + return; + } + if (iFoundCount == 1 && !iLen) { + return; + } + WideString wsIndex; + wsIndex = wsCondition.Mid(i, iLen - 1 - i); + int32_t iIndex = wsIndex.GetInteger(); + if (bRelative) { + iIndex += iCurIndex; + } + if (iFoundCount <= iIndex || iIndex < 0) { + if (rnd.m_dwStyles & XFA_RESOLVENODE_CreateNode) { + m_pNodeHelper->m_pCreateParent = ToNode(rnd.m_CurObject); + m_pNodeHelper->m_iCreateCount = iIndex - iFoundCount + 1; + } + rnd.m_Objects.clear(); + } else { + CXFA_Object* ret = rnd.m_Objects[iIndex]; + rnd.m_Objects.clear(); + rnd.m_Objects.push_back(ret); + } +} + +void CFXJSE_ResolveProcessor::DoPredicateFilter(int32_t iCurIndex, + WideString wsCondition, + int32_t iFoundCount, + CFXJSE_ResolveNodeData& rnd) { + ASSERT(iFoundCount == pdfium::CollectionSize(rnd.m_Objects)); + WideString wsExpression; + XFA_SCRIPTLANGTYPE eLangType = XFA_SCRIPTLANGTYPE_Unkown; + if (wsCondition.Left(2) == L".[" && wsCondition.Last() == L']') { + eLangType = XFA_SCRIPTLANGTYPE_Formcalc; + } else if (wsCondition.Left(2) == L".(" && wsCondition.Last() == L')') { + eLangType = XFA_SCRIPTLANGTYPE_Javascript; + } else { + return; + } + + CFXJSE_Engine* pContext = rnd.m_pSC; + wsExpression = wsCondition.Mid(2, wsCondition.GetLength() - 3); + for (int32_t i = iFoundCount - 1; i >= 0; i--) { + auto pRetValue = pdfium::MakeUnique(rnd.m_pSC->GetRuntime()); + bool bRet = pContext->RunScript(eLangType, wsExpression.AsStringView(), + pRetValue.get(), rnd.m_Objects[i]); + if (!bRet || !pRetValue->ToBoolean()) + rnd.m_Objects.erase(rnd.m_Objects.begin() + i); + } +} + +void CFXJSE_ResolveProcessor::FilterCondition(CFXJSE_ResolveNodeData& rnd, + WideString wsCondition) { + int32_t iCurrIndex = 0; + const std::vector* pArray = rnd.m_pSC->GetUpObjectArray(); + if (!pArray->empty()) { + CXFA_Node* curNode = pArray->back(); + bool bIsProperty = m_pNodeHelper->NodeIsProperty(curNode); + if (curNode->IsUnnamed() || + (bIsProperty && curNode->GetElementType() != XFA_Element::PageSet)) { + iCurrIndex = m_pNodeHelper->GetIndex(curNode, XFA_LOGIC_Transparent, + bIsProperty, true); + } else { + iCurrIndex = m_pNodeHelper->GetIndex(curNode, XFA_LOGIC_Transparent, + bIsProperty, false); + } + } + int32_t iFoundCount = pdfium::CollectionSize(rnd.m_Objects); + wsCondition.TrimLeft(); + wsCondition.TrimRight(); + int32_t iLen = wsCondition.GetLength(); + if (!iLen) { + if (rnd.m_dwStyles & XFA_RESOLVENODE_ALL) { + return; + } + if (iFoundCount == 1) { + return; + } + if (iFoundCount <= iCurrIndex) { + if (rnd.m_dwStyles & XFA_RESOLVENODE_CreateNode) { + m_pNodeHelper->m_pCreateParent = ToNode(rnd.m_CurObject); + m_pNodeHelper->m_iCreateCount = iCurrIndex - iFoundCount + 1; + } + rnd.m_Objects.clear(); + return; + } else { + CXFA_Object* ret = rnd.m_Objects[iCurrIndex]; + rnd.m_Objects.clear(); + rnd.m_Objects.push_back(ret); + return; + } + } + wchar_t wTypeChar = wsCondition[0]; + switch (wTypeChar) { + case '[': + ConditionArray(iCurrIndex, wsCondition, iFoundCount, rnd); + return; + case '(': + return; + case '"': + return; + case '.': + if (iLen > 1 && (wsCondition[1] == '[' || wsCondition[1] == '(')) { + DoPredicateFilter(iCurrIndex, wsCondition, iFoundCount, rnd); + } + default: + return; + } +} +void CFXJSE_ResolveProcessor::SetStylesForChild(uint32_t dwParentStyles, + CFXJSE_ResolveNodeData& rnd) { + uint32_t dwSubStyles = XFA_RESOLVENODE_Children; + if (dwParentStyles & XFA_RESOLVENODE_TagName) { + dwSubStyles |= XFA_RESOLVENODE_TagName; + } + dwSubStyles &= ~XFA_RESOLVENODE_Parent; + dwSubStyles &= ~XFA_RESOLVENODE_Siblings; + dwSubStyles &= ~XFA_RESOLVENODE_Properties; + dwSubStyles |= XFA_RESOLVENODE_ALL; + rnd.m_dwStyles = dwSubStyles; +} + +int32_t CFXJSE_ResolveProcessor::SetResultCreateNode( + XFA_RESOLVENODE_RS& resolveNodeRS, + WideString& wsLastCondition) { + if (m_pNodeHelper->m_pCreateParent) + resolveNodeRS.objects.push_back(m_pNodeHelper->m_pCreateParent); + else + m_pNodeHelper->CreateNode_ForCondition(wsLastCondition); + + resolveNodeRS.dwFlags = m_pNodeHelper->m_iCreateFlag; + if (resolveNodeRS.dwFlags == XFA_RESOLVENODE_RSTYPE_CreateNodeOne) { + if (m_pNodeHelper->m_iCurAllStart != -1) + resolveNodeRS.dwFlags = XFA_RESOLVENODE_RSTYPE_CreateNodeMidAll; + } + return pdfium::CollectionSize(resolveNodeRS.objects); +} + +void CFXJSE_ResolveProcessor::SetIndexDataBind(WideString& wsNextCondition, + int32_t& iIndex, + int32_t iCount) { + if (m_pNodeHelper->CreateNode_ForCondition(wsNextCondition)) { + if (m_pNodeHelper->m_eLastCreateType == XFA_Element::DataGroup) { + iIndex = 0; + } else { + iIndex = iCount - 1; + } + } else { + iIndex = iCount - 1; + } +} + +CFXJSE_ResolveNodeData::CFXJSE_ResolveNodeData(CFXJSE_Engine* pSC) + : m_pSC(pSC), + m_CurObject(nullptr), + m_wsName(), + m_uHashName(XFA_HASHCODE_None), + m_wsCondition(), + m_nLevel(0), + m_Objects(), + m_dwStyles(XFA_RESOLVENODE_Children), + m_pScriptAttribute(nullptr), + m_dwFlag(XFA_RESOVENODE_RSTYPE_Nodes) {} + +CFXJSE_ResolveNodeData::~CFXJSE_ResolveNodeData() {} diff --git a/fxjs/cfxjse_resolveprocessor.h b/fxjs/cfxjse_resolveprocessor.h new file mode 100644 index 0000000000..f376cca4c7 --- /dev/null +++ b/fxjs/cfxjse_resolveprocessor.h @@ -0,0 +1,80 @@ +// Copyright 2014 PDFium Authors. All rights reserved. +// 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 + +#ifndef FXJS_CFXJSE_RESOLVEPROCESSOR_H_ +#define FXJS_CFXJSE_RESOLVEPROCESSOR_H_ + +#include +#include + +#include "xfa/fxfa/parser/xfa_resolvenode_rs.h" + +class CXFA_NodeHelper; +class CFXJSE_Engine; + +class CFXJSE_ResolveNodeData { + public: + explicit CFXJSE_ResolveNodeData(CFXJSE_Engine* pSC); + ~CFXJSE_ResolveNodeData(); + + CFXJSE_Engine* m_pSC; + CXFA_Object* m_CurObject; + WideString m_wsName; + XFA_HashCode m_uHashName; + WideString m_wsCondition; + int32_t m_nLevel; + std::vector m_Objects; // Not owned. + uint32_t m_dwStyles; + const XFA_SCRIPTATTRIBUTEINFO* m_pScriptAttribute; + XFA_RESOVENODE_RSTYPE m_dwFlag; +}; + +class CFXJSE_ResolveProcessor { + public: + CFXJSE_ResolveProcessor(); + ~CFXJSE_ResolveProcessor(); + + int32_t Resolve(CFXJSE_ResolveNodeData& rnd); + int32_t GetFilter(const WideStringView& wsExpression, + int32_t nStart, + CFXJSE_ResolveNodeData& rnd); + int32_t SetResultCreateNode(XFA_RESOLVENODE_RS& resolveNodeRS, + WideString& wsLastCondition); + void SetIndexDataBind(WideString& wsNextCondition, + int32_t& iIndex, + int32_t iCount); + void SetCurStart(int32_t start) { m_iCurStart = start; } + + CXFA_NodeHelper* GetNodeHelper() const { return m_pNodeHelper.get(); } + + private: + int32_t ResolveForAttributeRs(CXFA_Object* curNode, + CFXJSE_ResolveNodeData& rnd, + const WideStringView& strAttr); + int32_t ResolveAnyChild(CFXJSE_ResolveNodeData& rnd); + int32_t ResolveDollar(CFXJSE_ResolveNodeData& rnd); + int32_t ResolveExcalmatory(CFXJSE_ResolveNodeData& rnd); + int32_t ResolveNumberSign(CFXJSE_ResolveNodeData& rnd); + int32_t ResolveAsterisk(CFXJSE_ResolveNodeData& rnd); + int32_t ResolveNormal(CFXJSE_ResolveNodeData& rnd); + int32_t ResolvePopStack(std::vector* stack); + void SetStylesForChild(uint32_t dwParentStyles, CFXJSE_ResolveNodeData& rnd); + + void ConditionArray(int32_t iCurIndex, + WideString wsCondition, + int32_t iFoundCount, + CFXJSE_ResolveNodeData& rnd); + void DoPredicateFilter(int32_t iCurIndex, + WideString wsCondition, + int32_t iFoundCount, + CFXJSE_ResolveNodeData& rnd); + void FilterCondition(CFXJSE_ResolveNodeData& rnd, WideString wsCondition); + + int32_t m_iCurStart; + std::unique_ptr m_pNodeHelper; +}; + +#endif // FXJS_CFXJSE_RESOLVEPROCESSOR_H_ -- cgit v1.2.3