diff options
Diffstat (limited to 'core/fxcrt')
-rw-r--r-- | core/fxcrt/cfx_readonlymemorystream.cpp | 34 | ||||
-rw-r--r-- | core/fxcrt/cfx_readonlymemorystream.h | 30 |
2 files changed, 64 insertions, 0 deletions
diff --git a/core/fxcrt/cfx_readonlymemorystream.cpp b/core/fxcrt/cfx_readonlymemorystream.cpp new file mode 100644 index 0000000000..1d09759208 --- /dev/null +++ b/core/fxcrt/cfx_readonlymemorystream.cpp @@ -0,0 +1,34 @@ +// Copyright 2018 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/cfx_readonlymemorystream.h" + +#include "core/fxcrt/fx_safe_types.h" + +CFX_ReadOnlyMemoryStream::CFX_ReadOnlyMemoryStream(const uint8_t* pBuf, + FX_FILESIZE size) + : m_pBuf(pBuf), m_size(size) {} + +CFX_ReadOnlyMemoryStream::~CFX_ReadOnlyMemoryStream() = default; + +FX_FILESIZE CFX_ReadOnlyMemoryStream::GetSize() { + return m_size; +} + +bool CFX_ReadOnlyMemoryStream::ReadBlock(void* buffer, + FX_FILESIZE offset, + size_t size) { + if (offset < 0) + return false; + + FX_SAFE_FILESIZE newPos = pdfium::base::checked_cast<FX_FILESIZE>(size); + newPos += offset; + if (!newPos.IsValid() || newPos.ValueOrDie() > m_size) + return false; + + memcpy(buffer, m_pBuf + offset, size); + return true; +} diff --git a/core/fxcrt/cfx_readonlymemorystream.h b/core/fxcrt/cfx_readonlymemorystream.h new file mode 100644 index 0000000000..e9d18222a4 --- /dev/null +++ b/core/fxcrt/cfx_readonlymemorystream.h @@ -0,0 +1,30 @@ +// Copyright 2018 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 CORE_FXCRT_CFX_READONLYMEMORYSTREAM_H_ +#define CORE_FXCRT_CFX_READONLYMEMORYSTREAM_H_ + +#include "core/fxcrt/fx_stream.h" +#include "core/fxcrt/retain_ptr.h" + +class CFX_ReadOnlyMemoryStream final : public IFX_SeekableReadStream { + public: + template <typename T, typename... Args> + friend RetainPtr<T> pdfium::MakeRetain(Args&&... args); + + // IFX_SeekableReadStream: + FX_FILESIZE GetSize() override; + bool ReadBlock(void* buffer, FX_FILESIZE offset, size_t size) override; + + private: + CFX_ReadOnlyMemoryStream(const uint8_t* pBuf, FX_FILESIZE size); + ~CFX_ReadOnlyMemoryStream() override; + + const uint8_t* const m_pBuf; + const FX_FILESIZE m_size; +}; + +#endif // CORE_FXCRT_CFX_READONLYMEMORYSTREAM_H_ |