decryptor_source.h

// Copyright 2016 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_DECRYPTOR_SOURCE_H_
#define PACKAGER_MEDIA_BASE_DECRYPTOR_SOURCE_H_
 
#include <cstdint>
#include <map>
#include <memory>
#include <vector>
 
#include <packager/macros/classes.h>
#include <packager/media/base/aes_decryptor.h>
#include <packager/media/base/decrypt_config.h>
#include <packager/media/base/key_source.h>
 
namespace shaka {
namespace media {
 
class DecryptorSource {
 public:
  explicit DecryptorSource(KeySource* key_source);
  ~DecryptorSource();
 
  bool DecryptSampleBuffer(const DecryptConfig* decrypt_config,
                           const uint8_t* encrypted_buffer,
                           size_t buffer_size,
                           uint8_t* decrypted_buffer);
 
 private:
  KeySource* key_source_;
  std::map<std::vector<uint8_t>, std::unique_ptr<AesCryptor>> decryptor_map_;
 
  DISALLOW_COPY_AND_ASSIGN(DecryptorSource);
};
 
}  // namespace media
}  // namespace shaka
 
#endif  // PACKAGER_MEDIA_BASE_DECRYPTOR_SOURCE_H_