Chromium Code Reviews| 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/ec_signature_creator.h" | 5 #include "crypto/ec_signature_creator.h" |
| 6 | 6 |
| 7 #include <cryptohi.h> | 7 #include <cryptohi.h> |
| 8 #include <pk11pub.h> | 8 #include <pk11pub.h> |
| 9 #include <secerr.h> | 9 #include <secerr.h> |
| 10 #include <sechash.h> | 10 #include <sechash.h> |
| 11 | 11 |
| 12 #include "base/logging.h" | 12 #include "base/logging.h" |
| 13 #include "crypto/ec_private_key.h" | 13 #include "crypto/ec_private_key.h" |
| 14 #include "crypto/nss_util.h" | 14 #include "crypto/nss_util.h" |
| 15 #include "crypto/scoped_nss_types.h" | 15 #include "crypto/scoped_nss_types.h" |
| 16 | 16 |
| 17 namespace crypto { | 17 namespace crypto { |
| 18 | 18 |
| 19 namespace { | 19 namespace { |
| 20 | 20 |
| 21 SECStatus SignData(PLArenaPool* arena, | 21 SECStatus SignData(SECItem* result, |
| 22 SECItem* result, | |
| 23 SECItem* input, | 22 SECItem* input, |
| 24 SECKEYPrivateKey* key, | 23 SECKEYPrivateKey* key, |
| 25 HASH_HashType hash_type) { | 24 HASH_HashType hash_type) { |
| 26 if (key->keyType != ecKey) { | 25 if (key->keyType != ecKey) { |
| 27 DLOG(FATAL) << "Should be using an EC key."; | 26 DLOG(FATAL) << "Should be using an EC key."; |
| 28 PORT_SetError(SEC_ERROR_INVALID_ARGS); | 27 PORT_SetError(SEC_ERROR_INVALID_ARGS); |
| 29 return SECFailure; | 28 return SECFailure; |
| 30 } | 29 } |
| 31 | 30 |
| 32 // Hash the input. | 31 // Hash the input. |
| (...skipping 32 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... | |
| 65 | 64 |
| 66 bool ECSignatureCreator::Sign(const uint8* data, | 65 bool ECSignatureCreator::Sign(const uint8* data, |
| 67 int data_len, | 66 int data_len, |
| 68 std::vector<uint8>* signature) { | 67 std::vector<uint8>* signature) { |
| 69 // Data to be signed | 68 // Data to be signed |
| 70 SECItem secret; | 69 SECItem secret; |
| 71 secret.type = siBuffer; | 70 secret.type = siBuffer; |
| 72 secret.len = data_len; | 71 secret.len = data_len; |
| 73 secret.data = const_cast<unsigned char*>(data); | 72 secret.data = const_cast<unsigned char*>(data); |
| 74 | 73 |
| 75 // |arena| is used to encode the cert. | 74 // SECItem to receive the output buffer. |
| 76 crypto::ScopedPLArenaPool arena(PORT_NewArena(DER_DEFAULT_CHUNKSIZE)); | 75 SECItem result; |
| 77 CHECK(arena.get() != NULL); | 76 result.type = siBuffer; |
| 78 | 77 result.len = 0; |
| 79 // Allocate space to contain the signed data. | 78 result.data = NULL; |
| 80 SECItem* result = SECITEM_AllocItem(arena.get(), NULL, 0); | |
| 81 if (!result) { | |
| 82 DLOG(ERROR) << "Unable to allocate space for signed data."; | |
| 83 return false; | |
| 84 } | |
| 85 | 79 |
| 86 // Sign the secret data and save it to |result|. | 80 // Sign the secret data and save it to |result|. |
| 87 SECStatus rv = | 81 SECStatus rv = |
| 88 SignData(arena.get(), result, &secret, key_->key(), HASH_AlgSHA1); | 82 SignData(&result, &secret, key_->key(), HASH_AlgSHA1); |
| 89 if (rv != SECSuccess) { | 83 if (rv != SECSuccess) { |
| 90 DLOG(ERROR) << "DerSignData: " << PORT_GetError(); | 84 DLOG(ERROR) << "DerSignData: " << PORT_GetError(); |
| 91 return false; | 85 return false; |
| 92 } | 86 } |
| 93 | 87 |
| 94 // Copy the signed data into the output vector. | 88 // Copy the signed data into the output vector. |
| 95 signature->assign(result->data, result->data + result->len); | 89 signature->assign(result.data, result.data + result.len); |
| 90 SECITEM_FreeItem(&result, PR_FALSE /* only free |result.data| */); | |
|
wtc
2012/01/31 18:43:57
You can also just do
PORT_Free(result.data);
to
| |
| 96 return true; | 91 return true; |
| 97 } | 92 } |
| 98 | 93 |
| 99 } // namespace crypto | 94 } // namespace crypto |
| OLD | NEW |