| 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/permission_type.h" | |
| 12 #include "content/public/browser/render_frame_host.h" | 11 #include "content/public/browser/render_frame_host.h" |
| 13 #include "content/public/browser/web_contents.h" | 12 #include "content/public/browser/web_contents.h" |
| 14 #include "content/public/browser/web_contents_observer.h" | 13 #include "content/public/browser/web_contents_observer.h" |
| 15 | 14 |
| 16 class DelegationTracker::DelegatedForChild : content::WebContentsObserver { | 15 class DelegationTracker::DelegatedForChild : content::WebContentsObserver { |
| 17 public: | 16 public: |
| 18 DelegatedForChild(content::RenderFrameHost* child_rfh, | 17 DelegatedForChild(content::RenderFrameHost* child_rfh, |
| 19 const std::vector<content::PermissionType>& permissions, | 18 const std::vector<ContentSettingsType>& permissions, |
| 20 const base::Callback<void(content::RenderFrameHost*)>& | 19 const base::Callback<void(content::RenderFrameHost*)>& |
| 21 rfh_destroyed_callback) | 20 rfh_destroyed_callback) |
| 22 : content::WebContentsObserver( | 21 : content::WebContentsObserver( |
| 23 content::WebContents::FromRenderFrameHost(child_rfh)), | 22 content::WebContents::FromRenderFrameHost(child_rfh)), |
| 24 child_rfh_(child_rfh), | 23 child_rfh_(child_rfh), |
| 25 permissions_(permissions.begin(), permissions.end()), | 24 permissions_(permissions.begin(), permissions.end()), |
| 26 rfh_destroyed_callback_(rfh_destroyed_callback) {} | 25 rfh_destroyed_callback_(rfh_destroyed_callback) {} |
| 27 | 26 |
| 28 ~DelegatedForChild() override {} | 27 ~DelegatedForChild() override {} |
| 29 | 28 |
| 30 bool HasPermission(const content::PermissionType& permission) { | 29 bool HasPermission(ContentSettingsType permission) { |
| 31 return permissions_.count(permission) == 1; | 30 return permissions_.count(permission) == 1; |
| 32 } | 31 } |
| 33 | 32 |
| 34 private: | 33 private: |
| 35 DISALLOW_COPY_AND_ASSIGN(DelegatedForChild); | 34 DISALLOW_COPY_AND_ASSIGN(DelegatedForChild); |
| 36 | 35 |
| 37 void ClearPermissions(content::RenderFrameHost* render_frame_host) { | 36 void ClearPermissions(content::RenderFrameHost* render_frame_host) { |
| 38 if (render_frame_host == child_rfh_) | 37 if (render_frame_host == child_rfh_) |
| 39 rfh_destroyed_callback_.Run(render_frame_host); // Will delete |this|. | 38 rfh_destroyed_callback_.Run(render_frame_host); // Will delete |this|. |
| 40 } | 39 } |
| (...skipping 10 matching lines...) Expand all Loading... |
| 51 | 50 |
| 52 void DidNavigateAnyFrame( | 51 void DidNavigateAnyFrame( |
| 53 content::RenderFrameHost* render_frame_host, | 52 content::RenderFrameHost* render_frame_host, |
| 54 const content::LoadCommittedDetails& details, | 53 const content::LoadCommittedDetails& details, |
| 55 const content::FrameNavigateParams& params) override { | 54 const content::FrameNavigateParams& params) override { |
| 56 ClearPermissions(render_frame_host); | 55 ClearPermissions(render_frame_host); |
| 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 |