2ca351bd582fc93761bb3e3ba7f4445e67b9b6ac
[WebKit-https.git] / Source / WebKit2 / WebProcess / Databases / IndexedDB / WebIDBConnectionToServer.cpp
1 /*
2  * Copyright (C) 2016 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 "WebIDBConnectionToServer.h"
28
29 #if ENABLE(INDEXED_DATABASE)
30
31 #include "DataReference.h"
32 #include "DatabaseToWebProcessConnectionMessages.h"
33 #include "NetworkConnectionToWebProcessMessages.h"
34 #include "NetworkProcessConnection.h"
35 #include "WebCoreArgumentCoders.h"
36 #include "WebIDBConnectionToClientMessages.h"
37 #include "WebIDBResult.h"
38 #include "WebProcess.h"
39 #include "WebToDatabaseProcessConnection.h"
40 #include <WebCore/IDBConnectionToServer.h>
41 #include <WebCore/IDBCursorInfo.h>
42 #include <WebCore/IDBDatabaseException.h>
43 #include <WebCore/IDBError.h>
44 #include <WebCore/IDBIndexInfo.h>
45 #include <WebCore/IDBKeyRangeData.h>
46 #include <WebCore/IDBObjectStoreInfo.h>
47 #include <WebCore/IDBOpenDBRequest.h>
48 #include <WebCore/IDBRequestData.h>
49 #include <WebCore/IDBResourceIdentifier.h>
50 #include <WebCore/IDBResultData.h>
51 #include <WebCore/IDBTransactionInfo.h>
52 #include <WebCore/IDBValue.h>
53
54 using namespace WebCore;
55
56 namespace WebKit {
57
58 Ref<WebIDBConnectionToServer> WebIDBConnectionToServer::create()
59 {
60     return adoptRef(*new WebIDBConnectionToServer);
61 }
62
63 WebIDBConnectionToServer::WebIDBConnectionToServer()
64 {
65     relaxAdoptionRequirement();
66
67     m_isOpenInServer = sendSync(Messages::DatabaseToWebProcessConnection::EstablishIDBConnectionToServer(), Messages::DatabaseToWebProcessConnection::EstablishIDBConnectionToServer::Reply(m_identifier));
68     m_connectionToServer = IDBClient::IDBConnectionToServer::create(*this);
69 }
70
71 WebIDBConnectionToServer::~WebIDBConnectionToServer()
72 {
73     if (m_isOpenInServer)
74         send(Messages::DatabaseToWebProcessConnection::RemoveIDBConnectionToServer(m_identifier));
75 }
76
77 IPC::Connection* WebIDBConnectionToServer::messageSenderConnection()
78 {
79     return &WebProcess::singleton().webToDatabaseProcessConnection()->connection();
80 }
81
82 IDBClient::IDBConnectionToServer& WebIDBConnectionToServer::coreConnectionToServer()
83 {
84     return *m_connectionToServer;
85 }
86
87 void WebIDBConnectionToServer::deleteDatabase(const IDBRequestData& requestData)
88 {
89     send(Messages::WebIDBConnectionToClient::DeleteDatabase(requestData));
90 }
91
92 void WebIDBConnectionToServer::openDatabase(const IDBRequestData& requestData)
93 {
94     send(Messages::WebIDBConnectionToClient::OpenDatabase(requestData));
95 }
96
97 void WebIDBConnectionToServer::abortTransaction(const IDBResourceIdentifier& transactionIdentifier)
98 {
99     send(Messages::WebIDBConnectionToClient::AbortTransaction(transactionIdentifier));
100 }
101
102 void WebIDBConnectionToServer::commitTransaction(const IDBResourceIdentifier& transactionIdentifier)
103 {
104     send(Messages::WebIDBConnectionToClient::CommitTransaction(transactionIdentifier));
105 }
106
107 void WebIDBConnectionToServer::didFinishHandlingVersionChangeTransaction(uint64_t databaseConnectionIdentifier, const IDBResourceIdentifier& transactionIdentifier)
108 {
109     send(Messages::WebIDBConnectionToClient::DidFinishHandlingVersionChangeTransaction(databaseConnectionIdentifier, transactionIdentifier));
110 }
111
112 void WebIDBConnectionToServer::createObjectStore(const IDBRequestData& requestData, const IDBObjectStoreInfo& info)
113 {
114     send(Messages::WebIDBConnectionToClient::CreateObjectStore(requestData, info));
115 }
116
117 void WebIDBConnectionToServer::deleteObjectStore(const IDBRequestData& requestData, const String& objectStoreName)
118 {
119     send(Messages::WebIDBConnectionToClient::DeleteObjectStore(requestData, objectStoreName));
120 }
121
122 void WebIDBConnectionToServer::renameObjectStore(const IDBRequestData& requestData, uint64_t objectStoreIdentifier, const String& newName)
123 {
124     send(Messages::WebIDBConnectionToClient::RenameObjectStore(requestData, objectStoreIdentifier, newName));
125 }
126
127 void WebIDBConnectionToServer::clearObjectStore(const IDBRequestData& requestData, uint64_t objectStoreIdentifier)
128 {
129     send(Messages::WebIDBConnectionToClient::ClearObjectStore(requestData, objectStoreIdentifier));
130 }
131
132 void WebIDBConnectionToServer::createIndex(const IDBRequestData& requestData, const IDBIndexInfo& info)
133 {
134     send(Messages::WebIDBConnectionToClient::CreateIndex(requestData, info));
135 }
136
137 void WebIDBConnectionToServer::deleteIndex(const IDBRequestData& requestData, uint64_t objectStoreIdentifier, const String& indexName)
138 {
139     send(Messages::WebIDBConnectionToClient::DeleteIndex(requestData, objectStoreIdentifier, indexName));
140 }
141
142 void WebIDBConnectionToServer::renameIndex(const IDBRequestData& requestData, uint64_t objectStoreIdentifier, uint64_t indexIdentifier, const String& newName)
143 {
144     send(Messages::WebIDBConnectionToClient::RenameIndex(requestData, objectStoreIdentifier, indexIdentifier, newName));
145 }
146
147 void WebIDBConnectionToServer::putOrAdd(const IDBRequestData& requestData, const IDBKeyData& keyData, const IDBValue& value, const IndexedDB::ObjectStoreOverwriteMode mode)
148 {
149     send(Messages::WebIDBConnectionToClient::PutOrAdd(requestData, keyData, value, static_cast<unsigned>(mode)));
150 }
151
152 void WebIDBConnectionToServer::getRecord(const IDBRequestData& requestData, const IDBGetRecordData& getRecordData)
153 {
154     send(Messages::WebIDBConnectionToClient::GetRecord(requestData, getRecordData));
155 }
156
157 void WebIDBConnectionToServer::getAllRecords(const IDBRequestData& requestData, const IDBGetAllRecordsData& getAllRecordsData)
158 {
159     send(Messages::WebIDBConnectionToClient::GetAllRecords(requestData, getAllRecordsData));
160 }
161
162 void WebIDBConnectionToServer::getCount(const IDBRequestData& requestData, const IDBKeyRangeData& range)
163 {
164     send(Messages::WebIDBConnectionToClient::GetCount(requestData, range));
165 }
166
167 void WebIDBConnectionToServer::deleteRecord(const IDBRequestData& requestData, const IDBKeyRangeData& range)
168 {
169     send(Messages::WebIDBConnectionToClient::DeleteRecord(requestData, range));
170 }
171
172 void WebIDBConnectionToServer::openCursor(const IDBRequestData& requestData, const IDBCursorInfo& info)
173 {
174     send(Messages::WebIDBConnectionToClient::OpenCursor(requestData, info));
175 }
176
177 void WebIDBConnectionToServer::iterateCursor(const IDBRequestData& requestData, const IDBKeyData& key, unsigned long count)
178 {
179     send(Messages::WebIDBConnectionToClient::IterateCursor(requestData, key, count));
180 }
181
182 void WebIDBConnectionToServer::establishTransaction(uint64_t databaseConnectionIdentifier, const IDBTransactionInfo& info)
183 {
184     send(Messages::WebIDBConnectionToClient::EstablishTransaction(databaseConnectionIdentifier, info));
185 }
186
187 void WebIDBConnectionToServer::databaseConnectionClosed(uint64_t databaseConnectionIdentifier)
188 {
189     send(Messages::WebIDBConnectionToClient::DatabaseConnectionClosed(databaseConnectionIdentifier));
190 }
191
192 void WebIDBConnectionToServer::abortOpenAndUpgradeNeeded(uint64_t databaseConnectionIdentifier, const IDBResourceIdentifier& transactionIdentifier)
193 {
194     send(Messages::WebIDBConnectionToClient::AbortOpenAndUpgradeNeeded(databaseConnectionIdentifier, transactionIdentifier));
195 }
196
197 void WebIDBConnectionToServer::didFireVersionChangeEvent(uint64_t databaseConnectionIdentifier, const IDBResourceIdentifier& requestIdentifier)
198 {
199     send(Messages::WebIDBConnectionToClient::DidFireVersionChangeEvent(databaseConnectionIdentifier, requestIdentifier));
200 }
201
202 void WebIDBConnectionToServer::openDBRequestCancelled(const IDBRequestData& requestData)
203 {
204     send(Messages::WebIDBConnectionToClient::OpenDBRequestCancelled(requestData));
205 }
206
207 void WebIDBConnectionToServer::confirmDidCloseFromServer(uint64_t databaseConnectionIdentifier)
208 {
209     send(Messages::WebIDBConnectionToClient::ConfirmDidCloseFromServer(databaseConnectionIdentifier));
210 }
211
212 void WebIDBConnectionToServer::getAllDatabaseNames(const WebCore::SecurityOriginData& topOrigin, const WebCore::SecurityOriginData& openingOrigin, uint64_t callbackID)
213 {
214     send(Messages::WebIDBConnectionToClient::GetAllDatabaseNames(m_identifier, topOrigin, openingOrigin, callbackID));
215 }
216
217 void WebIDBConnectionToServer::didDeleteDatabase(const IDBResultData& result)
218 {
219     m_connectionToServer->didDeleteDatabase(result);
220 }
221
222 void WebIDBConnectionToServer::didOpenDatabase(const IDBResultData& result)
223 {
224     m_connectionToServer->didOpenDatabase(result);
225 }
226
227 void WebIDBConnectionToServer::didAbortTransaction(const IDBResourceIdentifier& transactionIdentifier, const IDBError& error)
228 {
229     m_connectionToServer->didAbortTransaction(transactionIdentifier, error);
230 }
231
232 void WebIDBConnectionToServer::didCommitTransaction(const IDBResourceIdentifier& transactionIdentifier, const IDBError& error)
233 {
234     m_connectionToServer->didCommitTransaction(transactionIdentifier, error);
235 }
236
237 void WebIDBConnectionToServer::didCreateObjectStore(const IDBResultData& result)
238 {
239     m_connectionToServer->didCreateObjectStore(result);
240 }
241
242 void WebIDBConnectionToServer::didDeleteObjectStore(const IDBResultData& result)
243 {
244     m_connectionToServer->didDeleteObjectStore(result);
245 }
246
247 void WebIDBConnectionToServer::didRenameObjectStore(const IDBResultData& result)
248 {
249     m_connectionToServer->didRenameObjectStore(result);
250 }
251
252 void WebIDBConnectionToServer::didClearObjectStore(const IDBResultData& result)
253 {
254     m_connectionToServer->didClearObjectStore(result);
255 }
256
257 void WebIDBConnectionToServer::didCreateIndex(const IDBResultData& result)
258 {
259     m_connectionToServer->didCreateIndex(result);
260 }
261
262 void WebIDBConnectionToServer::didDeleteIndex(const IDBResultData& result)
263 {
264     m_connectionToServer->didDeleteIndex(result);
265 }
266
267 void WebIDBConnectionToServer::didRenameIndex(const IDBResultData& result)
268 {
269     m_connectionToServer->didRenameIndex(result);
270 }
271
272 void WebIDBConnectionToServer::didPutOrAdd(const IDBResultData& result)
273 {
274     m_connectionToServer->didPutOrAdd(result);
275 }
276
277 static void preregisterSandboxExtensionsIfNecessary(const WebIDBResult& result)
278 {
279     auto resultType = result.resultData().type();
280     if (resultType != IDBResultType::GetRecordSuccess && resultType != IDBResultType::OpenCursorSuccess && resultType != IDBResultType::IterateCursorSuccess && resultType != IDBResultType::GetAllRecordsSuccess) {
281         ASSERT(resultType == IDBResultType::Error);
282         return;
283     }
284
285     const auto filePaths = resultType == IDBResultType::GetAllRecordsSuccess ? result.resultData().getAllResult().allBlobFilePaths() : result.resultData().getResult().value().blobFilePaths();
286
287 #if ENABLE(SANDBOX_EXTENSIONS)
288     ASSERT(filePaths.size() == result.handles().size());
289 #endif
290
291     if (!filePaths.isEmpty())
292         WebProcess::singleton().networkConnection().connection().send(Messages::NetworkConnectionToWebProcess::PreregisterSandboxExtensionsForOptionallyFileBackedBlob(filePaths, result.handles()), 0);
293 }
294
295 void WebIDBConnectionToServer::didGetRecord(const WebIDBResult& result)
296 {
297     preregisterSandboxExtensionsIfNecessary(result);
298     m_connectionToServer->didGetRecord(result.resultData());
299 }
300
301 void WebIDBConnectionToServer::didGetAllRecords(const WebIDBResult& result)
302 {
303     if (result.resultData().getAllResult().type() == IndexedDB::GetAllType::Values)
304         preregisterSandboxExtensionsIfNecessary(result);
305     m_connectionToServer->didGetAllRecords(result.resultData());
306 }
307
308 void WebIDBConnectionToServer::didGetCount(const IDBResultData& result)
309 {
310     m_connectionToServer->didGetCount(result);
311 }
312
313 void WebIDBConnectionToServer::didDeleteRecord(const IDBResultData& result)
314 {
315     m_connectionToServer->didDeleteRecord(result);
316 }
317
318 void WebIDBConnectionToServer::didOpenCursor(const WebIDBResult& result)
319 {
320     preregisterSandboxExtensionsIfNecessary(result);
321     m_connectionToServer->didOpenCursor(result.resultData());
322 }
323
324 void WebIDBConnectionToServer::didIterateCursor(const WebIDBResult& result)
325 {
326     preregisterSandboxExtensionsIfNecessary(result);
327     m_connectionToServer->didIterateCursor(result.resultData());
328 }
329
330 void WebIDBConnectionToServer::fireVersionChangeEvent(uint64_t uniqueDatabaseConnectionIdentifier, const IDBResourceIdentifier& requestIdentifier, uint64_t requestedVersion)
331 {
332     m_connectionToServer->fireVersionChangeEvent(uniqueDatabaseConnectionIdentifier, requestIdentifier, requestedVersion);
333 }
334
335 void WebIDBConnectionToServer::didStartTransaction(const IDBResourceIdentifier& transactionIdentifier, const IDBError& error)
336 {
337     m_connectionToServer->didStartTransaction(transactionIdentifier, error);
338 }
339
340 void WebIDBConnectionToServer::didCloseFromServer(uint64_t databaseConnectionIdentifier, const IDBError& error)
341 {
342     m_connectionToServer->didCloseFromServer(databaseConnectionIdentifier, error);
343 }
344
345 void WebIDBConnectionToServer::notifyOpenDBRequestBlocked(const IDBResourceIdentifier& requestIdentifier, uint64_t oldVersion, uint64_t newVersion)
346 {
347     m_connectionToServer->notifyOpenDBRequestBlocked(requestIdentifier, oldVersion, newVersion);
348 }
349
350 void WebIDBConnectionToServer::didGetAllDatabaseNames(uint64_t callbackID, const Vector<String>& databaseNames)
351 {
352     m_connectionToServer->didGetAllDatabaseNames(callbackID, databaseNames);
353 }
354
355 void WebIDBConnectionToServer::connectionToServerLost()
356 {
357     m_connectionToServer->connectionToServerLost({ WebCore::IDBDatabaseException::UnknownError, ASCIILiteral("An internal error was encountered in the Indexed Database server") });
358 }
359
360 } // namespace WebKit
361
362 #endif // ENABLE(INDEXED_DATABASE)