OLD | NEW |
1 // Copyright 2015 The Chromium Authors. All rights reserved. | 1 // Copyright 2015 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_INCIDENT_REPORTING_SCRIPT_REQUEST_DETECTOR_
H_ | 5 #ifndef CHROME_BROWSER_SAFE_BROWSING_INCIDENT_REPORTING_RESOURCE_REQUEST_DETECTO
R_H_ |
6 #define CHROME_BROWSER_SAFE_BROWSING_INCIDENT_REPORTING_SCRIPT_REQUEST_DETECTOR_
H_ | 6 #define CHROME_BROWSER_SAFE_BROWSING_INCIDENT_REPORTING_RESOURCE_REQUEST_DETECTO
R_H_ |
7 | 7 |
8 #include "base/containers/hash_tables.h" | 8 #include "base/containers/hash_tables.h" |
9 #include "base/macros.h" | 9 #include "base/macros.h" |
10 #include "base/memory/scoped_ptr.h" | 10 #include "base/memory/scoped_ptr.h" |
11 #include "base/memory/weak_ptr.h" | 11 #include "base/memory/weak_ptr.h" |
12 #include "chrome/browser/safe_browsing/incident_reporting/incident_receiver.h" | 12 #include "chrome/browser/safe_browsing/incident_reporting/incident_receiver.h" |
13 | 13 |
14 namespace net { | 14 namespace net { |
15 class URLRequest; | 15 class URLRequest; |
16 } | 16 } |
17 | 17 |
18 namespace safe_browsing { | 18 namespace safe_browsing { |
19 | 19 |
20 class ClientIncidentReport_IncidentData_ScriptRequestIncident; | 20 class ClientIncidentReport_IncidentData_ResourceRequestIncident; |
21 | 21 |
22 // Observes network requests and reports suspicious activity. | 22 // Observes network requests and reports suspicious activity. |
23 class ScriptRequestDetector { | 23 class ResourceRequestDetector { |
24 public: | 24 public: |
25 explicit ScriptRequestDetector( | 25 explicit ResourceRequestDetector( |
26 scoped_ptr<IncidentReceiver> incident_receiver); | 26 scoped_ptr<IncidentReceiver> incident_receiver); |
27 ~ScriptRequestDetector(); | 27 ~ResourceRequestDetector(); |
28 | 28 |
29 // Analyzes the |request| and triggers an incident report on suspicious | 29 // Analyzes the |request| and triggers an incident report on suspicious |
30 // script inclusion. | 30 // script inclusion. |
31 void OnResourceRequest(const net::URLRequest* request); | 31 void OnResourceRequest(const net::URLRequest* request); |
32 | 32 |
33 protected: | 33 protected: |
34 // Testing hook. | 34 // Testing hook. |
35 void set_allow_null_profile_for_testing(bool allow_null_profile_for_testing); | 35 void set_allow_null_profile_for_testing(bool allow_null_profile_for_testing); |
36 | 36 |
37 private: | 37 private: |
38 void InitializeScriptSet(); | 38 void InitializeHashSets(); |
| 39 |
| 40 void DetectDomainRequests(const net::URLRequest* request); |
| 41 void DetectScriptRequests(const net::URLRequest* request); |
39 | 42 |
40 void ReportIncidentOnUIThread( | 43 void ReportIncidentOnUIThread( |
41 int render_process_id, | 44 int render_process_id, |
42 int render_frame_id, | 45 int render_frame_id, |
43 scoped_ptr<ClientIncidentReport_IncidentData_ScriptRequestIncident> | 46 scoped_ptr<ClientIncidentReport_IncidentData_ResourceRequestIncident> |
44 incident_data); | 47 incident_data); |
45 | 48 |
46 scoped_ptr<IncidentReceiver> incident_receiver_; | 49 scoped_ptr<IncidentReceiver> incident_receiver_; |
47 base::hash_set<std::string> script_set_; | 50 base::hash_set<std::string> script_set_; |
| 51 base::hash_set<std::string> domain_set_; |
48 bool allow_null_profile_for_testing_; | 52 bool allow_null_profile_for_testing_; |
49 | 53 |
50 base::WeakPtrFactory<ScriptRequestDetector> weak_ptr_factory_; | 54 base::WeakPtrFactory<ResourceRequestDetector> weak_ptr_factory_; |
51 | 55 |
52 DISALLOW_COPY_AND_ASSIGN(ScriptRequestDetector); | 56 DISALLOW_COPY_AND_ASSIGN(ResourceRequestDetector); |
53 }; | 57 }; |
54 | 58 |
55 } // namespace safe_browsing | 59 } // namespace safe_browsing |
56 | 60 |
57 #endif // CHROME_BROWSER_SAFE_BROWSING_INCIDENT_REPORTING_SCRIPT_REQUEST_DETECT
OR_H_ | 61 #endif // CHROME_BROWSER_SAFE_BROWSING_INCIDENT_REPORTING_RESOURCE_REQUEST_DETE
CTOR_H_ |
OLD | NEW |