Sweden-Number/programs/dxdiag/output.c

407 lines
13 KiB
C
Raw Normal View History

/*
* DxDiag file information output
*
* 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 COBJMACROS
#include <initguid.h>
#include <windows.h>
#include <msxml2.h>
#include <assert.h>
#include <stdio.h>
#include "wine/debug.h"
#include "wine/unicode.h"
#include "dxdiag_private.h"
WINE_DEFAULT_DEBUG_CHANNEL(dxdiag);
static char output_buffer[1024];
static const char crlf[2] = "\r\n";
static const WCHAR DxDiag[] = {'D','x','D','i','a','g',0};
static const WCHAR SystemInformation[] = {'S','y','s','t','e','m','I','n','f','o','r','m','a','t','i','o','n',0};
static const WCHAR Time[] = {'T','i','m','e',0};
static const WCHAR MachineName[] = {'M','a','c','h','i','n','e','N','a','m','e',0};
static const WCHAR OperatingSystem[] = {'O','p','e','r','a','t','i','n','g','S','y','s','t','e','m',0};
static const WCHAR Language[] = {'L','a','n','g','u','a','g','e',0};
static const WCHAR SystemManufacturer[] = {'S','y','s','t','e','m','M','a','n','u','f','a','c','t','u','r','e','r',0};
static const WCHAR SystemModel[] = {'S','y','s','t','e','m','M','o','d','e','l',0};
static const WCHAR BIOS[] = {'B','I','O','S',0};
static const WCHAR Processor[] = {'P','r','o','c','e','s','s','o','r',0};
static const WCHAR Memory[] = {'M','e','m','o','r','y',0};
static const WCHAR PageFile[] = {'P','a','g','e','F','i','l','e',0};
static const WCHAR WindowsDir[] = {'W','i','n','d','o','w','s','D','i','r',0};
static const WCHAR DirectXVersion[] = {'D','i','r','e','c','t','X','V','e','r','s','i','o','n',0};
static const WCHAR DXSetupParameters[] = {'D','X','S','e','t','u','p','P','a','r','a','m','e','t','e','r','s',0};
static const WCHAR DxDiagVersion[] = {'D','x','D','i','a','g','V','e','r','s','i','o','n',0};
static const WCHAR DxDiagUnicode[] = {'D','x','D','i','a','g','U','n','i','c','o','d','e',0};
static const WCHAR DxDiag64Bit[] = {'D','x','D','i','a','g','6','4','B','i','t',0};
struct text_information_field
{
const char *field_name;
const WCHAR *value;
};
struct xml_information_field
{
const WCHAR *tag_name;
const WCHAR *value;
};
static BOOL output_text_header(HANDLE hFile, const char *caption)
{
DWORD len = strlen(caption);
DWORD total_len = 3 * (len + sizeof(crlf));
char *ptr = output_buffer;
assert(total_len <= sizeof(output_buffer));
memset(ptr, '-', len);
ptr += len;
memcpy(ptr, crlf, sizeof(crlf));
ptr += sizeof(crlf);
memcpy(ptr, caption, len);
ptr += len;
memcpy(ptr, crlf, sizeof(crlf));
ptr += sizeof(crlf);
memset(ptr, '-', len);
ptr += len;
memcpy(ptr, crlf, sizeof(crlf));
return WriteFile(hFile, output_buffer, total_len, NULL, NULL);
}
static BOOL output_text_field(HANDLE hFile, const char *field_name, DWORD field_width, const WCHAR *value)
{
DWORD value_lenW = strlenW(value);
DWORD value_lenA = WideCharToMultiByte(CP_ACP, 0, value, value_lenW, NULL, 0, NULL, NULL);
DWORD total_len = field_width + sizeof(": ") - 1 + value_lenA + sizeof(crlf);
char sprintf_fmt[1 + 10 + 3 + 1];
char *ptr = output_buffer;
assert(total_len <= sizeof(output_buffer));
sprintf(sprintf_fmt, "%%%us: ", field_width);
ptr += sprintf(ptr, sprintf_fmt, field_name);
ptr += WideCharToMultiByte(CP_ACP, 0, value, value_lenW, ptr, value_lenA, NULL, NULL);
memcpy(ptr, crlf, sizeof(crlf));
ptr += sizeof(crlf);
return WriteFile(hFile, output_buffer, total_len, NULL, NULL);
}
static BOOL output_crlf(HANDLE hFile)
{
return WriteFile(hFile, crlf, sizeof(crlf), NULL, NULL);
}
static inline void fill_system_text_output_table(struct dxdiag_information *dxdiag_info, struct text_information_field *fields)
{
fields[0].field_name = "Time of this report";
fields[0].value = dxdiag_info->system_info.szTimeEnglish;
fields[1].field_name = "Machine name";
fields[1].value = dxdiag_info->system_info.szMachineNameEnglish;
fields[2].field_name = "Operating System";
fields[2].value = dxdiag_info->system_info.szOSExLongEnglish;
fields[3].field_name = "Language";
fields[3].value = dxdiag_info->system_info.szLanguagesEnglish;
fields[4].field_name = "System Manufacturer";
fields[4].value = dxdiag_info->system_info.szSystemManufacturerEnglish;
fields[5].field_name = "System Model";
fields[5].value = dxdiag_info->system_info.szSystemModelEnglish;
fields[6].field_name = "BIOS";
fields[6].value = dxdiag_info->system_info.szBIOSEnglish;
fields[7].field_name = "Processor";
fields[7].value = dxdiag_info->system_info.szProcessorEnglish;
fields[8].field_name = "Memory";
fields[8].value = dxdiag_info->system_info.szPhysicalMemoryEnglish;
fields[9].field_name = "Page File";
fields[9].value = dxdiag_info->system_info.szPageFileEnglish;
fields[10].field_name = "Windows Dir";
fields[10].value = dxdiag_info->system_info.szWindowsDir;
fields[11].field_name = "DirectX Version";
fields[11].value = dxdiag_info->system_info.szDirectXVersionLongEnglish;
fields[12].field_name = "DX Setup Parameters";
fields[12].value = dxdiag_info->system_info.szSetupParamEnglish;
fields[13].field_name = "DxDiag Version";
fields[13].value = dxdiag_info->system_info.szDxDiagVersion;
}
static BOOL output_text_information(struct dxdiag_information *dxdiag_info, const WCHAR *filename)
{
struct information_block
{
const char *caption;
const size_t field_width;
struct text_information_field fields[50];
} output_table[] =
{
{"System Information", 19},
};
HANDLE hFile;
size_t i;
fill_system_text_output_table(dxdiag_info, output_table[0].fields);
hFile = CreateFileW(filename, GENERIC_WRITE, FILE_SHARE_READ | FILE_SHARE_WRITE,
NULL, CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL);
if (hFile == INVALID_HANDLE_VALUE)
{
WINE_ERR("File creation failed, last error %u\n", GetLastError());
return FALSE;
}
for (i = 0; i < sizeof(output_table)/sizeof(output_table[0]); i++)
{
const struct text_information_field *fields = output_table[i].fields;
unsigned int j;
output_text_header(hFile, output_table[i].caption);
for (j = 0; fields[j].field_name; j++)
output_text_field(hFile, fields[j].field_name, output_table[i].field_width, fields[j].value);
output_crlf(hFile);
}
CloseHandle(hFile);
return FALSE;
}
static IXMLDOMElement *xml_create_element(IXMLDOMDocument *xmldoc, const WCHAR *name)
{
BSTR bstr = SysAllocString(name);
IXMLDOMElement *ret;
HRESULT hr;
if (!bstr)
return NULL;
hr = IXMLDOMDocument_createElement(xmldoc, bstr, &ret);
SysFreeString(bstr);
return SUCCEEDED(hr) ? ret : NULL;
}
static HRESULT xml_put_element_text(IXMLDOMElement *element, const WCHAR *text)
{
BSTR bstr = SysAllocString(text);
HRESULT hr;
if (!bstr)
return E_OUTOFMEMORY;
hr = IXMLDOMElement_put_text(element, bstr);
SysFreeString(bstr);
return hr;
}
static HRESULT save_xml_document(IXMLDOMDocument *xmldoc, const WCHAR *filename)
{
BSTR bstr = SysAllocString(filename);
VARIANT destVar;
HRESULT hr;
if (!bstr)
return E_OUTOFMEMORY;
V_VT(&destVar) = VT_BSTR;
V_BSTR(&destVar) = bstr;
hr = IXMLDOMDocument_save(xmldoc, destVar);
VariantClear(&destVar);
return hr;
}
static inline void fill_system_xml_output_table(struct dxdiag_information *dxdiag_info, struct xml_information_field *fields)
{
static const WCHAR zeroW[] = {'0',0};
static const WCHAR oneW[] = {'1',0};
fields[0].tag_name = Time;
fields[0].value = dxdiag_info->system_info.szTimeEnglish;
fields[1].tag_name = MachineName;
fields[1].value = dxdiag_info->system_info.szMachineNameEnglish;
fields[2].tag_name = OperatingSystem;
fields[2].value = dxdiag_info->system_info.szOSExLongEnglish;
fields[3].tag_name = Language;
fields[3].value = dxdiag_info->system_info.szLanguagesEnglish;
fields[4].tag_name = SystemManufacturer;
fields[4].value = dxdiag_info->system_info.szSystemManufacturerEnglish;
fields[5].tag_name = SystemModel;
fields[5].value = dxdiag_info->system_info.szSystemModelEnglish;
fields[6].tag_name = BIOS;
fields[6].value = dxdiag_info->system_info.szBIOSEnglish;
fields[7].tag_name = Processor;
fields[7].value = dxdiag_info->system_info.szProcessorEnglish;
fields[8].tag_name = Memory;
fields[8].value = dxdiag_info->system_info.szPhysicalMemoryEnglish;
fields[9].tag_name = PageFile;
fields[9].value = dxdiag_info->system_info.szPageFileEnglish;
fields[10].tag_name = WindowsDir;
fields[10].value = dxdiag_info->system_info.szWindowsDir;
fields[11].tag_name = DirectXVersion;
fields[11].value = dxdiag_info->system_info.szDirectXVersionLongEnglish;
fields[12].tag_name = DXSetupParameters;
fields[12].value = dxdiag_info->system_info.szSetupParamEnglish;
fields[13].tag_name = DxDiagVersion;
fields[13].value = dxdiag_info->system_info.szDxDiagVersion;
fields[14].tag_name = DxDiagUnicode;
fields[14].value = oneW;
fields[15].tag_name = DxDiag64Bit;
fields[15].value = dxdiag_info->system_info.win64 ? oneW : zeroW;
}
static BOOL output_xml_information(struct dxdiag_information *dxdiag_info, const WCHAR *filename)
{
struct information_block
{
const WCHAR *tag_name;
struct xml_information_field fields[50];
} output_table[] =
{
{SystemInformation},
};
IXMLDOMDocument *xmldoc = NULL;
IXMLDOMElement *dxdiag_element = NULL;
HRESULT hr;
size_t i;
fill_system_xml_output_table(dxdiag_info, output_table[0].fields);
hr = CoCreateInstance(&CLSID_DOMDocument, NULL, CLSCTX_INPROC_SERVER,
&IID_IXMLDOMDocument, (void **)&xmldoc);
if (FAILED(hr))
{
WINE_ERR("IXMLDOMDocument instance creation failed with 0x%08x\n", hr);
goto error;
}
if (!(dxdiag_element = xml_create_element(xmldoc, DxDiag)))
goto error;
hr = IXMLDOMDocument_appendChild(xmldoc, (IXMLDOMNode *)dxdiag_element, NULL);
if (FAILED(hr))
goto error;
for (i = 0; i < sizeof(output_table)/sizeof(output_table[0]); i++)
{
IXMLDOMElement *info_element = xml_create_element(xmldoc, output_table[i].tag_name);
const struct xml_information_field *fields = output_table[i].fields;
unsigned int j = 0;
if (!info_element)
goto error;
hr = IXMLDOMElement_appendChild(dxdiag_element, (IXMLDOMNode *)info_element, NULL);
if (FAILED(hr))
{
IXMLDOMElement_Release(info_element);
goto error;
}
for (j = 0; fields[j].tag_name; j++)
{
IXMLDOMElement *field_element = xml_create_element(xmldoc, fields[j].tag_name);
if (!field_element)
{
IXMLDOMElement_Release(info_element);
goto error;
}
hr = xml_put_element_text(field_element, fields[j].value);
if (FAILED(hr))
{
IXMLDOMElement_Release(field_element);
IXMLDOMElement_Release(info_element);
goto error;
}
hr = IXMLDOMElement_appendChild(info_element, (IXMLDOMNode *)field_element, NULL);
if (FAILED(hr))
{
IXMLDOMElement_Release(field_element);
IXMLDOMElement_Release(info_element);
goto error;
}
IXMLDOMElement_Release(field_element);
}
IXMLDOMElement_Release(info_element);
}
hr = save_xml_document(xmldoc, filename);
if (FAILED(hr))
goto error;
IXMLDOMElement_Release(dxdiag_element);
IXMLDOMDocument_Release(xmldoc);
return TRUE;
error:
if (dxdiag_element) IXMLDOMElement_Release(dxdiag_element);
if (xmldoc) IXMLDOMDocument_Release(xmldoc);
return FALSE;
}
static struct output_backend
{
const WCHAR filename_ext[5];
BOOL (*output_handler)(struct dxdiag_information *, const WCHAR *filename);
} output_backends[] =
{
/* OUTPUT_TEXT */
{
{'.','t','x','t',0},
output_text_information,
},
/* OUTPUT_XML */
{
{'.','x','m','l',0},
output_xml_information,
},
};
const WCHAR *get_output_extension(enum output_type type)
{
assert(type > OUTPUT_NONE && type <= sizeof(output_backends)/sizeof(output_backends[0]));
return output_backends[type - 1].filename_ext;
}
BOOL output_dxdiag_information(struct dxdiag_information *dxdiag_info, const WCHAR *filename, enum output_type type)
{
assert(type > OUTPUT_NONE && type <= sizeof(output_backends)/sizeof(output_backends[0]));
return output_backends[type - 1].output_handler(dxdiag_info, filename);
}