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