[WIP] Hack on Blob mojom

content/browser/blob_storage/blob_dispatcher_host.cc

 // Copyright 2016 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/browser/blob_storage/blob_dispatcher_host.h"
 
 #include <algorithm>
 
 #include "base/bind.h"
 #include "base/metrics/histogram_macros.h"
@@ skipping 29 matching lines @@
   BDH_TRACING_ENUM_LAST
 };
 
 } // namespace
 
 BlobDispatcherHost::BlobDispatcherHost(
     int process_id,
     scoped_refptr<ChromeBlobStorageContext> blob_storage_context,
     scoped_refptr<storage::FileSystemContext> file_system_context)
     : BrowserMessageFilter(BlobMsgStart),
+      BrowserAssociatedInterface<mojom::BlobFactory>(this, this),
       process_id_(process_id),
       file_system_context_(std::move(file_system_context)),
-      blob_storage_context_(std::move(blob_storage_context)) {}
+      blob_storage_context_(std::move(blob_storage_context)),
+      blob_bindings_(mojo::BindingSetDispatchMode::WITH_CONTEXT) {}
 
 BlobDispatcherHost::~BlobDispatcherHost() {
   ClearHostFromBlobStorageContext();
 }
 
 void BlobDispatcherHost::OnChannelClosing() {
   ClearHostFromBlobStorageContext();
   public_blob_urls_.clear();
   blobs_inuse_map_.clear();
 }
 
 bool BlobDispatcherHost::OnMessageReceived(const IPC::Message& message) {
   bool handled = true;
   IPC_BEGIN_MESSAGE_MAP(BlobDispatcherHost, message)
-    IPC_MESSAGE_HANDLER(BlobStorageMsg_RegisterBlobUUID, OnRegisterBlobUUID)
-    IPC_MESSAGE_HANDLER(BlobStorageMsg_StartBuildingBlob, OnStartBuildingBlob)
     IPC_MESSAGE_HANDLER(BlobStorageMsg_MemoryItemResponse, OnMemoryItemResponse)
-    IPC_MESSAGE_HANDLER(BlobStorageMsg_CancelBuildingBlob, OnCancelBuildingBlob)
     IPC_MESSAGE_HANDLER(BlobHostMsg_IncrementRefCount, OnIncrementBlobRefCount)
     IPC_MESSAGE_HANDLER(BlobHostMsg_DecrementRefCount, OnDecrementBlobRefCount)
     IPC_MESSAGE_HANDLER(BlobHostMsg_RegisterPublicURL, OnRegisterPublicBlobURL)
     IPC_MESSAGE_HANDLER(BlobHostMsg_RevokePublicURL, OnRevokePublicBlobURL)
     IPC_MESSAGE_UNHANDLED(handled = false)
   IPC_END_MESSAGE_MAP()
   return handled;
 }
 
-void BlobDispatcherHost::OnRegisterBlobUUID(
+void BlobDispatcherHost::BuildBlob(
     const std::string& uuid,
     const std::string& content_type,
     const std::string& content_disposition,
-    const std::set<std::string>& referenced_blob_uuids) {
+    const std::vector<std::string>& referenced_blob_uuids,
+    const std::vector<storage::DataElement>& item_descriptions,
+    const BuildBlobCallback& callback) {
   DCHECK_CURRENTLY_ON(BrowserThread::IO);
+
   BlobStorageContext* context = this->context();
   if (uuid.empty() || context->registry().HasEntry(uuid) ||
       async_builder_.IsBeingBuilt(uuid)) {
     bad_message::ReceivedBadMessage(this, bad_message::BDH_UUID_REGISTERED);
     return;
   }
   blobs_inuse_map_[uuid] = 1;
+  std::set<std::string> referenced_blob_uuid_set;
+  std::copy(referenced_blob_uuids.begin(), referenced_blob_uuids.end(),
+            std::inserter(referenced_blob_uuid_set,
+                          referenced_blob_uuid_set.end()));
   BlobTransportResult result = async_builder_.RegisterBlobUUID(
-      uuid, content_type, content_disposition, referenced_blob_uuids, context);
+      uuid, content_type, content_disposition, referenced_blob_uuid_set,
+      context);
   switch (result) {
     case BlobTransportResult::BAD_IPC:
       blobs_inuse_map_.erase(uuid);
       bad_message::ReceivedBadMessage(this,
                                       bad_message::BDH_CONSTRUCTION_FAILED);
       break;
     case BlobTransportResult::CANCEL_REFERENCED_BLOB_BROKEN:
       // The async builder builds the blob as broken, and we just need to send
       // the cancel message back to the renderer.
-      Send(new BlobStorageMsg_CancelBuildingBlob(
-          uuid, IPCBlobCreationCancelCode::REFERENCED_BLOB_BROKEN));
-      break;
+      callback.Run(
+          nullptr, mojom::BlobCreationCancelCode::REFERENCED_BLOB_BROKEN);
+      return;
     case BlobTransportResult::DONE:
       break;
     case BlobTransportResult::CANCEL_MEMORY_FULL:
     case BlobTransportResult::CANCEL_FILE_ERROR:
     case BlobTransportResult::CANCEL_UNKNOWN:
     case BlobTransportResult::PENDING_RESPONSES:
       NOTREACHED();
       break;
   }
-}
 
-void BlobDispatcherHost::OnStartBuildingBlob(
-    const std::string& uuid,
-    const std::vector<storage::DataElement>& descriptions) {
-  DCHECK_CURRENTLY_ON(BrowserThread::IO);
   if (uuid.empty()) {
-    SendIPCResponse(uuid, BlobTransportResult::BAD_IPC);
+    SendIPCResponse(callback, BlobTransportResult::BAD_IPC);
     return;
   }
   BlobStorageContext* context = this->context();
   const BlobStorageRegistry::Entry* entry = context->registry().GetEntry(uuid);
   if (!entry || entry->state == BlobStorageRegistry::BlobState::BROKEN) {
     // We ignore messages for blobs that don't exist to handle the case where
     // the renderer de-refs a blob that we're still constructing, and there are
     // no references to that blob. We ignore broken as well, in the case where
     // we decided to break a blob after RegisterBlobUUID is called.
     // Second, if the last dereference of the blob happened on a different host,
     // then we still haven't gotten rid of the 'building' state in the original
     // host. So we call cancel, and send the message just in case that happens.
     if (async_builder_.IsBeingBuilt(uuid)) {
       async_builder_.CancelBuildingBlob(
           uuid, IPCBlobCreationCancelCode::BLOB_DEREFERENCED_WHILE_BUILDING,
           context);
-      Send(new BlobStorageMsg_CancelBuildingBlob(
-          uuid, IPCBlobCreationCancelCode::BLOB_DEREFERENCED_WHILE_BUILDING));
+      callback.Run(
+          nullptr,
+          mojom::BlobCreationCancelCode::BLOB_DEREFERENCED_WHILE_BUILDING);
     }
     return;
   }
   if (!async_builder_.IsBeingBuilt(uuid)) {
-    SendIPCResponse(uuid, BlobTransportResult::BAD_IPC);
+    SendIPCResponse(callback, BlobTransportResult::BAD_IPC);
     return;
   }
 
   ChildProcessSecurityPolicyImpl* security_policy =
       ChildProcessSecurityPolicyImpl::GetInstance();
   for (const DataElement& item : descriptions) {
     if (item.type() == storage::DataElement::TYPE_FILE_FILESYSTEM) {
       FileSystemURL filesystem_url(
           file_system_context_->CrackURL(item.filesystem_url()));
       if (!FileSystemURLIsValid(file_system_context_.get(), filesystem_url) ||
           !security_policy->CanReadFileSystemFile(process_id_,
                                                   filesystem_url)) {
         async_builder_.CancelBuildingBlob(
             uuid, IPCBlobCreationCancelCode::FILE_WRITE_FAILED, context);
-        Send(new BlobStorageMsg_CancelBuildingBlob(
-            uuid, IPCBlobCreationCancelCode::FILE_WRITE_FAILED));
+        callback.Run(nullptr, mojom::BlobCreationCancelCode::FILE_WRITE_FAILED);
         return;
       }
     }
     if (item.type() == storage::DataElement::TYPE_FILE &&
         !security_policy->CanReadFile(process_id_, item.path())) {
       async_builder_.CancelBuildingBlob(
           uuid, IPCBlobCreationCancelCode::FILE_WRITE_FAILED, context);
-      Send(new BlobStorageMsg_CancelBuildingBlob(
-          uuid, IPCBlobCreationCancelCode::FILE_WRITE_FAILED));
+      callback.Run(nullptr, mojom::BlobCreationCancelCode::FILE_WRITE_FAILED);
       return;
     }
   }
 
   // |this| owns async_builder_ so using base::Unretained(this) is safe.
   BlobTransportResult result = async_builder_.StartBuildingBlob(
       uuid, descriptions, context->memory_available(), context,
       base::Bind(&BlobDispatcherHost::SendMemoryRequest, base::Unretained(this),
                  uuid));
-  SendIPCResponse(uuid, result);
+  SendIPCResponse(callback, result);
 }
 
 void BlobDispatcherHost::OnMemoryItemResponse(
     const std::string& uuid,
     const std::vector<storage::BlobItemBytesResponse>& responses) {
   DCHECK_CURRENTLY_ON(BrowserThread::IO);
   if (uuid.empty()) {
     SendIPCResponse(uuid, BlobTransportResult::BAD_IPC);
     return;
   }
@@ skipping 153 matching lines @@
     std::unique_ptr<std::vector<base::SharedMemoryHandle>> memory_handles,
     std::unique_ptr<std::vector<base::File>> files) {
   DCHECK_CURRENTLY_ON(BrowserThread::IO);
   std::vector<IPC::PlatformFileForTransit> file_handles;
   // TODO(dmurph): Support file-backed blob transportation.
   DCHECK(files->empty());
   Send(new BlobStorageMsg_RequestMemoryItem(uuid, *requests, *memory_handles,
                                             file_handles));
 }
 
-void BlobDispatcherHost::SendIPCResponse(const std::string& uuid,
+void BlobDispatcherHost::SendIPCResponse(const BuildBlobCallback& callback,
                                          storage::BlobTransportResult result) {
+  mojom::BlobAssociatedPtr new_blob;
+  mojom::BlobCreationCancelCode cancel_code =
+      mojom:BlobCreationCancelCode::UNKNOWN;
   switch (result) {
     case BlobTransportResult::BAD_IPC:
       bad_message::ReceivedBadMessage(this,
                                       bad_message::BDH_CONSTRUCTION_FAILED);
-      return;
+      cancel_code = mojom::BlobCreationCancelCode::UNKNOWN;
+      break;
     case BlobTransportResult::CANCEL_MEMORY_FULL:
-      Send(new BlobStorageMsg_CancelBuildingBlob(
-          uuid, IPCBlobCreationCancelCode::OUT_OF_MEMORY));
-      return;
+      cancel_code = mojom::BlobCreationCancelCode::OUT_OF_MEMORY;
+      break;
     case BlobTransportResult::CANCEL_FILE_ERROR:
-      Send(new BlobStorageMsg_CancelBuildingBlob(
-          uuid, IPCBlobCreationCancelCode::FILE_WRITE_FAILED));
-      return;
+      cancel_code = mojom::BlobCreationCancelCode::FILE_WRITE_FAILED;
+      break;
     case BlobTransportResult::CANCEL_REFERENCED_BLOB_BROKEN:
-      Send(new BlobStorageMsg_CancelBuildingBlob(
-          uuid, IPCBlobCreationCancelCode::REFERENCED_BLOB_BROKEN));
-      return;
+      cancel_code = mojom::BlobCreationCancelCode::REFERENCED_BLOB_BROKEN;
+      break;
     case BlobTransportResult::CANCEL_UNKNOWN:
-      Send(new BlobStorageMsg_CancelBuildingBlob(
-          uuid, IPCBlobCreationCancelCode::UNKNOWN));
-      return;
+      cancel_code = mojom::BlobCreationCancelCode::UNKNOWN;
+      break;
     case BlobTransportResult::PENDING_RESPONSES:
       return;
     case BlobTransportResult::DONE:
-      Send(new BlobStorageMsg_DoneBuildingBlob(uuid));
-      return;
+      new_blob = blob_bindings_.CreateInterfacePtrAndBind(this);
+      break;
+    default:
+      NOTREACHED();
+      break;
   }
-  NOTREACHED();
+
+  callback.Run(std::move(new_blob), cancel_code);
 }
 
 bool BlobDispatcherHost::IsInUseInHost(const std::string& uuid) {
   return blobs_inuse_map_.find(uuid) != blobs_inuse_map_.end();
 }
 
 bool BlobDispatcherHost::IsUrlRegisteredInHost(const GURL& blob_url) {
   return public_blob_urls_.find(blob_url) != public_blob_urls_.end();
 }
 
 void BlobDispatcherHost::ClearHostFromBlobStorageContext() {
   BlobStorageContext* context = this->context();
   for (const auto& url : public_blob_urls_) {
     context->RevokePublicBlobURL(url);
   }
   for (const auto& uuid_refnum_pair : blobs_inuse_map_) {
     for (int i = 0; i < uuid_refnum_pair.second; ++i)
       context->DecrementBlobRefCount(uuid_refnum_pair.first);
   }
   async_builder_.CancelAll(context);
 }
 
 }  // namespace content