| Index: net/tools/domain_security_preload_generator/trie/trie_bit_buffer.h
|
| diff --git a/net/tools/domain_security_preload_generator/trie/trie_bit_buffer.h b/net/tools/domain_security_preload_generator/trie/trie_bit_buffer.h
|
| new file mode 100644
|
| index 0000000000000000000000000000000000000000..f767fbc7d923b5e4f311e929bca97ea03af16cf6
|
| --- /dev/null
|
| +++ b/net/tools/domain_security_preload_generator/trie/trie_bit_buffer.h
|
| @@ -0,0 +1,85 @@
|
| +// Copyright (c) 2016 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 NET_TOOLS_DOMAIN_SECURITY_PRELOAD_GENERATOR_TRIE_TRIE_BIT_BUFFER_H_
|
| +#define NET_TOOLS_DOMAIN_SECURITY_PRELOAD_GENERATOR_TRIE_TRIE_BIT_BUFFER_H_
|
| +
|
| +#include <stdint.h>
|
| +
|
| +#include <vector>
|
| +
|
| +#include "net/tools/domain_security_preload_generator/huffman/huffman_frequency_tracker.h"
|
| +
|
| +namespace net {
|
| +
|
| +namespace transport_security_state {
|
| +
|
| +class BitWriter;
|
| +
|
| +// TrieBitBuffer acts as a buffer for TrieWriter. It can be used to write bits,
|
| +// characters, and positions. The characters are stored as their
|
| +// HuffmanRepresentation. Positions are references to other locations in the
|
| +// trie.
|
| +class TrieBitBuffer {
|
| + public:
|
| + TrieBitBuffer();
|
| + ~TrieBitBuffer();
|
| +
|
| + // Writes |bit| to the buffer.
|
| + void WriteBit(uint8_t bit);
|
| +
|
| + // Writes the |number_of_bits| least-significant bits from |bits| to the
|
| + // buffer.
|
| + void WriteBits(uint32_t bits, uint8_t number_of_bits);
|
| +
|
| + // Write a position to the buffer. Actually writes the difference between
|
| + // |position| and |last_position|. |*last_position| will be updated to equal
|
| + // the input |position|.
|
| + void WritePosition(uint32_t position, int32_t* last_position);
|
| +
|
| + // Writes the character in |byte| to the buffer using its Huffman
|
| + // representation in |table|. Optionally tracks usage of the character in
|
| + // |*tracker|.
|
| + void WriteChar(uint8_t byte,
|
| + const HuffmanRepresentationTable& table,
|
| + HuffmanFrequencyTracker* tracker);
|
| +
|
| + // Writes the entire buffer to |*writer|. Returns the position |*writer| was
|
| + // at before the buffer was written to it.
|
| + uint32_t WriteToBitWriter(BitWriter* writer);
|
| +
|
| + // Appends the buffered bits in |current_byte_| to |elements_|. Empty bits
|
| + // are filled with zero's.
|
| + void Flush();
|
| +
|
| + private:
|
| + // Represents either the |number_of_bits| least-significant bits in |bits| or
|
| + // a position (offset) in the trie.
|
| + struct BitsOrPosition {
|
| + uint8_t bits;
|
| + uint8_t number_of_bits;
|
| + uint32_t position;
|
| + };
|
| +
|
| + // Returns the minimum number of bits needed to represent |input|.
|
| + uint8_t BitLength(uint32_t input) const;
|
| +
|
| + // Append a new element to |elements_|.
|
| + void AppendBitsElement(uint8_t bits, uint8_t number_of_bits);
|
| + void AppendPositionElement(uint32_t position);
|
| +
|
| + // Buffers bits until they fill a whole byte.
|
| + uint8_t current_byte_ = 0;
|
| +
|
| + // The number of bits currently in |current_byte_|.
|
| + uint32_t used_ = 0;
|
| +
|
| + std::vector<BitsOrPosition> elements_;
|
| +};
|
| +
|
| +} // namespace transport_security_state
|
| +
|
| +} // namespace net
|
| +
|
| +#endif // NET_TOOLS_DOMAIN_SECURITY_PRELOAD_GENERATOR_TRIE_TRIE_BIT_BUFFER_H_
|
|
|