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
|
// Copyright 2023 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "chrome/app/chrome_main_linux.h"
#include <optional>
#include <string>
#include <vector>
#include "base/command_line.h"
#include "base/environment.h"
#include "base/files/file_util.h"
#include "base/logging.h"
#include "base/strings/string_tokenizer.h"
#include "base/strings/string_util.h"
#include "chrome/common/channel_info.h"
namespace {
void AppendExtraArgsFromEnvVar(const std::string& env_var_name,
std::vector<std::string>& out_args) {
std::unique_ptr<base::Environment> environment = base::Environment::Create();
std::optional<std::string> extra_args_str = environment->GetVar(env_var_name);
if (extra_args_str.has_value()) {
base::StringTokenizer tokenizer(extra_args_str.value(),
base::kWhitespaceASCII);
tokenizer.set_quote_chars("\"'");
while (tokenizer.GetNext()) {
std::string arg;
const std::string& token = tokenizer.token();
base::TrimString(token, "\"'", &arg);
out_args.push_back(arg);
}
}
}
#if BUILDFLAG(GOOGLE_CHROME_BRANDING)
// Reads the contents of the "CHROME_VERSION_EXTRA" file at the specified path.
std::string ReadChromeVersionExtra(const base::FilePath& path) {
std::string contents;
// Don't read the file if it's unexpectedly large.
if (!base::ReadFileToStringWithMaxSize(path, &contents, 128)) {
return std::string();
}
return std::string(base::TrimWhitespaceASCII(contents, base::TRIM_ALL));
}
#endif // BUILDFLAG(GOOGLE_CHROME_BRANDING)
} // namespace
void AppendExtraArgumentsToCommandLine(base::CommandLine* command_line) {
constexpr char kExtraFlagsVarName[] = "CHROME_EXTRA_FLAGS";
std::vector<std::string> extra_args = {""};
AppendExtraArgsFromEnvVar(kExtraFlagsVarName, extra_args);
std::string channel_suffix =
chrome::GetChannelSuffixForExtraFlagsEnvVarName();
if (!channel_suffix.empty()) {
AppendExtraArgsFromEnvVar(kExtraFlagsVarName + channel_suffix, extra_args);
}
command_line->AppendArguments(base::CommandLine(extra_args),
/*include_program=*/false);
}
void PossiblyDetermineFallbackChromeChannel(const char* launched_binary_path) {
#if BUILDFLAG(GOOGLE_CHROME_BRANDING)
std::unique_ptr<base::Environment> env(base::Environment::Create());
// If "CHROME_VERSION_EXTRA" is not set, then we may have been launched via
// the "chrome" binary directly, instead of the "google-chrome" wrapper
// script. Check for an adjacent "CHROME_VERSION_EXTRA" file for the channel.
// Note: Child processes inherit environment variables, so once this is set
// in the parent, child processes won't enter the if below.
if (!env->HasVar("CHROME_VERSION_EXTRA")) {
base::FilePath path = base::FilePath(launched_binary_path)
.DirName()
.Append("CHROME_VERSION_EXTRA");
std::string channel = ReadChromeVersionExtra(path);
if (!channel.empty()) {
LOG(WARNING) << "Read channel " << channel << " from " << path;
env->SetVar("CHROME_VERSION_EXTRA", channel);
}
}
#endif // BUILDFLAG(GOOGLE_CHROME_BRANDING)
}
|
