| 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 |