git.maemo.org Git - keepassx/blob - src/crypto/aescpp.h

   1 /*\r
   2  ---------------------------------------------------------------------------\r
   3  Copyright (c) 1998-2008, Brian Gladman, Worcester, UK. All rights reserved.\r
   4 \r
   5  LICENSE TERMS\r
   6 \r
   7  The redistribution and use of this software (with or without changes)\r
   8  is allowed without the payment of fees or royalties provided that:\r
   9 \r
  10   1. source code distributions include the above copyright notice, this\r
  11      list of conditions and the following disclaimer;\r
  12 \r
  13   2. binary distributions include the above copyright notice, this list\r
  14      of conditions and the following disclaimer in their documentation;\r
  15 \r
  16   3. the name of the copyright holder is not used to endorse products\r
  17      built using this software without specific written permission.\r
  18 \r
  19  DISCLAIMER\r
  20 \r
  21  This software is provided 'as is' with no explicit or implied warranties\r
  22  in respect of its properties, including, but not limited to, correctness\r
  23  and/or fitness for purpose.\r
  24  ---------------------------------------------------------------------------\r
  25  Issue Date: 20/12/2007\r
  26 \r
  27  This file contains the definitions required to use AES (Rijndael) in C++.\r
  28 */\r
  29 \r
  30 #ifndef _AESCPP_H\r
  31 #define _AESCPP_H\r
  32 \r
  33 #include "aes.h"\r
  34 \r
  35 #if defined( AES_ENCRYPT )\r
  36 \r
  37 class AESencrypt\r
  38 {\r
  39 public:\r
  40         AESencrypt(void) { aes_init(); };\r
  41 #if defined(AES_128)\r
  42         AESencrypt(const unsigned char key[])\r
  43                 {   aes_encrypt_key128(key, cx); }\r
  44         inline AES_RETURN key128(const unsigned char key[])\r
  45                 {   return aes_encrypt_key128(key, cx); }\r
  46 #endif\r
  47 #if defined(AES_192)\r
  48         inline AES_RETURN key192(const unsigned char key[])\r
  49                 {   return aes_encrypt_key192(key, cx); }\r
  50 #endif\r
  51 #if defined(AES_256)\r
  52         inline AES_RETURN key256(const unsigned char key[])\r
  53                 {   return aes_encrypt_key256(key, cx); }\r
  54 #endif\r
  55 #if defined(AES_VAR)\r
  56         inline AES_RETURN key(const unsigned char key[], int key_len)\r
  57                 {   return aes_encrypt_key(key, key_len, cx); }\r
  58 #endif\r
  59         inline AES_RETURN encrypt(const unsigned char in[], unsigned char out[]) const\r
  60                 {   return aes_encrypt(in, out, cx);  }\r
  61 #ifndef AES_MODES\r
  62         inline AES_RETURN ecb_encrypt(const unsigned char in[], unsigned char out[], int nb) const\r
  63                 {   while(nb--)\r
  64                         {   aes_encrypt(in, out, cx), in += AES_BLOCK_SIZE, out += AES_BLOCK_SIZE; }\r
  65                 }\r
  66 #endif\r
  67 #ifdef AES_MODES\r
  68         inline AES_RETURN mode_reset(void)   { return aes_mode_reset(cx); }\r
  69 \r
  70         inline AES_RETURN ecb_encrypt(const unsigned char in[], unsigned char out[], int nb) const\r
  71                 {   return aes_ecb_encrypt(in, out, nb, cx);  }\r
  72 \r
  73         inline AES_RETURN cbc_encrypt(const unsigned char in[], unsigned char out[], int nb,\r
  74                                unsigned char iv[]) const\r
  75                 {   return aes_cbc_encrypt(in, out, nb, iv, cx);  }\r
  76 \r
  77         inline AES_RETURN cfb_encrypt(const unsigned char in[], unsigned char out[], int nb,\r
  78                                unsigned char iv[])\r
  79                 {   return aes_cfb_encrypt(in, out, nb, iv, cx);  }\r
  80 \r
  81         inline AES_RETURN cfb_decrypt(const unsigned char in[], unsigned char out[], int nb,\r
  82                                unsigned char iv[])\r
  83                 {   return aes_cfb_decrypt(in, out, nb, iv, cx);  }\r
  84 \r
  85         inline AES_RETURN ofb_crypt(const unsigned char in[], unsigned char out[], int nb,\r
  86                              unsigned char iv[])\r
  87                 {   return aes_ofb_crypt(in, out, nb, iv, cx);  }\r
  88 \r
  89         typedef void ctr_fn(unsigned char ctr[]);\r
  90 \r
  91         inline AES_RETURN ctr_crypt(const unsigned char in[], unsigned char out[], int nb,\r
  92                             unsigned char iv[], ctr_fn cf)\r
  93                 {   return aes_ctr_crypt(in, out, nb, iv, cf, cx);  }\r
  94 \r
  95 #endif\r
  96 \r
  97 private:\r
  98         aes_encrypt_ctx cx[1];\r
  99 };\r
 100 \r
 101 #endif\r
 102 \r
 103 #if defined( AES_DECRYPT )\r
 104 \r
 105 class AESdecrypt\r
 106 {\r
 107 public:\r
 108         AESdecrypt(void) { aes_init(); };\r
 109 #if defined(AES_128)\r
 110         AESdecrypt(const unsigned char key[])\r
 111                 { aes_decrypt_key128(key, cx); }\r
 112         inline AES_RETURN key128(const unsigned char key[])\r
 113                 { return aes_decrypt_key128(key, cx); }\r
 114 #endif\r
 115 #if defined(AES_192)\r
 116         inline AES_RETURN key192(const unsigned char key[])\r
 117                 { return aes_decrypt_key192(key, cx); }\r
 118 #endif\r
 119 #if defined(AES_256)\r
 120         inline AES_RETURN key256(const unsigned char key[])\r
 121                 { return aes_decrypt_key256(key, cx); }\r
 122 #endif\r
 123 #if defined(AES_VAR)\r
 124         inline AES_RETURN key(const unsigned char key[], int key_len)\r
 125                 { return aes_decrypt_key(key, key_len, cx); }\r
 126 #endif\r
 127         inline AES_RETURN decrypt(const unsigned char in[], unsigned char out[]) const\r
 128                 {   return aes_decrypt(in, out, cx);  }\r
 129 #ifndef AES_MODES\r
 130         inline AES_RETURN ecb_decrypt(const unsigned char in[], unsigned char out[], int nb) const\r
 131                 {   while(nb--)\r
 132                         {   aes_decrypt(in, out, cx), in += AES_BLOCK_SIZE, out += AES_BLOCK_SIZE; }\r
 133                 }\r
 134 #endif\r
 135 #ifdef AES_MODES\r
 136 \r
 137         inline AES_RETURN ecb_decrypt(const unsigned char in[], unsigned char out[], int nb) const\r
 138                 {   return aes_ecb_decrypt(in, out, nb, cx);  }\r
 139 \r
 140         inline AES_RETURN cbc_decrypt(const unsigned char in[], unsigned char out[], int nb,\r
 141                                unsigned char iv[]) const\r
 142                 {   return aes_cbc_decrypt(in, out, nb, iv, cx);  }\r
 143 #endif\r
 144 private:\r
 145         aes_decrypt_ctx cx[1];\r
 146 };\r
 147 \r
 148 #endif\r
 149 \r
 150 #endif\r