ReactOS  0.4.12-dev-51-ge94618b
main.c
Go to the documentation of this file.
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 
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 },
79 };
80 
81 static const WCHAR *find_class( const WCHAR *alias )
82 {
83  unsigned int i;
84 
85  for (i = 0; i < ARRAY_SIZE(alias_map); 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, ARRAY_SIZE(buffer));
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;
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 );
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 ) + ARRAY_SIZE(select_allW);
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  {
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  {
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  {
282  return 1;
283  }
284  class = argv[i];
285  }
286  else
287  {
288  class = find_class( argv[i] );
289  if (!class)
290  {
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:
309  return 1;
310 }
static const WCHAR classW[]
Definition: lex.c:38
static int output_message(int msg)
Definition: main.c:156
static const WCHAR newlineW[]
Definition: reg.c:81
static int argc
Definition: ServiceArgs.c:12
static const WCHAR * find_class(const WCHAR *alias)
Definition: main.c:81
#define TRUE
Definition: types.h:120
static const WCHAR cpuW[]
Definition: main.c:39
static const WCHAR win32_logicaldiskW[]
Definition: main.c:54
WINE_UNICODE_INLINE unsigned int strlenW(const WCHAR *str)
Definition: unicode.h:212
#define WideCharToMultiByte
Definition: compat.h:101
BSTR WINAPI SysAllocStringLen(const OLECHAR *str, unsigned int len)
Definition: oleaut.c:343
WINE_DEFAULT_DEBUG_CHANNEL(wmic)
static const WCHAR win32_computersystemW[]
Definition: main.c:52
static WCHAR * find_prop(IWbemClassObject *class, const WCHAR *prop)
Definition: main.c:101
static const WCHAR osW[]
Definition: main.c:45
__wchar_t WCHAR
Definition: xmlstorage.h:180
static WCHAR * strdupW(const WCHAR *src)
Definition: main.c:92
BOOL WINAPI DECLSPEC_HOTPATCH WriteConsoleW(IN HANDLE hConsoleOutput, IN CONST VOID *lpBuffer, IN DWORD nNumberOfCharsToWrite, OUT LPDWORD lpNumberOfCharsWritten, LPVOID lpReserved)
Definition: readwrite.c:1449
GLuint GLuint GLsizei count
Definition: gl.h:1545
const GLint * first
Definition: glext.h:5794
#define RPC_C_AUTHN_LEVEL_DEFAULT
Definition: rpcdce.h:145
void WINAPI SysFreeString(BSTR str)
Definition: oleaut.c:275
#define WINE_TRACE
Definition: debug.h:353
HANDLE WINAPI GetStdHandle(IN DWORD nStdHandle)
Definition: console.c:152
GLuint buffer
Definition: glext.h:5915
HRESULT WINAPI VariantChangeType(VARIANTARG *pvargDest, VARIANTARG *pvargSrc, USHORT wFlags, VARTYPE vt)
Definition: variant.c:965
HRESULT WINAPI SafeArrayGetElement(SAFEARRAY *psa, LONG *rgIndices, void *pvData)
Definition: safearray.c:949
OLECHAR * BSTR
Definition: compat.h:1927
#define RPC_C_IMP_LEVEL_IMPERSONATE
Definition: rpcdce.h:176
HRESULT WINAPI CoInitializeSecurity(PSECURITY_DESCRIPTOR pSecDesc, LONG cAuthSvc, SOLE_AUTHENTICATION_SERVICE *asAuthSvc, void *pReserved1, DWORD dwAuthnLevel, DWORD dwImpLevel, void *pReserved2, DWORD dwCapabilities, void *pReserved3)
Definition: compobj.c:4034
int wmain(int argc, WCHAR *argv[])
Definition: main.c:247
static int output_string(const WCHAR *msg,...)
Definition: main.c:124
int WINAPI LoadStringW(_In_opt_ HINSTANCE hInstance, _In_ UINT uID, _Out_writes_to_(cchBufferMax, return+1) LPWSTR lpBuffer, _In_ int cchBufferMax)
HRESULT WINAPI VariantClear(VARIANTARG *pVarg)
Definition: variant.c:651
static char ** argv
Definition: ServiceArgs.c:11
GLsizei GLsizei GLuint * obj
Definition: glext.h:6042
static const struct @5 alias_map[]
static const WCHAR biosW[]
Definition: main.c:35
GLuint const GLubyte GLvoid * src
Definition: s_context.h:57
static const WCHAR getW[]
Definition: object.c:47
GLenum GLclampf GLint i
Definition: glfuncs.h:14
#define va_end(ap)
Definition: acmsvcex.h:90
long LONG
Definition: pedump.c:60
#define debugstr_w
Definition: kernel32.h:32
UINT WINAPI DECLSPEC_HOTPATCH GetConsoleOutputCP(VOID)
Definition: console.c:2453
static const WCHAR wqlW[]
Definition: query.c:30
const WCHAR * str
UINT msg
Definition: msvc.h:92
smooth NULL
Definition: ftsmooth.c:416
BSTR WINAPI SysAllocString(LPCOLESTR str)
Definition: oleaut.c:242
char * va_list
Definition: acmsvcex.h:78
BOOL WINAPI WriteFile(IN HANDLE hFile, IN LPCVOID lpBuffer, IN DWORD nNumberOfBytesToWrite OPTIONAL, OUT LPDWORD lpNumberOfBytesWritten OPTIONAL, IN LPOVERLAPPED lpOverlapped OPTIONAL)
Definition: rw.c:24
static const WCHAR win32_operatingsystemW[]
Definition: main.c:58
unsigned int BOOL
Definition: ntddk_ex.h:94
#define GetProcessHeap()
Definition: compat.h:395
PVOID WINAPI HeapAlloc(HANDLE, DWORD, SIZE_T)
static const WCHAR processW[]
Definition: main.c:47
LONG HRESULT
Definition: typedefs.h:77
static const WCHAR win32_biosW[]
Definition: main.c:50
#define vsprintfW
Definition: unicode.h:59
unsigned long DWORD
Definition: ntddk_ex.h:95
static const WCHAR computersystemW[]
Definition: main.c:37
GLbitfield flags
Definition: glext.h:7161
#define STRING_INVALID_QUERY
Definition: wmic.h:23
HRESULT WINAPI SafeArrayDestroy(SAFEARRAY *psa)
Definition: safearray.c:1349
int ret
static const WCHAR exitW[]
Definition: lex.c:50
GLenum GLsizei len
Definition: glext.h:6722
static const WCHAR logicaldiskW[]
Definition: main.c:41
#define STRING_CMDLINE_NOT_SUPPORTED
Definition: wmic.h:21
#define STD_OUTPUT_HANDLE
Definition: winbase.h:265
static int query_prop(const WCHAR *class, const WCHAR *propname)
Definition: main.c:165
GLsizei const GLfloat * value
Definition: glext.h:6069
const WCHAR * alias
Definition: main.c:67
#define V_BSTR(A)
Definition: oleauto.h:226
static const WCHAR nicW[]
Definition: main.c:43
#define strcmpiW(s1, s2)
Definition: unicode.h:39
HRESULT WINAPI DECLSPEC_HOTPATCH CoCreateInstance(REFCLSID rclsid, LPUNKNOWN pUnkOuter, DWORD dwClsContext, REFIID iid, LPVOID *ppv)
Definition: compobj.c:3237
#define S_OK
Definition: intsafe.h:59
HRESULT WINAPI SafeArrayGetUBound(SAFEARRAY *psa, UINT nDim, LONG *plUbound)
Definition: safearray.c:1035
WINE_UNICODE_INLINE WCHAR * strcpyW(WCHAR *dst, const WCHAR *src)
Definition: unicode.h:219
const GLdouble * v
Definition: gl.h:2040
void WINAPI DECLSPEC_HOTPATCH CoUninitialize(void)
Definition: compobj.c:1994
#define ARRAY_SIZE(a)
Definition: main.h:24
Definition: services.c:325
WINE_UNICODE_INLINE WCHAR * strcatW(WCHAR *dst, const WCHAR *src)
Definition: unicode.h:242
static const WCHAR win32_networkadapterW[]
Definition: main.c:56
#define va_start(ap, A)
Definition: acmsvcex.h:91
GLuint const GLubyte GLvoid const GLvoid * dst
Definition: s_context.h:57
#define STRING_INVALID_PATH
Definition: wmic.h:24
static const WCHAR win32_processorW[]
Definition: main.c:62
HRESULT WINAPI CoInitialize(LPVOID lpReserved)
Definition: compobj.c:1900
unsigned int ULONG
Definition: retypes.h:1
HMODULE WINAPI GetModuleHandleW(LPCWSTR lpModuleName)
Definition: loader.c:845
#define STRING_ALIAS_NOT_FOUND
Definition: wmic.h:22
GLuint64EXT * result
Definition: glext.h:11304
static const WCHAR win32_processW[]
Definition: main.c:60
#define HeapFree(x, y, z)
Definition: compat.h:394
static struct sockaddr_in sa
Definition: adnsresfilter.c:69
#define WINE_FIXME
Definition: debug.h:365