Add Stream support to WebBlobRegistry and FileAPIMessageFilter.

content/child/webblobregistry_impl.cc

 // Copyright 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 "content/child/webblobregistry_impl.h"
 
 #include "base/files/file_path.h"
 #include "base/memory/ref_counted.h"
 #include "base/memory/shared_memory.h"
 #include "base/message_loop/message_loop.h"
@@ skipping 14 matching lines @@
 namespace content {
 
 WebBlobRegistryImpl::WebBlobRegistryImpl(ThreadSafeSender* sender)
     : sender_(sender) {
 }
 
 WebBlobRegistryImpl::~WebBlobRegistryImpl() {
 }
 
 void WebBlobRegistryImpl::SendData(const WebURL& url,
-                                   const WebThreadSafeData& data,
-                                   webkit_blob::BlobData::Item* item) {
+                                   const WebThreadSafeData& data) {
   const size_t kLargeThresholdBytes = 250 * 1024;
   const size_t kMaxSharedMemoryBytes = 10 * 1024 * 1024;
 
   if (data.size() == 0)
     return;
   if (data.size() < kLargeThresholdBytes) {
-    item->SetToBytes(data.data(), data.size());
-    sender_->Send(new BlobHostMsg_AppendBlobDataItem(url, *item));
+    webkit_blob::BlobData::Item item;
+    item.SetToBytes(data.data(), data.size());
+    sender_->Send(new BlobHostMsg_AppendBlobDataItemToBlobOrStream(url, item));
   } else {
     // We handle larger amounts of data via SharedMemory instead of
     // writing it directly to the IPC channel.
     size_t data_size = data.size();
     const char* data_ptr = data.data();
     size_t shared_memory_size = std::min(
         data_size, kMaxSharedMemoryBytes);
     scoped_ptr<base::SharedMemory> shared_memory(
         ChildThread::AllocateSharedMemory(shared_memory_size,
                                           sender_.get()));
     CHECK(shared_memory.get());
     while (data_size) {
       size_t chunk_size = std::min(data_size, shared_memory_size);
       memcpy(shared_memory->memory(), data_ptr, chunk_size);
-      sender_->Send(new BlobHostMsg_SyncAppendSharedMemory(
+      sender_->Send(new BlobHostMsg_SyncAppendSharedMemoryToBlobOrStream(
           url, shared_memory->handle(), chunk_size));
       data_size -= chunk_size;
       data_ptr += chunk_size;
     }
   }
 }
 
 void WebBlobRegistryImpl::registerBlobURL(
     const WebURL& url, WebBlobData& data) {
   DCHECK(ChildThread::current()->message_loop() ==
          base::MessageLoop::current());
   sender_->Send(new BlobHostMsg_StartBuildingBlob(url));
   size_t i = 0;
   WebBlobData::Item data_item;
   while (data.itemAt(i++, data_item)) {
-    webkit_blob::BlobData::Item item;
     switch (data_item.type) {
       case WebBlobData::Item::TypeData: {
         // WebBlobData does not allow partial data items.
         DCHECK(!data_item.offset && data_item.length == -1);
-        SendData(url, data_item.data, &item);
+        SendData(url, data_item.data);
         break;
       }
       case WebBlobData::Item::TypeFile:
         if (data_item.length) {
+          webkit_blob::BlobData::Item item;
           item.SetToFilePathRange(
               base::FilePath::FromUTF16Unsafe(data_item.filePath),
               static_cast<uint64>(data_item.offset),
               static_cast<uint64>(data_item.length),
               base::Time::FromDoubleT(data_item.expectedModificationTime));
-          sender_->Send(new BlobHostMsg_AppendBlobDataItem(url, item));
+          sender_->Send(
+              new BlobHostMsg_AppendBlobDataItemToBlobOrStream(url, item));
         }
         break;
       case WebBlobData::Item::TypeBlob:
         if (data_item.length) {
+          webkit_blob::BlobData::Item item;
           item.SetToBlobUrlRange(
               data_item.blobURL,
               static_cast<uint64>(data_item.offset),
               static_cast<uint64>(data_item.length));
-          sender_->Send(new BlobHostMsg_AppendBlobDataItem(url, item));
+          sender_->Send(
+              new BlobHostMsg_AppendBlobDataItemToBlobOrStream(url, item));
         }
         break;
       case WebBlobData::Item::TypeURL:
         if (data_item.length) {
           // We only support filesystem URL as of now.
           DCHECK(GURL(data_item.url).SchemeIsFileSystem());
+          webkit_blob::BlobData::Item item;
           item.SetToFileSystemUrlRange(
               data_item.url,
               static_cast<uint64>(data_item.offset),
               static_cast<uint64>(data_item.length),
               base::Time::FromDoubleT(data_item.expectedModificationTime));
-          sender_->Send(new BlobHostMsg_AppendBlobDataItem(url, item));
+          sender_->Send(
+              new BlobHostMsg_AppendBlobDataItemToBlobOrStream(url, item));
         }
         break;
       default:
         NOTREACHED();
     }
   }
   sender_->Send(new BlobHostMsg_FinishBuildingBlob(
       url, data.contentType().utf8().data()));
 }
 
+void WebBlobRegistryImpl::registerStreamURL(
+    const WebURL& url, const WebString& content_type) {
+  DCHECK(ChildThread::current()->message_loop() ==

[jam, 2013/07/23 15:52:36]

nit: here and in the rest of the file, ChildThread::current() returns non-NULL
only when you're on that thread, so this can just be written as

DCHECK(ChildThread::current());

[tyoshino (SeeGerritForStatus), 2013/07/24 12:14:20]

oh. Done.

+         base::MessageLoop::current());
+  sender_->Send(new BlobHostMsg_StartBuildingStream(url, content_type.utf8()));
+}
+
 void WebBlobRegistryImpl::registerBlobURL(
     const WebURL& url, const WebURL& src_url) {
   DCHECK(ChildThread::current()->message_loop() ==
          base::MessageLoop::current());
-  sender_->Send(new BlobHostMsg_CloneBlob(url, src_url));
+  sender_->Send(new BlobHostMsg_CloneBlobOrStream(url, src_url));
+}
+
+void WebBlobRegistryImpl::addDataToStream(const WebKit::WebURL& url,
+                                          WebKit::WebThreadSafeData& data) {
+  DCHECK(ChildThread::current()->message_loop() ==
+         base::MessageLoop::current());
+  SendData(url, data);
+}
+
+void WebBlobRegistryImpl::finalizeStream(const WebURL& url) {
+  DCHECK(ChildThread::current()->message_loop() ==
+         base::MessageLoop::current());
+  sender_->Send(new BlobHostMsg_FinishBuildingStream(url));
 }
 
 void WebBlobRegistryImpl::unregisterBlobURL(const WebURL& url) {
   DCHECK(ChildThread::current()->message_loop() ==
          base::MessageLoop::current());
-  sender_->Send(new BlobHostMsg_RemoveBlob(url));
+  sender_->Send(new BlobHostMsg_RemoveBlobOrStream(url));
 }
 
 }  // namespace content