2 * IDxDiagContainer Implementation
4 * Copyright 2004 Raphael Junqueira
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.
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.
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
25 #include "dxdiag_private.h"
26 #include "wine/debug.h"
27 #include "wine/unicode.h"
29 WINE_DEFAULT_DEBUG_CHANNEL(dxdiag
);
31 /* IDxDiagContainer IUnknown parts follow: */
32 HRESULT WINAPI
IDxDiagContainerImpl_QueryInterface(PDXDIAGCONTAINER iface
, REFIID riid
, LPVOID
*ppobj
)
34 IDxDiagContainerImpl
*This
= (IDxDiagContainerImpl
*)iface
;
36 if (!ppobj
) return E_INVALIDARG
;
38 if (IsEqualGUID(riid
, &IID_IUnknown
)
39 || IsEqualGUID(riid
, &IID_IDxDiagContainer
)) {
40 IUnknown_AddRef(iface
);
45 WARN("(%p)->(%s,%p),not found\n",This
,debugstr_guid(riid
),ppobj
);
50 static ULONG WINAPI
IDxDiagContainerImpl_AddRef(PDXDIAGCONTAINER iface
) {
51 IDxDiagContainerImpl
*This
= (IDxDiagContainerImpl
*)iface
;
52 ULONG refCount
= InterlockedIncrement(&This
->ref
);
54 TRACE("(%p)->(ref before=%u)\n", This
, refCount
- 1);
61 static ULONG WINAPI
IDxDiagContainerImpl_Release(PDXDIAGCONTAINER iface
) {
62 IDxDiagContainerImpl
*This
= (IDxDiagContainerImpl
*)iface
;
63 ULONG refCount
= InterlockedDecrement(&This
->ref
);
65 TRACE("(%p)->(ref before=%u)\n", This
, refCount
+ 1);
68 HeapFree(GetProcessHeap(), 0, This
);
71 DXDIAGN_UnlockModule();
76 /* IDxDiagContainer Interface follow: */
77 static HRESULT WINAPI
IDxDiagContainerImpl_GetNumberOfChildContainers(PDXDIAGCONTAINER iface
, DWORD
* pdwCount
) {
78 IDxDiagContainerImpl
*This
= (IDxDiagContainerImpl
*)iface
;
79 TRACE("(%p)\n", iface
);
80 if (NULL
== pdwCount
) {
83 *pdwCount
= This
->nSubContainers
;
87 static HRESULT WINAPI
IDxDiagContainerImpl_EnumChildContainerNames(PDXDIAGCONTAINER iface
, DWORD dwIndex
, LPWSTR pwszContainer
, DWORD cchContainer
) {
88 IDxDiagContainerImpl
*This
= (IDxDiagContainerImpl
*)iface
;
89 IDxDiagContainerImpl_SubContainer
* p
= NULL
;
92 TRACE("(%p, %u, %p, %u)\n", iface
, dwIndex
, pwszContainer
, cchContainer
);
94 if (NULL
== pwszContainer
|| 0 == cchContainer
) {
98 p
= This
->subContainers
;
101 TRACE("Found container name %s, copying string\n", debugstr_w(p
->contName
));
102 lstrcpynW(pwszContainer
, p
->contName
, cchContainer
);
103 return (cchContainer
<= strlenW(p
->contName
)) ?
104 DXDIAG_E_INSUFFICIENT_BUFFER
: S_OK
;
110 TRACE("Failed to find container name at specified index\n");
111 *pwszContainer
= '\0';
115 static HRESULT
IDxDiagContainerImpl_GetChildContainerInternal(PDXDIAGCONTAINER iface
, LPCWSTR pwszContainer
, IDxDiagContainer
** ppInstance
) {
116 IDxDiagContainerImpl
*This
= (IDxDiagContainerImpl
*)iface
;
117 IDxDiagContainerImpl_SubContainer
* p
= NULL
;
119 p
= This
->subContainers
;
121 if (0 == lstrcmpW(p
->contName
, pwszContainer
)) {
122 *ppInstance
= p
->pCont
;
130 static HRESULT WINAPI
IDxDiagContainerImpl_GetChildContainer(PDXDIAGCONTAINER iface
, LPCWSTR pwszContainer
, IDxDiagContainer
** ppInstance
) {
131 IDxDiagContainerImpl
*This
= (IDxDiagContainerImpl
*)iface
;
132 IDxDiagContainer
* pContainer
= NULL
;
133 LPWSTR tmp
, orig_tmp
;
136 HRESULT hr
= E_INVALIDARG
;
138 TRACE("(%p, %s, %p)\n", iface
, debugstr_w(pwszContainer
), ppInstance
);
140 if (NULL
== ppInstance
|| NULL
== pwszContainer
) {
144 pContainer
= (PDXDIAGCONTAINER
) This
;
146 tmp_len
= strlenW(pwszContainer
) + 1;
147 orig_tmp
= tmp
= HeapAlloc(GetProcessHeap(), 0, tmp_len
* sizeof(WCHAR
));
148 if (NULL
== tmp
) return E_FAIL
;
149 lstrcpynW(tmp
, pwszContainer
, tmp_len
);
151 cur
= strchrW(tmp
, '.');
152 while (NULL
!= cur
) {
153 *cur
= '\0'; /* cut tmp string to '.' */
154 hr
= IDxDiagContainerImpl_GetChildContainerInternal(pContainer
, tmp
, &pContainer
);
155 if (FAILED(hr
) || NULL
== pContainer
)
157 cur
++; /* go after '.' (just replaced by \0) */
159 cur
= strchrW(tmp
, '.');
162 hr
= IDxDiagContainerImpl_GetChildContainerInternal(pContainer
, tmp
, ppInstance
);
164 IDxDiagContainerImpl_AddRef(*ppInstance
);
168 HeapFree(GetProcessHeap(), 0, orig_tmp
);
172 static HRESULT WINAPI
IDxDiagContainerImpl_GetNumberOfProps(PDXDIAGCONTAINER iface
, DWORD
* pdwCount
) {
173 IDxDiagContainerImpl
*This
= (IDxDiagContainerImpl
*)iface
;
174 TRACE("(%p)\n", iface
);
175 if (NULL
== pdwCount
) {
178 *pdwCount
= This
->nProperties
;
182 static HRESULT WINAPI
IDxDiagContainerImpl_EnumPropNames(PDXDIAGCONTAINER iface
, DWORD dwIndex
, LPWSTR pwszPropName
, DWORD cchPropName
) {
183 IDxDiagContainerImpl
*This
= (IDxDiagContainerImpl
*)iface
;
184 IDxDiagContainerImpl_Property
* p
= NULL
;
187 TRACE("(%p, %u, %s, %u)\n", iface
, dwIndex
, debugstr_w(pwszPropName
), cchPropName
);
189 if (NULL
== pwszPropName
) {
192 if (256 > cchPropName
) {
193 return DXDIAG_E_INSUFFICIENT_BUFFER
;
196 p
= This
->properties
;
199 if (cchPropName
<= strlenW(p
->vName
)) {
200 return DXDIAG_E_INSUFFICIENT_BUFFER
;
202 lstrcpynW(pwszPropName
, p
->vName
, cchPropName
);
211 static HRESULT WINAPI
IDxDiagContainerImpl_GetProp(PDXDIAGCONTAINER iface
, LPCWSTR pwszPropName
, VARIANT
* pvarProp
) {
212 IDxDiagContainerImpl
*This
= (IDxDiagContainerImpl
*)iface
;
213 IDxDiagContainerImpl_Property
* p
= NULL
;
215 TRACE("(%p, %s, %p)\n", iface
, debugstr_w(pwszPropName
), pvarProp
);
217 if (NULL
== pvarProp
|| NULL
== pwszPropName
) {
221 p
= This
->properties
;
223 if (0 == lstrcmpW(p
->vName
, pwszPropName
)) {
224 VariantCopy(pvarProp
, &p
->v
);
232 HRESULT WINAPI
IDxDiagContainerImpl_AddProp(PDXDIAGCONTAINER iface
, LPCWSTR pwszPropName
, VARIANT
* pVarProp
) {
233 IDxDiagContainerImpl
*This
= (IDxDiagContainerImpl
*)iface
;
234 IDxDiagContainerImpl_Property
* p
= NULL
;
235 IDxDiagContainerImpl_Property
* pNew
= NULL
;
237 TRACE("(%p, %s, %p)\n", iface
, debugstr_w(pwszPropName
), pVarProp
);
239 if (NULL
== pVarProp
|| NULL
== pwszPropName
) {
243 pNew
= HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY
, sizeof(IDxDiagContainerImpl_Property
));
245 return E_OUTOFMEMORY
;
247 VariantInit(&pNew
->v
);
248 VariantCopy(&pNew
->v
, pVarProp
);
249 pNew
->vName
= HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY
, (lstrlenW(pwszPropName
) + 1) * sizeof(WCHAR
));
250 lstrcpyW(pNew
->vName
, pwszPropName
);
253 p
= This
->properties
;
255 This
->properties
= pNew
;
257 while (NULL
!= p
->next
) {
266 HRESULT WINAPI
IDxDiagContainerImpl_AddChildContainer(PDXDIAGCONTAINER iface
, LPCWSTR pszContName
, PDXDIAGCONTAINER pSubCont
) {
267 IDxDiagContainerImpl
*This
= (IDxDiagContainerImpl
*)iface
;
268 IDxDiagContainerImpl_SubContainer
* p
= NULL
;
269 IDxDiagContainerImpl_SubContainer
* pNew
= NULL
;
271 TRACE("(%p, %s, %p)\n", iface
, debugstr_w(pszContName
), pSubCont
);
273 if (NULL
== pSubCont
|| NULL
== pszContName
) {
277 pNew
= HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY
, sizeof(IDxDiagContainerImpl_SubContainer
));
279 return E_OUTOFMEMORY
;
281 pNew
->pCont
= pSubCont
;
282 pNew
->contName
= HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY
, (lstrlenW(pszContName
) + 1) * sizeof(WCHAR
));
283 lstrcpyW(pNew
->contName
, pszContName
);
286 p
= This
->subContainers
;
288 This
->subContainers
= pNew
;
290 while (NULL
!= p
->next
) {
295 ++This
->nSubContainers
;
299 static const IDxDiagContainerVtbl DxDiagContainer_Vtbl
=
301 IDxDiagContainerImpl_QueryInterface
,
302 IDxDiagContainerImpl_AddRef
,
303 IDxDiagContainerImpl_Release
,
304 IDxDiagContainerImpl_GetNumberOfChildContainers
,
305 IDxDiagContainerImpl_EnumChildContainerNames
,
306 IDxDiagContainerImpl_GetChildContainer
,
307 IDxDiagContainerImpl_GetNumberOfProps
,
308 IDxDiagContainerImpl_EnumPropNames
,
309 IDxDiagContainerImpl_GetProp
313 HRESULT
DXDiag_CreateDXDiagContainer(REFIID riid
, LPVOID
*ppobj
) {
314 IDxDiagContainerImpl
* container
;
316 TRACE("(%p, %p)\n", debugstr_guid(riid
), ppobj
);
318 container
= HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY
, sizeof(IDxDiagContainerImpl
));
319 if (NULL
== container
) {
321 return E_OUTOFMEMORY
;
323 container
->lpVtbl
= &DxDiagContainer_Vtbl
;
324 container
->ref
= 0; /* will be inited with QueryInterface */
325 return IDxDiagContainerImpl_QueryInterface((PDXDIAGCONTAINER
)container
, riid
, ppobj
);