1 files changed, 82 insertions, 0 deletions
diff --git a/net/tools/dump_cache/cache_dumper.h b/net/tools/dump_cache/cache_dumper.h
new file mode 100644
index 0000000..8a11269
--- /dev/null
+++ b/net/tools/dump_cache/cache_dumper.h
@@ -0,0 +1,82 @@
+// Copyright (c) 2009 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.
+
+#ifndef NET_TOOLS_DUMP_CACHE_DUMPER_H_
+#define NET_TOOLS_DUMP_CACHE_DUMPER_H_
+
+#include <string>
+#include "base/file_path.h"
+#include "base/file_util.h"
+#include "net/disk_cache/backend_impl.h"
+
+#ifdef WIN32
+// Dumping the cache often creates very large filenames, which are tricky
+// on windows.  Most API calls don't support large filenames, including
+// most of the base library functions.  Unfortunately, adding "\\?\" into
+// the filename support is tricky.  Instead, if WIN32_LARGE_FILENAME_SUPPORT
+// is set, we use direct WIN32 APIs for manipulating the files.
+#define WIN32_LARGE_FILENAME_SUPPORT
+#endif
+
+// An abstract class for writing cache dump data.
+class CacheDumpWriter {
+ public:
+  // Creates an entry to be written.
+  // On success, populates the |entry|.
+  // Returns true on success, false otherwise.
+  virtual bool CreateEntry(const std::string& key,
+                           disk_cache::Entry** entry) = 0;
+
+  // Write to the current entry.
+  // Returns true on success, false otherwise.
+  virtual bool WriteEntry(disk_cache::Entry* entry, int stream, int offset,
+                          net::IOBuffer* buf, int buf_len) = 0;
+
+  // Close the current entry.
+  virtual void CloseEntry(disk_cache::Entry* entry, base::Time last_used,
+                          base::Time last_modified) = 0;
+};
+
+// Writes data to a cache.
+class CacheDumper : public CacheDumpWriter {
+ public:
+  CacheDumper(disk_cache::BackendImpl* cache) : cache_(cache) {};
+
+  virtual bool CreateEntry(const std::string& key, disk_cache::Entry** entry);
+  virtual bool WriteEntry(disk_cache::Entry* entry, int stream, int offset,
+                          net::IOBuffer* buf, int buf_len);
+  virtual void CloseEntry(disk_cache::Entry* entry, base::Time last_used,
+                          base::Time last_modified);
+
+ private:
+  disk_cache::BackendImpl* cache_;
+};
+
+// Writes data to a disk.
+class DiskDumper : public CacheDumpWriter {
+ public:
+  DiskDumper(const std::wstring& path) : path_(path), entry_(NULL) {
+    file_util::CreateDirectory(path);
+  };
+  virtual bool CreateEntry(const std::string& key, disk_cache::Entry** entry);
+  virtual bool WriteEntry(disk_cache::Entry* entry, int stream, int offset,
+                          net::IOBuffer* buf, int buf_len);
+  virtual void CloseEntry(disk_cache::Entry* entry, base::Time last_used,
+                          base::Time last_modified);
+
+ private:
+  std::wstring path_;
+  // This is a bit of a hack.  As we get a CreateEntry, we coin the current
+  // entry_path_ where we write that entry to disk.  Subsequent calls to
+  // WriteEntry() utilize this path for writing to disk.
+  FilePath entry_path_;
+  std::string entry_url_;
+#ifdef WIN32_LARGE_FILENAME_SUPPORT
+  HANDLE entry_;
+#else
+  FILE* entry_;
+#endif
+};
+
+#endif  // NET_TOOLS_DUMP_CACHE_DUMPER_H_