1 // Copyright (c) 2011 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/signature_creator.h"
6
7 #include "base/logging.h"
8 #include "base/memory/scoped_ptr.h"
9
10 namespace crypto {
11
12 // static
Create(RSAPrivateKey * key)13 SignatureCreator* SignatureCreator::Create(RSAPrivateKey* key) {
14 scoped_ptr<SignatureCreator> result(new SignatureCreator);
15 result->key_ = key;
16
17 if (!CryptCreateHash(key->provider(), CALG_SHA1, 0, 0,
18 result->hash_object_.receive())) {
19 NOTREACHED();
20 return NULL;
21 }
22
23 return result.release();
24 }
25
SignatureCreator()26 SignatureCreator::SignatureCreator() : hash_object_(0) {}
27
~SignatureCreator()28 SignatureCreator::~SignatureCreator() {}
29
Update(const uint8 * data_part,int data_part_len)30 bool SignatureCreator::Update(const uint8* data_part, int data_part_len) {
31 if (!CryptHashData(hash_object_, data_part, data_part_len, 0)) {
32 NOTREACHED();
33 return false;
34 }
35
36 return true;
37 }
38
Final(std::vector<uint8> * signature)39 bool SignatureCreator::Final(std::vector<uint8>* signature) {
40 DWORD signature_length = 0;
41 if (!CryptSignHash(hash_object_, AT_SIGNATURE, NULL, 0, NULL,
42 &signature_length)) {
43 return false;
44 }
45
46 std::vector<uint8> temp;
47 temp.resize(signature_length);
48 if (!CryptSignHash(hash_object_, AT_SIGNATURE, NULL, 0, &temp.front(),
49 &signature_length)) {
50 return false;
51 }
52
53 temp.resize(signature_length);
54 for (size_t i = temp.size(); i > 0; --i)
55 signature->push_back(temp[i - 1]);
56
57 return true;
58 }
59
60 } // namespace crypto
61