| OLD | NEW |
| 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 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 | 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/browsing_data/browsing_data_helper.h" | 5 #include "chrome/browser/browsing_data/browsing_data_helper.h" |
| 6 | 6 |
| 7 #include "base/macros.h" | 7 #include "base/macros.h" |
| 8 #include "chrome/common/url_constants.h" | 8 #include "chrome/common/url_constants.h" |
| 9 #include "extensions/common/constants.h" | 9 #include "extensions/common/constants.h" |
| 10 #include "extensions/features/features.h" | 10 #include "extensions/features/features.h" |
| (...skipping 43 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 54 bool Match(const GURL& origin, | 54 bool Match(const GURL& origin, |
| 55 int mask, | 55 int mask, |
| 56 storage::SpecialStoragePolicy* policy) { | 56 storage::SpecialStoragePolicy* policy) { |
| 57 return BrowsingDataHelper::DoesOriginMatchMask(origin, mask, policy); | 57 return BrowsingDataHelper::DoesOriginMatchMask(origin, mask, policy); |
| 58 } | 58 } |
| 59 | 59 |
| 60 private: | 60 private: |
| 61 DISALLOW_COPY_AND_ASSIGN(BrowsingDataHelperTest); | 61 DISALLOW_COPY_AND_ASSIGN(BrowsingDataHelperTest); |
| 62 }; | 62 }; |
| 63 | 63 |
| 64 TEST_F(BrowsingDataHelperTest, WebSafeSchemesAreWebSafe) { | 64 TEST_F(BrowsingDataHelperTest, WebStorageSchemesAreWebSchemes) { |
| 65 EXPECT_TRUE(IsWebScheme(url::kHttpScheme)); | 65 EXPECT_TRUE(IsWebScheme(url::kHttpScheme)); |
| 66 EXPECT_TRUE(IsWebScheme(url::kHttpsScheme)); | 66 EXPECT_TRUE(IsWebScheme(url::kHttpsScheme)); |
| 67 EXPECT_TRUE(IsWebScheme(url::kFileScheme)); |
| 67 EXPECT_TRUE(IsWebScheme(url::kFtpScheme)); | 68 EXPECT_TRUE(IsWebScheme(url::kFtpScheme)); |
| 68 EXPECT_TRUE(IsWebScheme(url::kDataScheme)); | 69 EXPECT_TRUE(IsWebScheme(url::kWsScheme)); |
| 69 EXPECT_TRUE(IsWebScheme("feed")); | 70 EXPECT_TRUE(IsWebScheme(url::kWssScheme)); |
| 70 EXPECT_TRUE(IsWebScheme(url::kBlobScheme)); | |
| 71 EXPECT_TRUE(IsWebScheme(url::kFileSystemScheme)); | |
| 72 EXPECT_FALSE(IsWebScheme("invalid-scheme-i-just-made-up")); | |
| 73 } | 71 } |
| 74 | 72 |
| 75 TEST_F(BrowsingDataHelperTest, ChromeSchemesAreNotWebSafe) { | 73 TEST_F(BrowsingDataHelperTest, ChromeSchemesAreNotWebSchemes) { |
| 76 EXPECT_FALSE(IsWebScheme(extensions::kExtensionScheme)); | 74 EXPECT_FALSE(IsWebScheme(extensions::kExtensionScheme)); |
| 77 EXPECT_FALSE(IsWebScheme(url::kAboutScheme)); | 75 EXPECT_FALSE(IsWebScheme(url::kAboutScheme)); |
| 78 EXPECT_FALSE(IsWebScheme(content::kChromeDevToolsScheme)); | 76 EXPECT_FALSE(IsWebScheme(content::kChromeDevToolsScheme)); |
| 79 EXPECT_FALSE(IsWebScheme(content::kChromeUIScheme)); | 77 EXPECT_FALSE(IsWebScheme(content::kChromeUIScheme)); |
| 80 EXPECT_FALSE(IsWebScheme(url::kJavaScriptScheme)); | 78 EXPECT_FALSE(IsWebScheme(url::kJavaScriptScheme)); |
| 81 EXPECT_FALSE(IsWebScheme(url::kMailToScheme)); | 79 EXPECT_FALSE(IsWebScheme(url::kMailToScheme)); |
| 82 EXPECT_FALSE(IsWebScheme(content::kViewSourceScheme)); | 80 EXPECT_FALSE(IsWebScheme(content::kViewSourceScheme)); |
| 83 } | 81 } |
| 84 | 82 |
| 85 TEST_F(BrowsingDataHelperTest, WebSafeSchemesAreNotExtensions) { | 83 TEST_F(BrowsingDataHelperTest, WebStorageSchemesAreNotExtensions) { |
| 86 EXPECT_FALSE(IsExtensionScheme(url::kHttpScheme)); | 84 EXPECT_FALSE(IsExtensionScheme(url::kHttpScheme)); |
| 87 EXPECT_FALSE(IsExtensionScheme(url::kHttpsScheme)); | 85 EXPECT_FALSE(IsExtensionScheme(url::kHttpsScheme)); |
| 86 EXPECT_FALSE(IsExtensionScheme(url::kFileScheme)); |
| 88 EXPECT_FALSE(IsExtensionScheme(url::kFtpScheme)); | 87 EXPECT_FALSE(IsExtensionScheme(url::kFtpScheme)); |
| 89 EXPECT_FALSE(IsExtensionScheme(url::kDataScheme)); | 88 EXPECT_FALSE(IsExtensionScheme(url::kWsScheme)); |
| 90 EXPECT_FALSE(IsExtensionScheme("feed")); | 89 EXPECT_FALSE(IsExtensionScheme(url::kWssScheme)); |
| 91 EXPECT_FALSE(IsExtensionScheme(url::kBlobScheme)); | |
| 92 EXPECT_FALSE(IsExtensionScheme(url::kFileSystemScheme)); | |
| 93 EXPECT_FALSE(IsExtensionScheme("invalid-scheme-i-just-made-up")); | |
| 94 } | 90 } |
| 95 | 91 |
| 96 TEST_F(BrowsingDataHelperTest, ChromeSchemesAreNotAllExtension) { | 92 TEST_F(BrowsingDataHelperTest, ChromeSchemesAreNotAllExtension) { |
| 97 EXPECT_TRUE(IsExtensionScheme(extensions::kExtensionScheme)); | 93 EXPECT_TRUE(IsExtensionScheme(extensions::kExtensionScheme)); |
| 98 | 94 |
| 99 EXPECT_FALSE(IsExtensionScheme(url::kAboutScheme)); | 95 EXPECT_FALSE(IsExtensionScheme(url::kAboutScheme)); |
| 100 EXPECT_FALSE(IsExtensionScheme(content::kChromeDevToolsScheme)); | 96 EXPECT_FALSE(IsExtensionScheme(content::kChromeDevToolsScheme)); |
| 101 EXPECT_FALSE(IsExtensionScheme(content::kChromeUIScheme)); | 97 EXPECT_FALSE(IsExtensionScheme(content::kChromeUIScheme)); |
| 102 EXPECT_FALSE(IsExtensionScheme(url::kJavaScriptScheme)); | 98 EXPECT_FALSE(IsExtensionScheme(url::kJavaScriptScheme)); |
| 103 EXPECT_FALSE(IsExtensionScheme(url::kMailToScheme)); | 99 EXPECT_FALSE(IsExtensionScheme(url::kMailToScheme)); |
| 104 EXPECT_FALSE(IsExtensionScheme(content::kViewSourceScheme)); | 100 EXPECT_FALSE(IsExtensionScheme(content::kViewSourceScheme)); |
| 105 } | 101 } |
| 106 | 102 |
| 103 TEST_F(BrowsingDataHelperTest, SchemesThatCantStoreDataDontMatchAnything) { |
| 104 EXPECT_FALSE(IsWebScheme(url::kDataScheme)); |
| 105 EXPECT_FALSE(IsExtensionScheme(url::kDataScheme)); |
| 106 |
| 107 EXPECT_FALSE(IsWebScheme("feed")); |
| 108 EXPECT_FALSE(IsExtensionScheme("feed")); |
| 109 |
| 110 EXPECT_FALSE(IsWebScheme(url::kBlobScheme)); |
| 111 EXPECT_FALSE(IsExtensionScheme(url::kBlobScheme)); |
| 112 |
| 113 EXPECT_FALSE(IsWebScheme(url::kFileSystemScheme)); |
| 114 EXPECT_FALSE(IsExtensionScheme(url::kFileSystemScheme)); |
| 115 |
| 116 EXPECT_FALSE(IsWebScheme("invalid-scheme-i-just-made-up")); |
| 117 EXPECT_FALSE(IsExtensionScheme("invalid-scheme-i-just-made-up")); |
| 118 } |
| 119 |
| 107 #if BUILDFLAG(ENABLE_EXTENSIONS) | 120 #if BUILDFLAG(ENABLE_EXTENSIONS) |
| 108 TEST_F(BrowsingDataHelperTest, TestMatches) { | 121 TEST_F(BrowsingDataHelperTest, TestMatches) { |
| 109 scoped_refptr<MockExtensionSpecialStoragePolicy> mock_policy = | 122 scoped_refptr<MockExtensionSpecialStoragePolicy> mock_policy = |
| 110 new MockExtensionSpecialStoragePolicy; | 123 new MockExtensionSpecialStoragePolicy; |
| 111 // Protect kOrigin1. | 124 // Protect kOrigin1. |
| 112 mock_policy->AddProtected(kOrigin1.GetOrigin()); | 125 mock_policy->AddProtected(kOrigin1.GetOrigin()); |
| 113 | 126 |
| 114 EXPECT_FALSE(Match(kOrigin1, kUnprotected, mock_policy.get())); | 127 EXPECT_FALSE(Match(kOrigin1, kUnprotected, mock_policy.get())); |
| 115 EXPECT_TRUE(Match(kOrigin2, kUnprotected, mock_policy.get())); | 128 EXPECT_TRUE(Match(kOrigin2, kUnprotected, mock_policy.get())); |
| 116 EXPECT_FALSE(Match(kOriginExt, kUnprotected, mock_policy.get())); | 129 EXPECT_FALSE(Match(kOriginExt, kUnprotected, mock_policy.get())); |
| (...skipping 48 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 165 EXPECT_TRUE(Match(kOriginExt, kExtension, nullptr)); | 178 EXPECT_TRUE(Match(kOriginExt, kExtension, nullptr)); |
| 166 EXPECT_FALSE(Match(kOriginExt, kUnprotected, nullptr)); | 179 EXPECT_FALSE(Match(kOriginExt, kUnprotected, nullptr)); |
| 167 EXPECT_FALSE(Match(kOriginExt, kProtected, nullptr)); | 180 EXPECT_FALSE(Match(kOriginExt, kProtected, nullptr)); |
| 168 | 181 |
| 169 EXPECT_FALSE(Match(kOriginDevTools, kExtension, nullptr)); | 182 EXPECT_FALSE(Match(kOriginDevTools, kExtension, nullptr)); |
| 170 EXPECT_FALSE(Match(kOriginDevTools, kUnprotected, nullptr)); | 183 EXPECT_FALSE(Match(kOriginDevTools, kUnprotected, nullptr)); |
| 171 EXPECT_FALSE(Match(kOriginDevTools, kProtected, nullptr)); | 184 EXPECT_FALSE(Match(kOriginDevTools, kProtected, nullptr)); |
| 172 } | 185 } |
| 173 | 186 |
| 174 } // namespace | 187 } // namespace |
| OLD | NEW |