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
  32 WINE_DEFAULT_DEBUG_CHANNEL(wmic);
  33
  34 static const struct
  35 {
  36     const WCHAR *alias;
  37     const WCHAR *class;
  38 }
  39 alias_map[] =
  40 {
  41     { L"bios", L"Win32_BIOS" },
  42     { L"computersystem", L"Win32_ComputerSystem" },
  43     { L"cpu", L"Win32_Processor" },
  44     { L"LogicalDisk", L"Win32_LogicalDisk" },
  45     { L"nic", L"Win32_NetworkAdapter" },
  46     { L"os", L"Win32_OperatingSystem" },
  47     { L"process", L"Win32_Process" },
  48     { L"baseboard", L"Win32_BaseBoard" },
  49     { L"diskdrive", L"Win32_DiskDrive" },
  50     { L"memorychip", L"Win32_PhysicalMemory" }
  51 };
  52
  53 static const WCHAR *find_class( const WCHAR *alias )
  54 {
  55     unsigned int i;
  56
  57     for (i = 0; i < ARRAY_SIZE(alias_map); i++)
  58     {
  59         if (!wcsicmp( alias, alias_map[i].alias )) return alias_map[i].class;
  60     }
  61     return NULL;
  62 }
  63
  64 static inline WCHAR *strdupW( const WCHAR *src )
  65 {
  66     WCHAR *dst;
  67     if (!src) return NULL;
  68     if (!(dst = HeapAlloc( GetProcessHeap(), 0, (lstrlenW( src ) + 1) * sizeof(WCHAR) ))) return NULL;
  69     lstrcpyW( dst, src );
  70     return dst;
  71 }
  72
  73 static WCHAR *find_prop( IWbemClassObject *class, const WCHAR *prop )
  74 {
  75     SAFEARRAY *sa;
  76     WCHAR *ret = NULL;
  77     LONG i, last_index = 0;
  78     BSTR str;
  79
  80     if (IWbemClassObject_GetNames( class, NULL, WBEM_FLAG_ALWAYS, NULL, &sa ) != S_OK) return NULL;
  81
  82     SafeArrayGetUBound( sa, 1, &last_index );
  83     for (i = 0; i <= last_index; i++)
  84     {
  85         SafeArrayGetElement( sa, &i, &str );
  86         if (!wcsicmp( str, prop ))
  87         {
  88             ret = strdupW( str );
  89             break;
  90         }
  91     }
  92     SafeArrayDestroy( sa );
  93     return ret;
  94 }
  95
  96 static int WINAPIV output_string( HANDLE handle, const WCHAR *msg, ... )
  97 {
  98     __ms_va_list va_args;
  99     int len;
 100     DWORD count;
 101     WCHAR buffer[8192];
 102
 103     __ms_va_start( va_args, msg );
 104     len = vswprintf( buffer, ARRAY_SIZE(buffer), msg, va_args );
 105     __ms_va_end( va_args );
 106
 107     if (!WriteConsoleW( handle, buffer, len, &count, NULL ))
 108         WriteFile( handle, buffer, len * sizeof(WCHAR), &count, FALSE );
 109
 110     return count;
 111 }
 112
 113 static int output_error( int msg )
 114 {
 115     WCHAR buffer[8192];
 116
 117     LoadStringW( GetModuleHandleW(NULL), msg, buffer, ARRAY_SIZE(buffer));
 118     return output_string( GetStdHandle(STD_ERROR_HANDLE), L"%s", buffer );
 119 }
 120
 121 static int output_header( const WCHAR *prop, ULONG column_width )
 122 {
 123     static const WCHAR bomW[] = {0xfeff};
 124     int len;
 125     DWORD count;
 126     WCHAR buffer[8192];
 127
 128     len = swprintf( buffer, ARRAY_SIZE(buffer), L"%-*s\r\n", column_width, prop );
 129
 130     if (!WriteConsoleW( GetStdHandle(STD_OUTPUT_HANDLE), buffer, len, &count, NULL )) /* redirected */
 131     {
 132         WriteFile( GetStdHandle(STD_OUTPUT_HANDLE), bomW, sizeof(bomW), &count, FALSE );
 133         WriteFile( GetStdHandle(STD_OUTPUT_HANDLE), buffer, len * sizeof(WCHAR), &count, FALSE );
 134         count += sizeof(bomW);
 135     }
 136
 137     return count;
 138 }
 139
 140 static int output_line( const WCHAR *str, ULONG column_width )
 141 {
 142     return output_string( GetStdHandle(STD_OUTPUT_HANDLE), L"%-*s\r\n", column_width, str );
 143 }
 144
 145 static int query_prop( const WCHAR *class, const WCHAR *propname )
 146 {
 147     HRESULT hr;
 148     IWbemLocator *locator = NULL;
 149     IWbemServices *services = NULL;
 150     IEnumWbemClassObject *result = NULL;
 151     LONG flags = WBEM_FLAG_RETURN_IMMEDIATELY | WBEM_FLAG_FORWARD_ONLY;
 152     BSTR path = NULL, wql = NULL, query = NULL;
 153     WCHAR *prop = NULL;
 154     BOOL first = TRUE;
 155     int len, ret = -1;
 156     IWbemClassObject *obj;
 157     ULONG count, width = 0;
 158     VARIANT v;
 159
 160     WINE_TRACE("%s, %s\n", debugstr_w(class), debugstr_w(propname));
 161
 162     CoInitialize( NULL );
 163     CoInitializeSecurity( NULL, -1, NULL, NULL, RPC_C_AUTHN_LEVEL_DEFAULT,
 164                           RPC_C_IMP_LEVEL_IMPERSONATE, NULL, EOAC_NONE, NULL );
 165
 166     hr = CoCreateInstance( &CLSID_WbemLocator, NULL, CLSCTX_INPROC_SERVER, &IID_IWbemLocator,
 167                            (void **)&locator );
 168     if (hr != S_OK) goto done;
 169
 170     if (!(path = SysAllocString(L"ROOT\\CIMV2" ))) goto done;
 171     hr = IWbemLocator_ConnectServer( locator, path, NULL, NULL, NULL, 0, NULL, NULL, &services );
 172     if (hr != S_OK) goto done;
 173
 174     len = lstrlenW( class ) + ARRAY_SIZE(L"SELECT * FROM ");
 175     if (!(query = SysAllocStringLen( NULL, len ))) goto done;
 176     lstrcpyW( query, L"SELECT * FROM " );
 177     lstrcatW( query, class );
 178
 179     if (!(wql = SysAllocString(L"WQL" ))) goto done;
 180     hr = IWbemServices_ExecQuery( services, wql, query, flags, NULL, &result );
 181     if (hr != S_OK) goto done;
 182
 183     for (;;) /* get column width */
 184     {
 185         IEnumWbemClassObject_Next( result, WBEM_INFINITE, 1, &obj, &count );
 186         if (!count) break;
 187
 188         if (!prop && !(prop = find_prop( obj, propname )))
 189         {
 190             output_error( STRING_INVALID_QUERY );
 191             goto done;
 192         }
 193         if (IWbemClassObject_Get( obj, prop, 0, &v, NULL, NULL ) == WBEM_S_NO_ERROR)
 194         {
 195             VariantChangeType( &v, &v, 0, VT_BSTR );
 196             width = max( lstrlenW( V_BSTR( &v ) ), width );
 197             VariantClear( &v );
 198         }
 199         IWbemClassObject_Release( obj );
 200     }
 201     width += 2;
 202
 203     IEnumWbemClassObject_Reset( result );
 204     for (;;)
 205     {
 206         IEnumWbemClassObject_Next( result, WBEM_INFINITE, 1, &obj, &count );
 207         if (!count) break;
 208
 209         if (first)
 210         {
 211             output_header( prop, width );
 212             first = FALSE;
 213         }
 214         if (IWbemClassObject_Get( obj, prop, 0, &v, NULL, NULL ) == WBEM_S_NO_ERROR)
 215         {
 216             VariantChangeType( &v, &v, 0, VT_BSTR );
 217             output_line( V_BSTR( &v ), width );
 218             VariantClear( &v );
 219         }
 220         IWbemClassObject_Release( obj );
 221     }
 222     ret = 0;
 223
 224 done:
 225     if (result) IEnumWbemClassObject_Release( result );
 226     if (services) IWbemServices_Release( services );
 227     if (locator) IWbemLocator_Release( locator );
 228     SysFreeString( path );
 229     SysFreeString( query );
 230     SysFreeString( wql );
 231     HeapFree( GetProcessHeap(), 0, prop );
 232     CoUninitialize();
 233     return ret;
 234 }
 235
 236 int __cdecl wmain(int argc, WCHAR *argv[])
 237 {
 238     const WCHAR *class, *value;
 239     int i;
 240
 241     for (i = 1; i < argc && argv[i][0] == '/'; i++)
 242         WINE_FIXME( "command line switch %s not supported\n", debugstr_w(argv[i]) );
 243
 244     if (i >= argc)
 245         goto not_supported;
 246
 247     if (!wcsicmp( argv[i], L"quit" ) || !wcsicmp( argv[i], L"exit" ))
 248     {
 249         return 0;
 250     }
 251
 252     if (!wcsicmp( argv[i], L"class") || !wcsicmp( argv[i], L"context" ))
 253     {
 254         WINE_FIXME( "command %s not supported\n", debugstr_w(argv[i]) );
 255         goto not_supported;
 256     }
 257
 258     if (!wcsicmp( argv[i], L"path" ))
 259     {
 260         if (++i >= argc)
 261         {
 262             output_error( STRING_INVALID_PATH );
 263             return 1;
 264         }
 265         class = argv[i];
 266     }
 267     else
 268     {
 269         class = find_class( argv[i] );
 270         if (!class)
 271         {
 272             output_error( STRING_ALIAS_NOT_FOUND );
 273             return 1;
 274         }
 275     }
 276
 277     if (++i >= argc)
 278         goto not_supported;
 279
 280     if (!wcsicmp( argv[i], L"get" ))
 281     {
 282         if (++i >= argc)
 283             goto not_supported;
 284         value = argv[i];
 285         return query_prop( class, value );
 286     }
 287
 288 not_supported:
 289     output_error( STRING_CMDLINE_NOT_SUPPORTED );
 290     return 1;
 291 }