1 /*
2 * DxDiag information collection
3 *
4 * Copyright 2011 Andrew Nguyen
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 #define WIN32_LEAN_AND_MEAN
22 #include <windows.h>
23 #include <dxdiag.h>
24
25 #include "wine/debug.h"
26 #include "wine/unicode.h"
27
28 #include "dxdiag_private.h"
29
30 WINE_DEFAULT_DEBUG_CHANNEL(dxdiag);
31
32 static const WCHAR szTimeEnglish[] = {'s','z','T','i','m','e','E','n','g','l','i','s','h',0};
33 static const WCHAR szTimeLocalized[] = {'s','z','T','i','m','e','L','o','c','a','l','i','z','e','d',0};
34 static const WCHAR szMachineNameEnglish[] = {'s','z','M','a','c','h','i','n','e','N','a','m','e','E','n','g','l','i','s','h',0};
35 static const WCHAR szOSExLongEnglish[] = {'s','z','O','S','E','x','L','o','n','g','E','n','g','l','i','s','h',0};
36 static const WCHAR szOSExLocalized[] = {'s','z','O','S','E','x','L','o','c','a','l','i','z','e','d',0};
37 static const WCHAR szLanguagesEnglish[] = {'s','z','L','a','n','g','u','a','g','e','s','E','n','g','l','i','s','h',0};
38 static const WCHAR szLanguagesLocalized[] = {'s','z','L','a','n','g','u','a','g','e','s','L','o','c','a','l','i','z','e','d',0};
39 static const WCHAR szSystemManufacturerEnglish[] = {'s','z','S','y','s','t','e','m','M','a','n','u','f','a','c','t','u','r','e','r','E','n','g','l','i','s','h',0};
40 static const WCHAR szSystemModelEnglish[] = {'s','z','S','y','s','t','e','m','M','o','d','e','l','E','n','g','l','i','s','h',0};
41 static const WCHAR szBIOSEnglish[] = {'s','z','B','I','O','S','E','n','g','l','i','s','h',0};
42 static const WCHAR szProcessorEnglish[] = {'s','z','P','r','o','c','e','s','s','o','r','E','n','g','l','i','s','h',0};
43 static const WCHAR szPhysicalMemoryEnglish[] = {'s','z','P','h','y','s','i','c','a','l','M','e','m','o','r','y','E','n','g','l','i','s','h',0};
44 static const WCHAR szPageFileEnglish[] = {'s','z','P','a','g','e','F','i','l','e','E','n','g','l','i','s','h',0};
45 static const WCHAR szPageFileLocalized[] = {'s','z','P','a','g','e','F','i','l','e','L','o','c','a','l','i','z','e','d',0};
46 static const WCHAR szWindowsDir[] = {'s','z','W','i','n','d','o','w','s','D','i','r',0};
47 static const WCHAR szDirectXVersionLongEnglish[] = {'s','z','D','i','r','e','c','t','X','V','e','r','s','i','o','n','L','o','n','g','E','n','g','l','i','s','h',0};
48 static const WCHAR szSetupParamEnglish[] = {'s','z','S','e','t','u','p','P','a','r','a','m','E','n','g','l','i','s','h',0};
49 static const WCHAR szDxDiagVersion[] = {'s','z','D','x','D','i','a','g','V','e','r','s','i','o','n',0};
50
51 struct property_list
52 {
53 const WCHAR *property_name;
54 WCHAR **output;
55 };
56
57 static BOOL property_to_string(IDxDiagContainer *container, const WCHAR *property, WCHAR **output)
58 {
59 VARIANT var;
60 HRESULT hr;
61 BOOL ret = FALSE;
62
63 VariantInit(&var);
64
65 hr = IDxDiagContainer_GetProp(container, property, &var);
66 if (SUCCEEDED(hr))
67 {
68 if (V_VT(&var) == VT_BSTR)
69 {
70 WCHAR *bstr = V_BSTR(&var);
71
72 *output = HeapAlloc(GetProcessHeap(), 0, (strlenW(bstr) + 1) * sizeof(WCHAR));
73 if (*output)
74 {
75 strcpyW(*output, bstr);
76 ret = TRUE;
77 }
78 }
79 }
80
81 VariantClear(&var);
82 return ret;
83 }
84
85 static void free_system_information(struct dxdiag_information *dxdiag_info)
86 {
87 struct system_information *system_info = &dxdiag_info->system_info;
88
89 HeapFree(GetProcessHeap(), 0, system_info->szTimeEnglish);
90 HeapFree(GetProcessHeap(), 0, system_info->szTimeLocalized);
91 HeapFree(GetProcessHeap(), 0, system_info->szMachineNameEnglish);
92 HeapFree(GetProcessHeap(), 0, system_info->szOSExLongEnglish);
93 HeapFree(GetProcessHeap(), 0, system_info->szOSExLocalized);
94 HeapFree(GetProcessHeap(), 0, system_info->szLanguagesEnglish);
95 HeapFree(GetProcessHeap(), 0, system_info->szLanguagesLocalized);
96 HeapFree(GetProcessHeap(), 0, system_info->szSystemManufacturerEnglish);
97 HeapFree(GetProcessHeap(), 0, system_info->szSystemModelEnglish);
98 HeapFree(GetProcessHeap(), 0, system_info->szBIOSEnglish);
99 HeapFree(GetProcessHeap(), 0, system_info->szProcessorEnglish);
100 HeapFree(GetProcessHeap(), 0, system_info->szPhysicalMemoryEnglish);
101 HeapFree(GetProcessHeap(), 0, system_info->szPageFileEnglish);
102 HeapFree(GetProcessHeap(), 0, system_info->szPageFileLocalized);
103 HeapFree(GetProcessHeap(), 0, system_info->szWindowsDir);
104 HeapFree(GetProcessHeap(), 0, system_info->szDirectXVersionLongEnglish);
105 HeapFree(GetProcessHeap(), 0, system_info->szSetupParamEnglish);
106 HeapFree(GetProcessHeap(), 0, system_info->szDxDiagVersion);
107 }
108
109 static inline void fill_system_property_list(struct dxdiag_information *dxdiag_info, struct property_list *list)
110 {
111 struct system_information *system_info = &dxdiag_info->system_info;
112
113 list[0].property_name = szTimeEnglish;
114 list[0].output = &system_info->szTimeEnglish;
115 list[1].property_name = szTimeLocalized;
116 list[1].output = &system_info->szTimeLocalized;
117 list[2].property_name = szMachineNameEnglish;
118 list[2].output = &system_info->szMachineNameEnglish;
119 list[3].property_name = szOSExLongEnglish;
120 list[3].output = &system_info->szOSExLongEnglish;
121 list[4].property_name = szOSExLocalized;
122 list[4].output = &system_info->szOSExLocalized;
123 list[5].property_name = szLanguagesEnglish;
124 list[5].output = &system_info->szLanguagesEnglish;
125 list[6].property_name = szLanguagesLocalized;
126 list[6].output = &system_info->szLanguagesLocalized;
127 list[7].property_name = szSystemManufacturerEnglish;
128 list[7].output = &system_info->szSystemManufacturerEnglish;
129 list[8].property_name = szSystemModelEnglish;
130 list[8].output = &system_info->szSystemModelEnglish;
131 list[9].property_name = szBIOSEnglish;
132 list[9].output = &system_info->szBIOSEnglish;
133 list[10].property_name = szProcessorEnglish;
134 list[10].output = &system_info->szProcessorEnglish;
135 list[11].property_name = szPhysicalMemoryEnglish;
136 list[11].output = &system_info->szPhysicalMemoryEnglish;
137 list[12].property_name = szPageFileEnglish;
138 list[12].output = &system_info->szPageFileEnglish;
139 list[13].property_name = szPageFileLocalized;
140 list[13].output = &system_info->szPageFileLocalized;
141 list[14].property_name = szWindowsDir;
142 list[14].output = &system_info->szWindowsDir;
143 list[15].property_name = szDirectXVersionLongEnglish;
144 list[15].output = &system_info->szDirectXVersionLongEnglish;
145 list[16].property_name = szSetupParamEnglish;
146 list[16].output = &system_info->szSetupParamEnglish;
147 list[17].property_name = szDxDiagVersion;
148 list[17].output = &system_info->szDxDiagVersion;
149 }
150
151 static BOOL fill_system_information(IDxDiagContainer *container, struct dxdiag_information *dxdiag_info)
152 {
153 struct system_information *system_info = &dxdiag_info->system_info;
154 size_t i;
155 struct property_list property_list[18];
156
157 fill_system_property_list(dxdiag_info, property_list);
158
159 for (i = 0; i < sizeof(property_list)/sizeof(property_list[0]); i++)
160 {
161 if (!property_to_string(container, property_list[i].property_name, property_list[i].output))
162 {
163 WINE_ERR("Failed to retrieve property %s\n", wine_dbgstr_w(property_list[i].property_name));
164 return FALSE;
165 }
166 }
167
168 #ifdef _WIN64
169 system_info->win64 = TRUE;
170 #else
171 system_info->win64 = FALSE;
172 #endif
173
174 return TRUE;
175 }
176
177 static const WCHAR DxDiag_SystemInfo[] = {'D','x','D','i','a','g','_','S','y','s','t','e','m','I','n','f','o',0};
178
179 static const struct information_fillers
180 {
181 const WCHAR *child_container_name;
182 BOOL (*fill_function)(IDxDiagContainer *, struct dxdiag_information *);
183 void (*free_function)(struct dxdiag_information *);
184 } filler_list[] =
185 {
186 {DxDiag_SystemInfo, fill_system_information, free_system_information},
187 };
188
189 void free_dxdiag_information(struct dxdiag_information *system_info)
190 {
191 size_t i;
192
193 if (!system_info)
194 return;
195
196 for (i = 0; i < sizeof(filler_list)/sizeof(filler_list[0]); i++)
197 filler_list[i].free_function(system_info);
198
199 HeapFree(GetProcessHeap(), 0, system_info);
200 }
201
202 struct dxdiag_information *collect_dxdiag_information(BOOL whql_check)
203 {
204 IDxDiagProvider *pddp = NULL;
205 IDxDiagContainer *root = NULL;
206 struct dxdiag_information *ret = NULL;
207 DXDIAG_INIT_PARAMS params = {sizeof(DXDIAG_INIT_PARAMS), DXDIAG_DX9_SDK_VERSION};
208 HRESULT hr;
209 size_t i;
210
211 /* Initialize the DxDiag COM instances. */
212 hr = CoCreateInstance(&CLSID_DxDiagProvider, NULL, CLSCTX_INPROC_SERVER,
213 &IID_IDxDiagProvider, (void **)&pddp);
214 if (FAILED(hr))
215 {
216 WINE_ERR("IDxDiagProvider instance creation failed with 0x%08x\n", hr);
217 goto error;
218 }
219
220 params.bAllowWHQLChecks = whql_check;
221 hr = IDxDiagProvider_Initialize(pddp, ¶ms);
222 if (FAILED(hr))
223 goto error;
224
225 hr = IDxDiagProvider_GetRootContainer(pddp, &root);
226 if (FAILED(hr))
227 goto error;
228
229 ret = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(*ret));
230 if (!ret)
231 goto error;
232
233 for (i = 0; i < sizeof(filler_list)/sizeof(filler_list[0]); i++)
234 {
235 IDxDiagContainer *child;
236 BOOL success;
237
238 hr = IDxDiagContainer_GetChildContainer(root, filler_list[i].child_container_name, &child);
239 if (FAILED(hr))
240 goto error;
241
242 success = filler_list[i].fill_function(child, ret);
243 IDxDiagContainer_Release(child);
244 if (!success)
245 goto error;
246 }
247
248 IDxDiagContainer_Release(root);
249 IDxDiagProvider_Release(pddp);
250 return ret;
251
252 error:
253 free_dxdiag_information(ret);
254 if (root) IDxDiagContainer_Release(root);
255 if (pddp) IDxDiagProvider_Release(pddp);
256 return NULL;
257 }
258
This page was automatically generated by the
LXR engine.
Visit the LXR main site for more
information.