| OLD | NEW |
| 1 // Copyright 2012 The Chromium Authors. All rights reserved. | 1 // Copyright 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 #include "chrome/browser/extensions/blacklist.h" | 5 #include "chrome/browser/extensions/blacklist.h" |
| 6 | 6 |
| 7 #include <algorithm> | 7 #include <algorithm> |
| 8 #include <iterator> | 8 #include <iterator> |
| 9 | 9 |
| 10 #include "base/bind.h" | 10 #include "base/bind.h" |
| (...skipping 72 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 83 } | 83 } |
| 84 | 84 |
| 85 private: | 85 private: |
| 86 friend class base::RefCountedThreadSafe<SafeBrowsingClientImpl>; | 86 friend class base::RefCountedThreadSafe<SafeBrowsingClientImpl>; |
| 87 virtual ~SafeBrowsingClientImpl() {} | 87 virtual ~SafeBrowsingClientImpl() {} |
| 88 | 88 |
| 89 // Pass |database_manager| as a parameter to avoid touching | 89 // Pass |database_manager| as a parameter to avoid touching |
| 90 // SafeBrowsingService on the IO thread. | 90 // SafeBrowsingService on the IO thread. |
| 91 void StartCheck(scoped_refptr<SafeBrowsingDatabaseManager> database_manager, | 91 void StartCheck(scoped_refptr<SafeBrowsingDatabaseManager> database_manager, |
| 92 const std::set<std::string>& extension_ids) { | 92 const std::set<std::string>& extension_ids) { |
| 93 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO)); | 93 DCHECK_CURRENTLY_ON(BrowserThread::IO); |
| 94 if (database_manager->CheckExtensionIDs(extension_ids, this)) { | 94 if (database_manager->CheckExtensionIDs(extension_ids, this)) { |
| 95 // Definitely not blacklisted. Callback immediately. | 95 // Definitely not blacklisted. Callback immediately. |
| 96 callback_message_loop_->PostTask( | 96 callback_message_loop_->PostTask( |
| 97 FROM_HERE, | 97 FROM_HERE, |
| 98 base::Bind(callback_, std::set<std::string>())); | 98 base::Bind(callback_, std::set<std::string>())); |
| 99 return; | 99 return; |
| 100 } | 100 } |
| 101 // Something might be blacklisted, response will come in | 101 // Something might be blacklisted, response will come in |
| 102 // OnCheckExtensionsResult. | 102 // OnCheckExtensionsResult. |
| 103 AddRef(); // Balanced in OnCheckExtensionsResult | 103 AddRef(); // Balanced in OnCheckExtensionsResult |
| 104 } | 104 } |
| 105 | 105 |
| 106 virtual void OnCheckExtensionsResult( | 106 virtual void OnCheckExtensionsResult( |
| 107 const std::set<std::string>& hits) OVERRIDE { | 107 const std::set<std::string>& hits) OVERRIDE { |
| 108 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO)); | 108 DCHECK_CURRENTLY_ON(BrowserThread::IO); |
| 109 callback_message_loop_->PostTask(FROM_HERE, base::Bind(callback_, hits)); | 109 callback_message_loop_->PostTask(FROM_HERE, base::Bind(callback_, hits)); |
| 110 Release(); // Balanced in StartCheck. | 110 Release(); // Balanced in StartCheck. |
| 111 } | 111 } |
| 112 | 112 |
| 113 scoped_refptr<base::MessageLoopProxy> callback_message_loop_; | 113 scoped_refptr<base::MessageLoopProxy> callback_message_loop_; |
| 114 OnResultCallback callback_; | 114 OnResultCallback callback_; |
| 115 | 115 |
| 116 DISALLOW_COPY_AND_ASSIGN(SafeBrowsingClientImpl); | 116 DISALLOW_COPY_AND_ASSIGN(SafeBrowsingClientImpl); |
| 117 }; | 117 }; |
| 118 | 118 |
| (...skipping 58 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 177 if (!prefs->GetInstalledExtensionInfo(*it)) | 177 if (!prefs->GetInstalledExtensionInfo(*it)) |
| 178 prefs->DeleteExtensionPrefs(*it); | 178 prefs->DeleteExtensionPrefs(*it); |
| 179 } | 179 } |
| 180 } | 180 } |
| 181 | 181 |
| 182 Blacklist::~Blacklist() { | 182 Blacklist::~Blacklist() { |
| 183 } | 183 } |
| 184 | 184 |
| 185 void Blacklist::GetBlacklistedIDs(const std::set<std::string>& ids, | 185 void Blacklist::GetBlacklistedIDs(const std::set<std::string>& ids, |
| 186 const GetBlacklistedIDsCallback& callback) { | 186 const GetBlacklistedIDsCallback& callback) { |
| 187 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI)); | 187 DCHECK_CURRENTLY_ON(BrowserThread::UI); |
| 188 | 188 |
| 189 if (ids.empty() || !g_database_manager.Get().get().get()) { | 189 if (ids.empty() || !g_database_manager.Get().get().get()) { |
| 190 base::MessageLoopProxy::current()->PostTask( | 190 base::MessageLoopProxy::current()->PostTask( |
| 191 FROM_HERE, base::Bind(callback, BlacklistStateMap())); | 191 FROM_HERE, base::Bind(callback, BlacklistStateMap())); |
| 192 return; | 192 return; |
| 193 } | 193 } |
| 194 | 194 |
| 195 // Constructing the SafeBrowsingClientImpl begins the process of asking | 195 // Constructing the SafeBrowsingClientImpl begins the process of asking |
| 196 // safebrowsing for the blacklisted extensions. The set of blacklisted | 196 // safebrowsing for the blacklisted extensions. The set of blacklisted |
| 197 // extensions returned by SafeBrowsing will then be passed to | 197 // extensions returned by SafeBrowsing will then be passed to |
| (...skipping 13 matching lines...) Expand all Loading... |
| 211 void Blacklist::IsBlacklisted(const std::string& extension_id, | 211 void Blacklist::IsBlacklisted(const std::string& extension_id, |
| 212 const IsBlacklistedCallback& callback) { | 212 const IsBlacklistedCallback& callback) { |
| 213 std::set<std::string> check; | 213 std::set<std::string> check; |
| 214 check.insert(extension_id); | 214 check.insert(extension_id); |
| 215 GetBlacklistedIDs(check, base::Bind(&CheckOneExtensionState, callback)); | 215 GetBlacklistedIDs(check, base::Bind(&CheckOneExtensionState, callback)); |
| 216 } | 216 } |
| 217 | 217 |
| 218 void Blacklist::GetBlacklistStateForIDs( | 218 void Blacklist::GetBlacklistStateForIDs( |
| 219 const GetBlacklistedIDsCallback& callback, | 219 const GetBlacklistedIDsCallback& callback, |
| 220 const std::set<std::string>& blacklisted_ids) { | 220 const std::set<std::string>& blacklisted_ids) { |
| 221 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI)); | 221 DCHECK_CURRENTLY_ON(BrowserThread::UI); |
| 222 | 222 |
| 223 std::set<std::string> ids_unknown_state; | 223 std::set<std::string> ids_unknown_state; |
| 224 BlacklistStateMap extensions_state; | 224 BlacklistStateMap extensions_state; |
| 225 for (std::set<std::string>::const_iterator it = blacklisted_ids.begin(); | 225 for (std::set<std::string>::const_iterator it = blacklisted_ids.begin(); |
| 226 it != blacklisted_ids.end(); ++it) { | 226 it != blacklisted_ids.end(); ++it) { |
| 227 BlacklistStateMap::const_iterator cache_it = | 227 BlacklistStateMap::const_iterator cache_it = |
| 228 blacklist_state_cache_.find(*it); | 228 blacklist_state_cache_.find(*it); |
| 229 if (cache_it == blacklist_state_cache_.end() || | 229 if (cache_it == blacklist_state_cache_.end() || |
| 230 cache_it->second == BLACKLISTED_UNKNOWN) // Do not return UNKNOWN | 230 cache_it->second == BLACKLISTED_UNKNOWN) // Do not return UNKNOWN |
| 231 // from cache, retry request. | 231 // from cache, retry request. |
| (...skipping 28 matching lines...) Expand all Loading... |
| 260 extensions_state[*it] = cache_it->second; | 260 extensions_state[*it] = cache_it->second; |
| 261 // If for some reason we still haven't cached the state of this extension, | 261 // If for some reason we still haven't cached the state of this extension, |
| 262 // we silently skip it. | 262 // we silently skip it. |
| 263 } | 263 } |
| 264 | 264 |
| 265 callback.Run(extensions_state); | 265 callback.Run(extensions_state); |
| 266 } | 266 } |
| 267 | 267 |
| 268 void Blacklist::RequestExtensionsBlacklistState( | 268 void Blacklist::RequestExtensionsBlacklistState( |
| 269 const std::set<std::string>& ids, const base::Callback<void()>& callback) { | 269 const std::set<std::string>& ids, const base::Callback<void()>& callback) { |
| 270 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI)); | 270 DCHECK_CURRENTLY_ON(BrowserThread::UI); |
| 271 if (!state_fetcher_) | 271 if (!state_fetcher_) |
| 272 state_fetcher_.reset(new BlacklistStateFetcher()); | 272 state_fetcher_.reset(new BlacklistStateFetcher()); |
| 273 | 273 |
| 274 state_requests_.push_back( | 274 state_requests_.push_back( |
| 275 make_pair(std::vector<std::string>(ids.begin(), ids.end()), callback)); | 275 make_pair(std::vector<std::string>(ids.begin(), ids.end()), callback)); |
| 276 for (std::set<std::string>::const_iterator it = ids.begin(); | 276 for (std::set<std::string>::const_iterator it = ids.begin(); |
| 277 it != ids.end(); | 277 it != ids.end(); |
| 278 ++it) { | 278 ++it) { |
| 279 state_fetcher_->Request( | 279 state_fetcher_->Request( |
| 280 *it, | 280 *it, |
| 281 base::Bind(&Blacklist::OnBlacklistStateReceived, AsWeakPtr(), *it)); | 281 base::Bind(&Blacklist::OnBlacklistStateReceived, AsWeakPtr(), *it)); |
| 282 } | 282 } |
| 283 } | 283 } |
| 284 | 284 |
| 285 void Blacklist::OnBlacklistStateReceived(const std::string& id, | 285 void Blacklist::OnBlacklistStateReceived(const std::string& id, |
| 286 BlacklistState state) { | 286 BlacklistState state) { |
| 287 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI)); | 287 DCHECK_CURRENTLY_ON(BrowserThread::UI); |
| 288 blacklist_state_cache_[id] = state; | 288 blacklist_state_cache_[id] = state; |
| 289 | 289 |
| 290 // Go through the opened requests and call the callbacks for those requests | 290 // Go through the opened requests and call the callbacks for those requests |
| 291 // for which we already got all the required blacklist states. | 291 // for which we already got all the required blacklist states. |
| 292 StateRequestsList::iterator requests_it = state_requests_.begin(); | 292 StateRequestsList::iterator requests_it = state_requests_.begin(); |
| 293 while (requests_it != state_requests_.end()) { | 293 while (requests_it != state_requests_.end()) { |
| 294 const std::vector<std::string>& ids = requests_it->first; | 294 const std::vector<std::string>& ids = requests_it->first; |
| 295 | 295 |
| 296 bool have_all_in_cache = true; | 296 bool have_all_in_cache = true; |
| 297 for (std::vector<std::string>::const_iterator ids_it = ids.begin(); | 297 for (std::vector<std::string>::const_iterator ids_it = ids.begin(); |
| (...skipping 17 matching lines...) Expand all Loading... |
| 315 void Blacklist::SetBlacklistStateFetcherForTest( | 315 void Blacklist::SetBlacklistStateFetcherForTest( |
| 316 BlacklistStateFetcher* fetcher) { | 316 BlacklistStateFetcher* fetcher) { |
| 317 state_fetcher_.reset(fetcher); | 317 state_fetcher_.reset(fetcher); |
| 318 } | 318 } |
| 319 | 319 |
| 320 BlacklistStateFetcher* Blacklist::ResetBlacklistStateFetcherForTest() { | 320 BlacklistStateFetcher* Blacklist::ResetBlacklistStateFetcherForTest() { |
| 321 return state_fetcher_.release(); | 321 return state_fetcher_.release(); |
| 322 } | 322 } |
| 323 | 323 |
| 324 void Blacklist::AddObserver(Observer* observer) { | 324 void Blacklist::AddObserver(Observer* observer) { |
| 325 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI)); | 325 DCHECK_CURRENTLY_ON(BrowserThread::UI); |
| 326 observers_.AddObserver(observer); | 326 observers_.AddObserver(observer); |
| 327 } | 327 } |
| 328 | 328 |
| 329 void Blacklist::RemoveObserver(Observer* observer) { | 329 void Blacklist::RemoveObserver(Observer* observer) { |
| 330 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI)); | 330 DCHECK_CURRENTLY_ON(BrowserThread::UI); |
| 331 observers_.RemoveObserver(observer); | 331 observers_.RemoveObserver(observer); |
| 332 } | 332 } |
| 333 | 333 |
| 334 // static | 334 // static |
| 335 void Blacklist::SetDatabaseManager( | 335 void Blacklist::SetDatabaseManager( |
| 336 scoped_refptr<SafeBrowsingDatabaseManager> database_manager) { | 336 scoped_refptr<SafeBrowsingDatabaseManager> database_manager) { |
| 337 g_database_manager.Get().set(database_manager); | 337 g_database_manager.Get().set(database_manager); |
| 338 } | 338 } |
| 339 | 339 |
| 340 // static | 340 // static |
| 341 scoped_refptr<SafeBrowsingDatabaseManager> Blacklist::GetDatabaseManager() { | 341 scoped_refptr<SafeBrowsingDatabaseManager> Blacklist::GetDatabaseManager() { |
| 342 return g_database_manager.Get().get(); | 342 return g_database_manager.Get().get(); |
| 343 } | 343 } |
| 344 | 344 |
| 345 void Blacklist::Observe(int type, | 345 void Blacklist::Observe(int type, |
| 346 const content::NotificationSource& source, | 346 const content::NotificationSource& source, |
| 347 const content::NotificationDetails& details) { | 347 const content::NotificationDetails& details) { |
| 348 DCHECK_EQ(chrome::NOTIFICATION_SAFE_BROWSING_UPDATE_COMPLETE, type); | 348 DCHECK_EQ(chrome::NOTIFICATION_SAFE_BROWSING_UPDATE_COMPLETE, type); |
| 349 FOR_EACH_OBSERVER(Observer, observers_, OnBlacklistUpdated()); | 349 FOR_EACH_OBSERVER(Observer, observers_, OnBlacklistUpdated()); |
| 350 } | 350 } |
| 351 | 351 |
| 352 } // namespace extensions | 352 } // namespace extensions |
| OLD | NEW |