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
|
/*
* Copyright (C) 2022 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.
*/
#pragma once
#include <WebCore/SharedWorkerIdentifier.h>
#include <WebCore/SharedWorkerKey.h>
#include <WebCore/SharedWorkerObjectIdentifier.h>
#include <WebCore/TransferredMessagePort.h>
#include <WebCore/WorkerFetchResult.h>
#include <WebCore/WorkerInitializationData.h>
#include <WebCore/WorkerOptions.h>
#include <wtf/ListHashSet.h>
#include <wtf/WeakPtr.h>
namespace WebCore {
class RegistrableDomain;
}
namespace WebKit {
class WebSharedWorkerServer;
class WebSharedWorkerServerToContextConnection;
class WebSharedWorker : public CanMakeWeakPtr<WebSharedWorker> {
WTF_MAKE_FAST_ALLOCATED;
public:
WebSharedWorker(WebSharedWorkerServer&, const WebCore::SharedWorkerKey&, const WebCore::WorkerOptions&);
~WebSharedWorker();
static WebSharedWorker* fromIdentifier(WebCore::SharedWorkerIdentifier);
WebCore::SharedWorkerIdentifier identifier() const { return m_identifier; }
const WebCore::SharedWorkerKey& key() const { return m_key; }
const WebCore::WorkerOptions& workerOptions() const { return m_workerOptions; }
const WebCore::ClientOrigin& origin() const { return m_key.origin; }
const URL& url() const { return m_key.url; }
WebCore::RegistrableDomain registrableDomain() const;
WebSharedWorkerServerToContextConnection* contextConnection() const;
void addSharedWorkerObject(WebCore::SharedWorkerObjectIdentifier, const WebCore::TransferredMessagePort&);
void removeSharedWorkerObject(WebCore::SharedWorkerObjectIdentifier);
void suspend(WebCore::SharedWorkerObjectIdentifier);
void resume(WebCore::SharedWorkerObjectIdentifier);
unsigned sharedWorkerObjectsCount() const { return m_sharedWorkerObjects.size(); }
void forEachSharedWorkerObject(const Function<void(WebCore::SharedWorkerObjectIdentifier, const WebCore::TransferredMessagePort&)>&) const;
std::optional<WebCore::ProcessIdentifier> firstSharedWorkerObjectProcess() const;
void didCreateContextConnection(WebSharedWorkerServerToContextConnection&);
bool isRunning() const { return m_isRunning; }
void markAsRunning() { m_isRunning = true; }
const WebCore::WorkerInitializationData& initializationData() const { return m_initializationData; }
void setInitializationData(WebCore::WorkerInitializationData&& initializationData) { m_initializationData = WTFMove(initializationData); }
const WebCore::WorkerFetchResult& fetchResult() const { return m_fetchResult; }
void setFetchResult(WebCore::WorkerFetchResult&&);
bool didFinishFetching() const { return !!m_fetchResult.script; }
void launch(WebSharedWorkerServerToContextConnection&);
struct SharedWorkerObjectState {
bool isSuspended { false };
WebCore::TransferredMessagePort port;
};
struct Object {
WebCore::SharedWorkerObjectIdentifier identifier;
SharedWorkerObjectState state;
};
private:
WebSharedWorker(WebSharedWorker&&) = delete;
WebSharedWorker& operator=(WebSharedWorker&&) = delete;
WebSharedWorker(const WebSharedWorker&) = delete;
WebSharedWorker& operator=(const WebSharedWorker&) = delete;
void suspendIfNeeded();
void resumeIfNeeded();
WebSharedWorkerServer& m_server;
WebCore::SharedWorkerIdentifier m_identifier;
WebCore::SharedWorkerKey m_key;
WebCore::WorkerOptions m_workerOptions;
ListHashSet<Object> m_sharedWorkerObjects;
WebCore::WorkerFetchResult m_fetchResult;
WebCore::WorkerInitializationData m_initializationData;
bool m_isRunning { false };
bool m_isSuspended { false };
};
} // namespace WebKit
namespace WTF {
struct WebSharedWorkerObjectHash {
static unsigned hash(const WebKit::WebSharedWorker::Object& object) { return DefaultHash<WebCore::SharedWorkerObjectIdentifier>::hash(object.identifier); }
static bool equal(const WebKit::WebSharedWorker::Object& a, const WebKit::WebSharedWorker::Object& b) { return a.identifier == b.identifier; }
static constexpr bool safeToCompareToEmptyOrDeleted = true;
};
template<> struct HashTraits<WebKit::WebSharedWorker::Object> : SimpleClassHashTraits<WebSharedWorkerObjectHash> { };
template<> struct DefaultHash<WebKit::WebSharedWorker::Object> : WebSharedWorkerObjectHash { };
} // namespace WTF
|
