Add deletable file refs to Blobs

webkit/blob/blob_storage_controller.cc

 // Copyright (c) 2010 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 "webkit/blob/blob_storage_controller.h"
 
 #include "base/logging.h"
 #include "googleurl/src/gurl.h"
 #include "net/base/upload_data.h"
 #include "webkit/blob/blob_data.h"
 
 namespace webkit_blob {
 
 BlobStorageController::BlobStorageController() {
 }
 
 BlobStorageController::~BlobStorageController() {
 }
 
-void BlobStorageController::AppendStorageItems(
-    BlobData* target_blob_data, BlobData* src_blob_data,
-    uint64 offset, uint64 length) {
-  DCHECK(target_blob_data && src_blob_data &&
-         length != static_cast<uint64>(-1));
-
-  std::vector<BlobData::Item>::const_iterator iter =
-      src_blob_data->items().begin();
-  if (offset) {
-    for (; iter != src_blob_data->items().end(); ++iter) {
-      if (offset >= iter->length())
-        offset -= iter->length();
-      else
-        break;
-    }
-  }
-
-  for (; iter != src_blob_data->items().end() && length > 0; ++iter) {
-    uint64 current_length = iter->length() - offset;
-    uint64 new_length = current_length > length ? length : current_length;
-    if (iter->type() == BlobData::TYPE_DATA) {
-      target_blob_data->AppendData(iter->data(),
-                                   static_cast<uint32>(iter->offset() + offset),
-                                   static_cast<uint32>(new_length));
-    } else {
-      DCHECK(iter->type() == BlobData::TYPE_FILE);
-      target_blob_data->AppendFile(iter->file_path(),
-                                   iter->offset() + offset,
-                                   new_length,
-                                   iter->expected_modification_time());
-    }
-    length -= new_length;
-    offset = 0;
-  }
-}
-
 void BlobStorageController::RegisterBlobUrl(
     const GURL& url, const BlobData* blob_data) {
   scoped_refptr<BlobData> target_blob_data = new BlobData();
   target_blob_data->set_content_type(blob_data->content_type());
   target_blob_data->set_content_disposition(blob_data->content_disposition());
 
   // The blob data is stored in the "canonical" way. That is, it only contains a
   // list of Data and File items.
   // 1) The Data item is denoted by the raw data and the range.
   // 2) The File item is denoted by the file path, the range and the expected
   //    modification time.
   // All the Blob items in the passing blob data are resolved and expanded into
   // a set of Data and File items.
 
   for (std::vector<BlobData::Item>::const_iterator iter =
            blob_data->items().begin();
        iter != blob_data->items().end(); ++iter) {
     switch (iter->type()) {
       case BlobData::TYPE_DATA: {
         // WebBlobData does not allow partial data.
         DCHECK(!(iter->offset()) && iter->length() == iter->data().size());
         target_blob_data->AppendData(iter->data());
         break;
       }
       case BlobData::TYPE_FILE:
-        target_blob_data->AppendFile(iter->file_path(),
-                                     iter->offset(),
-                                     iter->length(),
-                                     iter->expected_modification_time());
+        AppendFileItem(target_blob_data,
+                       iter->file_path(),
+                       iter->offset(),
+                       iter->length(),
+                       iter->expected_modification_time());
         break;
       case BlobData::TYPE_BLOB: {
         BlobData* src_blob_data = GetBlobDataFromUrl(iter->blob_url());
         DCHECK(src_blob_data);
         if (src_blob_data)
           AppendStorageItems(target_blob_data.get(),
                              src_blob_data,
                              iter->offset(),
                              iter->length());
         break;
@@ skipping 77 matching lines @@
               item.expected_modification_time());
           break;
         default:
           NOTREACHED();
           break;
       }
     }
   }
 }
 
+void BlobStorageController::AppendStorageItems(
+    BlobData* target_blob_data, BlobData* src_blob_data,
+    uint64 offset, uint64 length) {
+  DCHECK(target_blob_data && src_blob_data &&
+         length != static_cast<uint64>(-1));
+
+  std::vector<BlobData::Item>::const_iterator iter =
+      src_blob_data->items().begin();
+  if (offset) {
+    for (; iter != src_blob_data->items().end(); ++iter) {
+      if (offset >= iter->length())
+        offset -= iter->length();
+      else
+        break;
+    }
+  }
+
+  for (; iter != src_blob_data->items().end() && length > 0; ++iter) {
+    uint64 current_length = iter->length() - offset;
+    uint64 new_length = current_length > length ? length : current_length;
+    if (iter->type() == BlobData::TYPE_DATA) {
+      target_blob_data->AppendData(iter->data(),
+                                   static_cast<uint32>(iter->offset() + offset),
+                                   static_cast<uint32>(new_length));
+    } else {
+      DCHECK(iter->type() == BlobData::TYPE_FILE);
+      AppendFileItem(target_blob_data,
+                     iter->file_path(),
+                     iter->offset() + offset,
+                     new_length,
+                     iter->expected_modification_time());
+    }
+    length -= new_length;
+    offset = 0;
+  }
+}
+
+void BlobStorageController::AppendFileItem(
+    BlobData* target_blob_data,
+    const FilePath& file_path, uint64 offset, uint64 length,
+    const base::Time& expected_modification_time) {
+  target_blob_data->AppendFile(file_path, offset, length,
+                               expected_modification_time);
+
+  // It may be a temporary file that should be deleted when no longer needed.
+  scoped_refptr<DeletableFileReference> deletable_file =
+      DeletableFileReference::Get(file_path);
+  if (deletable_file)
+    target_blob_data->AttachDeletableFileReference(deletable_file);
+}
+
 }  // namespace webkit_blob