diff options
Diffstat (limited to 'core/fxcrt/cfx_binarybuf.h')
-rw-r--r-- | core/fxcrt/cfx_binarybuf.h | 52 |
1 files changed, 52 insertions, 0 deletions
diff --git a/core/fxcrt/cfx_binarybuf.h b/core/fxcrt/cfx_binarybuf.h new file mode 100644 index 0000000000..3081d02902 --- /dev/null +++ b/core/fxcrt/cfx_binarybuf.h @@ -0,0 +1,52 @@ +// 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_ |