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 |