Hans Leidekker : wbemprox: Use separate critical sections for tables and table list.

[Alexandre Julliard]

Module: wine
Branch: master
Commit: e7ff2688c8042fb6d7af163480b1d2ec849d0dee
URL:    https://gitlab.winehq.org/wine/wine/-/commit/e7ff2688c8042fb6d7af163480b1d2e...

Author: Hans Leidekker <hans(a)codeweavers.com>
Date:   Tue Mar  5 12:44:43 2024 +0100

wbemprox: Use separate critical sections for tables and table list.

---

 dlls/wbemprox/builtin.c          | 13 +++++---
 dlls/wbemprox/table.c            | 66 ++++++++++++++++++++++++++--------------
 dlls/wbemprox/wbemprox_private.h |  2 ++
 3 files changed, 55 insertions(+), 26 deletions(-)

diff --git a/dlls/wbemprox/builtin.c b/dlls/wbemprox/builtin.c
index 1095450e053..f6991ac2d8a 100644
--- a/dlls/wbemprox/builtin.c
+++ b/dlls/wbemprox/builtin.c
@@ -4374,9 +4374,9 @@ static struct table wmi_builtin_classes[] =
 
 static const struct
 {
-    const WCHAR *name;
-    struct table *classes;
-    unsigned int table_count;
+    const WCHAR  *name;
+    struct table *tables;
+    unsigned int  table_count;
 }
 builtin_namespaces[WBEMPROX_NAMESPACE_LAST] =
 {
@@ -4395,7 +4395,12 @@ void init_table_list( void )
     {
         list_init( &tables[ns] );
         for (i = 0; i < builtin_namespaces[ns].table_count; i++)
-            list_add_tail( &tables[ns], &builtin_namespaces[ns].classes[i].entry );
+        {
+            struct table *table = &builtin_namespaces[ns].tables[i];
+            InitializeCriticalSectionEx( &table->cs, 0, RTL_CRITICAL_SECTION_FLAG_FORCE_DEBUG_INFO );
+            table->cs.DebugInfo->Spare[0] = (DWORD_PTR)(__FILE__ ": table.cs" );
+            list_add_tail( &tables[ns], &table->entry );
+        }
         table_list[ns] = &tables[ns];
     }
 }
diff --git a/dlls/wbemprox/table.c b/dlls/wbemprox/table.c
index 573cf035245..37aa546ad2b 100644
--- a/dlls/wbemprox/table.c
+++ b/dlls/wbemprox/table.c
@@ -18,6 +18,7 @@
 
 #define COBJMACROS
 
+#include <assert.h>
 #include <stdarg.h>
 
 #include "windef.h"
@@ -29,14 +30,14 @@
 
 WINE_DEFAULT_DEBUG_CHANNEL(wbemprox);
 
-static CRITICAL_SECTION table_cs;
+static CRITICAL_SECTION table_list_cs;
 static CRITICAL_SECTION_DEBUG table_debug =
 {
-    0, 0, &table_cs,
+    0, 0, &table_list_cs,
     { &table_debug.ProcessLocksList, &table_debug.ProcessLocksList },
-      0, 0, { (DWORD_PTR)(__FILE__ ": table_cs") }
+      0, 0, { (DWORD_PTR)(__FILE__ ": table_list_cs") }
 };
-static CRITICAL_SECTION table_cs = { &table_debug, -1, 0, 0, 0, 0 };
+static CRITICAL_SECTION table_list_cs = { &table_debug, -1, 0, 0, 0, 0 };
 
 HRESULT get_column_index( const struct table *table, const WCHAR *name, UINT *column )
 {
@@ -337,31 +338,49 @@ void free_columns( struct column *columns, UINT num_cols )
 void free_table( struct table *table )
 {
     if (!table) return;
+    assert( table->flags & TABLE_FLAG_DYNAMIC );
+
+    TRACE("destroying %p\n", table);
 
-    EnterCriticalSection( &table_cs );
     clear_table( table );
-    if (table->flags & TABLE_FLAG_DYNAMIC)
-    {
-        TRACE("destroying %p\n", table);
-        free( (WCHAR *)table->name );
-        free_columns( (struct column *)table->columns, table->num_cols );
-        free( table->data );
-        list_remove( &table->entry );
-        free( table );
-    }
-    LeaveCriticalSection( &table_cs );
+    free( (WCHAR *)table->name );
+    free_columns( (struct column *)table->columns, table->num_cols );
+    free( table->data );
+
+    table->cs.DebugInfo->Spare[0] = 0;
+    DeleteCriticalSection( &table->cs );
+    free( table );
 }
 
 void release_table( struct table *table )
 {
-    if (!InterlockedDecrement( &table->refs )) free_table( table );
-    LeaveCriticalSection( &table_cs );
+    if (!--table->refs)
+    {
+        clear_table( table );
+        if (table->flags & TABLE_FLAG_DYNAMIC)
+        {
+            EnterCriticalSection( &table_list_cs );
+            list_remove( &table->entry );
+            table->removed = TRUE;
+            LeaveCriticalSection( &table_list_cs );
+
+            LeaveCriticalSection( &table->cs );
+            free_table( table );
+            return;
+        }
+    }
+    LeaveCriticalSection( &table->cs );
 }
 
 struct table *grab_table( struct table *table )
 {
-    EnterCriticalSection( &table_cs );
-    InterlockedIncrement( &table->refs );
+    EnterCriticalSection( &table->cs );
+    if (table->removed)
+    {
+        LeaveCriticalSection( &table->cs );
+        return NULL;
+    }
+    table->refs++;
     return table;
 }
 
@@ -398,7 +417,10 @@ struct table *create_table( const WCHAR *name, UINT num_cols, const struct colum
     table->fill               = fill;
     table->flags              = TABLE_FLAG_DYNAMIC;
     table->refs               = 0;
+    table->removed            = FALSE;
     list_init( &table->entry );
+    InitializeCriticalSectionEx( &table->cs, 0, RTL_CRITICAL_SECTION_FLAG_FORCE_DEBUG_INFO );
+    table->cs.DebugInfo->Spare[0] = (DWORD_PTR)(__FILE__ ": table.cs");
     return table;
 }
 
@@ -408,18 +430,18 @@ BOOL add_table( enum wbm_namespace ns, struct table *table )
 
     if (ns == WBEMPROX_NAMESPACE_LAST) return FALSE;
 
-    EnterCriticalSection( &table_cs );
+    EnterCriticalSection( &table_list_cs );
     LIST_FOR_EACH_ENTRY( iter, table_list[ns], struct table, entry )
     {
         if (!wcsicmp( iter->name, table->name ))
         {
             TRACE("table %s already exists\n", debugstr_w(table->name));
-            LeaveCriticalSection( &table_cs );
+            LeaveCriticalSection( &table_list_cs );
             return FALSE;
         }
     }
     list_add_tail( table_list[ns], &table->entry );
-    LeaveCriticalSection( &table_cs );
+    LeaveCriticalSection( &table_list_cs );
 
     TRACE("added %p\n", table);
     return TRUE;
diff --git a/dlls/wbemprox/wbemprox_private.h b/dlls/wbemprox/wbemprox_private.h
index 655e3db27c2..5e585049aa6 100644
--- a/dlls/wbemprox/wbemprox_private.h
+++ b/dlls/wbemprox/wbemprox_private.h
@@ -121,6 +121,8 @@ struct table
     UINT flags;
     struct list entry;
     LONG refs;
+    CRITICAL_SECTION cs;
+    BOOL removed;
 };
 
 struct property