Piotr Caban : msvcr110: Add Context::IsCurrentTaskCollectionCanceling implementation.

[Alexandre Julliard]

Module: wine
Branch: master
Commit: 3f55fb72caea5e32d81abfec9a4ea7d64a91a5b8
URL:    https://gitlab.winehq.org/wine/wine/-/commit/3f55fb72caea5e32d81abfec9a4ea7d...

Author: Piotr Caban <piotr(a)codeweavers.com>
Date:   Fri Aug 25 19:47:25 2023 +0200

msvcr110: Add Context::IsCurrentTaskCollectionCanceling implementation.

---

 dlls/msvcrt/concurrency.c | 50 +++++++++++++++++++++++++++++++++++++++--------
 1 file changed, 42 insertions(+), 8 deletions(-)

diff --git a/dlls/msvcrt/concurrency.c b/dlls/msvcrt/concurrency.c
index eb9e8f9ee32..395188c257d 100644
--- a/dlls/msvcrt/concurrency.c
+++ b/dlls/msvcrt/concurrency.c
@@ -103,6 +103,7 @@ typedef struct {
     unsigned int id;
     union allocator_cache_entry *allocator_cache[8];
     LONG blocked;
+    struct _StructuredTaskCollection *task_collection;
 } ExternalContextBase;
 extern const vtable_ptr ExternalContextBase_vtable;
 static void ExternalContextBase_ctor(ExternalContextBase*);
@@ -175,7 +176,7 @@ typedef struct
 } SpinWait;
 
 #define FINISHED_INITIAL 0x80000000
-typedef struct
+typedef struct _StructuredTaskCollection
 {
     void *unk1;
     unsigned int unk2;
@@ -187,6 +188,8 @@ typedef struct
     void *event;
 } _StructuredTaskCollection;
 
+bool __thiscall _StructuredTaskCollection__IsCanceling(_StructuredTaskCollection*);
+
 typedef enum
 {
     TASK_COLLECTION_SUCCESS = 1,
@@ -851,7 +854,17 @@ void __cdecl Context__SpinYield(void)
 /* ?IsCurrentTaskCollectionCanceling(a)Context@Concurrency@@SA_NXZ */
 bool __cdecl Context_IsCurrentTaskCollectionCanceling(void)
 {
-    FIXME("()\n");
+    ExternalContextBase *ctx = (ExternalContextBase*)try_get_current_context();
+
+    TRACE("()\n");
+
+    if (ctx && ctx->context.vtable != &ExternalContextBase_vtable) {
+        ERR("unknown context set\n");
+        return FALSE;
+    }
+
+    if (ctx && ctx->task_collection)
+        return _StructuredTaskCollection__IsCanceling(ctx->task_collection);
     return FALSE;
 }
 
@@ -2166,23 +2179,44 @@ static LONG CALLBACK execute_chore_except(EXCEPTION_POINTERS *pexc, void *_data)
     return EXCEPTION_EXECUTE_HANDLER;
 }
 
+static void CALLBACK execute_chore_finally(BOOL normal, void *data)
+{
+    ExternalContextBase *ctx = (ExternalContextBase*)try_get_current_context();
+    _StructuredTaskCollection *old_collection = data;
+
+    if (ctx && ctx->context.vtable == &ExternalContextBase_vtable)
+        ctx->task_collection = old_collection;
+}
+
 static void execute_chore(_UnrealizedChore *chore,
         _StructuredTaskCollection *task_collection)
 {
+    ExternalContextBase *ctx = (ExternalContextBase*)try_get_current_context();
     struct execute_chore_data data = { chore, task_collection };
+    _StructuredTaskCollection *old_collection;
 
     TRACE("(%p %p)\n", chore, task_collection);
 
-    __TRY
+    if (ctx && ctx->context.vtable == &ExternalContextBase_vtable)
     {
-        if (!((ULONG_PTR)task_collection->exception & ~STRUCTURED_TASK_COLLECTION_STATUS_MASK) &&
-                chore->chore_proc)
-            chore->chore_proc(chore);
+        old_collection = ctx->task_collection;
+        ctx->task_collection = task_collection;
     }
-    __EXCEPT_CTX(execute_chore_except, &data)
+
+    __TRY
     {
+        __TRY
+        {
+            if (!((ULONG_PTR)task_collection->exception & ~STRUCTURED_TASK_COLLECTION_STATUS_MASK) &&
+                    chore->chore_proc)
+                chore->chore_proc(chore);
+        }
+        __EXCEPT_CTX(execute_chore_except, &data)
+        {
+        }
+        __ENDTRY
     }
-    __ENDTRY
+    __FINALLY_CTX(execute_chore_finally, old_collection)
 }
 
 static void CALLBACK chore_wrapper_finally(BOOL normal, void *data)