2006-06-15 Zoltan Varga <vargaz@gmail.com>
[mono.git] / mono / metadata / image.c
index 9d0369296ac1c8fd94a403d6d09256d2a918281b..e7925f62fb48618739b925d37ba563dd30ec787c 100644 (file)
@@ -22,7 +22,9 @@
 #include "tabledefs.h"
 #include "tokentype.h"
 #include "metadata-internals.h"
+#include "loader.h"
 #include <mono/io-layer/io-layer.h>
+#include <mono/utils/mono-logger.h>
 #include <sys/types.h>
 #include <sys/stat.h>
 #include <unistd.h>
  */
 static GHashTable *loaded_images_hash;
 static GHashTable *loaded_images_guid_hash;
+static GHashTable *loaded_images_refonly_hash;
+static GHashTable *loaded_images_refonly_guid_hash;
 
+static gboolean debug_assembly_unload = FALSE;
+
+#define mono_images_lock() EnterCriticalSection (&images_mutex)
+#define mono_images_unlock() LeaveCriticalSection (&images_mutex)
 static CRITICAL_SECTION images_mutex;
 
 guint32
@@ -107,7 +115,9 @@ canonicalize_path (const char *path)
                        if (backc > 0) {
                                backc--;
                        } else {
-                               if (dest != lastpos) strncpy (dest, lastpos, len + 1);
+                               if (dest != lastpos) 
+                                       /* The two strings can overlap */
+                                       memmove (dest, lastpos, len + 1);
                                dest += len + 1;
                        }
                }
@@ -131,6 +141,26 @@ mono_images_init (void)
 
        loaded_images_hash = g_hash_table_new (g_str_hash, g_str_equal);
        loaded_images_guid_hash = g_hash_table_new (g_str_hash, g_str_equal);
+       loaded_images_refonly_hash = g_hash_table_new (g_str_hash, g_str_equal);
+       loaded_images_refonly_guid_hash = g_hash_table_new (g_str_hash, g_str_equal);
+
+       debug_assembly_unload = getenv ("MONO_DEBUG_ASSEMBLY_UNLOAD") != NULL;
+}
+
+/**
+ * mono_images_cleanup:
+ *
+ *  Free all resources used by this module.
+ */
+void
+mono_images_cleanup (void)
+{
+       DeleteCriticalSection (&images_mutex);
+
+       g_hash_table_destroy (loaded_images_hash);
+       g_hash_table_destroy (loaded_images_guid_hash);
+       g_hash_table_destroy (loaded_images_refonly_hash);
+       g_hash_table_destroy (loaded_images_refonly_guid_hash);
 }
 
 /**
@@ -314,7 +344,12 @@ load_metadata_ptrs (MonoImage *image, MonoCLIImageInfo *iinfo)
        if (strncmp (ptr, "BSJB", 4) == 0){
                guint32 version_string_len;
 
-               ptr += 12;
+               ptr += 4;
+               image->md_version_major = read16 (ptr);
+               ptr += 4;
+               image->md_version_minor = read16 (ptr);
+               ptr += 4;
+
                version_string_len = read32 (ptr);
                ptr += 4;
                image->version = g_strndup (ptr, version_string_len);
@@ -367,7 +402,7 @@ load_metadata_ptrs (MonoImage *image, MonoCLIImageInfo *iinfo)
        g_assert (image->heap_guid.data);
        g_assert (image->heap_guid.size >= 16);
 
-       image->guid = mono_guid_to_string (image->heap_guid.data);
+       image->guid = mono_guid_to_string ((guint8*)image->heap_guid.data);
 
        return TRUE;
 }
@@ -450,6 +485,7 @@ load_modules (MonoImage *image, MonoImageOpenStatus *status)
        MonoTableInfo *t;
        int i;
        char *base_dir;
+       gboolean refonly = image->ref_only;
 
        if (image->modules)
                return;
@@ -466,8 +502,9 @@ load_modules (MonoImage *image, MonoImageOpenStatus *status)
                mono_metadata_decode_row (t, i, cols, MONO_MODULEREF_SIZE);
                name = mono_metadata_string_heap (image, cols [MONO_MODULEREF_NAME]);
                module_ref = g_build_filename (base_dir, name, NULL);
-               image->modules [i] = mono_image_open (module_ref, status);
+               image->modules [i] = mono_image_open_full (module_ref, status, refonly);
                if (image->modules [i]) {
+                       mono_image_addref (image->modules [i]);
                        /* g_print ("loaded module %s from %s (%p)\n", module_ref, image->name, image->assembly); */
                }
                /* 
@@ -551,9 +588,10 @@ register_guid (gpointer key, gpointer value, gpointer user_data)
 }
 
 static void
-build_guid_table (void)
+build_guid_table (gboolean refonly)
 {
-       g_hash_table_foreach (loaded_images_hash, register_guid, NULL);
+       GHashTable *loaded_images = refonly ? loaded_images_refonly_hash : loaded_images_hash;
+       g_hash_table_foreach (loaded_images, register_guid, NULL);
 }
 
 void
@@ -564,7 +602,6 @@ mono_image_init (MonoImage *image)
        image->class_cache = g_hash_table_new (NULL, NULL);
        image->field_cache = g_hash_table_new (NULL, NULL);
        image->name_cache = g_hash_table_new (g_str_hash, g_str_equal);
-       image->array_cache = g_hash_table_new (mono_aligned_addr_hash, NULL);
 
        image->delegate_begin_invoke_cache = 
                g_hash_table_new ((GHashFunc)mono_signature_hash, 
@@ -585,9 +622,19 @@ mono_image_init (MonoImage *image)
        image->synchronized_cache = g_hash_table_new (mono_aligned_addr_hash, NULL);
        image->unbox_wrapper_cache = g_hash_table_new (mono_aligned_addr_hash, NULL);
 
+       image->ldfld_wrapper_cache = g_hash_table_new (mono_aligned_addr_hash, NULL);
+       image->ldflda_wrapper_cache = g_hash_table_new (mono_aligned_addr_hash, NULL);
+       image->ldfld_remote_wrapper_cache = g_hash_table_new (mono_aligned_addr_hash, NULL);
+       image->stfld_wrapper_cache = g_hash_table_new (mono_aligned_addr_hash, NULL);
+       image->stfld_remote_wrapper_cache = g_hash_table_new (mono_aligned_addr_hash, NULL);
+       image->isinst_cache = g_hash_table_new (mono_aligned_addr_hash, NULL);
+       image->castclass_cache = g_hash_table_new (mono_aligned_addr_hash, NULL);
+       image->proxy_isinst_cache = g_hash_table_new (mono_aligned_addr_hash, NULL);
+
        image->typespec_cache = g_hash_table_new (NULL, NULL);
        image->memberref_signatures = g_hash_table_new (NULL, NULL);
        image->helper_signatures = g_hash_table_new (g_str_hash, g_str_equal);
+       image->method_signatures = g_hash_table_new (NULL, NULL);
 }
 
 static MonoImage *
@@ -750,8 +797,6 @@ done:
        if (status)
                *status = MONO_IMAGE_OK;
 
-       image->ref_count=1;
-
        return image;
 
 invalid_image:
@@ -761,7 +806,7 @@ invalid_image:
 
 static MonoImage *
 do_mono_image_open (const char *fname, MonoImageOpenStatus *status,
-                   gboolean care_about_cli)
+                   gboolean care_about_cli, gboolean refonly)
 {
        MonoCLIImageInfo *iinfo;
        MonoImage *image;
@@ -781,65 +826,81 @@ do_mono_image_open (const char *fname, MonoImageOpenStatus *status,
                return NULL;
        }
        image = g_new0 (MonoImage, 1);
-       image->ref_count = 1;
        image->file_descr = filed;
        image->raw_data_len = stat_buf.st_size;
        image->raw_data = mono_raw_buffer_load (fileno (filed), FALSE, 0, stat_buf.st_size);
        iinfo = g_new0 (MonoCLIImageInfo, 1);
        image->image_info = iinfo;
        image->name = canonicalize_path (fname);
+       image->ref_only = refonly;
+       image->ref_count = 1;
 
        return do_mono_image_load (image, status, care_about_cli);
 }
 
 MonoImage *
-mono_image_loaded (const char *name)
+mono_image_loaded_full (const char *name, gboolean refonly)
 {
        MonoImage *res;
+       GHashTable *loaded_images = refonly ? loaded_images_refonly_hash : loaded_images_hash;
         
-       EnterCriticalSection (&images_mutex);
-       res = g_hash_table_lookup (loaded_images_hash, name);
-       LeaveCriticalSection (&images_mutex);
+       mono_images_lock ();
+       res = g_hash_table_lookup (loaded_images, name);
+       mono_images_unlock ();
        return res;
 }
 
 MonoImage *
-mono_image_loaded_by_guid (const char *guid)
+mono_image_loaded (const char *name)
+{
+       return mono_image_loaded_full (name, FALSE);
+}
+
+MonoImage *
+mono_image_loaded_by_guid_full (const char *guid, gboolean refonly)
 {
        MonoImage *res;
+       GHashTable *loaded_images = refonly ? loaded_images_refonly_guid_hash : loaded_images_guid_hash;
 
-       EnterCriticalSection (&images_mutex);
-       res = g_hash_table_lookup (loaded_images_guid_hash, guid);
-       LeaveCriticalSection (&images_mutex);
+       mono_images_lock ();
+       res = g_hash_table_lookup (loaded_images, guid);
+       mono_images_unlock ();
        return res;
 }
 
+MonoImage *
+mono_image_loaded_by_guid (const char *guid)
+{
+       return mono_image_loaded_by_guid_full (guid, FALSE);
+}
+
 static MonoImage *
 register_image (MonoImage *image)
 {
        MonoImage *image2;
+       GHashTable *loaded_images = image->ref_only ? loaded_images_refonly_hash : loaded_images_hash;
 
-       EnterCriticalSection (&images_mutex);
-       image2 = g_hash_table_lookup (loaded_images_hash, image->name);
+       mono_images_lock ();
+       image2 = g_hash_table_lookup (loaded_images, image->name);
 
        if (image2) {
                /* Somebody else beat us to it */
                mono_image_addref (image2);
-               LeaveCriticalSection (&images_mutex);
+               mono_images_unlock ();
                mono_image_close (image);
                return image2;
        }
-       g_hash_table_insert (loaded_images_hash, image->name, image);
-       if (image->assembly_name && (g_hash_table_lookup (loaded_images_hash, image->assembly_name) == NULL))
-               g_hash_table_insert (loaded_images_hash, (char *) image->assembly_name, image); 
-       g_hash_table_insert (loaded_images_guid_hash, image->guid, image);
-       LeaveCriticalSection (&images_mutex);
+       g_hash_table_insert (loaded_images, image->name, image);
+       if (image->assembly_name && (g_hash_table_lookup (loaded_images, image->assembly_name) == NULL))
+               g_hash_table_insert (loaded_images, (char *) image->assembly_name, image);      
+       g_hash_table_insert (image->ref_only ? loaded_images_refonly_guid_hash : loaded_images_guid_hash, image->guid, image);
+       mono_images_unlock ();
 
        return image;
 }
 
 MonoImage *
-mono_image_open_from_data (char *data, guint32 data_len, gboolean need_copy, MonoImageOpenStatus *status)
+mono_image_open_from_data_full (char *data, guint32 data_len, gboolean need_copy, MonoImageOpenStatus *status, gboolean refonly)
 {
        MonoCLIImageInfo *iinfo;
        MonoImage *image;
@@ -862,13 +923,13 @@ mono_image_open_from_data (char *data, guint32 data_len, gboolean need_copy, Mon
        }
 
        image = g_new0 (MonoImage, 1);
-       image->ref_count = 1;
        image->raw_data = datac;
        image->raw_data_len = data_len;
        image->raw_data_allocated = need_copy;
        image->name = g_strdup_printf ("data-%p", datac);
        iinfo = g_new0 (MonoCLIImageInfo, 1);
        image->image_info = iinfo;
+       image->ref_only = refonly;
 
        image = do_mono_image_load (image, status, TRUE);
        if (image == NULL)
@@ -877,18 +938,17 @@ mono_image_open_from_data (char *data, guint32 data_len, gboolean need_copy, Mon
        return register_image (image);
 }
 
-/**
- * mono_image_open:
- * @fname: filename that points to the module we want to open
- * @status: An error condition is returned in this field
- *
- * Returns: An open image of type %MonoImage or NULL on error.
- * if NULL, then check the value of @status for details on the error
- */
 MonoImage *
-mono_image_open (const char *fname, MonoImageOpenStatus *status)
+mono_image_open_from_data (char *data, guint32 data_len, gboolean need_copy, MonoImageOpenStatus *status)
+{
+       return mono_image_open_from_data_full (data, data_len, need_copy, status, FALSE);
+}
+
+MonoImage *
+mono_image_open_full (const char *fname, MonoImageOpenStatus *status, gboolean refonly)
 {
        MonoImage *image;
+       GHashTable *loaded_images;
        char *absfname;
        
        g_return_val_if_fail (fname != NULL, NULL);
@@ -901,24 +961,41 @@ mono_image_open (const char *fname, MonoImageOpenStatus *status)
         * happen outside the mutex, and if multiple threads happen to load
         * the same image, we discard all but the first copy.
         */
-       EnterCriticalSection (&images_mutex);
-       image = g_hash_table_lookup (loaded_images_hash, absfname);
+       mono_images_lock ();
+       loaded_images = refonly ? loaded_images_refonly_hash : loaded_images_hash;
+       image = g_hash_table_lookup (loaded_images, absfname);
        g_free (absfname);
        
        if (image){
                mono_image_addref (image);
-               LeaveCriticalSection (&images_mutex);
+               mono_images_unlock ();
                return image;
        }
-       LeaveCriticalSection (&images_mutex);
+       mono_images_unlock ();
 
-       image = do_mono_image_open (fname, status, TRUE);
+       image = do_mono_image_open (fname, status, TRUE, refonly);
        if (image == NULL)
                return NULL;
 
        return register_image (image);
 }
 
+/**
+ * mono_image_open:
+ * @fname: filename that points to the module we want to open
+ * @status: An error condition is returned in this field
+ *
+ * Returns: An open image of type %MonoImage or NULL on error. 
+ * The caller holds a temporary reference to the returned image which should be cleared 
+ * when no longer needed by calling mono_image_close ().
+ * if NULL, then check the value of @status for details on the error
+ */
+MonoImage *
+mono_image_open (const char *fname, MonoImageOpenStatus *status)
+{
+       return mono_image_open_full (fname, status, FALSE);
+}
+
 /**
  * mono_pe_file_open:
  * @fname: filename that points to the module we want to open
@@ -935,7 +1012,7 @@ mono_pe_file_open (const char *fname, MonoImageOpenStatus *status)
 {
        g_return_val_if_fail (fname != NULL, NULL);
        
-       return(do_mono_image_open (fname, status, FALSE));
+       return(do_mono_image_open (fname, status, FALSE, FALSE));
 }
 
 static void
@@ -944,11 +1021,19 @@ free_hash_table (gpointer key, gpointer val, gpointer user_data)
        g_hash_table_destroy ((GHashTable*)val);
 }
 
+/*
 static void
 free_mr_signatures (gpointer key, gpointer val, gpointer user_data)
 {
        mono_metadata_free_method_signature ((MonoMethodSignature*)val);
 }
+*/
+
+static void
+free_blob_cache_entry (gpointer key, gpointer val, gpointer user_data)
+{
+       g_free (key);
+}
 
 static void
 free_remoting_wrappers (gpointer key, gpointer val, gpointer user_data)
@@ -956,6 +1041,12 @@ free_remoting_wrappers (gpointer key, gpointer val, gpointer user_data)
        g_free (val);
 }
 
+static void
+free_array_cache_entry (gpointer key, gpointer val, gpointer user_data)
+{
+       g_slist_free ((GSList*)val);
+}
+
 /**
  * mono_image_addref:
  * @image: The image file we wish to add a reference to
@@ -991,28 +1082,32 @@ void
 mono_image_close (MonoImage *image)
 {
        MonoImage *image2;
+       GHashTable *loaded_images, *loaded_images_guid;
        int i;
 
        g_return_if_fail (image != NULL);
 
-       EnterCriticalSection (&images_mutex);
-       /*g_print ("destroy image '%s' %p (dynamic: %d) refcount: %d\n", image->name, image, image->dynamic, image->ref_count);*/
-       g_assert (image->ref_count > 0);
-       if (--image->ref_count) {
-               LeaveCriticalSection (&images_mutex);
+       if (InterlockedDecrement (&image->ref_count) > 0)
                return;
-       }
-       image2 = g_hash_table_lookup (loaded_images_hash, image->name);
+
+       mono_trace (G_LOG_LEVEL_INFO, MONO_TRACE_ASSEMBLY, "Unloading image %s [%p].", image->name, image);
+       
+       mono_images_lock ();
+       loaded_images = image->ref_only ? loaded_images_refonly_hash : loaded_images_hash;
+       loaded_images_guid = image->ref_only ? loaded_images_refonly_guid_hash : loaded_images_guid_hash;
+       image2 = g_hash_table_lookup (loaded_images, image->name);
        if (image == image2) {
                /* This is not true if we are called from mono_image_open () */
-               g_hash_table_remove (loaded_images_hash, image->name);
-               g_hash_table_remove (loaded_images_guid_hash, image->guid);
-               /* Multiple images might have the same guid */
-               build_guid_table ();
+               g_hash_table_remove (loaded_images, image->name);
+               g_hash_table_remove (loaded_images_guid, image->guid);
        }
-       if (image->assembly_name && (g_hash_table_lookup (loaded_images_hash, image->assembly_name) == image))
-               g_hash_table_remove (loaded_images_hash, (char *) image->assembly_name);        
-       LeaveCriticalSection (&images_mutex);
+       if (image->assembly_name && (g_hash_table_lookup (loaded_images, image->assembly_name) == image))
+               g_hash_table_remove (loaded_images, (char *) image->assembly_name);     
+
+       /* Multiple images might have the same guid */
+       build_guid_table (image->ref_only);
+
+       mono_images_unlock ();
 
        if (image->file_descr) {
                fclose (image->file_descr);
@@ -1036,14 +1131,25 @@ mono_image_close (MonoImage *image)
 
                g_free (image->raw_data);
        }
-       g_free (image->name);
-       g_free (image->guid);
-       g_free (image->files);
+
+       if (debug_assembly_unload) {
+               image->name = g_strdup_printf ("%s - UNLOADED", image->name);
+       } else {
+               g_free (image->name);
+               g_free (image->guid);
+               g_free (image->version);
+               g_free (image->files);
+       }
 
        g_hash_table_destroy (image->method_cache);
        g_hash_table_destroy (image->class_cache);
        g_hash_table_destroy (image->field_cache);
-       g_hash_table_destroy (image->array_cache);
+       if (image->array_cache) {
+               g_hash_table_foreach (image->array_cache, free_array_cache_entry, NULL);
+               g_hash_table_destroy (image->array_cache);
+       }
+       if (image->ptr_cache)
+               g_hash_table_destroy (image->ptr_cache);
        g_hash_table_foreach (image->name_cache, free_hash_table, NULL);
        g_hash_table_destroy (image->name_cache);
        g_hash_table_destroy (image->native_wrapper_cache);
@@ -1057,11 +1163,26 @@ mono_image_close (MonoImage *image)
        g_hash_table_destroy (image->synchronized_cache);
        g_hash_table_destroy (image->unbox_wrapper_cache);
        g_hash_table_destroy (image->typespec_cache);
-       g_hash_table_foreach (image->memberref_signatures, free_mr_signatures, NULL);
+       g_hash_table_destroy (image->ldfld_wrapper_cache);
+       g_hash_table_destroy (image->ldflda_wrapper_cache);
+       g_hash_table_destroy (image->ldfld_remote_wrapper_cache);
+       g_hash_table_destroy (image->stfld_wrapper_cache);
+       g_hash_table_destroy (image->stfld_remote_wrapper_cache);
+       g_hash_table_destroy (image->isinst_cache);
+       g_hash_table_destroy (image->castclass_cache);
+       g_hash_table_destroy (image->proxy_isinst_cache);
+
+       /* The ownership of signatures is not well defined */
+       //g_hash_table_foreach (image->memberref_signatures, free_mr_signatures, NULL);
        g_hash_table_destroy (image->memberref_signatures);
-       g_hash_table_foreach (image->helper_signatures, free_mr_signatures, NULL);
+       //g_hash_table_foreach (image->helper_signatures, free_mr_signatures, NULL);
        g_hash_table_destroy (image->helper_signatures);
-       
+       g_hash_table_destroy (image->method_signatures);
+
+       if (image->interface_bitset) {
+               mono_unload_interface_ids (image->interface_bitset);
+               mono_bitset_free (image->interface_bitset);
+       }
        if (image->image_info){
                MonoCLIImageInfo *ii = image->image_info;
 
@@ -1076,10 +1197,18 @@ mono_image_close (MonoImage *image)
                if (image->modules [i])
                        mono_image_close (image->modules [i]);
        }
+       if (image->modules)
+               g_free (image->modules);
+       if (image->references)
+               g_free (image->references);
        /*g_print ("destroy image %p (dynamic: %d)\n", image, image->dynamic);*/
        if (!image->dynamic) {
-               mono_mempool_destroy (image->mempool);
-               g_free (image);
+               if (debug_assembly_unload)
+                       mono_mempool_invalidate (image->mempool);
+               else {
+                       mono_mempool_destroy (image->mempool);
+                       g_free (image);
+               }
        } else {
                /* Dynamic images are GC_MALLOCed */
                struct _MonoDynamicImage *di = (struct _MonoDynamicImage*)image;
@@ -1093,8 +1222,10 @@ mono_image_close (MonoImage *image)
                        g_hash_table_destroy (di->handleref);
                if (di->tokens)
                        mono_g_hash_table_destroy (di->tokens);
-               if (di->blob_cache)
+               if (di->blob_cache) {
+                       g_hash_table_foreach (di->blob_cache, free_blob_cache_entry, NULL);
                        g_hash_table_destroy (di->blob_cache);
+               }
                g_list_free (di->array_methods);
                if (di->gen_params)
                        g_ptr_array_free (di->gen_params, TRUE);
@@ -1215,6 +1346,16 @@ mono_image_walk_resource_tree (MonoCLIImageInfo *info, guint32 res_id,
        }
 }
 
+/**
+ * mono_image_lookup_resource:
+ * @image: the image to look up the resource in
+ * @res_id: A MONO_PE_RESOURCE_ID_ that represents the resource ID to lookup.
+ * @lang_id: The language id.
+ * @name: the resource name to lookup.
+ *
+ * Returns: NULL if not found, otherwise a pointer to the in-memory representation
+ * of the given resource.
+ */
 gpointer
 mono_image_lookup_resource (MonoImage *image, guint32 res_id, guint32 lang_id, gunichar2 *name)
 {
@@ -1273,12 +1414,33 @@ mono_image_lookup_resource (MonoImage *image, guint32 res_id, guint32 lang_id, g
        return(NULL);
 }
 
+/** 
+ * mono_image_get_entry_point:
+ * @image: the image where the entry point will be looked up.
+ *
+ * Use this routine to determine the metadata token for method that
+ * has been flagged as the entry point.
+ *
+ * Returns: the token for the entry point method in the image
+ */
 guint32
 mono_image_get_entry_point (MonoImage *image)
 {
        return ((MonoCLIImageInfo*)image->image_info)->cli_cli_header.ch_entry_point;
 }
 
+/**
+ * mono_image_get_resource:
+ * @image: the image where the resource will be looked up.
+ * @offset: The offset to add to the resource
+ * @size: a pointer to an int where the size of the resource will be stored
+ *
+ * This is a low-level routine that fetches a resource from the
+ * metadata that starts at a given @offset.  The @size parameter is
+ * filled with the data field as encoded in the metadata.
+ *
+ * Returns: the pointer to the resource whose offset is @offset.
+ */
 const char*
 mono_image_get_resource (MonoImage *image, guint32 offset, guint32 *size)
 {
@@ -1410,6 +1572,12 @@ mono_image_get_filename (MonoImage *image)
        return image->name;
 }
 
+const char*
+mono_image_get_guid (MonoImage *image)
+{
+       return image->guid;
+}
+
 const MonoTableInfo*
 mono_image_get_table_info (MonoImage *image, int table_id)
 {