protection_system_specific_info.h

// Copyright 2016 Google LLC. 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 <cstddef>
#include <cstdint>
#include <memory>
#include <vector>
 
#include <absl/log/check.h>
 
 
namespace shaka {
namespace media {
 
struct ProtectionSystemSpecificInfo {
  std::vector<uint8_t> system_id;
  std::vector<uint8_t> psshs;
 
  static bool ParseBoxes(const uint8_t* data,
                         size_t data_size,
                         std::vector<ProtectionSystemSpecificInfo>* pssh_boxes);
};
 
class PsshBoxBuilder {
 public:
  PsshBoxBuilder() = default;
  ~PsshBoxBuilder() = default;
 
  static std::unique_ptr<PsshBoxBuilder> ParseFromBox(const uint8_t* data,
                                                      size_t data_size);
 
  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_