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