| 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 "net/spdy/spdy_framer.h" | 7 #include "net/spdy/spdy_framer.h" |
| 8 #include "net/tools/quic/quic_client_session.h" | 8 #include "net/tools/quic/quic_client_session.h" |
| 9 #include "net/tools/quic/spdy_utils.h" | 9 #include "net/tools/quic/spdy_utils.h" |
| 10 | 10 |
| (...skipping 19 matching lines...) Expand all Loading... |
| 30 | 30 |
| 31 bool QuicSpdyClientStream::OnStreamFrame(const QuicStreamFrame& frame) { | 31 bool QuicSpdyClientStream::OnStreamFrame(const QuicStreamFrame& frame) { |
| 32 if (!write_side_closed()) { | 32 if (!write_side_closed()) { |
| 33 DVLOG(1) << "Got a response before the request was complete. " | 33 DVLOG(1) << "Got a response before the request was complete. " |
| 34 << "Aborting request."; | 34 << "Aborting request."; |
| 35 CloseWriteSide(); | 35 CloseWriteSide(); |
| 36 } | 36 } |
| 37 return QuicDataStream::OnStreamFrame(frame); | 37 return QuicDataStream::OnStreamFrame(frame); |
| 38 } | 38 } |
| 39 | 39 |
| 40 void QuicSpdyClientStream::OnStreamHeadersComplete(bool fin, | 40 void QuicSpdyClientStream::OnStreamHeadersComplete(bool fin, size_t frame_len) { |
| 41 size_t frame_len) { | |
| 42 header_bytes_read_ = frame_len; | 41 header_bytes_read_ = frame_len; |
| 43 QuicDataStream::OnStreamHeadersComplete(fin, frame_len); | 42 QuicDataStream::OnStreamHeadersComplete(fin, frame_len); |
| 44 } | 43 } |
| 45 | 44 |
| 46 uint32 QuicSpdyClientStream::ProcessData(const char* data, | 45 uint32 QuicSpdyClientStream::ProcessData(const char* data, uint32 data_len) { |
| 47 uint32 data_len) { | |
| 48 int total_bytes_processed = 0; | 46 int total_bytes_processed = 0; |
| 49 | 47 |
| 50 // Are we still reading the response headers. | 48 // Are we still reading the response headers. |
| 51 if (!response_headers_received_) { | 49 if (!response_headers_received_) { |
| 52 // Grow the read buffer if necessary. | 50 // Grow the read buffer if necessary. |
| 53 if (read_buf_->RemainingCapacity() < (int)data_len) { | 51 if (read_buf_->RemainingCapacity() < (int)data_len) { |
| 54 read_buf_->SetCapacity(read_buf_->capacity() + kHeaderBufInitialSize); | 52 read_buf_->SetCapacity(read_buf_->capacity() + kHeaderBufInitialSize); |
| 55 } | 53 } |
| 56 memcpy(read_buf_->data(), data, data_len); | 54 memcpy(read_buf_->data(), data, data_len); |
| 57 read_buf_->set_offset(read_buf_->offset() + data_len); | 55 read_buf_->set_offset(read_buf_->offset() + data_len); |
| 58 ParseResponseHeaders(); | 56 ParseResponseHeaders(); |
| 59 } else { | 57 } else { |
| 60 data_.append(data + total_bytes_processed, | 58 data_.append(data + total_bytes_processed, |
| 61 data_len - total_bytes_processed); | 59 data_len - total_bytes_processed); |
| 62 } | 60 } |
| 63 return data_len; | 61 return data_len; |
| 64 } | 62 } |
| 65 | 63 |
| 66 void QuicSpdyClientStream::OnFinRead() { | 64 void QuicSpdyClientStream::OnFinRead() { |
| 67 ReliableQuicStream::OnFinRead(); | 65 ReliableQuicStream::OnFinRead(); |
| 68 if (!response_headers_received_) { | 66 if (!response_headers_received_) { |
| 69 Reset(QUIC_BAD_APPLICATION_PAYLOAD); | 67 Reset(QUIC_BAD_APPLICATION_PAYLOAD); |
| 70 } else if ((headers().content_length_status() == | 68 } else if ((headers().content_length_status() == |
| 71 BalsaHeadersEnums::VALID_CONTENT_LENGTH) && | 69 BalsaHeadersEnums::VALID_CONTENT_LENGTH) && |
| 72 data_.size() != headers().content_length()) { | 70 data_.size() != headers().content_length()) { |
| 73 Reset(QUIC_BAD_APPLICATION_PAYLOAD); | 71 Reset(QUIC_BAD_APPLICATION_PAYLOAD); |
| 74 } | 72 } |
| 75 } | 73 } |
| 76 | 74 |
| 77 ssize_t QuicSpdyClientStream::SendRequest(const BalsaHeaders& headers, | 75 ssize_t QuicSpdyClientStream::SendRequest(const BalsaHeaders& headers, |
| 78 StringPiece body, | 76 StringPiece body, |
| 79 bool fin) { | 77 bool fin) { |
| 80 SpdyHeaderBlock header_block = | 78 SpdyHeaderBlock header_block = |
| 81 SpdyUtils::RequestHeadersToSpdyHeaders(headers); | 79 SpdyUtils::RequestHeadersToSpdyHeaders(headers); |
| 82 | 80 |
| 83 bool send_fin_with_headers = fin && body.empty(); | 81 bool send_fin_with_headers = fin && body.empty(); |
| 84 size_t bytes_sent = body.size(); | 82 size_t bytes_sent = body.size(); |
| 85 header_bytes_written_ = WriteHeaders( | 83 header_bytes_written_ = |
| 86 header_block, send_fin_with_headers, NULL); | 84 WriteHeaders(header_block, send_fin_with_headers, NULL); |
| 87 bytes_sent += header_bytes_written_; | 85 bytes_sent += header_bytes_written_; |
| 88 | 86 |
| 89 if (!body.empty()) { | 87 if (!body.empty()) { |
| 90 WriteOrBufferData(body, fin, NULL); | 88 WriteOrBufferData(body, fin, NULL); |
| 91 } | 89 } |
| 92 | 90 |
| 93 return bytes_sent; | 91 return bytes_sent; |
| 94 } | 92 } |
| 95 | 93 |
| 96 int QuicSpdyClientStream::ParseResponseHeaders() { | 94 int QuicSpdyClientStream::ParseResponseHeaders() { |
| 97 size_t read_buf_len = static_cast<size_t>(read_buf_->offset()); | 95 size_t read_buf_len = static_cast<size_t>(read_buf_->offset()); |
| 98 SpdyFramer framer(SPDY3); | 96 SpdyFramer framer(SPDY3); |
| 99 SpdyHeaderBlock headers; | 97 SpdyHeaderBlock headers; |
| 100 char* data = read_buf_->StartOfBuffer(); | 98 char* data = read_buf_->StartOfBuffer(); |
| 101 size_t len = framer.ParseHeaderBlockInBuffer(data, read_buf_->offset(), | 99 size_t len = |
| 102 &headers); | 100 framer.ParseHeaderBlockInBuffer(data, read_buf_->offset(), &headers); |
| 103 if (len == 0) { | 101 if (len == 0) { |
| 104 return -1; | 102 return -1; |
| 105 } | 103 } |
| 106 | 104 |
| 107 if (!SpdyUtils::FillBalsaResponseHeaders(headers, &headers_)) { | 105 if (!SpdyUtils::FillBalsaResponseHeaders(headers, &headers_)) { |
| 108 Reset(QUIC_BAD_APPLICATION_PAYLOAD); | 106 Reset(QUIC_BAD_APPLICATION_PAYLOAD); |
| 109 return -1; | 107 return -1; |
| 110 } | 108 } |
| 111 response_headers_received_ = true; | 109 response_headers_received_ = true; |
| 112 | 110 |
| 113 size_t delta = read_buf_len - len; | 111 size_t delta = read_buf_len - len; |
| 114 if (delta > 0) { | 112 if (delta > 0) { |
| 115 data_.append(data + len, delta); | 113 data_.append(data + len, delta); |
| 116 } | 114 } |
| 117 | 115 |
| 118 return len; | 116 return len; |
| 119 } | 117 } |
| 120 | 118 |
| 121 // Sends body data to the server and returns the number of bytes sent. | 119 // Sends body data to the server and returns the number of bytes sent. |
| 122 void QuicSpdyClientStream::SendBody(const string& data, bool fin) { | 120 void QuicSpdyClientStream::SendBody(const string& data, bool fin) { |
| 123 WriteOrBufferData(data, fin, NULL); | 121 WriteOrBufferData(data, fin, NULL); |
| 124 } | 122 } |
| 125 | 123 |
| 126 } // namespace tools | 124 } // namespace tools |
| 127 } // namespace net | 125 } // namespace net |
| OLD | NEW |