1f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles)// Copyright 2014 The Chromium Authors. All rights reserved. 2f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles)// Use of this source code is governed by a BSD-style license that can be 3f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles)// found in the LICENSE file. 4f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) 5f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles)#include "net/quic/quic_spdy_server_stream.h" 6f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) 7f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles)#include "base/memory/singleton.h" 8116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch#include "base/strings/string_number_conversions.h" 9f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles)#include "net/quic/quic_in_memory_cache.h" 10f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles)#include "net/quic/quic_session.h" 11116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch#include "net/quic/spdy_utils.h" 12f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles)#include "net/spdy/spdy_framer.h" 13116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch#include "net/spdy/spdy_header_block.h" 14116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch#include "net/spdy/spdy_http_utils.h" 15f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) 16f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles)using base::StringPiece; 17f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles)using std::string; 18f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) 19f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles)namespace net { 20f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) 21f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles)static const size_t kHeaderBufInitialSize = 4096; 22f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) 23f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles)QuicSpdyServerStream::QuicSpdyServerStream(QuicStreamId id, 24f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) QuicSession* session) 25f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) : QuicDataStream(id, session), 26f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) read_buf_(new GrowableIOBuffer()), 27f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) request_headers_received_(false) { 28116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch read_buf_->SetCapacity(kHeaderBufInitialSize); 29f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles)} 30f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) 31f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles)QuicSpdyServerStream::~QuicSpdyServerStream() { 32f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles)} 33f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) 34f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles)uint32 QuicSpdyServerStream::ProcessData(const char* data, uint32 data_len) { 35116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch if (data_len > INT_MAX) { 36116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch LOG(DFATAL) << "Data length too long: " << data_len; 37116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch return 0; 38116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch } 39f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) // Are we still reading the request headers. 40f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) if (!request_headers_received_) { 41f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) // Grow the read buffer if necessary. 42116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch while (read_buf_->RemainingCapacity() < static_cast<int>(data_len)) { 43116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch read_buf_->SetCapacity(read_buf_->capacity() * 2); 44f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) } 45f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) memcpy(read_buf_->data(), data, data_len); 46f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) read_buf_->set_offset(read_buf_->offset() + data_len); 47116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch // Try parsing the request headers. This will set request_headers_received_ 48116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch // if successful; if not, it will be tried again with more data. 49f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) ParseRequestHeaders(); 50f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) } else { 51116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch body_.append(data, data_len); 52f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) } 53f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) return data_len; 54f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles)} 55f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) 56f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles)void QuicSpdyServerStream::OnFinRead() { 57f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) ReliableQuicStream::OnFinRead(); 58f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) if (write_side_closed() || fin_buffered()) { 59f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) return; 60f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) } 61f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) 62f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) if (!request_headers_received_) { 63f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) SendErrorResponse(); // We're not done reading headers. 64116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch return; 65116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch } 66116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch 67116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch SpdyHeaderBlock::const_iterator it = headers_.find("content-length"); 68116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch size_t content_length; 69116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch if (it != headers_.end() && 70116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch (!base::StringToSizeT(it->second, &content_length) || 71116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch body_.size() != content_length)) { 72f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) SendErrorResponse(); // Invalid content length 73116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch return; 74f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) } 75116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch 76116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch SendResponse(); 77f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles)} 78f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) 79116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch// Try parsing the request headers. If successful, sets 80116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch// request_headers_received_. If not successful, it can just be tried again once 81116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch// there's more data. 82116680a4aac90f2aa7413d9095a592090648e557Ben Murdochvoid QuicSpdyServerStream::ParseRequestHeaders() { 83116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch SpdyFramer framer((kDefaultSpdyMajorVersion)); 84116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch const char* data = read_buf_->StartOfBuffer(); 85f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) size_t read_buf_len = static_cast<size_t>(read_buf_->offset()); 86116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch size_t len = framer.ParseHeaderBlockInBuffer(data, read_buf_len, &headers_); 87f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) if (len == 0) { 88116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch // Not enough data yet, presumably. (If we still don't succeed by the end of 89116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch // the stream, then we'll error in OnFinRead().) 90116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch return; 91f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) } 92f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) 93116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch // Headers received and parsed: extract the request URL. 94116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch request_url_ = GetUrlFromHeaderBlock(headers_, 95116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch kDefaultSpdyMajorVersion, 96116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch false); 97116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch if (!request_url_.is_valid()) { 98f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) SendErrorResponse(); 99116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch return; 100f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) } 101f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) 102116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch // Add any data past the headers to the request body. 103f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) size_t delta = read_buf_len - len; 104f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) if (delta > 0) { 105f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) body_.append(data + len, delta); 106f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) } 107f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) 108f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) request_headers_received_ = true; 109f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles)} 110f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) 111f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles)void QuicSpdyServerStream::SendResponse() { 112f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) // Find response in cache. If not found, send error response. 113f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) const QuicInMemoryCache::Response* response = 114116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch QuicInMemoryCache::GetInstance()->GetResponse(request_url_); 115f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) if (response == NULL) { 116f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) SendErrorResponse(); 117f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) return; 118f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) } 119f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) 120f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) if (response->response_type() == QuicInMemoryCache::CLOSE_CONNECTION) { 121f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) DVLOG(1) << "Special response: closing connection."; 122f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) CloseConnection(QUIC_NO_ERROR); 123f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) return; 124f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) } 125f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) 126f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) if (response->response_type() == QuicInMemoryCache::IGNORE_REQUEST) { 127f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) DVLOG(1) << "Special response: ignoring request."; 128f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) return; 129f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) } 130f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) 131f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) DVLOG(1) << "Sending response for stream " << id(); 132f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) SendHeadersAndBody(response->headers(), response->body()); 133f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles)} 134f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) 135f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles)void QuicSpdyServerStream::SendErrorResponse() { 136f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) DVLOG(1) << "Sending error response for stream " << id(); 137116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch scoped_refptr<HttpResponseHeaders> headers 138116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch = new HttpResponseHeaders(string("HTTP/1.1 500 Server Error") + '\0' + 139116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch "content-length: 3" + '\0' + '\0'); 1401320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci SendHeadersAndBody(*headers.get(), "bad"); 141f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles)} 142f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) 143f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles)void QuicSpdyServerStream::SendHeadersAndBody( 144116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch const HttpResponseHeaders& response_headers, 145f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) StringPiece body) { 146f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) // We only support SPDY and HTTP, and neither handles bidirectional streaming. 147f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) if (!read_side_closed()) { 148f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) CloseReadSide(); 149f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) } 150f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) 151116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch SpdyHeaderBlock header_block; 152116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch CreateSpdyHeadersFromHttpResponse(response_headers, 153116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch kDefaultSpdyMajorVersion, 154116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch &header_block); 155f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) 156f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) WriteHeaders(header_block, body.empty(), NULL); 157f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) 158f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) if (!body.empty()) { 159f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) WriteOrBufferData(body, true, NULL); 160f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) } 161f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles)} 162f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) 163f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles)} // namespace net 164