2 * Copyright 2012 Hans Leidekker for CodeWeavers
4 * This library is free software; you can redistribute it and/or
5 * modify it under the terms of the GNU Lesser General Public
6 * License as published by the Free Software Foundation; either
7 * version 2.1 of the License, or (at your option) any later version.
9 * This library is distributed in the hope that it will be useful,
10 * but WITHOUT ANY WARRANTY; without even the implied warranty of
11 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
12 * Lesser General Public License for more details.
14 * You should have received a copy of the GNU Lesser General Public
15 * License along with this library; if not, write to the Free Software
16 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
27 #include "wine/test.h"
29 static const WCHAR wqlW[] = {'w','q','l',0};
31 static HRESULT exec_query( IWbemServices *services, const WCHAR *str, IEnumWbemClassObject **result )
33 static const WCHAR captionW[] = {'C','a','p','t','i','o','n',0};
34 static const WCHAR descriptionW[] = {'D','e','s','c','r','i','p','t','i','o','n',0};
36 IWbemClassObject *obj;
37 BSTR wql = SysAllocString( wqlW ), query = SysAllocString( str );
38 LONG flags = WBEM_FLAG_RETURN_IMMEDIATELY | WBEM_FLAG_FORWARD_ONLY;
41 hr = IWbemServices_ExecQuery( services, wql, query, flags, NULL, result );
44 trace("%s\n", wine_dbgstr_w(str));
49 IEnumWbemClassObject_Next( *result, WBEM_INFINITE, 1, &obj, &count );
52 if (IWbemClassObject_Get( obj, captionW, 0, &var, NULL, NULL ) == WBEM_S_NO_ERROR)
54 trace("caption: %s\n", wine_dbgstr_w(V_BSTR(&var)));
57 if (IWbemClassObject_Get( obj, descriptionW, 0, &var, NULL, NULL ) == WBEM_S_NO_ERROR)
59 trace("description: %s\n", wine_dbgstr_w(V_BSTR(&var)));
62 IWbemClassObject_Release( obj );
66 SysFreeString( query );
70 static void test_select( IWbemServices *services )
72 static const WCHAR sqlW[] = {'S','Q','L',0};
73 static const WCHAR query1[] =
74 {'S','E','L','E','C','T',' ','H','O','T','F','I','X','I','D',' ','F','R','O','M',' ',
75 'W','i','n','3','2','_','Q','u','i','c','k','F','i','x','E','n','g','i','n','e','e','r','i','n','g',0};
76 static const WCHAR query2[] =
77 {'S','E','L','E','C','T',' ','*',' ','F','R','O','M',' ','W','i','n','3','2','_','B','I','O','S',0};
78 static const WCHAR query3[] =
79 {'S','E','L','E','C','T',' ','*',' ','F','R','O','M',' ','W','i','n','3','2','_',
80 'L','o','g','i','c','a','l','D','i','s','k',' ','W','H','E','R','E',' ',
81 '\"','N','T','F','S','\"',' ','=',' ','F','i','l','e','S','y','s','t','e','m',0};
82 static const WCHAR query4[] =
83 {'S','E','L','E','C','T',' ','a',' ','F','R','O','M',' ','b',0};
84 static const WCHAR query5[] =
85 {'S','E','L','E','C','T',' ','a',' ','F','R','O','M',' ','W','i','n','3','2','_','B','i','o','s',0};
86 static const WCHAR query6[] =
87 {'S','E','L','E','C','T',' ','D','e','s','c','r','i','p','t','i','o','n',' ','F','R','O','M',' ',
88 'W','i','n','3','2','_','B','i','o','s',0};
89 static const WCHAR query7[] =
90 {'S','E','L','E','C','T',' ','*',' ','F','R','O','M',' ','W','i','n','3','2','_',
91 'P','r','o','c','e','s','s',' ','W','H','E','R','E',' ','C','a','p','t','i','o','n',' ',
92 'L','I','K','E',' ','\'','%','%','R','E','G','E','D','I','T','%','\'',0};
93 static const WCHAR *test[] = { query1, query2, query3, query4, query5, query6, query7 };
95 IEnumWbemClassObject *result;
96 BSTR wql = SysAllocString( wqlW );
97 BSTR sql = SysAllocString( sqlW );
98 BSTR query = SysAllocString( query1 );
101 hr = IWbemServices_ExecQuery( services, NULL, NULL, 0, NULL, &result );
102 ok( hr == WBEM_E_INVALID_PARAMETER, "query failed %08x\n", hr );
104 hr = IWbemServices_ExecQuery( services, NULL, query, 0, NULL, &result );
105 ok( hr == WBEM_E_INVALID_PARAMETER, "query failed %08x\n", hr );
107 hr = IWbemServices_ExecQuery( services, wql, NULL, 0, NULL, &result );
108 ok( hr == WBEM_E_INVALID_PARAMETER, "query failed %08x\n", hr );
110 hr = IWbemServices_ExecQuery( services, sql, query, 0, NULL, &result );
111 ok( hr == WBEM_E_INVALID_QUERY_TYPE, "query failed %08x\n", hr );
113 hr = IWbemServices_ExecQuery( services, sql, NULL, 0, NULL, &result );
114 ok( hr == WBEM_E_INVALID_PARAMETER, "query failed %08x\n", hr );
116 for (i = 0; i < sizeof(test)/sizeof(test[0]); i++)
118 hr = exec_query( services, test[i], &result );
119 ok( hr == S_OK, "query %u failed: %08x\n", i, hr );
120 if (result) IEnumWbemClassObject_Release( result );
123 SysFreeString( wql );
124 SysFreeString( sql );
125 SysFreeString( query );
128 static void test_Win32_Service( IWbemServices *services )
130 static const WCHAR returnvalueW[] = {'R','e','t','u','r','n','V','a','l','u','e',0};
131 static const WCHAR pauseserviceW[] = {'P','a','u','s','e','S','e','r','v','i','c','e',0};
132 static const WCHAR resumeserviceW[] = {'R','e','s','u','m','e','S','e','r','v','i','c','e',0};
133 static const WCHAR startserviceW[] = {'S','t','a','r','t','S','e','r','v','i','c','e',0};
134 static const WCHAR stopserviceW[] = {'S','t','o','p','S','e','r','v','i','c','e',0};
135 static const WCHAR stateW[] = {'S','t','a','t','e',0};
136 static const WCHAR stoppedW[] = {'S','t','o','p','p','e','d',0};
137 static const WCHAR serviceW[] = {'W','i','n','3','2','_','S','e','r','v','i','c','e','.',
138 'N','a','m','e','=','"','S','p','o','o','l','e','r','"',0};
139 BSTR class = SysAllocString( serviceW ), method;
140 IWbemClassObject *service, *out;
141 VARIANT state, retval;
145 hr = IWbemServices_GetObject( services, class, 0, NULL, &service, NULL );
148 win_skip( "Win32_Service not available\n" );
152 VariantInit( &state );
153 hr = IWbemClassObject_Get( service, stateW, 0, &state, &type, NULL );
154 ok( hr == S_OK, "failed to get service state %08x\n", hr );
155 ok( V_VT( &state ) == VT_BSTR, "unexpected variant type 0x%x\n", V_VT( &state ) );
156 ok( type == CIM_STRING, "unexpected type 0x%x\n", type );
158 if (!lstrcmpW( V_BSTR( &state ), stoppedW ))
161 method = SysAllocString( startserviceW );
162 hr = IWbemServices_ExecMethod( services, class, method, 0, NULL, NULL, &out, NULL );
163 ok( hr == S_OK, "failed to execute method %08x\n", hr );
164 SysFreeString( method );
166 VariantInit( &retval );
167 hr = IWbemClassObject_Get( out, returnvalueW, 0, &retval, NULL, NULL );
168 ok( hr == S_OK, "failed to get return value %08x\n", hr );
169 ok( !V_I4( &retval ), "unexpected error %u\n", V_UI4( &retval ) );
170 IWbemClassObject_Release( out );
173 method = SysAllocString( pauseserviceW );
174 hr = IWbemServices_ExecMethod( services, class, method, 0, NULL, NULL, &out, NULL );
175 ok( hr == S_OK, "failed to execute method %08x\n", hr );
176 SysFreeString( method );
178 VariantInit( &retval );
179 hr = IWbemClassObject_Get( out, returnvalueW, 0, &retval, NULL, NULL );
180 ok( hr == S_OK, "failed to get return value %08x\n", hr );
181 ok( V_I4( &retval ), "unexpected success\n" );
182 IWbemClassObject_Release( out );
185 method = SysAllocString( resumeserviceW );
186 hr = IWbemServices_ExecMethod( services, class, method, 0, NULL, NULL, &out, NULL );
187 ok( hr == S_OK, "failed to execute method %08x\n", hr );
188 SysFreeString( method );
190 VariantInit( &retval );
191 hr = IWbemClassObject_Get( out, returnvalueW, 0, &retval, NULL, NULL );
192 ok( hr == S_OK, "failed to get return value %08x\n", hr );
193 ok( V_I4( &retval ), "unexpected success\n" );
194 IWbemClassObject_Release( out );
196 if (!lstrcmpW( V_BSTR( &state ), stoppedW ))
199 method = SysAllocString( stopserviceW );
200 hr = IWbemServices_ExecMethod( services, class, method, 0, NULL, NULL, &out, NULL );
201 ok( hr == S_OK, "failed to execute method %08x\n", hr );
202 SysFreeString( method );
204 VariantInit( &retval );
205 hr = IWbemClassObject_Get( out, returnvalueW, 0, &retval, NULL, NULL );
206 ok( hr == S_OK, "failed to get return value %08x\n", hr );
207 ok( !V_I4( &retval ), "unexpected error %u\n", V_UI4( &retval ) );
208 IWbemClassObject_Release( out );
210 VariantClear( &state );
211 IWbemClassObject_Release( service );
212 SysFreeString( class );
215 static void test_Win32_Process( IWbemServices *services )
217 static const WCHAR returnvalueW[] = {'R','e','t','u','r','n','V','a','l','u','e',0};
218 static const WCHAR getownerW[] = {'G','e','t','O','w','n','e','r',0};
219 static const WCHAR userW[] = {'U','s','e','r',0};
220 static const WCHAR domainW[] = {'D','o','m','a','i','n',0};
221 static const WCHAR processW[] = {'W','i','n','3','2','_','P','r','o','c','e','s','s',0};
222 static const WCHAR fmtW[] = {'W','i','n','3','2','_','P','r','o','c','e','s','s','.',
223 'H','a','n','d','l','e','=','"','%','u','"',0};
225 IWbemClassObject *process, *out;
226 VARIANT user, domain, retval;
230 class = SysAllocString( processW );
231 hr = IWbemServices_GetObject( services, class, 0, NULL, &process, NULL );
232 SysFreeString( class );
235 win_skip( "Win32_Process not available\n" );
238 hr = IWbemClassObject_GetMethod( process, getownerW, 0, NULL, NULL );
239 ok( hr == S_OK, "failed to get GetOwner method %08x\n", hr );
242 method = SysAllocString( getownerW );
243 class = SysAllocStringLen( NULL, sizeof(fmtW)/sizeof(fmtW[0]) + 10 );
244 wsprintfW( class, fmtW, GetCurrentProcessId() );
245 hr = IWbemServices_ExecMethod( services, class, method, 0, NULL, NULL, &out, NULL );
246 ok( hr == S_OK, "failed to execute method %08x\n", hr );
247 SysFreeString( method );
248 SysFreeString( class );
251 VariantInit( &retval );
252 hr = IWbemClassObject_Get( out, returnvalueW, 0, &retval, &type, NULL );
253 ok( hr == S_OK, "failed to get return value %08x\n", hr );
254 ok( V_VT( &retval ) == VT_I4, "unexpected variant type 0x%x\n", V_VT( &retval ) );
255 ok( !V_I4( &retval ), "unexpected error %u\n", V_I4( &retval ) );
256 ok( type == CIM_UINT32, "unexpected type 0x%x\n", type );
259 VariantInit( &user );
260 hr = IWbemClassObject_Get( out, userW, 0, &user, &type, NULL );
261 ok( hr == S_OK, "failed to get user %08x\n", hr );
262 ok( V_VT( &user ) == VT_BSTR, "unexpected variant type 0x%x\n", V_VT( &user ) );
263 ok( type == CIM_STRING, "unexpected type 0x%x\n", type );
264 trace("%s\n", wine_dbgstr_w(V_BSTR(&user)));
267 VariantInit( &domain );
268 hr = IWbemClassObject_Get( out, domainW, 0, &domain, &type, NULL );
269 ok( hr == S_OK, "failed to get domain %08x\n", hr );
270 ok( V_VT( &domain ) == VT_BSTR, "unexpected variant type 0x%x\n", V_VT( &domain ) );
271 ok( type == CIM_STRING, "unexpected type 0x%x\n", type );
272 trace("%s\n", wine_dbgstr_w(V_BSTR(&domain)));
274 VariantClear( &user );
275 VariantClear( &domain );
276 IWbemClassObject_Release( out );
279 static void test_StdRegProv( IWbemServices *services )
281 static const WCHAR enumkeyW[] = {'E','n','u','m','K','e','y',0};
282 static const WCHAR enumvaluesW[] = {'E','n','u','m','V','a','l','u','e','s',0};
283 static const WCHAR getstringvalueW[] = {'G','e','t','S','t','r','i','n','g','V','a','l','u','e',0};
284 static const WCHAR stdregprovW[] = {'S','t','d','R','e','g','P','r','o','v',0};
285 static const WCHAR defkeyW[] = {'h','D','e','f','K','e','y',0};
286 static const WCHAR subkeynameW[] = {'s','S','u','b','K','e','y','N','a','m','e',0};
287 static const WCHAR returnvalueW[] = {'R','e','t','u','r','n','V','a','l','u','e',0};
288 static const WCHAR namesW[] = {'s','N','a','m','e','s',0};
289 static const WCHAR typesW[] = {'T','y','p','e','s',0};
290 static const WCHAR valueW[] = {'s','V','a','l','u','e',0};
291 static const WCHAR valuenameW[] = {'s','V','a','l','u','e','N','a','m','e',0};
292 static const WCHAR programfilesW[] = {'P','r','o','g','r','a','m','F','i','l','e','s','D','i','r',0};
293 static const WCHAR windowsW[] =
294 {'S','o','f','t','w','a','r','e','\\','M','i','c','r','o','s','o','f','t','\\',
295 'W','i','n','d','o','w','s','\\','C','u','r','r','e','n','t','V','e','r','s','i','o','n',0};
296 BSTR class = SysAllocString( stdregprovW ), method;
297 IWbemClassObject *reg, *sig_in, *in, *out;
298 VARIANT defkey, subkey, retval, names, types, value, valuename;
302 hr = IWbemServices_GetObject( services, class, 0, NULL, ®, NULL );
305 win_skip( "StdRegProv not available\n" );
308 hr = IWbemClassObject_GetMethod( reg, enumkeyW, 0, &sig_in, NULL );
309 ok( hr == S_OK, "failed to get EnumKey method %08x\n", hr );
311 hr = IWbemClassObject_SpawnInstance( sig_in, 0, &in );
312 ok( hr == S_OK, "failed to spawn instance %08x\n", hr );
314 V_VT( &defkey ) = VT_I4;
315 V_I4( &defkey ) = 0x80000002;
316 hr = IWbemClassObject_Put( in, defkeyW, 0, &defkey, 0 );
317 ok( hr == S_OK, "failed to set root %08x\n", hr );
319 V_VT( &subkey ) = VT_BSTR;
320 V_BSTR( &subkey ) = SysAllocString( windowsW );
321 hr = IWbemClassObject_Put( in, subkeynameW, 0, &subkey, 0 );
322 ok( hr == S_OK, "failed to set subkey %08x\n", hr );
325 method = SysAllocString( enumkeyW );
326 hr = IWbemServices_ExecMethod( services, class, method, 0, NULL, in, &out, NULL );
327 ok( hr == S_OK, "failed to execute method %08x\n", hr );
328 SysFreeString( method );
331 VariantInit( &retval );
332 hr = IWbemClassObject_Get( out, returnvalueW, 0, &retval, &type, NULL );
333 ok( hr == S_OK, "failed to get return value %08x\n", hr );
334 ok( V_VT( &retval ) == VT_I4, "unexpected variant type 0x%x\n", V_VT( &retval ) );
335 ok( !V_I4( &retval ), "unexpected error %u\n", V_UI4( &retval ) );
336 ok( type == CIM_UINT32, "unexpected type 0x%x\n", type );
339 VariantInit( &names );
340 hr = IWbemClassObject_Get( out, namesW, 0, &names, &type, NULL );
341 ok( hr == S_OK, "failed to get names %08x\n", hr );
342 ok( V_VT( &names ) == (VT_BSTR|VT_ARRAY), "unexpected variant type 0x%x\n", V_VT( &names ) );
343 ok( type == (CIM_STRING|CIM_FLAG_ARRAY), "unexpected type 0x%x\n", type );
345 VariantClear( &names );
346 VariantClear( &subkey );
347 IWbemClassObject_Release( in );
348 IWbemClassObject_Release( out );
349 IWbemClassObject_Release( sig_in );
351 hr = IWbemClassObject_GetMethod( reg, enumvaluesW, 0, &sig_in, NULL );
352 ok( hr == S_OK, "failed to get EnumValues method %08x\n", hr );
354 hr = IWbemClassObject_SpawnInstance( sig_in, 0, &in );
355 ok( hr == S_OK, "failed to spawn instance %08x\n", hr );
357 V_VT( &defkey ) = VT_I4;
358 V_I4( &defkey ) = 0x80000002;
359 hr = IWbemClassObject_Put( in, defkeyW, 0, &defkey, 0 );
360 ok( hr == S_OK, "failed to set root %08x\n", hr );
362 V_VT( &subkey ) = VT_BSTR;
363 V_BSTR( &subkey ) = SysAllocString( windowsW );
364 hr = IWbemClassObject_Put( in, subkeynameW, 0, &subkey, 0 );
365 ok( hr == S_OK, "failed to set subkey %08x\n", hr );
368 method = SysAllocString( enumvaluesW );
369 hr = IWbemServices_ExecMethod( services, class, method, 0, NULL, in, &out, NULL );
370 ok( hr == S_OK, "failed to execute method %08x\n", hr );
371 SysFreeString( method );
374 VariantInit( &retval );
375 hr = IWbemClassObject_Get( out, returnvalueW, 0, &retval, &type, NULL );
376 ok( hr == S_OK, "failed to get return value %08x\n", hr );
377 ok( V_VT( &retval ) == VT_I4, "unexpected variant type 0x%x\n", V_VT( &retval ) );
378 ok( !V_I4( &retval ), "unexpected error %u\n", V_I4( &retval ) );
379 ok( type == CIM_UINT32, "unexpected type 0x%x\n", type );
382 VariantInit( &names );
383 hr = IWbemClassObject_Get( out, namesW, 0, &names, &type, NULL );
384 ok( hr == S_OK, "failed to get names %08x\n", hr );
385 ok( V_VT( &names ) == (VT_BSTR|VT_ARRAY), "unexpected variant type 0x%x\n", V_VT( &names ) );
386 ok( type == (CIM_STRING|CIM_FLAG_ARRAY), "unexpected type 0x%x\n", type );
389 VariantInit( &types );
390 hr = IWbemClassObject_Get( out, typesW, 0, &types, &type, NULL );
391 ok( hr == S_OK, "failed to get names %08x\n", hr );
392 ok( V_VT( &types ) == (VT_I4|VT_ARRAY), "unexpected variant type 0x%x\n", V_VT( &types ) );
393 ok( type == (CIM_SINT32|CIM_FLAG_ARRAY), "unexpected type 0x%x\n", type );
395 VariantClear( &types );
396 VariantClear( &names );
397 VariantClear( &subkey );
398 IWbemClassObject_Release( in );
399 IWbemClassObject_Release( out );
400 IWbemClassObject_Release( sig_in );
402 hr = IWbemClassObject_GetMethod( reg, getstringvalueW, 0, &sig_in, NULL );
403 ok( hr == S_OK, "failed to get GetStringValue method %08x\n", hr );
405 hr = IWbemClassObject_SpawnInstance( sig_in, 0, &in );
406 ok( hr == S_OK, "failed to spawn instance %08x\n", hr );
408 V_VT( &defkey ) = VT_I4;
409 V_I4( &defkey ) = 0x80000002;
410 hr = IWbemClassObject_Put( in, defkeyW, 0, &defkey, 0 );
411 ok( hr == S_OK, "failed to set root %08x\n", hr );
413 V_VT( &subkey ) = VT_BSTR;
414 V_BSTR( &subkey ) = SysAllocString( windowsW );
415 hr = IWbemClassObject_Put( in, subkeynameW, 0, &subkey, 0 );
416 ok( hr == S_OK, "failed to set subkey %08x\n", hr );
418 V_VT( &valuename ) = VT_BSTR;
419 V_BSTR( &valuename ) = SysAllocString( programfilesW );
420 hr = IWbemClassObject_Put( in, valuenameW, 0, &valuename, 0 );
421 ok( hr == S_OK, "failed to set value name %08x\n", hr );
424 method = SysAllocString( getstringvalueW );
425 hr = IWbemServices_ExecMethod( services, class, method, 0, NULL, in, &out, NULL );
426 ok( hr == S_OK, "failed to execute method %08x\n", hr );
427 SysFreeString( method );
430 VariantInit( &retval );
431 hr = IWbemClassObject_Get( out, returnvalueW, 0, &retval, &type, NULL );
432 ok( hr == S_OK, "failed to get return value %08x\n", hr );
433 ok( V_VT( &retval ) == VT_I4, "unexpected variant type 0x%x\n", V_VT( &retval ) );
434 ok( !V_I4( &retval ), "unexpected error %u\n", V_I4( &retval ) );
435 ok( type == CIM_UINT32, "unexpected type 0x%x\n", type );
438 VariantInit( &value );
439 hr = IWbemClassObject_Get( out, valueW, 0, &value, &type, NULL );
440 ok( hr == S_OK, "failed to get value %08x\n", hr );
441 ok( V_VT( &value ) == VT_BSTR, "unexpected variant type 0x%x\n", V_VT( &value ) );
442 ok( type == CIM_STRING, "unexpected type 0x%x\n", type );
444 VariantClear( &value );
445 VariantClear( &valuename );
446 VariantClear( &subkey );
447 IWbemClassObject_Release( in );
448 IWbemClassObject_Release( out );
449 IWbemClassObject_Release( sig_in );
451 IWbemClassObject_Release( reg );
452 SysFreeString( class );
457 static const WCHAR cimv2W[] = {'R','O','O','T','\\','C','I','M','V','2',0};
458 BSTR path = SysAllocString( cimv2W );
459 IWbemLocator *locator;
460 IWbemServices *services;
463 CoInitialize( NULL );
464 CoInitializeSecurity( NULL, -1, NULL, NULL, RPC_C_AUTHN_LEVEL_DEFAULT,
465 RPC_C_IMP_LEVEL_IMPERSONATE, NULL, EOAC_NONE, NULL );
466 hr = CoCreateInstance( &CLSID_WbemLocator, NULL, CLSCTX_INPROC_SERVER, &IID_IWbemLocator,
470 win_skip("can't create instance of WbemLocator\n");
473 hr = IWbemLocator_ConnectServer( locator, path, NULL, NULL, NULL, 0, NULL, NULL, &services );
474 ok( hr == S_OK, "failed to get IWbemServices interface %08x\n", hr );
476 hr = CoSetProxyBlanket( (IUnknown *)services, RPC_C_AUTHN_WINNT, RPC_C_AUTHZ_NONE, NULL,
477 RPC_C_AUTHN_LEVEL_CALL, RPC_C_IMP_LEVEL_IMPERSONATE, NULL, EOAC_NONE );
478 ok( hr == S_OK, "failed to set proxy blanket %08x\n", hr );
480 test_select( services );
481 test_Win32_Process( services );
482 test_Win32_Service( services );
483 test_StdRegProv( services );
485 SysFreeString( path );
486 IWbemServices_Release( services );
487 IWbemLocator_Release( locator );