| OLD | NEW |
| 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 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 | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "net/tools/quic/quic_in_memory_cache.h" | 5 #include "net/tools/quic/quic_in_memory_cache.h" |
| 6 | 6 |
| 7 #include "base/file_util.h" | 7 #include "base/file_util.h" |
| 8 #include "base/files/file_enumerator.h" |
| 8 #include "base/stl_util.h" | 9 #include "base/stl_util.h" |
| 9 | 10 |
| 10 using base::FilePath; | 11 using base::FilePath; |
| 11 using base::StringPiece; | 12 using base::StringPiece; |
| 12 using file_util::FileEnumerator; | |
| 13 using std::string; | 13 using std::string; |
| 14 | 14 |
| 15 // Specifies the directory used during QuicInMemoryCache | 15 // Specifies the directory used during QuicInMemoryCache |
| 16 // construction to seed the cache. Cache directory can be | 16 // construction to seed the cache. Cache directory can be |
| 17 // generated using `wget -p --save-headers <url> | 17 // generated using `wget -p --save-headers <url> |
| 18 | 18 |
| 19 namespace net { | 19 namespace net { |
| 20 namespace tools { | 20 namespace tools { |
| 21 | 21 |
| 22 std::string FLAGS_quic_in_memory_cache_dir = "/tmp/quic-data"; | 22 std::string FLAGS_quic_in_memory_cache_dir = "/tmp/quic-data"; |
| (...skipping 90 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 113 QuicInMemoryCache::QuicInMemoryCache() { | 113 QuicInMemoryCache::QuicInMemoryCache() { |
| 114 // If there's no defined cache dir, we have no initialization to do. | 114 // If there's no defined cache dir, we have no initialization to do. |
| 115 if (FLAGS_quic_in_memory_cache_dir.empty()) { | 115 if (FLAGS_quic_in_memory_cache_dir.empty()) { |
| 116 LOG(WARNING) << "No cache directory found. Skipping initialization."; | 116 LOG(WARNING) << "No cache directory found. Skipping initialization."; |
| 117 return; | 117 return; |
| 118 } | 118 } |
| 119 LOG(INFO) << "Attempting to initialize QuicInMemoryCache from directory: " | 119 LOG(INFO) << "Attempting to initialize QuicInMemoryCache from directory: " |
| 120 << FLAGS_quic_in_memory_cache_dir; | 120 << FLAGS_quic_in_memory_cache_dir; |
| 121 | 121 |
| 122 FilePath directory(FLAGS_quic_in_memory_cache_dir); | 122 FilePath directory(FLAGS_quic_in_memory_cache_dir); |
| 123 FileEnumerator file_list(directory, | 123 base::FileEnumerator file_list(directory, |
| 124 true, | 124 true, |
| 125 FileEnumerator::FILES); | 125 base::FileEnumerator::FILES); |
| 126 | 126 |
| 127 FilePath file = file_list.Next(); | 127 FilePath file = file_list.Next(); |
| 128 while (!file.empty()) { | 128 while (!file.empty()) { |
| 129 BalsaHeaders request_headers, response_headers; | 129 BalsaHeaders request_headers, response_headers; |
| 130 | 130 |
| 131 string file_contents; | 131 string file_contents; |
| 132 file_util::ReadFileToString(file, &file_contents); | 132 file_util::ReadFileToString(file, &file_contents); |
| 133 | 133 |
| 134 // Frame HTTP. | 134 // Frame HTTP. |
| 135 CachingBalsaVisitor caching_visitor; | 135 CachingBalsaVisitor caching_visitor; |
| (...skipping 61 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 197 STLDeleteValues(&responses_); | 197 STLDeleteValues(&responses_); |
| 198 } | 198 } |
| 199 | 199 |
| 200 string QuicInMemoryCache::GetKey(const BalsaHeaders& request_headers) const { | 200 string QuicInMemoryCache::GetKey(const BalsaHeaders& request_headers) const { |
| 201 return request_headers.GetHeader("host").as_string() + | 201 return request_headers.GetHeader("host").as_string() + |
| 202 request_headers.request_uri().as_string(); | 202 request_headers.request_uri().as_string(); |
| 203 } | 203 } |
| 204 | 204 |
| 205 } // namespace tools | 205 } // namespace tools |
| 206 } // namespace net | 206 } // namespace net |
| OLD | NEW |