OLD | NEW |
1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 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 | 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 "chrome/browser/extensions/api/declarative/rules_registry_service.h" | 5 #include "chrome/browser/extensions/api/declarative/rules_registry_service.h" |
6 | 6 |
7 #include "base/bind.h" | 7 #include "base/bind.h" |
8 #include "base/lazy_instance.h" | 8 #include "base/lazy_instance.h" |
9 #include "base/logging.h" | 9 #include "base/logging.h" |
10 #include "base/memory/scoped_ptr.h" | 10 #include "base/memory/scoped_ptr.h" |
11 #include "chrome/browser/chrome_notification_types.h" | 11 #include "chrome/browser/chrome_notification_types.h" |
12 #include "chrome/browser/extensions/api/declarative/rules_cache_delegate.h" | 12 #include "chrome/browser/extensions/api/declarative/rules_cache_delegate.h" |
13 #include "chrome/browser/extensions/api/declarative_content/content_rules_regist
ry.h" | 13 #include "chrome/browser/extensions/api/declarative_content/content_rules_regist
ry.h" |
14 #include "chrome/browser/extensions/api/declarative_webrequest/webrequest_consta
nts.h" | 14 #include "chrome/browser/extensions/api/declarative_webrequest/webrequest_consta
nts.h" |
15 #include "chrome/browser/extensions/api/declarative_webrequest/webrequest_rules_
registry.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" | 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" | 17 #include "content/public/browser/browser_thread.h" |
19 #include "content/public/browser/notification_details.h" | 18 #include "content/public/browser/notification_details.h" |
20 #include "content/public/browser/notification_service.h" | 19 #include "content/public/browser/notification_service.h" |
21 #include "content/public/browser/notification_source.h" | 20 #include "content/public/browser/notification_source.h" |
22 #include "content/public/browser/render_process_host.h" | 21 #include "content/public/browser/render_process_host.h" |
23 #include "extensions/browser/extension_registry.h" | 22 #include "extensions/browser/extension_registry.h" |
24 #include "extensions/common/extension.h" | 23 #include "extensions/common/extension.h" |
25 | 24 |
26 namespace extensions { | 25 namespace extensions { |
27 | 26 |
28 namespace { | 27 namespace { |
29 | 28 |
30 // Registers |web_request_rules_registry| on the IO thread. | 29 // Registers |web_request_rules_registry| on the IO thread. |
31 void RegisterToExtensionWebRequestEventRouterOnIO( | 30 void RegisterToExtensionWebRequestEventRouterOnIO( |
32 void* profile, | 31 content::BrowserContext* browser_context, |
33 const RulesRegistryService::WebViewKey& webview_key, | 32 const RulesRegistryService::WebViewKey& webview_key, |
34 scoped_refptr<WebRequestRulesRegistry> web_request_rules_registry) { | 33 scoped_refptr<WebRequestRulesRegistry> web_request_rules_registry) { |
35 ExtensionWebRequestEventRouter::GetInstance()->RegisterRulesRegistry( | 34 ExtensionWebRequestEventRouter::GetInstance()->RegisterRulesRegistry( |
36 profile, webview_key, web_request_rules_registry); | 35 browser_context, webview_key, web_request_rules_registry); |
37 } | 36 } |
38 | 37 |
39 bool IsWebView(const RulesRegistryService::WebViewKey& webview_key) { | 38 bool IsWebView(const RulesRegistryService::WebViewKey& webview_key) { |
40 return webview_key.embedder_process_id && webview_key.webview_instance_id; | 39 return webview_key.embedder_process_id && webview_key.webview_instance_id; |
41 } | 40 } |
42 | 41 |
43 } // namespace | 42 } // namespace |
44 | 43 |
45 RulesRegistryService::RulesRegistryService(content::BrowserContext* context) | 44 RulesRegistryService::RulesRegistryService(content::BrowserContext* context) |
46 : content_rules_registry_(NULL), | 45 : content_rules_registry_(NULL), |
47 extension_registry_observer_(this), | 46 extension_registry_observer_(this), |
48 profile_(Profile::FromBrowserContext(context)) { | 47 browser_context_(context) { |
49 if (profile_) { | 48 if (browser_context_) { |
50 extension_registry_observer_.Add(ExtensionRegistry::Get(profile_)); | 49 extension_registry_observer_.Add(ExtensionRegistry::Get(browser_context_)); |
51 registrar_.Add( | 50 registrar_.Add( |
52 this, content::NOTIFICATION_RENDERER_PROCESS_TERMINATED, | 51 this, content::NOTIFICATION_RENDERER_PROCESS_TERMINATED, |
53 content::NotificationService::AllBrowserContextsAndSources()); | 52 content::NotificationService::AllBrowserContextsAndSources()); |
54 EnsureDefaultRulesRegistriesRegistered(WebViewKey(0, 0)); | 53 EnsureDefaultRulesRegistriesRegistered(WebViewKey(0, 0)); |
55 } | 54 } |
56 } | 55 } |
57 | 56 |
58 RulesRegistryService::~RulesRegistryService() {} | 57 RulesRegistryService::~RulesRegistryService() {} |
59 | 58 |
60 void RulesRegistryService::EnsureDefaultRulesRegistriesRegistered( | 59 void RulesRegistryService::EnsureDefaultRulesRegistriesRegistered( |
61 const WebViewKey& webview_key) { | 60 const WebViewKey& webview_key) { |
62 if (!profile_) | 61 if (!browser_context_) |
63 return; | 62 return; |
64 | 63 |
65 RulesRegistryKey key(declarative_webrequest_constants::kOnRequest, | 64 RulesRegistryKey key(declarative_webrequest_constants::kOnRequest, |
66 webview_key); | 65 webview_key); |
67 // If we can find the key in the |rule_registries_| then we have already | 66 // If we can find the key in the |rule_registries_| then we have already |
68 // installed the default registries. | 67 // installed the default registries. |
69 if (ContainsKey(rule_registries_, key)) | 68 if (ContainsKey(rule_registries_, key)) |
70 return; | 69 return; |
71 | 70 |
72 | 71 |
73 RulesCacheDelegate* web_request_cache_delegate = NULL; | 72 RulesCacheDelegate* web_request_cache_delegate = NULL; |
74 if (!IsWebView(webview_key)) { | 73 if (!IsWebView(webview_key)) { |
75 web_request_cache_delegate = | 74 web_request_cache_delegate = |
76 new RulesCacheDelegate(true /*log_storage_init_delay*/); | 75 new RulesCacheDelegate(true /*log_storage_init_delay*/); |
77 cache_delegates_.push_back(web_request_cache_delegate); | 76 cache_delegates_.push_back(web_request_cache_delegate); |
78 } | 77 } |
79 scoped_refptr<WebRequestRulesRegistry> web_request_rules_registry( | 78 scoped_refptr<WebRequestRulesRegistry> web_request_rules_registry( |
80 new WebRequestRulesRegistry(profile_, | 79 new WebRequestRulesRegistry(browser_context_, |
81 web_request_cache_delegate, | 80 web_request_cache_delegate, |
82 webview_key)); | 81 webview_key)); |
83 | 82 |
84 RegisterRulesRegistry(web_request_rules_registry); | 83 RegisterRulesRegistry(web_request_rules_registry); |
85 content::BrowserThread::PostTask( | 84 content::BrowserThread::PostTask( |
86 content::BrowserThread::IO, FROM_HERE, | 85 content::BrowserThread::IO, FROM_HERE, |
87 base::Bind(&RegisterToExtensionWebRequestEventRouterOnIO, | 86 base::Bind(&RegisterToExtensionWebRequestEventRouterOnIO, |
88 profile_, webview_key, web_request_rules_registry)); | 87 browser_context_, webview_key, web_request_rules_registry)); |
89 | 88 |
90 // Only create a ContentRulesRegistry for regular pages and not webviews. | 89 // Only create a ContentRulesRegistry for regular pages and not webviews. |
91 if (!IsWebView(webview_key)) { | 90 if (!IsWebView(webview_key)) { |
92 RulesCacheDelegate* content_rules_cache_delegate = | 91 RulesCacheDelegate* content_rules_cache_delegate = |
93 new RulesCacheDelegate(false /*log_storage_init_delay*/); | 92 new RulesCacheDelegate(false /*log_storage_init_delay*/); |
94 cache_delegates_.push_back(content_rules_cache_delegate); | 93 cache_delegates_.push_back(content_rules_cache_delegate); |
95 scoped_refptr<ContentRulesRegistry> content_rules_registry( | 94 scoped_refptr<ContentRulesRegistry> content_rules_registry( |
96 new ContentRulesRegistry(profile_, content_rules_cache_delegate)); | 95 new ContentRulesRegistry(browser_context_, |
| 96 content_rules_cache_delegate)); |
97 RegisterRulesRegistry(content_rules_registry); | 97 RegisterRulesRegistry(content_rules_registry); |
98 content_rules_registry_ = content_rules_registry.get(); | 98 content_rules_registry_ = content_rules_registry.get(); |
99 } | 99 } |
100 } | 100 } |
101 | 101 |
102 void RulesRegistryService::Shutdown() { | 102 void RulesRegistryService::Shutdown() { |
103 // Release the references to all registries. This would happen soon during | 103 // Release the references to all registries. This would happen soon during |
104 // destruction of |*this|, but we need the ExtensionWebRequestEventRouter to | 104 // destruction of |*this|, but we need the ExtensionWebRequestEventRouter to |
105 // be the last to reference the WebRequestRulesRegistry objects, so that | 105 // be the last to reference the WebRequestRulesRegistry objects, so that |
106 // the posted task below causes their destruction on the IO thread, not on UI | 106 // the posted task below causes their destruction on the IO thread, not on UI |
107 // where the destruction of |*this| takes place. | 107 // where the destruction of |*this| takes place. |
108 // TODO(vabr): Remove once http://crbug.com/218451#c6 gets addressed. | 108 // TODO(vabr): Remove once http://crbug.com/218451#c6 gets addressed. |
109 rule_registries_.clear(); | 109 rule_registries_.clear(); |
110 content::BrowserThread::PostTask( | 110 content::BrowserThread::PostTask( |
111 content::BrowserThread::IO, FROM_HERE, | 111 content::BrowserThread::IO, FROM_HERE, |
112 base::Bind(&RegisterToExtensionWebRequestEventRouterOnIO, | 112 base::Bind(&RegisterToExtensionWebRequestEventRouterOnIO, |
113 profile_, WebViewKey(0, 0), | 113 browser_context_, WebViewKey(0, 0), |
114 scoped_refptr<WebRequestRulesRegistry>(NULL))); | 114 scoped_refptr<WebRequestRulesRegistry>(NULL))); |
115 } | 115 } |
116 | 116 |
117 static base::LazyInstance<BrowserContextKeyedAPIFactory<RulesRegistryService> > | 117 static base::LazyInstance<BrowserContextKeyedAPIFactory<RulesRegistryService> > |
118 g_factory = LAZY_INSTANCE_INITIALIZER; | 118 g_factory = LAZY_INSTANCE_INITIALIZER; |
119 | 119 |
120 // static | 120 // static |
121 BrowserContextKeyedAPIFactory<RulesRegistryService>* | 121 BrowserContextKeyedAPIFactory<RulesRegistryService>* |
122 RulesRegistryService::GetFactoryInstance() { | 122 RulesRegistryService::GetFactoryInstance() { |
123 return g_factory.Pointer(); | 123 return g_factory.Pointer(); |
(...skipping 100 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
224 const content::NotificationSource& source, | 224 const content::NotificationSource& source, |
225 const content::NotificationDetails& details) { | 225 const content::NotificationDetails& details) { |
226 DCHECK_EQ(content::NOTIFICATION_RENDERER_PROCESS_TERMINATED, type); | 226 DCHECK_EQ(content::NOTIFICATION_RENDERER_PROCESS_TERMINATED, type); |
227 | 227 |
228 content::RenderProcessHost* process = | 228 content::RenderProcessHost* process = |
229 content::Source<content::RenderProcessHost>(source).ptr(); | 229 content::Source<content::RenderProcessHost>(source).ptr(); |
230 RemoveWebViewRulesRegistries(process->GetID()); | 230 RemoveWebViewRulesRegistries(process->GetID()); |
231 } | 231 } |
232 | 232 |
233 } // namespace extensions | 233 } // namespace extensions |
OLD | NEW |