Chromium Code Reviews| OLD | NEW |
|---|---|
| 1 // Copyright 2014 The Chromium Authors. All rights reserved. | 1 // Copyright 2014 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/quic/quic_spdy_server_stream.h" | 5 #include "net/quic/quic_spdy_server_stream.h" |
| 6 | 6 |
| 7 #include "base/memory/singleton.h" | 7 #include "base/memory/singleton.h" |
| 8 #include "base/strings/string_number_conversions.h" | 8 #include "base/strings/string_number_conversions.h" |
| 9 #include "net/quic/quic_in_memory_cache.h" | 9 #include "net/quic/quic_in_memory_cache.h" |
| 10 #include "net/quic/quic_session.h" | 10 #include "net/quic/quic_session.h" |
| (...skipping 14 matching lines...) Expand all Loading... | |
| 25 : QuicDataStream(id, session), | 25 : QuicDataStream(id, session), |
| 26 read_buf_(new GrowableIOBuffer()), | 26 read_buf_(new GrowableIOBuffer()), |
| 27 request_headers_received_(false) { | 27 request_headers_received_(false) { |
| 28 read_buf_->SetCapacity(kHeaderBufInitialSize); | 28 read_buf_->SetCapacity(kHeaderBufInitialSize); |
| 29 } | 29 } |
| 30 | 30 |
| 31 QuicSpdyServerStream::~QuicSpdyServerStream() { | 31 QuicSpdyServerStream::~QuicSpdyServerStream() { |
| 32 } | 32 } |
| 33 | 33 |
| 34 uint32 QuicSpdyServerStream::ProcessData(const char* data, uint32 data_len) { | 34 uint32 QuicSpdyServerStream::ProcessData(const char* data, uint32 data_len) { |
| 35 if (data_len > INT_MAX) { | |
|
Ryan Hamilton
2014/07/02 19:31:31
Is INT_MAX the right type here, since int *might*
wtc
2014/07/02 19:39:08
This protects the cast of 'data_len' to 'int' on l
| |
| 36 LOG(DFATAL) << "Data length too long: " << data_len; | |
| 37 return 0; | |
| 38 } | |
| 35 // Are we still reading the request headers. | 39 // Are we still reading the request headers. |
| 36 if (!request_headers_received_) { | 40 if (!request_headers_received_) { |
| 37 // Grow the read buffer if necessary. | 41 // Grow the read buffer if necessary. |
| 38 while (read_buf_->RemainingCapacity() < (int)data_len) { | 42 while (read_buf_->RemainingCapacity() < (int)data_len) { |
|
wtc
2014/07/02 19:39:08
Please take the opportunity to change this to stat
dmz
2014/07/02 20:58:39
Done.
| |
| 39 read_buf_->SetCapacity(read_buf_->capacity() * 2); | 43 read_buf_->SetCapacity(read_buf_->capacity() * 2); |
| 40 } | 44 } |
| 41 memcpy(read_buf_->data(), data, data_len); | 45 memcpy(read_buf_->data(), data, data_len); |
| 42 read_buf_->set_offset(read_buf_->offset() + data_len); | 46 read_buf_->set_offset(read_buf_->offset() + data_len); |
| 43 // Try parsing the request headers. This will set request_headers_received_ | 47 // Try parsing the request headers. This will set request_headers_received_ |
| 44 // if successful; if not, it will be tried again with more data. | 48 // if successful; if not, it will be tried again with more data. |
| 45 ParseRequestHeaders(); | 49 ParseRequestHeaders(); |
| 46 } else { | 50 } else { |
| 47 body_.append(data, data_len); | 51 body_.append(data, data_len); |
| 48 } | 52 } |
| (...skipping 21 matching lines...) Expand all Loading... | |
| 70 } | 74 } |
| 71 | 75 |
| 72 SendResponse(); | 76 SendResponse(); |
| 73 } | 77 } |
| 74 | 78 |
| 75 // Try parsing the request headers. If successful, sets | 79 // Try parsing the request headers. If successful, sets |
| 76 // request_headers_received_. If not successful, it can just be tried again once | 80 // request_headers_received_. If not successful, it can just be tried again once |
| 77 // there's more data. | 81 // there's more data. |
| 78 void QuicSpdyServerStream::ParseRequestHeaders() { | 82 void QuicSpdyServerStream::ParseRequestHeaders() { |
| 79 SpdyFramer framer((kDefaultSpdyMajorVersion)); | 83 SpdyFramer framer((kDefaultSpdyMajorVersion)); |
| 80 char* data = read_buf_->StartOfBuffer(); | 84 const char* data = read_buf_->StartOfBuffer(); |
| 81 size_t read_buf_len = static_cast<size_t>(read_buf_->offset()); | 85 size_t read_buf_len = static_cast<size_t>(read_buf_->offset()); |
| 82 size_t len = framer.ParseHeaderBlockInBuffer(data, read_buf_len, &headers_); | 86 size_t len = framer.ParseHeaderBlockInBuffer(data, read_buf_len, &headers_); |
| 83 if (len == 0) { | 87 if (len == 0) { |
| 84 // Not enough data yet, presumably. (If we still don't succeed by the end of | 88 // Not enough data yet, presumably. (If we still don't succeed by the end of |
| 85 // the stream, then we'll error above.) | 89 // the stream, then we'll error in OnFinRead().) |
| 86 return; | 90 return; |
| 87 } | 91 } |
| 88 | 92 |
| 89 // Headers received and parsed: extract the request URL. | 93 // Headers received and parsed: extract the request URL. |
| 90 request_url_ = GetUrlFromHeaderBlock(headers_, | 94 request_url_ = GetUrlFromHeaderBlock(headers_, |
| 91 kDefaultSpdyMajorVersion, | 95 kDefaultSpdyMajorVersion, |
| 92 false); | 96 false); |
| 93 if (!request_url_.is_valid()) { | 97 if (!request_url_.is_valid()) { |
| 94 SendErrorResponse(); | 98 SendErrorResponse(); |
| 95 return; | 99 return; |
| (...skipping 22 matching lines...) Expand all Loading... | |
| 118 CloseConnection(QUIC_NO_ERROR); | 122 CloseConnection(QUIC_NO_ERROR); |
| 119 return; | 123 return; |
| 120 } | 124 } |
| 121 | 125 |
| 122 if (response->response_type() == QuicInMemoryCache::IGNORE_REQUEST) { | 126 if (response->response_type() == QuicInMemoryCache::IGNORE_REQUEST) { |
| 123 DVLOG(1) << "Special response: ignoring request."; | 127 DVLOG(1) << "Special response: ignoring request."; |
| 124 return; | 128 return; |
| 125 } | 129 } |
| 126 | 130 |
| 127 DVLOG(1) << "Sending response for stream " << id(); | 131 DVLOG(1) << "Sending response for stream " << id(); |
| 128 SendHeadersAndBody(response->headers(), response->body()); | 132 SendHeadersAndBody(*response->headers(), response->body()); |
|
wtc
2014/07/02 19:39:08
Hmm... If we are still dereferencing response->hea
dmz
2014/07/02 20:58:39
Alright, I've undone the change.
| |
| 129 } | 133 } |
| 130 | 134 |
| 131 void QuicSpdyServerStream::SendErrorResponse() { | 135 void QuicSpdyServerStream::SendErrorResponse() { |
| 132 DVLOG(1) << "Sending error response for stream " << id(); | 136 DVLOG(1) << "Sending error response for stream " << id(); |
| 133 scoped_refptr<HttpResponseHeaders> headers | 137 scoped_refptr<HttpResponseHeaders> headers |
| 134 = new HttpResponseHeaders(string("HTTP/1.1 500 Server Error") + '\0' + | 138 = new HttpResponseHeaders(string("HTTP/1.1 500 Server Error") + '\0' + |
| 135 "content-length: 3" + '\0' + '\0'); | 139 "content-length: 3" + '\0' + '\0'); |
| 136 SendHeadersAndBody(*headers, "bad"); | 140 SendHeadersAndBody(*headers, "bad"); |
| 137 } | 141 } |
| 138 | 142 |
| (...skipping 11 matching lines...) Expand all Loading... | |
| 150 &header_block); | 154 &header_block); |
| 151 | 155 |
| 152 WriteHeaders(header_block, body.empty(), NULL); | 156 WriteHeaders(header_block, body.empty(), NULL); |
| 153 | 157 |
| 154 if (!body.empty()) { | 158 if (!body.empty()) { |
| 155 WriteOrBufferData(body, true, NULL); | 159 WriteOrBufferData(body, true, NULL); |
| 156 } | 160 } |
| 157 } | 161 } |
| 158 | 162 |
| 159 } // namespace net | 163 } // namespace net |
| OLD | NEW |