cbc.hpp

/*
  ANCH Framework: ANother C++ Hack is a C++ framework based on C++11 standard
  Copyright (C) 2012 Vincent Lachenal
 
  This file is part of ANCH Framework.
 
  ANCH Framework is free software: you can redistribute it and/or modify
  it under the terms of the GNU Lesser General Public License as published by
  the Free Software Foundation, either version 3 of the License, or
  (at your option) any later version.
 
  ANCH Framework is distributed in the hope that it will be useful,
  but WITHOUT ANY WARRANTY; without even the implied warranty of
  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  GNU Lesser General Public License for more details.
 
  You should have received a copy of the GNU Lesser General Public License
  along with ANCH Framework.  If not, see <http://www.gnu.org/licenses/>.
*/
#pragma once
 
#include "crypto/cipher/bcModOp.hpp"
 
namespace anch {
  namespace crypto {

    template<typename Cipher, typename Padding>
    class CBC: public BlockCipherModeOfOperation<CBC<Cipher,Padding>,Cipher> {
 
      // Attributes +
    private:
      std::array<uint8_t,Cipher::getBlockSize()> _initVect;

      std::array<uint8_t,Cipher::getBlockSize()> _ctxtVect;
      // Attributes -
 
 
      // Constructors +
    public:
      CBC(const std::array<uint8_t,Cipher::getBlockSize()>& initVect, unsigned int nbThread = 1);
      // Constructors -
 
 
      // Destructor +
      virtual ~CBC();
      // Destructor -
 
 
      // Methods +
    protected:
      virtual std::size_t cipherBlock(std::array<uint8_t,Cipher::getBlockSize()>& input,
                                      std::streamsize nbRead,
                                      std::array<uint8_t,Cipher::getBlockSize()>& output,
                                      uint32_t, Cipher& cipher) override;

      virtual std::size_t decipherBlock(std::array<uint8_t,Cipher::getBlockSize()>& input,
                                        std::array<uint8_t,Cipher::getBlockSize()>& prevInput,
                                        std::streamsize nbRead,
                                        bool lastBlock,
                                        std::array<uint8_t,Cipher::getBlockSize()>& output,
                                        uint32_t, Cipher& cipher) override;

      virtual const std::array<uint8_t,Cipher::getBlockSize()>& reset();
      // Methods -
 
    };
 
    // Constructors +
    template<typename Cipher, typename Padding>
    CBC<Cipher,Padding>::CBC(const std::array<uint8_t,Cipher::getBlockSize()>& initVect, unsigned int nbThread):
      BlockCipherModeOfOperation<CBC<Cipher,Padding>,Cipher>(false, true, nbThread),
      _initVect(initVect),
      _ctxtVect() {
      // Nothing to do
    }
    // Constructors -
 
 
    // Destructor +
    template<typename Cipher, typename Padding>
    CBC<Cipher,Padding>::~CBC() {
      // Nothing to do
    }
    // Destructor -
 
 
    // Methods +
    template<typename Cipher, typename Padding>
    std::size_t
    CBC<Cipher,Padding>::cipherBlock(std::array<uint8_t,Cipher::getBlockSize()>& input,
                                     std::streamsize nbRead,
                                     std::array<uint8_t,Cipher::getBlockSize()>& output,
                                     uint32_t, Cipher& cipher) {
      if(static_cast<std::size_t>(nbRead) != Cipher::getBlockSize()) {
        Padding::pad(input.data(), static_cast<std::size_t>(nbRead), Cipher::getBlockSize());
      }
      std::array<uint8_t,Cipher::getBlockSize()> data;
      for(std::size_t i = 0 ; i < Cipher::getBlockSize() ; ++i) {
        data[i] = input[i] ^ _ctxtVect[i];
      }
      cipher.cipher(data, output);
      for(std::size_t i = 0 ; i < Cipher::getBlockSize() ; ++i) {
        _ctxtVect[i] = output[i];
      }
      return Cipher::getBlockSize(); // This mode pad data => the number of bytes to write will always be a complete block
    }
 
    template<typename Cipher, typename Padding>
    std::size_t
    CBC<Cipher,Padding>::decipherBlock(std::array<uint8_t,Cipher::getBlockSize()>& input,
                                       std::array<uint8_t,Cipher::getBlockSize()>& prevInput,
                                       std::streamsize nbRead,
                                       bool lastBlock,
                                       std::array<uint8_t,Cipher::getBlockSize()>& output,
                                       uint32_t, Cipher& cipher) {
      if(lastBlock && static_cast<std::size_t>(nbRead) != Cipher::getBlockSize()) {
        throw InvalidBlockException("Invalid block size");
      }
      std::array<uint8_t,Cipher::getBlockSize()> data;
      cipher.decipher(input, data);
      for(std::size_t i = 0 ; i < Cipher::getBlockSize() ; ++i) {
        output[i] = data[i] ^ prevInput[i];
      }
      if(lastBlock) {
        return Padding::length(output.data(), Cipher::getBlockSize());
      } else {
        return Cipher::getBlockSize();
      }
    }
 
    template<typename Cipher, typename Padding>
    const std::array<uint8_t,Cipher::getBlockSize()>&
    CBC<Cipher,Padding>::reset() {
      _ctxtVect = _initVect;
      return _initVect;
    }
    // Methods -
 
  }
}