OLD | NEW |
1 // Copyright 2016 The Chromium Authors. All rights reserved. | 1 // Copyright 2016 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 COMPONENTS_SAFE_BROWSING_DB_V4_LOCAL_DATABASE_MANAGER_H_ | 5 #ifndef COMPONENTS_SAFE_BROWSING_DB_V4_LOCAL_DATABASE_MANAGER_H_ |
6 #define COMPONENTS_SAFE_BROWSING_DB_V4_LOCAL_DATABASE_MANAGER_H_ | 6 #define COMPONENTS_SAFE_BROWSING_DB_V4_LOCAL_DATABASE_MANAGER_H_ |
7 | 7 |
8 // A class that provides the interface between the SafeBrowsing protocol manager | 8 // A class that provides the interface between the SafeBrowsing protocol manager |
9 // and database that holds the downloaded updates. | 9 // and database that holds the downloaded updates. |
10 | 10 |
11 #include <memory> | 11 #include <memory> |
12 | 12 |
13 #include "components/safe_browsing_db/database_manager.h" | 13 #include "components/safe_browsing_db/database_manager.h" |
14 #include "components/safe_browsing_db/hit_report.h" | 14 #include "components/safe_browsing_db/hit_report.h" |
15 #include "components/safe_browsing_db/v4_database.h" | 15 #include "components/safe_browsing_db/v4_database.h" |
16 #include "components/safe_browsing_db/v4_protocol_manager_util.h" | 16 #include "components/safe_browsing_db/v4_protocol_manager_util.h" |
17 #include "components/safe_browsing_db/v4_update_protocol_manager.h" | 17 #include "components/safe_browsing_db/v4_update_protocol_manager.h" |
18 #include "url/gurl.h" | 18 #include "url/gurl.h" |
19 | 19 |
20 using content::ResourceType; | 20 using content::ResourceType; |
21 | 21 |
22 namespace safe_browsing { | 22 namespace safe_browsing { |
23 | 23 |
24 // Manages the local, on-disk database of updates downloaded from the | 24 // Manages the local, on-disk database of updates downloaded from the |
25 // SafeBrowsing service and interfaces with the protocol manager. | 25 // SafeBrowsing service and interfaces with the protocol manager. |
26 class V4LocalDatabaseManager : public SafeBrowsingDatabaseManager { | 26 class V4LocalDatabaseManager : public SafeBrowsingDatabaseManager { |
27 public: | 27 public: |
28 // Construct V4LocalDatabaseManager. | 28 // Construct V4LocalDatabaseManager. |
29 // Must be initialized by calling StartOnIOThread() before using. | 29 // Must be initialized by calling StartOnIOThread() before using. |
30 V4LocalDatabaseManager(const base::FilePath& base_path); | 30 V4LocalDatabaseManager(); |
31 | 31 |
32 // | 32 // |
33 // SafeBrowsingDatabaseManager implementation | 33 // SafeBrowsingDatabaseManager implementation |
34 // | 34 // |
35 | 35 |
36 bool IsSupported() const override; | 36 bool IsSupported() const override; |
37 safe_browsing::ThreatSource GetThreatSource() const override; | 37 safe_browsing::ThreatSource GetThreatSource() const override; |
38 bool ChecksAreAlwaysAsync() const override; | 38 bool ChecksAreAlwaysAsync() const override; |
39 bool CanCheckResourceType(content::ResourceType resource_type) const override; | 39 bool CanCheckResourceType(content::ResourceType resource_type) const override; |
40 bool CanCheckUrl(const GURL& url) const override; | 40 bool CanCheckUrl(const GURL& url) const override; |
41 bool IsDownloadProtectionEnabled() const override; | 41 bool IsDownloadProtectionEnabled() const override; |
42 bool CheckBrowseUrl(const GURL& url, Client* client) override; | 42 bool CheckBrowseUrl(const GURL& url, Client* client) override; |
43 void CancelCheck(Client* client) override; | 43 void CancelCheck(Client* client) override; |
44 void StartOnIOThread(net::URLRequestContextGetter* request_context_getter, | 44 void StartOnIOThread( |
45 const V4ProtocolConfig& config) override; | 45 net::URLRequestContextGetter* request_context_getter, |
| 46 const V4ProtocolConfig& config) override; |
46 void StopOnIOThread(bool shutdown) override; | 47 void StopOnIOThread(bool shutdown) override; |
47 bool CheckDownloadUrl(const std::vector<GURL>& url_chain, | 48 bool CheckDownloadUrl(const std::vector<GURL>& url_chain, |
48 Client* client) override; | 49 Client* client) override; |
49 bool CheckExtensionIDs(const std::set<std::string>& extension_ids, | 50 bool CheckExtensionIDs(const std::set<std::string>& extension_ids, |
50 Client* client) override; | 51 Client* client) override; |
51 bool MatchCsdWhitelistUrl(const GURL& url) override; | 52 bool MatchCsdWhitelistUrl(const GURL& url) override; |
52 bool MatchMalwareIP(const std::string& ip_address) override; | 53 bool MatchMalwareIP(const std::string& ip_address) override; |
53 bool MatchDownloadWhitelistUrl(const GURL& url) override; | 54 bool MatchDownloadWhitelistUrl(const GURL& url) override; |
54 bool MatchDownloadWhitelistString(const std::string& str) override; | 55 bool MatchDownloadWhitelistString(const std::string& str) override; |
55 bool MatchInclusionWhitelistUrl(const GURL& url) override; | 56 bool MatchInclusionWhitelistUrl(const GURL& url) override; |
56 bool MatchModuleWhitelistString(const std::string& str) override; | 57 bool MatchModuleWhitelistString(const std::string& str) override; |
57 bool CheckResourceUrl(const GURL& url, Client* client) override; | 58 bool CheckResourceUrl(const GURL& url, Client* client) override; |
58 bool IsMalwareKillSwitchOn() override; | 59 bool IsMalwareKillSwitchOn() override; |
59 bool IsCsdWhitelistKillSwitchOn() override; | 60 bool IsCsdWhitelistKillSwitchOn() override; |
60 | 61 |
61 private: | 62 private: |
62 ~V4LocalDatabaseManager() override; | 63 ~V4LocalDatabaseManager() override; |
63 | 64 |
64 // The callback called each time the protocol manager downloads updates | 65 // The callback called each time the protocol manager downloads updates |
65 // successfully. | 66 // successfully. |
66 void UpdateRequestCompleted(const std::vector<ListUpdateResponse>& responses); | 67 void UpdateRequestCompleted(const std::vector<ListUpdateResponse>& responses); |
67 | 68 |
68 void SetupUpdateProtocolManager( | |
69 net::URLRequestContextGetter* request_context_getter, | |
70 const V4ProtocolConfig& config); | |
71 | |
72 void SetupDatabase(); | |
73 | |
74 void DatabaseReady(std::unique_ptr<V4Database> v4_database); | |
75 | |
76 void OnCloseDatabase(); | |
77 | |
78 // The base directory under which to create the files that contain hashes. | |
79 const base::FilePath base_path_; | |
80 | |
81 // Whether the service is running. | |
82 bool enabled_; | 69 bool enabled_; |
83 | 70 |
84 // Stores the current status of the lists to download from the SafeBrowsing | 71 // Stores the current status of the lists to download from the SafeBrowsing |
85 // servers. | 72 // servers. |
86 // TODO(vakh): current_list_states_ doesn't really belong here. | 73 // TODO(vakh): current_list_states_ doesn't really belong here. |
87 // It should come through the database, from the various V4Stores. | 74 // It should come through the database, from the various V4Stores. |
88 base::hash_map<UpdateListIdentifier, std::string> current_list_states_; | 75 base::hash_map<UpdateListIdentifier, std::string> current_list_states_; |
89 | 76 |
90 // The protocol manager that downloads the hash prefix updates. | 77 // The protocol manager that downloads the hash prefix updates. |
91 std::unique_ptr<V4UpdateProtocolManager> v4_update_protocol_manager_; | 78 std::unique_ptr<V4UpdateProtocolManager> v4_update_protocol_manager_; |
92 | 79 |
93 // The database that manages the stores containing the hash prefix updates. | 80 // The database that manages the stores containing the hash prefix updates. |
94 // All writes to this variable must happen on the IO thread only. | |
95 std::unique_ptr<V4Database> v4_database_; | 81 std::unique_ptr<V4Database> v4_database_; |
96 | 82 |
97 // The sequenced task runner for running safe browsing database operations. | |
98 scoped_refptr<base::SequencedTaskRunner> task_runner_; | |
99 | |
100 friend class base::RefCountedThreadSafe<V4LocalDatabaseManager>; | 83 friend class base::RefCountedThreadSafe<V4LocalDatabaseManager>; |
101 DISALLOW_COPY_AND_ASSIGN(V4LocalDatabaseManager); | 84 DISALLOW_COPY_AND_ASSIGN(V4LocalDatabaseManager); |
102 }; // class V4LocalDatabaseManager | 85 }; // class V4LocalDatabaseManager |
103 | 86 |
104 } // namespace safe_browsing | 87 } // namespace safe_browsing |
105 | 88 |
106 #endif // COMPONENTS_SAFE_BROWSING_DB_V4_LOCAL_DATABASE_MANAGER_H_ | 89 #endif // COMPONENTS_SAFE_BROWSING_DB_V4_LOCAL_DATABASE_MANAGER_H_ |
OLD | NEW |