[WMIC] Sync with Wine Staging 3.17. CORE-15127
[reactos.git] / base / applications / cmdutils / 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 < 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;
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 ) + 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 {
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 }