1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
|
// 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/third_party/quiche/src/http2/hpack/decoder/hpack_string_collector.h"
#include <stddef.h>
#include <iosfwd>
#include <ostream>
#include "testing/gtest/include/gtest/gtest.h"
#include "net/third_party/quiche/src/http2/platform/api/http2_string_utils.h"
#include "net/third_party/quiche/src/http2/platform/api/http2_test_helpers.h"
namespace http2 {
namespace test {
namespace {
std::ostream& operator<<(std::ostream& out,
HpackStringCollector::CollectorState v) {
switch (v) {
case HpackStringCollector::CollectorState::kGenesis:
return out << "kGenesis";
case HpackStringCollector::CollectorState::kStarted:
return out << "kStarted";
case HpackStringCollector::CollectorState::kEnded:
return out << "kEnded";
}
return out << "UnknownCollectorState";
}
} // namespace
HpackStringCollector::HpackStringCollector() {
Clear();
}
HpackStringCollector::HpackStringCollector(const std::string& str, bool huffman)
: s(str), len(str.size()), huffman_encoded(huffman), state(kEnded) {}
void HpackStringCollector::Clear() {
s = "";
len = 0;
huffman_encoded = false;
state = kGenesis;
}
bool HpackStringCollector::IsClear() const {
return s.empty() && len == 0 && huffman_encoded == false && state == kGenesis;
}
bool HpackStringCollector::IsInProgress() const {
return state == kStarted;
}
bool HpackStringCollector::HasEnded() const {
return state == kEnded;
}
void HpackStringCollector::OnStringStart(bool huffman, size_t length) {
EXPECT_TRUE(IsClear()) << ToString();
state = kStarted;
huffman_encoded = huffman;
len = length;
}
void HpackStringCollector::OnStringData(const char* data, size_t length) {
quiche::QuicheStringPiece sp(data, length);
EXPECT_TRUE(IsInProgress()) << ToString();
EXPECT_LE(sp.size(), len) << ToString();
Http2StrAppend(&s, sp);
EXPECT_LE(s.size(), len) << ToString();
}
void HpackStringCollector::OnStringEnd() {
EXPECT_TRUE(IsInProgress()) << ToString();
EXPECT_EQ(s.size(), len) << ToString();
state = kEnded;
}
::testing::AssertionResult HpackStringCollector::Collected(
quiche::QuicheStringPiece str,
bool is_huffman_encoded) const {
VERIFY_TRUE(HasEnded());
VERIFY_EQ(str.size(), len);
VERIFY_EQ(is_huffman_encoded, huffman_encoded);
VERIFY_EQ(str, s);
return ::testing::AssertionSuccess();
}
std::string HpackStringCollector::ToString() const {
std::stringstream ss;
ss << *this;
return ss.str();
}
bool operator==(const HpackStringCollector& a, const HpackStringCollector& b) {
return a.s == b.s && a.len == b.len &&
a.huffman_encoded == b.huffman_encoded && a.state == b.state;
}
bool operator!=(const HpackStringCollector& a, const HpackStringCollector& b) {
return !(a == b);
}
std::ostream& operator<<(std::ostream& out, const HpackStringCollector& v) {
out << "HpackStringCollector(state=" << v.state;
if (v.state == HpackStringCollector::kGenesis) {
return out << ")";
}
if (v.huffman_encoded) {
out << ", Huffman Encoded";
}
out << ", Length=" << v.len;
if (!v.s.empty() && v.len != v.s.size()) {
out << " (" << v.s.size() << ")";
}
return out << ", String=\"" << Http2HexEscape(v.s) << "\")";
}
} // namespace test
} // namespace http2
|
