2005-01-31 Zoltan Varga <vargaz@freemail.hu>
[mono.git] / mono / metadata / appdomain.c
index 90dbf53ffb00f513de731bcc79760c13051ae28e..1fac818ce60ba3c023303dfa6b650232d5d49315 100644 (file)
 #include <mono/metadata/tabledefs.h>
 #include <mono/metadata/gc-internal.h>
 #include <mono/metadata/marshal.h>
+#include <mono/metadata/threadpool.h>
 #include <mono/utils/mono-uri.h>
 
-#define MONO_CORLIB_VERSION 24
+#define MONO_CORLIB_VERSION 32
 
 CRITICAL_SECTION mono_delegate_section;
 
@@ -41,16 +42,20 @@ mono_domain_assembly_preload (MonoAssemblyName *aname,
                              gchar **assemblies_path,
                              gpointer user_data);
 
+static MonoAssembly *
+mono_domain_assembly_search (MonoAssemblyName *aname,
+                                                        gpointer user_data);
+
 static void
 mono_domain_fire_assembly_load (MonoAssembly *assembly, gpointer user_data);
 
-static MonoMethod *
-look_for_method_by_name (MonoClass *klass, const gchar *name);
+static void
+add_assemblies_to_domain (MonoDomain *domain, MonoAssembly *ass, GHashTable *hash);
 
 static void
 mono_domain_unload (MonoDomain *domain);
 
-/*
+/**
  * mono_runtime_init:
  * @domain: domain returned by mono_init ()
  *
@@ -72,9 +77,11 @@ mono_runtime_init (MonoDomain *domain, MonoThreadStartCB start_cb,
        MonoString *arg;
        
        MONO_GC_PRE_INIT ();
+       mono_thread_pool_init ();
        mono_marshal_init ();
-       
+
        mono_install_assembly_preload_hook (mono_domain_assembly_preload, NULL);
+       mono_install_assembly_search_hook (mono_domain_assembly_search, NULL);
        mono_install_assembly_load_hook (mono_domain_fire_assembly_load, NULL);
        mono_install_lookup_dynamic_token (mono_reflection_lookup_dynamic_token);
 
@@ -117,6 +124,9 @@ mono_runtime_init (MonoDomain *domain, MonoThreadStartCB start_cb,
 
        mono_network_init ();
 
+       /* mscorlib is loaded before we install the load hook */
+       mono_domain_fire_assembly_load (mono_defaults.corlib->assembly, NULL);
+
        return;
 }
 
@@ -178,7 +188,7 @@ mono_runtime_cleanup (MonoDomain *domain)
 static MonoDomainFunc quit_function = NULL;
 
 void
-mono_runtime_install_cleanup (MonoDomainFunc func)
+mono_install_runtime_cleanup (MonoDomainFunc func)
 {
        quit_function = func;
 }
@@ -229,7 +239,7 @@ mono_domain_try_type_resolve (MonoDomain *domain, char *name, MonoObject *tb)
                klass = domain->domain->mbr.obj.vtable->klass;
                g_assert (klass);
 
-               method = look_for_method_by_name (klass, "DoTypeResolve");
+               method = mono_class_get_method_from_name (klass, "DoTypeResolve", -1);
                if (method == NULL) {
                        g_warning ("Method AppDomain.DoTypeResolve not found.\n");
                        return NULL;
@@ -243,19 +253,38 @@ 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
+ * @force: force setting.
  *
  *   Set the current appdomain to @domain. If @force is set, set it even
  * if it is being unloaded.
+ *
  * Returns:
- *   - TRUE on success
- *   - FALSE if the domain is being unloaded
+ *   TRUE on success;
+ *   FALSE if the domain is unloaded
  */
 gboolean
 mono_domain_set (MonoDomain *domain, gboolean force)
 {
-       if (!force && (domain->state == MONO_APPDOMAIN_UNLOADING || domain->state == MONO_APPDOMAIN_UNLOADED))
+       if (!force && domain->state == MONO_APPDOMAIN_UNLOADED)
                return FALSE;
 
        mono_domain_set_internal (domain);
@@ -273,7 +302,9 @@ ves_icall_System_AppDomain_GetData (MonoAppDomain *ad, MonoString *name)
        MONO_ARCH_SAVE_REGS;
 
        g_assert (ad != NULL);
-       g_assert (name != NULL);
+
+       if (name == NULL)
+               mono_raise_exception (mono_get_exception_argument_null ("name"));
 
        str = mono_string_to_utf8 (name);
 
@@ -317,7 +348,9 @@ ves_icall_System_AppDomain_SetData (MonoAppDomain *ad, MonoString *name, MonoObj
        MONO_ARCH_SAVE_REGS;
 
        g_assert (ad != NULL);
-       g_assert (name != NULL);
+
+       if (name == NULL)
+               mono_raise_exception (mono_get_exception_argument_null ("name"));
 
        mono_domain_lock (add);
 
@@ -359,22 +392,23 @@ 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);
+       return root->domain;
 }
 
 MonoAppDomain *
 ves_icall_System_AppDomain_createDomain (MonoString *friendly_name, MonoAppDomainSetup *setup)
 {
-       /*MonoDomain *domain = mono_domain_get (); */
+       MonoDomain *domain = mono_domain_get ();
        MonoClass *adclass;
        MonoAppDomain *ad;
        MonoDomain *data;
+       GSList *tmp;
        
        MONO_ARCH_SAVE_REGS;
 
@@ -390,34 +424,33 @@ ves_icall_System_AppDomain_createDomain (MonoString *friendly_name, MonoAppDomai
        data->friendly_name = mono_string_to_utf8 (friendly_name);
        data->out_of_memory_ex = mono_exception_from_name_domain (data, mono_defaults.corlib, "System", "OutOfMemoryException");
 
+       if (!setup->application_base) {
+               /* Inherit from the root domain since MS.NET does this */
+               MonoDomain *root = mono_get_root_domain ();
+               setup->application_base = mono_string_new_utf16 (data, mono_string_chars (root->setup->application_base), mono_string_length (root->setup->application_base));
+       }
+
        mono_context_init (data);
 
-       /* FIXME: what to do next ? */
+       /* The new appdomain should have all assemblies loaded */
+       mono_domain_lock (domain);
+       /*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;
-} add_assembly_helper_t;
-
-static void
-add_assembly (gpointer key, gpointer value, gpointer user_data)
-{
-       add_assembly_helper_t *ah = (add_assembly_helper_t *) user_data;
-
-       mono_array_set (ah->res, gpointer, ah->idx++, mono_assembly_get_object (ah->domain, value));
-}
-
 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;
 
@@ -425,37 +458,28 @@ ves_icall_System_AppDomain_GetAssemblies (MonoAppDomain *ad)
                System_Reflection_Assembly = mono_class_from_name (
                        mono_defaults.corlib, "System.Reflection", "Assembly");
 
-       res = mono_array_new (domain, System_Reflection_Assembly, g_hash_table_size (domain->assemblies));
-
-       ah.domain = domain;
-       ah.res = res;
-       ah.idx = 0;
+       count = 0;
+       /* Need to skip internal assembly builders created by remoting */
        mono_domain_lock (domain);
-       g_hash_table_foreach (domain->assemblies, 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;
 }
 
-/*
- * Used to find methods in AppDomain class.
- * It only works if there are no multiple signatures for any given method name
- */
-static MonoMethod *
-look_for_method_by_name (MonoClass *klass, const gchar *name)
-{
-       gint i;
-       MonoMethod *method;
-
-       for (i = 0; i < klass->method.count; i++) {
-               method = klass->methods [i];
-               if (!strcmp (method->name, name))
-                       return method;
-       }
-
-       return NULL;
-}
-
 static MonoReflectionAssembly *
 try_assembly_resolve (MonoDomain *domain, MonoString *fname)
 {
@@ -468,7 +492,7 @@ try_assembly_resolve (MonoDomain *domain, MonoString *fname)
        klass = domain->domain->mbr.obj.vtable->klass;
        g_assert (klass);
        
-       method = look_for_method_by_name (klass, "DoAssemblyResolve");
+       method = mono_class_get_method_from_name (klass, "DoAssemblyResolve", -1);
        if (method == NULL) {
                g_warning ("Method AppDomain.DoAssemblyResolve not found.\n");
                return NULL;
@@ -478,27 +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, 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, (gpointer) ass->aname.name, 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
@@ -512,14 +552,15 @@ mono_domain_fire_assembly_load (MonoAssembly *assembly, gpointer user_data)
 
        klass = domain->domain->mbr.obj.vtable->klass;
 
-       
-       method = look_for_method_by_name (klass, "DoAssemblyLoad");
+       method = mono_class_get_method_from_name (klass, "DoAssemblyLoad", -1);
        if (method == NULL) {
                g_warning ("Method AppDomain.DoAssemblyLoad not found.\n");
                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);
@@ -775,6 +816,32 @@ mono_domain_assembly_preload (MonoAssemblyName *aname,
                result = real_load (assemblies_path, aname->culture, aname->name);
        }
 
+       return result;
+}
+
+/*
+ * Check whenever a given assembly was already loaded in the current appdomain.
+ */
+static MonoAssembly *
+mono_domain_assembly_search (MonoAssemblyName *aname,
+                                                        gpointer user_data)
+{
+       MonoDomain *domain = mono_domain_get ();
+       GSList *tmp;
+       MonoAssembly *ass;
+
+       mono_domain_lock (domain);
+       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))
+                       continue;
+
+               mono_domain_unlock (domain);
+               return ass;
+       }
+       mono_domain_unlock (domain);
+
        return NULL;
 }
 
@@ -1022,10 +1089,8 @@ ves_icall_System_AppDomain_ExecuteAssembly (MonoAppDomain *ad, MonoString *file,
        assembly = mono_assembly_open (filename, NULL);
        g_free (filename);
 
-       if (!assembly) {
-               mono_raise_exception ((MonoException *)mono_exception_from_name (
-                       mono_defaults.corlib, "System.IO", "FileNotFoundException"));
-       }
+       if (!assembly)
+               mono_raise_exception (mono_get_exception_file_not_found (file));
 
        image = assembly->image;
 
@@ -1137,7 +1202,7 @@ ves_icall_System_AppDomain_InternalSetContext (MonoAppContext *mc)
        MONO_ARCH_SAVE_REGS;
 
        mono_context_set (mc);
-       
+
        return old_context;
 }
 
@@ -1205,10 +1270,12 @@ unload_thread_main (void *arg)
 
        /* Clear references to our vtables in class->cached_vtable */
        mono_domain_lock (domain);
-       mono_g_hash_table_foreach (domain->class_vtable_hash, clear_cached_vtable,
-                                                          domain);
+       g_hash_table_foreach (domain->class_vtable_hash, clear_cached_vtable,
+                                                 domain);
        mono_domain_unlock (domain);
 
+       mono_threads_clear_cached_culture (domain);
+
        domain->state = MONO_APPDOMAIN_UNLOADED;
 
        /* printf ("UNLOADED %s.\n", domain->friendly_name); */
@@ -1235,13 +1302,14 @@ static void
 mono_domain_unload (MonoDomain *domain)
 {
        HANDLE thread_handle;
-       guint32 tid;
+       guint32 tid, res;
        MonoAppDomainState prev_state;
        MonoMethod *method;
        MonoObject *exc;
        unload_data thread_data;
+       MonoDomain *caller_domain = mono_domain_get ();
 
-       /* printf ("UNLOAD STARTING FOR %s.\n", domain->friendly_name); */
+       /* printf ("UNLOAD STARTING FOR %s (%p) IN THREAD 0x%x.\n", domain->friendly_name, domain, GetCurrentThreadId ()); */
 
        /* Atomically change our state to UNLOADING */
        prev_state = InterlockedCompareExchange (&domain->state,
@@ -1257,8 +1325,9 @@ mono_domain_unload (MonoDomain *domain)
                        g_assert_not_reached ();
        }
 
+       mono_domain_set (domain, FALSE);
        /* Notify OnDomainUnload listeners */
-       method = look_for_method_by_name (domain->domain->mbr.obj.vtable->klass, "DoDomainUnload");     
+       method = mono_class_get_method_from_name (domain->domain->mbr.obj.vtable->klass, "DoDomainUnload", -1); 
        g_assert (method);
 
        exc = NULL;
@@ -1266,6 +1335,7 @@ mono_domain_unload (MonoDomain *domain)
        if (exc) {
                /* Roll back the state change */
                domain->state = MONO_APPDOMAIN_CREATED;
+               mono_domain_set (caller_domain, FALSE);
                mono_raise_exception ((MonoException*)exc);
        }
 
@@ -1284,28 +1354,36 @@ mono_domain_unload (MonoDomain *domain)
 #if 0
        thread_handle = CreateThread (NULL, 0, unload_thread_main, &thread_data, 0, &tid);
 #else
-#if defined(PLATFORM_WIN32) && defined (HAVE_GC_BOEHM)
+#if defined(PLATFORM_WIN32) && defined (HAVE_BOEHM_GC)
        thread_handle = GC_CreateThread (NULL, 0, unload_thread_main, &thread_data, CREATE_SUSPENDED, &tid);
 #else
        thread_handle = CreateThread (NULL, 0, unload_thread_main, &thread_data, CREATE_SUSPENDED, &tid);
 #endif
        ResumeThread (thread_handle);
 #endif
-       
-       while (WaitForSingleObjectEx (thread_handle, INFINITE, FALSE) == WAIT_IO_COMPLETION)
-               ; /* wait for the thread */
-       
+
+       /* Wait for the thread */       
+       while ((res = WaitForSingleObjectEx (thread_handle, INFINITE, TRUE) == WAIT_IO_COMPLETION)) {
+               if (mono_thread_has_appdomain_ref (mono_thread_current (), domain) && (mono_thread_interruption_requested ()))
+                       /* The unload thread tries to abort us */
+                       /* The icall wrapper will execute the abort */
+                       return;
+       }
+
+       mono_domain_set (caller_domain, FALSE);
 
        if (thread_data.failure_reason) {
                MonoException *ex;
 
-               ex = mono_get_exception_cannot_unload_appdomain (thread_data.failure_reason);
                /* Roll back the state change */
                domain->state = MONO_APPDOMAIN_CREATED;
 
                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);
 
                mono_raise_exception (ex);
        }