| 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 #ifndef COMPONENTS_RAPPOR_BYTE_VECTOR_UTILS_H_ | 5 #ifndef COMPONENTS_RAPPOR_BYTE_VECTOR_UTILS_H_ |
| 6 #define COMPONENTS_RAPPOR_BYTE_VECTOR_UTILS_H_ | 6 #define COMPONENTS_RAPPOR_BYTE_VECTOR_UTILS_H_ |
| 7 | 7 |
| 8 #include <vector> | 8 #include <vector> |
| 9 | 9 |
| 10 #include "base/basictypes.h" | 10 #include "base/basictypes.h" |
| (...skipping 25 matching lines...) Expand all Loading... |
| 36 | 36 |
| 37 // Counts the number of bits set in the byte vector. | 37 // Counts the number of bits set in the byte vector. |
| 38 int CountBits(const ByteVector& vector); | 38 int CountBits(const ByteVector& vector); |
| 39 | 39 |
| 40 // A utility object for generating random binary data with different | 40 // A utility object for generating random binary data with different |
| 41 // likelihood of bits being true, using entropy from crypto::RandBytes(). | 41 // likelihood of bits being true, using entropy from crypto::RandBytes(). |
| 42 class ByteVectorGenerator { | 42 class ByteVectorGenerator { |
| 43 public: | 43 public: |
| 44 explicit ByteVectorGenerator(size_t byte_count); | 44 explicit ByteVectorGenerator(size_t byte_count); |
| 45 | 45 |
| 46 ~ByteVectorGenerator(); | 46 virtual ~ByteVectorGenerator(); |
| 47 | 47 |
| 48 // Generates a random byte vector where the bits are independent random | 48 // Generates a random byte vector where the bits are independent random |
| 49 // variables which are true with the given |probability|. | 49 // variables which are true with the given |probability|. |
| 50 ByteVector GetWeightedRandomByteVector(Probability probability); | 50 ByteVector GetWeightedRandomByteVector(Probability probability); |
| 51 | 51 |
| 52 protected: | 52 protected: |
| 53 // Size of vectors to be generated. | 53 // Size of vectors to be generated. |
| 54 size_t byte_count() const { return byte_count_; } | 54 size_t byte_count() const { return byte_count_; } |
| 55 | 55 |
| 56 // Generates a random vector of bytes from a uniform distribution. | 56 // Generates a random vector of bytes from a uniform distribution. |
| (...skipping 16 matching lines...) Expand all Loading... |
| 73 // based on a secret seed. | 73 // based on a secret seed. |
| 74 class HmacByteVectorGenerator : public ByteVectorGenerator { | 74 class HmacByteVectorGenerator : public ByteVectorGenerator { |
| 75 public: | 75 public: |
| 76 // Constructor takes the size of the vector to generate, along with a | 76 // Constructor takes the size of the vector to generate, along with a |
| 77 // |entropy_input| and |personalization_string| to seed the pseudo-random | 77 // |entropy_input| and |personalization_string| to seed the pseudo-random |
| 78 // number generator. The string parameters are treated as byte arrays. | 78 // number generator. The string parameters are treated as byte arrays. |
| 79 HmacByteVectorGenerator(size_t byte_count, | 79 HmacByteVectorGenerator(size_t byte_count, |
| 80 const std::string& entropy_input, | 80 const std::string& entropy_input, |
| 81 const std::string& personalization_string); | 81 const std::string& personalization_string); |
| 82 | 82 |
| 83 ~HmacByteVectorGenerator(); | 83 virtual ~HmacByteVectorGenerator(); |
| 84 | 84 |
| 85 // Generates a random string suitable for passing to the constructor as | 85 // Generates a random string suitable for passing to the constructor as |
| 86 // |entropy_input|. | 86 // |entropy_input|. |
| 87 static std::string GenerateEntropyInput(); | 87 static std::string GenerateEntropyInput(); |
| 88 | 88 |
| 89 // Key size required for 128-bit security strength (including nonce). | 89 // Key size required for 128-bit security strength (including nonce). |
| 90 static const size_t kEntropyInputSize; | 90 static const size_t kEntropyInputSize; |
| 91 | 91 |
| 92 protected: | 92 protected: |
| 93 // Generate byte vector generator that streams from the next request instead | 93 // Generate byte vector generator that streams from the next request instead |
| (...skipping 12 matching lines...) Expand all Loading... |
| 106 | 106 |
| 107 // Total number of bytes streamed from the HMAC_DRBG Generate Process. | 107 // Total number of bytes streamed from the HMAC_DRBG Generate Process. |
| 108 size_t generated_bytes_; | 108 size_t generated_bytes_; |
| 109 | 109 |
| 110 DISALLOW_ASSIGN(HmacByteVectorGenerator); | 110 DISALLOW_ASSIGN(HmacByteVectorGenerator); |
| 111 }; | 111 }; |
| 112 | 112 |
| 113 } // namespace rappor | 113 } // namespace rappor |
| 114 | 114 |
| 115 #endif // COMPONENTS_RAPPOR_BYTE_VECTOR_UTILS_H_ | 115 #endif // COMPONENTS_RAPPOR_BYTE_VECTOR_UTILS_H_ |
| OLD | NEW |