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
|
// 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 "components/feature_engagement/internal/blocked_iph_features.h"
#include <vector>
#include "base/base_switches.h"
#include "base/check_op.h"
#include "base/command_line.h"
#include "base/feature_list.h"
#include "base/no_destructor.h"
#include "base/strings/strcat.h"
#include "base/strings/string_util.h"
namespace feature_engagement {
const char BlockedIphFeatures::kPropagateIPHForTestingSwitch[] =
"propagate-iph-for-testing";
BlockedIphFeatures::BlockedIphFeatures() = default;
BlockedIphFeatures::~BlockedIphFeatures() = default;
void BlockedIphFeatures::IncrementGlobalBlockCount() {
++global_block_count_;
}
void BlockedIphFeatures::DecrementGlobalBlockCount() {
DCHECK_GT(global_block_count_, 0U);
--global_block_count_;
}
void BlockedIphFeatures::IncrementFeatureAllowedCount(
const std::string& feature_name) {
DCHECK(!feature_name.empty());
++allow_feature_counts_[feature_name];
}
void BlockedIphFeatures::DecrementFeatureAllowedCount(
const std::string& feature_name) {
DCHECK_GT(allow_feature_counts_[feature_name], 0U);
--allow_feature_counts_[feature_name];
}
bool BlockedIphFeatures::AreAnyFeaturesBlocked() const {
return global_block_count_ > 0U;
}
bool BlockedIphFeatures::IsFeatureBlocked(
const std::string& feature_name) const {
// If nobody has requested blocks, then nothing is blocked.
if (global_block_count_ == 0U) {
return false;
}
// All features are blocked unless explicitly allowed.
const auto it = allow_feature_counts_.find(feature_name);
return it == allow_feature_counts_.end() || it->second == 0U;
}
void BlockedIphFeatures::MaybeWriteToCommandLine(
base::CommandLine& command_line) const {
if (global_block_count_ == 0U) {
return;
}
std::vector<std::string> features;
for (const auto& [name, count] : allow_feature_counts_) {
if (count > 0U) {
features.push_back(name);
}
}
std::string value_string = base::JoinString(features, ",");
command_line.AppendSwitchASCII(kPropagateIPHForTestingSwitch, value_string);
if (!features.empty()) {
if (command_line.HasSwitch(switches::kEnableFeatures)) {
const std::string old_value =
command_line.GetSwitchValueASCII(switches::kEnableFeatures);
if (!old_value.empty()) {
value_string = base::StrCat({old_value, ",", value_string});
}
command_line.RemoveSwitch(switches::kEnableFeatures);
}
command_line.AppendSwitchASCII(switches::kEnableFeatures, value_string);
}
}
void BlockedIphFeatures::MaybeReadFromCommandLine() {
if (read_from_command_line_) {
return;
}
read_from_command_line_ = true;
const base::CommandLine* const command_line =
base::CommandLine::ForCurrentProcess();
if (command_line->HasSwitch(kPropagateIPHForTestingSwitch)) {
IncrementGlobalBlockCount();
const std::string value =
command_line->GetSwitchValueASCII(kPropagateIPHForTestingSwitch);
if (!value.empty()) {
auto features = base::FeatureList::SplitFeatureListString(value);
for (auto& feature_name : features) {
if (!feature_name.empty()) {
IncrementFeatureAllowedCount(std::string(feature_name));
}
}
}
}
}
// static
BlockedIphFeatures* BlockedIphFeatures::GetInstance() {
static base::NoDestructor<BlockedIphFeatures> instance;
auto* const test_features = instance.get();
{
base::AutoLock lock(test_features->GetLock());
test_features->MaybeReadFromCommandLine();
}
return test_features;
}
} // namespace feature_engagement
|
