C/C++内存操作经典源码（仅供分享）

//计算数组维数
#define CountArray(Array) (sizeof(Array)/sizeof(Array[0]))

//计算字符长度
#define CountString(String) ((UINT)((lstrlen(String)+1)*sizeof(TCHAR)))

//安全删除指针
#define SafeDelete(pData) { try { if(pData) delete pData; } catch (...) { } pData=NULL; }

//安全删除指针
#define SafeDeleteArray(pData) { try { delete [] pData; } catch (...) { } pData=NULL; }



//************************************
// <p>Description: 插入数据</p>
// <p>Parameters: </p>
// <p> WORD wIdentifier</p>
// <p> void * const pBuffer</p>
// <p> WORD wDataSize</p>
//
// <p>Returns: bool</p>
//************************************
bool CDataStorage::AddData(WORD wIdentifier, void * const pBuffer, WORD wDataSize)
{
//初始化数据
tagDataHead DataHead;
DataHead.wDataSize=wDataSize;
DataHead.wIdentifier=wIdentifier;
DataHead.dwInsertTime=(DWORD)time(NULL)/1000L;
DWORD dwCopySize=sizeof(DataHead)+wDataSize;

try
{
//效验缓冲区
if ((m_dwDataSize+dwCopySize)>m_dwBufferSize) throw 0;
else if ((m_dwInsertPos==m_dwTerminalPos)&&((m_dwInsertPos+dwCopySize)>m_dwBufferSize))
{
if (m_dwDataQueryPos>=dwCopySize) m_dwInsertPos=0;
else throw 0;
}
else if ((m_dwInsertPos<m_dwTerminalPos)&&((m_dwInsertPos+dwCopySize)>m_dwDataQueryPos)) throw 0;
}
catch (...)
{
try
{
//申请内存
DWORD dwNewBufferSize=m_dwBufferSize+__max(m_dwBufferSize/2L,(wDataSize+sizeof(tagDataHead))*10L);
BYTE * pNewQueueServiceBuffer=new BYTE [dwNewBufferSize];
if (pNewQueueServiceBuffer==NULL) return false;

//拷贝数据
if (m_pDataStorageBuffer!=NULL)
{
_ASSERT(m_dwTerminalPos>=m_dwDataSize);
_ASSERT(m_dwTerminalPos>=m_dwDataQueryPos);
DWORD dwPartOneSize=m_dwTerminalPos-m_dwDataQueryPos;
if (dwPartOneSize>0L) CopyMemory(pNewQueueServiceBuffer,m_pDataStorageBuffer+m_dwDataQueryPos,dwPartOneSize);
if (m_dwDataSize>dwPartOneSize)
{
_ASSERT((m_dwInsertPos+dwPartOneSize)==m_dwDataSize);
CopyMemory(pNewQueueServiceBuffer+dwPartOneSize,m_pDataStorageBuffer,m_dwInsertPos);
}
}

//调整变量
m_dwDataQueryPos=0L;
m_dwInsertPos=m_dwDataSize;
m_dwTerminalPos=m_dwDataSize;
m_dwBufferSize=dwNewBufferSize;
SafeDeleteArray(m_pDataStorageBuffer);
m_pDataStorageBuffer=pNewQueueServiceBuffer;
}
catch (...) { return false; }
}

try
{
//拷贝数据
CopyMemory(m_pDataStorageBuffer+m_dwInsertPos,&DataHead,sizeof(DataHead));
if (wDataSize>0)
{
_ASSERT(pBuffer!=NULL);
CopyMemory(m_pDataStorageBuffer+m_dwInsertPos+sizeof(DataHead),pBuffer,wDataSize);
}

//调整数据
m_dwDataPacketCount++;
m_dwDataSize+=dwCopySize;
m_dwInsertPos+=dwCopySize;
m_dwTerminalPos=__max(m_dwTerminalPos,m_dwInsertPos);

return true;
}
catch (...) { }

return false;
}