RAPPOR implementation

components/rappor/byte_vector_utils.h

+// Copyright (c) 2013 The Chromium Authors. All rights reserved.
+// Use of this source code is governed by a BSD-style license that can be
+// found in the LICENSE file.
+
+#ifndef COMPONENTS_RAPPOR_BYTE_VECTOR_UTILS_H_
+#define COMPONENTS_RAPPOR_BYTE_VECTOR_UTILS_H_
+
+#include <stddef.h>
+#include <stdint.h>
+#include <vector>
+
+#include "components/rappor/rappor_parameters.h"
+#include "crypto/hmac.h"
+
+namespace rappor {
+
+// A vector of one byte integers used to store a set of binary bits.
+typedef std::vector<uint8_t> ByteVector;
+
+// Bitwise operations on ByteVectors.  These methods modify the lhs to be the
+// result of the bitwise AND, OR, or AND NOT.
+ByteVector* ByteVectorAnd(ByteVector* lhs, const ByteVector& rhs);
+ByteVector* ByteVectorOr(ByteVector* lhs, const ByteVector& rhs);
+ByteVector* ByteVectorMerge(ByteVector* lhs,
+                            const ByteVector& rhs,
+                            const ByteVector& mask);
+
+// A utility object for generating random binary data with different
+// likelihood of bits being true.
+class ByteVectorGenerator {
+ public:
+  explicit ByteVectorGenerator(size_t byte_count);
+  // Generates a random byte vector where each bit is independently set with
+  // the given probability.
+  ByteVector GetWeightedRandomByteVector(Probability probability);
+
+ protected:
+  // Generates a random byte from a uniform distribution.
+  virtual uint8_t RandByte();
+
+ private:
+  // Generates a random vector of bytes from a uniform distribution.
+  ByteVector GetRandomByteVector();
+  size_t byte_count_;
+};
+
+// A ByteVectorGenerator that uses a psuedo-random function to generate
+// deterministicly random bits.
+class HmacByteVectorGenerator : public ByteVectorGenerator {
+ public:
+  HmacByteVectorGenerator(size_t byte_count, const std::string& secret);
+
+ protected:
+  virtual uint8_t RandByte();
+
+ private:
+  crypto::HMAC hmac_;
+  uint64_t hmac_state_;
+};
+
+}  // namespace rappor
+
+#endif  // COMPONENTS_RAPPOR_BYTE_VECTOR_UTILS_H_