111 lines
2.5 KiB
C++
111 lines
2.5 KiB
C++
|
/*++
|
||
|
|
|
||
|
|
Copyright (C) 2000-2001 Microsoft Corporation
|
||
|
|
|
||
|
|
--*/
|
||
|
|
|
||
|
|
#include <wbemcomn.h>
|
||
|
|
#include "longstg.h"
|
||
|
|
#include "absfile.h"
|
||
|
|
|
||
|
|
DWORD CAbstractFile::Write(DWORD dwOffs, LPVOID pMem, DWORD dwBytes,
|
||
|
|
DWORD *pdwWritten)
|
||
|
|
{
|
||
|
|
long lRes = m_pStage->WriteFile(m_nId, dwOffs, (BYTE*)pMem, dwBytes,
|
||
|
|
pdwWritten);
|
||
|
|
return lRes;
|
||
|
|
}
|
||
|
|
|
||
|
|
DWORD CAbstractFile::Read(DWORD dwOffs, LPVOID pMem, DWORD dwBytes,
|
||
|
|
DWORD *pdwRead)
|
||
|
|
{
|
||
|
|
long lRes = m_pStage->ReadFile(m_nId, dwOffs, (BYTE*)pMem, dwBytes,
|
||
|
|
pdwRead);
|
||
|
|
return lRes;
|
||
|
|
}
|
||
|
|
|
||
|
|
DWORD CAbstractFile::SetFileLength(DWORD dwLen)
|
||
|
|
{
|
||
|
|
long lRes = m_pStage->SetFileLength(m_nId, dwLen);
|
||
|
|
return lRes;
|
||
|
|
}
|
||
|
|
|
||
|
|
DWORD CAbstractFile::GetFileLength(DWORD* pdwLength)
|
||
|
|
{
|
||
|
|
long lRes = m_pStage->GetFileLength(m_nId, pdwLength);
|
||
|
|
return lRes;
|
||
|
|
}
|
||
|
|
|
||
|
|
void CAbstractFile::Touch()
|
||
|
|
{
|
||
|
|
m_pStage->TouchTransaction();
|
||
|
|
}
|
||
|
|
|
||
|
|
long CAbstractFileSource::Create(LPCWSTR wszFileName, long lMaxFileSize,
|
||
|
|
long lAbortTransactionFileSize)
|
||
|
|
{
|
||
|
|
m_Stage.SetMaxFileSize(lMaxFileSize, lAbortTransactionFileSize);
|
||
|
|
return m_Stage.Create(wszFileName);
|
||
|
|
}
|
||
|
|
|
||
|
|
long CAbstractFileSource::Start()
|
||
|
|
{
|
||
|
|
return m_Stage.Initialize();
|
||
|
|
}
|
||
|
|
|
||
|
|
long CAbstractFileSource::Stop(DWORD dwShutDownFlags)
|
||
|
|
{
|
||
|
|
return m_Stage.Uninitialize(dwShutDownFlags);
|
||
|
|
}
|
||
|
|
|
||
|
|
DWORD CAbstractFileSource::Register(HANDLE hFile, int nID,
|
||
|
|
bool bSupportsOverwrites,
|
||
|
|
CAbstractFile **pFile)
|
||
|
|
{
|
||
|
|
long lRes = m_Stage.RegisterFile(nID, hFile, bSupportsOverwrites);
|
||
|
|
if(lRes != ERROR_SUCCESS)
|
||
|
|
return lRes;
|
||
|
|
|
||
|
|
*pFile = new CAbstractFile(&m_Stage, nID);
|
||
|
|
if(*pFile == NULL)
|
||
|
|
return ERROR_OUTOFMEMORY;
|
||
|
|
|
||
|
|
return ERROR_SUCCESS;
|
||
|
|
}
|
||
|
|
|
||
|
|
DWORD CAbstractFileSource::CloseAll()
|
||
|
|
{
|
||
|
|
long lRes = m_Stage.CloseAllFiles();
|
||
|
|
if(lRes == ERROR_SUCCESS)
|
||
|
|
return TRUE;
|
||
|
|
else
|
||
|
|
return FALSE;
|
||
|
|
}
|
||
|
|
|
||
|
|
DWORD CAbstractFileSource::Begin(DWORD *pdwTransId)
|
||
|
|
{
|
||
|
|
if(pdwTransId)
|
||
|
|
*pdwTransId = 0;
|
||
|
|
long lRes = m_Stage.BeginTransaction();
|
||
|
|
return lRes;
|
||
|
|
}
|
||
|
|
|
||
|
|
DWORD CAbstractFileSource::Commit(DWORD dwTransId)
|
||
|
|
{
|
||
|
|
_ASSERT(dwTransId == 0, L"");
|
||
|
|
long lRes = m_Stage.CommitTransaction();
|
||
|
|
return lRes;
|
||
|
|
}
|
||
|
|
|
||
|
|
DWORD CAbstractFileSource::Rollback(DWORD dwTransId, bool* pbNonEmpty)
|
||
|
|
{
|
||
|
|
_ASSERT(dwTransId == 0, L"");
|
||
|
|
long lRes = m_Stage.AbortTransaction(pbNonEmpty);
|
||
|
|
return lRes;
|
||
|
|
}
|
||
|
|
|
||
|
|
void CAbstractFileSource::Dump(FILE* f)
|
||
|
|
{
|
||
|
|
m_Stage.Dump(f);
|
||
|
|
}
|