/* * Win32_Service methods implementation * * 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 "config.h" #include <stdarg.h> #include "windef.h" #include "winbase.h" #include "wbemcli.h" #include "winsvc.h" #include "wine/debug.h" #include "wbemprox_private.h" WINE_DEFAULT_DEBUG_CHANNEL(wbemprox); static UINT map_error( DWORD error ) { switch (error) { case ERROR_SUCCESS: return 0; case ERROR_ACCESS_DENIED: return 2; case ERROR_DEPENDENT_SERVICES_RUNNING: return 3; case ERROR_INVALID_SERVICE_CONTROL: return 4; case ERROR_SERVICE_CANNOT_ACCEPT_CTRL: return 5; case ERROR_SERVICE_NOT_ACTIVE: return 6; case ERROR_SERVICE_REQUEST_TIMEOUT: return 7; case ERROR_SERVICE_ALREADY_RUNNING: return 10; default: WARN("unknown error %u\n", error); break; } return 8; } static HRESULT control_service( const WCHAR *name, DWORD control, VARIANT *retval ) { SC_HANDLE manager, service = NULL; SERVICE_STATUS status; UINT error = 0; if (!(manager = OpenSCManagerW( NULL, NULL, SC_MANAGER_ENUMERATE_SERVICE ))) { error = map_error( GetLastError() ); goto done; } if (!(service = OpenServiceW( manager, name, SERVICE_STOP|SERVICE_START|SERVICE_PAUSE_CONTINUE ))) { error = map_error( GetLastError() ); goto done; } if (!ControlService( service, control, &status )) error = map_error( GetLastError() ); CloseServiceHandle( service ); done: set_variant( VT_UI4, error, NULL, retval ); if (manager) CloseServiceHandle( manager ); return S_OK; } HRESULT service_pause_service( IWbemClassObject *obj, IWbemClassObject *in, IWbemClassObject **out ) { VARIANT name, retval; IWbemClassObject *sig, *out_params = NULL; HRESULT hr; TRACE("%p, %p, %p\n", obj, in, out); hr = IWbemClassObject_Get( obj, prop_nameW, 0, &name, NULL, NULL ); if (hr != S_OK) return hr; hr = create_signature( class_serviceW, method_pauseserviceW, PARAM_OUT, &sig ); if (hr != S_OK) { VariantClear( &name ); return hr; } if (out) { hr = IWbemClassObject_SpawnInstance( sig, 0, &out_params ); if (hr != S_OK) { VariantClear( &name ); IWbemClassObject_Release( sig ); return hr; } } hr = control_service( V_BSTR(&name), SERVICE_CONTROL_PAUSE, &retval ); if (hr != S_OK) goto done; if (out_params) hr = IWbemClassObject_Put( out_params, param_returnvalueW, 0, &retval, CIM_UINT32 ); done: VariantClear( &name ); IWbemClassObject_Release( sig ); if (hr == S_OK && out) { *out = out_params; IWbemClassObject_AddRef( out_params ); } if (out_params) IWbemClassObject_Release( out_params ); return hr; } HRESULT service_resume_service( IWbemClassObject *obj, IWbemClassObject *in, IWbemClassObject **out ) { VARIANT name, retval; IWbemClassObject *sig, *out_params = NULL; HRESULT hr; TRACE("%p, %p, %p\n", obj, in, out); hr = IWbemClassObject_Get( obj, prop_nameW, 0, &name, NULL, NULL ); if (hr != S_OK) return hr; hr = create_signature( class_serviceW, method_resumeserviceW, PARAM_OUT, &sig ); if (hr != S_OK) { VariantClear( &name ); return hr; } if (out) { hr = IWbemClassObject_SpawnInstance( sig, 0, &out_params ); if (hr != S_OK) { VariantClear( &name ); IWbemClassObject_Release( sig ); return hr; } } hr = control_service( V_BSTR(&name), SERVICE_CONTROL_CONTINUE, &retval ); if (hr != S_OK) goto done; if (out_params) hr = IWbemClassObject_Put( out_params, param_returnvalueW, 0, &retval, CIM_UINT32 ); done: VariantClear( &name ); IWbemClassObject_Release( sig ); if (hr == S_OK && out) { *out = out_params; IWbemClassObject_AddRef( out_params ); } if (out_params) IWbemClassObject_Release( out_params ); return hr; } static HRESULT start_service( const WCHAR *name, VARIANT *retval ) { SC_HANDLE manager, service = NULL; UINT error = 0; if (!(manager = OpenSCManagerW( NULL, NULL, SC_MANAGER_ENUMERATE_SERVICE ))) { error = map_error( GetLastError() ); goto done; } if (!(service = OpenServiceW( manager, name, SERVICE_START ))) { error = map_error( GetLastError() ); goto done; } if (!StartServiceW( service, 0, NULL )) error = map_error( GetLastError() ); CloseServiceHandle( service ); done: set_variant( VT_UI4, error, NULL, retval ); if (manager) CloseServiceHandle( manager ); return S_OK; } HRESULT service_start_service( IWbemClassObject *obj, IWbemClassObject *in, IWbemClassObject **out ) { VARIANT name, retval; IWbemClassObject *sig, *out_params = NULL; HRESULT hr; TRACE("%p, %p, %p\n", obj, in, out); hr = IWbemClassObject_Get( obj, prop_nameW, 0, &name, NULL, NULL ); if (hr != S_OK) return hr; hr = create_signature( class_serviceW, method_startserviceW, PARAM_OUT, &sig ); if (hr != S_OK) { VariantClear( &name ); return hr; } if (out) { hr = IWbemClassObject_SpawnInstance( sig, 0, &out_params ); if (hr != S_OK) { VariantClear( &name ); IWbemClassObject_Release( sig ); return hr; } } hr = start_service( V_BSTR(&name), &retval ); if (hr != S_OK) goto done; if (out_params) hr = IWbemClassObject_Put( out_params, param_returnvalueW, 0, &retval, CIM_UINT32 ); done: VariantClear( &name ); IWbemClassObject_Release( sig ); if (hr == S_OK && out) { *out = out_params; IWbemClassObject_AddRef( out_params ); } if (out_params) IWbemClassObject_Release( out_params ); return hr; } HRESULT service_stop_service( IWbemClassObject *obj, IWbemClassObject *in, IWbemClassObject **out ) { VARIANT name, retval; IWbemClassObject *sig, *out_params = NULL; HRESULT hr; TRACE("%p, %p, %p\n", obj, in, out); hr = IWbemClassObject_Get( obj, prop_nameW, 0, &name, NULL, NULL ); if (hr != S_OK) return hr; hr = create_signature( class_serviceW, method_stopserviceW, PARAM_OUT, &sig ); if (hr != S_OK) { VariantClear( &name ); return hr; } if (out) { hr = IWbemClassObject_SpawnInstance( sig, 0, &out_params ); if (hr != S_OK) { VariantClear( &name ); IWbemClassObject_Release( sig ); return hr; } } hr = control_service( V_BSTR(&name), SERVICE_CONTROL_STOP, &retval ); if (hr != S_OK) goto done; if (out_params) hr = IWbemClassObject_Put( out_params, param_returnvalueW, 0, &retval, CIM_UINT32 ); done: VariantClear( &name ); IWbemClassObject_Release( sig ); if (hr == S_OK && out) { *out = out_params; IWbemClassObject_AddRef( out_params ); } if (out_params) IWbemClassObject_Release( out_params ); return hr; }