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" |
| 16 #include "extensions/browser/extensions_browser_client.h" |
15 #include "extensions/browser/mojo/service_registration.h" | 17 #include "extensions/browser/mojo/service_registration.h" |
16 #include "extensions/browser/view_type_utils.h" | 18 #include "extensions/browser/view_type_utils.h" |
17 #include "extensions/common/constants.h" | 19 #include "extensions/common/constants.h" |
18 #include "extensions/common/extension_messages.h" | 20 #include "extensions/common/extension_messages.h" |
19 | 21 |
20 namespace extensions { | 22 namespace extensions { |
| 23 namespace { |
| 24 |
| 25 const Extension* GetExtensionForRenderFrame( |
| 26 content::RenderFrameHost* render_frame_host) { |
| 27 content::SiteInstance* site_instance = render_frame_host->GetSiteInstance(); |
| 28 GURL url = render_frame_host->GetLastCommittedURL(); |
| 29 if (!url.is_empty()) { |
| 30 if (site_instance->GetSiteURL().GetOrigin() != url.GetOrigin()) |
| 31 return nullptr; |
| 32 } else { |
| 33 url = site_instance->GetSiteURL(); |
| 34 } |
| 35 content::BrowserContext* browser_context = site_instance->GetBrowserContext(); |
| 36 if (!url.SchemeIs(kExtensionScheme)) |
| 37 return nullptr; |
| 38 |
| 39 return ExtensionRegistry::Get(browser_context) |
| 40 ->enabled_extensions() |
| 41 .GetExtensionOrAppByURL(url); |
| 42 } |
| 43 |
| 44 } // namespace |
21 | 45 |
22 ExtensionWebContentsObserver::ExtensionWebContentsObserver( | 46 ExtensionWebContentsObserver::ExtensionWebContentsObserver( |
23 content::WebContents* web_contents) | 47 content::WebContents* web_contents) |
24 : content::WebContentsObserver(web_contents), | 48 : content::WebContentsObserver(web_contents), |
25 browser_context_(web_contents->GetBrowserContext()) { | 49 browser_context_(web_contents->GetBrowserContext()) { |
26 NotifyRenderViewType(web_contents->GetRenderViewHost()); | 50 NotifyRenderViewType(web_contents->GetRenderViewHost()); |
27 } | 51 } |
28 | 52 |
29 ExtensionWebContentsObserver::~ExtensionWebContentsObserver() {} | 53 ExtensionWebContentsObserver::~ExtensionWebContentsObserver() {} |
30 | 54 |
(...skipping 54 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
85 case Manifest::TYPE_SHARED_MODULE: | 109 case Manifest::TYPE_SHARED_MODULE: |
86 break; | 110 break; |
87 | 111 |
88 case Manifest::NUM_LOAD_TYPES: | 112 case Manifest::NUM_LOAD_TYPES: |
89 NOTREACHED(); | 113 NOTREACHED(); |
90 } | 114 } |
91 } | 115 } |
92 | 116 |
93 void ExtensionWebContentsObserver::RenderFrameCreated( | 117 void ExtensionWebContentsObserver::RenderFrameCreated( |
94 content::RenderFrameHost* render_frame_host) { | 118 content::RenderFrameHost* render_frame_host) { |
95 RegisterCoreExtensionServices(render_frame_host); | 119 const Extension* extension = GetExtensionForRenderFrame(render_frame_host); |
| 120 if (extension) { |
| 121 ExtensionsBrowserClient::Get()->RegisterMojoServices(render_frame_host, |
| 122 extension); |
| 123 } |
96 } | 124 } |
97 | 125 |
98 void ExtensionWebContentsObserver::NotifyRenderViewType( | 126 void ExtensionWebContentsObserver::NotifyRenderViewType( |
99 content::RenderViewHost* render_view_host) { | 127 content::RenderViewHost* render_view_host) { |
100 if (render_view_host) { | 128 if (render_view_host) { |
101 render_view_host->Send(new ExtensionMsg_NotifyRenderViewType( | 129 render_view_host->Send(new ExtensionMsg_NotifyRenderViewType( |
102 render_view_host->GetRoutingID(), GetViewType(web_contents()))); | 130 render_view_host->GetRoutingID(), GetViewType(web_contents()))); |
103 } | 131 } |
104 } | 132 } |
105 | 133 |
(...skipping 17 matching lines...) Expand all Loading... |
123 // site, so we can ignore that wrinkle here. | 151 // site, so we can ignore that wrinkle here. |
124 const GURL& site = render_view_host->GetSiteInstance()->GetSiteURL(); | 152 const GURL& site = render_view_host->GetSiteInstance()->GetSiteURL(); |
125 | 153 |
126 if (!site.SchemeIs(kExtensionScheme)) | 154 if (!site.SchemeIs(kExtensionScheme)) |
127 return std::string(); | 155 return std::string(); |
128 | 156 |
129 return site.host(); | 157 return site.host(); |
130 } | 158 } |
131 | 159 |
132 } // namespace extensions | 160 } // namespace extensions |
OLD | NEW |