- Modifications and additions to the original source code (contained in this file)

-*/

-

-#include "Buffer.h"

-#include "Exception.h"

-

-namespace VeraCrypt

-{

- Buffer::Buffer () : DataPtr (nullptr), DataSize (0)

- {

- }

-

- Buffer::Buffer (size_t size) : DataPtr (nullptr), DataSize (0)

- {

- Allocate (size);

- }

-

- Buffer::~Buffer ()

- {

- if (DataPtr != nullptr)

- Free ();

- }

-

- void Buffer::Allocate (size_t size)

- {

- if (size < 1)

- throw ParameterIncorrect (SRC_POS);

-

- if (DataPtr != nullptr)

- {

- if (DataSize == size)

- return;

- Free();

- }

-

- try

- {

- DataPtr = static_cast<byte *> (Memory::Allocate (size));

- DataSize = size;

- }

- catch (...)

- {

- DataPtr = nullptr;

- DataSize = 0;

- throw;

- }

- }

-

- void Buffer::CopyFrom (const ConstBufferPtr &bufferPtr)

- {

- if (!IsAllocated ())

- {

- if (bufferPtr.Size())

- Allocate (bufferPtr.Size());

- }

- else if (bufferPtr.Size() > DataSize)

- throw ParameterTooLarge (SRC_POS);

-

- if (bufferPtr.Size())

- Memory::Copy (DataPtr, bufferPtr.Get(), bufferPtr.Size());

- }

-

- void Buffer::Erase ()

- {

- if (DataSize > 0)

- Memory::Erase (DataPtr, DataSize);

- }

-

- void Buffer::Free ()

- {

- if (DataPtr == nullptr)

- throw NotInitialized (SRC_POS);

-

- Memory::Free (DataPtr);

- DataPtr = nullptr;

- DataSize = 0;

- }

-

- BufferPtr Buffer::GetRange (size_t offset, size_t size) const

- {

- if (offset + size > DataSize)

- throw ParameterIncorrect (SRC_POS);

-

- return BufferPtr (DataPtr + offset, size);

- }

-

- void Buffer::Zero ()

- {

- if (DataSize > 0)

- Memory::Zero (DataPtr, DataSize);

- }

-

- SecureBuffer::SecureBuffer (size_t size)

- {

- Allocate (size);

- }

-

- SecureBuffer::~SecureBuffer ()

- {

- if (DataPtr != nullptr && DataSize != 0)

- Free ();

- }

-

- void SecureBuffer::Allocate (size_t size)

- {

- Buffer::Allocate (size);

- }

-

- void SecureBuffer::Free ()

- {

- if (DataPtr == nullptr)

- throw NotInitialized (SRC_POS);

-

- Erase ();

- Buffer::Free ();

- }

-

- void BufferPtr::CopyFrom (const ConstBufferPtr &bufferPtr) const

- {

- if (bufferPtr.Size() > DataSize)

- throw ParameterTooLarge (SRC_POS);

-

- Memory::Copy (DataPtr, bufferPtr.Get(), bufferPtr.Size());

- }

-

- BufferPtr BufferPtr::GetRange (size_t offset, size_t size) const

- {

- if (offset + size > DataSize)

- throw ParameterIncorrect (SRC_POS);

-

- return BufferPtr (DataPtr + offset, size);

- }

-

- ConstBufferPtr ConstBufferPtr::GetRange (size_t offset, size_t size) const

- {

- if (offset + size > DataSize)

- throw ParameterIncorrect (SRC_POS);

-

- return ConstBufferPtr (DataPtr + offset, size);

- }

-}

- Modifications and additions to the original source code (contained in this file)

-*/

-

-#ifndef TC_HEADER_Platform_Buffer

-#define TC_HEADER_Platform_Buffer

-

-#include "PlatformBase.h"

-#include "Memory.h"

-

-namespace VeraCrypt

-{

-

- class ConstBufferPtr

- {

- public:

- ConstBufferPtr ()

- : DataPtr (nullptr), DataSize (0) { }

- ConstBufferPtr (const byte *data, size_t size)

- : DataPtr (data), DataSize (size) { }

- virtual ~ConstBufferPtr () { }

-

- operator const byte * () const { return DataPtr; }

-

- bool IsDataEqual (const ConstBufferPtr &other) const { return Memory::Compare (DataPtr, DataSize, other.DataPtr, other.DataSize) == 0; }

- const byte *Get () const { return DataPtr; }

- ConstBufferPtr GetRange (size_t offset, size_t size) const;

- void Set (const byte *data, size_t size) { DataPtr = data; DataSize = size; }

- size_t Size () const { return DataSize; }

-

- protected:

- const byte *DataPtr;

- size_t DataSize;

- };

-

-

- class BufferPtr

- {

- public:

- BufferPtr ()

- : DataPtr (nullptr), DataSize (0) { }

- BufferPtr (byte *data, size_t size)

- : DataPtr (data), DataSize (size) { }

- virtual ~BufferPtr () { }

-

- operator byte * () const { return DataPtr; }

- void CopyFrom (const ConstBufferPtr &bufferPtr) const;

- void Erase () const { Zero(); }

- byte *Get () const { return DataPtr; }

- BufferPtr GetRange (size_t offset, size_t size) const;

- void Set (byte *data, size_t size) { DataPtr = data; DataSize = size; }

- size_t Size () const { return DataSize; }

- void Zero () const { Memory::Zero (DataPtr, DataSize); }

-

- operator ConstBufferPtr () const { return ConstBufferPtr (DataPtr, DataSize); }

-

- protected:

- byte *DataPtr;

- size_t DataSize;

- };

-

- class Buffer

- {

- public:

- Buffer ();

- Buffer (size_t size);

- Buffer (const ConstBufferPtr &bufferPtr) { CopyFrom (bufferPtr); }

- virtual ~Buffer ();

-

- virtual void Allocate (size_t size);

- virtual void CopyFrom (const ConstBufferPtr &bufferPtr);

- virtual byte *Ptr () const { return DataPtr; }

- virtual void Erase ();

- virtual void Free ();

- virtual BufferPtr GetRange (size_t offset, size_t size) const;

- virtual size_t Size () const { return DataSize; }

- virtual bool IsAllocated () const { return DataSize != 0; }

- virtual void Zero ();

-

- virtual operator byte * () const { return DataPtr; }

- virtual operator BufferPtr () const { return BufferPtr (DataPtr, DataSize); }

- virtual operator ConstBufferPtr () const { return ConstBufferPtr (DataPtr, DataSize); }

-

- protected:

- byte *DataPtr;

- size_t DataSize;

-

- private:

- Buffer (const Buffer &);

- Buffer &operator= (const Buffer &);

- };

-

- class SecureBuffer : public Buffer

- {

- public:

- SecureBuffer () { }

- SecureBuffer (size_t size);

- SecureBuffer (const ConstBufferPtr &bufferPtr) { CopyFrom (bufferPtr); }

- virtual ~SecureBuffer ();

-

- virtual void Allocate (size_t size);

- virtual void Free ();

-

- private:

- SecureBuffer (const SecureBuffer &);

- SecureBuffer &operator= (const SecureBuffer &);

- };

-

-}

-

-#endif // TC_HEADER_Platform_Buffer

- Modifications and additions to the original source code (contained in this file)

-*/

-

-#ifndef TC_HEADER_Platform_Directory

-#define TC_HEADER_Platform_Directory

-

-#include "PlatformBase.h"

-#include "FilesystemPath.h"

-

-namespace VeraCrypt

-{

- class Directory

- {

- public:

- static void Create (const DirectoryPath &path);

- static DirectoryPath AppendSeparator (const DirectoryPath &path);

- static FilePathList GetFilePaths (const DirectoryPath &path = L".", bool regularFilesOnly = true);

-

- private:

- Directory ();

- };

-}

-

-#endif // TC_HEADER_Platform_Directory

- Modifications and additions to the original source code (contained in this file)

-*/

-

-#include "Event.h"

-

-namespace VeraCrypt

-{

- void Event::Connect (const EventConnectorBase &connector)

- {

- ScopeLock lock (HandlersMutex);

- ConnectedHandlers.push_back (shared_ptr <EventConnectorBase> (connector.CloneNew()));

- }

-

- void Event::Disconnect (void *handler)

- {

- ScopeLock lock (HandlersMutex);

-

- EventHandlerList newConnectedHandlers;

- foreach (shared_ptr <EventConnectorBase> h, ConnectedHandlers)

- {

- if (h->GetHandler() != handler)

- newConnectedHandlers.push_back (h);

- }

-

- ConnectedHandlers = newConnectedHandlers;

- }

-

- void Event::Raise ()

- {

- EventArgs args;

- Raise (args);

- }

-

- void Event::Raise (EventArgs &args)

- {

- ScopeLock lock (HandlersMutex);

- foreach_ref (EventConnectorBase &handler, ConnectedHandlers)

- {

- handler (args);

- }

- }

-}

- Modifications and additions to the original source code (contained in this file)

-*/

-

-#ifndef TC_HEADER_Platform_Event

-#define TC_HEADER_Platform_Event

-

-#include "PlatformBase.h"

-#include "ForEach.h"

-#include "Mutex.h"

-#include "SharedPtr.h"

-

-namespace VeraCrypt

-{

- struct EventArgs

- {

- virtual ~EventArgs () { }

- };

-

- class EventConnectorBase

- {

- public:

- virtual ~EventConnectorBase () { }

- virtual void operator() (EventArgs &args) = 0;

-

- virtual EventConnectorBase *CloneNew () const = 0;

- virtual void *GetHandler () const = 0;

- };

-

- typedef list < shared_ptr <EventConnectorBase> > EventHandlerList;

-

- template <class T>

- class EventConnector : public EventConnectorBase

- {

- public:

- typedef void (T::*EventHandlerFunction) (EventArgs &);

-

- EventConnector (T *handler, EventHandlerFunction function)

- : Handler (handler), Function (function) { }

-

- virtual void operator() (EventArgs &args) { (Handler->*Function) (args); }

-

- virtual EventConnectorBase *CloneNew () const { return new EventConnector <T> (*this); }

- virtual void *GetHandler () const { return Handler; }

-

- protected:

- T *Handler;

- EventHandlerFunction Function;

- };

-

- class Event

- {

- public:

- Event () { }

- virtual ~Event () { }

-

- void Connect (const EventConnectorBase &connector);

- void Disconnect (void *handler);

- void Raise ();

- void Raise (EventArgs &args);

-

- protected:

- EventHandlerList ConnectedHandlers;

- Mutex HandlersMutex;

-

- private:

- Event (const Event &);

- Event &operator= (const Event &);

- };

-

- struct ExceptionEventArgs : public EventArgs

- {

- ExceptionEventArgs (exception &ex) : mException (ex) { }

- exception &mException;

-

- private:

- ExceptionEventArgs (const ExceptionEventArgs &);

- ExceptionEventArgs &operator= (const ExceptionEventArgs &);

- };

-}

-

-#endif // TC_HEADER_Platform_Event

- Modifications and additions to the original source code (contained in this file)

-*/

-

-#include "Exception.h"

-#include "SerializerFactory.h"

-

-namespace VeraCrypt

-{

- void Exception::Deserialize (shared_ptr <Stream> stream)

- {

- Serializer sr (stream);

- sr.Deserialize ("Message", Message);

- sr.Deserialize ("Subject", Subject);

- }

-

- void Exception::Serialize (shared_ptr <Stream> stream) const

- {

- Serializable::Serialize (stream);

- Serializer sr (stream);

- sr.Serialize ("Message", Message);

- sr.Serialize ("Subject", Subject);

- }

-

- void ExecutedProcessFailed::Deserialize (shared_ptr <Stream> stream)

- {

- Exception::Deserialize (stream);

- Serializer sr (stream);

- sr.Deserialize ("Command", Command);

- sr.Deserialize ("ExitCode", ExitCode);

- sr.Deserialize ("ErrorOutput", ErrorOutput);

- }

-

- void ExecutedProcessFailed::Serialize (shared_ptr <Stream> stream) const

- {

- Exception::Serialize (stream);

- Serializer sr (stream);

- sr.Serialize ("Command", Command);

- sr.Serialize ("ExitCode", ExitCode);

- sr.Serialize ("ErrorOutput", ErrorOutput);

- }

-

-#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 (Exception);

-}

- Modifications and additions to the original source code (contained in this file)

-*/

-

-#ifndef TC_HEADER_Platform_Exception

-#define TC_HEADER_Platform_Exception

-

-#include <exception>

-#include "PlatformBase.h"

-#include "Serializable.h"

-

-namespace VeraCrypt

-{

-#define TC_SERIALIZABLE_EXCEPTION(TYPE) TC_SERIALIZABLE (TYPE); \

- virtual Exception *CloneNew () { return new TYPE (*this); } \

- virtual void Throw () const { throw *this; }

-

- struct Exception : public exception, public Serializable

- {

- public:

- Exception () { }

- Exception (const string &message) : Message (message) { }

- Exception (const string &message, const wstring &subject) : Message (message), Subject (subject) { }

- virtual ~Exception () throw () { }

-

- TC_SERIALIZABLE_EXCEPTION (Exception);

-

- virtual const char *what () const throw () { return Message.c_str(); }

- virtual const wstring &GetSubject() const { return Subject; }

-

- protected:

- string Message;

- wstring Subject;

- };

-

- struct ExecutedProcessFailed : public Exception

- {

- ExecutedProcessFailed () { }

- ExecutedProcessFailed (const string &message, const string &command, int exitCode, const string &errorOutput)

- : Exception (message), Command (command), ExitCode (exitCode), ErrorOutput (errorOutput) { }

- virtual ~ExecutedProcessFailed () throw () { }

-

- TC_SERIALIZABLE_EXCEPTION (ExecutedProcessFailed);

-

- string GetCommand () const { return Command; }

- int64 GetExitCode () const { return ExitCode; }

- string GetErrorOutput () const { return ErrorOutput; }

-

- protected:

- string Command;

- int64 ExitCode;

- string ErrorOutput;

- };

-

-#define TC_EXCEPTION_DECL(NAME,BASE) \

- struct NAME : public BASE \

- { \

- NAME () { } \

- NAME (const string &message) : BASE (message) { } \

- NAME (const string &message, const wstring &subject) : BASE (message, subject) { } \

- virtual Exception *CloneNew () { return new NAME (*this); } \

- static Serializable *GetNewSerializable () { return new NAME (); } \

- virtual void Throw () const { throw *this; } \

- }

-

-#define TC_EXCEPTION_NODECL(dummy) //

-#define TC_EXCEPTION(NAME) TC_EXCEPTION_DECL(NAME,Exception)

-

-#ifdef TC_EXCEPTION_SET

-#undef TC_EXCEPTION_SET

-#endif

-#define TC_EXCEPTION_SET \

- TC_EXCEPTION_NODECL (Exception); \

- TC_EXCEPTION_NODECL (ExecutedProcessFailed); \

- TC_EXCEPTION (AlreadyInitialized); \

- TC_EXCEPTION (AssertionFailed); \

- TC_EXCEPTION (ExternalException); \

- TC_EXCEPTION (InsufficientData); \

- TC_EXCEPTION (NotApplicable); \

- TC_EXCEPTION (NotImplemented); \

- TC_EXCEPTION (NotInitialized); \

- TC_EXCEPTION (ParameterIncorrect); \

- TC_EXCEPTION (ParameterTooLarge); \

- TC_EXCEPTION (PartitionDeviceRequired); \

- TC_EXCEPTION (StringConversionFailed); \

- TC_EXCEPTION (TestFailed); \

- TC_EXCEPTION (TimeOut); \

- TC_EXCEPTION (UnsupportedTrueCryptFormat); \

- TC_EXCEPTION (UserAbort)

-

- TC_EXCEPTION_SET;

-

-#undef TC_EXCEPTION

-}

-

-#ifdef assert

-# undef assert

-#endif

-

-#ifdef DEBUG

-# define assert(condition) do { if (!(condition)) throw AssertionFailed (SRC_POS); } while (false)

-#else

-# define assert(condition) ((void) 0)

-#endif

-

-#endif // TC_HEADER_Platform_Exception

- Modifications and additions to the original source code (contained in this file)

-*/

-

-#ifndef TC_HEADER_Platform_File

-#define TC_HEADER_Platform_File

-

-#include "PlatformBase.h"

-#include "Buffer.h"

-#include "FilesystemPath.h"

-#include "SystemException.h"

-

-namespace VeraCrypt

-{

- class File

- {

- public:

- enum FileOpenMode

- {

- CreateReadWrite,

- CreateWrite,

- OpenRead,

- OpenWrite,

- OpenReadWrite

- };

-

- enum FileShareMode

- {

- ShareNone,

- ShareRead,

- ShareReadWrite,

- ShareReadWriteIgnoreLock

- };

-

- enum FileOpenFlags

- {

- // Bitmap

- FlagsNone = 0,

- PreserveTimestamps = 1 << 0,

- DisableWriteCaching = 1 << 1

- };

-

-#ifdef TC_WINDOWS

- typedef FILE* SystemFileHandleType;

-#else

- typedef int SystemFileHandleType;

-#endif

-

- File () : FileIsOpen (false), SharedHandle (false) { }

- virtual ~File ();

-

- void AssignSystemHandle (SystemFileHandleType openFileHandle, bool sharedHandle = true)

- {

- if (FileIsOpen)

- Close();

- FileHandle = openFileHandle;

- FileIsOpen = true;

- SharedHandle = sharedHandle;

- }

-

- void Close ();

- static void Copy (const FilePath &sourcePath, const FilePath &destinationPath, bool preserveTimestamps = true);

- void Delete ();

- void Flush () const;

- uint32 GetDeviceSectorSize () const;

- static size_t GetOptimalReadSize () { return OptimalReadSize; }

- static size_t GetOptimalWriteSize () { return OptimalWriteSize; }

- uint64 GetPartitionDeviceStartOffset () const;

- bool IsOpen () const { return FileIsOpen; }

- FilePath GetPath () const;

- uint64 Length () const;

- void Open (const FilePath &path, FileOpenMode mode = OpenRead, FileShareMode shareMode = ShareReadWrite, FileOpenFlags flags = FlagsNone);

- uint64 Read (const BufferPtr &buffer) const;

- void ReadCompleteBuffer (const BufferPtr &buffer) const;

- uint64 ReadAt (const BufferPtr &buffer, uint64 position) const;

- void SeekAt (uint64 position) const;

- void SeekEnd (int ofset) const;

- void Write (const ConstBufferPtr &buffer) const;

- void Write (const ConstBufferPtr &buffer, size_t length) const { Write (buffer.GetRange (0, length)); }

- void WriteAt (const ConstBufferPtr &buffer, uint64 position) const;

-

- protected:

- void ValidateState () const;

-

- static const size_t OptimalReadSize = 256 * 1024;

- static const size_t OptimalWriteSize = 256 * 1024;

-

- bool FileIsOpen;

- FileOpenFlags mFileOpenFlags;

- bool SharedHandle;

- FilePath Path;

- SystemFileHandleType FileHandle;

-

-#ifdef TC_WINDOWS

-#else

- time_t AccTime;

- time_t ModTime;

-#endif

-

- private:

- File (const File &);

- File &operator= (const File &);

- };

-}

-

-#endif // TC_HEADER_Platform_File

- Modifications and additions to the original source code (contained in this file)

-*/

-

-#include "File.h"

-#ifdef TC_UNIX

-#include <sys/types.h>

-#include <sys/stat.h>

-#include <unistd.h>

-#include <utime.h>

-#endif

-

-namespace VeraCrypt

-{

- File::~File ()

- {

- try

- {

- if (FileIsOpen)

- Close();

- }

- catch (...) { }

- }

-

- void File::Copy (const FilePath &sourcePath, const FilePath &destinationPath, bool preserveTimestamps)

- {

- File source;

- source.Open (sourcePath);

-

- File destination;

- destination.Open (destinationPath, CreateWrite);

-

- SecureBuffer buffer (OptimalReadSize);

- uint64 len;

-

- while ((len = source.Read (buffer)) > 0)

- {

- destination.Write (buffer, static_cast <size_t> (len));

- }

-

- if (preserveTimestamps)

- {

- destination.Flush();

-#ifndef TC_WINDOWS

- struct stat statData;

- throw_sys_sub_if (stat (string (sourcePath).c_str(), &statData) == -1, wstring (sourcePath));

-

- struct utimbuf u;

- u.actime = statData.st_atime;

- u.modtime = statData.st_mtime;

- throw_sys_sub_if (utime (string (destinationPath).c_str(), &u) == -1, wstring (destinationPath));

-#endif

- }

- }

-

- FilePath File::GetPath () const

- {

- if_debug (ValidateState());

- return Path;

- }

-

- void File::ReadCompleteBuffer (const BufferPtr &buffer) const

- {

- size_t dataLeft = buffer.Size();

- size_t offset = 0;

-

- while (dataLeft > 0)

- {

- size_t dataRead = static_cast <size_t> (Read (buffer.GetRange (offset, dataLeft)));

- if (dataRead == 0)

- throw InsufficientData (SRC_POS);

-

- dataLeft -= dataRead;

- offset += dataRead;

- }

- }

-

- void File::ValidateState () const

- {

- if (!FileIsOpen)

- throw NotInitialized (SRC_POS);

- }

-}

- Modifications and additions to the original source code (contained in this file)

-*/

-

-#ifndef TC_HEADER_Platform_FileStream

-#define TC_HEADER_Platform_FileStream

-

-#include "PlatformBase.h"

-#include "File.h"

-#include "SharedPtr.h"

-#include "Stream.h"

-

-namespace VeraCrypt

-{

- class FileStream : public Stream

- {

- public:

- FileStream (shared_ptr <File> file) : DataFile (file) { }

- FileStream (File::SystemFileHandleType openFileHandle) { DataFile.reset (new File ()); DataFile->AssignSystemHandle (openFileHandle); }

- virtual ~FileStream () { }

-

- virtual uint64 Read (const BufferPtr &buffer)

- {

- return DataFile->Read (buffer);

- }

-

- virtual void ReadCompleteBuffer (const BufferPtr &buffer)

- {

- DataFile->ReadCompleteBuffer (buffer);

- }

-

- virtual string ReadToEnd ()

- {

- string str;

- vector <char> buffer (4096);

- uint64 len;

-

- while ((len = DataFile->Read (BufferPtr (reinterpret_cast <byte *> (&buffer[0]), buffer.size()))) > 0)

- str.insert (str.end(), buffer.begin(), buffer.begin() + static_cast <int> (len));

-

- return str;

- }

-

- virtual void Write (const ConstBufferPtr &data)

- {

- DataFile->Write (data);

- }

-

- protected:

- shared_ptr <File> DataFile;

- };

-}

-

-#endif // TC_HEADER_Platform_FileStream

- Modifications and additions to the original source code (contained in this file)

-*/

-

-#ifndef TC_HEADER_Platform_FilesystemPath

-#define TC_HEADER_Platform_FilesystemPath

-

-#include "PlatformBase.h"

-#include "Platform/User.h"

-#include "SharedPtr.h"

-#include "StringConverter.h"

-

-namespace VeraCrypt

-{

- struct FilesystemPathType

- {

- enum Enum

- {

- Unknown,

- File,

- Directory,

- SymbolickLink,

- BlockDevice,

- CharacterDevice

- };

- };

-

- class FilesystemPath

- {

- public:

- FilesystemPath () { }

- FilesystemPath (const char *path) : Path (StringConverter::ToWide (path)) { }

- FilesystemPath (string path) : Path (StringConverter::ToWide (path)) { }

- FilesystemPath (const wchar_t *path) : Path (path) { }

- FilesystemPath (wstring path) : Path (path) { }

- virtual ~FilesystemPath () { }

-

- bool operator== (const FilesystemPath &other) const { return Path == other.Path; }

- bool operator!= (const FilesystemPath &other) const { return Path != other.Path; }

- operator string () const { return StringConverter::ToSingle (Path); }

- operator wstring () const { return Path; }

-

- void Delete () const;

- UserId GetOwner () const;

- FilesystemPathType::Enum GetType () const;

- bool IsBlockDevice () const throw () { try { return GetType() == FilesystemPathType::BlockDevice; } catch (...) { return false; }; }

- bool IsCharacterDevice () const throw () { try { return GetType() == FilesystemPathType::CharacterDevice; } catch (...) { return false; }; }

- bool IsDevice () const throw () { return IsBlockDevice() || IsCharacterDevice(); }

- bool IsDirectory () const throw () { try { return GetType() == FilesystemPathType::Directory; } catch (...) { return false; } }

- bool IsEmpty () const throw () { try { return Path.empty(); } catch (...) { return false; } }

- bool IsFile () const throw () { try { return GetType() == FilesystemPathType::File; } catch (...) { return false; } }

- FilesystemPath ToBaseName () const;

- FilesystemPath ToHostDriveOfPartition () const;

-

- static const int MaxSize = 260;

-

- protected:

- wstring Path;

- };

-

- typedef FilesystemPath DevicePath;

- typedef FilesystemPath DirectoryPath;

- typedef FilesystemPath FilePath;

-

- typedef list < shared_ptr <DirectoryPath> > DirectoryPathList;

- typedef list < shared_ptr <FilePath> > FilePathList;

-}

-

-#endif // TC_HEADER_Platform_FilesystemPath

- Modifications and additions to the original source code (contained in this file)

-*/

-

-#ifndef TC_HEADER_Platform_Finally

-#define TC_HEADER_Platform_Finally

-

-#include "PlatformBase.h"

-

-// Execute code when leaving scope

-#define finally_do(code) \

-struct TC_JOIN(Finally,__LINE__) \

-{ \

- TC_JOIN(~Finally,__LINE__) () { try { code } catch (...) { } } \

-} \

-TC_UNUSED_VAR \

-TC_JOIN(finally,__LINE__)

-

-// Execute code with argument 'finally_arg' when leaving scope

-#define finally_do_arg(argType, arg, code) \

-struct TC_JOIN(Finally,__LINE__) \

-{ \

- TC_JOIN(Finally,__LINE__) (argType a) : finally_arg (a) { } \

- TC_JOIN(~Finally,__LINE__) () { try { code } catch (...) { } } \

- argType finally_arg; \

-} \

-TC_UNUSED_VAR \

-TC_JOIN(finally,__LINE__) (arg)

-

-#define finally_do_arg2(argType, arg, argType2, arg2, code) \

-struct TC_JOIN(Finally,__LINE__) \

-{ \

- TC_JOIN(Finally,__LINE__) (argType a, argType2 a2) : finally_arg (a), finally_arg2 (a2) { } \

- TC_JOIN(~Finally,__LINE__) () { try { code } catch (...) { } } \

- argType finally_arg; \

- argType2 finally_arg2; \

-} \

-TC_UNUSED_VAR \

-TC_JOIN(finally,__LINE__) (arg, arg2)

-

-

-#endif // TC_HEADER_Platform_Finally

- Modifications and additions to the original source code (contained in this file)

-*/

-

-#ifndef TC_HEADER_Platform_ForEach

-#define TC_HEADER_Platform_ForEach

-

-namespace VeraCrypt

-{

- class ForEach

- {

- public:

- struct Container

- {

- Container () : InnerContinue (true), InnerEndCondition (false) { }

- virtual ~Container () { }

-

- void Continue () const { InnerContinue = true; }

- bool InnerIsNotEnd () const { return InnerEndCondition = !InnerEndCondition; }

- virtual bool IsNotEnd () const = 0;

- virtual void Next () const = 0;

-

- mutable bool InnerContinue;

- mutable bool InnerEndCondition;

- };

-

- protected:

- template <class T>

- struct ContainerForward : Container

- {

- ContainerForward (const T &container)

- : ContainerCopy (container), EndIterator (ContainerCopy.end()), Iterator (ContainerCopy.begin()) { }

-

- virtual bool IsNotEnd () const { bool r = InnerContinue && Iterator != EndIterator; InnerContinue = false; return r; }

- virtual void Next () const { ++Iterator; }

-

- const T ContainerCopy; // Support for temporary objects

- typename T::const_iterator EndIterator;

- mutable typename T::const_iterator Iterator;

-

- private:

- ContainerForward &operator= (const ContainerForward &);

- };

-

- template <class T>

- struct ContainerReverse : Container

- {

- ContainerReverse (const T &container)

- : ContainerCopy (container), EndIterator (ContainerCopy.rend()), Iterator (ContainerCopy.rbegin()) { }

-

- virtual bool IsNotEnd () const { bool r = InnerContinue && Iterator != EndIterator; InnerContinue = false; return r; }

- virtual void Next () const { ++Iterator; }

-

- const T ContainerCopy;

- typename T::const_reverse_iterator EndIterator;

- mutable typename T::const_reverse_iterator Iterator;

-

- private:

- ContainerReverse &operator= (const ContainerReverse &);

- };

-

- public:

- template <class T>

- static ContainerForward <T> GetContainerForward (const T &container)

- {

- return ContainerForward <T> (container);

- }

-

- template <class T>

- static ContainerReverse <T> GetContainerReverse (const T &container)

- {

- return ContainerReverse <T> (container);

- }

-

- protected:

- template <class T>

- struct TypeWrapper { };

-

- public:

- template <class T>

- static TypeWrapper <T> ToTypeWrapper (const T &x) { return TypeWrapper <T> (); }

-

- struct TypeWrapperDummy

- {

- template <class T>

- operator TypeWrapper <T> () const { return TypeWrapper <T> (); }

- };

-

- template <class T>

- static const ContainerForward <T> &GetContainerForward (const Container &forEachContainer, const TypeWrapper <T> &)

- {

- return static_cast <const ContainerForward <T> &> (forEachContainer);

- }

-

- template <class T>

- static const ContainerReverse <T> &GetContainerReverse (const Container &forEachContainer, const TypeWrapper <T> &)

- {

- return static_cast <const ContainerReverse <T> &> (forEachContainer);

- }

- };

-}

-

-

-#define FOREACH_TEMPLATE(dereference,listType,variable,listInstance) \

- for (const ForEach::Container &forEachContainer = ForEach::GetContainer##listType (listInstance); forEachContainer.IsNotEnd(); forEachContainer.Next()) \

- for (variable = dereference(ForEach::GetContainer##listType (forEachContainer, (true ? ForEach::TypeWrapperDummy() : ForEach::ToTypeWrapper (listInstance))).Iterator); forEachContainer.InnerIsNotEnd(); forEachContainer.Continue())

-

-#define foreach(variable,listInstance) FOREACH_TEMPLATE(*, Forward, variable, listInstance)

-#define foreach_ref(variable,listInstance) FOREACH_TEMPLATE(**, Forward, variable, listInstance)

-#define foreach_reverse(variable,listInstance) FOREACH_TEMPLATE(*, Reverse, variable, listInstance)

-#define foreach_reverse_ref(variable,listInstance) FOREACH_TEMPLATE(**, Reverse, variable, listInstance)

-

-

-#endif // TC_HEADER_Platform_ForEach

- Modifications and additions to the original source code (contained in this file)

-*/

-

-#ifndef TC_HEADER_Platform_Functor

-#define TC_HEADER_Platform_Functor

-

-#include "PlatformBase.h"

-

-namespace VeraCrypt

-{

- struct Functor

- {

- virtual ~Functor () { }

- virtual void operator() () = 0;

- };

-

- struct GetStringFunctor

- {

- virtual ~GetStringFunctor () { }

- virtual void operator() (string &str) = 0;

- };

-}

-

-#endif // TC_HEADER_Platform_Functor

- Modifications and additions to the original source code (contained in this file)

-*/

-

-#include "Common/Tcdefs.h"

-#include "Memory.h"

-#include "Exception.h"

-

-namespace VeraCrypt

-{

- void *Memory::Allocate (std::size_t size)

- {

- if (size < 1)

- throw ParameterIncorrect (SRC_POS);

-

- void *bufPtr = malloc (size);

- if (!bufPtr)

- throw bad_alloc();

-

- return bufPtr;

- }

-

- int Memory::Compare (const void *memory1, size_t size1, const void *memory2, size_t size2)

- {

- if (size1 > size2)

- return 1;

- else if (size1 < size2)

- return -1;

-

- return memcmp (memory1, memory2, size1);

- }

-

- void Memory::Copy (void *memoryDestination, const void *memorySource, size_t size)

- {

- assert (memoryDestination != nullptr && memorySource != nullptr);

- memcpy (memoryDestination, memorySource, size);

- }

-

- void Memory::Erase (void *memory, size_t size)

- {

- burn (memory, size);

- }

-

- void Memory::Zero (void *memory, size_t size)

- {

- memset (memory, 0, size);

- }

-

- void Memory::Free (void *memory)

- {

- assert (memory != nullptr);

- free (memory);

- }

-}

- Modifications and additions to the original source code (contained in this file)

-*/

-

-#ifndef TC_HEADER_Platform_Memory

-#define TC_HEADER_Platform_Memory

-

-#include <new>

-#include <memory.h>

-#include "PlatformBase.h"

-

-#ifdef TC_WINDOWS

-

-# ifndef LITTLE_ENDIAN

-# define LITTLE_ENDIAN 1234

-# endif

-# ifndef BYTE_ORDER

-# define BYTE_ORDER LITTLE_ENDIAN

-# endif

-

-#elif !defined(BYTE_ORDER)

-

-# ifdef TC_MACOSX

-# include <machine/endian.h>

-# elif defined (TC_BSD)

-# include <sys/endian.h>

-# elif defined (TC_SOLARIS)

-# include <sys/types.h>

-# define LITTLE_ENDIAN 1234

-# define BIG_ENDIAN 4321

-# ifdef _BIG_ENDIAN

-# define BYTE_ORDER BIG_ENDIAN

-# else

-# define BYTE_ORDER LITTLE_ENDIAN

-# endif

-# else

-# include <endian.h>

-# endif

-

-# ifndef BYTE_ORDER

-# ifndef __BYTE_ORDER

-# error Byte ordering cannot be determined (BYTE_ORDER undefined).

-# endif

-

-# define BYTE_ORDER __BYTE_ORDER

-# endif

-

-# ifndef LITTLE_ENDIAN

-# define LITTLE_ENDIAN __LITTLE_ENDIAN

-# endif

-

-# ifndef BIG_ENDIAN

-# define BIG_ENDIAN __BIG_ENDIAN

-# endif

-

-#endif // !BYTE_ORDER

-

-#if BYTE_ORDER != BIG_ENDIAN && BYTE_ORDER != LITTLE_ENDIAN

-# error Unsupported byte ordering detected.

-#endif

-

-namespace VeraCrypt

-{

- class Memory

- {

- public:

- static void *Allocate (size_t size);

- static int Compare (const void *memory1, size_t size1, const void *memory2, size_t size2);

- static void Copy (void *memoryDestination, const void *memorySource, size_t size);

- static void Erase (void *memory, size_t size);

- static void Free (void *memory);

- static void Zero (void *memory, size_t size);

- };

-

- class Endian

- {

- public:

- static byte Big (const byte &x)

- {

- return x;

- }

-

- static uint16 Big (const uint16 &x)

- {

-#if BYTE_ORDER == BIG_ENDIAN

- return x;

-#else

- return MirrorBytes (x);

-#endif

- }

-

- static uint32 Big (const uint32 &x)

- {

-#if BYTE_ORDER == BIG_ENDIAN

- return x;

-#else

- return MirrorBytes (x);

-#endif

- }

-

- static uint64 Big (const uint64 &x)

- {

-#if BYTE_ORDER == BIG_ENDIAN

- return x;

-#else

- return MirrorBytes (x);

-#endif

- }

-

- static byte Little (const byte &x)

- {

- return x;

- }

-

- static uint16 Little (const uint16 &x)

- {

-#if BYTE_ORDER == LITTLE_ENDIAN

- return x;

-#else

- return MirrorBytes (x);

-#endif

- }

-

- static uint32 Little (const uint32 &x)

- {

-#if BYTE_ORDER == LITTLE_ENDIAN

- return x;

-#else

- return MirrorBytes (x);

-#endif

- }

-

- static uint64 Little (const uint64 &x)

- {

-#if BYTE_ORDER == LITTLE_ENDIAN

- return x;

-#else

- return MirrorBytes (x);

-#endif

- }

-

- protected:

- static uint16 MirrorBytes (const uint16 &x)

- {

- return (x << 8) | (x >> 8);

- }

-

- static uint32 MirrorBytes (const uint32 &x)

- {

- uint32 n = (byte) x;

- n <<= 8; n |= (byte) (x >> 8);

- n <<= 8; n |= (byte) (x >> 16);

- return (n << 8) | (byte) (x >> 24);

- }

-

- static uint64 MirrorBytes (const uint64 &x)

- {

- uint64 n = (byte) x;

- n <<= 8; n |= (byte) (x >> 8);

- n <<= 8; n |= (byte) (x >> 16);

- n <<= 8; n |= (byte) (x >> 24);

- n <<= 8; n |= (byte) (x >> 32);

- n <<= 8; n |= (byte) (x >> 40);

- n <<= 8; n |= (byte) (x >> 48);

- return (n << 8) | (byte) (x >> 56);

- }

- };

-}

-

-#endif // TC_HEADER_Platform_Memory

- Modifications and additions to the original source code (contained in this file)

-*/

-

-#include "Exception.h"

-#include "MemoryStream.h"

-

-namespace VeraCrypt

-{

- MemoryStream::MemoryStream (const ConstBufferPtr &data) :

- ReadPosition (0)

- {

- Data = vector <byte> (data.Size());

- BufferPtr (&Data[0], Data.size()).CopyFrom (data);

- }

-

- uint64 MemoryStream::Read (const BufferPtr &buffer)

- {

- if (Data.size() == 0)

- throw ParameterIncorrect (SRC_POS);

-

- ConstBufferPtr streamBuf (*this);

- size_t len = buffer.Size();

- if (streamBuf.Size() - ReadPosition < len)

- len = streamBuf.Size() - ReadPosition;

-

- BufferPtr(buffer).CopyFrom (streamBuf.GetRange (ReadPosition, len));

- ReadPosition += len;

- return len;

- }

-

- void MemoryStream::ReadCompleteBuffer (const BufferPtr &buffer)

- {

- if (Read (buffer) != buffer.Size())

- throw InsufficientData (SRC_POS);

- }

-

- void MemoryStream::Write (const ConstBufferPtr &data)

- {

- for (uint64 i = 0; i < data.Size(); i++)

- Data.push_back (data[i]);

- }

-}

- Modifications and additions to the original source code (contained in this file)

-*/

-

-#ifndef TC_HEADER_Platform_MemoryStream

-#define TC_HEADER_Platform_MemoryStream

-

-#include "PlatformBase.h"

-#include "Stream.h"

-

-namespace VeraCrypt

-{

- class MemoryStream : public Stream

- {

- public:

- MemoryStream () : ReadPosition (0) { }

- MemoryStream (const ConstBufferPtr &data);

- virtual ~MemoryStream () { }

-

- operator ConstBufferPtr () const { return ConstBufferPtr (&Data[0], Data.size()); }

-

- virtual uint64 Read (const BufferPtr &buffer);

- virtual void ReadCompleteBuffer (const BufferPtr &buffer);

- virtual void Write (const ConstBufferPtr &data);

-

- protected:

- vector <byte> Data;

- size_t ReadPosition;

- };

-}

-

-#endif // TC_HEADER_Platform_MemoryStream

- Modifications and additions to the original source code (contained in this file)

-*/

-

-#ifndef TC_HEADER_Platform_Mutex

-#define TC_HEADER_Platform_Mutex

-

-#ifdef TC_WINDOWS

-# include "System.h"

-#else

-# include <pthread.h>

-#endif

-#include "PlatformBase.h"

-

-namespace VeraCrypt

-{

- class Mutex

- {

-#ifdef TC_WINDOWS

- typedef CRITICAL_SECTION SystemMutex_t;

-#else

- typedef pthread_mutex_t SystemMutex_t;

-#endif

-

- public:

- Mutex ();

- ~Mutex ();

-

- SystemMutex_t *GetSystemHandle () { return &SystemMutex; }

- void Lock ();

- void Unlock ();

-

- protected:

- bool Initialized;

- SystemMutex_t SystemMutex;

-

- private:

- Mutex (const Mutex &);

- Mutex &operator= (const Mutex &);

- };

-

- class ScopeLock

- {

- public:

- ScopeLock (Mutex &mutex) : ScopeMutex (mutex) { mutex.Lock(); }

- ~ScopeLock () { ScopeMutex.Unlock(); }

-

- protected:

- Mutex &ScopeMutex;

-

- private:

- ScopeLock (const ScopeLock &);

- ScopeLock &operator= (const ScopeLock &);

- };

-}

-

-#endif // TC_HEADER_Platform_Mutex

- Modifications and additions to the original source code (contained in this file)

-*/

-

-#ifndef TC_HEADER_Platform

-#define TC_HEADER_Platform

-

-#include "PlatformBase.h"

-#include "Buffer.h"

-#include "Exception.h"

-#include "Directory.h"

-#include "Event.h"

-#include "File.h"

-#include "FilesystemPath.h"

-#include "Finally.h"

-#include "ForEach.h"

-#include "Functor.h"

-#include "Memory.h"

-#include "Mutex.h"

-#include "SharedPtr.h"

-#include "SystemException.h"

-#include "Thread.h"

-

-#endif // TC_HEADER_Platform

-#

-# Derived from source code of TrueCrypt 7.1a, which is

-# Copyright (c) 2008-2012 TrueCrypt Developers Association and which is governed

-# by the TrueCrypt License 3.0.

-#

-# Modifications and additions to the original source code (contained in this file)

-# and all other portions of this file are Copyright (c) 2013-2016 IDRIX

-# and are governed by the Apache License 2.0 the full text of which is

-# contained in the file License.txt included in VeraCrypt binary and source

-# code distribution packages.

-#

-

-OBJS := Buffer.o

-OBJS += Exception.o

-OBJS += Event.o

-OBJS += FileCommon.o

-OBJS += MemoryStream.o

-OBJS += Memory.o

-OBJS += PlatformTest.o

-OBJS += Serializable.o

-OBJS += Serializer.o

-OBJS += SerializerFactory.o

-OBJS += StringConverter.o

-OBJS += TextReader.o

-OBJS += Unix/Directory.o

-OBJS += Unix/File.o

-OBJS += Unix/FilesystemPath.o

-OBJS += Unix/Mutex.o

-OBJS += Unix/Pipe.o

-OBJS += Unix/Poller.o

-OBJS += Unix/Process.o

-OBJS += Unix/SyncEvent.o

-OBJS += Unix/SystemException.o

-OBJS += Unix/SystemInfo.o

-OBJS += Unix/SystemLog.o

-OBJS += Unix/Thread.o

-OBJS += Unix/Time.o

-

-include $(BUILD_INC)/Makefile.inc

- Modifications and additions to the original source code (contained in this file)

-*/

-

-#ifndef TC_HEADER_Platform_PlatformBase

-#define TC_HEADER_Platform_PlatformBase

-

-#include <cstddef>

-#include <list>

-#include <map>

-#include <memory>

-#include <sstream>

-#include <string>

-#include <vector>

-

-#ifndef _MSC_VER

-#include <inttypes.h>

-#endif

-

-using namespace std;

-

-#ifdef nullptr

-#undef nullptr

-#endif

-

-#if !(defined(_MSC_VER) && _MSC_VER >= 1600)

-#define nullptr 0

-#endif

-

-namespace VeraCrypt

-{

-#ifdef _MSC_VER

-# ifndef TC_INT_TYPES_DEFINED

- typedef __int8 int8;

- typedef __int16 int16;

- typedef __int32 int32;

- typedef __int64 int64;

- typedef unsigned __int8 byte;

- typedef unsigned __int16 uint16;

- typedef unsigned __int32 uint32;

- typedef unsigned __int64 uint64;

-# endif

-#else

- typedef int8_t int8;

- typedef int16_t int16;

- typedef int32_t int32;

- typedef int64_t int64;

- typedef uint8_t byte;

- typedef uint16_t uint16;

- typedef uint32_t uint32;

- typedef uint64_t uint64;

-#endif

-}

-

-#if (defined(_WIN32) || defined(_WIN64)) && !defined(TC_WINDOWS)

-# define TC_WINDOWS

-#endif

-

-#if defined(_DEBUG) && !defined(DEBUG)

-# define DEBUG

-#endif

-

-#ifndef TC_TO_STRING

-# define TC_TO_STRING2(n) #n

-# define TC_TO_STRING(n) TC_TO_STRING2(n)

-#endif

-

-#define TC_JOIN_ARGS(a,b) a##b

-#define TC_JOIN(a,b) TC_JOIN_ARGS(a,b)

-

-#ifdef __GNUC__

- template <class T> string GetFunctionName (T pos)

- {

- string s (pos);

- size_t p = s.find ('(');

- if (p == string::npos)

- return s;

- s = s.substr (0, p);

- p = s.find_last_of (" ");

- if (p == string::npos)

- return s;

- return s.substr (p + 1);

- }

-# define SRC_POS (GetFunctionName (__PRETTY_FUNCTION__) += ":" TC_TO_STRING(__LINE__))

-# define TC_UNUSED_VAR __attribute__ ((unused))

-#else

-# define SRC_POS (__FUNCTION__ ":" TC_TO_STRING(__LINE__))

-# define TC_UNUSED_VAR

-#endif

-

-#ifdef trace_point

-# undef trace_point

-#endif

-

-#ifdef trace_msg

-# undef trace_msg

-#endif

-

-#ifdef DEBUG

-# define if_debug(...) __VA_ARGS__

-

-# ifdef TC_WINDOWS

-# define trace_point OutputDebugStringA ((string (SRC_POS) + "\n").c_str())

-# define trace_msg(stream_args) do { stringstream s; s << (SRC_POS) << ": " << stream_args << endl; OutputDebugStringA (s.str().c_str()); } while (0)

-# define trace_msgw(stream_args) do { wstringstream s; s << (SRC_POS) << L": " << stream_args << endl; OutputDebugStringW (s.str().c_str()); } while (0)

-# else

-# include <iostream>

-# define trace_point cerr << (SRC_POS) << endl

-# define trace_msg(stream_args) cerr << (SRC_POS) << ": " << stream_args << endl

-# define trace_msgw(stream_args) cerr << (SRC_POS); wcerr << L": " << stream_args << endl

-# endif

-

-# include "Platform/SystemLog.h"

-# define trace_log_point SystemLog::WriteError (SRC_POS)

-# define trace_log_msg(stream_args) do { stringstream s; s << (SRC_POS) << ": " << stream_args; SystemLog::WriteError (s.str()); } while (0)

-

-#else

-# define if_debug(...)

-# define trace_point

-# define trace_msg(...)

-# define trace_msgw(...)

-# define trace_log_point

-# define trace_log_msg(...)

-#endif

-

-#define trace_val(VAL) trace_msg (#VAL << '=' << (VAL));

-

-#define array_capacity(arr) (sizeof (arr) / sizeof ((arr)[0]))

-

-#endif // TC_HEADER_Platform_PlatformBase

- Modifications and additions to the original source code (contained in this file)

-*/

-

-#include "PlatformTest.h"

-#include "Exception.h"

-#include "FileStream.h"

-#include "Finally.h"

-#include "ForEach.h"

-#include "MemoryStream.h"

-#include "Mutex.h"

-#include "Serializable.h"

-#include "SharedPtr.h"

-#include "StringConverter.h"

-#include "SyncEvent.h"

-#include "Thread.h"

-#include "Common/Tcdefs.h"

-

-namespace VeraCrypt

-{

- // make_shared_auto, File, Stream, MemoryStream, Endian, Serializer, Serializable

- void PlatformTest::SerializerTest ()

- {

- shared_ptr <Stream> stream (new MemoryStream);

-

-#if 0

- make_shared_auto (File, file);

- finally_do_arg (File&, *file, { if (finally_arg.IsOpen()) finally_arg.Delete(); });

-

- try

- {

- file->Open ("veracrypt-serializer-test.tmp", File::CreateReadWrite);

- stream = shared_ptr <Stream> (new FileStream (file));

- }

- catch (...) { }

-#endif

-

- Serializer ser (stream);

-

- uint32 i32 = 0x12345678;

- uint64 i64 = 0x0123456789abcdefULL;

- string str = "string test";

- wstring wstr = L"wstring test";

-

- string convStr = "test";

- StringConverter::ToSingle (wstr, convStr);

- if (convStr != "wstring test")

- throw TestFailed (SRC_POS);

-

- StringConverter::Erase (convStr);

- if (convStr != " ")

- throw TestFailed (SRC_POS);

-

- wstring wEraseTest = L"erase test";

- StringConverter::Erase (wEraseTest);

- if (wEraseTest != L" ")

- throw TestFailed (SRC_POS);

-

- list <string> stringList;

- stringList.push_back (str + "1");

- stringList.push_back (str + "2");

- stringList.push_back (str + "3");

-

- list <wstring> wstringList;

- wstringList.push_back (wstr + L"1");

- wstringList.push_back (wstr + L"2");

- wstringList.push_back (wstr + L"3");

-

- Buffer buffer (10);

- for (size_t i = 0; i < buffer.Size(); i++)

- buffer[i] = (byte) i;

-

- ser.Serialize ("int32", i32);

- ser.Serialize ("int64", i64);

- ser.Serialize ("string", str);

- ser.Serialize ("wstring", wstr);

- ser.Serialize ("stringList", stringList);

- ser.Serialize ("wstringList", wstringList);

- ser.Serialize ("buffer", ConstBufferPtr (buffer));

-

- ExecutedProcessFailed ex (SRC_POS, "cmd", -123, "error output");

- ex.Serialize (stream);

-

- list < shared_ptr <ExecutedProcessFailed> > exList;

- exList.push_back (make_shared <ExecutedProcessFailed> (ExecutedProcessFailed (SRC_POS, "cmd", -123, "error output1")));

- exList.push_back (make_shared <ExecutedProcessFailed> (ExecutedProcessFailed (SRC_POS, "cmd", -234, "error output2")));

- exList.push_back (make_shared <ExecutedProcessFailed> (ExecutedProcessFailed (SRC_POS, "cmd", -567, "error output3")));

- Serializable::SerializeList (stream, exList);

-

-#if 0

- if (file->IsOpen())

- file->SeekAt (0);

-#endif

-

- uint32 di32;

- ser.Deserialize ("int32", di32);

- if (i32 != di32)

- throw TestFailed (SRC_POS);

-

- uint64 di64;

- ser.Deserialize ("int64", di64);

- if (i64 != di64)

- throw TestFailed (SRC_POS);

-

- string dstr;

- ser.Deserialize ("string", dstr);

- if (str != dstr)

- throw TestFailed (SRC_POS);

-

- wstring dwstr;

- ser.Deserialize ("wstring", dwstr);

- if (str != dstr)

- throw TestFailed (SRC_POS);

-

- int i = 1;

- foreach (string item, ser.DeserializeStringList ("stringList"))

- {

- stringstream s;

- s << str << i++;

- if (item != s.str())

- throw TestFailed (SRC_POS);

- }

-

- i = 1;

- foreach (wstring item, ser.DeserializeWStringList ("wstringList"))

- {

- wstringstream s;

- s << wstr << i++;

- if (item != s.str())

- throw TestFailed (SRC_POS);

- }

-

- Buffer dbuffer (10);

- ser.Deserialize ("buffer", buffer);

- for (size_t i = 0; i < buffer.Size(); i++)

- if (buffer[i] != (byte) i)

- throw TestFailed (SRC_POS);

-

- shared_ptr <ExecutedProcessFailed> dex = Serializable::DeserializeNew <ExecutedProcessFailed> (stream);

- if (!dex

- || dex->GetCommand() != "cmd"

- || dex->GetExitCode() != -123

- || dex->GetErrorOutput() != "error output")

- throw TestFailed (SRC_POS);

-

- list < shared_ptr <ExecutedProcessFailed> > dexList;

- Serializable::DeserializeList (stream, dexList);

- i = 1;

- foreach_ref (const ExecutedProcessFailed &ex, dexList)

- {

- stringstream s;

- s << "error output" << i++;

- if (ex.GetErrorOutput() != s.str())

- throw TestFailed (SRC_POS);

- }

- }

-

- // shared_ptr, Mutex, ScopeLock, SyncEvent, Thread

- static struct

- {

- shared_ptr <int> SharedIntPtr;

- Mutex IntMutex;

- SyncEvent ExitAllowedEvent;

- } ThreadTestData;

-

- void PlatformTest::ThreadTest ()

- {

- Mutex mutex;

- mutex.Lock();

- mutex.Unlock();

-

- const int maxThreads = 3;

- ThreadTestData.SharedIntPtr.reset (new int (0));

-

- for (int i = 0; i < maxThreads; i++)

- {

- Thread t;

- t.Start (&ThreadTestProc, (void *) &ThreadTestData);

- }

-

- for (int i = 0; i < 50; i++)

- {

- {

- ScopeLock sl (ThreadTestData.IntMutex);

- if (*ThreadTestData.SharedIntPtr == maxThreads)

- break;

- }

-

- Thread::Sleep(100);

- }

-

- if (*ThreadTestData.SharedIntPtr != maxThreads)

- throw TestFailed (SRC_POS);

-

- for (int i = 0; i < 60000; i++)

- {

- ThreadTestData.ExitAllowedEvent.Signal();

- Thread::Sleep(1);

-

- ScopeLock sl (ThreadTestData.IntMutex);

- if (*ThreadTestData.SharedIntPtr == 0)

- break;

- }

-

- if (*ThreadTestData.SharedIntPtr != 0)

- throw TestFailed (SRC_POS);

- }

-

- TC_THREAD_PROC PlatformTest::ThreadTestProc (void *arg)

- {

-

- if (arg != (void *) &ThreadTestData)

- return 0;

-

- {

- ScopeLock sl (ThreadTestData.IntMutex);

- ++(*ThreadTestData.SharedIntPtr);

- }

-

- ThreadTestData.ExitAllowedEvent.Wait();

-

- {

- ScopeLock sl (ThreadTestData.IntMutex);

- --(*ThreadTestData.SharedIntPtr);

- }

-

- return 0;

- }

-

- bool PlatformTest::TestAll ()

- {

- // Integer types

- if (sizeof (byte) != 1 || sizeof (int8) != 1 || sizeof (__int8) != 1) throw TestFailed (SRC_POS);

- if (sizeof (uint16) != 2 || sizeof (int16) != 2 || sizeof (__int16) != 2) throw TestFailed (SRC_POS);

- if (sizeof (uint32) != 4 || sizeof (int32) != 4 || sizeof (__int32) != 4) throw TestFailed (SRC_POS);

- if (sizeof (uint64) != 8 || sizeof (int64) != 8) throw TestFailed (SRC_POS);

-

- // Exception handling

- TestFlag = false;

- try

- {

- try

- {

- throw TestFailed (SRC_POS);

- }

- catch (...)

- {

- throw;

- }

- return false;

- }

- catch (Exception &)

- {

- TestFlag = true;

- }

- if (!TestFlag)

- return false;

-

- // RTTI

- RttiTest rtti;

- RttiTestBase &rttiBaseRef = rtti;

- RttiTestBase *rttiBasePtr = &rtti;

-

- if (typeid (rttiBaseRef) != typeid (rtti))

- throw TestFailed (SRC_POS);

-

- if (typeid (*rttiBasePtr) != typeid (rtti))

- throw TestFailed (SRC_POS);

-

- if (dynamic_cast <RttiTest *> (rttiBasePtr) == nullptr)

- throw TestFailed (SRC_POS);

-

- try

- {

- dynamic_cast <RttiTest &> (rttiBaseRef);

- }

- catch (...)

- {

- throw TestFailed (SRC_POS);

- }

-

- // finally

- TestFlag = false;

- {

- finally_do ({ TestFlag = true; });

- if (TestFlag)

- throw TestFailed (SRC_POS);

- }

- if (!TestFlag)

- throw TestFailed (SRC_POS);

-

- TestFlag = false;

- {

- finally_do_arg (bool*, &TestFlag, { *finally_arg = true; });

- if (TestFlag)

- throw TestFailed (SRC_POS);

- }

- if (!TestFlag)

- throw TestFailed (SRC_POS);

-

- TestFlag = false;

- int tesFlag2 = 0;

- {

- finally_do_arg2 (bool*, &TestFlag, int*, &tesFlag2, { *finally_arg = true; *finally_arg2 = 2; });

- if (TestFlag || tesFlag2 != 0)

- throw TestFailed (SRC_POS);

- }

- if (!TestFlag || tesFlag2 != 2)

- throw TestFailed (SRC_POS);

-

- // uint64, vector, list, string, wstring, stringstream, wstringstream

- // shared_ptr, make_shared, StringConverter, foreach

- list <shared_ptr <uint64> > numList;

-

- numList.push_front (make_shared <uint64> (StringConverter::ToUInt64 (StringConverter::FromNumber ((uint64) 0xFFFFffffFFFFfffeULL))));

- numList.push_front (make_shared <uint64> (StringConverter::ToUInt32 (StringConverter::GetTrailingNumber ("str2"))));

- numList.push_front (make_shared <uint64> (3));

-

- list <wstring> testList;

- wstringstream wstream (L"test");

- foreach_reverse_ref (uint64 n, numList)

- {

- wstream.str (L"");

- wstream << L"str" << n;

- testList.push_back (wstream.str());

- }

-

- stringstream sstream;

- sstream << "dummy";

- sstream.str ("");

- sstream << "str18446744073709551614,str2" << " str" << StringConverter::Trim (StringConverter::ToSingle (L"\t 3 \r\n"));

- foreach (const string &s, StringConverter::Split (sstream.str(), ", "))

- {

- if (testList.front() != StringConverter::ToWide (s))

- throw TestFailed (SRC_POS);

- testList.pop_front();

- }

-

- SerializerTest();

- ThreadTest();

-

- return true;

- }

-

- bool PlatformTest::TestFlag;

-}

- Modifications and additions to the original source code (contained in this file)

-*/

-

-#ifndef TC_HEADER_Platform_PlatformTest

-#define TC_HEADER_Platform_PlatformTest

-

-#include "PlatformBase.h"

-#include "Thread.h"

-

-namespace VeraCrypt

-{

- class PlatformTest

- {

- public:

- static bool TestAll ();

-

- protected:

- class RttiTestBase

- {

- public:

- virtual ~RttiTestBase () { };

- };

-

- class RttiTest : public RttiTestBase {

- public:

- virtual ~RttiTest () { };

- };

-

- PlatformTest ();

- static void SerializerTest ();

- static void ThreadTest ();

- static TC_THREAD_PROC ThreadTestProc (void *param);

-

- static bool TestFlag;

- };

-}

-

-#endif // TC_HEADER_Platform_PlatformTest

- Modifications and additions to the original source code (contained in this file)

-*/

-

-#include "Serializable.h"

-#include "SerializerFactory.h"

-

-namespace VeraCrypt

-{

- string Serializable::DeserializeHeader (shared_ptr <Stream> stream)

- {

- Serializer sr (stream);

- return sr.DeserializeString ("SerializableName");

- }

-

- Serializable *Serializable::DeserializeNew (shared_ptr <Stream> stream)

- {

- string name = Serializable::DeserializeHeader (stream);

- Serializable *serializable = SerializerFactory::GetNewSerializable (name);

- serializable->Deserialize (stream);

-

- return serializable;

- }

-

- void Serializable::Serialize (shared_ptr <Stream> stream) const

- {

- Serializer sr (stream);

- Serializable::SerializeHeader (sr, SerializerFactory::GetName (typeid (*this)));

- }

-

- void Serializable::SerializeHeader (Serializer &serializer, const string &name)

- {

- serializer.Serialize ("SerializableName", name);

- }

-}

- Modifications and additions to the original source code (contained in this file)

-*/

-

-#ifndef TC_HEADER_Platform_Serializable

-#define TC_HEADER_Platform_Serializable

-

-#include <stdexcept>

-#include "PlatformBase.h"

-#include "ForEach.h"

-#include "Serializer.h"

-#include "SerializerFactory.h"

-

-namespace VeraCrypt

-{

- class Serializable

- {

- public:

- virtual ~Serializable () { }

-

- virtual void Deserialize (shared_ptr <Stream> stream) = 0;

- static string DeserializeHeader (shared_ptr <Stream> stream);

- static Serializable *DeserializeNew (shared_ptr <Stream> stream);

-

- template <class T>

- static shared_ptr <T> DeserializeNew (shared_ptr <Stream> stream)

- {

- shared_ptr <T> p (dynamic_cast <T *> (DeserializeNew (stream)));

- if (!p)

- throw std::runtime_error (SRC_POS);

- return p;

- }

-

- template <class T>

- static void DeserializeList (shared_ptr <Stream> stream, list < shared_ptr <T> > &dataList)

- {

- if (DeserializeHeader (stream) != string ("list<") + SerializerFactory::GetName (typeid (T)) + ">")

- throw std::runtime_error (SRC_POS);

-

- Serializer sr (stream);

- uint64 listSize;

- sr.Deserialize ("ListSize", listSize);

-

- for (size_t i = 0; i < listSize; i++)

- {

- shared_ptr <T> p (dynamic_cast <T *> (DeserializeNew (stream)));

- if (!p)

- throw std::runtime_error (SRC_POS);

- dataList.push_back (p);

- }

- }

-

- virtual void Serialize (shared_ptr <Stream> stream) const;

-

- template <class T>

- static void SerializeList (shared_ptr <Stream> stream, const list < shared_ptr <T> > &dataList)

- {

- Serializer sr (stream);

- SerializeHeader (sr, string ("list<") + SerializerFactory::GetName (typeid (T)) + ">");

-

- sr.Serialize ("ListSize", (uint64) dataList.size());

- foreach_ref (const T &item, dataList)

- item.Serialize (stream);

- }

-

- static void SerializeHeader (Serializer &serializer, const string &name);

-

- protected:

- Serializable () { }

- };

-}

-

-#define TC_SERIALIZABLE(TYPE) \

- static Serializable *GetNewSerializable () { return new TYPE(); } \

- virtual void Deserialize (shared_ptr <Stream> stream); \

- virtual void Serialize (shared_ptr <Stream> stream) const

-

-#endif // TC_HEADER_Platform_Serializable

- Modifications and additions to the original source code (contained in this file)

-*/

-

-#include "Exception.h"

-#include "ForEach.h"

-#include "Memory.h"

-#include "Serializer.h"

-

-namespace VeraCrypt

-{

- template <typename T>

- T Serializer::Deserialize ()

- {

- uint64 size;

- DataStream->ReadCompleteBuffer (BufferPtr ((byte *) &size, sizeof (size)));

-

- if (Endian::Big (size) != sizeof (T))

- throw ParameterIncorrect (SRC_POS);

-

- T data;

- DataStream->ReadCompleteBuffer (BufferPtr ((byte *) &data, sizeof (data)));

-

- return Endian::Big (data);

- }

-

- void Serializer::Deserialize (const string &name, bool &data)

- {

- ValidateName (name);

- data = Deserialize <byte> () == 1;

- }

-

- void Serializer::Deserialize (const string &name, byte &data)

- {

- ValidateName (name);

- data = Deserialize <byte> ();

- }

-

- void Serializer::Deserialize (const string &name, int32 &data)

- {

- ValidateName (name);

- data = (int32) Deserialize <uint32> ();

- }

-

- void Serializer::Deserialize (const string &name, int64 &data)

- {

- ValidateName (name);

- data = (int64) Deserialize <uint64> ();

- }

-

- void Serializer::Deserialize (const string &name, uint32 &data)

- {

- ValidateName (name);

- data = Deserialize <uint32> ();

- }

-

- void Serializer::Deserialize (const string &name, uint64 &data)

- {

- ValidateName (name);

- data = Deserialize <uint64> ();

- }

-

- void Serializer::Deserialize (const string &name, string &data)

- {

- ValidateName (name);

- data = DeserializeString ();

- }

-

- void Serializer::Deserialize (const string &name, wstring &data)

- {

- ValidateName (name);

- data = DeserializeWString ();

- }

-

- void Serializer::Deserialize (const string &name, const BufferPtr &data)

- {

- ValidateName (name);

-

- uint64 size = Deserialize <uint64> ();

- if (data.Size() != size)

- throw ParameterIncorrect (SRC_POS);

-

- DataStream->ReadCompleteBuffer (data);

- }

-

- bool Serializer::DeserializeBool (const string &name)

- {

- bool data;

- Deserialize (name, data);

- return data;

- }

-

- int32 Serializer::DeserializeInt32 (const string &name)

- {

- ValidateName (name);

- return Deserialize <uint32> ();

- }

-

- int64 Serializer::DeserializeInt64 (const string &name)

- {

- ValidateName (name);

- return Deserialize <uint64> ();

- }

-

- uint32 Serializer::DeserializeUInt32 (const string &name)

- {

- ValidateName (name);

- return Deserialize <uint32> ();

- }

-

- uint64 Serializer::DeserializeUInt64 (const string &name)

- {

- ValidateName (name);

- return Deserialize <uint64> ();

- }

-

- string Serializer::DeserializeString ()

- {

- uint64 size = Deserialize <uint64> ();

-

- vector <char> data ((size_t) size);

- DataStream->ReadCompleteBuffer (BufferPtr ((byte *) &data[0], (size_t) size));

-

- return string (&data[0]);

- }

-

- string Serializer::DeserializeString (const string &name)

- {

- ValidateName (name);

- return DeserializeString ();

- }

-

- list <string> Serializer::DeserializeStringList (const string &name)

- {

- ValidateName (name);

- list <string> deserializedList;

- uint64 listSize = Deserialize <uint64> ();

-

- for (size_t i = 0; i < listSize; i++)

- deserializedList.push_back (DeserializeString ());

-

- return deserializedList;

- }

-

- wstring Serializer::DeserializeWString ()

- {

- uint64 size = Deserialize <uint64> ();

-

- vector <wchar_t> data ((size_t) size / sizeof (wchar_t));

- DataStream->ReadCompleteBuffer (BufferPtr ((byte *) &data[0], (size_t) size));

-

- return wstring (&data[0]);

- }

-

- list <wstring> Serializer::DeserializeWStringList (const string &name)

- {

- ValidateName (name);

- list <wstring> deserializedList;

- uint64 listSize = Deserialize <uint64> ();

-

- for (size_t i = 0; i < listSize; i++)

- deserializedList.push_back (DeserializeWString ());

-

- return deserializedList;

- }

-

- wstring Serializer::DeserializeWString (const string &name)

- {

- ValidateName (name);

- return DeserializeWString ();

- }

-

- template <typename T>

- void Serializer::Serialize (T data)

- {

- uint64 size = Endian::Big (uint64 (sizeof (data)));

- DataStream->Write (ConstBufferPtr ((byte *) &size, sizeof (size)));

-

- data = Endian::Big (data);

- DataStream->Write (ConstBufferPtr ((byte *) &data, sizeof (data)));

- }

-

- void Serializer::Serialize (const string &name, bool data)

- {

- SerializeString (name);

- byte d = data ? 1 : 0;

- Serialize (d);

- }

-

- void Serializer::Serialize (const string &name, byte data)

- {

- SerializeString (name);

- Serialize (data);

- }

-

- void Serializer::Serialize (const string &name, const char *data)

- {

- Serialize (name, string (data));

- }

-

- void Serializer::Serialize (const string &name, int32 data)

- {

- SerializeString (name);

- Serialize ((uint32) data);

- }

-

- void Serializer::Serialize (const string &name, int64 data)

- {

- SerializeString (name);

- Serialize ((uint64) data);

- }

-

- void Serializer::Serialize (const string &name, uint32 data)

- {

- SerializeString (name);

- Serialize (data);

- }

-

- void Serializer::Serialize (const string &name, uint64 data)

- {

- SerializeString (name);

- Serialize (data);

- }

-

- void Serializer::Serialize (const string &name, const string &data)

- {

- SerializeString (name);

- SerializeString (data);

- }

-

- void Serializer::Serialize (const string &name, const wchar_t *data)

- {

- Serialize (name, wstring (data));

- }

-

- void Serializer::Serialize (const string &name, const wstring &data)

- {

- SerializeString (name);

- SerializeWString (data);

- }

-

- void Serializer::Serialize (const string &name, const list <string> &stringList)

- {

- SerializeString (name);

-

- uint64 listSize = stringList.size();

- Serialize (listSize);

-

- foreach (const string &item, stringList)

- SerializeString (item);

- }

-

- void Serializer::Serialize (const string &name, const list <wstring> &stringList)

- {

- SerializeString (name);

-

- uint64 listSize = stringList.size();

- Serialize (listSize);

-

- foreach (const wstring &item, stringList)

- SerializeWString (item);

- }

-

- void Serializer::Serialize (const string &name, const ConstBufferPtr &data)

- {

- SerializeString (name);

-

- uint64 size = data.Size();

- Serialize (size);

-

- DataStream->Write (data);

- }

-

- void Serializer::SerializeString (const string &data)

- {

- Serialize ((uint64) data.size() + 1);

- DataStream->Write (ConstBufferPtr ((byte *) (data.data() ? data.data() : data.c_str()), data.size() + 1));

- }

-

- void Serializer::SerializeWString (const wstring &data)

- {

- uint64 size = (data.size() + 1) * sizeof (wchar_t);

- Serialize (size);

- DataStream->Write (ConstBufferPtr ((byte *) (data.data() ? data.data() : data.c_str()), (size_t) size));

- }

-

- void Serializer::ValidateName (const string &name)

- {

- string dName = DeserializeString();

- if (dName != name)

- {

- throw ParameterIncorrect (SRC_POS);

- }

- }

-}

- Modifications and additions to the original source code (contained in this file)

-*/

-

-#ifndef TC_HEADER_Platform_Serializer

-#define TC_HEADER_Platform_Serializer

-

-#include "PlatformBase.h"

-#include "Buffer.h"

-#include "SharedPtr.h"

-#include "Stream.h"

-

-namespace VeraCrypt

-{

- class Serializer

- {

- public:

- Serializer (shared_ptr <Stream> stream) : DataStream (stream) { }

- virtual ~Serializer () { }

-

- void Deserialize (const string &name, bool &data);

- void Deserialize (const string &name, byte &data);

- void Deserialize (const string &name, int32 &data);

- void Deserialize (const string &name, int64 &data);

- void Deserialize (const string &name, uint32 &data);

- void Deserialize (const string &name, uint64 &data);

- void Deserialize (const string &name, string &data);

- void Deserialize (const string &name, wstring &data);

- void Deserialize (const string &name, const BufferPtr &data);

- bool DeserializeBool (const string &name);

- int32 DeserializeInt32 (const string &name);

- int64 DeserializeInt64 (const string &name);

- uint32 DeserializeUInt32 (const string &name);

- uint64 DeserializeUInt64 (const string &name);

- string DeserializeString (const string &name);

- list <string> DeserializeStringList (const string &name);

- wstring DeserializeWString (const string &name);

- list <wstring> DeserializeWStringList (const string &name);

- void Serialize (const string &name, bool data);

- void Serialize (const string &name, byte data);

- void Serialize (const string &name, const char *data);

- void Serialize (const string &name, int32 data);

- void Serialize (const string &name, int64 data);

- void Serialize (const string &name, uint32 data);

- void Serialize (const string &name, uint64 data);

- void Serialize (const string &name, const string &data);

- void Serialize (const string &name, const wstring &data);

- void Serialize (const string &name, const wchar_t *data);

- void Serialize (const string &name, const list <string> &stringList);

- void Serialize (const string &name, const list <wstring> &stringList);

- void Serialize (const string &name, const ConstBufferPtr &data);

-

- protected:

- template <typename T> T Deserialize ();

- string DeserializeString ();

- wstring DeserializeWString ();

- template <typename T> void Serialize (T data);

- void SerializeString (const string &data);

- void SerializeWString (const wstring &data);

- void ValidateName (const string &name);

-

- shared_ptr <Stream> DataStream;

-

- private:

- Serializer (const Serializer &);

- Serializer &operator= (const Serializer &);

- };

-}

-

-#endif // TC_HEADER_Platform_Serializer

- Modifications and additions to the original source code (contained in this file)

-*/

-

-#include <stdexcept>

-#include "SerializerFactory.h"

-

-namespace VeraCrypt

-{

- void SerializerFactory::Deinitialize ()

- {

- if (--UseCount == 0)

- {

- delete NameToTypeMap;

- delete TypeToNameMap;

- }

- }

-

- string SerializerFactory::GetName (const type_info &typeInfo)

- {

- string typeName = StringConverter::GetTypeName (typeInfo);

- if (TypeToNameMap->find (typeName) == TypeToNameMap->end())

- throw std::runtime_error (SRC_POS);

-

- return (*TypeToNameMap)[typeName];

- }

-

- Serializable *SerializerFactory::GetNewSerializable (const string &typeName)

- {

- if (NameToTypeMap->find (typeName) == NameToTypeMap->end())

- throw std::runtime_error (SRC_POS);

-

- return (*NameToTypeMap)[typeName].GetNewPtr();

- }

-

- void SerializerFactory::Initialize ()

- {

- if (UseCount == 0)

- {

- NameToTypeMap = new map <string, SerializerFactory::MapEntry>;

- TypeToNameMap = new map <string, string>;

- }

-

- ++UseCount;

- }

-

- map <string, SerializerFactory::MapEntry> *SerializerFactory::NameToTypeMap;

- map <string, string> *SerializerFactory::TypeToNameMap;

- int SerializerFactory::UseCount;

-}

- Modifications and additions to the original source code (contained in this file)

-*/

-

-#ifndef TC_HEADER_Platform_SerializerFactory

-#define TC_HEADER_Platform_SerializerFactory

-

-#include <typeinfo>

-#include "PlatformBase.h"

-#include "StringConverter.h"

-

-namespace VeraCrypt

-{

- class Serializable;

-

- class SerializerFactory

- {

- public:

- ~SerializerFactory ();

-

- static void Deinitialize ();

- static string GetName (const type_info &typeInfo);

- static Serializable *GetNewSerializable (const string &typeName);

- static void Initialize ();

-

- struct MapEntry

- {

- MapEntry () { }

- MapEntry (const string &typeName, Serializable* (*getNewPtr) ())

- : TypeName (typeName), GetNewPtr (getNewPtr) { }

-

- MapEntry &operator= (const MapEntry &right)

- {

- TypeName = right.TypeName;

- GetNewPtr = right.GetNewPtr;

- return *this;

- }

-

- string TypeName;

- Serializable* (*GetNewPtr) ();

- };

-

- static map <string, MapEntry> *NameToTypeMap;

- static map <string, string> *TypeToNameMap;

-

- protected:

- SerializerFactory ();

-

- static int UseCount;

- };

-

-}

-

-#define TC_SERIALIZER_FACTORY_ADD_EXCEPTION_SET(TYPE) \

- struct TYPE##SerializerFactoryInitializer \

- { \

- TYPE##SerializerFactoryInitializer () \

- { \

- SerializerFactory::Initialize(); \

- TC_EXCEPTION_SET; \

- } \

- ~TYPE##SerializerFactoryInitializer () \

- { \

- SerializerFactory::Deinitialize(); \

- } \

- }; \

- static TYPE##SerializerFactoryInitializer TYPE##SerializerFactoryInitializer

-

-#define TC_SERIALIZER_FACTORY_ADD_CLASS(TYPE) \

- struct TYPE##SerializerFactoryInitializer \

- { \

- TYPE##SerializerFactoryInitializer () \

- { \

- SerializerFactory::Initialize(); \

- TC_SERIALIZER_FACTORY_ADD (TYPE); \

- } \

- ~TYPE##SerializerFactoryInitializer () \

- { \

- SerializerFactory::Deinitialize(); \

- } \

- }; \

- static TYPE##SerializerFactoryInitializer TYPE##SerializerFactoryInitializerInst

-

-#define TC_SERIALIZER_FACTORY_ADD(TYPE) \

- (*SerializerFactory::NameToTypeMap)[#TYPE] = SerializerFactory::MapEntry (StringConverter::GetTypeName (typeid (TYPE)), &TYPE::GetNewSerializable); \

- (*SerializerFactory::TypeToNameMap)[StringConverter::GetTypeName (typeid (TYPE))] = #TYPE

-

-

-#endif // TC_HEADER_Platform_SerializerFactory

- Modifications and additions to the original source code (contained in this file)

-*/

-

-#ifndef TC_HEADER_Platform_SharedPtr

-#define TC_HEADER_Platform_SharedPtr

-

-#include <stdexcept>

-#include "SharedVal.h"

-

-#ifdef nullptr

-

-namespace VeraCrypt

-{

- template <class T>

- class SharedPtr

- {

- public:

- explicit SharedPtr ()

- : Pointer (nullptr), UseCount (nullptr) { }

-

- explicit SharedPtr (T *pointer)

- : Pointer (pointer), UseCount (new SharedVal <uint64> (1)) { }

-

- SharedPtr (const SharedPtr &source)

- {

- CopyFrom (source);

- }

-

- ~SharedPtr ()

- {

- Release();

- }

-

- SharedPtr &operator= (const SharedPtr &source)

- {

- if (&source == this)

- return *this;

-

- Release();

- CopyFrom (source);

- return *this;

- }

-

- bool operator == (const SharedPtr &other)

- {

- return get() == other.get();

- }

-

- bool operator != (const SharedPtr &other)

- {

- return get() != other.get();

- }

-

- T &operator* () const

- {

-#ifdef DEBUG

- if (Pointer == nullptr)

- throw std::runtime_error (SRC_POS);

-#endif

- return *Pointer;

- }

-

- T *operator-> () const

- {

-#ifdef DEBUG

- if (Pointer == nullptr)

- throw std::runtime_error (SRC_POS);

-#endif

- return Pointer;

- }

-

- operator bool () const

- {

- return Pointer != nullptr;

- }

-

- T *get () const

- {

- return Pointer;

- }

-

- void reset ()

- {

- Release();

- }

-

- void reset (T *pointer)

- {

- *this = SharedPtr (pointer);

- }

-

- uint64 use_count () const

- {

- if (!UseCount)

- return 0;

-

- return *UseCount;

- }

-

- protected:

- void CopyFrom (const SharedPtr &source)

- {

- Pointer = source.Pointer;

- UseCount = source.UseCount;

-

- if (UseCount)

- UseCount->Increment();

- }

-

- void Release ()

- {

- if (UseCount != nullptr)

- {

- if (UseCount->Decrement() == 0)

- {

- if (Pointer != nullptr)

- delete Pointer;

- delete UseCount;

- }

-

- Pointer = nullptr;

- UseCount = nullptr;

- }

- }

-

- T *Pointer;

- SharedVal <uint64> *UseCount;

- };

-

-#ifdef shared_ptr

-#undef shared_ptr

-#endif

-#define shared_ptr VeraCrypt::SharedPtr

-

-#ifdef make_shared

-#undef make_shared

-#endif

-

- template <class T> shared_ptr <T> make_shared ()

- {

- return shared_ptr <T> (new T ());

- }

-

- template <class T, class A> shared_ptr <T> make_shared (const A &arg)

- {

- return shared_ptr <T> (new T (arg));

- }

-

-#define make_shared VeraCrypt::make_shared

-

-}

-

-#endif // nullptr

-

-#define make_shared_auto(typeName,instanceName) shared_ptr <typeName> instanceName (new typeName ())

-

-#endif // TC_HEADER_Platform_SharedPtr

- Modifications and additions to the original source code (contained in this file)

-*/

-

-#ifndef TC_HEADER_Platform_SharedVal

-#define TC_HEADER_Platform_SharedVal

-

-#include "PlatformBase.h"

-#include "Mutex.h"

-

-namespace VeraCrypt

-{

- template <class T>

- class SharedVal

- {

- public:

- SharedVal () { }

- explicit SharedVal (T value) : Value (value) { }

- virtual ~SharedVal () { }

-

- operator T ()

- {

- return Get ();

- }

-

- T Decrement ()

- {

- ValMutex.Lock();

- T r = --Value;

- ValMutex.Unlock();

- return r;

- }

-

- T Get ()

- {

- ValMutex.Lock();

- T r = Value;

- ValMutex.Unlock();

- return r;

- }

-

- T Increment ()

- {

- ValMutex.Lock();

- T r = ++Value;

- ValMutex.Unlock();

- return r;

- }

-

- void Set (T value)

- {

- ValMutex.Lock();

- Value = value;

- ValMutex.Unlock();

- }

-

- protected:

- volatile T Value;

- Mutex ValMutex;

-

- private:

- SharedVal (const SharedVal &);

- SharedVal &operator= (const SharedVal &);

- };

-}

-

-#endif // TC_HEADER_Platform_SharedVal

- Modifications and additions to the original source code (contained in this file)

-*/

-

-#ifndef TC_HEADER_Platform_Stream

-#define TC_HEADER_Platform_Stream

-

-#include "PlatformBase.h"

-#include "Buffer.h"

-

-namespace VeraCrypt

-{

- class Stream

- {

- public:

- virtual ~Stream () { }

- virtual uint64 Read (const BufferPtr &buffer) = 0;

- virtual void ReadCompleteBuffer (const BufferPtr &buffer) = 0;

- virtual void Write (const ConstBufferPtr &data) = 0;

-

- protected:

- Stream () { };

-

- private:

- Stream (const Stream &);

- Stream &operator= (const Stream &);

- };

-}

-

-#endif // TC_HEADER_Platform_Stream

- Modifications and additions to the original source code (contained in this file)

-*/

-

-#ifdef __GNUC__

-# include <cxxabi.h>

-#endif

-#include <locale>

-#include <typeinfo>

-#include "Buffer.h"

-#include "Exception.h"

-#include "ForEach.h"

-#include "StringConverter.h"

-#include "SystemException.h"

-

-namespace VeraCrypt

-{

- void StringConverter::Erase (string &str)

- {

- for (size_t i = 0; i < str.size(); ++i)

- {

- str[i] = ' ';

- }

- }

-

- void StringConverter::Erase (wstring &str)

- {

- for (size_t i = 0; i < str.size(); ++i)

- {

- str[i] = ' ';

- }

- }

-

- wstring StringConverter::FromNumber (double number)

- {

- wstringstream s;

- s << number;

- return s.str();

- }

-

- wstring StringConverter::FromNumber (int32 number)

- {

- wstringstream s;

- s << number;

- return s.str();

- }

-

- wstring StringConverter::FromNumber (uint32 number)

- {

- wstringstream s;

- s << number;

- return s.str();

- }

-

- wstring StringConverter::FromNumber (int64 number)

- {

- wstringstream s;

- s << number;

- return s.str();

- }

-

- wstring StringConverter::FromNumber (uint64 number)

- {

- wstringstream s;

- s << number;

- return s.str();

- }

-

- string StringConverter::GetTrailingNumber (const string &str)

- {

- size_t start = str.find_last_not_of ("0123456789");

- if (start == string::npos)

- return str;

-

- string s = str.substr (start + 1);

- if (s.empty ())

- throw ParameterIncorrect (SRC_POS);

-

- return s;

- }

-

- string StringConverter::GetTypeName (const type_info &typeInfo)

- {

- try

- {

-#ifdef _MSC_VER

- // type_info::name() leaks memory as of MS VC++ 8.0

- string rawName (typeInfo.raw_name());

-

- size_t cut1 = (rawName.find (".?A") != string::npos) ? 4 : string::npos;

- size_t cut2 = rawName.find ("@");

- size_t cut3 = rawName.find ("@@");

-

- if (cut1 == string::npos || cut2 == string::npos || cut3 == string::npos)

- return typeInfo.name();

-

- return rawName.substr (cut2 + 1, cut3 - cut2 - 1) + "::" + rawName.substr (cut1, cut2 - cut1);

-

-#elif defined (__GNUC__)

- int status;

- char *name = abi::__cxa_demangle (typeInfo.name(), nullptr, nullptr, &status);

-

- if (name)

- {

- string s (name);

- free (name);

- return s;

- }

-#endif

- }

- catch (...) { }

-

- return typeInfo.name();

- }

-

- wstring StringConverter::QuoteSpaces (const wstring &str)

- {

- if (str.find (L' ') == string::npos)

- return str;

-

- wstring escaped (L"'");

- foreach (wchar_t c, str)

- {

- if (c == L'\'')

- escaped += L'\'';

- escaped += c;

- }

- return escaped + L'\'';

- }

-

- vector <string> StringConverter::Split (const string &str, const string &separators, bool returnEmptyFields)

- {

- vector <string> elements;

-

- if (!returnEmptyFields)

- {

- size_t p = 0;

- while ((p = str.find_first_not_of (separators, p)) != string::npos)

- {

- size_t end = str.find_first_of (separators, p);

- if (end == string::npos)

- {

- elements.push_back (str.substr (p));

- break;

- }

-

- elements.push_back (str.substr (p, end - p));

- p = end;

- }

- }

- else

- {

- string element;

- elements.push_back (element);

- foreach (char c, str)

- {

- if (separators.find (c) != string::npos)

- {

- element.erase();

- elements.push_back (element);

- }

- else

- {

- elements.back() += c;

- }

- }

- }

-

- return elements;

- }

-

- string StringConverter::StripTrailingNumber (const string &str)

- {

- size_t start = str.find_last_not_of ("0123456789");

- if (start == string::npos)

- return "";

-

- return str.substr (0, start + 1);

- }

-

- wstring StringConverter::ToExceptionString (const exception &ex)

- {

- const SystemException *sysEx = dynamic_cast <const SystemException *> (&ex);

- if (sysEx)

- return ToWide (sysEx->what()) + L": " + sysEx->SystemText() + L": " + sysEx->GetSubject();

-

- if (ex.what() && !string (ex.what()).empty())

- return ToWide (GetTypeName (typeid (ex)) + ": " + ex.what());

-

- return ToWide (GetTypeName (typeid (ex)));

- }

-

- string StringConverter::ToLower (const string &str)

- {

- string s;

- foreach (char c, str)

- s += tolower (c, locale());

- return s;

- }

-

- string StringConverter::ToSingle (const wstring &wstr, bool noThrow)

- {

- string str;

- ToSingle (wstr, str, noThrow);

- return str;

- }

-

- void StringConverter::ToSingle (const wstring &wstr, string &str, bool noThrow)

- {

- try

- {

- mbstate_t mbState;

- Memory::Zero (&mbState, sizeof (mbState));

- const wchar_t *src = wstr.c_str();

-

- size_t size = wcsrtombs (nullptr, &src, 0, &mbState);

- if (size == (size_t) -1)

- throw StringConversionFailed (SRC_POS, wstr);

-

- vector <char> buf (size + 1);

- Memory::Zero (&mbState, sizeof (mbState));

-

- if ((size = wcsrtombs (&buf[0], &src, buf.size(), &mbState)) == (size_t) -1)

- throw StringConversionFailed (SRC_POS, wstr);

-

- str.clear();

- str.insert (0, &buf.front(), size);

- Memory::Erase (&buf.front(), buf.size());

- }

- catch (...)

- {

- if (!noThrow)

- throw;

- }

- }

-

- uint32 StringConverter::ToUInt32 (const string &str)

- {

- uint32 n;

- stringstream ss (str);

-

- ss >> n;

- if (ss.fail() || n == 0xffffFFFFU)

- throw ParameterIncorrect (SRC_POS);

-

- return n;

- }

-

- uint32 StringConverter::ToUInt32 (const wstring &str)

- {

- uint32 n;

- wstringstream ss (str);

-

- ss >> n;

- if (ss.fail() || n == 0xffffFFFFU)

- throw ParameterIncorrect (SRC_POS);

-

- return n;

- int32 StringConverter::ToInt32 (const string &str)

- {

- int32 n;

- stringstream ss (str);

-

- ss >> n;

- if (ss.fail() || n == 0x7fffFFFF || n == -0x7fffFFFF)

- throw ParameterIncorrect (SRC_POS);

-

- return n;

- }

-

- int32 StringConverter::ToInt32 (const wstring &str)

- {

- int32 n;

- wstringstream ss (str);

-

- ss >> n;

- if (ss.fail() || n == 0x7fffFFFF || n == -0x7fffFFFF)

- throw ParameterIncorrect (SRC_POS);

-

- return n;

- }

-

- uint64 StringConverter::ToUInt64 (const string &str)

- {

- uint64 n;

- stringstream ss (str);

-

- ss >> n;

- if (ss.fail() || n == 0xffffFFFFffffFFFFULL)

- throw ParameterIncorrect (SRC_POS);

-

- return n;

- }

-

- uint64 StringConverter::ToUInt64 (const wstring &str)

- {

- uint64 n;

- wstringstream ss (str);

-

- ss >> n;

- if (ss.fail() || n == 0xffffFFFFffffFFFFULL)

- throw ParameterIncorrect (SRC_POS);

-

- return n;

- int64 StringConverter::ToInt64 (const string &str)

- {

- int64 n;

- stringstream ss (str);

-

- ss >> n;

- if (ss.fail() || n == 0x7fffFFFFffffFFFFLL || n == -0x7fffFFFFffffFFFFLL)

- throw ParameterIncorrect (SRC_POS);

-

- return n;

- }

-

- int64 StringConverter::ToInt64 (const wstring &str)

- {

- int64 n;

- wstringstream ss (str);

-

- ss >> n;

- if (ss.fail() || n == 0x7fffFFFFffffFFFFLL || n == -0x7fffFFFFffffFFFFLL)

- throw ParameterIncorrect (SRC_POS);

-

- return n;

- }

-

- string StringConverter::ToUpper (const string &str)

- {

- string s;

- foreach (char c, str)

- s += toupper (c, locale());

- return s;

- }

-

- wstring StringConverter::ToWide (const string &str, bool noThrow)

- {

- try

- {

- mbstate_t mbState;

- Memory::Zero (&mbState, sizeof (mbState));

- const char *src = str.c_str();

-

- size_t size = mbsrtowcs (nullptr, &src, 0, &mbState);

- if (size == (size_t) -1)

- throw StringConversionFailed (SRC_POS);

-

- vector <wchar_t> buf (size + 1);

- Memory::Zero (&mbState, sizeof (mbState));

-

- if ((size = mbsrtowcs (&buf[0], &src, buf.size(), &mbState)) == (size_t) -1)

- throw StringConversionFailed (SRC_POS);

-

- wstring s;

- s.insert (s.begin(), buf.begin(), buf.begin() + size);

- return s;

- }

- catch (...)

- {

- if (noThrow)

- return L"";

- throw;

- }

- }

-

- void StringConverter::ToWideBuffer (const wstring &str, wchar_t *buffer, size_t bufferSize)

- {

- if (str.length() < 1)

- {

- buffer[0] = 0;

- return;

- }

-

- BufferPtr (

- (byte *) buffer,

- bufferSize).CopyFrom (

- ConstBufferPtr ((byte *) (wstring (str).c_str()),

- (str.length() + 1) * sizeof (wchar_t)

- )

- );

- }

-

- string StringConverter::Trim (const string &str)

- {

- size_t start = 0;

- size_t end = str.size();

- if (end < 1)

- return str;

-

- foreach (char c, str)

- {

- if (c > ' ')

- break;

- ++start;

- }

-

- foreach_reverse (char c, str)

- {

- if (c > ' ')

- break;

- --end;

- }

-

- return str.substr (start, end - start);

- }

-}

- Modifications and additions to the original source code (contained in this file)

-*/

-

-#ifndef TC_HEADER_Platform_StringConverter

-#define TC_HEADER_Platform_StringConverter

-

-#include <stdlib.h>

-#include "PlatformBase.h"

-

-namespace VeraCrypt

-{

- class StringConverter

- {

- public:

- static void Erase (string &str);

- static void Erase (wstring &str);

- static wstring FromNumber (double number);

- static wstring FromNumber (int32 number);

- static wstring FromNumber (uint32 number);

- static wstring FromNumber (int64 number);

- static wstring FromNumber (uint64 number);

- static string GetTrailingNumber (const string &str);

- static string GetTypeName (const type_info &typeInfo);

- static wstring QuoteSpaces (const wstring &str);

- static vector <string> Split (const string &str, const string &separators = " \t\r\n", bool returnEmptyFields = false);

- static string StripTrailingNumber (const string &str);

- static wstring ToExceptionString (const exception &ex);

- static string ToLower (const string &str);

- static uint32 ToUInt32 (const string &str);

- static int32 ToInt32 (const string &str);

- static int32 ToInt32 (const wstring &str);

- static uint64 ToUInt64 (const string &str);

- static int64 ToInt64 (const string &str);

- static int64 ToInt64 (const wstring &str);

- static string ToSingle (double number) { return ToSingle (FromNumber (number)); }

- static string ToSingle (int32 number) { return ToSingle (FromNumber (number)); }

- static string ToSingle (uint32 number) { return ToSingle (FromNumber (number)); }

- static string ToSingle (int64 number) { return ToSingle (FromNumber (number)); }

- static string ToSingle (uint64 number) { return ToSingle (FromNumber (number)); }

- static string ToSingle (const wstring &wstr, bool noThrow = false);

- static void ToSingle (const wstring &wstr, string &str, bool noThrow = false);

- static string ToUpper (const string &str);

- static wstring ToWide (double number) { return FromNumber (number); }

- static wstring ToWide (int32 number) { return FromNumber (number); }

- static wstring ToWide (uint32 number) { return FromNumber (number); }

- static wstring ToWide (int64 number) { return FromNumber (number); }

- static wstring ToWide (uint64 number) { return FromNumber (number); }

- static wstring ToWide (const string &str, bool noThrow = false);

- static void ToWideBuffer (const wstring &str, wchar_t *buffer, size_t bufferSize);

- static string Trim (const string &str);

-

- private:

- StringConverter ();

- };

-}

-

-#endif // TC_HEADER_Platform_StringConverter

- Modifications and additions to the original source code (contained in this file)

-*/

-

-#ifndef TC_HEADER_Platform_SyncEvent

-#define TC_HEADER_Platform_SyncEvent

-

-#ifdef TC_WINDOWS

-# include "System.h"

-#else

-# include <pthread.h>

-#endif

-#include "PlatformBase.h"

-#include "Mutex.h"

-

-namespace VeraCrypt

-{

- class SyncEvent

- {

- public:

- SyncEvent ();

- ~SyncEvent ();

-

- void Signal ();

- void Wait ();

-

- protected:

- bool Initialized;

-#ifdef TC_WINDOWS

- HANDLE SystemSyncEvent;

-#else

- volatile bool Signaled;

- pthread_cond_t SystemSyncEvent;

- Mutex EventMutex;

-#endif

-

- private:

- SyncEvent (const SyncEvent &);

- SyncEvent &operator= (const SyncEvent &);

- };

-}

-

-#endif // TC_HEADER_Platform_SyncEvent