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 #include "chrome/browser/extensions/extension_special_storage_policy.h" | 5 #include "chrome/browser/extensions/extension_special_storage_policy.h" |
6 | 6 |
7 #include "base/logging.h" | 7 #include "base/logging.h" |
8 #include "chrome/common/extensions/extension.h" | 8 #include "chrome/common/extensions/extension.h" |
9 #include "chrome/common/url_constants.h" | 9 #include "chrome/common/url_constants.h" |
10 | 10 |
11 ExtensionSpecialStoragePolicy::ExtensionSpecialStoragePolicy() {} | 11 ExtensionSpecialStoragePolicy::ExtensionSpecialStoragePolicy() {} |
12 | 12 |
13 ExtensionSpecialStoragePolicy::~ExtensionSpecialStoragePolicy() {} | 13 ExtensionSpecialStoragePolicy::~ExtensionSpecialStoragePolicy() {} |
14 | 14 |
15 bool ExtensionSpecialStoragePolicy::IsStorageProtected(const GURL& origin) { | 15 bool ExtensionSpecialStoragePolicy::IsStorageProtected(const GURL& origin) { |
16 if (origin.SchemeIs(chrome::kExtensionScheme)) | 16 if (origin.SchemeIs(chrome::kExtensionScheme)) |
17 return true; | 17 return true; |
18 base::AutoLock locker(lock_); | 18 base::AutoLock locker(lock_); |
19 return protected_apps_.Contains(origin); | 19 return protected_apps_.Contains(origin); |
20 } | 20 } |
21 | 21 |
22 bool ExtensionSpecialStoragePolicy::IsStorageUnlimited(const GURL& origin) { | 22 bool ExtensionSpecialStoragePolicy::IsStorageUnlimited(const GURL& origin) { |
23 base::AutoLock locker(lock_); | 23 base::AutoLock locker(lock_); |
24 return unlimited_extensions_.Contains(origin); | 24 return unlimited_extensions_.Contains(origin); |
25 } | 25 } |
26 | 26 |
27 bool ExtensionSpecialStoragePolicy::IsLocalFileSystemAccessAllowed( | 27 bool ExtensionSpecialStoragePolicy::IsFileHandler( |
28 const GURL& origin) { | 28 const std::string& extension_id) { |
29 base::AutoLock locker(lock_); | 29 base::AutoLock locker(lock_); |
30 return local_filesystem_extensions_.Contains(origin); | 30 return file_handler_extensions_.ContainsExtension(extension_id); |
31 } | 31 } |
32 | 32 |
33 void ExtensionSpecialStoragePolicy::GrantRightsForExtension( | 33 void ExtensionSpecialStoragePolicy::GrantRightsForExtension( |
34 const Extension* extension) { | 34 const Extension* extension) { |
35 DCHECK(extension); | 35 DCHECK(extension); |
36 if (!extension->is_hosted_app() && | 36 if (!extension->is_hosted_app() && |
37 !extension->HasApiPermission(Extension::kUnlimitedStoragePermission) && | 37 !extension->HasApiPermission(Extension::kUnlimitedStoragePermission) && |
38 !extension->HasApiPermission(Extension::kFileSystemPermission)) { | 38 !extension->HasApiPermission(Extension::kFileBrowserHandlerPermission)) { |
39 return; | 39 return; |
40 } | 40 } |
41 base::AutoLock locker(lock_); | 41 base::AutoLock locker(lock_); |
42 if (extension->is_hosted_app()) | 42 if (extension->is_hosted_app()) |
43 protected_apps_.Add(extension); | 43 protected_apps_.Add(extension); |
44 if (extension->HasApiPermission(Extension::kUnlimitedStoragePermission)) | 44 if (extension->HasApiPermission(Extension::kUnlimitedStoragePermission)) |
45 unlimited_extensions_.Add(extension); | 45 unlimited_extensions_.Add(extension); |
46 if (extension->HasApiPermission(Extension::kFileSystemPermission)) | 46 if (extension->HasApiPermission(Extension::kFileBrowserHandlerPermission)) |
47 local_filesystem_extensions_.Add(extension); | 47 file_handler_extensions_.Add(extension); |
48 } | 48 } |
49 | 49 |
50 void ExtensionSpecialStoragePolicy::RevokeRightsForExtension( | 50 void ExtensionSpecialStoragePolicy::RevokeRightsForExtension( |
51 const Extension* extension) { | 51 const Extension* extension) { |
52 DCHECK(extension); | 52 DCHECK(extension); |
53 if (!extension->is_hosted_app() && | 53 if (!extension->is_hosted_app() && |
54 !extension->HasApiPermission(Extension::kUnlimitedStoragePermission) && | 54 !extension->HasApiPermission(Extension::kUnlimitedStoragePermission) && |
55 !extension->HasApiPermission(Extension::kFileSystemPermission)) { | 55 !extension->HasApiPermission(Extension::kFileBrowserHandlerPermission)) { |
56 return; | 56 return; |
57 } | 57 } |
58 base::AutoLock locker(lock_); | 58 base::AutoLock locker(lock_); |
59 if (extension->is_hosted_app()) | 59 if (extension->is_hosted_app()) |
60 protected_apps_.Remove(extension); | 60 protected_apps_.Remove(extension); |
61 if (extension->HasApiPermission(Extension::kUnlimitedStoragePermission)) | 61 if (extension->HasApiPermission(Extension::kUnlimitedStoragePermission)) |
62 unlimited_extensions_.Remove(extension); | 62 unlimited_extensions_.Remove(extension); |
63 if (extension->HasApiPermission(Extension::kFileSystemPermission)) | 63 if (extension->HasApiPermission(Extension::kFileBrowserHandlerPermission)) |
64 local_filesystem_extensions_.Remove(extension); | 64 file_handler_extensions_.Remove(extension); |
65 } | 65 } |
66 | 66 |
67 void ExtensionSpecialStoragePolicy::RevokeRightsForAllExtensions() { | 67 void ExtensionSpecialStoragePolicy::RevokeRightsForAllExtensions() { |
68 base::AutoLock locker(lock_); | 68 base::AutoLock locker(lock_); |
69 protected_apps_.Clear(); | 69 protected_apps_.Clear(); |
70 unlimited_extensions_.Clear(); | 70 unlimited_extensions_.Clear(); |
71 local_filesystem_extensions_.Clear(); | 71 file_handler_extensions_.Clear(); |
72 } | 72 } |
73 | 73 |
74 //----------------------------------------------------------------------------- | 74 //----------------------------------------------------------------------------- |
75 // SpecialCollection helper class | 75 // SpecialCollection helper class |
76 //----------------------------------------------------------------------------- | 76 //----------------------------------------------------------------------------- |
77 | 77 |
78 ExtensionSpecialStoragePolicy::SpecialCollection::SpecialCollection() {} | 78 ExtensionSpecialStoragePolicy::SpecialCollection::SpecialCollection() {} |
79 | 79 |
80 ExtensionSpecialStoragePolicy::SpecialCollection::~SpecialCollection() {} | 80 ExtensionSpecialStoragePolicy::SpecialCollection::~SpecialCollection() {} |
81 | 81 |
82 bool ExtensionSpecialStoragePolicy::SpecialCollection::Contains( | 82 bool ExtensionSpecialStoragePolicy::SpecialCollection::Contains( |
83 const GURL& origin) { | 83 const GURL& origin) { |
84 CachedResults::const_iterator found = cached_resuts_.find(origin); | 84 CachedResults::const_iterator found = cached_results_.find(origin); |
85 if (found != cached_resuts_.end()) | 85 if (found != cached_results_.end()) |
86 return found->second; | 86 return found->second; |
87 | 87 |
88 for (Extensions::const_iterator iter = extensions_.begin(); | 88 for (Extensions::const_iterator iter = extensions_.begin(); |
89 iter != extensions_.end(); ++iter) { | 89 iter != extensions_.end(); ++iter) { |
90 if (iter->second->OverlapsWithOrigin(origin)) { | 90 if (iter->second->OverlapsWithOrigin(origin)) { |
91 cached_resuts_[origin] = true; | 91 cached_results_[origin] = true; |
92 return true; | 92 return true; |
93 } | 93 } |
94 } | 94 } |
95 cached_resuts_[origin] = false; | 95 cached_results_[origin] = false; |
96 return false; | 96 return false; |
97 } | 97 } |
98 | 98 |
| 99 bool ExtensionSpecialStoragePolicy::SpecialCollection::ContainsExtension( |
| 100 const std::string& extension_id) { |
| 101 return extensions_.find(extension_id) != extensions_.end(); |
| 102 } |
| 103 |
99 void ExtensionSpecialStoragePolicy::SpecialCollection::Add( | 104 void ExtensionSpecialStoragePolicy::SpecialCollection::Add( |
100 const Extension* extension) { | 105 const Extension* extension) { |
101 cached_resuts_.clear(); | 106 cached_results_.clear(); |
102 extensions_[extension->id()] = extension; | 107 extensions_[extension->id()] = extension; |
103 } | 108 } |
104 | 109 |
105 void ExtensionSpecialStoragePolicy::SpecialCollection::Remove( | 110 void ExtensionSpecialStoragePolicy::SpecialCollection::Remove( |
106 const Extension* extension) { | 111 const Extension* extension) { |
107 cached_resuts_.clear(); | 112 cached_results_.clear(); |
108 extensions_.erase(extension->id()); | 113 extensions_.erase(extension->id()); |
109 } | 114 } |
110 | 115 |
111 void ExtensionSpecialStoragePolicy::SpecialCollection::Clear() { | 116 void ExtensionSpecialStoragePolicy::SpecialCollection::Clear() { |
112 cached_resuts_.clear(); | 117 cached_results_.clear(); |
113 extensions_.clear(); | 118 extensions_.clear(); |
114 } | 119 } |
OLD | NEW |