57 lines
895 B
C++
57 lines
895 B
C++
#include "dataobj.h"
|
||
|
||
HRESULT
|
||
GetStgMedpUnkForRelease(LPSTGMEDIUM pSTM, IUnknown **pp_unk)
|
||
{
|
||
CStgMedIf *p_smi = new CStgMedIf(pSTM);
|
||
HRESULT hr = p_smi->QueryInterface(IID_IUnknown, (PPVOID)pp_unk);
|
||
return hr;
|
||
}
|
||
|
||
CStgMedIf::CStgMedIf(LPSTGMEDIUM pSTM)
|
||
{
|
||
m_pSTM = pSTM;
|
||
m_cRef = 0;
|
||
}
|
||
|
||
|
||
STDMETHODIMP
|
||
CStgMedIf::QueryInterface(
|
||
REFIID riid,
|
||
LPLPVOID ppv
|
||
)
|
||
{
|
||
*ppv = NULL;
|
||
if (IsEqualIID(riid, IID_IUnknown))
|
||
{
|
||
*ppv = (LPVOID)this;
|
||
((LPUNKNOWN)*ppv)->AddRef();
|
||
return NOERROR;
|
||
}
|
||
return ResultFromScode(E_NOINTERFACE);
|
||
}
|
||
|
||
|
||
STDMETHODIMP_(ULONG)
|
||
CStgMedIf::AddRef(void)
|
||
{
|
||
return ++m_cRef;
|
||
}
|
||
|
||
|
||
STDMETHODIMP_(ULONG)
|
||
CStgMedIf::Release(void)
|
||
{
|
||
ULONG cRefT;
|
||
|
||
cRefT = --m_cRef;
|
||
|
||
if (0==m_cRef)
|
||
{
|
||
m_pSTM->pUnkForRelease = 0;
|
||
ReleaseStgMedium(m_pSTM);
|
||
delete this;
|
||
}
|
||
return cRefT;
|
||
}
|
||
|