You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
80 lines
2.4 KiB
80 lines
2.4 KiB
// 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 XFA_FXFA_XFA_CHECKSUM_H_ |
|
#define XFA_FXFA_XFA_CHECKSUM_H_ |
|
|
|
#include <memory> |
|
|
|
#include "core/fdrm/crypto/fx_crypt.h" |
|
#include "xfa/fde/xml/cfx_saxreader.h" |
|
#include "xfa/fxfa/fxfa.h" |
|
|
|
class CXFA_SAXReaderHandler; |
|
class CXFA_ChecksumContext; |
|
|
|
class CXFA_SAXContext { |
|
public: |
|
CXFA_SAXContext() : m_eNode(CFX_SAXItem::Type::Unknown) {} |
|
|
|
CFX_ByteTextBuf m_TextBuf; |
|
CFX_ByteString m_bsTagName; |
|
CFX_SAXItem::Type m_eNode; |
|
}; |
|
|
|
class CXFA_SAXReaderHandler { |
|
public: |
|
explicit CXFA_SAXReaderHandler(CXFA_ChecksumContext* pContext); |
|
~CXFA_SAXReaderHandler(); |
|
|
|
CXFA_SAXContext* OnTagEnter(const CFX_ByteStringC& bsTagName, |
|
CFX_SAXItem::Type eType, |
|
uint32_t dwStartPos); |
|
void OnTagAttribute(CXFA_SAXContext* pTag, |
|
const CFX_ByteStringC& bsAttri, |
|
const CFX_ByteStringC& bsValue); |
|
void OnTagBreak(CXFA_SAXContext* pTag); |
|
void OnTagData(CXFA_SAXContext* pTag, |
|
CFX_SAXItem::Type eType, |
|
const CFX_ByteStringC& bsData, |
|
uint32_t dwStartPos); |
|
void OnTagClose(CXFA_SAXContext* pTag, uint32_t dwEndPos); |
|
void OnTagEnd(CXFA_SAXContext* pTag, |
|
const CFX_ByteStringC& bsTagName, |
|
uint32_t dwEndPos); |
|
|
|
void OnTargetData(CXFA_SAXContext* pTag, |
|
CFX_SAXItem::Type eType, |
|
const CFX_ByteStringC& bsData, |
|
uint32_t dwStartPos); |
|
|
|
protected: |
|
void UpdateChecksum(bool bCheckSpace); |
|
|
|
CXFA_ChecksumContext* m_pContext; |
|
CXFA_SAXContext m_SAXContext; |
|
}; |
|
|
|
class CXFA_ChecksumContext { |
|
public: |
|
CXFA_ChecksumContext(); |
|
~CXFA_ChecksumContext(); |
|
|
|
void StartChecksum(); |
|
void Update(const CFX_ByteStringC& bsText); |
|
bool UpdateChecksum(const CFX_RetainPtr<IFX_SeekableReadStream>& pSrcFile, |
|
FX_FILESIZE offset = 0, |
|
size_t size = 0); |
|
void FinishChecksum(); |
|
CFX_ByteString GetChecksum() const; |
|
|
|
protected: |
|
std::unique_ptr<CFX_SAXReader> m_pSAXReader; |
|
std::unique_ptr<CRYPT_sha1_context> m_pByteContext; |
|
CFX_ByteString m_bsChecksum; |
|
}; |
|
|
|
#endif // XFA_FXFA_XFA_CHECKSUM_H_
|
|
|