Landing Recent QUIC changes until 12/18/2015 13:57 UTC

net/tools/quic/quic_simple_server_stream.cc

 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "net/tools/quic/quic_simple_server_stream.h"
 
 #include "base/logging.h"
 #include "base/stl_util.h"
 #include "base/strings/string_number_conversions.h"
 #include "base/strings/string_piece.h"
@@ skipping 78 matching lines @@
       content_length_ != static_cast<int>(body_.size())) {
     DVLOG(1) << "Content length (" << content_length_ << ") != body size ("
              << body_.size() << ").";
     SendErrorResponse();
     return;
   }
 
   SendResponse();
 }
 
+void QuicSimpleServerStream::PushResponse(
+    SpdyHeaderBlock push_request_headers) {
+  if (id() % 2 != 0) {
+    LOG(DFATAL) << "Client initiated stream shouldn't be used "
+                << "as promised stream.";
+    return;
+  }
+  // Change the stream state to emulate a client request.
+  request_headers_ = push_request_headers;
+  content_length_ = 0;
+  DVLOG(1) << "Stream " << id() << ": Ready to receive server push response.";
+
+  // Set as if stream decompresed the headers and received fin.
+  QuicSpdyStream::OnInitialHeadersComplete(/*fin=*/true, 0);
+}
+
 void QuicSimpleServerStream::SendResponse() {
   if (!ContainsKey(request_headers_, ":authority") ||
       !ContainsKey(request_headers_, ":path")) {
     DVLOG(1) << "Request headers do not contain :authority or :path.";
     SendErrorResponse();
     return;
   }
 
   // Find response in cache. If not found, send error response.
   const QuicInMemoryCache::Response* response =
       QuicInMemoryCache::GetInstance()->GetResponse(
           request_headers_[":authority"], request_headers_[":path"]);
   if (response == nullptr) {
     DVLOG(1) << "Response not found in cache.";
     SendErrorResponse();
     return;
   }
 
   if (response->response_type() == QuicInMemoryCache::CLOSE_CONNECTION) {
     DVLOG(1) << "Special response: closing connection.";
-    CloseConnection(QUIC_NO_ERROR);
+    CloseConnectionWithDetails(QUIC_NO_ERROR, "Toy server forcing close");
     return;
   }
 
   if (response->response_type() == QuicInMemoryCache::IGNORE_REQUEST) {
     DVLOG(1) << "Special response: ignoring request.";
     return;
   }
 
   // Examing response status, if it was not pure integer as typical h2 response
   // status, send error response.
@@ skipping 74 matching lines @@
   // Send the trailers. A FIN is always sent with trailers.
   DVLOG(1) << "Writing trailers (fin = true): "
            << response_trailers.DebugString();
   WriteTrailers(response_trailers, nullptr);
 }
 
 const char* const QuicSimpleServerStream::kErrorResponseBody = "bad";
 
 }  // namespace tools
 }  // namespace net