VeraCrypt
aboutsummaryrefslogtreecommitdiff
path: root/src/Volume/VolumePassword.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/Volume/VolumePassword.cpp')
-rw-r--r--src/Volume/VolumePassword.cpp167
1 files changed, 167 insertions, 0 deletions
diff --git a/src/Volume/VolumePassword.cpp b/src/Volume/VolumePassword.cpp
new file mode 100644
index 00000000..86b17e22
--- /dev/null
+++ b/src/Volume/VolumePassword.cpp
@@ -0,0 +1,167 @@
+/*
+ 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.
+*/
+
+#include "VolumePassword.h"
+#include "Platform/SerializerFactory.h"
+#include "Platform/StringConverter.h"
+
+namespace TrueCrypt
+{
+ VolumePassword::VolumePassword () : PasswordSize (0), Unportable (false)
+ {
+ AllocateBuffer ();
+ }
+
+ VolumePassword::VolumePassword (const char *password, size_t size)
+ {
+ Set ((const byte *) password, size);
+ }
+
+ VolumePassword::VolumePassword (const byte *password, size_t size)
+ {
+ Set (password, size);
+ }
+
+ VolumePassword::VolumePassword (const wchar_t *password, size_t charCount)
+ {
+ Set (password, charCount);
+ }
+
+ VolumePassword::VolumePassword (const wstring &password)
+ {
+ Set (password.c_str(), password.size());
+ }
+
+ VolumePassword::~VolumePassword ()
+ {
+ }
+
+ void VolumePassword::AllocateBuffer ()
+ {
+ if (!PasswordBuffer.IsAllocated ())
+ PasswordBuffer.Allocate (MaxSize);
+ }
+
+ void VolumePassword::CheckPortability () const
+ {
+ if (Unportable || !IsPortable())
+ throw UnportablePassword (SRC_POS);
+ }
+
+ void VolumePassword::Deserialize (shared_ptr <Stream> stream)
+ {
+ Serializer sr (stream);
+ uint64 passwordSize;
+ sr.Deserialize ("PasswordSize", passwordSize);
+ PasswordSize = static_cast <size_t> (passwordSize);
+ sr.Deserialize ("PasswordBuffer", BufferPtr (PasswordBuffer));
+
+ Buffer wipeBuffer (128 * 1024);
+ sr.Deserialize ("WipeData", wipeBuffer);
+ }
+
+ bool VolumePassword::IsPortable () const
+ {
+ for (size_t i = 0; i < PasswordSize; i++)
+ {
+ if (PasswordBuffer[i] >= 0x7f || PasswordBuffer[i] < 0x20)
+ return false;
+ }
+ return true;
+ }
+
+ void VolumePassword::Serialize (shared_ptr <Stream> stream) const
+ {
+ Serializable::Serialize (stream);
+ Serializer sr (stream);
+ sr.Serialize ("PasswordSize", static_cast <uint64> (PasswordSize));
+ sr.Serialize ("PasswordBuffer", ConstBufferPtr (PasswordBuffer));
+
+ // Wipe password from an eventual pipe buffer
+ Buffer wipeBuffer (128 * 1024);
+ wipeBuffer.Zero();
+ sr.Serialize ("WipeData", ConstBufferPtr (wipeBuffer));
+ }
+
+ void VolumePassword::Set (const byte *password, size_t size)
+ {
+ AllocateBuffer ();
+
+ if (size > MaxSize)
+ throw PasswordTooLong (SRC_POS);
+
+ PasswordBuffer.CopyFrom (ConstBufferPtr (password, size));
+ PasswordSize = size;
+
+ Unportable = !IsPortable();
+ }
+
+ void VolumePassword::Set (const wchar_t *password, size_t charCount)
+ {
+ if (charCount > MaxSize)
+ throw PasswordTooLong (SRC_POS);
+
+ union Conv
+ {
+ byte b[sizeof (wchar_t)];
+ wchar_t c;
+ };
+
+ Conv conv;
+ conv.c = L'A';
+
+ int lsbPos = -1;
+ for (size_t i = 0; i < sizeof (conv.b); ++i)
+ {
+ if (conv.b[i] == L'A')
+ {
+ lsbPos = i;
+ break;
+ }
+ }
+
+ if (lsbPos == -1)
+ throw ParameterIncorrect (SRC_POS);
+
+ bool unportable = false;
+ byte passwordBuf[MaxSize];
+ for (size_t i = 0; i < charCount; ++i)
+ {
+ conv.c = password[i];
+ passwordBuf[i] = conv.b[lsbPos];
+ for (int j = 0; j < (int) sizeof (wchar_t); ++j)
+ {
+ if (j != lsbPos && conv.b[j] != 0)
+ unportable = true;
+ }
+ }
+
+ Set (passwordBuf, charCount);
+
+ if (unportable)
+ Unportable = true;
+ }
+
+ void VolumePassword::Set (const ConstBufferPtr &password)
+ {
+ Set (password, password.Size());
+ }
+
+ void VolumePassword::Set (const VolumePassword &password)
+ {
+ Set (password.DataPtr(), password.Size());
+ }
+
+ TC_SERIALIZER_FACTORY_ADD_CLASS (VolumePassword);
+
+#define TC_EXCEPTION(TYPE) TC_SERIALIZER_FACTORY_ADD(TYPE)
+#undef TC_EXCEPTION_NODECL
+#define TC_EXCEPTION_NODECL(TYPE) TC_SERIALIZER_FACTORY_ADD(TYPE)
+
+ TC_SERIALIZER_FACTORY_ADD_EXCEPTION_SET (PasswordException);
+}