1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
|
// Copyright (c) 2012 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#ifndef CHROME_BROWSER_SAFE_BROWSING_MALWARE_DETAILS_HISTORY_H_
#define CHROME_BROWSER_SAFE_BROWSING_MALWARE_DETAILS_HISTORY_H_
// This class gets redirect chain for urls from the history service.
#include <string>
#include <vector>
#include "base/callback.h"
#include "base/containers/hash_tables.h"
#include "base/memory/linked_ptr.h"
#include "base/memory/ref_counted.h"
#include "base/sequenced_task_runner_helpers.h"
#include "base/task/cancelable_task_tracker.h"
#include "chrome/browser/history/history_service.h"
#include "content/public/browser/browser_thread.h"
#include "content/public/browser/notification_observer.h"
#include "content/public/browser/notification_registrar.h"
#include "net/base/completion_callback.h"
namespace safe_browsing {
typedef std::vector<GURL> RedirectChain;
}
class MalwareDetailsRedirectsCollector
: public base::RefCountedThreadSafe<
MalwareDetailsRedirectsCollector,
content::BrowserThread::DeleteOnUIThread>,
public content::NotificationObserver {
public:
explicit MalwareDetailsRedirectsCollector(Profile* profile);
// Collects urls' redirects chain information from the history service.
// We get access to history service via web_contents in UI thread.
// Notice the callback will be posted to the IO thread.
void StartHistoryCollection(const std::vector<GURL>& urls,
const base::Closure& callback);
// Returns whether or not StartCacheCollection has been called.
bool HasStarted() const;
// Returns the redirect urls we get from history service
const std::vector<safe_browsing::RedirectChain>& GetCollectedUrls() const;
// content::NotificationObserver
void Observe(int type,
const content::NotificationSource& source,
const content::NotificationDetails& details) override;
private:
friend struct content::BrowserThread::DeleteOnThread<
content::BrowserThread::UI>;
friend class base::DeleteHelper<MalwareDetailsRedirectsCollector>;
~MalwareDetailsRedirectsCollector() override;
void StartGetRedirects(const std::vector<GURL>& urls);
void GetRedirects(const GURL& url);
void OnGotQueryRedirectsTo(const GURL& url,
const history::RedirectList* redirect_list);
// Posts the callback method back to IO thread when redirects collecting
// is all done.
void AllDone();
Profile* profile_;
base::CancelableTaskTracker request_tracker_;
// Method we call when we are done. The caller must be alive for the
// whole time, we are modifying its state (see above).
base::Closure callback_;
// Sets to true once StartHistoryCollection is called
bool has_started_;
// The urls we need to get redirects for
std::vector<GURL> urls_;
// The iterator goes over urls_
std::vector<GURL>::iterator urls_it_;
// The collected directs from history service
std::vector<safe_browsing::RedirectChain> redirects_urls_;
content::NotificationRegistrar registrar_;
DISALLOW_COPY_AND_ASSIGN(MalwareDetailsRedirectsCollector);
};
#endif // CHROME_BROWSER_SAFE_BROWSING_MALWARE_DETAILS_HISTORY_H_
|
