| // Copyright 2017 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_BINARYBUF_H_ |
| #define CORE_FXCRT_CFX_BINARYBUF_H_ |
| |
| #include <memory> |
| |
| #include "core/fxcrt/fx_memory.h" |
| #include "core/fxcrt/fx_string.h" |
| #include "core/fxcrt/fx_system.h" |
| |
| class CFX_BinaryBuf { |
| public: |
| CFX_BinaryBuf(); |
| explicit CFX_BinaryBuf(FX_STRSIZE size); |
| ~CFX_BinaryBuf(); |
| |
| uint8_t* GetBuffer() const { return m_pBuffer.get(); } |
| FX_STRSIZE GetSize() const { return m_DataSize; } |
| |
| void Clear(); |
| void EstimateSize(FX_STRSIZE size, FX_STRSIZE alloc_step = 0); |
| void AppendBlock(const void* pBuf, FX_STRSIZE size); |
| void AppendString(const CFX_ByteString& str) { |
| AppendBlock(str.c_str(), str.GetLength()); |
| } |
| |
| void AppendByte(uint8_t byte) { |
| ExpandBuf(1); |
| m_pBuffer.get()[m_DataSize++] = byte; |
| } |
| |
| void InsertBlock(FX_STRSIZE pos, const void* pBuf, FX_STRSIZE size); |
| void Delete(FX_STRSIZE start_index, FX_STRSIZE count); |
| |
| // Releases ownership of |m_pBuffer| and returns it. |
| std::unique_ptr<uint8_t, FxFreeDeleter> DetachBuffer(); |
| |
| protected: |
| void ExpandBuf(FX_STRSIZE size); |
| |
| FX_STRSIZE m_AllocStep; |
| FX_STRSIZE m_AllocSize; |
| FX_STRSIZE m_DataSize; |
| std::unique_ptr<uint8_t, FxFreeDeleter> m_pBuffer; |
| }; |
| |
| #endif // CORE_FXCRT_CFX_BINARYBUF_H_ |