chrome/browser/supervised_user/experimental/supervised_user_blacklist.cc

   1 // Copyright 2014 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/supervised_user/experimental/supervised_user_blacklist.h"
   6
   7 #include <algorithm>
   8 #include <cstring>
   9 #include <fstream>
  10
  11 #include "base/files/file_path.h"
  12 #include "base/files/memory_mapped_file.h"
  13 #include "content/public/browser/browser_thread.h"
  14 #include "url/gurl.h"
  15
  16 using content::BrowserThread;
  17
  18 namespace {
  19
  20 scoped_ptr<std::vector<SupervisedUserBlacklist::Hash> >
  21 ReadFromBinaryFileOnFileThread(const base::FilePath& path) {
  22   DCHECK(BrowserThread::GetBlockingPool()->RunsTasksOnCurrentThread());
  23
  24   scoped_ptr<std::vector<SupervisedUserBlacklist::Hash> > host_hashes(
  25       new std::vector<SupervisedUserBlacklist::Hash>);
  26
  27   base::MemoryMappedFile file;
  28   file.Initialize(path);
  29   if (!file.IsValid())
  30     return host_hashes.Pass();
  31
  32   size_t size = file.length();
  33   if (size <= 0 || size % base::kSHA1Length != 0)
  34     return host_hashes.Pass();
  35
  36   size_t hash_count = size / base::kSHA1Length;
  37   host_hashes->resize(hash_count);
  38
  39   for (size_t i = 0; i < hash_count; i++) {
  40     memcpy((*host_hashes.get())[i].data,
  41            file.data() + i * base::kSHA1Length,
  42            base::kSHA1Length);
  43   }
  44
  45   std::sort(host_hashes->begin(), host_hashes->end());
  46
  47   return host_hashes.Pass();
  48 }
  49
  50 } // namespace
  51
  52 SupervisedUserBlacklist::Hash::Hash(const std::string& host) {
  53   const unsigned char* host_bytes =
  54       reinterpret_cast<const unsigned char*>(host.c_str());
  55   base::SHA1HashBytes(host_bytes, host.length(), data);
  56 }
  57
  58 bool SupervisedUserBlacklist::Hash::operator<(const Hash& rhs) const {
  59   return memcmp(data, rhs.data, base::kSHA1Length) < 0;
  60 }
  61
  62 SupervisedUserBlacklist::SupervisedUserBlacklist() : weak_ptr_factory_(this) {}
  63
  64 SupervisedUserBlacklist::~SupervisedUserBlacklist() {}
  65
  66 bool SupervisedUserBlacklist::HasURL(const GURL& url) const {
  67   Hash hash(url.host());
  68   return std::binary_search(host_hashes_.begin(), host_hashes_.end(), hash);
  69 }
  70
  71 size_t SupervisedUserBlacklist::GetEntryCount() const {
  72   return host_hashes_.size();
  73 }
  74
  75 void SupervisedUserBlacklist::ReadFromFile(const base::FilePath& path,
  76                                            const base::Closure& done_callback) {
  77   base::PostTaskAndReplyWithResult(
  78       BrowserThread::GetBlockingPool()->GetTaskRunnerWithShutdownBehavior(
  79           base::SequencedWorkerPool::CONTINUE_ON_SHUTDOWN).get(),
  80       FROM_HERE,
  81       base::Bind(&ReadFromBinaryFileOnFileThread, path),
  82       base::Bind(&SupervisedUserBlacklist::OnReadFromFileCompleted,
  83                  weak_ptr_factory_.GetWeakPtr(),
  84                  done_callback));
  85 }
  86
  87 void SupervisedUserBlacklist::OnReadFromFileCompleted(
  88     const base::Closure& done_callback,
  89     scoped_ptr<std::vector<Hash> > host_hashes) {
  90   host_hashes_.swap(*host_hashes);
  91   LOG_IF(WARNING, host_hashes_.empty()) << "Got empty blacklist";
  92
  93   if (!done_callback.is_null())
  94     done_callback.Run();
  95 }