d3dx9_36/tests: Avoid a potential out of bound access.
[wine] / dlls / ddraw / light.c
1 /* Direct3D Light
2  * Copyright (c) 1998 / 2002 Lionel ULMER
3  * Copyright (c) 2006        Stefan DÖSINGER
4  *
5  * This file contains the implementation of Direct3DLight.
6  *
7  * This library is free software; you can redistribute it and/or
8  * modify it under the terms of the GNU Lesser General Public
9  * License as published by the Free Software Foundation; either
10  * version 2.1 of the License, or (at your option) any later version.
11  *
12  * This library is distributed in the hope that it will be useful,
13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
15  * Lesser General Public License for more details.
16  *
17  * You should have received a copy of the GNU Lesser General Public
18  * License along with this library; if not, write to the Free Software
19  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
20  */
21
22 #include "config.h"
23 #include "wine/port.h"
24
25 #include "ddraw_private.h"
26
27 WINE_DEFAULT_DEBUG_CHANNEL(ddraw);
28
29 /*****************************************************************************
30  * light_update
31  *
32  * Updates the Direct3DDevice7 lighting parameters
33  *
34  *****************************************************************************/
35 static void light_update(struct d3d_light *light)
36 {
37     struct d3d_device *device;
38
39     TRACE("light %p.\n", light);
40
41     if (!light->active_viewport || !light->active_viewport->active_device) return;
42     device = light->active_viewport->active_device;
43
44     IDirect3DDevice7_SetLight(&device->IDirect3DDevice7_iface, light->dwLightIndex, &light->light7);
45 }
46
47 /*****************************************************************************
48  * light_activate
49  *
50  * Uses the Direct3DDevice7::LightEnable method to active the light
51  *
52  *****************************************************************************/
53 void light_activate(struct d3d_light *light)
54 {
55     struct d3d_device *device;
56
57     TRACE("light %p.\n", light);
58
59     if (!light->active_viewport || !light->active_viewport->active_device) return;
60     device = light->active_viewport->active_device;
61
62     light_update(light);
63     if (!(light->light.dwFlags & D3DLIGHT_ACTIVE))
64     {
65         IDirect3DDevice7_LightEnable(&device->IDirect3DDevice7_iface, light->dwLightIndex, TRUE);
66         light->light.dwFlags |= D3DLIGHT_ACTIVE;
67     }
68 }
69
70 /*****************************************************************************
71  *
72  * light_deactivate
73  *
74  * Uses the Direct3DDevice7::LightEnable method to deactivate the light
75  *
76  *****************************************************************************/
77 void light_deactivate(struct d3d_light *light)
78 {
79     struct d3d_device *device;
80
81     TRACE("light %p.\n", light);
82
83     if (!light->active_viewport || !light->active_viewport->active_device) return;
84     device = light->active_viewport->active_device;
85
86     /* If was not active, activate it */
87     if (light->light.dwFlags & D3DLIGHT_ACTIVE)
88     {
89         IDirect3DDevice7_LightEnable(&device->IDirect3DDevice7_iface, light->dwLightIndex, FALSE);
90         light->light.dwFlags &= ~D3DLIGHT_ACTIVE;
91     }
92 }
93
94 static inline struct d3d_light *impl_from_IDirect3DLight(IDirect3DLight *iface)
95 {
96     return CONTAINING_RECORD(iface, struct d3d_light, IDirect3DLight_iface);
97 }
98
99 /*****************************************************************************
100  * IDirect3DLight::QueryInterface
101  *
102  * Queries the object for different interfaces. Unimplemented for this
103  * object at the moment
104  *
105  * Params:
106  *  riid: Interface id asked for
107  *  obj: Address to return the resulting pointer at.
108  *
109  * Returns:
110  *  E_NOINTERFACE, because it's a stub
111  *****************************************************************************/
112 static HRESULT WINAPI d3d_light_QueryInterface(IDirect3DLight *iface, REFIID riid, void **object)
113 {
114     FIXME("iface %p, riid %s, object %p stub!\n", iface, debugstr_guid(riid), object);
115
116     *object = NULL;
117     return E_NOINTERFACE;
118 }
119
120 static ULONG WINAPI d3d_light_AddRef(IDirect3DLight *iface)
121 {
122     struct d3d_light *light = impl_from_IDirect3DLight(iface);
123     ULONG ref = InterlockedIncrement(&light->ref);
124
125     TRACE("%p increasing refcount to %u.\n", light, ref);
126
127     return ref;
128 }
129
130 static ULONG WINAPI d3d_light_Release(IDirect3DLight *iface)
131 {
132     struct d3d_light *light = impl_from_IDirect3DLight(iface);
133     ULONG ref = InterlockedDecrement(&light->ref);
134
135     TRACE("%p decreasing refcount to %u.\n", light, ref);
136
137     if (!ref)
138     {
139         HeapFree(GetProcessHeap(), 0, light);
140         return 0;
141     }
142     return ref;
143 }
144
145 /*****************************************************************************
146  * IDirect3DLight Methods.
147  *****************************************************************************/
148
149 /*****************************************************************************
150  * IDirect3DLight::Initialize
151  *
152  * Initializes the interface. This implementation is a no-op, because
153  * initialization takes place at creation time
154  *
155  * Params:
156  *  Direct3D: Pointer to an IDirect3D interface.
157  *
158  * Returns:
159  *  D3D_OK
160  *
161  *****************************************************************************/
162 static HRESULT WINAPI d3d_light_Initialize(IDirect3DLight *iface, IDirect3D *d3d)
163 {
164     TRACE("iface %p, d3d %p.\n", iface, d3d);
165
166     return D3D_OK;
167 }
168
169 /*****************************************************************************
170  * IDirect3DLight::SetLight
171  *
172  * Assigns a lighting value to this object
173  *
174  * Params:
175  *  Light: Lighting parameter to set
176  *
177  * Returns:
178  *  D3D_OK on success
179  *  DDERR_INVALIDPARAMS if Light is NULL
180  *
181  *****************************************************************************/
182 static void dump_light(const D3DLIGHT2 *light)
183 {
184     TRACE("    - dwSize : %d\n", light->dwSize);
185 }
186
187 static const float zero_value[] = {
188     0.0, 0.0, 0.0, 0.0
189 };
190
191 static HRESULT WINAPI d3d_light_SetLight(IDirect3DLight *iface, D3DLIGHT *lpLight)
192 {
193     struct d3d_light *light = impl_from_IDirect3DLight(iface);
194     D3DLIGHT7 *light7 = &light->light7;
195
196     TRACE("iface %p, light %p.\n", iface, lpLight);
197
198     if (TRACE_ON(ddraw))
199     {
200         TRACE("  Light definition :\n");
201         dump_light((LPD3DLIGHT2) lpLight);
202     }
203
204     if ( (lpLight->dltType == 0) || (lpLight->dltType > D3DLIGHT_PARALLELPOINT) )
205          return DDERR_INVALIDPARAMS;
206
207     if ( lpLight->dltType == D3DLIGHT_PARALLELPOINT )
208         FIXME("D3DLIGHT_PARALLELPOINT not supported\n");
209
210     /* Translate D3DLIGH2 structure to D3DLIGHT7 */
211     light7->dltType        = lpLight->dltType;
212     light7->dcvDiffuse     = lpLight->dcvColor;
213     if ((((LPD3DLIGHT2)lpLight)->dwFlags & D3DLIGHT_NO_SPECULAR) != 0)
214       light7->dcvSpecular    = lpLight->dcvColor;
215     else
216       light7->dcvSpecular    = *(const D3DCOLORVALUE*)zero_value;
217     light7->dcvAmbient     = lpLight->dcvColor;
218     light7->dvPosition     = lpLight->dvPosition;
219     light7->dvDirection    = lpLight->dvDirection;
220     light7->dvRange        = lpLight->dvRange;
221     light7->dvFalloff      = lpLight->dvFalloff;
222     light7->dvAttenuation0 = lpLight->dvAttenuation0;
223     light7->dvAttenuation1 = lpLight->dvAttenuation1;
224     light7->dvAttenuation2 = lpLight->dvAttenuation2;
225     light7->dvTheta        = lpLight->dvTheta;
226     light7->dvPhi          = lpLight->dvPhi;
227
228     wined3d_mutex_lock();
229     memcpy(&light->light, lpLight, lpLight->dwSize);
230     if (light->light.dwFlags & D3DLIGHT_ACTIVE)
231         light_update(light);
232     wined3d_mutex_unlock();
233
234     return D3D_OK;
235 }
236
237 /*****************************************************************************
238  * IDirect3DLight::GetLight
239  *
240  * Returns the parameters currently assigned to the IDirect3DLight object
241  *
242  * Params:
243  *  Light: Pointer to an D3DLIGHT structure to store the parameters
244  *
245  * Returns:
246  *  D3D_OK on success
247  *  DDERR_INVALIDPARAMS if Light is NULL
248  *****************************************************************************/
249 static HRESULT WINAPI d3d_light_GetLight(IDirect3DLight *iface, D3DLIGHT *lpLight)
250 {
251     struct d3d_light *light = impl_from_IDirect3DLight(iface);
252
253     TRACE("iface %p, light %p.\n", iface, lpLight);
254
255     if (TRACE_ON(ddraw))
256     {
257         TRACE("  Returning light definition :\n");
258         dump_light(&light->light);
259     }
260
261     wined3d_mutex_lock();
262     memcpy(lpLight, &light->light, lpLight->dwSize);
263     wined3d_mutex_unlock();
264
265     return DD_OK;
266 }
267
268 static const struct IDirect3DLightVtbl d3d_light_vtbl =
269 {
270     /*** IUnknown Methods ***/
271     d3d_light_QueryInterface,
272     d3d_light_AddRef,
273     d3d_light_Release,
274     /*** IDirect3DLight Methods ***/
275     d3d_light_Initialize,
276     d3d_light_SetLight,
277     d3d_light_GetLight
278 };
279
280 void d3d_light_init(struct d3d_light *light, struct ddraw *ddraw)
281 {
282     light->IDirect3DLight_iface.lpVtbl = &d3d_light_vtbl;
283     light->ref = 1;
284     light->ddraw = ddraw;
285 }
286
287 struct d3d_light *unsafe_impl_from_IDirect3DLight(IDirect3DLight *iface)
288 {
289     if (!iface)
290         return NULL;
291     assert(iface->lpVtbl == &d3d_light_vtbl);
292
293     return impl_from_IDirect3DLight(iface);
294 }