urlmon: Fixed misc.c tests on IE9 and make image_tiff_filter behave like IE9.
[wine] / dlls / d3dxof / d3dxof.c
1 /*
2  * Implementation of DirectX File Interfaces
3  *
4  * Copyright 2004, 2008, 2010 Christian Costa
5  *
6  * This library is free software; you can redistribute it and/or
7  * modify it under the terms of the GNU Lesser General Public
8  * License as published by the Free Software Foundation; either
9  * version 2.1 of the License, or (at your option) any later version.
10  *
11  * This library is distributed in the hope that it will be useful,
12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
14  * Lesser General Public License for more details.
15  *
16  * You should have received a copy of the GNU Lesser General Public
17  * License along with this library; if not, write to the Free Software
18  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
19  */
20
21 #include "config.h"
22 #include "wine/debug.h"
23
24 #define COBJMACROS
25
26 #include "winbase.h"
27 #include "wingdi.h"
28
29 #include "d3dxof_private.h"
30 #include "dxfile.h"
31
32 #include <stdio.h>
33
34 WINE_DEFAULT_DEBUG_CHANNEL(d3dxof);
35
36 static const struct IDirectXFileVtbl IDirectXFile_Vtbl;
37 static const struct IDirectXFileBinaryVtbl IDirectXFileBinary_Vtbl;
38 static const struct IDirectXFileDataVtbl IDirectXFileData_Vtbl;
39 static const struct IDirectXFileDataReferenceVtbl IDirectXFileDataReference_Vtbl;
40 static const struct IDirectXFileEnumObjectVtbl IDirectXFileEnumObject_Vtbl;
41 static const struct IDirectXFileObjectVtbl IDirectXFileObject_Vtbl;
42 static const struct IDirectXFileSaveObjectVtbl IDirectXFileSaveObject_Vtbl;
43
44 static HRESULT IDirectXFileDataReferenceImpl_Create(IDirectXFileDataReferenceImpl** ppObj);
45 static HRESULT IDirectXFileEnumObjectImpl_Create(IDirectXFileEnumObjectImpl** ppObj);
46 static HRESULT IDirectXFileSaveObjectImpl_Create(IDirectXFileSaveObjectImpl** ppObj);
47
48 HRESULT IDirectXFileImpl_Create(IUnknown* pUnkOuter, LPVOID* ppObj)
49 {
50     IDirectXFileImpl* object;
51
52     TRACE("(%p,%p)\n", pUnkOuter, ppObj);
53       
54     object = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(IDirectXFileImpl));
55     if (!object)
56     {
57         ERR("Out of memory\n");
58         return DXFILEERR_BADALLOC;
59     }
60
61     object->IDirectXFile_iface.lpVtbl = &IDirectXFile_Vtbl;
62     object->ref = 1;
63
64     *ppObj = &object->IDirectXFile_iface;
65
66     return S_OK;
67 }
68
69 static inline IDirectXFileImpl *impl_from_IDirectXFile(IDirectXFile *iface)
70 {
71     return CONTAINING_RECORD(iface, IDirectXFileImpl, IDirectXFile_iface);
72 }
73
74 /*** IUnknown methods ***/
75 static HRESULT WINAPI IDirectXFileImpl_QueryInterface(IDirectXFile* iface, REFIID riid, void** ppvObject)
76 {
77   IDirectXFileImpl *This = impl_from_IDirectXFile(iface);
78
79   TRACE("(%p/%p)->(%s,%p)\n", iface, This, debugstr_guid(riid), ppvObject);
80
81   if (IsEqualGUID(riid, &IID_IUnknown)
82       || IsEqualGUID(riid, &IID_IDirectXFile))
83   {
84     IUnknown_AddRef(iface);
85     *ppvObject = &This->IDirectXFile_iface;
86     return S_OK;
87   }
88
89   ERR("(%p)->(%s,%p),not found\n",This,debugstr_guid(riid),ppvObject);
90   return E_NOINTERFACE;
91 }
92
93 static ULONG WINAPI IDirectXFileImpl_AddRef(IDirectXFile* iface)
94 {
95   IDirectXFileImpl *This = impl_from_IDirectXFile(iface);
96   ULONG ref = InterlockedIncrement(&This->ref);
97
98   TRACE("(%p/%p): AddRef from %d\n", iface, This, ref - 1);
99
100   return ref;
101 }
102
103 static ULONG WINAPI IDirectXFileImpl_Release(IDirectXFile* iface)
104 {
105   IDirectXFileImpl *This = impl_from_IDirectXFile(iface);
106   ULONG ref = InterlockedDecrement(&This->ref);
107
108   TRACE("(%p/%p): ReleaseRef to %d\n", iface, This, ref);
109
110   if (!ref)
111     HeapFree(GetProcessHeap(), 0, This);
112
113   return ref;
114 }
115
116 /*** IDirectXFile methods ***/
117 static HRESULT WINAPI IDirectXFileImpl_CreateEnumObject(IDirectXFile* iface, LPVOID pvSource, DXFILELOADOPTIONS dwLoadOptions, LPDIRECTXFILEENUMOBJECT* ppEnumObj)
118 {
119   IDirectXFileImpl *This = impl_from_IDirectXFile(iface);
120   IDirectXFileEnumObjectImpl* object;
121   HRESULT hr;
122   LPBYTE file_buffer;
123   DWORD file_size;
124
125   TRACE("(%p/%p)->(%p,%x,%p)\n", This, iface, pvSource, dwLoadOptions, ppEnumObj);
126
127   if (!ppEnumObj)
128     return DXFILEERR_BADVALUE;
129
130   /* Only lowest 4 bits are relevant in DXFILELOADOPTIONS */
131   dwLoadOptions &= 0xF;
132
133   hr = IDirectXFileEnumObjectImpl_Create(&object);
134   if (FAILED(hr))
135     return hr;
136
137   if (dwLoadOptions == DXFILELOAD_FROMFILE)
138   {
139     HANDLE hFile, file_mapping;
140
141     TRACE("Open source file '%s'\n", (char*)pvSource);
142
143     hFile = CreateFileA(pvSource, GENERIC_READ, FILE_SHARE_READ, NULL, OPEN_EXISTING, 0, NULL);
144     if (hFile == INVALID_HANDLE_VALUE)
145     {
146       TRACE("File '%s' not found\n", (char*)pvSource);
147       return DXFILEERR_FILENOTFOUND;
148     }
149
150     file_size = GetFileSize(hFile, NULL);
151
152     file_mapping = CreateFileMappingA(hFile, NULL, PAGE_READONLY, 0, 0, NULL);
153     CloseHandle(hFile);
154     if (!file_mapping)
155     {
156       hr = DXFILEERR_BADFILETYPE;
157       goto error;
158     }
159
160     object->mapped_memory = MapViewOfFile(file_mapping, FILE_MAP_READ, 0, 0, 0);
161     CloseHandle(file_mapping);
162     if (!object->mapped_memory)
163     {
164       hr = DXFILEERR_BADFILETYPE;
165       goto error;
166     }
167     file_buffer = object->mapped_memory;
168   }
169   else if (dwLoadOptions == DXFILELOAD_FROMRESOURCE)
170   {
171     HRSRC resource_info;
172     HGLOBAL resource_data;
173     LPDXFILELOADRESOURCE lpdxflr = pvSource;
174
175     TRACE("Source in resource (module = %p, name = %s, type = %s\n", lpdxflr->hModule, debugstr_a(lpdxflr->lpName), debugstr_a(lpdxflr->lpType));
176
177     resource_info = FindResourceA(lpdxflr->hModule, lpdxflr->lpName, lpdxflr->lpType);
178     if (!resource_info)
179     {
180       hr = DXFILEERR_RESOURCENOTFOUND;
181       goto error;
182     }
183
184     file_size = SizeofResource(lpdxflr->hModule, resource_info);
185
186     resource_data = LoadResource(lpdxflr->hModule, resource_info);
187     if (!resource_data)
188     {
189       hr = DXFILEERR_BADRESOURCE;
190       goto error;
191     }
192
193     file_buffer = LockResource(resource_data);
194     if (!file_buffer)
195     {
196       hr = DXFILEERR_BADRESOURCE;
197       goto error;
198     }
199   }
200   else if (dwLoadOptions == DXFILELOAD_FROMMEMORY)
201   {
202     LPDXFILELOADMEMORY lpdxflm = pvSource;
203
204     TRACE("Source in memory at %p with size %d\n", lpdxflm->lpMemory, lpdxflm->dSize);
205
206     file_buffer = lpdxflm->lpMemory;
207     file_size = lpdxflm->dSize;
208   }
209   else
210   {
211     FIXME("Source type %d is not handled yet\n", dwLoadOptions);
212     hr = DXFILEERR_NOTDONEYET;
213     goto error;
214   }
215
216   TRACE("File size is %d bytes\n", file_size);
217
218   object->pDirectXFile = This;
219
220   object->buf.pdxf = This;
221   object->buf.token_present = FALSE;
222   object->buf.buffer = file_buffer;
223   object->buf.rem_bytes = file_size;
224   hr = parse_header(&object->buf, &object->decomp_buffer);
225   if (FAILED(hr))
226     goto error;
227
228   if (!parse_templates(&object->buf)) {
229     hr = DXFILEERR_BADVALUE;
230     goto error;
231   }
232
233   if (TRACE_ON(d3dxof))
234   {
235     int i;
236     TRACE("Registered templates (%d):\n", This->nb_xtemplates);
237     for (i = 0; i < This->nb_xtemplates; i++)
238       DPRINTF("%s - %s\n", This->xtemplates[i].name, debugstr_guid(&This->xtemplates[i].class_id));
239   }
240
241   *ppEnumObj = &object->IDirectXFileEnumObject_iface;
242
243   return DXFILE_OK;
244
245 error:
246   IDirectXFileEnumObject_Release(&object->IDirectXFileEnumObject_iface);
247   *ppEnumObj = NULL;
248
249   return hr;
250 }
251
252 static HRESULT WINAPI IDirectXFileImpl_CreateSaveObject(IDirectXFile* iface, LPCSTR szFileName, DXFILEFORMAT dwFileFormat, LPDIRECTXFILESAVEOBJECT* ppSaveObj)
253 {
254   IDirectXFileImpl *This = impl_from_IDirectXFile(iface);
255   IDirectXFileSaveObjectImpl *object;
256   HRESULT hr;
257
258   FIXME("(%p/%p)->(%s,%x,%p) partial stub!\n", This, iface, szFileName, dwFileFormat, ppSaveObj);
259
260   if (!szFileName || !ppSaveObj)
261     return E_POINTER;
262
263   hr = IDirectXFileSaveObjectImpl_Create(&object);
264   if (SUCCEEDED(hr))
265     *ppSaveObj = &object->IDirectXFileSaveObject_iface;
266   return hr;
267 }
268
269 static HRESULT WINAPI IDirectXFileImpl_RegisterTemplates(IDirectXFile* iface, LPVOID pvData, DWORD cbSize)
270 {
271   IDirectXFileImpl *This = impl_from_IDirectXFile(iface);
272   parse_buffer buf;
273   HRESULT hr;
274   LPBYTE decomp_buffer = NULL;
275
276   buf.buffer = pvData;
277   buf.rem_bytes = cbSize;
278   buf.txt = FALSE;
279   buf.token_present = FALSE;
280   buf.pdxf = This;
281
282   TRACE("(%p/%p)->(%p,%d)\n", This, iface, pvData, cbSize);
283
284   if (!pvData)
285     return DXFILEERR_BADVALUE;
286
287   hr = parse_header(&buf, &decomp_buffer);
288   if (FAILED(hr))
289     goto cleanup;
290
291   if (!parse_templates(&buf)) {
292     hr = DXFILEERR_BADVALUE;
293     goto cleanup;
294   }
295
296   if (TRACE_ON(d3dxof))
297   {
298     int i;
299     TRACE("Registered templates (%d):\n", This->nb_xtemplates);
300     for (i = 0; i < This->nb_xtemplates; i++)
301       DPRINTF("%s - %s\n", This->xtemplates[i].name, debugstr_guid(&This->xtemplates[i].class_id));
302   }
303
304   hr = DXFILE_OK;
305 cleanup:
306   HeapFree(GetProcessHeap(), 0, decomp_buffer);
307   return hr;
308 }
309
310 static const IDirectXFileVtbl IDirectXFile_Vtbl =
311 {
312   IDirectXFileImpl_QueryInterface,
313   IDirectXFileImpl_AddRef,
314   IDirectXFileImpl_Release,
315   IDirectXFileImpl_CreateEnumObject,
316   IDirectXFileImpl_CreateSaveObject,
317   IDirectXFileImpl_RegisterTemplates
318 };
319
320 static HRESULT IDirectXFileBinaryImpl_Create(IDirectXFileBinaryImpl** ppObj)
321 {
322     IDirectXFileBinaryImpl* object;
323
324     TRACE("(%p)\n", ppObj);
325
326     object = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(IDirectXFileBinaryImpl));
327     if (!object)
328     {
329         ERR("Out of memory\n");
330         return DXFILEERR_BADALLOC;
331     }
332
333     object->IDirectXFileBinary_iface.lpVtbl = &IDirectXFileBinary_Vtbl;
334     object->ref = 1;
335
336     *ppObj = object;
337
338     return DXFILE_OK;
339 }
340
341 static inline IDirectXFileBinaryImpl *impl_from_IDirectXFileBinary(IDirectXFileBinary *iface)
342 {
343     return CONTAINING_RECORD(iface, IDirectXFileBinaryImpl, IDirectXFileBinary_iface);
344 }
345
346 /*** IUnknown methods ***/
347 static HRESULT WINAPI IDirectXFileBinaryImpl_QueryInterface(IDirectXFileBinary* iface, REFIID riid, void** ppvObject)
348 {
349   IDirectXFileBinaryImpl *This = impl_from_IDirectXFileBinary(iface);
350
351   TRACE("(%p/%p)->(%s,%p)\n", iface, This, debugstr_guid(riid), ppvObject);
352
353   if (IsEqualGUID(riid, &IID_IUnknown)
354       || IsEqualGUID(riid, &IID_IDirectXFileObject)
355       || IsEqualGUID(riid, &IID_IDirectXFileBinary))
356   {
357     IUnknown_AddRef(iface);
358     *ppvObject = &This->IDirectXFileBinary_iface;
359     return S_OK;
360   }
361
362   /* Do not print an error for interfaces that can be queried to retrieve the type of the object */
363   if (!IsEqualGUID(riid, &IID_IDirectXFileData)
364       && !IsEqualGUID(riid, &IID_IDirectXFileDataReference))
365     ERR("(%p)->(%s,%p),not found\n",This,debugstr_guid(riid),ppvObject);
366
367   return E_NOINTERFACE;
368 }
369
370 static ULONG WINAPI IDirectXFileBinaryImpl_AddRef(IDirectXFileBinary* iface)
371 {
372   IDirectXFileBinaryImpl *This = impl_from_IDirectXFileBinary(iface);
373   ULONG ref = InterlockedIncrement(&This->ref);
374
375   TRACE("(%p/%p): AddRef from %d\n", iface, This, ref - 1);
376
377   return ref;
378 }
379
380 static ULONG WINAPI IDirectXFileBinaryImpl_Release(IDirectXFileBinary* iface)
381 {
382   IDirectXFileBinaryImpl *This = impl_from_IDirectXFileBinary(iface);
383   ULONG ref = InterlockedDecrement(&This->ref);
384
385   TRACE("(%p/%p): ReleaseRef to %d\n", iface, This, ref);
386
387   if (!ref)
388     HeapFree(GetProcessHeap(), 0, This);
389
390   return ref;
391 }
392
393 /*** IDirectXFileObject methods ***/
394 static HRESULT WINAPI IDirectXFileBinaryImpl_GetName(IDirectXFileBinary* iface, LPSTR pstrNameBuf, LPDWORD pdwBufLen)
395
396 {
397   IDirectXFileBinaryImpl *This = impl_from_IDirectXFileBinary(iface);
398
399   FIXME("(%p/%p)->(%p,%p) stub!\n", This, iface, pstrNameBuf, pdwBufLen); 
400
401   return DXFILEERR_BADVALUE;
402 }
403
404 static HRESULT WINAPI IDirectXFileBinaryImpl_GetId(IDirectXFileBinary* iface, LPGUID pGuid)
405 {
406   IDirectXFileBinaryImpl *This = impl_from_IDirectXFileBinary(iface);
407
408   FIXME("(%p/%p)->(%p) stub!\n", This, iface, pGuid); 
409
410   return DXFILEERR_BADVALUE;
411 }
412
413 /*** IDirectXFileBinary methods ***/
414 static HRESULT WINAPI IDirectXFileBinaryImpl_GetSize(IDirectXFileBinary* iface, DWORD* pcbSize)
415 {
416   IDirectXFileBinaryImpl *This = impl_from_IDirectXFileBinary(iface);
417
418   FIXME("(%p/%p)->(%p) stub!\n", This, iface, pcbSize); 
419
420   return DXFILEERR_BADVALUE;
421 }
422
423 static HRESULT WINAPI IDirectXFileBinaryImpl_GetMimeType(IDirectXFileBinary* iface, LPCSTR* pszMimeType)
424 {
425   IDirectXFileBinaryImpl *This = impl_from_IDirectXFileBinary(iface);
426
427   FIXME("(%p/%p)->(%p) stub!\n", This, iface, pszMimeType);
428
429   return DXFILEERR_BADVALUE;
430 }
431
432 static HRESULT WINAPI IDirectXFileBinaryImpl_Read(IDirectXFileBinary* iface, LPVOID pvData, DWORD cbSize, LPDWORD pcbRead)
433 {
434   IDirectXFileBinaryImpl *This = impl_from_IDirectXFileBinary(iface);
435
436   FIXME("(%p/%p)->(%p, %d, %p) stub!\n", This, iface, pvData, cbSize, pcbRead);
437
438   return DXFILEERR_BADVALUE;
439 }
440
441 static const IDirectXFileBinaryVtbl IDirectXFileBinary_Vtbl =
442 {
443     IDirectXFileBinaryImpl_QueryInterface,
444     IDirectXFileBinaryImpl_AddRef,
445     IDirectXFileBinaryImpl_Release,
446     IDirectXFileBinaryImpl_GetName,
447     IDirectXFileBinaryImpl_GetId,
448     IDirectXFileBinaryImpl_GetSize,
449     IDirectXFileBinaryImpl_GetMimeType,
450     IDirectXFileBinaryImpl_Read
451 };
452
453 static HRESULT IDirectXFileDataImpl_Create(IDirectXFileDataImpl** ppObj)
454 {
455     IDirectXFileDataImpl* object;
456
457     TRACE("(%p)\n", ppObj);
458       
459     object = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(IDirectXFileDataImpl));
460     if (!object)
461     {
462         ERR("Out of memory\n");
463         return DXFILEERR_BADALLOC;
464     }
465
466     object->IDirectXFileData_iface.lpVtbl = &IDirectXFileData_Vtbl;
467     object->ref = 1;
468
469     *ppObj = object;
470     
471     return S_OK;
472 }
473
474 static inline IDirectXFileDataImpl *impl_from_IDirectXFileData(IDirectXFileData *iface)
475 {
476     return CONTAINING_RECORD(iface, IDirectXFileDataImpl, IDirectXFileData_iface);
477 }
478
479 /*** IUnknown methods ***/
480 static HRESULT WINAPI IDirectXFileDataImpl_QueryInterface(IDirectXFileData* iface, REFIID riid, void** ppvObject)
481 {
482   IDirectXFileDataImpl *This = impl_from_IDirectXFileData(iface);
483
484   TRACE("(%p/%p)->(%s,%p)\n", iface, This, debugstr_guid(riid), ppvObject);
485
486   if (IsEqualGUID(riid, &IID_IUnknown)
487       || IsEqualGUID(riid, &IID_IDirectXFileObject)
488       || IsEqualGUID(riid, &IID_IDirectXFileData))
489   {
490     IUnknown_AddRef(iface);
491     *ppvObject = &This->IDirectXFileData_iface;
492     return S_OK;
493   }
494
495   /* Do not print an error for interfaces that can be queried to retrieve the type of the object */
496   if (!IsEqualGUID(riid, &IID_IDirectXFileBinary)
497       && !IsEqualGUID(riid, &IID_IDirectXFileDataReference))
498     ERR("(%p)->(%s,%p),not found\n",This,debugstr_guid(riid),ppvObject);
499
500   return E_NOINTERFACE;
501 }
502
503 static ULONG WINAPI IDirectXFileDataImpl_AddRef(IDirectXFileData* iface)
504 {
505   IDirectXFileDataImpl *This = impl_from_IDirectXFileData(iface);
506   ULONG ref = InterlockedIncrement(&This->ref);
507
508   TRACE("(%p/%p): AddRef from %d\n", iface, This, ref - 1);
509
510   return ref;
511 }
512
513 static ULONG WINAPI IDirectXFileDataImpl_Release(IDirectXFileData* iface)
514 {
515   IDirectXFileDataImpl *This = impl_from_IDirectXFileData(iface);
516   ULONG ref = InterlockedDecrement(&This->ref);
517
518   TRACE("(%p/%p): ReleaseRef to %d\n", iface, This, ref);
519
520   if (!ref)
521   {
522     if (!This->level && !This->from_ref)
523     {
524       HeapFree(GetProcessHeap(), 0, This->pstrings);
525       HeapFree(GetProcessHeap(), 0, This->pobj->pdata);
526       HeapFree(GetProcessHeap(), 0, This->pobj);
527     }
528     HeapFree(GetProcessHeap(), 0, This);
529   }
530
531   return ref;
532 }
533
534 /*** IDirectXFileObject methods ***/
535 static HRESULT WINAPI IDirectXFileDataImpl_GetName(IDirectXFileData* iface, LPSTR pstrNameBuf, LPDWORD pdwBufLen)
536 {
537   IDirectXFileDataImpl *This = impl_from_IDirectXFileData(iface);
538   DWORD len;
539
540   TRACE("(%p/%p)->(%p,%p)\n", This, iface, pstrNameBuf, pdwBufLen);
541
542   if (!pdwBufLen)
543     return DXFILEERR_BADVALUE;
544
545   len = strlen(This->pobj->name);
546   if (len)
547     len++;
548
549   if (pstrNameBuf) {
550     if (*pdwBufLen < len)
551       return DXFILEERR_BADVALUE;
552     CopyMemory(pstrNameBuf, This->pobj->name, len);
553   }
554   *pdwBufLen = len;
555
556   return DXFILE_OK;
557 }
558
559 static HRESULT WINAPI IDirectXFileDataImpl_GetId(IDirectXFileData* iface, LPGUID pGuid)
560 {
561   IDirectXFileDataImpl *This = impl_from_IDirectXFileData(iface);
562
563   TRACE("(%p/%p)->(%p)\n", This, iface, pGuid);
564
565   if (!pGuid)
566     return DXFILEERR_BADVALUE;
567
568   memcpy(pGuid, &This->pobj->class_id, 16);
569
570   return DXFILE_OK;
571 }
572
573 /*** IDirectXFileData methods ***/
574 static HRESULT WINAPI IDirectXFileDataImpl_GetData(IDirectXFileData* iface, LPCSTR szMember, DWORD* pcbSize, void** ppvData)
575 {
576   IDirectXFileDataImpl *This = impl_from_IDirectXFileData(iface);
577
578   TRACE("(%p/%p)->(%s,%p,%p)\n", This, iface, debugstr_a(szMember), pcbSize, ppvData);
579
580   if (!pcbSize || !ppvData)
581     return DXFILEERR_BADVALUE;
582
583   if (szMember)
584   {
585     FIXME("Specifying a member is not supported yet!\n");
586     return DXFILEERR_BADVALUE;
587   }
588
589   *pcbSize = This->pobj->size;
590   *ppvData = This->pobj->root->pdata + This->pobj->pos_data;
591
592   return DXFILE_OK;
593 }
594
595 static HRESULT WINAPI IDirectXFileDataImpl_GetType(IDirectXFileData* iface, const GUID** pguid)
596 {
597   IDirectXFileDataImpl *This = impl_from_IDirectXFileData(iface);
598   static GUID guid;
599
600   TRACE("(%p/%p)->(%p)\n", This, iface, pguid);
601
602   if (!pguid)
603     return DXFILEERR_BADVALUE;
604
605   memcpy(&guid, &This->pobj->type, 16);
606   *pguid = &guid;
607
608   return DXFILE_OK;
609 }
610
611 static HRESULT WINAPI IDirectXFileDataImpl_GetNextObject(IDirectXFileData* iface, LPDIRECTXFILEOBJECT* ppChildObj)
612 {
613   HRESULT hr;
614   IDirectXFileDataImpl *This = impl_from_IDirectXFileData(iface);
615
616   TRACE("(%p/%p)->(%p)\n", This, iface, ppChildObj);
617
618   if (This->cur_enum_object >= This->pobj->nb_children)
619     return DXFILEERR_NOMOREOBJECTS;
620
621   if (This->from_ref && (This->level >= 1))
622   {
623     /* Only 2 levels can be enumerated if the object is obtained from a reference */
624     return DXFILEERR_NOMOREOBJECTS;
625   }
626
627   if (This->pobj->children[This->cur_enum_object]->binary)
628   {
629     IDirectXFileBinaryImpl *object;
630
631     hr = IDirectXFileBinaryImpl_Create(&object);
632     if (FAILED(hr))
633       return hr;
634
635     *ppChildObj = (LPDIRECTXFILEOBJECT)&object->IDirectXFileBinary_iface;
636   }
637   else if (This->pobj->children[This->cur_enum_object]->ptarget)
638   {
639     IDirectXFileDataReferenceImpl *object;
640
641     hr = IDirectXFileDataReferenceImpl_Create(&object);
642     if (FAILED(hr))
643       return hr;
644
645     object->ptarget = This->pobj->children[This->cur_enum_object++]->ptarget;
646
647     *ppChildObj = (LPDIRECTXFILEOBJECT)&object->IDirectXFileDataReference_iface;
648   }
649   else
650   {
651     IDirectXFileDataImpl *object;
652
653     hr = IDirectXFileDataImpl_Create(&object);
654     if (FAILED(hr))
655       return hr;
656
657     object->pobj = This->pobj->children[This->cur_enum_object++];
658     object->cur_enum_object = 0;
659     object->from_ref = This->from_ref;
660     object->level = This->level + 1;
661
662     *ppChildObj = (LPDIRECTXFILEOBJECT)&object->IDirectXFileData_iface;
663   }
664
665   return DXFILE_OK;
666 }
667
668 static HRESULT WINAPI IDirectXFileDataImpl_AddDataObject(IDirectXFileData* iface, LPDIRECTXFILEDATA pDataObj)
669 {
670   IDirectXFileDataImpl *This = impl_from_IDirectXFileData(iface);
671
672   FIXME("(%p/%p)->(%p) stub!\n", This, iface, pDataObj); 
673
674   return DXFILEERR_BADVALUE;
675 }
676
677 static HRESULT WINAPI IDirectXFileDataImpl_AddDataReference(IDirectXFileData* iface, LPCSTR szRef, const GUID* pguidRef)
678 {
679   IDirectXFileDataImpl *This = impl_from_IDirectXFileData(iface);
680
681   FIXME("(%p/%p)->(%s,%p) stub!\n", This, iface, szRef, pguidRef); 
682
683   return DXFILEERR_BADVALUE;
684 }
685
686 static HRESULT WINAPI IDirectXFileDataImpl_AddBinaryObject(IDirectXFileData* iface, LPCSTR szName, const GUID* pguid, LPCSTR szMimeType, LPVOID pvData, DWORD cbSize)
687 {
688   IDirectXFileDataImpl *This = impl_from_IDirectXFileData(iface);
689
690   FIXME("(%p/%p)->(%s,%p,%s,%p,%d) stub!\n", This, iface, szName, pguid, szMimeType, pvData, cbSize);
691
692   return DXFILEERR_BADVALUE;
693 }
694
695 static const IDirectXFileDataVtbl IDirectXFileData_Vtbl =
696 {
697     IDirectXFileDataImpl_QueryInterface,
698     IDirectXFileDataImpl_AddRef,
699     IDirectXFileDataImpl_Release,
700     IDirectXFileDataImpl_GetName,
701     IDirectXFileDataImpl_GetId,
702     IDirectXFileDataImpl_GetData,
703     IDirectXFileDataImpl_GetType,
704     IDirectXFileDataImpl_GetNextObject,
705     IDirectXFileDataImpl_AddDataObject,
706     IDirectXFileDataImpl_AddDataReference,
707     IDirectXFileDataImpl_AddBinaryObject
708 };
709
710 static HRESULT IDirectXFileDataReferenceImpl_Create(IDirectXFileDataReferenceImpl** ppObj)
711 {
712     IDirectXFileDataReferenceImpl* object;
713
714     TRACE("(%p)\n", ppObj);
715       
716     object = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(IDirectXFileDataReferenceImpl));
717     if (!object)
718     {
719         ERR("Out of memory\n");
720         return DXFILEERR_BADALLOC;
721     }
722
723     object->IDirectXFileDataReference_iface.lpVtbl = &IDirectXFileDataReference_Vtbl;
724     object->ref = 1;
725
726     *ppObj = object;
727     
728     return S_OK;
729 }
730
731 static inline IDirectXFileDataReferenceImpl *impl_from_IDirectXFileDataReference(IDirectXFileDataReference *iface)
732 {
733     return CONTAINING_RECORD(iface, IDirectXFileDataReferenceImpl, IDirectXFileDataReference_iface);
734 }
735
736 /*** IUnknown methods ***/
737 static HRESULT WINAPI IDirectXFileDataReferenceImpl_QueryInterface(IDirectXFileDataReference* iface, REFIID riid, void** ppvObject)
738 {
739   IDirectXFileDataReferenceImpl *This = impl_from_IDirectXFileDataReference(iface);
740
741   TRACE("(%p/%p)->(%s,%p)\n", iface, This, debugstr_guid(riid), ppvObject);
742
743   if (IsEqualGUID(riid, &IID_IUnknown)
744       || IsEqualGUID(riid, &IID_IDirectXFileObject)
745       || IsEqualGUID(riid, &IID_IDirectXFileDataReference))
746   {
747     IUnknown_AddRef(iface);
748     *ppvObject = &This->IDirectXFileDataReference_iface;
749     return S_OK;
750   }
751
752   /* Do not print an error for interfaces that can be queried to retrieve the type of the object */
753   if (!IsEqualGUID(riid, &IID_IDirectXFileData)
754       && !IsEqualGUID(riid, &IID_IDirectXFileBinary))
755     ERR("(%p)->(%s,%p),not found\n",This,debugstr_guid(riid),ppvObject);
756
757   return E_NOINTERFACE;
758 }
759
760 static ULONG WINAPI IDirectXFileDataReferenceImpl_AddRef(IDirectXFileDataReference* iface)
761 {
762   IDirectXFileDataReferenceImpl *This = impl_from_IDirectXFileDataReference(iface);
763   ULONG ref = InterlockedIncrement(&This->ref);
764
765   TRACE("(%p/%p): AddRef from %d\n", iface, This, ref - 1);
766
767   return ref;
768 }
769
770 static ULONG WINAPI IDirectXFileDataReferenceImpl_Release(IDirectXFileDataReference* iface)
771 {
772   IDirectXFileDataReferenceImpl *This = impl_from_IDirectXFileDataReference(iface);
773   ULONG ref = InterlockedDecrement(&This->ref);
774
775   TRACE("(%p/%p): ReleaseRef to %d\n", iface, This, ref);
776
777   if (!ref)
778     HeapFree(GetProcessHeap(), 0, This);
779
780   return ref;
781 }
782
783 /*** IDirectXFileObject methods ***/
784 static HRESULT WINAPI IDirectXFileDataReferenceImpl_GetName(IDirectXFileDataReference* iface, LPSTR pstrNameBuf, LPDWORD pdwBufLen)
785 {
786   IDirectXFileDataReferenceImpl *This = impl_from_IDirectXFileDataReference(iface);
787   DWORD len;
788
789   TRACE("(%p/%p)->(%p,%p)\n", This, iface, pstrNameBuf, pdwBufLen);
790
791   if (!pdwBufLen)
792     return DXFILEERR_BADVALUE;
793
794   len = strlen(This->ptarget->name);
795   if (len)
796     len++;
797
798   if (pstrNameBuf) {
799     if (*pdwBufLen < len)
800       return DXFILEERR_BADVALUE;
801     CopyMemory(pstrNameBuf, This->ptarget->name, len);
802   }
803   *pdwBufLen = len;
804
805   return DXFILE_OK;
806 }
807
808 static HRESULT WINAPI IDirectXFileDataReferenceImpl_GetId(IDirectXFileDataReference* iface, LPGUID pGuid)
809 {
810   IDirectXFileDataReferenceImpl *This = impl_from_IDirectXFileDataReference(iface);
811
812   TRACE("(%p/%p)->(%p)\n", This, iface, pGuid);
813
814   if (!pGuid)
815     return DXFILEERR_BADVALUE;
816
817   memcpy(pGuid, &This->ptarget->class_id, 16);
818
819   return DXFILE_OK;
820 }
821
822 /*** IDirectXFileDataReference ***/
823 static HRESULT WINAPI IDirectXFileDataReferenceImpl_Resolve(IDirectXFileDataReference* iface, LPDIRECTXFILEDATA* ppDataObj)
824 {
825   IDirectXFileDataReferenceImpl *This = impl_from_IDirectXFileDataReference(iface);
826   IDirectXFileDataImpl *object;
827   HRESULT hr;
828
829   TRACE("(%p/%p)->(%p)\n", This, iface, ppDataObj);
830
831   if (!ppDataObj)
832     return DXFILEERR_BADVALUE;
833
834   hr = IDirectXFileDataImpl_Create(&object);
835   if (FAILED(hr))
836     return hr;
837
838   object->pobj = This->ptarget;
839   object->cur_enum_object = 0;
840   object->level = 0;
841   object->from_ref = TRUE;
842
843   *ppDataObj = (LPDIRECTXFILEDATA)object;
844
845   return DXFILE_OK;
846 }
847
848 static const IDirectXFileDataReferenceVtbl IDirectXFileDataReference_Vtbl =
849 {
850     IDirectXFileDataReferenceImpl_QueryInterface,
851     IDirectXFileDataReferenceImpl_AddRef,
852     IDirectXFileDataReferenceImpl_Release,
853     IDirectXFileDataReferenceImpl_GetName,
854     IDirectXFileDataReferenceImpl_GetId,
855     IDirectXFileDataReferenceImpl_Resolve
856 };
857
858 static HRESULT IDirectXFileEnumObjectImpl_Create(IDirectXFileEnumObjectImpl** ppObj)
859 {
860     IDirectXFileEnumObjectImpl* object;
861
862     TRACE("(%p)\n", ppObj);
863       
864     object = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(IDirectXFileEnumObjectImpl));
865     if (!object)
866     {
867         ERR("Out of memory\n");
868         return DXFILEERR_BADALLOC;
869     }
870
871     object->IDirectXFileEnumObject_iface.lpVtbl = &IDirectXFileEnumObject_Vtbl;
872     object->ref = 1;
873
874     *ppObj = object;
875     
876     return S_OK;
877 }
878
879 static inline IDirectXFileEnumObjectImpl *impl_from_IDirectXFileEnumObject(IDirectXFileEnumObject *iface)
880 {
881     return CONTAINING_RECORD(iface, IDirectXFileEnumObjectImpl, IDirectXFileEnumObject_iface);
882 }
883
884 /*** IUnknown methods ***/
885 static HRESULT WINAPI IDirectXFileEnumObjectImpl_QueryInterface(IDirectXFileEnumObject* iface, REFIID riid, void** ppvObject)
886 {
887   IDirectXFileEnumObjectImpl *This = impl_from_IDirectXFileEnumObject(iface);
888
889   TRACE("(%p/%p)->(%s,%p)\n", iface, This, debugstr_guid(riid), ppvObject);
890
891   if (IsEqualGUID(riid, &IID_IUnknown)
892       || IsEqualGUID(riid, &IID_IDirectXFileEnumObject))
893   {
894     IUnknown_AddRef(iface);
895     *ppvObject = &This->IDirectXFileEnumObject_iface;
896     return S_OK;
897   }
898
899   ERR("(%p)->(%s,%p),not found\n",This,debugstr_guid(riid),ppvObject);
900   return E_NOINTERFACE;
901 }
902
903 static ULONG WINAPI IDirectXFileEnumObjectImpl_AddRef(IDirectXFileEnumObject* iface)
904 {
905   IDirectXFileEnumObjectImpl *This = impl_from_IDirectXFileEnumObject(iface);
906   ULONG ref = InterlockedIncrement(&This->ref);
907
908   TRACE("(%p/%p): AddRef from %d\n", iface, This, ref - 1);
909
910   return ref;
911 }
912
913 static ULONG WINAPI IDirectXFileEnumObjectImpl_Release(IDirectXFileEnumObject* iface)
914 {
915   IDirectXFileEnumObjectImpl *This = impl_from_IDirectXFileEnumObject(iface);
916   ULONG ref = InterlockedDecrement(&This->ref);
917
918   TRACE("(%p/%p): ReleaseRef to %d\n", iface, This, ref);
919
920   if (!ref)
921   {
922     int i;
923     for (i = 0; i < This->nb_xobjects; i++)
924       IDirectXFileData_Release(This->pRefObjects[i]);
925     if (This->mapped_memory)
926       UnmapViewOfFile(This->mapped_memory);
927     HeapFree(GetProcessHeap(), 0, This->decomp_buffer);
928     HeapFree(GetProcessHeap(), 0, This);
929   }
930
931   return ref;
932 }
933
934 /*** IDirectXFileEnumObject methods ***/
935 static HRESULT WINAPI IDirectXFileEnumObjectImpl_GetNextDataObject(IDirectXFileEnumObject* iface, LPDIRECTXFILEDATA* ppDataObj)
936 {
937   IDirectXFileEnumObjectImpl *This = impl_from_IDirectXFileEnumObject(iface);
938   IDirectXFileDataImpl* object;
939   HRESULT hr;
940   LPBYTE pstrings = NULL;
941
942   TRACE("(%p/%p)->(%p)\n", This, iface, ppDataObj);
943
944   if (This->nb_xobjects >= MAX_OBJECTS)
945   {
946     ERR("Too many objects\n");
947     return DXFILEERR_NOMOREOBJECTS;
948   }
949
950   /* Check if there are templates defined before the object */
951   if (!parse_templates(&This->buf)) {
952     hr = DXFILEERR_BADVALUE;
953     goto error;
954   }
955
956   if (!This->buf.rem_bytes)
957     return DXFILEERR_NOMOREOBJECTS;
958
959   hr = IDirectXFileDataImpl_Create(&object);
960   if (FAILED(hr))
961     return hr;
962
963   This->buf.pxo_globals = This->xobjects;
964   This->buf.nb_pxo_globals = This->nb_xobjects;
965   This->buf.level = 0;
966
967   This->buf.pxo_tab = HeapAlloc(GetProcessHeap(), 0, sizeof(xobject)*MAX_SUBOBJECTS);
968   if (!This->buf.pxo_tab)
969   {
970     ERR("Out of memory\n");
971     hr = DXFILEERR_BADALLOC;
972     goto error;
973   }
974   This->buf.pxo = This->xobjects[This->nb_xobjects] = This->buf.pxo_tab;
975
976   This->buf.pxo->pdata = This->buf.pdata = NULL;
977   This->buf.capacity = 0;
978   This->buf.cur_pos_data = 0;
979   This->buf.pxo->nb_subobjects = 1;
980
981   pstrings = HeapAlloc(GetProcessHeap(), 0, MAX_STRINGS_BUFFER);
982   if (!pstrings)
983   {
984     ERR("Out of memory\n");
985     hr = DXFILEERR_BADALLOC;
986     goto error;
987   }
988   This->buf.cur_pstrings = This->buf.pstrings = object->pstrings = pstrings;
989
990   if (!parse_object(&This->buf))
991   {
992     WARN("Object is not correct\n");
993     hr = DXFILEERR_PARSEERROR;
994     goto error;
995   }
996
997   object->pstrings = pstrings;
998   object->pobj = This->buf.pxo;
999   object->cur_enum_object = 0;
1000   object->level = 0;
1001   object->from_ref = FALSE;
1002
1003   *ppDataObj = (LPDIRECTXFILEDATA)object;
1004
1005   /* Get a reference to created object */
1006   This->pRefObjects[This->nb_xobjects] = (LPDIRECTXFILEDATA)object;
1007   IDirectXFileData_AddRef(This->pRefObjects[This->nb_xobjects]);
1008
1009   This->nb_xobjects++;
1010
1011   return DXFILE_OK;
1012
1013 error:
1014
1015   HeapFree(GetProcessHeap(), 0, This->buf.pxo_tab);
1016   HeapFree(GetProcessHeap(), 0, This->buf.pdata);
1017   HeapFree(GetProcessHeap(), 0, pstrings);
1018
1019   return hr;
1020 }
1021
1022 static HRESULT WINAPI IDirectXFileEnumObjectImpl_GetDataObjectById(IDirectXFileEnumObject* iface, REFGUID rguid, LPDIRECTXFILEDATA* ppDataObj)
1023 {
1024   IDirectXFileEnumObjectImpl *This = impl_from_IDirectXFileEnumObject(iface);
1025
1026   FIXME("(%p/%p)->(%p,%p) stub!\n", This, iface, rguid, ppDataObj); 
1027
1028   return DXFILEERR_BADVALUE;
1029 }
1030
1031 static HRESULT WINAPI IDirectXFileEnumObjectImpl_GetDataObjectByName(IDirectXFileEnumObject* iface, LPCSTR szName, LPDIRECTXFILEDATA* ppDataObj)
1032 {
1033   IDirectXFileEnumObjectImpl *This = impl_from_IDirectXFileEnumObject(iface);
1034
1035   FIXME("(%p/%p)->(%s,%p) stub!\n", This, iface, szName, ppDataObj); 
1036
1037   return DXFILEERR_BADVALUE;
1038 }
1039
1040 static const IDirectXFileEnumObjectVtbl IDirectXFileEnumObject_Vtbl =
1041 {
1042     IDirectXFileEnumObjectImpl_QueryInterface,
1043     IDirectXFileEnumObjectImpl_AddRef,
1044     IDirectXFileEnumObjectImpl_Release,
1045     IDirectXFileEnumObjectImpl_GetNextDataObject,
1046     IDirectXFileEnumObjectImpl_GetDataObjectById,
1047     IDirectXFileEnumObjectImpl_GetDataObjectByName
1048 };
1049
1050 static HRESULT IDirectXFileSaveObjectImpl_Create(IDirectXFileSaveObjectImpl** ppObj)
1051 {
1052     IDirectXFileSaveObjectImpl* object;
1053
1054     TRACE("(%p)\n", ppObj);
1055
1056     object = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(IDirectXFileSaveObjectImpl));
1057     if (!object)
1058     {
1059         ERR("Out of memory\n");
1060         return DXFILEERR_BADALLOC;
1061     }
1062
1063     object->IDirectXFileSaveObject_iface.lpVtbl = &IDirectXFileSaveObject_Vtbl;
1064     object->ref = 1;
1065
1066     *ppObj = object;
1067
1068     return S_OK;
1069 }
1070
1071 static inline IDirectXFileSaveObjectImpl *impl_from_IDirectXFileSaveObject(IDirectXFileSaveObject *iface)
1072 {
1073     return CONTAINING_RECORD(iface, IDirectXFileSaveObjectImpl, IDirectXFileSaveObject_iface);
1074 }
1075
1076 /*** IUnknown methods ***/
1077 static HRESULT WINAPI IDirectXFileSaveObjectImpl_QueryInterface(IDirectXFileSaveObject* iface, REFIID riid, void** ppvObject)
1078 {
1079   IDirectXFileSaveObjectImpl *This = impl_from_IDirectXFileSaveObject(iface);
1080
1081   TRACE("(%p/%p)->(%s,%p)\n", iface, This, debugstr_guid(riid), ppvObject);
1082
1083   if (IsEqualGUID(riid, &IID_IUnknown)
1084       || IsEqualGUID(riid, &IID_IDirectXFileSaveObject))
1085   {
1086     IUnknown_AddRef(iface);
1087     *ppvObject = &This->IDirectXFileSaveObject_iface;
1088     return S_OK;
1089   }
1090
1091   ERR("(%p)->(%s,%p),not found\n",This,debugstr_guid(riid),ppvObject);
1092   return E_NOINTERFACE;
1093 }
1094
1095 static ULONG WINAPI IDirectXFileSaveObjectImpl_AddRef(IDirectXFileSaveObject* iface)
1096 {
1097   IDirectXFileSaveObjectImpl *This = impl_from_IDirectXFileSaveObject(iface);
1098   ULONG ref = InterlockedIncrement(&This->ref);
1099
1100   TRACE("(%p/%p): AddRef from %d\n", iface, This, ref - 1);
1101
1102   return ref;
1103 }
1104
1105 static ULONG WINAPI IDirectXFileSaveObjectImpl_Release(IDirectXFileSaveObject* iface)
1106 {
1107   IDirectXFileSaveObjectImpl *This = impl_from_IDirectXFileSaveObject(iface);
1108   ULONG ref = InterlockedDecrement(&This->ref);
1109
1110   TRACE("(%p/%p): ReleaseRef to %d\n", iface, This, ref);
1111
1112   if (!ref)
1113     HeapFree(GetProcessHeap(), 0, This);
1114
1115   return ref;
1116 }
1117
1118 static HRESULT WINAPI IDirectXFileSaveObjectImpl_SaveTemplates(IDirectXFileSaveObject* iface, DWORD cTemplates, const GUID** ppguidTemplates)
1119 {
1120   IDirectXFileSaveObjectImpl *This = impl_from_IDirectXFileSaveObject(iface);
1121
1122   FIXME("(%p/%p)->(%d,%p) stub!\n", This, iface, cTemplates, ppguidTemplates);
1123
1124   return DXFILEERR_BADVALUE;
1125 }
1126
1127 static HRESULT WINAPI IDirectXFileSaveObjectImpl_CreateDataObject(IDirectXFileSaveObject* iface, REFGUID rguidTemplate, LPCSTR szName, const GUID* pguid, DWORD cbSize, LPVOID pvData, LPDIRECTXFILEDATA* ppDataObj)
1128 {
1129   IDirectXFileSaveObjectImpl *This = impl_from_IDirectXFileSaveObject(iface);
1130
1131   FIXME("(%p/%p)->(%p,%s,%p,%d,%p,%p) stub!\n", This, iface, rguidTemplate, szName, pguid, cbSize, pvData, ppDataObj);
1132
1133   return DXFILEERR_BADVALUE;
1134 }
1135
1136 static HRESULT WINAPI IDirectXFileSaveObjectImpl_SaveData(IDirectXFileSaveObject* iface, LPDIRECTXFILEDATA ppDataObj)
1137 {
1138   IDirectXFileSaveObjectImpl *This = impl_from_IDirectXFileSaveObject(iface);
1139
1140   FIXME("(%p/%p)->(%p) stub!\n", This, iface, ppDataObj); 
1141
1142   return DXFILEERR_BADVALUE;
1143 }
1144
1145 static const IDirectXFileSaveObjectVtbl IDirectXFileSaveObject_Vtbl =
1146 {
1147     IDirectXFileSaveObjectImpl_QueryInterface,
1148     IDirectXFileSaveObjectImpl_AddRef,
1149     IDirectXFileSaveObjectImpl_Release,
1150     IDirectXFileSaveObjectImpl_SaveTemplates,
1151     IDirectXFileSaveObjectImpl_CreateDataObject,
1152     IDirectXFileSaveObjectImpl_SaveData
1153 };