apps/elftosb/common/RijndaelCTR.h

/*
 * File:    RijndaelCTR.h
 *
 * Copyright (c) Freescale Semiconductor, Inc. All rights reserved.
 * See included license file for license details.
 */
#if !defined(_RijndaelCTR_h_)
#define _RijndaelCTR_h_

#include "AESKey.h"
#include "AESCounter.h"
#include <string.h>

class RijndaelCTR
{
public:
    enum
    {
        BLOCK_SIZE = 16 
    };

    typedef uint8_t block_t[BLOCK_SIZE];

public:
    RijndaelCTR(const AESKey<128> &key, const AESCounter<128> &counter);

    void encrypt(const uint8_t *data, unsigned length, uint8_t *dest);

    void decrypt(const uint8_t *data, unsigned length, uint8_t *dest);

    RijndaelCTR &operator=(const RijndaelCTR &other)
    {
        m_key = other.m_key;
        m_counter = other.m_counter;
        return *this;
    }

protected:
    AESKey<128> m_key;         
    AESCounter<128> m_counter; 

    void incrementCounter(AESCounter<128>::counter_t *counter);
    void baseProcess(const uint8_t *data, unsigned length, uint8_t *dest);

private:
    RijndaelCTR() {}
};

#endif // _RijndaelCTR_h_