87 lines
2.7 KiB
C++
87 lines
2.7 KiB
C++
// Copyright 2016 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
|
|
|
|
#ifndef PACKAGER_MEDIA_BASE_PROTECTION_SYSTEM_SPECIFIC_INFO_H_
|
|
#define PACKAGER_MEDIA_BASE_PROTECTION_SYSTEM_SPECIFIC_INFO_H_
|
|
|
|
#include <stdint.h>
|
|
#include <memory>
|
|
#include <vector>
|
|
|
|
#include "packager/base/logging.h"
|
|
|
|
#define NO_PROTECTION_SYSTEM_FLAG 0x00
|
|
#define COMMON_PROTECTION_SYSTEM_FLAG 0x01
|
|
#define PLAYREADY_PROTECTION_SYSTEM_FLAG 0x02
|
|
#define WIDEVINE_PROTECTION_SYSTEM_FLAG 0x04
|
|
#define FAIRPLAY_PROTECTION_SYSTEM_FLAG 0x08
|
|
#define MARLIN_PROTECTION_SYSTEM_FLAG 0x10
|
|
|
|
namespace shaka {
|
|
namespace media {
|
|
|
|
struct ProtectionSystemSpecificInfo {
|
|
std::vector<uint8_t> system_id;
|
|
std::vector<uint8_t> psshs;
|
|
|
|
/// Parses multiple PSSH boxes from @a data. These boxes should be
|
|
/// concatenated together. Any non-PSSH box is an error.
|
|
/// @return true on success; false on failure.
|
|
static bool ParseBoxes(
|
|
const uint8_t* data,
|
|
size_t data_size,
|
|
std::vector<ProtectionSystemSpecificInfo>* pssh_boxes);
|
|
};
|
|
|
|
class PsshBoxBuilder {
|
|
public:
|
|
PsshBoxBuilder() = default;
|
|
~PsshBoxBuilder() = default;
|
|
|
|
/// Parses the given PSSH box into this object.
|
|
/// @return nullptr on failure.
|
|
static std::unique_ptr<PsshBoxBuilder> ParseFromBox(const uint8_t* data,
|
|
size_t data_size);
|
|
|
|
/// Creates a PSSH box for the current data.
|
|
std::vector<uint8_t> CreateBox() const;
|
|
|
|
uint8_t pssh_box_version() const { return version_; }
|
|
const std::vector<uint8_t>& system_id() const { return system_id_; }
|
|
const std::vector<std::vector<uint8_t>>& key_ids() const { return key_ids_; }
|
|
const std::vector<uint8_t>& pssh_data() const { return pssh_data_; }
|
|
|
|
void set_pssh_box_version(uint8_t version) {
|
|
DCHECK_LT(version, 2);
|
|
version_ = version;
|
|
}
|
|
void set_system_id(const uint8_t* system_id, size_t system_id_size) {
|
|
DCHECK_EQ(16u, system_id_size);
|
|
system_id_.assign(system_id, system_id + system_id_size);
|
|
}
|
|
void add_key_id(const std::vector<uint8_t>& key_id) {
|
|
key_ids_.push_back(key_id);
|
|
}
|
|
void clear_key_ids() { key_ids_.clear(); }
|
|
void set_pssh_data(const std::vector<uint8_t>& pssh_data) {
|
|
pssh_data_ = pssh_data;
|
|
}
|
|
|
|
private:
|
|
PsshBoxBuilder(const PsshBoxBuilder&) = delete;
|
|
PsshBoxBuilder& operator=(const PsshBoxBuilder&) = delete;
|
|
|
|
uint8_t version_ = 0;
|
|
std::vector<uint8_t> system_id_;
|
|
std::vector<std::vector<uint8_t>> key_ids_;
|
|
std::vector<uint8_t> pssh_data_;
|
|
};
|
|
|
|
} // namespace media
|
|
} // namespace shaka
|
|
|
|
#endif // PACKAGER_MEDIA_BASE_PROTECTION_SYSTEM_SPECIFIC_INFO_H_
|