Delete enumerations at Simple Cache backend deletion.

net/disk_cache/simple/simple_backend_impl.cc

 // Copyright (c) 2013 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 "net/disk_cache/simple/simple_backend_impl.h"
 
 #include <algorithm>
 #include <cstdlib>
 #include <functional>
 
@@ skipping 465 matching lines @@
 int SimpleBackendImpl::OpenNextEntry(void** iter,
                                      Entry** next_entry,
                                      const CompletionCallback& callback) {
   CompletionCallback get_next_entry =
       base::Bind(&SimpleBackendImpl::GetNextEntryInIterator, AsWeakPtr(), iter,
                  next_entry, callback);
   return index_->ExecuteWhenReady(get_next_entry);
 }
 
 void SimpleBackendImpl::EndEnumeration(void** iter) {
-  SimpleIndex::HashList* entry_list =
-      static_cast<SimpleIndex::HashList*>(*iter);
-  delete entry_list;
+  active_enumerations_.Remove(IteratorToEnumerationId(iter));
   *iter = NULL;
 }
 
 void SimpleBackendImpl::GetStats(
     std::vector<std::pair<std::string, std::string> >* stats) {
   std::pair<std::string, std::string> item;
   item.first = "Cache type";
   item.second = "Simple Cache";
   stats->push_back(item);
 }
 
 void SimpleBackendImpl::OnExternalCacheHit(const std::string& key) {
   index_->UseIfExists(simple_util::GetEntryHashKey(key));
 }
 
+// static
+SimpleBackendImpl::ActiveEnumerationMap::KeyType
+    SimpleBackendImpl::IteratorToEnumerationId(void** iter) {
+  COMPILE_ASSERT(sizeof(ptrdiff_t) >= sizeof(*iter),
+                 integer_type_must_fit_ptr_type_for_cast_to_be_reversible);
+  const ptrdiff_t ptrdiff_enumeration_id = reinterpret_cast<ptrdiff_t>(*iter);
+  const ActiveEnumerationMap::KeyType enumeration_id = ptrdiff_enumeration_id;
+  DCHECK_EQ(enumeration_id, ptrdiff_enumeration_id);
+  return enumeration_id;
+}
+
+// static
+void* SimpleBackendImpl::EnumerationIdToIterator(
+    ActiveEnumerationMap::KeyType enumeration_id) {
+  const ptrdiff_t ptrdiff_enumeration_id = enumeration_id;
+  DCHECK_EQ(enumeration_id, ptrdiff_enumeration_id);
+  COMPILE_ASSERT(sizeof(ptrdiff_t) >= sizeof(void*),
+                 integer_type_must_fit_ptr_type_for_cast_to_be_reversible);
+  return reinterpret_cast<void*>(ptrdiff_enumeration_id);
+}
+
 void SimpleBackendImpl::InitializeIndex(const CompletionCallback& callback,
                                         const DiskStatResult& result) {
   if (result.net_error == net::OK) {
     index_->SetMaxSize(result.max_size);
     index_->Initialize(result.cache_dir_mtime);
   }
   callback.Run(result.net_error);
 }
 
 SimpleBackendImpl::DiskStatResult SimpleBackendImpl::InitCacheStructureOnDisk(
@@ skipping 102 matching lines @@
 
 void SimpleBackendImpl::GetNextEntryInIterator(
     void** iter,
     Entry** next_entry,
     const CompletionCallback& callback,
     int error_code) {
   if (error_code != net::OK) {
     callback.Run(error_code);
     return;
   }
+  std::vector<uint64>* entry_list = NULL;
   if (*iter == NULL) {
-    *iter = index()->GetAllHashes().release();
+    const ActiveEnumerationMap::KeyType new_enumeration_id =
+        active_enumerations_.Add(
+            entry_list = index()->GetAllHashes().release());
+    *iter = EnumerationIdToIterator(new_enumeration_id);
+  } else {
+    entry_list = active_enumerations_.Lookup(IteratorToEnumerationId(iter));
   }
-  SimpleIndex::HashList* entry_list =
-      static_cast<SimpleIndex::HashList*>(*iter);
   while (entry_list->size() > 0) {
     uint64 entry_hash = entry_list->back();
     entry_list->pop_back();
     if (index()->Has(entry_hash)) {
       *next_entry = NULL;
       CompletionCallback continue_iteration = base::Bind(
           &SimpleBackendImpl::CheckIterationReturnValue,
           AsWeakPtr(),
           iter,
           next_entry,
@@ skipping 87 matching lines @@
                    this));
   callback.Run(result);
 }
 
 void SimpleBackendImpl::FlushWorkerPoolForTesting() {
   if (g_sequenced_worker_pool)
     g_sequenced_worker_pool->FlushForTesting();
 }
 
 }  // namespace disk_cache