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_BLOOM_FILTER_H_ | 5 #ifndef COMPONENTS_RAPPOR_BLOOM_FILTER_H_ |
6 #define COMPONENTS_RAPPOR_BLOOM_FILTER_H_ | 6 #define COMPONENTS_RAPPOR_BLOOM_FILTER_H_ |
7 | 7 |
8 #include <string> | 8 #include <string> |
9 | 9 |
10 #include "base/basictypes.h" | 10 #include "base/basictypes.h" |
11 #include "base/macros.h" | 11 #include "base/macros.h" |
12 #include "components/rappor/byte_vector_utils.h" | 12 #include "components/rappor/byte_vector_utils.h" |
13 | 13 |
14 namespace rappor { | 14 namespace rappor { |
15 | 15 |
16 // BloomFilter is a simple Bloom filter for keeping track of a set of strings. | 16 // BloomFilter is a simple Bloom filter for keeping track of a set of strings. |
17 class BloomFilter { | 17 class BloomFilter { |
18 public: | 18 public: |
19 // Constructs a BloomFilter using |bytes_size| bytes of Bloom filter bits, | 19 // Constructs a BloomFilter using |bytes_size| bytes of Bloom filter bits, |
20 // and |hash_function_count| hash functions to set bits in the filter. The | 20 // and |hash_function_count| hash functions to set bits in the filter. The |
21 // hash functions will be generated by using seeds in the range | 21 // hash functions will be generated by using seeds in the range |
22 // |hash_seed_offset| to (|hash_seed_offset| + |hash_function_count|). | 22 // |hash_seed_offset| to (|hash_seed_offset| + |hash_function_count|). |
23 BloomFilter(uint32_t bytes_size, | 23 BloomFilter(uint32_t bytes_size, |
24 uint32_t hash_function_count, | 24 uint32_t hash_function_count, |
25 uint32_t hash_seed_offset); | 25 uint32_t hash_seed_offset); |
26 ~BloomFilter(); | 26 ~BloomFilter(); |
27 | 27 |
28 // Add a single string to the Bloom filter. | 28 // Sets the Bloom filter bits to contain a single string. |
29 void AddString(const std::string& str); | 29 void SetString(const std::string& str); |
30 | 30 |
31 // Returns the current value of the Bloom filter's bit array. | 31 // Returns the current value of the Bloom filter's bit array. |
32 const ByteVector& bytes() const { return bytes_; }; | 32 const ByteVector& bytes() const { return bytes_; }; |
33 | 33 |
| 34 // Sets bytes for testing purposes. |
| 35 void SetBytesForTesting(const ByteVector& bytes); |
| 36 |
34 private: | 37 private: |
35 // Stores the byte array of the Bloom filter. | 38 // Stores the byte array of the Bloom filter. |
36 ByteVector bytes_; | 39 ByteVector bytes_; |
37 | 40 |
38 // The number of bits to set for each string added. | 41 // The number of bits to set for each string added. |
39 uint32_t hash_function_count_; | 42 uint32_t hash_function_count_; |
40 | 43 |
41 // A number add to a hash function index to get a seed for that hash function. | 44 // A number add to a hash function index to get a seed for that hash function. |
42 uint32_t hash_seed_offset_; | 45 uint32_t hash_seed_offset_; |
43 | 46 |
44 DISALLOW_COPY_AND_ASSIGN(BloomFilter); | 47 DISALLOW_COPY_AND_ASSIGN(BloomFilter); |
45 }; | 48 }; |
46 | 49 |
47 } // namespace rappor | 50 } // namespace rappor |
48 | 51 |
49 #endif // COMPONENTS_RAPPOR_BLOOM_FILTER_H_ | 52 #endif // COMPONENTS_RAPPOR_BLOOM_FILTER_H_ |
OLD | NEW |