1 // Copyright (c) 2012 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 "chrome/browser/extensions/api/declarative/rules_registry_service.h"
6
7 #include "base/bind.h"
8 #include "base/lazy_instance.h"
9 #include "base/logging.h"
10 #include "base/memory/scoped_ptr.h"
11 #include "chrome/browser/chrome_notification_types.h"
12 #include "chrome/browser/extensions/api/declarative/rules_cache_delegate.h"
13 #include "chrome/browser/extensions/api/declarative_content/content_rules_registry.h"
14 #include "chrome/browser/extensions/api/declarative_webrequest/webrequest_constants.h"
15 #include "chrome/browser/extensions/api/declarative_webrequest/webrequest_rules_registry.h"
16 #include "chrome/browser/extensions/api/web_request/web_request_api.h"
17 #include "chrome/browser/profiles/profile.h"
18 #include "content/public/browser/browser_thread.h"
19 #include "content/public/browser/notification_details.h"
20 #include "content/public/browser/notification_service.h"
21 #include "content/public/browser/notification_source.h"
22 #include "content/public/browser/render_process_host.h"
23 #include "extensions/browser/extension_registry.h"
24 #include "extensions/common/extension.h"
25
26 namespace extensions {
27
28 namespace {
29
30 // Registers |web_request_rules_registry| on the IO thread.
RegisterToExtensionWebRequestEventRouterOnIO(void * profile,const RulesRegistryService::WebViewKey & webview_key,scoped_refptr<WebRequestRulesRegistry> web_request_rules_registry)31 void RegisterToExtensionWebRequestEventRouterOnIO(
32 void* profile,
33 const RulesRegistryService::WebViewKey& webview_key,
34 scoped_refptr<WebRequestRulesRegistry> web_request_rules_registry) {
35 ExtensionWebRequestEventRouter::GetInstance()->RegisterRulesRegistry(
36 profile, webview_key, web_request_rules_registry);
37 }
38
IsWebView(const RulesRegistryService::WebViewKey & webview_key)39 bool IsWebView(const RulesRegistryService::WebViewKey& webview_key) {
40 return webview_key.embedder_process_id && webview_key.webview_instance_id;
41 }
42
43 } // namespace
44
RulesRegistryService(content::BrowserContext * context)45 RulesRegistryService::RulesRegistryService(content::BrowserContext* context)
46 : content_rules_registry_(NULL),
47 extension_registry_observer_(this),
48 profile_(Profile::FromBrowserContext(context)) {
49 if (profile_) {
50 extension_registry_observer_.Add(ExtensionRegistry::Get(profile_));
51 registrar_.Add(
52 this, content::NOTIFICATION_RENDERER_PROCESS_TERMINATED,
53 content::NotificationService::AllBrowserContextsAndSources());
54 EnsureDefaultRulesRegistriesRegistered(WebViewKey(0, 0));
55 }
56 }
57
~RulesRegistryService()58 RulesRegistryService::~RulesRegistryService() {}
59
EnsureDefaultRulesRegistriesRegistered(const WebViewKey & webview_key)60 void RulesRegistryService::EnsureDefaultRulesRegistriesRegistered(
61 const WebViewKey& webview_key) {
62 if (!profile_)
63 return;
64
65 RulesRegistryKey key(declarative_webrequest_constants::kOnRequest,
66 webview_key);
67 // If we can find the key in the |rule_registries_| then we have already
68 // installed the default registries.
69 if (ContainsKey(rule_registries_, key))
70 return;
71
72
73 RulesCacheDelegate* web_request_cache_delegate = NULL;
74 if (!IsWebView(webview_key)) {
75 web_request_cache_delegate =
76 new RulesCacheDelegate(true /*log_storage_init_delay*/);
77 cache_delegates_.push_back(web_request_cache_delegate);
78 }
79 scoped_refptr<WebRequestRulesRegistry> web_request_rules_registry(
80 new WebRequestRulesRegistry(profile_,
81 web_request_cache_delegate,
82 webview_key));
83
84 RegisterRulesRegistry(web_request_rules_registry);
85 content::BrowserThread::PostTask(
86 content::BrowserThread::IO, FROM_HERE,
87 base::Bind(&RegisterToExtensionWebRequestEventRouterOnIO,
88 profile_, webview_key, web_request_rules_registry));
89
90 #if defined(ENABLE_EXTENSIONS)
91 // Only create a ContentRulesRegistry for regular pages and not webviews.
92 if (!IsWebView(webview_key)) {
93 RulesCacheDelegate* content_rules_cache_delegate =
94 new RulesCacheDelegate(false /*log_storage_init_delay*/);
95 cache_delegates_.push_back(content_rules_cache_delegate);
96 scoped_refptr<ContentRulesRegistry> content_rules_registry(
97 new ContentRulesRegistry(profile_, content_rules_cache_delegate));
98 RegisterRulesRegistry(content_rules_registry);
99 content_rules_registry_ = content_rules_registry.get();
100 }
101 #endif // defined(ENABLE_EXTENSIONS)
102 }
103
Shutdown()104 void RulesRegistryService::Shutdown() {
105 // Release the references to all registries. This would happen soon during
106 // destruction of |*this|, but we need the ExtensionWebRequestEventRouter to
107 // be the last to reference the WebRequestRulesRegistry objects, so that
108 // the posted task below causes their destruction on the IO thread, not on UI
109 // where the destruction of |*this| takes place.
110 // TODO(vabr): Remove once http://crbug.com/218451#c6 gets addressed.
111 rule_registries_.clear();
112 content::BrowserThread::PostTask(
113 content::BrowserThread::IO, FROM_HERE,
114 base::Bind(&RegisterToExtensionWebRequestEventRouterOnIO,
115 profile_, WebViewKey(0, 0),
116 scoped_refptr<WebRequestRulesRegistry>(NULL)));
117 }
118
119 static base::LazyInstance<BrowserContextKeyedAPIFactory<RulesRegistryService> >
120 g_factory = LAZY_INSTANCE_INITIALIZER;
121
122 // static
123 BrowserContextKeyedAPIFactory<RulesRegistryService>*
GetFactoryInstance()124 RulesRegistryService::GetFactoryInstance() {
125 return g_factory.Pointer();
126 }
127
128 // static
Get(content::BrowserContext * context)129 RulesRegistryService* RulesRegistryService::Get(
130 content::BrowserContext* context) {
131 return BrowserContextKeyedAPIFactory<RulesRegistryService>::Get(context);
132 }
133
RegisterRulesRegistry(scoped_refptr<RulesRegistry> rule_registry)134 void RulesRegistryService::RegisterRulesRegistry(
135 scoped_refptr<RulesRegistry> rule_registry) {
136 const std::string event_name(rule_registry->event_name());
137 RulesRegistryKey key(event_name, rule_registry->webview_key());
138 DCHECK(rule_registries_.find(key) == rule_registries_.end());
139 rule_registries_[key] = rule_registry;
140 }
141
GetRulesRegistry(const WebViewKey & webview_key,const std::string & event_name)142 scoped_refptr<RulesRegistry> RulesRegistryService::GetRulesRegistry(
143 const WebViewKey& webview_key,
144 const std::string& event_name) {
145 EnsureDefaultRulesRegistriesRegistered(webview_key);
146
147 RulesRegistryKey key(event_name, webview_key);
148 RulesRegistryMap::const_iterator i = rule_registries_.find(key);
149 if (i == rule_registries_.end())
150 return scoped_refptr<RulesRegistry>();
151 return i->second;
152 }
153
RemoveWebViewRulesRegistries(int process_id)154 void RulesRegistryService::RemoveWebViewRulesRegistries(int process_id) {
155 DCHECK_NE(0, process_id);
156
157 std::set<RulesRegistryKey> registries_to_delete;
158 for (RulesRegistryMap::iterator it = rule_registries_.begin();
159 it != rule_registries_.end(); ++it) {
160 const RulesRegistryKey& key = it->first;
161 const WebViewKey& webview_key = key.webview_key;
162 int embedder_process_id = webview_key.embedder_process_id;
163 // |process_id| will always be non-zero.
164 // |embedder_process_id| will only be non-zero if the key corresponds to a
165 // webview registry.
166 // Thus, |embedder_process_id| == |process_id| ==> the process ID is a
167 // webview embedder.
168 if (embedder_process_id != process_id)
169 continue;
170
171 // Modifying the container while iterating is bad so we'll save the keys we
172 // wish to delete in another container, and delete them in another loop.
173 registries_to_delete.insert(key);
174 }
175 for (std::set<RulesRegistryKey>::iterator it = registries_to_delete.begin();
176 it != registries_to_delete.end(); ++it) {
177 rule_registries_.erase(*it);
178 }
179 }
180
SimulateExtensionUninstalled(const std::string & extension_id)181 void RulesRegistryService::SimulateExtensionUninstalled(
182 const std::string& extension_id) {
183 NotifyRegistriesHelper(&RulesRegistry::OnExtensionUninstalled, extension_id);
184 }
185
NotifyRegistriesHelper(void (RulesRegistry::* notification_callback)(const std::string &),const std::string & extension_id)186 void RulesRegistryService::NotifyRegistriesHelper(
187 void (RulesRegistry::*notification_callback)(const std::string&),
188 const std::string& extension_id) {
189 RulesRegistryMap::iterator i;
190 for (i = rule_registries_.begin(); i != rule_registries_.end(); ++i) {
191 scoped_refptr<RulesRegistry> registry = i->second;
192 if (content::BrowserThread::CurrentlyOn(registry->owner_thread())) {
193 (registry->*notification_callback)(extension_id);
194 } else {
195 content::BrowserThread::PostTask(
196 registry->owner_thread(),
197 FROM_HERE,
198 base::Bind(notification_callback, registry, extension_id));
199 }
200 }
201 }
202
OnExtensionLoaded(content::BrowserContext * browser_context,const Extension * extension)203 void RulesRegistryService::OnExtensionLoaded(
204 content::BrowserContext* browser_context,
205 const Extension* extension) {
206 NotifyRegistriesHelper(&RulesRegistry::OnExtensionLoaded, extension->id());
207 }
208
OnExtensionUnloaded(content::BrowserContext * browser_context,const Extension * extension,UnloadedExtensionInfo::Reason reason)209 void RulesRegistryService::OnExtensionUnloaded(
210 content::BrowserContext* browser_context,
211 const Extension* extension,
212 UnloadedExtensionInfo::Reason reason) {
213 NotifyRegistriesHelper(&RulesRegistry::OnExtensionUnloaded, extension->id());
214 }
215
OnExtensionUninstalled(content::BrowserContext * browser_context,const Extension * extension)216 void RulesRegistryService::OnExtensionUninstalled(
217 content::BrowserContext* browser_context,
218 const Extension* extension) {
219 NotifyRegistriesHelper(&RulesRegistry::OnExtensionUninstalled,
220 extension->id());
221 }
222
Observe(int type,const content::NotificationSource & source,const content::NotificationDetails & details)223 void RulesRegistryService::Observe(
224 int type,
225 const content::NotificationSource& source,
226 const content::NotificationDetails& details) {
227 DCHECK_EQ(content::NOTIFICATION_RENDERER_PROCESS_TERMINATED, type);
228
229 content::RenderProcessHost* process =
230 content::Source<content::RenderProcessHost>(source).ptr();
231 RemoveWebViewRulesRegistries(process->GetID());
232 }
233
234 } // namespace extensions
235