| OLD | NEW |
| 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 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 | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "crypto/signature_creator.h" | 5 #include "crypto/signature_creator.h" |
| 6 | 6 |
| 7 #include <openssl/evp.h> | 7 #include <openssl/evp.h> |
| 8 #include <openssl/rsa.h> |
| 8 | 9 |
| 9 #include "base/logging.h" | 10 #include "base/logging.h" |
| 10 #include "base/memory/scoped_ptr.h" | 11 #include "base/memory/scoped_ptr.h" |
| 11 #include "base/stl_util.h" | 12 #include "base/stl_util.h" |
| 12 #include "crypto/openssl_util.h" | 13 #include "crypto/openssl_util.h" |
| 13 #include "crypto/rsa_private_key.h" | 14 #include "crypto/rsa_private_key.h" |
| 14 | 15 |
| 15 namespace crypto { | 16 namespace crypto { |
| 16 | 17 |
| 17 // static | 18 // static |
| 18 SignatureCreator* SignatureCreator::Create(RSAPrivateKey* key) { | 19 SignatureCreator* SignatureCreator::Create(RSAPrivateKey* key) { |
| 19 OpenSSLErrStackTracer err_tracer(FROM_HERE); | 20 OpenSSLErrStackTracer err_tracer(FROM_HERE); |
| 20 scoped_ptr<SignatureCreator> result(new SignatureCreator); | 21 scoped_ptr<SignatureCreator> result(new SignatureCreator); |
| 21 result->key_ = key; | 22 result->key_ = key; |
| 22 if (!EVP_SignInit_ex(result->sign_context_, EVP_sha1(), NULL)) | 23 if (!EVP_SignInit_ex(result->sign_context_, EVP_sha1(), NULL)) |
| 23 return NULL; | 24 return NULL; |
| 24 return result.release(); | 25 return result.release(); |
| 25 } | 26 } |
| 26 | 27 |
| 28 // static |
| 29 bool SignatureCreator::Sign(RSAPrivateKey* key, |
| 30 const uint8* data, |
| 31 int data_len, |
| 32 std::vector<uint8>* signature) { |
| 33 RSA* rsa_key = EVP_PKEY_get1_RSA(key->key()); |
| 34 if (!rsa_key) |
| 35 return false; |
| 36 signature->resize(RSA_size(rsa_key)); |
| 37 |
| 38 unsigned int len = 0; |
| 39 bool success = RSA_sign(NID_sha1, data, data_len, vector_as_array(signature), |
| 40 &len, rsa_key); |
| 41 if (!success) { |
| 42 signature->clear(); |
| 43 return false; |
| 44 } |
| 45 signature->resize(len); |
| 46 return true; |
| 47 } |
| 48 |
| 27 SignatureCreator::SignatureCreator() | 49 SignatureCreator::SignatureCreator() |
| 28 : sign_context_(EVP_MD_CTX_create()) { | 50 : sign_context_(EVP_MD_CTX_create()) { |
| 29 } | 51 } |
| 30 | 52 |
| 31 SignatureCreator::~SignatureCreator() { | 53 SignatureCreator::~SignatureCreator() { |
| 32 EVP_MD_CTX_destroy(sign_context_); | 54 EVP_MD_CTX_destroy(sign_context_); |
| 33 } | 55 } |
| 34 | 56 |
| 35 bool SignatureCreator::Update(const uint8* data_part, int data_part_len) { | 57 bool SignatureCreator::Update(const uint8* data_part, int data_part_len) { |
| 36 OpenSSLErrStackTracer err_tracer(FROM_HERE); | 58 OpenSSLErrStackTracer err_tracer(FROM_HERE); |
| 37 return EVP_SignUpdate(sign_context_, data_part, data_part_len) == 1; | 59 return EVP_SignUpdate(sign_context_, data_part, data_part_len) == 1; |
| 38 } | 60 } |
| 39 | 61 |
| 40 bool SignatureCreator::Final(std::vector<uint8>* signature) { | 62 bool SignatureCreator::Final(std::vector<uint8>* signature) { |
| 41 OpenSSLErrStackTracer err_tracer(FROM_HERE); | 63 OpenSSLErrStackTracer err_tracer(FROM_HERE); |
| 42 EVP_PKEY* key = key_->key(); | 64 EVP_PKEY* key = key_->key(); |
| 43 signature->resize(EVP_PKEY_size(key)); | 65 signature->resize(EVP_PKEY_size(key)); |
| 44 | 66 |
| 45 unsigned int len = 0; | 67 unsigned int len = 0; |
| 46 int rv = EVP_SignFinal(sign_context_, vector_as_array(signature), &len, key); | 68 int rv = EVP_SignFinal(sign_context_, vector_as_array(signature), &len, key); |
| 47 if (!rv) { | 69 if (!rv) { |
| 48 signature->clear(); | 70 signature->clear(); |
| 49 return false; | 71 return false; |
| 50 } | 72 } |
| 51 signature->resize(len); | 73 signature->resize(len); |
| 52 return true; | 74 return true; |
| 53 } | 75 } |
| 54 | 76 |
| 55 } // namespace crypto | 77 } // namespace crypto |
| OLD | NEW |