| // Copyright 2017 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 "core/fxcrt/xml/cfx_saxreaderhandler.h" |
| |
| #include <string> |
| |
| #include "core/fxcrt/cfx_checksumcontext.h" |
| |
| CFX_SAXReaderHandler::CFX_SAXReaderHandler(CFX_ChecksumContext* pContext) |
| : m_pContext(pContext) { |
| ASSERT(m_pContext); |
| } |
| |
| CFX_SAXReaderHandler::~CFX_SAXReaderHandler() {} |
| |
| CFX_SAXContext* CFX_SAXReaderHandler::OnTagEnter( |
| const ByteStringView& bsTagName, |
| CFX_SAXItem::Type eType, |
| uint32_t dwStartPos) { |
| UpdateChecksum(true); |
| if (eType != CFX_SAXItem::Type::Tag && |
| eType != CFX_SAXItem::Type::Instruction) { |
| return nullptr; |
| } |
| |
| m_SAXContext.m_eNode = eType; |
| m_SAXContext.m_TextBuf << "<"; |
| if (eType == CFX_SAXItem::Type::Instruction) |
| m_SAXContext.m_TextBuf << "?"; |
| |
| m_SAXContext.m_TextBuf << bsTagName; |
| m_SAXContext.m_bsTagName = bsTagName; |
| return &m_SAXContext; |
| } |
| |
| void CFX_SAXReaderHandler::OnTagAttribute(CFX_SAXContext* pTag, |
| const ByteStringView& bsAttri, |
| const ByteStringView& bsValue) { |
| if (!pTag) |
| return; |
| pTag->m_TextBuf << " " << bsAttri << "=\"" << bsValue << "\""; |
| } |
| |
| void CFX_SAXReaderHandler::OnTagBreak(CFX_SAXContext* pTag) { |
| if (!pTag) |
| return; |
| |
| pTag->m_TextBuf << ">"; |
| UpdateChecksum(false); |
| } |
| |
| void CFX_SAXReaderHandler::OnTagData(CFX_SAXContext* pTag, |
| CFX_SAXItem::Type eType, |
| const ByteStringView& bsData, |
| uint32_t dwStartPos) { |
| if (!pTag) |
| return; |
| |
| if (eType == CFX_SAXItem::Type::CharData) |
| pTag->m_TextBuf << "<![CDATA["; |
| |
| pTag->m_TextBuf << bsData; |
| if (eType == CFX_SAXItem::Type::CharData) |
| pTag->m_TextBuf << "]]>"; |
| } |
| |
| void CFX_SAXReaderHandler::OnTagClose(CFX_SAXContext* pTag, uint32_t dwEndPos) { |
| if (!pTag) |
| return; |
| |
| if (pTag->m_eNode == CFX_SAXItem::Type::Instruction) |
| pTag->m_TextBuf << "?>"; |
| else if (pTag->m_eNode == CFX_SAXItem::Type::Tag) |
| pTag->m_TextBuf << "></" << pTag->m_bsTagName.AsStringView() << ">"; |
| |
| UpdateChecksum(false); |
| } |
| |
| void CFX_SAXReaderHandler::OnTagEnd(CFX_SAXContext* pTag, |
| const ByteStringView& bsTagName, |
| uint32_t dwEndPos) { |
| if (!pTag) |
| return; |
| |
| pTag->m_TextBuf << "</" << bsTagName << ">"; |
| UpdateChecksum(false); |
| } |
| |
| void CFX_SAXReaderHandler::OnTargetData(CFX_SAXContext* pTag, |
| CFX_SAXItem::Type eType, |
| const ByteStringView& bsData, |
| uint32_t dwStartPos) { |
| if (!pTag && eType != CFX_SAXItem::Type::Comment) |
| return; |
| |
| if (eType == CFX_SAXItem::Type::Comment) { |
| m_SAXContext.m_TextBuf << "<!--" << bsData << "-->"; |
| UpdateChecksum(false); |
| } else { |
| pTag->m_TextBuf << " " << bsData; |
| } |
| } |
| |
| void CFX_SAXReaderHandler::UpdateChecksum(bool bCheckSpace) { |
| int32_t iLength = m_SAXContext.m_TextBuf.tellp(); |
| if (iLength < 1) |
| return; |
| |
| std::string sBuffer = m_SAXContext.m_TextBuf.str(); |
| const uint8_t* pBuffer = reinterpret_cast<const uint8_t*>(sBuffer.c_str()); |
| bool bUpdata = true; |
| if (bCheckSpace) { |
| bUpdata = false; |
| for (int32_t i = 0; i < iLength; i++) { |
| bUpdata = (pBuffer[i] > 0x20); |
| if (bUpdata) |
| break; |
| } |
| } |
| if (bUpdata) |
| m_pContext->Update(ByteStringView(pBuffer, iLength)); |
| |
| m_SAXContext.m_TextBuf.str(""); |
| } |