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
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
|
/* $Id: VBoxWatchdogInternal.h $ */
/** @file
* VBoxWatchdog - VirtualBox Watchdog Service.
*/
/*
* Copyright (C) 2011-2022 Oracle and/or its affiliates.
*
* This file is part of VirtualBox base platform packages, as
* available from https://www.virtualbox.org.
*
* 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, in version 3 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, see <https://www.gnu.org/licenses>.
*
* SPDX-License-Identifier: GPL-3.0-only
*/
#ifndef VBOX_INCLUDED_SRC_VBoxBalloonCtrl_VBoxWatchdogInternal_h
#define VBOX_INCLUDED_SRC_VBoxBalloonCtrl_VBoxWatchdogInternal_h
#ifndef RT_WITHOUT_PRAGMA_ONCE
# pragma once
#endif
#ifndef VBOX_ONLY_DOCS
# include <iprt/getopt.h>
# include <iprt/time.h>
# include <VBox/err.h>
# include <VBox/com/com.h>
# include <VBox/com/string.h>
# include <VBox/com/Guid.h>
# include <VBox/com/array.h>
# include <VBox/com/ErrorInfo.h>
# include <VBox/com/VirtualBox.h>
#endif /* !VBOX_ONLY_DOCS */
#include <map>
#include <vector>
using namespace com;
////////////////////////////////////////////////////////////////////////////////
//
// definitions
//
////////////////////////////////////////////////////////////////////////////////
/** Command handler argument. */
struct HandlerArg
{
int argc;
char **argv;
};
/**
* A module's payload for a machine entry.
* The payload data is not (yet) thread safe -- so only
* use this in one module at a time only!
*/
typedef struct VBOXWATCHDOG_MODULE_PAYLOAD
{
/** Pointer to allocated payload. Can be NULL if
* a module doesn't have an own payload. */
void *pvData;
/** Size of payload (in bytes). */
size_t cbData;
/** @todo Add mutex for locking + getPayloadLocked(). */
} VBOXWATCHDOG_MODULE_PAYLOAD, *PVBOXWATCHDOG_MODULE_PAYLOAD;
/**
* Map containing a module's individual payload -- the module itself
* is responsible for allocating/handling/destroying this payload.
* Primary key is the module name.
*/
typedef std::map<const char*, VBOXWATCHDOG_MODULE_PAYLOAD> mapPayload;
typedef std::map<const char*, VBOXWATCHDOG_MODULE_PAYLOAD>::iterator mapPayloadIter;
typedef std::map<const char*, VBOXWATCHDOG_MODULE_PAYLOAD>::const_iterator mapPayloadIterConst;
/** Group list (plus additional per-group flags, not used yet) for one VM.
* Primary key is the group name, secondary specify flags (if any). */
typedef std::map<Utf8Str, uint32_t> mapGroups;
typedef std::map<Utf8Str, uint32_t>::iterator mapGroupsIter;
typedef std::map<Utf8Str, uint32_t>::const_iterator mapGroupsIterConst;
/** A machine's internal entry.
* Primary key is the machine's UUID. */
typedef struct VBOXWATCHDOG_MACHINE
{
ComPtr<IMachine> machine;
/** The machine's name. For logging. */
Bstr strName;
#ifndef VBOX_WATCHDOG_GLOBAL_PERFCOL
ComPtr<IPerformanceCollector> collector;
#endif
/** The group(s) this machine belongs to. */
mapGroups groups;
/** Map containing the individual module payloads. */
mapPayload payload;
} VBOXWATCHDOG_MACHINE, *PVBOXWATCHDOG_MACHINE;
typedef std::map<Bstr, VBOXWATCHDOG_MACHINE> mapVM;
typedef std::map<Bstr, VBOXWATCHDOG_MACHINE>::iterator mapVMIter;
typedef std::map<Bstr, VBOXWATCHDOG_MACHINE>::const_iterator mapVMIterConst;
/** Members of a VM group; currently only represented by the machine's UUID.
* Primary key is the machine's UUID. */
typedef std::vector<Bstr> vecGroupMembers;
typedef std::vector<Bstr>::iterator vecGroupMembersIter;
typedef std::vector<Bstr>::const_iterator vecGroupMembersIterConst;
/** A VM group. Can contain none, one or more group members.
* Primary key is the group's name. */
typedef std::map<Utf8Str, vecGroupMembers> mapGroup;
typedef std::map<Utf8Str, vecGroupMembers>::iterator mapGroupIter;
typedef std::map<Utf8Str, vecGroupMembers>::const_iterator mapGroupIterConst;
/**
* A module descriptor.
*/
typedef struct
{
/** The short module name. */
const char *pszName;
/** The longer module name. */
const char *pszDescription;
/** A comma-separated list of modules this module
* depends on. Might be NULL if no dependencies. */
const char *pszDepends;
/** Priority (lower is higher, 0 is invalid) of
* module execution. */
uint32_t uPriority;
/** The usage options stuff for the --help screen. */
const char *pszUsage;
/** The option descriptions for the --help screen. */
const char *pszOptions;
/**
* Called before parsing arguments.
* @returns VBox status code.
*/
DECLCALLBACKMEMBER(int, pfnPreInit,(void));
/**
* Tries to parse the given command line options.
*
* @returns 0 if we parsed, -1 if it didn't and anything else means exit.
* @param argc Argument count.
* @param argv Arguments.
* @param piConsumed How many parameters this callback consumed from the
* remaining arguments passed in.
*/
DECLCALLBACKMEMBER(int, pfnOption,(int argc, char *argv[], int *piConsumed));
/**
* Called before parsing arguments.
* @returns VBox status code.
*/
DECLCALLBACKMEMBER(int, pfnInit,(void));
/** Called from the watchdog's main function. Non-blocking.
*
* @returns VBox status code.
*/
DECLCALLBACKMEMBER(int, pfnMain,(void));
/**
* Stop the module.
*/
DECLCALLBACKMEMBER(int, pfnStop,(void));
/**
* Does termination cleanups.
*
* @remarks This may be called even if pfnInit hasn't been called!
*/
DECLCALLBACKMEMBER(void, pfnTerm,(void));
/** @name Callbacks.
* @{
*/
/**
*
* @returns VBox status code.
*/
DECLCALLBACKMEMBER(int, pfnOnMachineRegistered,(const Bstr &strUuid));
/**
*
* @returns VBox status code.
*/
DECLCALLBACKMEMBER(int, pfnOnMachineUnregistered,(const Bstr &strUuid));
/**
*
* @returns VBox status code.
*/
DECLCALLBACKMEMBER(int, pfnOnMachineStateChanged,(const Bstr &strUuid, MachineState_T enmState));
/**
*
* @returns VBox status code.
*/
DECLCALLBACKMEMBER(int, pfnOnServiceStateChanged,(bool fAvailable));
/** @} */
} VBOXMODULE;
/** Pointer to a VBOXMODULE. */
typedef VBOXMODULE *PVBOXMODULE;
/** Pointer to a const VBOXMODULE. */
typedef VBOXMODULE const *PCVBOXMODULE;
RT_C_DECLS_BEGIN
extern bool g_fDryrun;
extern bool g_fVerbose;
extern ComPtr<IVirtualBox> g_pVirtualBox;
extern ComPtr<ISession> g_pSession;
extern mapVM g_mapVM;
extern mapGroup g_mapGroup;
# ifdef VBOX_WATCHDOG_GLOBAL_PERFCOL
extern ComPtr<IPerformanceCollector> g_pPerfCollector;
# endif
extern VBOXMODULE g_ModBallooning;
extern VBOXMODULE g_ModAPIMonitor;
extern void serviceLog(const char *pszFormat, ...);
#define serviceLogVerbose(a) if (g_fVerbose) { serviceLog a; }
int groupAdd(mapGroups &groups, const char *pszGroupsToAdd, uint32_t fFlags);
extern int getMetric(PVBOXWATCHDOG_MACHINE pMachine, const Bstr& strName, LONG *pulData);
void* payloadFrom(PVBOXWATCHDOG_MACHINE pMachine, const char *pszModule);
int payloadAlloc(PVBOXWATCHDOG_MACHINE pMachine, const char *pszModule, size_t cbSize, void **ppszPayload);
void payloadFree(PVBOXWATCHDOG_MACHINE pMachine, const char *pszModule);
PVBOXWATCHDOG_MACHINE getMachine(const Bstr& strUuid);
MachineState_T getMachineState(const PVBOXWATCHDOG_MACHINE pMachine);
int cfgGetValueStr(const ComPtr<IVirtualBox> &rptrVBox, const ComPtr<IMachine> &rptrMachine,
const char *pszGlobal, const char *pszVM, Utf8Str &strValue, Utf8Str strDefault);
int cfgGetValueU32(const ComPtr<IVirtualBox> &rptrVBox, const ComPtr<IMachine> &rptrMachine,
const char *pszGlobal, const char *pszVM, uint32_t *puValue, uint32_t uDefault);
RT_C_DECLS_END
#endif /* !VBOX_INCLUDED_SRC_VBoxBalloonCtrl_VBoxWatchdogInternal_h */
|
