Improved WorkQueue / WorkItem.

 * make use of C++ paradigms instead of C

Signed-off-by: Toni Uhlig <matzeton@googlemail.com>

2 files changed, 15 insertions, 7 deletions

diff --git a/CRT/DriverThread.cpp b/CRT/DriverThread.cpp
index e2bb2ec..ecedc48 100644
--- a/CRT/DriverThread.cpp
+++ b/CRT/DriverThread.cpp
@@ -192,9 +192,9 @@ void DriverThread::WorkQueue::Stop(void)
     KeSetEvent(&m_wakeEvent, 0, FALSE);
 }
 
-void DriverThread::WorkQueue::Enqueue(PSLIST_ENTRY workItem)
+void DriverThread::WorkQueue::Enqueue(WorkItem * item)
 {
-    if (InterlockedPushEntrySList(&m_work, workItem) == NULL)
+    if (InterlockedPushEntrySList(&m_work, &item->QueueEntry) == NULL)
     {
         // Work queue was empty. So, signal the work queue event in case the
         // worker thread is waiting on the event for more operations.
@@ -243,7 +243,8 @@ NTSTATUS DriverThread::WorkQueue::WorkerInterceptorRoutine(PVOID workerContext)
         {
             PSLIST_ENTRY arg = listEntry;
             listEntry = listEntry->Next;
-            if (wq->m_workerRoutine(arg) != STATUS_SUCCESS)
+            DriverThread::WorkItem * wi = CONTAINING_RECORD(arg, DriverThread::WorkItem, QueueEntry);
+            if (wq->m_workerRoutine(wi) != STATUS_SUCCESS)
             {
                 wq->m_stopWorker = TRUE;
             }
diff --git a/CRT/DriverThread.hpp b/CRT/DriverThread.hpp
index 6ae9d2c..d863a66 100644
--- a/CRT/DriverThread.hpp
+++ b/CRT/DriverThread.hpp
@@ -5,11 +5,11 @@
 
 extern "C" void InterceptorThreadRoutine(PVOID threadContext);
 
-typedef NTSTATUS (*threadRoutine_t)(PVOID);
-typedef NTSTATUS (*workerRoutine_t)(PSLIST_ENTRY);
-
 namespace DriverThread
 {
+class WorkItem;
+typedef NTSTATUS (*threadRoutine_t)(PVOID);
+typedef NTSTATUS (*workerRoutine_t)(WorkItem * item);
 
 class Mutex
 {
@@ -79,6 +79,13 @@ private:
     KSEMAPHORE m_semaphore;
 };
 
+class WorkItem
+{
+public:
+    SLIST_ENTRY QueueEntry;
+    PSLIST_ENTRY WorkListEntry;
+};
+
 class WorkQueue
 {
 public:
@@ -86,7 +93,7 @@ public:
     ~WorkQueue(void);
     NTSTATUS Start(workerRoutine_t workerRoutine);
     void Stop(void);
-    void Enqueue(PSLIST_ENTRY workItem);
+    void Enqueue(WorkItem * item);
 
 private:
     Mutex m_mutex;