2005-01-31 Zoltan Varga <vargaz@freemail.hu>
[mono.git] / mono / metadata / appdomain.c
index 43357d446e6d271824a33d7499602e584b09f764..1fac818ce60ba3c023303dfa6b650232d5d49315 100644 (file)
@@ -50,7 +50,7 @@ static void
 mono_domain_fire_assembly_load (MonoAssembly *assembly, gpointer user_data);
 
 static void
-add_assemblies_to_domain (MonoDomain *domain, MonoAssembly *ass);
+add_assemblies_to_domain (MonoDomain *domain, MonoAssembly *ass, GHashTable *hash);
 
 static void
 mono_domain_unload (MonoDomain *domain);
@@ -253,6 +253,22 @@ mono_domain_try_type_resolve (MonoDomain *domain, char *name, MonoObject *tb)
        return (MonoReflectionAssembly *) mono_runtime_invoke (method, domain->domain, params, NULL);
 }
 
+/**
+ * mono_domain_owns_vtable_slot:
+ *
+ *   Returns whenever VTABLE_SLOT is inside a vtable which belongs to DOMAIN.
+ */
+gboolean
+mono_domain_owns_vtable_slot (MonoDomain *domain, gpointer vtable_slot)
+{
+       gboolean res;
+
+       mono_domain_lock (domain);
+       res = mono_mempool_contains_addr (domain->mp, vtable_slot);
+       mono_domain_unlock (domain);
+       return res;
+}
+
 /**
  * mono_domain_set:
  * @domain: domain
@@ -376,19 +392,13 @@ ves_icall_System_AppDomain_getCurDomain ()
 }
 
 MonoAppDomain *
-ves_icall_System_AppDomain_getDomainByID (gint32 domain_id)
+ves_icall_System_AppDomain_getRootDomain ()
 {
-       MonoDomain *add = mono_domain_get_by_id (domain_id);
+       MonoDomain *root = mono_get_root_domain ();
 
        MONO_ARCH_SAVE_REGS;
 
-       return ((add == NULL) ? NULL : add->domain);
-}
-
-static void
-add_assembly_to_domain (gpointer key, gpointer value, gpointer user_data)
-{
-       add_assemblies_to_domain ((MonoDomain*)user_data, (MonoAssembly*)value);
+       return root->domain;
 }
 
 MonoAppDomain *
@@ -398,6 +408,7 @@ ves_icall_System_AppDomain_createDomain (MonoString *friendly_name, MonoAppDomai
        MonoClass *adclass;
        MonoAppDomain *ad;
        MonoDomain *data;
+       GSList *tmp;
        
        MONO_ARCH_SAVE_REGS;
 
@@ -423,45 +434,23 @@ ves_icall_System_AppDomain_createDomain (MonoString *friendly_name, MonoAppDomai
 
        /* The new appdomain should have all assemblies loaded */
        mono_domain_lock (domain);
-       g_hash_table_foreach (domain->assemblies_by_name, add_assembly_to_domain, data);
+       /*g_print ("copy assemblies from domain %p (%s)\n", domain, domain->friendly_name);*/
+       for (tmp = domain->domain_assemblies; tmp; tmp = tmp->next)
+               add_assemblies_to_domain (data, tmp->data, NULL);
        mono_domain_unlock (domain);
 
        return ad;
 }
 
-typedef struct {
-       MonoArray *res;
-       MonoDomain *domain;
-       int idx, len;
-} add_assembly_helper_t;
-
-static void
-filter_assembly (gpointer key, gpointer value, gpointer user_data)
-{
-       add_assembly_helper_t *ah = (add_assembly_helper_t *) user_data;
-       MonoAssembly *ass = (MonoAssembly*)value;
-
-       if (!ass->corlib_internal)
-               ah->len ++;
-}
-
-static void
-add_assembly (gpointer key, gpointer value, gpointer user_data)
-{
-       add_assembly_helper_t *ah = (add_assembly_helper_t *) user_data;
-       MonoAssembly *ass = (MonoAssembly*)value;
-
-       if (!ass->corlib_internal)
-               mono_array_set (ah->res, gpointer, ah->idx++, mono_assembly_get_object (ah->domain, ass));
-}
-
 MonoArray *
 ves_icall_System_AppDomain_GetAssemblies (MonoAppDomain *ad)
 {
        MonoDomain *domain = ad->data; 
+       MonoAssembly* ass;
        static MonoClass *System_Reflection_Assembly;
        MonoArray *res;
-       add_assembly_helper_t ah;
+       GSList *tmp;
+       int i, count;
        
        MONO_ARCH_SAVE_REGS;
 
@@ -469,20 +458,23 @@ ves_icall_System_AppDomain_GetAssemblies (MonoAppDomain *ad)
                System_Reflection_Assembly = mono_class_from_name (
                        mono_defaults.corlib, "System.Reflection", "Assembly");
 
-       ah.domain = domain;
-       ah.idx = 0;
-       ah.len = 0;
-
+       count = 0;
        /* Need to skip internal assembly builders created by remoting */
        mono_domain_lock (domain);
-       g_hash_table_foreach (domain->assemblies_by_name, filter_assembly, &ah);
-       mono_domain_unlock (domain);
-
-       res = mono_array_new (domain, System_Reflection_Assembly, ah.len);
-       ah.res = res;
-
-       mono_domain_lock (domain);
-       g_hash_table_foreach (domain->assemblies_by_name, add_assembly, &ah);
+       for (tmp = domain->domain_assemblies; tmp; tmp = tmp->next) {
+               ass = tmp->data;
+               if (!ass->corlib_internal)
+                       count++;
+       }
+       res = mono_array_new (domain, System_Reflection_Assembly, count);
+       i = 0;
+       for (tmp = domain->domain_assemblies; tmp; tmp = tmp->next) {
+               ass = tmp->data;
+               if (ass->corlib_internal)
+                       continue;
+               mono_array_set (res, gpointer, i, mono_assembly_get_object (domain, ass));
+               ++i;
+       }
        mono_domain_unlock (domain);
 
        return res;
@@ -510,28 +502,43 @@ try_assembly_resolve (MonoDomain *domain, MonoString *fname)
        return (MonoReflectionAssembly *) mono_runtime_invoke (method, domain->domain, params, NULL);
 }
 
+/*
+ * LOCKING: assumes domain is already locked.
+ */
 static void
-add_assemblies_to_domain (MonoDomain *domain, MonoAssembly *ass)
+add_assemblies_to_domain (MonoDomain *domain, MonoAssembly *ass, GHashTable *ht)
 {
        gint i;
+       GSList *tmp;
+       gboolean destroy_ht = FALSE;
 
        if (!ass->aname.name)
                return;
 
-       mono_domain_lock (domain);
-
-       if (g_hash_table_lookup (domain->assemblies_by_name, ass->aname.name)) {
-               mono_domain_unlock (domain);
-               return; /* This is ok while no lazy loading of assemblies */
+       if (!ht) {
+               ht = g_hash_table_new (mono_aligned_addr_hash, NULL);
+               destroy_ht = TRUE;
        }
 
-       g_hash_table_insert (domain->assemblies_by_name, (gpointer) ass->aname.name, ass);
-       domain->assemblies = g_list_prepend (domain->assemblies, ass);
-       mono_domain_unlock (domain);
+       /* FIXME: handle lazy loaded assemblies */
+       for (tmp = domain->domain_assemblies; tmp; tmp = tmp->next) {
+               g_hash_table_insert (ht, tmp->data, tmp->data);
+       }
+       if (!g_hash_table_lookup (ht, ass)) {
+               mono_assembly_addref (ass);
+               g_hash_table_insert (ht, ass, ass);
+               domain->domain_assemblies = g_slist_prepend (domain->domain_assemblies, ass);
+       }
 
-       if (ass->image->references)
-               for (i = 0; ass->image->references [i] != NULL; i++)
-                       add_assemblies_to_domain (domain, ass->image->references [i]);
+       if (ass->image->references) {
+               for (i = 0; ass->image->references [i] != NULL; i++) {
+                       if (!g_hash_table_lookup (ht, ass->image->references [i])) {
+                               add_assemblies_to_domain (domain, ass->image->references [i], ht);
+                       }
+               }
+       }
+       if (destroy_ht)
+               g_hash_table_destroy (ht);
 }
 
 static void
@@ -551,7 +558,9 @@ mono_domain_fire_assembly_load (MonoAssembly *assembly, gpointer user_data)
                return;
        }
 
-       add_assemblies_to_domain (domain, assembly);
+       mono_domain_lock (domain);
+       add_assemblies_to_domain (domain, assembly, NULL);
+       mono_domain_unlock (domain);
 
        ref_assembly = mono_assembly_get_object (domain, assembly);
        g_assert (ref_assembly);
@@ -818,11 +827,11 @@ mono_domain_assembly_search (MonoAssemblyName *aname,
                                                         gpointer user_data)
 {
        MonoDomain *domain = mono_domain_get ();
-       GList *tmp;
+       GSList *tmp;
        MonoAssembly *ass;
 
        mono_domain_lock (domain);
-       for (tmp = domain->assemblies; tmp; tmp = tmp->next) {
+       for (tmp = domain->domain_assemblies; tmp; tmp = tmp->next) {
                ass = tmp->data;
                /* Dynamic assemblies can't match here in MS.NET */
                if (ass->dynamic || !mono_assembly_names_equal (aname, &ass->aname))
@@ -1262,7 +1271,7 @@ unload_thread_main (void *arg)
        /* Clear references to our vtables in class->cached_vtable */
        mono_domain_lock (domain);
        g_hash_table_foreach (domain->class_vtable_hash, clear_cached_vtable,
-                                                          domain);
+                                                 domain);
        mono_domain_unlock (domain);
 
        mono_threads_clear_cached_culture (domain);
@@ -1372,6 +1381,7 @@ mono_domain_unload (MonoDomain *domain)
                g_warning (thread_data.failure_reason);
 
                g_free (thread_data.failure_reason);
+               thread_data.failure_reason = NULL;
 
                ex = mono_get_exception_cannot_unload_appdomain (thread_data.failure_reason);