OLD | NEW |
1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2011 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 #ifndef CHROME_BROWSER_EXTENSIONS_EXTENSION_UPDATER_H_ | 5 #ifndef CHROME_BROWSER_EXTENSIONS_EXTENSION_UPDATER_H_ |
6 #define CHROME_BROWSER_EXTENSIONS_EXTENSION_UPDATER_H_ | 6 #define CHROME_BROWSER_EXTENSIONS_EXTENSION_UPDATER_H_ |
7 #pragma once | 7 #pragma once |
8 | 8 |
9 #include <deque> | 9 #include <deque> |
10 #include <map> | 10 #include <map> |
11 #include <set> | 11 #include <set> |
| 12 #include <stack> |
12 #include <string> | 13 #include <string> |
13 #include <vector> | 14 #include <vector> |
14 | 15 |
15 #include "base/gtest_prod_util.h" | 16 #include "base/gtest_prod_util.h" |
16 #include "base/memory/ref_counted.h" | 17 #include "base/memory/ref_counted.h" |
17 #include "base/memory/scoped_ptr.h" | 18 #include "base/memory/scoped_ptr.h" |
18 #include "base/memory/weak_ptr.h" | 19 #include "base/memory/weak_ptr.h" |
19 #include "base/scoped_temp_dir.h" | 20 #include "base/scoped_temp_dir.h" |
20 #include "base/task.h" | 21 #include "base/task.h" |
21 #include "base/time.h" | 22 #include "base/time.h" |
(...skipping 135 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
157 }; | 158 }; |
158 | 159 |
159 // A class for doing auto-updates of installed Extensions. Used like this: | 160 // A class for doing auto-updates of installed Extensions. Used like this: |
160 // | 161 // |
161 // ExtensionUpdater* updater = new ExtensionUpdater(my_extensions_service, | 162 // ExtensionUpdater* updater = new ExtensionUpdater(my_extensions_service, |
162 // pref_service, | 163 // pref_service, |
163 // update_frequency_secs); | 164 // update_frequency_secs); |
164 // updater->Start(); | 165 // updater->Start(); |
165 // .... | 166 // .... |
166 // updater->Stop(); | 167 // updater->Stop(); |
167 class ExtensionUpdater : public URLFetcher::Delegate { | 168 class ExtensionUpdater : public URLFetcher::Delegate, |
| 169 public NotificationObserver { |
168 public: | 170 public: |
169 // Holds a pointer to the passed |service|, using it for querying installed | 171 // Holds a pointer to the passed |service|, using it for querying installed |
170 // extensions and installing updated ones. The |frequency_seconds| parameter | 172 // extensions and installing updated ones. The |frequency_seconds| parameter |
171 // controls how often update checks are scheduled. | 173 // controls how often update checks are scheduled. |
172 ExtensionUpdater(ExtensionServiceInterface* service, | 174 ExtensionUpdater(ExtensionServiceInterface* service, |
173 ExtensionPrefs* extension_prefs, | 175 ExtensionPrefs* extension_prefs, |
174 PrefService* prefs, | 176 PrefService* prefs, |
175 Profile* profile, | 177 Profile* profile, |
176 int frequency_seconds); | 178 int frequency_seconds); |
177 | 179 |
(...skipping 36 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
214 ExtensionFetch(const std::string& i, const GURL& u, | 216 ExtensionFetch(const std::string& i, const GURL& u, |
215 const std::string& h, const std::string& v); | 217 const std::string& h, const std::string& v); |
216 ~ExtensionFetch(); | 218 ~ExtensionFetch(); |
217 | 219 |
218 std::string id; | 220 std::string id; |
219 GURL url; | 221 GURL url; |
220 std::string package_hash; | 222 std::string package_hash; |
221 std::string version; | 223 std::string version; |
222 }; | 224 }; |
223 | 225 |
| 226 // FetchedCRXFile holds information about a CRX file we fetched to disk, |
| 227 // but have not yet installed. |
| 228 struct FetchedCRXFile { |
| 229 FetchedCRXFile(); |
| 230 FetchedCRXFile(const std::string& id, |
| 231 const FilePath& path, |
| 232 const GURL& download_url); |
| 233 ~FetchedCRXFile(); |
| 234 |
| 235 std::string id; |
| 236 FilePath path; |
| 237 GURL download_url; |
| 238 }; |
| 239 |
224 // These are needed for unit testing, to help identify the correct mock | 240 // These are needed for unit testing, to help identify the correct mock |
225 // URLFetcher objects. | 241 // URLFetcher objects. |
226 static const int kManifestFetcherId = 1; | 242 static const int kManifestFetcherId = 1; |
227 static const int kExtensionFetcherId = 2; | 243 static const int kExtensionFetcherId = 2; |
228 | 244 |
229 static const char* kBlacklistAppID; | 245 static const char* kBlacklistAppID; |
230 | 246 |
231 // Does common work from constructors. | 247 // Does common work from constructors. |
232 void Init(); | 248 void Init(); |
233 | 249 |
(...skipping 69 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
303 | 319 |
304 // Send a notification if we're finished updating. | 320 // Send a notification if we're finished updating. |
305 void NotifyIfFinished(); | 321 void NotifyIfFinished(); |
306 | 322 |
307 // Adds a set of ids to in_progress_ids_. | 323 // Adds a set of ids to in_progress_ids_. |
308 void AddToInProgress(const std::set<std::string>& ids); | 324 void AddToInProgress(const std::set<std::string>& ids); |
309 | 325 |
310 // Removes a set of ids from in_progress_ids_. | 326 // Removes a set of ids from in_progress_ids_. |
311 void RemoveFromInProgress(const std::set<std::string>& ids); | 327 void RemoveFromInProgress(const std::set<std::string>& ids); |
312 | 328 |
| 329 // If a CRX file has been fetched but not installed, and no install is |
| 330 // currently running, start installing. Returns true if an install was |
| 331 // started. |
| 332 bool MaybeInstallCRXFile(); |
| 333 |
| 334 // NotificationObserver implementation. |
| 335 virtual void Observe(NotificationType type, |
| 336 const NotificationSource& source, |
| 337 const NotificationDetails& details); |
| 338 |
313 // Whether Start() has been called but not Stop(). | 339 // Whether Start() has been called but not Stop(). |
314 bool alive_; | 340 bool alive_; |
315 | 341 |
316 base::WeakPtrFactory<ExtensionUpdater> weak_ptr_factory_; | 342 base::WeakPtrFactory<ExtensionUpdater> weak_ptr_factory_; |
317 | 343 |
318 // Outstanding url fetch requests for manifests and updates. | 344 // Outstanding url fetch requests for manifests and updates. |
319 scoped_ptr<URLFetcher> manifest_fetcher_; | 345 scoped_ptr<URLFetcher> manifest_fetcher_; |
320 scoped_ptr<URLFetcher> extension_fetcher_; | 346 scoped_ptr<URLFetcher> extension_fetcher_; |
321 | 347 |
322 // Pending manifests and extensions to be fetched when the appropriate fetcher | 348 // Pending manifests and extensions to be fetched when the appropriate fetcher |
(...skipping 18 matching lines...) Expand all Loading... |
341 bool will_check_soon_; | 367 bool will_check_soon_; |
342 | 368 |
343 ExtensionPrefs* extension_prefs_; | 369 ExtensionPrefs* extension_prefs_; |
344 PrefService* prefs_; | 370 PrefService* prefs_; |
345 Profile* profile_; | 371 Profile* profile_; |
346 bool blacklist_checks_enabled_; | 372 bool blacklist_checks_enabled_; |
347 | 373 |
348 // The ids of extensions that have in-progress update checks. | 374 // The ids of extensions that have in-progress update checks. |
349 std::set<std::string> in_progress_ids_; | 375 std::set<std::string> in_progress_ids_; |
350 | 376 |
| 377 // Observes CRX installs we initiate. |
| 378 NotificationRegistrar registrar_; |
| 379 |
| 380 // True when a CrxInstaller is doing an install. Used in MaybeUpdateCrxFile() |
| 381 // to keep more than one install from running at once. |
| 382 bool crx_install_is_running_; |
| 383 |
| 384 // Fetched CRX files waiting to be installed. |
| 385 std::stack<FetchedCRXFile> fetched_crx_files_; |
| 386 |
351 FRIEND_TEST(ExtensionUpdaterTest, TestStartUpdateCheckMemory); | 387 FRIEND_TEST(ExtensionUpdaterTest, TestStartUpdateCheckMemory); |
352 FRIEND_TEST(ExtensionUpdaterTest, TestAfterStopBehavior); | 388 FRIEND_TEST(ExtensionUpdaterTest, TestAfterStopBehavior); |
353 | 389 |
354 DISALLOW_COPY_AND_ASSIGN(ExtensionUpdater); | 390 DISALLOW_COPY_AND_ASSIGN(ExtensionUpdater); |
355 }; | 391 }; |
356 | 392 |
357 #endif // CHROME_BROWSER_EXTENSIONS_EXTENSION_UPDATER_H_ | 393 #endif // CHROME_BROWSER_EXTENSIONS_EXTENSION_UPDATER_H_ |
OLD | NEW |