Change the QuicInMemoryCache to store a SpdyHeaderBlock instead of BalsaHeaders.

net/tools/quic/quic_in_memory_cache.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_in_memory_cache.h"
 
 #include "base/files/file_enumerator.h"
 #include "base/files/file_util.h"
 #include "base/stl_util.h"
 #include "base/strings/string_number_conversions.h"
+#include "net/tools/balsa/balsa_frame.h"
 #include "net/tools/balsa/balsa_headers.h"
+#include "net/tools/balsa/noop_balsa_visitor.h"
+#include "net/tools/quic/spdy_utils.h"
 
 using base::FilePath;
 using base::StringPiece;
 using std::string;
 
 namespace net {
 namespace tools {
 
 // Specifies the directory used during QuicInMemoryCache
 // construction to seed the cache. Cache directory can be
@@ skipping 23 matching lines @@
   bool done_framing() const { return done_framing_; }
   const string& body() const { return body_; }
 
  private:
   bool done_framing_;
   string body_;
 };
 
 }  // namespace
 
+QuicInMemoryCache::Response::Response() : response_type_(REGULAR_RESPONSE) {}
+
+QuicInMemoryCache::Response::~Response() {}
+
 // static
 QuicInMemoryCache* QuicInMemoryCache::GetInstance() {
   return Singleton<QuicInMemoryCache>::get();
 }
 
 const QuicInMemoryCache::Response* QuicInMemoryCache::GetResponse(
     StringPiece host,
     StringPiece path) const {
   ResponseMap::const_iterator it = responses_.find(GetKey(host, path));
   if (it == responses_.end()) {
     return nullptr;
   }
   return it->second;
 }
 
 void QuicInMemoryCache::AddSimpleResponse(StringPiece host,
                                           StringPiece path,
                                           int response_code,
                                           StringPiece response_detail,
                                           StringPiece body) {
-  BalsaHeaders response_headers;
-  response_headers.SetRequestFirstlineFromStringPieces(
-      "HTTP/1.1", base::IntToString(response_code), response_detail);
-  response_headers.AppendHeader("content-length",
-                                base::IntToString(body.length()));
-
+  SpdyHeaderBlock response_headers;
+  response_headers[":version"] = "HTTP/1.1";
+  string status = base::IntToString(response_code) + " ";
+  response_detail.AppendToString(&status);
+  response_headers[":status"] = status;
+  response_headers["content-length"] = base::IntToString(body.length());
   AddResponse(host, path, response_headers, body);
 }
 
 void QuicInMemoryCache::AddResponse(StringPiece host,
                                     StringPiece path,
-                                    const BalsaHeaders& response_headers,
+                                    const SpdyHeaderBlock& response_headers,
                                     StringPiece response_body) {
   AddResponseImpl(host, path, REGULAR_RESPONSE, response_headers,
                   response_body);
 }
 
 void QuicInMemoryCache::AddSpecialResponse(StringPiece host,
                                            StringPiece path,
                                            SpecialResponseType response_type) {
-  AddResponseImpl(host, path, response_type, BalsaHeaders(), "");
+  AddResponseImpl(host, path, response_type, SpdyHeaderBlock(), "");
 }
 
 QuicInMemoryCache::QuicInMemoryCache() {
   Initialize();
 }
 
 void QuicInMemoryCache::ResetForTests() {
   STLDeleteValues(&responses_);
   Initialize();
 }
@@ skipping 62 matching lines @@
         base.remove_prefix(7);
       }
     }
     int path_start = base.find_first_of('/');
     DCHECK_LT(0, path_start);
     StringPiece host(base.substr(0, path_start));
     StringPiece path(base.substr(path_start));
     if (path[path.length() - 1] == ',') {
       path.remove_suffix(1);
     }
-    AddResponse(host, path, response_headers, caching_visitor.body());
+    AddResponse(host, path,
+                SpdyUtils::ResponseHeadersToSpdyHeaders(response_headers),
+                caching_visitor.body());
   }
 }
 
 QuicInMemoryCache::~QuicInMemoryCache() {
   STLDeleteValues(&responses_);
 }
 
 void QuicInMemoryCache::AddResponseImpl(
     StringPiece host,
     StringPiece path,
     SpecialResponseType response_type,
-    const BalsaHeaders& response_headers,
+    const SpdyHeaderBlock& response_headers,
     StringPiece response_body) {
   string key = GetKey(host, path);
   VLOG(1) << "Adding response for: " << key;
   if (ContainsKey(responses_, key)) {
     LOG(DFATAL) << "Response for '" << key << "' already exists!";
     return;
   }
   Response* new_response = new Response();
   new_response->set_response_type(response_type);
   new_response->set_headers(response_headers);
   new_response->set_body(response_body);
   responses_[key] = new_response;
 }
 
 string QuicInMemoryCache::GetKey(StringPiece host, StringPiece path) const {
   return host.as_string() + path.as_string();
 }
 
 }  // namespace tools
 }  // namespace net