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 #ifndef CHROME_BROWSER_EXTENSIONS_EXTENSION_SPECIAL_STORAGE_POLICY_H_ 6 #define CHROME_BROWSER_EXTENSIONS_EXTENSION_SPECIAL_STORAGE_POLICY_H_ 7 8 #include <map> 9 #include <string> 10 11 #include "base/synchronization/lock.h" 12 #include "extensions/common/extension_set.h" 13 #include "url/gurl.h" 14 #include "webkit/browser/quota/special_storage_policy.h" 15 16 class CookieSettings; 17 18 namespace extensions { 19 class Extension; 20 } 21 22 // Special rights are granted to 'extensions' and 'applications'. The 23 // storage subsystems and the browsing data remover query this interface 24 // to determine which origins have these rights. 25 class ExtensionSpecialStoragePolicy : public quota::SpecialStoragePolicy { 26 public: 27 explicit ExtensionSpecialStoragePolicy(CookieSettings* cookie_settings); 28 29 // quota::SpecialStoragePolicy methods used by storage subsystems and the 30 // browsing data remover. These methods are safe to call on any thread. 31 virtual bool IsStorageProtected(const GURL& origin) OVERRIDE; 32 virtual bool IsStorageUnlimited(const GURL& origin) OVERRIDE; 33 virtual bool IsStorageSessionOnly(const GURL& origin) OVERRIDE; 34 virtual bool CanQueryDiskSize(const GURL& origin) OVERRIDE; 35 virtual bool IsFileHandler(const std::string& extension_id) OVERRIDE; 36 virtual bool HasIsolatedStorage(const GURL& origin) OVERRIDE; 37 virtual bool HasSessionOnlyOrigins() OVERRIDE; 38 39 // Methods used by the ExtensionService to populate this class. 40 void GrantRightsForExtension(const extensions::Extension* extension); 41 void RevokeRightsForExtension(const extensions::Extension* extension); 42 void RevokeRightsForAllExtensions(); 43 44 // Decides whether the storage for |extension|'s web extent needs protection. 45 bool NeedsProtection(const extensions::Extension* extension); 46 47 // Returns the set of extensions protecting this origin. The caller does not 48 // take ownership of the return value. 49 const extensions::ExtensionSet* ExtensionsProtectingOrigin( 50 const GURL& origin); 51 52 protected: 53 virtual ~ExtensionSpecialStoragePolicy(); 54 55 private: 56 class SpecialCollection { 57 public: 58 SpecialCollection(); 59 ~SpecialCollection(); 60 61 bool Contains(const GURL& origin); 62 const extensions::ExtensionSet* ExtensionsContaining(const GURL& origin); 63 bool ContainsExtension(const std::string& extension_id); 64 bool Add(const extensions::Extension* extension); 65 bool Remove(const extensions::Extension* extension); 66 void Clear(); 67 68 private: 69 typedef std::map<GURL, extensions::ExtensionSet*> CachedResults; 70 71 void ClearCache(); 72 73 extensions::ExtensionSet extensions_; 74 CachedResults cached_results_; 75 }; 76 77 void NotifyGranted(const GURL& origin, int change_flags); 78 void NotifyRevoked(const GURL& origin, int change_flags); 79 void NotifyCleared(); 80 81 base::Lock lock_; // Synchronize all access to the collections. 82 SpecialCollection protected_apps_; 83 SpecialCollection installed_apps_; 84 SpecialCollection unlimited_extensions_; 85 SpecialCollection file_handler_extensions_; 86 SpecialCollection isolated_extensions_; 87 scoped_refptr<CookieSettings> cookie_settings_; 88 }; 89 90 #endif // CHROME_BROWSER_EXTENSIONS_EXTENSION_SPECIAL_STORAGE_POLICY_H_ 91