| // Copyright (c) 2010 The Chromium Authors. All rights reserved. |
| // Use of this source code is governed by a BSD-style license that can be |
| // found in the LICENSE file. |
| |
| #ifndef BASE_CRYPTO_SYMMETRIC_KEY_H_ |
| #define BASE_CRYPTO_SYMMETRIC_KEY_H_ |
| #pragma once |
| |
| #include <string> |
| |
| #include "base/basictypes.h" |
| |
| #if defined(USE_NSS) |
| #include "base/crypto/scoped_nss_types.h" |
| #elif defined(OS_MACOSX) |
| #include <Security/cssmtype.h> |
| #elif defined(OS_WIN) |
| #include "base/crypto/scoped_capi_types.h" |
| #endif |
| |
| namespace base { |
| |
| // Wraps a platform-specific symmetric key and allows it to be held in a |
| // scoped_ptr. |
| class SymmetricKey { |
| public: |
| enum Algorithm { |
| AES, |
| HMAC_SHA1, |
| }; |
| |
| virtual ~SymmetricKey(); |
| |
| // Generates a random key suitable to be used with |cipher| and of |
| // |key_size_in_bits| bits. |
| // The caller is responsible for deleting the returned SymmetricKey. |
| static SymmetricKey* GenerateRandomKey(Algorithm algorithm, |
| size_t key_size_in_bits); |
| |
| // Derives a key from the supplied password and salt using PBKDF2. The caller |
| // is responsible for deleting the returned SymmetricKey. |
| static SymmetricKey* DeriveKeyFromPassword(Algorithm algorithm, |
| const std::string& password, |
| const std::string& salt, |
| size_t iterations, |
| size_t key_size_in_bits); |
| |
| // Imports a raw key. For this call to be successful, |raw_key| must have been |
| // generated by either GenerateRandomKey or DeriveKeyFromPassword, and |
| // must have been exported with GetRawKey. The caller owns the returned |
| // SymmetricKey. |
| static SymmetricKey* Import(Algorithm algorithm, const std::string& raw_key); |
| |
| #if defined(USE_NSS) |
| PK11SymKey* key() const { return key_.get(); } |
| #elif defined(OS_MACOSX) |
| CSSM_DATA cssm_data() const; |
| #elif defined(OS_WIN) |
| HCRYPTKEY key() const { return key_.get(); } |
| #endif |
| |
| // Extracts the raw key from the platform specific data. |
| // Warning: |raw_key| holds the raw key as bytes and thus must be handled |
| // carefully. |
| bool GetRawKey(std::string* raw_key); |
| |
| private: |
| #if defined(USE_OPENSSL) |
| // TODO(joth): Add a constructor that accepts OpenSSL symmetric key data, and |
| // the appropriate data members to store it in. |
| #elif defined(USE_NSS) |
| explicit SymmetricKey(PK11SymKey* key); |
| ScopedPK11SymKey key_; |
| #elif defined(OS_MACOSX) |
| SymmetricKey(const void* key_data, size_t key_size_in_bits); |
| std::string key_; |
| #elif defined(OS_WIN) |
| SymmetricKey(HCRYPTPROV provider, HCRYPTKEY key, |
| const void* key_data, size_t key_size_in_bytes); |
| |
| ScopedHCRYPTPROV provider_; |
| ScopedHCRYPTKEY key_; |
| |
| // Contains the raw key, if it is known during initialization and when it |
| // is likely that the associated |provider_| will be unable to export the |
| // |key_|. This is the case of HMAC keys when the key size exceeds 16 bytes |
| // when using the default RSA provider. |
| // TODO(rsleevi): See if KP_EFFECTIVE_KEYLEN is the reason why CryptExportKey |
| // fails with NTE_BAD_KEY/NTE_BAD_LEN |
| std::string raw_key_; |
| #endif |
| |
| DISALLOW_COPY_AND_ASSIGN(SymmetricKey); |
| }; |
| |
| } // namespace base |
| |
| #endif // BASE_CRYPTO_SYMMETRIC_KEY_H_ |