chrome/common/safe_browsing/binary_feature_extractor.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/common/safe_browsing/binary_feature_extractor.h"
   6
   7 #include "base/files/file.h"
   8 #include "base/files/file_path.h"
   9 #include "base/memory/scoped_ptr.h"
  10 #include "chrome/common/safe_browsing/csd.pb.h"
  11 #include "crypto/secure_hash.h"
  12 #include "crypto/sha2.h"
  13
  14 namespace safe_browsing {
  15
  16 void BinaryFeatureExtractor::ExtractDigest(
  17     const base::FilePath& file_path,
  18     ClientDownloadRequest_Digests* digests) {
  19   base::File file(file_path, base::File::FLAG_OPEN | base::File::FLAG_READ);
  20   if (file.IsValid()) {
  21     const int kBufferSize = 1 << 12;
  22     scoped_ptr<char[]> buf(new char[kBufferSize]);
  23     scoped_ptr<crypto::SecureHash> ctx(
  24         crypto::SecureHash::Create(crypto::SecureHash::SHA256));
  25     int len = 0;
  26     while (true) {
  27       len = file.ReadAtCurrentPos(buf.get(), kBufferSize);
  28       if (len <= 0)
  29         break;
  30       ctx->Update(buf.get(), len);
  31     }
  32     if (!len) {
  33       uint8_t hash[crypto::kSHA256Length];
  34       ctx->Finish(hash, sizeof(hash));
  35       digests->set_sha256(hash, sizeof(hash));
  36     }
  37   }
  38 }
  39
  40 }  // namespace safe_browsing