1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
|
// Copyright (c) 2006-2008 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include <windows.h>
#include <atlbase.h>
#pragma comment(lib, "wbemuuid.lib")
#include "base/wmi_util.h"
bool WMIUtil::CreateLocalConnection(bool set_blanket,
IWbemServices** wmi_services) {
CComPtr<IWbemLocator> wmi_locator;
HRESULT hr = wmi_locator.CoCreateInstance(CLSID_WbemLocator, NULL,
CLSCTX_INPROC_SERVER);
if (FAILED(hr))
return false;
CComPtr<IWbemServices> wmi_services_r;
hr = wmi_locator->ConnectServer(CComBSTR(L"ROOT\\CIMV2"), NULL, NULL, 0, NULL,
0, 0, &wmi_services_r);
if (FAILED(hr))
return false;
if (set_blanket) {
hr = ::CoSetProxyBlanket(wmi_services_r,
RPC_C_AUTHN_WINNT,
RPC_C_AUTHZ_NONE,
NULL,
RPC_C_AUTHN_LEVEL_CALL,
RPC_C_IMP_LEVEL_IMPERSONATE,
NULL,
EOAC_NONE);
if (FAILED(hr))
return false;
}
*wmi_services = wmi_services_r.Detach();
return true;
}
bool WMIUtil::CreateClassMethodObject(IWbemServices* wmi_services,
const std::wstring& class_name,
const std::wstring& method_name,
IWbemClassObject** class_instance) {
// We attempt to instantiate a COM object that represents a WMI object plus
// a method rolled into one entity.
CComBSTR b_class_name(class_name.c_str());
CComBSTR b_method_name(method_name.c_str());
CComPtr<IWbemClassObject> class_object = NULL;
HRESULT hr;
hr = wmi_services->GetObject(b_class_name, 0, NULL, &class_object, NULL);
if (FAILED(hr))
return false;
CComPtr<IWbemClassObject> params_def = NULL;
hr = class_object->GetMethod(b_method_name, 0, ¶ms_def, NULL);
if (FAILED(hr))
return false;
if (NULL == params_def) {
// You hit this special case if the WMI class is not a CIM class. MSDN
// sometimes tells you this. Welcome to WMI hell.
return false;
}
hr = params_def->SpawnInstance(0, class_instance);
return(SUCCEEDED(hr));
}
bool SetParameter(IWbemClassObject* class_method,
const std::wstring& parameter_name, VARIANT* parameter) {
HRESULT hr = class_method->Put(parameter_name.c_str(), 0, parameter, 0);
return SUCCEEDED(hr);
}
// The code in Launch() basically calls the Create Method of the Win32_Process
// CIM class is documented here:
// http://msdn2.microsoft.com/en-us/library/aa389388(VS.85).aspx
bool WMIProcessUtil::Launch(const std::wstring& command_line, int* process_id) {
CComPtr<IWbemServices> wmi_local;
if (!WMIUtil::CreateLocalConnection(true, &wmi_local))
return false;
const wchar_t class_name[] = L"Win32_Process";
const wchar_t method_name[] = L"Create";
CComPtr<IWbemClassObject> process_create;
if (!WMIUtil::CreateClassMethodObject(wmi_local, class_name, method_name,
&process_create))
return false;
CComVariant b_command_line(command_line.c_str());
if (!SetParameter(process_create, L"CommandLine", &b_command_line))
return false;
CComPtr<IWbemClassObject> out_params;
HRESULT hr = wmi_local->ExecMethod(CComBSTR(class_name),
CComBSTR(method_name), 0, NULL,
process_create, &out_params, NULL);
if (FAILED(hr))
return false;
CComVariant ret_value;
hr = out_params->Get(L"ReturnValue", 0, &ret_value, NULL, 0);
if (FAILED(hr) || (0 != ret_value.uintVal))
return false;
CComVariant pid;
hr = out_params->Get(L"ProcessId", 0, &pid, NULL, 0);
if (FAILED(hr) || (0 == pid.intVal))
return false;
if (process_id)
*process_id = pid.intVal;
return true;
}
|
