| 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_simple_client_stream.h" | 5 #include "net/tools/quic/quic_spdy_client_stream.h" |
| 6 | 6 |
| 7 #include "net/http/http_response_info.h" | |
| 8 #include "net/spdy/spdy_framer.h" | 7 #include "net/spdy/spdy_framer.h" |
| 9 #include "net/spdy/spdy_http_utils.h" | 8 #include "net/tools/quic/quic_client_session.h" |
| 10 #include "net/tools/quic/quic_simple_client_session.h" | 9 #include "net/tools/quic/spdy_utils.h" |
| 11 | 10 |
| 12 using base::StringPiece; | 11 using base::StringPiece; |
| 13 using std::string; | 12 using std::string; |
| 14 | 13 |
| 15 namespace net { | 14 namespace net { |
| 16 namespace tools { | 15 namespace tools { |
| 17 | 16 |
| 18 static const size_t kHeaderBufInitialSize = 4096; | 17 static const size_t kHeaderBufInitialSize = 4096; |
| 19 | 18 |
| 20 QuicSimpleClientStream::QuicSimpleClientStream(QuicStreamId id, | 19 QuicSpdyClientStream::QuicSpdyClientStream(QuicStreamId id, |
| 21 QuicSimpleClientSession* session) | 20 QuicClientSession* session) |
| 22 : QuicDataStream(id, session), | 21 : QuicDataStream(id, session), |
| 23 read_buf_(new GrowableIOBuffer()), | 22 read_buf_(new GrowableIOBuffer()), |
| 24 response_headers_received_(false), | 23 response_headers_received_(false), |
| 25 header_bytes_read_(0), | 24 header_bytes_read_(0), |
| 26 header_bytes_written_(0) { | 25 header_bytes_written_(0) { |
| 27 read_buf_->SetCapacity(kHeaderBufInitialSize); | |
| 28 } | 26 } |
| 29 | 27 |
| 30 QuicSimpleClientStream::~QuicSimpleClientStream() { | 28 QuicSpdyClientStream::~QuicSpdyClientStream() { |
| 31 } | 29 } |
| 32 | 30 |
| 33 void QuicSimpleClientStream::OnStreamFrame(const QuicStreamFrame& frame) { | 31 void QuicSpdyClientStream::OnStreamFrame(const QuicStreamFrame& frame) { |
| 34 if (!write_side_closed()) { | 32 if (!write_side_closed()) { |
| 35 DVLOG(1) << "Got a response before the request was complete. " | 33 DVLOG(1) << "Got a response before the request was complete. " |
| 36 << "Aborting request."; | 34 << "Aborting request."; |
| 37 CloseWriteSide(); | 35 CloseWriteSide(); |
| 38 } | 36 } |
| 39 QuicDataStream::OnStreamFrame(frame); | 37 QuicDataStream::OnStreamFrame(frame); |
| 40 } | 38 } |
| 41 | 39 |
| 42 void QuicSimpleClientStream::OnStreamHeadersComplete(bool fin, | 40 void QuicSpdyClientStream::OnStreamHeadersComplete(bool fin, |
| 43 size_t frame_len) { | 41 size_t frame_len) { |
| 44 header_bytes_read_ = frame_len; | 42 header_bytes_read_ = frame_len; |
| 45 QuicDataStream::OnStreamHeadersComplete(fin, frame_len); | 43 QuicDataStream::OnStreamHeadersComplete(fin, frame_len); |
| 46 } | 44 } |
| 47 | 45 |
| 48 uint32 QuicSimpleClientStream::ProcessData(const char* data, | 46 uint32 QuicSpdyClientStream::ProcessData(const char* data, |
| 49 uint32 data_len) { | 47 uint32 data_len) { |
| 50 int total_bytes_processed = 0; | 48 int total_bytes_processed = 0; |
| 51 | 49 |
| 52 // Are we still reading the response headers. | 50 // Are we still reading the response headers. |
| 53 if (!response_headers_received_) { | 51 if (!response_headers_received_) { |
| 54 // Grow the read buffer if necessary. | 52 // Grow the read buffer if necessary. |
| 55 if (read_buf_->RemainingCapacity() < (int)data_len) { | 53 if (read_buf_->RemainingCapacity() < (int)data_len) { |
| 56 read_buf_->SetCapacity(read_buf_->capacity() + kHeaderBufInitialSize); | 54 read_buf_->SetCapacity(read_buf_->capacity() + kHeaderBufInitialSize); |
| 57 } | 55 } |
| 58 memcpy(read_buf_->data(), data, data_len); | 56 memcpy(read_buf_->data(), data, data_len); |
| 59 read_buf_->set_offset(read_buf_->offset() + data_len); | 57 read_buf_->set_offset(read_buf_->offset() + data_len); |
| 60 ParseResponseHeaders(); | 58 ParseResponseHeaders(); |
| 61 } else { | 59 } else { |
| 62 data_.append(data + total_bytes_processed, | 60 data_.append(data + total_bytes_processed, |
| 63 data_len - total_bytes_processed); | 61 data_len - total_bytes_processed); |
| 64 } | 62 } |
| 65 return data_len; | 63 return data_len; |
| 66 } | 64 } |
| 67 | 65 |
| 68 void QuicSimpleClientStream::OnFinRead() { | 66 void QuicSpdyClientStream::OnFinRead() { |
| 69 ReliableQuicStream::OnFinRead(); | 67 ReliableQuicStream::OnFinRead(); |
| 70 if (!response_headers_received_) { | 68 if (!response_headers_received_) { |
| 71 Reset(QUIC_BAD_APPLICATION_PAYLOAD); | 69 Reset(QUIC_BAD_APPLICATION_PAYLOAD); |
| 72 } else if (headers()->GetContentLength() != -1 && | 70 } else if ((headers().content_length_status() == |
| 73 data_.size() != | 71 BalsaHeadersEnums::VALID_CONTENT_LENGTH) && |
| 74 static_cast<size_t>(headers()->GetContentLength())) { | 72 data_.size() != headers().content_length()) { |
| 75 Reset(QUIC_BAD_APPLICATION_PAYLOAD); | 73 Reset(QUIC_BAD_APPLICATION_PAYLOAD); |
| 76 } | 74 } |
| 77 } | 75 } |
| 78 | 76 |
| 79 size_t QuicSimpleClientStream::SendRequest(const HttpRequestInfo& headers, | 77 ssize_t QuicSpdyClientStream::SendRequest(const BalsaHeaders& headers, |
| 80 StringPiece body, | 78 StringPiece body, |
| 81 bool fin) { | 79 bool fin) { |
| 82 SpdyHeaderBlock header_block; | 80 SpdyHeaderBlock header_block = |
| 83 CreateSpdyHeadersFromHttpRequest(headers, | 81 SpdyUtils::RequestHeadersToSpdyHeaders(headers); |
| 84 headers.extra_headers, | |
| 85 SPDY3, | |
| 86 /*direct=*/ true, | |
| 87 &header_block); | |
| 88 | 82 |
| 89 bool send_fin_with_headers = fin && body.empty(); | 83 bool send_fin_with_headers = fin && body.empty(); |
| 90 size_t bytes_sent = body.size(); | 84 size_t bytes_sent = body.size(); |
| 91 header_bytes_written_ = | 85 header_bytes_written_ = |
| 92 WriteHeaders(header_block, send_fin_with_headers, nullptr); | 86 WriteHeaders(header_block, send_fin_with_headers, nullptr); |
| 93 bytes_sent += header_bytes_written_; | 87 bytes_sent += header_bytes_written_; |
| 94 | 88 |
| 95 if (!body.empty()) { | 89 if (!body.empty()) { |
| 96 WriteOrBufferData(body, fin, nullptr); | 90 WriteOrBufferData(body, fin, nullptr); |
| 97 } | 91 } |
| 98 | 92 |
| 99 return bytes_sent; | 93 return bytes_sent; |
| 100 } | 94 } |
| 101 | 95 |
| 102 int QuicSimpleClientStream::ParseResponseHeaders() { | 96 int QuicSpdyClientStream::ParseResponseHeaders() { |
| 103 size_t read_buf_len = static_cast<size_t>(read_buf_->offset()); | 97 size_t read_buf_len = static_cast<size_t>(read_buf_->offset()); |
| 104 SpdyFramer framer(SPDY3); | 98 SpdyFramer framer(SPDY3); |
| 105 SpdyHeaderBlock headers; | 99 SpdyHeaderBlock headers; |
| 106 char* data = read_buf_->StartOfBuffer(); | 100 char* data = read_buf_->StartOfBuffer(); |
| 107 size_t len = framer.ParseHeaderBlockInBuffer(data, read_buf_->offset(), | 101 size_t len = framer.ParseHeaderBlockInBuffer(data, read_buf_->offset(), |
| 108 &headers); | 102 &headers); |
| 109 if (len == 0) { | 103 if (len == 0) { |
| 110 return -1; | 104 return -1; |
| 111 } | 105 } |
| 112 | 106 |
| 113 HttpResponseInfo info; | 107 if (!SpdyUtils::FillBalsaResponseHeaders(headers, &headers_)) { |
| 114 if (!SpdyHeadersToHttpResponse(headers, SPDY3, &info)) { | |
| 115 Reset(QUIC_BAD_APPLICATION_PAYLOAD); | 108 Reset(QUIC_BAD_APPLICATION_PAYLOAD); |
| 116 return -1; | 109 return -1; |
| 117 } | 110 } |
| 118 headers_ = info.headers; | |
| 119 | |
| 120 response_headers_received_ = true; | 111 response_headers_received_ = true; |
| 121 | 112 |
| 122 size_t delta = read_buf_len - len; | 113 size_t delta = read_buf_len - len; |
| 123 if (delta > 0) { | 114 if (delta > 0) { |
| 124 data_.append(data + len, delta); | 115 data_.append(data + len, delta); |
| 125 } | 116 } |
| 126 | 117 |
| 127 return static_cast<int>(len); | 118 return len; |
| 128 } | 119 } |
| 129 | 120 |
| 130 void QuicSimpleClientStream::SendBody(const string& data, bool fin) { | 121 void QuicSpdyClientStream::SendBody(const string& data, bool fin) { |
| 131 SendBody(data, fin, nullptr); | 122 SendBody(data, fin, nullptr); |
| 132 } | 123 } |
| 133 | 124 |
| 134 void QuicSimpleClientStream::SendBody( | 125 void QuicSpdyClientStream::SendBody( |
| 135 const string& data, | 126 const string& data, |
| 136 bool fin, | 127 bool fin, |
| 137 QuicAckNotifier::DelegateInterface* delegate) { | 128 QuicAckNotifier::DelegateInterface* delegate) { |
| 138 WriteOrBufferData(data, fin, delegate); | 129 WriteOrBufferData(data, fin, delegate); |
| 139 } | 130 } |
| 140 | 131 |
| 141 } // namespace tools | 132 } // namespace tools |
| 142 } // namespace net | 133 } // namespace net |
| OLD | NEW |