wined3d: Add a separate function for query initialization.
[wine] / dlls / dmloader / loader.c
1 /* IDirectMusicLoaderImpl
2  *
3  * Copyright (C) 2003-2004 Rok Mandeljc
4  *
5  * This program is free software; you can redistribute it and/or
6  * modify it under the terms of the GNU Lesser General Public
7  * License as published by the Free Software Foundation; either
8  * version 2.1 of the License, or (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 GNU
13  * Lesser General Public License for more details.
14  *
15  * You should have received a copy of the GNU Lesser General Public
16  * License along with this program; if not, write to the Free Software
17  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
18  */
19
20 #include "dmloader_private.h"
21
22 WINE_DEFAULT_DEBUG_CHANNEL(dmloader);
23
24 static HRESULT DMUSIC_InitLoaderSettings (LPDIRECTMUSICLOADER8 iface);
25 static HRESULT DMUSIC_GetLoaderSettings (LPDIRECTMUSICLOADER8 iface, REFGUID pClassID, WCHAR* wszSearchPath, LPBOOL pbCache);
26 static HRESULT DMUSIC_SetLoaderSettings (LPDIRECTMUSICLOADER8 iface, REFGUID pClassID, WCHAR* wszSearchPath, LPBOOL pbCache);
27
28 static HRESULT DMUSIC_CopyDescriptor (LPDMUS_OBJECTDESC pDst, LPDMUS_OBJECTDESC pSrc) {
29         TRACE(": copy\n%s\n", debugstr_DMUS_OBJECTDESC(pSrc));
30         /* copy field by field */
31         if (pSrc->dwValidData & DMUS_OBJ_CLASS) pDst->guidClass = pSrc->guidClass;
32         if (pSrc->dwValidData & DMUS_OBJ_OBJECT) pDst->guidObject = pSrc->guidObject;
33         if (pSrc->dwValidData & DMUS_OBJ_DATE) pDst->ftDate = pSrc->ftDate;
34         if (pSrc->dwValidData & DMUS_OBJ_VERSION) pDst->vVersion = pSrc->vVersion;
35         if (pSrc->dwValidData & DMUS_OBJ_NAME) strcpyW (pDst->wszName, pSrc->wszName);
36         if (pSrc->dwValidData & DMUS_OBJ_CATEGORY) strcpyW (pDst->wszCategory, pSrc->wszCategory);
37         if (pSrc->dwValidData & DMUS_OBJ_FILENAME) strcpyW (pDst->wszFileName, pSrc->wszFileName);
38         if (pSrc->dwValidData & DMUS_OBJ_STREAM) IStream_Clone (pSrc->pStream, &pDst->pStream);
39         if (pSrc->dwValidData & DMUS_OBJ_MEMORY) {
40                 pDst->pbMemData = pSrc->pbMemData;
41                 pDst->llMemLength = pSrc->llMemLength;
42         }
43         /* set flags */
44         pDst->dwValidData |= pSrc->dwValidData;
45         return S_OK;
46 }
47
48
49 static BOOL DMUSIC_IsValidLoadableClass (REFCLSID pClassID) {
50         if (IsEqualCLSID(pClassID, &CLSID_DirectMusicAudioPathConfig) ||
51                 IsEqualCLSID(pClassID, &CLSID_DirectMusicBand) ||
52                 IsEqualCLSID(pClassID, &CLSID_DirectMusicContainer) ||
53                 IsEqualCLSID(pClassID, &CLSID_DirectMusicCollection) ||
54                 IsEqualCLSID(pClassID, &CLSID_DirectMusicChordMap) ||
55                 IsEqualCLSID(pClassID, &CLSID_DirectMusicSegment) ||
56                 IsEqualCLSID(pClassID, &CLSID_DirectMusicScript) ||
57                 IsEqualCLSID(pClassID, &CLSID_DirectMusicSong) ||
58                 IsEqualCLSID(pClassID, &CLSID_DirectMusicStyle) ||
59                 IsEqualCLSID(pClassID, &CLSID_DirectMusicGraph) ||
60                 IsEqualCLSID(pClassID, &CLSID_DirectSoundWave) ||
61                 IsEqualCLSID(pClassID, &GUID_DirectMusicAllTypes))
62                 return TRUE;
63         else
64                 return FALSE;
65 }
66
67 /*****************************************************************************
68  * IDirectMusicLoaderImpl implementation
69  */
70 /* IUnknown/IDirectMusicLoader(8) part: */
71
72 static HRESULT WINAPI IDirectMusicLoaderImpl_IDirectMusicLoader_QueryInterface (LPDIRECTMUSICLOADER8 iface, REFIID riid, LPVOID *ppobj) {
73         ICOM_THIS_MULTI(IDirectMusicLoaderImpl, LoaderVtbl, iface);
74
75         TRACE("(%p, %s, %p)\n",This, debugstr_dmguid(riid), ppobj);
76         if (IsEqualIID (riid, &IID_IUnknown) || 
77             IsEqualIID (riid, &IID_IDirectMusicLoader) ||
78             IsEqualIID (riid, &IID_IDirectMusicLoader8)) {
79                 IDirectMusicLoader_AddRef (iface);
80                 *ppobj = This;
81                 return S_OK;
82         }
83         
84         WARN(": not found\n");
85         return E_NOINTERFACE;
86 }
87
88 static ULONG WINAPI IDirectMusicLoaderImpl_IDirectMusicLoader_AddRef (LPDIRECTMUSICLOADER8 iface) {
89         ICOM_THIS_MULTI(IDirectMusicLoaderImpl, LoaderVtbl, iface);
90         TRACE("(%p): AddRef from %d\n", This, This->dwRef);
91         return InterlockedIncrement (&This->dwRef);
92 }
93
94 static ULONG WINAPI IDirectMusicLoaderImpl_IDirectMusicLoader_Release (LPDIRECTMUSICLOADER8 iface) {
95         ICOM_THIS_MULTI(IDirectMusicLoaderImpl, LoaderVtbl, iface);
96         
97         DWORD dwRef = InterlockedDecrement (&This->dwRef);
98         TRACE("(%p): ReleaseRef to %d\n", This, This->dwRef);
99         if (dwRef == 0) {
100             /* firstly, release the cache */
101             IDirectMusicLoader8_ClearCache (iface, &GUID_DirectMusicAllTypes);
102             /* FIXME: release all allocated entries */
103             /* destroy critical section */
104             /*This->CritSect.DebugInfo->Spare[0] = 0;
105               DeleteCriticalSection (&This->CritSect); */
106             HeapFree (GetProcessHeap(), 0, This);
107
108             /* decrease number of instances */
109             InterlockedDecrement (&dwDirectMusicLoader);
110         }
111         
112         return dwRef;
113 }
114
115 static HRESULT WINAPI IDirectMusicLoaderImpl_IDirectMusicLoader_GetObject (LPDIRECTMUSICLOADER8 iface, LPDMUS_OBJECTDESC pDesc, REFIID riid, LPVOID* ppv) {
116         ICOM_THIS_MULTI(IDirectMusicLoaderImpl, LoaderVtbl, iface);
117         HRESULT result = S_OK;
118         HRESULT ret = S_OK; /* used at the end of function, to determine whether everything went OK */
119         
120         struct list *pEntry;
121         LPWINE_LOADER_ENTRY pObjectEntry = NULL;
122         LPSTREAM pStream;
123         IPersistStream* pPersistStream = NULL;
124
125         LPDIRECTMUSICOBJECT pObject;
126         DMUS_OBJECTDESC GotDesc;
127         BOOL bCache;
128
129         TRACE("(%p, %p, %s, %p): pDesc:\n%s\n", This, pDesc, debugstr_dmguid(riid), ppv, debugstr_DMUS_OBJECTDESC(pDesc));
130         
131         /* sometimes it happens that guidClass is missing... which is a BadThingTM */
132         if (!(pDesc->dwValidData & DMUS_OBJ_CLASS)) {
133           ERR(": guidClass not valid but needed\n");
134           *ppv = NULL;
135           return DMUS_E_LOADER_NOCLASSID;
136         }
137         
138         /* OK, first we iterate thru the list of objects we know about; these are either loaded (GetObject, LoadObjectFromFile)
139            or set via SetObject; */
140         TRACE(": looking if we have object in the cache or if it can be found via alias\n");
141         LIST_FOR_EACH(pEntry, This->pObjects) {
142                 LPWINE_LOADER_ENTRY pExistingEntry = LIST_ENTRY(pEntry, WINE_LOADER_ENTRY, entry);
143                 if ((pDesc->dwValidData & DMUS_OBJ_OBJECT) &&
144                         (pExistingEntry->Desc.dwValidData & DMUS_OBJ_OBJECT) &&
145                         IsEqualGUID (&pDesc->guidObject, &pExistingEntry->Desc.guidObject)) {
146                         TRACE(": found it by object GUID\n");
147                         /* I suppose such stuff can happen only when GUID for object is given (GUID_DefaultGMCollection) */
148                         if (pExistingEntry->bInvalidDefaultDLS) {
149                                 TRACE(": found faulty default DLS collection... enabling M$ compliant behaviour\n");
150                                 return DMUS_E_LOADER_NOFILENAME;        
151                         }
152                         if (pExistingEntry->Desc.dwValidData & DMUS_OBJ_LOADED) {
153                                 TRACE(": already loaded\n");
154                                 return IDirectMusicObject_QueryInterface (pExistingEntry->pObject, riid, ppv);
155                         } else {
156                                 TRACE(": not loaded yet\n");
157                                 pObjectEntry = pExistingEntry;
158                         }
159                 }
160                 else if ((pDesc->dwValidData & (DMUS_OBJ_FILENAME | DMUS_OBJ_FULLPATH)) &&
161                                 (pExistingEntry->Desc.dwValidData & (DMUS_OBJ_FILENAME | DMUS_OBJ_FULLPATH)) &&
162                                 !strncmpW (pDesc->wszFileName, pExistingEntry->Desc.wszFileName, DMUS_MAX_FILENAME)) {
163                         TRACE(": found it by fullpath filename\n");
164                         if (pExistingEntry->Desc.dwValidData & DMUS_OBJ_LOADED) {
165                                 TRACE(": already loaded\n");
166                                 return IDirectMusicObject_QueryInterface (pExistingEntry->pObject, riid, ppv);
167                         } else {
168                                 TRACE(": not loaded yet\n");
169                                 pObjectEntry = pExistingEntry;
170                         }
171                 }
172                 else if ((pDesc->dwValidData & (DMUS_OBJ_NAME | DMUS_OBJ_CATEGORY)) &&
173                                 (pExistingEntry->Desc.dwValidData & (DMUS_OBJ_NAME | DMUS_OBJ_CATEGORY)) &&
174                                 !strncmpW (pDesc->wszName, pExistingEntry->Desc.wszName, DMUS_MAX_NAME) &&
175                                 !strncmpW (pDesc->wszCategory, pExistingEntry->Desc.wszCategory, DMUS_MAX_CATEGORY)) {
176                         TRACE(": found it by name and category\n");
177                         if (pExistingEntry->Desc.dwValidData & DMUS_OBJ_LOADED) {
178                                 TRACE(": already loaded\n");
179                                 return IDirectMusicObject_QueryInterface (pExistingEntry->pObject, riid, ppv);
180                         } else {
181                                 TRACE(": not loaded yet\n");
182                                 pObjectEntry = pExistingEntry;
183                         }
184                 }
185                 else if ((pDesc->dwValidData & DMUS_OBJ_NAME) &&
186                                 (pExistingEntry->Desc.dwValidData & DMUS_OBJ_NAME) &&
187                                 !strncmpW (pDesc->wszName, pExistingEntry->Desc.wszName, DMUS_MAX_NAME)) {
188                         TRACE(": found it by name\n");
189                         if (pExistingEntry->Desc.dwValidData & DMUS_OBJ_LOADED) {
190                                 TRACE(": already loaded\n");
191                                 return IDirectMusicObject_QueryInterface (pExistingEntry->pObject, riid, ppv);
192                         } else {
193                                 TRACE(": not loaded yet\n");
194                                 pObjectEntry = pExistingEntry;
195                         }
196                 }
197                 else if ((pDesc->dwValidData & DMUS_OBJ_FILENAME) &&
198                                 (pExistingEntry->Desc.dwValidData & DMUS_OBJ_FILENAME) &&
199                                 !strncmpW (pDesc->wszFileName, pExistingEntry->Desc.wszFileName, DMUS_MAX_FILENAME)) {
200                         TRACE(": found it by filename\n");                              
201                         if (pExistingEntry->Desc.dwValidData & DMUS_OBJ_LOADED) {
202                                 TRACE(": already loaded\n");
203                                 return IDirectMusicObject_QueryInterface (pExistingEntry->pObject, riid, ppv);
204                         } else {
205                                 TRACE(": not loaded yet\n");
206                                 pObjectEntry = pExistingEntry;
207                         }
208                 }
209         }
210         
211         /* basically, if we found alias, we use its descriptor to load...
212            else we use info we were given */
213         if (pObjectEntry) {
214                 TRACE(": found alias entry for requested object... using stored info\n");
215                 /* I think in certain cases it can happen that entry's descriptor lacks info about
216                    where to load from (e.g.: if we loaded from stream and then released object
217                    from cache; then only its CLSID, GUID and perhaps name are left); so just
218                    overwrite whatever info the entry has (since it ought to be 100% correct) */
219                 DMUSIC_CopyDescriptor (pDesc, &pObjectEntry->Desc);
220                 /*pDesc = &pObjectEntry->Desc; */ /* FIXME: is this OK? */
221         } else {
222                 TRACE(": no cache/alias entry found for requested object\n");
223         }
224         
225         if (pDesc->dwValidData & DMUS_OBJ_URL) {
226                 TRACE(": loading from URLs not supported yet\n");
227                 return DMUS_E_LOADER_FORMATNOTSUPPORTED;
228         }
229         else if (pDesc->dwValidData & DMUS_OBJ_FILENAME) {
230                 /* load object from file */
231                 /* generate filename; if it's full path, don't add search 
232                    directory path, otherwise do */
233                 WCHAR wszFileName[MAX_PATH];
234
235                 if (pDesc->dwValidData & DMUS_OBJ_FULLPATH) {
236                         lstrcpyW(wszFileName, pDesc->wszFileName);
237                 } else {
238                         WCHAR *p, wszSearchPath[MAX_PATH];
239                         DMUSIC_GetLoaderSettings (iface, &pDesc->guidClass, wszSearchPath, NULL);
240                         lstrcpyW(wszFileName, wszSearchPath);
241                         p = wszFileName + lstrlenW(wszFileName);
242                         if (p > wszFileName && p[-1] != '\\') *p++ = '\\';
243                         strcpyW(p, pDesc->wszFileName);
244                 }
245                 TRACE(": loading from file (%s)\n", debugstr_w(wszFileName));
246                 /* create stream and associate it with file */                  
247                 result = DMUSIC_CreateDirectMusicLoaderFileStream ((LPVOID*)&pStream);
248                 if (FAILED(result)) {
249                         ERR(": could not create file stream\n");
250                         return result;
251                 }
252                 result = IDirectMusicLoaderFileStream_Attach (pStream, wszFileName, iface);
253                 if (FAILED(result)) {
254                         ERR(": could not attach stream to file\n");
255                         IStream_Release (pStream);
256                         return result;
257                 }
258         }
259         else if (pDesc->dwValidData & DMUS_OBJ_MEMORY) {
260                 /* load object from resource */
261                 TRACE(": loading from resource\n");
262                 /* create stream and associate it with given resource */                        
263                 result = DMUSIC_CreateDirectMusicLoaderResourceStream ((LPVOID*)&pStream);
264                 if (FAILED(result)) {
265                         ERR(": could not create resource stream\n");
266                         return result;
267                 }
268                 result = IDirectMusicLoaderResourceStream_Attach (pStream, pDesc->pbMemData, pDesc->llMemLength, 0, iface);
269                 if (FAILED(result)) {
270                         ERR(": could not attach stream to resource\n");
271                         IStream_Release (pStream);
272                         return result;
273                 }
274         }
275         else if (pDesc->dwValidData & DMUS_OBJ_STREAM) {
276                 /* load object from stream */
277                 TRACE(": loading from stream\n");
278                 /* create universal stream and associate it with given one */                   
279                 result = DMUSIC_CreateDirectMusicLoaderGenericStream ((LPVOID*)&pStream);
280                 if (FAILED(result)) {
281                         ERR(": could not create generic stream\n");
282                         return result;
283                 }
284                 result = IDirectMusicLoaderGenericStream_Attach (pStream, pDesc->pStream, iface);
285                 if (FAILED(result)) {
286                         ERR(": failed to attach stream\n");
287                         IStream_Release (pStream);
288                         return result;
289                 }
290         } else {
291                 /* nowhere to load from */
292                 FIXME(": unknown/unsupported way of loading\n");
293                 return DMUS_E_LOADER_NOFILENAME; /* test shows this is returned */
294         }
295
296         /* create object */
297         result = CoCreateInstance (&pDesc->guidClass, NULL, CLSCTX_INPROC_SERVER, &IID_IDirectMusicObject, (LPVOID*)&pObject);
298         if (FAILED(result)) {
299                 ERR(": could not create object\n");
300                 return result;
301         }
302         /* acquire PersistStream interface */
303         result = IDirectMusicObject_QueryInterface (pObject, &IID_IPersistStream, (LPVOID*)&pPersistStream);
304         if (FAILED(result)) {
305                 ERR("failed to Query\n");
306                 return result;
307         }
308         /* load */
309         result = IPersistStream_Load (pPersistStream, pStream);
310         if (result != S_OK) {
311                 WARN(": failed to (completely) load object (%s)\n", debugstr_dmreturn(result));
312                 ret = DMUS_S_PARTIALLOAD /*result*/;
313         }
314         /* get descriptor */
315         DM_STRUCT_INIT(&GotDesc);
316         result = IDirectMusicObject_GetDescriptor (pObject, &GotDesc);
317         /* set filename (if we loaded via filename) */
318         if (pDesc->dwValidData & DMUS_OBJ_FILENAME) {
319                 GotDesc.dwValidData |= (pDesc->dwValidData & (DMUS_OBJ_FILENAME | DMUS_OBJ_FULLPATH));
320                 strcpyW (GotDesc.wszFileName, pDesc->wszFileName);
321         }
322         if (FAILED(result)) {
323                 ERR(": failed to get descriptor\n");
324                 return result;
325         }
326         /* release all loading related stuff */
327         IStream_Release (pStream);
328         IPersistStream_Release (pPersistStream);        
329                 
330         /* add object to cache/overwrite existing info (if cache is enabled) */
331         DMUSIC_GetLoaderSettings (iface, &pDesc->guidClass, NULL, &bCache);
332         if (bCache) {
333                 if (!pObjectEntry) {
334                         pObjectEntry = HeapAlloc (GetProcessHeap (), HEAP_ZERO_MEMORY, sizeof(WINE_LOADER_ENTRY));
335                         DM_STRUCT_INIT(&pObjectEntry->Desc);
336                         DMUSIC_CopyDescriptor (&pObjectEntry->Desc, &GotDesc);
337                         pObjectEntry->pObject = pObject;
338                         pObjectEntry->bInvalidDefaultDLS = FALSE;
339                         list_add_head (This->pObjects, &pObjectEntry->entry);
340                 } else {
341                         DMUSIC_CopyDescriptor (&pObjectEntry->Desc, &GotDesc);
342                         pObjectEntry->pObject = pObject;
343                         pObjectEntry->bInvalidDefaultDLS = FALSE;
344                 }
345                 TRACE(": filled in cache entry\n");
346         } else TRACE(": caching disabled\n");
347
348 #if 0
349         /* for debug purposes (e.g. to check if all files are cached) */
350         TRACE("*** Loader's cache ***\n");
351         int i = 0;
352         LIST_FOR_EACH (pEntry, This->pObjects) {
353                 i++;
354                 pObjectEntry = LIST_ENTRY(pEntry, WINE_LOADER_ENTRY, entry);
355                 TRACE(": entry nr. %i:\n%s\n  - bInvalidDefaultDLS = %i\n  - pObject = %p\n", i, debugstr_DMUS_OBJECTDESC(&pObjectEntry->Desc), pObjectEntry->bInvalidDefaultDLS, pObjectEntry->pObject);
356         }
357 #endif
358         
359         result = IDirectMusicObject_QueryInterface (pObject, riid, ppv);
360         if (!bCache) IDirectMusicObject_Release (pObject); /* since loader's reference is not needed */
361         /* if there was trouble with loading, and if no other error occurred,
362            we should return DMUS_S_PARTIALLOAD; else, error is returned */
363         if (result == S_OK)
364                 return ret;
365         else
366                 return result;
367 }
368
369 static HRESULT WINAPI IDirectMusicLoaderImpl_IDirectMusicLoader_SetObject (LPDIRECTMUSICLOADER8 iface, LPDMUS_OBJECTDESC pDesc) {
370         ICOM_THIS_MULTI(IDirectMusicLoaderImpl, LoaderVtbl, iface);
371         LPSTREAM pStream;
372         LPDIRECTMUSICOBJECT pObject;
373         DMUS_OBJECTDESC Desc;
374         struct list *pEntry;
375         LPWINE_LOADER_ENTRY pObjectEntry, pNewEntry;
376         HRESULT hr;
377
378         TRACE("(%p, %p): pDesc:\n%s\n", This, pDesc, debugstr_DMUS_OBJECTDESC(pDesc));
379
380         /* create stream and load additional info from it */
381         if (pDesc->dwValidData & DMUS_OBJ_FILENAME) {
382                 /* generate filename; if it's full path, don't add search 
383                    directory path, otherwise do */
384                 WCHAR wszFileName[MAX_PATH];
385
386                 if (pDesc->dwValidData & DMUS_OBJ_FULLPATH) {
387                         lstrcpyW(wszFileName, pDesc->wszFileName);
388                 } else {
389                         WCHAR *p;
390                         WCHAR wszSearchPath[MAX_PATH];
391                         DMUSIC_GetLoaderSettings (iface, &pDesc->guidClass, wszSearchPath, NULL);
392                         lstrcpyW(wszFileName, wszSearchPath);
393                         p = wszFileName + lstrlenW(wszFileName);
394                         if (p > wszFileName && p[-1] != '\\') *p++ = '\\';
395                         strcpyW(p, pDesc->wszFileName);
396                 }
397                 /* create stream */
398                 hr = DMUSIC_CreateDirectMusicLoaderFileStream ((LPVOID*)&pStream);
399                 if (FAILED(hr)) {
400                         ERR(": could not create file stream\n");
401                         return DMUS_E_LOADER_FAILEDOPEN;
402                 }
403                 /* attach stream */
404                 hr = IDirectMusicLoaderFileStream_Attach (pStream, wszFileName, iface);
405                 if (FAILED(hr)) {
406                         ERR(": could not attach stream to file\n");
407                         IStream_Release (pStream);
408                         return DMUS_E_LOADER_FAILEDOPEN;
409                 }
410         }
411         else if (pDesc->dwValidData & DMUS_OBJ_STREAM) {        
412                 /* create stream */
413                 hr = DMUSIC_CreateDirectMusicLoaderGenericStream ((LPVOID*)&pStream);
414                 if (FAILED(hr)) {
415                         ERR(": could not create generic stream\n");
416                         return DMUS_E_LOADER_FAILEDOPEN;
417                 }
418                 /* attach stream */
419                 hr = IDirectMusicLoaderGenericStream_Attach (pStream, pDesc->pStream, iface);
420                 if (FAILED(hr)) {
421                         ERR(": could not attach stream\n");
422                         IStream_Release (pStream);
423                         return DMUS_E_LOADER_FAILEDOPEN;
424                 }
425         }
426         else if (pDesc->dwValidData & DMUS_OBJ_MEMORY) {
427                 /* create stream */
428                 hr = DMUSIC_CreateDirectMusicLoaderResourceStream ((LPVOID*)&pStream);
429                 if (FAILED(hr)) {
430                         ERR(": could not create resource stream\n");
431                         return DMUS_E_LOADER_FAILEDOPEN;
432                 }
433                 /* attach stream */
434                 hr = IDirectMusicLoaderResourceStream_Attach (pStream, pDesc->pbMemData, pDesc->llMemLength, 0, iface);
435                 if (FAILED(hr)) {
436                         ERR(": could not attach stream to resource\n");
437                         IStream_Release (pStream);
438                         return DMUS_E_LOADER_FAILEDOPEN;
439                 }
440         }
441         else {
442                 ERR(": no way to get additional info\n");
443                 return DMUS_E_LOADER_FAILEDOPEN;
444         }
445
446         /* create object */
447         CoCreateInstance (&pDesc->guidClass, NULL, CLSCTX_INPROC_SERVER, &IID_IDirectMusicObject, (LPVOID*)&pObject);
448
449         /* *sigh*... some ms objects have lousy implementation of ParseDescriptor that clears input descriptor :( */
450 #ifdef NOW_WE_ARE_FREE
451         /* parse descriptor: we actually use input descriptor; fields that aren't available from stream remain,
452            otherwise real info is set */
453          IDirectMusicObject_ParseDescriptor (pObject, pStream, pDesc);
454 #endif
455         /* hmph... due to some trouble I had with certain tests, we store current position and then set it back */
456         DM_STRUCT_INIT(&Desc);
457         if (FAILED(IDirectMusicObject_ParseDescriptor (pObject, pStream, &Desc))) {
458                 ERR(": couldn't parse descriptor\n");
459                 return DMUS_E_LOADER_FORMATNOTSUPPORTED;
460         }
461
462         /* copy elements from parsed descriptor into input descriptor; this sets new info, overwriting if necessary,
463            but leaves info that's provided by input and not available from stream */    
464         DMUSIC_CopyDescriptor (pDesc, &Desc);
465         
466         /* release everything */
467         IDirectMusicObject_Release (pObject);
468         IStream_Release (pStream);      
469         
470         /* sometimes it happens that twisted programs call SetObject for same object twice...
471            in such cases, native loader returns S_OK and does nothing... a sound plan */
472         LIST_FOR_EACH (pEntry, This->pObjects) {
473                 pObjectEntry = LIST_ENTRY (pEntry, WINE_LOADER_ENTRY, entry);
474                 if (!memcmp (&pObjectEntry->Desc, pDesc, sizeof(DMUS_OBJECTDESC))) {
475                         TRACE(": exactly same entry already exists\n");
476                         return S_OK;
477                 }
478         }               
479         
480         /* add new entry */
481         TRACE(": adding alias entry with following info:\n%s\n", debugstr_DMUS_OBJECTDESC(pDesc));
482         pNewEntry = HeapAlloc (GetProcessHeap (), HEAP_ZERO_MEMORY, sizeof(WINE_LOADER_ENTRY));
483         /* use this function instead of pure memcpy due to streams (memcpy just copies pointer), 
484            which is basically used further by app that called SetDescriptor... better safety than exception */
485         DMUSIC_CopyDescriptor (&pNewEntry->Desc, pDesc);
486         list_add_head (This->pObjects, &pNewEntry->entry);
487
488         return S_OK;
489 }
490
491 static HRESULT WINAPI IDirectMusicLoaderImpl_IDirectMusicLoader_SetSearchDirectory (LPDIRECTMUSICLOADER8 iface, REFGUID rguidClass, WCHAR* pwzPath, BOOL fClear) {
492         WCHAR wszCurrentPath[MAX_PATH];
493         ICOM_THIS_MULTI(IDirectMusicLoaderImpl, LoaderVtbl, iface);
494         TRACE("(%p, %s, %s, %d)\n", This, debugstr_dmguid(rguidClass), debugstr_w(pwzPath), fClear);
495         FIXME(": fClear ignored\n");
496         DMUSIC_GetLoaderSettings (iface, rguidClass, wszCurrentPath, NULL);
497         if (!strncmpW(wszCurrentPath, pwzPath, MAX_PATH)) {
498           return S_FALSE;
499         }
500         /* FIXME: check if path is valid; else return DMUS_E_LOADER_BADPATH */
501         return DMUSIC_SetLoaderSettings (iface, rguidClass, pwzPath, NULL);
502 }
503
504 static HRESULT WINAPI IDirectMusicLoaderImpl_IDirectMusicLoader_ScanDirectory (LPDIRECTMUSICLOADER8 iface, REFGUID rguidClass, WCHAR* pwzFileExtension, WCHAR* pwzScanFileName) {
505         static const WCHAR wszAny[] = {'*',0};
506         WIN32_FIND_DATAW FileData;
507         HANDLE hSearch;
508         WCHAR wszSearchString[MAX_PATH];
509         WCHAR *p;
510         HRESULT result;
511         ICOM_THIS_MULTI(IDirectMusicLoaderImpl, LoaderVtbl, iface);
512         TRACE("(%p, %s, %p, %p)\n", This, debugstr_dmguid(rguidClass), pwzFileExtension, pwzScanFileName);
513         if (IsEqualGUID (rguidClass, &GUID_DirectMusicAllTypes) || !DMUSIC_IsValidLoadableClass(rguidClass)) {
514                 ERR(": rguidClass invalid CLSID\n");
515                 return REGDB_E_CLASSNOTREG;
516         }
517         
518         /* get search path for given class */
519         DMUSIC_GetLoaderSettings (iface, rguidClass, wszSearchString, NULL);
520         
521         p = wszSearchString + lstrlenW(wszSearchString);
522         if (p > wszSearchString && p[-1] != '\\') *p++ = '\\';
523         *p++ = '*'; /* any file */
524         if (strcmpW (pwzFileExtension, wszAny)) *p++ = '.'; /* if we have actual extension, put a dot */
525         strcpyW (p, pwzFileExtension);
526         
527         TRACE(": search string: %s\n", debugstr_w(wszSearchString));
528         
529         hSearch = FindFirstFileW (wszSearchString, &FileData);
530         if (hSearch == INVALID_HANDLE_VALUE) {
531                 TRACE(": no files found\n");
532                 return S_FALSE;
533         }
534         
535         do {
536                 DMUS_OBJECTDESC Desc;
537                 DM_STRUCT_INIT(&Desc);
538                 Desc.dwValidData = DMUS_OBJ_CLASS | DMUS_OBJ_FILENAME | DMUS_OBJ_FULLPATH | DMUS_OBJ_DATE;
539                 Desc.guidClass = *rguidClass;
540                 strcpyW (Desc.wszFileName, FileData.cFileName);
541                 FileTimeToLocalFileTime (&FileData.ftCreationTime, &Desc.ftDate);
542                 IDirectMusicLoader8_SetObject (iface, &Desc);
543                 
544                 if (!FindNextFileW (hSearch, &FileData)) {
545                         if (GetLastError () == ERROR_NO_MORE_FILES) {
546                                 TRACE(": search completed\n");
547                                 result = S_OK;
548                         } else {
549                                 ERR(": could not get next file\n");
550                                 result = E_FAIL;
551                         }
552                         FindClose (hSearch);
553                         return result;
554                 }
555         } while (1);
556 }
557
558 static HRESULT WINAPI IDirectMusicLoaderImpl_IDirectMusicLoader_CacheObject (LPDIRECTMUSICLOADER8 iface, IDirectMusicObject* pObject) {
559         DMUS_OBJECTDESC Desc;
560         HRESULT result = DMUS_E_LOADER_OBJECTNOTFOUND;
561         struct list *pEntry;
562         LPWINE_LOADER_ENTRY  pObjectEntry = NULL;
563
564         ICOM_THIS_MULTI(IDirectMusicLoaderImpl, LoaderVtbl, iface);
565         TRACE("(%p, %p)\n", This, pObject);
566         
567         /* get descriptor */
568         DM_STRUCT_INIT(&Desc);
569         IDirectMusicObject_GetDescriptor (pObject, &Desc);
570         
571         /* now iterate thru list and check if we have alias (without object), corresponding
572            to descriptor of input object */
573         LIST_FOR_EACH(pEntry, This->pObjects) {
574                 pObjectEntry = LIST_ENTRY(pEntry, WINE_LOADER_ENTRY, entry);
575                 if ((Desc.dwValidData & DMUS_OBJ_OBJECT) &&
576                         (pObjectEntry->Desc.dwValidData & DMUS_OBJ_OBJECT) &&
577                         IsEqualGUID (&Desc.guidObject, &pObjectEntry->Desc.guidObject)) {
578                         TRACE(": found it by object GUID\n");
579                         if ((pObjectEntry->Desc.dwValidData & DMUS_OBJ_LOADED) && pObjectEntry->pObject)
580                                 result = S_FALSE;
581                         else
582                                 result = S_OK;
583                         break;
584                 }
585                 else if ((Desc.dwValidData & (DMUS_OBJ_FILENAME | DMUS_OBJ_FULLPATH)) &&
586                                 (pObjectEntry->Desc.dwValidData & (DMUS_OBJ_FILENAME | DMUS_OBJ_FULLPATH)) &&
587                                 !strncmpW (Desc.wszFileName, pObjectEntry->Desc.wszFileName, DMUS_MAX_FILENAME)) {
588                         TRACE(": found it by fullpath filename\n");
589                         if ((pObjectEntry->Desc.dwValidData & DMUS_OBJ_LOADED) && pObjectEntry->pObject)
590                                 result = S_FALSE;
591                         else
592                                 result = S_OK;
593                         break;
594                 }
595                 else if ((Desc.dwValidData & (DMUS_OBJ_NAME | DMUS_OBJ_CATEGORY)) &&
596                                 (pObjectEntry->Desc.dwValidData & (DMUS_OBJ_NAME | DMUS_OBJ_CATEGORY)) &&
597                                 !strncmpW (Desc.wszName, pObjectEntry->Desc.wszName, DMUS_MAX_NAME) &&
598                                 !strncmpW (Desc.wszCategory, pObjectEntry->Desc.wszCategory, DMUS_MAX_CATEGORY)) {
599                         TRACE(": found it by name and category\n");
600                         if ((pObjectEntry->Desc.dwValidData & DMUS_OBJ_LOADED) && pObjectEntry->pObject)
601                                 result = S_FALSE;
602                         else
603                                 result = S_OK;
604                         break;
605                 }
606                 else if ((Desc.dwValidData & DMUS_OBJ_NAME) &&
607                                 (pObjectEntry->Desc.dwValidData & DMUS_OBJ_NAME) &&
608                                 !strncmpW (Desc.wszName, pObjectEntry->Desc.wszName, DMUS_MAX_NAME)) {
609                         TRACE(": found it by name\n");
610                         if ((pObjectEntry->Desc.dwValidData & DMUS_OBJ_LOADED) && pObjectEntry->pObject)
611                                 result = S_FALSE;
612                         else
613                                 result = S_OK;
614                         break;
615                 }
616                 else if ((Desc.dwValidData & DMUS_OBJ_FILENAME) &&
617                                 (pObjectEntry->Desc.dwValidData & DMUS_OBJ_FILENAME) &&
618                                 !strncmpW (Desc.wszFileName, pObjectEntry->Desc.wszFileName, DMUS_MAX_FILENAME)) {
619                         TRACE(": found it by filename\n");                              
620                         if ((pObjectEntry->Desc.dwValidData & DMUS_OBJ_LOADED) && pObjectEntry->pObject)
621                                 result = S_FALSE;
622                         else
623                                 result = S_OK;
624                         break;
625                 }
626         }
627         
628         /* if we found such alias, then set everything */
629         if (result == S_OK) {
630                 pObjectEntry->Desc.dwValidData &= DMUS_OBJ_LOADED;
631                 pObjectEntry->pObject = pObject;
632                 IDirectMusicObject_AddRef (pObjectEntry->pObject);
633         }
634         
635         return result;
636 }
637
638 static HRESULT WINAPI IDirectMusicLoaderImpl_IDirectMusicLoader_ReleaseObject (LPDIRECTMUSICLOADER8 iface, IDirectMusicObject* pObject) {
639         DMUS_OBJECTDESC Desc;
640         struct list *pEntry;
641         LPWINE_LOADER_ENTRY pObjectEntry = NULL;
642         HRESULT result = S_FALSE;
643
644         ICOM_THIS_MULTI(IDirectMusicLoaderImpl, LoaderVtbl, iface);
645         TRACE("(%p, %p)\n", This, pObject);
646         
647         if(!pObject) return E_POINTER;
648
649         /* get descriptor */
650         DM_STRUCT_INIT(&Desc);
651         IDirectMusicObject_GetDescriptor (pObject, &Desc);
652         
653         /* iterate thru the list of objects we know about; check only those with DMUS_OBJ_LOADED */
654         TRACE(": looking for the object in cache\n");
655         LIST_FOR_EACH(pEntry, This->pObjects) {
656                 pObjectEntry = LIST_ENTRY(pEntry, WINE_LOADER_ENTRY, entry);
657                 if ((Desc.dwValidData & DMUS_OBJ_OBJECT) &&
658                         (pObjectEntry->Desc.dwValidData & (DMUS_OBJ_OBJECT | DMUS_OBJ_LOADED)) &&
659                         IsEqualGUID (&Desc.guidObject, &pObjectEntry->Desc.guidObject)) {
660                         TRACE(": found it by object GUID\n%s\n", debugstr_DMUS_OBJECTDESC(&pObjectEntry->Desc));
661                         result = S_OK;
662                         break;
663                 }
664                 else if ((Desc.dwValidData & (DMUS_OBJ_FILENAME | DMUS_OBJ_FULLPATH)) &&
665                                 (pObjectEntry->Desc.dwValidData & (DMUS_OBJ_FILENAME | DMUS_OBJ_FULLPATH | DMUS_OBJ_LOADED)) &&
666                                 !strncmpW (Desc.wszFileName, pObjectEntry->Desc.wszFileName, DMUS_MAX_FILENAME)) {
667                         TRACE(": found it by fullpath filename\n");
668                         result = S_OK;                  
669                         break;
670                 }
671                 else if ((Desc.dwValidData & (DMUS_OBJ_NAME | DMUS_OBJ_CATEGORY)) &&
672                                 (pObjectEntry->Desc.dwValidData & (DMUS_OBJ_NAME | DMUS_OBJ_CATEGORY | DMUS_OBJ_LOADED)) &&
673                                 !strncmpW (Desc.wszName, pObjectEntry->Desc.wszName, DMUS_MAX_NAME) &&
674                                 !strncmpW (Desc.wszCategory, pObjectEntry->Desc.wszCategory, DMUS_MAX_CATEGORY)) {
675                         TRACE(": found it by name and category\n");
676                         result = S_OK;                  
677                         break;
678                 }
679                 else if ((Desc.dwValidData & DMUS_OBJ_NAME) &&
680                                 (pObjectEntry->Desc.dwValidData & (DMUS_OBJ_NAME | DMUS_OBJ_LOADED)) &&
681                                 !strncmpW (Desc.wszName, pObjectEntry->Desc.wszName, DMUS_MAX_NAME)) {
682                         TRACE(": found it by name\n");
683                         result = S_OK;
684                         break;
685                 }
686                 else if ((Desc.dwValidData & DMUS_OBJ_FILENAME) &&
687                                 (pObjectEntry->Desc.dwValidData & (DMUS_OBJ_FILENAME | DMUS_OBJ_LOADED)) &&
688                                 !strncmpW (Desc.wszFileName, pObjectEntry->Desc.wszFileName, DMUS_MAX_FILENAME)) {
689                         TRACE(": found it by filename\n");
690                         result = S_OK;                  
691                         break;
692                 }
693         }
694         if (result == S_OK) {
695                 /*TRACE(": releasing:\n%s  - bInvalidDefaultDLS = %i\n  - pObject = %p\n", debugstr_DMUS_OBJECTDESC(&pObjectEntry->Desc), pObjectEntry->bInvalidDefaultDLS, pObjectEntry->pObject); */
696                 IDirectMusicObject_Release (pObjectEntry->pObject);
697                 pObjectEntry->pObject = NULL;
698                 pObjectEntry->Desc.dwValidData &= ~DMUS_OBJ_LOADED;
699         } 
700         return result;
701 }
702
703 static HRESULT WINAPI IDirectMusicLoaderImpl_IDirectMusicLoader_ClearCache (LPDIRECTMUSICLOADER8 iface, REFGUID rguidClass) {
704         struct list *pEntry;
705         LPWINE_LOADER_ENTRY pObjectEntry;
706         ICOM_THIS_MULTI(IDirectMusicLoaderImpl, LoaderVtbl, iface);
707         TRACE("(%p, %s)\n", This, debugstr_dmguid(rguidClass));
708         
709         LIST_FOR_EACH (pEntry, This->pObjects) {
710                 pObjectEntry = LIST_ENTRY (pEntry, WINE_LOADER_ENTRY, entry);
711                 
712                 if ((IsEqualGUID (rguidClass, &GUID_DirectMusicAllTypes) || IsEqualGUID (rguidClass, &pObjectEntry->Desc.guidClass)) &&
713                         (pObjectEntry->Desc.dwValidData & DMUS_OBJ_LOADED)) {
714                         /* basically, wrap to ReleaseObject for each object found */
715                         IDirectMusicLoader8_ReleaseObject (iface, pObjectEntry->pObject);
716                 }
717         }
718         
719         return S_OK;
720 }
721
722 static HRESULT WINAPI IDirectMusicLoaderImpl_IDirectMusicLoader_EnableCache (LPDIRECTMUSICLOADER8 iface, REFGUID rguidClass, BOOL fEnable) {
723         ICOM_THIS_MULTI(IDirectMusicLoaderImpl, LoaderVtbl, iface);
724         BOOL bCurrent;
725         TRACE("(%p, %s, %d)\n", This, debugstr_dmguid(rguidClass), fEnable);
726         DMUSIC_GetLoaderSettings (iface, rguidClass, NULL, &bCurrent);
727         if (bCurrent == fEnable)
728                 return S_FALSE;
729         else
730                 return DMUSIC_SetLoaderSettings (iface, rguidClass, NULL, &fEnable);
731 }
732
733 static HRESULT WINAPI IDirectMusicLoaderImpl_IDirectMusicLoader_EnumObject (LPDIRECTMUSICLOADER8 iface, REFGUID rguidClass, DWORD dwIndex, LPDMUS_OBJECTDESC pDesc) {
734         DWORD dwCount = 0;
735         struct list *pEntry;
736         LPWINE_LOADER_ENTRY pObjectEntry;
737         ICOM_THIS_MULTI(IDirectMusicLoaderImpl, LoaderVtbl, iface);
738         TRACE("(%p, %s, %d, %p)\n", This, debugstr_dmguid(rguidClass), dwIndex, pDesc);
739
740         DM_STRUCT_INIT(pDesc);
741
742         LIST_FOR_EACH (pEntry, This->pObjects) {
743                 pObjectEntry = LIST_ENTRY (pEntry, WINE_LOADER_ENTRY, entry);
744
745                 if (IsEqualGUID (rguidClass, &GUID_DirectMusicAllTypes) || IsEqualGUID (rguidClass, &pObjectEntry->Desc.guidClass)) {
746                         if (dwCount == dwIndex) {
747                                 *pDesc = pObjectEntry->Desc;
748                                 /* we aren't supposed to reveal this info */
749                                 pDesc->dwValidData &= ~(DMUS_OBJ_MEMORY | DMUS_OBJ_STREAM);
750                                 pDesc->pbMemData = NULL;
751                                 pDesc->llMemLength = 0;
752                                 pDesc->pStream = NULL;
753                                 return S_OK;
754                         }
755                         dwCount++;
756                 }
757         }
758         
759         TRACE(": not found\n"); 
760         return S_FALSE;
761 }
762
763 static void WINAPI IDirectMusicLoaderImpl_IDirectMusicLoader_CollectGarbage (LPDIRECTMUSICLOADER8 iface) {
764         ICOM_THIS_MULTI(IDirectMusicLoaderImpl, LoaderVtbl, iface);
765         FIXME("(%p): stub\n", This);
766 }
767
768 static HRESULT WINAPI IDirectMusicLoaderImpl_IDirectMusicLoader_ReleaseObjectByUnknown (LPDIRECTMUSICLOADER8 iface, IUnknown* pObject) {
769         ICOM_THIS_MULTI(IDirectMusicLoaderImpl, LoaderVtbl, iface);
770         HRESULT result;
771         LPDIRECTMUSICOBJECT pObjectInterface;
772         
773         TRACE("(%p, %p)\n", This, pObject);
774         
775         if (IsBadReadPtr (pObject, sizeof(LPUNKNOWN))) {
776                 ERR(": pObject bad write pointer\n");
777                 return E_POINTER;
778         }
779         /* we simply get IDirectMusicObject interface */
780         result = IUnknown_QueryInterface (pObject, &IID_IDirectMusicObject, (LPVOID*)&pObjectInterface);
781         if (FAILED(result)) return result;
782         /* and release it in old-fashioned way */
783         result = IDirectMusicLoader8_ReleaseObject (iface, pObjectInterface);
784         IDirectMusicObject_Release (pObjectInterface);
785         
786         return result;
787 }
788
789 static HRESULT WINAPI IDirectMusicLoaderImpl_IDirectMusicLoader_LoadObjectFromFile (LPDIRECTMUSICLOADER8 iface, REFGUID rguidClassID, REFIID iidInterfaceID, WCHAR* pwzFilePath, void** ppObject) {
790         ICOM_THIS_MULTI(IDirectMusicLoaderImpl, LoaderVtbl, iface);
791         DMUS_OBJECTDESC ObjDesc;
792         WCHAR wszLoaderSearchPath[MAX_PATH];
793
794         TRACE("(%p, %s, %s, %s, %p): wrapping to IDirectMusicLoaderImpl_GetObject\n", This, debugstr_dmguid(rguidClassID), debugstr_dmguid(iidInterfaceID), debugstr_w(pwzFilePath), ppObject);
795
796         DM_STRUCT_INIT(&ObjDesc);       
797         ObjDesc.dwValidData = DMUS_OBJ_FILENAME | DMUS_OBJ_FULLPATH | DMUS_OBJ_CLASS; /* I believe I've read somewhere in MSDN that this function requires either full path or relative path */
798         ObjDesc.guidClass = *rguidClassID;
799         /* OK, MSDN says that search order is the following:
800             - current directory (DONE)
801             - windows search path (FIXME: how do I get that?)
802             - loader's search path (DONE)
803         */
804         DMUSIC_GetLoaderSettings (iface, rguidClassID, wszLoaderSearchPath, NULL);
805     /* search in current directory */
806         if (!SearchPathW (NULL, pwzFilePath, NULL, sizeof(ObjDesc.wszFileName)/sizeof(WCHAR), ObjDesc.wszFileName, NULL) &&
807         /* search in loader's search path */
808                 !SearchPathW (wszLoaderSearchPath, pwzFilePath, NULL, sizeof(ObjDesc.wszFileName)/sizeof(WCHAR), ObjDesc.wszFileName, NULL)) {
809                 /* cannot find file */
810                 TRACE(": cannot find file\n");
811                 return DMUS_E_LOADER_FAILEDOPEN;
812         }
813         
814         TRACE(": full file path = %s\n", debugstr_w (ObjDesc.wszFileName));
815         
816         return IDirectMusicLoaderImpl_IDirectMusicLoader_GetObject (iface, &ObjDesc, iidInterfaceID, ppObject);
817 }
818
819 static const IDirectMusicLoader8Vtbl DirectMusicLoader_Loader_Vtbl = {
820         IDirectMusicLoaderImpl_IDirectMusicLoader_QueryInterface,
821         IDirectMusicLoaderImpl_IDirectMusicLoader_AddRef,
822         IDirectMusicLoaderImpl_IDirectMusicLoader_Release,
823         IDirectMusicLoaderImpl_IDirectMusicLoader_GetObject,
824         IDirectMusicLoaderImpl_IDirectMusicLoader_SetObject,
825         IDirectMusicLoaderImpl_IDirectMusicLoader_SetSearchDirectory,
826         IDirectMusicLoaderImpl_IDirectMusicLoader_ScanDirectory,
827         IDirectMusicLoaderImpl_IDirectMusicLoader_CacheObject,
828         IDirectMusicLoaderImpl_IDirectMusicLoader_ReleaseObject,
829         IDirectMusicLoaderImpl_IDirectMusicLoader_ClearCache,
830         IDirectMusicLoaderImpl_IDirectMusicLoader_EnableCache,
831         IDirectMusicLoaderImpl_IDirectMusicLoader_EnumObject,
832         IDirectMusicLoaderImpl_IDirectMusicLoader_CollectGarbage,
833         IDirectMusicLoaderImpl_IDirectMusicLoader_ReleaseObjectByUnknown,
834         IDirectMusicLoaderImpl_IDirectMusicLoader_LoadObjectFromFile
835 };
836
837 /* help function for DMUSIC_SetDefaultDLS */
838 static HRESULT DMUSIC_GetDefaultGMPath (WCHAR wszPath[MAX_PATH]) {
839         HKEY hkDM;
840         DWORD returnType, sizeOfReturnBuffer = MAX_PATH;
841         char szPath[MAX_PATH];
842
843         if ((RegOpenKeyExA (HKEY_LOCAL_MACHINE, "Software\\Microsoft\\DirectMusic" , 0, KEY_READ, &hkDM) != ERROR_SUCCESS) ||
844             (RegQueryValueExA (hkDM, "GMFilePath", NULL, &returnType, (LPBYTE) szPath, &sizeOfReturnBuffer) != ERROR_SUCCESS)) {
845                 WARN(": registry entry missing\n" );
846                 return E_FAIL;
847         }
848         /* FIXME: Check return types to ensure we're interpreting data right */
849         MultiByteToWideChar (CP_ACP, 0, szPath, -1, wszPath, MAX_PATH);
850
851         return S_OK;
852 }
853
854 /* for ClassFactory */
855 HRESULT WINAPI DMUSIC_CreateDirectMusicLoaderImpl (LPCGUID lpcGUID, LPVOID *ppobj, LPUNKNOWN pUnkOuter) {
856         IDirectMusicLoaderImpl *obj;
857         DMUS_OBJECTDESC Desc;
858         LPWINE_LOADER_ENTRY pDefaultDLSEntry;
859         struct list *pEntry;
860
861         TRACE("(%s, %p, %p)\n", debugstr_dmguid(lpcGUID), ppobj, pUnkOuter);
862         obj = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(IDirectMusicLoaderImpl));
863         if (NULL == obj) {
864                 *ppobj = NULL;
865                 return E_OUTOFMEMORY;
866         }
867         obj->LoaderVtbl = &DirectMusicLoader_Loader_Vtbl;
868         obj->dwRef = 0; /* will be inited with QueryInterface */
869         /* init critical section */
870         /* init cache/alias list */
871         /*InitializeCriticalSection (&obj->CritSect);
872         obj->CritSect.DebugInfo->Spare[0] = (DWORD_PTR)(__FILE__ ": IDirectMusicLoaderImpl.CritSect"); */
873         obj->pObjects = HeapAlloc (GetProcessHeap (), HEAP_ZERO_MEMORY, sizeof(struct list));
874         list_init (obj->pObjects);
875         /* init settings */
876         obj->pClassSettings = HeapAlloc (GetProcessHeap (), HEAP_ZERO_MEMORY, sizeof(struct list));
877         list_init (obj->pClassSettings);
878         DMUSIC_InitLoaderSettings ((LPDIRECTMUSICLOADER8)obj);
879
880         /* set default DLS collection (via SetObject... so that loading via DMUS_OBJ_OBJECT is possible) */
881         DM_STRUCT_INIT(&Desc);
882         Desc.dwValidData = DMUS_OBJ_CLASS | DMUS_OBJ_FILENAME | DMUS_OBJ_FULLPATH | DMUS_OBJ_OBJECT;
883         Desc.guidClass = CLSID_DirectMusicCollection;
884         Desc.guidObject = GUID_DefaultGMCollection;
885         DMUSIC_GetDefaultGMPath (Desc.wszFileName);
886         IDirectMusicLoader_SetObject ((LPDIRECTMUSICLOADER8)obj, &Desc);
887         /* and now the workaroundTM for "invalid" default DLS; basically, 
888            my tests showed that if GUID chunk is present in default DLS 
889            collection, loader treats it as "invalid" and returns 
890            DMUS_E_LOADER_NOFILENAME for all requests for it; basically, we check 
891            if out input guidObject was overwritten */
892         pEntry = list_head (obj->pObjects);
893         pDefaultDLSEntry = LIST_ENTRY (pEntry, WINE_LOADER_ENTRY, entry);
894         if (!IsEqualGUID(&Desc.guidObject, &GUID_DefaultGMCollection)) {
895                 pDefaultDLSEntry->bInvalidDefaultDLS = TRUE;
896         }
897
898         /* increase number of instances */
899         InterlockedIncrement (&dwDirectMusicLoader);
900         
901         return IDirectMusicLoaderImpl_IDirectMusicLoader_QueryInterface ((LPDIRECTMUSICLOADER8)obj, lpcGUID, ppobj);
902 }
903
904 /* help function for retrieval of search path and caching option for certain class */
905 static HRESULT DMUSIC_GetLoaderSettings (LPDIRECTMUSICLOADER8 iface, REFGUID pClassID, WCHAR* wszSearchPath, LPBOOL pbCache) {
906         ICOM_THIS_MULTI(IDirectMusicLoaderImpl, LoaderVtbl, iface);
907         struct list *pEntry;
908         TRACE(": (%p, %s, %p, %p)\n", This, debugstr_dmguid(pClassID), wszSearchPath, pbCache);
909         
910         LIST_FOR_EACH(pEntry, This->pClassSettings) {
911                 LPWINE_LOADER_OPTION pOptionEntry = LIST_ENTRY(pEntry, WINE_LOADER_OPTION, entry);
912                 if (IsEqualCLSID (pClassID, &pOptionEntry->guidClass)) {
913                         if (wszSearchPath)
914                                 strcpyW(wszSearchPath, pOptionEntry->wszSearchPath);
915                         if (pbCache)
916                                 *pbCache = pOptionEntry->bCache;
917                         return S_OK;
918                 }
919         }
920         return S_FALSE;
921 }
922
923 /* help function for setting search path and caching option for certain class */
924 static HRESULT DMUSIC_SetLoaderSettings (LPDIRECTMUSICLOADER8 iface, REFGUID pClassID, WCHAR* wszSearchPath, LPBOOL pbCache) {
925         ICOM_THIS_MULTI(IDirectMusicLoaderImpl, LoaderVtbl, iface);
926         struct list *pEntry;
927         HRESULT result = S_FALSE; /* in case pClassID != GUID_DirectMusicAllTypes and not a valid CLSID */
928         TRACE(": (%p, %s, %p, %p)\n", This, debugstr_dmguid(pClassID), wszSearchPath, pbCache);
929         
930         LIST_FOR_EACH(pEntry, This->pClassSettings) {
931                 LPWINE_LOADER_OPTION pOptionEntry = LIST_ENTRY(pEntry, WINE_LOADER_OPTION, entry);
932                 /* well, either we have GUID_DirectMusicAllTypes and need to set it to all,
933                    or specific CLSID is given and we set it only to it */
934                 if (IsEqualGUID (pClassID, &GUID_DirectMusicAllTypes) ||
935                         IsEqualCLSID (pClassID, &pOptionEntry->guidClass)) {
936                         if (wszSearchPath)
937                                 strcpyW(pOptionEntry->wszSearchPath, wszSearchPath);
938                         if (pbCache)
939                                 pOptionEntry->bCache = *pbCache;
940                         result = S_OK;
941                 }
942         }
943
944         return result;
945 }
946
947 static HRESULT DMUSIC_InitLoaderSettings (LPDIRECTMUSICLOADER8 iface) {
948         ICOM_THIS_MULTI(IDirectMusicLoaderImpl, LoaderVtbl, iface);
949         
950         /* hard-coded list of classes */
951         static REFCLSID classes[] = {
952                 &CLSID_DirectMusicAudioPathConfig,
953                 &CLSID_DirectMusicBand,
954                 &CLSID_DirectMusicContainer,
955                 &CLSID_DirectMusicCollection,
956                 &CLSID_DirectMusicChordMap,
957                 &CLSID_DirectMusicSegment,
958                 &CLSID_DirectMusicScript,
959                 &CLSID_DirectMusicSong,
960                 &CLSID_DirectMusicStyle,
961                 &CLSID_DirectMusicGraph,
962                 &CLSID_DirectSoundWave
963         };
964         
965         unsigned int i;
966         WCHAR wszCurrent[MAX_PATH];
967
968         TRACE(": (%p)\n", This);
969         GetCurrentDirectoryW (MAX_PATH, wszCurrent);
970
971         for (i = 0; i < sizeof(classes)/sizeof(REFCLSID); i++) {
972                 LPWINE_LOADER_OPTION pNewSetting = HeapAlloc (GetProcessHeap (), HEAP_ZERO_MEMORY, sizeof(WINE_LOADER_OPTION));
973                 pNewSetting->guidClass = *classes[i];
974                 strcpyW (pNewSetting->wszSearchPath, wszCurrent);
975                 pNewSetting->bCache = TRUE;
976                 list_add_tail (This->pClassSettings, &pNewSetting->entry);
977         }
978
979         return S_OK;
980 }