| OLD | NEW |
| 1 // Copyright 2014 The Chromium Authors. All rights reserved. | 1 // Copyright 2014 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/permissions/permission_context_base.h" | 5 #include "chrome/browser/permissions/permission_context_base.h" |
| 6 | 6 |
| 7 #include <map> | 7 #include <map> |
| 8 #include <set> | 8 #include <set> |
| 9 #include <string> | 9 #include <string> |
| 10 #include <utility> | 10 #include <utility> |
| (...skipping 149 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 160 respond_permission_.Run(); | 160 respond_permission_.Run(); |
| 161 respond_permission_.Reset(); | 161 respond_permission_.Reset(); |
| 162 } else { | 162 } else { |
| 163 // Stop the run loop from spinning indefinitely if no response callback | 163 // Stop the run loop from spinning indefinitely if no response callback |
| 164 // has been set, as is the case with TestParallelRequests. | 164 // has been set, as is the case with TestParallelRequests. |
| 165 quit_closure_.Run(); | 165 quit_closure_.Run(); |
| 166 quit_closure_.Reset(); | 166 quit_closure_.Reset(); |
| 167 } | 167 } |
| 168 } | 168 } |
| 169 | 169 |
| 170 // Permission request will need to be responded to, so pass a callback to be | 170 // Set the callback to run if the permission is being responded to in the |
| 171 // run once the request has completed and the decision has been made. | 171 // test. This is left empty where no response is needed, such as in parallel |
| 172 // requests, permissions blacklisting, invalid origin, and killswitch. |
| 172 void SetRespondPermissionCallback(base::Closure callback) { | 173 void SetRespondPermissionCallback(base::Closure callback) { |
| 173 respond_permission_ = callback; | 174 respond_permission_ = callback; |
| 174 } | 175 } |
| 175 | 176 |
| 176 protected: | 177 protected: |
| 177 void UpdateTabContext(const PermissionRequestID& id, | 178 void UpdateTabContext(const PermissionRequestID& id, |
| 178 const GURL& requesting_origin, | 179 const GURL& requesting_origin, |
| 179 bool allowed) override { | 180 bool allowed) override { |
| 180 tab_context_updated_ = true; | 181 tab_context_updated_ = true; |
| 181 } | 182 } |
| (...skipping 458 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 640 ContentSettingsType content_settings_type, | 641 ContentSettingsType content_settings_type, |
| 641 scoped_refptr<safe_browsing::SafeBrowsingDatabaseManager> db_manager, | 642 scoped_refptr<safe_browsing::SafeBrowsingDatabaseManager> db_manager, |
| 642 const GURL& url, | 643 const GURL& url, |
| 643 int timeout, | 644 int timeout, |
| 644 ContentSetting response) { | 645 ContentSetting response) { |
| 645 NavigateAndCommit(url); | 646 NavigateAndCommit(url); |
| 646 base::test::ScopedFeatureList scoped_feature_list; | 647 base::test::ScopedFeatureList scoped_feature_list; |
| 647 scoped_feature_list.InitAndEnableFeature(features::kPermissionsBlacklist); | 648 scoped_feature_list.InitAndEnableFeature(features::kPermissionsBlacklist); |
| 648 TestPermissionContext permission_context(profile(), permission_type, | 649 TestPermissionContext permission_context(profile(), permission_type, |
| 649 content_settings_type); | 650 content_settings_type); |
| 650 permission_context.SetSafeBrowsingDatabaseManagerAndTimeoutForTest( | 651 PermissionDecisionAutoBlocker::GetForProfile(profile()) |
| 651 db_manager, timeout); | 652 ->SetSafeBrowsingDatabaseManagerAndTimeoutForTesting(db_manager, |
| 653 timeout); |
| 652 const PermissionRequestID id( | 654 const PermissionRequestID id( |
| 653 web_contents()->GetRenderProcessHost()->GetID(), | 655 web_contents()->GetRenderProcessHost()->GetID(), |
| 654 web_contents()->GetMainFrame()->GetRoutingID(), -1); | 656 web_contents()->GetMainFrame()->GetRoutingID(), -1); |
| 655 // The response callback needs to be set here to test a response being made | 657 // The response callback needs to be set here to test a response being made |
| 656 // in the case of a site not being blacklisted or a safe browsing timeout. | 658 // in the case of a site not being blacklisted or a safe browsing timeout. |
| 657 permission_context.SetRespondPermissionCallback(base::Bind( | 659 if (response == CONTENT_SETTING_ALLOW) { |
| 658 &PermissionContextBaseTests::RespondToPermission, | 660 permission_context.SetRespondPermissionCallback( |
| 659 base::Unretained(this), &permission_context, id, url, false, response)); | 661 base::Bind(&PermissionContextBaseTests::RespondToPermission, |
| 662 base::Unretained(this), &permission_context, id, url, |
| 663 true /* persist */, response)); |
| 664 } |
| 665 |
| 660 permission_context.RequestPermission( | 666 permission_context.RequestPermission( |
| 661 web_contents(), id, url, true /* user_gesture */, | 667 web_contents(), id, url, true /* user_gesture */, |
| 662 base::Bind(&TestPermissionContext::TrackPermissionDecision, | 668 base::Bind(&TestPermissionContext::TrackPermissionDecision, |
| 663 base::Unretained(&permission_context))); | 669 base::Unretained(&permission_context))); |
| 670 EXPECT_EQ(response, permission_context.GetPermissionStatus(url, url)); |
| 664 | 671 |
| 665 ASSERT_EQ(1u, permission_context.decisions().size()); | 672 if (response == CONTENT_SETTING_ALLOW) { |
| 666 EXPECT_EQ(response, permission_context.decisions()[0]); | 673 ASSERT_EQ(1u, permission_context.decisions().size()); |
| 674 EXPECT_EQ(response, permission_context.decisions()[0]); |
| 675 } |
| 667 } | 676 } |
| 668 | 677 |
| 669 private: | 678 private: |
| 670 // ChromeRenderViewHostTestHarness: | 679 // ChromeRenderViewHostTestHarness: |
| 671 void SetUp() override { | 680 void SetUp() override { |
| 672 ChromeRenderViewHostTestHarness::SetUp(); | 681 ChromeRenderViewHostTestHarness::SetUp(); |
| 673 #if defined(OS_ANDROID) | 682 #if defined(OS_ANDROID) |
| 674 InfoBarService::CreateForWebContents(web_contents()); | 683 InfoBarService::CreateForWebContents(web_contents()); |
| 675 #else | 684 #else |
| 676 PermissionRequestManager::CreateForWebContents(web_contents()); | 685 PermissionRequestManager::CreateForWebContents(web_contents()); |
| (...skipping 152 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 829 const GURL url("https://www.example.com"); | 838 const GURL url("https://www.example.com"); |
| 830 std::set<std::string> blacklisted_permissions{ | 839 std::set<std::string> blacklisted_permissions{ |
| 831 PermissionUtil::GetPermissionString( | 840 PermissionUtil::GetPermissionString( |
| 832 content::PermissionType::GEOLOCATION)}; | 841 content::PermissionType::GEOLOCATION)}; |
| 833 db_manager->BlacklistUrlPermissions(url, blacklisted_permissions); | 842 db_manager->BlacklistUrlPermissions(url, blacklisted_permissions); |
| 834 TestPermissionsBlacklisting(content::PermissionType::GEOLOCATION, | 843 TestPermissionsBlacklisting(content::PermissionType::GEOLOCATION, |
| 835 CONTENT_SETTINGS_TYPE_GEOLOCATION, db_manager, | 844 CONTENT_SETTINGS_TYPE_GEOLOCATION, db_manager, |
| 836 url, 2000 /* timeout */, CONTENT_SETTING_BLOCK); | 845 url, 2000 /* timeout */, CONTENT_SETTING_BLOCK); |
| 837 } | 846 } |
| 838 | 847 |
| 839 // Tests that a URL with a blacklisted permission is permitted to request a | 848 // Tests that a URL that is blacklisted for one permission can still request |
| 840 // non-blacklisted permission. | 849 // another and grant another. |
| 841 TEST_F(PermissionContextBaseTests, TestPermissionsBlacklistingAllowed) { | 850 TEST_F(PermissionContextBaseTests, TestPermissionsBlacklistingAllowed) { |
| 842 scoped_refptr<MockSafeBrowsingDatabaseManager> db_manager = | 851 scoped_refptr<MockSafeBrowsingDatabaseManager> db_manager = |
| 843 new MockSafeBrowsingDatabaseManager(true /* perform_callback */); | 852 new MockSafeBrowsingDatabaseManager(true /* perform_callback */); |
| 844 const GURL url("https://www.example.com"); | 853 const GURL url("https://www.example.com"); |
| 845 std::set<std::string> blacklisted_permissions{ | 854 std::set<std::string> blacklisted_permissions{ |
| 846 PermissionUtil::GetPermissionString( | 855 PermissionUtil::GetPermissionString( |
| 847 content::PermissionType::GEOLOCATION)}; | 856 content::PermissionType::GEOLOCATION)}; |
| 848 db_manager->BlacklistUrlPermissions(url, blacklisted_permissions); | 857 db_manager->BlacklistUrlPermissions(url, blacklisted_permissions); |
| 849 TestPermissionsBlacklisting( | |
| 850 content::PermissionType::GEOLOCATION, CONTENT_SETTINGS_TYPE_GEOLOCATION, | |
| 851 db_manager, url, 2000 /* timeout in ms */, CONTENT_SETTING_BLOCK); | |
| 852 TestPermissionsBlacklisting(content::PermissionType::NOTIFICATIONS, | 858 TestPermissionsBlacklisting(content::PermissionType::NOTIFICATIONS, |
| 853 CONTENT_SETTINGS_TYPE_NOTIFICATIONS, db_manager, | 859 CONTENT_SETTINGS_TYPE_NOTIFICATIONS, db_manager, |
| 854 url, 2000 /* timeout in ms */, | 860 url, 2000 /* timeout */, CONTENT_SETTING_ALLOW); |
| 855 CONTENT_SETTING_ALLOW); | |
| 856 } | 861 } |
| 857 | |
| 858 // Tests that a URL with a blacklisted permisison is permitted to request that | |
| 859 // permission if Safe Browsing has timed out. | |
| 860 TEST_F(PermissionContextBaseTests, TestSafeBrowsingTimeout) { | |
| 861 scoped_refptr<MockSafeBrowsingDatabaseManager> db_manager = | |
| 862 new MockSafeBrowsingDatabaseManager(false /* perform_callback */); | |
| 863 const GURL url("https://www.example.com"); | |
| 864 std::set<std::string> blacklisted_permissions{ | |
| 865 PermissionUtil::GetPermissionString( | |
| 866 content::PermissionType::GEOLOCATION)}; | |
| 867 db_manager->BlacklistUrlPermissions(url, blacklisted_permissions); | |
| 868 TestPermissionsBlacklisting(content::PermissionType::GEOLOCATION, | |
| 869 CONTENT_SETTINGS_TYPE_GEOLOCATION, db_manager, | |
| 870 url, 0 /* timeout in ms */, CONTENT_SETTING_ASK); | |
| 871 } | |
| OLD | NEW |