ServiceWorker: Implement 'source' and 'origin' attributes of ExtendableMessageEvent

content/child/service_worker/web_service_worker_impl.cc

 // Copyright 2013 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "content/child/service_worker/web_service_worker_impl.h"
 
 #include <utility>
 
 #include "base/macros.h"
 #include "content/child/service_worker/service_worker_dispatcher.h"
 #include "content/child/service_worker/service_worker_handle_reference.h"
+#include "content/child/service_worker/web_service_worker_provider_impl.h"
 #include "content/child/thread_safe_sender.h"
 #include "content/child/webmessageportchannel_impl.h"
 #include "content/common/service_worker/service_worker_messages.h"
+#include "third_party/WebKit/public/platform/WebSecurityOrigin.h"
 #include "third_party/WebKit/public/platform/WebString.h"
 #include "third_party/WebKit/public/platform/modules/serviceworker/WebServiceWorkerProxy.h"
 #include "third_party/WebKit/public/web/WebRuntimeFeatures.h"
 
 using blink::WebMessagePortChannel;
 using blink::WebMessagePortChannelArray;
 using blink::WebMessagePortChannelClient;
 using blink::WebRuntimeFeatures;
+using blink::WebSecurityOrigin;
 using blink::WebString;
 
 namespace content {
 
 namespace {
 
 class HandleImpl : public blink::WebServiceWorker::Handle {
  public:
   explicit HandleImpl(const scoped_refptr<WebServiceWorkerImpl>& worker)
       : worker_(worker) {}
   ~HandleImpl() override {}
 
   blink::WebServiceWorker* serviceWorker() override { return worker_.get(); }
 
  private:
   scoped_refptr<WebServiceWorkerImpl> worker_;
 
   DISALLOW_COPY_AND_ASSIGN(HandleImpl);
 };
 
 void SendPostMessageToWorkerOnMainThread(
     ThreadSafeSender* thread_safe_sender,
     int handle_id,
+    int provider_id,
     const base::string16& message,
+    const url::Origin& source_origin,
     scoped_ptr<WebMessagePortChannelArray> channels) {
   if (WebRuntimeFeatures::isServiceWorkerExtendableMessageEventEnabled()) {
     thread_safe_sender->Send(new ServiceWorkerHostMsg_PostMessageToWorker(
-        handle_id, message,
+        handle_id, provider_id, message, source_origin,
         WebMessagePortChannelImpl::ExtractMessagePortIDs(std::move(channels))));
   } else {
     thread_safe_sender->Send(
         new ServiceWorkerHostMsg_DeprecatedPostMessageToWorker(
             handle_id, message,
             WebMessagePortChannelImpl::ExtractMessagePortIDs(
                 std::move(channels))));
   }
 }
 
@@ skipping 32 matching lines @@
 }
 
 blink::WebURL WebServiceWorkerImpl::url() const {
   return handle_ref_->url();
 }
 
 blink::WebServiceWorkerState WebServiceWorkerImpl::state() const {
   return state_;
 }
 
-void WebServiceWorkerImpl::postMessage(const WebString& message,
-                                       WebMessagePortChannelArray* channels) {
+void WebServiceWorkerImpl::postMessage(
+    blink::WebServiceWorkerProvider* provider,
+    const WebString& message,
+    const WebSecurityOrigin& source_origin,
+    WebMessagePortChannelArray* channels) {
+  WebServiceWorkerProviderImpl* provider_impl =
+      static_cast<WebServiceWorkerProviderImpl*>(provider);
   ServiceWorkerDispatcher* dispatcher =
       ServiceWorkerDispatcher::GetThreadSpecificInstance();
   DCHECK(dispatcher);
 
   // This may send channels for MessagePorts, and all internal book-keeping
   // messages for MessagePort (e.g. QueueMessages) are sent from main thread
   // (with thread hopping), so we need to do the same thread hopping here not
   // to overtake those messages.
   dispatcher->main_thread_task_runner()->PostTask(
-      FROM_HERE, base::Bind(&SendPostMessageToWorkerOnMainThread,
-                            thread_safe_sender_, handle_ref_->handle_id(),
-                            // We cast WebString to string16 before crossing
-                            // threads for thread-safety.
-                            static_cast<base::string16>(message),
-                            base::Passed(make_scoped_ptr(channels))));
+      FROM_HERE,
+      base::Bind(&SendPostMessageToWorkerOnMainThread, thread_safe_sender_,
+                 handle_ref_->handle_id(), provider_impl->provider_id(),
+                 // We cast WebString to string16 before crossing
+                 // threads for thread-safety.
+                 static_cast<base::string16>(message),
+                 url::Origin(source_origin),
+                 base::Passed(make_scoped_ptr(channels))));
 }
 
 void WebServiceWorkerImpl::terminate() {
   thread_safe_sender_->Send(
       new ServiceWorkerHostMsg_TerminateWorker(handle_ref_->handle_id()));
 }
 
 // static
 blink::WebPassOwnPtr<blink::WebServiceWorker::Handle>
 WebServiceWorkerImpl::CreateHandle(
     const scoped_refptr<WebServiceWorkerImpl>& worker) {
   if (!worker)
     return nullptr;
   return blink::adoptWebPtr(new HandleImpl(worker));
 }
 
 WebServiceWorkerImpl::~WebServiceWorkerImpl() {
   ServiceWorkerDispatcher* dispatcher =
       ServiceWorkerDispatcher::GetThreadSpecificInstance();
   if (dispatcher)
     dispatcher->RemoveServiceWorker(handle_ref_->handle_id());
 }
 
 }  // namespace content