Source/WebKit2/DatabaseProcess/IndexedDB/DatabaseProcessIDBConnection.cpp

   1 /*
   2  * Copyright (C) 2013 Apple Inc. All rights reserved.
   3  *
   4  * Redistribution and use in source and binary forms, with or without
   5  * modification, are permitted provided that the following conditions
   6  * are met:
   7  * 1. Redistributions of source code must retain the above copyright
   8  *    notice, this list of conditions and the following disclaimer.
   9  * 2. Redistributions in binary form must reproduce the above copyright
  10  *    notice, this list of conditions and the following disclaimer in the
  11  *    documentation and/or other materials provided with the distribution.
  12  *
  13  * THIS SOFTWARE IS PROVIDED BY APPLE INC. AND ITS CONTRIBUTORS ``AS IS''
  14  * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO,
  15  * THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
  16  * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL APPLE INC. OR ITS CONTRIBUTORS
  17  * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
  18  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
  19  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
  20  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
  21  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
  22  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF
  23  * THE POSSIBILITY OF SUCH DAMAGE.
  24  */
  25
  26 #include "config.h"
  27 #include "DatabaseProcessIDBConnection.h"
  28
  29 #if ENABLE(INDEXED_DATABASE) && ENABLE(DATABASE_PROCESS)
  30
  31 #include "DatabaseProcess.h"
  32 #include "DatabaseToWebProcessConnection.h"
  33 #include "IDBTransactionIdentifier.h"
  34 #include "Logging.h"
  35 #include "UniqueIDBDatabase.h"
  36 #include "WebCoreArgumentCoders.h"
  37 #include "WebIDBServerConnectionMessages.h"
  38 #include <WebCore/IDBDatabaseMetadata.h>
  39 #include <WebCore/IndexedDB.h>
  40
  41 using namespace WebCore;
  42
  43 namespace WebKit {
  44
  45 DatabaseProcessIDBConnection::DatabaseProcessIDBConnection(DatabaseToWebProcessConnection& connection, uint64_t serverConnectionIdentifier)
  46     : m_connection(connection)
  47     , m_serverConnectionIdentifier(serverConnectionIdentifier)
  48 {
  49 }
  50
  51 DatabaseProcessIDBConnection::~DatabaseProcessIDBConnection()
  52 {
  53     ASSERT(!m_uniqueIDBDatabase);
  54 }
  55
  56 void DatabaseProcessIDBConnection::disconnectedFromWebProcess()
  57 {
  58     m_uniqueIDBDatabase->unregisterConnection(*this);
  59     m_uniqueIDBDatabase.clear();
  60 }
  61
  62 void DatabaseProcessIDBConnection::establishConnection(const String& databaseName, const SecurityOriginData& openingOrigin, const SecurityOriginData& mainFrameOrigin)
  63 {
  64     m_uniqueIDBDatabase = DatabaseProcess::shared().getOrCreateUniqueIDBDatabase(UniqueIDBDatabaseIdentifier(databaseName, openingOrigin, mainFrameOrigin));
  65     m_uniqueIDBDatabase->registerConnection(*this);
  66 }
  67
  68 void DatabaseProcessIDBConnection::getOrEstablishIDBDatabaseMetadata(uint64_t requestID)
  69 {
  70     ASSERT(m_uniqueIDBDatabase);
  71
  72     LOG(IDB, "DatabaseProcess getOrEstablishIDBDatabaseMetadata request ID %llu", requestID);
  73
  74     RefPtr<DatabaseProcessIDBConnection> connection(this);
  75     m_uniqueIDBDatabase->getOrEstablishIDBDatabaseMetadata([connection, requestID](bool success, const IDBDatabaseMetadata& metadata) {
  76         connection->send(Messages::WebIDBServerConnection::DidGetOrEstablishIDBDatabaseMetadata(requestID, success, metadata));
  77     });
  78 }
  79
  80 void DatabaseProcessIDBConnection::openTransaction(uint64_t requestID, int64_t transactionID, const Vector<int64_t>& objectStoreIDs, uint64_t intMode)
  81 {
  82     ASSERT(m_uniqueIDBDatabase);
  83
  84     LOG(IDB, "DatabaseProcess openTransaction request ID %llu", requestID);
  85
  86     if (intMode > IndexedDB::TransactionModeMaximum) {
  87         send(Messages::WebIDBServerConnection::DidOpenTransaction(requestID, false));
  88         return;
  89     }
  90
  91     IndexedDB::TransactionMode mode = static_cast<IndexedDB::TransactionMode>(intMode);
  92     RefPtr<DatabaseProcessIDBConnection> connection(this);
  93     m_uniqueIDBDatabase->openTransaction(IDBTransactionIdentifier(*this, transactionID), objectStoreIDs, mode, [connection, requestID](bool success) {
  94         connection->send(Messages::WebIDBServerConnection::DidOpenTransaction(requestID, success));
  95     });
  96 }
  97
  98 void DatabaseProcessIDBConnection::beginTransaction(uint64_t requestID, int64_t transactionID)
  99 {
 100     ASSERT(m_uniqueIDBDatabase);
 101
 102     LOG(IDB, "DatabaseProcess beginTransaction request ID %llu", requestID);
 103
 104     RefPtr<DatabaseProcessIDBConnection> connection(this);
 105     m_uniqueIDBDatabase->beginTransaction(IDBTransactionIdentifier(*this, transactionID), [connection, requestID](bool success) {
 106         connection->send(Messages::WebIDBServerConnection::DidBeginTransaction(requestID, success));
 107     });
 108 }
 109
 110 void DatabaseProcessIDBConnection::commitTransaction(uint64_t requestID, int64_t transactionID)
 111 {
 112     ASSERT(m_uniqueIDBDatabase);
 113
 114     LOG(IDB, "DatabaseProcess commitTransaction request ID %llu", requestID);
 115
 116     RefPtr<DatabaseProcessIDBConnection> connection(this);
 117     m_uniqueIDBDatabase->commitTransaction(IDBTransactionIdentifier(*this, transactionID), [connection, requestID](bool success) {
 118         connection->send(Messages::WebIDBServerConnection::DidCommitTransaction(requestID, success));
 119     });
 120 }
 121
 122 void DatabaseProcessIDBConnection::resetTransaction(uint64_t requestID, int64_t transactionID)
 123 {
 124     ASSERT(m_uniqueIDBDatabase);
 125
 126     LOG(IDB, "DatabaseProcess resetTransaction request ID %llu", requestID);
 127
 128     RefPtr<DatabaseProcessIDBConnection> connection(this);
 129     m_uniqueIDBDatabase->resetTransaction(IDBTransactionIdentifier(*this, transactionID), [connection, requestID](bool success) {
 130         connection->send(Messages::WebIDBServerConnection::DidResetTransaction(requestID, success));
 131     });
 132 }
 133
 134 void DatabaseProcessIDBConnection::rollbackTransaction(uint64_t requestID, int64_t transactionID)
 135 {
 136     ASSERT(m_uniqueIDBDatabase);
 137
 138     LOG(IDB, "DatabaseProcess rollbackTransaction request ID %llu", requestID);
 139
 140     RefPtr<DatabaseProcessIDBConnection> connection(this);
 141     m_uniqueIDBDatabase->rollbackTransaction(IDBTransactionIdentifier(*this, transactionID), [connection, requestID](bool success) {
 142         connection->send(Messages::WebIDBServerConnection::DidRollbackTransaction(requestID, success));
 143     });
 144 }
 145
 146 void DatabaseProcessIDBConnection::changeDatabaseVersion(uint64_t requestID, int64_t transactionID, uint64_t newVersion)
 147 {
 148     ASSERT(m_uniqueIDBDatabase);
 149
 150     LOG(IDB, "DatabaseProcess changeDatabaseVersion request ID %llu, new version %llu", requestID, newVersion);
 151
 152     RefPtr<DatabaseProcessIDBConnection> connection(this);
 153     m_uniqueIDBDatabase->changeDatabaseVersion(IDBTransactionIdentifier(*this, transactionID), newVersion, [connection, requestID](bool success) {
 154         connection->send(Messages::WebIDBServerConnection::DidChangeDatabaseVersion(requestID, success));
 155     });
 156 }
 157
 158 void DatabaseProcessIDBConnection::createObjectStore(uint64_t requestID, int64_t transactionID, WebCore::IDBObjectStoreMetadata metadata)
 159 {
 160     ASSERT(m_uniqueIDBDatabase);
 161
 162     LOG(IDB, "DatabaseProcess createObjectStore request ID %llu, object store name '%s'", requestID, metadata.name.utf8().data());
 163     RefPtr<DatabaseProcessIDBConnection> connection(this);
 164     m_uniqueIDBDatabase->createObjectStore(IDBTransactionIdentifier(*this, transactionID), metadata, [connection, requestID](bool success) {
 165         connection->send(Messages::WebIDBServerConnection::DidCreateObjectStore(requestID, success));
 166     });
 167 }
 168
 169 void DatabaseProcessIDBConnection::deleteObjectStore(uint64_t requestID, int64_t transactionID, int64_t objectStoreID)
 170 {
 171     ASSERT(m_uniqueIDBDatabase);
 172
 173     LOG(IDB, "DatabaseProcess deleteObjectStore request ID %llu, object store id %lli", requestID, objectStoreID);
 174     RefPtr<DatabaseProcessIDBConnection> connection(this);
 175     m_uniqueIDBDatabase->deleteObjectStore(IDBTransactionIdentifier(*this, transactionID), objectStoreID, [connection, requestID](bool success) {
 176         connection->send(Messages::WebIDBServerConnection::DidDeleteObjectStore(requestID, success));
 177     });
 178 }
 179
 180 void DatabaseProcessIDBConnection::putRecord(uint64_t requestID, int64_t transactionID, int64_t objectStoreID, const WebCore::IDBKeyData& key, const IPC::DataReference& value, int64_t putMode, const Vector<int64_t>& indexIDs, const Vector<Vector<WebCore::IDBKeyData>>& indexKeys)
 181 {
 182     ASSERT(m_uniqueIDBDatabase);
 183
 184     LOG(IDB, "DatabaseProcess putRecord request ID %llu, object store id %lli", requestID, objectStoreID);
 185     RefPtr<DatabaseProcessIDBConnection> connection(this);
 186     m_uniqueIDBDatabase->putRecord(IDBTransactionIdentifier(*this, transactionID), objectStoreID, key, value, putMode, indexIDs, indexKeys, [connection, requestID](const IDBKeyData& keyData, uint32_t errorCode, const String& errorMessage) {
 187         connection->send(Messages::WebIDBServerConnection::DidPutRecord(requestID, keyData, errorCode, errorMessage));
 188     });
 189 }
 190
 191
 192 IPC::Connection* DatabaseProcessIDBConnection::messageSenderConnection()
 193 {
 194     return m_connection->connection();
 195 }
 196
 197 } // namespace WebKit
 198
 199 #endif // ENABLE(INDEXED_DATABASE) && ENABLE(DATABASE_PROCESS)