src/chrome/browser/extensions/extension_special_storage_policy.h

   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 "storage/browser/quota/special_storage_policy.h"
  14 #include "url/gurl.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 storage::SpecialStoragePolicy {
  26  public:
  27   explicit ExtensionSpecialStoragePolicy(CookieSettings* cookie_settings);
  28
  29   // storage::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_