OLD | NEW |
| (Empty) |
1 // Copyright 2013 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 "apps/shell/shell_content_browser_client.h" | |
6 | |
7 #include "apps/shell/shell_browser_context.h" | |
8 #include "apps/shell/shell_browser_main_parts.h" | |
9 #include "apps/shell/shell_extension_system.h" | |
10 #include "base/command_line.h" | |
11 #include "chrome/browser/extensions/extension_protocols.h" | |
12 #include "chrome/browser/extensions/extension_resource_protocols.h" | |
13 #include "content/public/browser/browser_thread.h" | |
14 #include "content/public/browser/render_process_host.h" | |
15 #include "content/public/browser/site_instance.h" | |
16 #include "content/public/common/content_switches.h" | |
17 #include "content/public/common/url_constants.h" | |
18 #include "content/shell/browser/shell_browser_context.h" | |
19 #include "extensions/browser/extension_registry.h" | |
20 #include "extensions/browser/info_map.h" | |
21 #include "extensions/browser/process_map.h" | |
22 #include "extensions/common/constants.h" | |
23 #include "extensions/common/extension.h" | |
24 #include "extensions/common/switches.h" | |
25 #include "url/gurl.h" | |
26 | |
27 using content::BrowserThread; | |
28 using extensions::ExtensionRegistry; | |
29 | |
30 namespace apps { | |
31 | |
32 ShellContentBrowserClient::ShellContentBrowserClient() | |
33 : browser_main_parts_(NULL) { | |
34 } | |
35 | |
36 ShellContentBrowserClient::~ShellContentBrowserClient() { | |
37 } | |
38 | |
39 content::BrowserMainParts* ShellContentBrowserClient::CreateBrowserMainParts( | |
40 const content::MainFunctionParams& parameters) { | |
41 browser_main_parts_ = new ShellBrowserMainParts(parameters); | |
42 return browser_main_parts_; | |
43 } | |
44 | |
45 net::URLRequestContextGetter* | |
46 ShellContentBrowserClient::CreateRequestContext( | |
47 content::BrowserContext* content_browser_context, | |
48 content::ProtocolHandlerMap* protocol_handlers) { | |
49 // Handle chrome-extension: and chrome-extension-resource: requests. | |
50 extensions::InfoMap* extension_info_map = | |
51 browser_main_parts_->extension_system()->info_map(); | |
52 (*protocol_handlers)[extensions::kExtensionScheme] = | |
53 linked_ptr<net::URLRequestJobFactory::ProtocolHandler>( | |
54 CreateExtensionProtocolHandler(false /*is_incognito*/, | |
55 extension_info_map)); | |
56 (*protocol_handlers)[extensions::kExtensionResourceScheme] = | |
57 linked_ptr<net::URLRequestJobFactory::ProtocolHandler>( | |
58 CreateExtensionResourceProtocolHandler()); | |
59 // Let content::ShellBrowserContext handle the rest of the setup. | |
60 return browser_main_parts_->browser_context()->CreateRequestContext( | |
61 protocol_handlers); | |
62 } | |
63 | |
64 bool ShellContentBrowserClient::IsHandledURL(const GURL& url) { | |
65 if (!url.is_valid()) | |
66 return false; | |
67 // Keep in sync with ProtocolHandlers added in CreateRequestContext() and in | |
68 // content::ShellURLRequestContextGetter::GetURLRequestContext(). | |
69 static const char* const kProtocolList[] = { | |
70 chrome::kBlobScheme, | |
71 chrome::kChromeUIScheme, | |
72 chrome::kChromeDevToolsScheme, | |
73 content::kDataScheme, | |
74 content::kFileScheme, | |
75 content::kFileSystemScheme, | |
76 extensions::kExtensionScheme, | |
77 extensions::kExtensionResourceScheme, | |
78 }; | |
79 for (size_t i = 0; i < arraysize(kProtocolList); ++i) { | |
80 if (url.scheme() == kProtocolList[i]) | |
81 return true; | |
82 } | |
83 return false; | |
84 } | |
85 | |
86 void ShellContentBrowserClient::SiteInstanceGotProcess( | |
87 content::SiteInstance* site_instance) { | |
88 // If this isn't an extension renderer there's nothing to do. | |
89 const extensions::Extension* extension = GetExtension(site_instance); | |
90 if (!extension) | |
91 return; | |
92 | |
93 extensions::ProcessMap::Get(browser_main_parts_->browser_context()) | |
94 ->Insert(extension->id(), | |
95 site_instance->GetProcess()->GetID(), | |
96 site_instance->GetId()); | |
97 | |
98 BrowserThread::PostTask( | |
99 BrowserThread::IO, | |
100 FROM_HERE, | |
101 base::Bind(&extensions::InfoMap::RegisterExtensionProcess, | |
102 browser_main_parts_->extension_system()->info_map(), | |
103 extension->id(), | |
104 site_instance->GetProcess()->GetID(), | |
105 site_instance->GetId())); | |
106 } | |
107 | |
108 void ShellContentBrowserClient::SiteInstanceDeleting( | |
109 content::SiteInstance* site_instance) { | |
110 // If this isn't an extension renderer there's nothing to do. | |
111 const extensions::Extension* extension = GetExtension(site_instance); | |
112 if (!extension) | |
113 return; | |
114 | |
115 extensions::ProcessMap::Get(browser_main_parts_->browser_context()) | |
116 ->Remove(extension->id(), | |
117 site_instance->GetProcess()->GetID(), | |
118 site_instance->GetId()); | |
119 | |
120 BrowserThread::PostTask( | |
121 BrowserThread::IO, | |
122 FROM_HERE, | |
123 base::Bind(&extensions::InfoMap::UnregisterExtensionProcess, | |
124 browser_main_parts_->extension_system()->info_map(), | |
125 extension->id(), | |
126 site_instance->GetProcess()->GetID(), | |
127 site_instance->GetId())); | |
128 } | |
129 | |
130 void ShellContentBrowserClient::AppendExtraCommandLineSwitches( | |
131 CommandLine* command_line, int child_process_id) { | |
132 std::string process_type = | |
133 command_line->GetSwitchValueASCII(switches::kProcessType); | |
134 if (process_type == switches::kRendererProcess) { | |
135 // TODO(jamescook): Should we check here if the process is in the extension | |
136 // service process map, or can we assume all renderers are extension | |
137 // renderers? | |
138 command_line->AppendSwitch(extensions::switches::kExtensionProcess); | |
139 } | |
140 } | |
141 | |
142 const extensions::Extension* ShellContentBrowserClient::GetExtension( | |
143 content::SiteInstance* site_instance) { | |
144 ExtensionRegistry* registry = | |
145 ExtensionRegistry::Get(site_instance->GetBrowserContext()); | |
146 return registry->enabled_extensions().GetExtensionOrAppByURL( | |
147 site_instance->GetSiteURL()); | |
148 } | |
149 | |
150 } // namespace apps | |
OLD | NEW |