//+------------------------------------------------------------------------- // // Microsoft Windows // Copyright (C) Microsoft Corporation, 1992 - 1993. // // File: psetstg.cxx // // Contents: Implementation of common class for OFS and DocFile // IPropertySetStorage // // Classes: CPropertySetStorage // CEnumSTATPROPSETSTG // // History: 1-Mar-95 BillMo Created. // 09-May-96 MikeHill Don't delete a CPropertyStorage object, // it deletes itself in the Release. // 22-May-96 MikeHill Set STATPROPSETSTG.dwOSVersion. // 06-Jun-96 MikeHill Added input validation. // 15-Jul-96 MikeHill - Handle STATSTG as OLECHAR (not WCHAR). // - Added CDocFilePropertySetStorage imp. // 07-Feb-97 Danl - Removed CDocFilePropertySetStorage. // - Moved _Create, _Open, & _Delete // into Create, Open & Delete // 10-Mar-98 MikeHill Pass grfMode into CPropertyStorage // create/open methods. // 06-May-98 MikeHill Use CoTaskMem rather than new/delete. // 5/18/98 MikeHill // - Cleaned up constructor by moving parameters to a // new Init method. // 6/11/98 MikeHill // - Dbg outs. // - Don't use out-parm as temp-parm in Create, Open. // // Notes: // //-------------------------------------------------------------------------- #include #include #ifdef _MAC_NODOC ASSERTDATA // File-specific data for FnAssert #endif // // debugging support // #if DBG CHAR * DbgFmtId(REFFMTID rfmtid, CHAR *pszBuf) { PropSprintfA(pszBuf, CB_DBGBUF, "rfmtid=%08X-%04X-%04X-%02X%02X%02X%02X%02X%02X%02X%02X", rfmtid.Data1, rfmtid.Data2, rfmtid.Data3, rfmtid.Data4[0], rfmtid.Data4[1], rfmtid.Data4[2], rfmtid.Data4[3], rfmtid.Data4[4], rfmtid.Data4[5], rfmtid.Data4[6], rfmtid.Data4[7]); return(pszBuf); } CHAR * DbgMode(DWORD grfMode, CHAR *psz) { *psz = 0; if (grfMode & STGM_TRANSACTED) StringCbCatA(psz, CB_DBGBUF, "STGM_TRANSACTED | "); else StringCbCatA(psz, CB_DBGBUF, "STGM_DIRECT | "); if (grfMode & STGM_SIMPLE) StringCbCatA(psz, CB_DBGBUF, "STGM_SIMPLE | "); switch (grfMode & 3) { case STGM_READ: StringCbCatA(psz, CB_DBGBUF, "STGM_READ |"); break; case STGM_WRITE: StringCbCatA(psz, CB_DBGBUF, "STGM_WRITE |"); break; case STGM_READWRITE: StringCbCatA(psz, CB_DBGBUF, "STGM_READWRITE |"); break; default: StringCbCatA(psz, CB_DBGBUF, "BAD grfMode |"); break; } switch (grfMode & 0x70) { case STGM_SHARE_DENY_NONE: StringCbCatA(psz, CB_DBGBUF, "STGM_SHARE_DENY_NONE |"); break; case STGM_SHARE_DENY_READ: StringCbCatA(psz, CB_DBGBUF, "STGM_SHARE_DENY_READ |"); break; case STGM_SHARE_DENY_WRITE: StringCbCatA(psz, CB_DBGBUF, "STGM_SHARE_DENY_WRITE |"); break; case STGM_SHARE_EXCLUSIVE: StringCbCatA(psz, CB_DBGBUF, "STGM_SHARE_EXCLUSIVE |"); break; default: StringCbCatA(psz, CB_DBGBUF, "BAD grfMode | "); break; } if (grfMode & STGM_PRIORITY) StringCbCatA(psz, CB_DBGBUF, "STGM_PRIORITY | "); if (grfMode & STGM_DELETEONRELEASE) StringCbCatA(psz, CB_DBGBUF, "STGM_DELETEONRELEASE | "); if (grfMode & STGM_NOSCRATCH) StringCbCatA(psz, CB_DBGBUF, "STGM_NOSCRATCH | "); if (grfMode & STGM_CREATE) StringCbCatA(psz, CB_DBGBUF, "STGM_CREATE | "); if (grfMode & STGM_CONVERT) StringCbCatA(psz, CB_DBGBUF, "STGM_CONVERT | "); if (grfMode & STGM_FAILIFTHERE) StringCbCatA(psz, CB_DBGBUF, "STGM_FAILIFTHERE | "); return(psz); } CHAR * DbgFlags(DWORD grfFlags, CHAR *psz) { StringCbCopyA(psz, CB_DBGBUF, "grfFlags="); if (grfFlags & PROPSETFLAG_NONSIMPLE) StringCbCatA(psz, CB_DBGBUF, "PROPSETFLAG_NONSIMPLE |"); else StringCbCatA(psz, CB_DBGBUF, "PROPSETFLAG_SIMPLE |"); if (grfFlags & PROPSETFLAG_ANSI) StringCbCatA(psz, CB_DBGBUF, "PROPSETFLAG_ANSI |"); else StringCbCatA(psz, CB_DBGBUF, "PROPSETFLAG_WIDECHAR |"); return(psz); } #endif //+---------------------------------------------------------------------------- // // Member: CPropertySetStorage::Init // // Synopsis: This method is used to provide the IStorage in which // property sets (IPropertyStorage's) will be created/opened. // // Arguments: [IStorage*] // The containing storage. // [IBlockingLock*] // The locking mechanism this CPropertySetStorage should // use. May be NULL. // [BOOL] (fControlLifetimes) // If true, we must addref the IStorage. E.g. in the docfile // implementation, CPropertySetStorage is a base class // for CExposedStorage, and this flag is set false. // //+---------------------------------------------------------------------------- void CPropertySetStorage::Init( IStorage *pstg, IBlockingLock *pBlockingLock, BOOL fControlLifetimes ) { DfpAssert( NULL == _pstg && NULL == _pBlockingLock ); _pstg = pstg; _pBlockingLock = pBlockingLock; if( fControlLifetimes ) { _fContainingStgIsRefed = TRUE; pstg->AddRef(); if( NULL != pBlockingLock ) pBlockingLock->AddRef(); } } //+------------------------------------------------------------------- // // Member: CPropertySetStorage::Create // // Synopsis: Create a property set for outermost client of // IPropertSetStorage. // // Arguments: Passed through to CPropertyStorage ctor. // // Returns: S_OK or failure code. // // Notes: Create a new CPropertyStorage object which will // implement IPropertyStorage. The _pprivstg parameter // passed into CPropertyStorage::CPropertyStorage is // used to create (via QI) a matching type of mapped // stream (for OFS or DocFile properties.) // //-------------------------------------------------------------------- HRESULT CPropertySetStorage::Create( REFFMTID rfmtid, const CLSID * pclsid, DWORD grfFlags, DWORD grfMode, IPropertyStorage ** ppprstg) { // ------ // Locals // ------ HRESULT hr; IStream *pstmPropSet = NULL; IStorage *pstgPropSet = NULL; CPropertyStorage *pcPropStg = NULL; BOOL fCreated = FALSE; INT nPass = 0; BOOL fLocked = FALSE; CPropSetName psn; DBGBUF(buf1); DBGBUF(buf2); DBGBUF(buf3); propXTrace( "CPropertySetStorage::Create" ); // ---------- // Validation // ---------- if (S_OK != (hr = Validate())) goto Exit; Lock(); fLocked = TRUE; GEN_VDATEREADPTRIN_LABEL(&rfmtid, FMTID, E_INVALIDARG, Exit, hr); GEN_VDATEPTRIN_LABEL(pclsid, CLSID, E_INVALIDARG, Exit, hr); GEN_VDATEPTROUT_LABEL( ppprstg, IPropertyStorage*, E_INVALIDARG, Exit, hr); propTraceParameters(( "%s, %s, %s", static_cast(CStringize(rfmtid)), static_cast(CStringize( SGrfFlags(grfFlags))), static_cast(CStringize(SGrfMode(grfMode))) )); // We don't support PROPSETFLAG_UNBUFFERED from the IPropertySetStorage // interface. This may only be used in the StgOpenPropStg // and StgCreatePropStg APIs. This was done because we can't support // the flag on IPropertySetStorage::Open, so it would be inconsistent // to support it on the Create method. if( grfFlags & PROPSETFLAG_UNBUFFERED ) { hr = STG_E_INVALIDFLAG; goto Exit; } hr = CheckFlagsOnCreateOrOpen(TRUE,grfMode); if (hr != S_OK) { goto Exit; } psn = CPropSetName(rfmtid); *ppprstg = NULL; // -------------------------------- // Create a child Stream or Storage // -------------------------------- // We'll make one or two passes (two if we the stream/storage // already exists and we have to delete it). while( nPass <= 1 ) { if( PROPSETFLAG_NONSIMPLE & grfFlags ) { // The Child should be a Storage hr = _pstg->CreateStorage( psn.GetPropSetName(), grfMode, 0L, 0L, &pstgPropSet ); if( SUCCEEDED(hr) ) { fCreated = TRUE; if( NULL != pclsid ) { // We should also set the CLSID of the Storage. hr = pstgPropSet->SetClass(*pclsid); if( FAILED(hr) && E_NOTIMPL != hr ) goto Exit; hr = S_OK; } } } // if( PROPSETFLAG_NONSIMPLE & grfFlags ) else { // The Child should be a Stream if( IsEqualGUID( rfmtid, FMTID_UserDefinedProperties )) { hr = CreateUserDefinedStream( _pstg, psn, grfMode, &fCreated, &pstmPropSet ); } else { hr = _pstg->CreateStream(psn.GetPropSetName(), grfMode & ~STGM_TRANSACTED, 0, 0, &pstmPropSet); if( hr == S_OK ) fCreated = TRUE; } } // If the create failed because the element already existed, // and if STGM_CREATE was set, then let's delete the existing // element and try again. if( hr == STG_E_FILEALREADYEXISTS && grfMode & STGM_CREATE && 0 == nPass ) { hr = _pstg->DestroyElement( psn.GetPropSetName() ); if( FAILED(hr) ) goto Exit; nPass++; } // If we failed for any other reason, then it's fatal. else if( FAILED(hr) ) goto Exit; // Otherwise (we succeeded), we can move on. else break; } // while( nPass <= 1 ) // --------------------------- // Create the Property Storage // --------------------------- // Create a CPropertyStorage pcPropStg = new CPropertyStorage( _MSOpts ); if( NULL == pcPropStg ) { hr = E_OUTOFMEMORY; goto Exit; } // Initialize the property set. if( PROPSETFLAG_NONSIMPLE & grfFlags ) // We need a non-simple IPropertyStorage hr = pcPropStg->Create( pstgPropSet, rfmtid, pclsid, grfFlags, grfMode); else // We need a simple IPropertyStorage hr = pcPropStg->Create( pstmPropSet, rfmtid, pclsid, grfFlags, grfMode ); if( FAILED(hr) ) goto Exit; // ---- // Exit // ---- *ppprstg = static_cast(pcPropStg); pcPropStg = NULL; hr = S_OK; Exit: // On failure ... if( FAILED(hr) ) { // If an entry was created, attempt to delete it. if( fCreated ) _pstg->DestroyElement( psn.GetPropSetName() ); } if( NULL != pcPropStg ) delete pcPropStg; if( NULL != pstmPropSet ) pstmPropSet->Release(); if( NULL != pstgPropSet ) pstgPropSet->Release(); if( fLocked ) Unlock(); if( STG_E_FILEALREADYEXISTS == hr ) propSuppressExitErrors(); return( hr ); } //+------------------------------------------------------------------- // // Member: CPropertySetStorage::Open // // Synopsis: Open a property set for outermost client. // // Arguments: passed through to CPropertyStorage ctor. // // Returns: S_OK or error. // //-------------------------------------------------------------------- HRESULT CPropertySetStorage::Open( REFFMTID rfmtid, DWORD grfMode, IPropertyStorage ** ppprstg) { HRESULT hr; IUnknown *punkPropSet = NULL; BOOL fSimple = TRUE; BOOL fLocked = FALSE; CPropertyStorage *pcPropStg = NULL; CPropSetName psn; DBGBUF(buf1); DBGBUF(buf2); propXTrace( "CPropertySetStorage::Open" ); // ---------- // Validation // ---------- // Validate 'this' if (S_OK != (hr = Validate())) goto Exit; Lock(); fLocked = TRUE; // Validate inputs GEN_VDATEREADPTRIN_LABEL(&rfmtid, FMTID, E_INVALIDARG, Exit, hr); GEN_VDATEPTROUT_LABEL( ppprstg, IPropertyStorage*, E_INVALIDARG, Exit, hr); propTraceParameters(( "%s, %s, %p", DbgFmtId(rfmtid, buf1), DbgMode(grfMode, buf2), ppprstg )); psn = CPropSetName(rfmtid); *ppprstg = NULL; // -------------------------------- // Open the child Stream or Storage // -------------------------------- hr = _pstg->OpenStream( psn.GetPropSetName(), 0L, grfMode & ~STGM_TRANSACTED, 0L, (IStream**) &punkPropSet ); if( STG_E_FILENOTFOUND == hr ) { fSimple = FALSE; hr = _pstg->OpenStorage( psn.GetPropSetName(), NULL, grfMode, NULL, 0L, (IStorage**) &punkPropSet ); } if( FAILED(hr) ) goto Exit; // ------------------------- // Open the Property Storage // ------------------------- // Create an empty CPropertyStorage object. pcPropStg = new CPropertyStorage( _MSOpts ); if( NULL == pcPropStg ) { hr = E_OUTOFMEMORY; goto Exit; } // Open the property set and load it into the CPropertyStorage // object. We pas the default grfFlags, letting the Open method // infer its correct value from the property set. if( !fSimple ) { hr = pcPropStg->Open(static_cast(punkPropSet), // Addref-ed rfmtid, PROPSETFLAG_DEFAULT, // Flags are inferred grfMode ); } else { hr = pcPropStg->Open( static_cast(punkPropSet), // Addref-ed rfmtid, PROPSETFLAG_DEFAULT, // Flags are inferred grfMode, FALSE ); // Don't delete this property set } if( FAILED(hr) ) goto Exit; // ---- // Exit // ---- *ppprstg = static_cast(pcPropStg); pcPropStg = NULL; hr = S_OK; Exit: if( NULL != pcPropStg ) delete pcPropStg; if( NULL != punkPropSet ) punkPropSet->Release(); if( fLocked ) Unlock(); if( STG_E_FILENOTFOUND == hr ) propSuppressExitErrors(); return(hr); } //+------------------------------------------------------------------- // // Member: CPropertySetStorage::Delete // // Synopsis: Delete the specified property set. // // Arguments: [rfmtid] -- format id of property set to delete. // // Returns: S_OK if successful, error otherwise. // // Notes: Get the matching name and try the element deletion. // //-------------------------------------------------------------------- HRESULT CPropertySetStorage::Delete( REFFMTID rfmtid) { HRESULT hr = S_OK; IStream *pstm = NULL; BOOL fLocked = FALSE; CPropSetName psn; DBGBUF(buf); propXTrace( "CPropertySetStorage::Delete" ); // ---------- // Validation // ---------- // Validate 'this' if (S_OK != (hr = Validate())) goto Exit; Lock(); fLocked = TRUE; // Validate the input GEN_VDATEREADPTRIN_LABEL(&rfmtid, FMTID, E_INVALIDARG, Exit, hr); propTraceParameters(( "%s", DbgFmtId(rfmtid, buf) )); psn = CPropSetName(rfmtid); // -------------------------- // Delete the PropertyStorage // -------------------------- // Check for the special-case if( IsEqualIID( rfmtid, FMTID_UserDefinedProperties )) { // This property set is actually the second section of the Document // Summary Information property set. We must delete this // section, but we can't delete the Stream because it // still contains the first section. CPropertyStorage* pprstg; // Open the Stream. hr = _pstg->OpenStream( psn.GetPropSetName(), NULL, STGM_READWRITE | STGM_SHARE_EXCLUSIVE, 0L, &pstm ); if( FAILED(hr) ) goto Exit; // Create a CPropertyStorage pprstg = new CPropertyStorage( _MSOpts ); if( NULL == pprstg ) { hr = E_OUTOFMEMORY; goto Exit; } // Use the CPropertyStorage to delete the section. hr = pprstg->Open( pstm, rfmtid, PROPSETFLAG_DEFAULT, STGM_READWRITE | STGM_SHARE_EXCLUSIVE, TRUE ); // Delete this section pprstg->Release(); // Deletes *pprstg pprstg = NULL; if( FAILED(hr) ) goto Exit; } // if( IsEqualIID( rfmtid, FMTID_DocSummaryInformation2 )) else { // This is not a special case, so we can just delete // the Stream. Note that if the rfmtid represents the first // section of the DocumentSummaryInformation set, we might be // deleting the second section here as well. That is a documented // side-effect. hr = _pstg->DestroyElement(psn.GetPropSetName()); if( FAILED(hr) ) goto Exit; } // if( IsEqualIID( rfmtid, FMTID_UserDefinedProperties )) ... else // ---- // Exit // ---- hr = S_OK; Exit: if( NULL != pstm ) pstm->Release(); if( fLocked ) Unlock(); if( STG_E_FILENOTFOUND == hr ) propSuppressExitErrors(); return(hr); } //+------------------------------------------------------------------- // // Member: CPropertySetStorage::Enum // // Synopsis: Create an enumerator over the property set // // Arguments: [ppenum] -- where to return the pointer to the // enumerator. // // Returns: S_OK if ok, error otherwise. // // Notes: [ppenum] is NULL on error. // //-------------------------------------------------------------------- HRESULT CPropertySetStorage::Enum( IEnumSTATPROPSETSTG ** ppenum) { HRESULT hr; BOOL fLocked = FALSE; propXTrace( "CPropertySetStorage::Enum" ); // ---------- // Validation // ---------- // Validate 'this' if (S_OK != (hr = Validate())) goto Exit; Lock(); fLocked = TRUE; // Validate the input GEN_VDATEPTROUT_LABEL( ppenum, IEnumSTATPROPSETSTG*, E_INVALIDARG, Exit, hr); *ppenum = NULL; propTraceParameters(( "%p", ppenum )); // -------------------- // Create the enuerator // -------------------- hr = STG_E_INSUFFICIENTMEMORY; *ppenum = new CEnumSTATPROPSETSTG(_pstg, &hr); if (FAILED(hr)) { delete (CEnumSTATPROPSETSTG*) *ppenum; *ppenum = NULL; } // ---- // Exit // ---- Exit: if( fLocked ) Unlock(); return(hr); } //+------------------------------------------------------------------- // // Member: CEnumSTATPROPSETSTG::CEnumSTATPROPSETSTG // // Synopsis: Constructor which is used to implement // IPropertySetStorage::Enum // // Arguments: [pstg] -- the storage of the container to enumerate. // [phr] -- place to return HRESULT, S_OK or error. // // Notes: We use an STATSTG enumerator over the actual storage // to get the information about the property sets. // //-------------------------------------------------------------------- CEnumSTATPROPSETSTG::CEnumSTATPROPSETSTG(IStorage *pstg, HRESULT *phr) { HRESULT & hr = *phr; _ulSig = ENUMSTATPROPSETSTG_SIG; _cRefs = 1; hr = pstg->EnumElements(FALSE, NULL, 0, &_penumSTATSTG); if (FAILED(hr)) _penumSTATSTG = NULL; _cstatTotalInArray = 0; _istatNextToRead = 0; } //+------------------------------------------------------------------- // // Member: CEnumSTATPROPSETSTG::CEnumSTATPROPSETSTG // // Synopsis: Copy constructor which is used to implement // IEnumSTATPROPSETSTG::Clone. // // Arguments: [Other] -- The CEnumSTATPROPSETSTG to clone. // [phr] -- place to return HRESULT, S_OK or error. // //-------------------------------------------------------------------- CEnumSTATPROPSETSTG::CEnumSTATPROPSETSTG( CEnumSTATPROPSETSTG &Other, HRESULT *phr) { HRESULT & hr = *phr; _ulSig = ENUMSTATPROPSETSTG_SIG; _cRefs = 1; _cstatTotalInArray = 0; _istatNextToRead = Other._istatNextToRead; hr = Other._penumSTATSTG->Clone(&_penumSTATSTG); if (hr == S_OK) { // Copy the data in the buffer memcpy(_statarray, Other._statarray, sizeof(_statarray)); _cstatTotalInArray = Other._cstatTotalInArray; // Copy the strings in the buffer for (ULONG i=0; i<_cstatTotalInArray; i++) { ULONG cbName = sizeof(OLECHAR)*( ocslen(Other._statarray[i].pwcsName) + 1 ); _statarray[i].pwcsName = reinterpret_cast ( CoTaskMemAlloc( cbName )); if (_statarray[i].pwcsName == NULL) { _cstatTotalInArray = i; hr = STG_E_INSUFFICIENTMEMORY; break; } else { StringCbCopy(_statarray[i].pwcsName, cbName, Other._statarray[i].pwcsName ); } } } // note: destructor will cleanup the the strings or enumerator left behind // in the error case } //+------------------------------------------------------------------- // // Member: CEnumSTATPROPSETSTG::~CEnumSTATPROPSETSTG // // Synopsis: Delete the enumerator. // // Notes: Just releases the contained IEnumSTATSTG // //-------------------------------------------------------------------- CEnumSTATPROPSETSTG::~CEnumSTATPROPSETSTG() { _ulSig = ENUMSTATPROPSETSTG_SIGDEL; if (_penumSTATSTG != NULL) _penumSTATSTG->Release(); CleanupStatArray(); } //+------------------------------------------------------------------- // // Member: CEnumSTATPROPSETSTG::QueryInterface, AddRef, Release // // Synopsis: IUnknown // // Arguments: The usual thing. // //-------------------------------------------------------------------- HRESULT CEnumSTATPROPSETSTG::QueryInterface( REFIID riid, void **ppvObject) { HRESULT hr; if (S_OK != (hr = Validate())) return(hr); *ppvObject = NULL; if (IsEqualIID(riid, IID_IEnumSTATPROPSETSTG)) { *ppvObject = (IEnumSTATPROPSETSTG *)this; CEnumSTATPROPSETSTG::AddRef(); } else if (IsEqualIID(riid, IID_IUnknown)) { *ppvObject = (IUnknown *)this; CEnumSTATPROPSETSTG::AddRef(); } else { hr = E_NOINTERFACE; } return(hr); } //+------------------------------------------------------------------- // // Member: CEnumSTATPROPSETSTG::AddRef // //-------------------------------------------------------------------- ULONG CEnumSTATPROPSETSTG::AddRef(void) { if (S_OK != Validate()) return(0); InterlockedIncrement(&_cRefs); return(_cRefs); } //+------------------------------------------------------------------- // // Member: CEnumSTATPROPSETSTG::Release // //-------------------------------------------------------------------- ULONG CEnumSTATPROPSETSTG::Release(void) { LONG lRet; if (S_OK != Validate()) return(0); lRet = InterlockedDecrement(&_cRefs); if (lRet == 0) { delete this; } else if (lRet <0) { lRet = 0; } return(lRet); } //+------------------------------------------------------------------- // // Member: CEnumSTATPROPSETSTG::Next // // Synopsis: Implement IEnumSTATPROPSETSTG for ofs and docfile. // // Arguments: [celt] -- Count of elements to attempt to retrieve. // [rgelt] -- Where to put the results. Must be valid for at least // celt * sizeof(STATPROPSETSTG) bytes in length. // [pceltFetched] -- Count of elements returned is put here if // the pointer is non-null. If celt > 1, pceltFetched must // be valid non-NULL. If pcelt is non-NULL, it must be valid. // if pcelt is NULL, celt must be 1. // // Returns: S_OK if ok, error otherwise. // // Notes: We use a stack buffer to get more stuff per call to // underlying storage IEnumSTATSTG::Next. We then copy // data from the STATSTG's to STATPROPSETSTG's. // // An outer loop enumerates into statarray and then an // inner loop copies each batch into the [rgelt] buffer. // //-------------------------------------------------------------------- HRESULT CEnumSTATPROPSETSTG::Next(ULONG celt, STATPROPSETSTG * rgelt, ULONG * pceltFetched) { HRESULT hr; ULONG celtCallerTotal; // ---------- // Validation // ---------- // Validate 'this' if (S_OK != (hr = Validate())) return(hr); // Validate inputs if (NULL == pceltFetched) { if (1 != celt) return(STG_E_INVALIDPARAMETER); } else { VDATEPTROUT( pceltFetched, ULONG ); *pceltFetched = 0; } if (0 == celt) return(hr); if( !IsValidPtrOut(rgelt, celt * sizeof(rgelt[0])) ) return( E_INVALIDARG ); // ----------------------- // Perform the enumeration // ----------------------- celtCallerTotal = 0; // // we do this loop until we have what the caller wanted or error, or // no more. // do { // // If our internal buffer is empty, we (re)load it // if (_istatNextToRead == _cstatTotalInArray) { if (_cstatTotalInArray != 0) CleanupStatArray(); hr = _penumSTATSTG->Next(sizeof(_statarray)/sizeof(_statarray[0]), _statarray, &_cstatTotalInArray); } // S_OK or S_FALSE indicate that we got something if (SUCCEEDED(hr)) { // // we loop reading out of this buffer until either we have // all that the caller asked for, or we have exhausted the // buffer. // for (; celtCallerTotal < celt && _istatNextToRead < _cstatTotalInArray ; _istatNextToRead++) { OLECHAR *pocsName = _statarray[_istatNextToRead].pwcsName; BOOL fDone = FALSE; DfpAssert(pocsName != NULL); if (pocsName[0] == OC_PROPSET0) { // SPEC: if no matching fmtid then return GUID_NULL // *** get fmtid *** // if (!NT_SUCCESS(PrPropertySetNameToGuid( ocslen(pocsName), pocsName, &rgelt->fmtid))) { ZeroMemory(&rgelt->fmtid, sizeof(rgelt->fmtid)); } // *** get clsid *** // // *** get grfFlags *** // // SPEC: don't support returning PROPSETFLAG_ANSI if (_statarray[_istatNextToRead].type == STGTY_STORAGE) { rgelt->clsid = _statarray[_istatNextToRead].clsid; rgelt->grfFlags = PROPSETFLAG_NONSIMPLE; } else { // SPEC: don't get the clsid for !PROPSET_NONSIMPLE ZeroMemory(&rgelt->clsid, sizeof(rgelt->clsid)); rgelt->grfFlags = 0; } // *** get mtime *** // rgelt->mtime = _statarray[_istatNextToRead].mtime; // *** get ctime *** // rgelt->ctime = _statarray[_istatNextToRead].ctime; // *** get atime *** // rgelt->atime = _statarray[_istatNextToRead].atime; // *** default the OS Version *** // rgelt->dwOSVersion = PROPSETHDR_OSVERSION_UNKNOWN; rgelt ++; celtCallerTotal ++; } } // for (; celtCallerTotal < celt && ... } // if (SUCCEEDED(hr)) } while (celtCallerTotal < celt && hr == S_OK); if (SUCCEEDED(hr)) { hr = celt == celtCallerTotal ? S_OK : S_FALSE; DfpAssert(hr == S_OK || celtCallerTotal < celt); if (pceltFetched != NULL) *pceltFetched = celtCallerTotal; } return(hr); } //+------------------------------------------------------------------- // // Member: CEnumSTATPROPSETSTG::Skip // // Synopsis: Skip the requested number of elements. // // Arguments: [celt] -- number to skip. // // Returns: S_OK if all skipped, S_FALSE if less than requested // number skipped, error otherwise. // // Notes: // //-------------------------------------------------------------------- HRESULT CEnumSTATPROPSETSTG::Skip(ULONG celt) { HRESULT hr; STATPROPSETSTG stat; ULONG celtCallerTotal = 0; if (S_OK != (hr = Validate())) return(hr); do { hr = Next(1, &stat, NULL); } while ( hr == S_OK && ++celtCallerTotal < celt ); if (SUCCEEDED(hr)) hr = celt == celtCallerTotal ? S_OK : S_FALSE; return(hr); } //+------------------------------------------------------------------- // // Member: CEnumSTATPROPSETSTG::CleanupStatArray // // Synopsis: Free any strings in the array. // //-------------------------------------------------------------------- VOID CEnumSTATPROPSETSTG::CleanupStatArray() { for (ULONG i=0; i<_cstatTotalInArray; i++) { CoTaskMemFree( _statarray[i].pwcsName ); _statarray[i].pwcsName = NULL; } _istatNextToRead = 0; _cstatTotalInArray = 0; } //+------------------------------------------------------------------- // // Member: CEnumSTATPROPSETSTG::Reset // // Synopsis: Reset the enumerator. // // Notes: Merely resetting the underlying enumerator should be // adequate, // //-------------------------------------------------------------------- HRESULT CEnumSTATPROPSETSTG::Reset() { HRESULT hr; if (S_OK != (hr = Validate())) return(hr); hr = _penumSTATSTG->Reset(); if (hr == S_OK) { CleanupStatArray(); } return(hr); } //+------------------------------------------------------------------- // // Member: CEnumSTATPROPSETSTG::Clone // // Synopsis: Copy the enumeration state of this enumerator. // // Arguments: [ppenum] -- where to put the pointer to the clone // // Returns: S_OK if ok, error otherwise. // // Notes: We end up just calling IEnumSTATSTG::Clone in the // CEnumSTATPROPSETSTG constructor. // //-------------------------------------------------------------------- HRESULT CEnumSTATPROPSETSTG::Clone(IEnumSTATPROPSETSTG ** ppenum) { HRESULT hr; // ---------- // Validation // ---------- // Validate 'this' if (S_OK != (hr = Validate())) return(hr); // Validate inputs VDATEPTROUT( ppenum, IEnumSTATPROPSETSTG* ); // -------------------- // Clone the enumerator // -------------------- hr = STG_E_INSUFFICIENTMEMORY; *ppenum = new CEnumSTATPROPSETSTG(*this, &hr); if (FAILED(hr)) { delete (CEnumSTATPROPSETSTG*) *ppenum; *ppenum = NULL; } return(hr); } //+------------------------------------------------------------------- // // Member: CPropertySetStorage:: // QueryInterface, AddRef, and Release // // Synopsis: IUnknown members. // //-------------------------------------------------------------------- HRESULT CPropertySetStorage::QueryInterface( REFIID riid, void **ppvObject) { HRESULT hr = S_OK; // ---------- // Validation // ---------- if (S_OK != (hr = Validate())) return(hr); VDATEREADPTRIN( &riid, IID ); VDATEPTROUT( ppvObject, void* ); // ----- // Query // ----- if( IID_IPropertySetStorage == riid || IID_IUnknown == riid ) { *ppvObject = this; this->AddRef(); } else { *ppvObject = NULL; hr = E_NOINTERFACE; } return( hr ); } ULONG CPropertySetStorage::AddRef(void) { LONG lRet; // We don't call Validate here, because AddRef/Release should // always work. lRet = InterlockedIncrement( &_cReferences ); return( lRet ); } ULONG CPropertySetStorage::Release(void) { LONG lRet; // We don't call Validate here, because AddRef/Release should // always work. lRet = InterlockedDecrement( &_cReferences ); if( 0 == lRet ) delete this; else if( 0 > lRet ) lRet = 0; return( lRet ); } //+---------------------------------------------------------------------------- // // Function: Lock & Unlock // // Synopsis: This methods take and release the CPropertySetStorage's // critical section. // // Inputs: None // // Returns: Nothing // //+---------------------------------------------------------------------------- VOID CPropertySetStorage::Lock() { #ifndef _MAC if (NULL == _pBlockingLock) { DfpAssert (_fInitCriticalSection); EnterCriticalSection( &_CriticalSection ); } else { _pBlockingLock->Lock(INFINITE); } #endif } VOID CPropertySetStorage::Unlock() { #ifndef _MAC if (NULL == _pBlockingLock) { DfpAssert (_fInitCriticalSection); LeaveCriticalSection( &_CriticalSection ); } else { _pBlockingLock->Unlock(); } #endif } //+------------------------------------------------------------------- // // Member: CPropertyStorage::CreateUserDefinedStream // // Synopsis: Open the "DocumentSummaryInformation" stream in order // to create the UserDefined section of that property set. // Create the stream only if it doesn't already exist. // // Arguments: [pstg] -- container storage // [psn] -- the property set name // [grfMode] -- mode of the property set // [fCreated] -- TRUE if Stream is created, FALSE if opened. // [ppStream] -- On return, the Stream for the property set // // Notes: This special case is necessary because this DocSumInfo // property set is the only one in which we support more // than one section. For this property set, if the caller // Creates the second section, we must not *Create* the Stream, // because that would lose the first Section. So, we must open it. // // This routine is only called when creating the second // Section. The first Section is created normally (note // that if the client creates the first section, the second // section is lost). // // Also note that it may not be possible to open the Stream, // since it may already be opened. This is significant // because it may not be obvious to the caller. I.e., // to a client of IPropertyStorage, the 2 sections are // distinct property sets, and you would think that you could // open them for simultaneous write. // //-------------------------------------------------------------------- HRESULT CPropertySetStorage::CreateUserDefinedStream( IStorage * pstg, CPropSetName & psn, DWORD grfMode, BOOL * pfCreated, IStream ** ppStream ) { HRESULT hr; DWORD grfOpenMode; // Calculate the STGM flags to use for the Open. Create & Convert // don't have meaning for the Open, and Transacted isn't supported // by IPropertyStorage on the simple stream. grfOpenMode = grfMode & ~(STGM_CREATE | STGM_CONVERT | STGM_TRANSACTED); *pfCreated = FALSE; // Try an Open hr = pstg->OpenStream( psn.GetPropSetName(), NULL, grfOpenMode, 0L, ppStream ); // If the file wasn't there, try a create. if(( hr == STG_E_FILENOTFOUND ) || ( hr == STG_E_INVALIDFUNCTION)) { hr = pstg->CreateStream(psn.GetPropSetName(), grfMode, 0, 0, ppStream); if( SUCCEEDED( hr )) { *pfCreated = TRUE; } } return( hr ); } // CPropertySetStorage::CreateUserDefinedStream