VirtualBox

source: vbox/trunk/src/VBox/Main/testcase/tstCollector.cpp@ 25414

Last change on this file since 25414 was 16018, checked in by vboxsync, 15 years ago

Main: FreeBSD stubbing.
  • Property svn:eol-style set to native
  • Property svn:keywords set to Author Date Id Revision
File size: 11.3 KB
Line 
1/* $Id: tstCollector.cpp 16018 2009-01-18 00:30:41Z vboxsync $ */
2
3/** @file
4 *
5 * Collector classes test cases.
6 */
7
8/*
9 * Copyright (C) 2008 Sun Microsystems, Inc.
10 *
11 * This file is part of VirtualBox Open Source Edition (OSE), as
12 * available from http://www.virtualbox.org. This file is free software;
13 * you can redistribute it and/or modify it under the terms of the GNU
14 * General Public License (GPL) as published by the Free Software
15 * Foundation, in version 2 as it comes in the "COPYING" file of the
16 * VirtualBox OSE distribution. VirtualBox OSE is distributed in the
17 * hope that it will be useful, but WITHOUT ANY WARRANTY of any kind.
18 *
19 * Please contact Sun Microsystems, Inc., 4150 Network Circle, Santa
20 * Clara, CA 95054 USA or visit http://www.sun.com if you need
21 * additional information or have any questions.
22 */
23
24#ifdef RT_OS_DARWIN
25#include "../darwin/PerformanceDarwin.cpp"
26#endif
27#ifdef RT_OS_FREEBSD
28#include "../freebsd/PerformanceFreeBSD.cpp"
29#endif
30#ifdef RT_OS_LINUX
31#include "../linux/PerformanceLinux.cpp"
32#endif
33#ifdef RT_OS_OS2
34#include "../os2/PerformanceOS2.cpp"
35#endif
36#ifdef RT_OS_SOLARIS
37#include "../solaris/PerformanceSolaris.cpp"
38#endif
39#ifdef RT_OS_WINDOWS
40#define _WIN32_DCOM
41#include <objidl.h>
42#include <objbase.h>
43#include "../win/PerformanceWin.cpp"
44#endif
45
46#include <iprt/initterm.h>
47#include <iprt/stream.h>
48#include <iprt/env.h>
49#include <iprt/err.h>
50#include <iprt/process.h>
51#include <iprt/thread.h>
52#include <iprt/time.h>
53
54#define RUN_TIME_MS 1000
55
56#define N_CALLS(n, fn) \
57 for (int call = 0; call < n; ++call) \
58 rc = collector->fn; \
59 if (RT_FAILURE(rc)) \
60 RTPrintf("tstCollector: "#fn" -> %Rrc\n", rc)
61
62#define CALLS_PER_SECOND(fn) \
63 nCalls = 0; \
64 start = RTTimeMilliTS(); \
65 do { \
66 rc = collector->fn; \
67 if (RT_FAILURE(rc)) \
68 break; \
69 ++nCalls; \
70 } while(RTTimeMilliTS() - start < RUN_TIME_MS); \
71 if (RT_FAILURE(rc)) \
72 { \
73 RTPrintf("tstCollector: "#fn" -> %Rrc\n", rc); \
74 } \
75 else \
76 RTPrintf("%70s -- %u calls per second\n", #fn, nCalls)
77
78void measurePerformance(pm::CollectorHAL *collector, const char *pszName, int cVMs)
79{
80
81 static const char * const args[] = { pszName, "-child", NULL };
82 pm::CollectorHints hints;
83 std::vector<RTPROCESS> processes;
84
85 hints.collectHostCpuLoad();
86 hints.collectHostRamUsage();
87 /* Start fake VMs */
88 for (int i = 0; i < cVMs; ++i)
89 {
90 RTPROCESS pid;
91 int rc = RTProcCreate(pszName, args, RTENV_DEFAULT, 0, &pid);
92 if (RT_FAILURE(rc))
93 {
94 hints.getProcesses(processes);
95 std::for_each(processes.begin(), processes.end(), std::ptr_fun(RTProcTerminate));
96 RTPrintf("tstCollector: RTProcCreate() -> %Rrc\n", rc);
97 return;
98 }
99 hints.collectProcessCpuLoad(pid);
100 hints.collectProcessRamUsage(pid);
101 }
102
103 hints.getProcesses(processes);
104 RTThreadSleep(30000); // Let children settle for half a minute
105
106 int rc;
107 ULONG tmp;
108 uint64_t tmp64;
109 uint64_t start;
110 unsigned int nCalls;
111 /* Pre-collect */
112 CALLS_PER_SECOND(preCollect(hints));
113 /* Host CPU load */
114 CALLS_PER_SECOND(getRawHostCpuLoad(&tmp64, &tmp64, &tmp64));
115 /* Process CPU load */
116 CALLS_PER_SECOND(getRawProcessCpuLoad(processes[nCalls%cVMs], &tmp64, &tmp64, &tmp64));
117 /* Host CPU speed */
118 CALLS_PER_SECOND(getHostCpuMHz(&tmp));
119 /* Host RAM usage */
120 CALLS_PER_SECOND(getHostMemoryUsage(&tmp, &tmp, &tmp));
121 /* Process RAM usage */
122 CALLS_PER_SECOND(getProcessMemoryUsage(processes[nCalls%cVMs], &tmp));
123
124 start = RTTimeNanoTS();
125
126 int times;
127 for (times = 0; times < 100; times++)
128 {
129 /* Pre-collect */
130 N_CALLS(1, preCollect(hints));
131 /* Host CPU load */
132 N_CALLS(1, getRawHostCpuLoad(&tmp64, &tmp64, &tmp64));
133 /* Host CPU speed */
134 N_CALLS(1, getHostCpuMHz(&tmp));
135 /* Host RAM usage */
136 N_CALLS(1, getHostMemoryUsage(&tmp, &tmp, &tmp));
137 /* Process CPU load */
138 N_CALLS(cVMs, getRawProcessCpuLoad(processes[call], &tmp64, &tmp64, &tmp64));
139 /* Process RAM usage */
140 N_CALLS(cVMs, getProcessMemoryUsage(processes[call], &tmp));
141 }
142 printf("\n%u VMs -- %.2f%% of CPU time\n", cVMs, (RTTimeNanoTS() - start) / 10000000. / times);
143
144 /* Shut down fake VMs */
145 std::for_each(processes.begin(), processes.end(), std::ptr_fun(RTProcTerminate));
146}
147
148int main(int argc, char *argv[])
149{
150 /*
151 * Initialize the VBox runtime without loading
152 * the support driver.
153 */
154 int rc = RTR3Init();
155 if (RT_FAILURE(rc))
156 {
157 RTPrintf("tstCollector: RTR3Init() -> %d\n", rc);
158 return 1;
159 }
160 if (argc > 1 && !strcmp(argv[1], "-child"))
161 {
162 /* We have spawned ourselves as a child process -- scratch the leg */
163 RTThreadSleep(1000000);
164 return 1;
165 }
166#ifdef RT_OS_WINDOWS
167 HRESULT hRes = CoInitialize(NULL);
168 /*
169 * Need to initialize security to access performance enumerators.
170 */
171 hRes = CoInitializeSecurity(
172 NULL,
173 -1,
174 NULL,
175 NULL,
176 RPC_C_AUTHN_LEVEL_NONE,
177 RPC_C_IMP_LEVEL_IMPERSONATE,
178 NULL, EOAC_NONE, 0);
179#endif
180
181 pm::CollectorHAL *collector = pm::createHAL();
182 if (!collector)
183 {
184 RTPrintf("tstCollector: createMetricFactory() failed\n", rc);
185 return 1;
186 }
187#if 1
188 pm::CollectorHints hints;
189 hints.collectHostCpuLoad();
190 hints.collectHostRamUsage();
191 hints.collectProcessCpuLoad(RTProcSelf());
192 hints.collectProcessRamUsage(RTProcSelf());
193
194 uint64_t start;
195
196 uint64_t hostUserStart, hostKernelStart, hostIdleStart;
197 uint64_t hostUserStop, hostKernelStop, hostIdleStop, hostTotal;
198
199 uint64_t processUserStart, processKernelStart, processTotalStart;
200 uint64_t processUserStop, processKernelStop, processTotalStop;
201
202 RTPrintf("tstCollector: TESTING - CPU load, sleeping for 5 sec\n");
203
204 rc = collector->preCollect(hints);
205 if (RT_FAILURE(rc))
206 {
207 RTPrintf("tstCollector: preCollect() -> %Rrc\n", rc);
208 return 1;
209 }
210 rc = collector->getRawHostCpuLoad(&hostUserStart, &hostKernelStart, &hostIdleStart);
211 if (RT_FAILURE(rc))
212 {
213 RTPrintf("tstCollector: getRawHostCpuLoad() -> %Rrc\n", rc);
214 return 1;
215 }
216 rc = collector->getRawProcessCpuLoad(RTProcSelf(), &processUserStart, &processKernelStart, &processTotalStart);
217 if (RT_FAILURE(rc))
218 {
219 RTPrintf("tstCollector: getRawProcessCpuLoad() -> %Rrc\n", rc);
220 return 1;
221 }
222
223 RTThreadSleep(5000); // Sleep for 5 seconds
224
225 rc = collector->preCollect(hints);
226 if (RT_FAILURE(rc))
227 {
228 RTPrintf("tstCollector: preCollect() -> %Rrc\n", rc);
229 return 1;
230 }
231 rc = collector->getRawHostCpuLoad(&hostUserStop, &hostKernelStop, &hostIdleStop);
232 if (RT_FAILURE(rc))
233 {
234 RTPrintf("tstCollector: getRawHostCpuLoad() -> %Rrc\n", rc);
235 return 1;
236 }
237 rc = collector->getRawProcessCpuLoad(RTProcSelf(), &processUserStop, &processKernelStop, &processTotalStop);
238 if (RT_FAILURE(rc))
239 {
240 RTPrintf("tstCollector: getRawProcessCpuLoad() -> %Rrc\n", rc);
241 return 1;
242 }
243 hostTotal = hostUserStop - hostUserStart
244 + hostKernelStop - hostKernelStart
245 + hostIdleStop - hostIdleStart;
246 /*printf("tstCollector: host cpu user = %f sec\n", (hostUserStop - hostUserStart) / 10000000.);
247 printf("tstCollector: host cpu kernel = %f sec\n", (hostKernelStop - hostKernelStart) / 10000000.);
248 printf("tstCollector: host cpu idle = %f sec\n", (hostIdleStop - hostIdleStart) / 10000000.);
249 printf("tstCollector: host cpu total = %f sec\n", hostTotal / 10000000.);*/
250 RTPrintf("tstCollector: host cpu user = %llu %%\n", (hostUserStop - hostUserStart) * 100 / hostTotal);
251 RTPrintf("tstCollector: host cpu kernel = %llu %%\n", (hostKernelStop - hostKernelStart) * 100 / hostTotal);
252 RTPrintf("tstCollector: host cpu idle = %llu %%\n", (hostIdleStop - hostIdleStart) * 100 / hostTotal);
253 RTPrintf("tstCollector: process cpu user = %llu %%\n", (processUserStop - processUserStart) * 100 / (processTotalStop - processTotalStart));
254 RTPrintf("tstCollector: process cpu kernel = %llu %%\n\n", (processKernelStop - processKernelStart) * 100 / (processTotalStop - processTotalStart));
255
256 RTPrintf("tstCollector: TESTING - CPU load, looping for 5 sec\n");
257 rc = collector->preCollect(hints);
258 if (RT_FAILURE(rc))
259 {
260 RTPrintf("tstCollector: preCollect() -> %Rrc\n", rc);
261 return 1;
262 }
263 rc = collector->getRawHostCpuLoad(&hostUserStart, &hostKernelStart, &hostIdleStart);
264 if (RT_FAILURE(rc))
265 {
266 RTPrintf("tstCollector: getRawHostCpuLoad() -> %Rrc\n", rc);
267 return 1;
268 }
269 rc = collector->getRawProcessCpuLoad(RTProcSelf(), &processUserStart, &processKernelStart, &processTotalStart);
270 if (RT_FAILURE(rc))
271 {
272 RTPrintf("tstCollector: getRawProcessCpuLoad() -> %Rrc\n", rc);
273 return 1;
274 }
275 start = RTTimeMilliTS();
276 while(RTTimeMilliTS() - start < 5000)
277 ; // Loop for 5 seconds
278 rc = collector->preCollect(hints);
279 if (RT_FAILURE(rc))
280 {
281 RTPrintf("tstCollector: preCollect() -> %Rrc\n", rc);
282 return 1;
283 }
284 rc = collector->getRawHostCpuLoad(&hostUserStop, &hostKernelStop, &hostIdleStop);
285 if (RT_FAILURE(rc))
286 {
287 RTPrintf("tstCollector: getRawHostCpuLoad() -> %Rrc\n", rc);
288 return 1;
289 }
290 rc = collector->getRawProcessCpuLoad(RTProcSelf(), &processUserStop, &processKernelStop, &processTotalStop);
291 if (RT_FAILURE(rc))
292 {
293 RTPrintf("tstCollector: getRawProcessCpuLoad() -> %Rrc\n", rc);
294 return 1;
295 }
296 hostTotal = hostUserStop - hostUserStart
297 + hostKernelStop - hostKernelStart
298 + hostIdleStop - hostIdleStart;
299 RTPrintf("tstCollector: host cpu user = %llu %%\n", (hostUserStop - hostUserStart) * 100 / hostTotal);
300 RTPrintf("tstCollector: host cpu kernel = %llu %%\n", (hostKernelStop - hostKernelStart) * 100 / hostTotal);
301 RTPrintf("tstCollector: host cpu idle = %llu %%\n", (hostIdleStop - hostIdleStart) * 100 / hostTotal);
302 RTPrintf("tstCollector: process cpu user = %llu %%\n", (processUserStop - processUserStart) * 100 / (processTotalStop - processTotalStart));
303 RTPrintf("tstCollector: process cpu kernel = %llu %%\n\n", (processKernelStop - processKernelStart) * 100 / (processTotalStop - processTotalStart));
304
305 RTPrintf("tstCollector: TESTING - Memory usage\n");
306
307 ULONG total, used, available, processUsed;
308
309 rc = collector->getHostMemoryUsage(&total, &used, &available);
310 if (RT_FAILURE(rc))
311 {
312 RTPrintf("tstCollector: getHostMemoryUsage() -> %Rrc\n", rc);
313 return 1;
314 }
315 rc = collector->getProcessMemoryUsage(RTProcSelf(), &processUsed);
316 if (RT_FAILURE(rc))
317 {
318 RTPrintf("tstCollector: getProcessMemoryUsage() -> %Rrc\n", rc);
319 return 1;
320 }
321 RTPrintf("tstCollector: host mem total = %lu kB\n", total);
322 RTPrintf("tstCollector: host mem used = %lu kB\n", used);
323 RTPrintf("tstCollector: host mem available = %lu kB\n", available);
324 RTPrintf("tstCollector: process mem used = %lu kB\n", processUsed);
325#endif
326 RTPrintf("\ntstCollector: TESTING - Performance\n\n");
327
328 measurePerformance(collector, argv[0], 100);
329
330 delete collector;
331
332 printf ("\ntstCollector FINISHED.\n");
333
334 return rc;
335}
336
Note: See TracBrowser for help on using the repository browser.

© 2023 Oracle
Contact – Privacy policy – Terms of Use