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

extensions/browser/guest_view/guest_view_message_filter.cc

Index: extensions/browser/guest_view/guest_view_message_filter.cc
diff --git a/extensions/browser/guest_view/guest_view_message_filter.cc b/extensions/browser/guest_view/guest_view_message_filter.cc
index e7c505c43d41af08114cb94df061653f1201279e..1567df7fa929a7c143b8856e6cf345e4960c8bba 100644
--- a/extensions/browser/guest_view/guest_view_message_filter.cc
+++ b/extensions/browser/guest_view/guest_view_message_filter.cc
@@ -6,11 +6,15 @@
 
 #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"
 
@@ -39,6 +43,7 @@ void GuestViewMessageFilter::OverrideThreadForMessage(
     BrowserThread::ID* thread) {
   switch (message.type()) {
     case GuestViewHostMsg_AttachGuest::ID:
+    case GuestViewHostMsg_CanExecuteContentScript::ID:
     case GuestViewHostMsg_CreateMimeHandlerViewGuest::ID:
     case GuestViewHostMsg_ResizeGuest::ID:
       *thread = BrowserThread::UI;
@@ -61,6 +66,8 @@ bool GuestViewMessageFilter::OnMessageReceived(const IPC::Message& message) {
     IPC_MESSAGE_HANDLER(GuestViewHostMsg_CreateMimeHandlerViewGuest,
                         OnCreateMimeHandlerViewGuest)
     IPC_MESSAGE_HANDLER(GuestViewHostMsg_ResizeGuest, OnResizeGuest)
+    IPC_MESSAGE_HANDLER(GuestViewHostMsg_CanExecuteContentScript,
+                        OnCanExecuteContentScript)
     IPC_MESSAGE_UNHANDLED(handled = false)
   IPC_END_MESSAGE_MAP()
   return handled;
@@ -133,6 +140,34 @@ void GuestViewMessageFilter::OnResizeGuest(int render_frame_id,
   mhvg->SetSize(set_size_params);
 }
 
+void GuestViewMessageFilter::OnCanExecuteContentScript(int render_view_id,
+                                                       int script_id,
+                                                       bool* allowed) {
+  *allowed = false;
+  auto render_view_host =
+      content::RenderViewHost::FromID(render_process_id_, render_view_id);
+  if (!render_view_host)
+    return;
+
+  auto web_contents =
+      content::WebContents::FromRenderViewHost(render_view_host);
+  if (!web_contents)
+    return;
+
+  auto guest = WebViewGuest::FromWebContents(web_contents);
+  if (!guest)
+    return;
+
+  WebViewContentScriptManager* manager =
+      WebViewContentScriptManager::Get(browser_context_);
+  if (!manager)
+    return;
+
+  *allowed = manager->OwnsUserScript(
+      guest->embedder_web_contents()->GetRenderProcessHost()->GetID(),
+      guest->view_instance_id(), script_id);
+}
+
 void GuestViewMessageFilter::MimeHandlerViewGuestCreatedCallback(
     int element_instance_id,
     int embedder_render_process_id,