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 16using content::BrowserThread; 17 18namespace { 19 20scoped_ptr<std::vector<SupervisedUserBlacklist::Hash> > 21ReadFromBinaryFileOnFileThread(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 52SupervisedUserBlacklist::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 58bool SupervisedUserBlacklist::Hash::operator<(const Hash& rhs) const { 59 return memcmp(data, rhs.data, base::kSHA1Length) < 0; 60} 61 62SupervisedUserBlacklist::SupervisedUserBlacklist() : weak_ptr_factory_(this) {} 63 64SupervisedUserBlacklist::~SupervisedUserBlacklist() {} 65 66bool 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 71size_t SupervisedUserBlacklist::GetEntryCount() const { 72 return host_hashes_.size(); 73} 74 75void SupervisedUserBlacklist::ReadFromFile(const base::FilePath& path, 76 const base::Closure& done_callback) { 77 base::PostTaskAndReplyWithResult( 78 BrowserThread::GetBlockingPool(), 79 FROM_HERE, 80 base::Bind(&ReadFromBinaryFileOnFileThread, path), 81 base::Bind(&SupervisedUserBlacklist::OnReadFromFileCompleted, 82 weak_ptr_factory_.GetWeakPtr(), 83 done_callback)); 84} 85 86void SupervisedUserBlacklist::OnReadFromFileCompleted( 87 const base::Closure& done_callback, 88 scoped_ptr<std::vector<Hash> > host_hashes) { 89 host_hashes_.swap(*host_hashes); 90 LOG_IF(WARNING, host_hashes_.empty()) << "Got empty blacklist"; 91 92 if (!done_callback.is_null()) 93 done_callback.Run(); 94} 95