// $Id: rijndael.h 65 2008-02-25 13:16:37Z tb $

#ifndef RIJNDAEL_H
#define RIJNDAEL_H

#include <inttypes.h>
#include <stdlib.h>

/**
 * AES state context to encrypt input data blocks in ECB mode.
 */
class RijndaelEncryptECB
{
protected:

    /// key-length-dependent number of rounds
    int		nr_;

    /// encrypt key schedule
    uint32_t	ek_[60];

public:    

    /// Set the encryption key. The key must be 16, 24 or 32 bytes long.
    void set_key(const unsigned char* key, unsigned int keylen);

    /// Encrypt a block of 16 bytes using the current cipher state.
    void encrypt_block(const uint8_t src[16], uint8_t dst[16]) const;

    /// Encrypt a length of n*16 bytes. Len must be a multiple of 16 or this
    /// function will assert().
    void encrypt(const void* src, void* dst, size_t len) const;
};

/**
 * AES state context to decrypt input data blocks in ECB mode.
 */
class RijndaelDecryptECB
{
protected:

    /// key-length-dependent number of rounds
    int		nr_;

    /// decrypt key schedule
    uint32_t	dk_[60];

public:

    /// Set the encryption key. The key must be 16, 24 or 32 bytes long.
    void set_key(const unsigned char* key, unsigned int keylen);

    /// Decrypt a block of 16 bytes using the current cipher state.
    void decrypt_block(const uint8_t src[16], uint8_t dst[16]) const;

    /// Decrypt a length of n*16 bytes. Len must be a multiple of 16 or this
    /// function will assert().
    void decrypt(const void* src, void* dst, size_t len) const;
};

/**
 * AES state context to encrypt input data blocks in CBC mode.
 */
class RijndaelEncryptCBC
{
protected:

    /// key-length-dependent number of rounds
    int		nr_;

    /// encrypt key schedule
    uint32_t	ek_[60];

    /// cbc initialisation vector
    uint8_t	cbciv_[16];

public:    

    /// Set the encryption key. The key must be 16, 24 or 32 bytes long.
    void set_key(const unsigned char* key, unsigned int keylen);

    /// Set the initial cbc vector. The vector is always 16 bytes long.
    void set_cbciv(const uint8_t iv[16]);

    /// Encrypt a block of 16 bytes using the current cipher state.
    void encrypt_block(const uint8_t src[16], uint8_t dst[16]);

    /// Encrypt a length of n*16 bytes. Len must be a multiple of 16 or this
    /// function will assert().
    void encrypt(const void* src, void* dst, size_t len);
};

/**
 * AES state context to decrypt input data blocks in CBC mode.
 */
class RijndaelDecryptCBC
{
protected:

    /// key-length-dependent number of rounds
    int		nr_;

    /// decrypt key schedule
    uint32_t	dk_[60];

    /// cbc initialisation vector
    uint8_t	cbciv_[16];

    /// temporary cbc block
    uint8_t	cbcivsave_[16];

public:

    /// Set the encryption key. The key must be 16, 24 or 32 bytes long.
    void set_key(const unsigned char* key, unsigned int keylen);

    /// Set the initial cbc vector. The vector is always 16 bytes long.
    void set_cbciv(const uint8_t iv[16]);

    /// Decrypt a block of 16 bytes using the current cipher state.
    void decrypt_block(const uint8_t src[16], uint8_t dst[16]);

    /// Decrypt a length of n*16 bytes. Len must be a multiple of 16 or this
    /// function will assert().
    void decrypt(const void* src, void* dst, size_t len);
};

#endif // RIJNDAEL_H