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.
5 #include "crypto/symmetric_key.h"
10 #include "base/logging.h"
11 #include "crypto/nss_util.h"
15 SymmetricKey::~SymmetricKey() {}
18 SymmetricKey
* SymmetricKey::GenerateRandomKey(Algorithm algorithm
,
19 size_t key_size_in_bits
) {
20 DCHECK_EQ(AES
, algorithm
);
23 if (key_size_in_bits
== 0)
26 ScopedPK11Slot
slot(PK11_GetBestSlot(CKM_AES_KEY_GEN
, NULL
));
30 PK11SymKey
* sym_key
= PK11_KeyGen(slot
.get(), CKM_AES_KEY_GEN
, NULL
,
31 key_size_in_bits
/ 8, NULL
);
35 return new SymmetricKey(sym_key
);
39 SymmetricKey
* SymmetricKey::DeriveKeyFromPassword(Algorithm algorithm
,
40 const std::string
& password
,
41 const std::string
& salt
,
43 size_t key_size_in_bits
) {
45 if (salt
.empty() || iterations
== 0 || key_size_in_bits
== 0)
48 SECItem password_item
;
49 password_item
.type
= siBuffer
;
50 password_item
.data
= reinterpret_cast<unsigned char*>(
51 const_cast<char *>(password
.data()));
52 password_item
.len
= password
.size();
55 salt_item
.type
= siBuffer
;
56 salt_item
.data
= reinterpret_cast<unsigned char*>(
57 const_cast<char *>(salt
.data()));
58 salt_item
.len
= salt
.size();
60 SECOidTag cipher_algorithm
=
61 algorithm
== AES
? SEC_OID_AES_256_CBC
: SEC_OID_HMAC_SHA1
;
62 ScopedSECAlgorithmID
alg_id(PK11_CreatePBEV2AlgorithmID(SEC_OID_PKCS5_PBKDF2
,
71 ScopedPK11Slot
slot(PK11_GetBestSlot(SEC_OID_PKCS5_PBKDF2
, NULL
));
75 PK11SymKey
* sym_key
= PK11_PBEKeyGen(slot
.get(), alg_id
.get(), &password_item
,
80 return new SymmetricKey(sym_key
);
84 SymmetricKey
* SymmetricKey::Import(Algorithm algorithm
,
85 const std::string
& raw_key
) {
87 CK_MECHANISM_TYPE cipher
=
88 algorithm
== AES
? CKM_AES_CBC
: CKM_SHA_1_HMAC
;
91 key_item
.type
= siBuffer
;
92 key_item
.data
= reinterpret_cast<unsigned char*>(
93 const_cast<char *>(raw_key
.data()));
94 key_item
.len
= raw_key
.size();
96 ScopedPK11Slot
slot(PK11_GetBestSlot(cipher
, NULL
));
100 // The exact value of the |origin| argument doesn't matter to NSS as long as
101 // it's not PK11_OriginFortezzaHack, so we pass PK11_OriginUnwrap as a
103 PK11SymKey
* sym_key
= PK11_ImportSymKey(slot
.get(), cipher
, PK11_OriginUnwrap
,
104 CKA_ENCRYPT
, &key_item
, NULL
);
108 return new SymmetricKey(sym_key
);
111 bool SymmetricKey::GetRawKey(std::string
* raw_key
) {
112 SECStatus rv
= PK11_ExtractKeyValue(key_
.get());
113 if (SECSuccess
!= rv
)
116 SECItem
* key_item
= PK11_GetKeyData(key_
.get());
120 raw_key
->assign(reinterpret_cast<char*>(key_item
->data
), key_item
->len
);
124 #if defined(OS_CHROMEOS)
126 SymmetricKey
* SymmetricKey::CreateFromKey(PK11SymKey
* key
) {
127 return new SymmetricKey(key
);
131 SymmetricKey::SymmetricKey(PK11SymKey
* key
) : key_(key
) {
135 } // namespace crypto