OLD | NEW |
| (Empty) |
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/settings/managed_value_store_cache.h" | |
6 | |
7 #include <set> | |
8 | |
9 #include "base/bind.h" | |
10 #include "base/bind_helpers.h" | |
11 #include "base/callback.h" | |
12 #include "base/file_util.h" | |
13 #include "base/logging.h" | |
14 #include "base/message_loop_proxy.h" | |
15 #include "chrome/browser/extensions/event_names.h" | |
16 #include "chrome/browser/extensions/extension_service.h" | |
17 #include "chrome/browser/extensions/settings/policy_value_store.h" | |
18 #include "chrome/browser/extensions/settings/settings_storage_factory.h" | |
19 #include "chrome/browser/value_store/value_store_change.h" | |
20 #include "chrome/common/extensions/extension.h" | |
21 #include "content/public/browser/browser_thread.h" | |
22 | |
23 using content::BrowserThread; | |
24 | |
25 namespace extensions { | |
26 | |
27 ManagedValueStoreCache::ManagedValueStoreCache( | |
28 policy::PolicyService* policy_service, | |
29 EventRouter* event_router, | |
30 const scoped_refptr<SettingsStorageFactory>& factory, | |
31 const scoped_refptr<SettingsObserverList>& observers, | |
32 const FilePath& profile_path) | |
33 : ALLOW_THIS_IN_INITIALIZER_LIST(weak_factory_(this)), | |
34 weak_this_on_ui_(weak_factory_.GetWeakPtr()), | |
35 policy_service_(policy_service), | |
36 event_router_(event_router), | |
37 storage_factory_(factory), | |
38 observers_(observers), | |
39 base_path_(profile_path.AppendASCII( | |
40 ExtensionService::kManagedSettingsDirectoryName)) { | |
41 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI)); | |
42 // |event_router| can be NULL on unit_tests. | |
43 if (event_router_) | |
44 event_router_->RegisterObserver(this, event_names::kOnSettingsChanged); | |
45 policy_service_->AddObserver(policy::POLICY_DOMAIN_EXTENSIONS, this); | |
46 } | |
47 | |
48 ManagedValueStoreCache::~ManagedValueStoreCache() { | |
49 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::FILE)); | |
50 DCHECK(!event_router_); | |
51 // Delete the PolicyValueStores on FILE. | |
52 store_map_.clear(); | |
53 } | |
54 | |
55 void ManagedValueStoreCache::ShutdownOnUI() { | |
56 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI)); | |
57 policy_service_->RemoveObserver(policy::POLICY_DOMAIN_EXTENSIONS, this); | |
58 policy_service_ = NULL; | |
59 if (event_router_) | |
60 event_router_->UnregisterObserver(this); | |
61 event_router_ = NULL; | |
62 weak_factory_.InvalidateWeakPtrs(); | |
63 } | |
64 | |
65 void ManagedValueStoreCache::RunWithValueStoreForExtension( | |
66 const StorageCallback& callback, | |
67 scoped_refptr<const Extension> extension) { | |
68 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::FILE)); | |
69 PolicyValueStore* store = GetStoreFor(extension->id()); | |
70 if (store) { | |
71 callback.Run(store); | |
72 } else { | |
73 // First time that an extension calls storage.managed.get(). Create the | |
74 // store and load it with the current policy, and don't send event | |
75 // notifications. | |
76 CreateStoreFor( | |
77 extension->id(), | |
78 false, | |
79 base::Bind(&ManagedValueStoreCache::RunWithValueStoreForExtension, | |
80 base::Unretained(this), | |
81 callback, | |
82 extension)); | |
83 } | |
84 } | |
85 | |
86 void ManagedValueStoreCache::DeleteStorageSoon( | |
87 const std::string& extension_id) { | |
88 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::FILE)); | |
89 PolicyValueStore* store = GetStoreFor(extension_id); | |
90 if (!store) { | |
91 // It's possible that the store exists, but hasn't been loaded yet | |
92 // (because the extension is unloaded, for example). Open the database to | |
93 // clear it if it exists. | |
94 // TODO(joaodasilva): move this check to a ValueStore method. | |
95 if (file_util::DirectoryExists(base_path_.AppendASCII(extension_id))) { | |
96 CreateStoreFor( | |
97 extension_id, | |
98 false, | |
99 base::Bind(&ManagedValueStoreCache::DeleteStorageSoon, | |
100 base::Unretained(this), | |
101 extension_id)); | |
102 } | |
103 } else { | |
104 store->DeleteStorage(); | |
105 store_map_.erase(extension_id); | |
106 } | |
107 } | |
108 | |
109 void ManagedValueStoreCache::OnPolicyUpdated(policy::PolicyDomain domain, | |
110 const std::string& component_id, | |
111 const policy::PolicyMap& previous, | |
112 const policy::PolicyMap& current) { | |
113 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI)); | |
114 BrowserThread::PostTask( | |
115 BrowserThread::FILE, FROM_HERE, | |
116 base::Bind(&ManagedValueStoreCache::UpdatePolicyOnFILE, | |
117 base::Unretained(this), | |
118 std::string(component_id), | |
119 base::Passed(current.DeepCopy()))); | |
120 } | |
121 | |
122 void ManagedValueStoreCache::UpdatePolicyOnFILE( | |
123 const std::string& extension_id, | |
124 scoped_ptr<policy::PolicyMap> current_policy) { | |
125 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::FILE)); | |
126 PolicyValueStore* store = GetStoreFor(extension_id); | |
127 if (!store) { | |
128 // The extension hasn't executed any storage.managed.* calls, and isn't | |
129 // listening for onChanged() either. Ignore this notification in that case. | |
130 return; | |
131 } | |
132 // Update the policy on the backing store, and fire notifications if it | |
133 // changed. | |
134 store->SetCurrentPolicy(*current_policy, true); | |
135 } | |
136 | |
137 void ManagedValueStoreCache::OnListenerAdded( | |
138 const EventListenerInfo& details) { | |
139 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI)); | |
140 DCHECK_EQ(std::string(event_names::kOnSettingsChanged), details.event_name); | |
141 // This is invoked on several occasions: | |
142 // | |
143 // 1. when an extension first registers to observe storage.onChanged; in this | |
144 // case the backend doesn't have any previous data persisted, and it won't | |
145 // trigger a notification. | |
146 // | |
147 // 2. when the browser starts up and all existing extensions re-register for | |
148 // the onChanged event. In this case, if the current policy differs from | |
149 // the persisted version then a notification will be sent. | |
150 // | |
151 // 3. a policy update just occurred and sent a notification, and an extension | |
152 // with EventPages that is observing onChanged just woke up and registed | |
153 // again. In this case the policy update already persisted the current | |
154 // policy version, and |store| already exists. | |
155 BrowserThread::PostTask( | |
156 BrowserThread::FILE, FROM_HERE, | |
157 base::Bind(&ManagedValueStoreCache::CreateForExtensionOnFILE, | |
158 base::Unretained(this), | |
159 details.extension_id)); | |
160 } | |
161 | |
162 void ManagedValueStoreCache::CreateForExtensionOnFILE( | |
163 const std::string& extension_id) { | |
164 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::FILE)); | |
165 PolicyValueStore* store = GetStoreFor(extension_id); | |
166 if (!store) | |
167 CreateStoreFor(extension_id, true, base::Closure()); | |
168 } | |
169 | |
170 PolicyValueStore* ManagedValueStoreCache::GetStoreFor( | |
171 const std::string& extension_id) { | |
172 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::FILE)); | |
173 PolicyValueStoreMap::iterator it = store_map_.find(extension_id); | |
174 if (it == store_map_.end()) | |
175 return NULL; | |
176 return it->second.get(); | |
177 } | |
178 | |
179 void ManagedValueStoreCache::CreateStoreFor( | |
180 const std::string& extension_id, | |
181 bool notify_if_changed, | |
182 const base::Closure& continuation) { | |
183 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::FILE)); | |
184 DCHECK(!GetStoreFor(extension_id)); | |
185 // Creating or loading an existing database requires an immediate update | |
186 // with the current policy for the corresponding extension, which must be | |
187 // retrieved on UI. | |
188 BrowserThread::PostTask( | |
189 BrowserThread::UI, FROM_HERE, | |
190 base::Bind(&ManagedValueStoreCache::GetInitialPolicy, | |
191 weak_this_on_ui_, | |
192 extension_id, | |
193 notify_if_changed, | |
194 continuation)); | |
195 } | |
196 | |
197 void ManagedValueStoreCache::GetInitialPolicy( | |
198 const std::string& extension_id, | |
199 bool notify_if_changed, | |
200 const base::Closure& continuation) { | |
201 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI)); | |
202 const policy::PolicyMap& policy = policy_service_->GetPolicies( | |
203 policy::POLICY_DOMAIN_EXTENSIONS, extension_id); | |
204 // Now post back to FILE to create the database. | |
205 BrowserThread::PostTask( | |
206 BrowserThread::FILE, FROM_HERE, | |
207 base::Bind(&ManagedValueStoreCache::CreateStoreWithInitialPolicy, | |
208 base::Unretained(this), | |
209 extension_id, | |
210 notify_if_changed, | |
211 base::Passed(policy.DeepCopy()), | |
212 continuation)); | |
213 } | |
214 | |
215 void ManagedValueStoreCache::CreateStoreWithInitialPolicy( | |
216 const std::string& extension_id, | |
217 bool notify_if_changed, | |
218 scoped_ptr<policy::PolicyMap> initial_policy, | |
219 const base::Closure& continuation) { | |
220 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::FILE)); | |
221 // If a 2nd call to CreateStoreFor() is issued before the 1st gets to execute | |
222 // its UI task, then the 2nd will enter this function but the store has | |
223 // already been created. Check for that. | |
224 PolicyValueStore* store = GetStoreFor(extension_id); | |
225 | |
226 if (!store) { | |
227 // Create it now. | |
228 | |
229 // If the database doesn't exist yet then this is the initial install, | |
230 // and no notifications should be issued in that case. | |
231 // TODO(joaodasilva): move this check to a ValueStore method. | |
232 if (!file_util::DirectoryExists(base_path_.AppendASCII(extension_id))) | |
233 notify_if_changed = false; | |
234 | |
235 store = new PolicyValueStore( | |
236 extension_id, | |
237 observers_, | |
238 make_scoped_ptr(storage_factory_->Create(base_path_, extension_id))); | |
239 store_map_[extension_id] = make_linked_ptr(store); | |
240 } | |
241 | |
242 // Send the latest policy to the store. | |
243 store->SetCurrentPolicy(*initial_policy, notify_if_changed); | |
244 | |
245 // And finally resume from where this process started. | |
246 if (!continuation.is_null()) | |
247 continuation.Run(); | |
248 } | |
249 | |
250 } // namespace extensions | |
OLD | NEW |