235 lines
7.9 KiB
C
235 lines
7.9 KiB
C
/*
|
|
* DxDiag information collection
|
|
*
|
|
* Copyright 2011 Andrew Nguyen
|
|
*
|
|
* This library is free software; you can redistribute it and/or
|
|
* modify it under the terms of the GNU Lesser General Public
|
|
* License as published by the Free Software Foundation; either
|
|
* version 2.1 of the License, or (at your option) any later version.
|
|
*
|
|
* This library is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
|
* Lesser General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU Lesser General Public
|
|
* License along with this library; if not, write to the Free Software
|
|
* Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
|
|
*/
|
|
|
|
#define WIN32_LEAN_AND_MEAN
|
|
#include <windows.h>
|
|
#include <dxdiag.h>
|
|
|
|
#include "wine/debug.h"
|
|
#include "dxdiag_private.h"
|
|
|
|
WINE_DEFAULT_DEBUG_CHANNEL(dxdiag);
|
|
|
|
struct property_list
|
|
{
|
|
const WCHAR *property_name;
|
|
WCHAR **output;
|
|
};
|
|
|
|
static BOOL property_to_string(IDxDiagContainer *container, const WCHAR *property, WCHAR **output)
|
|
{
|
|
VARIANT var;
|
|
HRESULT hr;
|
|
BOOL ret = FALSE;
|
|
|
|
VariantInit(&var);
|
|
|
|
hr = IDxDiagContainer_GetProp(container, property, &var);
|
|
if (SUCCEEDED(hr))
|
|
{
|
|
if (V_VT(&var) == VT_BSTR)
|
|
{
|
|
WCHAR *bstr = V_BSTR(&var);
|
|
|
|
*output = HeapAlloc(GetProcessHeap(), 0, (lstrlenW(bstr) + 1) * sizeof(WCHAR));
|
|
if (*output)
|
|
{
|
|
lstrcpyW(*output, bstr);
|
|
ret = TRUE;
|
|
}
|
|
}
|
|
}
|
|
|
|
VariantClear(&var);
|
|
return ret;
|
|
}
|
|
|
|
static void free_system_information(struct dxdiag_information *dxdiag_info)
|
|
{
|
|
struct system_information *system_info = &dxdiag_info->system_info;
|
|
|
|
HeapFree(GetProcessHeap(), 0, system_info->szTimeEnglish);
|
|
HeapFree(GetProcessHeap(), 0, system_info->szTimeLocalized);
|
|
HeapFree(GetProcessHeap(), 0, system_info->szMachineNameEnglish);
|
|
HeapFree(GetProcessHeap(), 0, system_info->szOSExLongEnglish);
|
|
HeapFree(GetProcessHeap(), 0, system_info->szOSExLocalized);
|
|
HeapFree(GetProcessHeap(), 0, system_info->szLanguagesEnglish);
|
|
HeapFree(GetProcessHeap(), 0, system_info->szLanguagesLocalized);
|
|
HeapFree(GetProcessHeap(), 0, system_info->szSystemManufacturerEnglish);
|
|
HeapFree(GetProcessHeap(), 0, system_info->szSystemModelEnglish);
|
|
HeapFree(GetProcessHeap(), 0, system_info->szBIOSEnglish);
|
|
HeapFree(GetProcessHeap(), 0, system_info->szProcessorEnglish);
|
|
HeapFree(GetProcessHeap(), 0, system_info->szPhysicalMemoryEnglish);
|
|
HeapFree(GetProcessHeap(), 0, system_info->szPageFileEnglish);
|
|
HeapFree(GetProcessHeap(), 0, system_info->szPageFileLocalized);
|
|
HeapFree(GetProcessHeap(), 0, system_info->szWindowsDir);
|
|
HeapFree(GetProcessHeap(), 0, system_info->szDirectXVersionLongEnglish);
|
|
HeapFree(GetProcessHeap(), 0, system_info->szSetupParamEnglish);
|
|
HeapFree(GetProcessHeap(), 0, system_info->szDxDiagVersion);
|
|
}
|
|
|
|
static inline void fill_system_property_list(struct dxdiag_information *dxdiag_info, struct property_list *list)
|
|
{
|
|
struct system_information *system_info = &dxdiag_info->system_info;
|
|
|
|
list[0].property_name = L"szTimeEnglish";
|
|
list[0].output = &system_info->szTimeEnglish;
|
|
list[1].property_name = L"szTimeLocalized";
|
|
list[1].output = &system_info->szTimeLocalized;
|
|
list[2].property_name = L"szMachineNameEnglish";
|
|
list[2].output = &system_info->szMachineNameEnglish;
|
|
list[3].property_name = L"szOSExLongEnglish";
|
|
list[3].output = &system_info->szOSExLongEnglish;
|
|
list[4].property_name = L"szOSExLocalized";
|
|
list[4].output = &system_info->szOSExLocalized;
|
|
list[5].property_name = L"szLanguagesEnglish";
|
|
list[5].output = &system_info->szLanguagesEnglish;
|
|
list[6].property_name = L"szLanguagesLocalized";
|
|
list[6].output = &system_info->szLanguagesLocalized;
|
|
list[7].property_name = L"szSystemManufacturerEnglish";
|
|
list[7].output = &system_info->szSystemManufacturerEnglish;
|
|
list[8].property_name = L"szSystemModelEnglish";
|
|
list[8].output = &system_info->szSystemModelEnglish;
|
|
list[9].property_name = L"szBIOSEnglish";
|
|
list[9].output = &system_info->szBIOSEnglish;
|
|
list[10].property_name = L"szProcessorEnglish";
|
|
list[10].output = &system_info->szProcessorEnglish;
|
|
list[11].property_name = L"szPhysicalMemoryEnglish";
|
|
list[11].output = &system_info->szPhysicalMemoryEnglish;
|
|
list[12].property_name = L"szPageFileEnglish";
|
|
list[12].output = &system_info->szPageFileEnglish;
|
|
list[13].property_name = L"szPageFileLocalized";
|
|
list[13].output = &system_info->szPageFileLocalized;
|
|
list[14].property_name = L"szWindowsDir";
|
|
list[14].output = &system_info->szWindowsDir;
|
|
list[15].property_name = L"szDirectXVersionLongEnglish";
|
|
list[15].output = &system_info->szDirectXVersionLongEnglish;
|
|
list[16].property_name = L"szSetupParamEnglish";
|
|
list[16].output = &system_info->szSetupParamEnglish;
|
|
list[17].property_name = L"szDxDiagVersion";
|
|
list[17].output = &system_info->szDxDiagVersion;
|
|
}
|
|
|
|
static BOOL fill_system_information(IDxDiagContainer *container, struct dxdiag_information *dxdiag_info)
|
|
{
|
|
struct system_information *system_info = &dxdiag_info->system_info;
|
|
size_t i;
|
|
struct property_list property_list[18];
|
|
|
|
fill_system_property_list(dxdiag_info, property_list);
|
|
|
|
for (i = 0; i < ARRAY_SIZE(property_list); i++)
|
|
{
|
|
if (!property_to_string(container, property_list[i].property_name, property_list[i].output))
|
|
{
|
|
WINE_ERR("Failed to retrieve property %s\n", wine_dbgstr_w(property_list[i].property_name));
|
|
return FALSE;
|
|
}
|
|
}
|
|
|
|
#ifdef _WIN64
|
|
system_info->win64 = TRUE;
|
|
#else
|
|
system_info->win64 = FALSE;
|
|
#endif
|
|
|
|
return TRUE;
|
|
}
|
|
|
|
static const struct information_fillers
|
|
{
|
|
const WCHAR *child_container_name;
|
|
BOOL (*fill_function)(IDxDiagContainer *, struct dxdiag_information *);
|
|
void (*free_function)(struct dxdiag_information *);
|
|
} filler_list[] =
|
|
{
|
|
{L"DxDiag_SystemInfo", fill_system_information, free_system_information},
|
|
};
|
|
|
|
void free_dxdiag_information(struct dxdiag_information *system_info)
|
|
{
|
|
size_t i;
|
|
|
|
if (!system_info)
|
|
return;
|
|
|
|
for (i = 0; i < ARRAY_SIZE(filler_list); i++)
|
|
filler_list[i].free_function(system_info);
|
|
|
|
HeapFree(GetProcessHeap(), 0, system_info);
|
|
}
|
|
|
|
struct dxdiag_information *collect_dxdiag_information(BOOL whql_check)
|
|
{
|
|
IDxDiagProvider *pddp = NULL;
|
|
IDxDiagContainer *root = NULL;
|
|
struct dxdiag_information *ret = NULL;
|
|
DXDIAG_INIT_PARAMS params = {sizeof(DXDIAG_INIT_PARAMS), DXDIAG_DX9_SDK_VERSION};
|
|
HRESULT hr;
|
|
size_t i;
|
|
|
|
/* Initialize the DxDiag COM instances. */
|
|
hr = CoCreateInstance(&CLSID_DxDiagProvider, NULL, CLSCTX_INPROC_SERVER,
|
|
&IID_IDxDiagProvider, (void **)&pddp);
|
|
if (FAILED(hr))
|
|
{
|
|
WINE_ERR("IDxDiagProvider instance creation failed with 0x%08lx\n", hr);
|
|
goto error;
|
|
}
|
|
|
|
params.bAllowWHQLChecks = whql_check;
|
|
hr = IDxDiagProvider_Initialize(pddp, ¶ms);
|
|
if (FAILED(hr))
|
|
goto error;
|
|
|
|
hr = IDxDiagProvider_GetRootContainer(pddp, &root);
|
|
if (FAILED(hr))
|
|
goto error;
|
|
|
|
ret = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(*ret));
|
|
if (!ret)
|
|
goto error;
|
|
|
|
for (i = 0; i < ARRAY_SIZE(filler_list); i++)
|
|
{
|
|
IDxDiagContainer *child;
|
|
BOOL success;
|
|
|
|
hr = IDxDiagContainer_GetChildContainer(root, filler_list[i].child_container_name, &child);
|
|
if (FAILED(hr))
|
|
goto error;
|
|
|
|
success = filler_list[i].fill_function(child, ret);
|
|
IDxDiagContainer_Release(child);
|
|
if (!success)
|
|
goto error;
|
|
}
|
|
|
|
IDxDiagContainer_Release(root);
|
|
IDxDiagProvider_Release(pddp);
|
|
return ret;
|
|
|
|
error:
|
|
free_dxdiag_information(ret);
|
|
if (root) IDxDiagContainer_Release(root);
|
|
if (pddp) IDxDiagProvider_Release(pddp);
|
|
return NULL;
|
|
}
|