OLD | NEW |
1 // Copyright 2013 The Chromium Authors. All rights reserved. | 1 // Copyright 2013 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/android/webapps/single_tab_mode_tab_helper.h" | 5 #include "chrome/browser/android/webapps/single_tab_mode_tab_helper.h" |
6 | 6 |
7 #include "base/lazy_instance.h" | 7 #include "base/lazy_instance.h" |
8 #include "chrome/browser/android/tab_android.h" | 8 #include "chrome/browser/android/tab_android.h" |
9 #include "content/public/browser/browser_thread.h" | 9 #include "content/public/browser/browser_thread.h" |
| 10 #include "content/public/browser/render_frame_host.h" |
10 #include "content/public/browser/render_process_host.h" | 11 #include "content/public/browser/render_process_host.h" |
11 #include "content/public/browser/render_view_host.h" | |
12 #include "content/public/browser/web_contents.h" | 12 #include "content/public/browser/web_contents.h" |
13 | 13 |
14 DEFINE_WEB_CONTENTS_USER_DATA_KEY(SingleTabModeTabHelper); | 14 DEFINE_WEB_CONTENTS_USER_DATA_KEY(SingleTabModeTabHelper); |
15 | 15 |
16 namespace { | 16 namespace { |
17 | 17 |
18 typedef std::pair<int32_t, int32_t> RenderWidgetHostID; | 18 typedef std::pair<int32_t, int32_t> RenderFrameHostID; |
19 typedef std::set<RenderWidgetHostID> SingleTabIDSet; | 19 typedef std::set<RenderFrameHostID> SingleTabIDSet; |
20 base::LazyInstance<SingleTabIDSet> g_blocked_ids = LAZY_INSTANCE_INITIALIZER; | 20 base::LazyInstance<SingleTabIDSet> g_blocked_ids = LAZY_INSTANCE_INITIALIZER; |
21 | 21 |
22 void AddPairOnIOThread(int32_t process_id, int32_t routing_id) { | 22 void AddPairOnIOThread(int32_t process_id, int32_t frame_routing_id) { |
23 DCHECK_CURRENTLY_ON(content::BrowserThread::IO); | 23 DCHECK_CURRENTLY_ON(content::BrowserThread::IO); |
24 RenderWidgetHostID single_tab_pair(process_id, routing_id); | 24 RenderFrameHostID single_tab_pair(process_id, frame_routing_id); |
25 g_blocked_ids.Get().insert(single_tab_pair); | 25 g_blocked_ids.Get().insert(single_tab_pair); |
26 } | 26 } |
27 | 27 |
28 void RemovePairOnIOThread(int32_t process_id, int32_t routing_id) { | 28 void RemovePairOnIOThread(int32_t process_id, int32_t frame_routing_id) { |
29 DCHECK_CURRENTLY_ON(content::BrowserThread::IO); | 29 DCHECK_CURRENTLY_ON(content::BrowserThread::IO); |
30 RenderWidgetHostID single_tab_pair(process_id, routing_id); | 30 RenderFrameHostID single_tab_pair(process_id, frame_routing_id); |
31 SingleTabIDSet::iterator itr = g_blocked_ids.Get().find(single_tab_pair); | 31 SingleTabIDSet::iterator itr = g_blocked_ids.Get().find(single_tab_pair); |
32 DCHECK(itr != g_blocked_ids.Get().end()); | 32 DCHECK(itr != g_blocked_ids.Get().end()); |
33 g_blocked_ids.Get().erase(itr); | 33 g_blocked_ids.Get().erase(itr); |
34 } | 34 } |
35 | 35 |
36 void AddPair(content::RenderViewHost* render_view_host) { | 36 void AddPair(content::RenderFrameHost* render_frame_host) { |
37 if (!render_view_host) | 37 if (!render_frame_host) |
38 return; | 38 return; |
39 | 39 |
40 int32_t process_id = render_view_host->GetProcess()->GetID(); | 40 int32_t process_id = render_frame_host->GetProcess()->GetID(); |
41 int32_t routing_id = render_view_host->GetRoutingID(); | 41 int32_t frame_routing_id = render_frame_host->GetRoutingID(); |
42 content::BrowserThread::PostTask( | 42 content::BrowserThread::PostTask( |
43 content::BrowserThread::IO, FROM_HERE, | 43 content::BrowserThread::IO, FROM_HERE, |
44 base::Bind(&AddPairOnIOThread, process_id, routing_id)); | 44 base::Bind(&AddPairOnIOThread, process_id, frame_routing_id)); |
45 } | 45 } |
46 | 46 |
47 void RemovePair(content::RenderViewHost* render_view_host) { | 47 void RemovePair(content::RenderFrameHost* render_frame_host) { |
48 if (!render_view_host) | 48 if (!render_frame_host) |
49 return; | 49 return; |
50 | 50 |
51 int32_t process_id = render_view_host->GetProcess()->GetID(); | 51 int32_t process_id = render_frame_host->GetProcess()->GetID(); |
52 int32_t routing_id = render_view_host->GetRoutingID(); | 52 int32_t frame_routing_id = render_frame_host->GetRoutingID(); |
53 content::BrowserThread::PostTask( | 53 content::BrowserThread::PostTask( |
54 content::BrowserThread::IO, FROM_HERE, | 54 content::BrowserThread::IO, FROM_HERE, |
55 base::Bind(&RemovePairOnIOThread, process_id, routing_id)); | 55 base::Bind(&RemovePairOnIOThread, process_id, frame_routing_id)); |
56 } | 56 } |
57 | 57 |
58 } // namespace | 58 } // namespace |
59 | 59 |
60 SingleTabModeTabHelper::SingleTabModeTabHelper( | 60 SingleTabModeTabHelper::SingleTabModeTabHelper( |
61 content::WebContents* web_contents) | 61 content::WebContents* web_contents) |
62 : content::WebContentsObserver(web_contents), | 62 : content::WebContentsObserver(web_contents), |
63 block_all_new_windows_(false) { | 63 block_all_new_windows_(false) { |
64 } | 64 } |
65 | 65 |
66 SingleTabModeTabHelper::~SingleTabModeTabHelper() { | 66 SingleTabModeTabHelper::~SingleTabModeTabHelper() { |
67 } | 67 } |
68 | 68 |
69 void SingleTabModeTabHelper::RenderViewCreated( | 69 void SingleTabModeTabHelper::RenderFrameCreated( |
70 content::RenderViewHost* render_view_host) { | 70 content::RenderFrameHost* render_frame_host) { |
71 if (!block_all_new_windows_) | 71 if (!block_all_new_windows_) |
72 return; | 72 return; |
73 AddPair(render_view_host); | 73 AddPair(render_frame_host); |
74 } | 74 } |
75 | 75 |
76 void SingleTabModeTabHelper::RenderViewDeleted( | 76 void SingleTabModeTabHelper::RenderFrameDeleted( |
77 content::RenderViewHost* render_view_host) { | 77 content::RenderFrameHost* render_frame_host) { |
78 if (!block_all_new_windows_) | 78 if (!block_all_new_windows_) |
79 return; | 79 return; |
80 RemovePair(render_view_host); | 80 RemovePair(render_frame_host); |
81 } | 81 } |
82 | 82 |
83 void SingleTabModeTabHelper::PermanentlyBlockAllNewWindows() { | 83 void SingleTabModeTabHelper::PermanentlyBlockAllNewWindows() { |
84 block_all_new_windows_ = true; | 84 block_all_new_windows_ = true; |
85 AddPair(web_contents()->GetRenderViewHost()); | 85 for (content::RenderFrameHost* frame : web_contents()->GetAllFrames()) |
| 86 AddPair(frame); |
86 } | 87 } |
87 | 88 |
88 bool SingleTabModeTabHelper::IsRegistered(int32_t process_id, | 89 bool SingleTabModeTabHelper::IsRegistered(int32_t process_id, |
89 int32_t routing_id) { | 90 int32_t frame_routing_id) { |
90 DCHECK_CURRENTLY_ON(content::BrowserThread::IO); | 91 DCHECK_CURRENTLY_ON(content::BrowserThread::IO); |
91 RenderWidgetHostID single_tab_pair(process_id, routing_id); | 92 RenderFrameHostID single_tab_pair(process_id, frame_routing_id); |
92 SingleTabIDSet::iterator itr = g_blocked_ids.Get().find(single_tab_pair); | 93 SingleTabIDSet::iterator itr = g_blocked_ids.Get().find(single_tab_pair); |
93 return itr != g_blocked_ids.Get().end(); | 94 return itr != g_blocked_ids.Get().end(); |
94 } | 95 } |
95 | 96 |
96 void SingleTabModeTabHelper::HandleOpenUrl(const BlockedWindowParams ¶ms) { | 97 void SingleTabModeTabHelper::HandleOpenUrl(const BlockedWindowParams& params) { |
97 TabAndroid* tab = TabAndroid::FromWebContents(web_contents()); | 98 TabAndroid* tab = TabAndroid::FromWebContents(web_contents()); |
98 if (!tab) | 99 if (!tab) |
99 return; | 100 return; |
100 | 101 |
101 chrome::NavigateParams nav_params = | 102 chrome::NavigateParams nav_params = |
102 params.CreateNavigateParams(web_contents()); | 103 params.CreateNavigateParams(web_contents()); |
103 tab->HandlePopupNavigation(&nav_params); | 104 tab->HandlePopupNavigation(&nav_params); |
104 } | 105 } |
OLD | NEW |