OLD | NEW |
| (Empty) |
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 | |
3 // found in the LICENSE file. | |
4 | |
5 #include "net/spdy/hpack_encoding_context.h" | |
6 | |
7 #include <cstddef> | |
8 | |
9 #include "base/logging.h" | |
10 #include "base/macros.h" | |
11 #include "net/spdy/hpack_constants.h" | |
12 #include "net/spdy/hpack_entry.h" | |
13 | |
14 namespace net { | |
15 | |
16 using base::StringPiece; | |
17 | |
18 namespace { | |
19 | |
20 // An entry in the static table. Must be a POD in order to avoid | |
21 // static initializers, i.e. no user-defined constructors or | |
22 // destructors. | |
23 struct StaticEntry { | |
24 const char* const name; | |
25 const size_t name_len; | |
26 const char* const value; | |
27 const size_t value_len; | |
28 }; | |
29 | |
30 // The "constructor" for a StaticEntry that computes the lengths at | |
31 // compile time. | |
32 #define STATIC_ENTRY(name, value) \ | |
33 { name, arraysize(name) - 1, value, arraysize(value) - 1 } | |
34 | |
35 const StaticEntry kStaticTable[] = { | |
36 STATIC_ENTRY(":authority" , ""), // 1 | |
37 STATIC_ENTRY(":method" , "GET"), // 2 | |
38 STATIC_ENTRY(":method" , "POST"), // 3 | |
39 STATIC_ENTRY(":path" , "/"), // 4 | |
40 STATIC_ENTRY(":path" , "/index.html"), // 5 | |
41 STATIC_ENTRY(":scheme" , "http"), // 6 | |
42 STATIC_ENTRY(":scheme" , "https"), // 7 | |
43 STATIC_ENTRY(":status" , "200"), // 8 | |
44 STATIC_ENTRY(":status" , "500"), // 9 | |
45 STATIC_ENTRY(":status" , "404"), // 10 | |
46 STATIC_ENTRY(":status" , "403"), // 11 | |
47 STATIC_ENTRY(":status" , "400"), // 12 | |
48 STATIC_ENTRY(":status" , "401"), // 13 | |
49 STATIC_ENTRY("accept-charset" , ""), // 14 | |
50 STATIC_ENTRY("accept-encoding" , ""), // 15 | |
51 STATIC_ENTRY("accept-language" , ""), // 16 | |
52 STATIC_ENTRY("accept-ranges" , ""), // 17 | |
53 STATIC_ENTRY("accept" , ""), // 18 | |
54 STATIC_ENTRY("access-control-allow-origin" , ""), // 19 | |
55 STATIC_ENTRY("age" , ""), // 20 | |
56 STATIC_ENTRY("allow" , ""), // 21 | |
57 STATIC_ENTRY("authorization" , ""), // 22 | |
58 STATIC_ENTRY("cache-control" , ""), // 23 | |
59 STATIC_ENTRY("content-disposition" , ""), // 24 | |
60 STATIC_ENTRY("content-encoding" , ""), // 25 | |
61 STATIC_ENTRY("content-language" , ""), // 26 | |
62 STATIC_ENTRY("content-length" , ""), // 27 | |
63 STATIC_ENTRY("content-location" , ""), // 28 | |
64 STATIC_ENTRY("content-range" , ""), // 29 | |
65 STATIC_ENTRY("content-type" , ""), // 30 | |
66 STATIC_ENTRY("cookie" , ""), // 31 | |
67 STATIC_ENTRY("date" , ""), // 32 | |
68 STATIC_ENTRY("etag" , ""), // 33 | |
69 STATIC_ENTRY("expect" , ""), // 34 | |
70 STATIC_ENTRY("expires" , ""), // 35 | |
71 STATIC_ENTRY("from" , ""), // 36 | |
72 STATIC_ENTRY("host" , ""), // 37 | |
73 STATIC_ENTRY("if-match" , ""), // 38 | |
74 STATIC_ENTRY("if-modified-since" , ""), // 39 | |
75 STATIC_ENTRY("if-none-match" , ""), // 40 | |
76 STATIC_ENTRY("if-range" , ""), // 41 | |
77 STATIC_ENTRY("if-unmodified-since" , ""), // 42 | |
78 STATIC_ENTRY("last-modified" , ""), // 43 | |
79 STATIC_ENTRY("link" , ""), // 44 | |
80 STATIC_ENTRY("location" , ""), // 45 | |
81 STATIC_ENTRY("max-forwards" , ""), // 46 | |
82 STATIC_ENTRY("proxy-authenticate" , ""), // 47 | |
83 STATIC_ENTRY("proxy-authorization" , ""), // 48 | |
84 STATIC_ENTRY("range" , ""), // 49 | |
85 STATIC_ENTRY("referer" , ""), // 50 | |
86 STATIC_ENTRY("refresh" , ""), // 51 | |
87 STATIC_ENTRY("retry-after" , ""), // 52 | |
88 STATIC_ENTRY("server" , ""), // 53 | |
89 STATIC_ENTRY("set-cookie" , ""), // 54 | |
90 STATIC_ENTRY("strict-transport-security" , ""), // 55 | |
91 STATIC_ENTRY("transfer-encoding" , ""), // 56 | |
92 STATIC_ENTRY("user-agent" , ""), // 57 | |
93 STATIC_ENTRY("vary" , ""), // 58 | |
94 STATIC_ENTRY("via" , ""), // 59 | |
95 STATIC_ENTRY("www-authenticate" , ""), // 60 | |
96 }; | |
97 | |
98 #undef STATIC_ENTRY | |
99 | |
100 const size_t kStaticEntryCount = arraysize(kStaticTable); | |
101 | |
102 } // namespace | |
103 | |
104 // Must match HpackEntry::kUntouched. | |
105 const uint32 HpackEncodingContext::kUntouched = 0x7fffffff; | |
106 | |
107 HpackEncodingContext::HpackEncodingContext() | |
108 : settings_header_table_size_(kDefaultHeaderTableSizeSetting) { | |
109 DCHECK_EQ(HpackEncodingContext::kUntouched, HpackEntry::kUntouched); | |
110 } | |
111 | |
112 HpackEncodingContext::~HpackEncodingContext() {} | |
113 | |
114 uint32 HpackEncodingContext::GetMutableEntryCount() const { | |
115 return header_table_.GetEntryCount(); | |
116 } | |
117 | |
118 uint32 HpackEncodingContext::GetEntryCount() const { | |
119 return GetMutableEntryCount() + kStaticEntryCount; | |
120 } | |
121 | |
122 StringPiece HpackEncodingContext::GetNameAt(uint32 index) const { | |
123 CHECK_GE(index, 1u); | |
124 CHECK_LE(index, GetEntryCount()); | |
125 if (index > header_table_.GetEntryCount()) { | |
126 const StaticEntry& entry = | |
127 kStaticTable[index - header_table_.GetEntryCount() - 1]; | |
128 return StringPiece(entry.name, entry.name_len); | |
129 } | |
130 return header_table_.GetEntry(index).name(); | |
131 } | |
132 | |
133 StringPiece HpackEncodingContext::GetValueAt(uint32 index) const { | |
134 CHECK_GE(index, 1u); | |
135 CHECK_LE(index, GetEntryCount()); | |
136 if (index > header_table_.GetEntryCount()) { | |
137 const StaticEntry& entry = | |
138 kStaticTable[index - header_table_.GetEntryCount() - 1]; | |
139 return StringPiece(entry.value, entry.value_len); | |
140 } | |
141 return header_table_.GetEntry(index).value(); | |
142 } | |
143 | |
144 bool HpackEncodingContext::IsReferencedAt(uint32 index) const { | |
145 CHECK_GE(index, 1u); | |
146 CHECK_LE(index, GetEntryCount()); | |
147 if (index > header_table_.GetEntryCount()) | |
148 return false; | |
149 return header_table_.GetEntry(index).IsReferenced(); | |
150 } | |
151 | |
152 uint32 HpackEncodingContext::GetTouchCountAt(uint32 index) const { | |
153 CHECK_GE(index, 1u); | |
154 CHECK_LE(index, GetEntryCount()); | |
155 if (index > header_table_.GetEntryCount()) | |
156 return 0; | |
157 return header_table_.GetEntry(index).TouchCount(); | |
158 } | |
159 | |
160 void HpackEncodingContext::SetReferencedAt(uint32 index, bool referenced) { | |
161 header_table_.GetMutableEntry(index)->SetReferenced(referenced); | |
162 } | |
163 | |
164 void HpackEncodingContext::AddTouchesAt(uint32 index, uint32 touch_count) { | |
165 header_table_.GetMutableEntry(index)->AddTouches(touch_count); | |
166 } | |
167 | |
168 void HpackEncodingContext::ClearTouchesAt(uint32 index) { | |
169 header_table_.GetMutableEntry(index)->ClearTouches(); | |
170 } | |
171 | |
172 void HpackEncodingContext::ApplyHeaderTableSizeSetting(uint32 size) { | |
173 settings_header_table_size_ = size; | |
174 if (size < header_table_.max_size()) { | |
175 // Implicit maximum-size context update. | |
176 CHECK(ProcessContextUpdateNewMaximumSize(size)); | |
177 } | |
178 } | |
179 | |
180 bool HpackEncodingContext::ProcessContextUpdateNewMaximumSize(uint32 size) { | |
181 if (size > settings_header_table_size_) { | |
182 return false; | |
183 } | |
184 header_table_.SetMaxSize(size); | |
185 return true; | |
186 } | |
187 | |
188 bool HpackEncodingContext::ProcessContextUpdateEmptyReferenceSet() { | |
189 for (size_t i = 1; i <= header_table_.GetEntryCount(); ++i) { | |
190 HpackEntry* entry = header_table_.GetMutableEntry(i); | |
191 if (entry->IsReferenced()) { | |
192 entry->SetReferenced(false); | |
193 } | |
194 } | |
195 return true; | |
196 } | |
197 | |
198 bool HpackEncodingContext::ProcessIndexedHeader(uint32 index, uint32* new_index, | |
199 std::vector<uint32>* removed_referenced_indices) { | |
200 CHECK_GT(index, 0u); | |
201 CHECK_LT(index, GetEntryCount()); | |
202 | |
203 if (index <= header_table_.GetEntryCount()) { | |
204 *new_index = index; | |
205 removed_referenced_indices->clear(); | |
206 HpackEntry* entry = header_table_.GetMutableEntry(index); | |
207 entry->SetReferenced(!entry->IsReferenced()); | |
208 } else { | |
209 // TODO(akalin): Make HpackEntry know about owned strings and | |
210 // non-owned strings so that it can potentially avoid copies here. | |
211 HpackEntry entry(GetNameAt(index), GetValueAt(index)); | |
212 | |
213 header_table_.TryAddEntry(entry, new_index, removed_referenced_indices); | |
214 if (*new_index >= 1) { | |
215 header_table_.GetMutableEntry(*new_index)->SetReferenced(true); | |
216 } | |
217 } | |
218 return true; | |
219 } | |
220 | |
221 bool HpackEncodingContext::ProcessLiteralHeaderWithIncrementalIndexing( | |
222 StringPiece name, | |
223 StringPiece value, | |
224 uint32* index, | |
225 std::vector<uint32>* removed_referenced_indices) { | |
226 HpackEntry entry(name, value); | |
227 header_table_.TryAddEntry(entry, index, removed_referenced_indices); | |
228 if (*index >= 1) { | |
229 header_table_.GetMutableEntry(*index)->SetReferenced(true); | |
230 } | |
231 return true; | |
232 } | |
233 | |
234 } // namespace net | |
OLD | NEW |