2005-12-22 Zoltan Varga <vargaz@gmail.com>
[mono.git] / mono / metadata / assembly.c
index 0bf8486aff7259ce0f6514bd09510a917ec45471..dbae3e620219b4092d34ca2ad784216ab1384a18 100644 (file)
 #include <mono/metadata/mono-config.h>
 #include <mono/utils/mono-digest.h>
 #include <mono/utils/mono-logger.h>
-#ifdef PLATFORM_WIN32
 #include <mono/os/util.h>
-#ifdef _MSC_VER
-       /* not used on Windows - see mono_set_rootdir () */
-       #define MONO_ASSEMBLIES         NULL
-#endif
-#endif
 
 /* AssemblyVersionMap: an assembly name and the assembly version set on which it is based */
 typedef struct  {
@@ -42,10 +36,10 @@ typedef struct  {
        guint8 version_set_index;
 } AssemblyVersionMap;
 
-/* the default search path is just MONO_ASSEMBLIES */
+/* the default search path is empty, the first slot is replaced with the computed value */
 static const char*
 default_path [] = {
-       MONO_ASSEMBLIES,
+       NULL,
        NULL
 };
 
@@ -116,6 +110,8 @@ static GList *loaded_assemblies = NULL;
 static MonoAssembly *corlib;
 
 /* This protects loaded_assemblies and image->references */
+#define mono_assemblies_lock() EnterCriticalSection (&assemblies_mutex)
+#define mono_assemblies_unlock() LeaveCriticalSection (&assemblies_mutex)
 static CRITICAL_SECTION assemblies_mutex;
 
 /* A hastable of thread->assembly list mappings */
@@ -127,12 +123,12 @@ static GHashTable *assemblies_refonly_loading;
 /* If defined, points to the bundled assembly information */
 const MonoBundledAssembly **bundles;
 
-/* Reflection only private hook functions */
-static MonoAssembly* mono_assembly_refonly_invoke_search_hook (MonoAssemblyName *aname);
-
 /* Loaded assembly binding info */
 static GSList *loaded_assembly_bindings = NULL;
 
+static MonoAssembly*
+mono_assembly_invoke_search_hook_internal (MonoAssemblyName *aname, gboolean refonly, gboolean postload);
+
 static gchar*
 encode_public_tok (const guchar *token, gint32 len)
 {
@@ -367,7 +363,7 @@ search_loaded (MonoAssemblyName* aname, gboolean refonly)
        MonoAssembly *ass;
        GList *loading;
 
-       ass = refonly ? mono_assembly_refonly_invoke_search_hook (aname) : mono_assembly_invoke_search_hook (aname);
+       ass = mono_assembly_invoke_search_hook_internal (aname, refonly, FALSE);
        if (ass)
                return ass;
        
@@ -434,6 +430,41 @@ mono_assembly_getrootdir (void)
        return default_path [0];
 }
 
+/**
+ * mono_set_dirs:
+ * @assembly_dir: the base directory for assemblies
+ * @config_dir: the base directory for configuration files
+ *
+ * This routine is used internally and by developers embedding
+ * the runtime into their own applications.  There are a number
+ * of cases to consider: Mono as a system-installed package that
+ * is available on the location preconfigured or Mono in a relocated
+ * location.
+ *
+ * If you are using a system-installed Mono, you can pass NULL
+ * to both parameters.  If you are not, you should compute both
+ * directory values and call this routine.
+ *
+ * The values for a given PREFIX are:
+ *
+ *    assembly_dir: PREFIX/lib
+ *    config_dir:   PREFIX/etc
+ *
+ * Notice that embedders that use Mono in a relocated way must
+ * compute the location at runtime, as they will be in control
+ * of where Mono is installed.
+ */
+void
+mono_set_dirs (const char *assembly_dir, const char *config_dir)
+{
+       if (assembly_dir == NULL)
+               assembly_dir = MONO_ASSEMBLIES;
+       if (config_dir == NULL)
+               config_dir = MONO_CFG_DIR;
+       mono_assembly_setrootdir (assembly_dir);
+       mono_set_config_dir (config_dir);
+}
+
 /**
  * mono_assemblies_init:
  *
@@ -442,9 +473,12 @@ mono_assembly_getrootdir (void)
 void
 mono_assemblies_init (void)
 {
-#ifdef PLATFORM_WIN32
-       mono_set_rootdir ();
-#endif
+       /*
+        * Initialize our internal paths if we have not been initialized yet.
+        * This happens when embedders use Mono.
+        */
+       if (mono_assembly_getrootdir () == NULL)
+               mono_set_rootdir ();
 
        check_path_env ();
        check_extra_gac_path_env ();
@@ -505,6 +539,24 @@ mono_assembly_fill_assembly_name (MonoImage *image, MonoAssemblyName *aname)
        return TRUE;
 }
 
+/*
+ * mono_stringify_assembly_name:
+ *
+ *   Convert @aname into its string format. The returned string is dynamically
+ * allocated and should be freed by the caller.
+ */
+char*
+mono_stringify_assembly_name (MonoAssemblyName *aname)
+{
+       return g_strdup_printf (
+               "%s, Version=%d.%d.%d.%d, Culture=%s%s%s",
+               aname->name,
+               aname->major, aname->minor, aname->build, aname->revision,
+               aname->culture && *aname->culture? aname->culture: "neutral",
+               aname->public_key_token [0] ? ", PublicKeyToken=" : "",
+               aname->public_key_token [0] ? (char *)aname->public_key_token : "");
+}
+
 static gchar*
 assemblyref_public_tok (MonoImage *image, guint32 key_index, guint32 flags)
 {
@@ -633,9 +685,9 @@ mono_assembly_load_reference (MonoImage *image, int index)
         * image->references is shared between threads, so we need to access
         * it inside a critical section.
         */
-       EnterCriticalSection (&assemblies_mutex);
+       mono_assemblies_lock ();
        reference = image->references [index];
-       LeaveCriticalSection (&assemblies_mutex);
+       mono_assemblies_unlock ();
        if (reference)
                return;
 
@@ -679,7 +731,7 @@ mono_assembly_load_reference (MonoImage *image, int index)
                g_free (extra_msg);
        }
 
-       EnterCriticalSection (&assemblies_mutex);
+       mono_assemblies_lock ();
        if (reference == NULL) {
                /* Flag as not found */
                reference = (gpointer)-1;
@@ -689,7 +741,7 @@ mono_assembly_load_reference (MonoImage *image, int index)
 
        if (!image->references [index])
                image->references [index] = reference;
-       LeaveCriticalSection (&assemblies_mutex);
+       mono_assemblies_unlock ();
 
        if (image->references [index] != reference) {
                /* Somebody loaded it before us */
@@ -755,42 +807,37 @@ typedef struct AssemblySearchHook AssemblySearchHook;
 struct AssemblySearchHook {
        AssemblySearchHook *next;
        MonoAssemblySearchFunc func;
+       gboolean refonly;
+       gboolean postload;
        gpointer user_data;
 };
 
 AssemblySearchHook *assembly_search_hook = NULL;
-static AssemblySearchHook *assembly_refonly_search_hook = NULL;
 
-MonoAssembly*
-mono_assembly_invoke_search_hook (MonoAssemblyName *aname)
+static MonoAssembly*
+mono_assembly_invoke_search_hook_internal (MonoAssemblyName *aname, gboolean refonly, gboolean postload)
 {
        AssemblySearchHook *hook;
 
        for (hook = assembly_search_hook; hook; hook = hook->next) {
-               MonoAssembly *ass = hook->func (aname, hook->user_data);
-               if (ass)
-                       return ass;
+               if ((hook->refonly == refonly) && (hook->postload == postload)) {
+                       MonoAssembly *ass = hook->func (aname, hook->user_data);
+                       if (ass)
+                               return ass;
+               }
        }
 
        return NULL;
 }
 
-static MonoAssembly*
-mono_assembly_refonly_invoke_search_hook (MonoAssemblyName *aname)
+MonoAssembly*
+mono_assembly_invoke_search_hook (MonoAssemblyName *aname)
 {
-       AssemblySearchHook *hook;
-
-       for (hook = assembly_refonly_search_hook; hook; hook = hook->next) {
-               MonoAssembly *ass = hook->func (aname, hook->user_data);
-               if (ass)
-                       return ass;
-       }
-
-       return NULL;
+       return mono_assembly_invoke_search_hook_internal (aname, FALSE, FALSE);
 }
 
-void          
-mono_install_assembly_search_hook (MonoAssemblySearchFunc func, gpointer user_data)
+static void
+mono_install_assembly_search_hook_internal (MonoAssemblySearchFunc func, gpointer user_data, gboolean refonly, gboolean postload)
 {
        AssemblySearchHook *hook;
        
@@ -799,22 +846,34 @@ mono_install_assembly_search_hook (MonoAssemblySearchFunc func, gpointer user_da
        hook = g_new0 (AssemblySearchHook, 1);
        hook->func = func;
        hook->user_data = user_data;
+       hook->refonly = refonly;
+       hook->postload = postload;
        hook->next = assembly_search_hook;
        assembly_search_hook = hook;
+}
+
+void          
+mono_install_assembly_search_hook (MonoAssemblySearchFunc func, gpointer user_data)
+{
+       mono_install_assembly_search_hook_internal (func, user_data, FALSE, FALSE);
 }      
 
 void
 mono_install_assembly_refonly_search_hook (MonoAssemblySearchFunc func, gpointer user_data)
 {
-       AssemblySearchHook *hook;
+       mono_install_assembly_search_hook_internal (func, user_data, TRUE, FALSE);
+}
 
-       g_return_if_fail (func != NULL);
+void          
+mono_install_assembly_postload_search_hook (MonoAssemblySearchFunc func, gpointer user_data)
+{
+       mono_install_assembly_search_hook_internal (func, user_data, FALSE, TRUE);
+}      
 
-       hook = g_new0 (AssemblySearchHook, 1);
-       hook->func = func;
-       hook->user_data = user_data;
-       hook->next = assembly_refonly_search_hook;
-       assembly_refonly_search_hook = hook;
+void
+mono_install_assembly_postload_refonly_search_hook (MonoAssemblySearchFunc func, gpointer user_data)
+{
+       mono_install_assembly_search_hook_internal (func, user_data, TRUE, TRUE);
 }
 
 typedef struct AssemblyPreLoadHook AssemblyPreLoadHook;
@@ -960,14 +1019,14 @@ mono_assembly_open_from_bundle (const char *filename, MonoImageOpenStatus *statu
         * we do a very simple search for bundled assemblies: it's not a general 
         * purpose assembly loading mechanism.
         */
-       EnterCriticalSection (&assemblies_mutex);
+       mono_assemblies_lock ();
        for (i = 0; !image && bundles [i]; ++i) {
                if (strcmp (bundles [i]->name, name) == 0) {
                        image = mono_image_open_from_data_full ((char*)bundles [i]->data, bundles [i]->size, FALSE, status, refonly);
                        break;
                }
        }
-       LeaveCriticalSection (&assemblies_mutex);
+       mono_assemblies_unlock ();
        g_free (name);
        if (image) {
                mono_image_addref (image);
@@ -987,9 +1046,9 @@ do_mono_assembly_open (const char *filename, MonoImageOpenStatus *status, gboole
                if (image != NULL)
                        return image;
        }
-       EnterCriticalSection (&assemblies_mutex);
+       mono_assemblies_lock ();
        image = mono_image_open_full (filename, status, refonly);
-       LeaveCriticalSection (&assemblies_mutex);
+       mono_assemblies_unlock ();
 
        return image;
 }
@@ -1127,7 +1186,7 @@ mono_assembly_load_from_full (MonoImage *image, const char*fname,
        /* 
         * Atomically search the loaded list and add ourselves to it if necessary.
         */
-       EnterCriticalSection (&assemblies_mutex);
+       mono_assemblies_lock ();
        if (ass->aname.name) {
                /* avoid loading the same assembly twice for now... */
                ass2 = search_loaded (&ass->aname, refonly);
@@ -1136,7 +1195,7 @@ mono_assembly_load_from_full (MonoImage *image, const char*fname,
                        g_free (base_dir);
                        mono_image_close (image);
                        *status = MONO_IMAGE_OK;
-                       LeaveCriticalSection (&assemblies_mutex);
+                       mono_assemblies_unlock ();
                        return ass2;
                }
        }
@@ -1144,13 +1203,13 @@ mono_assembly_load_from_full (MonoImage *image, const char*fname,
        loading = g_hash_table_lookup (ass_loading, GetCurrentThread ());
        loading = g_list_prepend (loading, ass);
        g_hash_table_insert (ass_loading, GetCurrentThread (), loading);
-       LeaveCriticalSection (&assemblies_mutex);
+       mono_assemblies_unlock ();
 
        image->assembly = ass;
 
        mono_assembly_load_references (image, status);
 
-       EnterCriticalSection (&assemblies_mutex);
+       mono_assemblies_lock ();
 
        loading = g_hash_table_lookup (ass_loading, GetCurrentThread ());
        loading = g_list_remove (loading, ass);
@@ -1160,7 +1219,7 @@ mono_assembly_load_from_full (MonoImage *image, const char*fname,
        else
                g_hash_table_insert (ass_loading, GetCurrentThread (), loading);
        if (*status != MONO_IMAGE_OK) {
-               LeaveCriticalSection (&assemblies_mutex);
+               mono_assemblies_unlock ();
                mono_assembly_close (ass);
                return NULL;
        }
@@ -1169,14 +1228,14 @@ mono_assembly_load_from_full (MonoImage *image, const char*fname,
                ass2 = search_loaded (&ass->aname, refonly);
                if (ass2) {
                        /* Somebody else has loaded the assembly before us */
-                       LeaveCriticalSection (&assemblies_mutex);
+                       mono_assemblies_unlock ();
                        mono_assembly_close (ass);
                        return ass2;
                }
        }
 
        loaded_assemblies = g_list_prepend (loaded_assemblies, ass);
-       LeaveCriticalSection (&assemblies_mutex);
+       mono_assemblies_unlock ();
 
        mono_assembly_invoke_load_hook (ass);
 
@@ -1353,7 +1412,7 @@ parse_assembly_directory_name (const char *name, const char *dirname, MonoAssemb
 }
 
 gboolean
-mono_assembly_name_parse_full (const char *name, MonoAssemblyName *aname, gboolean save_public_key)
+mono_assembly_name_parse_full (const char *name, MonoAssemblyName *aname, gboolean save_public_key, gboolean *is_version_defined)
 {
        gchar *dllname;
        gchar *version = NULL;
@@ -1364,7 +1423,12 @@ mono_assembly_name_parse_full (const char *name, MonoAssemblyName *aname, gboole
        gchar *value;
        gchar **parts;
        gchar **tmp;
+       gboolean version_defined;
 
+       if (!is_version_defined)
+               is_version_defined = &version_defined;
+       *is_version_defined = FALSE;
+       
        parts = tmp = g_strsplit (name, ",", 4);
        if (!tmp || !*tmp) {
                g_strfreev (tmp);
@@ -1378,6 +1442,7 @@ mono_assembly_name_parse_full (const char *name, MonoAssemblyName *aname, gboole
        while (*tmp) {
                value = g_strstrip (*tmp);
                if (!g_ascii_strncasecmp (value, "Version=", 8)) {
+                       *is_version_defined = TRUE;
                        version = g_strstrip (value + 8);
                        tmp++;
                        continue;
@@ -1422,7 +1487,7 @@ mono_assembly_name_parse_full (const char *name, MonoAssemblyName *aname, gboole
 gboolean
 mono_assembly_name_parse (const char *name, MonoAssemblyName *aname)
 {
-       return mono_assembly_name_parse_full (name, aname, FALSE);
+       return mono_assembly_name_parse_full (name, aname, FALSE, NULL);
 }
 
 static MonoAssembly*
@@ -1872,7 +1937,10 @@ mono_assembly_load_full (MonoAssemblyName *aname, const char *basedir, MonoImage
                        return result;
        }
 
-       return NULL;
+       /* Try a postload search hook */
+       result = mono_assembly_invoke_search_hook_internal (aname, refonly, TRUE);
+
+       return result;
 }
 
 MonoAssembly*
@@ -1889,9 +1957,9 @@ mono_assembly_loaded_full (MonoAssemblyName *aname, gboolean refonly)
 
        aname = mono_assembly_remap_version (aname, &maped_aname);
 
-       EnterCriticalSection (&assemblies_mutex);
+       mono_assemblies_lock ();
        res = search_loaded (aname, refonly);
-       LeaveCriticalSection (&assemblies_mutex);
+       mono_assemblies_unlock ();
 
        return res;
 }
@@ -1917,9 +1985,9 @@ mono_assembly_close (MonoAssembly *assembly)
        if (InterlockedDecrement (&assembly->ref_count))
                return;
        
-       EnterCriticalSection (&assemblies_mutex);
+       mono_assemblies_lock ();
        loaded_assemblies = g_list_remove (loaded_assemblies, assembly);
-       LeaveCriticalSection (&assemblies_mutex);
+       mono_assemblies_unlock ();
        /* assemblies belong to domains, so the domain code takes care of unloading the
         * referenced assemblies
         */
@@ -1953,9 +2021,9 @@ mono_assembly_foreach (GFunc func, gpointer user_data)
         * We make a copy of the list to avoid calling the callback inside the 
         * lock, which could lead to deadlocks.
         */
-       EnterCriticalSection (&assemblies_mutex);
+       mono_assemblies_lock ();
        copy = g_list_copy (loaded_assemblies);
-       LeaveCriticalSection (&assemblies_mutex);
+       mono_assemblies_unlock ();
 
        g_list_foreach (loaded_assemblies, func, user_data);