1 files changed, 92 insertions, 0 deletions
diff --git a/src/Volume/VolumePassword.h b/src/Volume/VolumePassword.h
new file mode 100644
index 00000000..e43c50ec
--- /dev/null
+++ b/src/Volume/VolumePassword.h
@@ -0,0 +1,92 @@
+/*
+ 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_Password
+#define TC_HEADER_Encryption_Password
+
+#include "Platform/Platform.h"
+#include "Platform/Serializable.h"
+
+namespace TrueCrypt
+{
+	class VolumePassword : public Serializable
+	{
+	public:
+		VolumePassword ();
+		VolumePassword (const byte *password, size_t size);   
+		VolumePassword (const char *password, size_t size);
+		VolumePassword (const wchar_t *password, size_t charCount);
+		VolumePassword (const wstring &password);
+		VolumePassword (const VolumePassword &password) { Set (password); }
+		virtual ~VolumePassword ();
+
+		bool operator== (const VolumePassword &other) const { return ConstBufferPtr (DataPtr(), Size()).IsDataEqual (ConstBufferPtr (other.DataPtr(), other.Size())); }
+		bool operator!= (const VolumePassword &other) const { return !(*this == other); }
+		VolumePassword &operator= (const VolumePassword &password) { Set (password); return *this; }
+
+		operator BufferPtr () const { return BufferPtr (PasswordBuffer); }
+
+		void CheckPortability () const;
+		byte *DataPtr () const { return PasswordBuffer; }
+		bool IsEmpty () const { return PasswordSize == 0; }
+		size_t Size () const { return PasswordSize; }
+		void Set (const byte *password, size_t size);
+		void Set (const wchar_t *password, size_t charCount);
+		void Set (const ConstBufferPtr &password);
+		void Set (const VolumePassword &password);
+
+		TC_SERIALIZABLE (VolumePassword);
+
+		static const size_t MaxSize = 64;
+		static const size_t WarningSizeThreshold = 12;
+
+	protected:
+		void AllocateBuffer ();
+		bool IsPortable () const;
+
+		SecureBuffer PasswordBuffer;
+
+		size_t PasswordSize;
+		bool Unportable;
+	};
+
+	struct PasswordException : public Exception
+	{
+	protected:
+		PasswordException () { }
+		PasswordException (const string &message) : Exception (message) { }
+		PasswordException (const string &message, const wstring &subject) : Exception (message, subject) { }
+	};
+
+	TC_EXCEPTION_DECL (PasswordIncorrect, PasswordException);
+	TC_EXCEPTION_DECL (PasswordKeyfilesIncorrect, PasswordIncorrect);
+	TC_EXCEPTION_DECL (PasswordOrKeyboardLayoutIncorrect, PasswordException);
+	TC_EXCEPTION_DECL (PasswordOrMountOptionsIncorrect, PasswordException);
+	TC_EXCEPTION_DECL (ProtectionPasswordIncorrect, PasswordIncorrect);
+	TC_EXCEPTION_DECL (ProtectionPasswordKeyfilesIncorrect, PasswordIncorrect);
+
+#define TC_EXCEPTION(NAME) TC_EXCEPTION_DECL(NAME,PasswordException)
+
+#undef TC_EXCEPTION_SET
+#define TC_EXCEPTION_SET \
+	TC_EXCEPTION_NODECL (PasswordIncorrect); \
+	TC_EXCEPTION_NODECL (PasswordKeyfilesIncorrect); \
+	TC_EXCEPTION_NODECL (PasswordOrKeyboardLayoutIncorrect); \
+	TC_EXCEPTION_NODECL (PasswordOrMountOptionsIncorrect); \
+	TC_EXCEPTION_NODECL (ProtectionPasswordIncorrect); \
+	TC_EXCEPTION_NODECL (ProtectionPasswordKeyfilesIncorrect); \
+	TC_EXCEPTION (PasswordEmpty); \
+	TC_EXCEPTION (PasswordTooLong); \
+	TC_EXCEPTION (UnportablePassword);
+
+	TC_EXCEPTION_SET;
+
+#undef TC_EXCEPTION
+}
+
+#endif // TC_HEADER_Encryption_Password