1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
|
/*
* Copyright (C) 2021 Apple Inc. All rights reserved.
*
* Redistribution and use in source and binary forms, with or without
* modification, are permitted provided that the following conditions
* are met:
* 1. Redistributions of source code must retain the above copyright
* notice, this list of conditions and the following disclaimer.
* 2. Redistributions in binary form must reproduce the above copyright
* notice, this list of conditions and the following disclaimer in the
* documentation and/or other materials provided with the distribution.
*
* THIS SOFTWARE IS PROVIDED BY APPLE INC. AND ITS CONTRIBUTORS ``AS IS''
* AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO,
* THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
* PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL APPLE INC. OR ITS CONTRIBUTORS
* BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
* CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
* SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
* INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
* CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
* ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF
* THE POSSIBILITY OF SUCH DAMAGE.
*/
#include "config.h"
#include "WorkerStorageConnection.h"
#include "ClientOrigin.h"
#include "WorkerFileSystemStorageConnection.h"
#include "WorkerGlobalScope.h"
#include "WorkerLoaderProxy.h"
#include "WorkerThread.h"
#include <wtf/Scope.h>
namespace WebCore {
Ref<WorkerStorageConnection> WorkerStorageConnection::create(WorkerGlobalScope& scope)
{
return adoptRef(*new WorkerStorageConnection(scope));
}
WorkerStorageConnection::WorkerStorageConnection(WorkerGlobalScope& scope)
: m_scope(scope)
{
}
void WorkerStorageConnection::scopeClosed()
{
auto getPersistedCallbacks = std::exchange(m_getPersistedCallbacks, { });
for (auto& callback : getPersistedCallbacks.values())
callback(false);
auto getDirectoryCallbacks = std::exchange(m_getDirectoryCallbacks, { });
for (auto& callback : getDirectoryCallbacks.values())
callback(Exception { InvalidStateError });
m_scope = nullptr;
}
void WorkerStorageConnection::getPersisted(const ClientOrigin& origin, StorageConnection::PersistCallback&& completionHandler)
{
ASSERT(m_scope);
auto callbackIdentifier = ++m_lastCallbackIdentifier;
m_getPersistedCallbacks.add(callbackIdentifier, WTFMove(completionHandler));
callOnMainThread([callbackIdentifier, workerThread = Ref { m_scope->thread() }, origin = origin.isolatedCopy()]() mutable {
auto mainThreadConnection = workerThread->workerLoaderProxy().storageConnection();
auto mainThreadCallback = [callbackIdentifier, workerThread = WTFMove(workerThread)](bool result) mutable {
workerThread->runLoop().postTaskForMode([callbackIdentifier, result] (auto& scope) mutable {
downcast<WorkerGlobalScope>(scope).storageConnection().didGetPersisted(callbackIdentifier, result);
}, WorkerRunLoop::defaultMode());
};
if (!mainThreadConnection)
return mainThreadCallback(false);
mainThreadConnection->getPersisted(origin, WTFMove(mainThreadCallback));
});
}
void WorkerStorageConnection::didGetPersisted(uint64_t callbackIdentifier, bool persisted)
{
if (auto callback = m_getPersistedCallbacks.take(callbackIdentifier))
callback(persisted);
}
void WorkerStorageConnection::fileSystemGetDirectory(const ClientOrigin& origin, StorageConnection::GetDirectoryCallback&& completionHandler)
{
ASSERT(m_scope);
auto callbackIdentifier = ++m_lastCallbackIdentifier;
m_getDirectoryCallbacks.add(callbackIdentifier, WTFMove(completionHandler));
callOnMainThread([callbackIdentifier, workerThread = Ref { m_scope->thread() }, origin = origin.isolatedCopy()]() mutable {
auto mainThreadConnection = workerThread->workerLoaderProxy().storageConnection();
auto mainThreadCallback = [callbackIdentifier, workerThread = WTFMove(workerThread)](auto result) mutable {
workerThread->runLoop().postTaskForMode([callbackIdentifier, result = crossThreadCopy(result)] (auto& scope) mutable {
downcast<WorkerGlobalScope>(scope).storageConnection().didGetDirectory(callbackIdentifier, WTFMove(result));
}, WorkerRunLoop::defaultMode());
};
mainThreadConnection->fileSystemGetDirectory(origin, WTFMove(mainThreadCallback));
});
}
void WorkerStorageConnection::didGetDirectory(uint64_t callbackIdentifier, ExceptionOr<StorageConnection::DirectoryInfo>&& result)
{
RefPtr<FileSystemStorageConnection> mainThreadFileSystemStorageConnection = result.hasException() ? nullptr : result.returnValue().second;
auto releaseConnectionScope = makeScopeExit([connection = mainThreadFileSystemStorageConnection]() mutable {
if (connection)
callOnMainThread([connection = WTFMove(connection)]() { });
});
auto callback = m_getDirectoryCallbacks.take(callbackIdentifier);
if (!callback)
return;
if (result.hasException())
return callback(WTFMove(result));
if (!m_scope)
return callback(Exception { InvalidStateError });
releaseConnectionScope.release();
auto& workerFileSystemStorageConnection = m_scope->getFileSystemStorageConnection(Ref { *mainThreadFileSystemStorageConnection });
callback(StorageConnection::DirectoryInfo { result.returnValue().first, Ref { workerFileSystemStorageConnection } });
}
} // namespace WebCore
|
