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 |