| OLD | NEW |
| (Empty) |
| 1 // Copyright (c) 2012 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 #ifndef CHROME_COMMON_METRICS_ENTROPY_PROVIDER_H_ | |
| 6 #define CHROME_COMMON_METRICS_ENTROPY_PROVIDER_H_ | |
| 7 | |
| 8 #include <functional> | |
| 9 #include <string> | |
| 10 #include <vector> | |
| 11 | |
| 12 #include "base/basictypes.h" | |
| 13 #include "base/compiler_specific.h" | |
| 14 #include "base/metrics/field_trial.h" | |
| 15 #include "third_party/mt19937ar/mt19937ar.h" | |
| 16 | |
| 17 namespace metrics { | |
| 18 | |
| 19 // Internals of entropy_provider.cc exposed for testing. | |
| 20 namespace internal { | |
| 21 | |
| 22 // A functor that generates random numbers based on a seed, using the Mersenne | |
| 23 // Twister algorithm. Suitable for use with std::random_shuffle(). | |
| 24 struct SeededRandGenerator : std::unary_function<uint32, uint32> { | |
| 25 explicit SeededRandGenerator(uint32 seed); | |
| 26 ~SeededRandGenerator(); | |
| 27 | |
| 28 // Returns a random number in range [0, range). | |
| 29 uint32 operator()(uint32 range); | |
| 30 | |
| 31 MersenneTwister mersenne_twister_; | |
| 32 }; | |
| 33 | |
| 34 // Fills |mapping| to create a bijection of values in the range of | |
| 35 // [0, |mapping.size()|), permuted based on |randomization_seed|. | |
| 36 void PermuteMappingUsingRandomizationSeed(uint32 randomization_seed, | |
| 37 std::vector<uint16>* mapping); | |
| 38 | |
| 39 } // namespace internal | |
| 40 | |
| 41 // SHA1EntropyProvider is an entropy provider suitable for high entropy | |
| 42 // sources. It works by taking the first 64 bits of the SHA1 hash of the | |
| 43 // entropy source concatenated with the trial name and using that for the | |
| 44 // final entropy value. | |
| 45 class SHA1EntropyProvider : public base::FieldTrial::EntropyProvider { | |
| 46 public: | |
| 47 // Creates a SHA1EntropyProvider with the given |entropy_source|, which | |
| 48 // should contain a large amount of entropy - for example, a textual | |
| 49 // representation of a persistent randomly-generated 128-bit value. | |
| 50 explicit SHA1EntropyProvider(const std::string& entropy_source); | |
| 51 virtual ~SHA1EntropyProvider(); | |
| 52 | |
| 53 // base::FieldTrial::EntropyProvider implementation: | |
| 54 virtual double GetEntropyForTrial(const std::string& trial_name, | |
| 55 uint32 randomization_seed) const OVERRIDE; | |
| 56 | |
| 57 private: | |
| 58 std::string entropy_source_; | |
| 59 | |
| 60 DISALLOW_COPY_AND_ASSIGN(SHA1EntropyProvider); | |
| 61 }; | |
| 62 | |
| 63 // PermutedEntropyProvider is an entropy provider suitable for low entropy | |
| 64 // sources (below 16 bits). It uses the field trial name to generate a | |
| 65 // permutation of a mapping array from an initial entropy value to a new value. | |
| 66 // Note: This provider's performance is O(2^n), where n is the number of bits | |
| 67 // in the entropy source. | |
| 68 class PermutedEntropyProvider : public base::FieldTrial::EntropyProvider { | |
| 69 public: | |
| 70 // Creates a PermutedEntropyProvider with the given |low_entropy_source|, | |
| 71 // which should have a value in the range of [0, low_entropy_source_max). | |
| 72 PermutedEntropyProvider(uint16 low_entropy_source, | |
| 73 size_t low_entropy_source_max); | |
| 74 virtual ~PermutedEntropyProvider(); | |
| 75 | |
| 76 // base::FieldTrial::EntropyProvider implementation: | |
| 77 virtual double GetEntropyForTrial(const std::string& trial_name, | |
| 78 uint32 randomization_seed) const OVERRIDE; | |
| 79 | |
| 80 protected: | |
| 81 // Performs the permutation algorithm and returns the permuted value that | |
| 82 // corresponds to |low_entropy_source_|. | |
| 83 virtual uint16 GetPermutedValue(uint32 randomization_seed) const; | |
| 84 | |
| 85 private: | |
| 86 uint16 low_entropy_source_; | |
| 87 size_t low_entropy_source_max_; | |
| 88 | |
| 89 DISALLOW_COPY_AND_ASSIGN(PermutedEntropyProvider); | |
| 90 }; | |
| 91 | |
| 92 } // namespace metrics | |
| 93 | |
| 94 #endif // CHROME_COMMON_METRICS_ENTROPY_PROVIDER_H_ | |
| OLD | NEW |