81 lines
2.8 KiB
C++
81 lines
2.8 KiB
C++
// Copyright 2014 Google Inc. All rights reserved.
|
|
//
|
|
// Use of this source code is governed by a BSD-style
|
|
// license that can be found in the LICENSE file or at
|
|
// https://developers.google.com/open-source/licenses/bsd
|
|
|
|
#include "packager/media/base/request_signer.h"
|
|
|
|
#include "packager/base/logging.h"
|
|
#include "packager/base/sha1.h"
|
|
#include "packager/base/strings/string_number_conversions.h"
|
|
#include "packager/media/base/aes_encryptor.h"
|
|
#include "packager/media/base/rsa_key.h"
|
|
|
|
namespace shaka {
|
|
namespace media {
|
|
|
|
RequestSigner::RequestSigner(const std::string& signer_name)
|
|
: signer_name_(signer_name) {}
|
|
RequestSigner::~RequestSigner() {}
|
|
|
|
AesRequestSigner::AesRequestSigner(const std::string& signer_name,
|
|
std::unique_ptr<AesCbcEncryptor> encryptor)
|
|
: RequestSigner(signer_name), aes_cbc_encryptor_(std::move(encryptor)) {
|
|
DCHECK(aes_cbc_encryptor_);
|
|
}
|
|
AesRequestSigner::~AesRequestSigner() {}
|
|
|
|
AesRequestSigner* AesRequestSigner::CreateSigner(const std::string& signer_name,
|
|
const std::string& aes_key_hex,
|
|
const std::string& iv_hex) {
|
|
std::vector<uint8_t> aes_key;
|
|
if (!base::HexStringToBytes(aes_key_hex, &aes_key)) {
|
|
LOG(ERROR) << "Failed to convert hex string to bytes: " << aes_key_hex;
|
|
return NULL;
|
|
}
|
|
std::vector<uint8_t> iv;
|
|
if (!base::HexStringToBytes(iv_hex, &iv)) {
|
|
LOG(ERROR) << "Failed to convert hex string to bytes: " << iv_hex;
|
|
return NULL;
|
|
}
|
|
|
|
std::unique_ptr<AesCbcEncryptor> encryptor(
|
|
new AesCbcEncryptor(kPkcs5Padding, AesCryptor::kUseConstantIv));
|
|
if (!encryptor->InitializeWithIv(aes_key, iv))
|
|
return NULL;
|
|
return new AesRequestSigner(signer_name, std::move(encryptor));
|
|
}
|
|
|
|
bool AesRequestSigner::GenerateSignature(const std::string& message,
|
|
std::string* signature) {
|
|
aes_cbc_encryptor_->Crypt(base::SHA1HashString(message), signature);
|
|
return true;
|
|
}
|
|
|
|
RsaRequestSigner::RsaRequestSigner(
|
|
const std::string& signer_name,
|
|
std::unique_ptr<RsaPrivateKey> rsa_private_key)
|
|
: RequestSigner(signer_name), rsa_private_key_(std::move(rsa_private_key)) {
|
|
DCHECK(rsa_private_key_);
|
|
}
|
|
RsaRequestSigner::~RsaRequestSigner() {}
|
|
|
|
RsaRequestSigner* RsaRequestSigner::CreateSigner(
|
|
const std::string& signer_name,
|
|
const std::string& pkcs1_rsa_key) {
|
|
std::unique_ptr<RsaPrivateKey> rsa_private_key(
|
|
RsaPrivateKey::Create(pkcs1_rsa_key));
|
|
if (!rsa_private_key)
|
|
return NULL;
|
|
return new RsaRequestSigner(signer_name, std::move(rsa_private_key));
|
|
}
|
|
|
|
bool RsaRequestSigner::GenerateSignature(const std::string& message,
|
|
std::string* signature) {
|
|
return rsa_private_key_->GenerateSignature(message, signature);
|
|
}
|
|
|
|
} // namespace media
|
|
} // namespace shaka
|