aes_encryptor.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
//
// AES Encryptor implementation using openssl.

#ifndef MEDIA_BASE_AES_ENCRYPTOR_H_
#define MEDIA_BASE_AES_ENCRYPTOR_H_

#include <string>
#include <vector>

#include "packager/base/memory/scoped_ptr.h"
#include "packager/base/stl_util.h"

struct aes_key_st;
typedef struct aes_key_st AES_KEY;

namespace edash_packager {
namespace media {

class AesEncryptor {
 public:
  AesEncryptor();
  virtual ~AesEncryptor();

  virtual bool InitializeWithRandomIv(const std::vector<uint8_t>& key,
                                      uint8_t iv_size);

  virtual bool InitializeWithIv(const std::vector<uint8_t>& key,
                                const std::vector<uint8_t>& iv) = 0;

  virtual size_t NumPaddingBytes(size_t size) = 0;

  virtual bool EncryptData(const uint8_t* plaintext,
                           size_t plaintext_size,
                           uint8_t* ciphertext) = 0;

  bool Encrypt(const std::vector<uint8_t>& plaintext,
               std::vector<uint8_t>* ciphertext);

  bool Encrypt(const std::string& plaintext, std::string* ciphertext);

  virtual void UpdateIv() = 0;

  virtual bool SetIv(const std::vector<uint8_t>& iv) = 0;

  const std::vector<uint8_t>& iv() const { return iv_; }

 protected:
  // Initialization vector, with size 8 or 16.
  std::vector<uint8_t> iv_;
  // Openssl AES_KEY.
  scoped_ptr<AES_KEY> aes_key_;

 private:
  DISALLOW_COPY_AND_ASSIGN(AesEncryptor);
};

// Class which implements AES-CTR counter-mode encryption/decryption.
class AesCtrEncryptor : public AesEncryptor {
 public:
  AesCtrEncryptor();
  ~AesCtrEncryptor() override;

  bool InitializeWithIv(const std::vector<uint8_t>& key,
                        const std::vector<uint8_t>& iv) override;

  size_t NumPaddingBytes(size_t size) override;

  bool EncryptData(const uint8_t* plaintext,
                   size_t plaintext_size,
                   uint8_t* ciphertext) override;

  void UpdateIv() override;

  bool SetIv(const std::vector<uint8_t>& iv) override;

  uint32_t block_offset() const { return block_offset_; }

 private:
  // Current block offset.
  uint32_t block_offset_;
  // Current AES-CTR counter.
  std::vector<uint8_t> counter_;
  // Encrypted counter.
  std::vector<uint8_t> encrypted_counter_;
  // Keep track of whether the counter has overflowed.
  bool counter_overflow_;

  DISALLOW_COPY_AND_ASSIGN(AesCtrEncryptor);
};

// Class which implements AES-CBC (Cipher block chaining) encryption with
// PKCS#5 padding.
class AesCbcPkcs5Encryptor : public AesEncryptor {
 public:
  AesCbcPkcs5Encryptor();
  ~AesCbcPkcs5Encryptor() override;

  bool InitializeWithIv(const std::vector<uint8_t>& key,
                        const std::vector<uint8_t>& iv) override;

  size_t NumPaddingBytes(size_t size) override;

  bool EncryptData(const uint8_t* plaintext,
                   size_t plaintext_size,
                   uint8_t* ciphertext) override;

  void UpdateIv() override;

  bool SetIv(const std::vector<uint8_t>& iv) override;

 private:
  DISALLOW_COPY_AND_ASSIGN(AesCbcPkcs5Encryptor);
};

// Class which implements AES-CBC (Cipher block chaining) encryption with
// Ciphertext stealing.
class AesCbcCtsEncryptor : public AesEncryptor {
 public:
  AesCbcCtsEncryptor();
  ~AesCbcCtsEncryptor() override;

  bool InitializeWithIv(const std::vector<uint8_t>& key,
                        const std::vector<uint8_t>& iv) override;

  size_t NumPaddingBytes(size_t size) override;

  bool EncryptData(const uint8_t* plaintext,
                   size_t plaintext_size,
                   uint8_t* ciphertext) override;

  void UpdateIv() override;

  bool SetIv(const std::vector<uint8_t>& iv) override;

 private:
  DISALLOW_COPY_AND_ASSIGN(AesCbcCtsEncryptor);
};

}  // namespace media
}  // namespace edash_packager

#endif  // MEDIA_BASE_AES_ENCRYPTOR_H_