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