[IndexedDB] Hold referenced to IDB::put blobs in the IDBRequest

content/browser/indexed_db/indexed_db_dispatcher_host.cc

 // Copyright (c) 2012 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/indexed_db/indexed_db_dispatcher_host.h"
 
 #include "base/bind.h"
 #include "base/files/file_path.h"
 #include "base/guid.h"
 #include "base/memory/ptr_util.h"
 #include "base/process/process.h"
 #include "base/sequenced_task_runner.h"
 #include "base/stl_util.h"
 #include "base/strings/utf_string_conversions.h"
+#include "content/browser/blob_storage/blob_dispatcher_host.h"
 #include "content/browser/indexed_db/indexed_db_callbacks.h"
 #include "content/browser/indexed_db/indexed_db_connection.h"
 #include "content/browser/indexed_db/indexed_db_context_impl.h"
 #include "content/browser/indexed_db/indexed_db_database_callbacks.h"
 #include "content/browser/indexed_db/indexed_db_pending_connection.h"
 #include "content/public/browser/browser_thread.h"
 #include "storage/browser/blob/blob_data_builder.h"
 #include "storage/browser/blob/blob_storage_context.h"
 #include "storage/browser/database/database_util.h"
 #include "url/origin.h"
@@ skipping 43 matching lines @@
   DISALLOW_COPY_AND_ASSIGN(IDBThreadHelper);
 };
 
 IndexedDBDispatcherHost::IndexedDBDispatcherHost(
     int ipc_process_id,
     scoped_refptr<net::URLRequestContextGetter> request_context_getter,
     scoped_refptr<IndexedDBContextImpl> indexed_db_context,
     scoped_refptr<ChromeBlobStorageContext> blob_storage_context)
     : indexed_db_context_(std::move(indexed_db_context)),
       blob_storage_context_(std::move(blob_storage_context)),
+      blob_dispatcher_host_(nullptr),
       idb_runner_(indexed_db_context_->TaskRunner()),
       ipc_process_id_(ipc_process_id),
       weak_factory_(this) {
   // Can be null in unittests.
   idb_helper_ = idb_runner_
                     ? new IDBThreadHelper(ipc_process_id_,
                                           std::move(request_context_getter),
                                           indexed_db_context_)
                     : nullptr;
   DCHECK(indexed_db_context_.get());
@@ skipping 14 matching lines @@
     ::indexed_db::mojom::DatabaseAssociatedRequest request) {
   database_bindings_.AddBinding(std::move(database), std::move(request));
 }
 
 void IndexedDBDispatcherHost::AddCursorBinding(
     std::unique_ptr<::indexed_db::mojom::Cursor> cursor,
     ::indexed_db::mojom::CursorAssociatedRequest request) {
   cursor_bindings_.AddBinding(std::move(cursor), std::move(request));
 }
 
+void IndexedDBDispatcherHost::SetBlobDispatcherHost(
+    scoped_refptr<BlobDispatcherHost> host) {
+  blob_dispatcher_host_ = std::move(host);

[pwnall, 2017/04/21 01:39:22]

Would it make sense to DCHECK that the host_ is nullptr? My intuition is that
this gets called exactly once for a host.

+}
+
+void IndexedDBDispatcherHost::AckBlobRecievedFromIDB(const std::string& uuid) {
+  DCHECK(blob_dispatcher_host_);
+  blob_dispatcher_host_->AckBlobRecievedFromIDB(uuid);
+}
+
 std::string IndexedDBDispatcherHost::HoldBlobData(
     const IndexedDBBlobInfo& blob_info) {
   DCHECK_CURRENTLY_ON(BrowserThread::IO);
   std::string uuid = blob_info.uuid();
   storage::BlobStorageContext* context = blob_storage_context_->context();
   std::unique_ptr<storage::BlobDataHandle> blob_data_handle;
   if (uuid.empty()) {
     uuid = base::GenerateGUID();
     storage::BlobDataBuilder blob_data_builder(uuid);
     blob_data_builder.set_content_type(base::UTF16ToUTF8(blob_info.type()));
@@ skipping 155 matching lines @@
   DCHECK(indexed_db_context_->TaskRunner()->RunsTasksOnCurrentThread());
 
   base::FilePath indexed_db_path = indexed_db_context_->data_path();
   DCHECK(request_context_getter_);
   indexed_db_context_->GetIDBFactory()->DeleteDatabase(
       name, request_context_getter_, callbacks, origin, indexed_db_path,
       force_close);
 }
 
 }  // namespace content