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.
7 #include "base/memory/scoped_ptr.h"
9 #include "crypto/rsa_private_key.h"
10 #include "crypto/signature_creator.h"
11 #include "crypto/signature_verifier.h"
12 #include "testing/gtest/include/gtest/gtest.h"
16 // This is the algorithm ID for SHA-1 with RSA encryption.
17 const uint8 kSHA1WithRSAAlgorithmID
[] = {
18 0x30, 0x0d, 0x06, 0x09, 0x2a, 0x86, 0x48, 0x86,
19 0xf7, 0x0d, 0x01, 0x01, 0x05, 0x05, 0x00
24 TEST(SignatureCreatorTest
, BasicTest
) {
25 // Do a verify round trip.
26 scoped_ptr
<crypto::RSAPrivateKey
> key_original(
27 crypto::RSAPrivateKey::Create(1024));
28 ASSERT_TRUE(key_original
.get());
30 std::vector
<uint8
> key_info
;
31 key_original
->ExportPrivateKey(&key_info
);
32 scoped_ptr
<crypto::RSAPrivateKey
> key(
33 crypto::RSAPrivateKey::CreateFromPrivateKeyInfo(key_info
));
34 ASSERT_TRUE(key
.get());
36 scoped_ptr
<crypto::SignatureCreator
> signer(
37 crypto::SignatureCreator::Create(key
.get()));
38 ASSERT_TRUE(signer
.get());
40 std::string
data("Hello, World!");
41 ASSERT_TRUE(signer
->Update(reinterpret_cast<const uint8
*>(data
.c_str()),
44 std::vector
<uint8
> signature
;
45 ASSERT_TRUE(signer
->Final(&signature
));
47 std::vector
<uint8
> public_key_info
;
48 ASSERT_TRUE(key_original
->ExportPublicKey(&public_key_info
));
50 crypto::SignatureVerifier verifier
;
51 ASSERT_TRUE(verifier
.VerifyInit(
52 kSHA1WithRSAAlgorithmID
, sizeof(kSHA1WithRSAAlgorithmID
),
53 &signature
.front(), signature
.size(),
54 &public_key_info
.front(), public_key_info
.size()));
56 verifier
.VerifyUpdate(reinterpret_cast<const uint8
*>(data
.c_str()),
58 ASSERT_TRUE(verifier
.VerifyFinal());
61 TEST(SignatureCreatorTest
, SignDigestTest
) {
62 // Do a verify round trip.
63 scoped_ptr
<crypto::RSAPrivateKey
> key_original(
64 crypto::RSAPrivateKey::Create(1024));
65 ASSERT_TRUE(key_original
.get());
67 std::vector
<uint8
> key_info
;
68 key_original
->ExportPrivateKey(&key_info
);
69 scoped_ptr
<crypto::RSAPrivateKey
> key(
70 crypto::RSAPrivateKey::CreateFromPrivateKeyInfo(key_info
));
71 ASSERT_TRUE(key
.get());
73 std::string
data("Hello, World!");
74 std::string sha1
= base::SHA1HashString(data
);
75 // Sign sha1 of the input data.
76 std::vector
<uint8
> signature
;
77 ASSERT_TRUE(crypto::SignatureCreator::Sign(
79 reinterpret_cast<const uint8
*>(sha1
.c_str()),
83 std::vector
<uint8
> public_key_info
;
84 ASSERT_TRUE(key_original
->ExportPublicKey(&public_key_info
));
86 // Verify the input data.
87 crypto::SignatureVerifier verifier
;
88 ASSERT_TRUE(verifier
.VerifyInit(
89 kSHA1WithRSAAlgorithmID
, sizeof(kSHA1WithRSAAlgorithmID
),
90 &signature
.front(), signature
.size(),
91 &public_key_info
.front(), public_key_info
.size()));
93 verifier
.VerifyUpdate(reinterpret_cast<const uint8
*>(data
.c_str()),
95 ASSERT_TRUE(verifier
.VerifyFinal());