wintrust: Add tests for encoding/decoding SPC links.
[wine] / dlls / ddraw / main.c
1 /*        DirectDraw Base Functions
2  *
3  * Copyright 1997-1999 Marcus Meissner
4  * Copyright 1998 Lionel Ulmer
5  * Copyright 2000-2001 TransGaming Technologies Inc.
6  * Copyright 2006 Stefan Dösinger
7  *
8  * This file contains the (internal) driver registration functions,
9  * driver enumeration APIs and DirectDraw creation functions.
10  *
11  * This library is free software; you can redistribute it and/or
12  * modify it under the terms of the GNU Lesser General Public
13  * License as published by the Free Software Foundation; either
14  * version 2.1 of the License, or (at your option) any later version.
15  *
16  * This library is distributed in the hope that it will be useful,
17  * but WITHOUT ANY WARRANTY; without even the implied warranty of
18  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
19  * Lesser General Public License for more details.
20  *
21  * You should have received a copy of the GNU Lesser General Public
22  * License along with this library; if not, write to the Free Software
23  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
24  */
25
26 #include "config.h"
27 #include "wine/port.h"
28 #include "wine/debug.h"
29
30 #include <assert.h>
31 #include <stdarg.h>
32 #include <string.h>
33 #include <stdlib.h>
34
35 #define COBJMACROS
36
37 #include "windef.h"
38 #include "winbase.h"
39 #include "winerror.h"
40 #include "wingdi.h"
41 #include "wine/exception.h"
42 #include "winreg.h"
43
44 #include "ddraw.h"
45 #include "d3d.h"
46
47 #include "ddraw_private.h"
48
49 typedef IWineD3D* (WINAPI *fnWineDirect3DCreate)(UINT, UINT, IUnknown *);
50
51 static HMODULE hWineD3D = (HMODULE) -1;
52 static fnWineDirect3DCreate pWineDirect3DCreate;
53
54 WINE_DEFAULT_DEBUG_CHANNEL(ddraw);
55
56 /* The configured default surface */
57 WINED3DSURFTYPE DefaultSurfaceType = SURFACE_UNKNOWN;
58
59 /* DDraw list and critical section */
60 static struct list global_ddraw_list = LIST_INIT(global_ddraw_list);
61
62 static CRITICAL_SECTION_DEBUG ddraw_cs_debug =
63 {
64     0, 0, &ddraw_cs,
65     { &ddraw_cs_debug.ProcessLocksList,
66     &ddraw_cs_debug.ProcessLocksList },
67     0, 0, { (DWORD_PTR)(__FILE__ ": ddraw_cs") }
68 };
69 CRITICAL_SECTION ddraw_cs = { &ddraw_cs_debug, -1, 0, 0, 0, 0 };
70
71 /***********************************************************************
72  *
73  * Helper function for DirectDrawCreate and friends
74  * Creates a new DDraw interface with the given REFIID
75  *
76  * Interfaces that can be created:
77  *  IDirectDraw, IDirectDraw2, IDirectDraw4, IDirectDraw7
78  *  IDirect3D, IDirect3D2, IDirect3D3, IDirect3D7. (Does Windows return
79  *  IDirect3D interfaces?)
80  *
81  * Arguments:
82  *  guid: ID of the requested driver, NULL for the default driver.
83  *        The GUID can be queried with DirectDrawEnumerate(Ex)A/W
84  *  DD: Used to return the pointer to the created object
85  *  UnkOuter: For aggregation, which is unsupported. Must be NULL
86  *  iid: requested version ID.
87  *
88  * Returns:
89  *  DD_OK if the Interface was created successfully
90  *  CLASS_E_NOAGGREGATION if UnkOuter is not NULL
91  *  E_OUTOFMEMORY if some allocation failed
92  *
93  ***********************************************************************/
94 static HRESULT
95 DDRAW_Create(const GUID *guid,
96              void **DD,
97              IUnknown *UnkOuter,
98              REFIID iid)
99 {
100     IDirectDrawImpl *This = NULL;
101     HRESULT hr;
102     IWineD3D *wineD3D = NULL;
103     IWineD3DDevice *wineD3DDevice = NULL;
104     HDC hDC;
105     WINED3DDEVTYPE devicetype;
106
107     TRACE("(%s,%p,%p)\n", debugstr_guid(guid), DD, UnkOuter);
108
109     *DD = NULL;
110
111     /* We don't care about this guids. Well, there's no special guid anyway
112      * OK, we could
113      */
114     if (guid == (GUID *) DDCREATE_EMULATIONONLY)
115     {
116         /* Use the reference device id. This doesn't actually change anything,
117          * WineD3D always uses OpenGL for D3D rendering. One could make it request
118          * indirect rendering
119          */
120         devicetype = WINED3DDEVTYPE_REF;
121     }
122     else if(guid == (GUID *) DDCREATE_HARDWAREONLY)
123     {
124         devicetype = WINED3DDEVTYPE_HAL;
125     }
126     else
127     {
128         devicetype = 0;
129     }
130
131     /* DDraw doesn't support aggregation, according to msdn */
132     if (UnkOuter != NULL)
133         return CLASS_E_NOAGGREGATION;
134
135     /* DirectDraw creation comes here */
136     This = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(IDirectDrawImpl));
137     if(!This)
138     {
139         ERR("Out of memory when creating DirectDraw\n");
140         return E_OUTOFMEMORY;
141     }
142
143     /* The interfaces:
144      * IDirectDraw and IDirect3D are the same object,
145      * QueryInterface is used to get other interfaces.
146      */
147     ICOM_INIT_INTERFACE(This, IDirectDraw,  IDirectDraw1_Vtbl);
148     ICOM_INIT_INTERFACE(This, IDirectDraw2, IDirectDraw2_Vtbl);
149     ICOM_INIT_INTERFACE(This, IDirectDraw3, IDirectDraw3_Vtbl);
150     ICOM_INIT_INTERFACE(This, IDirectDraw4, IDirectDraw4_Vtbl);
151     ICOM_INIT_INTERFACE(This, IDirectDraw7, IDirectDraw7_Vtbl);
152     ICOM_INIT_INTERFACE(This, IDirect3D,  IDirect3D1_Vtbl);
153     ICOM_INIT_INTERFACE(This, IDirect3D2, IDirect3D2_Vtbl);
154     ICOM_INIT_INTERFACE(This, IDirect3D3, IDirect3D3_Vtbl);
155     ICOM_INIT_INTERFACE(This, IDirect3D7, IDirect3D7_Vtbl);
156
157     /* See comments in IDirectDrawImpl_CreateNewSurface for a description
158      * of this member.
159      * Read from a registry key, should add a winecfg option later
160      */
161     This->ImplType = DefaultSurfaceType;
162
163     /* Get the current screen settings */
164     hDC = GetDC(0);
165     This->orig_bpp = GetDeviceCaps(hDC, BITSPIXEL) * GetDeviceCaps(hDC, PLANES);
166     ReleaseDC(0, hDC);
167     This->orig_width = GetSystemMetrics(SM_CXSCREEN);
168     This->orig_height = GetSystemMetrics(SM_CYSCREEN);
169
170     if (hWineD3D == (HMODULE) -1)
171     {
172         hWineD3D = LoadLibraryA("wined3d");
173         if (hWineD3D)
174         {
175             pWineDirect3DCreate = (fnWineDirect3DCreate) GetProcAddress(hWineD3D, "WineDirect3DCreate");
176             pWineDirect3DCreateClipper = (fnWineDirect3DCreateClipper) GetProcAddress(hWineD3D, "WineDirect3DCreateClipper");
177         }
178     }
179
180     if (!hWineD3D)
181     {
182         ERR("Couldn't load WineD3D - OpenGL libs not present?\n");
183         hr = DDERR_NODIRECTDRAWSUPPORT;
184         goto err_out;
185     }
186
187     /* Initialize WineD3D
188      *
189      * All Rendering (2D and 3D) is relayed to WineD3D,
190      * but DirectDraw specific management, like DDSURFACEDESC and DDPIXELFORMAT
191      * structure handling is handled in this lib.
192      */
193     wineD3D = pWineDirect3DCreate(0 /* SDKVersion */, 7 /* DXVersion */, (IUnknown *) This /* Parent */);
194     if(!wineD3D)
195     {
196         ERR("Failed to initialise WineD3D\n");
197         hr = E_OUTOFMEMORY;
198         goto err_out;
199     }
200     This->wineD3D = wineD3D;
201     TRACE("WineD3D created at %p\n", wineD3D);
202
203     /* Initialized member...
204      *
205      * It is set to false at creation time, and set to true in
206      * IDirectDraw7::Initialize. Its sole purpose is to return DD_OK on
207      * initialize only once
208      */
209     This->initialized = FALSE;
210
211     /* Initialize WineD3DDevice
212      *
213      * It is used for screen setup, surface and palette creation
214      * When a Direct3DDevice7 is created, the D3D capabilities of WineD3D are
215      * initialized
216      */
217     hr = IWineD3D_CreateDevice(wineD3D,
218                                0 /*D3D_ADAPTER_DEFAULT*/,
219                                devicetype,
220                                NULL, /* FocusWindow, don't know yet */
221                                0, /* BehaviorFlags */
222                                &wineD3DDevice,
223                                (IUnknown *) ICOM_INTERFACE(This, IDirectDraw7));
224     if(FAILED(hr))
225     {
226         ERR("Failed to create a wineD3DDevice, result = %x\n", hr);
227         goto err_out;
228     }
229     This->wineD3DDevice = wineD3DDevice;
230     TRACE("wineD3DDevice created at %p\n", This->wineD3DDevice);
231
232     /* Register the window class
233      *
234      * It is used to create a hidden window for D3D
235      * rendering, if the application didn't pass one.
236      * It can also be used for Creating a device window
237      * from SetCooperativeLevel
238      *
239      * The name: DDRAW_<address>. The classname is
240      * 32 bit long, so a 64 bit address will fit nicely
241      * (Will this be compiled for 64 bit anyway?)
242      *
243      */
244     sprintf(This->classname, "DDRAW_%p", This);
245
246     memset(&This->wnd_class, 0, sizeof(This->wnd_class));
247     This->wnd_class.style = CS_HREDRAW | CS_VREDRAW;
248     This->wnd_class.lpfnWndProc = DefWindowProcA;
249     This->wnd_class.cbClsExtra = 0;
250     This->wnd_class.cbWndExtra = 0;
251     This->wnd_class.hInstance = GetModuleHandleA(0);
252     This->wnd_class.hIcon = 0;
253     This->wnd_class.hCursor = 0;
254     This->wnd_class.hbrBackground = (HBRUSH) GetStockObject(BLACK_BRUSH);
255     This->wnd_class.lpszMenuName = NULL;
256     This->wnd_class.lpszClassName = This->classname;
257     if(!RegisterClassA(&This->wnd_class))
258     {
259         ERR("RegisterClassA failed!\n");
260         goto err_out;
261     }
262
263     /* Get the amount of video memory */
264     This->total_vidmem = IWineD3DDevice_GetAvailableTextureMem(This->wineD3DDevice);
265
266     /* Initialize the caps */
267     This->caps.dwSize = sizeof(This->caps);
268 /* do not report DDCAPS_OVERLAY and friends since we don't support overlays */
269 #define BLIT_CAPS (DDCAPS_BLT | DDCAPS_BLTCOLORFILL | DDCAPS_BLTDEPTHFILL \
270           | DDCAPS_BLTSTRETCH | DDCAPS_CANBLTSYSMEM | DDCAPS_CANCLIP      \
271           | DDCAPS_CANCLIPSTRETCHED | DDCAPS_COLORKEY                     \
272           | DDCAPS_COLORKEYHWASSIST | DDCAPS_ALIGNBOUNDARYSRC )
273 #define CKEY_CAPS (DDCKEYCAPS_DESTBLT | DDCKEYCAPS_SRCBLT)
274 #define FX_CAPS (DDFXCAPS_BLTALPHA | DDFXCAPS_BLTMIRRORLEFTRIGHT        \
275                 | DDFXCAPS_BLTMIRRORUPDOWN | DDFXCAPS_BLTROTATION90     \
276                 | DDFXCAPS_BLTSHRINKX | DDFXCAPS_BLTSHRINKXN            \
277                 | DDFXCAPS_BLTSHRINKY | DDFXCAPS_BLTSHRINKXN            \
278                 | DDFXCAPS_BLTSTRETCHX | DDFXCAPS_BLTSTRETCHXN          \
279                 | DDFXCAPS_BLTSTRETCHY | DDFXCAPS_BLTSTRETCHYN)
280     This->caps.dwCaps |= DDCAPS_GDI | DDCAPS_PALETTE | BLIT_CAPS;
281
282     This->caps.dwCaps2 |= DDCAPS2_CERTIFIED | DDCAPS2_NOPAGELOCKREQUIRED |
283                           DDCAPS2_PRIMARYGAMMA | DDCAPS2_WIDESURFACES |
284                           DDCAPS2_CANRENDERWINDOWED;
285     This->caps.dwCKeyCaps |= CKEY_CAPS;
286     This->caps.dwFXCaps |= FX_CAPS;
287     This->caps.dwPalCaps |= DDPCAPS_8BIT | DDPCAPS_PRIMARYSURFACE;
288     This->caps.dwVidMemTotal = This->total_vidmem;
289     This->caps.dwVidMemFree = This->total_vidmem;
290     This->caps.dwSVBCaps |= BLIT_CAPS;
291     This->caps.dwSVBCKeyCaps |= CKEY_CAPS;
292     This->caps.dwSVBFXCaps |= FX_CAPS;
293     This->caps.dwVSBCaps |= BLIT_CAPS;
294     This->caps.dwVSBCKeyCaps |= CKEY_CAPS;
295     This->caps.dwVSBFXCaps |= FX_CAPS;
296     This->caps.dwSSBCaps |= BLIT_CAPS;
297     This->caps.dwSSBCKeyCaps |= CKEY_CAPS;
298     This->caps.dwSSBFXCaps |= FX_CAPS;
299     This->caps.ddsCaps.dwCaps |= DDSCAPS_ALPHA | DDSCAPS_BACKBUFFER |
300                                  DDSCAPS_FLIP | DDSCAPS_FRONTBUFFER |
301                                  DDSCAPS_OFFSCREENPLAIN | DDSCAPS_PALETTE |
302                                  DDSCAPS_PRIMARYSURFACE | DDSCAPS_SYSTEMMEMORY |
303                                  DDSCAPS_VIDEOMEMORY | DDSCAPS_VISIBLE;
304     /* Hacks for D3D code */
305     /* TODO: Check if WineD3D has 3D enabled
306        Need opengl surfaces or auto for 3D
307      */
308     if(This->ImplType == 0 || This->ImplType == SURFACE_OPENGL)
309     {
310         This->caps.dwCaps |= DDCAPS_3D;
311         This->caps.ddsCaps.dwCaps |= DDSCAPS_3DDEVICE | DDSCAPS_MIPMAP | DDSCAPS_TEXTURE | DDSCAPS_ZBUFFER;
312     }
313     This->caps.ddsOldCaps.dwCaps = This->caps.ddsCaps.dwCaps;
314
315 #undef BLIT_CAPS
316 #undef CKEY_CAPS
317 #undef FX_CAPS
318
319     list_init(&This->surface_list);
320     list_add_head(&global_ddraw_list, &This->ddraw_list_entry);
321
322     /* Call QueryInterface to get the pointer to the requested interface. This also initializes
323      * The required refcount
324      */
325     hr = IDirectDraw7_QueryInterface( ICOM_INTERFACE(This, IDirectDraw7), iid, DD);
326     if(SUCCEEDED(hr)) return DD_OK;
327
328 err_out:
329     /* Let's hope we never need this ;) */
330     if(wineD3DDevice) IWineD3DDevice_Release(wineD3DDevice);
331     if(wineD3D) IWineD3D_Release(wineD3D);
332     if(This) HeapFree(GetProcessHeap(), 0, This->decls);
333     HeapFree(GetProcessHeap(), 0, This);
334     return hr;
335 }
336
337 /***********************************************************************
338  * DirectDrawCreate (DDRAW.@)
339  *
340  * Creates legacy DirectDraw Interfaces. Can't create IDirectDraw7
341  * interfaces in theory
342  *
343  * Arguments, return values: See DDRAW_Create
344  *
345  ***********************************************************************/
346 HRESULT WINAPI
347 DirectDrawCreate(GUID *GUID,
348                  LPDIRECTDRAW *DD,
349                  IUnknown *UnkOuter)
350 {
351     HRESULT hr;
352     TRACE("(%s,%p,%p)\n", debugstr_guid(GUID), DD, UnkOuter);
353
354     EnterCriticalSection(&ddraw_cs);
355     hr = DDRAW_Create(GUID, (void **) DD, UnkOuter, &IID_IDirectDraw);
356     LeaveCriticalSection(&ddraw_cs);
357     return hr;
358 }
359
360 /***********************************************************************
361  * DirectDrawCreateEx (DDRAW.@)
362  *
363  * Only creates new IDirectDraw7 interfaces, supposed to fail if legacy
364  * interfaces are requested.
365  *
366  * Arguments, return values: See DDRAW_Create
367  *
368  ***********************************************************************/
369 HRESULT WINAPI
370 DirectDrawCreateEx(GUID *GUID,
371                    LPVOID *DD,
372                    REFIID iid,
373                    IUnknown *UnkOuter)
374 {
375     HRESULT hr;
376     TRACE("(%s,%p,%s,%p)\n", debugstr_guid(GUID), DD, debugstr_guid(iid), UnkOuter);
377
378     if (!IsEqualGUID(iid, &IID_IDirectDraw7))
379         return DDERR_INVALIDPARAMS;
380
381     EnterCriticalSection(&ddraw_cs);
382     hr = DDRAW_Create(GUID, DD, UnkOuter, iid);
383     LeaveCriticalSection(&ddraw_cs);
384     return hr;
385 }
386
387 /***********************************************************************
388  * DirectDrawEnumerateA (DDRAW.@)
389  *
390  * Enumerates legacy ddraw drivers, ascii version. We only have one
391  * driver, which relays to WineD3D. If we were sufficiently cool,
392  * we could offer various interfaces, which use a different default surface
393  * implementation, but I think it's better to offer this choice in
394  * winecfg, because some apps use the default driver, so we would need
395  * a winecfg option anyway, and there shouldn't be 2 ways to set one setting
396  *
397  * Arguments:
398  *  Callback: Callback function from the app
399  *  Context: Argument to the call back.
400  *
401  * Returns:
402  *  DD_OK on success
403  *  E_INVALIDARG if the Callback caused a page fault
404  *
405  *
406  ***********************************************************************/
407 HRESULT WINAPI
408 DirectDrawEnumerateA(LPDDENUMCALLBACKA Callback,
409                      LPVOID Context)
410 {
411     BOOL stop = FALSE;
412
413     TRACE(" Enumerating default DirectDraw HAL interface\n");
414     /* We only have one driver */
415     __TRY
416     {
417         static CHAR driver_desc[] = "DirectDraw HAL",
418         driver_name[] = "display";
419
420         stop = !Callback(NULL, driver_desc, driver_name, Context);
421     }
422     __EXCEPT_PAGE_FAULT
423     {
424         return E_INVALIDARG;
425     }
426     __ENDTRY
427
428     TRACE(" End of enumeration\n");
429     return DD_OK;
430 }
431
432 /***********************************************************************
433  * DirectDrawEnumerateExA (DDRAW.@)
434  *
435  * Enumerates DirectDraw7 drivers, ascii version. See
436  * the comments above DirectDrawEnumerateA for more details.
437  *
438  * The Flag member is not supported right now.
439  *
440  ***********************************************************************/
441 HRESULT WINAPI
442 DirectDrawEnumerateExA(LPDDENUMCALLBACKEXA Callback,
443                        LPVOID Context,
444                        DWORD Flags)
445 {
446     BOOL stop = FALSE;
447     TRACE("Enumerating default DirectDraw HAL interface\n");
448
449     /* We only have one driver by now */
450     __TRY
451     {
452         static CHAR driver_desc[] = "DirectDraw HAL",
453         driver_name[] = "display";
454
455         /* QuickTime expects the description "DirectDraw HAL" */
456         stop = !Callback(NULL, driver_desc, driver_name, Context, 0);
457     }
458     __EXCEPT_PAGE_FAULT
459     {
460         return E_INVALIDARG;
461     }
462     __ENDTRY;
463
464     TRACE("End of enumeration\n");
465     return DD_OK;
466 }
467
468 /***********************************************************************
469  * DirectDrawEnumerateW (DDRAW.@)
470  *
471  * Enumerates legacy drivers, unicode version. See
472  * the comments above DirectDrawEnumerateA for more details.
473  *
474  * The Flag member is not supported right now.
475  *
476  ***********************************************************************/
477
478 /***********************************************************************
479  * DirectDrawEnumerateExW (DDRAW.@)
480  *
481  * Enumerates DirectDraw7 drivers, unicode version. See
482  * the comments above DirectDrawEnumerateA for more details.
483  *
484  * The Flag member is not supported right now.
485  *
486  ***********************************************************************/
487
488 /***********************************************************************
489  * Classfactory implementation.
490  ***********************************************************************/
491
492 /***********************************************************************
493  * CF_CreateDirectDraw
494  *
495  * DDraw creation function for the class factory
496  *
497  * Params:
498  *  UnkOuter: Set to NULL
499  *  iid: ID of the wanted interface
500  *  obj: Address to pass the interface pointer back
501  *
502  * Returns
503  *  DD_OK / DDERR*, see DDRAW_Create
504  *
505  ***********************************************************************/
506 static HRESULT
507 CF_CreateDirectDraw(IUnknown* UnkOuter, REFIID iid,
508                     void **obj)
509 {
510     HRESULT hr;
511
512     TRACE("(%p,%s,%p)\n", UnkOuter, debugstr_guid(iid), obj);
513
514     EnterCriticalSection(&ddraw_cs);
515     hr = DDRAW_Create(NULL, obj, UnkOuter, iid);
516     LeaveCriticalSection(&ddraw_cs);
517     return hr;
518 }
519
520 /***********************************************************************
521  * CF_CreateDirectDraw
522  *
523  * Clipper creation function for the class factory
524  *
525  * Params:
526  *  UnkOuter: Set to NULL
527  *  iid: ID of the wanted interface
528  *  obj: Address to pass the interface pointer back
529  *
530  * Returns
531  *  DD_OK / DDERR*, see DDRAW_Create
532  *
533  ***********************************************************************/
534 static HRESULT
535 CF_CreateDirectDrawClipper(IUnknown* UnkOuter, REFIID riid,
536                               void **obj)
537 {
538     HRESULT hr;
539     IDirectDrawClipper *Clip;
540
541     EnterCriticalSection(&ddraw_cs);
542     hr = DirectDrawCreateClipper(0, &Clip, UnkOuter);
543     if (hr != DD_OK)
544     {
545         LeaveCriticalSection(&ddraw_cs);
546         return hr;
547     }
548
549     hr = IDirectDrawClipper_QueryInterface(Clip, riid, obj);
550     IDirectDrawClipper_Release(Clip);
551
552     LeaveCriticalSection(&ddraw_cs);
553     return hr;
554 }
555
556 static const struct object_creation_info object_creation[] =
557 {
558     { &CLSID_DirectDraw,        CF_CreateDirectDraw },
559     { &CLSID_DirectDraw7,       CF_CreateDirectDraw },
560     { &CLSID_DirectDrawClipper, CF_CreateDirectDrawClipper }
561 };
562
563 /*******************************************************************************
564  * IDirectDrawClassFactory::QueryInterface
565  *
566  * QueryInterface for the class factory
567  *
568  * PARAMS
569  *    riid   Reference to identifier of queried interface
570  *    ppv    Address to return the interface pointer at
571  *
572  * RETURNS
573  *    Success: S_OK
574  *    Failure: E_NOINTERFACE
575  *
576  *******************************************************************************/
577 static HRESULT WINAPI
578 IDirectDrawClassFactoryImpl_QueryInterface(IClassFactory *iface,
579                     REFIID riid,
580                     void **obj)
581 {
582     ICOM_THIS_FROM(IClassFactoryImpl, IClassFactory, iface);
583
584     TRACE("(%p)->(%s,%p)\n", This, debugstr_guid(riid), obj);
585
586     if (IsEqualGUID(riid, &IID_IUnknown)
587         || IsEqualGUID(riid, &IID_IClassFactory))
588     {
589         IClassFactory_AddRef(iface);
590         *obj = This;
591         return S_OK;
592     }
593
594     WARN("(%p)->(%s,%p),not found\n",This,debugstr_guid(riid),obj);
595     return E_NOINTERFACE;
596 }
597
598 /*******************************************************************************
599  * IDirectDrawClassFactory::AddRef
600  *
601  * AddRef for the class factory
602  *
603  * RETURNS
604  *  The new refcount
605  *
606  *******************************************************************************/
607 static ULONG WINAPI
608 IDirectDrawClassFactoryImpl_AddRef(IClassFactory *iface)
609 {
610     ICOM_THIS_FROM(IClassFactoryImpl, IClassFactory, iface);
611     ULONG ref = InterlockedIncrement(&This->ref);
612
613     TRACE("(%p)->() incrementing from %d.\n", This, ref - 1);
614
615     return ref;
616 }
617
618 /*******************************************************************************
619  * IDirectDrawClassFactory::Release
620  *
621  * Release for the class factory. If the refcount falls to 0, the object
622  * is destroyed
623  *
624  * RETURNS
625  *  The new refcount
626  *
627  *******************************************************************************/
628 static ULONG WINAPI
629 IDirectDrawClassFactoryImpl_Release(IClassFactory *iface)
630 {
631     ICOM_THIS_FROM(IClassFactoryImpl, IClassFactory, iface);
632     ULONG ref = InterlockedDecrement(&This->ref);
633     TRACE("(%p)->() decrementing from %d.\n", This, ref+1);
634
635     if (ref == 0)
636         HeapFree(GetProcessHeap(), 0, This);
637
638     return ref;
639 }
640
641
642 /*******************************************************************************
643  * IDirectDrawClassFactory::CreateInstance
644  *
645  * What is this? Seems to create DirectDraw objects...
646  *
647  * Params
648  *  The ususal things???
649  *
650  * RETURNS
651  *  ???
652  *
653  *******************************************************************************/
654 static HRESULT WINAPI
655 IDirectDrawClassFactoryImpl_CreateInstance(IClassFactory *iface,
656                                            IUnknown *UnkOuter,
657                                            REFIID riid,
658                                            void **obj)
659 {
660     ICOM_THIS_FROM(IClassFactoryImpl, IClassFactory, iface);
661
662     TRACE("(%p)->(%p,%s,%p)\n",This,UnkOuter,debugstr_guid(riid),obj);
663
664     return This->pfnCreateInstance(UnkOuter, riid, obj);
665 }
666
667 /*******************************************************************************
668  * IDirectDrawClassFactory::LockServer
669  *
670  * What is this?
671  *
672  * Params
673  *  ???
674  *
675  * RETURNS
676  *  S_OK, because it's a stub
677  *
678  *******************************************************************************/
679 static HRESULT WINAPI
680 IDirectDrawClassFactoryImpl_LockServer(IClassFactory *iface,BOOL dolock)
681 {
682     ICOM_THIS_FROM(IClassFactoryImpl, IClassFactory, iface);
683     FIXME("(%p)->(%d),stub!\n",This,dolock);
684     return S_OK;
685 }
686
687 /*******************************************************************************
688  * The class factory VTable
689  *******************************************************************************/
690 static const IClassFactoryVtbl IClassFactory_Vtbl =
691 {
692     IDirectDrawClassFactoryImpl_QueryInterface,
693     IDirectDrawClassFactoryImpl_AddRef,
694     IDirectDrawClassFactoryImpl_Release,
695     IDirectDrawClassFactoryImpl_CreateInstance,
696     IDirectDrawClassFactoryImpl_LockServer
697 };
698
699 /*******************************************************************************
700  * DllGetClassObject [DDRAW.@]
701  * Retrieves class object from a DLL object
702  *
703  * NOTES
704  *    Docs say returns STDAPI
705  *
706  * PARAMS
707  *    rclsid [I] CLSID for the class object
708  *    riid   [I] Reference to identifier of interface for class object
709  *    ppv    [O] Address of variable to receive interface pointer for riid
710  *
711  * RETURNS
712  *    Success: S_OK
713  *    Failure: CLASS_E_CLASSNOTAVAILABLE, E_OUTOFMEMORY, E_INVALIDARG,
714  *             E_UNEXPECTED
715  */
716 HRESULT WINAPI DllGetClassObject(REFCLSID rclsid, REFIID riid, LPVOID *ppv)
717 {
718     unsigned int i;
719     IClassFactoryImpl *factory;
720
721     TRACE("(%s,%s,%p)\n", debugstr_guid(rclsid), debugstr_guid(riid), ppv);
722
723     if ( !IsEqualGUID( &IID_IClassFactory, riid )
724          && ! IsEqualGUID( &IID_IUnknown, riid) )
725         return E_NOINTERFACE;
726
727     for (i=0; i < sizeof(object_creation)/sizeof(object_creation[0]); i++)
728     {
729         if (IsEqualGUID(object_creation[i].clsid, rclsid))
730             break;
731     }
732
733     if (i == sizeof(object_creation)/sizeof(object_creation[0]))
734     {
735         FIXME("%s: no class found.\n", debugstr_guid(rclsid));
736         return CLASS_E_CLASSNOTAVAILABLE;
737     }
738
739     factory = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(*factory));
740     if (factory == NULL) return E_OUTOFMEMORY;
741
742     ICOM_INIT_INTERFACE(factory, IClassFactory, IClassFactory_Vtbl);
743     factory->ref = 1;
744
745     factory->pfnCreateInstance = object_creation[i].pfnCreateInstance;
746
747     *ppv = ICOM_INTERFACE(factory, IClassFactory);
748     return S_OK;
749 }
750
751
752 /*******************************************************************************
753  * DllCanUnloadNow [DDRAW.@]  Determines whether the DLL is in use.
754  *
755  * RETURNS
756  *    Success: S_OK
757  *    Failure: S_FALSE
758  */
759 HRESULT WINAPI DllCanUnloadNow(void)
760 {
761     HRESULT hr;
762     FIXME("(void): stub\n");
763
764     EnterCriticalSection(&ddraw_cs);
765     hr = S_FALSE;
766     LeaveCriticalSection(&ddraw_cs);
767
768     return hr;
769 }
770
771 /*******************************************************************************
772  * DestroyCallback
773  *
774  * Callback function for the EnumSurfaces call in DllMain.
775  * Dumps some surface info and releases the surface
776  *
777  * Params:
778  *  surf: The enumerated surface
779  *  desc: it's description
780  *  context: Pointer to the ddraw impl
781  *
782  * Returns:
783  *  DDENUMRET_OK;
784  *******************************************************************************/
785 static HRESULT WINAPI
786 DestroyCallback(IDirectDrawSurface7 *surf,
787                 DDSURFACEDESC2 *desc,
788                 void *context)
789 {
790     IDirectDrawSurfaceImpl *Impl = ICOM_OBJECT(IDirectDrawSurfaceImpl, IDirectDrawSurface7, surf);
791     IDirectDrawImpl *ddraw = (IDirectDrawImpl *) context;
792     ULONG ref;
793
794     ref = IDirectDrawSurface7_Release(surf);  /* For the EnumSurfaces */
795     WARN("Surface %p has an reference count of %d\n", Impl, ref);
796
797     /* Skip surfaces which are attached somewhere or which are
798      * part of a complex compound. They will get released when destroying
799      * the root
800      */
801     if( (!Impl->is_complex_root) || (Impl->first_attached != Impl) )
802         return DDENUMRET_OK;
803     /* Skip our depth stencil surface, it will be released with the render target */
804     if( Impl == ddraw->DepthStencilBuffer)
805         return DDENUMRET_OK;
806
807     /* Destroy the surface */
808     while(ref) ref = IDirectDrawSurface7_Release(surf);
809
810     return DDENUMRET_OK;
811 }
812
813 /***********************************************************************
814  * get_config_key
815  *
816  * Reads a config key from the registry. Taken from WineD3D
817  *
818  ***********************************************************************/
819 static inline DWORD get_config_key(HKEY defkey, HKEY appkey, const char* name, char* buffer, DWORD size)
820 {
821     if (0 != appkey && !RegQueryValueExA( appkey, name, 0, NULL, (LPBYTE) buffer, &size )) return 0;
822     if (0 != defkey && !RegQueryValueExA( defkey, name, 0, NULL, (LPBYTE) buffer, &size )) return 0;
823     return ERROR_FILE_NOT_FOUND;
824 }
825
826 /***********************************************************************
827  * DllMain (DDRAW.0)
828  *
829  * Could be used to register DirectDraw drivers, if we have more than
830  * one. Also used to destroy any objects left at unload if the
831  * app didn't release them properly(Gothic 2, Diablo 2, Moto racer, ...)
832  *
833  ***********************************************************************/
834 BOOL WINAPI
835 DllMain(HINSTANCE hInstDLL,
836         DWORD Reason,
837         LPVOID lpv)
838 {
839     TRACE("(%p,%x,%p)\n", hInstDLL, Reason, lpv);
840     if (Reason == DLL_PROCESS_ATTACH)
841     {
842         char buffer[MAX_PATH+10];
843         DWORD size = sizeof(buffer);
844         HKEY hkey = 0;
845         HKEY appkey = 0;
846         DWORD len;
847
848        /* @@ Wine registry key: HKCU\Software\Wine\Direct3D */
849        if ( RegOpenKeyA( HKEY_CURRENT_USER, "Software\\Wine\\Direct3D", &hkey ) ) hkey = 0;
850
851        len = GetModuleFileNameA( 0, buffer, MAX_PATH );
852        if (len && len < MAX_PATH)
853        {
854             HKEY tmpkey;
855             /* @@ Wine registry key: HKCU\Software\Wine\AppDefaults\app.exe\Direct3D */
856             if (!RegOpenKeyA( HKEY_CURRENT_USER, "Software\\Wine\\AppDefaults", &tmpkey ))
857             {
858                 char *p, *appname = buffer;
859                 if ((p = strrchr( appname, '/' ))) appname = p + 1;
860                 if ((p = strrchr( appname, '\\' ))) appname = p + 1;
861                 strcat( appname, "\\Direct3D" );
862                 TRACE("appname = [%s]\n", appname);
863                 if (RegOpenKeyA( tmpkey, appname, &appkey )) appkey = 0;
864                 RegCloseKey( tmpkey );
865             }
866        }
867
868        if ( 0 != hkey || 0 != appkey )
869        {
870             if ( !get_config_key( hkey, appkey, "DirectDrawRenderer", buffer, size) )
871             {
872                 if (!strcmp(buffer,"gdi"))
873                 {
874                     TRACE("Defaulting to GDI surfaces\n");
875                     DefaultSurfaceType = SURFACE_GDI;
876                 }
877                 else if (!strcmp(buffer,"opengl"))
878                 {
879                     TRACE("Defaulting to opengl surfaces\n");
880                     DefaultSurfaceType = SURFACE_OPENGL;
881                 }
882                 else
883                 {
884                     ERR("Unknown default surface type. Supported are:\n gdi, opengl\n");
885                 }
886             }
887         }
888
889         DisableThreadLibraryCalls(hInstDLL);
890     }
891     else if (Reason == DLL_PROCESS_DETACH)
892     {
893         if(!list_empty(&global_ddraw_list))
894         {
895             struct list *entry, *entry2;
896             WARN("There are still existing DirectDraw interfaces. Wine bug or buggy application?\n");
897
898             /* We remove elemets from this loop */
899             LIST_FOR_EACH_SAFE(entry, entry2, &global_ddraw_list)
900             {
901                 HRESULT hr;
902                 DDSURFACEDESC2 desc;
903                 int i;
904                 IDirectDrawImpl *ddraw = LIST_ENTRY(entry, IDirectDrawImpl, ddraw_list_entry);
905
906                 WARN("DDraw %p has a refcount of %d\n", ddraw, ddraw->ref7 + ddraw->ref4 + ddraw->ref3 + ddraw->ref2 + ddraw->ref1);
907
908                 /* Add references to each interface to avoid freeing them unexpectadely */
909                 IDirectDraw_AddRef(ICOM_INTERFACE(ddraw, IDirectDraw));
910                 IDirectDraw2_AddRef(ICOM_INTERFACE(ddraw, IDirectDraw2));
911                 IDirectDraw3_AddRef(ICOM_INTERFACE(ddraw, IDirectDraw3));
912                 IDirectDraw4_AddRef(ICOM_INTERFACE(ddraw, IDirectDraw4));
913                 IDirectDraw7_AddRef(ICOM_INTERFACE(ddraw, IDirectDraw7));
914
915                 /* Does a D3D device exist? Destroy it
916                     * TODO: Destroy all Vertex buffers, Lights, Materials
917                     * and execture buffers too
918                     */
919                 if(ddraw->d3ddevice)
920                 {
921                     WARN("DDraw %p has d3ddevice %p attached\n", ddraw, ddraw->d3ddevice);
922                     while(IDirect3DDevice7_Release(ICOM_INTERFACE(ddraw->d3ddevice, IDirect3DDevice7)));
923                 }
924
925                 /* Try to release the objects
926                     * Do an EnumSurfaces to find any hanging surfaces
927                     */
928                 memset(&desc, 0, sizeof(desc));
929                 desc.dwSize = sizeof(desc);
930                 for(i = 0; i <= 1; i++)
931                 {
932                     hr = IDirectDraw7_EnumSurfaces(ICOM_INTERFACE(ddraw, IDirectDraw7),
933                                                     DDENUMSURFACES_ALL,
934                                                     &desc,
935                                                     (void *) ddraw,
936                                                     DestroyCallback);
937                     if(hr != D3D_OK)
938                         ERR("(%p) EnumSurfaces failed, prepare for trouble\n", ddraw);
939                 }
940
941                 /* Check the surface count */
942                 if(ddraw->surfaces > 0)
943                     ERR("DDraw %p still has %d surfaces attached\n", ddraw, ddraw->surfaces);
944
945                 /* Release all hanging references to destroy the objects. This
946                     * restores the screen mode too
947                     */
948                 while(IDirectDraw_Release(ICOM_INTERFACE(ddraw, IDirectDraw)));
949                 while(IDirectDraw2_Release(ICOM_INTERFACE(ddraw, IDirectDraw2)));
950                 while(IDirectDraw3_Release(ICOM_INTERFACE(ddraw, IDirectDraw3)));
951                 while(IDirectDraw4_Release(ICOM_INTERFACE(ddraw, IDirectDraw4)));
952                 while(IDirectDraw7_Release(ICOM_INTERFACE(ddraw, IDirectDraw7)));
953             }
954         }
955     }
956
957     return TRUE;
958 }