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
|
// Copyright 2022 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "base/at_exit.h"
#include "base/command_line.h"
#include "base/feature_list.h"
#include "base/metrics/field_trial.h"
#include "components/variations/client_filterable_state.h"
#include "components/variations/entropy_provider.h"
#include "components/variations/proto/study.pb.h"
#include "components/variations/variations_layers.h"
#include "components/variations/variations_seed_processor.h"
#include "components/variations/variations_test_utils.h"
#include "testing/libfuzzer/proto/lpm_interface.h"
namespace variations {
namespace {
class TestOverrideStringCallback {
public:
typedef std::map<uint32_t, std::u16string> OverrideMap;
TestOverrideStringCallback()
: callback_(base::BindRepeating(&TestOverrideStringCallback::Override,
base::Unretained(this))) {}
TestOverrideStringCallback(const TestOverrideStringCallback&) = delete;
TestOverrideStringCallback& operator=(const TestOverrideStringCallback&) =
delete;
virtual ~TestOverrideStringCallback() = default;
const VariationsSeedProcessor::UIStringOverrideCallback& callback() const {
return callback_;
}
const OverrideMap& overrides() const { return overrides_; }
private:
void Override(uint32_t hash, const std::u16string& string) {
overrides_[hash] = string;
}
VariationsSeedProcessor::UIStringOverrideCallback callback_;
OverrideMap overrides_;
};
struct Environment {
Environment() { base::CommandLine::Init(0, nullptr); }
base::AtExitManager at_exit_manager;
};
std::unique_ptr<ClientFilterableState> MockChromeClientFilterableState() {
auto client_state = CreateDummyClientFilterableState();
client_state->locale = "en-CA";
client_state->reference_date = base::Time::Now();
client_state->version = base::Version("20.0.0.0");
client_state->channel = Study::STABLE;
client_state->form_factor = Study::PHONE;
client_state->platform = Study::PLATFORM_ANDROID;
return client_state;
}
} // namespace
void CreateTrialsFromStudyFuzzer(const VariationsSeed& seed) {
base::FieldTrialList field_trial_list;
base::FeatureList feature_list;
TestOverrideStringCallback override_callback;
// TODO(b/244252663): Add more coverage of client_state and entropy
// provider arguments.
auto client_state = MockChromeClientFilterableState();
EntropyProviders entropy_providers(
"client_id", {7999, 8000},
// The following is a test value for limited entropy randomization source
"00000000000000000000000000000001");
VariationsLayers layers(seed, entropy_providers);
VariationsSeedProcessor().CreateTrialsFromSeed(
seed, *client_state, override_callback.callback(), entropy_providers,
layers, &feature_list);
}
DEFINE_PROTO_FUZZER(const VariationsSeed& seed) {
static Environment env;
CreateTrialsFromStudyFuzzer(seed);
}
} // namespace variations
|
