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 |