OLD | NEW |
| (Empty) |
1 // Copyright 2013 The Chromium Authors. All rights reserved. | |
2 // Use of this source code is governed by a BSD-style license that can be | |
3 // found in the LICENSE file. | |
4 | |
5 #include "chrome/common/metrics/caching_permuted_entropy_provider.h" | |
6 | |
7 #include <string> | |
8 | |
9 #include "base/base64.h" | |
10 #include "base/logging.h" | |
11 #include "base/prefs/pref_registry_simple.h" | |
12 #include "base/prefs/pref_service.h" | |
13 #include "chrome/common/pref_names.h" | |
14 | |
15 namespace metrics { | |
16 | |
17 CachingPermutedEntropyProvider::CachingPermutedEntropyProvider( | |
18 PrefService* local_state, | |
19 uint16 low_entropy_source, | |
20 size_t low_entropy_source_max) | |
21 : PermutedEntropyProvider(low_entropy_source, low_entropy_source_max), | |
22 local_state_(local_state) { | |
23 ReadFromLocalState(); | |
24 } | |
25 | |
26 CachingPermutedEntropyProvider::~CachingPermutedEntropyProvider() { | |
27 } | |
28 | |
29 // static | |
30 void CachingPermutedEntropyProvider::RegisterPrefs( | |
31 PrefRegistrySimple* registry) { | |
32 registry->RegisterStringPref(prefs::kMetricsPermutedEntropyCache, | |
33 std::string()); | |
34 } | |
35 | |
36 // static | |
37 void CachingPermutedEntropyProvider::ClearCache(PrefService* local_state) { | |
38 local_state->ClearPref(prefs::kMetricsPermutedEntropyCache); | |
39 } | |
40 | |
41 uint16 CachingPermutedEntropyProvider::GetPermutedValue( | |
42 uint32 randomization_seed) const { | |
43 DCHECK(thread_checker_.CalledOnValidThread()); | |
44 | |
45 uint16 value = 0; | |
46 if (!FindValue(randomization_seed, &value)) { | |
47 value = PermutedEntropyProvider::GetPermutedValue(randomization_seed); | |
48 AddToCache(randomization_seed, value); | |
49 } | |
50 return value; | |
51 } | |
52 | |
53 void CachingPermutedEntropyProvider::ReadFromLocalState() const { | |
54 const std::string base64_cache_data = | |
55 local_state_->GetString(prefs::kMetricsPermutedEntropyCache); | |
56 std::string cache_data; | |
57 if (!base::Base64Decode(base64_cache_data, &cache_data) || | |
58 !cache_.ParseFromString(cache_data)) { | |
59 local_state_->ClearPref(prefs::kMetricsPermutedEntropyCache); | |
60 NOTREACHED(); | |
61 } | |
62 } | |
63 | |
64 void CachingPermutedEntropyProvider::UpdateLocalState() const { | |
65 std::string serialized; | |
66 cache_.SerializeToString(&serialized); | |
67 | |
68 std::string base64_encoded; | |
69 base::Base64Encode(serialized, &base64_encoded); | |
70 local_state_->SetString(prefs::kMetricsPermutedEntropyCache, base64_encoded); | |
71 } | |
72 | |
73 void CachingPermutedEntropyProvider::AddToCache(uint32 randomization_seed, | |
74 uint16 value) const { | |
75 PermutedEntropyCache::Entry* entry; | |
76 const int kMaxSize = 25; | |
77 if (cache_.entry_size() >= kMaxSize) { | |
78 // If the cache is full, evict the first entry, swapping later entries in | |
79 // to take its place. This effectively creates a FIFO cache, which is good | |
80 // enough here because the expectation is that there shouldn't be more than | |
81 // |kMaxSize| field trials at any given time, so eviction should happen very | |
82 // rarely, only as new trials are introduced, evicting old expired trials. | |
83 for (int i = 1; i < kMaxSize; ++i) | |
84 cache_.mutable_entry()->SwapElements(i - 1, i); | |
85 entry = cache_.mutable_entry(kMaxSize - 1); | |
86 } else { | |
87 entry = cache_.add_entry(); | |
88 } | |
89 | |
90 entry->set_randomization_seed(randomization_seed); | |
91 entry->set_value(value); | |
92 | |
93 UpdateLocalState(); | |
94 } | |
95 | |
96 bool CachingPermutedEntropyProvider::FindValue(uint32 randomization_seed, | |
97 uint16* value) const { | |
98 for (int i = 0; i < cache_.entry_size(); ++i) { | |
99 if (cache_.entry(i).randomization_seed() == randomization_seed) { | |
100 *value = cache_.entry(i).value(); | |
101 return true; | |
102 } | |
103 } | |
104 return false; | |
105 } | |
106 | |
107 } // namespace metrics | |
OLD | NEW |