Port IndexedDB open() and database callbacks to Mojo.

content/child/indexed_db/indexed_db_dispatcher.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/indexed_db/indexed_db_dispatcher.h"
 
 #include <utility>
 
 #include "base/format_macros.h"
 #include "base/lazy_instance.h"
 #include "base/strings/stringprintf.h"
 #include "base/threading/thread_local.h"
+#include "content/child/child_thread_impl.h"
 #include "content/child/indexed_db/indexed_db_key_builders.h"
 #include "content/child/indexed_db/webidbcursor_impl.h"
 #include "content/child/indexed_db/webidbdatabase_impl.h"
 #include "content/child/thread_safe_sender.h"
 #include "content/common/indexed_db/indexed_db_messages.h"
 #include "ipc/ipc_channel.h"
-#include "third_party/WebKit/public/platform/modules/indexeddb/WebIDBDatabaseCallbacks.h"
+#include "ipc/ipc_sync_channel.h"
+#include "services/shell/public/cpp/interface_provider.h"
 #include "third_party/WebKit/public/platform/modules/indexeddb/WebIDBDatabaseError.h"
 #include "third_party/WebKit/public/platform/modules/indexeddb/WebIDBDatabaseException.h"
 #include "third_party/WebKit/public/platform/modules/indexeddb/WebIDBObservation.h"
 #include "third_party/WebKit/public/platform/modules/indexeddb/WebIDBValue.h"
 
 using blink::WebBlobInfo;
 using blink::WebData;
 using blink::WebIDBCallbacks;
 using blink::WebIDBCursor;
 using blink::WebIDBDatabase;
-using blink::WebIDBDatabaseCallbacks;
 using blink::WebIDBDatabaseError;
 using blink::WebIDBKey;
 using blink::WebIDBMetadata;
 using blink::WebIDBObservation;
 using blink::WebIDBObserver;
 using blink::WebIDBValue;
 using blink::WebString;
 using blink::WebVector;
 using base::ThreadLocalPointer;
 
 namespace content {
 static base::LazyInstance<ThreadLocalPointer<IndexedDBDispatcher> >::Leaky
     g_idb_dispatcher_tls = LAZY_INSTANCE_INITIALIZER;
 
 namespace {
 
 IndexedDBDispatcher* const kHasBeenDeleted =
     reinterpret_cast<IndexedDBDispatcher*>(0x1);
 
 }  // unnamed namespace
 
 IndexedDBDispatcher::IndexedDBDispatcher(ThreadSafeSender* thread_safe_sender)
     : thread_safe_sender_(thread_safe_sender) {
+  ChildThreadImpl* child_thread = ChildThreadImpl::current();
+  if (child_thread)
+    child_thread->channel()->GetRemoteAssociatedInterface(&database_factory_);
   g_idb_dispatcher_tls.Pointer()->Set(this);
 }
 
 IndexedDBDispatcher::~IndexedDBDispatcher() {
   // Clear any pending callbacks - which may result in dispatch requests -
   // before marking the dispatcher as deleted.
   pending_callbacks_.Clear();
-  pending_database_callbacks_.Clear();
 
   DCHECK(pending_callbacks_.IsEmpty());
-  DCHECK(pending_database_callbacks_.IsEmpty());
 
   g_idb_dispatcher_tls.Pointer()->Set(kHasBeenDeleted);
 }
 
 IndexedDBDispatcher* IndexedDBDispatcher::ThreadSpecificInstance(
     ThreadSafeSender* thread_safe_sender) {
   if (g_idb_dispatcher_tls.Pointer()->Get() == kHasBeenDeleted) {
     NOTREACHED() << "Re-instantiating TLS IndexedDBDispatcher.";
     g_idb_dispatcher_tls.Pointer()->Set(NULL);
   }
@@ skipping 85 matching lines @@
     IPC_MESSAGE_HANDLER(IndexedDBMsg_CallbacksSuccessStringList,
                         OnSuccessStringList)
     IPC_MESSAGE_HANDLER(IndexedDBMsg_CallbacksSuccessArray, OnSuccessArray)
     IPC_MESSAGE_HANDLER(IndexedDBMsg_CallbacksSuccessValue, OnSuccessValue)
     IPC_MESSAGE_HANDLER(IndexedDBMsg_CallbacksSuccessInteger, OnSuccessInteger)
     IPC_MESSAGE_HANDLER(IndexedDBMsg_CallbacksSuccessUndefined,
                         OnSuccessUndefined)
     IPC_MESSAGE_HANDLER(IndexedDBMsg_CallbacksError, OnError)
     IPC_MESSAGE_HANDLER(IndexedDBMsg_CallbacksIntBlocked, OnIntBlocked)
     IPC_MESSAGE_HANDLER(IndexedDBMsg_CallbacksUpgradeNeeded, OnUpgradeNeeded)
-    IPC_MESSAGE_HANDLER(IndexedDBMsg_DatabaseCallbacksForcedClose,
-                        OnForcedClose)
-    IPC_MESSAGE_HANDLER(IndexedDBMsg_DatabaseCallbacksVersionChange,
-                        OnVersionChange)
-    IPC_MESSAGE_HANDLER(IndexedDBMsg_DatabaseCallbacksAbort, OnAbort)
-    IPC_MESSAGE_HANDLER(IndexedDBMsg_DatabaseCallbacksComplete, OnComplete)
     IPC_MESSAGE_HANDLER(IndexedDBMsg_DatabaseCallbacksChanges,
                         OnDatabaseChanges)
     IPC_MESSAGE_UNHANDLED(handled = false)
   IPC_END_MESSAGE_MAP()
   // If a message gets here, IndexedDBMessageFilter already determined that it
   // is an IndexedDB message.
   DCHECK(handled) << "Didn't handle a message defined at line "
                   << IPC_MESSAGE_ID_LINE(msg.type());
 }
 
@@ skipping 80 matching lines @@
 }
 
 void IndexedDBDispatcher::RequestIDBCursorPrefetchReset(int used_prefetches,
                                                         int unused_prefetches,
                                                         int32_t ipc_cursor_id) {
   Send(new IndexedDBHostMsg_CursorPrefetchReset(
       ipc_cursor_id, used_prefetches, unused_prefetches));
 }
 
 void IndexedDBDispatcher::RequestIDBFactoryOpen(
-    const base::string16& name,
+    const std::string& name,
     int64_t version,
     int64_t transaction_id,
-    WebIDBCallbacks* callbacks_ptr,
-    WebIDBDatabaseCallbacks* database_callbacks_ptr,
+    WebIDBCallbacks* callbacks,
+    mojo::ScopedInterfaceEndpointHandle* client_interface_endpoint,
     const url::Origin& origin) {
-  std::unique_ptr<WebIDBCallbacks> callbacks(callbacks_ptr);
-  std::unique_ptr<WebIDBDatabaseCallbacks> database_callbacks(
-      database_callbacks_ptr);
+  indexed_db::mojom::DatabaseClientAssociatedPtrInfo clientInfo;
+  indexed_db::mojom::DatabaseClientAssociatedRequest clientRequest;
+  database_factory_.associated_group()->CreateAssociatedInterface(
+      mojo::AssociatedGroup::WILL_PASS_PTR, &clientInfo, &clientRequest);
+  *client_interface_endpoint = clientRequest.PassHandle();
 
-  IndexedDBHostMsg_FactoryOpen_Params params;
-  params.ipc_thread_id = CurrentWorkerId();
-  params.ipc_callbacks_id = pending_callbacks_.Add(callbacks.release());
-  params.ipc_database_callbacks_id =
-      pending_database_callbacks_.Add(database_callbacks.release());
-  params.origin = origin;
-  params.name = name;
-  params.transaction_id = transaction_id;
-  params.version = version;
-  Send(new IndexedDBHostMsg_FactoryOpen(params));
+  database_factory_->Open(name, version, transaction_id, origin,
+                          std::move(clientInfo), CurrentWorkerId(),
+                          pending_callbacks_.Add(callbacks));
 }
 
 void IndexedDBDispatcher::RequestIDBFactoryGetDatabaseNames(
     WebIDBCallbacks* callbacks_ptr,
     const url::Origin& origin) {
   std::unique_ptr<WebIDBCallbacks> callbacks(callbacks_ptr);
 
   IndexedDBHostMsg_FactoryGetDatabaseNames_Params params;
   params.ipc_thread_id = CurrentWorkerId();
   params.ipc_callbacks_id = pending_callbacks_.Add(callbacks.release());
   params.origin = origin;
   Send(new IndexedDBHostMsg_FactoryGetDatabaseNames(params));
 }
 
 void IndexedDBDispatcher::RequestIDBFactoryDeleteDatabase(
     const base::string16& name,
     WebIDBCallbacks* callbacks_ptr,
     const url::Origin& origin) {
   std::unique_ptr<WebIDBCallbacks> callbacks(callbacks_ptr);
 
   IndexedDBHostMsg_FactoryDeleteDatabase_Params params;
   params.ipc_thread_id = CurrentWorkerId();
   params.ipc_callbacks_id = pending_callbacks_.Add(callbacks.release());
   params.origin = origin;
   params.name = name;
   Send(new IndexedDBHostMsg_FactoryDeleteDatabase(params));
 }
 
-void IndexedDBDispatcher::RequestIDBDatabaseClose(
-    int32_t ipc_database_id,
-    int32_t ipc_database_callbacks_id) {
+void IndexedDBDispatcher::RequestIDBDatabaseClose(int32_t ipc_database_id) {
   Send(new IndexedDBHostMsg_DatabaseClose(ipc_database_id));
-  // There won't be pending database callbacks if the transaction was aborted in
-  // the initial upgradeneeded event handler.
-  if (pending_database_callbacks_.Lookup(ipc_database_callbacks_id))
-    pending_database_callbacks_.Remove(ipc_database_callbacks_id);
 }
 
 void IndexedDBDispatcher::NotifyIDBDatabaseVersionChangeIgnored(
     int32_t ipc_database_id) {
   Send(new IndexedDBHostMsg_DatabaseVersionChangeIgnored(ipc_database_id));
 }
 
 void IndexedDBDispatcher::RequestIDBDatabaseCreateTransaction(
     int32_t ipc_database_id,
     int64_t transaction_id,
-    WebIDBDatabaseCallbacks* database_callbacks_ptr,
     WebVector<long long> object_store_ids,
     blink::WebIDBTransactionMode mode) {
-  std::unique_ptr<WebIDBDatabaseCallbacks> database_callbacks(
-      database_callbacks_ptr);
   IndexedDBHostMsg_DatabaseCreateTransaction_Params params;
   params.ipc_thread_id = CurrentWorkerId();
   params.ipc_database_id = ipc_database_id;
   params.transaction_id = transaction_id;
-  params.ipc_database_callbacks_id =
-      pending_database_callbacks_.Add(database_callbacks.release());
   params.object_store_ids
       .assign(object_store_ids.data(),
               object_store_ids.data() + object_store_ids.size());
   params.mode = mode;
 
   Send(new IndexedDBHostMsg_DatabaseCreateTransaction(params));
 }
 
 void IndexedDBDispatcher::RequestIDBDatabaseGet(
     int32_t ipc_database_id,
@@ skipping 183 matching lines @@
 }
 
 void IndexedDBDispatcher::DatabaseDestroyed(int32_t ipc_database_id) {
   DCHECK_EQ(databases_.count(ipc_database_id), 1u);
   databases_.erase(ipc_database_id);
 }
 
 void IndexedDBDispatcher::OnSuccessIDBDatabase(
     int32_t ipc_thread_id,
     int32_t ipc_callbacks_id,
-    int32_t ipc_database_callbacks_id,
     int32_t ipc_object_id,
     const IndexedDBDatabaseMetadata& idb_metadata) {
   DCHECK_EQ(ipc_thread_id, CurrentWorkerId());
   WebIDBCallbacks* callbacks = pending_callbacks_.Lookup(ipc_callbacks_id);
   if (!callbacks)
     return;
   WebIDBMetadata metadata(ConvertMetadata(idb_metadata));
   // If an upgrade was performed, count will be non-zero.
   WebIDBDatabase* database = NULL;
 
   // Back-end will send kNoDatabase if it was already sent in OnUpgradeNeeded.
   // May already be deleted and removed from the table, but do not recreate..
   if (ipc_object_id != kNoDatabase) {
     DCHECK(!databases_.count(ipc_object_id));
-    database = databases_[ipc_object_id] = new WebIDBDatabaseImpl(
-        ipc_object_id, ipc_database_callbacks_id, thread_safe_sender_.get());
+    database = databases_[ipc_object_id] =
+        new WebIDBDatabaseImpl(ipc_object_id, thread_safe_sender_.get());
   }
 
   callbacks->onSuccess(database, metadata);
   pending_callbacks_.Remove(ipc_callbacks_id);
 }
 
 void IndexedDBDispatcher::OnSuccessIndexedDBKey(int32_t ipc_thread_id,
                                                 int32_t ipc_callbacks_id,
                                                 const IndexedDBKey& key) {
   DCHECK_EQ(ipc_thread_id, CurrentWorkerId());
@@ skipping 183 matching lines @@
 
 void IndexedDBDispatcher::OnUpgradeNeeded(
     const IndexedDBMsg_CallbacksUpgradeNeeded_Params& p) {
   DCHECK_EQ(p.ipc_thread_id, CurrentWorkerId());
   WebIDBCallbacks* callbacks = pending_callbacks_.Lookup(p.ipc_callbacks_id);
   DCHECK(callbacks);
   WebIDBMetadata metadata(ConvertMetadata(p.idb_metadata));
   DCHECK(!databases_.count(p.ipc_database_id));
   databases_[p.ipc_database_id] =
       new WebIDBDatabaseImpl(p.ipc_database_id,
-                             p.ipc_database_callbacks_id,
                              thread_safe_sender_.get());
   callbacks->onUpgradeNeeded(
       p.old_version,
       databases_[p.ipc_database_id],
       metadata,
       static_cast<blink::WebIDBDataLoss>(p.data_loss),
       WebString::fromUTF8(p.data_loss_message));
 }
 
 void IndexedDBDispatcher::OnError(int32_t ipc_thread_id,
                                   int32_t ipc_callbacks_id,
                                   int code,
                                   const base::string16& message) {
   DCHECK_EQ(ipc_thread_id, CurrentWorkerId());
   WebIDBCallbacks* callbacks = pending_callbacks_.Lookup(ipc_callbacks_id);
   if (!callbacks)
     return;
   if (message.empty())
     callbacks->onError(WebIDBDatabaseError(code));
   else
     callbacks->onError(WebIDBDatabaseError(code, message));
   pending_callbacks_.Remove(ipc_callbacks_id);
   cursor_transaction_ids_.erase(ipc_callbacks_id);
 }
 
-void IndexedDBDispatcher::OnAbort(int32_t ipc_thread_id,
-                                  int32_t ipc_database_callbacks_id,
-                                  int64_t transaction_id,
-                                  int code,
-                                  const base::string16& message) {
-  DCHECK_EQ(ipc_thread_id, CurrentWorkerId());
-  WebIDBDatabaseCallbacks* callbacks =
-      pending_database_callbacks_.Lookup(ipc_database_callbacks_id);
-  if (!callbacks)
-    return;
-  if (message.empty())
-    callbacks->onAbort(transaction_id, WebIDBDatabaseError(code));
-  else
-    callbacks->onAbort(transaction_id, WebIDBDatabaseError(code, message));
-}
-
-void IndexedDBDispatcher::OnComplete(int32_t ipc_thread_id,
-                                     int32_t ipc_database_callbacks_id,
-                                     int64_t transaction_id) {
-  DCHECK_EQ(ipc_thread_id, CurrentWorkerId());
-  WebIDBDatabaseCallbacks* callbacks =
-      pending_database_callbacks_.Lookup(ipc_database_callbacks_id);
-  if (!callbacks)
-    return;
-  callbacks->onComplete(transaction_id);
-}
-
 void IndexedDBDispatcher::OnDatabaseChanges(
     int32_t ipc_thread_id,
     int32_t ipc_database_id,
     const IndexedDBMsg_ObserverChanges& changes) {
   DCHECK_EQ(ipc_thread_id, CurrentWorkerId());
   std::vector<WebIDBObservation> observations(
       ConvertObservations(changes.observations));
   for (auto& it : changes.observation_index) {
     WebIDBObserver* observer = observers_.Lookup(it.first);
     // An observer can be removed from the renderer, but still exist in the
     // backend. Moreover, observer might have recorded some changes before being
     // removed from the backend and thus, have its id be present in changes.
     if (!observer)
       continue;
     observer->onChange(observations, std::move(it.second));
   }
 }
 
-void IndexedDBDispatcher::OnForcedClose(int32_t ipc_thread_id,
-                                        int32_t ipc_database_callbacks_id) {
-  DCHECK_EQ(ipc_thread_id, CurrentWorkerId());
-  WebIDBDatabaseCallbacks* callbacks =
-      pending_database_callbacks_.Lookup(ipc_database_callbacks_id);
-  if (!callbacks)
-    return;
-  callbacks->onForcedClose();
-}
-
-void IndexedDBDispatcher::OnVersionChange(int32_t ipc_thread_id,
-                                          int32_t ipc_database_callbacks_id,
-                                          int64_t old_version,
-                                          int64_t new_version) {
-  DCHECK_EQ(ipc_thread_id, CurrentWorkerId());
-  WebIDBDatabaseCallbacks* callbacks =
-      pending_database_callbacks_.Lookup(ipc_database_callbacks_id);
-  // callbacks would be NULL if a versionchange event is received after close
-  // has been called.
-  if (!callbacks)
-    return;
-  callbacks->onVersionChange(old_version, new_version);
-}
-
 void IndexedDBDispatcher::ResetCursorPrefetchCaches(
     int64_t transaction_id,
     int32_t ipc_exception_cursor_id) {
   typedef std::map<int32_t, WebIDBCursorImpl*>::iterator Iterator;
   for (Iterator i = cursors_.begin(); i != cursors_.end(); ++i) {
     if (i->first == ipc_exception_cursor_id ||
         i->second->transaction_id() != transaction_id)
       continue;
     i->second->ResetPrefetchCache();
   }
 }
 
 }  // namespace content