82 lines
2.9 KiB
C++
82 lines
2.9 KiB
C++
// Copyright 2014 Google Inc. All Rights Reserved.
|
|
|
|
#ifndef CORE_INCLUDE_INITIALIZATION_DATA_H_
|
|
#define CORE_INCLUDE_INITIALIZATION_DATA_H_
|
|
|
|
#include <string>
|
|
|
|
#include "license_protocol.pb.h"
|
|
#include "wv_cdm_types.h"
|
|
|
|
namespace wvcdm {
|
|
|
|
class WvCdmEngineTest;
|
|
|
|
class InitializationData {
|
|
public:
|
|
InitializationData(const std::string& type = std::string(),
|
|
const CdmInitData& data = CdmInitData());
|
|
|
|
bool is_supported() const { return is_cenc_ || is_webm_ || is_hls_; }
|
|
bool is_cenc() const { return is_cenc_; }
|
|
bool is_hls() const { return is_hls_; }
|
|
bool is_webm() const { return is_webm_; }
|
|
|
|
bool IsEmpty() const { return data_.empty(); }
|
|
|
|
const std::string& type() const { return type_; }
|
|
const CdmInitData& data() const { return data_; }
|
|
std::vector<uint8_t> hls_iv() const { return hls_iv_; }
|
|
CdmHlsMethod hls_method() const { return hls_method_; }
|
|
std::vector<video_widevine::SubLicense> ExtractEmbeddedKeys() const;
|
|
|
|
private:
|
|
// Parse a blob of multiple concatenated PSSH atoms to extract the first
|
|
// Widevine PSSH.
|
|
bool ExtractWidevinePssh(const CdmInitData& init_data, CdmInitData* output);
|
|
|
|
bool ExtractHlsAttributes(const std::string& attribute_list,
|
|
CdmHlsMethod* method, std::vector<uint8_t>* iv,
|
|
std::string* uri);
|
|
static bool ConstructWidevineInitData(CdmHlsMethod method,
|
|
const std::string& uri,
|
|
CdmInitData* output);
|
|
static bool ExtractQuotedAttribute(const std::string& attribute_list,
|
|
const std::string& key,
|
|
std::string* value);
|
|
static bool ExtractHexAttribute(const std::string& attribute_list,
|
|
const std::string& key,
|
|
std::vector<uint8_t>* value);
|
|
static bool ExtractAttribute(const std::string& attribute_list,
|
|
const std::string& key, std::string* value);
|
|
|
|
static std::vector<std::string> ExtractKeyFormatVersions(
|
|
const std::string& key_format_versions);
|
|
|
|
// For testing only:
|
|
#if defined(UNIT_TEST)
|
|
FRIEND_TEST(HlsAttributeExtractionTest, ExtractAttribute);
|
|
FRIEND_TEST(HlsConstructionTest, InitData);
|
|
FRIEND_TEST(HlsInitDataConstructionTest, InvalidUriDataFormat);
|
|
FRIEND_TEST(HlsInitDataConstructionTest, InvalidUriBase64Encode);
|
|
FRIEND_TEST(HlsHexAttributeExtractionTest, ExtractHexAttribute);
|
|
FRIEND_TEST(HlsKeyFormatVersionsExtractionTest, ExtractKeyFormatVersions);
|
|
FRIEND_TEST(HlsParseTest, Parse);
|
|
FRIEND_TEST(HlsQuotedAttributeExtractionTest, ExtractQuotedAttribute);
|
|
FRIEND_TEST(HlsTest, ExtractHlsAttributes);
|
|
#endif
|
|
|
|
std::string type_;
|
|
CdmInitData data_;
|
|
bool is_cenc_;
|
|
bool is_hls_;
|
|
bool is_webm_;
|
|
|
|
std::vector<uint8_t> hls_iv_;
|
|
CdmHlsMethod hls_method_;
|
|
};
|
|
|
|
} // namespace wvcdm
|
|
|
|
#endif // CORE_INCLUDE_INITIALIZATION_DATA_H_
|