OLD | NEW |
1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 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/spdy_frame_builder.h" | 5 #include "net/spdy/spdy_frame_builder.h" |
6 | 6 |
7 #include <limits> | 7 #include <limits> |
8 | 8 |
9 #include "base/logging.h" | 9 #include "base/logging.h" |
10 #include "net/spdy/spdy_framer.h" | 10 #include "net/spdy/spdy_framer.h" |
(...skipping 42 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
53 return false; | 53 return false; |
54 } | 54 } |
55 | 55 |
56 length_ += length; | 56 length_ += length; |
57 return true; | 57 return true; |
58 } | 58 } |
59 | 59 |
60 bool SpdyFrameBuilder::WriteControlFrameHeader(const SpdyFramer& framer, | 60 bool SpdyFrameBuilder::WriteControlFrameHeader(const SpdyFramer& framer, |
61 SpdyFrameType type, | 61 SpdyFrameType type, |
62 uint8 flags) { | 62 uint8 flags) { |
63 DCHECK_GE(type, FIRST_CONTROL_TYPE); | |
64 DCHECK_LE(type, LAST_CONTROL_TYPE); | |
65 DCHECK_GT(4, framer.protocol_version()); | 63 DCHECK_GT(4, framer.protocol_version()); |
| 64 DCHECK_NE(-1, |
| 65 SpdyConstants::SerializeFrameType(framer.protocol_version(), type)); |
66 bool success = true; | 66 bool success = true; |
67 FlagsAndLength flags_length = CreateFlagsAndLength( | 67 FlagsAndLength flags_length = CreateFlagsAndLength( |
68 flags, capacity_ - framer.GetControlFrameHeaderSize()); | 68 flags, capacity_ - framer.GetControlFrameHeaderSize()); |
69 success &= WriteUInt16(kControlFlagMask | framer.protocol_version()); | 69 success &= WriteUInt16(kControlFlagMask | framer.protocol_version()); |
70 success &= WriteUInt16(type); | 70 success &= WriteUInt16( |
| 71 SpdyConstants::SerializeFrameType(framer.protocol_version(), type)); |
71 success &= WriteBytes(&flags_length, sizeof(flags_length)); | 72 success &= WriteBytes(&flags_length, sizeof(flags_length)); |
72 DCHECK_EQ(framer.GetControlFrameHeaderSize(), length()); | 73 DCHECK_EQ(framer.GetControlFrameHeaderSize(), length()); |
73 return success; | 74 return success; |
74 } | 75 } |
75 | 76 |
76 bool SpdyFrameBuilder::WriteDataFrameHeader(const SpdyFramer& framer, | 77 bool SpdyFrameBuilder::WriteDataFrameHeader(const SpdyFramer& framer, |
77 SpdyStreamId stream_id, | 78 SpdyStreamId stream_id, |
78 uint8 flags) { | 79 uint8 flags) { |
79 if (framer.protocol_version() >= 4) { | 80 if (framer.protocol_version() >= 4) { |
80 return WriteFramePrefix(framer, DATA, flags, stream_id); | 81 return WriteFramePrefix(framer, DATA, flags, stream_id); |
81 } | 82 } |
82 DCHECK_EQ(0u, stream_id & ~kStreamIdMask); | 83 DCHECK_EQ(0u, stream_id & ~kStreamIdMask); |
83 bool success = true; | 84 bool success = true; |
84 success &= WriteUInt32(stream_id); | 85 success &= WriteUInt32(stream_id); |
85 size_t length_field = capacity_ - framer.GetDataFrameMinimumSize(); | 86 size_t length_field = capacity_ - framer.GetDataFrameMinimumSize(); |
86 DCHECK_EQ(0u, length_field & ~static_cast<size_t>(kLengthMask)); | 87 DCHECK_EQ(0u, length_field & ~static_cast<size_t>(kLengthMask)); |
87 FlagsAndLength flags_length; | 88 FlagsAndLength flags_length; |
88 flags_length.length_ = htonl(length_field); | 89 flags_length.length_ = htonl(length_field); |
89 DCHECK_EQ(0, flags & ~kDataFlagsMask); | 90 DCHECK_EQ(0, flags & ~kDataFlagsMask); |
90 flags_length.flags_[0] = flags; | 91 flags_length.flags_[0] = flags; |
91 success &= WriteBytes(&flags_length, sizeof(flags_length)); | 92 success &= WriteBytes(&flags_length, sizeof(flags_length)); |
92 DCHECK_EQ(framer.GetDataFrameMinimumSize(), length()); | 93 DCHECK_EQ(framer.GetDataFrameMinimumSize(), length()); |
93 return success; | 94 return success; |
94 } | 95 } |
95 | 96 |
96 bool SpdyFrameBuilder::WriteFramePrefix(const SpdyFramer& framer, | 97 bool SpdyFrameBuilder::WriteFramePrefix(const SpdyFramer& framer, |
97 SpdyFrameType type, | 98 SpdyFrameType type, |
98 uint8 flags, | 99 uint8 flags, |
99 SpdyStreamId stream_id) { | 100 SpdyStreamId stream_id) { |
100 DCHECK_LE(DATA, type); | 101 DCHECK_NE(-1, |
101 DCHECK_GE(LAST_CONTROL_TYPE, type); | 102 SpdyConstants::SerializeFrameType(framer.protocol_version(), type)); |
102 DCHECK_EQ(0u, stream_id & ~kStreamIdMask); | 103 DCHECK_EQ(0u, stream_id & ~kStreamIdMask); |
103 DCHECK_LE(4, framer.protocol_version()); | 104 DCHECK_LE(4, framer.protocol_version()); |
104 bool success = true; | 105 bool success = true; |
105 // Upstream DCHECK's that capacity_ is under the maximum frame size at this | 106 // Upstream DCHECK's that capacity_ is under the maximum frame size at this |
106 // point. Chromium does not, because of the large additional zlib inflation | 107 // point. Chromium does not, because of the large additional zlib inflation |
107 // factor we use. (Frame size is is still checked by OverwriteLength() below). | 108 // factor we use. (Frame size is is still checked by OverwriteLength() below). |
108 if (type != DATA) { | 109 if (type != DATA) { |
109 success &= WriteUInt16(capacity_ - framer.GetControlFrameHeaderSize()); | 110 success &= WriteUInt16(capacity_ - framer.GetControlFrameHeaderSize()); |
110 } else { | 111 } else { |
111 success &= WriteUInt16(capacity_ - framer.GetDataFrameMinimumSize()); | 112 success &= WriteUInt16(capacity_ - framer.GetDataFrameMinimumSize()); |
112 } | 113 } |
113 success &= WriteUInt8(type); | 114 success &= WriteUInt8( |
| 115 SpdyConstants::SerializeFrameType(framer.protocol_version(), type)); |
114 success &= WriteUInt8(flags); | 116 success &= WriteUInt8(flags); |
115 success &= WriteUInt32(stream_id); | 117 success &= WriteUInt32(stream_id); |
116 DCHECK_EQ(framer.GetDataFrameMinimumSize(), length()); | 118 DCHECK_EQ(framer.GetDataFrameMinimumSize(), length()); |
117 return success; | 119 return success; |
118 } | 120 } |
119 | 121 |
120 bool SpdyFrameBuilder::WriteString(const std::string& value) { | 122 bool SpdyFrameBuilder::WriteString(const std::string& value) { |
121 if (value.size() > 0xffff) { | 123 if (value.size() > 0xffff) { |
122 DCHECK(false) << "Tried to write string with length > 16bit."; | 124 DCHECK(false) << "Tried to write string with length > 16bit."; |
123 return false; | 125 return false; |
(...skipping 78 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
202 | 204 |
203 if (length_ + length > capacity_) { | 205 if (length_ + length > capacity_) { |
204 DCHECK(false); | 206 DCHECK(false); |
205 return false; | 207 return false; |
206 } | 208 } |
207 | 209 |
208 return true; | 210 return true; |
209 } | 211 } |
210 | 212 |
211 } // namespace net | 213 } // namespace net |
OLD | NEW |