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
|
/*
* Copyright (c) 2010, 2012, Oracle and/or its affiliates. All rights reserved.
*
* This program is free software; you can redistribute it and/or
* modify it under the terms of the GNU General Public License as
* published by the Free Software Foundation; version 2 of the
* License.
*
* This program 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 General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program; if not, write to the Free Software
* Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
* 02110-1301 USA
*/
#ifndef _WMI_H_
#define _WMI_H_
#include "grtpp.h"
#define _WIN32_DCOM
#include <atlbase.h>
#include <wbemidl.h>
#include <comutil.h>
#pragma comment(lib, "wbemuuid.lib")
/**
* Interface functions for Workbench modules to work with Windows Management Instrumentation.
* This code can only be used from the Windows version of WB.
*/
namespace wmi
{
class WmiServices;
/**
* Helper class to for fast monitoring of certain properties (like CPU load).
*/
class WmiMonitor
{
private:
WmiServices *_owner;
IWbemServices* _services;
CComPtr<IWbemRefresher> _refresher;
CComPtr<IWbemHiPerfEnum> _enumerator;
long _enumeratorId;
CComBSTR _propertyName;
long _propertyHandle;
long _namePropertyHandle;
bool _findTotal; // For multiple CPUs there is a total counter which we will return.
public:
WmiMonitor(IWbemServices* services, const std::string& parameter);
~WmiMonitor();
std::string readValue();
inline WmiServices* owner() { return _owner; }
};
/**
* Main class providing WMI based services (stats, service control, queries).
*/
class WmiServices
{
private:
CComPtr<IWbemServices> _services;
protected:
static void allocate_locator();
static void deallocate_locator();
public:
WmiServices(const std::string& server, const std::string& user, const std::string& password);
~WmiServices();
grt::DictListRef query(grt::GRT* grt, const std::string& query);
std::string serviceControl(const std::string& service, const std::string& action);
std::string systemStat(const std::string& what);
WmiMonitor* startMonitoring(const std::string& parameter);
void stopMonitoring(WmiMonitor* monitor);
};
}
#endif // #ifndef _WMI_H_
|
