playready_key_source.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_PLAYREADY_SOURCE_H_
#define PACKAGER_MEDIA_BASE_PLAYREADY_SOURCE_H_
 
#include <cstdint>
#include <memory>
#include <string>
#include <vector>
 
#include <packager/macros/classes.h>
#include <packager/media/base/key_source.h>
 
namespace shaka {
namespace media {
 
class PlayReadyKeySource : public KeySource {
 public:
  PlayReadyKeySource(const std::string& server_url,
                     ProtectionSystem protection_systems);
  ~PlayReadyKeySource() override;
 
  Status FetchKeys(EmeInitDataType init_data_type,
                   const std::vector<uint8_t>& init_data) override;
  Status GetKey(const std::string& stream_label, EncryptionKey* key) override;
  Status GetKey(const std::vector<uint8_t>& key_id,
                EncryptionKey* key) override;
  Status GetCryptoPeriodKey(uint32_t crypto_period_index,
                            int32_t crypto_period_duration_in_seconds,
                            const std::string& stream_label,
                            EncryptionKey* key) override;
  virtual Status FetchKeysWithProgramIdentifier(const std::string& program_identifier);
 
  static std::unique_ptr<PlayReadyKeySource> CreateFromKeyAndKeyId(
      const std::vector<uint8_t>& key_id,
      const std::vector<uint8_t>& key);
 
 private:
  Status GetKeyInternal();
  Status GetCryptoPeriodKeyInternal();
 
  // Indicates whether PlayReady protection system should be generated.
  bool generate_playready_protection_system_ = true;
 
  std::unique_ptr<EncryptionKey> encryption_key_;
  std::string server_url_;
 
  DISALLOW_COPY_AND_ASSIGN(PlayReadyKeySource);
};
 
}  // namespace media
}  // namespace shaka
 
#endif  // PACKAGER_MEDIA_BASE_PLAYREADY_SOURCE_H_