Chromium Code Reviews| OLD | NEW |
|---|---|
| 1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. | 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 | 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 // Utility class for calculating the HMAC for a given message. We currently | 5 // Utility class for calculating the HMAC for a given message. We currently |
| 6 // only support SHA1 for the hash algorithm, but this can be extended easily. | 6 // only support SHA1 for the hash algorithm, but this can be extended easily. |
| 7 | 7 |
| 8 #ifndef CRYPTO_HMAC_H_ | 8 #ifndef CRYPTO_HMAC_H_ |
| 9 #define CRYPTO_HMAC_H_ | 9 #define CRYPTO_HMAC_H_ |
| 10 #pragma once | 10 #pragma once |
| 11 | 11 |
| 12 #include <string> | 12 #include <string> |
| 13 | 13 |
| 14 #include "base/basictypes.h" | 14 #include "base/basictypes.h" |
| 15 #include "base/memory/scoped_ptr.h" | 15 #include "base/memory/scoped_ptr.h" |
| 16 #include "base/string_piece.h" | |
| 16 | 17 |
| 17 namespace crypto { | 18 namespace crypto { |
| 18 | 19 |
| 19 // Simplify the interface and reduce includes by abstracting out the internals. | 20 // Simplify the interface and reduce includes by abstracting out the internals. |
| 20 struct HMACPlatformData; | 21 struct HMACPlatformData; |
| 21 | 22 |
| 22 class HMAC { | 23 class HMAC { |
| 23 public: | 24 public: |
| 24 // The set of supported hash functions. Extend as required. | 25 // The set of supported hash functions. Extend as required. |
| 25 enum HashAlgorithm { | 26 enum HashAlgorithm { |
| 26 SHA1, | 27 SHA1, |
| 27 SHA256, | 28 SHA256, |
| 28 }; | 29 }; |
| 29 | 30 |
| 30 explicit HMAC(HashAlgorithm hash_alg); | 31 explicit HMAC(HashAlgorithm hash_alg); |
| 31 ~HMAC(); | 32 ~HMAC(); |
| 32 | 33 |
| 33 // Returns the length of digest that this HMAC will create. | 34 // Returns the length of digest that this HMAC will create. |
| 34 size_t DigestLength() const; | 35 size_t DigestLength() const; |
| 35 | 36 |
| 36 // TODO(abarth): Add a PreferredKeyLength() member function. | 37 // TODO(abarth): Add a PreferredKeyLength() member function. |
| 37 | 38 |
| 38 // Initializes this instance using |key| of the length |key_length|. Call Init | 39 // Initializes this instance using |key| of the length |key_length|. Call Init |
| 39 // only once. It returns false on the second or later calls. | 40 // only once. It returns false on the second or later calls. |
| 40 // TODO(abarth): key_length should be a size_t. | 41 // TODO(abarth): key_length should be a size_t. |
| 41 bool Init(const unsigned char* key, int key_length); | 42 bool Init(const unsigned char* key, int key_length); |
| 42 | 43 |
| 43 // Initializes this instance using |key|. Call Init only once. It returns | 44 // Initializes this instance using |key|. Call Init only once. It returns |
| 44 // false on the second or later calls. | 45 // false on the second or later calls. |
| 45 bool Init(const std::string& key) { | 46 bool Init(const std::string& key) { |
|
wtc
2011/06/02 19:56:48
Can you also change 'key' to a StringPiece, and re
| |
| 46 return Init(reinterpret_cast<const unsigned char*>(key.data()), | 47 return Init(reinterpret_cast<const unsigned char*>(key.data()), |
| 47 static_cast<int>(key.size())); | 48 static_cast<int>(key.size())); |
| 48 } | 49 } |
| 49 | 50 |
| 50 // Calculates the HMAC for the message in |data| using the algorithm supplied | 51 // Calculates the HMAC for the message in |data| using the algorithm supplied |
| 51 // to the constructor and the key supplied to the Init method. The HMAC is | 52 // to the constructor and the key supplied to the Init method. The HMAC is |
| 52 // returned in |digest|, which has |digest_length| bytes of storage available. | 53 // returned in |digest|, which has |digest_length| bytes of storage available. |
| 53 // TODO(abarth): digest_length should be a size_t. | 54 // TODO(abarth): digest_length should be a size_t. |
| 54 bool Sign(const std::string& data, | 55 bool Sign(const base::StringPiece& data, unsigned char* digest, |
| 55 unsigned char* digest, | |
| 56 int digest_length) const; | 56 int digest_length) const; |
| 57 | 57 |
| 58 // TODO(albertb): Add a Verify method. | 58 // TODO(albertb): Add a Verify method. |
| 59 | 59 |
| 60 private: | 60 private: |
| 61 HashAlgorithm hash_alg_; | 61 HashAlgorithm hash_alg_; |
| 62 scoped_ptr<HMACPlatformData> plat_; | 62 scoped_ptr<HMACPlatformData> plat_; |
| 63 | 63 |
| 64 DISALLOW_COPY_AND_ASSIGN(HMAC); | 64 DISALLOW_COPY_AND_ASSIGN(HMAC); |
| 65 }; | 65 }; |
| 66 | 66 |
| 67 } // namespace crypto | 67 } // namespace crypto |
| 68 | 68 |
| 69 #endif // CRYPTO_HMAC_H_ | 69 #endif // CRYPTO_HMAC_H_ |
| OLD | NEW |