OLD | NEW |
| (Empty) |
1 // Copyright (c) 2012 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/managed_mode/managed_mode_navigation_observer.h" | |
6 | |
7 #include "base/bind.h" | |
8 #include "base/callback.h" | |
9 #include "base/metrics/histogram.h" | |
10 #include "base/strings/string_util.h" | |
11 #include "chrome/browser/history/history_service.h" | |
12 #include "chrome/browser/history/history_service_factory.h" | |
13 #include "chrome/browser/history/history_types.h" | |
14 #include "chrome/browser/infobars/infobar_service.h" | |
15 #include "chrome/browser/managed_mode/managed_mode_interstitial.h" | |
16 #include "chrome/browser/managed_mode/managed_mode_resource_throttle.h" | |
17 #include "chrome/browser/managed_mode/managed_mode_url_filter.h" | |
18 #include "chrome/browser/managed_mode/managed_user_service.h" | |
19 #include "chrome/browser/managed_mode/managed_user_service_factory.h" | |
20 #include "chrome/browser/profiles/profile.h" | |
21 #include "chrome/browser/tab_contents/tab_util.h" | |
22 #include "components/infobars/core/confirm_infobar_delegate.h" | |
23 #include "components/infobars/core/infobar.h" | |
24 #include "content/public/browser/browser_thread.h" | |
25 #include "content/public/browser/navigation_entry.h" | |
26 #include "content/public/browser/render_process_host.h" | |
27 #include "content/public/browser/render_view_host.h" | |
28 #include "content/public/browser/user_metrics.h" | |
29 #include "grit/generated_resources.h" | |
30 #include "ui/base/l10n/l10n_util.h" | |
31 | |
32 #if !defined(OS_ANDROID) | |
33 #include "chrome/browser/ui/browser.h" | |
34 #include "chrome/browser/ui/browser_commands.h" | |
35 #include "chrome/browser/ui/browser_finder.h" | |
36 #include "chrome/browser/ui/browser_list.h" | |
37 #include "chrome/browser/ui/host_desktop.h" | |
38 #include "chrome/browser/ui/tabs/tab_strip_model.h" | |
39 #endif | |
40 | |
41 using base::Time; | |
42 using content::NavigationEntry; | |
43 | |
44 namespace { | |
45 | |
46 | |
47 // Helpers -------------------------------------------------------------------- | |
48 | |
49 #if !defined(OS_ANDROID) | |
50 // TODO(bauerb): Get rid of the platform-specific #ifdef here. | |
51 // http://crbug.com/313377 | |
52 void GoBackToSafety(content::WebContents* web_contents) { | |
53 // For now, just go back one page (the user didn't retreat from that page, | |
54 // so it should be okay). | |
55 content::NavigationController* controller = | |
56 &web_contents->GetController(); | |
57 if (controller->CanGoBack()) { | |
58 controller->GoBack(); | |
59 return; | |
60 } | |
61 | |
62 // If we can't go back (because we opened a new tab), try to close the tab. | |
63 // If this is the last tab on this desktop, open a new window. | |
64 chrome::HostDesktopType host_desktop_type = | |
65 chrome::GetHostDesktopTypeForNativeView(web_contents->GetNativeView()); | |
66 const BrowserList* browser_list = BrowserList::GetInstance(host_desktop_type); | |
67 if (browser_list->size() == 1) { | |
68 Browser* browser = browser_list->get(0); | |
69 DCHECK(browser == chrome::FindBrowserWithWebContents(web_contents)); | |
70 if (browser->tab_strip_model()->count() == 1) | |
71 chrome::NewEmptyWindow(browser->profile(), browser->host_desktop_type()); | |
72 } | |
73 | |
74 web_contents->GetDelegate()->CloseContents(web_contents); | |
75 } | |
76 #endif | |
77 | |
78 // ManagedModeWarningInfoBarDelegate ------------------------------------------ | |
79 | |
80 class ManagedModeWarningInfoBarDelegate : public ConfirmInfoBarDelegate { | |
81 public: | |
82 // Creates a managed mode warning infobar and delegate and adds the infobar to | |
83 // |infobar_service|. Returns the infobar if it was successfully added. | |
84 static infobars::InfoBar* Create(InfoBarService* infobar_service); | |
85 | |
86 private: | |
87 ManagedModeWarningInfoBarDelegate(); | |
88 virtual ~ManagedModeWarningInfoBarDelegate(); | |
89 | |
90 // ConfirmInfoBarDelegate: | |
91 virtual bool ShouldExpire(const NavigationDetails& details) const OVERRIDE; | |
92 virtual void InfoBarDismissed() OVERRIDE; | |
93 virtual base::string16 GetMessageText() const OVERRIDE; | |
94 virtual int GetButtons() const OVERRIDE; | |
95 virtual base::string16 GetButtonLabel(InfoBarButton button) const OVERRIDE; | |
96 virtual bool Accept() OVERRIDE; | |
97 | |
98 DISALLOW_COPY_AND_ASSIGN(ManagedModeWarningInfoBarDelegate); | |
99 }; | |
100 | |
101 // static | |
102 infobars::InfoBar* ManagedModeWarningInfoBarDelegate::Create( | |
103 InfoBarService* infobar_service) { | |
104 return infobar_service->AddInfoBar(ConfirmInfoBarDelegate::CreateInfoBar( | |
105 scoped_ptr<ConfirmInfoBarDelegate>( | |
106 new ManagedModeWarningInfoBarDelegate()))); | |
107 } | |
108 | |
109 ManagedModeWarningInfoBarDelegate::ManagedModeWarningInfoBarDelegate() | |
110 : ConfirmInfoBarDelegate() { | |
111 } | |
112 | |
113 ManagedModeWarningInfoBarDelegate::~ManagedModeWarningInfoBarDelegate() { | |
114 } | |
115 | |
116 bool ManagedModeWarningInfoBarDelegate::ShouldExpire( | |
117 const NavigationDetails& details) const { | |
118 // ManagedModeNavigationObserver removes us below. | |
119 return false; | |
120 } | |
121 | |
122 void ManagedModeWarningInfoBarDelegate::InfoBarDismissed() { | |
123 content::WebContents* web_contents = | |
124 InfoBarService::WebContentsFromInfoBar(infobar()); | |
125 ManagedModeNavigationObserver::FromWebContents( | |
126 web_contents)->WarnInfoBarDismissed(); | |
127 } | |
128 | |
129 base::string16 ManagedModeWarningInfoBarDelegate::GetMessageText() const { | |
130 return l10n_util::GetStringUTF16(IDS_MANAGED_USER_WARN_INFOBAR_MESSAGE); | |
131 } | |
132 | |
133 int ManagedModeWarningInfoBarDelegate::GetButtons() const { | |
134 return BUTTON_OK; | |
135 } | |
136 | |
137 base::string16 ManagedModeWarningInfoBarDelegate::GetButtonLabel( | |
138 InfoBarButton button) const { | |
139 DCHECK_EQ(BUTTON_OK, button); | |
140 return l10n_util::GetStringUTF16(IDS_MANAGED_USER_WARN_INFOBAR_GO_BACK); | |
141 } | |
142 | |
143 bool ManagedModeWarningInfoBarDelegate::Accept() { | |
144 #if defined(OS_ANDROID) | |
145 // TODO(bauerb): Get rid of the platform-specific #ifdef here. | |
146 // http://crbug.com/313377 | |
147 NOTIMPLEMENTED(); | |
148 #else | |
149 GoBackToSafety(InfoBarService::WebContentsFromInfoBar(infobar())); | |
150 #endif | |
151 | |
152 return false; | |
153 } | |
154 | |
155 | |
156 } // namespace | |
157 | |
158 // ManagedModeNavigationObserver ---------------------------------------------- | |
159 | |
160 DEFINE_WEB_CONTENTS_USER_DATA_KEY(ManagedModeNavigationObserver); | |
161 | |
162 ManagedModeNavigationObserver::~ManagedModeNavigationObserver() { | |
163 } | |
164 | |
165 ManagedModeNavigationObserver::ManagedModeNavigationObserver( | |
166 content::WebContents* web_contents) | |
167 : WebContentsObserver(web_contents), | |
168 warn_infobar_(NULL) { | |
169 Profile* profile = | |
170 Profile::FromBrowserContext(web_contents->GetBrowserContext()); | |
171 managed_user_service_ = ManagedUserServiceFactory::GetForProfile(profile); | |
172 url_filter_ = managed_user_service_->GetURLFilterForUIThread(); | |
173 } | |
174 | |
175 void ManagedModeNavigationObserver::WarnInfoBarDismissed() { | |
176 DCHECK(warn_infobar_); | |
177 warn_infobar_ = NULL; | |
178 } | |
179 | |
180 void ManagedModeNavigationObserver::ProvisionalChangeToMainFrameUrl( | |
181 const GURL& url, | |
182 content::RenderFrameHost* render_frame_host) { | |
183 ManagedModeURLFilter::FilteringBehavior behavior = | |
184 url_filter_->GetFilteringBehaviorForURL(url); | |
185 | |
186 if (behavior == ManagedModeURLFilter::WARN || !warn_infobar_) | |
187 return; | |
188 | |
189 // If we shouldn't have a warn infobar remove it here. | |
190 InfoBarService::FromWebContents(web_contents())->RemoveInfoBar(warn_infobar_); | |
191 warn_infobar_ = NULL; | |
192 } | |
193 | |
194 void ManagedModeNavigationObserver::DidCommitProvisionalLoadForFrame( | |
195 int64 frame_id, | |
196 const base::string16& frame_unique_name, | |
197 bool is_main_frame, | |
198 const GURL& url, | |
199 content::PageTransition transition_type, | |
200 content::RenderViewHost* render_view_host) { | |
201 if (!is_main_frame) | |
202 return; | |
203 | |
204 DVLOG(1) << "DidCommitProvisionalLoadForFrame " << url.spec(); | |
205 ManagedModeURLFilter::FilteringBehavior behavior = | |
206 url_filter_->GetFilteringBehaviorForURL(url); | |
207 | |
208 if (behavior == ManagedModeURLFilter::WARN && !warn_infobar_) { | |
209 warn_infobar_ = ManagedModeWarningInfoBarDelegate::Create( | |
210 InfoBarService::FromWebContents(web_contents())); | |
211 } | |
212 } | |
213 | |
214 // static | |
215 void ManagedModeNavigationObserver::OnRequestBlocked( | |
216 int render_process_host_id, | |
217 int render_view_id, | |
218 const GURL& url, | |
219 const base::Callback<void(bool)>& callback) { | |
220 content::WebContents* web_contents = | |
221 tab_util::GetWebContentsByID(render_process_host_id, render_view_id); | |
222 if (!web_contents) { | |
223 content::BrowserThread::PostTask( | |
224 content::BrowserThread::IO, FROM_HERE, base::Bind(callback, false)); | |
225 return; | |
226 } | |
227 | |
228 ManagedModeNavigationObserver* navigation_observer = | |
229 ManagedModeNavigationObserver::FromWebContents(web_contents); | |
230 if (navigation_observer) | |
231 navigation_observer->OnRequestBlockedInternal(url); | |
232 | |
233 // Show the interstitial. | |
234 ManagedModeInterstitial::Show(web_contents, url, callback); | |
235 } | |
236 | |
237 void ManagedModeNavigationObserver::OnRequestBlockedInternal(const GURL& url) { | |
238 Time timestamp = Time::Now(); // TODO(bauerb): Use SaneTime when available. | |
239 // Create a history entry for the attempt and mark it as such. | |
240 history::HistoryAddPageArgs add_page_args( | |
241 url, timestamp, web_contents(), 0, | |
242 url, history::RedirectList(), | |
243 content::PAGE_TRANSITION_BLOCKED, history::SOURCE_BROWSED, | |
244 false); | |
245 | |
246 // Add the entry to the history database. | |
247 Profile* profile = | |
248 Profile::FromBrowserContext(web_contents()->GetBrowserContext()); | |
249 HistoryService* history_service = | |
250 HistoryServiceFactory::GetForProfile(profile, Profile::IMPLICIT_ACCESS); | |
251 | |
252 // |history_service| is null if saving history is disabled. | |
253 if (history_service) | |
254 history_service->AddPage(add_page_args); | |
255 | |
256 scoped_ptr<NavigationEntry> entry(NavigationEntry::Create()); | |
257 entry->SetVirtualURL(url); | |
258 entry->SetTimestamp(timestamp); | |
259 blocked_navigations_.push_back(entry.release()); | |
260 ManagedUserService* managed_user_service = | |
261 ManagedUserServiceFactory::GetForProfile(profile); | |
262 managed_user_service->DidBlockNavigation(web_contents()); | |
263 } | |
OLD | NEW |