pssh_generator.h

// Copyright 2017 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_PSSH_GENERATOR_H_
#define PACKAGER_MEDIA_BASE_PSSH_GENERATOR_H_
 
#include <cstdint>
#include <optional>
#include <vector>
 
#include <packager/media/base/protection_system_specific_info.h>
#include <packager/status.h>
 
// TODO(hmchen): move pssh related files into a sperate folder.
namespace shaka {
namespace media {
 
class PsshGenerator {
 public:
  PsshGenerator(const std::vector<uint8_t>& system_id, uint8_t box_version);
  virtual ~PsshGenerator();
 
  virtual bool SupportMultipleKeys() = 0;
 
  Status GeneratePsshFromKeyIds(
      const std::vector<std::vector<uint8_t>>& key_ids,
      ProtectionSystemSpecificInfo* info) const;
 
  Status GeneratePsshFromKeyIdAndKey(const std::vector<uint8_t>& key_id,
                                     const std::vector<uint8_t>& key,
                                     ProtectionSystemSpecificInfo* info) const;
 
 private:
  virtual std::optional<std::vector<uint8_t>> GeneratePsshDataFromKeyIds(
      const std::vector<std::vector<uint8_t>>& key_ids) const = 0;
 
  virtual std::optional<std::vector<uint8_t>> GeneratePsshDataFromKeyIdAndKey(
      const std::vector<uint8_t>& key_id,
      const std::vector<uint8_t>& key) const = 0;
 
  std::vector<uint8_t> system_id_;
  uint8_t box_version_ = 0;
};
 
}  // namespace media
}  // namespace shaka
 
#endif  // PACKAGER_MEDIA_BASE_PSSH_GENERATOR_H_