Give automation API access to browser plugin objects.

chrome/browser/extensions/api/automation_internal/automation_util.cc

 // Copyright 2014 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 "chrome/browser/extensions/api/automation_internal/automation_util.h"
 
 #include <string>
 #include <utility>
 
 #include "base/values.h"
 #include "chrome/browser/accessibility/ax_tree_id_registry.h"
 #include "chrome/common/extensions/api/automation_internal.h"
+#include "content/public/browser/browser_context.h"
+#include "content/public/browser/browser_plugin_guest_manager.h"
+#include "content/public/browser/render_frame_host.h"
+#include "content/public/browser/render_process_host.h"
+#include "content/public/browser/web_contents.h"
 #include "extensions/browser/event_router.h"
 #include "ui/accessibility/ax_enums.h"
 #include "ui/accessibility/ax_node_data.h"
 
 namespace extensions {
 
 namespace {
 
 void PopulateNodeData(const ui::AXNodeData& node_data,
     linked_ptr< api::automation_internal::AXNodeData>& out_node_data) {
@@ skipping 122 matching lines @@
     ax_event_params.target_id = event.id;
 
     AXTreeUpdate& ax_tree_update = ax_event_params.update;
     ax_tree_update.node_id_to_clear = event.node_id_to_clear;
     for (size_t i = 0; i < event.nodes.size(); ++i) {
       ui::AXNodeData src = event.nodes[i];
       src.location.Offset(location_offset);
       linked_ptr<api::automation_internal::AXNodeData> out_node(
           new api::automation_internal::AXNodeData());
       PopulateNodeData(src, out_node);
+      if (src.HasBoolAttribute(ui::AX_ATTR_IS_AX_TREE_HOST)) {
+        const auto& iter = event.node_to_browser_plugin_instance_id_map.find(
+            src.id);
+        if (iter != event.node_to_browser_plugin_instance_id_map.end()) {
+          int instance_id = iter->second;
+          content::BrowserPluginGuestManager* guest_manager =
+              browser_context->GetGuestManager();
+          content::WebContents* guest_web_contents =
+              guest_manager->GetGuestByInstanceID(event.process_id,
+                                                  instance_id);
+          if (guest_web_contents) {
+            content::RenderFrameHost* guest_rfh =
+                guest_web_contents->GetMainFrame();
+            int guest_tree_id =
+                AXTreeIDRegistry::GetInstance()->GetOrCreateAXTreeID(
+                    guest_rfh->GetProcess()->GetID(),
+                    guest_rfh->GetRoutingID());
+            out_node->int_attributes->additional_properties.SetInteger(
+                ToString(ui::AX_ATTR_CHILD_TREE_ID),
+                guest_tree_id);
+          }
+        }
+      }
       ax_tree_update.nodes.push_back(out_node);
     }
 
     // TODO(dtseng/aboxhall): Why are we sending only one update at a time? We
     // should match the behavior from renderer -> browser and send a
     // collection of tree updates over (to the extension); see
     // |AccessibilityHostMsg_EventParams| and |AccessibilityHostMsg_Events|.
     DispatchEventInternal(
         browser_context,
         api::automation_internal::OnAccessibilityEvent::kEventName,
@@ skipping 11 matching lines @@
   DispatchEventInternal(
       browser_context,
       api::automation_internal::OnAccessibilityTreeDestroyed::kEventName,
       api::automation_internal::OnAccessibilityTreeDestroyed::Create(tree_id));
   AXTreeIDRegistry::GetInstance()->RemoveAXTreeID(tree_id);
 }
 
 }  // namespace automation_util
 
 }  // namespace extensions