[PATCH v9 05/12] rtworkq: Introduce a platform startup count.

[Conor McCarthy]

From: Conor McCarthy

---
 dlls/mfplat/tests/mfplat.c   |  6 ------
 dlls/rtworkq/queue.c         | 11 +++++++----
 dlls/rtworkq/tests/rtworkq.c |  4 ----
 3 files changed, 7 insertions(+), 14 deletions(-)

diff --git a/dlls/mfplat/tests/mfplat.c b/dlls/mfplat/tests/mfplat.c
index 5ca8fa7dbbc..a31cdb4a2dc 100644
--- a/dlls/mfplat/tests/mfplat.c
+++ b/dlls/mfplat/tests/mfplat.c
@@ -3997,7 +3997,6 @@ void test_startup_counts(void)
     hr = MFLockPlatform();
     ok(hr == S_OK, "Failed to lock, %#lx.\n", hr);
     hr = MFAllocateWorkQueue(&queue);
-    todo_wine
     ok(hr == MF_E_SHUTDOWN, "Unexpected hr %#lx.\n", hr);
     hr = MFUnlockPlatform();
     ok(hr == S_OK, "Failed to unlock, %#lx.\n", hr);
@@ -4020,7 +4019,6 @@ void test_startup_counts(void)
     hr = MFStartup(MF_VERSION, MFSTARTUP_FULL);
     ok(hr == S_OK, "Failed to start up, hr %#lx.\n", hr);
     /* Startup only locks once. */
-    todo_wine
     check_platform_lock_count(1);
     hr = MFShutdown();
     ok(hr == S_OK, "Failed to shut down, hr %#lx.\n", hr);
@@ -4071,7 +4069,6 @@ void test_startup_counts(void)

/* Platform is in shutdown state if either the lock count or the startup count is <= 0. */
     hr = MFAllocateWorkQueue(&queue);
-    todo_wine
     ok(hr == MF_E_SHUTDOWN, "Unexpected hr %#lx.\n", hr);

/* Platform can be unlocked after shutdown. */
@@ -4155,11 +4152,9 @@ void test_startup_counts(void)
     ok(hr == S_OK, "Failed to shut down, hr %#lx.\n", hr);

hr = MFAllocateWorkQueue(&queue);
-    todo_wine
     ok(hr == MF_E_SHUTDOWN, "Unexpected hr %#lx.\n", hr);

hr = MFCancelWorkItem(key);
-    todo_wine
     ok(hr == MF_E_SHUTDOWN, "Unexpected hr %#lx.\n", hr);

res = wait_async_callback_result(&callback->IMFAsyncCallback_iface, 0, &result);
@@ -4169,7 +4164,6 @@ void test_startup_counts(void)
     ok(hr == S_OK, "Failed to start up, hr %#lx.\n", hr);

/* Shutdown while a scheduled item is pending leaks the internal AsyncResult. */
-    todo_wine
     check_platform_lock_count(2);

hr = MFShutdown();
diff --git a/dlls/rtworkq/queue.c b/dlls/rtworkq/queue.c
index 29ea3c884f2..2250f66bbb9 100644
--- a/dlls/rtworkq/queue.c
+++ b/dlls/rtworkq/queue.c
@@ -66,6 +66,7 @@ static CRITICAL_SECTION_DEBUG queues_critsect_debug =
 };
 static CRITICAL_SECTION queues_section = { &queues_critsect_debug, -1, 0, 0, 0, 0 };

+static LONG startup_count;
 static LONG platform_lock;
 static CO_MTA_USAGE_COOKIE mta_cookie;

@@ -960,7 +961,7 @@ static HRESULT alloc_user_queue(const struct queue_desc *desc, DWORD *queue_id)

*queue_id = RTWQ_CALLBACK_QUEUE_UNDEFINED;

-    if (platform_lock <= 0)
+    if (startup_count <= 0 || platform_lock <= 0)
         return RTWQ_E_SHUTDOWN;

if (!(queue = calloc(1, sizeof(*queue))))
@@ -1212,8 +1213,9 @@ static void init_system_queues(void)

HRESULT WINAPI RtwqStartup(void)
 {
-    if (InterlockedIncrement(&platform_lock) == 1)
+    if (InterlockedIncrement(&startup_count) == 1)
     {
+        RtwqLockPlatform();
         init_system_queues();
     }

@@ -1240,12 +1242,13 @@ static void shutdown_system_queues(void)

HRESULT WINAPI RtwqShutdown(void)
 {
-    if (platform_lock <= 0)
+    if (startup_count <= 0)
         return S_OK;

-    if (InterlockedExchangeAdd(&platform_lock, -1) == 1)
+    if (InterlockedExchangeAdd(&startup_count, -1) == 1)
     {
         shutdown_system_queues();
+        RtwqUnlockPlatform();
     }

return S_OK;
diff --git a/dlls/rtworkq/tests/rtworkq.c b/dlls/rtworkq/tests/rtworkq.c
index 2c8d5f3c297..8d89a5047a4 100644
--- a/dlls/rtworkq/tests/rtworkq.c
+++ b/dlls/rtworkq/tests/rtworkq.c
@@ -298,7 +298,6 @@ static void test_work_queue(void)
     hr = RtwqLockPlatform();
     ok(hr == S_OK, "Failed to lock, %#lx.\n", hr);
     hr = RtwqAllocateWorkQueue(RTWQ_STANDARD_WORKQUEUE, &queue);
-    todo_wine
     ok(hr == RTWQ_E_SHUTDOWN, "Unexpected hr %#lx.\n", hr);
     hr = RtwqUnlockPlatform();
     ok(hr == S_OK, "Failed to unlock, %#lx.\n", hr);
@@ -321,7 +320,6 @@ static void test_work_queue(void)
     hr = RtwqStartup();
     ok(hr == S_OK, "Failed to start up, hr %#lx.\n", hr);
     /* Startup only locks once. */
-    todo_wine
     check_platform_lock_count(1);
     hr = RtwqShutdown();
     ok(hr == S_OK, "Failed to shut down, hr %#lx.\n", hr);
@@ -372,7 +370,6 @@ static void test_work_queue(void)

/* Platform is in shutdown state if either the lock count or the startup count is <= 0. */
     hr = RtwqAllocateWorkQueue(RTWQ_STANDARD_WORKQUEUE, &queue);
-    todo_wine
     ok(hr == RTWQ_E_SHUTDOWN, "Unexpected hr %#lx.\n", hr);

/* Platform can be unlocked after shutdown. */
@@ -469,7 +466,6 @@ static void test_work_queue(void)
     ok(hr == RTWQ_E_SHUTDOWN, "Unexpected hr %#lx.\n", hr);

hr = RtwqCancelWorkItem(key);
-    todo_wine
     ok(hr == RTWQ_E_SHUTDOWN, "Unexpected hr %#lx.\n", hr);

res = wait_async_callback_result(&test_callback->IRtwqAsyncCallback_iface, 0, &callback_result);