mirror of
https://github.com/klzgrad/naiveproxy.git
synced 2024-12-01 01:36:09 +03:00
86 lines
2.7 KiB
C++
86 lines
2.7 KiB
C++
|
// Copyright 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.
|
||
|
|
|
||
|
|
#include "net/http2/hpack/tools/hpack_block_builder.h"
|
||
|
|
|
||
|
|
#include "net/http2/tools/http2_bug_tracker.h"
|
||
|
|
#include "testing/gtest/include/gtest/gtest.h"
|
||
|
|
|
||
|
|
namespace net {
|
||
|
|
namespace test {
|
||
|
|
|
||
|
|
void HpackBlockBuilder::AppendHighBitsAndVarint(uint8_t high_bits,
|
||
|
|
uint8_t prefix_length,
|
||
|
|
uint64_t varint) {
|
||
|
|
EXPECT_LE(4, prefix_length);
|
||
|
|
EXPECT_LE(prefix_length, 7);
|
||
|
|
|
||
|
|
// prefix_mask defines the sequence of low-order bits of the first byte
|
||
|
|
// that encode the prefix of the value. It is also the marker in those bits
|
||
|
|
// of the first byte indicating that at least one extension byte is needed.
|
||
|
|
uint8_t prefix_mask = (1 << prefix_length) - 1;
|
||
|
|
EXPECT_EQ(0, high_bits & prefix_mask);
|
||
|
|
|
||
|
|
if (varint < prefix_mask) {
|
||
|
|
uint8_t b = high_bits | static_cast<uint8_t>(varint);
|
||
|
|
buffer_.push_back(static_cast<char>(b));
|
||
|
|
return;
|
||
|
|
}
|
||
|
|
|
||
|
|
// We need extension bytes.
|
||
|
|
buffer_.push_back(static_cast<char>(high_bits | prefix_mask));
|
||
|
|
varint -= prefix_mask;
|
||
|
|
while (varint >= 128) {
|
||
|
|
uint8_t b = static_cast<uint8_t>((varint % 128) + 128);
|
||
|
|
buffer_.push_back(static_cast<char>(b));
|
||
|
|
varint = varint / 128;
|
||
|
|
}
|
||
|
|
char c = static_cast<char>(varint);
|
||
|
|
buffer_.push_back(c);
|
||
|
|
}
|
||
|
|
|
||
|
|
void HpackBlockBuilder::AppendEntryTypeAndVarint(HpackEntryType entry_type,
|
||
|
|
uint64_t varint) {
|
||
|
|
uint8_t high_bits;
|
||
|
|
uint8_t prefix_length; // Bits of the varint prefix in the first byte.
|
||
|
|
switch (entry_type) {
|
||
|
|
case HpackEntryType::kIndexedHeader:
|
||
|
|
high_bits = 0x80;
|
||
|
|
prefix_length = 7;
|
||
|
|
break;
|
||
|
|
case HpackEntryType::kDynamicTableSizeUpdate:
|
||
|
|
high_bits = 0x20;
|
||
|
|
prefix_length = 5;
|
||
|
|
break;
|
||
|
|
case HpackEntryType::kIndexedLiteralHeader:
|
||
|
|
high_bits = 0x40;
|
||
|
|
prefix_length = 6;
|
||
|
|
break;
|
||
|
|
case HpackEntryType::kUnindexedLiteralHeader:
|
||
|
|
high_bits = 0x00;
|
||
|
|
prefix_length = 4;
|
||
|
|
break;
|
||
|
|
case HpackEntryType::kNeverIndexedLiteralHeader:
|
||
|
|
high_bits = 0x10;
|
||
|
|
prefix_length = 4;
|
||
|
|
break;
|
||
|
|
default:
|
||
|
|
HTTP2_BUG << "Unreached, entry_type=" << entry_type;
|
||
|
|
high_bits = 0;
|
||
|
|
prefix_length = 0;
|
||
|
|
}
|
||
|
|
AppendHighBitsAndVarint(high_bits, prefix_length, varint);
|
||
|
|
}
|
||
|
|
|
||
|
|
void HpackBlockBuilder::AppendString(bool is_huffman_encoded,
|
||
|
|
Http2StringPiece str) {
|
||
|
|
uint8_t high_bits = is_huffman_encoded ? 0x80 : 0;
|
||
|
|
uint8_t prefix_length = 7;
|
||
|
|
AppendHighBitsAndVarint(high_bits, prefix_length, str.size());
|
||
|
|
buffer_.append(str.data(), str.size());
|
||
|
|
}
|
||
|
|
|
||
|
|
} // namespace test
|
||
|
|
} // namespace net
|