mirror of https://github.com/tongzx/nt5src
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
532 lines
8.4 KiB
532 lines
8.4 KiB
#pragma hdrstop
|
|
|
|
#include <ctballs.hxx>
|
|
#include <sem32.hxx>
|
|
|
|
|
|
// We need a semaphore to synchronize loads and releases.
|
|
CMutexSem mxsLoadRelease;
|
|
|
|
SAFE_INTERFACE_PTR(XIStream, IStream)
|
|
|
|
#define XPOS L"XPOS"
|
|
#define YPOS L"YPOS"
|
|
|
|
|
|
HRESULT ReadPos(IStorage *pstg, LPOLESTR pwszStream, ULONG *pulPos)
|
|
{
|
|
HRESULT hr;
|
|
|
|
BEGIN_BLOCK
|
|
|
|
XIStream xstrm;
|
|
|
|
// Read the streams for xpos and ypos
|
|
hr = pstg->OpenStream(pwszStream, NULL,
|
|
STGM_READ | STGM_SHARE_EXCLUSIVE, NULL, &xstrm);
|
|
|
|
if (FAILED(hr))
|
|
{
|
|
EXIT_BLOCK;
|
|
}
|
|
|
|
ULONG cb;
|
|
|
|
hr = xstrm->Read(pulPos, sizeof(*pulPos), &cb);
|
|
|
|
if (FAILED(hr))
|
|
{
|
|
EXIT_BLOCK;
|
|
}
|
|
|
|
hr = ResultFromScode(S_OK);
|
|
|
|
END_BLOCK
|
|
|
|
return hr;
|
|
}
|
|
|
|
HRESULT WritePos(IStorage *pstg, LPOLESTR pwszStream, ULONG ulPos)
|
|
{
|
|
HRESULT hr;
|
|
|
|
BEGIN_BLOCK
|
|
|
|
XIStream xstrm;
|
|
|
|
// Read the streams for xpos and ypos
|
|
hr = pstg->CreateStream(pwszStream,
|
|
STGM_CREATE | STGM_WRITE | STGM_SHARE_EXCLUSIVE, NULL, NULL,
|
|
&xstrm);
|
|
|
|
if (FAILED(hr))
|
|
{
|
|
EXIT_BLOCK;
|
|
}
|
|
|
|
|
|
ULONG cb;
|
|
|
|
hr = xstrm->Write(&ulPos, sizeof(ulPos), &cb);
|
|
|
|
if (FAILED(hr))
|
|
{
|
|
EXIT_BLOCK;
|
|
}
|
|
|
|
hr = ResultFromScode(S_OK);
|
|
|
|
END_BLOCK
|
|
|
|
return hr;
|
|
}
|
|
|
|
CTestBalls::CTestBalls(REFCLSID rclsid)
|
|
: _rclsid(rclsid), _fDirty(FALSE), _xPos(0), _yPos(0),
|
|
_fSaveInprogress(FALSE), _pstg(NULL), _dwRegister(0), _cRefs(1)
|
|
{
|
|
// Use as a flag for whether a file name has been assigned
|
|
_awszCurFile[0] = 0;
|
|
|
|
GlobalRefs(TRUE);
|
|
}
|
|
|
|
CTestBalls::~CTestBalls(void)
|
|
{
|
|
if (_pstg != NULL)
|
|
{
|
|
// Release the storage because we are done with it
|
|
ULONG ulCnt = _pstg->Release();
|
|
|
|
#if 0
|
|
// this test is not valud when running stress
|
|
if (ulCnt != 0)
|
|
{
|
|
DebugBreak();
|
|
}
|
|
#endif
|
|
}
|
|
|
|
if (_dwRegister)
|
|
{
|
|
IRunningObjectTable *prot;
|
|
GetRunningObjectTable(NULL, &prot);
|
|
prot->Revoke(_dwRegister);
|
|
prot->Release();
|
|
}
|
|
|
|
GlobalRefs(FALSE);
|
|
}
|
|
|
|
STDMETHODIMP CTestBalls::QueryInterface(REFIID iid, void **ppv)
|
|
{
|
|
HRESULT hr = ResultFromScode(S_OK);
|
|
|
|
// We support IUnknown, IPersistFile and IBalls
|
|
if (IsEqualIID(iid, IID_IUnknown))
|
|
{
|
|
*ppv = (IBalls *) this;
|
|
}
|
|
else if (IsEqualIID(iid, IID_IPersistFile))
|
|
{
|
|
*ppv = (IPersistFile *) this;
|
|
}
|
|
else if (IsEqualIID(iid, IID_IPersistStorage))
|
|
{
|
|
*ppv = (IPersistStorage *) this;
|
|
}
|
|
else if (IsEqualIID(iid, IID_IBalls))
|
|
{
|
|
*ppv = (IBalls *) this;
|
|
}
|
|
else
|
|
{
|
|
*ppv = NULL;
|
|
hr = ResultFromScode(E_NOINTERFACE);
|
|
}
|
|
|
|
if (SUCCEEDED(hr))
|
|
{
|
|
AddRef();
|
|
}
|
|
|
|
return hr;
|
|
}
|
|
|
|
STDMETHODIMP_(ULONG) CTestBalls::AddRef(void)
|
|
{
|
|
InterlockedIncrement(&_cRefs);
|
|
return _cRefs;
|
|
}
|
|
|
|
STDMETHODIMP_(ULONG) CTestBalls::Release(void)
|
|
{
|
|
CLock lck(mxsLoadRelease);
|
|
|
|
if (InterlockedDecrement(&_cRefs) == 0)
|
|
{
|
|
delete this;
|
|
return 0;
|
|
}
|
|
|
|
return _cRefs;
|
|
}
|
|
|
|
STDMETHODIMP CTestBalls::GetClassID(LPCLSID lpClassID)
|
|
{
|
|
*lpClassID = _rclsid;
|
|
return ResultFromScode(S_OK);
|
|
}
|
|
|
|
STDMETHODIMP CTestBalls::IsDirty()
|
|
{
|
|
return (_fDirty) ? ResultFromScode(S_OK) : ResultFromScode(S_FALSE);
|
|
}
|
|
|
|
STDMETHODIMP CTestBalls::Load(LPCOLESTR lpszFileName, DWORD grfMode)
|
|
{
|
|
CLock lck(mxsLoadRelease);
|
|
|
|
HRESULT hr;
|
|
|
|
BEGIN_BLOCK
|
|
|
|
hr = StgOpenStorage(lpszFileName, NULL,
|
|
STGM_READWRITE | STGM_SHARE_EXCLUSIVE, NULL, NULL, &_pstg);
|
|
|
|
if (FAILED(hr))
|
|
{
|
|
#if 0
|
|
// this test is not valid when running stress
|
|
if (hr == STG_E_LOCKVIOLATION)
|
|
{
|
|
DebugBreak();
|
|
}
|
|
#endif
|
|
EXIT_BLOCK;
|
|
}
|
|
|
|
// Get the saved xposition
|
|
hr = GetData();
|
|
|
|
if (FAILED(hr))
|
|
{
|
|
EXIT_BLOCK;
|
|
}
|
|
|
|
// Since everything went Ok save the file name
|
|
olestrcpy(_awszCurFile, lpszFileName);
|
|
|
|
// Create a file moniker for the object
|
|
IMoniker *pmk;
|
|
CreateFileMoniker(lpszFileName, &pmk);
|
|
|
|
// Register it in the running object table.
|
|
IRunningObjectTable *prot;
|
|
GetRunningObjectTable(NULL, &prot);
|
|
prot->Register(NULL, (IPersistFile *) this, pmk, &_dwRegister);
|
|
|
|
// Release the temporary objects
|
|
pmk->Release();
|
|
prot->Release();
|
|
|
|
END_BLOCK
|
|
|
|
return hr;
|
|
}
|
|
|
|
STDMETHODIMP CTestBalls::Save(LPCOLESTR lpszFileName, BOOL fRemember)
|
|
{
|
|
HRESULT hr;
|
|
|
|
BEGIN_BLOCK
|
|
|
|
IStorage *pstgNew;
|
|
|
|
// Save the data
|
|
if (olestrcmp(lpszFileName, _awszCurFile) == 0)
|
|
{
|
|
pstgNew = _pstg;
|
|
_fDirty = FALSE;
|
|
}
|
|
else
|
|
{
|
|
hr = StgCreateDocfile(lpszFileName,
|
|
STGM_CREATE | STGM_READWRITE | STGM_SHARE_EXCLUSIVE,
|
|
NULL, &pstgNew);
|
|
}
|
|
|
|
if (FAILED(hr))
|
|
{
|
|
EXIT_BLOCK;
|
|
}
|
|
|
|
WriteClassStg(pstgNew, _rclsid);
|
|
|
|
hr = SaveData(pstgNew);
|
|
|
|
if (FAILED(hr))
|
|
{
|
|
EXIT_BLOCK;
|
|
}
|
|
|
|
if (fRemember)
|
|
{
|
|
// Save the file name
|
|
olestrcpy(_awszCurFile, lpszFileName);
|
|
|
|
|
|
// Replace the storage
|
|
if (_pstg && pstgNew != _pstg)
|
|
{
|
|
_pstg->Release();
|
|
}
|
|
_pstg = pstgNew;
|
|
|
|
_fDirty = FALSE;
|
|
}
|
|
else
|
|
{
|
|
pstgNew->Release();
|
|
}
|
|
|
|
_fSaveInprogress = TRUE;
|
|
|
|
hr = ResultFromScode(S_OK);
|
|
|
|
END_BLOCK;
|
|
|
|
return hr;
|
|
}
|
|
|
|
STDMETHODIMP CTestBalls::SaveCompleted(LPCOLESTR lpszFileName)
|
|
{
|
|
_fSaveInprogress = FALSE;
|
|
return ResultFromScode(S_OK);
|
|
}
|
|
|
|
STDMETHODIMP CTestBalls::GetCurFile(LPOLESTR FAR * lpszFileName)
|
|
{
|
|
// Allocate a buffer for the file and copy in the data
|
|
if (_awszCurFile[0] == 0)
|
|
{
|
|
return ResultFromScode(S_FALSE);
|
|
}
|
|
|
|
|
|
IMalloc *pIMalloc;
|
|
|
|
HRESULT hr = CoGetMalloc(MEMCTX_TASK, &pIMalloc);
|
|
|
|
if (SUCCEEDED(hr))
|
|
{
|
|
*lpszFileName = (WCHAR *) pIMalloc->Alloc(
|
|
olestrlen((_awszCurFile) + 1) * sizeof(WCHAR));
|
|
|
|
olestrcpy(*lpszFileName, _awszCurFile);
|
|
|
|
hr = ResultFromScode(S_OK);
|
|
}
|
|
|
|
return hr;
|
|
}
|
|
|
|
STDMETHODIMP CTestBalls::MoveBall(ULONG xPos, ULONG yPos)
|
|
{
|
|
if (!_fSaveInprogress)
|
|
{
|
|
_fDirty = TRUE;
|
|
_xPos = xPos;
|
|
_yPos = yPos;
|
|
return S_OK;
|
|
}
|
|
|
|
// Can't change state because a save is still pending
|
|
return ResultFromScode(E_UNEXPECTED);
|
|
}
|
|
|
|
STDMETHODIMP CTestBalls::GetBallPos(ULONG *xPos, ULONG *yPos)
|
|
{
|
|
*xPos = _xPos;
|
|
*yPos = _yPos;
|
|
return S_OK;
|
|
}
|
|
|
|
STDMETHODIMP CTestBalls::IsOverLapped(IBalls *pIBall)
|
|
{
|
|
ULONG xPos;
|
|
ULONG yPos;
|
|
|
|
HRESULT hr = pIBall->GetBallPos(&xPos, &yPos);
|
|
|
|
if (SUCCEEDED(hr))
|
|
{
|
|
if ((xPos == _xPos) && (yPos == _yPos))
|
|
{
|
|
hr = ResultFromScode(S_OK);
|
|
}
|
|
else
|
|
{
|
|
hr = ResultFromScode(S_FALSE);
|
|
}
|
|
}
|
|
|
|
return hr;
|
|
}
|
|
|
|
STDMETHODIMP CTestBalls::IsContainedIn(ICube *pICube)
|
|
{
|
|
ULONG xPos;
|
|
ULONG yPos;
|
|
|
|
HRESULT hr = pICube->GetCubePos(&xPos, &yPos);
|
|
|
|
if (SUCCEEDED(hr))
|
|
{
|
|
if ((xPos == _xPos) && (yPos == _yPos))
|
|
{
|
|
hr = ResultFromScode(S_OK);
|
|
}
|
|
else
|
|
{
|
|
hr = ResultFromScode(S_FALSE);
|
|
}
|
|
}
|
|
|
|
return hr;
|
|
|
|
}
|
|
|
|
STDMETHODIMP CTestBalls::Clone(IBalls **ppIBall)
|
|
{
|
|
CTestBalls *ptballs = new CTestBalls(_rclsid);
|
|
|
|
ptballs->_xPos = _xPos;
|
|
ptballs->_yPos = _yPos;
|
|
ptballs->_fDirty = _fDirty;
|
|
_pstg->AddRef();
|
|
ptballs->_pstg = _pstg;
|
|
lstrcpy(ptballs->_awszCurFile, _awszCurFile);
|
|
return ResultFromScode(S_OK);
|
|
}
|
|
|
|
STDMETHODIMP CTestBalls::Echo(IUnknown *punkIn, IUnknown**ppunkOut)
|
|
{
|
|
*ppunkOut = punkIn;
|
|
return S_OK;
|
|
}
|
|
|
|
STDMETHODIMP CTestBalls::InitNew(LPSTORAGE pStg)
|
|
{
|
|
pStg->AddRef();
|
|
_pstg = pStg;
|
|
WriteClassStg(_pstg, _rclsid);
|
|
|
|
return ResultFromScode(S_OK);
|
|
}
|
|
|
|
STDMETHODIMP CTestBalls::Load(LPSTORAGE pStg)
|
|
{
|
|
HRESULT hr;
|
|
|
|
_pstg = pStg;
|
|
|
|
hr = GetData();
|
|
|
|
if (SUCCEEDED(hr))
|
|
{
|
|
_pstg->AddRef();
|
|
}
|
|
else
|
|
{
|
|
_pstg = NULL;
|
|
}
|
|
|
|
return hr;
|
|
}
|
|
|
|
STDMETHODIMP CTestBalls::Save(
|
|
LPSTORAGE pStgSave,
|
|
BOOL fSameAsLoad)
|
|
{
|
|
HRESULT hr;
|
|
|
|
if (!fSameAsLoad)
|
|
{
|
|
if (_pstg)
|
|
_pstg->Release();
|
|
|
|
_pstg = pStgSave;
|
|
|
|
pStgSave->AddRef();
|
|
}
|
|
else
|
|
{
|
|
pStgSave = _pstg;
|
|
}
|
|
|
|
WriteClassStg(pStgSave, _rclsid);
|
|
|
|
hr = SaveData(pStgSave);
|
|
|
|
_fSaveInprogress = TRUE;
|
|
|
|
return hr;
|
|
}
|
|
|
|
STDMETHODIMP CTestBalls::SaveCompleted(LPSTORAGE pStgSaved)
|
|
{
|
|
_fSaveInprogress = FALSE;
|
|
return ResultFromScode(S_OK);
|
|
}
|
|
|
|
STDMETHODIMP CTestBalls::HandsOffStorage(void)
|
|
{
|
|
// Figure out what to do here!
|
|
return ResultFromScode(E_UNEXPECTED);
|
|
}
|
|
|
|
HRESULT CTestBalls::GetData(void)
|
|
{
|
|
HRESULT hr;
|
|
|
|
BEGIN_BLOCK
|
|
|
|
// Get the saved xposition
|
|
hr = ReadPos(_pstg, XPOS, &_xPos);
|
|
|
|
if (FAILED(hr))
|
|
{
|
|
EXIT_BLOCK;
|
|
}
|
|
|
|
// Get the saved yposition
|
|
hr = ReadPos(_pstg, YPOS, &_yPos);
|
|
|
|
END_BLOCK
|
|
|
|
return hr;
|
|
}
|
|
|
|
HRESULT CTestBalls::SaveData(IStorage *pstg)
|
|
{
|
|
HRESULT hr;
|
|
|
|
BEGIN_BLOCK
|
|
|
|
// Get the saved xposition
|
|
hr = WritePos(pstg, XPOS, _xPos);
|
|
|
|
if (FAILED(hr))
|
|
{
|
|
EXIT_BLOCK;
|
|
}
|
|
|
|
// Get the saved yposition
|
|
hr = WritePos(pstg, YPOS, _yPos);
|
|
|
|
|
|
END_BLOCK
|
|
|
|
return hr;
|
|
}
|