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