| 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/tools/quic/quic_spdy_client_stream.h" | 5 #include "net/tools/quic/quic_spdy_client_stream.h" |
| 6 | 6 |
| 7 #include "base/logging.h" | 7 #include "base/logging.h" |
| 8 #include "base/stl_util.h" | 8 #include "base/stl_util.h" |
| 9 #include "base/strings/string_number_conversions.h" | 9 #include "base/strings/string_number_conversions.h" |
| 10 #include "net/quic/quic_alarm.h" | 10 #include "net/quic/quic_alarm.h" |
| (...skipping 50 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 61 Reset(QUIC_BAD_APPLICATION_PAYLOAD); | 61 Reset(QUIC_BAD_APPLICATION_PAYLOAD); |
| 62 return; | 62 return; |
| 63 } | 63 } |
| 64 | 64 |
| 65 MarkHeadersConsumed(decompressed_headers().length()); | 65 MarkHeadersConsumed(decompressed_headers().length()); |
| 66 DVLOG(1) << "headers complete for stream " << id(); | 66 DVLOG(1) << "headers complete for stream " << id(); |
| 67 | 67 |
| 68 session_->OnInitialHeadersComplete(id(), response_headers_); | 68 session_->OnInitialHeadersComplete(id(), response_headers_); |
| 69 } | 69 } |
| 70 | 70 |
| 71 void QuicSpdyClientStream::OnInitialHeadersComplete( |
| 72 bool fin, |
| 73 size_t frame_len, |
| 74 const QuicHeaderList& header_list) { |
| 75 QuicSpdyStream::OnInitialHeadersComplete(fin, frame_len, header_list); |
| 76 |
| 77 DCHECK(headers_decompressed()); |
| 78 header_bytes_read_ += frame_len; |
| 79 if (!SpdyUtils::CopyAndValidateHeaders(header_list, &content_length_, |
| 80 &response_headers_)) { |
| 81 DLOG(ERROR) << "Failed to parse header list: " << header_list.DebugString(); |
| 82 Reset(QUIC_BAD_APPLICATION_PAYLOAD); |
| 83 return; |
| 84 } |
| 85 |
| 86 if (!ParseHeaderStatusCode(&response_headers_, &response_code_)) { |
| 87 DLOG(ERROR) << "Received invalid response code: " |
| 88 << response_headers_[":status"].as_string(); |
| 89 Reset(QUIC_BAD_APPLICATION_PAYLOAD); |
| 90 return; |
| 91 } |
| 92 |
| 93 ConsumeHeaderList(); |
| 94 DVLOG(1) << "headers complete for stream " << id(); |
| 95 |
| 96 session_->OnInitialHeadersComplete(id(), response_headers_); |
| 97 } |
| 98 |
| 71 void QuicSpdyClientStream::OnTrailingHeadersComplete(bool fin, | 99 void QuicSpdyClientStream::OnTrailingHeadersComplete(bool fin, |
| 72 size_t frame_len) { | 100 size_t frame_len) { |
| 73 QuicSpdyStream::OnTrailingHeadersComplete(fin, frame_len); | 101 QuicSpdyStream::OnTrailingHeadersComplete(fin, frame_len); |
| 74 MarkTrailersConsumed(decompressed_trailers().length()); | 102 MarkTrailersConsumed(decompressed_trailers().length()); |
| 75 } | 103 } |
| 76 | 104 |
| 77 void QuicSpdyClientStream::OnPromiseHeadersComplete(QuicStreamId promised_id, | 105 void QuicSpdyClientStream::OnPromiseHeadersComplete(QuicStreamId promised_id, |
| 78 size_t frame_len) { | 106 size_t frame_len) { |
| 79 header_bytes_read_ += frame_len; | 107 header_bytes_read_ += frame_len; |
| 80 int64_t content_length = -1; | 108 int64_t content_length = -1; |
| 81 SpdyHeaderBlock promise_headers; | 109 SpdyHeaderBlock promise_headers; |
| 82 if (!SpdyUtils::ParseHeaders(decompressed_headers().data(), | 110 if (!SpdyUtils::ParseHeaders(decompressed_headers().data(), |
| 83 decompressed_headers().length(), &content_length, | 111 decompressed_headers().length(), &content_length, |
| 84 &promise_headers)) { | 112 &promise_headers)) { |
| 85 DLOG(ERROR) << "Failed to parse promise headers: " | 113 DLOG(ERROR) << "Failed to parse promise headers: " |
| 86 << decompressed_headers(); | 114 << decompressed_headers(); |
| 87 Reset(QUIC_BAD_APPLICATION_PAYLOAD); | 115 Reset(QUIC_BAD_APPLICATION_PAYLOAD); |
| 88 return; | 116 return; |
| 89 } | 117 } |
| 90 MarkHeadersConsumed(decompressed_headers().length()); | 118 MarkHeadersConsumed(decompressed_headers().length()); |
| 91 | 119 |
| 92 session_->HandlePromised(id(), promised_id, promise_headers); | 120 session_->HandlePromised(id(), promised_id, promise_headers); |
| 93 } | 121 } |
| 94 | 122 |
| 123 void QuicSpdyClientStream::OnPromiseHeaderList( |
| 124 QuicStreamId promised_id, |
| 125 size_t frame_len, |
| 126 const QuicHeaderList& header_list) { |
| 127 header_bytes_read_ += frame_len; |
| 128 int64_t content_length = -1; |
| 129 SpdyHeaderBlock promise_headers; |
| 130 if (!SpdyUtils::CopyAndValidateHeaders(header_list, &content_length, |
| 131 &promise_headers)) { |
| 132 DLOG(ERROR) << "Failed to parse promise headers: " |
| 133 << header_list.DebugString(); |
| 134 Reset(QUIC_BAD_APPLICATION_PAYLOAD); |
| 135 return; |
| 136 } |
| 137 |
| 138 session_->HandlePromised(id(), promised_id, promise_headers); |
| 139 } |
| 140 |
| 95 void QuicSpdyClientStream::OnDataAvailable() { | 141 void QuicSpdyClientStream::OnDataAvailable() { |
| 96 if (FLAGS_quic_supports_push_promise) { | 142 if (FLAGS_quic_supports_push_promise) { |
| 97 // For push streams, visitor will not be set until the rendezvous | 143 // For push streams, visitor will not be set until the rendezvous |
| 98 // between server promise and client request is complete. | 144 // between server promise and client request is complete. |
| 99 if (visitor() == nullptr) | 145 if (visitor() == nullptr) |
| 100 return; | 146 return; |
| 101 } | 147 } |
| 102 | 148 |
| 103 while (HasBytesToRead()) { | 149 while (HasBytesToRead()) { |
| 104 struct iovec iov; | 150 struct iovec iov; |
| (...skipping 30 matching lines...) Expand all Loading... |
| 135 bytes_sent += header_bytes_written_; | 181 bytes_sent += header_bytes_written_; |
| 136 | 182 |
| 137 if (!body.empty()) { | 183 if (!body.empty()) { |
| 138 WriteOrBufferData(body, fin, nullptr); | 184 WriteOrBufferData(body, fin, nullptr); |
| 139 } | 185 } |
| 140 | 186 |
| 141 return bytes_sent; | 187 return bytes_sent; |
| 142 } | 188 } |
| 143 | 189 |
| 144 } // namespace net | 190 } // namespace net |
| OLD | NEW |