VeraCrypt
aboutsummaryrefslogtreecommitdiff
path: root/src/Volume/EncryptionThreadPool.h
diff options
context:
space:
mode:
Diffstat (limited to 'src/Volume/EncryptionThreadPool.h')
-rw-r--r--src/Volume/EncryptionThreadPool.h87
1 files changed, 87 insertions, 0 deletions
diff --git a/src/Volume/EncryptionThreadPool.h b/src/Volume/EncryptionThreadPool.h
new file mode 100644
index 00000000..70d87021
--- /dev/null
+++ b/src/Volume/EncryptionThreadPool.h
@@ -0,0 +1,87 @@
+/*
+ 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_Volume_EncryptionThreadPool
+#define TC_HEADER_Volume_EncryptionThreadPool
+
+#include "Platform/Platform.h"
+#include "EncryptionMode.h"
+
+namespace TrueCrypt
+{
+ class EncryptionThreadPool
+ {
+ public:
+ struct WorkType
+ {
+ enum Enum
+ {
+ EncryptDataUnits,
+ DecryptDataUnits,
+ DeriveKey
+ };
+ };
+
+ struct WorkItem
+ {
+ struct State
+ {
+ enum Enum
+ {
+ Free,
+ Ready,
+ Busy
+ };
+ };
+
+ struct WorkItem *FirstFragment;
+ auto_ptr <Exception> ItemException;
+ SyncEvent ItemCompletedEvent;
+ SharedVal <size_t> OutstandingFragmentCount;
+ SharedVal <State::Enum> State;
+ WorkType::Enum Type;
+
+ union
+ {
+ struct
+ {
+ const EncryptionMode *Mode;
+ byte *Data;
+ uint64 StartUnitNo;
+ uint64 UnitCount;
+ size_t SectorSize;
+ } Encryption;
+ };
+ };
+
+ static void DoWork (WorkType::Enum type, const EncryptionMode *mode, byte *data, uint64 startUnitNo, uint64 unitCount, size_t sectorSize);
+ static bool IsRunning () { return ThreadPoolRunning; }
+ static void Start ();
+ static void Stop ();
+
+ protected:
+ static void WorkThreadProc ();
+
+ static const size_t MaxThreadCount = 32;
+ static const size_t QueueSize = MaxThreadCount * 2;
+
+ static Mutex DequeueMutex;
+ static volatile size_t DequeuePosition;
+ static volatile size_t EnqueuePosition;
+ static Mutex EnqueueMutex;
+ static list < shared_ptr <Thread> > RunningThreads;
+ static volatile bool StopPending;
+ static size_t ThreadCount;
+ static volatile bool ThreadPoolRunning;
+ static SyncEvent WorkItemCompletedEvent;
+ static WorkItem WorkItemQueue[QueueSize];
+ static SyncEvent WorkItemReadyEvent;
+ };
+}
+
+#endif // TC_HEADER_Volume_EncryptionThreadPool