Add a way to be notified when a deletable file reference is destroyed so that

callers can clean up any other state associated with it.

I used this capability to revoke the renderer's permissions for a downloaded
file. The old code path where it would revoke the permissions as soon as the
download was complete (as indicated by the renderer) doesn't account for the
case where the file is stored in a blob and that file reference outlives the
resource request.

BUG=81319
TEST=PPAPI.URLLoader
Review URL: http://codereview.chromium.org/6930042

git-svn-id: svn://svn.chromium.org/chrome/trunk/src@84500 0039d316-1c4b-4281-b951-d872f2087c98

2 files changed, 18 insertions, 1 deletions

diff --git a/webkit/blob/deletable_file_reference.cc b/webkit/blob/deletable_file_reference.cc
index 3f2daed..a47013d 100644
--- a/webkit/blob/deletable_file_reference.cc
+++ b/webkit/blob/deletable_file_reference.cc
@@ -1,4 +1,4 @@
-// Copyright (c) 2010 The Chromium Authors. All rights reserved.
+// Copyright (c) 2011 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.
 
@@ -55,6 +55,11 @@ scoped_refptr<DeletableFileReference> DeletableFileReference::GetOrCreate(
   return reference;
 }
 
+void DeletableFileReference::AddDeletionCallback(
+    const DeletionCallback& callback) {
+  deletion_callbacks_.push_back(callback);
+}
+
 DeletableFileReference::DeletableFileReference(
     const FilePath& path, base::MessageLoopProxy* file_thread)
     : path_(path), file_thread_(file_thread) {
@@ -62,6 +67,9 @@ DeletableFileReference::DeletableFileReference(
 }
 
 DeletableFileReference::~DeletableFileReference() {
+  for (size_t i = 0; i < deletion_callbacks_.size(); i++)
+    deletion_callbacks_[i].Run(path_);
+
   DCHECK(g_deletable_file_map.Get().find(path_)->second == this);
   g_deletable_file_map.Get().erase(path_);
   base::FileUtilProxy::Delete(file_thread_, path_, false /* recursive */, NULL);
diff --git a/webkit/blob/deletable_file_reference.h b/webkit/blob/deletable_file_reference.h
index b432bf0c..a4301bf 100644
--- a/webkit/blob/deletable_file_reference.h
+++ b/webkit/blob/deletable_file_reference.h
@@ -6,6 +6,9 @@
 #define WEBKIT_BLOB_DELETABLE_FILE_REFERENCE_H_
 #pragma once
 
+#include <vector>
+
+#include "base/callback.h"
 #include "base/file_path.h"
 #include "base/memory/ref_counted.h"
 
@@ -19,6 +22,8 @@ namespace webkit_blob {
 // to be deleted upon final release.
 class DeletableFileReference : public base::RefCounted<DeletableFileReference> {
  public:
+  typedef base::Callback<void(const FilePath&)> DeletionCallback;
+
   // Returns a DeletableFileReference for the given path, if no reference
   // for this path exists returns NULL.
   static scoped_refptr<DeletableFileReference> Get(const FilePath& path);
@@ -31,6 +36,8 @@ class DeletableFileReference : public base::RefCounted<DeletableFileReference> {
   // The full file path.
   const FilePath& path() const { return path_; }
 
+  void AddDeletionCallback(const DeletionCallback& callback);
+
  private:
   friend class base::RefCounted<DeletableFileReference>;
 
@@ -41,6 +48,8 @@ class DeletableFileReference : public base::RefCounted<DeletableFileReference> {
   const FilePath path_;
   scoped_refptr<base::MessageLoopProxy> file_thread_;
 
+  std::vector<DeletionCallback> deletion_callbacks_;
+
   DISALLOW_COPY_AND_ASSIGN(DeletableFileReference);
 };