1// Copyright 2014 The Chromium Authors. All rights reserved.
2// Use of this source code is governed by a BSD-style license that can be
3// found in the LICENSE file.
4
5#include "chrome/browser/task_manager/extension_information.h"
6
7#include "base/strings/string16.h"
8#include "base/strings/utf_string_conversions.h"
9#include "chrome/browser/browser_process.h"
10#include "chrome/browser/profiles/profile.h"
11#include "chrome/browser/profiles/profile_manager.h"
12#include "chrome/browser/task_manager/renderer_resource.h"
13#include "chrome/browser/task_manager/task_manager_util.h"
14#include "content/public/browser/render_frame_host.h"
15#include "content/public/browser/render_process_host.h"
16#include "content/public/browser/render_view_host.h"
17#include "content/public/browser/web_contents.h"
18#include "extensions/browser/extension_system.h"
19#include "extensions/browser/guest_view/guest_view_base.h"
20#include "extensions/browser/process_manager.h"
21#include "extensions/browser/view_type_utils.h"
22#include "extensions/common/extension.h"
23#include "grit/theme_resources.h"
24#include "ui/base/l10n/l10n_util.h"
25#include "ui/base/resource/resource_bundle.h"
26#include "ui/gfx/image/image_skia.h"
27
28using content::WebContents;
29using extensions::Extension;
30
31namespace {
32
33const Extension* GetExtensionForWebContents(WebContents* web_contents) {
34  Profile* profile =
35      Profile::FromBrowserContext(web_contents->GetBrowserContext());
36  extensions::ProcessManager* process_manager =
37      extensions::ExtensionSystem::Get(profile)->process_manager();
38  return process_manager->GetExtensionForRenderViewHost(
39      web_contents->GetRenderViewHost());
40}
41
42}  // namespace
43
44namespace task_manager {
45
46class ExtensionProcessResource : public RendererResource {
47 public:
48  explicit ExtensionProcessResource(const Extension* extension,
49                                    content::RenderViewHost* render_view_host);
50  virtual ~ExtensionProcessResource();
51
52  // Resource methods:
53  virtual base::string16 GetTitle() const OVERRIDE;
54  virtual gfx::ImageSkia GetIcon() const OVERRIDE;
55  virtual Type GetType() const OVERRIDE;
56
57 private:
58  // Returns true if the associated extension has a background page.
59  bool IsBackground() const;
60
61  // The icon painted for the extension process.
62  static gfx::ImageSkia* default_icon_;
63
64  // Cached data about the extension.
65  base::string16 title_;
66
67  DISALLOW_COPY_AND_ASSIGN(ExtensionProcessResource);
68};
69
70gfx::ImageSkia* ExtensionProcessResource::default_icon_ = NULL;
71
72ExtensionProcessResource::ExtensionProcessResource(
73    const Extension* extension,
74    content::RenderViewHost* render_view_host)
75    : RendererResource(render_view_host->GetProcess()->GetHandle(),
76                       render_view_host) {
77  if (!default_icon_) {
78    ResourceBundle& rb = ResourceBundle::GetSharedInstance();
79    default_icon_ = rb.GetImageSkiaNamed(IDR_PLUGINS_FAVICON);
80  }
81  base::string16 extension_name = base::UTF8ToUTF16(extension->name());
82  DCHECK(!extension_name.empty());
83
84  Profile* profile = Profile::FromBrowserContext(
85      render_view_host->GetProcess()->GetBrowserContext());
86  int message_id = util::GetMessagePrefixID(extension->is_app(),
87                                            true,  // is_extension
88                                            profile->IsOffTheRecord(),
89                                            false,  // is_prerender
90                                            IsBackground());
91  title_ = l10n_util::GetStringFUTF16(message_id, extension_name);
92}
93
94ExtensionProcessResource::~ExtensionProcessResource() {}
95
96base::string16 ExtensionProcessResource::GetTitle() const { return title_; }
97
98gfx::ImageSkia ExtensionProcessResource::GetIcon() const {
99  return *default_icon_;
100}
101
102Resource::Type ExtensionProcessResource::GetType() const { return EXTENSION; }
103
104bool ExtensionProcessResource::IsBackground() const {
105  WebContents* web_contents =
106      WebContents::FromRenderViewHost(render_view_host());
107  extensions::ViewType view_type = extensions::GetViewType(web_contents);
108  return view_type == extensions::VIEW_TYPE_EXTENSION_BACKGROUND_PAGE;
109}
110
111////////////////////////////////////////////////////////////////////////////////
112// ExtensionInformation class
113////////////////////////////////////////////////////////////////////////////////
114
115ExtensionInformation::ExtensionInformation() {}
116
117ExtensionInformation::~ExtensionInformation() {}
118
119void ExtensionInformation::GetAll(const NewWebContentsCallback& callback) {
120  // Add all the existing extension views from all Profiles, including those
121  // from incognito split mode.
122  ProfileManager* profile_manager = g_browser_process->profile_manager();
123  std::vector<Profile*> profiles(profile_manager->GetLoadedProfiles());
124  size_t num_default_profiles = profiles.size();
125  for (size_t i = 0; i < num_default_profiles; ++i) {
126    if (profiles[i]->HasOffTheRecordProfile()) {
127      profiles.push_back(profiles[i]->GetOffTheRecordProfile());
128    }
129  }
130
131  for (size_t i = 0; i < profiles.size(); ++i) {
132    extensions::ProcessManager* process_manager =
133        extensions::ExtensionSystem::Get(profiles[i])->process_manager();
134    if (process_manager) {
135      const extensions::ProcessManager::ViewSet all_views =
136          process_manager->GetAllViews();
137      extensions::ProcessManager::ViewSet::const_iterator jt =
138          all_views.begin();
139      for (; jt != all_views.end(); ++jt) {
140        WebContents* web_contents = WebContents::FromRenderViewHost(*jt);
141        if (CheckOwnership(web_contents))
142          callback.Run(web_contents);
143      }
144    }
145  }
146}
147
148bool ExtensionInformation::CheckOwnership(WebContents* web_contents) {
149  // Don't add WebContents that belong to a guest (those are handled by
150  // GuestInformation). Otherwise they will be added twice.
151  if (extensions::GuestViewBase::IsGuest(web_contents))
152    return false;
153
154  // Extension WebContents tracked by this class will always host extension
155  // content.
156  if (!GetExtensionForWebContents(web_contents))
157    return false;
158
159  extensions::ViewType view_type = extensions::GetViewType(web_contents);
160
161  // Don't add tab contents (those are handled by TabContentsResourceProvider)
162  // or background contents (handled by BackgroundInformation) or panels
163  // (handled by PanelInformation)
164  return (view_type != extensions::VIEW_TYPE_INVALID &&
165          view_type != extensions::VIEW_TYPE_TAB_CONTENTS &&
166          view_type != extensions::VIEW_TYPE_BACKGROUND_CONTENTS &&
167          view_type != extensions::VIEW_TYPE_PANEL);
168}
169
170scoped_ptr<RendererResource> ExtensionInformation::MakeResource(
171    WebContents* web_contents) {
172  const Extension* extension = GetExtensionForWebContents(web_contents);
173  if (!extension) {
174    NOTREACHED();
175    return scoped_ptr<RendererResource>();
176  }
177  return scoped_ptr<RendererResource>(new ExtensionProcessResource(
178      extension, web_contents->GetRenderViewHost()));
179}
180
181}  // namespace task_manager
182