rsaenh/tests: Fix a test failure on W2K and below.
[wine/wine64.git] / dlls / ole32 / storage32.c
blob3c88a36f766f054ab551be9819e5eeb69b1ae7b4
1 /*
2 * Compound Storage (32 bit version)
3 * Storage implementation
5 * This file contains the compound file implementation
6 * of the storage interface.
8 * Copyright 1999 Francis Beaudet
9 * Copyright 1999 Sylvain St-Germain
10 * Copyright 1999 Thuy Nguyen
11 * Copyright 2005 Mike McCormack
13 * This library is free software; you can redistribute it and/or
14 * modify it under the terms of the GNU Lesser General Public
15 * License as published by the Free Software Foundation; either
16 * version 2.1 of the License, or (at your option) any later version.
18 * This library is distributed in the hope that it will be useful,
19 * but WITHOUT ANY WARRANTY; without even the implied warranty of
20 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
21 * Lesser General Public License for more details.
23 * You should have received a copy of the GNU Lesser General Public
24 * License along with this library; if not, write to the Free Software
25 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
27 * NOTES
28 * The compound file implementation of IStorage used for create
29 * and manage substorages and streams within a storage object
30 * residing in a compound file object.
33 #include <assert.h>
34 #include <stdarg.h>
35 #include <stdio.h>
36 #include <stdlib.h>
37 #include <string.h>
39 #define COBJMACROS
40 #define NONAMELESSUNION
41 #define NONAMELESSSTRUCT
43 #include "windef.h"
44 #include "winbase.h"
45 #include "winnls.h"
46 #include "winuser.h"
47 #include "wine/unicode.h"
48 #include "wine/debug.h"
50 #include "storage32.h"
51 #include "ole2.h" /* For Write/ReadClassStm */
53 #include "winreg.h"
54 #include "wine/wingdi16.h"
56 WINE_DEFAULT_DEBUG_CHANNEL(storage);
58 /* Used for OleConvertIStorageToOLESTREAM and OleConvertOLESTREAMToIStorage */
59 #define OLESTREAM_ID 0x501
60 #define OLESTREAM_MAX_STR_LEN 255
63 * These are signatures to detect the type of Document file.
65 static const BYTE STORAGE_magic[8] ={0xd0,0xcf,0x11,0xe0,0xa1,0xb1,0x1a,0xe1};
66 static const BYTE STORAGE_oldmagic[8] ={0xd0,0xcf,0x11,0xe0,0x0e,0x11,0xfc,0x0d};
68 static const char rootPropertyName[] = "Root Entry";
70 /****************************************************************************
71 * Storage32InternalImpl definitions.
73 * Definition of the implementation structure for the IStorage32 interface.
74 * This one implements the IStorage32 interface for storage that are
75 * inside another storage.
77 struct StorageInternalImpl
79 struct StorageBaseImpl base;
81 * There is no specific data for this class.
84 typedef struct StorageInternalImpl StorageInternalImpl;
86 /* Method definitions for the Storage32InternalImpl class. */
87 static StorageInternalImpl* StorageInternalImpl_Construct(StorageImpl* ancestorStorage,
88 DWORD openFlags, ULONG rootTropertyIndex);
89 static void StorageImpl_Destroy(StorageBaseImpl* iface);
90 static BOOL StorageImpl_ReadBigBlock(StorageImpl* This, ULONG blockIndex, void* buffer);
91 static BOOL StorageImpl_WriteBigBlock(StorageImpl* This, ULONG blockIndex, const void* buffer);
92 static void StorageImpl_SetNextBlockInChain(StorageImpl* This, ULONG blockIndex, ULONG nextBlock);
93 static HRESULT StorageImpl_LoadFileHeader(StorageImpl* This);
94 static void StorageImpl_SaveFileHeader(StorageImpl* This);
96 static void Storage32Impl_AddBlockDepot(StorageImpl* This, ULONG blockIndex);
97 static ULONG Storage32Impl_AddExtBlockDepot(StorageImpl* This);
98 static ULONG Storage32Impl_GetNextExtendedBlock(StorageImpl* This, ULONG blockIndex);
99 static ULONG Storage32Impl_GetExtDepotBlock(StorageImpl* This, ULONG depotIndex);
100 static void Storage32Impl_SetExtDepotBlock(StorageImpl* This, ULONG depotIndex, ULONG blockIndex);
102 static ULONG BlockChainStream_GetHeadOfChain(BlockChainStream* This);
103 static ULARGE_INTEGER BlockChainStream_GetSize(BlockChainStream* This);
104 static ULONG BlockChainStream_GetCount(BlockChainStream* This);
106 static ULARGE_INTEGER SmallBlockChainStream_GetSize(SmallBlockChainStream* This);
107 static BOOL StorageImpl_WriteDWordToBigBlock( StorageImpl* This,
108 ULONG blockIndex, ULONG offset, DWORD value);
109 static BOOL StorageImpl_ReadDWordFromBigBlock( StorageImpl* This,
110 ULONG blockIndex, ULONG offset, DWORD* value);
112 /* OLESTREAM memory structure to use for Get and Put Routines */
113 /* Used for OleConvertIStorageToOLESTREAM and OleConvertOLESTREAMToIStorage */
114 typedef struct
116 DWORD dwOleID;
117 DWORD dwTypeID;
118 DWORD dwOleTypeNameLength;
119 CHAR strOleTypeName[OLESTREAM_MAX_STR_LEN];
120 CHAR *pstrOleObjFileName;
121 DWORD dwOleObjFileNameLength;
122 DWORD dwMetaFileWidth;
123 DWORD dwMetaFileHeight;
124 CHAR strUnknown[8]; /* don't know what this 8 byte information in OLE stream is. */
125 DWORD dwDataLength;
126 BYTE *pData;
127 }OLECONVERT_OLESTREAM_DATA;
129 /* CompObj Stream structure */
130 /* Used for OleConvertIStorageToOLESTREAM and OleConvertOLESTREAMToIStorage */
131 typedef struct
133 BYTE byUnknown1[12];
134 CLSID clsid;
135 DWORD dwCLSIDNameLength;
136 CHAR strCLSIDName[OLESTREAM_MAX_STR_LEN];
137 DWORD dwOleTypeNameLength;
138 CHAR strOleTypeName[OLESTREAM_MAX_STR_LEN];
139 DWORD dwProgIDNameLength;
140 CHAR strProgIDName[OLESTREAM_MAX_STR_LEN];
141 BYTE byUnknown2[16];
142 }OLECONVERT_ISTORAGE_COMPOBJ;
145 /* Ole Presentation Stream structure */
146 /* Used for OleConvertIStorageToOLESTREAM and OleConvertOLESTREAMToIStorage */
147 typedef struct
149 BYTE byUnknown1[28];
150 DWORD dwExtentX;
151 DWORD dwExtentY;
152 DWORD dwSize;
153 BYTE *pData;
154 }OLECONVERT_ISTORAGE_OLEPRES;
158 /***********************************************************************
159 * Forward declaration of internal functions used by the method DestroyElement
161 static HRESULT deleteStorageProperty(
162 StorageImpl *parentStorage,
163 ULONG foundPropertyIndexToDelete,
164 StgProperty propertyToDelete);
166 static HRESULT deleteStreamProperty(
167 StorageImpl *parentStorage,
168 ULONG foundPropertyIndexToDelete,
169 StgProperty propertyToDelete);
171 static HRESULT findPlaceholder(
172 StorageImpl *storage,
173 ULONG propertyIndexToStore,
174 ULONG storagePropertyIndex,
175 INT typeOfRelation);
177 static HRESULT adjustPropertyChain(
178 StorageImpl *This,
179 StgProperty propertyToDelete,
180 StgProperty parentProperty,
181 ULONG parentPropertyId,
182 INT typeOfRelation);
184 /***********************************************************************
185 * Declaration of the functions used to manipulate StgProperty
188 static ULONG getFreeProperty(
189 StorageImpl *storage);
191 static void updatePropertyChain(
192 StorageImpl *storage,
193 ULONG newPropertyIndex,
194 StgProperty newProperty);
196 static LONG propertyNameCmp(
197 const OLECHAR *newProperty,
198 const OLECHAR *currentProperty);
201 /***********************************************************************
202 * Declaration of miscellaneous functions...
204 static HRESULT validateSTGM(DWORD stgmValue);
206 static DWORD GetShareModeFromSTGM(DWORD stgm);
207 static DWORD GetAccessModeFromSTGM(DWORD stgm);
208 static DWORD GetCreationModeFromSTGM(DWORD stgm);
210 extern const IPropertySetStorageVtbl IPropertySetStorage_Vtbl;
213 /****************************************************************************
214 * IEnumSTATSTGImpl definitions.
216 * Definition of the implementation structure for the IEnumSTATSTGImpl interface.
217 * This class allows iterating through the content of a storage and to find
218 * specific items inside it.
220 struct IEnumSTATSTGImpl
222 const IEnumSTATSTGVtbl *lpVtbl; /* Needs to be the first item in the struct
223 * since we want to cast this in an IEnumSTATSTG pointer */
225 LONG ref; /* Reference count */
226 StorageImpl* parentStorage; /* Reference to the parent storage */
227 ULONG firstPropertyNode; /* Index of the root of the storage to enumerate */
230 * The current implementation of the IEnumSTATSTGImpl class uses a stack
231 * to walk the property sets to get the content of a storage. This stack
232 * is implemented by the following 3 data members
234 ULONG stackSize;
235 ULONG stackMaxSize;
236 ULONG* stackToVisit;
238 #define ENUMSTATSGT_SIZE_INCREMENT 10
242 static IEnumSTATSTGImpl* IEnumSTATSTGImpl_Construct(StorageImpl* This, ULONG firstPropertyNode);
243 static void IEnumSTATSTGImpl_Destroy(IEnumSTATSTGImpl* This);
244 static void IEnumSTATSTGImpl_PushSearchNode(IEnumSTATSTGImpl* This, ULONG nodeToPush);
245 static ULONG IEnumSTATSTGImpl_PopSearchNode(IEnumSTATSTGImpl* This, BOOL remove);
246 static ULONG IEnumSTATSTGImpl_FindProperty(IEnumSTATSTGImpl* This, const OLECHAR* lpszPropName,
247 StgProperty* buffer);
248 static INT IEnumSTATSTGImpl_FindParentProperty(IEnumSTATSTGImpl *This, ULONG childProperty,
249 StgProperty *currentProperty, ULONG *propertyId);
251 /************************************************************************
252 ** Block Functions
255 static ULONG BLOCK_GetBigBlockOffset(ULONG index)
257 if (index == 0xffffffff)
258 index = 0;
259 else
260 index ++;
262 return index * BIG_BLOCK_SIZE;
265 /************************************************************************
266 ** Storage32BaseImpl implementation
268 static HRESULT StorageImpl_ReadAt(StorageImpl* This,
269 ULARGE_INTEGER offset,
270 void* buffer,
271 ULONG size,
272 ULONG* bytesRead)
274 return BIGBLOCKFILE_ReadAt(This->bigBlockFile,offset,buffer,size,bytesRead);
277 static HRESULT StorageImpl_WriteAt(StorageImpl* This,
278 ULARGE_INTEGER offset,
279 const void* buffer,
280 const ULONG size,
281 ULONG* bytesWritten)
283 return BIGBLOCKFILE_WriteAt(This->bigBlockFile,offset,buffer,size,bytesWritten);
286 /************************************************************************
287 * Storage32BaseImpl_QueryInterface (IUnknown)
289 * This method implements the common QueryInterface for all IStorage32
290 * implementations contained in this file.
292 * See Windows documentation for more details on IUnknown methods.
294 static HRESULT WINAPI StorageBaseImpl_QueryInterface(
295 IStorage* iface,
296 REFIID riid,
297 void** ppvObject)
299 StorageBaseImpl *This = (StorageBaseImpl *)iface;
301 * Perform a sanity check on the parameters.
303 if ( (This==0) || (ppvObject==0) )
304 return E_INVALIDARG;
307 * Initialize the return parameter.
309 *ppvObject = 0;
312 * Compare the riid with the interface IDs implemented by this object.
314 if (IsEqualGUID(&IID_IUnknown, riid) ||
315 IsEqualGUID(&IID_IStorage, riid))
317 *ppvObject = (IStorage*)This;
319 else if (IsEqualGUID(&IID_IPropertySetStorage, riid))
321 *ppvObject = (IStorage*)&This->pssVtbl;
325 * Check that we obtained an interface.
327 if ((*ppvObject)==0)
328 return E_NOINTERFACE;
331 * Query Interface always increases the reference count by one when it is
332 * successful
334 IStorage_AddRef(iface);
336 return S_OK;
339 /************************************************************************
340 * Storage32BaseImpl_AddRef (IUnknown)
342 * This method implements the common AddRef for all IStorage32
343 * implementations contained in this file.
345 * See Windows documentation for more details on IUnknown methods.
347 static ULONG WINAPI StorageBaseImpl_AddRef(
348 IStorage* iface)
350 StorageBaseImpl *This = (StorageBaseImpl *)iface;
351 ULONG ref = InterlockedIncrement(&This->ref);
353 TRACE("(%p) AddRef to %d\n", This, ref);
355 return ref;
358 /************************************************************************
359 * Storage32BaseImpl_Release (IUnknown)
361 * This method implements the common Release for all IStorage32
362 * implementations contained in this file.
364 * See Windows documentation for more details on IUnknown methods.
366 static ULONG WINAPI StorageBaseImpl_Release(
367 IStorage* iface)
369 StorageBaseImpl *This = (StorageBaseImpl *)iface;
371 * Decrease the reference count on this object.
373 ULONG ref = InterlockedDecrement(&This->ref);
375 TRACE("(%p) ReleaseRef to %d\n", This, ref);
378 * If the reference count goes down to 0, perform suicide.
380 if (ref == 0)
383 * Since we are using a system of base-classes, we want to call the
384 * destructor of the appropriate derived class. To do this, we are
385 * using virtual functions to implement the destructor.
387 This->v_destructor(This);
390 return ref;
393 /************************************************************************
394 * Storage32BaseImpl_OpenStream (IStorage)
396 * This method will open the specified stream object from the current storage.
398 * See Windows documentation for more details on IStorage methods.
400 static HRESULT WINAPI StorageBaseImpl_OpenStream(
401 IStorage* iface,
402 const OLECHAR* pwcsName, /* [string][in] */
403 void* reserved1, /* [unique][in] */
404 DWORD grfMode, /* [in] */
405 DWORD reserved2, /* [in] */
406 IStream** ppstm) /* [out] */
408 StorageBaseImpl *This = (StorageBaseImpl *)iface;
409 IEnumSTATSTGImpl* propertyEnumeration;
410 StgStreamImpl* newStream;
411 StgProperty currentProperty;
412 ULONG foundPropertyIndex;
413 HRESULT res = STG_E_UNKNOWN;
415 TRACE("(%p, %s, %p, %x, %d, %p)\n",
416 iface, debugstr_w(pwcsName), reserved1, grfMode, reserved2, ppstm);
419 * Perform a sanity check on the parameters.
421 if ( (pwcsName==NULL) || (ppstm==0) )
423 res = E_INVALIDARG;
424 goto end;
428 * Initialize the out parameter
430 *ppstm = NULL;
433 * Validate the STGM flags
435 if ( FAILED( validateSTGM(grfMode) ) ||
436 STGM_SHARE_MODE(grfMode) != STGM_SHARE_EXCLUSIVE)
438 res = STG_E_INVALIDFLAG;
439 goto end;
443 * As documented.
445 if ( (grfMode & STGM_DELETEONRELEASE) || (grfMode & STGM_TRANSACTED) )
447 res = STG_E_INVALIDFUNCTION;
448 goto end;
452 * Check that we're compatible with the parent's storage mode, but
453 * only if we are not in transacted mode
455 if(!(This->ancestorStorage->base.openFlags & STGM_TRANSACTED)) {
456 if ( STGM_ACCESS_MODE( grfMode ) > STGM_ACCESS_MODE( This->openFlags ) )
458 res = STG_E_ACCESSDENIED;
459 goto end;
464 * Create a property enumeration to search the properties
466 propertyEnumeration = IEnumSTATSTGImpl_Construct(
467 This->ancestorStorage,
468 This->rootPropertySetIndex);
471 * Search the enumeration for the property with the given name
473 foundPropertyIndex = IEnumSTATSTGImpl_FindProperty(
474 propertyEnumeration,
475 pwcsName,
476 &currentProperty);
479 * Delete the property enumeration since we don't need it anymore
481 IEnumSTATSTGImpl_Destroy(propertyEnumeration);
484 * If it was found, construct the stream object and return a pointer to it.
486 if ( (foundPropertyIndex!=PROPERTY_NULL) &&
487 (currentProperty.propertyType==PROPTYPE_STREAM) )
489 newStream = StgStreamImpl_Construct(This, grfMode, foundPropertyIndex);
491 if (newStream!=0)
493 newStream->grfMode = grfMode;
494 *ppstm = (IStream*)newStream;
497 * Since we are returning a pointer to the interface, we have to
498 * nail down the reference.
500 IStream_AddRef(*ppstm);
502 res = S_OK;
503 goto end;
506 res = E_OUTOFMEMORY;
507 goto end;
510 res = STG_E_FILENOTFOUND;
512 end:
513 if (res == S_OK)
514 TRACE("<-- IStream %p\n", *ppstm);
515 TRACE("<-- %08x\n", res);
516 return res;
519 /************************************************************************
520 * Storage32BaseImpl_OpenStorage (IStorage)
522 * This method will open a new storage object from the current storage.
524 * See Windows documentation for more details on IStorage methods.
526 static HRESULT WINAPI StorageBaseImpl_OpenStorage(
527 IStorage* iface,
528 const OLECHAR* pwcsName, /* [string][unique][in] */
529 IStorage* pstgPriority, /* [unique][in] */
530 DWORD grfMode, /* [in] */
531 SNB snbExclude, /* [unique][in] */
532 DWORD reserved, /* [in] */
533 IStorage** ppstg) /* [out] */
535 StorageBaseImpl *This = (StorageBaseImpl *)iface;
536 StorageInternalImpl* newStorage;
537 IEnumSTATSTGImpl* propertyEnumeration;
538 StgProperty currentProperty;
539 ULONG foundPropertyIndex;
540 HRESULT res = STG_E_UNKNOWN;
542 TRACE("(%p, %s, %p, %x, %p, %d, %p)\n",
543 iface, debugstr_w(pwcsName), pstgPriority,
544 grfMode, snbExclude, reserved, ppstg);
547 * Perform a sanity check on the parameters.
549 if ( (This==0) || (pwcsName==NULL) || (ppstg==0) )
551 res = E_INVALIDARG;
552 goto end;
555 /* as documented */
556 if (snbExclude != NULL)
558 res = STG_E_INVALIDPARAMETER;
559 goto end;
563 * Validate the STGM flags
565 if ( FAILED( validateSTGM(grfMode) ))
567 res = STG_E_INVALIDFLAG;
568 goto end;
572 * As documented.
574 if ( STGM_SHARE_MODE(grfMode) != STGM_SHARE_EXCLUSIVE ||
575 (grfMode & STGM_DELETEONRELEASE) ||
576 (grfMode & STGM_PRIORITY) )
578 res = STG_E_INVALIDFUNCTION;
579 goto end;
583 * Check that we're compatible with the parent's storage mode,
584 * but only if we are not transacted
586 if(!(This->ancestorStorage->base.openFlags & STGM_TRANSACTED)) {
587 if ( STGM_ACCESS_MODE( grfMode ) > STGM_ACCESS_MODE( This->openFlags ) )
589 res = STG_E_ACCESSDENIED;
590 goto end;
595 * Initialize the out parameter
597 *ppstg = NULL;
600 * Create a property enumeration to search the properties
602 propertyEnumeration = IEnumSTATSTGImpl_Construct(
603 This->ancestorStorage,
604 This->rootPropertySetIndex);
607 * Search the enumeration for the property with the given name
609 foundPropertyIndex = IEnumSTATSTGImpl_FindProperty(
610 propertyEnumeration,
611 pwcsName,
612 &currentProperty);
615 * Delete the property enumeration since we don't need it anymore
617 IEnumSTATSTGImpl_Destroy(propertyEnumeration);
620 * If it was found, construct the stream object and return a pointer to it.
622 if ( (foundPropertyIndex!=PROPERTY_NULL) &&
623 (currentProperty.propertyType==PROPTYPE_STORAGE) )
626 * Construct a new Storage object
628 newStorage = StorageInternalImpl_Construct(
629 This->ancestorStorage,
630 grfMode,
631 foundPropertyIndex);
633 if (newStorage != 0)
635 *ppstg = (IStorage*)newStorage;
638 * Since we are returning a pointer to the interface,
639 * we have to nail down the reference.
641 StorageBaseImpl_AddRef(*ppstg);
643 res = S_OK;
644 goto end;
647 res = STG_E_INSUFFICIENTMEMORY;
648 goto end;
651 res = STG_E_FILENOTFOUND;
653 end:
654 TRACE("<-- %08x\n", res);
655 return res;
658 /************************************************************************
659 * Storage32BaseImpl_EnumElements (IStorage)
661 * This method will create an enumerator object that can be used to
662 * retrieve information about all the properties in the storage object.
664 * See Windows documentation for more details on IStorage methods.
666 static HRESULT WINAPI StorageBaseImpl_EnumElements(
667 IStorage* iface,
668 DWORD reserved1, /* [in] */
669 void* reserved2, /* [size_is][unique][in] */
670 DWORD reserved3, /* [in] */
671 IEnumSTATSTG** ppenum) /* [out] */
673 StorageBaseImpl *This = (StorageBaseImpl *)iface;
674 IEnumSTATSTGImpl* newEnum;
676 TRACE("(%p, %d, %p, %d, %p)\n",
677 iface, reserved1, reserved2, reserved3, ppenum);
680 * Perform a sanity check on the parameters.
682 if ( (This==0) || (ppenum==0))
683 return E_INVALIDARG;
686 * Construct the enumerator.
688 newEnum = IEnumSTATSTGImpl_Construct(
689 This->ancestorStorage,
690 This->rootPropertySetIndex);
692 if (newEnum!=0)
694 *ppenum = (IEnumSTATSTG*)newEnum;
697 * Don't forget to nail down a reference to the new object before
698 * returning it.
700 IEnumSTATSTG_AddRef(*ppenum);
702 return S_OK;
705 return E_OUTOFMEMORY;
708 /************************************************************************
709 * Storage32BaseImpl_Stat (IStorage)
711 * This method will retrieve information about this storage object.
713 * See Windows documentation for more details on IStorage methods.
715 static HRESULT WINAPI StorageBaseImpl_Stat(
716 IStorage* iface,
717 STATSTG* pstatstg, /* [out] */
718 DWORD grfStatFlag) /* [in] */
720 StorageBaseImpl *This = (StorageBaseImpl *)iface;
721 StgProperty curProperty;
722 BOOL readSuccessful;
723 HRESULT res = STG_E_UNKNOWN;
725 TRACE("(%p, %p, %x)\n",
726 iface, pstatstg, grfStatFlag);
729 * Perform a sanity check on the parameters.
731 if ( (This==0) || (pstatstg==0))
733 res = E_INVALIDARG;
734 goto end;
738 * Read the information from the property.
740 readSuccessful = StorageImpl_ReadProperty(
741 This->ancestorStorage,
742 This->rootPropertySetIndex,
743 &curProperty);
745 if (readSuccessful)
747 StorageUtl_CopyPropertyToSTATSTG(
748 pstatstg,
749 &curProperty,
750 grfStatFlag);
752 pstatstg->grfMode = This->openFlags;
753 pstatstg->grfStateBits = This->stateBits;
755 res = S_OK;
756 goto end;
759 res = E_FAIL;
761 end:
762 if (res == S_OK)
764 TRACE("<-- STATSTG: pwcsName: %s, type: %d, cbSize.Low/High: %d/%d, grfMode: %08x, grfLocksSupported: %d, grfStateBits: %08x\n", debugstr_w(pstatstg->pwcsName), pstatstg->type, pstatstg->cbSize.u.LowPart, pstatstg->cbSize.u.HighPart, pstatstg->grfMode, pstatstg->grfLocksSupported, pstatstg->grfStateBits);
766 TRACE("<-- %08x\n", res);
767 return res;
770 /************************************************************************
771 * Storage32BaseImpl_RenameElement (IStorage)
773 * This method will rename the specified element.
775 * See Windows documentation for more details on IStorage methods.
777 * Implementation notes: The method used to rename consists of creating a clone
778 * of the deleted StgProperty object setting it with the new name and to
779 * perform a DestroyElement of the old StgProperty.
781 static HRESULT WINAPI StorageBaseImpl_RenameElement(
782 IStorage* iface,
783 const OLECHAR* pwcsOldName, /* [in] */
784 const OLECHAR* pwcsNewName) /* [in] */
786 StorageBaseImpl *This = (StorageBaseImpl *)iface;
787 IEnumSTATSTGImpl* propertyEnumeration;
788 StgProperty currentProperty;
789 ULONG foundPropertyIndex;
791 TRACE("(%p, %s, %s)\n",
792 iface, debugstr_w(pwcsOldName), debugstr_w(pwcsNewName));
795 * Create a property enumeration to search the properties
797 propertyEnumeration = IEnumSTATSTGImpl_Construct(This->ancestorStorage,
798 This->rootPropertySetIndex);
801 * Search the enumeration for the new property name
803 foundPropertyIndex = IEnumSTATSTGImpl_FindProperty(propertyEnumeration,
804 pwcsNewName,
805 &currentProperty);
807 if (foundPropertyIndex != PROPERTY_NULL)
810 * There is already a property with the new name
812 IEnumSTATSTGImpl_Destroy(propertyEnumeration);
813 return STG_E_FILEALREADYEXISTS;
816 IEnumSTATSTG_Reset((IEnumSTATSTG*)propertyEnumeration);
819 * Search the enumeration for the old property name
821 foundPropertyIndex = IEnumSTATSTGImpl_FindProperty(propertyEnumeration,
822 pwcsOldName,
823 &currentProperty);
826 * Delete the property enumeration since we don't need it anymore
828 IEnumSTATSTGImpl_Destroy(propertyEnumeration);
830 if (foundPropertyIndex != PROPERTY_NULL)
832 StgProperty renamedProperty;
833 ULONG renamedPropertyIndex;
836 * Setup a new property for the renamed property
838 renamedProperty.sizeOfNameString =
839 ( lstrlenW(pwcsNewName)+1 ) * sizeof(WCHAR);
841 if (renamedProperty.sizeOfNameString > PROPERTY_NAME_BUFFER_LEN)
842 return STG_E_INVALIDNAME;
844 strcpyW(renamedProperty.name, pwcsNewName);
846 renamedProperty.propertyType = currentProperty.propertyType;
847 renamedProperty.startingBlock = currentProperty.startingBlock;
848 renamedProperty.size.u.LowPart = currentProperty.size.u.LowPart;
849 renamedProperty.size.u.HighPart = currentProperty.size.u.HighPart;
851 renamedProperty.previousProperty = PROPERTY_NULL;
852 renamedProperty.nextProperty = PROPERTY_NULL;
855 * Bring the dirProperty link in case it is a storage and in which
856 * case the renamed storage elements don't require to be reorganized.
858 renamedProperty.dirProperty = currentProperty.dirProperty;
860 /* call CoFileTime to get the current time
861 renamedProperty.timeStampS1
862 renamedProperty.timeStampD1
863 renamedProperty.timeStampS2
864 renamedProperty.timeStampD2
865 renamedProperty.propertyUniqueID
869 * Obtain a free property in the property chain
871 renamedPropertyIndex = getFreeProperty(This->ancestorStorage);
874 * Save the new property into the new property spot
876 StorageImpl_WriteProperty(
877 This->ancestorStorage,
878 renamedPropertyIndex,
879 &renamedProperty);
882 * Find a spot in the property chain for our newly created property.
884 updatePropertyChain(
885 (StorageImpl*)This,
886 renamedPropertyIndex,
887 renamedProperty);
890 * At this point the renamed property has been inserted in the tree,
891 * now, before Destroying the old property we must zero its dirProperty
892 * otherwise the DestroyProperty below will zap it all and we do not want
893 * this to happen.
894 * Also, we fake that the old property is a storage so the DestroyProperty
895 * will not do a SetSize(0) on the stream data.
897 * This means that we need to tweak the StgProperty if it is a stream or a
898 * non empty storage.
900 StorageImpl_ReadProperty(This->ancestorStorage,
901 foundPropertyIndex,
902 &currentProperty);
904 currentProperty.dirProperty = PROPERTY_NULL;
905 currentProperty.propertyType = PROPTYPE_STORAGE;
906 StorageImpl_WriteProperty(
907 This->ancestorStorage,
908 foundPropertyIndex,
909 &currentProperty);
912 * Invoke Destroy to get rid of the ole property and automatically redo
913 * the linking of its previous and next members...
915 IStorage_DestroyElement((IStorage*)This->ancestorStorage, pwcsOldName);
918 else
921 * There is no property with the old name
923 return STG_E_FILENOTFOUND;
926 return S_OK;
929 /************************************************************************
930 * Storage32BaseImpl_CreateStream (IStorage)
932 * This method will create a stream object within this storage
934 * See Windows documentation for more details on IStorage methods.
936 static HRESULT WINAPI StorageBaseImpl_CreateStream(
937 IStorage* iface,
938 const OLECHAR* pwcsName, /* [string][in] */
939 DWORD grfMode, /* [in] */
940 DWORD reserved1, /* [in] */
941 DWORD reserved2, /* [in] */
942 IStream** ppstm) /* [out] */
944 StorageBaseImpl *This = (StorageBaseImpl *)iface;
945 IEnumSTATSTGImpl* propertyEnumeration;
946 StgStreamImpl* newStream;
947 StgProperty currentProperty, newStreamProperty;
948 ULONG foundPropertyIndex, newPropertyIndex;
950 TRACE("(%p, %s, %x, %d, %d, %p)\n",
951 iface, debugstr_w(pwcsName), grfMode,
952 reserved1, reserved2, ppstm);
955 * Validate parameters
957 if (ppstm == 0)
958 return STG_E_INVALIDPOINTER;
960 if (pwcsName == 0)
961 return STG_E_INVALIDNAME;
963 if (reserved1 || reserved2)
964 return STG_E_INVALIDPARAMETER;
967 * Validate the STGM flags
969 if ( FAILED( validateSTGM(grfMode) ))
970 return STG_E_INVALIDFLAG;
972 if (STGM_SHARE_MODE(grfMode) != STGM_SHARE_EXCLUSIVE)
973 return STG_E_INVALIDFLAG;
976 * As documented.
978 if ((grfMode & STGM_DELETEONRELEASE) ||
979 (grfMode & STGM_TRANSACTED))
980 return STG_E_INVALIDFUNCTION;
983 * Check that we're compatible with the parent's storage mode
984 * if not in transacted mode
986 if(!(This->ancestorStorage->base.openFlags & STGM_TRANSACTED)) {
987 if ( STGM_ACCESS_MODE( grfMode ) > STGM_ACCESS_MODE( This->openFlags ) )
988 return STG_E_ACCESSDENIED;
992 * Initialize the out parameter
994 *ppstm = 0;
997 * Create a property enumeration to search the properties
999 propertyEnumeration = IEnumSTATSTGImpl_Construct(This->ancestorStorage,
1000 This->rootPropertySetIndex);
1002 foundPropertyIndex = IEnumSTATSTGImpl_FindProperty(propertyEnumeration,
1003 pwcsName,
1004 &currentProperty);
1006 IEnumSTATSTGImpl_Destroy(propertyEnumeration);
1008 if (foundPropertyIndex != PROPERTY_NULL)
1011 * An element with this name already exists
1013 if (STGM_CREATE_MODE(grfMode) == STGM_CREATE)
1015 StgStreamImpl *strm;
1017 LIST_FOR_EACH_ENTRY(strm, &This->strmHead, StgStreamImpl, StrmListEntry)
1019 if (strm->ownerProperty == foundPropertyIndex)
1021 TRACE("Stream deleted %p\n", strm);
1022 strm->parentStorage = NULL;
1023 list_remove(&strm->StrmListEntry);
1026 IStorage_DestroyElement(iface, pwcsName);
1028 else
1029 return STG_E_FILEALREADYEXISTS;
1031 else if (STGM_ACCESS_MODE(This->openFlags) == STGM_READ)
1033 WARN("read-only storage\n");
1034 return STG_E_ACCESSDENIED;
1038 * memset the empty property
1040 memset(&newStreamProperty, 0, sizeof(StgProperty));
1042 newStreamProperty.sizeOfNameString =
1043 ( lstrlenW(pwcsName)+1 ) * sizeof(WCHAR);
1045 if (newStreamProperty.sizeOfNameString > PROPERTY_NAME_BUFFER_LEN)
1046 return STG_E_INVALIDNAME;
1048 strcpyW(newStreamProperty.name, pwcsName);
1050 newStreamProperty.propertyType = PROPTYPE_STREAM;
1051 newStreamProperty.startingBlock = BLOCK_END_OF_CHAIN;
1052 newStreamProperty.size.u.LowPart = 0;
1053 newStreamProperty.size.u.HighPart = 0;
1055 newStreamProperty.previousProperty = PROPERTY_NULL;
1056 newStreamProperty.nextProperty = PROPERTY_NULL;
1057 newStreamProperty.dirProperty = PROPERTY_NULL;
1059 /* call CoFileTime to get the current time
1060 newStreamProperty.timeStampS1
1061 newStreamProperty.timeStampD1
1062 newStreamProperty.timeStampS2
1063 newStreamProperty.timeStampD2
1066 /* newStreamProperty.propertyUniqueID */
1069 * Get a free property or create a new one
1071 newPropertyIndex = getFreeProperty(This->ancestorStorage);
1074 * Save the new property into the new property spot
1076 StorageImpl_WriteProperty(
1077 This->ancestorStorage,
1078 newPropertyIndex,
1079 &newStreamProperty);
1082 * Find a spot in the property chain for our newly created property.
1084 updatePropertyChain(
1085 (StorageImpl*)This,
1086 newPropertyIndex,
1087 newStreamProperty);
1090 * Open the stream to return it.
1092 newStream = StgStreamImpl_Construct(This, grfMode, newPropertyIndex);
1094 if (newStream != 0)
1096 *ppstm = (IStream*)newStream;
1099 * Since we are returning a pointer to the interface, we have to nail down
1100 * the reference.
1102 IStream_AddRef(*ppstm);
1104 else
1106 return STG_E_INSUFFICIENTMEMORY;
1109 return S_OK;
1112 /************************************************************************
1113 * Storage32BaseImpl_SetClass (IStorage)
1115 * This method will write the specified CLSID in the property of this
1116 * storage.
1118 * See Windows documentation for more details on IStorage methods.
1120 static HRESULT WINAPI StorageBaseImpl_SetClass(
1121 IStorage* iface,
1122 REFCLSID clsid) /* [in] */
1124 StorageBaseImpl *This = (StorageBaseImpl *)iface;
1125 HRESULT hRes = E_FAIL;
1126 StgProperty curProperty;
1127 BOOL success;
1129 TRACE("(%p, %p)\n", iface, clsid);
1131 success = StorageImpl_ReadProperty(This->ancestorStorage,
1132 This->rootPropertySetIndex,
1133 &curProperty);
1134 if (success)
1136 curProperty.propertyUniqueID = *clsid;
1138 success = StorageImpl_WriteProperty(This->ancestorStorage,
1139 This->rootPropertySetIndex,
1140 &curProperty);
1141 if (success)
1142 hRes = S_OK;
1145 return hRes;
1148 /************************************************************************
1149 ** Storage32Impl implementation
1152 /************************************************************************
1153 * Storage32Impl_CreateStorage (IStorage)
1155 * This method will create the storage object within the provided storage.
1157 * See Windows documentation for more details on IStorage methods.
1159 static HRESULT WINAPI StorageImpl_CreateStorage(
1160 IStorage* iface,
1161 const OLECHAR *pwcsName, /* [string][in] */
1162 DWORD grfMode, /* [in] */
1163 DWORD reserved1, /* [in] */
1164 DWORD reserved2, /* [in] */
1165 IStorage **ppstg) /* [out] */
1167 StorageImpl* const This=(StorageImpl*)iface;
1169 IEnumSTATSTGImpl *propertyEnumeration;
1170 StgProperty currentProperty;
1171 StgProperty newProperty;
1172 ULONG foundPropertyIndex;
1173 ULONG newPropertyIndex;
1174 HRESULT hr;
1176 TRACE("(%p, %s, %x, %d, %d, %p)\n",
1177 iface, debugstr_w(pwcsName), grfMode,
1178 reserved1, reserved2, ppstg);
1181 * Validate parameters
1183 if (ppstg == 0)
1184 return STG_E_INVALIDPOINTER;
1186 if (pwcsName == 0)
1187 return STG_E_INVALIDNAME;
1190 * Initialize the out parameter
1192 *ppstg = NULL;
1195 * Validate the STGM flags
1197 if ( FAILED( validateSTGM(grfMode) ) ||
1198 (grfMode & STGM_DELETEONRELEASE) )
1200 WARN("bad grfMode: 0x%x\n", grfMode);
1201 return STG_E_INVALIDFLAG;
1205 * Check that we're compatible with the parent's storage mode
1207 if ( STGM_ACCESS_MODE( grfMode ) > STGM_ACCESS_MODE( This->base.openFlags ) )
1209 WARN("access denied\n");
1210 return STG_E_ACCESSDENIED;
1214 * Create a property enumeration and search the properties
1216 propertyEnumeration = IEnumSTATSTGImpl_Construct( This->base.ancestorStorage,
1217 This->base.rootPropertySetIndex);
1219 foundPropertyIndex = IEnumSTATSTGImpl_FindProperty(propertyEnumeration,
1220 pwcsName,
1221 &currentProperty);
1222 IEnumSTATSTGImpl_Destroy(propertyEnumeration);
1224 if (foundPropertyIndex != PROPERTY_NULL)
1227 * An element with this name already exists
1229 if (STGM_CREATE_MODE(grfMode) == STGM_CREATE)
1230 IStorage_DestroyElement(iface, pwcsName);
1231 else
1233 WARN("file already exists\n");
1234 return STG_E_FILEALREADYEXISTS;
1237 else if (STGM_ACCESS_MODE(This->base.openFlags) == STGM_READ)
1239 WARN("read-only storage\n");
1240 return STG_E_ACCESSDENIED;
1244 * memset the empty property
1246 memset(&newProperty, 0, sizeof(StgProperty));
1248 newProperty.sizeOfNameString = (lstrlenW(pwcsName)+1)*sizeof(WCHAR);
1250 if (newProperty.sizeOfNameString > PROPERTY_NAME_BUFFER_LEN)
1252 FIXME("name too long\n");
1253 return STG_E_INVALIDNAME;
1256 strcpyW(newProperty.name, pwcsName);
1258 newProperty.propertyType = PROPTYPE_STORAGE;
1259 newProperty.startingBlock = BLOCK_END_OF_CHAIN;
1260 newProperty.size.u.LowPart = 0;
1261 newProperty.size.u.HighPart = 0;
1263 newProperty.previousProperty = PROPERTY_NULL;
1264 newProperty.nextProperty = PROPERTY_NULL;
1265 newProperty.dirProperty = PROPERTY_NULL;
1267 /* call CoFileTime to get the current time
1268 newProperty.timeStampS1
1269 newProperty.timeStampD1
1270 newProperty.timeStampS2
1271 newProperty.timeStampD2
1274 /* newStorageProperty.propertyUniqueID */
1277 * Obtain a free property in the property chain
1279 newPropertyIndex = getFreeProperty(This->base.ancestorStorage);
1282 * Save the new property into the new property spot
1284 StorageImpl_WriteProperty(
1285 This->base.ancestorStorage,
1286 newPropertyIndex,
1287 &newProperty);
1290 * Find a spot in the property chain for our newly created property.
1292 updatePropertyChain(
1293 This,
1294 newPropertyIndex,
1295 newProperty);
1298 * Open it to get a pointer to return.
1300 hr = IStorage_OpenStorage(iface, pwcsName, 0, grfMode, 0, 0, ppstg);
1302 if( (hr != S_OK) || (*ppstg == NULL))
1304 return hr;
1308 return S_OK;
1312 /***************************************************************************
1314 * Internal Method
1316 * Get a free property or create a new one.
1318 static ULONG getFreeProperty(
1319 StorageImpl *storage)
1321 ULONG currentPropertyIndex = 0;
1322 ULONG newPropertyIndex = PROPERTY_NULL;
1323 BOOL readSuccessful = TRUE;
1324 StgProperty currentProperty;
1329 * Start by reading the root property
1331 readSuccessful = StorageImpl_ReadProperty(storage->base.ancestorStorage,
1332 currentPropertyIndex,
1333 &currentProperty);
1334 if (readSuccessful)
1336 if (currentProperty.sizeOfNameString == 0)
1339 * The property existis and is available, we found it.
1341 newPropertyIndex = currentPropertyIndex;
1344 else
1347 * We exhausted the property list, we will create more space below
1349 newPropertyIndex = currentPropertyIndex;
1351 currentPropertyIndex++;
1353 } while (newPropertyIndex == PROPERTY_NULL);
1356 * grow the property chain
1358 if (! readSuccessful)
1360 StgProperty emptyProperty;
1361 ULARGE_INTEGER newSize;
1362 ULONG propertyIndex;
1363 ULONG lastProperty = 0;
1364 ULONG blockCount = 0;
1367 * obtain the new count of property blocks
1369 blockCount = BlockChainStream_GetCount(
1370 storage->base.ancestorStorage->rootBlockChain)+1;
1373 * initialize the size used by the property stream
1375 newSize.u.HighPart = 0;
1376 newSize.u.LowPart = storage->bigBlockSize * blockCount;
1379 * add a property block to the property chain
1381 BlockChainStream_SetSize(storage->base.ancestorStorage->rootBlockChain, newSize);
1384 * memset the empty property in order to initialize the unused newly
1385 * created property
1387 memset(&emptyProperty, 0, sizeof(StgProperty));
1390 * initialize them
1392 lastProperty = storage->bigBlockSize / PROPSET_BLOCK_SIZE * blockCount;
1394 for(
1395 propertyIndex = newPropertyIndex;
1396 propertyIndex < lastProperty;
1397 propertyIndex++)
1399 StorageImpl_WriteProperty(
1400 storage->base.ancestorStorage,
1401 propertyIndex,
1402 &emptyProperty);
1406 return newPropertyIndex;
1409 /****************************************************************************
1411 * Internal Method
1413 * Case insensitive comparison of StgProperty.name by first considering
1414 * their size.
1416 * Returns <0 when newProperty < currentProperty
1417 * >0 when newProperty > currentProperty
1418 * 0 when newProperty == currentProperty
1420 static LONG propertyNameCmp(
1421 const OLECHAR *newProperty,
1422 const OLECHAR *currentProperty)
1424 LONG diff = lstrlenW(newProperty) - lstrlenW(currentProperty);
1426 if (diff == 0)
1429 * We compare the string themselves only when they are of the same length
1431 diff = lstrcmpiW( newProperty, currentProperty);
1434 return diff;
1437 /****************************************************************************
1439 * Internal Method
1441 * Properly link this new element in the property chain.
1443 static void updatePropertyChain(
1444 StorageImpl *storage,
1445 ULONG newPropertyIndex,
1446 StgProperty newProperty)
1448 StgProperty currentProperty;
1451 * Read the root property
1453 StorageImpl_ReadProperty(storage->base.ancestorStorage,
1454 storage->base.rootPropertySetIndex,
1455 &currentProperty);
1457 if (currentProperty.dirProperty != PROPERTY_NULL)
1460 * The root storage contains some element, therefore, start the research
1461 * for the appropriate location.
1463 BOOL found = 0;
1464 ULONG current, next, previous, currentPropertyId;
1467 * Keep the StgProperty sequence number of the storage first property
1469 currentPropertyId = currentProperty.dirProperty;
1472 * Read
1474 StorageImpl_ReadProperty(storage->base.ancestorStorage,
1475 currentProperty.dirProperty,
1476 &currentProperty);
1478 previous = currentProperty.previousProperty;
1479 next = currentProperty.nextProperty;
1480 current = currentPropertyId;
1482 while (found == 0)
1484 LONG diff = propertyNameCmp( newProperty.name, currentProperty.name);
1486 if (diff < 0)
1488 if (previous != PROPERTY_NULL)
1490 StorageImpl_ReadProperty(storage->base.ancestorStorage,
1491 previous,
1492 &currentProperty);
1493 current = previous;
1495 else
1497 currentProperty.previousProperty = newPropertyIndex;
1498 StorageImpl_WriteProperty(storage->base.ancestorStorage,
1499 current,
1500 &currentProperty);
1501 found = 1;
1504 else if (diff > 0)
1506 if (next != PROPERTY_NULL)
1508 StorageImpl_ReadProperty(storage->base.ancestorStorage,
1509 next,
1510 &currentProperty);
1511 current = next;
1513 else
1515 currentProperty.nextProperty = newPropertyIndex;
1516 StorageImpl_WriteProperty(storage->base.ancestorStorage,
1517 current,
1518 &currentProperty);
1519 found = 1;
1522 else
1525 * Trying to insert an item with the same name in the
1526 * subtree structure.
1528 assert(FALSE);
1531 previous = currentProperty.previousProperty;
1532 next = currentProperty.nextProperty;
1535 else
1538 * The root storage is empty, link the new property to its dir property
1540 currentProperty.dirProperty = newPropertyIndex;
1541 StorageImpl_WriteProperty(storage->base.ancestorStorage,
1542 storage->base.rootPropertySetIndex,
1543 &currentProperty);
1548 /*************************************************************************
1549 * CopyTo (IStorage)
1551 static HRESULT WINAPI StorageImpl_CopyTo(
1552 IStorage* iface,
1553 DWORD ciidExclude, /* [in] */
1554 const IID* rgiidExclude, /* [size_is][unique][in] */
1555 SNB snbExclude, /* [unique][in] */
1556 IStorage* pstgDest) /* [unique][in] */
1558 IEnumSTATSTG *elements = 0;
1559 STATSTG curElement, strStat;
1560 HRESULT hr;
1561 IStorage *pstgTmp, *pstgChild;
1562 IStream *pstrTmp, *pstrChild;
1564 if ((ciidExclude != 0) || (rgiidExclude != NULL) || (snbExclude != NULL))
1565 FIXME("Exclude option not implemented\n");
1567 TRACE("(%p, %d, %p, %p, %p)\n",
1568 iface, ciidExclude, rgiidExclude,
1569 snbExclude, pstgDest);
1572 * Perform a sanity check
1574 if ( pstgDest == 0 )
1575 return STG_E_INVALIDPOINTER;
1578 * Enumerate the elements
1580 hr = IStorage_EnumElements( iface, 0, 0, 0, &elements );
1582 if ( hr != S_OK )
1583 return hr;
1586 * set the class ID
1588 IStorage_Stat( iface, &curElement, STATFLAG_NONAME);
1589 IStorage_SetClass( pstgDest, &curElement.clsid );
1594 * Obtain the next element
1596 hr = IEnumSTATSTG_Next( elements, 1, &curElement, NULL );
1598 if ( hr == S_FALSE )
1600 hr = S_OK; /* done, every element has been copied */
1601 break;
1604 if (curElement.type == STGTY_STORAGE)
1607 * open child source storage
1609 hr = IStorage_OpenStorage( iface, curElement.pwcsName, NULL,
1610 STGM_READ|STGM_SHARE_EXCLUSIVE,
1611 NULL, 0, &pstgChild );
1613 if (hr != S_OK)
1614 break;
1617 * Check if destination storage is not a child of the source
1618 * storage, which will cause an infinite loop
1620 if (pstgChild == pstgDest)
1622 IEnumSTATSTG_Release(elements);
1624 return STG_E_ACCESSDENIED;
1628 * create a new storage in destination storage
1630 hr = IStorage_CreateStorage( pstgDest, curElement.pwcsName,
1631 STGM_FAILIFTHERE|STGM_WRITE|STGM_SHARE_EXCLUSIVE,
1632 0, 0,
1633 &pstgTmp );
1635 * if it already exist, don't create a new one use this one
1637 if (hr == STG_E_FILEALREADYEXISTS)
1639 hr = IStorage_OpenStorage( pstgDest, curElement.pwcsName, NULL,
1640 STGM_WRITE|STGM_SHARE_EXCLUSIVE,
1641 NULL, 0, &pstgTmp );
1644 if (hr != S_OK)
1645 break;
1649 * do the copy recursively
1651 hr = IStorage_CopyTo( pstgChild, ciidExclude, rgiidExclude,
1652 snbExclude, pstgTmp );
1654 IStorage_Release( pstgTmp );
1655 IStorage_Release( pstgChild );
1657 else if (curElement.type == STGTY_STREAM)
1660 * create a new stream in destination storage. If the stream already
1661 * exist, it will be deleted and a new one will be created.
1663 hr = IStorage_CreateStream( pstgDest, curElement.pwcsName,
1664 STGM_CREATE|STGM_WRITE|STGM_SHARE_EXCLUSIVE,
1665 0, 0, &pstrTmp );
1667 if (hr != S_OK)
1668 break;
1671 * open child stream storage
1673 hr = IStorage_OpenStream( iface, curElement.pwcsName, NULL,
1674 STGM_READ|STGM_SHARE_EXCLUSIVE,
1675 0, &pstrChild );
1677 if (hr != S_OK)
1678 break;
1681 * Get the size of the source stream
1683 IStream_Stat( pstrChild, &strStat, STATFLAG_NONAME );
1686 * Set the size of the destination stream.
1688 IStream_SetSize(pstrTmp, strStat.cbSize);
1691 * do the copy
1693 hr = IStream_CopyTo( pstrChild, pstrTmp, strStat.cbSize,
1694 NULL, NULL );
1696 IStream_Release( pstrTmp );
1697 IStream_Release( pstrChild );
1699 else
1701 WARN("unknown element type: %d\n", curElement.type);
1704 } while (hr == S_OK);
1707 * Clean-up
1709 IEnumSTATSTG_Release(elements);
1711 return hr;
1714 /*************************************************************************
1715 * MoveElementTo (IStorage)
1717 static HRESULT WINAPI StorageImpl_MoveElementTo(
1718 IStorage* iface,
1719 const OLECHAR *pwcsName, /* [string][in] */
1720 IStorage *pstgDest, /* [unique][in] */
1721 const OLECHAR *pwcsNewName,/* [string][in] */
1722 DWORD grfFlags) /* [in] */
1724 FIXME("(%p %s %p %s %u): stub\n", iface,
1725 debugstr_w(pwcsName), pstgDest,
1726 debugstr_w(pwcsNewName), grfFlags);
1727 return E_NOTIMPL;
1730 /*************************************************************************
1731 * Commit (IStorage)
1733 * Ensures that any changes made to a storage object open in transacted mode
1734 * are reflected in the parent storage
1736 * NOTES
1737 * Wine doesn't implement transacted mode, which seems to be a basic
1738 * optimization, so we can ignore this stub for now.
1740 static HRESULT WINAPI StorageImpl_Commit(
1741 IStorage* iface,
1742 DWORD grfCommitFlags)/* [in] */
1744 FIXME("(%p %d): stub\n", iface, grfCommitFlags);
1745 return S_OK;
1748 /*************************************************************************
1749 * Revert (IStorage)
1751 * Discard all changes that have been made since the last commit operation
1753 static HRESULT WINAPI StorageImpl_Revert(
1754 IStorage* iface)
1756 FIXME("(%p): stub\n", iface);
1757 return E_NOTIMPL;
1760 /*************************************************************************
1761 * DestroyElement (IStorage)
1763 * Strategy: This implementation is built this way for simplicity not for speed.
1764 * I always delete the topmost element of the enumeration and adjust
1765 * the deleted element pointer all the time. This takes longer to
1766 * do but allow to reinvoke DestroyElement whenever we encounter a
1767 * storage object. The optimisation resides in the usage of another
1768 * enumeration strategy that would give all the leaves of a storage
1769 * first. (postfix order)
1771 static HRESULT WINAPI StorageImpl_DestroyElement(
1772 IStorage* iface,
1773 const OLECHAR *pwcsName)/* [string][in] */
1775 StorageImpl* const This=(StorageImpl*)iface;
1777 IEnumSTATSTGImpl* propertyEnumeration;
1778 HRESULT hr = S_OK;
1779 BOOL res;
1780 StgProperty propertyToDelete;
1781 StgProperty parentProperty;
1782 ULONG foundPropertyIndexToDelete;
1783 ULONG typeOfRelation;
1784 ULONG parentPropertyId = 0;
1786 TRACE("(%p, %s)\n",
1787 iface, debugstr_w(pwcsName));
1790 * Perform a sanity check on the parameters.
1792 if (pwcsName==NULL)
1793 return STG_E_INVALIDPOINTER;
1796 * Create a property enumeration to search the property with the given name
1798 propertyEnumeration = IEnumSTATSTGImpl_Construct(
1799 This->base.ancestorStorage,
1800 This->base.rootPropertySetIndex);
1802 foundPropertyIndexToDelete = IEnumSTATSTGImpl_FindProperty(
1803 propertyEnumeration,
1804 pwcsName,
1805 &propertyToDelete);
1807 IEnumSTATSTGImpl_Destroy(propertyEnumeration);
1809 if ( foundPropertyIndexToDelete == PROPERTY_NULL )
1811 return STG_E_FILENOTFOUND;
1815 * Find the parent property of the property to delete (the one that
1816 * link to it). If This->dirProperty == foundPropertyIndexToDelete,
1817 * the parent is This. Otherwise, the parent is one of its sibling...
1821 * First, read This's StgProperty..
1823 res = StorageImpl_ReadProperty(
1824 This->base.ancestorStorage,
1825 This->base.rootPropertySetIndex,
1826 &parentProperty);
1828 assert(res);
1831 * Second, check to see if by any chance the actual storage (This) is not
1832 * the parent of the property to delete... We never know...
1834 if ( parentProperty.dirProperty == foundPropertyIndexToDelete )
1837 * Set data as it would have been done in the else part...
1839 typeOfRelation = PROPERTY_RELATION_DIR;
1840 parentPropertyId = This->base.rootPropertySetIndex;
1842 else
1845 * Create a property enumeration to search the parent properties, and
1846 * delete it once done.
1848 IEnumSTATSTGImpl* propertyEnumeration2;
1850 propertyEnumeration2 = IEnumSTATSTGImpl_Construct(
1851 This->base.ancestorStorage,
1852 This->base.rootPropertySetIndex);
1854 typeOfRelation = IEnumSTATSTGImpl_FindParentProperty(
1855 propertyEnumeration2,
1856 foundPropertyIndexToDelete,
1857 &parentProperty,
1858 &parentPropertyId);
1860 IEnumSTATSTGImpl_Destroy(propertyEnumeration2);
1863 if ( propertyToDelete.propertyType == PROPTYPE_STORAGE )
1865 hr = deleteStorageProperty(
1866 This,
1867 foundPropertyIndexToDelete,
1868 propertyToDelete);
1870 else if ( propertyToDelete.propertyType == PROPTYPE_STREAM )
1872 hr = deleteStreamProperty(
1873 This,
1874 foundPropertyIndexToDelete,
1875 propertyToDelete);
1878 if (hr!=S_OK)
1879 return hr;
1882 * Adjust the property chain
1884 hr = adjustPropertyChain(
1885 This,
1886 propertyToDelete,
1887 parentProperty,
1888 parentPropertyId,
1889 typeOfRelation);
1891 return hr;
1895 /************************************************************************
1896 * StorageImpl_Stat (IStorage)
1898 * This method will retrieve information about this storage object.
1900 * See Windows documentation for more details on IStorage methods.
1902 static HRESULT WINAPI StorageImpl_Stat( IStorage* iface,
1903 STATSTG* pstatstg, /* [out] */
1904 DWORD grfStatFlag) /* [in] */
1906 StorageImpl* const This = (StorageImpl*)iface;
1907 HRESULT result = StorageBaseImpl_Stat( iface, pstatstg, grfStatFlag );
1909 if ( SUCCEEDED(result) && ((grfStatFlag & STATFLAG_NONAME) == 0) && This->pwcsName )
1911 CoTaskMemFree(pstatstg->pwcsName);
1912 pstatstg->pwcsName = CoTaskMemAlloc((lstrlenW(This->pwcsName)+1)*sizeof(WCHAR));
1913 strcpyW(pstatstg->pwcsName, This->pwcsName);
1916 return result;
1919 /******************************************************************************
1920 * Internal stream list handlers
1923 void StorageBaseImpl_AddStream(StorageBaseImpl * stg, StgStreamImpl * strm)
1925 TRACE("Stream added (stg=%p strm=%p)\n", stg, strm);
1926 list_add_tail(&stg->strmHead,&strm->StrmListEntry);
1929 void StorageBaseImpl_RemoveStream(StorageBaseImpl * stg, StgStreamImpl * strm)
1931 TRACE("Stream removed (stg=%p strm=%p)\n", stg,strm);
1932 list_remove(&(strm->StrmListEntry));
1935 static void StorageBaseImpl_DeleteAll(StorageBaseImpl * stg)
1937 struct list *cur, *cur2;
1938 StgStreamImpl *strm=NULL;
1940 LIST_FOR_EACH_SAFE(cur, cur2, &stg->strmHead) {
1941 strm = LIST_ENTRY(cur,StgStreamImpl,StrmListEntry);
1942 TRACE("Streams deleted (stg=%p strm=%p next=%p prev=%p)\n", stg,strm,cur->next,cur->prev);
1943 strm->parentStorage = NULL;
1944 list_remove(cur);
1949 /*********************************************************************
1951 * Internal Method
1953 * Perform the deletion of a complete storage node
1956 static HRESULT deleteStorageProperty(
1957 StorageImpl *parentStorage,
1958 ULONG indexOfPropertyToDelete,
1959 StgProperty propertyToDelete)
1961 IEnumSTATSTG *elements = 0;
1962 IStorage *childStorage = 0;
1963 STATSTG currentElement;
1964 HRESULT hr;
1965 HRESULT destroyHr = S_OK;
1968 * Open the storage and enumerate it
1970 hr = StorageBaseImpl_OpenStorage(
1971 (IStorage*)parentStorage,
1972 propertyToDelete.name,
1974 STGM_WRITE | STGM_SHARE_EXCLUSIVE,
1977 &childStorage);
1979 if (hr != S_OK)
1981 return hr;
1985 * Enumerate the elements
1987 IStorage_EnumElements( childStorage, 0, 0, 0, &elements);
1992 * Obtain the next element
1994 hr = IEnumSTATSTG_Next(elements, 1, &currentElement, NULL);
1995 if (hr==S_OK)
1997 destroyHr = StorageImpl_DestroyElement(childStorage, currentElement.pwcsName);
1999 CoTaskMemFree(currentElement.pwcsName);
2003 * We need to Reset the enumeration every time because we delete elements
2004 * and the enumeration could be invalid
2006 IEnumSTATSTG_Reset(elements);
2008 } while ((hr == S_OK) && (destroyHr == S_OK));
2011 * Invalidate the property by zeroing its name member.
2013 propertyToDelete.sizeOfNameString = 0;
2015 StorageImpl_WriteProperty(parentStorage->base.ancestorStorage,
2016 indexOfPropertyToDelete,
2017 &propertyToDelete);
2019 IStorage_Release(childStorage);
2020 IEnumSTATSTG_Release(elements);
2022 return destroyHr;
2025 /*********************************************************************
2027 * Internal Method
2029 * Perform the deletion of a stream node
2032 static HRESULT deleteStreamProperty(
2033 StorageImpl *parentStorage,
2034 ULONG indexOfPropertyToDelete,
2035 StgProperty propertyToDelete)
2037 IStream *pis;
2038 HRESULT hr;
2039 ULARGE_INTEGER size;
2041 size.u.HighPart = 0;
2042 size.u.LowPart = 0;
2044 hr = StorageBaseImpl_OpenStream(
2045 (IStorage*)parentStorage,
2046 (OLECHAR*)propertyToDelete.name,
2047 NULL,
2048 STGM_WRITE | STGM_SHARE_EXCLUSIVE,
2050 &pis);
2052 if (hr!=S_OK)
2054 return(hr);
2058 * Zap the stream
2060 hr = IStream_SetSize(pis, size);
2062 if(hr != S_OK)
2064 return hr;
2068 * Release the stream object.
2070 IStream_Release(pis);
2073 * Invalidate the property by zeroing its name member.
2075 propertyToDelete.sizeOfNameString = 0;
2078 * Here we should re-read the property so we get the updated pointer
2079 * but since we are here to zap it, I don't do it...
2081 StorageImpl_WriteProperty(
2082 parentStorage->base.ancestorStorage,
2083 indexOfPropertyToDelete,
2084 &propertyToDelete);
2086 return S_OK;
2089 /*********************************************************************
2091 * Internal Method
2093 * Finds a placeholder for the StgProperty within the Storage
2096 static HRESULT findPlaceholder(
2097 StorageImpl *storage,
2098 ULONG propertyIndexToStore,
2099 ULONG storePropertyIndex,
2100 INT typeOfRelation)
2102 StgProperty storeProperty;
2103 BOOL res = TRUE;
2106 * Read the storage property
2108 res = StorageImpl_ReadProperty(
2109 storage->base.ancestorStorage,
2110 storePropertyIndex,
2111 &storeProperty);
2113 if(! res)
2115 return E_FAIL;
2118 if (typeOfRelation == PROPERTY_RELATION_PREVIOUS)
2120 if (storeProperty.previousProperty != PROPERTY_NULL)
2122 return findPlaceholder(
2123 storage,
2124 propertyIndexToStore,
2125 storeProperty.previousProperty,
2126 typeOfRelation);
2128 else
2130 storeProperty.previousProperty = propertyIndexToStore;
2133 else if (typeOfRelation == PROPERTY_RELATION_NEXT)
2135 if (storeProperty.nextProperty != PROPERTY_NULL)
2137 return findPlaceholder(
2138 storage,
2139 propertyIndexToStore,
2140 storeProperty.nextProperty,
2141 typeOfRelation);
2143 else
2145 storeProperty.nextProperty = propertyIndexToStore;
2148 else if (typeOfRelation == PROPERTY_RELATION_DIR)
2150 if (storeProperty.dirProperty != PROPERTY_NULL)
2152 return findPlaceholder(
2153 storage,
2154 propertyIndexToStore,
2155 storeProperty.dirProperty,
2156 typeOfRelation);
2158 else
2160 storeProperty.dirProperty = propertyIndexToStore;
2164 res = StorageImpl_WriteProperty(
2165 storage->base.ancestorStorage,
2166 storePropertyIndex,
2167 &storeProperty);
2169 if(!res)
2171 return E_FAIL;
2174 return S_OK;
2177 /*************************************************************************
2179 * Internal Method
2181 * This method takes the previous and the next property link of a property
2182 * to be deleted and find them a place in the Storage.
2184 static HRESULT adjustPropertyChain(
2185 StorageImpl *This,
2186 StgProperty propertyToDelete,
2187 StgProperty parentProperty,
2188 ULONG parentPropertyId,
2189 INT typeOfRelation)
2191 ULONG newLinkProperty = PROPERTY_NULL;
2192 BOOL needToFindAPlaceholder = FALSE;
2193 ULONG storeNode = PROPERTY_NULL;
2194 ULONG toStoreNode = PROPERTY_NULL;
2195 INT relationType = 0;
2196 HRESULT hr = S_OK;
2197 BOOL res = TRUE;
2199 if (typeOfRelation == PROPERTY_RELATION_PREVIOUS)
2201 if (propertyToDelete.previousProperty != PROPERTY_NULL)
2204 * Set the parent previous to the property to delete previous
2206 newLinkProperty = propertyToDelete.previousProperty;
2208 if (propertyToDelete.nextProperty != PROPERTY_NULL)
2211 * We also need to find a storage for the other link, setup variables
2212 * to do this at the end...
2214 needToFindAPlaceholder = TRUE;
2215 storeNode = propertyToDelete.previousProperty;
2216 toStoreNode = propertyToDelete.nextProperty;
2217 relationType = PROPERTY_RELATION_NEXT;
2220 else if (propertyToDelete.nextProperty != PROPERTY_NULL)
2223 * Set the parent previous to the property to delete next
2225 newLinkProperty = propertyToDelete.nextProperty;
2229 * Link it for real...
2231 parentProperty.previousProperty = newLinkProperty;
2234 else if (typeOfRelation == PROPERTY_RELATION_NEXT)
2236 if (propertyToDelete.previousProperty != PROPERTY_NULL)
2239 * Set the parent next to the property to delete next previous
2241 newLinkProperty = propertyToDelete.previousProperty;
2243 if (propertyToDelete.nextProperty != PROPERTY_NULL)
2246 * We also need to find a storage for the other link, setup variables
2247 * to do this at the end...
2249 needToFindAPlaceholder = TRUE;
2250 storeNode = propertyToDelete.previousProperty;
2251 toStoreNode = propertyToDelete.nextProperty;
2252 relationType = PROPERTY_RELATION_NEXT;
2255 else if (propertyToDelete.nextProperty != PROPERTY_NULL)
2258 * Set the parent next to the property to delete next
2260 newLinkProperty = propertyToDelete.nextProperty;
2264 * Link it for real...
2266 parentProperty.nextProperty = newLinkProperty;
2268 else /* (typeOfRelation == PROPERTY_RELATION_DIR) */
2270 if (propertyToDelete.previousProperty != PROPERTY_NULL)
2273 * Set the parent dir to the property to delete previous
2275 newLinkProperty = propertyToDelete.previousProperty;
2277 if (propertyToDelete.nextProperty != PROPERTY_NULL)
2280 * We also need to find a storage for the other link, setup variables
2281 * to do this at the end...
2283 needToFindAPlaceholder = TRUE;
2284 storeNode = propertyToDelete.previousProperty;
2285 toStoreNode = propertyToDelete.nextProperty;
2286 relationType = PROPERTY_RELATION_NEXT;
2289 else if (propertyToDelete.nextProperty != PROPERTY_NULL)
2292 * Set the parent dir to the property to delete next
2294 newLinkProperty = propertyToDelete.nextProperty;
2298 * Link it for real...
2300 parentProperty.dirProperty = newLinkProperty;
2304 * Write back the parent property
2306 res = StorageImpl_WriteProperty(
2307 This->base.ancestorStorage,
2308 parentPropertyId,
2309 &parentProperty);
2310 if(! res)
2312 return E_FAIL;
2316 * If a placeholder is required for the other link, then, find one and
2317 * get out of here...
2319 if (needToFindAPlaceholder)
2321 hr = findPlaceholder(
2322 This,
2323 toStoreNode,
2324 storeNode,
2325 relationType);
2328 return hr;
2332 /******************************************************************************
2333 * SetElementTimes (IStorage)
2335 static HRESULT WINAPI StorageImpl_SetElementTimes(
2336 IStorage* iface,
2337 const OLECHAR *pwcsName,/* [string][in] */
2338 const FILETIME *pctime, /* [in] */
2339 const FILETIME *patime, /* [in] */
2340 const FILETIME *pmtime) /* [in] */
2342 FIXME("(%s,...), stub!\n",debugstr_w(pwcsName));
2343 return S_OK;
2346 /******************************************************************************
2347 * SetStateBits (IStorage)
2349 static HRESULT WINAPI StorageImpl_SetStateBits(
2350 IStorage* iface,
2351 DWORD grfStateBits,/* [in] */
2352 DWORD grfMask) /* [in] */
2354 StorageImpl* const This = (StorageImpl*)iface;
2355 This->base.stateBits = (This->base.stateBits & ~grfMask) | (grfStateBits & grfMask);
2356 return S_OK;
2360 * Virtual function table for the IStorage32Impl class.
2362 static const IStorageVtbl Storage32Impl_Vtbl =
2364 StorageBaseImpl_QueryInterface,
2365 StorageBaseImpl_AddRef,
2366 StorageBaseImpl_Release,
2367 StorageBaseImpl_CreateStream,
2368 StorageBaseImpl_OpenStream,
2369 StorageImpl_CreateStorage,
2370 StorageBaseImpl_OpenStorage,
2371 StorageImpl_CopyTo,
2372 StorageImpl_MoveElementTo,
2373 StorageImpl_Commit,
2374 StorageImpl_Revert,
2375 StorageBaseImpl_EnumElements,
2376 StorageImpl_DestroyElement,
2377 StorageBaseImpl_RenameElement,
2378 StorageImpl_SetElementTimes,
2379 StorageBaseImpl_SetClass,
2380 StorageImpl_SetStateBits,
2381 StorageImpl_Stat
2384 static HRESULT StorageImpl_Construct(
2385 StorageImpl* This,
2386 HANDLE hFile,
2387 LPCOLESTR pwcsName,
2388 ILockBytes* pLkbyt,
2389 DWORD openFlags,
2390 BOOL fileBased,
2391 BOOL fileCreate)
2393 HRESULT hr = S_OK;
2394 StgProperty currentProperty;
2395 BOOL readSuccessful;
2396 ULONG currentPropertyIndex;
2398 if ( FAILED( validateSTGM(openFlags) ))
2399 return STG_E_INVALIDFLAG;
2401 memset(This, 0, sizeof(StorageImpl));
2404 * Initialize stream list
2407 list_init(&This->base.strmHead);
2410 * Initialize the virtual function table.
2412 This->base.lpVtbl = &Storage32Impl_Vtbl;
2413 This->base.pssVtbl = &IPropertySetStorage_Vtbl;
2414 This->base.v_destructor = StorageImpl_Destroy;
2415 This->base.openFlags = (openFlags & ~STGM_CREATE);
2418 * This is the top-level storage so initialize the ancestor pointer
2419 * to this.
2421 This->base.ancestorStorage = This;
2424 * Initialize the physical support of the storage.
2426 This->hFile = hFile;
2429 * Store copy of file path.
2431 if(pwcsName) {
2432 This->pwcsName = HeapAlloc(GetProcessHeap(), 0,
2433 (lstrlenW(pwcsName)+1)*sizeof(WCHAR));
2434 if (!This->pwcsName)
2435 return STG_E_INSUFFICIENTMEMORY;
2436 strcpyW(This->pwcsName, pwcsName);
2440 * Initialize the big block cache.
2442 This->bigBlockSize = DEF_BIG_BLOCK_SIZE;
2443 This->smallBlockSize = DEF_SMALL_BLOCK_SIZE;
2444 This->bigBlockFile = BIGBLOCKFILE_Construct(hFile,
2445 pLkbyt,
2446 openFlags,
2447 This->bigBlockSize,
2448 fileBased);
2450 if (This->bigBlockFile == 0)
2451 return E_FAIL;
2453 if (fileCreate)
2455 ULARGE_INTEGER size;
2456 BYTE bigBlockBuffer[BIG_BLOCK_SIZE];
2459 * Initialize all header variables:
2460 * - The big block depot consists of one block and it is at block 0
2461 * - The properties start at block 1
2462 * - There is no small block depot
2464 memset( This->bigBlockDepotStart,
2465 BLOCK_UNUSED,
2466 sizeof(This->bigBlockDepotStart));
2468 This->bigBlockDepotCount = 1;
2469 This->bigBlockDepotStart[0] = 0;
2470 This->rootStartBlock = 1;
2471 This->smallBlockDepotStart = BLOCK_END_OF_CHAIN;
2472 This->bigBlockSizeBits = DEF_BIG_BLOCK_SIZE_BITS;
2473 This->smallBlockSizeBits = DEF_SMALL_BLOCK_SIZE_BITS;
2474 This->extBigBlockDepotStart = BLOCK_END_OF_CHAIN;
2475 This->extBigBlockDepotCount = 0;
2477 StorageImpl_SaveFileHeader(This);
2480 * Add one block for the big block depot and one block for the properties
2482 size.u.HighPart = 0;
2483 size.u.LowPart = This->bigBlockSize * 3;
2484 BIGBLOCKFILE_SetSize(This->bigBlockFile, size);
2487 * Initialize the big block depot
2489 memset(bigBlockBuffer, BLOCK_UNUSED, This->bigBlockSize);
2490 StorageUtl_WriteDWord(bigBlockBuffer, 0, BLOCK_SPECIAL);
2491 StorageUtl_WriteDWord(bigBlockBuffer, sizeof(ULONG), BLOCK_END_OF_CHAIN);
2492 StorageImpl_WriteBigBlock(This, 0, bigBlockBuffer);
2494 else
2497 * Load the header for the file.
2499 hr = StorageImpl_LoadFileHeader(This);
2501 if (FAILED(hr))
2503 BIGBLOCKFILE_Destructor(This->bigBlockFile);
2505 return hr;
2510 * There is no block depot cached yet.
2512 This->indexBlockDepotCached = 0xFFFFFFFF;
2515 * Start searching for free blocks with block 0.
2517 This->prevFreeBlock = 0;
2520 * Create the block chain abstractions.
2522 if(!(This->rootBlockChain =
2523 BlockChainStream_Construct(This, &This->rootStartBlock, PROPERTY_NULL)))
2524 return STG_E_READFAULT;
2526 if(!(This->smallBlockDepotChain =
2527 BlockChainStream_Construct(This, &This->smallBlockDepotStart,
2528 PROPERTY_NULL)))
2529 return STG_E_READFAULT;
2532 * Write the root property (memory only)
2534 if (fileCreate)
2536 StgProperty rootProp;
2538 * Initialize the property chain
2540 memset(&rootProp, 0, sizeof(rootProp));
2541 MultiByteToWideChar( CP_ACP, 0, rootPropertyName, -1, rootProp.name,
2542 sizeof(rootProp.name)/sizeof(WCHAR) );
2543 rootProp.sizeOfNameString = (strlenW(rootProp.name)+1) * sizeof(WCHAR);
2544 rootProp.propertyType = PROPTYPE_ROOT;
2545 rootProp.previousProperty = PROPERTY_NULL;
2546 rootProp.nextProperty = PROPERTY_NULL;
2547 rootProp.dirProperty = PROPERTY_NULL;
2548 rootProp.startingBlock = BLOCK_END_OF_CHAIN;
2549 rootProp.size.u.HighPart = 0;
2550 rootProp.size.u.LowPart = 0;
2552 StorageImpl_WriteProperty(This, 0, &rootProp);
2556 * Find the ID of the root in the property sets.
2558 currentPropertyIndex = 0;
2562 readSuccessful = StorageImpl_ReadProperty(
2563 This,
2564 currentPropertyIndex,
2565 &currentProperty);
2567 if (readSuccessful)
2569 if ( (currentProperty.sizeOfNameString != 0 ) &&
2570 (currentProperty.propertyType == PROPTYPE_ROOT) )
2572 This->base.rootPropertySetIndex = currentPropertyIndex;
2576 currentPropertyIndex++;
2578 } while (readSuccessful && (This->base.rootPropertySetIndex == PROPERTY_NULL) );
2580 if (!readSuccessful)
2582 /* TODO CLEANUP */
2583 return STG_E_READFAULT;
2587 * Create the block chain abstraction for the small block root chain.
2589 if(!(This->smallBlockRootChain =
2590 BlockChainStream_Construct(This, NULL, This->base.rootPropertySetIndex)))
2591 return STG_E_READFAULT;
2593 return hr;
2596 static void StorageImpl_Destroy(StorageBaseImpl* iface)
2598 StorageImpl *This = (StorageImpl*) iface;
2599 TRACE("(%p)\n", This);
2601 StorageBaseImpl_DeleteAll(&This->base);
2603 HeapFree(GetProcessHeap(), 0, This->pwcsName);
2605 BlockChainStream_Destroy(This->smallBlockRootChain);
2606 BlockChainStream_Destroy(This->rootBlockChain);
2607 BlockChainStream_Destroy(This->smallBlockDepotChain);
2609 BIGBLOCKFILE_Destructor(This->bigBlockFile);
2610 HeapFree(GetProcessHeap(), 0, This);
2613 /******************************************************************************
2614 * Storage32Impl_GetNextFreeBigBlock
2616 * Returns the index of the next free big block.
2617 * If the big block depot is filled, this method will enlarge it.
2620 static ULONG StorageImpl_GetNextFreeBigBlock(
2621 StorageImpl* This)
2623 ULONG depotBlockIndexPos;
2624 BYTE depotBuffer[BIG_BLOCK_SIZE];
2625 BOOL success;
2626 ULONG depotBlockOffset;
2627 ULONG blocksPerDepot = This->bigBlockSize / sizeof(ULONG);
2628 ULONG nextBlockIndex = BLOCK_SPECIAL;
2629 int depotIndex = 0;
2630 ULONG freeBlock = BLOCK_UNUSED;
2632 depotIndex = This->prevFreeBlock / blocksPerDepot;
2633 depotBlockOffset = (This->prevFreeBlock % blocksPerDepot) * sizeof(ULONG);
2636 * Scan the entire big block depot until we find a block marked free
2638 while (nextBlockIndex != BLOCK_UNUSED)
2640 if (depotIndex < COUNT_BBDEPOTINHEADER)
2642 depotBlockIndexPos = This->bigBlockDepotStart[depotIndex];
2645 * Grow the primary depot.
2647 if (depotBlockIndexPos == BLOCK_UNUSED)
2649 depotBlockIndexPos = depotIndex*blocksPerDepot;
2652 * Add a block depot.
2654 Storage32Impl_AddBlockDepot(This, depotBlockIndexPos);
2655 This->bigBlockDepotCount++;
2656 This->bigBlockDepotStart[depotIndex] = depotBlockIndexPos;
2659 * Flag it as a block depot.
2661 StorageImpl_SetNextBlockInChain(This,
2662 depotBlockIndexPos,
2663 BLOCK_SPECIAL);
2665 /* Save new header information.
2667 StorageImpl_SaveFileHeader(This);
2670 else
2672 depotBlockIndexPos = Storage32Impl_GetExtDepotBlock(This, depotIndex);
2674 if (depotBlockIndexPos == BLOCK_UNUSED)
2677 * Grow the extended depot.
2679 ULONG extIndex = BLOCK_UNUSED;
2680 ULONG numExtBlocks = depotIndex - COUNT_BBDEPOTINHEADER;
2681 ULONG extBlockOffset = numExtBlocks % (blocksPerDepot - 1);
2683 if (extBlockOffset == 0)
2685 /* We need an extended block.
2687 extIndex = Storage32Impl_AddExtBlockDepot(This);
2688 This->extBigBlockDepotCount++;
2689 depotBlockIndexPos = extIndex + 1;
2691 else
2692 depotBlockIndexPos = depotIndex * blocksPerDepot;
2695 * Add a block depot and mark it in the extended block.
2697 Storage32Impl_AddBlockDepot(This, depotBlockIndexPos);
2698 This->bigBlockDepotCount++;
2699 Storage32Impl_SetExtDepotBlock(This, depotIndex, depotBlockIndexPos);
2701 /* Flag the block depot.
2703 StorageImpl_SetNextBlockInChain(This,
2704 depotBlockIndexPos,
2705 BLOCK_SPECIAL);
2707 /* If necessary, flag the extended depot block.
2709 if (extIndex != BLOCK_UNUSED)
2710 StorageImpl_SetNextBlockInChain(This, extIndex, BLOCK_EXTBBDEPOT);
2712 /* Save header information.
2714 StorageImpl_SaveFileHeader(This);
2718 success = StorageImpl_ReadBigBlock(This, depotBlockIndexPos, depotBuffer);
2720 if (success)
2722 while ( ( (depotBlockOffset/sizeof(ULONG) ) < blocksPerDepot) &&
2723 ( nextBlockIndex != BLOCK_UNUSED))
2725 StorageUtl_ReadDWord(depotBuffer, depotBlockOffset, &nextBlockIndex);
2727 if (nextBlockIndex == BLOCK_UNUSED)
2729 freeBlock = (depotIndex * blocksPerDepot) +
2730 (depotBlockOffset/sizeof(ULONG));
2733 depotBlockOffset += sizeof(ULONG);
2737 depotIndex++;
2738 depotBlockOffset = 0;
2742 * make sure that the block physically exists before using it
2744 BIGBLOCKFILE_EnsureExists(This->bigBlockFile, freeBlock);
2746 This->prevFreeBlock = freeBlock;
2748 return freeBlock;
2751 /******************************************************************************
2752 * Storage32Impl_AddBlockDepot
2754 * This will create a depot block, essentially it is a block initialized
2755 * to BLOCK_UNUSEDs.
2757 static void Storage32Impl_AddBlockDepot(StorageImpl* This, ULONG blockIndex)
2759 BYTE blockBuffer[BIG_BLOCK_SIZE];
2762 * Initialize blocks as free
2764 memset(blockBuffer, BLOCK_UNUSED, This->bigBlockSize);
2765 StorageImpl_WriteBigBlock(This, blockIndex, blockBuffer);
2768 /******************************************************************************
2769 * Storage32Impl_GetExtDepotBlock
2771 * Returns the index of the block that corresponds to the specified depot
2772 * index. This method is only for depot indexes equal or greater than
2773 * COUNT_BBDEPOTINHEADER.
2775 static ULONG Storage32Impl_GetExtDepotBlock(StorageImpl* This, ULONG depotIndex)
2777 ULONG depotBlocksPerExtBlock = (This->bigBlockSize / sizeof(ULONG)) - 1;
2778 ULONG numExtBlocks = depotIndex - COUNT_BBDEPOTINHEADER;
2779 ULONG extBlockCount = numExtBlocks / depotBlocksPerExtBlock;
2780 ULONG extBlockOffset = numExtBlocks % depotBlocksPerExtBlock;
2781 ULONG blockIndex = BLOCK_UNUSED;
2782 ULONG extBlockIndex = This->extBigBlockDepotStart;
2784 assert(depotIndex >= COUNT_BBDEPOTINHEADER);
2786 if (This->extBigBlockDepotStart == BLOCK_END_OF_CHAIN)
2787 return BLOCK_UNUSED;
2789 while (extBlockCount > 0)
2791 extBlockIndex = Storage32Impl_GetNextExtendedBlock(This, extBlockIndex);
2792 extBlockCount--;
2795 if (extBlockIndex != BLOCK_UNUSED)
2796 StorageImpl_ReadDWordFromBigBlock(This, extBlockIndex,
2797 extBlockOffset * sizeof(ULONG), &blockIndex);
2799 return blockIndex;
2802 /******************************************************************************
2803 * Storage32Impl_SetExtDepotBlock
2805 * Associates the specified block index to the specified depot index.
2806 * This method is only for depot indexes equal or greater than
2807 * COUNT_BBDEPOTINHEADER.
2809 static void Storage32Impl_SetExtDepotBlock(StorageImpl* This, ULONG depotIndex, ULONG blockIndex)
2811 ULONG depotBlocksPerExtBlock = (This->bigBlockSize / sizeof(ULONG)) - 1;
2812 ULONG numExtBlocks = depotIndex - COUNT_BBDEPOTINHEADER;
2813 ULONG extBlockCount = numExtBlocks / depotBlocksPerExtBlock;
2814 ULONG extBlockOffset = numExtBlocks % depotBlocksPerExtBlock;
2815 ULONG extBlockIndex = This->extBigBlockDepotStart;
2817 assert(depotIndex >= COUNT_BBDEPOTINHEADER);
2819 while (extBlockCount > 0)
2821 extBlockIndex = Storage32Impl_GetNextExtendedBlock(This, extBlockIndex);
2822 extBlockCount--;
2825 if (extBlockIndex != BLOCK_UNUSED)
2827 StorageImpl_WriteDWordToBigBlock(This, extBlockIndex,
2828 extBlockOffset * sizeof(ULONG),
2829 blockIndex);
2833 /******************************************************************************
2834 * Storage32Impl_AddExtBlockDepot
2836 * Creates an extended depot block.
2838 static ULONG Storage32Impl_AddExtBlockDepot(StorageImpl* This)
2840 ULONG numExtBlocks = This->extBigBlockDepotCount;
2841 ULONG nextExtBlock = This->extBigBlockDepotStart;
2842 BYTE depotBuffer[BIG_BLOCK_SIZE];
2843 ULONG index = BLOCK_UNUSED;
2844 ULONG nextBlockOffset = This->bigBlockSize - sizeof(ULONG);
2845 ULONG blocksPerDepotBlock = This->bigBlockSize / sizeof(ULONG);
2846 ULONG depotBlocksPerExtBlock = blocksPerDepotBlock - 1;
2848 index = (COUNT_BBDEPOTINHEADER + (numExtBlocks * depotBlocksPerExtBlock)) *
2849 blocksPerDepotBlock;
2851 if ((numExtBlocks == 0) && (nextExtBlock == BLOCK_END_OF_CHAIN))
2854 * The first extended block.
2856 This->extBigBlockDepotStart = index;
2858 else
2860 unsigned int i;
2862 * Follow the chain to the last one.
2864 for (i = 0; i < (numExtBlocks - 1); i++)
2866 nextExtBlock = Storage32Impl_GetNextExtendedBlock(This, nextExtBlock);
2870 * Add the new extended block to the chain.
2872 StorageImpl_WriteDWordToBigBlock(This, nextExtBlock, nextBlockOffset,
2873 index);
2877 * Initialize this block.
2879 memset(depotBuffer, BLOCK_UNUSED, This->bigBlockSize);
2880 StorageImpl_WriteBigBlock(This, index, depotBuffer);
2882 return index;
2885 /******************************************************************************
2886 * Storage32Impl_FreeBigBlock
2888 * This method will flag the specified block as free in the big block depot.
2890 static void StorageImpl_FreeBigBlock(
2891 StorageImpl* This,
2892 ULONG blockIndex)
2894 StorageImpl_SetNextBlockInChain(This, blockIndex, BLOCK_UNUSED);
2896 if (blockIndex < This->prevFreeBlock)
2897 This->prevFreeBlock = blockIndex;
2900 /************************************************************************
2901 * Storage32Impl_GetNextBlockInChain
2903 * This method will retrieve the block index of the next big block in
2904 * in the chain.
2906 * Params: This - Pointer to the Storage object.
2907 * blockIndex - Index of the block to retrieve the chain
2908 * for.
2909 * nextBlockIndex - receives the return value.
2911 * Returns: This method returns the index of the next block in the chain.
2912 * It will return the constants:
2913 * BLOCK_SPECIAL - If the block given was not part of a
2914 * chain.
2915 * BLOCK_END_OF_CHAIN - If the block given was the last in
2916 * a chain.
2917 * BLOCK_UNUSED - If the block given was not past of a chain
2918 * and is available.
2919 * BLOCK_EXTBBDEPOT - This block is part of the extended
2920 * big block depot.
2922 * See Windows documentation for more details on IStorage methods.
2924 static HRESULT StorageImpl_GetNextBlockInChain(
2925 StorageImpl* This,
2926 ULONG blockIndex,
2927 ULONG* nextBlockIndex)
2929 ULONG offsetInDepot = blockIndex * sizeof (ULONG);
2930 ULONG depotBlockCount = offsetInDepot / This->bigBlockSize;
2931 ULONG depotBlockOffset = offsetInDepot % This->bigBlockSize;
2932 BYTE depotBuffer[BIG_BLOCK_SIZE];
2933 BOOL success;
2934 ULONG depotBlockIndexPos;
2935 int index;
2937 *nextBlockIndex = BLOCK_SPECIAL;
2939 if(depotBlockCount >= This->bigBlockDepotCount)
2941 WARN("depotBlockCount %d, bigBlockDepotCount %d\n", depotBlockCount,
2942 This->bigBlockDepotCount);
2943 return STG_E_READFAULT;
2947 * Cache the currently accessed depot block.
2949 if (depotBlockCount != This->indexBlockDepotCached)
2951 This->indexBlockDepotCached = depotBlockCount;
2953 if (depotBlockCount < COUNT_BBDEPOTINHEADER)
2955 depotBlockIndexPos = This->bigBlockDepotStart[depotBlockCount];
2957 else
2960 * We have to look in the extended depot.
2962 depotBlockIndexPos = Storage32Impl_GetExtDepotBlock(This, depotBlockCount);
2965 success = StorageImpl_ReadBigBlock(This, depotBlockIndexPos, depotBuffer);
2967 if (!success)
2968 return STG_E_READFAULT;
2970 for (index = 0; index < NUM_BLOCKS_PER_DEPOT_BLOCK; index++)
2972 StorageUtl_ReadDWord(depotBuffer, index*sizeof(ULONG), nextBlockIndex);
2973 This->blockDepotCached[index] = *nextBlockIndex;
2977 *nextBlockIndex = This->blockDepotCached[depotBlockOffset/sizeof(ULONG)];
2979 return S_OK;
2982 /******************************************************************************
2983 * Storage32Impl_GetNextExtendedBlock
2985 * Given an extended block this method will return the next extended block.
2987 * NOTES:
2988 * The last ULONG of an extended block is the block index of the next
2989 * extended block. Extended blocks are marked as BLOCK_EXTBBDEPOT in the
2990 * depot.
2992 * Return values:
2993 * - The index of the next extended block
2994 * - BLOCK_UNUSED: there is no next extended block.
2995 * - Any other return values denotes failure.
2997 static ULONG Storage32Impl_GetNextExtendedBlock(StorageImpl* This, ULONG blockIndex)
2999 ULONG nextBlockIndex = BLOCK_SPECIAL;
3000 ULONG depotBlockOffset = This->bigBlockSize - sizeof(ULONG);
3002 StorageImpl_ReadDWordFromBigBlock(This, blockIndex, depotBlockOffset,
3003 &nextBlockIndex);
3005 return nextBlockIndex;
3008 /******************************************************************************
3009 * Storage32Impl_SetNextBlockInChain
3011 * This method will write the index of the specified block's next block
3012 * in the big block depot.
3014 * For example: to create the chain 3 -> 1 -> 7 -> End of Chain
3015 * do the following
3017 * Storage32Impl_SetNextBlockInChain(This, 3, 1);
3018 * Storage32Impl_SetNextBlockInChain(This, 1, 7);
3019 * Storage32Impl_SetNextBlockInChain(This, 7, BLOCK_END_OF_CHAIN);
3022 static void StorageImpl_SetNextBlockInChain(
3023 StorageImpl* This,
3024 ULONG blockIndex,
3025 ULONG nextBlock)
3027 ULONG offsetInDepot = blockIndex * sizeof (ULONG);
3028 ULONG depotBlockCount = offsetInDepot / This->bigBlockSize;
3029 ULONG depotBlockOffset = offsetInDepot % This->bigBlockSize;
3030 ULONG depotBlockIndexPos;
3032 assert(depotBlockCount < This->bigBlockDepotCount);
3033 assert(blockIndex != nextBlock);
3035 if (depotBlockCount < COUNT_BBDEPOTINHEADER)
3037 depotBlockIndexPos = This->bigBlockDepotStart[depotBlockCount];
3039 else
3042 * We have to look in the extended depot.
3044 depotBlockIndexPos = Storage32Impl_GetExtDepotBlock(This, depotBlockCount);
3047 StorageImpl_WriteDWordToBigBlock(This, depotBlockIndexPos, depotBlockOffset,
3048 nextBlock);
3050 * Update the cached block depot, if necessary.
3052 if (depotBlockCount == This->indexBlockDepotCached)
3054 This->blockDepotCached[depotBlockOffset/sizeof(ULONG)] = nextBlock;
3058 /******************************************************************************
3059 * Storage32Impl_LoadFileHeader
3061 * This method will read in the file header, i.e. big block index -1.
3063 static HRESULT StorageImpl_LoadFileHeader(
3064 StorageImpl* This)
3066 HRESULT hr = STG_E_FILENOTFOUND;
3067 BYTE headerBigBlock[BIG_BLOCK_SIZE];
3068 BOOL success;
3069 int index;
3071 TRACE("\n");
3073 * Get a pointer to the big block of data containing the header.
3075 success = StorageImpl_ReadBigBlock(This, -1, headerBigBlock);
3078 * Extract the information from the header.
3080 if (success)
3083 * Check for the "magic number" signature and return an error if it is not
3084 * found.
3086 if (memcmp(headerBigBlock, STORAGE_oldmagic, sizeof(STORAGE_oldmagic))==0)
3088 return STG_E_OLDFORMAT;
3091 if (memcmp(headerBigBlock, STORAGE_magic, sizeof(STORAGE_magic))!=0)
3093 return STG_E_INVALIDHEADER;
3096 StorageUtl_ReadWord(
3097 headerBigBlock,
3098 OFFSET_BIGBLOCKSIZEBITS,
3099 &This->bigBlockSizeBits);
3101 StorageUtl_ReadWord(
3102 headerBigBlock,
3103 OFFSET_SMALLBLOCKSIZEBITS,
3104 &This->smallBlockSizeBits);
3106 StorageUtl_ReadDWord(
3107 headerBigBlock,
3108 OFFSET_BBDEPOTCOUNT,
3109 &This->bigBlockDepotCount);
3111 StorageUtl_ReadDWord(
3112 headerBigBlock,
3113 OFFSET_ROOTSTARTBLOCK,
3114 &This->rootStartBlock);
3116 StorageUtl_ReadDWord(
3117 headerBigBlock,
3118 OFFSET_SBDEPOTSTART,
3119 &This->smallBlockDepotStart);
3121 StorageUtl_ReadDWord(
3122 headerBigBlock,
3123 OFFSET_EXTBBDEPOTSTART,
3124 &This->extBigBlockDepotStart);
3126 StorageUtl_ReadDWord(
3127 headerBigBlock,
3128 OFFSET_EXTBBDEPOTCOUNT,
3129 &This->extBigBlockDepotCount);
3131 for (index = 0; index < COUNT_BBDEPOTINHEADER; index ++)
3133 StorageUtl_ReadDWord(
3134 headerBigBlock,
3135 OFFSET_BBDEPOTSTART + (sizeof(ULONG)*index),
3136 &(This->bigBlockDepotStart[index]));
3140 * Make the bitwise arithmetic to get the size of the blocks in bytes.
3142 This->bigBlockSize = 0x000000001 << (DWORD)This->bigBlockSizeBits;
3143 This->smallBlockSize = 0x000000001 << (DWORD)This->smallBlockSizeBits;
3146 * Right now, the code is making some assumptions about the size of the
3147 * blocks, just make sure they are what we're expecting.
3149 if (This->bigBlockSize != DEF_BIG_BLOCK_SIZE ||
3150 This->smallBlockSize != DEF_SMALL_BLOCK_SIZE)
3152 WARN("Broken OLE storage file\n");
3153 hr = STG_E_INVALIDHEADER;
3155 else
3156 hr = S_OK;
3159 return hr;
3162 /******************************************************************************
3163 * Storage32Impl_SaveFileHeader
3165 * This method will save to the file the header, i.e. big block -1.
3167 static void StorageImpl_SaveFileHeader(
3168 StorageImpl* This)
3170 BYTE headerBigBlock[BIG_BLOCK_SIZE];
3171 int index;
3172 BOOL success;
3175 * Get a pointer to the big block of data containing the header.
3177 success = StorageImpl_ReadBigBlock(This, -1, headerBigBlock);
3180 * If the block read failed, the file is probably new.
3182 if (!success)
3185 * Initialize for all unknown fields.
3187 memset(headerBigBlock, 0, BIG_BLOCK_SIZE);
3190 * Initialize the magic number.
3192 memcpy(headerBigBlock, STORAGE_magic, sizeof(STORAGE_magic));
3195 * And a bunch of things we don't know what they mean
3197 StorageUtl_WriteWord(headerBigBlock, 0x18, 0x3b);
3198 StorageUtl_WriteWord(headerBigBlock, 0x1a, 0x3);
3199 StorageUtl_WriteWord(headerBigBlock, 0x1c, (WORD)-2);
3200 StorageUtl_WriteDWord(headerBigBlock, 0x38, (DWORD)0x1000);
3204 * Write the information to the header.
3206 StorageUtl_WriteWord(
3207 headerBigBlock,
3208 OFFSET_BIGBLOCKSIZEBITS,
3209 This->bigBlockSizeBits);
3211 StorageUtl_WriteWord(
3212 headerBigBlock,
3213 OFFSET_SMALLBLOCKSIZEBITS,
3214 This->smallBlockSizeBits);
3216 StorageUtl_WriteDWord(
3217 headerBigBlock,
3218 OFFSET_BBDEPOTCOUNT,
3219 This->bigBlockDepotCount);
3221 StorageUtl_WriteDWord(
3222 headerBigBlock,
3223 OFFSET_ROOTSTARTBLOCK,
3224 This->rootStartBlock);
3226 StorageUtl_WriteDWord(
3227 headerBigBlock,
3228 OFFSET_SBDEPOTSTART,
3229 This->smallBlockDepotStart);
3231 StorageUtl_WriteDWord(
3232 headerBigBlock,
3233 OFFSET_SBDEPOTCOUNT,
3234 This->smallBlockDepotChain ?
3235 BlockChainStream_GetCount(This->smallBlockDepotChain) : 0);
3237 StorageUtl_WriteDWord(
3238 headerBigBlock,
3239 OFFSET_EXTBBDEPOTSTART,
3240 This->extBigBlockDepotStart);
3242 StorageUtl_WriteDWord(
3243 headerBigBlock,
3244 OFFSET_EXTBBDEPOTCOUNT,
3245 This->extBigBlockDepotCount);
3247 for (index = 0; index < COUNT_BBDEPOTINHEADER; index ++)
3249 StorageUtl_WriteDWord(
3250 headerBigBlock,
3251 OFFSET_BBDEPOTSTART + (sizeof(ULONG)*index),
3252 (This->bigBlockDepotStart[index]));
3256 * Write the big block back to the file.
3258 StorageImpl_WriteBigBlock(This, -1, headerBigBlock);
3261 /******************************************************************************
3262 * Storage32Impl_ReadProperty
3264 * This method will read the specified property from the property chain.
3266 BOOL StorageImpl_ReadProperty(
3267 StorageImpl* This,
3268 ULONG index,
3269 StgProperty* buffer)
3271 BYTE currentProperty[PROPSET_BLOCK_SIZE];
3272 ULARGE_INTEGER offsetInPropSet;
3273 HRESULT readRes;
3274 ULONG bytesRead;
3276 offsetInPropSet.u.HighPart = 0;
3277 offsetInPropSet.u.LowPart = index * PROPSET_BLOCK_SIZE;
3279 readRes = BlockChainStream_ReadAt(
3280 This->rootBlockChain,
3281 offsetInPropSet,
3282 PROPSET_BLOCK_SIZE,
3283 currentProperty,
3284 &bytesRead);
3286 if (SUCCEEDED(readRes))
3288 /* replace the name of root entry (often "Root Entry") by the file name */
3289 WCHAR *propName = (index == This->base.rootPropertySetIndex) ?
3290 This->filename : (WCHAR *)currentProperty+OFFSET_PS_NAME;
3292 memset(buffer->name, 0, sizeof(buffer->name));
3293 memcpy(
3294 buffer->name,
3295 propName,
3296 PROPERTY_NAME_BUFFER_LEN );
3297 TRACE("storage name: %s\n", debugstr_w(buffer->name));
3299 memcpy(&buffer->propertyType, currentProperty + OFFSET_PS_PROPERTYTYPE, 1);
3301 StorageUtl_ReadWord(
3302 currentProperty,
3303 OFFSET_PS_NAMELENGTH,
3304 &buffer->sizeOfNameString);
3306 StorageUtl_ReadDWord(
3307 currentProperty,
3308 OFFSET_PS_PREVIOUSPROP,
3309 &buffer->previousProperty);
3311 StorageUtl_ReadDWord(
3312 currentProperty,
3313 OFFSET_PS_NEXTPROP,
3314 &buffer->nextProperty);
3316 StorageUtl_ReadDWord(
3317 currentProperty,
3318 OFFSET_PS_DIRPROP,
3319 &buffer->dirProperty);
3321 StorageUtl_ReadGUID(
3322 currentProperty,
3323 OFFSET_PS_GUID,
3324 &buffer->propertyUniqueID);
3326 StorageUtl_ReadDWord(
3327 currentProperty,
3328 OFFSET_PS_TSS1,
3329 &buffer->timeStampS1);
3331 StorageUtl_ReadDWord(
3332 currentProperty,
3333 OFFSET_PS_TSD1,
3334 &buffer->timeStampD1);
3336 StorageUtl_ReadDWord(
3337 currentProperty,
3338 OFFSET_PS_TSS2,
3339 &buffer->timeStampS2);
3341 StorageUtl_ReadDWord(
3342 currentProperty,
3343 OFFSET_PS_TSD2,
3344 &buffer->timeStampD2);
3346 StorageUtl_ReadDWord(
3347 currentProperty,
3348 OFFSET_PS_STARTBLOCK,
3349 &buffer->startingBlock);
3351 StorageUtl_ReadDWord(
3352 currentProperty,
3353 OFFSET_PS_SIZE,
3354 &buffer->size.u.LowPart);
3356 buffer->size.u.HighPart = 0;
3359 return SUCCEEDED(readRes) ? TRUE : FALSE;
3362 /*********************************************************************
3363 * Write the specified property into the property chain
3365 BOOL StorageImpl_WriteProperty(
3366 StorageImpl* This,
3367 ULONG index,
3368 const StgProperty* buffer)
3370 BYTE currentProperty[PROPSET_BLOCK_SIZE];
3371 ULARGE_INTEGER offsetInPropSet;
3372 HRESULT writeRes;
3373 ULONG bytesWritten;
3375 offsetInPropSet.u.HighPart = 0;
3376 offsetInPropSet.u.LowPart = index * PROPSET_BLOCK_SIZE;
3378 memset(currentProperty, 0, PROPSET_BLOCK_SIZE);
3380 memcpy(
3381 currentProperty + OFFSET_PS_NAME,
3382 buffer->name,
3383 PROPERTY_NAME_BUFFER_LEN );
3385 memcpy(currentProperty + OFFSET_PS_PROPERTYTYPE, &buffer->propertyType, 1);
3387 StorageUtl_WriteWord(
3388 currentProperty,
3389 OFFSET_PS_NAMELENGTH,
3390 buffer->sizeOfNameString);
3392 StorageUtl_WriteDWord(
3393 currentProperty,
3394 OFFSET_PS_PREVIOUSPROP,
3395 buffer->previousProperty);
3397 StorageUtl_WriteDWord(
3398 currentProperty,
3399 OFFSET_PS_NEXTPROP,
3400 buffer->nextProperty);
3402 StorageUtl_WriteDWord(
3403 currentProperty,
3404 OFFSET_PS_DIRPROP,
3405 buffer->dirProperty);
3407 StorageUtl_WriteGUID(
3408 currentProperty,
3409 OFFSET_PS_GUID,
3410 &buffer->propertyUniqueID);
3412 StorageUtl_WriteDWord(
3413 currentProperty,
3414 OFFSET_PS_TSS1,
3415 buffer->timeStampS1);
3417 StorageUtl_WriteDWord(
3418 currentProperty,
3419 OFFSET_PS_TSD1,
3420 buffer->timeStampD1);
3422 StorageUtl_WriteDWord(
3423 currentProperty,
3424 OFFSET_PS_TSS2,
3425 buffer->timeStampS2);
3427 StorageUtl_WriteDWord(
3428 currentProperty,
3429 OFFSET_PS_TSD2,
3430 buffer->timeStampD2);
3432 StorageUtl_WriteDWord(
3433 currentProperty,
3434 OFFSET_PS_STARTBLOCK,
3435 buffer->startingBlock);
3437 StorageUtl_WriteDWord(
3438 currentProperty,
3439 OFFSET_PS_SIZE,
3440 buffer->size.u.LowPart);
3442 writeRes = BlockChainStream_WriteAt(This->rootBlockChain,
3443 offsetInPropSet,
3444 PROPSET_BLOCK_SIZE,
3445 currentProperty,
3446 &bytesWritten);
3447 return SUCCEEDED(writeRes) ? TRUE : FALSE;
3450 static BOOL StorageImpl_ReadBigBlock(
3451 StorageImpl* This,
3452 ULONG blockIndex,
3453 void* buffer)
3455 ULARGE_INTEGER ulOffset;
3456 DWORD read;
3458 ulOffset.u.HighPart = 0;
3459 ulOffset.u.LowPart = BLOCK_GetBigBlockOffset(blockIndex);
3461 StorageImpl_ReadAt(This, ulOffset, buffer, This->bigBlockSize, &read);
3462 return (read == This->bigBlockSize);
3465 static BOOL StorageImpl_ReadDWordFromBigBlock(
3466 StorageImpl* This,
3467 ULONG blockIndex,
3468 ULONG offset,
3469 DWORD* value)
3471 ULARGE_INTEGER ulOffset;
3472 DWORD read;
3473 DWORD tmp;
3475 ulOffset.u.HighPart = 0;
3476 ulOffset.u.LowPart = BLOCK_GetBigBlockOffset(blockIndex);
3477 ulOffset.u.LowPart += offset;
3479 StorageImpl_ReadAt(This, ulOffset, &tmp, sizeof(DWORD), &read);
3480 *value = le32toh(tmp);
3481 return (read == sizeof(DWORD));
3484 static BOOL StorageImpl_WriteBigBlock(
3485 StorageImpl* This,
3486 ULONG blockIndex,
3487 const void* buffer)
3489 ULARGE_INTEGER ulOffset;
3490 DWORD wrote;
3492 ulOffset.u.HighPart = 0;
3493 ulOffset.u.LowPart = BLOCK_GetBigBlockOffset(blockIndex);
3495 StorageImpl_WriteAt(This, ulOffset, buffer, This->bigBlockSize, &wrote);
3496 return (wrote == This->bigBlockSize);
3499 static BOOL StorageImpl_WriteDWordToBigBlock(
3500 StorageImpl* This,
3501 ULONG blockIndex,
3502 ULONG offset,
3503 DWORD value)
3505 ULARGE_INTEGER ulOffset;
3506 DWORD wrote;
3508 ulOffset.u.HighPart = 0;
3509 ulOffset.u.LowPart = BLOCK_GetBigBlockOffset(blockIndex);
3510 ulOffset.u.LowPart += offset;
3512 value = htole32(value);
3513 StorageImpl_WriteAt(This, ulOffset, &value, sizeof(DWORD), &wrote);
3514 return (wrote == sizeof(DWORD));
3517 /******************************************************************************
3518 * Storage32Impl_SmallBlocksToBigBlocks
3520 * This method will convert a small block chain to a big block chain.
3521 * The small block chain will be destroyed.
3523 BlockChainStream* Storage32Impl_SmallBlocksToBigBlocks(
3524 StorageImpl* This,
3525 SmallBlockChainStream** ppsbChain)
3527 ULONG bbHeadOfChain = BLOCK_END_OF_CHAIN;
3528 ULARGE_INTEGER size, offset;
3529 ULONG cbRead, cbWritten;
3530 ULARGE_INTEGER cbTotalRead;
3531 ULONG propertyIndex;
3532 HRESULT resWrite = S_OK;
3533 HRESULT resRead;
3534 StgProperty chainProperty;
3535 BYTE *buffer;
3536 BlockChainStream *bbTempChain = NULL;
3537 BlockChainStream *bigBlockChain = NULL;
3540 * Create a temporary big block chain that doesn't have
3541 * an associated property. This temporary chain will be
3542 * used to copy data from small blocks to big blocks.
3544 bbTempChain = BlockChainStream_Construct(This,
3545 &bbHeadOfChain,
3546 PROPERTY_NULL);
3547 if(!bbTempChain) return NULL;
3549 * Grow the big block chain.
3551 size = SmallBlockChainStream_GetSize(*ppsbChain);
3552 BlockChainStream_SetSize(bbTempChain, size);
3555 * Copy the contents of the small block chain to the big block chain
3556 * by small block size increments.
3558 offset.u.LowPart = 0;
3559 offset.u.HighPart = 0;
3560 cbTotalRead.QuadPart = 0;
3562 buffer = HeapAlloc(GetProcessHeap(),0,DEF_SMALL_BLOCK_SIZE);
3565 resRead = SmallBlockChainStream_ReadAt(*ppsbChain,
3566 offset,
3567 This->smallBlockSize,
3568 buffer,
3569 &cbRead);
3570 if (FAILED(resRead))
3571 break;
3573 if (cbRead > 0)
3575 cbTotalRead.QuadPart += cbRead;
3577 resWrite = BlockChainStream_WriteAt(bbTempChain,
3578 offset,
3579 cbRead,
3580 buffer,
3581 &cbWritten);
3583 if (FAILED(resWrite))
3584 break;
3586 offset.u.LowPart += This->smallBlockSize;
3588 } while (cbTotalRead.QuadPart < size.QuadPart);
3589 HeapFree(GetProcessHeap(),0,buffer);
3591 if (FAILED(resRead) || FAILED(resWrite))
3593 ERR("conversion failed: resRead = 0x%08x, resWrite = 0x%08x\n", resRead, resWrite);
3594 BlockChainStream_Destroy(bbTempChain);
3595 return NULL;
3599 * Destroy the small block chain.
3601 propertyIndex = (*ppsbChain)->ownerPropertyIndex;
3602 size.u.HighPart = 0;
3603 size.u.LowPart = 0;
3604 SmallBlockChainStream_SetSize(*ppsbChain, size);
3605 SmallBlockChainStream_Destroy(*ppsbChain);
3606 *ppsbChain = 0;
3609 * Change the property information. This chain is now a big block chain
3610 * and it doesn't reside in the small blocks chain anymore.
3612 StorageImpl_ReadProperty(This, propertyIndex, &chainProperty);
3614 chainProperty.startingBlock = bbHeadOfChain;
3616 StorageImpl_WriteProperty(This, propertyIndex, &chainProperty);
3619 * Destroy the temporary propertyless big block chain.
3620 * Create a new big block chain associated with this property.
3622 BlockChainStream_Destroy(bbTempChain);
3623 bigBlockChain = BlockChainStream_Construct(This,
3624 NULL,
3625 propertyIndex);
3627 return bigBlockChain;
3630 static void StorageInternalImpl_Destroy( StorageBaseImpl *iface)
3632 StorageInternalImpl* This = (StorageInternalImpl*) iface;
3634 StorageBaseImpl_Release((IStorage*)This->base.ancestorStorage);
3635 HeapFree(GetProcessHeap(), 0, This);
3638 /******************************************************************************
3640 ** Storage32InternalImpl_Commit
3642 ** The non-root storages cannot be opened in transacted mode thus this function
3643 ** does nothing.
3645 static HRESULT WINAPI StorageInternalImpl_Commit(
3646 IStorage* iface,
3647 DWORD grfCommitFlags) /* [in] */
3649 return S_OK;
3652 /******************************************************************************
3654 ** Storage32InternalImpl_Revert
3656 ** The non-root storages cannot be opened in transacted mode thus this function
3657 ** does nothing.
3659 static HRESULT WINAPI StorageInternalImpl_Revert(
3660 IStorage* iface)
3662 return S_OK;
3665 static void IEnumSTATSTGImpl_Destroy(IEnumSTATSTGImpl* This)
3667 IStorage_Release((IStorage*)This->parentStorage);
3668 HeapFree(GetProcessHeap(), 0, This->stackToVisit);
3669 HeapFree(GetProcessHeap(), 0, This);
3672 static HRESULT WINAPI IEnumSTATSTGImpl_QueryInterface(
3673 IEnumSTATSTG* iface,
3674 REFIID riid,
3675 void** ppvObject)
3677 IEnumSTATSTGImpl* const This=(IEnumSTATSTGImpl*)iface;
3680 * Perform a sanity check on the parameters.
3682 if (ppvObject==0)
3683 return E_INVALIDARG;
3686 * Initialize the return parameter.
3688 *ppvObject = 0;
3691 * Compare the riid with the interface IDs implemented by this object.
3693 if (IsEqualGUID(&IID_IUnknown, riid) ||
3694 IsEqualGUID(&IID_IEnumSTATSTG, riid))
3696 *ppvObject = (IEnumSTATSTG*)This;
3697 IEnumSTATSTG_AddRef((IEnumSTATSTG*)This);
3698 return S_OK;
3701 return E_NOINTERFACE;
3704 static ULONG WINAPI IEnumSTATSTGImpl_AddRef(
3705 IEnumSTATSTG* iface)
3707 IEnumSTATSTGImpl* const This=(IEnumSTATSTGImpl*)iface;
3708 return InterlockedIncrement(&This->ref);
3711 static ULONG WINAPI IEnumSTATSTGImpl_Release(
3712 IEnumSTATSTG* iface)
3714 IEnumSTATSTGImpl* const This=(IEnumSTATSTGImpl*)iface;
3716 ULONG newRef;
3718 newRef = InterlockedDecrement(&This->ref);
3721 * If the reference count goes down to 0, perform suicide.
3723 if (newRef==0)
3725 IEnumSTATSTGImpl_Destroy(This);
3728 return newRef;
3731 static HRESULT WINAPI IEnumSTATSTGImpl_Next(
3732 IEnumSTATSTG* iface,
3733 ULONG celt,
3734 STATSTG* rgelt,
3735 ULONG* pceltFetched)
3737 IEnumSTATSTGImpl* const This=(IEnumSTATSTGImpl*)iface;
3739 StgProperty currentProperty;
3740 STATSTG* currentReturnStruct = rgelt;
3741 ULONG objectFetched = 0;
3742 ULONG currentSearchNode;
3745 * Perform a sanity check on the parameters.
3747 if ( (rgelt==0) || ( (celt!=1) && (pceltFetched==0) ) )
3748 return E_INVALIDARG;
3751 * To avoid the special case, get another pointer to a ULONG value if
3752 * the caller didn't supply one.
3754 if (pceltFetched==0)
3755 pceltFetched = &objectFetched;
3758 * Start the iteration, we will iterate until we hit the end of the
3759 * linked list or until we hit the number of items to iterate through
3761 *pceltFetched = 0;
3764 * Start with the node at the top of the stack.
3766 currentSearchNode = IEnumSTATSTGImpl_PopSearchNode(This, FALSE);
3768 while ( ( *pceltFetched < celt) &&
3769 ( currentSearchNode!=PROPERTY_NULL) )
3772 * Remove the top node from the stack
3774 IEnumSTATSTGImpl_PopSearchNode(This, TRUE);
3777 * Read the property from the storage.
3779 StorageImpl_ReadProperty(This->parentStorage,
3780 currentSearchNode,
3781 &currentProperty);
3784 * Copy the information to the return buffer.
3786 StorageUtl_CopyPropertyToSTATSTG(currentReturnStruct,
3787 &currentProperty,
3788 STATFLAG_DEFAULT);
3791 * Step to the next item in the iteration
3793 (*pceltFetched)++;
3794 currentReturnStruct++;
3797 * Push the next search node in the search stack.
3799 IEnumSTATSTGImpl_PushSearchNode(This, currentProperty.nextProperty);
3802 * continue the iteration.
3804 currentSearchNode = IEnumSTATSTGImpl_PopSearchNode(This, FALSE);
3807 if (*pceltFetched == celt)
3808 return S_OK;
3810 return S_FALSE;
3814 static HRESULT WINAPI IEnumSTATSTGImpl_Skip(
3815 IEnumSTATSTG* iface,
3816 ULONG celt)
3818 IEnumSTATSTGImpl* const This=(IEnumSTATSTGImpl*)iface;
3820 StgProperty currentProperty;
3821 ULONG objectFetched = 0;
3822 ULONG currentSearchNode;
3825 * Start with the node at the top of the stack.
3827 currentSearchNode = IEnumSTATSTGImpl_PopSearchNode(This, FALSE);
3829 while ( (objectFetched < celt) &&
3830 (currentSearchNode!=PROPERTY_NULL) )
3833 * Remove the top node from the stack
3835 IEnumSTATSTGImpl_PopSearchNode(This, TRUE);
3838 * Read the property from the storage.
3840 StorageImpl_ReadProperty(This->parentStorage,
3841 currentSearchNode,
3842 &currentProperty);
3845 * Step to the next item in the iteration
3847 objectFetched++;
3850 * Push the next search node in the search stack.
3852 IEnumSTATSTGImpl_PushSearchNode(This, currentProperty.nextProperty);
3855 * continue the iteration.
3857 currentSearchNode = IEnumSTATSTGImpl_PopSearchNode(This, FALSE);
3860 if (objectFetched == celt)
3861 return S_OK;
3863 return S_FALSE;
3866 static HRESULT WINAPI IEnumSTATSTGImpl_Reset(
3867 IEnumSTATSTG* iface)
3869 IEnumSTATSTGImpl* const This=(IEnumSTATSTGImpl*)iface;
3871 StgProperty rootProperty;
3872 BOOL readSuccessful;
3875 * Re-initialize the search stack to an empty stack
3877 This->stackSize = 0;
3880 * Read the root property from the storage.
3882 readSuccessful = StorageImpl_ReadProperty(
3883 This->parentStorage,
3884 This->firstPropertyNode,
3885 &rootProperty);
3887 if (readSuccessful)
3889 assert(rootProperty.sizeOfNameString!=0);
3892 * Push the search node in the search stack.
3894 IEnumSTATSTGImpl_PushSearchNode(This, rootProperty.dirProperty);
3897 return S_OK;
3900 static HRESULT WINAPI IEnumSTATSTGImpl_Clone(
3901 IEnumSTATSTG* iface,
3902 IEnumSTATSTG** ppenum)
3904 IEnumSTATSTGImpl* const This=(IEnumSTATSTGImpl*)iface;
3906 IEnumSTATSTGImpl* newClone;
3909 * Perform a sanity check on the parameters.
3911 if (ppenum==0)
3912 return E_INVALIDARG;
3914 newClone = IEnumSTATSTGImpl_Construct(This->parentStorage,
3915 This->firstPropertyNode);
3919 * The new clone enumeration must point to the same current node as
3920 * the ole one.
3922 newClone->stackSize = This->stackSize ;
3923 newClone->stackMaxSize = This->stackMaxSize ;
3924 newClone->stackToVisit =
3925 HeapAlloc(GetProcessHeap(), 0, sizeof(ULONG) * newClone->stackMaxSize);
3927 memcpy(
3928 newClone->stackToVisit,
3929 This->stackToVisit,
3930 sizeof(ULONG) * newClone->stackSize);
3932 *ppenum = (IEnumSTATSTG*)newClone;
3935 * Don't forget to nail down a reference to the clone before
3936 * returning it.
3938 IEnumSTATSTGImpl_AddRef(*ppenum);
3940 return S_OK;
3943 static INT IEnumSTATSTGImpl_FindParentProperty(
3944 IEnumSTATSTGImpl *This,
3945 ULONG childProperty,
3946 StgProperty *currentProperty,
3947 ULONG *thisNodeId)
3949 ULONG currentSearchNode;
3950 ULONG foundNode;
3953 * To avoid the special case, get another pointer to a ULONG value if
3954 * the caller didn't supply one.
3956 if (thisNodeId==0)
3957 thisNodeId = &foundNode;
3960 * Start with the node at the top of the stack.
3962 currentSearchNode = IEnumSTATSTGImpl_PopSearchNode(This, FALSE);
3965 while (currentSearchNode!=PROPERTY_NULL)
3968 * Store the current node in the returned parameters
3970 *thisNodeId = currentSearchNode;
3973 * Remove the top node from the stack
3975 IEnumSTATSTGImpl_PopSearchNode(This, TRUE);
3978 * Read the property from the storage.
3980 StorageImpl_ReadProperty(
3981 This->parentStorage,
3982 currentSearchNode,
3983 currentProperty);
3985 if (currentProperty->previousProperty == childProperty)
3986 return PROPERTY_RELATION_PREVIOUS;
3988 else if (currentProperty->nextProperty == childProperty)
3989 return PROPERTY_RELATION_NEXT;
3991 else if (currentProperty->dirProperty == childProperty)
3992 return PROPERTY_RELATION_DIR;
3995 * Push the next search node in the search stack.
3997 IEnumSTATSTGImpl_PushSearchNode(This, currentProperty->nextProperty);
4000 * continue the iteration.
4002 currentSearchNode = IEnumSTATSTGImpl_PopSearchNode(This, FALSE);
4005 return PROPERTY_NULL;
4008 static ULONG IEnumSTATSTGImpl_FindProperty(
4009 IEnumSTATSTGImpl* This,
4010 const OLECHAR* lpszPropName,
4011 StgProperty* currentProperty)
4013 ULONG currentSearchNode;
4016 * Start with the node at the top of the stack.
4018 currentSearchNode = IEnumSTATSTGImpl_PopSearchNode(This, FALSE);
4020 while (currentSearchNode!=PROPERTY_NULL)
4023 * Remove the top node from the stack
4025 IEnumSTATSTGImpl_PopSearchNode(This, TRUE);
4028 * Read the property from the storage.
4030 StorageImpl_ReadProperty(This->parentStorage,
4031 currentSearchNode,
4032 currentProperty);
4034 if ( propertyNameCmp(
4035 (const OLECHAR*)currentProperty->name, lpszPropName) == 0)
4036 return currentSearchNode;
4039 * Push the next search node in the search stack.
4041 IEnumSTATSTGImpl_PushSearchNode(This, currentProperty->nextProperty);
4044 * continue the iteration.
4046 currentSearchNode = IEnumSTATSTGImpl_PopSearchNode(This, FALSE);
4049 return PROPERTY_NULL;
4052 static void IEnumSTATSTGImpl_PushSearchNode(
4053 IEnumSTATSTGImpl* This,
4054 ULONG nodeToPush)
4056 StgProperty rootProperty;
4057 BOOL readSuccessful;
4060 * First, make sure we're not trying to push an unexisting node.
4062 if (nodeToPush==PROPERTY_NULL)
4063 return;
4066 * First push the node to the stack
4068 if (This->stackSize == This->stackMaxSize)
4070 This->stackMaxSize += ENUMSTATSGT_SIZE_INCREMENT;
4072 This->stackToVisit = HeapReAlloc(
4073 GetProcessHeap(),
4075 This->stackToVisit,
4076 sizeof(ULONG) * This->stackMaxSize);
4079 This->stackToVisit[This->stackSize] = nodeToPush;
4080 This->stackSize++;
4083 * Read the root property from the storage.
4085 readSuccessful = StorageImpl_ReadProperty(
4086 This->parentStorage,
4087 nodeToPush,
4088 &rootProperty);
4090 if (readSuccessful)
4092 assert(rootProperty.sizeOfNameString!=0);
4095 * Push the previous search node in the search stack.
4097 IEnumSTATSTGImpl_PushSearchNode(This, rootProperty.previousProperty);
4101 static ULONG IEnumSTATSTGImpl_PopSearchNode(
4102 IEnumSTATSTGImpl* This,
4103 BOOL remove)
4105 ULONG topNode;
4107 if (This->stackSize == 0)
4108 return PROPERTY_NULL;
4110 topNode = This->stackToVisit[This->stackSize-1];
4112 if (remove)
4113 This->stackSize--;
4115 return topNode;
4119 * Virtual function table for the IEnumSTATSTGImpl class.
4121 static const IEnumSTATSTGVtbl IEnumSTATSTGImpl_Vtbl =
4123 IEnumSTATSTGImpl_QueryInterface,
4124 IEnumSTATSTGImpl_AddRef,
4125 IEnumSTATSTGImpl_Release,
4126 IEnumSTATSTGImpl_Next,
4127 IEnumSTATSTGImpl_Skip,
4128 IEnumSTATSTGImpl_Reset,
4129 IEnumSTATSTGImpl_Clone
4132 /******************************************************************************
4133 ** IEnumSTATSTGImpl implementation
4136 static IEnumSTATSTGImpl* IEnumSTATSTGImpl_Construct(
4137 StorageImpl* parentStorage,
4138 ULONG firstPropertyNode)
4140 IEnumSTATSTGImpl* newEnumeration;
4142 newEnumeration = HeapAlloc(GetProcessHeap(), 0, sizeof(IEnumSTATSTGImpl));
4144 if (newEnumeration!=0)
4147 * Set-up the virtual function table and reference count.
4149 newEnumeration->lpVtbl = &IEnumSTATSTGImpl_Vtbl;
4150 newEnumeration->ref = 0;
4153 * We want to nail-down the reference to the storage in case the
4154 * enumeration out-lives the storage in the client application.
4156 newEnumeration->parentStorage = parentStorage;
4157 IStorage_AddRef((IStorage*)newEnumeration->parentStorage);
4159 newEnumeration->firstPropertyNode = firstPropertyNode;
4162 * Initialize the search stack
4164 newEnumeration->stackSize = 0;
4165 newEnumeration->stackMaxSize = ENUMSTATSGT_SIZE_INCREMENT;
4166 newEnumeration->stackToVisit =
4167 HeapAlloc(GetProcessHeap(), 0, sizeof(ULONG)*ENUMSTATSGT_SIZE_INCREMENT);
4170 * Make sure the current node of the iterator is the first one.
4172 IEnumSTATSTGImpl_Reset((IEnumSTATSTG*)newEnumeration);
4175 return newEnumeration;
4179 * Virtual function table for the Storage32InternalImpl class.
4181 static const IStorageVtbl Storage32InternalImpl_Vtbl =
4183 StorageBaseImpl_QueryInterface,
4184 StorageBaseImpl_AddRef,
4185 StorageBaseImpl_Release,
4186 StorageBaseImpl_CreateStream,
4187 StorageBaseImpl_OpenStream,
4188 StorageImpl_CreateStorage,
4189 StorageBaseImpl_OpenStorage,
4190 StorageImpl_CopyTo,
4191 StorageImpl_MoveElementTo,
4192 StorageInternalImpl_Commit,
4193 StorageInternalImpl_Revert,
4194 StorageBaseImpl_EnumElements,
4195 StorageImpl_DestroyElement,
4196 StorageBaseImpl_RenameElement,
4197 StorageImpl_SetElementTimes,
4198 StorageBaseImpl_SetClass,
4199 StorageImpl_SetStateBits,
4200 StorageBaseImpl_Stat
4203 /******************************************************************************
4204 ** Storage32InternalImpl implementation
4207 static StorageInternalImpl* StorageInternalImpl_Construct(
4208 StorageImpl* ancestorStorage,
4209 DWORD openFlags,
4210 ULONG rootPropertyIndex)
4212 StorageInternalImpl* newStorage;
4215 * Allocate space for the new storage object
4217 newStorage = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(StorageInternalImpl));
4219 if (newStorage!=0)
4222 * Initialize the stream list
4224 list_init(&newStorage->base.strmHead);
4227 * Initialize the virtual function table.
4229 newStorage->base.lpVtbl = &Storage32InternalImpl_Vtbl;
4230 newStorage->base.v_destructor = StorageInternalImpl_Destroy;
4231 newStorage->base.openFlags = (openFlags & ~STGM_CREATE);
4234 * Keep the ancestor storage pointer and nail a reference to it.
4236 newStorage->base.ancestorStorage = ancestorStorage;
4237 StorageBaseImpl_AddRef((IStorage*)(newStorage->base.ancestorStorage));
4240 * Keep the index of the root property set for this storage,
4242 newStorage->base.rootPropertySetIndex = rootPropertyIndex;
4244 return newStorage;
4247 return 0;
4250 /******************************************************************************
4251 ** StorageUtl implementation
4254 void StorageUtl_ReadWord(const BYTE* buffer, ULONG offset, WORD* value)
4256 WORD tmp;
4258 memcpy(&tmp, buffer+offset, sizeof(WORD));
4259 *value = le16toh(tmp);
4262 void StorageUtl_WriteWord(BYTE* buffer, ULONG offset, WORD value)
4264 value = htole16(value);
4265 memcpy(buffer+offset, &value, sizeof(WORD));
4268 void StorageUtl_ReadDWord(const BYTE* buffer, ULONG offset, DWORD* value)
4270 DWORD tmp;
4272 memcpy(&tmp, buffer+offset, sizeof(DWORD));
4273 *value = le32toh(tmp);
4276 void StorageUtl_WriteDWord(BYTE* buffer, ULONG offset, DWORD value)
4278 value = htole32(value);
4279 memcpy(buffer+offset, &value, sizeof(DWORD));
4282 void StorageUtl_ReadULargeInteger(const BYTE* buffer, ULONG offset,
4283 ULARGE_INTEGER* value)
4285 #ifdef WORDS_BIGENDIAN
4286 ULARGE_INTEGER tmp;
4288 memcpy(&tmp, buffer + offset, sizeof(ULARGE_INTEGER));
4289 value->u.LowPart = htole32(tmp.u.HighPart);
4290 value->u.HighPart = htole32(tmp.u.LowPart);
4291 #else
4292 memcpy(value, buffer + offset, sizeof(ULARGE_INTEGER));
4293 #endif
4296 void StorageUtl_WriteULargeInteger(BYTE* buffer, ULONG offset,
4297 const ULARGE_INTEGER *value)
4299 #ifdef WORDS_BIGENDIAN
4300 ULARGE_INTEGER tmp;
4302 tmp.u.LowPart = htole32(value->u.HighPart);
4303 tmp.u.HighPart = htole32(value->u.LowPart);
4304 memcpy(buffer + offset, &tmp, sizeof(ULARGE_INTEGER));
4305 #else
4306 memcpy(buffer + offset, value, sizeof(ULARGE_INTEGER));
4307 #endif
4310 void StorageUtl_ReadGUID(const BYTE* buffer, ULONG offset, GUID* value)
4312 StorageUtl_ReadDWord(buffer, offset, &(value->Data1));
4313 StorageUtl_ReadWord(buffer, offset+4, &(value->Data2));
4314 StorageUtl_ReadWord(buffer, offset+6, &(value->Data3));
4316 memcpy(value->Data4, buffer+offset+8, sizeof(value->Data4));
4319 void StorageUtl_WriteGUID(BYTE* buffer, ULONG offset, const GUID* value)
4321 StorageUtl_WriteDWord(buffer, offset, value->Data1);
4322 StorageUtl_WriteWord(buffer, offset+4, value->Data2);
4323 StorageUtl_WriteWord(buffer, offset+6, value->Data3);
4325 memcpy(buffer+offset+8, value->Data4, sizeof(value->Data4));
4328 void StorageUtl_CopyPropertyToSTATSTG(
4329 STATSTG* destination,
4330 const StgProperty* source,
4331 int statFlags)
4334 * The copy of the string occurs only when the flag is not set
4336 if( ((statFlags & STATFLAG_NONAME) != 0) ||
4337 (source->name == NULL) ||
4338 (source->name[0] == 0) )
4340 destination->pwcsName = 0;
4342 else
4344 destination->pwcsName =
4345 CoTaskMemAlloc((lstrlenW(source->name)+1)*sizeof(WCHAR));
4347 strcpyW((LPWSTR)destination->pwcsName, source->name);
4350 switch (source->propertyType)
4352 case PROPTYPE_STORAGE:
4353 case PROPTYPE_ROOT:
4354 destination->type = STGTY_STORAGE;
4355 break;
4356 case PROPTYPE_STREAM:
4357 destination->type = STGTY_STREAM;
4358 break;
4359 default:
4360 destination->type = STGTY_STREAM;
4361 break;
4364 destination->cbSize = source->size;
4366 currentReturnStruct->mtime = {0}; TODO
4367 currentReturnStruct->ctime = {0};
4368 currentReturnStruct->atime = {0};
4370 destination->grfMode = 0;
4371 destination->grfLocksSupported = 0;
4372 destination->clsid = source->propertyUniqueID;
4373 destination->grfStateBits = 0;
4374 destination->reserved = 0;
4377 /******************************************************************************
4378 ** BlockChainStream implementation
4381 BlockChainStream* BlockChainStream_Construct(
4382 StorageImpl* parentStorage,
4383 ULONG* headOfStreamPlaceHolder,
4384 ULONG propertyIndex)
4386 BlockChainStream* newStream;
4387 ULONG blockIndex;
4389 newStream = HeapAlloc(GetProcessHeap(), 0, sizeof(BlockChainStream));
4391 newStream->parentStorage = parentStorage;
4392 newStream->headOfStreamPlaceHolder = headOfStreamPlaceHolder;
4393 newStream->ownerPropertyIndex = propertyIndex;
4394 newStream->lastBlockNoInSequence = 0xFFFFFFFF;
4395 newStream->tailIndex = BLOCK_END_OF_CHAIN;
4396 newStream->numBlocks = 0;
4398 blockIndex = BlockChainStream_GetHeadOfChain(newStream);
4400 while (blockIndex != BLOCK_END_OF_CHAIN)
4402 newStream->numBlocks++;
4403 newStream->tailIndex = blockIndex;
4405 if(FAILED(StorageImpl_GetNextBlockInChain(
4406 parentStorage,
4407 blockIndex,
4408 &blockIndex)))
4410 HeapFree(GetProcessHeap(), 0, newStream);
4411 return NULL;
4415 return newStream;
4418 void BlockChainStream_Destroy(BlockChainStream* This)
4420 HeapFree(GetProcessHeap(), 0, This);
4423 /******************************************************************************
4424 * BlockChainStream_GetHeadOfChain
4426 * Returns the head of this stream chain.
4427 * Some special chains don't have properties, their heads are kept in
4428 * This->headOfStreamPlaceHolder.
4431 static ULONG BlockChainStream_GetHeadOfChain(BlockChainStream* This)
4433 StgProperty chainProperty;
4434 BOOL readSuccessful;
4436 if (This->headOfStreamPlaceHolder != 0)
4437 return *(This->headOfStreamPlaceHolder);
4439 if (This->ownerPropertyIndex != PROPERTY_NULL)
4441 readSuccessful = StorageImpl_ReadProperty(
4442 This->parentStorage,
4443 This->ownerPropertyIndex,
4444 &chainProperty);
4446 if (readSuccessful)
4448 return chainProperty.startingBlock;
4452 return BLOCK_END_OF_CHAIN;
4455 /******************************************************************************
4456 * BlockChainStream_GetCount
4458 * Returns the number of blocks that comprises this chain.
4459 * This is not the size of the stream as the last block may not be full!
4462 static ULONG BlockChainStream_GetCount(BlockChainStream* This)
4464 ULONG blockIndex;
4465 ULONG count = 0;
4467 blockIndex = BlockChainStream_GetHeadOfChain(This);
4469 while (blockIndex != BLOCK_END_OF_CHAIN)
4471 count++;
4473 if(FAILED(StorageImpl_GetNextBlockInChain(
4474 This->parentStorage,
4475 blockIndex,
4476 &blockIndex)))
4477 return 0;
4480 return count;
4483 /******************************************************************************
4484 * BlockChainStream_ReadAt
4486 * Reads a specified number of bytes from this chain at the specified offset.
4487 * bytesRead may be NULL.
4488 * Failure will be returned if the specified number of bytes has not been read.
4490 HRESULT BlockChainStream_ReadAt(BlockChainStream* This,
4491 ULARGE_INTEGER offset,
4492 ULONG size,
4493 void* buffer,
4494 ULONG* bytesRead)
4496 ULONG blockNoInSequence = offset.u.LowPart / This->parentStorage->bigBlockSize;
4497 ULONG offsetInBlock = offset.u.LowPart % This->parentStorage->bigBlockSize;
4498 ULONG bytesToReadInBuffer;
4499 ULONG blockIndex;
4500 BYTE* bufferWalker;
4502 TRACE("(%p)-> %i %p %i %p\n",This, offset.u.LowPart, buffer, size, bytesRead);
4505 * Find the first block in the stream that contains part of the buffer.
4507 if ( (This->lastBlockNoInSequence == 0xFFFFFFFF) ||
4508 (This->lastBlockNoInSequenceIndex == BLOCK_END_OF_CHAIN) ||
4509 (blockNoInSequence < This->lastBlockNoInSequence) )
4511 blockIndex = BlockChainStream_GetHeadOfChain(This);
4512 This->lastBlockNoInSequence = blockNoInSequence;
4514 else
4516 ULONG temp = blockNoInSequence;
4518 blockIndex = This->lastBlockNoInSequenceIndex;
4519 blockNoInSequence -= This->lastBlockNoInSequence;
4520 This->lastBlockNoInSequence = temp;
4523 while ( (blockNoInSequence > 0) && (blockIndex != BLOCK_END_OF_CHAIN))
4525 if(FAILED(StorageImpl_GetNextBlockInChain(This->parentStorage, blockIndex, &blockIndex)))
4526 return STG_E_DOCFILECORRUPT;
4527 blockNoInSequence--;
4530 if ((blockNoInSequence > 0) && (blockIndex == BLOCK_END_OF_CHAIN))
4531 return STG_E_DOCFILECORRUPT; /* We failed to find the starting block */
4533 This->lastBlockNoInSequenceIndex = blockIndex;
4536 * Start reading the buffer.
4538 *bytesRead = 0;
4539 bufferWalker = buffer;
4541 while ( (size > 0) && (blockIndex != BLOCK_END_OF_CHAIN) )
4543 ULARGE_INTEGER ulOffset;
4544 DWORD bytesReadAt;
4546 * Calculate how many bytes we can copy from this big block.
4548 bytesToReadInBuffer =
4549 min(This->parentStorage->bigBlockSize - offsetInBlock, size);
4551 TRACE("block %i\n",blockIndex);
4552 ulOffset.u.HighPart = 0;
4553 ulOffset.u.LowPart = BLOCK_GetBigBlockOffset(blockIndex) +
4554 offsetInBlock;
4556 StorageImpl_ReadAt(This->parentStorage,
4557 ulOffset,
4558 bufferWalker,
4559 bytesToReadInBuffer,
4560 &bytesReadAt);
4562 * Step to the next big block.
4564 if( size > bytesReadAt && FAILED(StorageImpl_GetNextBlockInChain(This->parentStorage, blockIndex, &blockIndex)))
4565 return STG_E_DOCFILECORRUPT;
4567 bufferWalker += bytesReadAt;
4568 size -= bytesReadAt;
4569 *bytesRead += bytesReadAt;
4570 offsetInBlock = 0; /* There is no offset on the next block */
4572 if (bytesToReadInBuffer != bytesReadAt)
4573 break;
4576 return (size == 0) ? S_OK : STG_E_READFAULT;
4579 /******************************************************************************
4580 * BlockChainStream_WriteAt
4582 * Writes the specified number of bytes to this chain at the specified offset.
4583 * bytesWritten may be NULL.
4584 * Will fail if not all specified number of bytes have been written.
4586 HRESULT BlockChainStream_WriteAt(BlockChainStream* This,
4587 ULARGE_INTEGER offset,
4588 ULONG size,
4589 const void* buffer,
4590 ULONG* bytesWritten)
4592 ULONG blockNoInSequence = offset.u.LowPart / This->parentStorage->bigBlockSize;
4593 ULONG offsetInBlock = offset.u.LowPart % This->parentStorage->bigBlockSize;
4594 ULONG bytesToWrite;
4595 ULONG blockIndex;
4596 const BYTE* bufferWalker;
4599 * Find the first block in the stream that contains part of the buffer.
4601 if ( (This->lastBlockNoInSequence == 0xFFFFFFFF) ||
4602 (This->lastBlockNoInSequenceIndex == BLOCK_END_OF_CHAIN) ||
4603 (blockNoInSequence < This->lastBlockNoInSequence) )
4605 blockIndex = BlockChainStream_GetHeadOfChain(This);
4606 This->lastBlockNoInSequence = blockNoInSequence;
4608 else
4610 ULONG temp = blockNoInSequence;
4612 blockIndex = This->lastBlockNoInSequenceIndex;
4613 blockNoInSequence -= This->lastBlockNoInSequence;
4614 This->lastBlockNoInSequence = temp;
4617 while ( (blockNoInSequence > 0) && (blockIndex != BLOCK_END_OF_CHAIN))
4619 if(FAILED(StorageImpl_GetNextBlockInChain(This->parentStorage, blockIndex,
4620 &blockIndex)))
4621 return STG_E_DOCFILECORRUPT;
4622 blockNoInSequence--;
4625 This->lastBlockNoInSequenceIndex = blockIndex;
4627 /* BlockChainStream_SetSize should have already been called to ensure we have
4628 * enough blocks in the chain to write into */
4629 if (blockIndex == BLOCK_END_OF_CHAIN)
4631 ERR("not enough blocks in chain to write data\n");
4632 return STG_E_DOCFILECORRUPT;
4635 *bytesWritten = 0;
4636 bufferWalker = (const BYTE*)buffer;
4638 while ( (size > 0) && (blockIndex != BLOCK_END_OF_CHAIN) )
4640 ULARGE_INTEGER ulOffset;
4641 DWORD bytesWrittenAt;
4643 * Calculate how many bytes we can copy from this big block.
4645 bytesToWrite =
4646 min(This->parentStorage->bigBlockSize - offsetInBlock, size);
4648 TRACE("block %i\n",blockIndex);
4649 ulOffset.u.HighPart = 0;
4650 ulOffset.u.LowPart = BLOCK_GetBigBlockOffset(blockIndex) +
4651 offsetInBlock;
4653 StorageImpl_WriteAt(This->parentStorage,
4654 ulOffset,
4655 bufferWalker,
4656 bytesToWrite,
4657 &bytesWrittenAt);
4660 * Step to the next big block.
4662 if(size > bytesWrittenAt && FAILED(StorageImpl_GetNextBlockInChain(This->parentStorage, blockIndex,
4663 &blockIndex)))
4664 return STG_E_DOCFILECORRUPT;
4666 bufferWalker += bytesWrittenAt;
4667 size -= bytesWrittenAt;
4668 *bytesWritten += bytesWrittenAt;
4669 offsetInBlock = 0; /* There is no offset on the next block */
4671 if (bytesWrittenAt != bytesToWrite)
4672 break;
4675 return (size == 0) ? S_OK : STG_E_WRITEFAULT;
4678 /******************************************************************************
4679 * BlockChainStream_Shrink
4681 * Shrinks this chain in the big block depot.
4683 static BOOL BlockChainStream_Shrink(BlockChainStream* This,
4684 ULARGE_INTEGER newSize)
4686 ULONG blockIndex, extraBlock;
4687 ULONG numBlocks;
4688 ULONG count = 1;
4691 * Reset the last accessed block cache.
4693 This->lastBlockNoInSequence = 0xFFFFFFFF;
4694 This->lastBlockNoInSequenceIndex = BLOCK_END_OF_CHAIN;
4697 * Figure out how many blocks are needed to contain the new size
4699 numBlocks = newSize.u.LowPart / This->parentStorage->bigBlockSize;
4701 if ((newSize.u.LowPart % This->parentStorage->bigBlockSize) != 0)
4702 numBlocks++;
4704 blockIndex = BlockChainStream_GetHeadOfChain(This);
4707 * Go to the new end of chain
4709 while (count < numBlocks)
4711 if(FAILED(StorageImpl_GetNextBlockInChain(This->parentStorage, blockIndex,
4712 &blockIndex)))
4713 return FALSE;
4714 count++;
4717 /* Get the next block before marking the new end */
4718 if(FAILED(StorageImpl_GetNextBlockInChain(This->parentStorage, blockIndex,
4719 &extraBlock)))
4720 return FALSE;
4722 /* Mark the new end of chain */
4723 StorageImpl_SetNextBlockInChain(
4724 This->parentStorage,
4725 blockIndex,
4726 BLOCK_END_OF_CHAIN);
4728 This->tailIndex = blockIndex;
4729 This->numBlocks = numBlocks;
4732 * Mark the extra blocks as free
4734 while (extraBlock != BLOCK_END_OF_CHAIN)
4736 if(FAILED(StorageImpl_GetNextBlockInChain(This->parentStorage, extraBlock,
4737 &blockIndex)))
4738 return FALSE;
4739 StorageImpl_FreeBigBlock(This->parentStorage, extraBlock);
4740 extraBlock = blockIndex;
4743 return TRUE;
4746 /******************************************************************************
4747 * BlockChainStream_Enlarge
4749 * Grows this chain in the big block depot.
4751 static BOOL BlockChainStream_Enlarge(BlockChainStream* This,
4752 ULARGE_INTEGER newSize)
4754 ULONG blockIndex, currentBlock;
4755 ULONG newNumBlocks;
4756 ULONG oldNumBlocks = 0;
4758 blockIndex = BlockChainStream_GetHeadOfChain(This);
4761 * Empty chain. Create the head.
4763 if (blockIndex == BLOCK_END_OF_CHAIN)
4765 blockIndex = StorageImpl_GetNextFreeBigBlock(This->parentStorage);
4766 StorageImpl_SetNextBlockInChain(This->parentStorage,
4767 blockIndex,
4768 BLOCK_END_OF_CHAIN);
4770 if (This->headOfStreamPlaceHolder != 0)
4772 *(This->headOfStreamPlaceHolder) = blockIndex;
4774 else
4776 StgProperty chainProp;
4777 assert(This->ownerPropertyIndex != PROPERTY_NULL);
4779 StorageImpl_ReadProperty(
4780 This->parentStorage,
4781 This->ownerPropertyIndex,
4782 &chainProp);
4784 chainProp.startingBlock = blockIndex;
4786 StorageImpl_WriteProperty(
4787 This->parentStorage,
4788 This->ownerPropertyIndex,
4789 &chainProp);
4792 This->tailIndex = blockIndex;
4793 This->numBlocks = 1;
4797 * Figure out how many blocks are needed to contain this stream
4799 newNumBlocks = newSize.u.LowPart / This->parentStorage->bigBlockSize;
4801 if ((newSize.u.LowPart % This->parentStorage->bigBlockSize) != 0)
4802 newNumBlocks++;
4805 * Go to the current end of chain
4807 if (This->tailIndex == BLOCK_END_OF_CHAIN)
4809 currentBlock = blockIndex;
4811 while (blockIndex != BLOCK_END_OF_CHAIN)
4813 This->numBlocks++;
4814 currentBlock = blockIndex;
4816 if(FAILED(StorageImpl_GetNextBlockInChain(This->parentStorage, currentBlock,
4817 &blockIndex)))
4818 return FALSE;
4821 This->tailIndex = currentBlock;
4824 currentBlock = This->tailIndex;
4825 oldNumBlocks = This->numBlocks;
4828 * Add new blocks to the chain
4830 if (oldNumBlocks < newNumBlocks)
4832 while (oldNumBlocks < newNumBlocks)
4834 blockIndex = StorageImpl_GetNextFreeBigBlock(This->parentStorage);
4836 StorageImpl_SetNextBlockInChain(
4837 This->parentStorage,
4838 currentBlock,
4839 blockIndex);
4841 StorageImpl_SetNextBlockInChain(
4842 This->parentStorage,
4843 blockIndex,
4844 BLOCK_END_OF_CHAIN);
4846 currentBlock = blockIndex;
4847 oldNumBlocks++;
4850 This->tailIndex = blockIndex;
4851 This->numBlocks = newNumBlocks;
4854 return TRUE;
4857 /******************************************************************************
4858 * BlockChainStream_SetSize
4860 * Sets the size of this stream. The big block depot will be updated.
4861 * The file will grow if we grow the chain.
4863 * TODO: Free the actual blocks in the file when we shrink the chain.
4864 * Currently, the blocks are still in the file. So the file size
4865 * doesn't shrink even if we shrink streams.
4867 BOOL BlockChainStream_SetSize(
4868 BlockChainStream* This,
4869 ULARGE_INTEGER newSize)
4871 ULARGE_INTEGER size = BlockChainStream_GetSize(This);
4873 if (newSize.u.LowPart == size.u.LowPart)
4874 return TRUE;
4876 if (newSize.u.LowPart < size.u.LowPart)
4878 BlockChainStream_Shrink(This, newSize);
4880 else
4882 BlockChainStream_Enlarge(This, newSize);
4885 return TRUE;
4888 /******************************************************************************
4889 * BlockChainStream_GetSize
4891 * Returns the size of this chain.
4892 * Will return the block count if this chain doesn't have a property.
4894 static ULARGE_INTEGER BlockChainStream_GetSize(BlockChainStream* This)
4896 StgProperty chainProperty;
4898 if(This->headOfStreamPlaceHolder == NULL)
4901 * This chain is a data stream read the property and return
4902 * the appropriate size
4904 StorageImpl_ReadProperty(
4905 This->parentStorage,
4906 This->ownerPropertyIndex,
4907 &chainProperty);
4909 return chainProperty.size;
4911 else
4914 * this chain is a chain that does not have a property, figure out the
4915 * size by making the product number of used blocks times the
4916 * size of them
4918 ULARGE_INTEGER result;
4919 result.u.HighPart = 0;
4921 result.u.LowPart =
4922 BlockChainStream_GetCount(This) *
4923 This->parentStorage->bigBlockSize;
4925 return result;
4929 /******************************************************************************
4930 ** SmallBlockChainStream implementation
4933 SmallBlockChainStream* SmallBlockChainStream_Construct(
4934 StorageImpl* parentStorage,
4935 ULONG propertyIndex)
4937 SmallBlockChainStream* newStream;
4939 newStream = HeapAlloc(GetProcessHeap(), 0, sizeof(SmallBlockChainStream));
4941 newStream->parentStorage = parentStorage;
4942 newStream->ownerPropertyIndex = propertyIndex;
4944 return newStream;
4947 void SmallBlockChainStream_Destroy(
4948 SmallBlockChainStream* This)
4950 HeapFree(GetProcessHeap(), 0, This);
4953 /******************************************************************************
4954 * SmallBlockChainStream_GetHeadOfChain
4956 * Returns the head of this chain of small blocks.
4958 static ULONG SmallBlockChainStream_GetHeadOfChain(
4959 SmallBlockChainStream* This)
4961 StgProperty chainProperty;
4962 BOOL readSuccessful;
4964 if (This->ownerPropertyIndex)
4966 readSuccessful = StorageImpl_ReadProperty(
4967 This->parentStorage,
4968 This->ownerPropertyIndex,
4969 &chainProperty);
4971 if (readSuccessful)
4973 return chainProperty.startingBlock;
4978 return BLOCK_END_OF_CHAIN;
4981 /******************************************************************************
4982 * SmallBlockChainStream_GetNextBlockInChain
4984 * Returns the index of the next small block in this chain.
4986 * Return Values:
4987 * - BLOCK_END_OF_CHAIN: end of this chain
4988 * - BLOCK_UNUSED: small block 'blockIndex' is free
4990 static HRESULT SmallBlockChainStream_GetNextBlockInChain(
4991 SmallBlockChainStream* This,
4992 ULONG blockIndex,
4993 ULONG* nextBlockInChain)
4995 ULARGE_INTEGER offsetOfBlockInDepot;
4996 DWORD buffer;
4997 ULONG bytesRead;
4998 HRESULT res;
5000 *nextBlockInChain = BLOCK_END_OF_CHAIN;
5002 offsetOfBlockInDepot.u.HighPart = 0;
5003 offsetOfBlockInDepot.u.LowPart = blockIndex * sizeof(ULONG);
5006 * Read those bytes in the buffer from the small block file.
5008 res = BlockChainStream_ReadAt(
5009 This->parentStorage->smallBlockDepotChain,
5010 offsetOfBlockInDepot,
5011 sizeof(DWORD),
5012 &buffer,
5013 &bytesRead);
5015 if (SUCCEEDED(res))
5017 StorageUtl_ReadDWord((BYTE *)&buffer, 0, nextBlockInChain);
5018 return S_OK;
5021 return res;
5024 /******************************************************************************
5025 * SmallBlockChainStream_SetNextBlockInChain
5027 * Writes the index of the next block of the specified block in the small
5028 * block depot.
5029 * To set the end of chain use BLOCK_END_OF_CHAIN as nextBlock.
5030 * To flag a block as free use BLOCK_UNUSED as nextBlock.
5032 static void SmallBlockChainStream_SetNextBlockInChain(
5033 SmallBlockChainStream* This,
5034 ULONG blockIndex,
5035 ULONG nextBlock)
5037 ULARGE_INTEGER offsetOfBlockInDepot;
5038 DWORD buffer;
5039 ULONG bytesWritten;
5041 offsetOfBlockInDepot.u.HighPart = 0;
5042 offsetOfBlockInDepot.u.LowPart = blockIndex * sizeof(ULONG);
5044 StorageUtl_WriteDWord((BYTE *)&buffer, 0, nextBlock);
5047 * Read those bytes in the buffer from the small block file.
5049 BlockChainStream_WriteAt(
5050 This->parentStorage->smallBlockDepotChain,
5051 offsetOfBlockInDepot,
5052 sizeof(DWORD),
5053 &buffer,
5054 &bytesWritten);
5057 /******************************************************************************
5058 * SmallBlockChainStream_FreeBlock
5060 * Flag small block 'blockIndex' as free in the small block depot.
5062 static void SmallBlockChainStream_FreeBlock(
5063 SmallBlockChainStream* This,
5064 ULONG blockIndex)
5066 SmallBlockChainStream_SetNextBlockInChain(This, blockIndex, BLOCK_UNUSED);
5069 /******************************************************************************
5070 * SmallBlockChainStream_GetNextFreeBlock
5072 * Returns the index of a free small block. The small block depot will be
5073 * enlarged if necessary. The small block chain will also be enlarged if
5074 * necessary.
5076 static ULONG SmallBlockChainStream_GetNextFreeBlock(
5077 SmallBlockChainStream* This)
5079 ULARGE_INTEGER offsetOfBlockInDepot;
5080 DWORD buffer;
5081 ULONG bytesRead;
5082 ULONG blockIndex = 0;
5083 ULONG nextBlockIndex = BLOCK_END_OF_CHAIN;
5084 HRESULT res = S_OK;
5085 ULONG smallBlocksPerBigBlock;
5087 offsetOfBlockInDepot.u.HighPart = 0;
5090 * Scan the small block depot for a free block
5092 while (nextBlockIndex != BLOCK_UNUSED)
5094 offsetOfBlockInDepot.u.LowPart = blockIndex * sizeof(ULONG);
5096 res = BlockChainStream_ReadAt(
5097 This->parentStorage->smallBlockDepotChain,
5098 offsetOfBlockInDepot,
5099 sizeof(DWORD),
5100 &buffer,
5101 &bytesRead);
5104 * If we run out of space for the small block depot, enlarge it
5106 if (SUCCEEDED(res))
5108 StorageUtl_ReadDWord((BYTE *)&buffer, 0, &nextBlockIndex);
5110 if (nextBlockIndex != BLOCK_UNUSED)
5111 blockIndex++;
5113 else
5115 ULONG count =
5116 BlockChainStream_GetCount(This->parentStorage->smallBlockDepotChain);
5118 ULONG sbdIndex = This->parentStorage->smallBlockDepotStart;
5119 ULONG nextBlock, newsbdIndex;
5120 BYTE smallBlockDepot[BIG_BLOCK_SIZE];
5122 nextBlock = sbdIndex;
5123 while (nextBlock != BLOCK_END_OF_CHAIN)
5125 sbdIndex = nextBlock;
5126 StorageImpl_GetNextBlockInChain(This->parentStorage, sbdIndex, &nextBlock);
5129 newsbdIndex = StorageImpl_GetNextFreeBigBlock(This->parentStorage);
5130 if (sbdIndex != BLOCK_END_OF_CHAIN)
5131 StorageImpl_SetNextBlockInChain(
5132 This->parentStorage,
5133 sbdIndex,
5134 newsbdIndex);
5136 StorageImpl_SetNextBlockInChain(
5137 This->parentStorage,
5138 newsbdIndex,
5139 BLOCK_END_OF_CHAIN);
5142 * Initialize all the small blocks to free
5144 memset(smallBlockDepot, BLOCK_UNUSED, This->parentStorage->bigBlockSize);
5145 StorageImpl_WriteBigBlock(This->parentStorage, newsbdIndex, smallBlockDepot);
5147 if (count == 0)
5150 * We have just created the small block depot.
5152 StgProperty rootProp;
5153 ULONG sbStartIndex;
5156 * Save it in the header
5158 This->parentStorage->smallBlockDepotStart = newsbdIndex;
5159 StorageImpl_SaveFileHeader(This->parentStorage);
5162 * And allocate the first big block that will contain small blocks
5164 sbStartIndex =
5165 StorageImpl_GetNextFreeBigBlock(This->parentStorage);
5167 StorageImpl_SetNextBlockInChain(
5168 This->parentStorage,
5169 sbStartIndex,
5170 BLOCK_END_OF_CHAIN);
5172 StorageImpl_ReadProperty(
5173 This->parentStorage,
5174 This->parentStorage->base.rootPropertySetIndex,
5175 &rootProp);
5177 rootProp.startingBlock = sbStartIndex;
5178 rootProp.size.u.HighPart = 0;
5179 rootProp.size.u.LowPart = This->parentStorage->bigBlockSize;
5181 StorageImpl_WriteProperty(
5182 This->parentStorage,
5183 This->parentStorage->base.rootPropertySetIndex,
5184 &rootProp);
5186 else
5187 StorageImpl_SaveFileHeader(This->parentStorage);
5191 smallBlocksPerBigBlock =
5192 This->parentStorage->bigBlockSize / This->parentStorage->smallBlockSize;
5195 * Verify if we have to allocate big blocks to contain small blocks
5197 if (blockIndex % smallBlocksPerBigBlock == 0)
5199 StgProperty rootProp;
5200 ULONG blocksRequired = (blockIndex / smallBlocksPerBigBlock) + 1;
5202 StorageImpl_ReadProperty(
5203 This->parentStorage,
5204 This->parentStorage->base.rootPropertySetIndex,
5205 &rootProp);
5207 if (rootProp.size.u.LowPart <
5208 (blocksRequired * This->parentStorage->bigBlockSize))
5210 rootProp.size.u.LowPart += This->parentStorage->bigBlockSize;
5212 BlockChainStream_SetSize(
5213 This->parentStorage->smallBlockRootChain,
5214 rootProp.size);
5216 StorageImpl_WriteProperty(
5217 This->parentStorage,
5218 This->parentStorage->base.rootPropertySetIndex,
5219 &rootProp);
5223 return blockIndex;
5226 /******************************************************************************
5227 * SmallBlockChainStream_ReadAt
5229 * Reads a specified number of bytes from this chain at the specified offset.
5230 * bytesRead may be NULL.
5231 * Failure will be returned if the specified number of bytes has not been read.
5233 HRESULT SmallBlockChainStream_ReadAt(
5234 SmallBlockChainStream* This,
5235 ULARGE_INTEGER offset,
5236 ULONG size,
5237 void* buffer,
5238 ULONG* bytesRead)
5240 HRESULT rc = S_OK;
5241 ULARGE_INTEGER offsetInBigBlockFile;
5242 ULONG blockNoInSequence =
5243 offset.u.LowPart / This->parentStorage->smallBlockSize;
5245 ULONG offsetInBlock = offset.u.LowPart % This->parentStorage->smallBlockSize;
5246 ULONG bytesToReadInBuffer;
5247 ULONG blockIndex;
5248 ULONG bytesReadFromBigBlockFile;
5249 BYTE* bufferWalker;
5252 * This should never happen on a small block file.
5254 assert(offset.u.HighPart==0);
5257 * Find the first block in the stream that contains part of the buffer.
5259 blockIndex = SmallBlockChainStream_GetHeadOfChain(This);
5261 while ( (blockNoInSequence > 0) && (blockIndex != BLOCK_END_OF_CHAIN))
5263 rc = SmallBlockChainStream_GetNextBlockInChain(This, blockIndex, &blockIndex);
5264 if(FAILED(rc))
5265 return rc;
5266 blockNoInSequence--;
5270 * Start reading the buffer.
5272 *bytesRead = 0;
5273 bufferWalker = buffer;
5275 while ( (size > 0) && (blockIndex != BLOCK_END_OF_CHAIN) )
5278 * Calculate how many bytes we can copy from this small block.
5280 bytesToReadInBuffer =
5281 min(This->parentStorage->smallBlockSize - offsetInBlock, size);
5284 * Calculate the offset of the small block in the small block file.
5286 offsetInBigBlockFile.u.HighPart = 0;
5287 offsetInBigBlockFile.u.LowPart =
5288 blockIndex * This->parentStorage->smallBlockSize;
5290 offsetInBigBlockFile.u.LowPart += offsetInBlock;
5293 * Read those bytes in the buffer from the small block file.
5294 * The small block has already been identified so it shouldn't fail
5295 * unless the file is corrupt.
5297 rc = BlockChainStream_ReadAt(This->parentStorage->smallBlockRootChain,
5298 offsetInBigBlockFile,
5299 bytesToReadInBuffer,
5300 bufferWalker,
5301 &bytesReadFromBigBlockFile);
5303 if (FAILED(rc))
5304 return rc;
5307 * Step to the next big block.
5309 rc = SmallBlockChainStream_GetNextBlockInChain(This, blockIndex, &blockIndex);
5310 if(FAILED(rc))
5311 return STG_E_DOCFILECORRUPT;
5313 bufferWalker += bytesReadFromBigBlockFile;
5314 size -= bytesReadFromBigBlockFile;
5315 *bytesRead += bytesReadFromBigBlockFile;
5316 offsetInBlock = (offsetInBlock + bytesReadFromBigBlockFile) % This->parentStorage->smallBlockSize;
5319 return (size == 0) ? S_OK : STG_E_READFAULT;
5322 /******************************************************************************
5323 * SmallBlockChainStream_WriteAt
5325 * Writes the specified number of bytes to this chain at the specified offset.
5326 * bytesWritten may be NULL.
5327 * Will fail if not all specified number of bytes have been written.
5329 HRESULT SmallBlockChainStream_WriteAt(
5330 SmallBlockChainStream* This,
5331 ULARGE_INTEGER offset,
5332 ULONG size,
5333 const void* buffer,
5334 ULONG* bytesWritten)
5336 ULARGE_INTEGER offsetInBigBlockFile;
5337 ULONG blockNoInSequence =
5338 offset.u.LowPart / This->parentStorage->smallBlockSize;
5340 ULONG offsetInBlock = offset.u.LowPart % This->parentStorage->smallBlockSize;
5341 ULONG bytesToWriteInBuffer;
5342 ULONG blockIndex;
5343 ULONG bytesWrittenToBigBlockFile;
5344 const BYTE* bufferWalker;
5345 HRESULT res;
5348 * This should never happen on a small block file.
5350 assert(offset.u.HighPart==0);
5353 * Find the first block in the stream that contains part of the buffer.
5355 blockIndex = SmallBlockChainStream_GetHeadOfChain(This);
5357 while ( (blockNoInSequence > 0) && (blockIndex != BLOCK_END_OF_CHAIN))
5359 if(FAILED(SmallBlockChainStream_GetNextBlockInChain(This, blockIndex, &blockIndex)))
5360 return STG_E_DOCFILECORRUPT;
5361 blockNoInSequence--;
5365 * Start writing the buffer.
5367 * Here, I'm casting away the constness on the buffer variable
5368 * This is OK since we don't intend to modify that buffer.
5370 *bytesWritten = 0;
5371 bufferWalker = (const BYTE*)buffer;
5372 while ( (size > 0) && (blockIndex != BLOCK_END_OF_CHAIN) )
5375 * Calculate how many bytes we can copy to this small block.
5377 bytesToWriteInBuffer =
5378 min(This->parentStorage->smallBlockSize - offsetInBlock, size);
5381 * Calculate the offset of the small block in the small block file.
5383 offsetInBigBlockFile.u.HighPart = 0;
5384 offsetInBigBlockFile.u.LowPart =
5385 blockIndex * This->parentStorage->smallBlockSize;
5387 offsetInBigBlockFile.u.LowPart += offsetInBlock;
5390 * Write those bytes in the buffer to the small block file.
5392 res = BlockChainStream_WriteAt(
5393 This->parentStorage->smallBlockRootChain,
5394 offsetInBigBlockFile,
5395 bytesToWriteInBuffer,
5396 bufferWalker,
5397 &bytesWrittenToBigBlockFile);
5398 if (FAILED(res))
5399 return res;
5402 * Step to the next big block.
5404 if(FAILED(SmallBlockChainStream_GetNextBlockInChain(This, blockIndex,
5405 &blockIndex)))
5406 return FALSE;
5407 bufferWalker += bytesWrittenToBigBlockFile;
5408 size -= bytesWrittenToBigBlockFile;
5409 *bytesWritten += bytesWrittenToBigBlockFile;
5410 offsetInBlock = (offsetInBlock + bytesWrittenToBigBlockFile) % This->parentStorage->smallBlockSize;
5413 return (size == 0) ? S_OK : STG_E_WRITEFAULT;
5416 /******************************************************************************
5417 * SmallBlockChainStream_Shrink
5419 * Shrinks this chain in the small block depot.
5421 static BOOL SmallBlockChainStream_Shrink(
5422 SmallBlockChainStream* This,
5423 ULARGE_INTEGER newSize)
5425 ULONG blockIndex, extraBlock;
5426 ULONG numBlocks;
5427 ULONG count = 0;
5429 numBlocks = newSize.u.LowPart / This->parentStorage->smallBlockSize;
5431 if ((newSize.u.LowPart % This->parentStorage->smallBlockSize) != 0)
5432 numBlocks++;
5434 blockIndex = SmallBlockChainStream_GetHeadOfChain(This);
5437 * Go to the new end of chain
5439 while (count < numBlocks)
5441 if(FAILED(SmallBlockChainStream_GetNextBlockInChain(This, blockIndex,
5442 &blockIndex)))
5443 return FALSE;
5444 count++;
5448 * If the count is 0, we have a special case, the head of the chain was
5449 * just freed.
5451 if (count == 0)
5453 StgProperty chainProp;
5455 StorageImpl_ReadProperty(This->parentStorage,
5456 This->ownerPropertyIndex,
5457 &chainProp);
5459 chainProp.startingBlock = BLOCK_END_OF_CHAIN;
5461 StorageImpl_WriteProperty(This->parentStorage,
5462 This->ownerPropertyIndex,
5463 &chainProp);
5466 * We start freeing the chain at the head block.
5468 extraBlock = blockIndex;
5470 else
5472 /* Get the next block before marking the new end */
5473 if(FAILED(SmallBlockChainStream_GetNextBlockInChain(This, blockIndex,
5474 &extraBlock)))
5475 return FALSE;
5477 /* Mark the new end of chain */
5478 SmallBlockChainStream_SetNextBlockInChain(
5479 This,
5480 blockIndex,
5481 BLOCK_END_OF_CHAIN);
5485 * Mark the extra blocks as free
5487 while (extraBlock != BLOCK_END_OF_CHAIN)
5489 if(FAILED(SmallBlockChainStream_GetNextBlockInChain(This, extraBlock,
5490 &blockIndex)))
5491 return FALSE;
5492 SmallBlockChainStream_FreeBlock(This, extraBlock);
5493 extraBlock = blockIndex;
5496 return TRUE;
5499 /******************************************************************************
5500 * SmallBlockChainStream_Enlarge
5502 * Grows this chain in the small block depot.
5504 static BOOL SmallBlockChainStream_Enlarge(
5505 SmallBlockChainStream* This,
5506 ULARGE_INTEGER newSize)
5508 ULONG blockIndex, currentBlock;
5509 ULONG newNumBlocks;
5510 ULONG oldNumBlocks = 0;
5512 blockIndex = SmallBlockChainStream_GetHeadOfChain(This);
5515 * Empty chain
5517 if (blockIndex == BLOCK_END_OF_CHAIN)
5520 StgProperty chainProp;
5522 StorageImpl_ReadProperty(This->parentStorage, This->ownerPropertyIndex,
5523 &chainProp);
5525 chainProp.startingBlock = SmallBlockChainStream_GetNextFreeBlock(This);
5527 StorageImpl_WriteProperty(This->parentStorage, This->ownerPropertyIndex,
5528 &chainProp);
5530 blockIndex = chainProp.startingBlock;
5531 SmallBlockChainStream_SetNextBlockInChain(
5532 This,
5533 blockIndex,
5534 BLOCK_END_OF_CHAIN);
5537 currentBlock = blockIndex;
5540 * Figure out how many blocks are needed to contain this stream
5542 newNumBlocks = newSize.u.LowPart / This->parentStorage->smallBlockSize;
5544 if ((newSize.u.LowPart % This->parentStorage->smallBlockSize) != 0)
5545 newNumBlocks++;
5548 * Go to the current end of chain
5550 while (blockIndex != BLOCK_END_OF_CHAIN)
5552 oldNumBlocks++;
5553 currentBlock = blockIndex;
5554 if(FAILED(SmallBlockChainStream_GetNextBlockInChain(This, currentBlock, &blockIndex)))
5555 return FALSE;
5559 * Add new blocks to the chain
5561 while (oldNumBlocks < newNumBlocks)
5563 blockIndex = SmallBlockChainStream_GetNextFreeBlock(This);
5564 SmallBlockChainStream_SetNextBlockInChain(This, currentBlock, blockIndex);
5566 SmallBlockChainStream_SetNextBlockInChain(
5567 This,
5568 blockIndex,
5569 BLOCK_END_OF_CHAIN);
5571 currentBlock = blockIndex;
5572 oldNumBlocks++;
5575 return TRUE;
5578 /******************************************************************************
5579 * SmallBlockChainStream_SetSize
5581 * Sets the size of this stream.
5582 * The file will grow if we grow the chain.
5584 * TODO: Free the actual blocks in the file when we shrink the chain.
5585 * Currently, the blocks are still in the file. So the file size
5586 * doesn't shrink even if we shrink streams.
5588 BOOL SmallBlockChainStream_SetSize(
5589 SmallBlockChainStream* This,
5590 ULARGE_INTEGER newSize)
5592 ULARGE_INTEGER size = SmallBlockChainStream_GetSize(This);
5594 if (newSize.u.LowPart == size.u.LowPart)
5595 return TRUE;
5597 if (newSize.u.LowPart < size.u.LowPart)
5599 SmallBlockChainStream_Shrink(This, newSize);
5601 else
5603 SmallBlockChainStream_Enlarge(This, newSize);
5606 return TRUE;
5609 /******************************************************************************
5610 * SmallBlockChainStream_GetSize
5612 * Returns the size of this chain.
5614 static ULARGE_INTEGER SmallBlockChainStream_GetSize(SmallBlockChainStream* This)
5616 StgProperty chainProperty;
5618 StorageImpl_ReadProperty(
5619 This->parentStorage,
5620 This->ownerPropertyIndex,
5621 &chainProperty);
5623 return chainProperty.size;
5626 /******************************************************************************
5627 * StgCreateDocfile [OLE32.@]
5628 * Creates a new compound file storage object
5630 * PARAMS
5631 * pwcsName [ I] Unicode string with filename (can be relative or NULL)
5632 * grfMode [ I] Access mode for opening the new storage object (see STGM_ constants)
5633 * reserved [ ?] unused?, usually 0
5634 * ppstgOpen [IO] A pointer to IStorage pointer to the new onject
5636 * RETURNS
5637 * S_OK if the file was successfully created
5638 * some STG_E_ value if error
5639 * NOTES
5640 * if pwcsName is NULL, create file with new unique name
5641 * the function can returns
5642 * STG_S_CONVERTED if the specified file was successfully converted to storage format
5643 * (unrealized now)
5645 HRESULT WINAPI StgCreateDocfile(
5646 LPCOLESTR pwcsName,
5647 DWORD grfMode,
5648 DWORD reserved,
5649 IStorage **ppstgOpen)
5651 StorageImpl* newStorage = 0;
5652 HANDLE hFile = INVALID_HANDLE_VALUE;
5653 HRESULT hr = STG_E_INVALIDFLAG;
5654 DWORD shareMode;
5655 DWORD accessMode;
5656 DWORD creationMode;
5657 DWORD fileAttributes;
5658 WCHAR tempFileName[MAX_PATH];
5660 TRACE("(%s, %x, %d, %p)\n",
5661 debugstr_w(pwcsName), grfMode,
5662 reserved, ppstgOpen);
5665 * Validate the parameters
5667 if (ppstgOpen == 0)
5668 return STG_E_INVALIDPOINTER;
5669 if (reserved != 0)
5670 return STG_E_INVALIDPARAMETER;
5672 /* if no share mode given then DENY_NONE is the default */
5673 if (STGM_SHARE_MODE(grfMode) == 0)
5674 grfMode |= STGM_SHARE_DENY_NONE;
5677 * Validate the STGM flags
5679 if ( FAILED( validateSTGM(grfMode) ))
5680 goto end;
5682 /* StgCreateDocFile seems to refuse readonly access, despite MSDN */
5683 switch(STGM_ACCESS_MODE(grfMode))
5685 case STGM_WRITE:
5686 case STGM_READWRITE:
5687 break;
5688 default:
5689 goto end;
5692 /* in direct mode, can only use SHARE_EXCLUSIVE */
5693 if (!(grfMode & STGM_TRANSACTED) && (STGM_SHARE_MODE(grfMode) != STGM_SHARE_EXCLUSIVE))
5694 goto end;
5696 /* but in transacted mode, any share mode is valid */
5699 * Generate a unique name.
5701 if (pwcsName == 0)
5703 WCHAR tempPath[MAX_PATH];
5704 static const WCHAR prefix[] = { 'S', 'T', 'O', 0 };
5706 memset(tempPath, 0, sizeof(tempPath));
5707 memset(tempFileName, 0, sizeof(tempFileName));
5709 if ((GetTempPathW(MAX_PATH, tempPath)) == 0 )
5710 tempPath[0] = '.';
5712 if (GetTempFileNameW(tempPath, prefix, 0, tempFileName) != 0)
5713 pwcsName = tempFileName;
5714 else
5716 hr = STG_E_INSUFFICIENTMEMORY;
5717 goto end;
5720 creationMode = TRUNCATE_EXISTING;
5722 else
5724 creationMode = GetCreationModeFromSTGM(grfMode);
5728 * Interpret the STGM value grfMode
5730 shareMode = FILE_SHARE_READ | FILE_SHARE_WRITE;
5731 accessMode = GetAccessModeFromSTGM(grfMode);
5733 if (grfMode & STGM_DELETEONRELEASE)
5734 fileAttributes = FILE_FLAG_RANDOM_ACCESS | FILE_FLAG_DELETE_ON_CLOSE;
5735 else
5736 fileAttributes = FILE_ATTRIBUTE_NORMAL | FILE_FLAG_RANDOM_ACCESS;
5738 if (STGM_SHARE_MODE(grfMode) && !(grfMode & STGM_SHARE_DENY_NONE))
5739 FIXME("Storage share mode not implemented.\n");
5741 if (grfMode & STGM_TRANSACTED)
5742 FIXME("Transacted mode not implemented.\n");
5745 * Initialize the "out" parameter.
5747 *ppstgOpen = 0;
5749 hFile = CreateFileW(pwcsName,
5750 accessMode,
5751 shareMode,
5752 NULL,
5753 creationMode,
5754 fileAttributes,
5757 if (hFile == INVALID_HANDLE_VALUE)
5759 if(GetLastError() == ERROR_FILE_EXISTS)
5760 hr = STG_E_FILEALREADYEXISTS;
5761 else
5762 hr = E_FAIL;
5763 goto end;
5767 * Allocate and initialize the new IStorage32object.
5769 newStorage = HeapAlloc(GetProcessHeap(), 0, sizeof(StorageImpl));
5771 if (newStorage == 0)
5773 hr = STG_E_INSUFFICIENTMEMORY;
5774 goto end;
5777 hr = StorageImpl_Construct(
5778 newStorage,
5779 hFile,
5780 pwcsName,
5781 NULL,
5782 grfMode,
5783 TRUE,
5784 TRUE);
5786 if (FAILED(hr))
5788 HeapFree(GetProcessHeap(), 0, newStorage);
5789 goto end;
5793 * Get an "out" pointer for the caller.
5795 hr = StorageBaseImpl_QueryInterface(
5796 (IStorage*)newStorage,
5797 (REFIID)&IID_IStorage,
5798 (void**)ppstgOpen);
5799 end:
5800 TRACE("<-- %p r = %08x\n", *ppstgOpen, hr);
5802 return hr;
5805 /******************************************************************************
5806 * StgCreateStorageEx [OLE32.@]
5808 HRESULT WINAPI StgCreateStorageEx(const WCHAR* pwcsName, DWORD grfMode, DWORD stgfmt, DWORD grfAttrs, STGOPTIONS* pStgOptions, void* reserved, REFIID riid, void** ppObjectOpen)
5810 TRACE("(%s, %x, %x, %x, %p, %p, %p, %p)\n", debugstr_w(pwcsName),
5811 grfMode, stgfmt, grfAttrs, pStgOptions, reserved, riid, ppObjectOpen);
5813 if (stgfmt != STGFMT_FILE && grfAttrs != 0)
5815 ERR("grfAttrs must be 0 if stgfmt != STGFMT_FILE\n");
5816 return STG_E_INVALIDPARAMETER;
5819 if (stgfmt == STGFMT_FILE && grfAttrs != 0 && grfAttrs != FILE_FLAG_NO_BUFFERING)
5821 ERR("grfAttrs must be 0 or FILE_FLAG_NO_BUFFERING if stgfmt == STGFMT_FILE\n");
5822 return STG_E_INVALIDPARAMETER;
5825 if (stgfmt == STGFMT_FILE)
5827 ERR("Cannot use STGFMT_FILE - this is NTFS only\n");
5828 return STG_E_INVALIDPARAMETER;
5831 if (stgfmt == STGFMT_STORAGE || stgfmt == STGFMT_DOCFILE)
5833 FIXME("Stub: calling StgCreateDocfile, but ignoring pStgOptions and grfAttrs\n");
5834 return StgCreateDocfile(pwcsName, grfMode, 0, (IStorage **)ppObjectOpen);
5837 ERR("Invalid stgfmt argument\n");
5838 return STG_E_INVALIDPARAMETER;
5841 /******************************************************************************
5842 * StgCreatePropSetStg [OLE32.@]
5844 HRESULT WINAPI StgCreatePropSetStg(IStorage *pstg, DWORD reserved,
5845 IPropertySetStorage **ppPropSetStg)
5847 HRESULT hr;
5849 TRACE("(%p, 0x%x, %p)\n", pstg, reserved, ppPropSetStg);
5850 if (reserved)
5851 hr = STG_E_INVALIDPARAMETER;
5852 else
5853 hr = StorageBaseImpl_QueryInterface(pstg, &IID_IPropertySetStorage,
5854 (void**)ppPropSetStg);
5855 return hr;
5858 /******************************************************************************
5859 * StgOpenStorageEx [OLE32.@]
5861 HRESULT WINAPI StgOpenStorageEx(const WCHAR* pwcsName, DWORD grfMode, DWORD stgfmt, DWORD grfAttrs, STGOPTIONS* pStgOptions, void* reserved, REFIID riid, void** ppObjectOpen)
5863 TRACE("(%s, %x, %x, %x, %p, %p, %p, %p)\n", debugstr_w(pwcsName),
5864 grfMode, stgfmt, grfAttrs, pStgOptions, reserved, riid, ppObjectOpen);
5866 if (stgfmt != STGFMT_DOCFILE && grfAttrs != 0)
5868 ERR("grfAttrs must be 0 if stgfmt != STGFMT_DOCFILE\n");
5869 return STG_E_INVALIDPARAMETER;
5872 switch (stgfmt)
5874 case STGFMT_FILE:
5875 ERR("Cannot use STGFMT_FILE - this is NTFS only\n");
5876 return STG_E_INVALIDPARAMETER;
5878 case STGFMT_STORAGE:
5879 break;
5881 case STGFMT_DOCFILE:
5882 if (grfAttrs && grfAttrs != FILE_FLAG_NO_BUFFERING)
5884 ERR("grfAttrs must be 0 or FILE_FLAG_NO_BUFFERING if stgfmt == STGFMT_DOCFILE\n");
5885 return STG_E_INVALIDPARAMETER;
5887 FIXME("Stub: calling StgOpenStorage, but ignoring pStgOptions and grfAttrs\n");
5888 break;
5890 case STGFMT_ANY:
5891 WARN("STGFMT_ANY assuming storage\n");
5892 break;
5894 default:
5895 return STG_E_INVALIDPARAMETER;
5898 return StgOpenStorage(pwcsName, NULL, grfMode, NULL, 0, (IStorage **)ppObjectOpen);
5902 /******************************************************************************
5903 * StgOpenStorage [OLE32.@]
5905 HRESULT WINAPI StgOpenStorage(
5906 const OLECHAR *pwcsName,
5907 IStorage *pstgPriority,
5908 DWORD grfMode,
5909 SNB snbExclude,
5910 DWORD reserved,
5911 IStorage **ppstgOpen)
5913 StorageImpl* newStorage = 0;
5914 HRESULT hr = S_OK;
5915 HANDLE hFile = 0;
5916 DWORD shareMode;
5917 DWORD accessMode;
5918 WCHAR fullname[MAX_PATH];
5920 TRACE("(%s, %p, %x, %p, %d, %p)\n",
5921 debugstr_w(pwcsName), pstgPriority, grfMode,
5922 snbExclude, reserved, ppstgOpen);
5925 * Perform sanity checks
5927 if (pwcsName == 0)
5929 hr = STG_E_INVALIDNAME;
5930 goto end;
5933 if (ppstgOpen == 0)
5935 hr = STG_E_INVALIDPOINTER;
5936 goto end;
5939 if (reserved)
5941 hr = STG_E_INVALIDPARAMETER;
5942 goto end;
5945 if (grfMode & STGM_PRIORITY)
5947 if (grfMode & (STGM_TRANSACTED|STGM_SIMPLE|STGM_NOSCRATCH|STGM_NOSNAPSHOT))
5948 return STG_E_INVALIDFLAG;
5949 if (grfMode & STGM_DELETEONRELEASE)
5950 return STG_E_INVALIDFUNCTION;
5951 if(STGM_ACCESS_MODE(grfMode) != STGM_READ)
5952 return STG_E_INVALIDFLAG;
5953 grfMode &= ~0xf0; /* remove the existing sharing mode */
5954 grfMode |= STGM_SHARE_DENY_NONE;
5956 /* STGM_PRIORITY stops other IStorage objects on the same file from
5957 * committing until the STGM_PRIORITY IStorage is closed. it also
5958 * stops non-transacted mode StgOpenStorage calls with write access from
5959 * succeeding. obviously, both of these cannot be achieved through just
5960 * file share flags */
5961 FIXME("STGM_PRIORITY mode not implemented correctly\n");
5965 * Validate the sharing mode
5967 if (!(grfMode & (STGM_TRANSACTED|STGM_PRIORITY)))
5968 switch(STGM_SHARE_MODE(grfMode))
5970 case STGM_SHARE_EXCLUSIVE:
5971 case STGM_SHARE_DENY_WRITE:
5972 break;
5973 default:
5974 hr = STG_E_INVALIDFLAG;
5975 goto end;
5979 * Validate the STGM flags
5981 if ( FAILED( validateSTGM(grfMode) ) ||
5982 (grfMode&STGM_CREATE))
5984 hr = STG_E_INVALIDFLAG;
5985 goto end;
5988 /* shared reading requires transacted mode */
5989 if( STGM_SHARE_MODE(grfMode) == STGM_SHARE_DENY_WRITE &&
5990 STGM_ACCESS_MODE(grfMode) == STGM_READWRITE &&
5991 !(grfMode&STGM_TRANSACTED) )
5993 hr = STG_E_INVALIDFLAG;
5994 goto end;
5998 * Interpret the STGM value grfMode
6000 shareMode = GetShareModeFromSTGM(grfMode);
6001 accessMode = GetAccessModeFromSTGM(grfMode);
6004 * Initialize the "out" parameter.
6006 *ppstgOpen = 0;
6008 hFile = CreateFileW( pwcsName,
6009 accessMode,
6010 shareMode,
6011 NULL,
6012 OPEN_EXISTING,
6013 FILE_ATTRIBUTE_NORMAL | FILE_FLAG_RANDOM_ACCESS,
6016 if (hFile==INVALID_HANDLE_VALUE)
6018 DWORD last_error = GetLastError();
6020 hr = E_FAIL;
6022 switch (last_error)
6024 case ERROR_FILE_NOT_FOUND:
6025 hr = STG_E_FILENOTFOUND;
6026 break;
6028 case ERROR_PATH_NOT_FOUND:
6029 hr = STG_E_PATHNOTFOUND;
6030 break;
6032 case ERROR_ACCESS_DENIED:
6033 case ERROR_WRITE_PROTECT:
6034 hr = STG_E_ACCESSDENIED;
6035 break;
6037 case ERROR_SHARING_VIOLATION:
6038 hr = STG_E_SHAREVIOLATION;
6039 break;
6041 default:
6042 hr = E_FAIL;
6045 goto end;
6049 * Refuse to open the file if it's too small to be a structured storage file
6050 * FIXME: verify the file when reading instead of here
6052 if (GetFileSize(hFile, NULL) < 0x100)
6054 CloseHandle(hFile);
6055 hr = STG_E_FILEALREADYEXISTS;
6056 goto end;
6060 * Allocate and initialize the new IStorage32object.
6062 newStorage = HeapAlloc(GetProcessHeap(), 0, sizeof(StorageImpl));
6064 if (newStorage == 0)
6066 hr = STG_E_INSUFFICIENTMEMORY;
6067 goto end;
6070 /* Initialize the storage */
6071 hr = StorageImpl_Construct(
6072 newStorage,
6073 hFile,
6074 pwcsName,
6075 NULL,
6076 grfMode,
6077 TRUE,
6078 FALSE );
6080 if (FAILED(hr))
6082 HeapFree(GetProcessHeap(), 0, newStorage);
6084 * According to the docs if the file is not a storage, return STG_E_FILEALREADYEXISTS
6086 if(hr == STG_E_INVALIDHEADER)
6087 hr = STG_E_FILEALREADYEXISTS;
6088 goto end;
6091 /* prepare the file name string given in lieu of the root property name */
6092 GetFullPathNameW(pwcsName, MAX_PATH, fullname, NULL);
6093 memcpy(newStorage->filename, fullname, PROPERTY_NAME_BUFFER_LEN);
6094 newStorage->filename[PROPERTY_NAME_BUFFER_LEN-1] = '\0';
6097 * Get an "out" pointer for the caller.
6099 hr = StorageBaseImpl_QueryInterface(
6100 (IStorage*)newStorage,
6101 (REFIID)&IID_IStorage,
6102 (void**)ppstgOpen);
6104 end:
6105 TRACE("<-- %08x, IStorage %p\n", hr, ppstgOpen ? *ppstgOpen : NULL);
6106 return hr;
6109 /******************************************************************************
6110 * StgCreateDocfileOnILockBytes [OLE32.@]
6112 HRESULT WINAPI StgCreateDocfileOnILockBytes(
6113 ILockBytes *plkbyt,
6114 DWORD grfMode,
6115 DWORD reserved,
6116 IStorage** ppstgOpen)
6118 StorageImpl* newStorage = 0;
6119 HRESULT hr = S_OK;
6122 * Validate the parameters
6124 if ((ppstgOpen == 0) || (plkbyt == 0))
6125 return STG_E_INVALIDPOINTER;
6128 * Allocate and initialize the new IStorage object.
6130 newStorage = HeapAlloc(GetProcessHeap(), 0, sizeof(StorageImpl));
6132 if (newStorage == 0)
6133 return STG_E_INSUFFICIENTMEMORY;
6135 hr = StorageImpl_Construct(
6136 newStorage,
6139 plkbyt,
6140 grfMode,
6141 FALSE,
6142 TRUE);
6144 if (FAILED(hr))
6146 HeapFree(GetProcessHeap(), 0, newStorage);
6147 return hr;
6151 * Get an "out" pointer for the caller.
6153 hr = StorageBaseImpl_QueryInterface(
6154 (IStorage*)newStorage,
6155 (REFIID)&IID_IStorage,
6156 (void**)ppstgOpen);
6158 return hr;
6161 /******************************************************************************
6162 * StgOpenStorageOnILockBytes [OLE32.@]
6164 HRESULT WINAPI StgOpenStorageOnILockBytes(
6165 ILockBytes *plkbyt,
6166 IStorage *pstgPriority,
6167 DWORD grfMode,
6168 SNB snbExclude,
6169 DWORD reserved,
6170 IStorage **ppstgOpen)
6172 StorageImpl* newStorage = 0;
6173 HRESULT hr = S_OK;
6176 * Perform a sanity check
6178 if ((plkbyt == 0) || (ppstgOpen == 0))
6179 return STG_E_INVALIDPOINTER;
6182 * Validate the STGM flags
6184 if ( FAILED( validateSTGM(grfMode) ))
6185 return STG_E_INVALIDFLAG;
6188 * Initialize the "out" parameter.
6190 *ppstgOpen = 0;
6193 * Allocate and initialize the new IStorage object.
6195 newStorage = HeapAlloc(GetProcessHeap(), 0, sizeof(StorageImpl));
6197 if (newStorage == 0)
6198 return STG_E_INSUFFICIENTMEMORY;
6200 hr = StorageImpl_Construct(
6201 newStorage,
6204 plkbyt,
6205 grfMode,
6206 FALSE,
6207 FALSE);
6209 if (FAILED(hr))
6211 HeapFree(GetProcessHeap(), 0, newStorage);
6212 return hr;
6216 * Get an "out" pointer for the caller.
6218 hr = StorageBaseImpl_QueryInterface(
6219 (IStorage*)newStorage,
6220 (REFIID)&IID_IStorage,
6221 (void**)ppstgOpen);
6223 return hr;
6226 /******************************************************************************
6227 * StgSetTimes [ole32.@]
6228 * StgSetTimes [OLE32.@]
6232 HRESULT WINAPI StgSetTimes(OLECHAR const *str, FILETIME const *pctime,
6233 FILETIME const *patime, FILETIME const *pmtime)
6235 IStorage *stg = NULL;
6236 HRESULT r;
6238 TRACE("%s %p %p %p\n", debugstr_w(str), pctime, patime, pmtime);
6240 r = StgOpenStorage(str, NULL, STGM_READWRITE | STGM_SHARE_DENY_WRITE,
6241 0, 0, &stg);
6242 if( SUCCEEDED(r) )
6244 r = IStorage_SetElementTimes(stg, NULL, pctime, patime, pmtime);
6245 IStorage_Release(stg);
6248 return r;
6251 /******************************************************************************
6252 * StgIsStorageILockBytes [OLE32.@]
6254 * Determines if the ILockBytes contains a storage object.
6256 HRESULT WINAPI StgIsStorageILockBytes(ILockBytes *plkbyt)
6258 BYTE sig[8];
6259 ULARGE_INTEGER offset;
6261 offset.u.HighPart = 0;
6262 offset.u.LowPart = 0;
6264 ILockBytes_ReadAt(plkbyt, offset, sig, sizeof(sig), NULL);
6266 if (memcmp(sig, STORAGE_magic, sizeof(STORAGE_magic)) == 0)
6267 return S_OK;
6269 return S_FALSE;
6272 /******************************************************************************
6273 * WriteClassStg [OLE32.@]
6275 * This method will store the specified CLSID in the specified storage object
6277 HRESULT WINAPI WriteClassStg(IStorage* pStg, REFCLSID rclsid)
6279 HRESULT hRes;
6281 if(!pStg)
6282 return E_INVALIDARG;
6284 hRes = IStorage_SetClass(pStg, rclsid);
6286 return hRes;
6289 /***********************************************************************
6290 * ReadClassStg (OLE32.@)
6292 * This method reads the CLSID previously written to a storage object with
6293 * the WriteClassStg.
6295 * PARAMS
6296 * pstg [I] IStorage pointer
6297 * pclsid [O] Pointer to where the CLSID is written
6299 * RETURNS
6300 * Success: S_OK.
6301 * Failure: HRESULT code.
6303 HRESULT WINAPI ReadClassStg(IStorage *pstg,CLSID *pclsid){
6305 STATSTG pstatstg;
6306 HRESULT hRes;
6308 TRACE("(%p, %p)\n", pstg, pclsid);
6310 if(!pstg || !pclsid)
6311 return E_INVALIDARG;
6314 * read a STATSTG structure (contains the clsid) from the storage
6316 hRes=IStorage_Stat(pstg,&pstatstg,STATFLAG_DEFAULT);
6318 if(SUCCEEDED(hRes))
6319 *pclsid=pstatstg.clsid;
6321 return hRes;
6324 /***********************************************************************
6325 * OleLoadFromStream (OLE32.@)
6327 * This function loads an object from stream
6329 HRESULT WINAPI OleLoadFromStream(IStream *pStm,REFIID iidInterface,void** ppvObj)
6331 CLSID clsid;
6332 HRESULT res;
6333 LPPERSISTSTREAM xstm;
6335 TRACE("(%p,%s,%p)\n",pStm,debugstr_guid(iidInterface),ppvObj);
6337 res=ReadClassStm(pStm,&clsid);
6338 if (FAILED(res))
6339 return res;
6340 res=CoCreateInstance(&clsid,NULL,CLSCTX_INPROC_SERVER,iidInterface,ppvObj);
6341 if (FAILED(res))
6342 return res;
6343 res=IUnknown_QueryInterface((IUnknown*)*ppvObj,&IID_IPersistStream,(LPVOID*)&xstm);
6344 if (FAILED(res)) {
6345 IUnknown_Release((IUnknown*)*ppvObj);
6346 return res;
6348 res=IPersistStream_Load(xstm,pStm);
6349 IPersistStream_Release(xstm);
6350 /* FIXME: all refcounts ok at this point? I think they should be:
6351 * pStm : unchanged
6352 * ppvObj : 1
6353 * xstm : 0 (released)
6355 return res;
6358 /***********************************************************************
6359 * OleSaveToStream (OLE32.@)
6361 * This function saves an object with the IPersistStream interface on it
6362 * to the specified stream.
6364 HRESULT WINAPI OleSaveToStream(IPersistStream *pPStm,IStream *pStm)
6367 CLSID clsid;
6368 HRESULT res;
6370 TRACE("(%p,%p)\n",pPStm,pStm);
6372 res=IPersistStream_GetClassID(pPStm,&clsid);
6374 if (SUCCEEDED(res)){
6376 res=WriteClassStm(pStm,&clsid);
6378 if (SUCCEEDED(res))
6380 res=IPersistStream_Save(pPStm,pStm,TRUE);
6383 TRACE("Finished Save\n");
6384 return res;
6387 /****************************************************************************
6388 * This method validate a STGM parameter that can contain the values below
6390 * The stgm modes in 0x0000ffff are not bit masks, but distinct 4 bit values.
6391 * The stgm values contained in 0xffff0000 are bitmasks.
6393 * STGM_DIRECT 0x00000000
6394 * STGM_TRANSACTED 0x00010000
6395 * STGM_SIMPLE 0x08000000
6397 * STGM_READ 0x00000000
6398 * STGM_WRITE 0x00000001
6399 * STGM_READWRITE 0x00000002
6401 * STGM_SHARE_DENY_NONE 0x00000040
6402 * STGM_SHARE_DENY_READ 0x00000030
6403 * STGM_SHARE_DENY_WRITE 0x00000020
6404 * STGM_SHARE_EXCLUSIVE 0x00000010
6406 * STGM_PRIORITY 0x00040000
6407 * STGM_DELETEONRELEASE 0x04000000
6409 * STGM_CREATE 0x00001000
6410 * STGM_CONVERT 0x00020000
6411 * STGM_FAILIFTHERE 0x00000000
6413 * STGM_NOSCRATCH 0x00100000
6414 * STGM_NOSNAPSHOT 0x00200000
6416 static HRESULT validateSTGM(DWORD stgm)
6418 DWORD access = STGM_ACCESS_MODE(stgm);
6419 DWORD share = STGM_SHARE_MODE(stgm);
6420 DWORD create = STGM_CREATE_MODE(stgm);
6422 if (stgm&~STGM_KNOWN_FLAGS)
6424 ERR("unknown flags %08x\n", stgm);
6425 return E_FAIL;
6428 switch (access)
6430 case STGM_READ:
6431 case STGM_WRITE:
6432 case STGM_READWRITE:
6433 break;
6434 default:
6435 return E_FAIL;
6438 switch (share)
6440 case STGM_SHARE_DENY_NONE:
6441 case STGM_SHARE_DENY_READ:
6442 case STGM_SHARE_DENY_WRITE:
6443 case STGM_SHARE_EXCLUSIVE:
6444 break;
6445 default:
6446 return E_FAIL;
6449 switch (create)
6451 case STGM_CREATE:
6452 case STGM_FAILIFTHERE:
6453 break;
6454 default:
6455 return E_FAIL;
6459 * STGM_DIRECT | STGM_TRANSACTED | STGM_SIMPLE
6461 if ( (stgm & STGM_TRANSACTED) && (stgm & STGM_SIMPLE) )
6462 return E_FAIL;
6465 * STGM_CREATE | STGM_CONVERT
6466 * if both are false, STGM_FAILIFTHERE is set to TRUE
6468 if ( create == STGM_CREATE && (stgm & STGM_CONVERT) )
6469 return E_FAIL;
6472 * STGM_NOSCRATCH requires STGM_TRANSACTED
6474 if ( (stgm & STGM_NOSCRATCH) && !(stgm & STGM_TRANSACTED) )
6475 return E_FAIL;
6478 * STGM_NOSNAPSHOT requires STGM_TRANSACTED and
6479 * not STGM_SHARE_EXCLUSIVE or STGM_SHARE_DENY_WRITE`
6481 if ( (stgm & STGM_NOSNAPSHOT) &&
6482 (!(stgm & STGM_TRANSACTED) ||
6483 share == STGM_SHARE_EXCLUSIVE ||
6484 share == STGM_SHARE_DENY_WRITE) )
6485 return E_FAIL;
6487 return S_OK;
6490 /****************************************************************************
6491 * GetShareModeFromSTGM
6493 * This method will return a share mode flag from a STGM value.
6494 * The STGM value is assumed valid.
6496 static DWORD GetShareModeFromSTGM(DWORD stgm)
6498 switch (STGM_SHARE_MODE(stgm))
6500 case STGM_SHARE_DENY_NONE:
6501 return FILE_SHARE_READ | FILE_SHARE_WRITE;
6502 case STGM_SHARE_DENY_READ:
6503 return FILE_SHARE_WRITE;
6504 case STGM_SHARE_DENY_WRITE:
6505 return FILE_SHARE_READ;
6506 case STGM_SHARE_EXCLUSIVE:
6507 return 0;
6509 ERR("Invalid share mode!\n");
6510 assert(0);
6511 return 0;
6514 /****************************************************************************
6515 * GetAccessModeFromSTGM
6517 * This method will return an access mode flag from a STGM value.
6518 * The STGM value is assumed valid.
6520 static DWORD GetAccessModeFromSTGM(DWORD stgm)
6522 switch (STGM_ACCESS_MODE(stgm))
6524 case STGM_READ:
6525 return GENERIC_READ;
6526 case STGM_WRITE:
6527 case STGM_READWRITE:
6528 return GENERIC_READ | GENERIC_WRITE;
6530 ERR("Invalid access mode!\n");
6531 assert(0);
6532 return 0;
6535 /****************************************************************************
6536 * GetCreationModeFromSTGM
6538 * This method will return a creation mode flag from a STGM value.
6539 * The STGM value is assumed valid.
6541 static DWORD GetCreationModeFromSTGM(DWORD stgm)
6543 switch(STGM_CREATE_MODE(stgm))
6545 case STGM_CREATE:
6546 return CREATE_ALWAYS;
6547 case STGM_CONVERT:
6548 FIXME("STGM_CONVERT not implemented!\n");
6549 return CREATE_NEW;
6550 case STGM_FAILIFTHERE:
6551 return CREATE_NEW;
6553 ERR("Invalid create mode!\n");
6554 assert(0);
6555 return 0;
6559 /*************************************************************************
6560 * OLECONVERT_LoadOLE10 [Internal]
6562 * Loads the OLE10 STREAM to memory
6564 * PARAMS
6565 * pOleStream [I] The OLESTREAM
6566 * pData [I] Data Structure for the OLESTREAM Data
6568 * RETURNS
6569 * Success: S_OK
6570 * Failure: CONVERT10_E_OLESTREAM_GET for invalid Get
6571 * CONVERT10_E_OLESTREAM_FMT if the OLEID is invalid
6573 * NOTES
6574 * This function is used by OleConvertOLESTREAMToIStorage only.
6576 * Memory allocated for pData must be freed by the caller
6578 static HRESULT OLECONVERT_LoadOLE10(LPOLESTREAM pOleStream, OLECONVERT_OLESTREAM_DATA *pData, BOOL bStrem1)
6580 DWORD dwSize;
6581 HRESULT hRes = S_OK;
6582 int nTryCnt=0;
6583 int max_try = 6;
6585 pData->pData = NULL;
6586 pData->pstrOleObjFileName = NULL;
6588 for( nTryCnt=0;nTryCnt < max_try; nTryCnt++)
6590 /* Get the OleID */
6591 dwSize = pOleStream->lpstbl->Get(pOleStream, (void *)&(pData->dwOleID), sizeof(pData->dwOleID));
6592 if(dwSize != sizeof(pData->dwOleID))
6594 hRes = CONVERT10_E_OLESTREAM_GET;
6596 else if(pData->dwOleID != OLESTREAM_ID)
6598 hRes = CONVERT10_E_OLESTREAM_FMT;
6600 else
6602 hRes = S_OK;
6603 break;
6607 if(hRes == S_OK)
6609 /* Get the TypeID...more info needed for this field */
6610 dwSize = pOleStream->lpstbl->Get(pOleStream, (void *)&(pData->dwTypeID), sizeof(pData->dwTypeID));
6611 if(dwSize != sizeof(pData->dwTypeID))
6613 hRes = CONVERT10_E_OLESTREAM_GET;
6616 if(hRes == S_OK)
6618 if(pData->dwTypeID != 0)
6620 /* Get the length of the OleTypeName */
6621 dwSize = pOleStream->lpstbl->Get(pOleStream, (void *) &(pData->dwOleTypeNameLength), sizeof(pData->dwOleTypeNameLength));
6622 if(dwSize != sizeof(pData->dwOleTypeNameLength))
6624 hRes = CONVERT10_E_OLESTREAM_GET;
6627 if(hRes == S_OK)
6629 if(pData->dwOleTypeNameLength > 0)
6631 /* Get the OleTypeName */
6632 dwSize = pOleStream->lpstbl->Get(pOleStream, (void *)pData->strOleTypeName, pData->dwOleTypeNameLength);
6633 if(dwSize != pData->dwOleTypeNameLength)
6635 hRes = CONVERT10_E_OLESTREAM_GET;
6639 if(bStrem1)
6641 dwSize = pOleStream->lpstbl->Get(pOleStream, (void *)&(pData->dwOleObjFileNameLength), sizeof(pData->dwOleObjFileNameLength));
6642 if(dwSize != sizeof(pData->dwOleObjFileNameLength))
6644 hRes = CONVERT10_E_OLESTREAM_GET;
6646 if(hRes == S_OK)
6648 if(pData->dwOleObjFileNameLength < 1) /* there is no file name exist */
6649 pData->dwOleObjFileNameLength = sizeof(pData->dwOleObjFileNameLength);
6650 pData->pstrOleObjFileName = HeapAlloc(GetProcessHeap(), 0, pData->dwOleObjFileNameLength);
6651 if(pData->pstrOleObjFileName)
6653 dwSize = pOleStream->lpstbl->Get(pOleStream, (void *)(pData->pstrOleObjFileName),pData->dwOleObjFileNameLength);
6654 if(dwSize != pData->dwOleObjFileNameLength)
6656 hRes = CONVERT10_E_OLESTREAM_GET;
6659 else
6660 hRes = CONVERT10_E_OLESTREAM_GET;
6663 else
6665 /* Get the Width of the Metafile */
6666 dwSize = pOleStream->lpstbl->Get(pOleStream, (void *)&(pData->dwMetaFileWidth), sizeof(pData->dwMetaFileWidth));
6667 if(dwSize != sizeof(pData->dwMetaFileWidth))
6669 hRes = CONVERT10_E_OLESTREAM_GET;
6671 if(hRes == S_OK)
6673 /* Get the Height of the Metafile */
6674 dwSize = pOleStream->lpstbl->Get(pOleStream, (void *)&(pData->dwMetaFileHeight), sizeof(pData->dwMetaFileHeight));
6675 if(dwSize != sizeof(pData->dwMetaFileHeight))
6677 hRes = CONVERT10_E_OLESTREAM_GET;
6681 if(hRes == S_OK)
6683 /* Get the Length of the Data */
6684 dwSize = pOleStream->lpstbl->Get(pOleStream, (void *)&(pData->dwDataLength), sizeof(pData->dwDataLength));
6685 if(dwSize != sizeof(pData->dwDataLength))
6687 hRes = CONVERT10_E_OLESTREAM_GET;
6691 if(hRes == S_OK) /* I don't know what this 8 byte information is. We have to figure out */
6693 if(!bStrem1) /* if it is a second OLE stream data */
6695 pData->dwDataLength -= 8;
6696 dwSize = pOleStream->lpstbl->Get(pOleStream, (void *)(pData->strUnknown), sizeof(pData->strUnknown));
6697 if(dwSize != sizeof(pData->strUnknown))
6699 hRes = CONVERT10_E_OLESTREAM_GET;
6703 if(hRes == S_OK)
6705 if(pData->dwDataLength > 0)
6707 pData->pData = HeapAlloc(GetProcessHeap(),0,pData->dwDataLength);
6709 /* Get Data (ex. IStorage, Metafile, or BMP) */
6710 if(pData->pData)
6712 dwSize = pOleStream->lpstbl->Get(pOleStream, (void *)pData->pData, pData->dwDataLength);
6713 if(dwSize != pData->dwDataLength)
6715 hRes = CONVERT10_E_OLESTREAM_GET;
6718 else
6720 hRes = CONVERT10_E_OLESTREAM_GET;
6726 return hRes;
6729 /*************************************************************************
6730 * OLECONVERT_SaveOLE10 [Internal]
6732 * Saves the OLE10 STREAM From memory
6734 * PARAMS
6735 * pData [I] Data Structure for the OLESTREAM Data
6736 * pOleStream [I] The OLESTREAM to save
6738 * RETURNS
6739 * Success: S_OK
6740 * Failure: CONVERT10_E_OLESTREAM_PUT for invalid Put
6742 * NOTES
6743 * This function is used by OleConvertIStorageToOLESTREAM only.
6746 static HRESULT OLECONVERT_SaveOLE10(OLECONVERT_OLESTREAM_DATA *pData, LPOLESTREAM pOleStream)
6748 DWORD dwSize;
6749 HRESULT hRes = S_OK;
6752 /* Set the OleID */
6753 dwSize = pOleStream->lpstbl->Put(pOleStream, (void *)&(pData->dwOleID), sizeof(pData->dwOleID));
6754 if(dwSize != sizeof(pData->dwOleID))
6756 hRes = CONVERT10_E_OLESTREAM_PUT;
6759 if(hRes == S_OK)
6761 /* Set the TypeID */
6762 dwSize = pOleStream->lpstbl->Put(pOleStream, (void *)&(pData->dwTypeID), sizeof(pData->dwTypeID));
6763 if(dwSize != sizeof(pData->dwTypeID))
6765 hRes = CONVERT10_E_OLESTREAM_PUT;
6769 if(pData->dwOleID == OLESTREAM_ID && pData->dwTypeID != 0 && hRes == S_OK)
6771 /* Set the Length of the OleTypeName */
6772 dwSize = pOleStream->lpstbl->Put(pOleStream, (void *)&(pData->dwOleTypeNameLength), sizeof(pData->dwOleTypeNameLength));
6773 if(dwSize != sizeof(pData->dwOleTypeNameLength))
6775 hRes = CONVERT10_E_OLESTREAM_PUT;
6778 if(hRes == S_OK)
6780 if(pData->dwOleTypeNameLength > 0)
6782 /* Set the OleTypeName */
6783 dwSize = pOleStream->lpstbl->Put(pOleStream, (void *) pData->strOleTypeName, pData->dwOleTypeNameLength);
6784 if(dwSize != pData->dwOleTypeNameLength)
6786 hRes = CONVERT10_E_OLESTREAM_PUT;
6791 if(hRes == S_OK)
6793 /* Set the width of the Metafile */
6794 dwSize = pOleStream->lpstbl->Put(pOleStream, (void *)&(pData->dwMetaFileWidth), sizeof(pData->dwMetaFileWidth));
6795 if(dwSize != sizeof(pData->dwMetaFileWidth))
6797 hRes = CONVERT10_E_OLESTREAM_PUT;
6801 if(hRes == S_OK)
6803 /* Set the height of the Metafile */
6804 dwSize = pOleStream->lpstbl->Put(pOleStream, (void *)&(pData->dwMetaFileHeight), sizeof(pData->dwMetaFileHeight));
6805 if(dwSize != sizeof(pData->dwMetaFileHeight))
6807 hRes = CONVERT10_E_OLESTREAM_PUT;
6811 if(hRes == S_OK)
6813 /* Set the length of the Data */
6814 dwSize = pOleStream->lpstbl->Put(pOleStream, (void *)&(pData->dwDataLength), sizeof(pData->dwDataLength));
6815 if(dwSize != sizeof(pData->dwDataLength))
6817 hRes = CONVERT10_E_OLESTREAM_PUT;
6821 if(hRes == S_OK)
6823 if(pData->dwDataLength > 0)
6825 /* Set the Data (eg. IStorage, Metafile, Bitmap) */
6826 dwSize = pOleStream->lpstbl->Put(pOleStream, (void *) pData->pData, pData->dwDataLength);
6827 if(dwSize != pData->dwDataLength)
6829 hRes = CONVERT10_E_OLESTREAM_PUT;
6834 return hRes;
6837 /*************************************************************************
6838 * OLECONVERT_GetOLE20FromOLE10[Internal]
6840 * This function copies OLE10 Data (the IStorage in the OLESTREAM) to disk,
6841 * opens it, and copies the content to the dest IStorage for
6842 * OleConvertOLESTREAMToIStorage
6845 * PARAMS
6846 * pDestStorage [I] The IStorage to copy the data to
6847 * pBuffer [I] Buffer that contains the IStorage from the OLESTREAM
6848 * nBufferLength [I] The size of the buffer
6850 * RETURNS
6851 * Nothing
6853 * NOTES
6857 static void OLECONVERT_GetOLE20FromOLE10(LPSTORAGE pDestStorage, const BYTE *pBuffer, DWORD nBufferLength)
6859 HRESULT hRes;
6860 HANDLE hFile;
6861 IStorage *pTempStorage;
6862 DWORD dwNumOfBytesWritten;
6863 WCHAR wstrTempDir[MAX_PATH], wstrTempFile[MAX_PATH];
6864 static const WCHAR wstrPrefix[] = {'s', 'i', 's', 0};
6866 /* Create a temp File */
6867 GetTempPathW(MAX_PATH, wstrTempDir);
6868 GetTempFileNameW(wstrTempDir, wstrPrefix, 0, wstrTempFile);
6869 hFile = CreateFileW(wstrTempFile, GENERIC_WRITE, 0, NULL, CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, 0);
6871 if(hFile != INVALID_HANDLE_VALUE)
6873 /* Write IStorage Data to File */
6874 WriteFile(hFile, pBuffer, nBufferLength, &dwNumOfBytesWritten, NULL);
6875 CloseHandle(hFile);
6877 /* Open and copy temp storage to the Dest Storage */
6878 hRes = StgOpenStorage(wstrTempFile, NULL, STGM_READ, NULL, 0, &pTempStorage);
6879 if(hRes == S_OK)
6881 hRes = StorageImpl_CopyTo(pTempStorage, 0, NULL, NULL, pDestStorage);
6882 StorageBaseImpl_Release(pTempStorage);
6884 DeleteFileW(wstrTempFile);
6889 /*************************************************************************
6890 * OLECONVERT_WriteOLE20ToBuffer [Internal]
6892 * Saves the OLE10 STREAM From memory
6894 * PARAMS
6895 * pStorage [I] The Src IStorage to copy
6896 * pData [I] The Dest Memory to write to.
6898 * RETURNS
6899 * The size in bytes allocated for pData
6901 * NOTES
6902 * Memory allocated for pData must be freed by the caller
6904 * Used by OleConvertIStorageToOLESTREAM only.
6907 static DWORD OLECONVERT_WriteOLE20ToBuffer(LPSTORAGE pStorage, BYTE **pData)
6909 HANDLE hFile;
6910 HRESULT hRes;
6911 DWORD nDataLength = 0;
6912 IStorage *pTempStorage;
6913 WCHAR wstrTempDir[MAX_PATH], wstrTempFile[MAX_PATH];
6914 static const WCHAR wstrPrefix[] = {'s', 'i', 's', 0};
6916 *pData = NULL;
6918 /* Create temp Storage */
6919 GetTempPathW(MAX_PATH, wstrTempDir);
6920 GetTempFileNameW(wstrTempDir, wstrPrefix, 0, wstrTempFile);
6921 hRes = StgCreateDocfile(wstrTempFile, STGM_CREATE | STGM_READWRITE | STGM_SHARE_EXCLUSIVE, 0, &pTempStorage);
6923 if(hRes == S_OK)
6925 /* Copy Src Storage to the Temp Storage */
6926 StorageImpl_CopyTo(pStorage, 0, NULL, NULL, pTempStorage);
6927 StorageBaseImpl_Release(pTempStorage);
6929 /* Open Temp Storage as a file and copy to memory */
6930 hFile = CreateFileW(wstrTempFile, GENERIC_READ, 0, NULL, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, 0);
6931 if(hFile != INVALID_HANDLE_VALUE)
6933 nDataLength = GetFileSize(hFile, NULL);
6934 *pData = HeapAlloc(GetProcessHeap(),0,nDataLength);
6935 ReadFile(hFile, *pData, nDataLength, &nDataLength, 0);
6936 CloseHandle(hFile);
6938 DeleteFileW(wstrTempFile);
6940 return nDataLength;
6943 /*************************************************************************
6944 * OLECONVERT_CreateOleStream [Internal]
6946 * Creates the "\001OLE" stream in the IStorage if necessary.
6948 * PARAMS
6949 * pStorage [I] Dest storage to create the stream in
6951 * RETURNS
6952 * Nothing
6954 * NOTES
6955 * This function is used by OleConvertOLESTREAMToIStorage only.
6957 * This stream is still unknown, MS Word seems to have extra data
6958 * but since the data is stored in the OLESTREAM there should be
6959 * no need to recreate the stream. If the stream is manually
6960 * deleted it will create it with this default data.
6963 void OLECONVERT_CreateOleStream(LPSTORAGE pStorage)
6965 HRESULT hRes;
6966 IStream *pStream;
6967 static const WCHAR wstrStreamName[] = {1,'O', 'l', 'e', 0};
6968 BYTE pOleStreamHeader [] =
6970 0x01, 0x00, 0x00, 0x02, 0x00, 0x00, 0x00, 0x00,
6971 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
6972 0x00, 0x00, 0x00, 0x00
6975 /* Create stream if not present */
6976 hRes = IStorage_CreateStream(pStorage, wstrStreamName,
6977 STGM_WRITE | STGM_SHARE_EXCLUSIVE, 0, 0, &pStream );
6979 if(hRes == S_OK)
6981 /* Write default Data */
6982 hRes = IStream_Write(pStream, pOleStreamHeader, sizeof(pOleStreamHeader), NULL);
6983 IStream_Release(pStream);
6987 /* write a string to a stream, preceded by its length */
6988 static HRESULT STREAM_WriteString( IStream *stm, LPCWSTR string )
6990 HRESULT r;
6991 LPSTR str;
6992 DWORD len = 0;
6994 if( string )
6995 len = WideCharToMultiByte( CP_ACP, 0, string, -1, NULL, 0, NULL, NULL);
6996 r = IStream_Write( stm, &len, sizeof(len), NULL);
6997 if( FAILED( r ) )
6998 return r;
6999 if(len == 0)
7000 return r;
7001 str = CoTaskMemAlloc( len );
7002 WideCharToMultiByte( CP_ACP, 0, string, -1, str, len, NULL, NULL);
7003 r = IStream_Write( stm, str, len, NULL);
7004 CoTaskMemFree( str );
7005 return r;
7008 /* read a string preceded by its length from a stream */
7009 static HRESULT STREAM_ReadString( IStream *stm, LPWSTR *string )
7011 HRESULT r;
7012 DWORD len, count = 0;
7013 LPSTR str;
7014 LPWSTR wstr;
7016 r = IStream_Read( stm, &len, sizeof(len), &count );
7017 if( FAILED( r ) )
7018 return r;
7019 if( count != sizeof(len) )
7020 return E_OUTOFMEMORY;
7022 TRACE("%d bytes\n",len);
7024 str = CoTaskMemAlloc( len );
7025 if( !str )
7026 return E_OUTOFMEMORY;
7027 count = 0;
7028 r = IStream_Read( stm, str, len, &count );
7029 if( FAILED( r ) )
7030 return r;
7031 if( count != len )
7033 CoTaskMemFree( str );
7034 return E_OUTOFMEMORY;
7037 TRACE("Read string %s\n",debugstr_an(str,len));
7039 len = MultiByteToWideChar( CP_ACP, 0, str, count, NULL, 0 );
7040 wstr = CoTaskMemAlloc( (len + 1)*sizeof (WCHAR) );
7041 if( wstr )
7042 MultiByteToWideChar( CP_ACP, 0, str, count, wstr, len );
7043 CoTaskMemFree( str );
7045 *string = wstr;
7047 return r;
7051 static HRESULT STORAGE_WriteCompObj( LPSTORAGE pstg, CLSID *clsid,
7052 LPCWSTR lpszUserType, LPCWSTR szClipName, LPCWSTR szProgIDName )
7054 IStream *pstm;
7055 HRESULT r = S_OK;
7056 static const WCHAR szwStreamName[] = {1, 'C', 'o', 'm', 'p', 'O', 'b', 'j', 0};
7058 static const BYTE unknown1[12] =
7059 { 0x01, 0x00, 0xFE, 0xFF, 0x03, 0x0A, 0x00, 0x00,
7060 0xFF, 0xFF, 0xFF, 0xFF};
7061 static const BYTE unknown2[16] =
7062 { 0xF4, 0x39, 0xB2, 0x71, 0x00, 0x00, 0x00, 0x00,
7063 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00 };
7065 TRACE("%p %s %s %s %s\n", pstg, debugstr_guid(clsid),
7066 debugstr_w(lpszUserType), debugstr_w(szClipName),
7067 debugstr_w(szProgIDName));
7069 /* Create a CompObj stream if it doesn't exist */
7070 r = IStorage_CreateStream(pstg, szwStreamName,
7071 STGM_WRITE | STGM_SHARE_EXCLUSIVE, 0, 0, &pstm );
7072 if( FAILED (r) )
7073 return r;
7075 /* Write CompObj Structure to stream */
7076 r = IStream_Write(pstm, unknown1, sizeof(unknown1), NULL);
7078 if( SUCCEEDED( r ) )
7079 r = WriteClassStm( pstm, clsid );
7081 if( SUCCEEDED( r ) )
7082 r = STREAM_WriteString( pstm, lpszUserType );
7083 if( SUCCEEDED( r ) )
7084 r = STREAM_WriteString( pstm, szClipName );
7085 if( SUCCEEDED( r ) )
7086 r = STREAM_WriteString( pstm, szProgIDName );
7087 if( SUCCEEDED( r ) )
7088 r = IStream_Write(pstm, unknown2, sizeof(unknown2), NULL);
7090 IStream_Release( pstm );
7092 return r;
7095 /***********************************************************************
7096 * WriteFmtUserTypeStg (OLE32.@)
7098 HRESULT WINAPI WriteFmtUserTypeStg(
7099 LPSTORAGE pstg, CLIPFORMAT cf, LPOLESTR lpszUserType)
7101 HRESULT r;
7102 WCHAR szwClipName[0x40];
7103 CLSID clsid = CLSID_NULL;
7104 LPWSTR wstrProgID = NULL;
7105 DWORD n;
7107 TRACE("(%p,%x,%s)\n",pstg,cf,debugstr_w(lpszUserType));
7109 /* get the clipboard format name */
7110 n = GetClipboardFormatNameW( cf, szwClipName, sizeof(szwClipName)/sizeof(szwClipName[0]) );
7111 szwClipName[n]=0;
7113 TRACE("Clipboard name is %s\n", debugstr_w(szwClipName));
7115 /* FIXME: There's room to save a CLSID and its ProgID, but
7116 the CLSID is not looked up in the registry and in all the
7117 tests I wrote it was CLSID_NULL. Where does it come from?
7120 /* get the real program ID. This may fail, but that's fine */
7121 ProgIDFromCLSID(&clsid, &wstrProgID);
7123 TRACE("progid is %s\n",debugstr_w(wstrProgID));
7125 r = STORAGE_WriteCompObj( pstg, &clsid,
7126 lpszUserType, szwClipName, wstrProgID );
7128 CoTaskMemFree(wstrProgID);
7130 return r;
7134 /******************************************************************************
7135 * ReadFmtUserTypeStg [OLE32.@]
7137 HRESULT WINAPI ReadFmtUserTypeStg (LPSTORAGE pstg, CLIPFORMAT* pcf, LPOLESTR* lplpszUserType)
7139 HRESULT r;
7140 IStream *stm = 0;
7141 static const WCHAR szCompObj[] = { 1, 'C','o','m','p','O','b','j', 0 };
7142 unsigned char unknown1[12];
7143 unsigned char unknown2[16];
7144 DWORD count;
7145 LPWSTR szProgIDName = NULL, szCLSIDName = NULL, szOleTypeName = NULL;
7146 CLSID clsid;
7148 TRACE("(%p,%p,%p)\n", pstg, pcf, lplpszUserType);
7150 r = IStorage_OpenStream( pstg, szCompObj, NULL,
7151 STGM_READ | STGM_SHARE_EXCLUSIVE, 0, &stm );
7152 if( FAILED ( r ) )
7154 WARN("Failed to open stream r = %08x\n", r);
7155 return r;
7158 /* read the various parts of the structure */
7159 r = IStream_Read( stm, unknown1, sizeof(unknown1), &count );
7160 if( FAILED( r ) || ( count != sizeof(unknown1) ) )
7161 goto end;
7162 r = ReadClassStm( stm, &clsid );
7163 if( FAILED( r ) )
7164 goto end;
7166 r = STREAM_ReadString( stm, &szCLSIDName );
7167 if( FAILED( r ) )
7168 goto end;
7170 r = STREAM_ReadString( stm, &szOleTypeName );
7171 if( FAILED( r ) )
7172 goto end;
7174 r = STREAM_ReadString( stm, &szProgIDName );
7175 if( FAILED( r ) )
7176 goto end;
7178 r = IStream_Read( stm, unknown2, sizeof(unknown2), &count );
7179 if( FAILED( r ) || ( count != sizeof(unknown2) ) )
7180 goto end;
7182 /* ok, success... now we just need to store what we found */
7183 if( pcf )
7184 *pcf = RegisterClipboardFormatW( szOleTypeName );
7185 CoTaskMemFree( szOleTypeName );
7187 if( lplpszUserType )
7188 *lplpszUserType = szCLSIDName;
7189 CoTaskMemFree( szProgIDName );
7191 end:
7192 IStream_Release( stm );
7194 return r;
7198 /*************************************************************************
7199 * OLECONVERT_CreateCompObjStream [Internal]
7201 * Creates a "\001CompObj" is the destination IStorage if necessary.
7203 * PARAMS
7204 * pStorage [I] The dest IStorage to create the CompObj Stream
7205 * if necessary.
7206 * strOleTypeName [I] The ProgID
7208 * RETURNS
7209 * Success: S_OK
7210 * Failure: REGDB_E_CLASSNOTREG if cannot reconstruct the stream
7212 * NOTES
7213 * This function is used by OleConvertOLESTREAMToIStorage only.
7215 * The stream data is stored in the OLESTREAM and there should be
7216 * no need to recreate the stream. If the stream is manually
7217 * deleted it will attempt to create it by querying the registry.
7221 HRESULT OLECONVERT_CreateCompObjStream(LPSTORAGE pStorage, LPCSTR strOleTypeName)
7223 IStream *pStream;
7224 HRESULT hStorageRes, hRes = S_OK;
7225 OLECONVERT_ISTORAGE_COMPOBJ IStorageCompObj;
7226 static const WCHAR wstrStreamName[] = {1,'C', 'o', 'm', 'p', 'O', 'b', 'j', 0};
7227 WCHAR bufferW[OLESTREAM_MAX_STR_LEN];
7229 BYTE pCompObjUnknown1[] = {0x01, 0x00, 0xFE, 0xFF, 0x03, 0x0A, 0x00, 0x00, 0xFF, 0xFF, 0xFF, 0xFF};
7230 BYTE pCompObjUnknown2[] = {0xF4, 0x39, 0xB2, 0x71};
7232 /* Initialize the CompObj structure */
7233 memset(&IStorageCompObj, 0, sizeof(IStorageCompObj));
7234 memcpy(IStorageCompObj.byUnknown1, pCompObjUnknown1, sizeof(pCompObjUnknown1));
7235 memcpy(IStorageCompObj.byUnknown2, pCompObjUnknown2, sizeof(pCompObjUnknown2));
7238 /* Create a CompObj stream if it doesn't exist */
7239 hStorageRes = IStorage_CreateStream(pStorage, wstrStreamName,
7240 STGM_WRITE | STGM_SHARE_EXCLUSIVE, 0, 0, &pStream );
7241 if(hStorageRes == S_OK)
7243 /* copy the OleTypeName to the compobj struct */
7244 IStorageCompObj.dwOleTypeNameLength = strlen(strOleTypeName)+1;
7245 strcpy(IStorageCompObj.strOleTypeName, strOleTypeName);
7247 /* copy the OleTypeName to the compobj struct */
7248 /* Note: in the test made, these were Identical */
7249 IStorageCompObj.dwProgIDNameLength = strlen(strOleTypeName)+1;
7250 strcpy(IStorageCompObj.strProgIDName, strOleTypeName);
7252 /* Get the CLSID */
7253 MultiByteToWideChar( CP_ACP, 0, IStorageCompObj.strProgIDName, -1,
7254 bufferW, OLESTREAM_MAX_STR_LEN );
7255 hRes = CLSIDFromProgID(bufferW, &(IStorageCompObj.clsid));
7257 if(hRes == S_OK)
7259 HKEY hKey;
7260 LONG hErr;
7261 /* Get the CLSID Default Name from the Registry */
7262 hErr = RegOpenKeyA(HKEY_CLASSES_ROOT, IStorageCompObj.strProgIDName, &hKey);
7263 if(hErr == ERROR_SUCCESS)
7265 char strTemp[OLESTREAM_MAX_STR_LEN];
7266 IStorageCompObj.dwCLSIDNameLength = OLESTREAM_MAX_STR_LEN;
7267 hErr = RegQueryValueA(hKey, NULL, strTemp, (LONG*) &(IStorageCompObj.dwCLSIDNameLength));
7268 if(hErr == ERROR_SUCCESS)
7270 strcpy(IStorageCompObj.strCLSIDName, strTemp);
7272 RegCloseKey(hKey);
7276 /* Write CompObj Structure to stream */
7277 hRes = IStream_Write(pStream, IStorageCompObj.byUnknown1, sizeof(IStorageCompObj.byUnknown1), NULL);
7279 WriteClassStm(pStream,&(IStorageCompObj.clsid));
7281 hRes = IStream_Write(pStream, &(IStorageCompObj.dwCLSIDNameLength), sizeof(IStorageCompObj.dwCLSIDNameLength), NULL);
7282 if(IStorageCompObj.dwCLSIDNameLength > 0)
7284 hRes = IStream_Write(pStream, IStorageCompObj.strCLSIDName, IStorageCompObj.dwCLSIDNameLength, NULL);
7286 hRes = IStream_Write(pStream, &(IStorageCompObj.dwOleTypeNameLength) , sizeof(IStorageCompObj.dwOleTypeNameLength), NULL);
7287 if(IStorageCompObj.dwOleTypeNameLength > 0)
7289 hRes = IStream_Write(pStream, IStorageCompObj.strOleTypeName , IStorageCompObj.dwOleTypeNameLength, NULL);
7291 hRes = IStream_Write(pStream, &(IStorageCompObj.dwProgIDNameLength) , sizeof(IStorageCompObj.dwProgIDNameLength), NULL);
7292 if(IStorageCompObj.dwProgIDNameLength > 0)
7294 hRes = IStream_Write(pStream, IStorageCompObj.strProgIDName , IStorageCompObj.dwProgIDNameLength, NULL);
7296 hRes = IStream_Write(pStream, IStorageCompObj.byUnknown2 , sizeof(IStorageCompObj.byUnknown2), NULL);
7297 IStream_Release(pStream);
7299 return hRes;
7303 /*************************************************************************
7304 * OLECONVERT_CreateOlePresStream[Internal]
7306 * Creates the "\002OlePres000" Stream with the Metafile data
7308 * PARAMS
7309 * pStorage [I] The dest IStorage to create \002OLEPres000 stream in.
7310 * dwExtentX [I] Width of the Metafile
7311 * dwExtentY [I] Height of the Metafile
7312 * pData [I] Metafile data
7313 * dwDataLength [I] Size of the Metafile data
7315 * RETURNS
7316 * Success: S_OK
7317 * Failure: CONVERT10_E_OLESTREAM_PUT for invalid Put
7319 * NOTES
7320 * This function is used by OleConvertOLESTREAMToIStorage only.
7323 static void OLECONVERT_CreateOlePresStream(LPSTORAGE pStorage, DWORD dwExtentX, DWORD dwExtentY , BYTE *pData, DWORD dwDataLength)
7325 HRESULT hRes;
7326 IStream *pStream;
7327 static const WCHAR wstrStreamName[] = {2, 'O', 'l', 'e', 'P', 'r', 'e', 's', '0', '0', '0', 0};
7328 BYTE pOlePresStreamHeader [] =
7330 0xFF, 0xFF, 0xFF, 0xFF, 0x03, 0x00, 0x00, 0x00,
7331 0x04, 0x00, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00,
7332 0xFF, 0xFF, 0xFF, 0xFF, 0x00, 0x00, 0x00, 0x00,
7333 0x00, 0x00, 0x00, 0x00
7336 BYTE pOlePresStreamHeaderEmpty [] =
7338 0x00, 0x00, 0x00, 0x00,
7339 0x04, 0x00, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00,
7340 0xFF, 0xFF, 0xFF, 0xFF, 0x00, 0x00, 0x00, 0x00,
7341 0x00, 0x00, 0x00, 0x00
7344 /* Create the OlePres000 Stream */
7345 hRes = IStorage_CreateStream(pStorage, wstrStreamName,
7346 STGM_CREATE | STGM_WRITE | STGM_SHARE_EXCLUSIVE, 0, 0, &pStream );
7348 if(hRes == S_OK)
7350 DWORD nHeaderSize;
7351 OLECONVERT_ISTORAGE_OLEPRES OlePres;
7353 memset(&OlePres, 0, sizeof(OlePres));
7354 /* Do we have any metafile data to save */
7355 if(dwDataLength > 0)
7357 memcpy(OlePres.byUnknown1, pOlePresStreamHeader, sizeof(pOlePresStreamHeader));
7358 nHeaderSize = sizeof(pOlePresStreamHeader);
7360 else
7362 memcpy(OlePres.byUnknown1, pOlePresStreamHeaderEmpty, sizeof(pOlePresStreamHeaderEmpty));
7363 nHeaderSize = sizeof(pOlePresStreamHeaderEmpty);
7365 /* Set width and height of the metafile */
7366 OlePres.dwExtentX = dwExtentX;
7367 OlePres.dwExtentY = -dwExtentY;
7369 /* Set Data and Length */
7370 if(dwDataLength > sizeof(METAFILEPICT16))
7372 OlePres.dwSize = dwDataLength - sizeof(METAFILEPICT16);
7373 OlePres.pData = &(pData[8]);
7375 /* Save OlePres000 Data to Stream */
7376 hRes = IStream_Write(pStream, OlePres.byUnknown1, nHeaderSize, NULL);
7377 hRes = IStream_Write(pStream, &(OlePres.dwExtentX), sizeof(OlePres.dwExtentX), NULL);
7378 hRes = IStream_Write(pStream, &(OlePres.dwExtentY), sizeof(OlePres.dwExtentY), NULL);
7379 hRes = IStream_Write(pStream, &(OlePres.dwSize), sizeof(OlePres.dwSize), NULL);
7380 if(OlePres.dwSize > 0)
7382 hRes = IStream_Write(pStream, OlePres.pData, OlePres.dwSize, NULL);
7384 IStream_Release(pStream);
7388 /*************************************************************************
7389 * OLECONVERT_CreateOle10NativeStream [Internal]
7391 * Creates the "\001Ole10Native" Stream (should contain a BMP)
7393 * PARAMS
7394 * pStorage [I] Dest storage to create the stream in
7395 * pData [I] Ole10 Native Data (ex. bmp)
7396 * dwDataLength [I] Size of the Ole10 Native Data
7398 * RETURNS
7399 * Nothing
7401 * NOTES
7402 * This function is used by OleConvertOLESTREAMToIStorage only.
7404 * Might need to verify the data and return appropriate error message
7407 static void OLECONVERT_CreateOle10NativeStream(LPSTORAGE pStorage, const BYTE *pData, DWORD dwDataLength)
7409 HRESULT hRes;
7410 IStream *pStream;
7411 static const WCHAR wstrStreamName[] = {1, 'O', 'l', 'e', '1', '0', 'N', 'a', 't', 'i', 'v', 'e', 0};
7413 /* Create the Ole10Native Stream */
7414 hRes = IStorage_CreateStream(pStorage, wstrStreamName,
7415 STGM_CREATE | STGM_WRITE | STGM_SHARE_EXCLUSIVE, 0, 0, &pStream );
7417 if(hRes == S_OK)
7419 /* Write info to stream */
7420 hRes = IStream_Write(pStream, &dwDataLength, sizeof(dwDataLength), NULL);
7421 hRes = IStream_Write(pStream, pData, dwDataLength, NULL);
7422 IStream_Release(pStream);
7427 /*************************************************************************
7428 * OLECONVERT_GetOLE10ProgID [Internal]
7430 * Finds the ProgID (or OleTypeID) from the IStorage
7432 * PARAMS
7433 * pStorage [I] The Src IStorage to get the ProgID
7434 * strProgID [I] the ProgID string to get
7435 * dwSize [I] the size of the string
7437 * RETURNS
7438 * Success: S_OK
7439 * Failure: REGDB_E_CLASSNOTREG if cannot reconstruct the stream
7441 * NOTES
7442 * This function is used by OleConvertIStorageToOLESTREAM only.
7446 static HRESULT OLECONVERT_GetOLE10ProgID(LPSTORAGE pStorage, char *strProgID, DWORD *dwSize)
7448 HRESULT hRes;
7449 IStream *pStream;
7450 LARGE_INTEGER iSeekPos;
7451 OLECONVERT_ISTORAGE_COMPOBJ CompObj;
7452 static const WCHAR wstrStreamName[] = {1,'C', 'o', 'm', 'p', 'O', 'b', 'j', 0};
7454 /* Open the CompObj Stream */
7455 hRes = IStorage_OpenStream(pStorage, wstrStreamName, NULL,
7456 STGM_READ | STGM_SHARE_EXCLUSIVE, 0, &pStream );
7457 if(hRes == S_OK)
7460 /*Get the OleType from the CompObj Stream */
7461 iSeekPos.u.LowPart = sizeof(CompObj.byUnknown1) + sizeof(CompObj.clsid);
7462 iSeekPos.u.HighPart = 0;
7464 IStream_Seek(pStream, iSeekPos, STREAM_SEEK_SET, NULL);
7465 IStream_Read(pStream, &CompObj.dwCLSIDNameLength, sizeof(CompObj.dwCLSIDNameLength), NULL);
7466 iSeekPos.u.LowPart = CompObj.dwCLSIDNameLength;
7467 IStream_Seek(pStream, iSeekPos, STREAM_SEEK_CUR , NULL);
7468 IStream_Read(pStream, &CompObj.dwOleTypeNameLength, sizeof(CompObj.dwOleTypeNameLength), NULL);
7469 iSeekPos.u.LowPart = CompObj.dwOleTypeNameLength;
7470 IStream_Seek(pStream, iSeekPos, STREAM_SEEK_CUR , NULL);
7472 IStream_Read(pStream, dwSize, sizeof(*dwSize), NULL);
7473 if(*dwSize > 0)
7475 IStream_Read(pStream, strProgID, *dwSize, NULL);
7477 IStream_Release(pStream);
7479 else
7481 STATSTG stat;
7482 LPOLESTR wstrProgID;
7484 /* Get the OleType from the registry */
7485 REFCLSID clsid = &(stat.clsid);
7486 IStorage_Stat(pStorage, &stat, STATFLAG_NONAME);
7487 hRes = ProgIDFromCLSID(clsid, &wstrProgID);
7488 if(hRes == S_OK)
7490 *dwSize = WideCharToMultiByte(CP_ACP, 0, wstrProgID, -1, strProgID, *dwSize, NULL, FALSE);
7494 return hRes;
7497 /*************************************************************************
7498 * OLECONVERT_GetOle10PresData [Internal]
7500 * Converts IStorage "/001Ole10Native" stream to a OLE10 Stream
7502 * PARAMS
7503 * pStorage [I] Src IStroage
7504 * pOleStream [I] Dest OleStream Mem Struct
7506 * RETURNS
7507 * Nothing
7509 * NOTES
7510 * This function is used by OleConvertIStorageToOLESTREAM only.
7512 * Memory allocated for pData must be freed by the caller
7516 static void OLECONVERT_GetOle10PresData(LPSTORAGE pStorage, OLECONVERT_OLESTREAM_DATA *pOleStreamData)
7519 HRESULT hRes;
7520 IStream *pStream;
7521 static const WCHAR wstrStreamName[] = {1, 'O', 'l', 'e', '1', '0', 'N', 'a', 't', 'i', 'v', 'e', 0};
7523 /* Initialize Default data for OLESTREAM */
7524 pOleStreamData[0].dwOleID = OLESTREAM_ID;
7525 pOleStreamData[0].dwTypeID = 2;
7526 pOleStreamData[1].dwOleID = OLESTREAM_ID;
7527 pOleStreamData[1].dwTypeID = 0;
7528 pOleStreamData[0].dwMetaFileWidth = 0;
7529 pOleStreamData[0].dwMetaFileHeight = 0;
7530 pOleStreamData[0].pData = NULL;
7531 pOleStreamData[1].pData = NULL;
7533 /* Open Ole10Native Stream */
7534 hRes = IStorage_OpenStream(pStorage, wstrStreamName, NULL,
7535 STGM_READ | STGM_SHARE_EXCLUSIVE, 0, &pStream );
7536 if(hRes == S_OK)
7539 /* Read Size and Data */
7540 IStream_Read(pStream, &(pOleStreamData->dwDataLength), sizeof(pOleStreamData->dwDataLength), NULL);
7541 if(pOleStreamData->dwDataLength > 0)
7543 pOleStreamData->pData = HeapAlloc(GetProcessHeap(),0,pOleStreamData->dwDataLength);
7544 IStream_Read(pStream, pOleStreamData->pData, pOleStreamData->dwDataLength, NULL);
7546 IStream_Release(pStream);
7552 /*************************************************************************
7553 * OLECONVERT_GetOle20PresData[Internal]
7555 * Converts IStorage "/002OlePres000" stream to a OLE10 Stream
7557 * PARAMS
7558 * pStorage [I] Src IStroage
7559 * pOleStreamData [I] Dest OleStream Mem Struct
7561 * RETURNS
7562 * Nothing
7564 * NOTES
7565 * This function is used by OleConvertIStorageToOLESTREAM only.
7567 * Memory allocated for pData must be freed by the caller
7569 static void OLECONVERT_GetOle20PresData(LPSTORAGE pStorage, OLECONVERT_OLESTREAM_DATA *pOleStreamData)
7571 HRESULT hRes;
7572 IStream *pStream;
7573 OLECONVERT_ISTORAGE_OLEPRES olePress;
7574 static const WCHAR wstrStreamName[] = {2, 'O', 'l', 'e', 'P', 'r', 'e', 's', '0', '0', '0', 0};
7576 /* Initialize Default data for OLESTREAM */
7577 pOleStreamData[0].dwOleID = OLESTREAM_ID;
7578 pOleStreamData[0].dwTypeID = 2;
7579 pOleStreamData[0].dwMetaFileWidth = 0;
7580 pOleStreamData[0].dwMetaFileHeight = 0;
7581 pOleStreamData[0].dwDataLength = OLECONVERT_WriteOLE20ToBuffer(pStorage, &(pOleStreamData[0].pData));
7582 pOleStreamData[1].dwOleID = OLESTREAM_ID;
7583 pOleStreamData[1].dwTypeID = 0;
7584 pOleStreamData[1].dwOleTypeNameLength = 0;
7585 pOleStreamData[1].strOleTypeName[0] = 0;
7586 pOleStreamData[1].dwMetaFileWidth = 0;
7587 pOleStreamData[1].dwMetaFileHeight = 0;
7588 pOleStreamData[1].pData = NULL;
7589 pOleStreamData[1].dwDataLength = 0;
7592 /* Open OlePress000 stream */
7593 hRes = IStorage_OpenStream(pStorage, wstrStreamName, NULL,
7594 STGM_READ | STGM_SHARE_EXCLUSIVE, 0, &pStream );
7595 if(hRes == S_OK)
7597 LARGE_INTEGER iSeekPos;
7598 METAFILEPICT16 MetaFilePict;
7599 static const char strMetafilePictName[] = "METAFILEPICT";
7601 /* Set the TypeID for a Metafile */
7602 pOleStreamData[1].dwTypeID = 5;
7604 /* Set the OleTypeName to Metafile */
7605 pOleStreamData[1].dwOleTypeNameLength = strlen(strMetafilePictName) +1;
7606 strcpy(pOleStreamData[1].strOleTypeName, strMetafilePictName);
7608 iSeekPos.u.HighPart = 0;
7609 iSeekPos.u.LowPart = sizeof(olePress.byUnknown1);
7611 /* Get Presentation Data */
7612 IStream_Seek(pStream, iSeekPos, STREAM_SEEK_SET, NULL);
7613 IStream_Read(pStream, &(olePress.dwExtentX), sizeof(olePress.dwExtentX), NULL);
7614 IStream_Read(pStream, &(olePress.dwExtentY), sizeof(olePress.dwExtentY), NULL);
7615 IStream_Read(pStream, &(olePress.dwSize), sizeof(olePress.dwSize), NULL);
7617 /*Set width and Height */
7618 pOleStreamData[1].dwMetaFileWidth = olePress.dwExtentX;
7619 pOleStreamData[1].dwMetaFileHeight = -olePress.dwExtentY;
7620 if(olePress.dwSize > 0)
7622 /* Set Length */
7623 pOleStreamData[1].dwDataLength = olePress.dwSize + sizeof(METAFILEPICT16);
7625 /* Set MetaFilePict struct */
7626 MetaFilePict.mm = 8;
7627 MetaFilePict.xExt = olePress.dwExtentX;
7628 MetaFilePict.yExt = olePress.dwExtentY;
7629 MetaFilePict.hMF = 0;
7631 /* Get Metafile Data */
7632 pOleStreamData[1].pData = HeapAlloc(GetProcessHeap(),0,pOleStreamData[1].dwDataLength);
7633 memcpy(pOleStreamData[1].pData, &MetaFilePict, sizeof(MetaFilePict));
7634 IStream_Read(pStream, &(pOleStreamData[1].pData[sizeof(MetaFilePict)]), pOleStreamData[1].dwDataLength-sizeof(METAFILEPICT16), NULL);
7636 IStream_Release(pStream);
7640 /*************************************************************************
7641 * OleConvertOLESTREAMToIStorage [OLE32.@]
7643 * Read info on MSDN
7645 * TODO
7646 * DVTARGETDEVICE parameter is not handled
7647 * Still unsure of some mem fields for OLE 10 Stream
7648 * Still some unknowns for the IStorage: "\002OlePres000", "\001CompObj",
7649 * and "\001OLE" streams
7652 HRESULT WINAPI OleConvertOLESTREAMToIStorage (
7653 LPOLESTREAM pOleStream,
7654 LPSTORAGE pstg,
7655 const DVTARGETDEVICE* ptd)
7657 int i;
7658 HRESULT hRes=S_OK;
7659 OLECONVERT_OLESTREAM_DATA pOleStreamData[2];
7661 TRACE("%p %p %p\n", pOleStream, pstg, ptd);
7663 memset(pOleStreamData, 0, sizeof(pOleStreamData));
7665 if(ptd != NULL)
7667 FIXME("DVTARGETDEVICE is not NULL, unhandled parameter\n");
7670 if(pstg == NULL || pOleStream == NULL)
7672 hRes = E_INVALIDARG;
7675 if(hRes == S_OK)
7677 /* Load the OLESTREAM to Memory */
7678 hRes = OLECONVERT_LoadOLE10(pOleStream, &pOleStreamData[0], TRUE);
7681 if(hRes == S_OK)
7683 /* Load the OLESTREAM to Memory (part 2)*/
7684 hRes = OLECONVERT_LoadOLE10(pOleStream, &pOleStreamData[1], FALSE);
7687 if(hRes == S_OK)
7690 if(pOleStreamData[0].dwDataLength > sizeof(STORAGE_magic))
7692 /* Do we have the IStorage Data in the OLESTREAM */
7693 if(memcmp(pOleStreamData[0].pData, STORAGE_magic, sizeof(STORAGE_magic)) ==0)
7695 OLECONVERT_GetOLE20FromOLE10(pstg, pOleStreamData[0].pData, pOleStreamData[0].dwDataLength);
7696 OLECONVERT_CreateOlePresStream(pstg, pOleStreamData[1].dwMetaFileWidth, pOleStreamData[1].dwMetaFileHeight, pOleStreamData[1].pData, pOleStreamData[1].dwDataLength);
7698 else
7700 /* It must be an original OLE 1.0 source */
7701 OLECONVERT_CreateOle10NativeStream(pstg, pOleStreamData[0].pData, pOleStreamData[0].dwDataLength);
7704 else
7706 /* It must be an original OLE 1.0 source */
7707 OLECONVERT_CreateOle10NativeStream(pstg, pOleStreamData[0].pData, pOleStreamData[0].dwDataLength);
7710 /* Create CompObj Stream if necessary */
7711 hRes = OLECONVERT_CreateCompObjStream(pstg, pOleStreamData[0].strOleTypeName);
7712 if(hRes == S_OK)
7714 /*Create the Ole Stream if necessary */
7715 OLECONVERT_CreateOleStream(pstg);
7720 /* Free allocated memory */
7721 for(i=0; i < 2; i++)
7723 HeapFree(GetProcessHeap(),0,pOleStreamData[i].pData);
7724 HeapFree(GetProcessHeap(),0,pOleStreamData[i].pstrOleObjFileName);
7725 pOleStreamData[i].pstrOleObjFileName = NULL;
7727 return hRes;
7730 /*************************************************************************
7731 * OleConvertIStorageToOLESTREAM [OLE32.@]
7733 * Read info on MSDN
7735 * Read info on MSDN
7737 * TODO
7738 * Still unsure of some mem fields for OLE 10 Stream
7739 * Still some unknowns for the IStorage: "\002OlePres000", "\001CompObj",
7740 * and "\001OLE" streams.
7743 HRESULT WINAPI OleConvertIStorageToOLESTREAM (
7744 LPSTORAGE pstg,
7745 LPOLESTREAM pOleStream)
7747 int i;
7748 HRESULT hRes = S_OK;
7749 IStream *pStream;
7750 OLECONVERT_OLESTREAM_DATA pOleStreamData[2];
7751 static const WCHAR wstrStreamName[] = {1, 'O', 'l', 'e', '1', '0', 'N', 'a', 't', 'i', 'v', 'e', 0};
7753 TRACE("%p %p\n", pstg, pOleStream);
7755 memset(pOleStreamData, 0, sizeof(pOleStreamData));
7757 if(pstg == NULL || pOleStream == NULL)
7759 hRes = E_INVALIDARG;
7761 if(hRes == S_OK)
7763 /* Get the ProgID */
7764 pOleStreamData[0].dwOleTypeNameLength = OLESTREAM_MAX_STR_LEN;
7765 hRes = OLECONVERT_GetOLE10ProgID(pstg, pOleStreamData[0].strOleTypeName, &(pOleStreamData[0].dwOleTypeNameLength));
7767 if(hRes == S_OK)
7769 /* Was it originally Ole10 */
7770 hRes = IStorage_OpenStream(pstg, wstrStreamName, 0, STGM_READ | STGM_SHARE_EXCLUSIVE, 0, &pStream);
7771 if(hRes == S_OK)
7773 IStream_Release(pStream);
7774 /* Get Presentation Data for Ole10Native */
7775 OLECONVERT_GetOle10PresData(pstg, pOleStreamData);
7777 else
7779 /* Get Presentation Data (OLE20) */
7780 OLECONVERT_GetOle20PresData(pstg, pOleStreamData);
7783 /* Save OLESTREAM */
7784 hRes = OLECONVERT_SaveOLE10(&(pOleStreamData[0]), pOleStream);
7785 if(hRes == S_OK)
7787 hRes = OLECONVERT_SaveOLE10(&(pOleStreamData[1]), pOleStream);
7792 /* Free allocated memory */
7793 for(i=0; i < 2; i++)
7795 HeapFree(GetProcessHeap(),0,pOleStreamData[i].pData);
7798 return hRes;
7801 /***********************************************************************
7802 * GetConvertStg (OLE32.@)
7804 HRESULT WINAPI GetConvertStg(IStorage *stg) {
7805 FIXME("unimplemented stub!\n");
7806 return E_FAIL;
7809 /******************************************************************************
7810 * StgIsStorageFile [OLE32.@]
7811 * Verify if the file contains a storage object
7813 * PARAMS
7814 * fn [ I] Filename
7816 * RETURNS
7817 * S_OK if file has magic bytes as a storage object
7818 * S_FALSE if file is not storage
7820 HRESULT WINAPI
7821 StgIsStorageFile(LPCOLESTR fn)
7823 HANDLE hf;
7824 BYTE magic[8];
7825 DWORD bytes_read;
7827 TRACE("%s\n", debugstr_w(fn));
7828 hf = CreateFileW(fn, GENERIC_READ,
7829 FILE_SHARE_DELETE | FILE_SHARE_READ | FILE_SHARE_WRITE,
7830 NULL, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, 0);
7832 if (hf == INVALID_HANDLE_VALUE)
7833 return STG_E_FILENOTFOUND;
7835 if (!ReadFile(hf, magic, 8, &bytes_read, NULL))
7837 WARN(" unable to read file\n");
7838 CloseHandle(hf);
7839 return S_FALSE;
7842 CloseHandle(hf);
7844 if (bytes_read != 8) {
7845 WARN(" too short\n");
7846 return S_FALSE;
7849 if (!memcmp(magic,STORAGE_magic,8)) {
7850 WARN(" -> YES\n");
7851 return S_OK;
7854 WARN(" -> Invalid header.\n");
7855 return S_FALSE;
7858 /***********************************************************************
7859 * WriteClassStm (OLE32.@)
7861 * Writes a CLSID to a stream.
7863 * PARAMS
7864 * pStm [I] Stream to write to.
7865 * rclsid [I] CLSID to write.
7867 * RETURNS
7868 * Success: S_OK.
7869 * Failure: HRESULT code.
7871 HRESULT WINAPI WriteClassStm(IStream *pStm,REFCLSID rclsid)
7873 TRACE("(%p,%p)\n",pStm,rclsid);
7875 if (!pStm || !rclsid)
7876 return E_INVALIDARG;
7878 return IStream_Write(pStm,rclsid,sizeof(CLSID),NULL);
7881 /***********************************************************************
7882 * ReadClassStm (OLE32.@)
7884 * Reads a CLSID from a stream.
7886 * PARAMS
7887 * pStm [I] Stream to read from.
7888 * rclsid [O] CLSID to read.
7890 * RETURNS
7891 * Success: S_OK.
7892 * Failure: HRESULT code.
7894 HRESULT WINAPI ReadClassStm(IStream *pStm,CLSID *pclsid)
7896 ULONG nbByte;
7897 HRESULT res;
7899 TRACE("(%p,%p)\n",pStm,pclsid);
7901 if (!pStm || !pclsid)
7902 return E_INVALIDARG;
7904 /* clear the output args */
7905 *pclsid = CLSID_NULL;
7907 res = IStream_Read(pStm,(void*)pclsid,sizeof(CLSID),&nbByte);
7909 if (FAILED(res))
7910 return res;
7912 if (nbByte != sizeof(CLSID))
7913 return STG_E_READFAULT;
7914 else
7915 return S_OK;