1 files changed, 125 insertions, 0 deletions
diff --git a/chromium/third_party/blink/renderer/core/workers/shared_worker_client_holder.cc b/chromium/third_party/blink/renderer/core/workers/shared_worker_client_holder.cc
new file mode 100644
index 00000000000..eab1ce5258c
--- /dev/null
+++ b/chromium/third_party/blink/renderer/core/workers/shared_worker_client_holder.cc
@@ -0,0 +1,125 @@
+/*
+ * Copyright (C) 2009 Google 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:
+ *
+ *     * Redistributions of source code must retain the above copyright
+ * notice, this list of conditions and the following disclaimer.
+ *     * 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.
+ *     * Neither the name of Google Inc. nor the names of its
+ * contributors may be used to endorse or promote products derived from
+ * this software without specific prior written permission.
+ *
+ * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND 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 THE COPYRIGHT
+ * OWNER OR 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 "third_party/blink/renderer/core/workers/shared_worker_client_holder.h"
+
+#include <memory>
+#include <utility>
+#include "base/logging.h"
+#include "services/service_manager/public/cpp/interface_provider.h"
+#include "third_party/blink/public/common/messaging/message_port_channel.h"
+#include "third_party/blink/public/mojom/worker/shared_worker_info.mojom-blink.h"
+#include "third_party/blink/public/platform/web_content_security_policy.h"
+#include "third_party/blink/public/platform/web_string.h"
+#include "third_party/blink/public/platform/web_url.h"
+#include "third_party/blink/public/web/blink.h"
+#include "third_party/blink/public/web/web_shared_worker.h"
+#include "third_party/blink/renderer/core/execution_context/execution_context.h"
+#include "third_party/blink/renderer/core/frame/csp/content_security_policy.h"
+#include "third_party/blink/renderer/core/workers/shared_worker.h"
+#include "third_party/blink/renderer/core/workers/shared_worker_client.h"
+
+namespace blink {
+
+const char SharedWorkerClientHolder::kSupplementName[] =
+    "SharedWorkerClientHolder";
+
+SharedWorkerClientHolder* SharedWorkerClientHolder::From(Document& document) {
+  DCHECK(IsMainThread());
+  SharedWorkerClientHolder* holder =
+      Supplement<Document>::From<SharedWorkerClientHolder>(document);
+  if (!holder) {
+    holder = MakeGarbageCollected<SharedWorkerClientHolder>(document);
+    Supplement<Document>::ProvideTo(document, holder);
+  }
+  return holder;
+}
+
+SharedWorkerClientHolder::SharedWorkerClientHolder(Document& document)
+    : ContextLifecycleObserver(&document) {
+  DCHECK(IsMainThread());
+  document.GetInterfaceProvider()->GetInterface(mojo::MakeRequest(&connector_));
+}
+
+void SharedWorkerClientHolder::Connect(
+    SharedWorker* worker,
+    MessagePortChannel port,
+    const KURL& url,
+    mojom::blink::BlobURLTokenPtr blob_url_token,
+    const String& name) {
+  DCHECK(IsMainThread());
+  DCHECK(!name.IsNull());
+
+  // TODO(estark): this is broken, as it only uses the first header
+  // when multiple might have been sent. Fix by making the
+  // mojom::blink::SharedWorkerInfo take a map that can contain multiple
+  // headers.
+  Vector<CSPHeaderAndType> headers =
+      worker->GetExecutionContext()->GetContentSecurityPolicy()->Headers();
+  WebString header = "";
+  auto header_type = mojom::ContentSecurityPolicyType::kReport;
+  if (headers.size() > 0) {
+    header = headers[0].first;
+    header_type =
+        static_cast<mojom::ContentSecurityPolicyType>(headers[0].second);
+  }
+
+  mojom::blink::SharedWorkerInfoPtr info(mojom::blink::SharedWorkerInfo::New(
+      url, name, header, header_type,
+      worker->GetExecutionContext()->GetSecurityContext().AddressSpace()));
+
+  mojom::blink::SharedWorkerClientPtr client_ptr;
+  client_set_.AddBinding(std::make_unique<SharedWorkerClient>(worker),
+                         mojo::MakeRequest(&client_ptr));
+
+  connector_->Connect(
+      std::move(info), std::move(client_ptr),
+      worker->GetExecutionContext()->IsSecureContext()
+          ? mojom::SharedWorkerCreationContextType::kSecure
+          : mojom::SharedWorkerCreationContextType::kNonsecure,
+      port.ReleaseHandle(),
+      mojom::blink::BlobURLTokenPtr(mojom::blink::BlobURLTokenPtrInfo(
+          blob_url_token.PassInterface().PassHandle(),
+          mojom::blink::BlobURLToken::Version_)));
+}
+
+void SharedWorkerClientHolder::ContextDestroyed(ExecutionContext*) {
+  DCHECK(IsMainThread());
+  // Close mojo connections which will signal disinterest in the associated
+  // shared worker.
+  client_set_.CloseAllBindings();
+}
+
+void SharedWorkerClientHolder::Trace(Visitor* visitor) {
+  Supplement<Document>::Trace(visitor);
+  ContextLifecycleObserver::Trace(visitor);
+}
+
+}  // namespace blink