OLD | NEW |
1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2011 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 #ifndef CHROME_BROWSER_SAFE_BROWSING_THREAT_DETAILS_H_ | 5 #ifndef CHROME_BROWSER_SAFE_BROWSING_THREAT_DETAILS_H_ |
6 #define CHROME_BROWSER_SAFE_BROWSING_THREAT_DETAILS_H_ | 6 #define CHROME_BROWSER_SAFE_BROWSING_THREAT_DETAILS_H_ |
7 | 7 |
8 // A class that encapsulates the detailed threat reports sent when | 8 // A class that encapsulates the detailed threat reports sent when |
9 // users opt-in to do so from the safe browsing warning page. | 9 // users opt-in to do so from the safe browsing warning page. |
10 | 10 |
(...skipping 11 matching lines...) Expand all Loading... |
22 #include "base/memory/scoped_ptr.h" | 22 #include "base/memory/scoped_ptr.h" |
23 #include "chrome/browser/safe_browsing/ui_manager.h" | 23 #include "chrome/browser/safe_browsing/ui_manager.h" |
24 #include "chrome/common/safe_browsing/csd.pb.h" | 24 #include "chrome/common/safe_browsing/csd.pb.h" |
25 #include "content/public/browser/web_contents_observer.h" | 25 #include "content/public/browser/web_contents_observer.h" |
26 #include "net/base/completion_callback.h" | 26 #include "net/base/completion_callback.h" |
27 | 27 |
28 namespace net { | 28 namespace net { |
29 class URLRequestContextGetter; | 29 class URLRequestContextGetter; |
30 } | 30 } |
31 | 31 |
32 class ThreatDetailsCacheCollector; | |
33 class ThreatDetailsRedirectsCollector; | |
34 class ThreatDetailsFactory; | |
35 class Profile; | 32 class Profile; |
36 struct SafeBrowsingHostMsg_ThreatDOMDetails_Node; | 33 struct SafeBrowsingHostMsg_ThreatDOMDetails_Node; |
37 | 34 |
38 namespace safe_browsing { | 35 namespace safe_browsing { |
| 36 |
39 // Maps a URL to its Resource. | 37 // Maps a URL to its Resource. |
| 38 class ThreatDetailsCacheCollector; |
| 39 class ThreatDetailsRedirectsCollector; |
| 40 class ThreatDetailsFactory; |
| 41 |
40 typedef base::hash_map< | 42 typedef base::hash_map< |
41 std::string, | 43 std::string, |
42 linked_ptr<safe_browsing::ClientSafeBrowsingReportRequest::Resource>> | 44 linked_ptr<ClientSafeBrowsingReportRequest::Resource>> |
43 ResourceMap; | 45 ResourceMap; |
44 } | |
45 | 46 |
46 class ThreatDetails : public base::RefCountedThreadSafe<ThreatDetails>, | 47 class ThreatDetails : public base::RefCountedThreadSafe<ThreatDetails>, |
47 public content::WebContentsObserver { | 48 public content::WebContentsObserver { |
48 public: | 49 public: |
49 typedef SafeBrowsingUIManager::UnsafeResource UnsafeResource; | 50 typedef SafeBrowsingUIManager::UnsafeResource UnsafeResource; |
50 | 51 |
51 // Constructs a new ThreatDetails instance, using the factory. | 52 // Constructs a new ThreatDetails instance, using the factory. |
52 static ThreatDetails* NewThreatDetails(SafeBrowsingUIManager* ui_manager, | 53 static ThreatDetails* NewThreatDetails(SafeBrowsingUIManager* ui_manager, |
53 content::WebContents* web_contents, | 54 content::WebContents* web_contents, |
54 const UnsafeResource& resource); | 55 const UnsafeResource& resource); |
(...skipping 28 matching lines...) Expand all Loading... |
83 | 84 |
84 ~ThreatDetails() override; | 85 ~ThreatDetails() override; |
85 | 86 |
86 // Called on the IO thread with the DOM details. | 87 // Called on the IO thread with the DOM details. |
87 virtual void AddDOMDetails( | 88 virtual void AddDOMDetails( |
88 const std::vector<SafeBrowsingHostMsg_ThreatDOMDetails_Node>& params); | 89 const std::vector<SafeBrowsingHostMsg_ThreatDOMDetails_Node>& params); |
89 | 90 |
90 Profile* profile_; | 91 Profile* profile_; |
91 | 92 |
92 // The report protocol buffer. | 93 // The report protocol buffer. |
93 scoped_ptr<safe_browsing::ClientSafeBrowsingReportRequest> report_; | 94 scoped_ptr<ClientSafeBrowsingReportRequest> report_; |
94 | 95 |
95 // Used to get a pointer to the HTTP cache. | 96 // Used to get a pointer to the HTTP cache. |
96 scoped_refptr<net::URLRequestContextGetter> request_context_getter_; | 97 scoped_refptr<net::URLRequestContextGetter> request_context_getter_; |
97 | 98 |
98 private: | 99 private: |
99 friend class base::RefCountedThreadSafe<ThreatDetails>; | 100 friend class base::RefCountedThreadSafe<ThreatDetails>; |
100 | 101 |
101 // Starts the collection of the report. | 102 // Starts the collection of the report. |
102 void StartCollection(); | 103 void StartCollection(); |
103 | 104 |
104 // Whether the url is "public" so we can add it to the report. | 105 // Whether the url is "public" so we can add it to the report. |
105 bool IsReportableUrl(const GURL& url) const; | 106 bool IsReportableUrl(const GURL& url) const; |
106 | 107 |
107 // Finds an existing Resource for the given url, or creates a new | 108 // Finds an existing Resource for the given url, or creates a new |
108 // one if not found, and adds it to |resources_|. Returns the | 109 // one if not found, and adds it to |resources_|. Returns the |
109 // found/created resource. | 110 // found/created resource. |
110 safe_browsing::ClientSafeBrowsingReportRequest::Resource* | 111 ClientSafeBrowsingReportRequest::Resource* FindOrCreateResource( |
111 FindOrCreateResource(const GURL& url); | 112 const GURL& url); |
112 | 113 |
113 // Adds a Resource to resources_ with the given parent-child | 114 // Adds a Resource to resources_ with the given parent-child |
114 // relationship. |parent| and |tagname| can be empty, |children| can be NULL. | 115 // relationship. |parent| and |tagname| can be empty, |children| can be NULL. |
115 void AddUrl(const GURL& url, | 116 void AddUrl(const GURL& url, |
116 const GURL& parent, | 117 const GURL& parent, |
117 const std::string& tagname, | 118 const std::string& tagname, |
118 const std::vector<GURL>* children); | 119 const std::vector<GURL>* children); |
119 | 120 |
120 // Message handler. | 121 // Message handler. |
121 void OnReceivedThreatDOMDetails( | 122 void OnReceivedThreatDOMDetails( |
122 const std::vector<SafeBrowsingHostMsg_ThreatDOMDetails_Node>& params); | 123 const std::vector<SafeBrowsingHostMsg_ThreatDOMDetails_Node>& params); |
123 | 124 |
124 void AddRedirectUrlList(const std::vector<GURL>& urls); | 125 void AddRedirectUrlList(const std::vector<GURL>& urls); |
125 | 126 |
126 scoped_refptr<SafeBrowsingUIManager> ui_manager_; | 127 scoped_refptr<SafeBrowsingUIManager> ui_manager_; |
127 | 128 |
128 const UnsafeResource resource_; | 129 const UnsafeResource resource_; |
129 | 130 |
130 // For every Url we collect we create a Resource message. We keep | 131 // For every Url we collect we create a Resource message. We keep |
131 // them in a map so we can avoid duplicates. | 132 // them in a map so we can avoid duplicates. |
132 safe_browsing::ResourceMap resources_; | 133 ResourceMap resources_; |
133 | 134 |
134 // Result from the cache extractor. | 135 // Result from the cache extractor. |
135 bool cache_result_; | 136 bool cache_result_; |
136 | 137 |
137 // Whether user did proceed with the safe browsing blocking page or | 138 // Whether user did proceed with the safe browsing blocking page or |
138 // not. | 139 // not. |
139 bool did_proceed_; | 140 bool did_proceed_; |
140 | 141 |
141 // How many times this user has visited this page before. | 142 // How many times this user has visited this page before. |
142 int num_visits_; | 143 int num_visits_; |
(...skipping 21 matching lines...) Expand all Loading... |
164 class ThreatDetailsFactory { | 165 class ThreatDetailsFactory { |
165 public: | 166 public: |
166 virtual ~ThreatDetailsFactory() {} | 167 virtual ~ThreatDetailsFactory() {} |
167 | 168 |
168 virtual ThreatDetails* CreateThreatDetails( | 169 virtual ThreatDetails* CreateThreatDetails( |
169 SafeBrowsingUIManager* ui_manager, | 170 SafeBrowsingUIManager* ui_manager, |
170 content::WebContents* web_contents, | 171 content::WebContents* web_contents, |
171 const SafeBrowsingUIManager::UnsafeResource& unsafe_resource) = 0; | 172 const SafeBrowsingUIManager::UnsafeResource& unsafe_resource) = 0; |
172 }; | 173 }; |
173 | 174 |
| 175 } // namespace safe_browsing |
| 176 |
174 #endif // CHROME_BROWSER_SAFE_BROWSING_THREAT_DETAILS_H_ | 177 #endif // CHROME_BROWSER_SAFE_BROWSING_THREAT_DETAILS_H_ |
OLD | NEW |