programs/dxdiag/main.c

   1 /*
   2  * DxDiag Implementation
   3  *
   4  * Copyright 2009 Austin English
   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 "dxdiag_private.h"
  27
  28 WINE_DEFAULT_DEBUG_CHANNEL(dxdiag);
  29
  30 HINSTANCE hInstance;
  31
  32 struct command_line_info
  33 {
  34     WCHAR outfile[MAX_PATH];
  35     enum output_type output_type;
  36     BOOL whql_check;
  37 };
  38
  39 static void usage(void)
  40 {
  41     WCHAR title[MAX_STRING_LEN];
  42     WCHAR usage[MAX_STRING_LEN];
  43
  44     LoadStringW(hInstance, STRING_DXDIAG_TOOL, title, ARRAY_SIZE(title));
  45     LoadStringW(hInstance, STRING_USAGE, usage, ARRAY_SIZE(usage));
  46
  47     MessageBoxW(NULL, usage, title, MB_OK | MB_ICONWARNING);
  48
  49     ExitProcess(0);
  50 }
  51
  52 static BOOL process_file_name(const WCHAR *cmdline, enum output_type output_type, WCHAR *filename, size_t filename_len)
  53 {
  54     const WCHAR *endptr;
  55     size_t len;
  56
  57     /* Skip any intervening spaces. */
  58     while (*cmdline == ' ')
  59         cmdline++;
  60
  61     /* Ignore filename quoting, if any. */
  62     if (*cmdline == '"' && (endptr = wcsrchr(cmdline, '"')))
  63     {
  64         /* Reject a string with only one quote. */
  65         if (cmdline == endptr)
  66             return FALSE;
  67
  68         cmdline++;
  69     }
  70     else
  71         endptr = cmdline + lstrlenW(cmdline);
  72
  73     len = endptr - cmdline;
  74     if (len == 0 || len >= filename_len)
  75         return FALSE;
  76
  77     memcpy(filename, cmdline, len * sizeof(WCHAR));
  78     filename[len] = '\0';
  79
  80     /* Append an extension appropriate for the output type if the filename does not have one. */
  81     if (!wcsrchr(filename, '.'))
  82     {
  83         const WCHAR *filename_ext = get_output_extension(output_type);
  84
  85         if (len + lstrlenW(filename_ext) >= filename_len)
  86             return FALSE;
  87
  88         lstrcatW(filename, filename_ext);
  89     }
  90
  91     return TRUE;
  92 }
  93
  94 /*
  95     Process options [/WHQL:ON|OFF][/X outfile|/T outfile]
  96     Returns TRUE if options were present, FALSE otherwise
  97     Only one of /X and /T is allowed, /WHQL must come before /X and /T,
  98     and the rest of the command line after /X or /T is interpreted as a
  99     filename. If a non-option portion of the command line is encountered,
 100     dxdiag assumes that the string is a filename for the /T option.
 101
 102     Native does not interpret quotes, but quotes are parsed here because of how
 103     Wine handles the command line.
 104 */
 105
 106 static BOOL process_command_line(const WCHAR *cmdline, struct command_line_info *info)
 107 {
 108     info->whql_check = FALSE;
 109     info->output_type = OUTPUT_NONE;
 110
 111     while (*cmdline)
 112     {
 113         /* Skip whitespace before arg */
 114         while (*cmdline == ' ')
 115             cmdline++;
 116
 117         /* If no option is specified, treat the command line as a filename. */
 118         if (*cmdline != '-' && *cmdline != '/')
 119         {
 120             info->output_type = OUTPUT_TEXT;
 121             return process_file_name(cmdline, OUTPUT_TEXT, info->outfile,
 122                                      ARRAY_SIZE(info->outfile));
 123         }
 124
 125         cmdline++;
 126
 127         switch (*cmdline)
 128         {
 129         case 'T':
 130         case 't':
 131             info->output_type = OUTPUT_TEXT;
 132             return process_file_name(cmdline + 1, OUTPUT_TEXT, info->outfile,
 133                                      ARRAY_SIZE(info->outfile));
 134         case 'X':
 135         case 'x':
 136             info->output_type = OUTPUT_XML;
 137             return process_file_name(cmdline + 1, OUTPUT_XML, info->outfile,
 138                                      ARRAY_SIZE(info->outfile));
 139         case 'W':
 140         case 'w':
 141             if (wcsnicmp(cmdline, L"whql:", 5))
 142                 return FALSE;
 143
 144             cmdline += 5;
 145
 146             if (!wcsnicmp(cmdline, L"off", 3))
 147             {
 148                 info->whql_check = FALSE;
 149                 cmdline += 2;
 150             }
 151             else if (!wcsnicmp(cmdline, L"on", 2))
 152             {
 153                 info->whql_check = TRUE;
 154                 cmdline++;
 155             }
 156             else
 157                 return FALSE;
 158
 159             break;
 160
 161         case 'd':
 162         case 'D':
 163             if (wcsnicmp(cmdline, L"dontskip", 8))
 164                 return FALSE;
 165             cmdline += 8;
 166             break;
 167
 168         default:
 169             return FALSE;
 170         }
 171
 172         cmdline++;
 173     }
 174
 175     return TRUE;
 176 }
 177
 178 int WINAPI wWinMain(HINSTANCE hInst, HINSTANCE hPrevInst, LPWSTR cmdline, int cmdshow)
 179 {
 180     struct command_line_info info;
 181     struct dxdiag_information *dxdiag_info;
 182
 183     hInstance = hInst;
 184
 185     if (!process_command_line(cmdline, &info))
 186         usage();
 187
 188     WINE_TRACE("WHQL check: %s\n", info.whql_check ? "TRUE" : "FALSE");
 189     WINE_TRACE("Output type: %d\n", info.output_type);
 190     if (info.output_type != OUTPUT_NONE)
 191         WINE_TRACE("Output filename: %s\n", debugstr_output_type(info.output_type));
 192
 193     CoInitialize(NULL);
 194
 195     dxdiag_info = collect_dxdiag_information(info.whql_check);
 196     if (!dxdiag_info)
 197     {
 198         WINE_ERR("DxDiag information collection failed\n");
 199         CoUninitialize();
 200         return 1;
 201     }
 202
 203     if (info.output_type != OUTPUT_NONE)
 204         output_dxdiag_information(dxdiag_info, info.outfile, info.output_type);
 205     else
 206         WINE_FIXME("Information dialog is not implemented\n");
 207
 208     free_dxdiag_information(dxdiag_info);
 209
 210     CoUninitialize();
 211     return 0;
 212 }