| 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 | 
|---|