133 lines
2.8 KiB
C++
133 lines
2.8 KiB
C++
//*********************************************************************
|
|
//* Microsoft Windows **
|
|
//* Copyright(c) Microsoft Corp., 1998 **
|
|
//*********************************************************************
|
|
|
|
|
|
#include "pre.h"
|
|
|
|
|
|
CStorage::CStorage(void)
|
|
{
|
|
// initialize all items
|
|
for (int i=0; i<MAX_STORGE_ITEM; i++)
|
|
{
|
|
m_pItem[i] = NULL;
|
|
}
|
|
}
|
|
|
|
CStorage::~CStorage(void)
|
|
{
|
|
// Clean up
|
|
for (int i=0; i<MAX_STORGE_ITEM; i++)
|
|
{
|
|
if (m_pItem[i])
|
|
{
|
|
if (m_pItem[i]->pData)
|
|
{
|
|
delete [] m_pItem[i]->pData;
|
|
}
|
|
delete m_pItem[i];
|
|
}
|
|
}
|
|
}
|
|
|
|
// Associate the data with a key and puts it in storage
|
|
BOOL CStorage::Set(
|
|
STORAGEKEY key,
|
|
void far * pData,
|
|
DWORD dwSize
|
|
)
|
|
{
|
|
|
|
// checks for existence of previous item
|
|
if (m_pItem[key])
|
|
{
|
|
// Checks if a new allocation is necessary
|
|
if (m_pItem[key]->dwSize < dwSize )
|
|
{
|
|
// Too small, new reallocation
|
|
if (m_pItem[key]->pData)
|
|
{
|
|
delete [] m_pItem[key]->pData;
|
|
m_pItem[key]->pData = (void*) new CHAR[dwSize];
|
|
}
|
|
}
|
|
}
|
|
else
|
|
{
|
|
// Allocate a new item
|
|
m_pItem[key] = new ITEM;
|
|
if (m_pItem[key])
|
|
{
|
|
m_pItem[key]->pData = (void*) new CHAR[dwSize];
|
|
}
|
|
else
|
|
{
|
|
return FALSE;
|
|
}
|
|
}
|
|
|
|
if (m_pItem[key]->pData)
|
|
{
|
|
memcpy( m_pItem[key]->pData, pData, dwSize );
|
|
m_pItem[key]->dwSize = dwSize;
|
|
return TRUE;
|
|
}
|
|
|
|
return FALSE;
|
|
}
|
|
|
|
|
|
|
|
// Get the data with the specified key
|
|
void* CStorage::Get(STORAGEKEY key)
|
|
{
|
|
if (key < MAX_STORGE_ITEM)
|
|
{
|
|
if (m_pItem[key])
|
|
{
|
|
return m_pItem[key]->pData;
|
|
}
|
|
}
|
|
return NULL;
|
|
}
|
|
|
|
// Compare the data with the specified key with the data
|
|
// pointed by pData with size dwSize
|
|
BOOL CStorage::Compare
|
|
(
|
|
STORAGEKEY key,
|
|
void far * pData,
|
|
DWORD dwSize
|
|
)
|
|
{
|
|
// Make sure key is within our range
|
|
if (key < MAX_STORGE_ITEM)
|
|
{
|
|
// make sure item is non-null
|
|
if (m_pItem[key])
|
|
{
|
|
// make sure item has data
|
|
if (m_pItem[key]->pData && pData)
|
|
{
|
|
if (m_pItem[key]->dwSize == dwSize)
|
|
{
|
|
if (memcmp(m_pItem[key]->pData,
|
|
pData,
|
|
dwSize) == 0)
|
|
{
|
|
return TRUE;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
return FALSE;
|
|
}
|
|
|
|
|
|
|
|
|