decryptor.h

Go to the documentation of this file.

// Copyright 2018 Google LLC
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     https://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

#ifndef SHAKA_EMBEDDED_UTIL_DECRYPTORS_H_
#define SHAKA_EMBEDDED_UTIL_DECRYPTORS_H_

#include <memory>
#include <vector>

#include "shaka/eme/configuration.h"
#include "src/util/macros.h"

#define AES_BLOCK_SIZE 16u

namespace shaka {
namespace util {

class Decryptor {
 public:
  Decryptor(eme::EncryptionScheme scheme, const std::vector<uint8_t>& key,
            const std::vector<uint8_t>& iv);
  ~Decryptor();

  SHAKA_NON_COPYABLE_OR_MOVABLE_TYPE(Decryptor);

  bool DecryptPartialBlock(const uint8_t* data, size_t data_size,
                           uint32_t block_offset, uint8_t* dest);

  bool Decrypt(const uint8_t* data, size_t data_size, uint8_t* dest);

 private:
  bool InitIfNeeded();

  const eme::EncryptionScheme scheme_;
  const std::vector<uint8_t> key_;
  std::vector<uint8_t> iv_;

  struct Impl;
  std::unique_ptr<Impl> extra_;
};

}  // namespace util
}  // namespace shaka

#endif  // SHAKA_EMBEDDED_UTIL_DECRYPTORS_H_