//+-------------------------------------------------------------- // // Microsoft Windows // Copyright (C) Microsoft Corporation, 1991 - 1992. // // File: docfile.c // // Contents: DocFile root functions (Stg* functions) // // History: 10-Dec-91 DrewB Created // //--------------------------------------------------------------- #include #pragma hdrstop #include #include #include #include #include #include #include #include #ifdef COORD #include #endif #ifdef _MAC #include #endif HRESULT IsNffAppropriate(const LPCWSTR pwcsName); //+-------------------------------------------------------------- // // Function: DfFromLB, private // // Synopsis: Starts a root Docfile on an ILockBytes // // Arguments: [plst] - LStream to start on // [df] - Permissions // [dwStartFlags] - Startup flags // [snbExclude] - Partial instantiation list // [ppdfExp] - DocFile return // [pcid] - Class ID return for opens // // Returns: Appropriate status code // // Modifies: [ppdfExp] // [pcid] // // History: 19-Mar-92 DrewB Created // 18-May-93 AlexT Added pMalloc // // Algorithm: Create and initialize a root transaction level // Create and initialize a public docfile // Create and initialize an exposed docfile // //--------------------------------------------------------------- #ifdef COORD SCODE DfFromLB(CPerContext *ppc, ILockBytes *plst, DFLAGS df, DWORD dwStartFlags, SNBW snbExclude, ITransaction *pTransaction, CExposedDocFile **ppdfExp, CLSID *pcid) #else SCODE DfFromLB(CPerContext *ppc, ILockBytes *plst, DFLAGS df, DWORD dwStartFlags, SNBW snbExclude, CExposedDocFile **ppdfExp, CLSID *pcid) #endif //COORD { SCODE sc, scConv; CRootPubDocFile *prpdf; #ifdef COORD CPubDocFile *ppubdf; CPubDocFile *ppubReturn; CWrappedDocFile *pwdf; CDocfileResource *pdfr = NULL; #endif CDFBasis *pdfb; ULONG ulOpenLock; IMalloc *pMalloc = ppc->GetMalloc(); ppc->AddRef(); olDebugOut((DEB_ITRACE, "In DfFromLB(%p, %p, %X, %lX, %p, %p, %p)\n", pMalloc, plst, df, dwStartFlags, snbExclude, ppdfExp, pcid)); //Take the mutex in the CPerContext, in case there is an IFillLockBytes // trying to write data while we're trying to open. CSafeSem _ss(ppc); olChk(_ss.Take()); #ifdef CHECKCID ULONG cbRead; olChk(plst->ReadAt(CBCLSIDOFFSET, pcid, sizeof(CLSID), &cbRead)); if (cbRead != sizeof(CLSID)) olErr(EH_Err, STG_E_INVALIDHEADER); if (!REFCLSIDEQ(*pcid, DOCFILE_CLASSID)) olErr(EH_Err, STG_E_INVALIDHEADER); #endif #ifdef COORD if (pTransaction != NULL) { //If we've passed in an ITransaction pointer, it indicates that we // want to open or create this docfile as part of a coordinated // transaction. First, we need to find out if there's a docfile // resource manager for that transaction currently existing in // this process. //First, check if we're opening transacted. If we aren't, then we're // not going to allow this docfile to participate in the // transaction. if (!P_TRANSACTED(df)) { //Is this the right error? olErr(EH_Err, STG_E_INVALIDFUNCTION); } XACTTRANSINFO xti; olChk(pTransaction->GetTransactionInfo(&xti)); EnterCriticalSection(&g_csResourceList); CDocfileResource *pdfrTemp = g_dfrHead.GetNext(); while (pdfrTemp != NULL) { if (IsEqualBOID(pdfrTemp->GetUOW(), xti.uow)) { //Direct hit. pdfr = pdfrTemp; break; } pdfrTemp = pdfrTemp->GetNext(); } if (pdfr == NULL) { ITransactionCoordinator *ptc; //If there isn't, we need to create one. olChkTo(EH_cs, pTransaction->QueryInterface( IID_ITransactionCoordinator, (void **)&ptc)); pdfr = new CDocfileResource; if (pdfr == NULL) { ptc->Release(); olErr(EH_cs, STG_E_INSUFFICIENTMEMORY); } sc = pdfr->Enlist(ptc); ptc->Release(); if (FAILED(sc)) { pdfr->Release();; olErr(EH_cs, sc); } //Add to list. pdfr->SetNext(g_dfrHead.GetNext()); if (g_dfrHead.GetNext() != NULL) g_dfrHead.GetNext()->SetPrev(pdfr); g_dfrHead.SetNext(pdfr); pdfr->SetPrev(&g_dfrHead); } else { //We'll release this reference below. pdfr->AddRef(); } LeaveCriticalSection(&g_csResourceList); } #endif // Make root olMem(prpdf = new (pMalloc) CRootPubDocFile(pMalloc)); olChkTo(EH_prpdf, scConv = prpdf->InitRoot(plst, dwStartFlags, df, snbExclude, &pdfb, &ulOpenLock, ppc->GetGlobal())); #ifdef COORD if (pTransaction != NULL) { //Set up a fake transaction level at the root. A pointer to // this will be held by the resource manager. The storage pointer // that is passed back to the caller will be a pointer to the // transaction level (non-root) below it. This will allow the // client to write and commit as many times as desired without // the changes ever actually hitting the file. CDfName dfnNull; // auto-initialized to 0 WCHAR wcZero = 0; dfnNull.Set(2, (BYTE *)&wcZero); olMemTo(EH_prpdfInit, pwdf = new (pMalloc) CWrappedDocFile( &dfnNull, ROOT_LUID, (df & ~DF_INDEPENDENT), pdfb, NULL)); olChkTo(EH_pwdf, pwdf->Init(prpdf->GetDF())); prpdf->GetDF()->AddRef(); olMemTo(EH_pwdfInit, ppubdf = new (pMalloc) CPubDocFile( prpdf, pwdf, (df | DF_COORD) & ~DF_INDEPENDENT, ROOT_LUID, pdfb, &dfnNull, 2, pdfb->GetBaseMultiStream())); olChkTo(EH_ppubdf, pwdf->InitPub(ppubdf)); ppubdf->AddXSMember(NULL, pwdf, ROOT_LUID); ppubReturn = ppubdf; } else { ppubReturn = prpdf; } #endif //COORD ppc->SetILBInfo(pdfb->GetBase(), pdfb->GetDirty(), pdfb->GetOriginal(), ulOpenLock); ppc->SetLockInfo(ulOpenLock != 0, df); // Make exposed #ifdef COORD //We don't need to AddRef ppc since it starts with a refcount of 1. olMemTo(EH_ppcInit, *ppdfExp = new (pMalloc) CExposedDocFile( ppubReturn, pdfb, ppc)); if (pTransaction != NULL) { CExposedDocFile *pexpdf; olMemTo(EH_ppcInit, pexpdf = new (pMalloc) CExposedDocFile( prpdf, pdfb, ppc)); ppc->AddRef(); sc = pdfr->Join(pexpdf); if (FAILED(sc)) { pexpdf->Release(); olErr(EH_ppcInit, sc); } pdfr->Release(); } #else olMemTo(EH_ppcInit, *ppdfExp = new (pMalloc) CExposedDocFile( prpdf, pdfb, ppc)); #endif //COORD olDebugOut((DEB_ITRACE, "Out DfFromLB => %p\n", *ppdfExp)); return scConv; EH_ppcInit: // The context will release this but we want to keep it around // so take a reference pdfb->GetOriginal()->AddRef(); pdfb->GetBase()->AddRef(); pdfb->GetDirty()->AddRef(); if (ulOpenLock > 0 && ppc->GetGlobal() == NULL) { // The global context doesn't exist, so we need to release // the open lock explicitly. ReleaseOpen(pdfb->GetOriginal(), df, ulOpenLock); } // The open lock has now been released (either explicitly or by ppc) ulOpenLock = 0; #ifdef COORD EH_ppubdf: if (pTransaction != NULL) { ppubdf->vRelease(); } EH_pwdfInit: if (pTransaction != NULL) { prpdf->GetDF()->Release(); } EH_pwdf: if (pTransaction != NULL) { pwdf->Release(); } EH_prpdfInit: #endif //COORD pdfb->GetDirty()->Release(); pdfb->GetBase()->Release(); if (ulOpenLock > 0) ReleaseOpen(pdfb->GetOriginal(), df, ulOpenLock); pdfb->SetDirty(NULL); pdfb->SetBase(NULL); EH_prpdf: prpdf->ReleaseLocks(plst); prpdf->vRelease(); #ifdef COORD if ((pTransaction != NULL) && (pdfr != NULL)) { pdfr->Release(); } goto EH_Err; EH_cs: LeaveCriticalSection(&g_csResourceList); #endif EH_Err: ppc->Release(); return sc; } //+-------------------------------------------------------------- // // Function: DfFromName, private // // Synopsis: Starts a root DocFile from a base name // // Arguments: [pwcsName] - Name // [df] - Permissions // [dwStartFlags] - Startup flags // [snbExclude] - Partial instantiation list // [ppdfExp] - Docfile return // [pcid] - Class ID return for opens // // Returns: Appropriate status code // // Modifies: [ppdfExp] // [pcid] // // History: 19-Mar-92 DrewB Created // 18-May-93 AlexT Add per file allocator // // Notes: [pwcsName] is treated as unsafe memory // //--------------------------------------------------------------- // This set of root startup flags is handled by the multistream // and doesn't need to be set for filestreams #define RSF_MSF (RSF_CONVERT | RSF_TRUNCATE | RSF_ENCRYPTED) #ifdef COORD SCODE DfFromName(WCHAR const *pwcsName, DFLAGS df, DWORD dwStartFlags, SNBW snbExclude, ITransaction *pTransaction, CExposedDocFile **ppdfExp, ULONG *pulSectorSize, CLSID *pcid) #else SCODE DfFromName(WCHAR const *pwcsName, DFLAGS df, DWORD dwStartFlags, SNBW snbExclude, CExposedDocFile **ppdfExp, ULONG *pulSectorSize, CLSID *pcid) #endif { IMalloc *pMalloc; CFileStream *plst; CPerContext *ppc; CFileStream *plst2 = NULL; SCODE sc; CMSFHeader *phdr = NULL; BOOL fCreated; olDebugOut((DEB_ITRACE, "In DfFromName(%ws, %lX, %lX, %p, %p, %p)\n", pwcsName, df, dwStartFlags, snbExclude, ppdfExp, pcid)); olHChk(DfCreateSharedAllocator(&pMalloc, df & DF_LARGE)); // Start an ILockBytes from the named file olMemTo(EH_Malloc, plst = new (pMalloc) CFileStream(pMalloc)); olChkTo(EH_plst, plst->InitGlobal(dwStartFlags & ~RSF_MSF, df)); sc = plst->InitFile(pwcsName); fCreated = SUCCEEDED(sc) && ((dwStartFlags & RSF_CREATE) || pwcsName == NULL); if (sc == STG_E_FILEALREADYEXISTS && (dwStartFlags & RSF_MSF)) { plst->SetStartFlags(dwStartFlags & ~(RSF_MSF | RSF_CREATE)); sc = plst->InitFile(pwcsName); } olChkTo(EH_plst, sc); if (!(dwStartFlags & RSF_CREATE)) { ULONG cbDiskSector = (dwStartFlags & RSF_NO_BUFFERING) ? plst->GetSectorSize() : HEADERSIZE; olMemTo (EH_plstInit, phdr = (CMSFHeader*) TaskMemAlloc (cbDiskSector)); ULARGE_INTEGER ulOffset = {0,0}; ULONG ulRead; olChkTo (EH_plstInit, plst->ReadAt(ulOffset,phdr,cbDiskSector,&ulRead)); if (ulRead < sizeof(CMSFHeaderData)) olErr (EH_plstInit, STG_E_FILEALREADYEXISTS); sc = phdr->Validate(); if (sc == STG_E_INVALIDHEADER) sc = STG_E_FILEALREADYEXISTS; olChkTo (EH_plstInit, sc); if (phdr->GetSectorShift() > SECTORSHIFT512) { IMalloc *pMalloc2 = NULL; CGlobalFileStream *pgfst = plst->GetGlobal(); #ifdef MULTIHEAP CSharedMemoryBlock *psmb; BYTE *pbBase; ULONG ulHeapName; #endif df |= DF_LARGE; // reallocate objects from task memory dwStartFlags |= (phdr->GetSectorShift() <<12) & RSF_SECTORSIZE_MASK; // create and initialize the task allocator #ifdef MULTIHEAP g_smAllocator.GetState (&psmb, &pbBase, &ulHeapName); #endif olChkTo(EH_taskmem, DfCreateSharedAllocator(&pMalloc2, TRUE)); pMalloc->Release(); pMalloc = pMalloc2; olMemTo(EH_taskmem, plst2 = new (pMalloc2) CFileStream(pMalloc2)); olChkTo(EH_taskmem, plst2->InitGlobal(dwStartFlags & ~RSF_MSF, df)); plst2->InitFromFileStream (plst); plst2->GetGlobal()->InitFromGlobalFileStream (pgfst); #ifdef MULTIHEAP g_smAllocator.SetState (psmb, pbBase, ulHeapName, NULL, NULL); g_smAllocator.Uninit(); // unmap newly created heap g_smAllocator.SetState (NULL, NULL, 0, NULL, NULL); #endif plst = plst2; // CFileStream was destroyed by Uninit } TaskMemFree ((BYTE*)phdr); phdr = NULL; } //Create the per context olMemTo(EH_plstInit, ppc = new (pMalloc) CPerContext(pMalloc)); olChkTo(EH_ppc, ppc->InitNewContext()); { #ifdef MULTIHEAP CSafeMultiHeap smh(ppc); #endif // Start up the docfile #ifdef COORD sc = DfFromLB(ppc, plst, df, dwStartFlags, snbExclude, pTransaction, ppdfExp, pcid); #else sc = DfFromLB(ppc, plst, df, dwStartFlags, snbExclude, ppdfExp, pcid); #endif //COORD //Either DfFromLB has AddRef'ed the per context or it has failed. //Either way we want to release our reference here. LONG lRet; lRet = ppc->Release(); #ifdef MULTIHEAP olAssert((FAILED(sc)) || (lRet != 0)); #endif if (FAILED(sc)) { if (fCreated || ((dwStartFlags & RSF_CREATE) && !P_TRANSACTED(df))) plst->Delete(); plst->Release(); #ifdef MULTIHEAP if (lRet == 0) { g_smAllocator.Uninit(); } #endif } else if (pulSectorSize != NULL) { *pulSectorSize = (*ppdfExp)->GetPub()->GetBaseMS()->GetSectorSize(); } } pMalloc->Release(); olDebugOut((DEB_ITRACE, "Out DfFromName => %p\n", *ppdfExp)); return sc; EH_ppc: delete ppc; EH_taskmem: if (plst2) plst2->Release(); EH_plstInit: if (fCreated || ((dwStartFlags & RSF_CREATE) && !P_TRANSACTED(df))) plst->Delete(); EH_plst: plst->Release(); EH_Malloc: #ifdef MULTIHEAP g_smAllocator.Uninit(); // unmap newly created heap #endif pMalloc->Release(); EH_Err: if (phdr != NULL) TaskMemFree ((BYTE*)phdr); return sc; } //+-------------------------------------------------------------- // // Function: DfCreateDocfile, public // // Synopsis: Creates a root Docfile on a file // // Arguments: [pwcsName] - Filename // [grfMode] - Permissions // [reserved] - security attributes // [grfAttrs] - Win32 CreateFile attributes // [ppstgOpen] - Docfile return // // Returns: Appropriate status code // // Modifies: [ppstgOpen] // // History: 14-Jan-92 DrewB Created // //--------------------------------------------------------------- STDAPI DfCreateDocfile (WCHAR const *pwcsName, #ifdef COORD ITransaction *pTransaction, #else void *pTransaction, #endif DWORD grfMode, #if WIN32 == 300 LPSECURITY_ATTRIBUTES reserved, #else LPSTGSECURITY reserved, #endif ULONG ulSectorSize, DWORD grfAttrs, IStorage **ppstgOpen) { SafeCExposedDocFile pdfExp; SCODE sc; DFLAGS df; DWORD dwSectorFlag = 0; olLog(("--------::In StgCreateDocFile(%ws, %lX, %lu, %p)\n", pwcsName, grfMode, reserved, ppstgOpen)); olDebugOut((DEB_TRACE, "In StgCreateDocfile(%ws, %lX, %lu, %p)\n", pwcsName, grfMode, reserved, ppstgOpen)); olAssert(sizeof(LPSTGSECURITY) == sizeof(DWORD)); olChkTo(EH_BadPtr, ValidatePtrBuffer(ppstgOpen)); *ppstgOpen = NULL; if (pwcsName) olChk(ValidateNameW(pwcsName, _MAX_PATH)); if (reserved != 0) olErr(EH_Err, STG_E_INVALIDPARAMETER); if (grfMode & STGM_SIMPLE) { if (pTransaction != NULL) { olErr(EH_Err, STG_E_INVALIDFLAG); } if (ulSectorSize > 512) olErr (EH_Err, STG_E_INVALIDPARAMETER); sc = DfCreateSimpDocfile(pwcsName, grfMode, 0, ppstgOpen); goto EH_Err; } olChk(VerifyPerms(grfMode, TRUE)); if ((grfMode & STGM_RDWR) == STGM_READ || (grfMode & (STGM_DELETEONRELEASE | STGM_CONVERT)) == (STGM_DELETEONRELEASE | STGM_CONVERT)) olErr(EH_Err, STG_E_INVALIDFLAG); df = ModeToDFlags(grfMode); if ((grfMode & (STGM_TRANSACTED | STGM_CONVERT)) == (STGM_TRANSACTED | STGM_CONVERT)) df |= DF_INDEPENDENT; if (ulSectorSize > 512) { df |= DF_LARGE; switch (ulSectorSize) { case 4096 : dwSectorFlag = RSF_SECTORSIZE4K; break; case 8192 : dwSectorFlag = RSF_SECTORSIZE8K; break; case 16384 : dwSectorFlag = RSF_SECTORSIZE16K; break; case 32768 : dwSectorFlag = RSF_SECTORSIZE32K; break; default : olErr (EH_Err, STG_E_INVALIDPARAMETER); } } else if (ulSectorSize != 0 && ulSectorSize != 512) olErr (EH_Err, STG_E_INVALIDPARAMETER); #if WIN32 != 200 // // When we create over NFF files, delete them first. // except when we want to preserve encryption information // if( (STGM_CREATE & grfMode) && !(FILE_ATTRIBUTE_ENCRYPTED & grfAttrs)) { if( SUCCEEDED( IsNffAppropriate( pwcsName ) ) ) { if(FALSE == DeleteFileW( pwcsName ) ) { DWORD dwErr = GetLastError(); if( dwErr != ERROR_FILE_NOT_FOUND && dwErr != ERROR_PATH_NOT_FOUND ) { olErr( EH_Err, Win32ErrorToScode( dwErr ) ); } } } } #endif // _CHICAGO_ DfInitSharedMemBase(); #ifdef COORD olChk(sc = DfFromName(pwcsName, df, RSF_CREATE | ((grfMode & STGM_CREATE) ? RSF_TRUNCATE : 0) | ((grfMode & STGM_CONVERT) ? RSF_CONVERT : 0) | ((grfMode & STGM_DELETEONRELEASE) ? RSF_DELETEONRELEASE : 0) | (dwSectorFlag) | ((grfAttrs & FILE_FLAG_NO_BUFFERING) ? RSF_NO_BUFFERING : 0) | ((grfAttrs & FILE_ATTRIBUTE_ENCRYPTED) ? RSF_ENCRYPTED : 0), NULL, pTransaction, &pdfExp, NULL, NULL)); #else olChk(sc = DfFromName(pwcsName, df, RSF_CREATE | ((grfMode & STGM_CREATE) ? RSF_TRUNCATE : 0) | ((grfMode & STGM_CONVERT) ? RSF_CONVERT : 0) | ((grfMode & STGM_DELETEONRELEASE) ? RSF_DELETEONRELEASE : 0) | (dwSectorFlag) | ((grfAttrs & FILE_FLAG_NO_BUFFERING) ? RSF_NO_BUFFERING : 0) | ((grfAttrs & FILE_ATTRIBUTE_ENCRYPTED) ? RSF_ENCRYPTED : 0), NULL, &pdfExp, NULL, NULL)); #endif //COORD TRANSFER_INTERFACE(pdfExp, IStorage, ppstgOpen); EH_Err: olDebugOut((DEB_TRACE, "Out StgCreateDocfile => %p, ret == %lx\n", *ppstgOpen, sc)); olLog(("--------::Out StgCreateDocFile(). *ppstgOpen == %p, ret == %lx\n", *ppstgOpen, sc)); EH_BadPtr: FreeLogFile(); return _OLERETURN(sc); } //+-------------------------------------------------------------- // // Function: StgCreateDocfile, public // // Synopsis: Creates a root Docfile on a file // // Arguments: [pwcsName] - Filename // [grfMode] - Permissions // [reserved] - security attributes // [ppstgOpen] - Docfile return // // Returns: Appropriate status code // // Modifies: [ppstgOpen] // // History: 14-Jan-92 DrewB Created // //--------------------------------------------------------------- STDAPI StgCreateDocfile(WCHAR const *pwcsName, DWORD grfMode, LPSTGSECURITY reserved, IStorage **ppstgOpen) { return DfCreateDocfile(pwcsName, NULL, grfMode, reserved, 0, 0, ppstgOpen); } //+-------------------------------------------------------------- // // Function: StgCreateDocfileOnILockBytes, public // // Synopsis: Creates a root Docfile on an lstream // // Arguments: [plkbyt] - LStream // [grfMode] - Permissions // [reserved] - Unused // [ppstgOpen] - Docfile return // // Returns: Appropriate status code // // Modifies: [ppstgOpen] // // History: 14-Jan-92 DrewB Created // //--------------------------------------------------------------- STDAPI StgCreateDocfileOnILockBytes(ILockBytes *plkbyt, DWORD grfMode, DWORD reserved, IStorage **ppstgOpen) { IMalloc *pMalloc; CPerContext *ppc; SafeCExposedDocFile pdfExp; SCODE sc; DFLAGS df; #ifdef MULTIHEAP CPerContext pcSharedMemory (NULL); #endif olLog(("--------::In StgCreateDocFileOnILockBytes(%p, %lX, %lu, %p)\n", plkbyt, grfMode, reserved, ppstgOpen)); olDebugOut((DEB_TRACE, "In StgCreateDocfileOnILockBytes(" "%p, %lX, %lu, %p)\n", plkbyt, grfMode, reserved, ppstgOpen)); olChk(ValidatePtrBuffer(ppstgOpen)); *ppstgOpen = NULL; olChk(ValidateInterface(plkbyt, IID_ILockBytes)); if (reserved != 0) olErr(EH_Err, STG_E_INVALIDPARAMETER); if ((grfMode & (STGM_CREATE | STGM_CONVERT)) == 0) olErr(EH_Err, STG_E_FILEALREADYEXISTS); olChk(VerifyPerms(grfMode, TRUE)); if (grfMode & STGM_DELETEONRELEASE) olErr(EH_Err, STG_E_INVALIDFUNCTION); df = ModeToDFlags(grfMode); if ((grfMode & (STGM_TRANSACTED | STGM_CONVERT)) == (STGM_TRANSACTED | STGM_CONVERT)) df |= DF_INDEPENDENT; DfInitSharedMemBase(); olHChk(DfCreateSharedAllocator(&pMalloc, TRUE)); #ifdef MULTIHEAP // Because a custom ILockbytes can call back into storage code, // possibly using another shared heap, we need a temporary // owner until the real CPerContext is allocated // new stack frame for CSafeMultiHeap constructor/destructor { pcSharedMemory.GetThreadAllocatorState(); CSafeMultiHeap smh(&pcSharedMemory); #endif //Create the per context olMem(ppc = new (pMalloc) CPerContext(pMalloc)); olChkTo(EH_ppc, ppc->InitNewContext()); #ifdef COORD sc = DfFromLB(ppc, plkbyt, df, RSF_CREATE | ((grfMode & STGM_CREATE) ? RSF_TRUNCATE : 0) | ((grfMode & STGM_CONVERT) ? RSF_CONVERT : 0), NULL, NULL, &pdfExp, NULL); #else sc = DfFromLB(ppc, plkbyt, df, RSF_CREATE | ((grfMode & STGM_CREATE) ? RSF_TRUNCATE : 0) | ((grfMode & STGM_CONVERT) ? RSF_CONVERT : 0), NULL, &pdfExp, NULL); #endif //COORD pMalloc->Release(); //Either DfFromLB has AddRef'ed the per context or it has failed. //Either way we want to release our reference here. ppc->Release(); olChkTo(EH_Truncate, sc); TRANSFER_INTERFACE(pdfExp, IStorage, ppstgOpen); // Success; since we hold on to the ILockBytes interface, // we must take a reference to it. plkbyt->AddRef(); olDebugOut((DEB_TRACE, "Out StgCreateDocfileOnILockBytes => %p\n", *ppstgOpen)); #ifdef MULTIHEAP } #endif EH_Err: olLog(("--------::Out StgCreateDocFileOnILockBytes(). " "*ppstgOpen == %p, ret == %lx\n", *ppstgOpen, sc)); FreeLogFile(); return ResultFromScode(sc); EH_ppc: delete ppc; goto EH_Err; EH_Truncate: if ((grfMode & STGM_CREATE) && (grfMode & STGM_TRANSACTED) == 0) { ULARGE_INTEGER ulSize; ULISet32(ulSize, 0); olHVerSucc(plkbyt->SetSize(ulSize)); } goto EH_Err; } //+-------------------------------------------------------------- // // Function: DfOpenDocfile, public // // Synopsis: Instantiates a root Docfile from a file, // converting if necessary // // Arguments: [pwcsName] - Name // [pstgPriority] - Priority mode reopen IStorage // [grfMode] - Permissions // [snbExclude] - Exclusions for priority reopen // [reserved] - security attributes // [grfAttrs] - Win32 CreateFile attributes // [ppstgOpen] - Docfile return // // Returns: Appropriate status code // // Modifies: [ppstgOpen] // // History: 14-Jan-92 DrewB Created // //--------------------------------------------------------------- STDAPI DfOpenDocfile(OLECHAR const *pwcsName, #ifdef COORD ITransaction *pTransaction, #else void *pTransaction, #endif IStorage *pstgPriority, DWORD grfMode, SNB snbExclude, LPSTGSECURITY reserved, ULONG *pulSectorSize, DWORD grfAttrs, IStorage **ppstgOpen) { SafeCExposedDocFile pdfExp; SCODE sc; WCHAR awcName[_MAX_PATH]; CLSID cid; olLog(("--------::In StgOpenStorage(%ws, %p, %lX, %p, %lu, %p)\n", pwcsName, pstgPriority, grfMode, snbExclude, reserved, ppstgOpen)); olDebugOut((DEB_TRACE, "In StgOpenStorage(" "%ws, %p, %lX, %p, %lu, %p)\n", pwcsName, pstgPriority, grfMode, snbExclude, reserved, ppstgOpen)); olAssert(sizeof(LPSTGSECURITY) == sizeof(DWORD)); olChk(ValidatePtrBuffer(ppstgOpen)); *ppstgOpen = NULL; if (pstgPriority == NULL) { olChk(ValidateNameW(pwcsName, _MAX_PATH)); StringCbCopyW(awcName, sizeof(awcName), pwcsName); } if (pstgPriority) { STATSTG stat; olChk(ValidateInterface(pstgPriority, IID_IStorage)); olHChk(pstgPriority->Stat(&stat, 0)); if (lstrlenW (stat.pwcsName) > _MAX_PATH) olErr (EH_Err, STG_E_INVALIDNAME); StringCbCopyW(awcName, sizeof(awcName), stat.pwcsName); TaskMemFree(stat.pwcsName); } #if WIN32 != 200 if (grfMode & STGM_SIMPLE) { sc = DfOpenSimpDocfile(pwcsName, grfMode, 0, ppstgOpen); goto EH_Err; } #endif olChk(VerifyPerms(grfMode, TRUE)); if (grfMode & (STGM_CREATE | STGM_CONVERT)) olErr(EH_Err, STG_E_INVALIDFLAG); if (snbExclude) { if ((grfMode & STGM_RDWR) != STGM_READWRITE) olErr(EH_Err, STG_E_ACCESSDENIED); olChk(ValidateSNB(snbExclude)); } if (reserved != 0) olErr(EH_Err, STG_E_INVALIDPARAMETER); if (grfMode & STGM_DELETEONRELEASE) olErr(EH_Err, STG_E_INVALIDFUNCTION); //Otherwise, try it as a docfile if (pstgPriority) olChk(pstgPriority->Release()); DfInitSharedMemBase(); #ifdef COORD olChk(DfFromName(awcName, ModeToDFlags(grfMode), RSF_OPEN | ((grfMode & STGM_DELETEONRELEASE) ? RSF_DELETEONRELEASE : 0) | ((grfAttrs & FILE_FLAG_NO_BUFFERING) ? RSF_NO_BUFFERING : 0), snbExclude, pTransaction, &pdfExp, pulSectorSize, &cid)); #else olChk(DfFromName(awcName, ModeToDFlags(grfMode), RSF_OPEN | ((grfMode & STGM_DELETEONRELEASE) ? RSF_DELETEONRELEASE : 0) | ((grfAttrs & FILE_FLAG_NO_BUFFERING) ? RSF_NO_BUFFERING : 0), snbExclude, &pdfExp, pulSectorSize, &cid)); #endif //COORD TRANSFER_INTERFACE(pdfExp, IStorage, ppstgOpen); olDebugOut((DEB_TRACE, "Out StgOpenStorage => %p\n", *ppstgOpen)); EH_Err: olLog(("--------::Out StgOpenStorage(). *ppstgOpen == %p, ret == %lx\n", *ppstgOpen, sc)); FreeLogFile(); return sc; } //+-------------------------------------------------------------- // // Function: StgOpenStorageOnILockBytes, public // // Synopsis: Instantiates a root Docfile from an LStream, // converting if necessary // // Arguments: [plkbyt] - Source LStream // [pstgPriority] - For priority reopens // [grfMode] - Permissions // [snbExclude] - For priority reopens // [reserved] // [ppstgOpen] - Docfile return // // Returns: Appropriate status code // // Modifies: [ppstgOpen] // // History: 14-Jan-92 DrewB Created // //--------------------------------------------------------------- STDAPI StgOpenStorageOnILockBytes(ILockBytes *plkbyt, IStorage *pstgPriority, DWORD grfMode, SNB snbExclude, DWORD reserved, IStorage **ppstgOpen) { IMalloc *pMalloc; CPerContext *ppc; SCODE sc; SafeCExposedDocFile pdfExp; #ifdef MULTIHEAP CPerContext pcSharedMemory(NULL); #endif CLSID cid; olLog(("--------::In StgOpenStorageOnILockBytes(" "%p, %p, %lX, %p, %lu, %p)\n", plkbyt, pstgPriority, grfMode, snbExclude, reserved, ppstgOpen)); olDebugOut((DEB_TRACE, "In StgOpenStorageOnILockBytes(" "%p, %p, %lX, %p, %lu, %p)\n", plkbyt, pstgPriority, grfMode, snbExclude, reserved, ppstgOpen)); olChk(ValidatePtrBuffer(ppstgOpen)); *ppstgOpen = NULL; olChk(ValidateInterface(plkbyt, IID_ILockBytes)); if (pstgPriority) olChk(ValidateInterface(pstgPriority, IID_IStorage)); olChk(VerifyPerms(grfMode, TRUE)); if (grfMode & (STGM_CREATE | STGM_CONVERT)) olErr(EH_Err, STG_E_INVALIDFLAG); if (grfMode & STGM_DELETEONRELEASE) olErr(EH_Err, STG_E_INVALIDFUNCTION); if (snbExclude) { if ((grfMode & STGM_RDWR) != STGM_READWRITE) olErr(EH_Err, STG_E_ACCESSDENIED); olChk(ValidateSNB(snbExclude)); } if (reserved != 0) olErr(EH_Err, STG_E_INVALIDPARAMETER); if (pstgPriority) olChk(pstgPriority->Release()); IFileLockBytes *pfl; if (SUCCEEDED(plkbyt->QueryInterface(IID_IFileLockBytes, (void **)&pfl)) && ((CFileStream *)plkbyt)->GetContextPointer() != NULL) { //Someone passed us the ILockBytes we gave them from //StgGetIFillLockBytesOnFile. It already contains a //context pointer, so reuse that rather than creating //a whole new shared heap. pfl->Release(); CFileStream *pfst = (CFileStream *)plkbyt; CPerContext *ppc2 = pfst->GetContextPointer(); #ifdef MULTIHEAP CSafeMultiHeap smh(ppc2); #endif #ifdef COORD olChk(DfFromLB(ppc2, pfst, ModeToDFlags(grfMode), pfst->GetStartFlags(), NULL, NULL, &pdfExp, NULL)); #else olChk(DfFromLB(ppc2, pfst, ModeToDFlags(grfMode), pfst->GetStartFlags(), NULL, &pdfExp, NULL)); #endif } else { DfInitSharedMemBase(); olHChk(DfCreateSharedAllocator(&pMalloc, TRUE)); #ifdef MULTIHEAP // Because a custom ILockbytes can call back into storage code, // possibly using another shared heap, we need a temporary // owner until the real CPerContext is allocated // new stack frame for CSafeMultiHeap constructor/destructor { pcSharedMemory.GetThreadAllocatorState(); CSafeMultiHeap smh(&pcSharedMemory); #endif //Create the per context olMem(ppc = new (pMalloc) CPerContext(pMalloc)); sc = ppc->InitNewContext(); if (FAILED(sc)) { delete ppc; olErr(EH_Err, sc); } #ifdef COORD sc = DfFromLB(ppc, plkbyt, ModeToDFlags(grfMode), RSF_OPEN, snbExclude, NULL, &pdfExp, &cid); #else sc = DfFromLB(ppc, plkbyt, ModeToDFlags(grfMode), RSF_OPEN, snbExclude, &pdfExp, &cid); #endif //COORD pMalloc->Release(); //Either DfFromLB has AddRef'ed the per context or it has failed. //Either way we want to release our reference here. ppc->Release(); olChk(sc); #ifdef MULTIHEAP } #endif } TRANSFER_INTERFACE(pdfExp, IStorage, ppstgOpen); // Success; since we hold on to the ILockBytes interface, // we must take a reference to it. plkbyt->AddRef(); olDebugOut((DEB_TRACE, "Out StgOpenStorageOnILockBytes => %p\n", *ppstgOpen)); EH_Err: olLog(("--------::Out StgOpenStorageOnILockBytes(). " "*ppstgOpen == %p, ret == %lx\n", *ppstgOpen, sc)); FreeLogFile(); return sc; } //+--------------------------------------------------------------------------- // // Function: DfGetClass, public // // Synopsis: Retrieves the class ID of the root entry of a docfile // // Arguments: [hFile] - Docfile file handle // [pclsid] - Class ID return // // Returns: Appropriate status code // // Modifies: [pclsid] // // History: 09-Feb-94 DrewB Created // //---------------------------------------------------------------------------- STDAPI DfGetClass(HANDLE hFile, CLSID *pclsid) { SCODE sc; DWORD dwCb; IMalloc *pMalloc; CFileStream *pfst; ULARGE_INTEGER uliOffset; ULONG ulOpenLock, ulAccessLock; BYTE bBuffer[sizeof(CMSFHeader)]; CMSFHeader *pmsh; CDirEntry *pde; olDebugOut((DEB_ITRACE, "In DfGetClass(%p, %p)\n", hFile, pclsid)); olAssert(sizeof(bBuffer) >= sizeof(CMSFHeader)); pmsh = (CMSFHeader *)bBuffer; olAssert(sizeof(bBuffer) >= sizeof(CDirEntry)); pde = (CDirEntry *)bBuffer; if (SetFilePointer(hFile, 0, NULL, FILE_BEGIN) != 0) { olErr(EH_Err, LAST_STG_SCODE); } if (!ReadFile(hFile, pmsh->GetData(), sizeof(CMSFHeaderData), &dwCb, NULL)) { olErr(EH_Err, LAST_STG_SCODE); } if (dwCb != sizeof(CMSFHeaderData)) { olErr(EH_Err, STG_E_INVALIDHEADER); } sc = pmsh->Validate(); olChk(sc); // Now we know it's a docfile DfInitSharedMemBase(); olHChk(DfCreateSharedAllocator(&pMalloc, TRUE)); olMemTo(EH_pMalloc, pfst = new (pMalloc) CFileStream(pMalloc)); olChkTo(EH_pfst, pfst->InitGlobal(0, 0)); olChkTo(EH_pfst, pfst->InitFromHandle(hFile)); // Take open and access locks to ensure that we're cooperating // with real opens olChkTo(EH_pfst, GetOpen(pfst, DF_READ, TRUE, &ulOpenLock)); olChkTo(EH_open, GetAccess(pfst, DF_READ, &ulAccessLock)); #ifdef LARGE_DOCFILE uliOffset.QuadPart = (ULONGLONG)(pmsh->GetDirStart() + 1) << pmsh->GetSectorShift(); #else uliOffset.HighPart = 0; uliOffset.LowPart = (pmsh->GetDirStart()+1) << pmsh->GetSectorShift(); #endif // The root directory entry is always the first directory entry // in the first directory sector // Ideally, we could read just the class ID directly into // pclsid. In practice, all the important things are declared // private or protected so it's easier to read the whole entry olChkTo(EH_access, GetScode(pfst->ReadAt(uliOffset, pde, sizeof(CDirEntry), &dwCb))); if (dwCb != sizeof(CDirEntry)) { sc = STG_E_READFAULT; } else { if (pde->GetFlags() != STGTY_ROOT) olErr (EH_access, STG_E_DOCFILECORRUPT); *pclsid = pde->GetClassId(); sc = S_OK; } olDebugOut((DEB_ITRACE, "Out DfGetClass\n")); EH_access: ReleaseAccess(pfst, DF_READ, ulAccessLock); EH_open: ReleaseOpen(pfst, DF_READ, ulOpenLock); EH_pfst: pfst->Release(); EH_pMalloc: pMalloc->Release(); EH_Err: return ResultFromScode(sc); }