OLD | NEW |
1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 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 | 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 // Safe Browsing Database Manager implementation that manages a local | 5 // Safe Browsing Database Manager implementation that manages a local |
6 // database. This is used by Desktop Chromium. | 6 // database. This is used by Desktop Chromium. |
7 | 7 |
8 #ifndef CHROME_BROWSER_SAFE_BROWSING_LOCAL_DATABASE_MANAGER_H_ | 8 #ifndef CHROME_BROWSER_SAFE_BROWSING_LOCAL_DATABASE_MANAGER_H_ |
9 #define CHROME_BROWSER_SAFE_BROWSING_LOCAL_DATABASE_MANAGER_H_ | 9 #define CHROME_BROWSER_SAFE_BROWSING_LOCAL_DATABASE_MANAGER_H_ |
10 | 10 |
11 #include <stddef.h> | 11 #include <stddef.h> |
12 | 12 |
13 #include <deque> | 13 #include <deque> |
14 #include <map> | 14 #include <map> |
15 #include <memory> | 15 #include <memory> |
16 #include <set> | 16 #include <set> |
17 #include <string> | 17 #include <string> |
18 #include <vector> | 18 #include <vector> |
19 | 19 |
20 #include "base/callback.h" | 20 #include "base/callback.h" |
21 #include "base/containers/hash_tables.h" | |
22 #include "base/gtest_prod_util.h" | 21 #include "base/gtest_prod_util.h" |
23 #include "base/macros.h" | 22 #include "base/macros.h" |
24 #include "base/memory/ref_counted.h" | 23 #include "base/memory/ref_counted.h" |
25 #include "base/memory/weak_ptr.h" | 24 #include "base/memory/weak_ptr.h" |
26 #include "base/synchronization/lock.h" | 25 #include "base/synchronization/lock.h" |
27 #include "base/time/time.h" | 26 #include "base/time/time.h" |
28 #include "chrome/browser/safe_browsing/protocol_manager.h" | 27 #include "chrome/browser/safe_browsing/protocol_manager.h" |
29 #include "chrome/browser/safe_browsing/safe_browsing_util.h" | 28 #include "chrome/browser/safe_browsing/safe_browsing_util.h" |
30 #include "components/safe_browsing/common/safe_browsing_prefs.h" | 29 #include "components/safe_browsing/common/safe_browsing_prefs.h" |
31 #include "components/safe_browsing_db/database_manager.h" | 30 #include "components/safe_browsing_db/database_manager.h" |
(...skipping 127 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
159 friend class base::RefCountedThreadSafe<LocalSafeBrowsingDatabaseManager>; | 158 friend class base::RefCountedThreadSafe<LocalSafeBrowsingDatabaseManager>; |
160 friend class SafeBrowsingServerTest; | 159 friend class SafeBrowsingServerTest; |
161 friend class SafeBrowsingServiceTest; | 160 friend class SafeBrowsingServiceTest; |
162 friend class SafeBrowsingServiceTestHelper; | 161 friend class SafeBrowsingServiceTestHelper; |
163 friend class LocalDatabaseManagerTest; | 162 friend class LocalDatabaseManagerTest; |
164 FRIEND_TEST_ALL_PREFIXES(LocalDatabaseManagerTest, GetUrlSeverestThreatType); | 163 FRIEND_TEST_ALL_PREFIXES(LocalDatabaseManagerTest, GetUrlSeverestThreatType); |
165 FRIEND_TEST_ALL_PREFIXES(LocalDatabaseManagerTest, | 164 FRIEND_TEST_ALL_PREFIXES(LocalDatabaseManagerTest, |
166 ServiceStopWithPendingChecks); | 165 ServiceStopWithPendingChecks); |
167 | 166 |
168 typedef std::vector<SafeBrowsingCheck*> GetHashRequestors; | 167 typedef std::vector<SafeBrowsingCheck*> GetHashRequestors; |
169 typedef base::hash_map<SBPrefix, GetHashRequestors> GetHashRequests; | 168 typedef std::map<SBPrefix, GetHashRequestors> GetHashRequests; |
170 | 169 |
171 // Clients that we've queued up for checking later once the database is ready. | 170 // Clients that we've queued up for checking later once the database is ready. |
172 struct QueuedCheck { | 171 struct QueuedCheck { |
173 QueuedCheck(const ListType check_type, | 172 QueuedCheck(const ListType check_type, |
174 Client* client, | 173 Client* client, |
175 const GURL& url, | 174 const GURL& url, |
176 const SBThreatTypeSet& expected_threats, | 175 const SBThreatTypeSet& expected_threats, |
177 const base::TimeTicks& start); | 176 const base::TimeTicks& start); |
178 QueuedCheck(const QueuedCheck& other); | 177 QueuedCheck(const QueuedCheck& other); |
179 ~QueuedCheck(); | 178 ~QueuedCheck(); |
(...skipping 200 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
380 | 379 |
381 // Timeout to use for safe browsing checks. | 380 // Timeout to use for safe browsing checks. |
382 base::TimeDelta check_timeout_; | 381 base::TimeDelta check_timeout_; |
383 | 382 |
384 DISALLOW_COPY_AND_ASSIGN(LocalSafeBrowsingDatabaseManager); | 383 DISALLOW_COPY_AND_ASSIGN(LocalSafeBrowsingDatabaseManager); |
385 }; // class LocalSafeBrowsingDatabaseManager | 384 }; // class LocalSafeBrowsingDatabaseManager |
386 | 385 |
387 } // namespace safe_browsing | 386 } // namespace safe_browsing |
388 | 387 |
389 #endif // CHROME_BROWSER_SAFE_BROWSING_LOCAL_DATABASE_MANAGER_H_ | 388 #endif // CHROME_BROWSER_SAFE_BROWSING_LOCAL_DATABASE_MANAGER_H_ |
OLD | NEW |