| 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/geolocation/geolocation_permission_context.h" | 5 #include "chrome/browser/geolocation/geolocation_permission_context.h" |
| 6 | 6 |
| 7 #include <set> | 7 #include <set> |
| 8 #include <string> | 8 #include <string> |
| 9 #include <utility> | 9 #include <utility> |
| 10 | 10 |
| 11 #include "base/bind.h" | 11 #include "base/bind.h" |
| 12 #include "base/command_line.h" | 12 #include "base/command_line.h" |
| 13 #include "base/containers/hash_tables.h" | 13 #include "base/containers/hash_tables.h" |
| 14 #include "base/id_map.h" | 14 #include "base/id_map.h" |
| 15 #include "base/memory/scoped_vector.h" | 15 #include "base/memory/scoped_vector.h" |
| 16 #include "base/synchronization/waitable_event.h" | 16 #include "base/synchronization/waitable_event.h" |
| 17 #include "base/test/simple_test_clock.h" | 17 #include "base/test/simple_test_clock.h" |
| 18 #include "base/time/clock.h" | 18 #include "base/time/clock.h" |
| 19 #include "chrome/browser/chrome_notification_types.h" | 19 #include "chrome/browser/chrome_notification_types.h" |
| 20 #include "chrome/browser/content_settings/tab_specific_content_settings.h" | 20 #include "chrome/browser/content_settings/tab_specific_content_settings.h" |
| 21 #include "chrome/browser/geolocation/geolocation_permission_context_factory.h" | 21 #include "chrome/browser/geolocation/geolocation_permission_context_factory.h" |
| 22 #include "chrome/browser/infobars/infobar_service.h" | 22 #include "chrome/browser/infobars/infobar_service.h" |
| 23 #include "chrome/browser/permissions/permission_request_id.h" |
| 23 #include "chrome/browser/ui/website_settings/mock_permission_bubble_view.h" | 24 #include "chrome/browser/ui/website_settings/mock_permission_bubble_view.h" |
| 24 #include "chrome/browser/ui/website_settings/permission_bubble_manager.h" | 25 #include "chrome/browser/ui/website_settings/permission_bubble_manager.h" |
| 25 #include "chrome/browser/ui/website_settings/permission_bubble_request.h" | 26 #include "chrome/browser/ui/website_settings/permission_bubble_request.h" |
| 26 #include "chrome/common/chrome_switches.h" | 27 #include "chrome/common/chrome_switches.h" |
| 27 #include "chrome/test/base/chrome_render_view_host_test_harness.h" | 28 #include "chrome/test/base/chrome_render_view_host_test_harness.h" |
| 28 #include "chrome/test/base/testing_profile.h" | 29 #include "chrome/test/base/testing_profile.h" |
| 29 #include "components/content_settings/core/browser/host_content_settings_map.h" | 30 #include "components/content_settings/core/browser/host_content_settings_map.h" |
| 30 #include "components/content_settings/core/common/permission_request_id.h" | |
| 31 #include "components/infobars/core/confirm_infobar_delegate.h" | 31 #include "components/infobars/core/confirm_infobar_delegate.h" |
| 32 #include "components/infobars/core/infobar.h" | 32 #include "components/infobars/core/infobar.h" |
| 33 #include "content/public/browser/browser_thread.h" | 33 #include "content/public/browser/browser_thread.h" |
| 34 #include "content/public/browser/navigation_details.h" | 34 #include "content/public/browser/navigation_details.h" |
| 35 #include "content/public/browser/navigation_entry.h" | 35 #include "content/public/browser/navigation_entry.h" |
| 36 #include "content/public/browser/notification_observer.h" | 36 #include "content/public/browser/notification_observer.h" |
| 37 #include "content/public/browser/notification_registrar.h" | 37 #include "content/public/browser/notification_registrar.h" |
| 38 #include "content/public/browser/notification_service.h" | 38 #include "content/public/browser/notification_service.h" |
| 39 #include "content/public/browser/render_frame_host.h" | 39 #include "content/public/browser/render_frame_host.h" |
| 40 #include "content/public/browser/web_contents.h" | 40 #include "content/public/browser/web_contents.h" |
| (...skipping 68 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 109 | 109 |
| 110 // GeolocationPermissionContextTests ------------------------------------------ | 110 // GeolocationPermissionContextTests ------------------------------------------ |
| 111 | 111 |
| 112 class GeolocationPermissionContextTests | 112 class GeolocationPermissionContextTests |
| 113 : public ChromeRenderViewHostTestHarness { | 113 : public ChromeRenderViewHostTestHarness { |
| 114 protected: | 114 protected: |
| 115 // ChromeRenderViewHostTestHarness: | 115 // ChromeRenderViewHostTestHarness: |
| 116 void SetUp() override; | 116 void SetUp() override; |
| 117 void TearDown() override; | 117 void TearDown() override; |
| 118 | 118 |
| 119 PermissionRequestID RequestID(int bridge_id); | 119 PermissionRequestID RequestID(int request_id); |
| 120 PermissionRequestID RequestIDForTab(int tab, int bridge_id); | 120 PermissionRequestID RequestIDForTab(int tab, int request_id); |
| 121 InfoBarService* infobar_service() { | 121 InfoBarService* infobar_service() { |
| 122 return InfoBarService::FromWebContents(web_contents()); | 122 return InfoBarService::FromWebContents(web_contents()); |
| 123 } | 123 } |
| 124 InfoBarService* infobar_service_for_tab(int tab) { | 124 InfoBarService* infobar_service_for_tab(int tab) { |
| 125 return InfoBarService::FromWebContents(extra_tabs_[tab]); | 125 return InfoBarService::FromWebContents(extra_tabs_[tab]); |
| 126 } | 126 } |
| 127 | 127 |
| 128 void RequestGeolocationPermission(content::WebContents* web_contents, | 128 void RequestGeolocationPermission(content::WebContents* web_contents, |
| 129 const PermissionRequestID& id, | 129 const PermissionRequestID& id, |
| 130 const GURL& requesting_frame, | 130 const GURL& requesting_frame, |
| 131 bool user_gesture); | 131 bool user_gesture); |
| 132 | 132 |
| 133 void PermissionResponse(const PermissionRequestID& id, | 133 void PermissionResponse(const PermissionRequestID& id, |
| 134 ContentSetting content_setting); | 134 ContentSetting content_setting); |
| 135 void CheckPermissionMessageSent(int bridge_id, bool allowed); | 135 void CheckPermissionMessageSent(int request_id, bool allowed); |
| 136 void CheckPermissionMessageSentForTab(int tab, int bridge_id, bool allowed); | 136 void CheckPermissionMessageSentForTab(int tab, int request_id, bool allowed); |
| 137 void CheckPermissionMessageSentInternal(MockRenderProcessHost* process, | 137 void CheckPermissionMessageSentInternal(MockRenderProcessHost* process, |
| 138 int bridge_id, | 138 int request_id, |
| 139 bool allowed); | 139 bool allowed); |
| 140 void AddNewTab(const GURL& url); | 140 void AddNewTab(const GURL& url); |
| 141 void CheckTabContentsState(const GURL& requesting_frame, | 141 void CheckTabContentsState(const GURL& requesting_frame, |
| 142 ContentSetting expected_content_setting); | 142 ContentSetting expected_content_setting); |
| 143 size_t GetBubblesQueueSize(PermissionBubbleManager* manager); | 143 size_t GetBubblesQueueSize(PermissionBubbleManager* manager); |
| 144 void AcceptBubble(PermissionBubbleManager* manager); | 144 void AcceptBubble(PermissionBubbleManager* manager); |
| 145 void DenyBubble(PermissionBubbleManager* manager); | 145 void DenyBubble(PermissionBubbleManager* manager); |
| 146 void CloseBubble(PermissionBubbleManager* manager); | 146 void CloseBubble(PermissionBubbleManager* manager); |
| 147 void BubbleManagerDocumentLoadCompleted(); | 147 void BubbleManagerDocumentLoadCompleted(); |
| 148 void BubbleManagerDocumentLoadCompleted(content::WebContents* web_contents); | 148 void BubbleManagerDocumentLoadCompleted(content::WebContents* web_contents); |
| 149 ContentSetting GetGeolocationContentSetting(GURL frame_0, GURL frame_1); | 149 ContentSetting GetGeolocationContentSetting(GURL frame_0, GURL frame_1); |
| 150 | 150 |
| 151 // owned by the browser context | 151 // owned by the browser context |
| 152 GeolocationPermissionContext* geolocation_permission_context_; | 152 GeolocationPermissionContext* geolocation_permission_context_; |
| 153 ClosedInfoBarTracker closed_infobar_tracker_; | 153 ClosedInfoBarTracker closed_infobar_tracker_; |
| 154 MockPermissionBubbleView bubble_view_; | 154 MockPermissionBubbleView bubble_view_; |
| 155 ScopedVector<content::WebContents> extra_tabs_; | 155 ScopedVector<content::WebContents> extra_tabs_; |
| 156 | 156 |
| 157 // A map between renderer child id and a pair represending the bridge id and | 157 // A map between renderer child id and a pair represending the bridge id and |
| 158 // whether the requested permission was allowed. | 158 // whether the requested permission was allowed. |
| 159 base::hash_map<int, std::pair<int, bool> > responses_; | 159 base::hash_map<int, std::pair<int, bool> > responses_; |
| 160 }; | 160 }; |
| 161 | 161 |
| 162 PermissionRequestID GeolocationPermissionContextTests::RequestID( | 162 PermissionRequestID GeolocationPermissionContextTests::RequestID( |
| 163 int bridge_id) { | 163 int request_id) { |
| 164 return PermissionRequestID( | 164 return PermissionRequestID( |
| 165 web_contents()->GetRenderProcessHost()->GetID(), | 165 web_contents()->GetRenderProcessHost()->GetID(), |
| 166 web_contents()->GetMainFrame()->GetRoutingID(), | 166 web_contents()->GetMainFrame()->GetRoutingID(), |
| 167 bridge_id, | 167 request_id, |
| 168 GURL()); | 168 GURL()); |
| 169 } | 169 } |
| 170 | 170 |
| 171 PermissionRequestID GeolocationPermissionContextTests::RequestIDForTab( | 171 PermissionRequestID GeolocationPermissionContextTests::RequestIDForTab( |
| 172 int tab, | 172 int tab, |
| 173 int bridge_id) { | 173 int request_id) { |
| 174 return PermissionRequestID( | 174 return PermissionRequestID( |
| 175 extra_tabs_[tab]->GetRenderProcessHost()->GetID(), | 175 extra_tabs_[tab]->GetRenderProcessHost()->GetID(), |
| 176 extra_tabs_[tab]->GetMainFrame()->GetRoutingID(), | 176 extra_tabs_[tab]->GetMainFrame()->GetRoutingID(), |
| 177 bridge_id, | 177 request_id, |
| 178 GURL()); | 178 GURL()); |
| 179 } | 179 } |
| 180 | 180 |
| 181 void GeolocationPermissionContextTests::RequestGeolocationPermission( | 181 void GeolocationPermissionContextTests::RequestGeolocationPermission( |
| 182 content::WebContents* web_contents, | 182 content::WebContents* web_contents, |
| 183 const PermissionRequestID& id, | 183 const PermissionRequestID& id, |
| 184 const GURL& requesting_frame, | 184 const GURL& requesting_frame, |
| 185 bool user_gesture) { | 185 bool user_gesture) { |
| 186 geolocation_permission_context_->RequestPermission( | 186 geolocation_permission_context_->RequestPermission( |
| 187 web_contents, id, requesting_frame, user_gesture, | 187 web_contents, id, requesting_frame, user_gesture, |
| 188 base::Bind(&GeolocationPermissionContextTests::PermissionResponse, | 188 base::Bind(&GeolocationPermissionContextTests::PermissionResponse, |
| 189 base::Unretained(this), id)); | 189 base::Unretained(this), id)); |
| 190 content::RunAllBlockingPoolTasksUntilIdle(); | 190 content::RunAllBlockingPoolTasksUntilIdle(); |
| 191 } | 191 } |
| 192 | 192 |
| 193 void GeolocationPermissionContextTests::PermissionResponse( | 193 void GeolocationPermissionContextTests::PermissionResponse( |
| 194 const PermissionRequestID& id, | 194 const PermissionRequestID& id, |
| 195 ContentSetting content_setting) { | 195 ContentSetting content_setting) { |
| 196 responses_[id.render_process_id()] = | 196 responses_[id.render_process_id()] = |
| 197 std::make_pair(id.bridge_id(), content_setting == CONTENT_SETTING_ALLOW); | 197 std::make_pair(id.request_id(), content_setting == CONTENT_SETTING_ALLOW); |
| 198 } | 198 } |
| 199 | 199 |
| 200 void GeolocationPermissionContextTests::CheckPermissionMessageSent( | 200 void GeolocationPermissionContextTests::CheckPermissionMessageSent( |
| 201 int bridge_id, | 201 int request_id, |
| 202 bool allowed) { | 202 bool allowed) { |
| 203 CheckPermissionMessageSentInternal(process(), bridge_id, allowed); | 203 CheckPermissionMessageSentInternal(process(), request_id, allowed); |
| 204 } | 204 } |
| 205 | 205 |
| 206 void GeolocationPermissionContextTests::CheckPermissionMessageSentForTab( | 206 void GeolocationPermissionContextTests::CheckPermissionMessageSentForTab( |
| 207 int tab, | 207 int tab, |
| 208 int bridge_id, | 208 int request_id, |
| 209 bool allowed) { | 209 bool allowed) { |
| 210 CheckPermissionMessageSentInternal(static_cast<MockRenderProcessHost*>( | 210 CheckPermissionMessageSentInternal(static_cast<MockRenderProcessHost*>( |
| 211 extra_tabs_[tab]->GetRenderProcessHost()), | 211 extra_tabs_[tab]->GetRenderProcessHost()), |
| 212 bridge_id, allowed); | 212 request_id, allowed); |
| 213 } | 213 } |
| 214 | 214 |
| 215 void GeolocationPermissionContextTests::CheckPermissionMessageSentInternal( | 215 void GeolocationPermissionContextTests::CheckPermissionMessageSentInternal( |
| 216 MockRenderProcessHost* process, | 216 MockRenderProcessHost* process, |
| 217 int bridge_id, | 217 int request_id, |
| 218 bool allowed) { | 218 bool allowed) { |
| 219 ASSERT_EQ(responses_.count(process->GetID()), 1U); | 219 ASSERT_EQ(responses_.count(process->GetID()), 1U); |
| 220 EXPECT_EQ(bridge_id, responses_[process->GetID()].first); | 220 EXPECT_EQ(request_id, responses_[process->GetID()].first); |
| 221 EXPECT_EQ(allowed, responses_[process->GetID()].second); | 221 EXPECT_EQ(allowed, responses_[process->GetID()].second); |
| 222 responses_.erase(process->GetID()); | 222 responses_.erase(process->GetID()); |
| 223 } | 223 } |
| 224 | 224 |
| 225 void GeolocationPermissionContextTests::AddNewTab(const GURL& url) { | 225 void GeolocationPermissionContextTests::AddNewTab(const GURL& url) { |
| 226 content::WebContents* new_tab = CreateTestWebContents(); | 226 content::WebContents* new_tab = CreateTestWebContents(); |
| 227 new_tab->GetController().LoadURL( | 227 new_tab->GetController().LoadURL( |
| 228 url, content::Referrer(), ui::PAGE_TRANSITION_TYPED, std::string()); | 228 url, content::Referrer(), ui::PAGE_TRANSITION_TYPED, std::string()); |
| 229 content::NavigationEntry* entry = new_tab->GetController().GetPendingEntry(); | 229 content::NavigationEntry* entry = new_tab->GetController().GetPendingEntry(); |
| 230 content::RenderFrameHostTester::For(new_tab->GetMainFrame()) | 230 content::RenderFrameHostTester::For(new_tab->GetMainFrame()) |
| (...skipping 741 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 972 13); | 972 13); |
| 973 EXPECT_EQ(map->GetLastUsage(requesting_frame_1.GetOrigin(), | 973 EXPECT_EQ(map->GetLastUsage(requesting_frame_1.GetOrigin(), |
| 974 requesting_frame_0.GetOrigin(), | 974 requesting_frame_0.GetOrigin(), |
| 975 CONTENT_SETTINGS_TYPE_GEOLOCATION).ToDoubleT(), | 975 CONTENT_SETTINGS_TYPE_GEOLOCATION).ToDoubleT(), |
| 976 11); | 976 11); |
| 977 } | 977 } |
| 978 | 978 |
| 979 INSTANTIATE_TEST_CASE_P(GeolocationPermissionContextTestsWithAndWithoutBubbles, | 979 INSTANTIATE_TEST_CASE_P(GeolocationPermissionContextTestsWithAndWithoutBubbles, |
| 980 GeolocationPermissionContextParamTests, | 980 GeolocationPermissionContextParamTests, |
| 981 ::testing::Values(false, true)); | 981 ::testing::Values(false, true)); |
| OLD | NEW |