1 files changed, 93 insertions, 0 deletions
diff --git a/src/Volume/EncryptionAlgorithm.h b/src/Volume/EncryptionAlgorithm.h
new file mode 100644
index 00000000..8118de6b
--- /dev/null
+++ b/src/Volume/EncryptionAlgorithm.h
@@ -0,0 +1,93 @@
+/*
+ Copyright (c) 2008 TrueCrypt Developers Association. All rights reserved.
+
+ Governed by the TrueCrypt License 3.0 the full text of which is contained in
+ the file License.txt included in TrueCrypt binary and source code distribution
+ packages.
+*/
+
+#ifndef TC_HEADER_Encryption_EncryptionAlgorithm
+#define TC_HEADER_Encryption_EncryptionAlgorithm
+
+#include "Platform/Platform.h"
+#include "Cipher.h"
+#include "EncryptionMode.h"
+
+namespace TrueCrypt
+{
+	class EncryptionAlgorithm;
+	typedef list < shared_ptr <EncryptionAlgorithm> > EncryptionAlgorithmList;
+
+	class EncryptionAlgorithm
+	{
+	public:
+		virtual ~EncryptionAlgorithm ();
+
+		virtual void Decrypt (byte *data, uint64 length) const;
+		virtual void Decrypt (const BufferPtr &data) const;
+		virtual void DecryptSectors (byte *data, uint64 sectorIndex, uint64 sectorCount, size_t sectorSize) const;
+		virtual void Encrypt (byte *data, uint64 length) const;
+		virtual void Encrypt (const BufferPtr &data) const;
+		virtual void EncryptSectors (byte *data, uint64 sectorIndex, uint64 sectorCount, size_t sectorSize) const;
+		static EncryptionAlgorithmList GetAvailableAlgorithms ();
+		virtual const CipherList &GetCiphers () const { return Ciphers; }
+		virtual shared_ptr <EncryptionAlgorithm> GetNew () const = 0;
+		virtual size_t GetMaxBlockSize () const;
+		virtual size_t GetMinBlockSize () const;
+		static size_t GetLargestKeySize (const EncryptionAlgorithmList &algorithms);
+		virtual size_t GetKeySize () const;
+		virtual shared_ptr <EncryptionMode> GetMode () const;
+		virtual wstring GetName () const;
+		bool IsDeprecated () const { return Deprecated; }
+		virtual bool IsModeSupported (const EncryptionMode &mode) const;
+		virtual bool IsModeSupported (const shared_ptr <EncryptionMode> mode) const;
+		virtual void SetKey (const ConstBufferPtr &key);
+		virtual void SetMode (shared_ptr <EncryptionMode> mode);
+
+	protected:
+		EncryptionAlgorithm ();
+
+		void ValidateState () const;
+
+		CipherList Ciphers;
+		bool Deprecated;
+		shared_ptr <EncryptionMode> Mode;
+		EncryptionModeList SupportedModes;
+
+	private:
+		EncryptionAlgorithm (const EncryptionAlgorithm &);
+		EncryptionAlgorithm &operator= (const EncryptionAlgorithm &);
+	};
+
+#define TC_ENCRYPTION_ALGORITHM(NAME) \
+	class NAME : public EncryptionAlgorithm \
+	{ \
+	public: \
+		NAME (); \
+		virtual ~NAME () { } \
+\
+		virtual shared_ptr <EncryptionAlgorithm> GetNew () const { return shared_ptr <EncryptionAlgorithm> (new NAME()); } \
+\
+	private: \
+		NAME (const NAME &); \
+		NAME &operator= (const NAME &); \
+	}
+
+	TC_ENCRYPTION_ALGORITHM (AES);
+	TC_ENCRYPTION_ALGORITHM (AESBlowfish);
+	TC_ENCRYPTION_ALGORITHM (AESBlowfishSerpent);
+	TC_ENCRYPTION_ALGORITHM (AESTwofish);
+	TC_ENCRYPTION_ALGORITHM (AESTwofishSerpent);
+	TC_ENCRYPTION_ALGORITHM (Blowfish);
+	TC_ENCRYPTION_ALGORITHM (Cast5);
+	TC_ENCRYPTION_ALGORITHM (Serpent);
+	TC_ENCRYPTION_ALGORITHM (SerpentAES);
+	TC_ENCRYPTION_ALGORITHM (TripleDES);
+	TC_ENCRYPTION_ALGORITHM (Twofish);
+	TC_ENCRYPTION_ALGORITHM (TwofishSerpent);
+	TC_ENCRYPTION_ALGORITHM (SerpentTwofishAES);
+
+#undef TC_ENCRYPTION_ALGORITHM
+}
+
+#endif // TC_HEADER_Encryption_EncryptionAlgorithm