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; |
+} |