| Index: core/src/fpdfapi/fpdf_parser/cpdf_stream_acc.cpp
|
| diff --git a/core/src/fpdfapi/fpdf_parser/cpdf_stream_acc.cpp b/core/src/fpdfapi/fpdf_parser/cpdf_stream_acc.cpp
|
| new file mode 100644
|
| index 0000000000000000000000000000000000000000..51ab1c0134c8d2e8ad8d7832faf5e6949fc9ab2b
|
| --- /dev/null
|
| +++ b/core/src/fpdfapi/fpdf_parser/cpdf_stream_acc.cpp
|
| @@ -0,0 +1,107 @@
|
| +// Copyright 2016 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/include/fpdfapi/cpdf_stream_acc.h"
|
| +
|
| +#include "core/include/fpdfapi/cpdf_dictionary.h"
|
| +#include "core/include/fpdfapi/fpdf_parser_decode.h"
|
| +
|
| +CPDF_StreamAcc::CPDF_StreamAcc()
|
| + : m_pData(nullptr),
|
| + m_dwSize(0),
|
| + m_bNewBuf(FALSE),
|
| + m_pImageParam(nullptr),
|
| + m_pStream(nullptr),
|
| + m_pSrcData(nullptr) {}
|
| +
|
| +void CPDF_StreamAcc::LoadAllData(const CPDF_Stream* pStream,
|
| + FX_BOOL bRawAccess,
|
| + FX_DWORD estimated_size,
|
| + FX_BOOL bImageAcc) {
|
| + if (!pStream)
|
| + return;
|
| +
|
| + m_pStream = pStream;
|
| + if (pStream->IsMemoryBased() &&
|
| + (!pStream->GetDict()->KeyExist("Filter") || bRawAccess)) {
|
| + m_dwSize = pStream->GetRawSize();
|
| + m_pData = pStream->GetRawData();
|
| + return;
|
| + }
|
| + uint8_t* pSrcData;
|
| + FX_DWORD dwSrcSize = pStream->GetRawSize();
|
| + if (dwSrcSize == 0)
|
| + return;
|
| +
|
| + if (!pStream->IsMemoryBased()) {
|
| + pSrcData = m_pSrcData = FX_Alloc(uint8_t, dwSrcSize);
|
| + if (!pStream->ReadRawData(0, pSrcData, dwSrcSize))
|
| + return;
|
| + } else {
|
| + pSrcData = pStream->GetRawData();
|
| + }
|
| + uint8_t* pDecryptedData = pSrcData;
|
| + FX_DWORD dwDecryptedSize = dwSrcSize;
|
| + if (!pStream->GetDict()->KeyExist("Filter") || bRawAccess) {
|
| + m_pData = pDecryptedData;
|
| + m_dwSize = dwDecryptedSize;
|
| + } else {
|
| + FX_BOOL bRet = PDF_DataDecode(
|
| + pDecryptedData, dwDecryptedSize, m_pStream->GetDict(), m_pData,
|
| + m_dwSize, m_ImageDecoder, m_pImageParam, estimated_size, bImageAcc);
|
| + if (!bRet) {
|
| + m_pData = pDecryptedData;
|
| + m_dwSize = dwDecryptedSize;
|
| + }
|
| + }
|
| + if (pSrcData != pStream->GetRawData() && pSrcData != m_pData) {
|
| + FX_Free(pSrcData);
|
| + }
|
| + if (pDecryptedData != pSrcData && pDecryptedData != m_pData) {
|
| + FX_Free(pDecryptedData);
|
| + }
|
| + m_pSrcData = nullptr;
|
| + m_bNewBuf = m_pData != pStream->GetRawData();
|
| +}
|
| +
|
| +CPDF_StreamAcc::~CPDF_StreamAcc() {
|
| + if (m_bNewBuf) {
|
| + FX_Free(m_pData);
|
| + }
|
| + FX_Free(m_pSrcData);
|
| +}
|
| +
|
| +const uint8_t* CPDF_StreamAcc::GetData() const {
|
| + if (m_bNewBuf) {
|
| + return m_pData;
|
| + }
|
| + if (!m_pStream) {
|
| + return nullptr;
|
| + }
|
| + return m_pStream->GetRawData();
|
| +}
|
| +
|
| +FX_DWORD CPDF_StreamAcc::GetSize() const {
|
| + if (m_bNewBuf) {
|
| + return m_dwSize;
|
| + }
|
| + if (!m_pStream) {
|
| + return 0;
|
| + }
|
| + return m_pStream->GetRawSize();
|
| +}
|
| +
|
| +uint8_t* CPDF_StreamAcc::DetachData() {
|
| + if (m_bNewBuf) {
|
| + uint8_t* p = m_pData;
|
| + m_pData = nullptr;
|
| + m_dwSize = 0;
|
| + return p;
|
| + }
|
| + uint8_t* p = FX_Alloc(uint8_t, m_dwSize);
|
| + FXSYS_memcpy(p, m_pData, m_dwSize);
|
| + return p;
|
| +}
|
|
|