diff options
Diffstat (limited to 'third_party/libwebrtc/webrtc/rtc_base/openssldigest.cc')
-rw-r--r-- | third_party/libwebrtc/webrtc/rtc_base/openssldigest.cc | 117 |
1 files changed, 117 insertions, 0 deletions
diff --git a/third_party/libwebrtc/webrtc/rtc_base/openssldigest.cc b/third_party/libwebrtc/webrtc/rtc_base/openssldigest.cc new file mode 100644 index 0000000000..3be82e61d3 --- /dev/null +++ b/third_party/libwebrtc/webrtc/rtc_base/openssldigest.cc @@ -0,0 +1,117 @@ +/* + * 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/openssldigest.h" + +#include "rtc_base/checks.h" +#include "rtc_base/openssl.h" + +namespace rtc { + +OpenSSLDigest::OpenSSLDigest(const std::string& algorithm) { + EVP_MD_CTX_init(&ctx_); + if (GetDigestEVP(algorithm, &md_)) { + EVP_DigestInit_ex(&ctx_, md_, nullptr); + } else { + md_ = nullptr; + } +} + +OpenSSLDigest::~OpenSSLDigest() { + EVP_MD_CTX_cleanup(&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 |