| OLD | NEW |
| 1 // Copyright 2016 The Chromium Authors. All rights reserved. | 1 // Copyright 2016 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/delegation_tracker.h" | 5 #include "chrome/browser/permissions/delegation_tracker.h" |
| 6 | 6 |
| 7 #include <unordered_set> | 7 #include <unordered_set> |
| 8 | 8 |
| 9 #include "base/memory/ptr_util.h" | 9 #include "base/memory/ptr_util.h" |
| 10 #include "chrome/browser/permissions/permission_util.h" | 10 #include "chrome/browser/permissions/permission_util.h" |
| 11 #include "content/public/browser/navigation_handle.h" | 11 #include "content/public/browser/navigation_handle.h" |
| 12 #include "content/public/browser/permission_type.h" | |
| 13 #include "content/public/browser/render_frame_host.h" | 12 #include "content/public/browser/render_frame_host.h" |
| 14 #include "content/public/browser/web_contents.h" | 13 #include "content/public/browser/web_contents.h" |
| 15 #include "content/public/browser/web_contents_observer.h" | 14 #include "content/public/browser/web_contents_observer.h" |
| 16 | 15 |
| 17 class DelegationTracker::DelegatedForChild : content::WebContentsObserver { | 16 class DelegationTracker::DelegatedForChild : content::WebContentsObserver { |
| 18 public: | 17 public: |
| 19 DelegatedForChild(content::RenderFrameHost* child_rfh, | 18 DelegatedForChild(content::RenderFrameHost* child_rfh, |
| 20 const std::vector<content::PermissionType>& permissions, | 19 const std::vector<ContentSettingsType>& permissions, |
| 21 const base::Callback<void(content::RenderFrameHost*)>& | 20 const base::Callback<void(content::RenderFrameHost*)>& |
| 22 rfh_destroyed_callback) | 21 rfh_destroyed_callback) |
| 23 : content::WebContentsObserver( | 22 : content::WebContentsObserver( |
| 24 content::WebContents::FromRenderFrameHost(child_rfh)), | 23 content::WebContents::FromRenderFrameHost(child_rfh)), |
| 25 child_rfh_(child_rfh), | 24 child_rfh_(child_rfh), |
| 26 permissions_(permissions.begin(), permissions.end()), | 25 permissions_(permissions.begin(), permissions.end()), |
| 27 rfh_destroyed_callback_(rfh_destroyed_callback) {} | 26 rfh_destroyed_callback_(rfh_destroyed_callback) {} |
| 28 | 27 |
| 29 ~DelegatedForChild() override {} | 28 ~DelegatedForChild() override {} |
| 30 | 29 |
| 31 bool HasPermission(const content::PermissionType& permission) { | 30 bool HasPermission(ContentSettingsType permission) { |
| 32 return permissions_.count(permission) == 1; | 31 return permissions_.count(permission) == 1; |
| 33 } | 32 } |
| 34 | 33 |
| 35 private: | 34 private: |
| 36 DISALLOW_COPY_AND_ASSIGN(DelegatedForChild); | 35 DISALLOW_COPY_AND_ASSIGN(DelegatedForChild); |
| 37 | 36 |
| 38 void ClearPermissions(content::RenderFrameHost* render_frame_host) { | 37 void ClearPermissions(content::RenderFrameHost* render_frame_host) { |
| 39 if (render_frame_host == child_rfh_) | 38 if (render_frame_host == child_rfh_) |
| 40 rfh_destroyed_callback_.Run(render_frame_host); // Will delete |this|. | 39 rfh_destroyed_callback_.Run(render_frame_host); // Will delete |this|. |
| 41 } | 40 } |
| 42 | 41 |
| 43 // WebContentsObserver | 42 // WebContentsObserver |
| 44 void RenderFrameHostChanged(content::RenderFrameHost* old_host, | 43 void RenderFrameHostChanged(content::RenderFrameHost* old_host, |
| 45 content::RenderFrameHost* new_host) override { | 44 content::RenderFrameHost* new_host) override { |
| 46 ClearPermissions(old_host); | 45 ClearPermissions(old_host); |
| 47 } | 46 } |
| 48 | 47 |
| 49 void FrameDeleted(content::RenderFrameHost* render_frame_host) override { | 48 void FrameDeleted(content::RenderFrameHost* render_frame_host) override { |
| 50 ClearPermissions(render_frame_host); | 49 ClearPermissions(render_frame_host); |
| 51 } | 50 } |
| 52 | 51 |
| 53 void DidFinishNavigation( | 52 void DidFinishNavigation( |
| 54 content::NavigationHandle* navigation_handle) override { | 53 content::NavigationHandle* navigation_handle) override { |
| 55 if (navigation_handle->HasCommitted()) | 54 if (navigation_handle->HasCommitted()) |
| 56 ClearPermissions(navigation_handle->GetRenderFrameHost()); | 55 ClearPermissions(navigation_handle->GetRenderFrameHost()); |
| 57 } | 56 } |
| 58 | 57 |
| 59 content::RenderFrameHost* child_rfh_; | 58 content::RenderFrameHost* child_rfh_; |
| 60 | 59 |
| 61 std::unordered_set<content::PermissionType, PermissionTypeHash> permissions_; | 60 std::unordered_set<ContentSettingsType, ContentSettingsTypeHash> permissions_; |
| 62 | 61 |
| 63 base::Callback<void(content::RenderFrameHost*)> rfh_destroyed_callback_; | 62 base::Callback<void(content::RenderFrameHost*)> rfh_destroyed_callback_; |
| 64 }; | 63 }; |
| 65 | 64 |
| 66 DelegationTracker::DelegationTracker() {} | 65 DelegationTracker::DelegationTracker() {} |
| 67 | 66 |
| 68 DelegationTracker::~DelegationTracker() {} | 67 DelegationTracker::~DelegationTracker() {} |
| 69 | 68 |
| 70 void DelegationTracker::SetDelegatedPermissions( | 69 void DelegationTracker::SetDelegatedPermissions( |
| 71 content::RenderFrameHost* child_rfh, | 70 content::RenderFrameHost* child_rfh, |
| 72 const std::vector<content::PermissionType>& permissions) { | 71 const std::vector<ContentSettingsType>& permissions) { |
| 73 DCHECK(child_rfh && child_rfh->GetParent()); | 72 DCHECK(child_rfh && child_rfh->GetParent()); |
| 74 delegated_permissions_[child_rfh] = base::MakeUnique<DelegatedForChild>( | 73 delegated_permissions_[child_rfh] = base::MakeUnique<DelegatedForChild>( |
| 75 child_rfh, permissions, | 74 child_rfh, permissions, |
| 76 base::Bind(&DelegationTracker::RenderFrameHostChanged, | 75 base::Bind(&DelegationTracker::RenderFrameHostChanged, |
| 77 base::Unretained(this))); | 76 base::Unretained(this))); |
| 78 } | 77 } |
| 79 | 78 |
| 80 bool DelegationTracker::IsGranted(content::RenderFrameHost* requesting_rfh, | 79 bool DelegationTracker::IsGranted(content::RenderFrameHost* requesting_rfh, |
| 81 const content::PermissionType& permission) { | 80 ContentSettingsType permission) { |
| 82 content::RenderFrameHost* child_rfh = requesting_rfh; | 81 content::RenderFrameHost* child_rfh = requesting_rfh; |
| 83 while (child_rfh->GetParent()) { | 82 while (child_rfh->GetParent()) { |
| 84 // Parents with unique origins can't delegate permission. | 83 // Parents with unique origins can't delegate permission. |
| 85 url::Origin parent_origin = | 84 url::Origin parent_origin = |
| 86 child_rfh->GetParent()->GetLastCommittedOrigin(); | 85 child_rfh->GetParent()->GetLastCommittedOrigin(); |
| 87 if (parent_origin.unique()) | 86 if (parent_origin.unique()) |
| 88 return false; | 87 return false; |
| 89 | 88 |
| 90 if (!child_rfh->GetLastCommittedOrigin().IsSameOriginWith(parent_origin)) { | 89 if (!child_rfh->GetLastCommittedOrigin().IsSameOriginWith(parent_origin)) { |
| 91 const auto& it = delegated_permissions_.find(child_rfh); | 90 const auto& it = delegated_permissions_.find(child_rfh); |
| 92 if (it == delegated_permissions_.end()) | 91 if (it == delegated_permissions_.end()) |
| 93 return false; | 92 return false; |
| 94 if (!it->second->HasPermission(permission)) | 93 if (!it->second->HasPermission(permission)) |
| 95 return false; | 94 return false; |
| 96 } | 95 } |
| 97 child_rfh = child_rfh->GetParent(); | 96 child_rfh = child_rfh->GetParent(); |
| 98 } | 97 } |
| 99 return true; | 98 return true; |
| 100 } | 99 } |
| 101 | 100 |
| 102 void DelegationTracker::RenderFrameHostChanged(content::RenderFrameHost* rfh) { | 101 void DelegationTracker::RenderFrameHostChanged(content::RenderFrameHost* rfh) { |
| 103 delegated_permissions_.erase(rfh); | 102 delegated_permissions_.erase(rfh); |
| 104 } | 103 } |
| OLD | NEW |