dmloader: complete rewrite and full implementation.
[wine] / dlls / dmloader / container.c
1 /* IDirectMusicContainerImpl
2  *
3  * Copyright (C) 2003-2004 Rok Mandeljc
4  *
5  * This program is free software; you can redistribute it and/or modify
6  * it under the terms of the GNU General Public License as published by
7  * the Free Software Foundation; either version 2 of the License, or
8  * (at your option) any later version.
9  *
10  * This program is distributed in the hope that it will be useful,
11  * but WITHOUT ANY WARRANTY; without even the implied warranty of
12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
13  * GNU Library General Public License for more details.
14  *
15  * You should have received a copy of the GNU General Public License
16  * along with this program; if not, write to the Free Software
17  * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
18  */
19
20 #include "dmloader_private.h"
21
22 WINE_DEFAULT_DEBUG_CHANNEL(dmloader);
23 WINE_DECLARE_DEBUG_CHANNEL(dmfile);
24 WINE_DECLARE_DEBUG_CHANNEL(dmdump);
25
26 #define DMUS_MAX_CATEGORY_SIZE DMUS_MAX_CATEGORY*sizeof(WCHAR)
27 #define DMUS_MAX_NAME_SIZE     DMUS_MAX_NAME*sizeof(WCHAR)
28 #define DMUS_MAX_FILENAME_SIZE DMUS_MAX_FILENAME*sizeof(WCHAR)
29
30 /*****************************************************************************
31  * IDirectMusicContainerImpl implementation
32  */
33 /* IUnknown/IDirectMusicContainer part: */
34 HRESULT WINAPI IDirectMusicContainerImpl_IDirectMusicContainer_QueryInterface (LPDIRECTMUSICCONTAINER iface, REFIID riid, LPVOID *ppobj) {
35         ICOM_THIS_MULTI(IDirectMusicContainerImpl, ContainerVtbl, iface);
36         
37         TRACE("(%p, %s, %p)\n", This, debugstr_dmguid(riid), ppobj);
38         if (IsEqualIID (riid, &IID_IUnknown) ||
39                 IsEqualIID (riid, &IID_IDirectMusicContainer)) {
40                 *ppobj = (LPVOID)&This->ContainerVtbl;
41                 IDirectMusicContainerImpl_IDirectMusicContainer_AddRef ((LPDIRECTMUSICCONTAINER)&This->ContainerVtbl);
42                 return S_OK;
43         } else if (IsEqualIID (riid, &IID_IDirectMusicObject)) {
44                 *ppobj = (LPVOID)&This->ObjectVtbl;
45                 IDirectMusicContainerImpl_IDirectMusicObject_AddRef ((LPDIRECTMUSICOBJECT)&This->ObjectVtbl);           
46                 return S_OK;
47         } else if (IsEqualIID (riid, &IID_IPersistStream)) {
48                 *ppobj = (LPVOID)&This->PersistStreamVtbl;
49                 IDirectMusicContainerImpl_IPersistStream_AddRef ((LPPERSISTSTREAM)&This->PersistStreamVtbl);            
50                 return S_OK;
51         }
52         
53         WARN(": not found\n");
54         return E_NOINTERFACE;
55 }
56
57 ULONG WINAPI IDirectMusicContainerImpl_IDirectMusicContainer_AddRef (LPDIRECTMUSICCONTAINER iface) {
58         ICOM_THIS_MULTI(IDirectMusicContainerImpl, ContainerVtbl, iface);
59         TRACE("(%p): AddRef from %ld\n", This, This->dwRef);
60         return InterlockedIncrement (&This->dwRef);
61 }
62
63 ULONG WINAPI IDirectMusicContainerImpl_IDirectMusicContainer_Release (LPDIRECTMUSICCONTAINER iface) {
64         ICOM_THIS_MULTI(IDirectMusicContainerImpl, ContainerVtbl, iface);
65         
66         DWORD dwRef = InterlockedDecrement (&This->dwRef);
67         TRACE("(%p): ReleaseRef to %ld\n", This, dwRef);
68         if (dwRef == 0) {
69                 DMUSIC_DestroyDirectMusicContainerImpl (iface);
70                 HeapFree(GetProcessHeap(), 0, This);
71         }
72         
73         return dwRef;
74 }
75
76 HRESULT WINAPI IDirectMusicContainerImpl_IDirectMusicContainer_EnumObject (LPDIRECTMUSICCONTAINER iface, REFGUID rguidClass, DWORD dwIndex, LPDMUS_OBJECTDESC pDesc, WCHAR* pwszAlias) {
77         ICOM_THIS_MULTI(IDirectMusicContainerImpl, ContainerVtbl, iface);       
78         struct list *pEntry;
79         LPWINE_CONTAINER_ENTRY pContainedObject;
80         DWORD dwCount = 0;
81
82         TRACE("(%p, %s, %ld, %p, %p)\n", This, debugstr_dmguid(rguidClass), dwIndex, pDesc, pwszAlias);
83
84         /* check if we can write to whole pDesc */
85         if (pDesc) {
86                 if (IsBadReadPtr (pDesc, sizeof(DWORD))) {
87                         ERR(": pDesc->dwSize bad read pointer\n");
88                         return E_POINTER;
89                 }
90                 if (pDesc->dwSize != sizeof(DMUS_OBJECTDESC)) {
91                         ERR(": invalid pDesc->dwSize\n");
92                         return E_INVALIDARG;
93                 }
94                 if (IsBadWritePtr (pDesc, sizeof(DMUS_OBJECTDESC))) {
95                         ERR(": pDesc bad write pointer\n");
96                         return E_POINTER;
97                 }
98         }
99         /* check if we wszAlias is big enough */
100         if (pwszAlias && IsBadWritePtr (pwszAlias, DMUS_MAX_FILENAME_SIZE)) {
101                 ERR(": wszAlias bad write pointer\n");
102                 return E_POINTER;               
103         }
104         
105         DM_STRUCT_INIT(pDesc);
106         
107         LIST_FOR_EACH (pEntry, This->pContainedObjects) {
108                 pContainedObject = LIST_ENTRY (pEntry, WINE_CONTAINER_ENTRY, entry);
109                 
110                 if (IsEqualGUID (rguidClass, &GUID_DirectMusicAllTypes) || IsEqualGUID (rguidClass, &pContainedObject->Desc.guidClass)) {
111                         if (dwCount == dwIndex) {
112                                 HRESULT result = S_OK;
113                                 if (pwszAlias) {
114                                         strncpyW (pwszAlias, pContainedObject->wszAlias, DMUS_MAX_FILENAME);
115                                         if (strlenW (pContainedObject->wszAlias) > DMUS_MAX_FILENAME)
116                                                 result = DMUS_S_STRING_TRUNCATED;
117                                 }
118                                 if (pDesc)
119                                         memcpy (pDesc, &pContainedObject->Desc, sizeof(DMUS_OBJECTDESC));
120                                 return result;
121                         }
122                         dwCount++;
123                 }
124         }               
125         
126         TRACE(": not found\n");
127         return S_FALSE;
128 }
129
130 ICOM_VTABLE(IDirectMusicContainer) DirectMusicContainer_Container_Vtbl = {
131     ICOM_MSVTABLE_COMPAT_DummyRTTIVALUE
132         IDirectMusicContainerImpl_IDirectMusicContainer_QueryInterface,
133         IDirectMusicContainerImpl_IDirectMusicContainer_AddRef,
134         IDirectMusicContainerImpl_IDirectMusicContainer_Release,
135         IDirectMusicContainerImpl_IDirectMusicContainer_EnumObject
136 };
137
138 /* IDirectMusicObject part: */
139 HRESULT WINAPI IDirectMusicContainerImpl_IDirectMusicObject_QueryInterface (LPDIRECTMUSICOBJECT iface, REFIID riid, LPVOID *ppobj) {
140         ICOM_THIS_MULTI(IDirectMusicContainerImpl, ObjectVtbl, iface);
141         return IDirectMusicContainerImpl_IDirectMusicContainer_QueryInterface ((LPDIRECTMUSICCONTAINER)&This->ContainerVtbl, riid, ppobj);
142 }
143
144 ULONG WINAPI IDirectMusicContainerImpl_IDirectMusicObject_AddRef (LPDIRECTMUSICOBJECT iface) {
145         ICOM_THIS_MULTI(IDirectMusicContainerImpl, ObjectVtbl, iface);
146         return IDirectMusicContainerImpl_IDirectMusicContainer_AddRef ((LPDIRECTMUSICCONTAINER)&This->ContainerVtbl);
147 }
148
149 ULONG WINAPI IDirectMusicContainerImpl_IDirectMusicObject_Release (LPDIRECTMUSICOBJECT iface) {
150         ICOM_THIS_MULTI(IDirectMusicContainerImpl, ObjectVtbl, iface);
151         return IDirectMusicContainerImpl_IDirectMusicContainer_Release ((LPDIRECTMUSICCONTAINER)&This->ContainerVtbl);
152 }
153
154 HRESULT WINAPI IDirectMusicContainerImpl_IDirectMusicObject_GetDescriptor (LPDIRECTMUSICOBJECT iface, LPDMUS_OBJECTDESC pDesc) {
155         ICOM_THIS_MULTI(IDirectMusicContainerImpl, ObjectVtbl, iface);
156         TRACE("(%p, %p):\n", This, pDesc);
157         
158         /* check if whe can write to whole pDesc */
159         if (IsBadReadPtr (pDesc, sizeof(DWORD))) {
160                 ERR(": pDesc->dwSize bad read pointer\n");
161                 return E_POINTER;
162         }
163         if (pDesc->dwSize != sizeof(DMUS_OBJECTDESC)) {
164                 ERR(": invalid pDesc->dwSize\n");
165                 return E_INVALIDARG;
166         }
167         if (IsBadWritePtr (pDesc, sizeof(DMUS_OBJECTDESC))) {
168                 ERR(": pDesc bad write pointer\n");
169                 return E_POINTER;
170         }
171         
172         DM_STRUCT_INIT(pDesc);
173         memcpy (pDesc, &This->Desc, sizeof(DMUS_OBJECTDESC));
174         
175         return S_OK;
176 }
177
178 HRESULT WINAPI IDirectMusicContainerImpl_IDirectMusicObject_SetDescriptor (LPDIRECTMUSICOBJECT iface, LPDMUS_OBJECTDESC pDesc) {
179         DWORD dwNewFlags = 0;
180         DWORD dwFlagDifference;
181         ICOM_THIS_MULTI(IDirectMusicContainerImpl, ObjectVtbl, iface);
182         TRACE("(%p, %p):\n", This, pDesc);
183
184         /* check if whe can read whole pDesc */
185         if (IsBadReadPtr (pDesc, sizeof(DWORD))) {
186                 ERR(": pDesc->dwSize bad read pointer\n");
187                 return E_POINTER;
188         }
189         if (pDesc->dwSize != sizeof(DMUS_OBJECTDESC)) {
190                 ERR(": invalid pDesc->dwSize\n");
191                 return E_INVALIDARG;
192         }
193         if (IsBadReadPtr (pDesc, sizeof(DMUS_OBJECTDESC))) {
194                 ERR(": pDesc bad read pointer\n");
195                 return E_POINTER;
196         }
197
198         if (pDesc->dwValidData & DMUS_OBJ_OBJECT) {
199                 memcpy (&This->Desc.guidObject, &pDesc->guidObject, sizeof(GUID));
200                 dwNewFlags |= DMUS_OBJ_OBJECT;
201         }
202         if (pDesc->dwValidData & DMUS_OBJ_NAME) {
203                 strncpyW (This->Desc.wszName, pDesc->wszName, DMUS_MAX_NAME);
204                 dwNewFlags |= DMUS_OBJ_NAME;
205         }
206         if (pDesc->dwValidData & DMUS_OBJ_CATEGORY) {
207                 strncpyW (This->Desc.wszCategory, pDesc->wszCategory, DMUS_MAX_CATEGORY);
208                 dwNewFlags |= DMUS_OBJ_CATEGORY;
209         }
210         if (pDesc->dwValidData & (DMUS_OBJ_FILENAME | DMUS_OBJ_FULLPATH)) {
211                 strncpyW (This->Desc.wszFileName, pDesc->wszFileName, DMUS_MAX_FILENAME);
212                 dwNewFlags |= (pDesc->dwValidData & (DMUS_OBJ_FILENAME | DMUS_OBJ_FULLPATH));
213         }
214         if (pDesc->dwValidData & DMUS_OBJ_VERSION) {
215                 This->Desc.vVersion.dwVersionLS = pDesc->vVersion.dwVersionLS;
216                 This->Desc.vVersion.dwVersionMS = pDesc->vVersion.dwVersionMS;
217                 dwNewFlags |= DMUS_OBJ_VERSION;
218         }
219         if (pDesc->dwValidData & DMUS_OBJ_DATE) {
220                 This->Desc.ftDate.dwHighDateTime = pDesc->ftDate.dwHighDateTime;
221                 This->Desc.ftDate.dwLowDateTime = pDesc->ftDate.dwLowDateTime;
222                 dwNewFlags |= DMUS_OBJ_DATE;
223         }
224         /* set new flags */
225         This->Desc.dwValidData |= dwNewFlags;
226         
227         dwFlagDifference = pDesc->dwValidData - dwNewFlags;
228         if (dwFlagDifference) {
229                 pDesc->dwValidData &= ~dwFlagDifference; /* and with bitwise complement */
230                 return S_FALSE;
231         } else return S_OK;
232 }
233
234 HRESULT WINAPI IDirectMusicContainerImpl_IDirectMusicObject_ParseDescriptor (LPDIRECTMUSICOBJECT iface, LPSTREAM pStream, LPDMUS_OBJECTDESC pDesc) {
235         ICOM_THIS_MULTI(IDirectMusicContainerImpl, ObjectVtbl, iface);
236         WINE_CHUNK Chunk;
237         DWORD StreamSize, StreamCount, ListSize[1], ListCount[1];
238         LARGE_INTEGER liMove; /* used when skipping chunks */
239
240         TRACE("(%p, %p, %p)\n", This, pStream, pDesc);
241         
242         /* check whether arguments are OK */
243         if (IsBadReadPtr (pStream, sizeof(LPVOID))) {
244                 ERR(": pStream bad read pointer\n");
245                 return E_POINTER;
246         }
247         /* check whether pDesc is OK */
248         if (IsBadReadPtr (pDesc, sizeof(DWORD))) {
249                 ERR(": pDesc->dwSize bad read pointer\n");
250                 return E_POINTER;
251         }
252         if (pDesc->dwSize != sizeof(DMUS_OBJECTDESC)) {
253                 ERR(": invalid pDesc->dwSize\n");
254                 return E_INVALIDARG;
255         }
256         if (IsBadWritePtr (pDesc, sizeof(DMUS_OBJECTDESC))) {
257                 ERR(": pDesc bad write pointer\n");
258                 return E_POINTER;
259         }
260
261         DM_STRUCT_INIT(pDesc);
262         
263         /* here we go... */
264         IStream_Read (pStream, &Chunk, sizeof(FOURCC)+sizeof(DWORD), NULL);
265         TRACE_(dmfile)(": %s chunk (size = 0x%08lX)", debugstr_fourcc (Chunk.fccID), Chunk.dwSize);
266         switch (Chunk.fccID) {  
267                 case FOURCC_RIFF: {
268                         IStream_Read (pStream, &Chunk.fccID, sizeof(FOURCC), NULL);                             
269                         TRACE_(dmfile)(": RIFF chunk of type %s", debugstr_fourcc(Chunk.fccID));
270                         StreamSize = Chunk.dwSize - sizeof(FOURCC);
271                         StreamCount = 0;
272                         if (Chunk.fccID == DMUS_FOURCC_CONTAINER_FORM) {
273                                 TRACE_(dmfile)(": container form\n");
274                                 /* set guidClass */
275                                 pDesc->dwValidData |= DMUS_OBJ_CLASS;
276                                 memcpy (&pDesc->guidClass, &CLSID_DirectMusicContainer, sizeof(CLSID));                         
277                                 do {
278                                         IStream_Read (pStream, &Chunk, sizeof(FOURCC)+sizeof(DWORD), NULL);
279                                         StreamCount += sizeof(FOURCC) + sizeof(DWORD) + Chunk.dwSize;
280                                         TRACE_(dmfile)(": %s chunk (size = 0x%08lX)", debugstr_fourcc (Chunk.fccID), Chunk.dwSize);
281                                         switch (Chunk.fccID) {
282                                                 case DMUS_FOURCC_GUID_CHUNK: {
283                                                         TRACE_(dmfile)(": GUID chunk\n");
284                                                         pDesc->dwValidData |= DMUS_OBJ_OBJECT;
285                                                         IStream_Read (pStream, &pDesc->guidObject, Chunk.dwSize, NULL);
286                                                         TRACE_(dmdump)(": GUID: %s\n", debugstr_guid(&pDesc->guidObject));
287                                                         break;
288                                                 }
289                                                 case DMUS_FOURCC_VERSION_CHUNK: {
290                                                         TRACE_(dmfile)(": version chunk\n");
291                                                         pDesc->dwValidData |= DMUS_OBJ_VERSION;
292                                                         IStream_Read (pStream, &pDesc->vVersion, Chunk.dwSize, NULL);
293                                                         TRACE_(dmdump)(": version: %s\n", debugstr_dmversion(&pDesc->vVersion));
294                                                         break;
295                                                 }
296                                                 case DMUS_FOURCC_DATE_CHUNK: {
297                                                         TRACE_(dmfile)(": date chunk\n");
298                                                         IStream_Read (pStream, &pDesc->ftDate, Chunk.dwSize, NULL);
299                                                         pDesc->dwValidData |= DMUS_OBJ_DATE;
300                                                         TRACE_(dmdump)(": date: %s\n", debugstr_filetime(&pDesc->ftDate));
301                                                         break;
302                                                 }                                                               
303                                                 case DMUS_FOURCC_CATEGORY_CHUNK: {
304                                                         TRACE_(dmfile)(": category chunk\n");
305                                                         /* if it happens that string is too long,
306                                                            read what we can and skip the rest*/
307                                                         if (Chunk.dwSize > DMUS_MAX_CATEGORY_SIZE) {
308                                                                 IStream_Read (pStream, pDesc->wszCategory, DMUS_MAX_CATEGORY_SIZE, NULL);
309                                                                 liMove.QuadPart = Chunk.dwSize - DMUS_MAX_CATEGORY_SIZE;
310                                                                 IStream_Seek (pStream, liMove, STREAM_SEEK_CUR, NULL);
311                                                         } else {
312                                                                 IStream_Read (pStream, pDesc->wszCategory, Chunk.dwSize, NULL);
313                                                         }
314                                                         pDesc->dwValidData |= DMUS_OBJ_CATEGORY;
315                                                         TRACE_(dmdump)(": category: %s\n", debugstr_w(pDesc->wszCategory));                                                     
316                                                         break;
317                                                 }
318                                                 case FOURCC_LIST: {
319                                                         IStream_Read (pStream, &Chunk.fccID, sizeof(FOURCC), NULL);                             
320                                                         TRACE_(dmfile)(": LIST chunk of type %s", debugstr_fourcc(Chunk.fccID));
321                                                         ListSize[0] = Chunk.dwSize - sizeof(FOURCC);
322                                                         ListCount[0] = 0;
323                                                         switch (Chunk.fccID) {
324                                                                 /* evil M$ UNFO list, which can (!?) contain INFO elements */
325                                                                 case DMUS_FOURCC_UNFO_LIST: {
326                                                                         TRACE_(dmfile)(": UNFO list\n");
327                                                                         do {
328                                                                                 IStream_Read (pStream, &Chunk, sizeof(FOURCC)+sizeof(DWORD), NULL);
329                                                                                 ListCount[0] += sizeof(FOURCC) + sizeof(DWORD) + Chunk.dwSize;
330                                                                                 TRACE_(dmfile)(": %s chunk (size = 0x%08lX)", debugstr_fourcc (Chunk.fccID), Chunk.dwSize);
331                                                                                 switch (Chunk.fccID) {
332                                                                                         /* don't ask me why, but M$ puts INFO elements in UNFO list sometimes
333                                              (though strings seem to be valid unicode) */
334                                                                                         case mmioFOURCC('I','N','A','M'):
335                                                                                         case DMUS_FOURCC_UNAM_CHUNK: {
336                                                                                                 TRACE_(dmfile)(": name chunk\n");
337                                                                                                 /* if it happens that string is too long,
338                                                                                                            read what we can and skip the rest*/
339                                                                                                 if (Chunk.dwSize > DMUS_MAX_NAME_SIZE) {
340                                                                                                         IStream_Read (pStream, pDesc->wszName, DMUS_MAX_NAME_SIZE, NULL);
341                                                                                                         liMove.QuadPart = Chunk.dwSize - DMUS_MAX_NAME_SIZE;
342                                                                                                         IStream_Seek (pStream, liMove, STREAM_SEEK_CUR, NULL);
343                                                                                                 } else {
344                                                                                                         IStream_Read (pStream, pDesc->wszName, Chunk.dwSize, NULL);
345                                                                                                 }
346                                                                                                 pDesc->dwValidData |= DMUS_OBJ_NAME;
347                                                                                                 TRACE_(dmdump)(": name: %s\n", debugstr_w(pDesc->wszName));
348                                                                                                 break;
349                                                                                         }
350                                                                                         default: {
351                                                                                                 TRACE_(dmfile)(": unknown chunk (irrevelant & skipping)\n");
352                                                                                                 liMove.QuadPart = Chunk.dwSize;
353                                                                                                 IStream_Seek (pStream, liMove, STREAM_SEEK_CUR, NULL);
354                                                                                                 break;                                          
355                                                                                         }
356                                                                                 }
357                                                                                 TRACE_(dmfile)(": ListCount[0] = 0x%08lX < ListSize[0] = 0x%08lX\n", ListCount[0], ListSize[0]);
358                                                                         } while (ListCount[0] < ListSize[0]);
359                                                                         break;
360                                                                 }
361                                                                 default: {
362                                                                         TRACE_(dmfile)(": unknown (skipping)\n");
363                                                                         liMove.QuadPart = Chunk.dwSize - sizeof(FOURCC);
364                                                                         IStream_Seek (pStream, liMove, STREAM_SEEK_CUR, NULL);
365                                                                         break;                                          
366                                                                 }
367                                                         }
368                                                         break;
369                                                 }       
370                                                 default: {
371                                                         TRACE_(dmfile)(": unknown chunk (irrevelant & skipping)\n");
372                                                         liMove.QuadPart = Chunk.dwSize;
373                                                         IStream_Seek (pStream, liMove, STREAM_SEEK_CUR, NULL);
374                                                         break;                                          
375                                                 }
376                                         }
377                                         TRACE_(dmfile)(": StreamCount[0] = 0x%08lX < StreamSize[0] = 0x%08lX\n", StreamCount, StreamSize);
378                                 } while (StreamCount < StreamSize);
379                                 break;
380                         } else {
381                                 TRACE_(dmfile)(": unexpected chunk; loading failed)\n");
382                                 liMove.QuadPart = StreamSize;
383                                 IStream_Seek (pStream, liMove, STREAM_SEEK_CUR, NULL); /* skip the rest of the chunk */
384                                 return E_FAIL;
385                         }
386                 
387                         TRACE_(dmfile)(": reading finished\n");
388                         break;
389                 }
390                 default: {
391                         TRACE_(dmfile)(": unexpected chunk; loading failed)\n");
392                         liMove.QuadPart = Chunk.dwSize;
393                         IStream_Seek (pStream, liMove, STREAM_SEEK_CUR, NULL); /* skip the rest of the chunk */
394                         return DMUS_E_INVALIDFILE;
395                 }
396         }       
397         
398         TRACE(": returning descriptor:\n%s\n", debugstr_DMUS_OBJECTDESC(pDesc));
399         return S_OK;    
400 }
401
402 ICOM_VTABLE(IDirectMusicObject) DirectMusicContainer_Object_Vtbl = {
403     ICOM_MSVTABLE_COMPAT_DummyRTTIVALUE
404         IDirectMusicContainerImpl_IDirectMusicObject_QueryInterface,
405         IDirectMusicContainerImpl_IDirectMusicObject_AddRef,
406         IDirectMusicContainerImpl_IDirectMusicObject_Release,
407         IDirectMusicContainerImpl_IDirectMusicObject_GetDescriptor,
408         IDirectMusicContainerImpl_IDirectMusicObject_SetDescriptor,
409         IDirectMusicContainerImpl_IDirectMusicObject_ParseDescriptor
410 };
411
412 /* IPersistStream part: */
413 HRESULT WINAPI IDirectMusicContainerImpl_IPersistStream_QueryInterface (LPPERSISTSTREAM iface, REFIID riid, LPVOID *ppobj) {
414         ICOM_THIS_MULTI(IDirectMusicContainerImpl, PersistStreamVtbl, iface);
415         return IDirectMusicContainerImpl_IDirectMusicContainer_QueryInterface ((LPDIRECTMUSICCONTAINER)&This->ContainerVtbl, riid, ppobj);
416 }
417
418 ULONG WINAPI IDirectMusicContainerImpl_IPersistStream_AddRef (LPPERSISTSTREAM iface) {
419         ICOM_THIS_MULTI(IDirectMusicContainerImpl, PersistStreamVtbl, iface);
420         return IDirectMusicContainerImpl_IDirectMusicContainer_AddRef ((LPDIRECTMUSICCONTAINER)&This->ContainerVtbl);
421 }
422
423 ULONG WINAPI IDirectMusicContainerImpl_IPersistStream_Release (LPPERSISTSTREAM iface) {
424         ICOM_THIS_MULTI(IDirectMusicContainerImpl, PersistStreamVtbl, iface);
425         return IDirectMusicContainerImpl_IDirectMusicContainer_Release ((LPDIRECTMUSICCONTAINER)&This->ContainerVtbl);
426 }
427
428 HRESULT WINAPI IDirectMusicContainerImpl_IPersistStream_GetClassID (LPPERSISTSTREAM iface, CLSID* pClassID) {
429         ICOM_THIS_MULTI(IDirectMusicContainerImpl, PersistStreamVtbl, iface);
430         
431         TRACE("(%p, %p)\n", This, pClassID);
432         if (IsBadWritePtr (pClassID, sizeof(CLSID))) {
433                 ERR(": pClassID bad write pointer\n");
434                 return E_POINTER;
435         }
436         
437         *pClassID = CLSID_DirectMusicContainer;
438         return S_OK;
439 }
440
441 HRESULT WINAPI IDirectMusicContainerImpl_IPersistStream_IsDirty (LPPERSISTSTREAM iface) {
442         /* FIXME: is implemented (somehow) */
443         return E_NOTIMPL;
444 }
445
446 HRESULT WINAPI IDirectMusicContainerImpl_IPersistStream_Load (LPPERSISTSTREAM iface, IStream* pStm) {
447         ICOM_THIS_MULTI(IDirectMusicContainerImpl, PersistStreamVtbl, iface);
448         WINE_CHUNK Chunk;
449         DWORD StreamSize, StreamCount, ListSize[3], ListCount[3];
450         LARGE_INTEGER liMove; /* used when skipping chunks */
451         ULARGE_INTEGER uliPos; /* needed when dealing with RIFF chunks */
452         LPDIRECTMUSICGETLOADER pGetLoader;
453         LPDIRECTMUSICLOADER pLoader;
454         HRESULT result = S_OK;
455
456         TRACE("(%p, %p):\n", This, pStm);
457         
458         /* check whether pStm is valid read pointer */
459         if (IsBadReadPtr (pStm, sizeof(LPVOID))) {
460                 ERR(": pStm bad read pointer\n");
461                 return E_POINTER;
462         }
463         /* if stream is already set, this means we're loaded already */
464         if (This->pStream) {
465                 TRACE(": stream is already set, which means container is already loaded\n");
466                 return DMUS_E_ALREADY_LOADED;
467         }
468
469         /* get loader since it will be needed later */
470         if (FAILED(IStream_QueryInterface (pStm, &IID_IDirectMusicGetLoader, (LPVOID*)&pGetLoader))) {
471                 ERR(": stream not supported\n");
472                 return DMUS_E_UNSUPPORTED_STREAM;
473         }
474         IDirectMusicGetLoader_GetLoader (pGetLoader, &pLoader);
475         IDirectMusicGetLoader_Release (pGetLoader);
476         
477         This->pStream = pStm;
478         IStream_AddRef (pStm); /* add count for later references */
479         
480         /* start with load */
481         IStream_Read (pStm, &Chunk, sizeof(FOURCC)+sizeof(DWORD), NULL);
482         TRACE_(dmfile)(": %s chunk (size = 0x%08lX)", debugstr_fourcc (Chunk.fccID), Chunk.dwSize);
483         switch (Chunk.fccID) {  
484                 case FOURCC_RIFF: {
485                         IStream_Read (pStm, &Chunk.fccID, sizeof(FOURCC), NULL);                                
486                         TRACE_(dmfile)(": RIFF chunk of type %s", debugstr_fourcc(Chunk.fccID));
487                         StreamSize = Chunk.dwSize - sizeof(FOURCC);
488                         StreamCount = 0;
489                         switch (Chunk.fccID) {
490                                 case DMUS_FOURCC_CONTAINER_FORM: {
491                                         TRACE_(dmfile)(": container form\n");
492                                         memcpy (&This->Desc.guidClass, &CLSID_DirectMusicContainer, sizeof(CLSID));
493                                         This->Desc.dwValidData |= DMUS_OBJ_CLASS;
494                                         do {
495                                                 IStream_Read (pStm, &Chunk, sizeof(FOURCC)+sizeof(DWORD), NULL);
496                                                 StreamCount += sizeof(FOURCC) + sizeof(DWORD) + Chunk.dwSize;
497                                                 TRACE_(dmfile)(": %s chunk (size = 0x%08lX)", debugstr_fourcc (Chunk.fccID), Chunk.dwSize);
498                                                 switch (Chunk.fccID) {
499                                                         case DMUS_FOURCC_CONTAINER_CHUNK: {
500                                                                 TRACE_(dmfile)(": container header chunk\n");
501                                                                 IStream_Read (pStm, &This->Header, Chunk.dwSize, NULL);
502                                                                 TRACE_(dmdump)(": container header chunk:\n%s", debugstr_DMUS_IO_CONTAINER_HEADER(&This->Header));
503                                                                 break;  
504                                                         }
505                                                         case DMUS_FOURCC_GUID_CHUNK: {
506                                                                 TRACE_(dmfile)(": GUID chunk\n");
507                                                                 IStream_Read (pStm, &This->Desc.guidObject, Chunk.dwSize, NULL);
508                                                                 This->Desc.dwValidData |= DMUS_OBJ_OBJECT;
509                                                                 TRACE_(dmdump)(": GUID: %s\n", debugstr_guid(&This->Desc.guidObject));
510                                                                 break;
511                                                         }
512                                                         case DMUS_FOURCC_VERSION_CHUNK: {
513                                                                 TRACE_(dmfile)(": version chunk\n");
514                                                                 IStream_Read (pStm, &This->Desc.vVersion, Chunk.dwSize, NULL);
515                                                                 This->Desc.dwValidData |= DMUS_OBJ_VERSION;
516                                                                 TRACE_(dmdump)(": version: %s\n", debugstr_dmversion(&This->Desc.vVersion));
517                                                                 break;
518                                                         }
519                                                         case DMUS_FOURCC_DATE_CHUNK: {
520                                                                 TRACE_(dmfile)(": date chunk\n");
521                                                                 IStream_Read (pStm, &This->Desc.ftDate, Chunk.dwSize, NULL);
522                                                                 This->Desc.dwValidData |= DMUS_OBJ_DATE;
523                                                                 TRACE_(dmdump)(": date: %s\n", debugstr_filetime(&This->Desc.ftDate));
524                                                                 break;
525                                                         }                                                       
526                                                         case DMUS_FOURCC_CATEGORY_CHUNK: {
527                                                                 TRACE_(dmfile)(": category chunk\n");
528                                                                 /* if it happens that string is too long,
529                                                                    read what we can and skip the rest*/
530                                                                 if (Chunk.dwSize > DMUS_MAX_CATEGORY_SIZE) {
531                                                                         IStream_Read (pStm, This->Desc.wszCategory, DMUS_MAX_CATEGORY_SIZE, NULL);
532                                                                         liMove.QuadPart = Chunk.dwSize - DMUS_MAX_CATEGORY_SIZE;
533                                                                         IStream_Seek (pStm, liMove, STREAM_SEEK_CUR, NULL);
534                                                                 } else {
535                                                                         IStream_Read (pStm, This->Desc.wszCategory, Chunk.dwSize, NULL);
536                                                                 }
537                                                                 This->Desc.dwValidData |= DMUS_OBJ_CATEGORY;
538                                                                 TRACE_(dmdump)(": category: %s\n", debugstr_w(This->Desc.wszCategory));                                                         
539                                                                 break;
540                                                         }
541                                                         case FOURCC_LIST: {
542                                                                 IStream_Read (pStm, &Chunk.fccID, sizeof(FOURCC), NULL);                                
543                                                                 TRACE_(dmfile)(": LIST chunk of type %s", debugstr_fourcc(Chunk.fccID));
544                                                                 ListSize[0] = Chunk.dwSize - sizeof(FOURCC);
545                                                                 ListCount[0] = 0;
546                                                                 switch (Chunk.fccID) {
547                                                                         case DMUS_FOURCC_UNFO_LIST: {
548                                                                                 TRACE_(dmfile)(": UNFO list\n");
549                                                                                 do {
550                                                                                         IStream_Read (pStm, &Chunk, sizeof(FOURCC)+sizeof(DWORD), NULL);
551                                                                                         ListCount[0] += sizeof(FOURCC) + sizeof(DWORD) + Chunk.dwSize;
552                                                                                         TRACE_(dmfile)(": %s chunk (size = 0x%08lX)", debugstr_fourcc (Chunk.fccID), Chunk.dwSize);
553                                                                                         switch (Chunk.fccID) {
554                                                                                                 /* don't ask me why, but M$ puts INFO elements in UNFO list sometimes
555                                               (though strings seem to be valid unicode) */
556                                                                                                 case mmioFOURCC('I','N','A','M'):
557                                                                                                 case DMUS_FOURCC_UNAM_CHUNK: {
558                                                                                                         TRACE_(dmfile)(": name chunk\n");
559                                                                                                         /* if it happens that string is too long,
560                                                                                                            read what we can and skip the rest*/
561                                                                                                         if (Chunk.dwSize > DMUS_MAX_NAME_SIZE) {
562                                                                                                                 IStream_Read (pStm, This->Desc.wszName, DMUS_MAX_NAME_SIZE, NULL);
563                                                                                                                 liMove.QuadPart = Chunk.dwSize - DMUS_MAX_NAME_SIZE;
564                                                                                                                 IStream_Seek (pStm, liMove, STREAM_SEEK_CUR, NULL);
565                                                                                                         } else {
566                                                                                                                 IStream_Read (pStm, This->Desc.wszName, Chunk.dwSize, NULL);
567                                                                                                         }
568                                                                                                         This->Desc.dwValidData |= DMUS_OBJ_NAME;
569                                                                                                         TRACE_(dmdump)(": name: %s\n", debugstr_w(This->Desc.wszName));
570                                                                                                         break;
571                                                                                                 }
572                                                                                                 default: {
573                                                                                                         TRACE_(dmfile)(": unknown chunk (irrevelant & skipping)\n");
574                                                                                                         liMove.QuadPart = Chunk.dwSize;
575                                                                                                         IStream_Seek (pStm, liMove, STREAM_SEEK_CUR, NULL);
576                                                                                                         break;                                          
577                                                                                                 }
578                                                                                         }
579                                                                                         TRACE_(dmfile)(": ListCount[0] = 0x%08lX < ListSize[0] = 0x%08lX\n", ListCount[0], ListSize[0]);
580                                                                                 } while (ListCount[0] < ListSize[0]);
581                                                                                 break;
582                                                                         }
583                                                                         case DMUS_FOURCC_CONTAINED_OBJECTS_LIST: {
584                                                                                 TRACE_(dmfile)(": contained objects list\n");
585                                                                                 do {
586                                                                                         IStream_Read (pStm, &Chunk, sizeof(FOURCC)+sizeof(DWORD), NULL);
587                                                                                         ListCount[0] += sizeof(FOURCC) + sizeof(DWORD) + Chunk.dwSize;
588                                                                                         TRACE_(dmfile)(": %s chunk (size = 0x%08lX)", debugstr_fourcc (Chunk.fccID), Chunk.dwSize);
589                                                                                         switch (Chunk.fccID) {
590                                                                                                 case FOURCC_LIST: {
591                                                                                                         IStream_Read (pStm, &Chunk.fccID, sizeof(FOURCC), NULL);                                
592                                                                                                         TRACE_(dmfile)(": LIST chunk of type %s", debugstr_fourcc(Chunk.fccID));
593                                                                                                         ListSize[1] = Chunk.dwSize - sizeof(FOURCC);
594                                                                                                         ListCount[1] = 0;
595                                                                                                         switch (Chunk.fccID) {
596                                                                                                                 case DMUS_FOURCC_CONTAINED_OBJECT_LIST: {
597                                                                                                                         LPWINE_CONTAINER_ENTRY pNewEntry;
598                                                                                                                         TRACE_(dmfile)(": contained object list\n");
599                                                                                                                         pNewEntry = HeapAlloc (GetProcessHeap (), HEAP_ZERO_MEMORY, sizeof(WINE_CONTAINER_ENTRY));
600                                                                                                                         DM_STRUCT_INIT(&pNewEntry->Desc);
601                                                                                                                         do {
602                                                                                                                                 IStream_Read (pStm, &Chunk, sizeof(FOURCC)+sizeof(DWORD), NULL);
603                                                                                                                                 ListCount[1] += sizeof(FOURCC) + sizeof(DWORD) + Chunk.dwSize;
604                                                                                                                                 TRACE_(dmfile)(": %s chunk (size = 0x%08lX)", debugstr_fourcc (Chunk.fccID), Chunk.dwSize);
605                                                                                                                                 switch (Chunk.fccID) {
606                                                                                                                                         case DMUS_FOURCC_CONTAINED_ALIAS_CHUNK: {
607                                                                                                                                                 TRACE_(dmfile)(": alias chunk\n");
608                                                                                                                                                 pNewEntry->wszAlias = HeapAlloc (GetProcessHeap (), HEAP_ZERO_MEMORY, Chunk.dwSize);
609                                                                                                                                                 IStream_Read (pStm, pNewEntry->wszAlias, Chunk.dwSize, NULL);
610                                                                                                                                                 TRACE_(dmdump)(": alias: %s\n", debugstr_w(pNewEntry->wszAlias));
611                                                                                                                                                 break;
612                                                                                                                                         }
613                                                                                                                                         case DMUS_FOURCC_CONTAINED_OBJECT_CHUNK: {
614                                                                                                                                                 DMUS_IO_CONTAINED_OBJECT_HEADER tmpObjectHeader;
615                                                                                                                                                 TRACE_(dmfile)(": contained object header chunk\n");
616                                                                                                                                                 IStream_Read (pStm, &tmpObjectHeader, Chunk.dwSize, NULL);
617                                                                                                                                                 TRACE_(dmdump)(": contained object header: \n%s", debugstr_DMUS_IO_CONTAINED_OBJECT_HEADER(&tmpObjectHeader));
618                                                                                                                                                 /* copy guidClass */
619                                                                                                                                                 pNewEntry->Desc.dwValidData |= DMUS_OBJ_CLASS;
620                                                                                                                                                 memcpy (&pNewEntry->Desc.guidClass, &tmpObjectHeader.guidClassID, sizeof(GUID));
621                                                                                                                                                 /* store flags */
622                                                                                                                                                 pNewEntry->dwFlags = tmpObjectHeader.dwFlags;
623                                                                                                                                                 break;
624                                                                                                                                         }
625                                                                                                                                         /* now read data... it may be safe to read everything after object header chunk, 
626                                                                                                                                                 but I'm not comfortable with MSDN's "the header is *normally* followed by ..." */
627                                                                                                                                         case FOURCC_LIST: {
628                                                                                                                                                 IStream_Read (pStm, &Chunk.fccID, sizeof(FOURCC), NULL);                                
629                                                                                                                                                 TRACE_(dmfile)(": LIST chunk of type %s", debugstr_fourcc(Chunk.fccID));
630                                                                                                                                                 ListSize[2] = Chunk.dwSize - sizeof(FOURCC);
631                                                                                                                                                 ListCount[2] = 0;
632                                                                                                                                                 switch (Chunk.fccID) {
633                                                                                                                                                         case DMUS_FOURCC_REF_LIST: {
634                                                                                                                                                                 TRACE_(dmfile)(": reference list\n");
635                                                                                                                                                                 pNewEntry->bIsRIFF = 0;
636                                                                                                                                                                 do {
637                                                                                                                                                                         IStream_Read (pStm, &Chunk, sizeof(FOURCC)+sizeof(DWORD), NULL);
638                                                                                                                                                                         ListCount[2] += sizeof(FOURCC) + sizeof(DWORD) + Chunk.dwSize;
639                                                                                                                                                                         TRACE_(dmfile)(": %s chunk (size = 0x%08lX)", debugstr_fourcc (Chunk.fccID), Chunk.dwSize);
640                                                                                                                                                                         switch (Chunk.fccID) {
641                                                                                                                                                                                 case DMUS_FOURCC_REF_CHUNK: {
642                                                                                                                                                                                         DMUS_IO_REFERENCE tmpReferenceHeader; /* temporary structure */
643                                                                                                                                                                                         TRACE_(dmfile)(": reference header chunk\n");
644                                                                                                                                                                                         memset (&tmpReferenceHeader, 0, sizeof(DMUS_IO_REFERENCE));
645                                                                                                                                                                                         IStream_Read (pStm, &tmpReferenceHeader, Chunk.dwSize, NULL);
646                                                                                                                                                                                         /* copy retrieved data to DMUS_OBJECTDESC */
647                                                                                                                                                                                         if (!IsEqualCLSID (&pNewEntry->Desc.guidClass, &tmpReferenceHeader.guidClassID)) ERR(": object header declares different CLSID than reference header?\n");
648                                                                                                                                                                                         /* it shouldn't be necessary to copy guidClass, since it was set in contained object header already...
649                                                                                                                                                                                            yet if they happen to be different, I'd rather stick to this one */
650                                                                                                                                                                                         memcpy (&pNewEntry->Desc.guidClass, &tmpReferenceHeader.guidClassID, sizeof(GUID));
651                                                                                                                                                                                         pNewEntry->Desc.dwValidData |= tmpReferenceHeader.dwValidData;
652                                                                                                                                                                                         break;                                                                                                                                  
653                                                                                                                                                                                 }
654                                                                                                                                                                                 case DMUS_FOURCC_GUID_CHUNK: {
655                                                                                                                                                                                         TRACE_(dmfile)(": guid chunk\n");
656                                                                                                                                                                                         /* no need to set flags since they were copied from reference header */
657                                                                                                                                                                                         IStream_Read (pStm, &pNewEntry->Desc.guidObject, Chunk.dwSize, NULL);
658                                                                                                                                                                                         break;
659                                                                                                                                                                                 }
660                                                                                                                                                                                 case DMUS_FOURCC_DATE_CHUNK: {
661                                                                                                                                                                                         TRACE_(dmfile)(": file date chunk\n");
662                                                                                                                                                                                         /* no need to set flags since they were copied from reference header */
663                                                                                                                                                                                         IStream_Read (pStm, &pNewEntry->Desc.ftDate, Chunk.dwSize, NULL);
664                                                                                                                                                                                         break;
665                                                                                                                                                                                 }
666                                                                                                                                                                                 case DMUS_FOURCC_NAME_CHUNK: {
667                                                                                                                                                                                         TRACE_(dmfile)(": name chunk\n");
668                                                                                                                                                                                         /* no need to set flags since they were copied from reference header */
669                                                                                                                                                                                         IStream_Read (pStm, pNewEntry->Desc.wszName, Chunk.dwSize, NULL);
670                                                                                                                                                                                         break;
671                                                                                                                                                                                 }
672                                                                                                                                                                                 case DMUS_FOURCC_FILE_CHUNK: {
673                                                                                                                                                                                         TRACE_(dmfile)(": file name chunk\n");
674                                                                                                                                                                                         /* no need to set flags since they were copied from reference header */
675                                                                                                                                                                                         IStream_Read (pStm, pNewEntry->Desc.wszFileName, Chunk.dwSize, NULL);
676                                                                                                                                                                                         break;
677                                                                                                                                                                                 }
678                                                                                                                                                                                 case DMUS_FOURCC_CATEGORY_CHUNK: {
679                                                                                                                                                                                         TRACE_(dmfile)(": category chunk\n");
680                                                                                                                                                                                         /* no need to set flags since they were copied from reference header */
681                                                                                                                                                                                         IStream_Read (pStm, pNewEntry->Desc.wszCategory, Chunk.dwSize, NULL);
682                                                                                                                                                                                         break;
683                                                                                                                                                                                 }
684                                                                                                                                                                                 case DMUS_FOURCC_VERSION_CHUNK: {
685                                                                                                                                                                                         TRACE_(dmfile)(": version chunk\n");
686                                                                                                                                                                                         /* no need to set flags since they were copied from reference header */
687                                                                                                                                                                                         IStream_Read (pStm, &pNewEntry->Desc.vVersion, Chunk.dwSize, NULL);
688                                                                                                                                                                                         break;
689                                                                                                                                                                                 }
690                                                                                                                                                                                 default: {
691                                                                                                                                                                                         TRACE_(dmfile)(": unknown chunk (skipping)\n");
692                                                                                                                                                                                         liMove.QuadPart = Chunk.dwSize;
693                                                                                                                                                                                         IStream_Seek (pStm, liMove, STREAM_SEEK_CUR, NULL); /* skip this chunk */
694                                                                                                                                                                                         break;
695                                                                                                                                                                                 }
696                                                                                                                                                                         }
697                                                                                                                                                                         TRACE_(dmfile)(": ListCount[2] = 0x%08lX < ListSize[2] = 0x%08lX\n", ListCount[2], ListSize[2]);
698                                                                                                                                                                 } while (ListCount[2] < ListSize[2]);
699                                                                                                                                                                 break;
700                                                                                                                                                         }
701                                                                                                                                                         default: {
702                                                                                                                                                                 TRACE_(dmfile)(": unexpected chunk; loading failed)\n");
703                                                                                                                                                                 return E_FAIL;
704                                                                                                                                                         }
705                                                                                                                                                 }
706                                                                                                                                                 break;
707                                                                                                                                         }
708                                                                                                                                         
709                                                                                                                                         case FOURCC_RIFF: {
710                                                                                                                                                 IStream_Read (pStm, &Chunk.fccID, sizeof(FOURCC), NULL);
711                                                                                                                                                 TRACE_(dmfile)(": RIFF chunk of type %s", debugstr_fourcc(Chunk.fccID));
712                                                                                                                                                 if (IS_VALID_DMFORM (Chunk.fccID)) {
713                                                                                                                                                         TRACE_(dmfile)(": valid DMUSIC form\n");
714                                                                                                                                                         pNewEntry->bIsRIFF = 1;
715                                                                                                                                                         /* we'll have to skip whole RIFF chunk after SetObject is called */
716                                                                                                                                                         liMove.QuadPart = 0;
717                                                                                                                                                         IStream_Seek (pStm, liMove, STREAM_SEEK_CUR, &uliPos);
718                                                                                                                                                         uliPos.QuadPart += (Chunk.dwSize - sizeof(FOURCC)); /* set uliPos at the end of RIFF chunk */                                                                                                                                                   
719                                                                                                                                                         /* move at the beginning of RIFF chunk */
720                                                                                                                                                         liMove.QuadPart = 0;
721                                                                                                                                                         liMove.QuadPart -= (sizeof(FOURCC)+sizeof(DWORD)+sizeof(FOURCC));
722                                                                                                                                                         IStream_Seek (pStm, liMove, STREAM_SEEK_CUR, NULL);
723                                                                                                                                                         /* put pointer to stream in descriptor */
724                                                                                                                                                         pNewEntry->Desc.dwValidData |= DMUS_OBJ_STREAM;
725                                                                                                                                                         pNewEntry->Desc.pStream = pStm; /* we don't have to worry about cloning, since SetObject will perform it */
726                                                                                                                                                         /* wait till we get on the end of object list */
727                                                                                                                                                 } else {
728                                                                                                                                                         TRACE_(dmfile)(": invalid DMUSIC form (skipping)\n");
729                                                                                                                                                         liMove.QuadPart = Chunk.dwSize - sizeof(FOURCC);
730                                                                                                                                                         IStream_Seek (pStm, liMove, STREAM_SEEK_CUR, NULL);
731                                                                                                                                                         /* FIXME: should we return E_FAIL? */
732                                                                                                                                                 }
733                                                                                                                                                 break;
734                                                                                                                                         }
735                                                                                                                                         default: {
736                                                                                                                                                 TRACE_(dmfile)(": unknown chunk (irrevelant & skipping)\n");
737                                                                                                                                                 liMove.QuadPart = Chunk.dwSize;
738                                                                                                                                                 IStream_Seek (pStm, liMove, STREAM_SEEK_CUR, NULL);
739                                                                                                                                                 break;                                          
740                                                                                                                                         }
741                                                                                                                                 }
742                                                                                                                                 TRACE_(dmfile)(": ListCount[1] = 0x%08lX < ListSize[1] = 0x%08lX\n", ListCount[1], ListSize[1]);
743                                                                                                                         } while (ListCount[1] < ListSize[1]);
744                                                                                                                         /* SetObject: this will fill descriptor with additional info    and add alias in loader's cache */
745                                                                                                                         IDirectMusicLoader_SetObject (pLoader, &pNewEntry->Desc);
746                                                                                                                         /* now that SetObject collected appropriate info into descriptor we can live happily ever after;
747                                                                                                                            or not, since we have to clean evidence of loading through stream... *sigh*
748                                                                                                                            and we have to skip rest of the chunk, if we loaded through RIFF */
749                                                                                                                         if (pNewEntry->bIsRIFF) {
750                                                                                                                                 liMove.QuadPart = uliPos.QuadPart;
751                                                                                                                                 IStream_Seek (pStm, liMove, STREAM_SEEK_SET, NULL);
752                                                                                                                                 pNewEntry->Desc.dwValidData &= ~DMUS_OBJ_STREAM; /* clear flag (and with bitwise complement) */
753                                                                                                                                 pNewEntry->Desc.pStream = NULL;                                                                                 
754                                                                                                                         }
755                                                                                                                         /* add entry to list of objects */
756                                                                                                                         list_add_tail (This->pContainedObjects, &pNewEntry->entry);
757                                                                                                                         break;
758                                                                                                                 }
759                                                                                                                 default: {
760                                                                                                                         TRACE_(dmfile)(": unknown (skipping)\n");
761                                                                                                                         liMove.QuadPart = Chunk.dwSize - sizeof(FOURCC);
762                                                                                                                         IStream_Seek (pStm, liMove, STREAM_SEEK_CUR, NULL);
763                                                                                                                         break;                                          
764                                                                                                                 }
765                                                                                                         }
766                                                                                                         break;
767                                                                                                 }
768                                                                                                 default: {
769                                                                                                         TRACE_(dmfile)(": unknown chunk (irrevelant & skipping)\n");
770                                                                                                         liMove.QuadPart = Chunk.dwSize;
771                                                                                                         IStream_Seek (pStm, liMove, STREAM_SEEK_CUR, NULL);
772                                                                                                         break;                                          
773                                                                                                 }
774                                                                                         }
775                                                                                         TRACE_(dmfile)(": ListCount[0] = 0x%08lX < ListSize[0] = 0x%08lX\n", ListCount[0], ListSize[0]);
776                                                                                 } while (ListCount[0] < ListSize[0]);
777                                                                                 break;
778                                                                         }                                                                       
779                                                                         default: {
780                                                                                 TRACE_(dmfile)(": unknown (skipping)\n");
781                                                                                 liMove.QuadPart = Chunk.dwSize - sizeof(FOURCC);
782                                                                                 IStream_Seek (pStm, liMove, STREAM_SEEK_CUR, NULL);
783                                                                                 break;                                          
784                                                                         }
785                                                                 }
786                                                                 break;
787                                                         }       
788                                                         default: {
789                                                                 TRACE_(dmfile)(": unknown chunk (irrevelant & skipping)\n");
790                                                                 liMove.QuadPart = Chunk.dwSize;
791                                                                 IStream_Seek (pStm, liMove, STREAM_SEEK_CUR, NULL);
792                                                                 break;                                          
793                                                         }
794                                                 }
795                                                 TRACE_(dmfile)(": StreamCount[0] = 0x%08lX < StreamSize[0] = 0x%08lX\n", StreamCount, StreamSize);
796                                         } while (StreamCount < StreamSize);
797                                         break;
798                                 }
799                                 default: {
800                                         TRACE_(dmfile)(": unexpected chunk; loading failed)\n");
801                                         liMove.QuadPart = StreamSize;
802                                         IStream_Seek (pStm, liMove, STREAM_SEEK_CUR, NULL); /* skip the rest of the chunk */
803                                         return E_FAIL;
804                                 }
805                         }
806                         TRACE_(dmfile)(": reading finished\n");
807                         This->Desc.dwValidData |= DMUS_OBJ_LOADED;
808                         break;
809                 }
810                 default: {
811                         TRACE_(dmfile)(": unexpected chunk; loading failed)\n");
812                         liMove.QuadPart = Chunk.dwSize;
813                         IStream_Seek (pStm, liMove, STREAM_SEEK_CUR, NULL); /* skip the rest of the chunk */
814                         return E_FAIL;
815                 }
816         }
817         
818         /* now, if DMUS_CONTAINER_NOLOADS is not set, we are supposed to load contained objects;
819            so when we call GetObject later, they'll already be in cache */
820         if (!(This->Header.dwFlags & DMUS_CONTAINER_NOLOADS)) {
821                 struct list *pEntry;
822                 LPWINE_CONTAINER_ENTRY pContainedObject;
823
824                 TRACE(": DMUS_CONTAINER_NOLOADS not set... load all objects\n");
825                 
826                 LIST_FOR_EACH (pEntry, This->pContainedObjects) {
827                         IDirectMusicObject* pObject;
828                         pContainedObject = LIST_ENTRY (pEntry, WINE_CONTAINER_ENTRY, entry);            
829                         /* get object from loader and then release it */
830                         if (SUCCEEDED(IDirectMusicLoader_GetObject (pLoader, &pContainedObject->Desc, &IID_IDirectMusicObject, (LPVOID*)&pObject))) {
831                                 pContainedObject->pObject = pObject; /* for final release */
832                                 IDirectMusicObject_Release (pObject); /* we don't really need this one */
833                         } else {
834                                 WARN(": failed to load contained object\n");
835                                 result = DMUS_S_PARTIALLOAD;
836                         }
837                 }
838         }
839         
840         IDirectMusicLoader_Release (pLoader); /* release loader */
841
842 #if 0
843         /* DEBUG: dumps whole container object tree: */
844         if (TRACE_ON(dmloader)) {
845                 int r = 0;
846                 LPWINE_CONTAINER_ENTRY tmpEntry;
847                 struct list *listEntry;
848
849                 TRACE("*** IDirectMusicContainer (%p) ***\n", This->ContainerVtbl);
850                 TRACE(" - Objects:\n");
851                 LIST_FOR_EACH (listEntry, This->pContainedObjects) {
852                         tmpEntry = LIST_ENTRY( listEntry, WINE_CONTAINER_ENTRY, entry );
853                         TRACE("    - Object[%i]:\n", r);
854                         TRACE("       - wszAlias: %s\n", debugstr_w(tmpEntry->wszAlias));
855                         TRACE("       - Object descriptor:\n%s\n", debugstr_DMUS_OBJECTDESC(&tmpEntry->Desc));
856                         r++;
857                 }
858         }
859 #endif
860
861         return result;
862 }
863
864 HRESULT WINAPI IDirectMusicContainerImpl_IPersistStream_Save (LPPERSISTSTREAM iface, IStream* pStm, BOOL fClearDirty) {
865         ERR(": should not be needed\n");
866         return E_NOTIMPL;
867 }
868
869 HRESULT WINAPI IDirectMusicContainerImpl_IPersistStream_GetSizeMax (LPPERSISTSTREAM iface, ULARGE_INTEGER* pcbSize) {
870         ERR(": should not be needed\n");
871         return E_NOTIMPL;
872 }
873
874 ICOM_VTABLE(IPersistStream) DirectMusicContainer_PersistStream_Vtbl = {
875     ICOM_MSVTABLE_COMPAT_DummyRTTIVALUE
876         IDirectMusicContainerImpl_IPersistStream_QueryInterface,
877         IDirectMusicContainerImpl_IPersistStream_AddRef,
878         IDirectMusicContainerImpl_IPersistStream_Release,
879         IDirectMusicContainerImpl_IPersistStream_GetClassID,
880         IDirectMusicContainerImpl_IPersistStream_IsDirty,
881         IDirectMusicContainerImpl_IPersistStream_Load,
882         IDirectMusicContainerImpl_IPersistStream_Save,
883         IDirectMusicContainerImpl_IPersistStream_GetSizeMax
884 };
885
886 /* for ClassFactory */
887 HRESULT WINAPI DMUSIC_CreateDirectMusicContainerImpl (LPCGUID lpcGUID, LPVOID* ppobj, LPUNKNOWN pUnkOuter) {
888         IDirectMusicContainerImpl* obj;
889
890         obj = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(IDirectMusicContainerImpl));
891         if (NULL == obj) {
892                 *ppobj = (LPVOID) NULL;
893                 return E_OUTOFMEMORY;
894         }
895         obj->ContainerVtbl = &DirectMusicContainer_Container_Vtbl;
896         obj->ObjectVtbl = &DirectMusicContainer_Object_Vtbl;
897         obj->PersistStreamVtbl = &DirectMusicContainer_PersistStream_Vtbl;
898         obj->dwRef = 0; /* will be inited by QueryInterface */
899         obj->pContainedObjects = HeapAlloc (GetProcessHeap (), HEAP_ZERO_MEMORY, sizeof(struct list));
900         list_init (obj->pContainedObjects);
901
902         /* increase number of instances */
903         InterlockedIncrement (&dwDirectMusicContainer);
904         
905         return IDirectMusicContainerImpl_IDirectMusicContainer_QueryInterface ((LPDIRECTMUSICCONTAINER)&obj->ContainerVtbl, lpcGUID, ppobj);
906 }
907
908 HRESULT WINAPI DMUSIC_DestroyDirectMusicContainerImpl (LPDIRECTMUSICCONTAINER iface) {
909         ICOM_THIS_MULTI(IDirectMusicContainerImpl, ContainerVtbl, iface);
910         LPDIRECTMUSICLOADER pLoader;
911         LPDIRECTMUSICGETLOADER pGetLoader;
912         struct list *pEntry;
913         LPWINE_CONTAINER_ENTRY pContainedObject;
914
915         /* get loader (from stream we loaded from) */
916         TRACE(": getting loader\n");
917         IStream_QueryInterface (This->pStream, &IID_IDirectMusicGetLoader, (LPVOID*)&pGetLoader);
918         IDirectMusicGetLoader_GetLoader (pGetLoader, &pLoader);
919         IDirectMusicGetLoader_Release (pGetLoader);
920         
921         /* release objects from loader's cache (if appropriate) */
922         TRACE(": releasing objects from loader's cache\n");
923         LIST_FOR_EACH (pEntry, This->pContainedObjects) {
924                 pContainedObject = LIST_ENTRY (pEntry, WINE_CONTAINER_ENTRY, entry);
925                 /* my tests indicate that container releases objects *only* 
926                    if they were loaded at it's load-time (makes sense, it doesn't
927                    have pointers to objects otherwise); BTW: native container seems
928                    ti ignore the flags (I won't) */
929                 if (pContainedObject->pObject && !(pContainedObject->dwFlags & DMUS_CONTAINED_OBJF_KEEP)) {
930                         /* flags say it shouldn't be kept in loader's cache */
931                         IDirectMusicLoader_ReleaseObject (pLoader, pContainedObject->pObject);
932                 }
933         }       
934         IDirectMusicLoader_Release (pLoader);
935         
936         /* release stream we loaded from */
937         IStream_Release (This->pStream);
938         
939         /* FIXME: release allocated entries */
940         
941         /* decrease number of instances */
942         InterlockedDecrement (&dwDirectMusicContainer); 
943         
944         return S_OK;
945 }