diff options
author | Allan Sandfeld Jensen <allan.jensen@theqtcompany.com> | 2016-05-09 14:22:11 +0200 |
---|---|---|
committer | Allan Sandfeld Jensen <allan.jensen@qt.io> | 2016-05-09 15:11:45 +0000 |
commit | 2ddb2d3e14eef3de7dbd0cef553d669b9ac2361c (patch) | |
tree | e75f511546c5fd1a173e87c1f9fb11d7ac8d1af3 /chromium/net/tools/cachetool | |
parent | a4f3d46271c57e8155ba912df46a05559d14726e (diff) | |
download | qtwebengine-chromium-2ddb2d3e14eef3de7dbd0cef553d669b9ac2361c.tar.gz |
BASELINE: Update Chromium to 51.0.2704.41
Also adds in all smaller components by reversing logic for exclusion.
Change-Id: Ibf90b506e7da088ea2f65dcf23f2b0992c504422
Reviewed-by: Joerg Bornemann <joerg.bornemann@theqtcompany.com>
Diffstat (limited to 'chromium/net/tools/cachetool')
-rw-r--r-- | chromium/net/tools/cachetool/cachetool.cc | 203 |
1 files changed, 203 insertions, 0 deletions
diff --git a/chromium/net/tools/cachetool/cachetool.cc b/chromium/net/tools/cachetool/cachetool.cc new file mode 100644 index 00000000000..c186400d9da --- /dev/null +++ b/chromium/net/tools/cachetool/cachetool.cc @@ -0,0 +1,203 @@ +// Copyright 2016 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 <iostream> + +#include "base/at_exit.h" +#include "base/command_line.h" +#include "base/files/file_path.h" +#include "base/logging.h" +#include "base/memory/scoped_ptr.h" +#include "base/message_loop/message_loop.h" +#include "base/strings/string_number_conversions.h" +#include "net/base/io_buffer.h" +#include "net/base/test_completion_callback.h" +#include "net/disk_cache/disk_cache.h" +#include "net/http/http_cache.h" +#include "net/http/http_response_headers.h" +#include "net/http/http_util.h" + +using disk_cache::Backend; +using disk_cache::Entry; + +namespace { + +// Print all of a cache's keys to stdout. +bool ListKeys(Backend* cache_backend) { + scoped_ptr<Backend::Iterator> entry_iterator = + cache_backend->CreateIterator(); + Entry* entry = nullptr; + net::TestCompletionCallback cb; + int rv = entry_iterator->OpenNextEntry(&entry, cb.callback()); + while (cb.GetResult(rv) == net::OK) { + std::string url = entry->GetKey(); + std::cout << url << std::endl; + entry->Close(); + entry = nullptr; + rv = entry_iterator->OpenNextEntry(&entry, cb.callback()); + } + return true; +} + +// Print a key's stream to stdout. +bool GetKeyStream(Backend* cache_backend, const std::string& key, int index) { + if (index < 0 || index > 2) { + std::cerr << "Invalid stream index." << std::endl; + return false; + } + + Entry* cache_entry; + net::TestCompletionCallback cb; + int rv = cache_backend->OpenEntry(key, &cache_entry, cb.callback()); + if (cb.GetResult(rv) != net::OK) { + std::cerr << "Couldn't find key's entry." << std::endl; + return false; + } + + const int kInitBufferSize = 8192; + scoped_refptr<net::GrowableIOBuffer> buffer(new net::GrowableIOBuffer()); + buffer->SetCapacity(kInitBufferSize); + while (true) { + rv = cache_entry->ReadData(index, buffer->offset(), buffer.get(), + buffer->capacity() - buffer->offset(), + cb.callback()); + rv = cb.GetResult(rv); + if (rv < 0) { + cache_entry->Close(); + std::cerr << "Stream read error." << std::endl; + return false; + } + buffer->set_offset(buffer->offset() + rv); + if (rv == 0) + break; + buffer->SetCapacity(buffer->offset() * 2); + } + cache_entry->Close(); + if (index == 0) { + net::HttpResponseInfo response_info; + bool truncated_response_info = false; + net::HttpCache::ParseResponseInfo(buffer->StartOfBuffer(), buffer->offset(), + &response_info, &truncated_response_info); + if (truncated_response_info) { + std::cerr << "Truncated HTTP response." << std::endl; + return false; + } + std::cout << net::HttpUtil::ConvertHeadersBackToHTTPResponse( + response_info.headers->raw_headers()); + } else { + std::cout.write(buffer->StartOfBuffer(), buffer->offset()); + } + return true; +} + +// Delete a specified key from the cache. +bool DeleteKey(Backend* cache_backend, const std::string& key) { + net::TestCompletionCallback cb; + int rv = cache_backend->DoomEntry(key, cb.callback()); + if (cb.GetResult(rv) != net::OK) { + std::cerr << "Couldn't delete key." << std::endl; + return false; + } + return true; +} + +// Print the command line help. +void PrintHelp() { + std::cout << "cachetool <cache_path> <cache_backend_type> <subcommand> ..." + << std::endl + << std::endl; + std::cout << "Available cache backend types: simple, blockfile" << std::endl; + std::cout << "Available subcommands:" << std::endl; + std::cout << " validate: Verify that the cache can be opened and return, " + << "confirming the cache exists and is of the right type." + << std::endl; + std::cout << " list_keys: List all keys in the cache." << std::endl; + std::cout << " get_stream <key> <index>: Print a particular stream for a" + << " given key." << std::endl; + std::cout << " delete_key <key>: Delete key from cache." << std::endl; + std::cout << "Expected values of <index> are:" << std::endl; + std::cout << " 0 (HTTP response headers)" << std::endl; + std::cout << " 1 (transport encoded content)" << std::endl; + std::cout << " 2 (compiled content)" << std::endl; +} + +} // namespace + +int main(int argc, char* argv[]) { + base::AtExitManager at_exit_manager; + base::MessageLoopForIO message_loop; + base::CommandLine::Init(argc, argv); + const base::CommandLine& command_line = + *base::CommandLine::ForCurrentProcess(); + + base::CommandLine::StringVector args = command_line.GetArgs(); + if (args.size() < 3U) { + PrintHelp(); + return 1; + } + + base::FilePath cache_path(args[0]); + std::string cache_backend_type(args[1]); + std::string subcommand(args[2]); + + net::BackendType backend_type; + if (cache_backend_type == "simple") { + backend_type = net::CACHE_BACKEND_SIMPLE; + } else if (cache_backend_type == "blockfile") { + backend_type = net::CACHE_BACKEND_BLOCKFILE; + } else { + std::cerr << "Unknown cache type." << std::endl; + PrintHelp(); + return 1; + } + + scoped_ptr<Backend> cache_backend; + net::TestCompletionCallback cb; + int rv = disk_cache::CreateCacheBackend( + net::DISK_CACHE, backend_type, cache_path, INT_MAX, false, + message_loop.task_runner(), nullptr, &cache_backend, cb.callback()); + if (cb.GetResult(rv) != net::OK) { + std::cerr << "Invalid cache." << std::endl; + return 1; + } + + bool successful_command = false; + if (subcommand == "validate") { + if (args.size() != 3) { + PrintHelp(); + return 1; + } + successful_command = true; + } else if (subcommand == "list_keys") { + if (args.size() != 3) { + PrintHelp(); + return 1; + } + successful_command = ListKeys(cache_backend.get()); + } else if (subcommand == "get_stream") { + if (args.size() != 5) { + PrintHelp(); + return 1; + } + std::string key(args[3]); + int index = 0; + if (!base::StringToInt(args[4], &index)) { + std::cerr << "<index> must be an integer." << std::endl; + PrintHelp(); + return 1; + } + successful_command = GetKeyStream(cache_backend.get(), key, index); + } else if (subcommand == "delete_key") { + if (args.size() != 4) { + PrintHelp(); + return 1; + } + std::string key(args[3]); + successful_command = DeleteKey(cache_backend.get(), key); + } else { + std::cerr << "Unknown subcommand." << std::endl; + PrintHelp(); + } + return !successful_command; +} |