OLD | NEW |
1 // Copyright (c) 2016 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2016 The Chromium Authors. All rights reserved. |
2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #include "net/spdy/header_coalescer.h" | 5 #include "net/spdy/header_coalescer.h" |
6 | 6 |
7 #include "base/strings/string_util.h" | 7 #include "base/strings/string_util.h" |
8 | 8 |
9 namespace net { | 9 namespace net { |
10 | 10 |
11 const size_t kMaxHeaderListSize = 256 * 1024; | 11 const size_t kMaxHeaderListSize = 256 * 1024; |
12 | 12 |
13 void HeaderCoalescer::OnHeader(base::StringPiece key, base::StringPiece value) { | 13 void HeaderCoalescer::OnHeader(base::StringPiece key, base::StringPiece value) { |
14 if (error_seen_) | 14 if (error_seen_) { |
15 return; | 15 return; |
| 16 } |
16 | 17 |
17 if (key.empty()) { | 18 if (key.empty()) { |
18 DVLOG(1) << "Header name must not be empty."; | 19 DVLOG(1) << "Header name must not be empty."; |
19 error_seen_ = true; | 20 error_seen_ = true; |
20 return; | 21 return; |
21 } | 22 } |
22 | 23 |
23 // 32 byte overhead according to RFC 7540 Section 6.5.2. | 24 // 32 byte overhead according to RFC 7540 Section 6.5.2. |
24 header_list_size_ += key.size() + value.size() + 32; | 25 header_list_size_ += key.size() + value.size() + 32; |
25 if (header_list_size_ > kMaxHeaderListSize) { | 26 if (header_list_size_ > kMaxHeaderListSize) { |
26 error_seen_ = true; | 27 error_seen_ = true; |
27 return; | 28 return; |
28 } | 29 } |
29 | 30 |
| 31 if (key[0] == ':') { |
| 32 if (protocol_version_ == HTTP2 && regular_header_seen_) { |
| 33 error_seen_ = true; |
| 34 return; |
| 35 } |
| 36 } else { |
| 37 regular_header_seen_ = true; |
| 38 } |
| 39 |
30 auto iter = headers_.find(key); | 40 auto iter = headers_.find(key); |
31 if (iter == headers_.end()) { | 41 if (iter == headers_.end()) { |
32 headers_[key] = value; | 42 headers_[key] = value; |
33 } else { | 43 } else { |
34 // This header had multiple values, so it must be reconstructed. | 44 // This header had multiple values, so it must be reconstructed. |
35 base::StringPiece v = iter->second; | 45 base::StringPiece v = iter->second; |
36 std::string s(v.data(), v.length()); | 46 std::string s(v.data(), v.length()); |
37 if (key == "cookie") { | 47 if (key == "cookie") { |
38 // Obeys section 8.1.2.5 in RFC 7540 for cookie reconstruction. | 48 // Obeys section 8.1.2.5 in RFC 7540 for cookie reconstruction. |
39 s.append("; "); | 49 s.append("; "); |
40 } else { | 50 } else { |
41 base::StringPiece("\0", 1).AppendToString(&s); | 51 base::StringPiece("\0", 1).AppendToString(&s); |
42 } | 52 } |
43 value.AppendToString(&s); | 53 value.AppendToString(&s); |
44 headers_.ReplaceOrAppendHeader(key, s); | 54 headers_.ReplaceOrAppendHeader(key, s); |
45 } | 55 } |
46 } | 56 } |
47 | 57 |
48 } // namespace net | 58 } // namespace net |
OLD | NEW |