172 lines
4.8 KiB
C++
172 lines
4.8 KiB
C++
#include <fusenetincludes.h>
|
|
#include <assemblyfile.h>
|
|
|
|
|
|
// ---------------------------------------------------------------------------
|
|
// CreateAssemblyFile
|
|
// ---------------------------------------------------------------------------
|
|
STDAPI
|
|
CreateAssemblyFileInfo(
|
|
LPASSEMBLY_FILE_INFO *ppAssemblyFile)
|
|
{
|
|
HRESULT hr = S_OK;
|
|
|
|
CAssemblyFileInfo *pAssemblyFile = NULL;
|
|
|
|
pAssemblyFile = new(CAssemblyFileInfo);
|
|
if (!pAssemblyFile)
|
|
{
|
|
hr = E_OUTOFMEMORY;
|
|
goto exit;
|
|
}
|
|
|
|
|
|
exit:
|
|
|
|
*ppAssemblyFile = static_cast<IAssemblyFileInfo*> (pAssemblyFile);
|
|
|
|
return hr;
|
|
}
|
|
|
|
// ---------------------------------------------------------------------------
|
|
// IsEqualAssemblyFileInfo
|
|
// ---------------------------------------------------------------------------
|
|
BOOL
|
|
IsEqualAssemblyFileInfo(LPASSEMBLY_FILE_INFO pAsmFileInfo1, LPASSEMBLY_FILE_INFO pAsmFileInfo2)
|
|
{
|
|
return (*(static_cast<CAssemblyFileInfo*> (pAsmFileInfo1))
|
|
== *(static_cast<CAssemblyFileInfo*> (pAsmFileInfo2)));
|
|
}
|
|
|
|
// ---------------------------------------------------------------------------
|
|
// ctor
|
|
// ---------------------------------------------------------------------------
|
|
CAssemblyFileInfo::CAssemblyFileInfo()
|
|
: _dwSig('ELIF'), _cRef(1), _hr(S_OK)
|
|
{
|
|
memset(_fi, 0, sizeof(_fi));
|
|
}
|
|
|
|
// ---------------------------------------------------------------------------
|
|
// dtor
|
|
// ---------------------------------------------------------------------------
|
|
CAssemblyFileInfo::~CAssemblyFileInfo()
|
|
{
|
|
for (DWORD i = ASM_FILE_NAME; i < ASM_FILE_MAX; i++)
|
|
SAFEDELETEARRAY(_fi[i].pwzProperty);
|
|
}
|
|
|
|
// ---------------------------------------------------------------------------
|
|
// operator==
|
|
// ---------------------------------------------------------------------------
|
|
BOOL
|
|
CAssemblyFileInfo::operator==( CAssemblyFileInfo& asmFIRHS )
|
|
{
|
|
BOOL bEqual = TRUE;
|
|
|
|
for (DWORD i = ASM_FILE_NAME; i < ASM_FILE_MAX; i++)
|
|
{
|
|
// BUGBUG?: case sensitivity and issues?
|
|
// BUGBUG?: locale?
|
|
if (((_fi[i].pwzProperty) && !((asmFIRHS._fi[i]).pwzProperty)) ||
|
|
(!(_fi[i].pwzProperty) && ((asmFIRHS._fi[i]).pwzProperty)) ||
|
|
((_fi[i].pwzProperty)
|
|
&& wcscmp(_fi[i].pwzProperty, (asmFIRHS._fi[i]).pwzProperty) != 0))
|
|
{
|
|
bEqual = FALSE;
|
|
break;
|
|
}
|
|
}
|
|
|
|
return bEqual;
|
|
}
|
|
|
|
// ---------------------------------------------------------------------------
|
|
// Set
|
|
// ---------------------------------------------------------------------------
|
|
HRESULT CAssemblyFileInfo::Set(DWORD dwId, LPCOLESTR pwzProperty)
|
|
{
|
|
DWORD ccProperty = lstrlen(pwzProperty) + 1;
|
|
LPWSTR pwzTmp = new WCHAR[ccProperty];
|
|
if (!pwzTmp)
|
|
{
|
|
_hr = E_OUTOFMEMORY;
|
|
goto exit;
|
|
}
|
|
memcpy(pwzTmp, pwzProperty, ccProperty * sizeof(WCHAR));
|
|
SAFEDELETEARRAY(_fi[dwId].pwzProperty);
|
|
_fi[dwId].pwzProperty = pwzTmp;
|
|
_fi[dwId].ccProperty = ccProperty;
|
|
|
|
exit:
|
|
return _hr;
|
|
}
|
|
|
|
// ---------------------------------------------------------------------------
|
|
// Get
|
|
// ---------------------------------------------------------------------------
|
|
HRESULT CAssemblyFileInfo::Get(DWORD dwId, LPOLESTR *ppwzProperty, LPDWORD pccProperty)
|
|
{
|
|
*ppwzProperty = NULL;
|
|
*pccProperty = 0;
|
|
DWORD ccProperty = _fi[dwId].ccProperty ? _fi[dwId].ccProperty : 1;
|
|
LPWSTR pwzProperty = new WCHAR[ccProperty];
|
|
if (!pwzProperty)
|
|
{
|
|
_hr = E_OUTOFMEMORY;
|
|
goto exit;
|
|
}
|
|
memcpy(pwzProperty, _fi[dwId].pwzProperty, ccProperty * sizeof(WCHAR));
|
|
*ppwzProperty = pwzProperty;
|
|
*pccProperty = ccProperty;
|
|
|
|
exit:
|
|
return _hr;
|
|
}
|
|
|
|
|
|
// IUnknown Boilerplate
|
|
|
|
// ---------------------------------------------------------------------------
|
|
// CAssemblyFileInfo::QI
|
|
// ---------------------------------------------------------------------------
|
|
STDMETHODIMP
|
|
CAssemblyFileInfo::QueryInterface(REFIID riid, void** ppvObj)
|
|
{
|
|
if ( IsEqualIID(riid, IID_IUnknown)
|
|
|| IsEqualIID(riid, IID_IAssemblyFileInfo)
|
|
)
|
|
{
|
|
*ppvObj = static_cast<IAssemblyFileInfo*> (this);
|
|
AddRef();
|
|
return S_OK;
|
|
}
|
|
else
|
|
{
|
|
*ppvObj = NULL;
|
|
return E_NOINTERFACE;
|
|
}
|
|
}
|
|
|
|
// ---------------------------------------------------------------------------
|
|
// CAssemblyFileInfo::AddRef
|
|
// ---------------------------------------------------------------------------
|
|
STDMETHODIMP_(ULONG)
|
|
CAssemblyFileInfo::AddRef()
|
|
{
|
|
return InterlockedIncrement ((LONG*) &_cRef);
|
|
}
|
|
|
|
// ---------------------------------------------------------------------------
|
|
// CAssemblyFileInfo::Release
|
|
// ---------------------------------------------------------------------------
|
|
STDMETHODIMP_(ULONG)
|
|
CAssemblyFileInfo::Release()
|
|
{
|
|
ULONG lRet = InterlockedDecrement ((LONG*) &_cRef);
|
|
if (!lRet)
|
|
delete this;
|
|
return lRet;
|
|
}
|
|
|