fixed_key_source.h

// Copyright 2016 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_FIXED_KEY_SOURCE_H_
#define MEDIA_BASE_FIXED_KEY_SOURCE_H_

#include <string>
#include <vector>

#include "packager/base/memory/scoped_ptr.h"
#include "packager/media/base/key_source.h"

namespace edash_packager {
namespace media {

class FixedKeySource : public KeySource {
 public:
  ~FixedKeySource() override;

  Status FetchKeys(const std::vector<uint8_t>& pssh_box) override;
  Status FetchKeys(const std::vector<std::vector<uint8_t>>& key_ids) override;
  Status FetchKeys(uint32_t asset_id) override;

  Status GetKey(TrackType track_type, EncryptionKey* key) override;
  Status GetKey(const std::vector<uint8_t>& key_id,
                EncryptionKey* key) override;
  Status GetCryptoPeriodKey(uint32_t crypto_period_index,
                            TrackType track_type,
                            EncryptionKey* key) override;

  static scoped_ptr<FixedKeySource> CreateFromHexStrings(
      const std::string& key_id_hex,
      const std::string& key_hex,
      const std::string& pssh_boxes_hex,
      const std::string& iv_hex);

 protected:
  // Allow default constructor for mock key sources.
  FixedKeySource();

 private:
  explicit FixedKeySource(scoped_ptr<EncryptionKey> key);

  scoped_ptr<EncryptionKey> encryption_key_;

  DISALLOW_COPY_AND_ASSIGN(FixedKeySource);
};

}  // namespace media
}  // namespace edash_packager

#endif  // MEDIA_BASE_FIXED_KEY_SOURCE_H_