components/ownership/owner_key_util_impl.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 "components/ownership/owner_key_util_impl.h"
   6
   7 #include <keythi.h>
   8 #include <limits>
   9
  10 #include "base/files/file_util.h"
  11 #include "base/logging.h"
  12 #include "base/sys_info.h"
  13 #include "build/build_config.h"
  14 #include "crypto/nss_key_util.h"
  15
  16 namespace ownership {
  17
  18 OwnerKeyUtilImpl::OwnerKeyUtilImpl(const base::FilePath& public_key_file)
  19     : public_key_file_(public_key_file) {
  20 }
  21
  22 OwnerKeyUtilImpl::~OwnerKeyUtilImpl() {
  23 }
  24
  25 bool OwnerKeyUtilImpl::ImportPublicKey(std::vector<uint8_t>* output) {
  26   // Get the file size (must fit in a 32 bit int for NSS).
  27   int64_t file_size;
  28   if (!base::GetFileSize(public_key_file_, &file_size)) {
  29 #if defined(OS_CHROMEOS)
  30     LOG_IF(ERROR, base::SysInfo::IsRunningOnChromeOS())
  31         << "Could not get size of " << public_key_file_.value();
  32 #endif  // defined(OS_CHROMEOS)
  33     return false;
  34   }
  35   if (file_size > static_cast<int64_t>(std::numeric_limits<int>::max())) {
  36     LOG(ERROR) << public_key_file_.value() << "is " << file_size
  37                << "bytes!!!  Too big!";
  38     return false;
  39   }
  40   int32_t safe_file_size = static_cast<int32_t>(file_size);
  41
  42   output->resize(safe_file_size);
  43
  44   if (safe_file_size == 0) {
  45     LOG(WARNING) << "Public key file is empty. This seems wrong.";
  46     return false;
  47   }
  48
  49   // Get the key data off of disk
  50   int data_read =
  51       base::ReadFile(public_key_file_, reinterpret_cast<char*>(output->data()),
  52                      safe_file_size);
  53   return data_read == safe_file_size;
  54 }
  55
  56 crypto::ScopedSECKEYPrivateKey OwnerKeyUtilImpl::FindPrivateKeyInSlot(
  57     const std::vector<uint8_t>& key,
  58     PK11SlotInfo* slot) {
  59   if (!slot)
  60     return nullptr;
  61
  62   crypto::ScopedSECKEYPrivateKey private_key(
  63       crypto::FindNSSKeyFromPublicKeyInfoInSlot(key, slot));
  64   if (!private_key || SECKEY_GetPrivateKeyType(private_key.get()) != rsaKey)
  65     return nullptr;
  66   return private_key;
  67 }
  68
  69 bool OwnerKeyUtilImpl::IsPublicKeyPresent() {
  70   return base::PathExists(public_key_file_);
  71 }
  72
  73 }  // namespace ownership