AES.h

#ifndef __AES_H__
#define __AES_H__

#include "AES_config.h"
/*
 ---------------------------------------------------------------------------
 Copyright (c) 1998-2008, Brian Gladman, Worcester, UK. All rights reserved.

 LICENSE TERMS

 The redistribution and use of this software (with or without changes)
 is allowed without the payment of fees or royalties provided that:

  1. source code distributions include the above copyright notice, this
     list of conditions and the following disclaimer;

  2. binary distributions include the above copyright notice, this list
     of conditions and the following disclaimer in their documentation;

  3. the name of the copyright holder is not used to endorse products
     built using this software without specific written permission.

 DISCLAIMER

 This software is provided 'as is' with no explicit or implied warranties
 in respect of its properties, including, but not limited to, correctness
 and/or fitness for purpose.
 ---------------------------------------------------------------------------
 Issue 09/09/2006

 This is an AES implementation that uses only 8-bit byte operations on the
 cipher state.
 */
 
 /* code was modified by george spanos <spaniakos@gmail.com>
 * 16/12/14
 */

class AES
{
 public:

/*  The following calls are for a precomputed key schedule

    NOTE: If the length_type used for the key length is an
    unsigned 8-bit character, a key length of 256 bits must
    be entered as a length in bytes (valid inputs are hence
    128, 192, 16, 24 and 32).
*/
    AES();
        
    byte set_key (byte key[], int keylen) ;
    
    void clean () ;  // delete key schedule after use
    
    void copy_n_bytes (byte * AESt, byte * src, byte n) ;

    byte encrypt (byte plain [N_BLOCK], byte cipher [N_BLOCK]) ;
    
    byte cbc_encrypt (byte * plain, byte * cipher, int n_block, byte iv [N_BLOCK]) ;
    
    byte cbc_encrypt (byte * plain, byte * cipher, int n_block) ;


    byte decrypt (byte cipher [N_BLOCK], byte plain [N_BLOCK]) ;
    
    byte cbc_decrypt (byte * cipher, byte * plain, int n_block, byte iv [N_BLOCK]) ;
    
    byte cbc_decrypt (byte * cipher, byte * plain, int n_block) ;
        
    void set_IV(unsigned long long int IVCl);
        
    void iv_inc();
        
    int get_size();
    
    void set_size(int sizel);
    
    void get_IV(byte *out);
        
    void calc_size_n_pad(int p_size);
    
    void padPlaintext(void* in,byte* out);
        
    bool CheckPad(byte* in,int size);

    void printArray(byte output[],bool p_pad = true);
    
    void printArray(byte output[],int sizel);
    
    void do_aes_encrypt(byte *plain,int size_p,byte *cipher,byte *key, int bits, byte ivl [N_BLOCK]);
    
    void do_aes_encrypt(byte *plain,int size_p,byte *cipher,byte *key, int bits);
    
    void do_aes_decrypt(byte *cipher,int size_c,byte *plain,byte *key, int bits, byte ivl [N_BLOCK]);
    
    void do_aes_decrypt(byte *cipher,int size_c,byte *plain,byte *key, int bits);

    #if defined(AES_LINUX)

        double millis();
    #endif
 private:
  int round ;
  byte key_sched [KEY_SCHEDULE_BYTES] ;
  unsigned long long int IVC;
  byte iv[16];
  int pad;
  int size;
  #if defined(AES_LINUX)
    timeval tv;
    byte arr_pad[15];
  #else
    byte arr_pad[15] = { 0x01,0x02,0x03,0x04,0x05,0x06,0x07,0x08,0x09,0x0a,0x0b,0x0c,0x0d,0x0e,0x0f };
  #endif
} ;


#endif