Home | History | Annotate | Download | only in crypto
      1 // Copyright (c) 2012 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 "crypto/ec_private_key.h"
      6 
      7 #include <openssl/ec.h>
      8 #include <openssl/evp.h>
      9 #include <openssl/pkcs12.h>
     10 #include <openssl/x509.h>
     11 
     12 #include "base/logging.h"
     13 #include "base/memory/scoped_ptr.h"
     14 #include "crypto/openssl_util.h"
     15 
     16 namespace crypto {
     17 
     18 namespace {
     19 
     20 // Function pointer definition, for injecting the required key export function
     21 // into ExportKeyWithBio, below. |bio| is a temporary memory BIO object, and
     22 // |key| is a handle to the input key object. Return 1 on success, 0 otherwise.
     23 // NOTE: Used with OpenSSL functions, which do not comply with the Chromium
     24 //       style guide, hence the unusual parameter placement / types.
     25 typedef int (*ExportBioFunction)(BIO* bio, const void* key);
     26 
     27 // Helper to export |key| into |output| via the specified ExportBioFunction.
     28 bool ExportKeyWithBio(const void* key,
     29                       ExportBioFunction export_fn,
     30                       std::vector<uint8>* output) {
     31   if (!key)
     32     return false;
     33 
     34   ScopedOpenSSL<BIO, BIO_free_all> bio(BIO_new(BIO_s_mem()));
     35   if (!bio.get())
     36     return false;
     37 
     38   if (!export_fn(bio.get(), key))
     39     return false;
     40 
     41   char* data = NULL;
     42   long len = BIO_get_mem_data(bio.get(), &data);
     43   if (!data || len < 0)
     44     return false;
     45 
     46   output->assign(data, data + len);
     47   return true;
     48 }
     49 
     50 // Function pointer definition, for injecting the required key export function
     51 // into ExportKey below. |key| is a pointer to the input key object,
     52 // and |data| is either NULL, or the address of an 'unsigned char*' pointer
     53 // that points to the start of the output buffer. The function must return
     54 // the number of bytes required to export the data, or -1 in case of error.
     55 typedef int (*ExportDataFunction)(const void* key, unsigned char** data);
     56 
     57 // Helper to export |key| into |output| via the specified export function.
     58 bool ExportKey(const void* key,
     59                ExportDataFunction export_fn,
     60                std::vector<uint8>* output) {
     61   if (!key)
     62     return false;
     63 
     64   int data_len = export_fn(key, NULL);
     65   if (data_len < 0)
     66     return false;
     67 
     68   output->resize(static_cast<size_t>(data_len));
     69   unsigned char* data = &(*output)[0];
     70   if (export_fn(key, &data) < 0)
     71     return false;
     72 
     73   return true;
     74 }
     75 
     76 }  // namespace
     77 
     78 ECPrivateKey::~ECPrivateKey() {
     79   if (key_)
     80     EVP_PKEY_free(key_);
     81 }
     82 
     83 // static
     84 bool ECPrivateKey::IsSupported() { return true; }
     85 
     86 // static
     87 ECPrivateKey* ECPrivateKey::Create() {
     88   OpenSSLErrStackTracer err_tracer(FROM_HERE);
     89 
     90   ScopedOpenSSL<EC_KEY, EC_KEY_free> ec_key(
     91       EC_KEY_new_by_curve_name(NID_X9_62_prime256v1));
     92   if (!ec_key.get() || !EC_KEY_generate_key(ec_key.get()))
     93     return NULL;
     94 
     95   scoped_ptr<ECPrivateKey> result(new ECPrivateKey());
     96   result->key_ = EVP_PKEY_new();
     97   if (!result->key_ || !EVP_PKEY_set1_EC_KEY(result->key_, ec_key.get()))
     98     return NULL;
     99 
    100   return result.release();
    101 }
    102 
    103 // static
    104 ECPrivateKey* ECPrivateKey::CreateFromEncryptedPrivateKeyInfo(
    105     const std::string& password,
    106     const std::vector<uint8>& encrypted_private_key_info,
    107     const std::vector<uint8>& subject_public_key_info) {
    108   // NOTE: The |subject_public_key_info| can be ignored here, it is only
    109   // useful for the NSS implementation (which uses the public key's SHA1
    110   // as a lookup key when storing the private one in its store).
    111   if (encrypted_private_key_info.empty())
    112     return NULL;
    113 
    114   OpenSSLErrStackTracer err_tracer(FROM_HERE);
    115   // Write the encrypted private key into a memory BIO.
    116   char* private_key_data = reinterpret_cast<char*>(
    117       const_cast<uint8*>(&encrypted_private_key_info[0]));
    118   int private_key_data_len =
    119       static_cast<int>(encrypted_private_key_info.size());
    120   ScopedOpenSSL<BIO, BIO_free_all> bio(
    121       BIO_new_mem_buf(private_key_data, private_key_data_len));
    122   if (!bio.get())
    123     return NULL;
    124 
    125   // Convert it, then decrypt it into a PKCS#8 object.
    126   ScopedOpenSSL<X509_SIG, X509_SIG_free> p8_encrypted(
    127       d2i_PKCS8_bio(bio.get(), NULL));
    128   if (!p8_encrypted.get())
    129     return NULL;
    130 
    131   ScopedOpenSSL<PKCS8_PRIV_KEY_INFO, PKCS8_PRIV_KEY_INFO_free> p8_decrypted(
    132       PKCS8_decrypt(p8_encrypted.get(),
    133                     password.c_str(),
    134                     static_cast<int>(password.size())));
    135   if (!p8_decrypted.get() && password.empty()) {
    136     // Hack for reading keys generated by ec_private_key_nss. Passing NULL
    137     // causes OpenSSL to use an empty password instead of "\0\0".
    138     p8_decrypted.reset(PKCS8_decrypt(p8_encrypted.get(), NULL, 0));
    139   }
    140   if (!p8_decrypted.get())
    141     return NULL;
    142 
    143   // Create a new EVP_PKEY for it.
    144   scoped_ptr<ECPrivateKey> result(new ECPrivateKey);
    145   result->key_ = EVP_PKCS82PKEY(p8_decrypted.get());
    146   if (!result->key_)
    147     return NULL;
    148 
    149   return result.release();
    150 }
    151 
    152 bool ECPrivateKey::ExportEncryptedPrivateKey(
    153     const std::string& password,
    154     int iterations,
    155     std::vector<uint8>* output) {
    156   OpenSSLErrStackTracer err_tracer(FROM_HERE);
    157   // Convert into a PKCS#8 object.
    158   ScopedOpenSSL<PKCS8_PRIV_KEY_INFO, PKCS8_PRIV_KEY_INFO_free> pkcs8(
    159       EVP_PKEY2PKCS8(key_));
    160   if (!pkcs8.get())
    161     return false;
    162 
    163   // Encrypt the object.
    164   // NOTE: NSS uses SEC_OID_PKCS12_V2_PBE_WITH_SHA1_AND_3KEY_TRIPLE_DES_CBC
    165   // so use NID_pbe_WithSHA1And3_Key_TripleDES_CBC which should be the OpenSSL
    166   // equivalent.
    167   ScopedOpenSSL<X509_SIG, X509_SIG_free> encrypted(
    168       PKCS8_encrypt(NID_pbe_WithSHA1And3_Key_TripleDES_CBC,
    169                     NULL,
    170                     password.c_str(),
    171                     static_cast<int>(password.size()),
    172                     NULL,
    173                     0,
    174                     iterations,
    175                     pkcs8.get()));
    176   if (!encrypted.get())
    177     return false;
    178 
    179   // Write it into |*output|
    180   return ExportKeyWithBio(encrypted.get(),
    181                           reinterpret_cast<ExportBioFunction>(i2d_PKCS8_bio),
    182                           output);
    183 }
    184 
    185 bool ECPrivateKey::ExportPublicKey(std::vector<uint8>* output) {
    186   OpenSSLErrStackTracer err_tracer(FROM_HERE);
    187   return ExportKeyWithBio(
    188       key_, reinterpret_cast<ExportBioFunction>(i2d_PUBKEY_bio), output);
    189 }
    190 
    191 bool ECPrivateKey::ExportValue(std::vector<uint8>* output) {
    192   OpenSSLErrStackTracer err_tracer(FROM_HERE);
    193   ScopedOpenSSL<EC_KEY, EC_KEY_free> ec_key(EVP_PKEY_get1_EC_KEY(key_));
    194   return ExportKey(ec_key.get(),
    195                    reinterpret_cast<ExportDataFunction>(i2d_ECPrivateKey),
    196                    output);
    197 }
    198 
    199 bool ECPrivateKey::ExportECParams(std::vector<uint8>* output) {
    200   OpenSSLErrStackTracer err_tracer(FROM_HERE);
    201   ScopedOpenSSL<EC_KEY, EC_KEY_free> ec_key(EVP_PKEY_get1_EC_KEY(key_));
    202   return ExportKey(ec_key.get(),
    203                    reinterpret_cast<ExportDataFunction>(i2d_ECParameters),
    204                    output);
    205 }
    206 
    207 ECPrivateKey::ECPrivateKey() : key_(NULL) {}
    208 
    209 }  // namespace crypto
    210