Implement <webview>.addContentScript/removeContentScript API [1]

extensions/browser/guest_view/guest_view_message_filter.cc

 // Copyright 2015 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 "extensions/browser/guest_view/guest_view_message_filter.h"
 
 #include "content/public/browser/browser_thread.h"
 #include "content/public/browser/render_frame_host.h"
+#include "content/public/browser/render_process_host.h"
 #include "content/public/browser/render_view_host.h"
 #include "extensions/browser/guest_view/guest_view_base.h"
 #include "extensions/browser/guest_view/guest_view_manager.h"
 #include "extensions/browser/guest_view/mime_handler_view/mime_handler_view_constants.h"
 #include "extensions/browser/guest_view/mime_handler_view/mime_handler_view_guest.h"
+#include "extensions/browser/guest_view/web_view/web_view_content_script_manager.h"
+#include "extensions/browser/guest_view/web_view/web_view_guest.h"
+#include "extensions/browser/guest_view/web_view/web_view_renderer_state.h"
 #include "extensions/common/guest_view/guest_view_messages.h"
 #include "ipc/ipc_message_macros.h"
 
 using content::BrowserContext;
 using content::BrowserThread;
 using content::RenderFrameHost;
 using content::WebContents;
 
 namespace extensions {
 
@@ skipping 30 matching lines @@
   BrowserThread::DeleteOnIOThread::Destruct(this);
 }
 
 bool GuestViewMessageFilter::OnMessageReceived(const IPC::Message& message) {
   bool handled = true;
   IPC_BEGIN_MESSAGE_MAP(GuestViewMessageFilter, message)
     IPC_MESSAGE_HANDLER(GuestViewHostMsg_AttachGuest, OnAttachGuest)
     IPC_MESSAGE_HANDLER(GuestViewHostMsg_CreateMimeHandlerViewGuest,
                         OnCreateMimeHandlerViewGuest)
     IPC_MESSAGE_HANDLER(GuestViewHostMsg_ResizeGuest, OnResizeGuest)
+    IPC_MESSAGE_HANDLER(GuestViewHostMsg_CanExecuteContentScriptSync,
+                        OnCanExecuteContentScript)
     IPC_MESSAGE_UNHANDLED(handled = false)
   IPC_END_MESSAGE_MAP()
   return handled;
 }
 
 void GuestViewMessageFilter::OnAttachGuest(
     int element_instance_id,
     int guest_instance_id,
     const base::DictionaryValue& params) {
   DCHECK_CURRENTLY_ON(BrowserThread::UI);
@@ skipping 52 matching lines @@
   auto mhvg = MimeHandlerViewGuest::FromWebContents(guest_web_contents);
   if (!mhvg)
     return;
 
   SetSizeParams set_size_params;
   set_size_params.enable_auto_size.reset(new bool(false));
   set_size_params.normal_size.reset(new gfx::Size(new_size));
   mhvg->SetSize(set_size_params);
 }
 
+void GuestViewMessageFilter::OnCanExecuteContentScript(int render_view_id,
+                                                       int script_id,
+                                                       bool* allowed) {
+  WebViewRendererState::WebViewInfo info;
+  WebViewRendererState::GetInstance()->GetInfo(render_process_id_,
+                                               render_view_id, &info);
+
+  *allowed =
+      info.content_script_ids.find(script_id) != info.content_script_ids.end();
+}
+
 void GuestViewMessageFilter::MimeHandlerViewGuestCreatedCallback(
     int element_instance_id,
     int embedder_render_process_id,
     int embedder_render_frame_id,
     const gfx::Size& element_size,
     WebContents* web_contents) {
   auto manager = GuestViewManager::FromBrowserContext(browser_context_);
   if (!manager)
     return;
 
@@ skipping 13 matching lines @@
   manager->AttachGuest(embedder_render_process_id,
                        element_instance_id,
                        guest_instance_id,
                        attach_params);
 
   rfh->Send(
       new GuestViewMsg_CreateMimeHandlerViewGuestACK(element_instance_id));
 }
 
 }  // namespace extensions