1 files changed, 77 insertions, 0 deletions
diff --git a/src/lib/block/aes/aes.h b/src/lib/block/aes/aes.h
new file mode 100644
index 000000000..5ddd39b08
--- /dev/null
+++ b/src/lib/block/aes/aes.h
@@ -0,0 +1,77 @@
+/*
+* AES
+* (C) 1999-2010 Jack Lloyd
+*
+* Distributed under the terms of the Botan license
+*/
+
+#ifndef BOTAN_AES_H__
+#define BOTAN_AES_H__
+
+#include <botan/block_cipher.h>
+
+namespace Botan {
+
+/**
+* AES-128
+*/
+class BOTAN_DLL AES_128 : public Block_Cipher_Fixed_Params<16, 16>
+   {
+   public:
+      void encrypt_n(const byte in[], byte out[], size_t blocks) const;
+      void decrypt_n(const byte in[], byte out[], size_t blocks) const;
+
+      void clear();
+
+      std::string name() const { return "AES-128"; }
+      BlockCipher* clone() const { return new AES_128; }
+   private:
+      void key_schedule(const byte key[], size_t length);
+
+      secure_vector<u32bit> EK, DK;
+      secure_vector<byte> ME, MD;
+   };
+
+/**
+* AES-192
+*/
+class BOTAN_DLL AES_192 : public Block_Cipher_Fixed_Params<16, 24>
+   {
+   public:
+      void encrypt_n(const byte in[], byte out[], size_t blocks) const;
+      void decrypt_n(const byte in[], byte out[], size_t blocks) const;
+
+      void clear();
+
+      std::string name() const { return "AES-192"; }
+      BlockCipher* clone() const { return new AES_192; }
+   private:
+      void key_schedule(const byte key[], size_t length);
+
+      secure_vector<u32bit> EK, DK;
+      secure_vector<byte> ME, MD;
+   };
+
+/**
+* AES-256
+*/
+class BOTAN_DLL AES_256 : public Block_Cipher_Fixed_Params<16, 32>
+   {
+   public:
+      void encrypt_n(const byte in[], byte out[], size_t blocks) const;
+      void decrypt_n(const byte in[], byte out[], size_t blocks) const;
+
+      void clear();
+
+      std::string name() const { return "AES-256"; }
+      BlockCipher* clone() const { return new AES_256; }
+   private:
+      void key_schedule(const byte key[], size_t length);
+
+      secure_vector<u32bit> EK, DK;
+      secure_vector<byte> ME, MD;
+   };
+
+}
+
+#endif