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
|
// Copyright 2019 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#ifdef UNSAFE_BUFFERS_BUILD
// TODO(crbug.com/390223051): Remove C-library calls to fix the errors.
#pragma allow_unsafe_libc_calls
#endif
#include "remoting/test/cli_util.h"
#include <string.h>
#include "base/command_line.h"
#include "base/run_loop.h"
#include "base/strings/string_number_conversions.h"
#include "base/task/thread_pool.h"
namespace {
void PrintOption(int option_number, const std::string& option_name) {
printf(" %d. %s\n", option_number, option_name.c_str());
}
} // namespace
namespace remoting {
namespace test {
CommandOption::CommandOption() = default;
CommandOption::CommandOption(const CommandOption&) = default;
CommandOption::CommandOption(CommandOption&&) = default;
CommandOption::CommandOption(
const std::string name,
const base::RepeatingCallback<void(base::OnceClosure on_done)>& command)
: name(name), command(command) {}
CommandOption::~CommandOption() = default;
CommandOption& CommandOption::operator=(const CommandOption&) = default;
CommandOption& CommandOption::operator=(CommandOption&&) = default;
void RunCommandOptionsLoop(const std::vector<CommandOption>& options) {
DCHECK_LT(0u, options.size());
while (true) {
base::RunLoop run_loop;
printf("\nOptions:\n");
int print_option_number = 1;
for (const auto& option : options) {
PrintOption(print_option_number, option.name);
print_option_number++;
}
int quit_option_number = print_option_number;
PrintOption(quit_option_number, "Quit");
printf("\nYour choice [number]: ");
int choice = 0;
base::StringToInt(test::ReadString(), &choice);
if (choice < 1 || choice > quit_option_number) {
fprintf(stderr, "Unknown option\n");
continue;
}
if (choice == quit_option_number) {
return;
}
auto& command = options[choice - 1].command;
command.Run(run_loop.QuitWhenIdleClosure());
run_loop.Run();
}
}
std::string ReadString() {
const int kMaxLen = 1024;
std::string str(kMaxLen, 0);
char* result = fgets(&str[0], kMaxLen, stdin);
if (!result) {
return std::string();
}
size_t newline_index = str.find('\n');
if (newline_index != std::string::npos) {
str[newline_index] = '\0';
}
str.resize(strlen(&str[0]));
return str;
}
bool ReadYNBool(bool default_value) {
std::string result = test::ReadString();
return result == "y" || result == "Y" || (default_value && result.empty());
}
std::string ReadStringFromCommandLineOrStdin(const std::string& switch_name,
const std::string& read_prompt) {
base::CommandLine* command_line = base::CommandLine::ForCurrentProcess();
if (command_line->HasSwitch(switch_name)) {
return command_line->GetSwitchValueASCII(switch_name);
}
printf("%s", read_prompt.c_str());
return ReadString();
}
void WaitForEnterKey(base::OnceClosure on_done) {
base::ThreadPool::PostTaskAndReply(FROM_HERE, {base::MayBlock()},
#pragma clang diagnostic push
#pragma clang diagnostic ignored "-Wunused-result"
base::BindOnce([]() { getchar(); }),
#pragma clang diagnostic pop
std::move(on_done));
}
} // namespace test
} // namespace remoting
|
