Merge 32634 - Make executeScript and insertCSS inject code into all frames....

chrome/renderer/render_view.cc

 // Copyright (c) 2009 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.
 
 #if defined(ENABLE_PEPPER)
 #define PEPPER_APIS_ENABLED
 #endif
 
 #include "chrome/renderer/render_view.h"
 
@@ skipping 2308 matching lines @@
     RenderThread::current()->user_script_slave()->InjectScripts(
         frame, UserScript::DOCUMENT_IDLE);
   }
 
   WebFrame* main_frame = webview()->mainFrame();
   if (frame == main_frame) {
     while (!pending_code_execution_queue_.empty()) {
       scoped_refptr<CodeExecutionInfo> info =
           pending_code_execution_queue_.front();
       ExecuteCodeImpl(main_frame, info->request_id, info->extension_id,
-                      info->is_js_code, info->code_string);
+                      info->is_js_code, info->code_string, info->all_frames);
       pending_code_execution_queue_.pop();
     }
   }
 }
 
 void RenderView::didHandleOnloadEvents(WebFrame* frame) {
   // Ignore
 }
 
 void RenderView::didFailLoad(WebFrame* frame, const WebURLError& error) {
@@ skipping 617 matching lines @@
 
 void RenderView::OnSetPageEncoding(const std::string& encoding_name) {
   webview()->setPageEncoding(WebString::fromUTF8(encoding_name));
 }
 
 void RenderView::OnResetPageEncodingToDefault() {
   WebString no_encoding;
   webview()->setPageEncoding(no_encoding);
 }
 
+bool RenderView::GetAllChildFrames(
+    WebFrame* parent_frame,
+    std::vector<WebFrame*>* frames_vector) const {
+  if (!parent_frame)
+    return false;
+  for (WebFrame* child_frame = parent_frame->firstChild(); child_frame;
+       child_frame = child_frame->nextSibling()) {
+    frames_vector->push_back(child_frame);
+    GetAllChildFrames(child_frame, frames_vector);
+  }
+  return true;
+}
+
 WebFrame* RenderView::GetChildFrame(const std::wstring& xpath) const {
   if (xpath.empty())
     return webview()->mainFrame();
 
   // xpath string can represent a frame deep down the tree (across multiple
   // frame DOMs).
   // Example, /html/body/table/tbody/tr/td/iframe\n/frameset/frame[0]
   // should break into 2 xpaths
   // /html/body/table/tbody/tr/td/iframe & /frameset/frame[0]
 
@@ skipping 761 matching lines @@
   print_helper_->Print(frame, script_initiated);
 }
 
 void RenderView::OnSetEditCommandsForNextKeyEvent(
     const EditCommands& edit_commands) {
   edit_commands_ = edit_commands;
 }
 
 void RenderView::OnExecuteCode(int request_id, const std::string& extension_id,
                                bool is_js_code,
-                               const std::string& code_string) {
+                               const std::string& code_string,
+                               bool all_frames) {
   WebFrame* main_frame = webview() ? webview()->mainFrame() : NULL;
   if (!main_frame) {
     Send(new ViewMsg_ExecuteCodeFinished(routing_id_, request_id, false));
     return;
   }
 
   WebDataSource* ds = main_frame->dataSource();
   NavigationState* navigation_state = NavigationState::FromDataSource(ds);
   if (!navigation_state->user_script_idle_scheduler()->has_run()) {
     scoped_refptr<CodeExecutionInfo> info = new CodeExecutionInfo(
-        request_id, extension_id, is_js_code, code_string);
+        request_id, extension_id, is_js_code, code_string, all_frames);
     pending_code_execution_queue_.push(info);
     return;
   }
 
   ExecuteCodeImpl(main_frame, request_id, extension_id, is_js_code,
-                  code_string);
+                  code_string, all_frames);
 }
 
 void RenderView::ExecuteCodeImpl(WebFrame* frame,
                                  int request_id,
                                  const std::string& extension_id,
                                  bool is_js_code,
-                                 const std::string& code_string) {
-  if (is_js_code) {
-    std::vector<WebScriptSource> sources;
-    sources.push_back(
-        WebScriptSource(WebString::fromUTF8(code_string)));
-    UserScriptSlave::InsertInitExtensionCode(&sources, extension_id);
-    frame->executeScriptInIsolatedWorld(
-        UserScriptSlave::GetIsolatedWorldId(extension_id),
-        &sources.front(), sources.size(), EXTENSION_GROUP_CONTENT_SCRIPTS);
-  } else {
-    frame->insertStyleText(WebString::fromUTF8(code_string), WebString());
+                                 const std::string& code_string,
+                                 bool all_frames) {
+  std::vector<WebFrame*> frame_vector;
+  frame_vector.push_back(frame);
+  if (all_frames)
+    GetAllChildFrames(frame, &frame_vector);
+
+  for (std::vector<WebFrame*>::iterator frame_it = frame_vector.begin();
+       frame_it != frame_vector.end(); ++frame_it) {
+    WebFrame* frame = *frame_it;
+    if (is_js_code) {
+      std::vector<WebScriptSource> sources;
+      sources.push_back(
+          WebScriptSource(WebString::fromUTF8(code_string)));
+      UserScriptSlave::InsertInitExtensionCode(&sources, extension_id);
+      frame->executeScriptInIsolatedWorld(
+          UserScriptSlave::GetIsolatedWorldId(extension_id),
+          &sources.front(), sources.size(), EXTENSION_GROUP_CONTENT_SCRIPTS);
+    } else {
+      frame->insertStyleText(WebString::fromUTF8(code_string), WebString());
+    }
   }
 
   Send(new ViewMsg_ExecuteCodeFinished(routing_id_, request_id, true));
 }
 
 void RenderView::Close() {
   // We need to grab a pointer to the doomed WebView before we destroy it.
   WebView* doomed = webview();
   RenderWidget::Close();
   Singleton<ViewMap>::get()->erase(doomed);
@@ skipping 20 matching lines @@
       new PluginMsg_SignalModalDialogEvent(host_window_));
 
   message->EnableMessagePumping();  // Runs a nested message loop.
   bool rv = Send(message);
 
   PluginChannelHost::Broadcast(
       new PluginMsg_ResetModalDialogEvent(host_window_));
 
   return rv;
 }