| 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 // Helper class loads models for client-side phishing detection | 5 // Helper class loads models for client-side phishing detection |
| 6 // from the the SafeBrowsing backends. | 6 // from the the SafeBrowsing backends. |
| 7 // | 7 // |
| 8 // This class is not thread-safe and expects all calls to be made on the UI | 8 // This class is not thread-safe and expects all calls to be made on the UI |
| 9 // thread. | 9 // thread. |
| 10 | 10 |
| 11 #ifndef CHROME_BROWSER_SAFE_BROWSING_CLIENT_SIDE_MODEL_LOADER_H_ | 11 #ifndef CHROME_BROWSER_SAFE_BROWSING_CLIENT_SIDE_MODEL_LOADER_H_ |
| 12 #define CHROME_BROWSER_SAFE_BROWSING_CLIENT_SIDE_MODEL_LOADER_H_ | 12 #define CHROME_BROWSER_SAFE_BROWSING_CLIENT_SIDE_MODEL_LOADER_H_ |
| 13 | 13 |
| 14 #include <stddef.h> | 14 #include <stddef.h> |
| 15 #include <stdint.h> | 15 #include <stdint.h> |
| 16 | 16 |
| 17 #include <memory> |
| 17 #include <string> | 18 #include <string> |
| 18 | 19 |
| 19 #include "base/callback.h" | 20 #include "base/callback.h" |
| 20 #include "base/gtest_prod_util.h" | 21 #include "base/gtest_prod_util.h" |
| 21 #include "base/macros.h" | 22 #include "base/macros.h" |
| 22 #include "base/memory/linked_ptr.h" | 23 #include "base/memory/linked_ptr.h" |
| 23 #include "base/memory/ref_counted.h" | 24 #include "base/memory/ref_counted.h" |
| 24 #include "base/memory/scoped_ptr.h" | |
| 25 #include "base/memory/weak_ptr.h" | 25 #include "base/memory/weak_ptr.h" |
| 26 #include "content/public/browser/browser_thread.h" | 26 #include "content/public/browser/browser_thread.h" |
| 27 #include "net/url_request/url_fetcher_delegate.h" | 27 #include "net/url_request/url_fetcher_delegate.h" |
| 28 #include "url/gurl.h" | 28 #include "url/gurl.h" |
| 29 | 29 |
| 30 namespace base { | 30 namespace base { |
| 31 class TimeDelta; | 31 class TimeDelta; |
| 32 } | 32 } |
| 33 | 33 |
| 34 namespace net { | 34 namespace net { |
| (...skipping 72 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 107 // valid hashes in the model. | 107 // valid hashes in the model. |
| 108 static bool ModelHasValidHashIds(const ClientSideModel& model); | 108 static bool ModelHasValidHashIds(const ClientSideModel& model); |
| 109 | 109 |
| 110 // The name of the model is the last component of the URL path. | 110 // The name of the model is the last component of the URL path. |
| 111 const std::string name_; | 111 const std::string name_; |
| 112 // Full URL of the model. | 112 // Full URL of the model. |
| 113 const GURL url_; | 113 const GURL url_; |
| 114 | 114 |
| 115 // If the model isn't yet loaded, model_str_ will be empty. | 115 // If the model isn't yet loaded, model_str_ will be empty. |
| 116 std::string model_str_; | 116 std::string model_str_; |
| 117 scoped_ptr<ClientSideModel> model_; | 117 std::unique_ptr<ClientSideModel> model_; |
| 118 scoped_ptr<net::URLFetcher> fetcher_; | 118 std::unique_ptr<net::URLFetcher> fetcher_; |
| 119 | 119 |
| 120 // Callback to invoke when we've got a new model. CSD will send it around. | 120 // Callback to invoke when we've got a new model. CSD will send it around. |
| 121 base::Closure update_renderers_callback_; | 121 base::Closure update_renderers_callback_; |
| 122 | 122 |
| 123 // Not owned, must outlive this obj or be NULL. | 123 // Not owned, must outlive this obj or be NULL. |
| 124 net::URLRequestContextGetter* request_context_getter_; | 124 net::URLRequestContextGetter* request_context_getter_; |
| 125 | 125 |
| 126 // Used to protect the delayed callback to StartFetchModel() | 126 // Used to protect the delayed callback to StartFetchModel() |
| 127 base::WeakPtrFactory<ModelLoader> weak_factory_; | 127 base::WeakPtrFactory<ModelLoader> weak_factory_; |
| 128 | 128 |
| 129 friend class ClientSideDetectionServiceTest; | 129 friend class ClientSideDetectionServiceTest; |
| 130 friend class ModelLoaderTest; | 130 friend class ModelLoaderTest; |
| 131 FRIEND_TEST_ALL_PREFIXES(ModelLoaderTest, FetchModelTest); | 131 FRIEND_TEST_ALL_PREFIXES(ModelLoaderTest, FetchModelTest); |
| 132 FRIEND_TEST_ALL_PREFIXES(ModelLoaderTest, ModelHasValidHashIds); | 132 FRIEND_TEST_ALL_PREFIXES(ModelLoaderTest, ModelHasValidHashIds); |
| 133 FRIEND_TEST_ALL_PREFIXES(ModelLoaderTest, ModelNamesTest); | 133 FRIEND_TEST_ALL_PREFIXES(ModelLoaderTest, ModelNamesTest); |
| 134 FRIEND_TEST_ALL_PREFIXES(ModelLoaderTest, RescheduleFetchTest); | 134 FRIEND_TEST_ALL_PREFIXES(ModelLoaderTest, RescheduleFetchTest); |
| 135 FRIEND_TEST_ALL_PREFIXES(ModelLoaderTest, UpdateRenderersTest); | 135 FRIEND_TEST_ALL_PREFIXES(ModelLoaderTest, UpdateRenderersTest); |
| 136 FRIEND_TEST_ALL_PREFIXES(ClientSideDetectionServiceTest, | 136 FRIEND_TEST_ALL_PREFIXES(ClientSideDetectionServiceTest, |
| 137 SetEnabledAndRefreshState); | 137 SetEnabledAndRefreshState); |
| 138 DISALLOW_COPY_AND_ASSIGN(ModelLoader); | 138 DISALLOW_COPY_AND_ASSIGN(ModelLoader); |
| 139 }; | 139 }; |
| 140 | 140 |
| 141 } // namespace safe_browsing | 141 } // namespace safe_browsing |
| 142 | 142 |
| 143 #endif // CHROME_BROWSER_SAFE_BROWSING_CLIENT_SIDE_MODEL_LOADER_H_ | 143 #endif // CHROME_BROWSER_SAFE_BROWSING_CLIENT_SIDE_MODEL_LOADER_H_ |
| OLD | NEW |