115 lines
2.3 KiB
C++
115 lines
2.3 KiB
C++
#pragma once
|
|
#include <Windows.h>
|
|
#include <string>
|
|
|
|
// Buffer 带引用计数的缓存.
|
|
class Buffer
|
|
{
|
|
private:
|
|
PBYTE buf;
|
|
ULONG len;
|
|
ULONG padding;
|
|
std::string md5;
|
|
ULONG *ref;
|
|
void AddRef()
|
|
{
|
|
(*ref)++;
|
|
}
|
|
void DelRef()
|
|
{
|
|
(*ref)--;
|
|
}
|
|
public:
|
|
LPBYTE &Buf()
|
|
{
|
|
return buf;
|
|
}
|
|
~Buffer()
|
|
{
|
|
DelRef();
|
|
if (*ref == 0) {
|
|
if (buf!=NULL) {
|
|
delete[] buf;
|
|
buf = NULL;
|
|
}
|
|
delete ref;
|
|
ref = NULL;
|
|
}
|
|
}
|
|
Buffer():buf(NULL), len(0), ref(new ULONG(1)), padding(0)
|
|
{
|
|
|
|
}
|
|
Buffer(const BYTE * b, int n, int padding=0, const std::string& md5="") :
|
|
len(n), ref(new ULONG(1)), padding(padding), md5(md5)
|
|
{
|
|
buf = new BYTE[n];
|
|
memcpy(buf, b, n);
|
|
}
|
|
Buffer(Buffer& o)
|
|
{
|
|
o.AddRef();
|
|
buf = o.buf;
|
|
len = o.len;
|
|
ref = o.ref;
|
|
}
|
|
Buffer operator =(Buffer &o)
|
|
{
|
|
o.AddRef();
|
|
buf = o.buf;
|
|
len = o.len;
|
|
ref = o.ref;
|
|
return *this;
|
|
}
|
|
char* c_str() const
|
|
{
|
|
return (char*)buf;
|
|
}
|
|
ULONG length(bool noPadding=false)const
|
|
{
|
|
return noPadding ? len - padding : len;
|
|
}
|
|
std::string MD5() const
|
|
{
|
|
return md5;
|
|
}
|
|
};
|
|
|
|
class CBuffer
|
|
{
|
|
public:
|
|
CBuffer(void);
|
|
~CBuffer(void);
|
|
|
|
ULONG ReadBuffer(PBYTE Buffer, ULONG ulLength);
|
|
ULONG GetBufferLength(); // 获得有效数据长度
|
|
ULONG GetBufferLen()
|
|
{
|
|
return GetBufferLength();
|
|
}
|
|
VOID ClearBuffer();
|
|
BOOL WriteBuffer(PBYTE Buffer, ULONG ulLength);
|
|
BOOL Write(PBYTE Buffer, ULONG ulLength)
|
|
{
|
|
return WriteBuffer(Buffer, ulLength);
|
|
}
|
|
BOOL WriteBuffer(CBuffer& buf)
|
|
{
|
|
return WriteBuffer(buf.GetBuffer(), buf.GetBufferLen());
|
|
}
|
|
LPBYTE GetBuffer(ULONG ulPos=0);
|
|
Buffer GetMyBuffer(ULONG ulPos);
|
|
BYTE GetBYTE(ULONG ulPos);
|
|
BOOL CopyBuffer(PVOID pDst, ULONG nLen, ULONG ulPos);
|
|
ULONG RemoveCompletedBuffer(ULONG ulLength);
|
|
std::string Skip(ULONG ulPos);
|
|
|
|
protected:
|
|
PBYTE m_Base;
|
|
PBYTE m_Ptr;
|
|
ULONG m_ulMaxLength;
|
|
CRITICAL_SECTION m_cs;
|
|
ULONG DeAllocateBuffer(ULONG ulLength); // 私有
|
|
ULONG ReAllocateBuffer(ULONG ulLength); // 私有
|
|
};
|