Home | History | Annotate | Download | only in extensions 
      1 // Copyright 2013 The Chromium Authors. All rights reserved.
      2 // Use of this source code is governed by a BSD-style license that can be
      3 // found in the LICENSE file.
      4 
      5 #include "chrome/browser/extensions/fake_safe_browsing_database_manager.h"
      6 
      7 #include <algorithm>
      8 #include <iterator>
      9 #include <vector>
     10 
     11 #include "base/bind_helpers.h"
     12 #include "base/memory/ref_counted.h"
     13 #include "base/message_loop/message_loop_proxy.h"
     14 #include "chrome/browser/safe_browsing/safe_browsing_service.h"
     15 #include "chrome/browser/safe_browsing/safe_browsing_util.h"
     16 
     17 namespace extensions {
     18 
     19 FakeSafeBrowsingDatabaseManager::FakeSafeBrowsingDatabaseManager()
     20     : SafeBrowsingDatabaseManager(
     21           make_scoped_refptr(SafeBrowsingService::CreateSafeBrowsingService())),
     22       enabled_(false) {
     23 }
     24 
     25 FakeSafeBrowsingDatabaseManager::~FakeSafeBrowsingDatabaseManager() {
     26 }
     27 
     28 bool FakeSafeBrowsingDatabaseManager::CheckExtensionIDs(
     29     const std::set<std::string>& extension_ids,
     30     Client* client) {
     31   if (!enabled_)
     32     return true;
     33 
     34   // Need to construct the full SafeBrowsingCheck rather than calling
     35   // OnCheckExtensionsResult directly because it's protected. Grr!
     36   std::vector<std::string> extension_ids_vector(extension_ids.begin(),
     37                                                 extension_ids.end());
     38   std::vector<SBFullHash> extension_id_hashes;
     39   std::transform(extension_ids_vector.begin(), extension_ids_vector.end(),
     40                  std::back_inserter(extension_id_hashes),
     41                  safe_browsing_util::StringToSBFullHash);
     42 
     43   scoped_ptr<SafeBrowsingCheck> safe_browsing_check(
     44       new SafeBrowsingCheck(std::vector<GURL>(),
     45                              extension_id_hashes,
     46                              client,
     47                              safe_browsing_util::EXTENSIONBLACKLIST));
     48 
     49   for (size_t i = 0; i < extension_ids_vector.size(); ++i) {
     50     const std::string& extension_id = extension_ids_vector[i];
     51     if (unsafe_ids_.count(extension_id))
     52       safe_browsing_check->full_hash_results[i] = SB_THREAT_TYPE_EXTENSION;
     53   }
     54 
     55   base::MessageLoopProxy::current()->PostTask(
     56       FROM_HERE,
     57       base::Bind(&FakeSafeBrowsingDatabaseManager::OnSafeBrowsingResult,
     58                  this,
     59                  base::Passed(&safe_browsing_check),
     60                  client));
     61   return false;
     62 }
     63 
     64 void FakeSafeBrowsingDatabaseManager::OnSafeBrowsingResult(
     65     scoped_ptr<SafeBrowsingCheck> result,
     66     Client* client) {
     67   client->OnSafeBrowsingResult(*result);
     68 }
     69 
     70 }  // namespace extensions
     71