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.
311 lines
8.6 KiB
311 lines
8.6 KiB
#include "shellprv.h"
|
|
#pragma hdrstop
|
|
|
|
class CPropertySetStg;
|
|
|
|
class CPropertyStg : public IPropertyStorage
|
|
{
|
|
public:
|
|
// IUnknown
|
|
STDMETHOD(QueryInterface) (THIS_ REFIID riid, void **ppv);
|
|
STDMETHOD_(ULONG,AddRef) (THIS);
|
|
STDMETHOD_(ULONG,Release) (THIS);
|
|
|
|
// IPropertyStorage
|
|
STDMETHODIMP ReadMultiple(ULONG cpspec, const PROPSPEC rgpspec[], PROPVARIANT rgpropvar[]);
|
|
STDMETHODIMP WriteMultiple(ULONG cpspec, const PROPSPEC rgpspec[], const PROPVARIANT rgpropvar[], PROPID propidNameFirst);
|
|
STDMETHODIMP DeleteMultiple(ULONG cpspec, const PROPSPEC rgpspec[]);
|
|
STDMETHODIMP ReadPropertyNames(ULONG cpropid, const PROPID rgpropid[], LPOLESTR rglpwstrName[]);
|
|
STDMETHODIMP WritePropertyNames(ULONG cpropid, const PROPID rgpropid[], const LPOLESTR rglpwstrName[]);
|
|
STDMETHODIMP DeletePropertyNames(ULONG cpropid, const PROPID rgpropid[]);
|
|
STDMETHODIMP Commit(DWORD grfCommitFlags);
|
|
STDMETHODIMP Revert(void);
|
|
STDMETHODIMP Enum(IEnumSTATPROPSTG **ppenum);
|
|
STDMETHODIMP SetTimes(const FILETIME *pctime, const FILETIME *patime,const FILETIME *pmtime);
|
|
STDMETHODIMP SetClass(REFCLSID clsid);
|
|
STDMETHODIMP Stat(STATPROPSETSTG *pstatpsstg);
|
|
|
|
CPropertyStg(CPropertySetStg *ppss, REFFMTID fmtid, DWORD grfMode);
|
|
void FMTIDPIDToSectionProp(REFFMTID fmtid, PROPID pid, LPTSTR pszSection, LPTSTR pszValueName);
|
|
|
|
private:
|
|
~CPropertyStg();
|
|
BOOL _SectionValueName(const PROPSPEC *ppspec,
|
|
LPTSTR pszSection, UINT cchSection, LPTSTR pszValueName, UINT cchValueName);
|
|
HRESULT _ReadProp(const PROPSPEC *ppspec, PROPVARIANT *ppropvar);
|
|
|
|
LONG _cRef;
|
|
CPropertySetStg *_ppss; // back ptr to parent
|
|
REFFMTID _fmtid;
|
|
DWORD _grfMode;
|
|
};
|
|
|
|
class CPropertySetStg : public IPropertySetStorage
|
|
{
|
|
friend CPropertyStg;
|
|
public:
|
|
// IUnknown
|
|
STDMETHOD(QueryInterface) (THIS_ REFIID riid, void **ppv);
|
|
STDMETHOD_(ULONG,AddRef) (THIS);
|
|
STDMETHOD_(ULONG,Release) (THIS);
|
|
|
|
// IPropertySetStorage
|
|
STDMETHODIMP Create(REFFMTID fmtid, const CLSID * pclsid, DWORD grfFlags, DWORD grfMode, IPropertyStorage** ppPropStg);
|
|
STDMETHODIMP Open(REFFMTID fmtid, DWORD grfMode, IPropertyStorage** ppPropStg);
|
|
STDMETHODIMP Delete(REFFMTID fmtid);
|
|
STDMETHODIMP Enum(IEnumSTATPROPSETSTG** ppenum);
|
|
|
|
CPropertySetStg(LPCTSTR pszFolder, DWORD grfMode);
|
|
|
|
LPCTSTR IniFile() { return _szIniFile; }
|
|
|
|
private:
|
|
~CPropertySetStg();
|
|
HRESULT _LoadPropHandler();
|
|
|
|
LONG _cRef;
|
|
DWORD _grfMode; // The mode that we opened the file in.
|
|
TCHAR _szIniFile[MAX_PATH]; // desktop.ini path
|
|
};
|
|
|
|
|
|
CPropertySetStg::CPropertySetStg(LPCTSTR pszFolder, DWORD grfMode) : _cRef(1), _grfMode(grfMode)
|
|
{
|
|
PathCombine(_szIniFile, pszFolder, TEXT("desktop.ini"));
|
|
}
|
|
|
|
CPropertySetStg::~CPropertySetStg()
|
|
{
|
|
}
|
|
|
|
STDMETHODIMP CPropertySetStg::QueryInterface(REFIID riid, void **ppv)
|
|
{
|
|
static const QITAB qit[] =
|
|
{
|
|
QITABENT(CPropertySetStg, IPropertySetStorage),
|
|
{ 0 },
|
|
};
|
|
return QISearch(this, qit, riid, ppv);
|
|
}
|
|
|
|
STDMETHODIMP_(ULONG) CPropertySetStg::AddRef()
|
|
{
|
|
return InterlockedIncrement(&_cRef);
|
|
}
|
|
|
|
STDMETHODIMP_(ULONG) CPropertySetStg::Release()
|
|
{
|
|
ASSERT( 0 != _cRef );
|
|
ULONG cRef = InterlockedDecrement(&_cRef);
|
|
if ( 0 == cRef )
|
|
{
|
|
delete this;
|
|
}
|
|
return cRef;
|
|
}
|
|
|
|
HRESULT CPropertySetStg::_LoadPropHandler()
|
|
{
|
|
return E_NOTIMPL;
|
|
}
|
|
|
|
STDMETHODIMP CPropertySetStg::Create(REFFMTID fmtid, const CLSID *pclsid, DWORD grfFlags,
|
|
DWORD grfMode, IPropertyStorage **pppropstg)
|
|
{
|
|
*pppropstg = NULL;
|
|
return E_NOTIMPL;
|
|
}
|
|
|
|
STDMETHODIMP CPropertySetStg::Open(REFFMTID fmtid, DWORD grfMode, IPropertyStorage **pppropstg)
|
|
{
|
|
*pppropstg = new CPropertyStg(this, fmtid, grfMode);
|
|
return *pppropstg ? S_OK : E_OUTOFMEMORY;
|
|
}
|
|
|
|
STDMETHODIMP CPropertySetStg::Delete(REFFMTID fmtid)
|
|
{
|
|
return STG_E_ACCESSDENIED;
|
|
}
|
|
|
|
STDMETHODIMP CPropertySetStg::Enum(IEnumSTATPROPSETSTG **ppenum)
|
|
{
|
|
*ppenum = NULL;
|
|
return E_NOTIMPL;
|
|
}
|
|
|
|
|
|
CPropertyStg::CPropertyStg(CPropertySetStg *ppss, REFFMTID fmtid, DWORD grfMode) : _ppss(ppss), _fmtid(fmtid), _grfMode(grfMode)
|
|
{
|
|
_ppss->AddRef();
|
|
}
|
|
|
|
CPropertyStg::~CPropertyStg()
|
|
{
|
|
_ppss->Release();
|
|
}
|
|
|
|
STDMETHODIMP CPropertyStg::QueryInterface(REFIID riid, void **ppv)
|
|
{
|
|
static const QITAB qit[] =
|
|
{
|
|
QITABENT(CPropertyStg, IPropertyStorage),
|
|
{ 0 },
|
|
};
|
|
return QISearch(this, qit, riid, ppv);
|
|
}
|
|
|
|
STDMETHODIMP_(ULONG) CPropertyStg::AddRef()
|
|
{
|
|
return InterlockedIncrement(&_cRef);
|
|
}
|
|
|
|
STDMETHODIMP_(ULONG) CPropertyStg::Release()
|
|
{
|
|
ASSERT( 0 != _cRef );
|
|
ULONG cRef = InterlockedDecrement(&_cRef);
|
|
if ( 0 == cRef )
|
|
{
|
|
delete this;
|
|
}
|
|
return cRef;
|
|
}
|
|
|
|
BOOL CPropertyStg::_SectionValueName(const PROPSPEC *ppspec,
|
|
LPTSTR pszSection, UINT cchSection,
|
|
LPTSTR pszValueName, UINT cchValueName)
|
|
{
|
|
*pszSection = *pszValueName = 0;
|
|
|
|
if (_fmtid == FMTID_SummaryInformation)
|
|
{
|
|
if (PIDSI_COMMENTS == ppspec->propid)
|
|
{
|
|
lstrcpyn(pszSection, TEXT(".ShellClassInfo"), cchSection);
|
|
lstrcpyn(pszValueName, TEXT("InfoTip"), cchValueName);
|
|
}
|
|
}
|
|
|
|
if (!*pszSection || !*pszValueName)
|
|
{
|
|
if (PID_CODEPAGE < ppspec->propid)
|
|
{
|
|
SHStringFromGUID(_fmtid, pszSection, cchSection);
|
|
if (PRSPEC_LPWSTR == ppspec->ulKind)
|
|
{
|
|
SHUnicodeToTChar(ppspec->lpwstr, pszValueName, cchValueName);
|
|
}
|
|
else if (PRSPEC_PROPID == ppspec->ulKind)
|
|
{
|
|
wnsprintf(pszValueName, cchValueName, TEXT("Prop%d"), ppspec->propid);
|
|
}
|
|
}
|
|
}
|
|
return (*pszSection && *pszValueName) ? TRUE : FALSE;
|
|
}
|
|
|
|
HRESULT CPropertyStg::_ReadProp(const PROPSPEC *ppspec, PROPVARIANT *ppropvar)
|
|
{
|
|
PropVariantInit(ppropvar); // init out param to VT_EMPTY
|
|
HRESULT hr = S_FALSE;
|
|
|
|
TCHAR szSection[128], szPropName[128];
|
|
if (_SectionValueName(ppspec, szSection, ARRAYSIZE(szSection), szPropName, ARRAYSIZE(szPropName)))
|
|
{
|
|
TCHAR szValue[128];
|
|
UINT cch = GetPrivateProfileString(szSection, szPropName, TEXT(""), szValue, ARRAYSIZE(szValue), _ppss->IniFile());
|
|
if (cch)
|
|
{
|
|
hr = SHStrDup(szValue, &ppropvar->pwszVal);
|
|
if (SUCCEEDED(hr))
|
|
ppropvar->vt = VT_LPWSTR;
|
|
}
|
|
}
|
|
return hr;
|
|
}
|
|
|
|
STDMETHODIMP CPropertyStg::ReadMultiple(ULONG cpspec, const PROPSPEC rgpspec[], PROPVARIANT rgpropvar[])
|
|
{
|
|
HRESULT hr = S_FALSE;
|
|
UINT cRead = 0;
|
|
for (UINT i = 0; i < cpspec; i++)
|
|
{
|
|
hr = _ReadProp(&rgpspec[i], &rgpropvar[i]);
|
|
if (S_OK == hr)
|
|
{
|
|
cRead++;
|
|
}
|
|
|
|
if (FAILED(hr))
|
|
{
|
|
FreePropVariantArray(i, rgpropvar);
|
|
cRead = 0;
|
|
break;
|
|
}
|
|
}
|
|
|
|
if (cRead)
|
|
hr = S_OK; // at least one non VT_EMPTY property read
|
|
|
|
return hr;
|
|
}
|
|
|
|
STDMETHODIMP CPropertyStg::WriteMultiple(ULONG cpspec, const PROPSPEC rgpspec[], const PROPVARIANT rgpropvar[], PROPID propidNameFirst)
|
|
{
|
|
return E_NOTIMPL;
|
|
}
|
|
|
|
STDMETHODIMP CPropertyStg::DeleteMultiple(ULONG cpspec, const PROPSPEC rgpspec[])
|
|
{
|
|
return E_NOTIMPL;
|
|
}
|
|
|
|
STDMETHODIMP CPropertyStg::ReadPropertyNames(ULONG cpropid, const PROPID rgpropid[], LPOLESTR rglpwstrName[])
|
|
{
|
|
return E_NOTIMPL;
|
|
}
|
|
|
|
STDMETHODIMP CPropertyStg::WritePropertyNames(ULONG cpropid, const PROPID rgpropid[], const LPOLESTR rglpwstrName[])
|
|
{
|
|
return E_NOTIMPL;
|
|
}
|
|
|
|
STDMETHODIMP CPropertyStg::DeletePropertyNames(ULONG cpropid, const PROPID rgpropid[])
|
|
{
|
|
return E_NOTIMPL;
|
|
}
|
|
|
|
STDMETHODIMP CPropertyStg::Commit(DWORD grfCommitFlags)
|
|
{
|
|
return E_NOTIMPL;
|
|
}
|
|
|
|
STDMETHODIMP CPropertyStg::Revert(void)
|
|
{
|
|
return E_NOTIMPL;
|
|
}
|
|
|
|
STDMETHODIMP CPropertyStg::Enum(IEnumSTATPROPSTG **ppenum)
|
|
{
|
|
return E_NOTIMPL;
|
|
}
|
|
|
|
STDMETHODIMP CPropertyStg::SetTimes(const FILETIME *pctime, const FILETIME *patime,const FILETIME *pmtime)
|
|
{
|
|
return E_NOTIMPL;
|
|
}
|
|
|
|
STDMETHODIMP CPropertyStg::SetClass(REFCLSID clsid)
|
|
{
|
|
return E_NOTIMPL;
|
|
}
|
|
|
|
STDMETHODIMP CPropertyStg::Stat(STATPROPSETSTG *pstatpsstg)
|
|
{
|
|
return E_NOTIMPL;
|
|
}
|
|
|
|
STDAPI SHCreatePropStgOnFolder(LPCTSTR pszFolder, DWORD grfMode, IPropertySetStorage **ppss)
|
|
{
|
|
*ppss = (IPropertySetStorage *)new CPropertySetStg(pszFolder, grfMode);
|
|
return *ppss ? S_OK : E_OUTOFMEMORY;
|
|
}
|