119 lines
2.9 KiB
C++
119 lines
2.9 KiB
C++
|
/*
|
||
|
|
* Copyright 2004 The WebRTC Project Authors. All rights reserved.
|
||
|
|
*
|
||
|
|
* Use of this source code is governed by a BSD-style license
|
||
|
|
* that can be found in the LICENSE file in the root of the source
|
||
|
|
* tree. An additional intellectual property rights grant can be found
|
||
|
|
* in the file PATENTS. All contributing project authors may
|
||
|
|
* be found in the AUTHORS file in the root of the source tree.
|
||
|
|
*/
|
||
|
|
|
||
|
|
#include "rtc_base/openssl_digest.h"
|
||
|
|
|
||
|
|
#include "rtc_base/checks.h" // RTC_DCHECK, RTC_CHECK
|
||
|
|
#include "rtc_base/openssl.h"
|
||
|
|
|
||
|
|
namespace rtc {
|
||
|
|
|
||
|
|
OpenSSLDigest::OpenSSLDigest(const std::string& algorithm) {
|
||
|
|
ctx_ = EVP_MD_CTX_new();
|
||
|
|
RTC_CHECK(ctx_ != nullptr);
|
||
|
|
EVP_MD_CTX_init(ctx_);
|
||
|
|
if (GetDigestEVP(algorithm, &md_)) {
|
||
|
|
EVP_DigestInit_ex(ctx_, md_, nullptr);
|
||
|
|
} else {
|
||
|
|
md_ = nullptr;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
OpenSSLDigest::~OpenSSLDigest() {
|
||
|
|
EVP_MD_CTX_destroy(ctx_);
|
||
|
|
}
|
||
|
|
|
||
|
|
size_t OpenSSLDigest::Size() const {
|
||
|
|
if (!md_) {
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
return EVP_MD_size(md_);
|
||
|
|
}
|
||
|
|
|
||
|
|
void OpenSSLDigest::Update(const void* buf, size_t len) {
|
||
|
|
if (!md_) {
|
||
|
|
return;
|
||
|
|
}
|
||
|
|
EVP_DigestUpdate(ctx_, buf, len);
|
||
|
|
}
|
||
|
|
|
||
|
|
size_t OpenSSLDigest::Finish(void* buf, size_t len) {
|
||
|
|
if (!md_ || len < Size()) {
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
unsigned int md_len;
|
||
|
|
EVP_DigestFinal_ex(ctx_, static_cast<unsigned char*>(buf), &md_len);
|
||
|
|
EVP_DigestInit_ex(ctx_, md_, nullptr); // prepare for future Update()s
|
||
|
|
RTC_DCHECK(md_len == Size());
|
||
|
|
return md_len;
|
||
|
|
}
|
||
|
|
|
||
|
|
bool OpenSSLDigest::GetDigestEVP(const std::string& algorithm,
|
||
|
|
const EVP_MD** mdp) {
|
||
|
|
const EVP_MD* md;
|
||
|
|
if (algorithm == DIGEST_MD5) {
|
||
|
|
md = EVP_md5();
|
||
|
|
} else if (algorithm == DIGEST_SHA_1) {
|
||
|
|
md = EVP_sha1();
|
||
|
|
} else if (algorithm == DIGEST_SHA_224) {
|
||
|
|
md = EVP_sha224();
|
||
|
|
} else if (algorithm == DIGEST_SHA_256) {
|
||
|
|
md = EVP_sha256();
|
||
|
|
} else if (algorithm == DIGEST_SHA_384) {
|
||
|
|
md = EVP_sha384();
|
||
|
|
} else if (algorithm == DIGEST_SHA_512) {
|
||
|
|
md = EVP_sha512();
|
||
|
|
} else {
|
||
|
|
return false;
|
||
|
|
}
|
||
|
|
|
||
|
|
// Can't happen
|
||
|
|
RTC_DCHECK(EVP_MD_size(md) >= 16);
|
||
|
|
*mdp = md;
|
||
|
|
return true;
|
||
|
|
}
|
||
|
|
|
||
|
|
bool OpenSSLDigest::GetDigestName(const EVP_MD* md, std::string* algorithm) {
|
||
|
|
RTC_DCHECK(md != nullptr);
|
||
|
|
RTC_DCHECK(algorithm != nullptr);
|
||
|
|
|
||
|
|
int md_type = EVP_MD_type(md);
|
||
|
|
if (md_type == NID_md5) {
|
||
|
|
*algorithm = DIGEST_MD5;
|
||
|
|
} else if (md_type == NID_sha1) {
|
||
|
|
*algorithm = DIGEST_SHA_1;
|
||
|
|
} else if (md_type == NID_sha224) {
|
||
|
|
*algorithm = DIGEST_SHA_224;
|
||
|
|
} else if (md_type == NID_sha256) {
|
||
|
|
*algorithm = DIGEST_SHA_256;
|
||
|
|
} else if (md_type == NID_sha384) {
|
||
|
|
*algorithm = DIGEST_SHA_384;
|
||
|
|
} else if (md_type == NID_sha512) {
|
||
|
|
*algorithm = DIGEST_SHA_512;
|
||
|
|
} else {
|
||
|
|
algorithm->clear();
|
||
|
|
return false;
|
||
|
|
}
|
||
|
|
|
||
|
|
return true;
|
||
|
|
}
|
||
|
|
|
||
|
|
bool OpenSSLDigest::GetDigestSize(const std::string& algorithm,
|
||
|
|
size_t* length) {
|
||
|
|
const EVP_MD* md;
|
||
|
|
if (!GetDigestEVP(algorithm, &md))
|
||
|
|
return false;
|
||
|
|
|
||
|
|
*length = EVP_MD_size(md);
|
||
|
|
return true;
|
||
|
|
}
|
||
|
|
|
||
|
|
} // namespace rtc
|