mirror of
https://github.com/mollyim/webrtc.git
synced 2025-05-13 05:40:42 +01:00
7fa9057a05
This is part of a large-scale effort to increase adoption of absl::string_view across the WebRTC code base. This CL converts the majority of "const std::string&"s in function parameters under rtc_base/ to absl::string_view. Bug: webrtc:13579 Change-Id: I2b1e3776aa42326aa405f76bb324a2d233b21dca Reviewed-on: https://webrtc-review.googlesource.com/c/src/+/254081 Reviewed-by: Niels Moller <nisse@webrtc.org> Reviewed-by: Mirko Bonadei <mbonadei@webrtc.org> Reviewed-by: Xavier Lepaul <xalep@webrtc.org> Reviewed-by: Anders Lilienthal <andersc@webrtc.org> Reviewed-by: Per Kjellander <perkj@webrtc.org> Commit-Queue: Ali Tofigh <alito@webrtc.org> Cr-Commit-Position: refs/heads/main@{#36239}
118 lines
2.9 KiB
C++
118 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 "absl/strings/string_view.h"
|
|
#include "rtc_base/checks.h" // RTC_DCHECK, RTC_CHECK
|
|
#include "rtc_base/openssl.h"
|
|
|
|
namespace rtc {
|
|
|
|
OpenSSLDigest::OpenSSLDigest(absl::string_view 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(absl::string_view 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(absl::string_view algorithm, size_t* length) {
|
|
const EVP_MD* md;
|
|
if (!GetDigestEVP(algorithm, &md))
|
|
return false;
|
|
|
|
*length = EVP_MD_size(md);
|
|
return true;
|
|
}
|
|
|
|
} // namespace rtc
|