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
19 namespace extensions {
20
ExtensionWebContentsObserver(content::WebContents * web_contents)21 ExtensionWebContentsObserver::ExtensionWebContentsObserver(
22 content::WebContents* web_contents)
23 : content::WebContentsObserver(web_contents),
24 browser_context_(web_contents->GetBrowserContext()) {
25 NotifyRenderViewType(web_contents->GetRenderViewHost());
26 }
27
~ExtensionWebContentsObserver()28 ExtensionWebContentsObserver::~ExtensionWebContentsObserver() {}
29
RenderViewCreated(content::RenderViewHost * render_view_host)30 void 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
NotifyRenderViewType(content::RenderViewHost * render_view_host)89 void 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
GetExtension(content::RenderViewHost * render_view_host)97 const 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
GetExtensionId(content::RenderViewHost * render_view_host)110 std::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