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 // Implementation of the MalwareDetailsRedirectsCollector class.
6
7 #include "chrome/browser/safe_browsing/malware_details_history.h"
8
9 #include "base/bind.h"
10 #include "base/bind_helpers.h"
11 #include "chrome/browser/chrome_notification_types.h"
12 #include "chrome/browser/history/history_service_factory.h"
13 #include "chrome/browser/profiles/profile.h"
14 #include "chrome/browser/safe_browsing/malware_details.h"
15 #include "content/public/browser/browser_thread.h"
16 #include "content/public/browser/notification_details.h"
17 #include "content/public/browser/notification_source.h"
18 #include "content/public/browser/render_view_host.h"
19 #include "content/public/browser/web_contents.h"
20
21 using content::BrowserThread;
22
MalwareDetailsRedirectsCollector(Profile * profile)23 MalwareDetailsRedirectsCollector::MalwareDetailsRedirectsCollector(
24 Profile* profile)
25 : profile_(profile),
26 has_started_(false) {
27 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI));
28 if (profile) {
29 registrar_.Add(this, chrome::NOTIFICATION_PROFILE_DESTROYED,
30 content::Source<Profile>(profile));
31 }
32 }
33
StartHistoryCollection(const std::vector<GURL> & urls,const base::Closure & callback)34 void MalwareDetailsRedirectsCollector::StartHistoryCollection(
35 const std::vector<GURL>& urls,
36 const base::Closure& callback) {
37 DVLOG(1) << "Num of urls to check in history service: " << urls.size();
38 has_started_ = true;
39 callback_ = callback;
40
41 if (urls.size() == 0) {
42 AllDone();
43 return;
44 }
45
46 BrowserThread::PostTask(
47 BrowserThread::UI, FROM_HERE,
48 base::Bind(&MalwareDetailsRedirectsCollector::StartGetRedirects,
49 this, urls));
50 }
51
HasStarted() const52 bool MalwareDetailsRedirectsCollector::HasStarted() const {
53 return has_started_;
54 }
55
56 const std::vector<safe_browsing::RedirectChain>&
GetCollectedUrls() const57 MalwareDetailsRedirectsCollector::GetCollectedUrls() const {
58 return redirects_urls_;
59 }
60
Observe(int type,const content::NotificationSource & source,const content::NotificationDetails & details)61 void MalwareDetailsRedirectsCollector::Observe(
62 int type,
63 const content::NotificationSource& source,
64 const content::NotificationDetails& details) {
65 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI));
66 DCHECK_EQ(type, chrome::NOTIFICATION_PROFILE_DESTROYED);
67 DVLOG(1) << "Profile gone.";
68 profile_ = NULL;
69 }
70
~MalwareDetailsRedirectsCollector()71 MalwareDetailsRedirectsCollector::~MalwareDetailsRedirectsCollector() {}
72
StartGetRedirects(const std::vector<GURL> & urls)73 void MalwareDetailsRedirectsCollector::StartGetRedirects(
74 const std::vector<GURL>& urls) {
75 // History access from profile needs to happen in UI thread
76 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI));
77 for (size_t i = 0; i < urls.size(); ++i) {
78 urls_.push_back(urls[i]);
79 }
80 urls_it_ = urls_.begin();
81 GetRedirects(*urls_it_);
82 }
83
GetRedirects(const GURL & url)84 void MalwareDetailsRedirectsCollector::GetRedirects(const GURL& url) {
85 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI));
86 if (!profile_) {
87 AllDone();
88 return;
89 }
90
91 HistoryService* history = HistoryServiceFactory::GetForProfile(
92 profile_, Profile::EXPLICIT_ACCESS);
93 if (!history) {
94 AllDone();
95 return;
96 }
97
98 history->QueryRedirectsTo(
99 url,
100 &request_consumer_,
101 base::Bind(&MalwareDetailsRedirectsCollector::OnGotQueryRedirectsTo,
102 base::Unretained(this)));
103 }
104
OnGotQueryRedirectsTo(HistoryService::Handle handle,GURL url,bool success,history::RedirectList * redirect_list)105 void MalwareDetailsRedirectsCollector::OnGotQueryRedirectsTo(
106 HistoryService::Handle handle,
107 GURL url,
108 bool success,
109 history::RedirectList* redirect_list) {
110
111 if (success && redirect_list->size() > 0) {
112 std::vector<GURL> urllist;
113 urllist.push_back(url);
114 for (size_t i = 0; i < redirect_list->size(); i++) {
115 urllist.push_back(redirect_list->at(i));
116 }
117 redirects_urls_.push_back(urllist);
118 }
119
120 // Proceed to next url
121 ++urls_it_;
122
123 if (urls_it_ == urls_.end()) {
124 AllDone();
125 return;
126 }
127
128 GetRedirects(*urls_it_);
129 }
130
AllDone()131 void MalwareDetailsRedirectsCollector::AllDone() {
132 DVLOG(1) << "AllDone";
133 BrowserThread::PostTask(BrowserThread::IO, FROM_HERE, callback_);
134 callback_.Reset();
135 }
136