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
|
// Copyright Epic Games, Inc. All Rights Reserved.
#include "bench_cmd.h"
#include "bench.h"
#include <zencore/except.h>
#include <zencore/filesystem.h>
#include <zencore/fmtutils.h>
#include <zencore/logging.h>
#include <zencore/process.h>
#include <zencore/string.h>
#include <zencore/thread.h>
#include <zencore/timer.h>
namespace zen {
BenchCommand::BenchCommand()
{
m_Options.add_options()("h,help", "Print help");
m_Options.add_options()("purge",
"Purge standby memory (system cache)",
cxxopts::value<bool>(m_PurgeStandbyLists)->default_value("false"));
m_Options.add_options()("single", "Do not spawn child processes", cxxopts::value<bool>(m_SingleProcess)->default_value("false"));
}
BenchCommand::~BenchCommand() = default;
int
BenchCommand::Run(const ZenCliOptions& GlobalOptions, int argc, char** argv)
{
ZEN_UNUSED(GlobalOptions);
if (!ParseOptions(argc, argv))
{
return 0;
}
#if ZEN_PLATFORM_WINDOWS
if (m_PurgeStandbyLists)
{
bool Ok = false;
zen::Stopwatch Timer;
try
{
zen::bench::util::EmptyStandByList();
Ok = true;
}
catch (const zen::bench::util::elevation_required_exception&)
{
ZEN_CONSOLE("purging standby lists requires elevation. Will try launch as elevated process");
}
catch (const std::exception& Ex)
{
ZEN_CONSOLE("ERROR: {}", Ex.what());
}
if (!Ok && !m_SingleProcess)
{
try
{
zen::CreateProcOptions Cpo;
Cpo.Flags = zen::CreateProcOptions::Flag_Elevated | zen::CreateProcOptions::Flag_NewConsole;
std::filesystem::path CurExe{zen::GetRunningExecutablePath()};
if (zen::CreateProcResult Cpr = zen::CreateProc(CurExe, fmt::format("bench --purge --single"), Cpo))
{
zen::ProcessHandle ProcHandle;
ProcHandle.Initialize(Cpr);
int ExitCode = ProcHandle.WaitExitCode();
if (ExitCode == 0)
{
Ok = true;
}
else
{
ZEN_CONSOLE("ERROR: Elevated child process failed with return code {}", ExitCode);
}
}
}
catch (const std::exception& Ex)
{
ZEN_CONSOLE("ERROR: {}", Ex.what());
}
}
if (Ok)
{
// TODO: could also add reporting on just how much memory was purged
ZEN_CONSOLE("purged standby lists! (took {})", zen::NiceTimeSpanMs(Timer.GetElapsedTimeMs()));
}
}
#endif
return 0;
}
} // namespace zen
|