OLD | NEW |
1 // Copyright 2015 The Chromium Authors. All rights reserved. | 1 // Copyright 2015 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/mojo/service_registration.h" | 5 #include "extensions/browser/mojo/service_registration.h" |
6 | 6 |
7 #include <string> | 7 #include <string> |
8 | 8 |
9 #include "base/bind.h" | 9 #include "base/bind.h" |
10 #include "base/command_line.h" | 10 #include "base/command_line.h" |
11 #include "content/public/browser/render_frame_host.h" | 11 #include "content/public/browser/render_frame_host.h" |
12 #include "content/public/browser/render_process_host.h" | 12 #include "content/public/browser/render_process_host.h" |
13 #include "content/public/browser/site_instance.h" | 13 #include "content/public/browser/site_instance.h" |
14 #include "content/public/common/service_registry.h" | 14 #include "content/public/common/service_registry.h" |
15 #include "extensions/browser/api/serial/serial_service_factory.h" | 15 #include "extensions/browser/api/serial/serial_service_factory.h" |
16 #include "extensions/browser/extension_registry.h" | 16 #include "extensions/browser/extension_registry.h" |
17 #include "extensions/browser/extensions_browser_client.h" | 17 #include "extensions/browser/extensions_browser_client.h" |
18 #include "extensions/browser/mojo/keep_alive_impl.h" | 18 #include "extensions/browser/mojo/keep_alive_impl.h" |
19 #include "extensions/browser/process_map.h" | 19 #include "extensions/browser/process_map.h" |
20 #include "extensions/common/constants.h" | 20 #include "extensions/common/constants.h" |
21 #include "extensions/common/extension_api.h" | 21 #include "extensions/common/extension_api.h" |
| 22 #include "extensions/common/permissions/permissions_data.h" |
22 #include "extensions/common/switches.h" | 23 #include "extensions/common/switches.h" |
23 | 24 |
24 namespace extensions { | 25 namespace extensions { |
25 namespace { | 26 namespace { |
26 | 27 |
27 const Extension* GetExtension(content::RenderFrameHost* render_frame_host) { | |
28 content::SiteInstance* site_instance = render_frame_host->GetSiteInstance(); | |
29 GURL url = render_frame_host->GetLastCommittedURL(); | |
30 if (!url.is_empty()) { | |
31 if (site_instance->GetSiteURL().GetOrigin() != url.GetOrigin()) | |
32 return nullptr; | |
33 } else { | |
34 url = site_instance->GetSiteURL(); | |
35 } | |
36 content::BrowserContext* browser_context = site_instance->GetBrowserContext(); | |
37 if (!url.SchemeIs(kExtensionScheme)) | |
38 return nullptr; | |
39 | |
40 return ExtensionRegistry::Get(browser_context) | |
41 ->enabled_extensions() | |
42 .GetExtensionOrAppByURL(url); | |
43 } | |
44 | |
45 bool ExtensionHasPermission(const Extension* extension, | 28 bool ExtensionHasPermission(const Extension* extension, |
46 content::RenderProcessHost* render_process_host, | 29 content::RenderProcessHost* render_process_host, |
47 const std::string& permission_name) { | 30 const std::string& permission_name) { |
48 Feature::Context context = | 31 Feature::Context context = |
49 ProcessMap::Get(render_process_host->GetBrowserContext()) | 32 ProcessMap::Get(render_process_host->GetBrowserContext()) |
50 ->GetMostLikelyContextType(extension, render_process_host->GetID()); | 33 ->GetMostLikelyContextType(extension, render_process_host->GetID()); |
51 | 34 |
52 return ExtensionAPI::GetSharedInstance() | 35 return ExtensionAPI::GetSharedInstance() |
53 ->IsAvailable(permission_name, extension, context, extension->url()) | 36 ->IsAvailable(permission_name, extension, context, extension->url()) |
54 .is_available(); | 37 .is_available(); |
55 } | 38 } |
56 | 39 |
57 } // namespace | 40 } // namespace |
58 | 41 |
59 void RegisterCoreExtensionServices( | 42 void RegisterServicesForFrame(content::RenderFrameHost* render_frame_host, |
60 content::RenderFrameHost* render_frame_host) { | 43 const Extension* extension) { |
61 const Extension* extension = GetExtension(render_frame_host); | 44 DCHECK(extension); |
62 if (!extension) | |
63 return; | |
64 | 45 |
65 content::ServiceRegistry* service_registry = | 46 content::ServiceRegistry* service_registry = |
66 render_frame_host->GetServiceRegistry(); | 47 render_frame_host->GetServiceRegistry(); |
67 if (base::CommandLine::ForCurrentProcess()->HasSwitch( | 48 if (base::CommandLine::ForCurrentProcess()->HasSwitch( |
68 switches::kEnableMojoSerialService)) { | 49 switches::kEnableMojoSerialService)) { |
69 if (ExtensionHasPermission(extension, render_frame_host->GetProcess(), | 50 if (ExtensionHasPermission(extension, render_frame_host->GetProcess(), |
70 "serial")) { | 51 "serial")) { |
71 service_registry->AddService(base::Bind(&BindToSerialServiceRequest)); | 52 service_registry->AddService(base::Bind(&BindToSerialServiceRequest)); |
72 } | 53 } |
73 } | 54 } |
74 service_registry->AddService(base::Bind( | 55 service_registry->AddService(base::Bind( |
75 KeepAliveImpl::Create, | 56 KeepAliveImpl::Create, |
76 render_frame_host->GetProcess()->GetBrowserContext(), extension)); | 57 render_frame_host->GetProcess()->GetBrowserContext(), extension)); |
77 } | 58 } |
78 | 59 |
79 } // namespace extensions | 60 } // namespace extensions |
OLD | NEW |