| 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_process_host.h" | 8 #include "content/public/browser/render_process_host.h" |
| 9 #include "content/public/browser/render_view_host.h" | 9 #include "content/public/browser/render_view_host.h" |
| 10 #include "content/public/browser/site_instance.h" | 10 #include "content/public/browser/site_instance.h" |
| 11 #include "content/public/browser/web_contents.h" | 11 #include "content/public/browser/web_contents.h" |
| 12 #include "content/public/common/url_constants.h" | 12 #include "content/public/common/url_constants.h" |
| 13 #include "extensions/browser/extension_prefs.h" | 13 #include "extensions/browser/extension_prefs.h" |
| 14 #include "extensions/browser/extension_registry.h" | 14 #include "extensions/browser/extension_registry.h" |
| 15 #include "extensions/browser/mojo/service_registration_manager.h" | 15 #include "extensions/browser/mojo/service_registration.h" |
| 16 #include "extensions/browser/view_type_utils.h" | 16 #include "extensions/browser/view_type_utils.h" |
| 17 #include "extensions/common/constants.h" | 17 #include "extensions/common/constants.h" |
| 18 #include "extensions/common/extension_messages.h" | 18 #include "extensions/common/extension_messages.h" |
| 19 | 19 |
| 20 namespace extensions { | 20 namespace extensions { |
| 21 | 21 |
| 22 ExtensionWebContentsObserver::ExtensionWebContentsObserver( | 22 ExtensionWebContentsObserver::ExtensionWebContentsObserver( |
| 23 content::WebContents* web_contents) | 23 content::WebContents* web_contents) |
| 24 : content::WebContentsObserver(web_contents), | 24 : content::WebContentsObserver(web_contents), |
| 25 browser_context_(web_contents->GetBrowserContext()) { | 25 browser_context_(web_contents->GetBrowserContext()) { |
| 26 NotifyRenderViewType(web_contents->GetRenderViewHost()); | 26 NotifyRenderViewType(web_contents->GetRenderViewHost()); |
| 27 } | 27 } |
| 28 | 28 |
| 29 ExtensionWebContentsObserver::~ExtensionWebContentsObserver() {} | 29 ExtensionWebContentsObserver::~ExtensionWebContentsObserver() {} |
| 30 | 30 |
| 31 void ExtensionWebContentsObserver::RenderViewCreated( | 31 void ExtensionWebContentsObserver::RenderViewCreated( |
| 32 content::RenderViewHost* render_view_host) { | 32 content::RenderViewHost* render_view_host) { |
| 33 NotifyRenderViewType(render_view_host); | 33 NotifyRenderViewType(render_view_host); |
| 34 | 34 |
| 35 // TODO(sammc): Call AddServicesToRenderFrame() for frames that aren't main | |
| 36 // frames. | |
| 37 ServiceRegistrationManager::GetSharedInstance()->AddServicesToRenderFrame( | |
| 38 render_view_host->GetMainFrame()); | |
| 39 | |
| 40 const Extension* extension = GetExtension(render_view_host); | 35 const Extension* extension = GetExtension(render_view_host); |
| 41 if (!extension) | 36 if (!extension) |
| 42 return; | 37 return; |
| 43 | 38 |
| 44 content::RenderProcessHost* process = render_view_host->GetProcess(); | 39 content::RenderProcessHost* process = render_view_host->GetProcess(); |
| 45 | 40 |
| 46 // Some extensions use chrome:// URLs. | 41 // Some extensions use chrome:// URLs. |
| 47 // This is a temporary solution. Replace it with access to chrome-static:// | 42 // This is a temporary solution. Replace it with access to chrome-static:// |
| 48 // once it is implemented. See: crbug.com/226927. | 43 // once it is implemented. See: crbug.com/226927. |
| 49 Manifest::Type type = extension->GetType(); | 44 Manifest::Type type = extension->GetType(); |
| (...skipping 38 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 88 case Manifest::TYPE_UNKNOWN: | 83 case Manifest::TYPE_UNKNOWN: |
| 89 case Manifest::TYPE_THEME: | 84 case Manifest::TYPE_THEME: |
| 90 case Manifest::TYPE_SHARED_MODULE: | 85 case Manifest::TYPE_SHARED_MODULE: |
| 91 break; | 86 break; |
| 92 | 87 |
| 93 case Manifest::NUM_LOAD_TYPES: | 88 case Manifest::NUM_LOAD_TYPES: |
| 94 NOTREACHED(); | 89 NOTREACHED(); |
| 95 } | 90 } |
| 96 } | 91 } |
| 97 | 92 |
| 93 void ExtensionWebContentsObserver::RenderFrameCreated( |
| 94 content::RenderFrameHost* render_frame_host) { |
| 95 RegisterCoreExtensionServices(render_frame_host); |
| 96 } |
| 97 |
| 98 void ExtensionWebContentsObserver::NotifyRenderViewType( | 98 void ExtensionWebContentsObserver::NotifyRenderViewType( |
| 99 content::RenderViewHost* render_view_host) { | 99 content::RenderViewHost* render_view_host) { |
| 100 if (render_view_host) { | 100 if (render_view_host) { |
| 101 render_view_host->Send(new ExtensionMsg_NotifyRenderViewType( | 101 render_view_host->Send(new ExtensionMsg_NotifyRenderViewType( |
| 102 render_view_host->GetRoutingID(), GetViewType(web_contents()))); | 102 render_view_host->GetRoutingID(), GetViewType(web_contents()))); |
| 103 } | 103 } |
| 104 } | 104 } |
| 105 | 105 |
| 106 const Extension* ExtensionWebContentsObserver::GetExtension( | 106 const Extension* ExtensionWebContentsObserver::GetExtension( |
| 107 content::RenderViewHost* render_view_host) { | 107 content::RenderViewHost* render_view_host) { |
| (...skipping 15 matching lines...) Expand all Loading... |
| 123 // site, so we can ignore that wrinkle here. | 123 // site, so we can ignore that wrinkle here. |
| 124 const GURL& site = render_view_host->GetSiteInstance()->GetSiteURL(); | 124 const GURL& site = render_view_host->GetSiteInstance()->GetSiteURL(); |
| 125 | 125 |
| 126 if (!site.SchemeIs(kExtensionScheme)) | 126 if (!site.SchemeIs(kExtensionScheme)) |
| 127 return std::string(); | 127 return std::string(); |
| 128 | 128 |
| 129 return site.host(); | 129 return site.host(); |
| 130 } | 130 } |
| 131 | 131 |
| 132 } // namespace extensions | 132 } // namespace extensions |
| OLD | NEW |