key_source.h

// Copyright 2014 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 MEDIA_BASE_KEY_SOURCE_H_
#define MEDIA_BASE_KEY_SOURCE_H_

#include <string>
#include <vector>

#include "packager/media/base/protection_system_specific_info.h"
#include "packager/media/base/status.h"

namespace shaka {
namespace media {

struct EncryptionKey {
  EncryptionKey();
  ~EncryptionKey();

  std::vector<ProtectionSystemSpecificInfo> key_system_info;
  std::vector<uint8_t> key_id;
  std::vector<uint8_t> key;
  std::vector<uint8_t> iv;
};

class KeySource {
 public:
  enum TrackType {
    TRACK_TYPE_UNKNOWN = 0,
    TRACK_TYPE_SD = 1,
    TRACK_TYPE_HD = 2,
    TRACK_TYPE_UHD1 = 3,
    TRACK_TYPE_UHD2 = 4,
    TRACK_TYPE_AUDIO = 5,
    TRACK_TYPE_UNSPECIFIED = 6,
    NUM_VALID_TRACK_TYPES = 6
  };

  KeySource();
  virtual ~KeySource();

  virtual Status FetchKeys(const std::vector<uint8_t>& pssh_box) = 0;

  virtual Status FetchKeys(
      const std::vector<std::vector<uint8_t>>& key_ids) = 0;

  virtual Status FetchKeys(uint32_t asset_id) = 0;

  virtual Status GetKey(TrackType track_type, EncryptionKey* key) = 0;

  virtual Status GetKey(const std::vector<uint8_t>& key_id,
                        EncryptionKey* key) = 0;

  virtual Status GetCryptoPeriodKey(uint32_t crypto_period_index,
                                    TrackType track_type,
                                    EncryptionKey* key) = 0;

  static TrackType GetTrackTypeFromString(const std::string& track_type_string);

  static std::string TrackTypeToString(TrackType track_type);

 private:
  DISALLOW_COPY_AND_ASSIGN(KeySource);
};

}  // namespace media
}  // namespace shaka

#endif  // MEDIA_BASE_KEY_SOURCE_H_