OLD | NEW |
---|---|
1 // Copyright 2014 The Chromium Authors. All rights reserved. | 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 | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #include "extensions/browser/extension_web_contents_observer.h" | 5 #include "extensions/browser/extension_web_contents_observer.h" |
6 | 6 |
7 #include "content/public/browser/child_process_security_policy.h" | 7 #include "content/public/browser/child_process_security_policy.h" |
8 #include "content/public/browser/render_frame_host.h" | |
8 #include "content/public/browser/render_process_host.h" | 9 #include "content/public/browser/render_process_host.h" |
9 #include "content/public/browser/render_view_host.h" | 10 #include "content/public/browser/render_view_host.h" |
10 #include "content/public/browser/site_instance.h" | 11 #include "content/public/browser/site_instance.h" |
11 #include "content/public/browser/web_contents.h" | 12 #include "content/public/browser/web_contents.h" |
12 #include "content/public/common/url_constants.h" | 13 #include "content/public/common/url_constants.h" |
13 #include "extensions/browser/extension_prefs.h" | 14 #include "extensions/browser/extension_prefs.h" |
14 #include "extensions/browser/extension_registry.h" | 15 #include "extensions/browser/extension_registry.h" |
15 #include "extensions/browser/mojo/service_registration.h" | 16 #include "extensions/browser/mojo/service_registration.h" |
16 #include "extensions/browser/view_type_utils.h" | 17 #include "extensions/browser/view_type_utils.h" |
17 #include "extensions/common/constants.h" | 18 #include "extensions/common/constants.h" |
18 #include "extensions/common/extension_messages.h" | 19 #include "extensions/common/extension_messages.h" |
19 | 20 |
20 namespace extensions { | 21 namespace extensions { |
22 namespace { | |
23 | |
24 const Extension* GetExtensionForRenderFrame( | |
mark a. foltz
2015/03/27 20:24:11
Is there no other way to map the render_frame_host
Kevin M
2015/03/27 22:42:37
I don't think so... the creator of the observer do
Sam McNally
2015/03/31 23:38:51
I don't think there's a way to do this as a simple
| |
25 content::RenderFrameHost* render_frame_host) { | |
26 content::SiteInstance* site_instance = render_frame_host->GetSiteInstance(); | |
27 GURL url = render_frame_host->GetLastCommittedURL(); | |
28 if (!url.is_empty()) { | |
29 if (site_instance->GetSiteURL().GetOrigin() != url.GetOrigin()) | |
30 return nullptr; | |
31 } else { | |
32 url = site_instance->GetSiteURL(); | |
33 } | |
34 content::BrowserContext* browser_context = site_instance->GetBrowserContext(); | |
35 if (!url.SchemeIs(kExtensionScheme)) | |
36 return nullptr; | |
37 | |
38 return ExtensionRegistry::Get(browser_context) | |
39 ->enabled_extensions() | |
40 .GetExtensionOrAppByURL(url); | |
41 } | |
42 | |
43 } // namespace | |
21 | 44 |
22 ExtensionWebContentsObserver::ExtensionWebContentsObserver( | 45 ExtensionWebContentsObserver::ExtensionWebContentsObserver( |
23 content::WebContents* web_contents) | 46 content::WebContents* web_contents) |
24 : content::WebContentsObserver(web_contents), | 47 : content::WebContentsObserver(web_contents), |
25 browser_context_(web_contents->GetBrowserContext()) { | 48 browser_context_(web_contents->GetBrowserContext()) { |
26 NotifyRenderViewType(web_contents->GetRenderViewHost()); | 49 NotifyRenderViewType(web_contents->GetRenderViewHost()); |
27 } | 50 } |
28 | 51 |
29 ExtensionWebContentsObserver::~ExtensionWebContentsObserver() {} | 52 ExtensionWebContentsObserver::~ExtensionWebContentsObserver() {} |
30 | 53 |
(...skipping 54 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... | |
85 case Manifest::TYPE_SHARED_MODULE: | 108 case Manifest::TYPE_SHARED_MODULE: |
86 break; | 109 break; |
87 | 110 |
88 case Manifest::NUM_LOAD_TYPES: | 111 case Manifest::NUM_LOAD_TYPES: |
89 NOTREACHED(); | 112 NOTREACHED(); |
90 } | 113 } |
91 } | 114 } |
92 | 115 |
93 void ExtensionWebContentsObserver::RenderFrameCreated( | 116 void ExtensionWebContentsObserver::RenderFrameCreated( |
94 content::RenderFrameHost* render_frame_host) { | 117 content::RenderFrameHost* render_frame_host) { |
95 RegisterCoreExtensionServices(render_frame_host); | 118 const Extension* extension = GetExtensionForRenderFrame(render_frame_host); |
119 if (extension) { | |
120 RegisterMojoServices(render_frame_host, | |
121 GetExtensionForRenderFrame(render_frame_host)); | |
122 } | |
96 } | 123 } |
97 | 124 |
98 void ExtensionWebContentsObserver::NotifyRenderViewType( | 125 void ExtensionWebContentsObserver::NotifyRenderViewType( |
99 content::RenderViewHost* render_view_host) { | 126 content::RenderViewHost* render_view_host) { |
100 if (render_view_host) { | 127 if (render_view_host) { |
101 render_view_host->Send(new ExtensionMsg_NotifyRenderViewType( | 128 render_view_host->Send(new ExtensionMsg_NotifyRenderViewType( |
102 render_view_host->GetRoutingID(), GetViewType(web_contents()))); | 129 render_view_host->GetRoutingID(), GetViewType(web_contents()))); |
103 } | 130 } |
104 } | 131 } |
105 | 132 |
(...skipping 16 matching lines...) Expand all Loading... | |
122 // (excluding bookmark apps) will have a chrome-extension:// URL for their | 149 // (excluding bookmark apps) will have a chrome-extension:// URL for their |
123 // site, so we can ignore that wrinkle here. | 150 // site, so we can ignore that wrinkle here. |
124 const GURL& site = render_view_host->GetSiteInstance()->GetSiteURL(); | 151 const GURL& site = render_view_host->GetSiteInstance()->GetSiteURL(); |
125 | 152 |
126 if (!site.SchemeIs(kExtensionScheme)) | 153 if (!site.SchemeIs(kExtensionScheme)) |
127 return std::string(); | 154 return std::string(); |
128 | 155 |
129 return site.host(); | 156 return site.host(); |
130 } | 157 } |
131 | 158 |
159 void ExtensionWebContentsObserver::RegisterMojoServices( | |
160 content::RenderFrameHost* render_frame_host, | |
161 const Extension* extension) { | |
162 DCHECK(render_frame_host); | |
163 DCHECK(extension); | |
164 | |
165 RegisterServicesForFrame(render_frame_host, extension); | |
166 } | |
167 | |
132 } // namespace extensions | 168 } // namespace extensions |
OLD | NEW |