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 |