programs/wmic/main.c

   1 /*
   2  * Copyright 2010 Louis Lenders
   3  * Copyright 2012 Hans Leidekker for CodeWeavers
   4  *
   5  * This library is free software; you can redistribute it and/or
   6  * modify it under the terms of the GNU Lesser General Public
   7  * License as published by the Free Software Foundation; either
   8  * version 2.1 of the License, or (at your option) any later version.
   9  *
  10  * This library is distributed in the hope that it will be useful,
  11  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  13  * Lesser General Public License for more details.
  14  *
  15  * You should have received a copy of the GNU Lesser General Public
  16  * License along with this library; if not, write to the Free Software
  17  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
  18  */
  19
  20 #define COBJMACROS
  21
  22 #include <stdio.h>
  23 #include "windows.h"
  24 #include "ocidl.h"
  25 #include "initguid.h"
  26 #include "objidl.h"
  27 #include "wbemcli.h"
  28 #include "wmic.h"
  29
  30 #include "wine/debug.h"
  31 #include "wine/unicode.h"
  32
  33 WINE_DEFAULT_DEBUG_CHANNEL(wmic);
  34
  35 static const WCHAR biosW[] =
  36     {'b','i','o','s',0};
  37 static const WCHAR computersystemW[] =
  38     {'c','o','m','p','u','t','e','r','s','y','s','t','e','m',0};
  39 static const WCHAR cpuW[] =
  40     {'c','p','u',0};
  41 static const WCHAR logicaldiskW[] =
  42     {'L','o','g','i','c','a','l','D','i','s','k',0};
  43 static const WCHAR nicW[] =
  44     {'n','i','c',0};
  45 static const WCHAR osW[] =
  46     {'o','s',0};
  47 static const WCHAR processW[] =
  48     {'p','r','o','c','e','s','s',0};
  49
  50 static const WCHAR win32_biosW[] =
  51     {'W','i','n','3','2','_','B','I','O','S',0};
  52 static const WCHAR win32_computersystemW[] =
  53     {'W','i','n','3','2','_','C','o','m','p','u','t','e','r','S','y','s','t','e','m',0};
  54 static const WCHAR win32_logicaldiskW[] =
  55     {'W','i','n','3','2','_','L','o','g','i','c','a','l','D','i','s','k',0};
  56 static const WCHAR win32_networkadapterW[] =
  57     {'W','i','n','3','2','_','N','e','t','w','o','r','k','A','d','a','p','t','e','r',0};
  58 static const WCHAR win32_operatingsystemW[] =
  59     {'W','i','n','3','2','_','O','p','e','r','a','t','i','n','g','S','y','s','t','e','m',0};
  60 static const WCHAR win32_processW[] =
  61     {'W','i','n','3','2','_','P','r','o','c','e','s','s',0};
  62 static const WCHAR win32_processorW[] =
  63     {'W','i','n','3','2','_','P','r','o','c','e','s','s','o','r',0};
  64
  65 static const struct
  66 {
  67     const WCHAR *alias;
  68     const WCHAR *class;
  69 }
  70 alias_map[] =
  71 {
  72     { biosW, win32_biosW },
  73     { computersystemW, win32_computersystemW },
  74     { cpuW, win32_processorW },
  75     { logicaldiskW, win32_logicaldiskW },
  76     { nicW, win32_networkadapterW },
  77     { osW, win32_operatingsystemW },
  78     { processW, win32_processW }
  79 };
  80
  81 static const WCHAR *find_class( const WCHAR *alias )
  82 {
  83     unsigned int i;
  84
  85     for (i = 0; i < sizeof(alias_map)/sizeof(alias_map[0]); i++)
  86     {
  87         if (!strcmpiW( alias, alias_map[i].alias )) return alias_map[i].class;
  88     }
  89     return NULL;
  90 }
  91
  92 static inline WCHAR *strdupW( const WCHAR *src )
  93 {
  94     WCHAR *dst;
  95     if (!src) return NULL;
  96     if (!(dst = HeapAlloc( GetProcessHeap(), 0, (strlenW( src ) + 1) * sizeof(WCHAR) ))) return NULL;
  97     strcpyW( dst, src );
  98     return dst;
  99 }
 100
 101 static WCHAR *find_prop( IWbemClassObject *class, const WCHAR *prop )
 102 {
 103     SAFEARRAY *sa;
 104     WCHAR *ret = NULL;
 105     LONG i, last_index = 0;
 106     BSTR str;
 107
 108     if (IWbemClassObject_GetNames( class, NULL, WBEM_FLAG_ALWAYS, NULL, &sa ) != S_OK) return NULL;
 109
 110     SafeArrayGetUBound( sa, 1, &last_index );
 111     for (i = 0; i <= last_index; i++)
 112     {
 113         SafeArrayGetElement( sa, &i, &str );
 114         if (!strcmpiW( str, prop ))
 115         {
 116             ret = strdupW( str );
 117             break;
 118         }
 119     }
 120     SafeArrayDestroy( sa );
 121     return ret;
 122 }
 123
 124 static int output_string( const WCHAR *msg, ... )
 125 {
 126     va_list va_args;
 127     int wlen;
 128     DWORD count, ret;
 129     WCHAR buffer[8192];
 130
 131     va_start( va_args, msg );
 132     vsprintfW( buffer, msg, va_args );
 133     va_end( va_args );
 134
 135     wlen = strlenW( buffer );
 136     ret = WriteConsoleW( GetStdHandle(STD_OUTPUT_HANDLE), buffer, wlen, &count, NULL );
 137     if (!ret)
 138     {
 139         DWORD len;
 140         char *msgA;
 141
 142         /* On Windows WriteConsoleW() fails if the output is redirected. So fall
 143          * back to WriteFile(), assuming the console encoding is still the right
 144          * one in that case.
 145          */
 146         len = WideCharToMultiByte( GetConsoleOutputCP(), 0, buffer, wlen, NULL, 0, NULL, NULL );
 147         if (!(msgA = HeapAlloc( GetProcessHeap(), 0, len * sizeof(char) ))) return 0;
 148
 149         WideCharToMultiByte( GetConsoleOutputCP(), 0, buffer, wlen, msgA, len, NULL, NULL );
 150         WriteFile( GetStdHandle(STD_OUTPUT_HANDLE), msgA, len, &count, FALSE );
 151         HeapFree( GetProcessHeap(), 0, msgA );
 152     }
 153     return count;
 154 }
 155
 156 static int output_message( int msg )
 157 {
 158     static const WCHAR fmtW[] = {'%','s',0};
 159     WCHAR buffer[8192];
 160
 161     LoadStringW( GetModuleHandleW(NULL), msg, buffer, sizeof(buffer)/sizeof(WCHAR) );
 162     return output_string( fmtW, buffer );
 163 }
 164
 165 static int query_prop( const WCHAR *class, const WCHAR *propname )
 166 {
 167     static const WCHAR select_allW[] = {'S','E','L','E','C','T',' ','*',' ','F','R','O','M',' ',0};
 168     static const WCHAR cimv2W[] = {'R','O','O','T','\\','C','I','M','V','2',0};
 169     static const WCHAR wqlW[] = {'W','Q','L',0};
 170     static const WCHAR newlineW[] = {'\n',0};
 171     static const WCHAR fmtW[] = {'%','s','\n',0};
 172     HRESULT hr;
 173     IWbemLocator *locator = NULL;
 174     IWbemServices *services = NULL;
 175     IEnumWbemClassObject *result = NULL;
 176     LONG flags = WBEM_FLAG_RETURN_IMMEDIATELY | WBEM_FLAG_FORWARD_ONLY;
 177     BSTR path = NULL, wql = NULL, query = NULL;
 178     WCHAR *prop = NULL;
 179     BOOL first = TRUE;
 180     int len, ret = -1;
 181
 182     WINE_TRACE("%s, %s\n", debugstr_w(class), debugstr_w(propname));
 183
 184     CoInitialize( NULL );
 185     CoInitializeSecurity( NULL, -1, NULL, NULL, RPC_C_AUTHN_LEVEL_DEFAULT,
 186                           RPC_C_IMP_LEVEL_IMPERSONATE, NULL, EOAC_NONE, NULL );
 187
 188     hr = CoCreateInstance( &CLSID_WbemLocator, NULL, CLSCTX_INPROC_SERVER, &IID_IWbemLocator,
 189                            (void **)&locator );
 190     if (hr != S_OK) goto done;
 191
 192     if (!(path = SysAllocString( cimv2W ))) goto done;
 193     hr = IWbemLocator_ConnectServer( locator, path, NULL, NULL, NULL, 0, NULL, NULL, &services );
 194     if (hr != S_OK) goto done;
 195
 196     len = strlenW( class ) + sizeof(select_allW) / sizeof(select_allW[0]);
 197     if (!(query = SysAllocStringLen( NULL, len ))) goto done;
 198     strcpyW( query, select_allW );
 199     strcatW( query, class );
 200
 201     if (!(wql = SysAllocString( wqlW ))) goto done;
 202     hr = IWbemServices_ExecQuery( services, wql, query, flags, NULL, &result );
 203     if (hr != S_OK) goto done;
 204
 205     for (;;)
 206     {
 207         IWbemClassObject *obj;
 208         ULONG count;
 209         VARIANT v;
 210
 211         IEnumWbemClassObject_Next( result, WBEM_INFINITE, 1, &obj, &count );
 212         if (!count) break;
 213
 214         if (first)
 215         {
 216             if (!(prop = find_prop( obj, propname )))
 217             {
 218                 output_message( STRING_INVALID_QUERY );
 219                 goto done;
 220             }
 221             output_string( fmtW, prop );
 222             first = FALSE;
 223         }
 224         if (IWbemClassObject_Get( obj, prop, 0, &v, NULL, NULL ) == WBEM_S_NO_ERROR)
 225         {
 226             VariantChangeType( &v, &v, 0, VT_BSTR );
 227             output_string( fmtW, V_BSTR( &v ) );
 228             VariantClear( &v );
 229         }
 230         IWbemClassObject_Release( obj );
 231     }
 232     output_string( newlineW );
 233     ret = 0;
 234
 235 done:
 236     if (result) IEnumWbemClassObject_Release( result );
 237     if (services) IWbemServices_Release( services );
 238     if (locator) IWbemLocator_Release( locator );
 239     SysFreeString( path );
 240     SysFreeString( query );
 241     SysFreeString( wql );
 242     HeapFree( GetProcessHeap(), 0, prop );
 243     CoUninitialize();
 244     return ret;
 245 }
 246
 247 int wmain(int argc, WCHAR *argv[])
 248 {
 249     static const WCHAR getW[] = {'g','e','t',0};
 250     static const WCHAR quitW[] = {'q','u','i','t',0};
 251     static const WCHAR exitW[] = {'e','x','i','t',0};
 252     static const WCHAR pathW[] = {'p','a','t','h',0};
 253     static const WCHAR classW[] = {'c','l','a','s','s',0};
 254     static const WCHAR contextW[] = {'c','o','n','t','e','x','t',0};
 255     const WCHAR *class, *value;
 256     int i;
 257
 258     for (i = 1; i < argc && argv[i][0] == '/'; i++)
 259         WINE_FIXME( "command line switch %s not supported\n", debugstr_w(argv[i]) );
 260
 261     if (i >= argc)
 262         goto not_supported;
 263
 264     if (!strcmpiW( argv[i], quitW ) ||
 265         !strcmpiW( argv[i], exitW ))
 266     {
 267         return 0;
 268     }
 269
 270     if (!strcmpiW( argv[i], classW) ||
 271         !strcmpiW( argv[i], contextW ))
 272     {
 273         WINE_FIXME( "command %s not supported\n", debugstr_w(argv[i]) );
 274         goto not_supported;
 275     }
 276
 277     if (!strcmpiW( argv[i], pathW ))
 278     {
 279         if (++i >= argc)
 280         {
 281             output_message( STRING_INVALID_PATH );
 282             return 1;
 283         }
 284         class = argv[i];
 285     }
 286     else
 287     {
 288         class = find_class( argv[i] );
 289         if (!class)
 290         {
 291             output_message( STRING_ALIAS_NOT_FOUND );
 292             return 1;
 293         }
 294     }
 295
 296     if (++i >= argc)
 297         goto not_supported;
 298
 299     if (!strcmpiW( argv[i], getW ))
 300     {
 301         if (++i >= argc)
 302             goto not_supported;
 303         value = argv[i];
 304         return query_prop( class, value );
 305     }
 306
 307 not_supported:
 308     output_message( STRING_CMDLINE_NOT_SUPPORTED );
 309     return 1;
 310 }