net/quic/quic_in_memory_cache.h

   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
   3 // found in the LICENSE file.
   4
   5 #ifndef NET_QUIC_QUIC_IN_MEMORY_CACHE_H_
   6 #define NET_QUIC_QUIC_IN_MEMORY_CACHE_H_
   7
   8 #include <string>
   9
  10 #include "base/containers/hash_tables.h"
  11 #include "base/files/file_util.h"
  12 #include "base/memory/singleton.h"
  13 #include "base/strings/string_piece.h"
  14 #include "net/http/http_response_headers.h"
  15
  16 template <typename T> struct DefaultSingletonTraits;
  17 class GURL;
  18
  19 namespace net {
  20
  21 extern base::FilePath::StringType g_quic_in_memory_cache_dir;
  22
  23 namespace test {
  24 class QuicInMemoryCachePeer;
  25 }  // namespace
  26
  27 class QuicServer;
  28
  29 // In-memory cache for HTTP responses.
  30 // Reads from disk cache generated by:
  31 // `wget -p --save_headers <url>`
  32 class QuicInMemoryCache {
  33  public:
  34   enum SpecialResponseType {
  35     REGULAR_RESPONSE,  // Send the headers and body like a server should.
  36     CLOSE_CONNECTION,  // Close the connection (sending the close packet).
  37     IGNORE_REQUEST,    // Do nothing, expect the client to time out.
  38   };
  39
  40   // Container for response header/body pairs.
  41   class Response {
  42    public:
  43     Response();
  44     ~Response();
  45
  46     SpecialResponseType response_type() const { return response_type_; }
  47     const HttpResponseHeaders& headers() const { return *headers_.get(); }
  48     const base::StringPiece body() const { return base::StringPiece(body_); }
  49
  50    private:
  51     friend class QuicInMemoryCache;
  52
  53     void set_headers(scoped_refptr<HttpResponseHeaders> headers) {
  54       headers_ = headers;
  55     }
  56     void set_body(base::StringPiece body) {
  57       body.CopyToString(&body_);
  58     }
  59
  60     SpecialResponseType response_type_;
  61     scoped_refptr<HttpResponseHeaders> headers_;
  62     std::string body_;
  63
  64     DISALLOW_COPY_AND_ASSIGN(Response);
  65   };
  66
  67   // Returns the singleton instance of the cache.
  68   static QuicInMemoryCache* GetInstance();
  69
  70   // Retrieve a response from this cache for a given request.
  71   // If no appropriate response exists, nullptr is returned.
  72   // Currently, responses are selected based on request URI only.
  73   const Response* GetResponse(const GURL& url) const;
  74
  75   // Adds a simple response to the cache.  The response headers will
  76   // only contain the "content-length" header with the length of |body|.
  77   void AddSimpleResponse(base::StringPiece path,
  78                          base::StringPiece version,
  79                          base::StringPiece response_code,
  80                          base::StringPiece response_detail,
  81                          base::StringPiece body);
  82
  83   // Add a response to the cache.
  84   void AddResponse(const GURL& url,
  85                    scoped_refptr<HttpResponseHeaders> response_headers,
  86                    base::StringPiece response_body);
  87
  88   // Simulate a special behavior at a particular path.
  89   void AddSpecialResponse(base::StringPiece path,
  90                           SpecialResponseType response_type);
  91
  92  private:
  93   typedef base::hash_map<std::string, Response*> ResponseMap;
  94   friend struct DefaultSingletonTraits<QuicInMemoryCache>;
  95   friend class test::QuicInMemoryCachePeer;
  96
  97   QuicInMemoryCache();
  98   ~QuicInMemoryCache();
  99
 100   void ResetForTests();
 101
 102   void Initialize();
 103
 104   std::string GetKey(const GURL& url) const;
 105
 106   // Cached responses.
 107   ResponseMap responses_;
 108
 109   DISALLOW_COPY_AND_ASSIGN(QuicInMemoryCache);
 110 };
 111
 112 }  // namespace net
 113
 114 #endif  // NET_QUIC_QUIC_IN_MEMORY_CACHE_H_