Fix class initialization during class enumeration

Author: alexey-zakharov

mono/metadata/unity-memory-info.c

@@ -1,5 +1,6 @@
 #include <config.h>
 #include <mono/utils/mono-publib.h>
+
 #include "unity-memory-info.h"
 #include <mono/metadata/assembly-internals.h>
 #include <mono/metadata/class-internals.h>
@@ -8,6 +9,7 @@
 #include <mono/metadata/object-internals.h>
 #include <mono/metadata/tokentype.h>
 #include <mono/metadata/gc-internals.h>
+#include <mono/utils/mono-conc-hashtable.h>
 #include <glib.h>
 #include <stdlib.h>
 
@@ -17,7 +19,7 @@ typedef struct ClassReportContext {
 } ClassReportContext;
 
 static void
-ReportHashMapClasses(gpointer key, gpointer value, gpointer user_data)
+ReportHashMapClass(gpointer key, gpointer value, gpointer user_data)
 {
 	ClassReportContext *context = (ClassReportContext *)user_data;
 	MonoClass *klass = (MonoClass *)value;
@@ -41,6 +43,34 @@ ReportHashMapListClasses(gpointer key, gpointer value, gpointer user_data)
 	}
 }
 
+static void
+ReportGenericClass(gpointer key, gpointer value, gpointer user_data)
+{
+	MonoGenericClass *genericClass = (MonoGenericClass *)key;
+	ClassReportContext *context = (ClassReportContext *)user_data;
+
+	if (genericClass->cached_class != NULL && genericClass->cached_class->inited)
+		context->callback(genericClass->cached_class, context->user_data);
+}
+
+// Iterate MonoInternalHashTable similarly to mono_internal_hash_table_apply, but allow passing custom context
+static void
+IterateMonoInternalHashTable (MonoInternalHashTable *table, gpointer user_data)
+{
+	ClassReportContext *context = (ClassReportContext *)user_data;
+	for (gint i = 0; i < table->size; i++) {
+		gpointer head = table->table [i];
+		while (head) {
+			MonoClass *klass = head;
+
+			if (klass->inited)
+				context->callback(klass, context->user_data);
+
+			head = *(table->next_value (head));
+		}
+	}
+}
+
 static void
 ReportClassesFromAssembly(MonoAssembly *assembly, void *user_data)
 {
@@ -80,48 +110,32 @@ ReportClassesFromAssembly(MonoAssembly *assembly, void *user_data)
 	}
 
 	// Iterate all image classes
-	for (i = 1; i < tdef->rows; ++i) {
-		MonoClass *klass;
-		MonoError error;
-
-		guint32 token = (i + 1) | MONO_TOKEN_TYPE_DEF;
-
-		klass = mono_class_get_checked(image, token, &error);
-
-		if (klass && klass->inited)
-			context->callback(klass, context->user_data);
-	}
+	IterateMonoInternalHashTable(&image->class_cache, user_data);
 
 	if (image->array_cache)
 		g_hash_table_foreach(image->array_cache, ReportHashMapListClasses, user_data);
 
 	if (image->szarray_cache)
-		g_hash_table_foreach(image->szarray_cache, ReportHashMapClasses, user_data);
+		g_hash_table_foreach(image->szarray_cache, ReportHashMapClass, user_data);
 
 	if (image->ptr_cache)
-		g_hash_table_foreach(image->ptr_cache, ReportHashMapClasses, user_data);
-}
-
-static void
-ReportGenericClasses(MonoGenericClass *genericClass, gpointer user_data)
-{
-	ClassReportContext *context = (ClassReportContext *)user_data;
-
-	if (genericClass->cached_class != NULL && genericClass->cached_class->inited)
-		context->callback(genericClass->cached_class, context->user_data);
+		g_hash_table_foreach(image->ptr_cache, ReportHashMapClass, user_data);
 }
 
 static void
 ReportImageSetClasses(MonoImageSet *imageSet, void* user_data)
 {
+	if (imageSet->gclass_cache)
+		mono_conc_hashtable_foreach(imageSet->gclass_cache, ReportGenericClass, user_data);
+
 	if (imageSet->array_cache)
 		g_hash_table_foreach(imageSet->array_cache, ReportHashMapListClasses, user_data);
 
 	if (imageSet->szarray_cache)
-		g_hash_table_foreach(imageSet->szarray_cache, ReportHashMapClasses, user_data);
+		g_hash_table_foreach(imageSet->szarray_cache, ReportHashMapClass, user_data);
 
 	if (imageSet->ptr_cache)
-		g_hash_table_foreach(imageSet->ptr_cache, ReportHashMapClasses, user_data);
+		g_hash_table_foreach(imageSet->ptr_cache, ReportHashMapClass, user_data);
 }
 
 MONO_API void
@@ -132,9 +146,7 @@ mono_unity_class_for_each(ClassReportFunc callback, void *user_data)
 	reportContext.user_data = user_data;
 	// Report all assembly classes and assembly specific arrays
 	mono_domain_assembly_foreach(mono_domain_get(), ReportClassesFromAssembly, &reportContext);
-	// Report all generic classes
-	mono_metadata_generic_class_foreach(ReportGenericClasses, &reportContext);
-	// Report all image set arrays
+	// Report all image set arrays which include generic classes
 	mono_metadata_image_set_foreach(ReportImageSetClasses, &reportContext);
 }