third-party libs

third-party/BugTrap-1.3-57bd515/source/MemStream.h

@@ -0,0 +1,154 @@
+/*
+ * This is a part of the BugTrap package.
+ * Copyright (c) 2005-2009 IntelleSoft.
+ * All rights reserved.
+ *
+ * Description: In-memory stream.
+ * Author: Maksim Pyatkovskiy.
+ *
+ * This source code is only intended as a supplement to the
+ * BugTrap package reference and related electronic documentation
+ * provided with the product. See these sources for detailed
+ * information regarding the BugTrap package.
+ */
+
+#pragma once
+
+#include "Stream.h"
+
+/// In-memory stream.
+class CMemStream : public CStream
+{
+public:
+	/// Initialize the object.
+	CMemStream(void);
+	/// Initialize the object.
+	explicit CMemStream(size_t nSize);
+	/// Destroy the object.
+	virtual ~CMemStream(void);
+	/// Copy constructor.
+	CMemStream(const CMemStream& rMemStream);
+	/// Assignment operator.
+	CMemStream& operator=(const CMemStream& rMemStream);
+	/// Get list of supported features.
+	virtual unsigned GetFeatures(void) const;
+	/// Get pointer to the buffer.
+	const BYTE* GetBuffer(void) const;
+	/// Return true if end of stream has been reached.
+	virtual bool IsEndOfStream(void) const;
+	/// Get data length.
+	virtual size_t GetLength(void) const;
+	/// Set number of bytes in the stream.
+	virtual size_t SetLength(size_t nLength);
+	/// Get current position.
+	virtual size_t GetPosition(void) const;
+	/// Set current position.
+	virtual size_t SetPosition(ptrdiff_t nOffset, int nMoveMethod);
+	/// Write data to the buffer.
+	virtual bool WriteByte(BYTE bValue);
+	/// Write data to the buffer.
+	virtual size_t WriteByte(BYTE bValue, size_t nCount);
+	/// Write data to the buffer.
+	virtual size_t WriteBytes(const BYTE* pBytes, size_t nCount);
+	/// Write data to the buffer.
+	size_t WriteStream(CMemStream* pMemStream);
+	/// Clear data in the buffer.
+	virtual void Close(void);
+	/// Read one byte from the stream.
+	virtual bool ReadByte(BYTE& bValue);
+	/// Read array of bytes from the stream.
+	virtual size_t ReadBytes(BYTE* arrBytes, size_t nNumBytes);
+
+private:
+	/// Initialize member variables.
+	void InitBuffer(void);
+	/// Initialize member variables.
+	void InitBuffer(size_t nSize);
+	/// Copy data from one buffer to another.
+	void CopyData(const CMemStream& rMemStream);
+	/// Ensures that the array has enough space.
+	void EnsureSize(size_t nSize, bool bAdaptiveGrowth);
+	/// Set current position.
+	void SetPositionPriv(ptrdiff_t nOffset, size_t nStartFrom);
+
+	/// Data length.
+	size_t m_nLength;
+	/// Data position.
+	size_t m_nPosition;
+	/// Size of allocated buffer.
+	size_t m_nSize;
+	/// Pointer to the buffer.
+	BYTE* m_pBuffer;
+};
+
+inline CMemStream::CMemStream(void)
+{
+	InitBuffer();
+}
+
+inline CMemStream::~CMemStream(void)
+{
+	delete[] m_pBuffer;
+}
+
+/**
+ * @return bit mask of supported features.
+ */
+inline unsigned CMemStream::GetFeatures(void) const
+{
+	return (SF_CLOSE | SF_GETLENGTH | SF_SETLENGTH | SF_GETPOSITION | SF_SETPOSITION);
+}
+
+/**
+ * @return pointer to the buffer.
+ */
+inline const BYTE* CMemStream::GetBuffer(void) const
+{
+	return m_pBuffer;
+}
+
+/**
+ * @return length of the data.
+ */
+inline size_t CMemStream::GetLength(void) const
+{
+	return m_nLength;
+}
+
+/**
+ * @return true if end of stream has been reached.
+ */
+inline bool CMemStream::IsEndOfStream(void) const
+{
+	return (m_nPosition >= m_nLength);
+}
+
+/**
+ * @return current stream position.
+ */
+inline size_t CMemStream::GetPosition(void) const
+{
+	return m_nPosition;
+}
+
+inline void CMemStream::Close(void)
+{
+	m_nLength = m_nPosition = 0;
+}
+
+/**
+ * @param pMemStream - another binary buffer.
+ * @return true if data has been written.
+ */
+inline size_t CMemStream::WriteStream(CMemStream* pMemStream)
+{
+	return WriteBytes(pMemStream->m_pBuffer + pMemStream->m_nPosition, pMemStream->m_nLength - pMemStream->m_nPosition);
+}
+
+/**
+ * @param nSize - initial buffer size.
+ */
+inline CMemStream::CMemStream(size_t nSize)
+{
+	InitBuffer(nSize);
+}