net/test/cert_test_util_nss.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 "net/test/cert_test_util.h"
   6
   7 #include <pk11pub.h>
   8 #include <secmodt.h>
   9
  10 #include "base/files/file_path.h"
  11 #include "base/files/file_util.h"
  12 #include "crypto/nss_util.h"
  13 #include "crypto/rsa_private_key.h"
  14 #include "net/cert/cert_type.h"
  15
  16 namespace net {
  17
  18 scoped_ptr<crypto::RSAPrivateKey> ImportSensitiveKeyFromFile(
  19     const base::FilePath& dir,
  20     const std::string& key_filename,
  21     PK11SlotInfo* slot) {
  22 #if defined(USE_OPENSSL)
  23   // TODO(davidben): Port RSAPrivateKey::CreateSensitiveFromPrivateKeyInfo away
  24   // from RSAPrivateKey so it doesn't make assumptions about the internal crypto
  25   // library. Instead, return a ScopedSECKEYPrivateKey or have this function
  26   // just return bool. https://crbug.com/478777
  27   NOTIMPLEMENTED();
  28   return nullptr;
  29 #else
  30   base::FilePath key_path = dir.AppendASCII(key_filename);
  31   std::string key_pkcs8;
  32   bool success = base::ReadFileToString(key_path, &key_pkcs8);
  33   if (!success) {
  34     LOG(ERROR) << "Failed to read file " << key_path.value();
  35     return scoped_ptr<crypto::RSAPrivateKey>();
  36   }
  37
  38   const uint8* key_pkcs8_begin =
  39       reinterpret_cast<const uint8*>(key_pkcs8.data());
  40   std::vector<uint8> key_vector(key_pkcs8_begin,
  41                                 key_pkcs8_begin + key_pkcs8.length());
  42
  43   scoped_ptr<crypto::RSAPrivateKey> private_key(
  44       crypto::RSAPrivateKey::CreateSensitiveFromPrivateKeyInfo(slot,
  45                                                                key_vector));
  46   LOG_IF(ERROR, !private_key) << "Could not create key from file "
  47                               << key_path.value();
  48   return private_key.Pass();
  49 #endif
  50 }
  51
  52 bool ImportClientCertToSlot(const scoped_refptr<X509Certificate>& cert,
  53                             PK11SlotInfo* slot) {
  54   std::string nickname = cert->GetDefaultNickname(USER_CERT);
  55   {
  56     crypto::AutoNSSWriteLock lock;
  57     SECStatus rv = PK11_ImportCert(slot,
  58                                    cert->os_cert_handle(),
  59                                    CK_INVALID_HANDLE,
  60                                    nickname.c_str(),
  61                                    PR_FALSE);
  62     if (rv != SECSuccess) {
  63       LOG(ERROR) << "Could not import cert";
  64       return false;
  65     }
  66   }
  67   return true;
  68 }
  69
  70 scoped_refptr<X509Certificate> ImportClientCertAndKeyFromFile(
  71     const base::FilePath& dir,
  72     const std::string& cert_filename,
  73     const std::string& key_filename,
  74     PK11SlotInfo* slot) {
  75   if (!ImportSensitiveKeyFromFile(dir, key_filename, slot)) {
  76     LOG(ERROR) << "Could not import private key from file " << key_filename;
  77     return NULL;
  78   }
  79
  80   scoped_refptr<X509Certificate> cert(ImportCertFromFile(dir, cert_filename));
  81
  82   if (!cert.get()) {
  83     LOG(ERROR) << "Failed to parse cert from file " << cert_filename;
  84     return NULL;
  85   }
  86
  87   if (!ImportClientCertToSlot(cert, slot))
  88     return NULL;
  89
  90   // |cert| continues to point to the original X509Certificate before the
  91   // import to |slot|. However this should not make a difference as NSS handles
  92   // state globally.
  93   return cert;
  94 }
  95
  96 }  // namespace net