Something went wrong. Try again.
Reactos
Something went wrong. Try again.
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333/* * Copyright 2010 Louis Lenders * Copyright 2012 Hans Leidekker for CodeWeavers * * 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 <stdio.h>#include "windows.h"#include "ocidl.h"#include "initguid.h"#include "objidl.h"#include "wbemcli.h"#include "wmic.h"
#include "wine/debug.h"#include "wine/unicode.h"
WINE_DEFAULT_DEBUG_CHANNEL(wmic);
static const WCHAR biosW[] = {'b','i','o','s',0};static const WCHAR computersystemW[] = {'c','o','m','p','u','t','e','r','s','y','s','t','e','m',0};static const WCHAR cpuW[] = {'c','p','u',0};static const WCHAR logicaldiskW[] = {'L','o','g','i','c','a','l','D','i','s','k',0};static const WCHAR nicW[] = {'n','i','c',0};static const WCHAR osW[] = {'o','s',0};static const WCHAR processW[] = {'p','r','o','c','e','s','s',0};
static const WCHAR win32_biosW[] = {'W','i','n','3','2','_','B','I','O','S',0};static const WCHAR win32_computersystemW[] = {'W','i','n','3','2','_','C','o','m','p','u','t','e','r','S','y','s','t','e','m',0};static const WCHAR win32_logicaldiskW[] = {'W','i','n','3','2','_','L','o','g','i','c','a','l','D','i','s','k',0};static const WCHAR win32_networkadapterW[] = {'W','i','n','3','2','_','N','e','t','w','o','r','k','A','d','a','p','t','e','r',0};static const WCHAR win32_operatingsystemW[] = {'W','i','n','3','2','_','O','p','e','r','a','t','i','n','g','S','y','s','t','e','m',0};static const WCHAR win32_processW[] = {'W','i','n','3','2','_','P','r','o','c','e','s','s',0};static const WCHAR win32_processorW[] = {'W','i','n','3','2','_','P','r','o','c','e','s','s','o','r',0};
static const struct{ const WCHAR *alias; const WCHAR *class;}alias_map[] ={ { biosW, win32_biosW }, { computersystemW, win32_computersystemW }, { cpuW, win32_processorW }, { logicaldiskW, win32_logicaldiskW }, { nicW, win32_networkadapterW }, { osW, win32_operatingsystemW }, { processW, win32_processW }};
static const WCHAR *find_class( const WCHAR *alias ){ unsigned int i;
for (i = 0; i < ARRAY_SIZE(alias_map); i++) { if (!strcmpiW( alias, alias_map[i].alias )) return alias_map[i].class; } return NULL;}
static inline WCHAR *strdupW( const WCHAR *src ){ WCHAR *dst; if (!src) return NULL; if (!(dst = HeapAlloc( GetProcessHeap(), 0, (strlenW( src ) + 1) * sizeof(WCHAR) ))) return NULL; strcpyW( dst, src ); return dst;}
static WCHAR *find_prop( IWbemClassObject *class, const WCHAR *prop ){ SAFEARRAY *sa; WCHAR *ret = NULL; LONG i, last_index = 0; BSTR str;
if (IWbemClassObject_GetNames( class, NULL, WBEM_FLAG_ALWAYS, NULL, &sa ) != S_OK) return NULL;
SafeArrayGetUBound( sa, 1, &last_index ); for (i = 0; i <= last_index; i++) { SafeArrayGetElement( sa, &i, &str ); if (!strcmpiW( str, prop )) { ret = strdupW( str ); break; } } SafeArrayDestroy( sa ); return ret;}
static int output_string( HANDLE handle, const WCHAR *msg, ... ){ va_list va_args; int len; DWORD count; WCHAR buffer[8192];
va_start( va_args, msg ); len = vsnprintfW( buffer, ARRAY_SIZE(buffer), msg, va_args ); va_end( va_args );
if (!WriteConsoleW( handle, buffer, len, &count, NULL )) WriteFile( handle, buffer, len * sizeof(WCHAR), &count, FALSE );
return count;}
static int output_error( int msg ){ static const WCHAR fmtW[] = {'%','s',0}; WCHAR buffer[8192];
LoadStringW( GetModuleHandleW(NULL), msg, buffer, ARRAY_SIZE(buffer)); return output_string( GetStdHandle(STD_ERROR_HANDLE), fmtW, buffer );}
static int output_header( const WCHAR *prop, ULONG column_width ){ static const WCHAR bomW[] = {0xfeff}, fmtW[] = {'%','-','*','s','\r','\n',0}; int len; DWORD count; WCHAR buffer[8192];
len = snprintfW( buffer, ARRAY_SIZE(buffer), fmtW, column_width, prop );
if (!WriteConsoleW( GetStdHandle(STD_OUTPUT_HANDLE), buffer, len, &count, NULL )) /* redirected */ { WriteFile( GetStdHandle(STD_OUTPUT_HANDLE), bomW, sizeof(bomW), &count, FALSE ); WriteFile( GetStdHandle(STD_OUTPUT_HANDLE), buffer, len * sizeof(WCHAR), &count, FALSE ); count += sizeof(bomW); }
return count;}
static int output_line( const WCHAR *str, ULONG column_width ){ static const WCHAR fmtW[] = {'%','-','*','s','\r','\n',0}; return output_string( GetStdHandle(STD_OUTPUT_HANDLE), fmtW, column_width, str );}
static int query_prop( const WCHAR *class, const WCHAR *propname ){ static const WCHAR select_allW[] = {'S','E','L','E','C','T',' ','*',' ','F','R','O','M',' ',0}; static const WCHAR cimv2W[] = {'R','O','O','T','\\','C','I','M','V','2',0}; static const WCHAR wqlW[] = {'W','Q','L',0}; HRESULT hr; IWbemLocator *locator = NULL; IWbemServices *services = NULL; IEnumWbemClassObject *result = NULL; LONG flags = WBEM_FLAG_RETURN_IMMEDIATELY | WBEM_FLAG_FORWARD_ONLY; BSTR path = NULL, wql = NULL, query = NULL; WCHAR *prop = NULL; BOOL first = TRUE; int len, ret = -1; IWbemClassObject *obj; ULONG count, width = 0; VARIANT v;
WINE_TRACE("%s, %s\n", debugstr_w(class), debugstr_w(propname));
CoInitialize( NULL ); CoInitializeSecurity( NULL, -1, NULL, NULL, RPC_C_AUTHN_LEVEL_DEFAULT, RPC_C_IMP_LEVEL_IMPERSONATE, NULL, EOAC_NONE, NULL );
hr = CoCreateInstance( &CLSID_WbemLocator, NULL, CLSCTX_INPROC_SERVER, &IID_IWbemLocator, (void **)&locator ); if (hr != S_OK) goto done;
if (!(path = SysAllocString( cimv2W ))) goto done; hr = IWbemLocator_ConnectServer( locator, path, NULL, NULL, NULL, 0, NULL, NULL, &services ); if (hr != S_OK) goto done;
len = strlenW( class ) + ARRAY_SIZE(select_allW); if (!(query = SysAllocStringLen( NULL, len ))) goto done; strcpyW( query, select_allW ); strcatW( query, class );
if (!(wql = SysAllocString( wqlW ))) goto done; hr = IWbemServices_ExecQuery( services, wql, query, flags, NULL, &result ); if (hr != S_OK) goto done;
for (;;) /* get column width */ { IEnumWbemClassObject_Next( result, WBEM_INFINITE, 1, &obj, &count ); if (!count) break;
if (!prop && !(prop = find_prop( obj, propname ))) { output_error( STRING_INVALID_QUERY ); goto done; } if (IWbemClassObject_Get( obj, prop, 0, &v, NULL, NULL ) == WBEM_S_NO_ERROR) { VariantChangeType( &v, &v, 0, VT_BSTR ); width = max( strlenW( V_BSTR( &v ) ), width ); VariantClear( &v ); } IWbemClassObject_Release( obj ); } width += 2;
IEnumWbemClassObject_Reset( result ); for (;;) { IEnumWbemClassObject_Next( result, WBEM_INFINITE, 1, &obj, &count ); if (!count) break;
if (first) { output_header( prop, width ); first = FALSE; } if (IWbemClassObject_Get( obj, prop, 0, &v, NULL, NULL ) == WBEM_S_NO_ERROR) { VariantChangeType( &v, &v, 0, VT_BSTR ); output_line( V_BSTR( &v ), width ); VariantClear( &v ); } IWbemClassObject_Release( obj ); } ret = 0;
done: if (result) IEnumWbemClassObject_Release( result ); if (services) IWbemServices_Release( services ); if (locator) IWbemLocator_Release( locator ); SysFreeString( path ); SysFreeString( query ); SysFreeString( wql ); HeapFree( GetProcessHeap(), 0, prop ); CoUninitialize(); return ret;}
int wmain(int argc, WCHAR *argv[]){ static const WCHAR getW[] = {'g','e','t',0}; static const WCHAR quitW[] = {'q','u','i','t',0}; static const WCHAR exitW[] = {'e','x','i','t',0}; static const WCHAR pathW[] = {'p','a','t','h',0}; static const WCHAR classW[] = {'c','l','a','s','s',0}; static const WCHAR contextW[] = {'c','o','n','t','e','x','t',0}; const WCHAR *class, *value; int i;
for (i = 1; i < argc && argv[i][0] == '/'; i++) WINE_FIXME( "command line switch %s not supported\n", debugstr_w(argv[i]) );
if (i >= argc) goto not_supported;
if (!strcmpiW( argv[i], quitW ) || !strcmpiW( argv[i], exitW )) { return 0; }
if (!strcmpiW( argv[i], classW) || !strcmpiW( argv[i], contextW )) { WINE_FIXME( "command %s not supported\n", debugstr_w(argv[i]) ); goto not_supported; }
if (!strcmpiW( argv[i], pathW )) { if (++i >= argc) { output_error( STRING_INVALID_PATH ); return 1; } class = argv[i]; } else { class = find_class( argv[i] ); if (!class) { output_error( STRING_ALIAS_NOT_FOUND ); return 1; } }
if (++i >= argc) goto not_supported;
if (!strcmpiW( argv[i], getW )) { if (++i >= argc) goto not_supported; value = argv[i]; return query_prop( class, value ); }
not_supported: output_error( STRING_CMDLINE_NOT_SUPPORTED ); return 1;}