key_source.h

// Copyright 2014 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_KEY_SOURCE_H_
#define PACKAGER_MEDIA_BASE_KEY_SOURCE_H_
 
#include <map>
#include <memory>
#include <string>
#include <vector>
 
#include <packager/macros/classes.h>
#include <packager/media/base/fourccs.h>
#include <packager/media/base/protection_system_specific_info.h>
#include <packager/media/base/pssh_generator.h>
#include <packager/status.h>
 
namespace shaka {
namespace media {
 
enum class EmeInitDataType {
  UNKNOWN,
  CENC,
  WEBM,
  KEYIDS,
  WIDEVINE_CLASSIC,
  MAX = WIDEVINE_CLASSIC
};
 
struct EncryptionKey {
  std::vector<ProtectionSystemSpecificInfo> key_system_info;
  std::vector<uint8_t> key_id;
  std::vector<std::vector<uint8_t>> key_ids;
  std::vector<uint8_t> key;
  std::vector<uint8_t> iv;
};
 
typedef std::map<std::string, std::unique_ptr<EncryptionKey>> EncryptionKeyMap;
 
class KeySource {
 public:
  KeySource();
 
  virtual ~KeySource();
 
  virtual Status FetchKeys(EmeInitDataType init_data_type,
                           const std::vector<uint8_t>& init_data) = 0;
 
  virtual Status GetKey(const std::string& stream_label,
                        EncryptionKey* key) = 0;
 
  virtual Status GetKey(const std::vector<uint8_t>& key_id,
                        EncryptionKey* key) = 0;
 
  virtual Status GetCryptoPeriodKey(uint32_t crypto_period_index,
                                    int32_t crypto_period_duration_in_seconds,
                                    const std::string& stream_label,
                                    EncryptionKey* key) = 0;
 
 private:
  DISALLOW_COPY_AND_ASSIGN(KeySource);
};
 
}  // namespace media
}  // namespace shaka
 
#endif  // PACKAGER_MEDIA_BASE_KEY_SOURCE_H_