// 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(""); }