wtc@chromium.org
bdc9256b3d
method. Overload the Init method for char* and std::string. Add DCHECKs to the destruction methods in ~HMAC in hmac_win.cc. The patch is written by Takeshi Yoshino <tyoshino@google.com>. Original code review: http://codereview.chromium.org/88062 R=wtc http://crbug.com/2297 TEST=base_unittests should pass. Safe browsing should continue to work. Review URL: http://codereview.chromium.org/113001 git-svn-id: svn://svn.chromium.org/chrome/trunk/src@15353 0039d316-1c4b-4281-b951-d872f2087c98
70 linhas
1.5 KiB
C++
70 linhas
1.5 KiB
C++
// Copyright (c) 2008 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.
|
|
|
|
#include "base/hmac.h"
|
|
|
|
#include <CommonCrypto/CommonHMAC.h>
|
|
|
|
#include "base/logging.h"
|
|
|
|
namespace base {
|
|
|
|
struct HMACPlatformData {
|
|
std::string key_;
|
|
};
|
|
|
|
HMAC::HMAC(HashAlgorithm hash_alg)
|
|
: hash_alg_(hash_alg), plat_(new HMACPlatformData()) {
|
|
// Only SHA-1 digest is supported now.
|
|
DCHECK(hash_alg_ == SHA1);
|
|
}
|
|
|
|
bool HMAC::Init(const unsigned char *key, int key_length) {
|
|
if (!plat_->key_.empty()) {
|
|
// Init must not be called more than once on the same HMAC object.
|
|
NOTREACHED();
|
|
return false;
|
|
}
|
|
|
|
plat_->key_.assign(reinterpret_cast<const char*>(key), key_length);
|
|
|
|
return true;
|
|
}
|
|
|
|
HMAC::~HMAC() {
|
|
// Zero out key copy.
|
|
plat_->key_.assign(plat_->key_.length(), std::string::value_type());
|
|
plat_->key_.clear();
|
|
plat_->key_.reserve(0);
|
|
}
|
|
|
|
bool HMAC::Sign(const std::string& data,
|
|
unsigned char* digest,
|
|
int digest_length) {
|
|
CCHmacAlgorithm algorithm;
|
|
int algorithm_digest_length;
|
|
switch (hash_alg_) {
|
|
case SHA1:
|
|
algorithm = kCCHmacAlgSHA1;
|
|
algorithm_digest_length = CC_SHA1_DIGEST_LENGTH;
|
|
break;
|
|
default:
|
|
NOTREACHED();
|
|
return false;
|
|
}
|
|
|
|
if (digest_length < algorithm_digest_length) {
|
|
NOTREACHED();
|
|
return false;
|
|
}
|
|
|
|
CCHmac(algorithm,
|
|
plat_->key_.data(), plat_->key_.length(), data.data(), data.length(),
|
|
digest);
|
|
|
|
return true;
|
|
}
|
|
|
|
} // namespace base
|