122 lines
3.9 KiB
C++
122 lines
3.9 KiB
C++
#pragma once
|
|
|
|
#include <Tick.h>
|
|
|
|
#ifdef _DEBUG
|
|
#define _BUFFER_TRACE
|
|
#endif
|
|
#include <_osdef.h>
|
|
#include <Tick.h>
|
|
#include <string.h>
|
|
|
|
#include "Lock.h"
|
|
|
|
using namespace wylib::time::tick64;
|
|
|
|
class CBufferAllocator
|
|
{
|
|
|
|
public:
|
|
static const DWORD SmallBufferSize = 64; //小块内存块大小
|
|
static const DWORD MiddleBufferSize = 256; //中等内存块大小
|
|
static const DWORD LargeBufferSize = 1024; //大块内存块大小
|
|
|
|
#pragma pack(push, 1)
|
|
typedef struct tagBuffer
|
|
{
|
|
size_t dwSize; //内存数据段大小
|
|
tagBuffer* pPrevBuffer;//指向上一个内存块的指针
|
|
TICKCOUNT dwFreeTick; //即将被销毁的时间
|
|
#ifdef _BUFFER_TRACE
|
|
tagBuffer* pNextBuffer;//指向上下个内存块的指针，用于调试
|
|
struct
|
|
{
|
|
const char* lpFile; //申请内存块的代码位置，用于调试
|
|
int nLine;
|
|
}al, fl;
|
|
BOOL boUsing; //是否使用的标记
|
|
#endif
|
|
}BUFFER, *PBUFFER;
|
|
#pragma pack(pop)
|
|
|
|
struct BufferStatic
|
|
{
|
|
size_t dwAllocSize;
|
|
size_t dwFreeSize;
|
|
};
|
|
|
|
typedef struct tagAllocatorMemoryInfo
|
|
{
|
|
BufferStatic SmallBuffer;
|
|
BufferStatic MiddleBuffer;
|
|
BufferStatic LargeBuffer;
|
|
BufferStatic SuperBuffer;
|
|
}ALLOCATOR_MEMORY_INFO, *PALLOCATPR_MEMORY_INFOR;
|
|
|
|
private:
|
|
//CRITICAL_SECTION m_BufferLock; //内存块申请锁
|
|
wylib::sync::lock::CCSLock m_BufferLock; //加锁
|
|
|
|
PBUFFER m_pLastSmallBuffer; //指向最后一个空闲的小型内存块的指针
|
|
PBUFFER m_pLastMiddleBuffer; //指向最后一个空闲的中型内存块的指针
|
|
PBUFFER m_pLastLargeBuffer; //指向最后一个空闲的大型内存块的指针
|
|
PBUFFER m_pLastSuperBuffer; //指向最后一个空闲的超级内存块的指针
|
|
#ifdef _BUFFER_TRACE
|
|
PBUFFER m_pLastAllocedBuffer; //指向最后一个被申请的内存块
|
|
#endif
|
|
size_t m_dwAllocedSmallSize; //已被申请的小型内存块大小
|
|
size_t m_dwAllocedMiddleSize; //已被申请的中型内存块大小
|
|
size_t m_dwAllocedLargeSize; //已被申请的大型内存块大小
|
|
size_t m_dwAllocedSuperSize; //已被申请的超级内存块大小
|
|
size_t m_dwFreedSmallSize; //已被释放的小型内存块大小
|
|
size_t m_dwFreedMiddleSize; //已被释放的中型内存块大小
|
|
size_t m_dwFreedLargeSize; //已被释放的大型内存块大小
|
|
size_t m_dwFreedSuperSize; //已被释放的超级内存块大小
|
|
public:
|
|
CBufferAllocator();
|
|
~CBufferAllocator();
|
|
|
|
#ifndef _BUFFER_TRACE
|
|
PVOID AllocBuffer(size_t dwSize);
|
|
PVOID ReallocBuffer(LPVOID ptr, size_t dwNewSize);
|
|
VOID FreeBuffer(PVOID ptr);
|
|
#else
|
|
PVOID _AllocBuffer(size_t dwSize, const char* lpFile, int nLine);
|
|
PVOID _ReallocBuffer(LPVOID ptr, size_t dwNewSize, const char* lpFile, int nLine);
|
|
VOID _FreeBuffer(PVOID ptr, const char* lpFile, int nLine);
|
|
#define AllocBuffer(dwSize) _AllocBuffer(dwSize, __FILE__, __LINE__)
|
|
#define ReallocBuffer(ptr, dwNewSize) _ReallocBuffer(ptr, dwNewSize, __FILE__, __LINE__)
|
|
#define FreeBuffer(ptr) _FreeBuffer(ptr, __FILE__, __LINE__)
|
|
#endif
|
|
|
|
VOID GetMemoryInfo(CBufferAllocator::PALLOCATPR_MEMORY_INFOR pMemoryInfo);
|
|
// bIgnoreInterval:是否忽略间隔。如果忽略时间间隔，不管是否时间间隔到，都强制删除
|
|
VOID CheckFreeBuffers(bool bIgnoreInterval = false);
|
|
};
|
|
|
|
class DataBuffer
|
|
{
|
|
public:
|
|
CBufferAllocator *m_pAllocator;
|
|
size_t m_dwSize;
|
|
char *pData;//内存块指针
|
|
char *ptr;//写内存指针
|
|
char *pEnd;//内存块结束指针
|
|
char *pReadPtr;//读内存指针
|
|
|
|
#ifdef _BUFFER_TRACE
|
|
DataBuffer(CBufferAllocator *pAllocator, size_t dwSize, const char *_FILE, const int _LINE);
|
|
#else
|
|
DataBuffer(CBufferAllocator *pAllocator, size_t dwSize);
|
|
#endif
|
|
~DataBuffer();
|
|
};
|
|
|
|
#ifdef _BUFFER_TRACE
|
|
#define DATA_BUFFER_VAR(_allocator, _name, _size) DataBuffer _name((_allocator), (_size), __FILE__, __LINE__)
|
|
#else
|
|
#define DATA_BUFFER_VAR(_allocator, _name, _size) DataBuffer _name((_allocator), (_size))
|
|
#endif
|
|
|
|
|