extension_web_contents_observer.cc revision cedac228d2dd51db4b79ea1e72c7f249408ee061 
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 "extensions/browser/extension_web_contents_observer.h"
6
7#include "content/public/browser/child_process_security_policy.h"
8#include "content/public/browser/render_process_host.h"
9#include "content/public/browser/render_view_host.h"
10#include "content/public/browser/site_instance.h"
11#include "content/public/browser/web_contents.h"
12#include "content/public/common/url_constants.h"
13#include "extensions/browser/extension_prefs.h"
14#include "extensions/browser/extension_registry.h"
15#include "extensions/browser/view_type_utils.h"
16#include "extensions/common/constants.h"
17#include "extensions/common/extension_messages.h"
18
19namespace extensions {
20
21ExtensionWebContentsObserver::ExtensionWebContentsObserver(
22    content::WebContents* web_contents)
23    : content::WebContentsObserver(web_contents),
24      browser_context_(web_contents->GetBrowserContext()) {
25  NotifyRenderViewType(web_contents->GetRenderViewHost());
26}
27
28ExtensionWebContentsObserver::~ExtensionWebContentsObserver() {}
29
30void ExtensionWebContentsObserver::RenderViewCreated(
31    content::RenderViewHost* render_view_host) {
32  NotifyRenderViewType(render_view_host);
33
34  const Extension* extension = GetExtension(render_view_host);
35  if (!extension)
36    return;
37
38  content::RenderProcessHost* process = render_view_host->GetProcess();
39
40  // Some extensions use chrome:// URLs.
41  // This is a temporary solution. Replace it with access to chrome-static://
42  // once it is implemented. See: crbug.com/226927.
43  Manifest::Type type = extension->GetType();
44  if (type == Manifest::TYPE_EXTENSION ||
45      type == Manifest::TYPE_LEGACY_PACKAGED_APP ||
46      (type == Manifest::TYPE_PLATFORM_APP &&
47       extension->location() == Manifest::COMPONENT)) {
48    content::ChildProcessSecurityPolicy::GetInstance()->GrantScheme(
49        process->GetID(), content::kChromeUIScheme);
50  }
51
52  // Some extensions use file:// URLs.
53  if (type == Manifest::TYPE_EXTENSION ||
54      type == Manifest::TYPE_LEGACY_PACKAGED_APP) {
55    ExtensionPrefs* prefs = ExtensionPrefs::Get(browser_context_);
56    if (prefs->AllowFileAccess(extension->id())) {
57      content::ChildProcessSecurityPolicy::GetInstance()->GrantScheme(
58          process->GetID(), url::kFileScheme);
59    }
60  }
61
62  switch (type) {
63    case Manifest::TYPE_EXTENSION:
64    case Manifest::TYPE_USER_SCRIPT:
65    case Manifest::TYPE_HOSTED_APP:
66    case Manifest::TYPE_LEGACY_PACKAGED_APP:
67    case Manifest::TYPE_PLATFORM_APP:
68      // Always send a Loaded message before ActivateExtension so that
69      // ExtensionDispatcher knows what Extension is active, not just its ID.
70      // This is important for classifying the Extension's JavaScript context
71      // correctly (see ExtensionDispatcher::ClassifyJavaScriptContext).
72      render_view_host->Send(
73          new ExtensionMsg_Loaded(std::vector<ExtensionMsg_Loaded_Params>(
74              1, ExtensionMsg_Loaded_Params(extension))));
75      render_view_host->Send(
76          new ExtensionMsg_ActivateExtension(extension->id()));
77      break;
78
79    case Manifest::TYPE_UNKNOWN:
80    case Manifest::TYPE_THEME:
81    case Manifest::TYPE_SHARED_MODULE:
82      break;
83
84    case Manifest::NUM_LOAD_TYPES:
85      NOTREACHED();
86  }
87}
88
89void ExtensionWebContentsObserver::NotifyRenderViewType(
90    content::RenderViewHost* render_view_host) {
91  if (render_view_host) {
92    render_view_host->Send(new ExtensionMsg_NotifyRenderViewType(
93        render_view_host->GetRoutingID(), GetViewType(web_contents())));
94  }
95}
96
97const Extension* ExtensionWebContentsObserver::GetExtension(
98    content::RenderViewHost* render_view_host) {
99  std::string extension_id = GetExtensionId(render_view_host);
100  if (extension_id.empty())
101    return NULL;
102
103  // May be null if the extension doesn't exist, for example if somebody typos
104  // a chrome-extension:// URL.
105  return ExtensionRegistry::Get(browser_context_)
106      ->GetExtensionById(extension_id, ExtensionRegistry::ENABLED);
107}
108
109// static
110std::string ExtensionWebContentsObserver::GetExtensionId(
111    content::RenderViewHost* render_view_host) {
112  // Note that due to ChromeContentBrowserClient::GetEffectiveURL(), hosted apps
113  // (excluding bookmark apps) will have a chrome-extension:// URL for their
114  // site, so we can ignore that wrinkle here.
115  const GURL& site = render_view_host->GetSiteInstance()->GetSiteURL();
116
117  if (!site.SchemeIs(kExtensionScheme))
118    return std::string();
119
120  return site.host();
121}
122
123}  // namespace extensions
124