Index: core/fxcrt/fx_extension.cpp |
diff --git a/core/fxcrt/fx_extension.cpp b/core/fxcrt/fx_extension.cpp |
index 1a95e8c3181fc9afe0d66254ba190931dabb5563..4c77244b3d3a7d656ba1449c32821615c9b8ce9b 100644 |
--- a/core/fxcrt/fx_extension.cpp |
+++ b/core/fxcrt/fx_extension.cpp |
@@ -16,11 +16,271 @@ |
#include <ctime> |
#endif |
+#ifdef PDF_ENABLE_XFA |
+ |
+CFX_CRTFileAccess::CFX_CRTFileAccess() : m_RefCount(0) {} |
+ |
+CFX_CRTFileAccess::~CFX_CRTFileAccess() {} |
+ |
+void CFX_CRTFileAccess::Release() { |
+ if (--m_RefCount == 0) |
+ delete this; |
+} |
+ |
+IFX_FileAccess* CFX_CRTFileAccess::Retain() { |
+ m_RefCount++; |
+ return (IFX_FileAccess*)this; |
+} |
+ |
+void CFX_CRTFileAccess::GetPath(CFX_WideString& wsPath) { |
+ wsPath = m_path; |
+} |
+ |
+IFX_FileStream* CFX_CRTFileAccess::CreateFileStream(uint32_t dwModes) { |
+ return FX_CreateFileStream(m_path.c_str(), dwModes); |
+} |
+ |
+FX_BOOL CFX_CRTFileAccess::Init(const CFX_WideStringC& wsPath) { |
+ m_path = wsPath; |
+ m_RefCount = 1; |
+ return TRUE; |
+} |
+ |
+#endif // PDF_ENABLE_XFA |
+ |
CFX_CRTFileStream::CFX_CRTFileStream(std::unique_ptr<IFXCRT_FileAccess> pFA) |
: m_pFile(std::move(pFA)), m_dwCount(1) {} |
CFX_CRTFileStream::~CFX_CRTFileStream() {} |
+CFX_MemoryStream::CFX_MemoryStream(FX_BOOL bConsecutive) |
+ : m_dwCount(1), |
+ m_nTotalSize(0), |
+ m_nCurSize(0), |
+ m_nCurPos(0), |
+ m_nGrowSize(FX_MEMSTREAM_BlockSize) { |
+ m_dwFlags = |
+ FX_MEMSTREAM_TakeOver | (bConsecutive ? FX_MEMSTREAM_Consecutive : 0); |
+} |
+ |
+CFX_MemoryStream::CFX_MemoryStream(uint8_t* pBuffer, |
+ size_t nSize, |
+ FX_BOOL bTakeOver) |
+ : m_dwCount(1), |
+ m_nTotalSize(nSize), |
+ m_nCurSize(nSize), |
+ m_nCurPos(0), |
+ m_nGrowSize(FX_MEMSTREAM_BlockSize) { |
+ m_Blocks.Add(pBuffer); |
+ m_dwFlags = |
+ FX_MEMSTREAM_Consecutive | (bTakeOver ? FX_MEMSTREAM_TakeOver : 0); |
+} |
+ |
+CFX_MemoryStream::~CFX_MemoryStream() { |
+ if (m_dwFlags & FX_MEMSTREAM_TakeOver) { |
+ for (int32_t i = 0; i < m_Blocks.GetSize(); i++) { |
+ FX_Free(m_Blocks[i]); |
+ } |
+ } |
+ m_Blocks.RemoveAll(); |
+} |
+ |
+IFX_FileStream* CFX_MemoryStream::Retain() { |
+ m_dwCount++; |
+ return this; |
+} |
+ |
+void CFX_MemoryStream::Release() { |
+ uint32_t nCount = --m_dwCount; |
+ if (nCount) { |
+ return; |
+ } |
+ delete this; |
+} |
+ |
+FX_FILESIZE CFX_MemoryStream::GetSize() { |
+ return (FX_FILESIZE)m_nCurSize; |
+} |
+ |
+FX_BOOL CFX_MemoryStream::IsEOF() { |
+ return m_nCurPos >= (size_t)GetSize(); |
+} |
+ |
+FX_FILESIZE CFX_MemoryStream::GetPosition() { |
+ return (FX_FILESIZE)m_nCurPos; |
+} |
+ |
+FX_BOOL CFX_MemoryStream::ReadBlock(void* buffer, |
+ FX_FILESIZE offset, |
+ size_t size) { |
+ if (!buffer || !size) { |
+ return FALSE; |
+ } |
+ |
+ FX_SAFE_SIZE_T newPos = size; |
+ newPos += offset; |
+ if (!newPos.IsValid() || newPos.ValueOrDefault(0) == 0 || |
+ newPos.ValueOrDie() > m_nCurSize) { |
+ return FALSE; |
+ } |
+ |
+ m_nCurPos = newPos.ValueOrDie(); |
+ if (m_dwFlags & FX_MEMSTREAM_Consecutive) { |
+ FXSYS_memcpy(buffer, m_Blocks[0] + (size_t)offset, size); |
+ return TRUE; |
+ } |
+ size_t nStartBlock = (size_t)offset / m_nGrowSize; |
+ offset -= (FX_FILESIZE)(nStartBlock * m_nGrowSize); |
+ while (size) { |
+ size_t nRead = m_nGrowSize - (size_t)offset; |
+ if (nRead > size) { |
+ nRead = size; |
+ } |
+ FXSYS_memcpy(buffer, m_Blocks[(int)nStartBlock] + (size_t)offset, nRead); |
+ buffer = ((uint8_t*)buffer) + nRead; |
+ size -= nRead; |
+ nStartBlock++; |
+ offset = 0; |
+ } |
+ return TRUE; |
+} |
+ |
+size_t CFX_MemoryStream::ReadBlock(void* buffer, size_t size) { |
+ if (m_nCurPos >= m_nCurSize) { |
+ return 0; |
+ } |
+ size_t nRead = std::min(size, m_nCurSize - m_nCurPos); |
+ if (!ReadBlock(buffer, (int32_t)m_nCurPos, nRead)) { |
+ return 0; |
+ } |
+ return nRead; |
+} |
+ |
+FX_BOOL CFX_MemoryStream::WriteBlock(const void* buffer, |
+ FX_FILESIZE offset, |
+ size_t size) { |
+ if (!buffer || !size) { |
+ return FALSE; |
+ } |
+ if (m_dwFlags & FX_MEMSTREAM_Consecutive) { |
+ FX_SAFE_SIZE_T newPos = size; |
+ newPos += offset; |
+ if (!newPos.IsValid()) |
+ return FALSE; |
+ |
+ m_nCurPos = newPos.ValueOrDie(); |
+ if (m_nCurPos > m_nTotalSize) { |
+ m_nTotalSize = (m_nCurPos + m_nGrowSize - 1) / m_nGrowSize * m_nGrowSize; |
+ if (m_Blocks.GetSize() < 1) { |
+ uint8_t* block = FX_Alloc(uint8_t, m_nTotalSize); |
+ m_Blocks.Add(block); |
+ } else { |
+ m_Blocks[0] = FX_Realloc(uint8_t, m_Blocks[0], m_nTotalSize); |
+ } |
+ if (!m_Blocks[0]) { |
+ m_Blocks.RemoveAll(); |
+ return FALSE; |
+ } |
+ } |
+ FXSYS_memcpy(m_Blocks[0] + (size_t)offset, buffer, size); |
+ if (m_nCurSize < m_nCurPos) { |
+ m_nCurSize = m_nCurPos; |
+ } |
+ return TRUE; |
+ } |
+ |
+ FX_SAFE_SIZE_T newPos = size; |
+ newPos += offset; |
+ if (!newPos.IsValid()) { |
+ return FALSE; |
+ } |
+ |
+ if (!ExpandBlocks(newPos.ValueOrDie())) { |
+ return FALSE; |
+ } |
+ m_nCurPos = newPos.ValueOrDie(); |
+ size_t nStartBlock = (size_t)offset / m_nGrowSize; |
+ offset -= (FX_FILESIZE)(nStartBlock * m_nGrowSize); |
+ while (size) { |
+ size_t nWrite = m_nGrowSize - (size_t)offset; |
+ if (nWrite > size) { |
+ nWrite = size; |
+ } |
+ FXSYS_memcpy(m_Blocks[(int)nStartBlock] + (size_t)offset, buffer, nWrite); |
+ buffer = ((uint8_t*)buffer) + nWrite; |
+ size -= nWrite; |
+ nStartBlock++; |
+ offset = 0; |
+ } |
+ return TRUE; |
+} |
+ |
+FX_BOOL CFX_MemoryStream::Flush() { |
+ return TRUE; |
+} |
+ |
+FX_BOOL CFX_MemoryStream::IsConsecutive() const { |
+ return !!(m_dwFlags & FX_MEMSTREAM_Consecutive); |
+} |
+ |
+void CFX_MemoryStream::EstimateSize(size_t nInitSize, size_t nGrowSize) { |
+ if (m_dwFlags & FX_MEMSTREAM_Consecutive) { |
+ if (m_Blocks.GetSize() < 1) { |
+ uint8_t* pBlock = |
+ FX_Alloc(uint8_t, std::max(nInitSize, static_cast<size_t>(4096))); |
+ m_Blocks.Add(pBlock); |
+ } |
+ m_nGrowSize = std::max(nGrowSize, static_cast<size_t>(4096)); |
+ } else if (m_Blocks.GetSize() < 1) { |
+ m_nGrowSize = std::max(nGrowSize, static_cast<size_t>(4096)); |
+ } |
+} |
+ |
+uint8_t* CFX_MemoryStream::GetBuffer() const { |
+ return m_Blocks.GetSize() ? m_Blocks[0] : nullptr; |
+} |
+ |
+void CFX_MemoryStream::AttachBuffer(uint8_t* pBuffer, |
+ size_t nSize, |
+ FX_BOOL bTakeOver) { |
+ if (!(m_dwFlags & FX_MEMSTREAM_Consecutive)) { |
+ return; |
+ } |
+ m_Blocks.RemoveAll(); |
+ m_Blocks.Add(pBuffer); |
+ m_nTotalSize = m_nCurSize = nSize; |
+ m_nCurPos = 0; |
+ m_dwFlags = |
+ FX_MEMSTREAM_Consecutive | (bTakeOver ? FX_MEMSTREAM_TakeOver : 0); |
+} |
+ |
+void CFX_MemoryStream::DetachBuffer() { |
+ if (!(m_dwFlags & FX_MEMSTREAM_Consecutive)) { |
+ return; |
+ } |
+ m_Blocks.RemoveAll(); |
+ m_nTotalSize = m_nCurSize = m_nCurPos = 0; |
+ m_dwFlags = FX_MEMSTREAM_TakeOver; |
+} |
+ |
+FX_BOOL CFX_MemoryStream::ExpandBlocks(size_t size) { |
+ if (m_nCurSize < size) { |
+ m_nCurSize = size; |
+ } |
+ if (size <= m_nTotalSize) { |
+ return TRUE; |
+ } |
+ int32_t iCount = m_Blocks.GetSize(); |
+ size = (size - m_nTotalSize + m_nGrowSize - 1) / m_nGrowSize; |
+ m_Blocks.SetSize(m_Blocks.GetSize() + (int32_t)size); |
+ while (size--) { |
+ uint8_t* pBlock = FX_Alloc(uint8_t, m_nGrowSize); |
+ m_Blocks.SetAt(iCount++, pBlock); |
+ m_nTotalSize += m_nGrowSize; |
+ } |
+ return TRUE; |
+} |
+ |
IFX_FileStream* CFX_CRTFileStream::Retain() { |
m_dwCount++; |
return this; |