2007-10-11 Mark Probst <mark.probst@gmail.com>
[mono.git] / mono / metadata / image.c
index 5b7911f6894042bb94304741c12c204bbba61048..d35eb58ed7d60ab6a017eb0111a8bcc1e67ce645 100644 (file)
 #include "tabledefs.h"
 #include "tokentype.h"
 #include "metadata-internals.h"
+#include "profiler-private.h"
 #include "loader.h"
 #include <mono/io-layer/io-layer.h>
 #include <mono/utils/mono-logger.h>
 #include <mono/utils/mono-path.h>
+#include <mono/utils/mono-io-portability.h>
 #include <mono/metadata/class-internals.h>
+#include <mono/metadata/assembly.h>
 #include <sys/types.h>
 #include <sys/stat.h>
 #ifdef HAVE_UNISTD_H
@@ -39,9 +42,7 @@
  * Keeps track of the various assemblies loaded
  */
 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;
 
@@ -111,9 +112,7 @@ mono_images_init (void)
        InitializeCriticalSection (&images_mutex);
 
        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;
 }
@@ -129,9 +128,7 @@ 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);
 }
 
 /**
@@ -562,22 +559,6 @@ mono_image_load_module (MonoImage *image, int idx)
        return image->modules [idx - 1];
 }
 
-static void
-register_guid (gpointer key, gpointer value, gpointer user_data)
-{
-       MonoImage *image = (MonoImage*)value;
-
-       if (!g_hash_table_lookup (loaded_images_guid_hash, image))
-               g_hash_table_insert (loaded_images_guid_hash, image->guid, image);
-}
-
-static void
-build_guid_table (gboolean refonly)
-{
-       GHashTable *loaded_images = refonly ? loaded_images_refonly_hash : loaded_images_hash;
-       g_hash_table_foreach (loaded_images, register_guid, NULL);
-}
-
 static gpointer
 class_key_extract (gpointer value)
 {
@@ -597,7 +578,7 @@ class_next_value (gpointer value)
 void
 mono_image_init (MonoImage *image)
 {
-       image->mempool = mono_mempool_new ();
+       image->mempool = mono_mempool_new_size (512);
        image->method_cache = g_hash_table_new (NULL, NULL);
        mono_internal_hash_table_init (&image->class_cache,
                                       g_direct_hash,
@@ -641,43 +622,35 @@ mono_image_init (MonoImage *image)
        image->method_signatures = g_hash_table_new (NULL, NULL);
 }
 
-static MonoImage *
-do_mono_image_load (MonoImage *image, MonoImageOpenStatus *status,
-                   gboolean care_about_cli)
-{
-       MonoCLIImageInfo *iinfo;
-       MonoDotNetHeader *header;
-       MonoMSDOSHeader msdos;
-       guint32 offset = 0;
-
-       mono_image_init (image);
-
-       iinfo = image->image_info;
-       header = &iinfo->cli_header;
-               
-       if (status)
-               *status = MONO_IMAGE_IMAGE_INVALID;
+#if G_BYTE_ORDER != G_LITTLE_ENDIAN
+#define SWAP64(x) (x) = GUINT64_FROM_LE ((x))
+#define SWAP32(x) (x) = GUINT32_FROM_LE ((x))
+#define SWAP16(x) (x) = GUINT16_FROM_LE ((x))
+#define SWAPPDE(x) do { (x).rva = GUINT32_FROM_LE ((x).rva); (x).size = GUINT32_FROM_LE ((x).size);} while (0)
+#else
+#define SWAP64(x)
+#define SWAP32(x)
+#define SWAP16(x)
+#define SWAPPDE(x)
+#endif
 
-       if (offset + sizeof (msdos) > image->raw_data_len)
-               goto invalid_image;
-       memcpy (&msdos, image->raw_data + offset, sizeof (msdos));
-       
-       if (!(msdos.msdos_sig [0] == 'M' && msdos.msdos_sig [1] == 'Z'))
-               goto invalid_image;
-       
-       msdos.pe_offset = GUINT32_FROM_LE (msdos.pe_offset);
+/*
+ * Returns < 0 to indicate an error.
+ */
+static int
+do_load_header (MonoImage *image, MonoDotNetHeader *header, int offset)
+{
+       MonoDotNetHeader64 header64;
 
-       offset = msdos.pe_offset;
+       if (offset + sizeof (MonoDotNetHeader32) > image->raw_data_len)
+               return -1;
 
-       if (offset + sizeof (MonoDotNetHeader) > image->raw_data_len)
-               goto invalid_image;
        memcpy (header, image->raw_data + offset, sizeof (MonoDotNetHeader));
-       offset += sizeof (MonoDotNetHeader);
 
-#if G_BYTE_ORDER != G_LITTLE_ENDIAN
-#define SWAP32(x) (x) = GUINT32_FROM_LE ((x))
-#define SWAP16(x) (x) = GUINT16_FROM_LE ((x))
-#define SWAPPDE(x) do { (x).rva = GUINT32_FROM_LE ((x).rva); (x).size = GUINT32_FROM_LE ((x).size);} while (0)
+       if (header->pesig [0] != 'P' || header->pesig [1] != 'E')
+               return -1;
+
+       /* endian swap the fields common between PE and PE+ */
        SWAP32 (header->coff.coff_time);
        SWAP32 (header->coff.coff_symptr);
        SWAP32 (header->coff.coff_symcount);
@@ -687,15 +660,71 @@ do_mono_image_load (MonoImage *image, MonoImageOpenStatus *status,
        SWAP16 (header->coff.coff_attributes);
        /* MonoPEHeader */
        SWAP32 (header->pe.pe_code_size);
-       SWAP32 (header->pe.pe_data_size);
        SWAP32 (header->pe.pe_uninit_data_size);
        SWAP32 (header->pe.pe_rva_entry_point);
        SWAP32 (header->pe.pe_rva_code_base);
        SWAP32 (header->pe.pe_rva_data_base);
        SWAP16 (header->pe.pe_magic);
 
+       /* now we are ready for the basic tests */
+
+       if (header->pe.pe_magic == 0x10B) {
+               offset += sizeof (MonoDotNetHeader);
+               SWAP32 (header->pe.pe_data_size);
+               if (header->coff.coff_opt_header_size != (sizeof (MonoDotNetHeader) - sizeof (MonoCOFFHeader) - 4))
+                       return -1;
+
+               SWAP32  (header->nt.pe_image_base);     /* must be 0x400000 */
+               SWAP32  (header->nt.pe_stack_reserve);
+               SWAP32  (header->nt.pe_stack_commit);
+               SWAP32  (header->nt.pe_heap_reserve);
+               SWAP32  (header->nt.pe_heap_commit);
+       } else if (header->pe.pe_magic == 0x20B) {
+               /* PE32+ file format */
+               if (header->coff.coff_opt_header_size != (sizeof (MonoDotNetHeader64) - sizeof (MonoCOFFHeader) - 4))
+                       return -1;
+               memcpy (&header64, image->raw_data + offset, sizeof (MonoDotNetHeader64));
+               offset += sizeof (MonoDotNetHeader64);
+               /* copy the fields already swapped. the last field, pe_data_size, is missing */
+               memcpy (&header64, header, sizeof (MonoDotNetHeader) - 4);
+               /* FIXME: we lose bits here, but we don't use this stuff internally, so we don't care much.
+                * will be fixed when we change MonoDotNetHeader to not match the 32 bit variant
+                */
+               SWAP64  (header64.nt.pe_image_base);
+               header->nt.pe_image_base = header64.nt.pe_image_base;
+               SWAP64  (header64.nt.pe_stack_reserve);
+               header->nt.pe_stack_reserve = header64.nt.pe_stack_reserve;
+               SWAP64  (header64.nt.pe_stack_commit);
+               header->nt.pe_stack_commit = header64.nt.pe_stack_commit;
+               SWAP64  (header64.nt.pe_heap_reserve);
+               header->nt.pe_heap_reserve = header64.nt.pe_heap_reserve;
+               SWAP64  (header64.nt.pe_heap_commit);
+               header->nt.pe_heap_commit = header64.nt.pe_heap_commit;
+
+               header->nt.pe_section_align = header64.nt.pe_section_align;
+               header->nt.pe_file_alignment = header64.nt.pe_file_alignment;
+               header->nt.pe_os_major = header64.nt.pe_os_major;
+               header->nt.pe_os_minor = header64.nt.pe_os_minor;
+               header->nt.pe_user_major = header64.nt.pe_user_major;
+               header->nt.pe_user_minor = header64.nt.pe_user_minor;
+               header->nt.pe_subsys_major = header64.nt.pe_subsys_major;
+               header->nt.pe_subsys_minor = header64.nt.pe_subsys_minor;
+               header->nt.pe_reserved_1 = header64.nt.pe_reserved_1;
+               header->nt.pe_image_size = header64.nt.pe_image_size;
+               header->nt.pe_header_size = header64.nt.pe_header_size;
+               header->nt.pe_checksum = header64.nt.pe_checksum;
+               header->nt.pe_subsys_required = header64.nt.pe_subsys_required;
+               header->nt.pe_dll_flags = header64.nt.pe_dll_flags;
+               header->nt.pe_loader_flags = header64.nt.pe_loader_flags;
+               header->nt.pe_data_dir_count = header64.nt.pe_data_dir_count;
+
+               /* copy the datadir */
+               memcpy (&header->datadir, &header64.datadir, sizeof (MonoPEDatadir));
+       } else {
+               return -1;
+       }
+
        /* MonoPEHeaderNT: not used yet */
-       SWAP32  (header->nt.pe_image_base);     /* must be 0x400000 */
        SWAP32  (header->nt.pe_section_align);       /* must be 8192 */
        SWAP32  (header->nt.pe_file_alignment);      /* must be 512 or 4096 */
        SWAP16  (header->nt.pe_os_major);            /* must be 4 */
@@ -710,10 +739,6 @@ do_mono_image_load (MonoImage *image, MonoImageOpenStatus *status,
        SWAP32  (header->nt.pe_checksum);
        SWAP16  (header->nt.pe_subsys_required);
        SWAP16  (header->nt.pe_dll_flags);
-       SWAP32  (header->nt.pe_stack_reserve);
-       SWAP32  (header->nt.pe_stack_commit);
-       SWAP32  (header->nt.pe_heap_reserve);
-       SWAP32  (header->nt.pe_heap_commit);
        SWAP32  (header->nt.pe_loader_flags);
        SWAP32  (header->nt.pe_data_dir_count);
 
@@ -735,19 +760,49 @@ do_mono_image_load (MonoImage *image, MonoImageOpenStatus *status,
        SWAPPDE (header->datadir.pe_cli_header);
        SWAPPDE (header->datadir.pe_reserved);
 
-#undef SWAP32
-#undef SWAP16
-#undef SWAPPDE
-#endif
+       return offset;
+}
 
-       if (header->coff.coff_machine != 0x14c)
+static MonoImage *
+do_mono_image_load (MonoImage *image, MonoImageOpenStatus *status,
+                   gboolean care_about_cli)
+{
+       MonoCLIImageInfo *iinfo;
+       MonoDotNetHeader *header;
+       MonoMSDOSHeader msdos;
+       gint32 offset = 0;
+
+       mono_profiler_module_event (image, MONO_PROFILE_START_LOAD);
+
+       mono_image_init (image);
+
+       iinfo = image->image_info;
+       header = &iinfo->cli_header;
+               
+       if (status)
+               *status = MONO_IMAGE_IMAGE_INVALID;
+
+       if (offset + sizeof (msdos) > image->raw_data_len)
                goto invalid_image;
+       memcpy (&msdos, image->raw_data + offset, sizeof (msdos));
+       
+       if (!(msdos.msdos_sig [0] == 'M' && msdos.msdos_sig [1] == 'Z'))
+               goto invalid_image;
+       
+       msdos.pe_offset = GUINT32_FROM_LE (msdos.pe_offset);
+
+       offset = msdos.pe_offset;
 
-       if (header->coff.coff_opt_header_size != (sizeof (MonoDotNetHeader) - sizeof (MonoCOFFHeader) - 4))
+       offset = do_load_header (image, header, offset);
+       if (offset < 0)
                goto invalid_image;
 
-       if (header->pesig[0] != 'P' || header->pesig[1] != 'E' || header->pe.pe_magic != 0x10B)
+       /*
+        * this tests for a x86 machine type, but itanium, amd64 and others could be used, too.
+        * we skip this test.
+       if (header->coff.coff_machine != 0x14c)
                goto invalid_image;
+       */
 
 #if 0
        /*
@@ -796,12 +851,14 @@ do_mono_image_load (MonoImage *image, MonoImageOpenStatus *status,
        load_modules (image);
 
 done:
+       mono_profiler_module_loaded (image, MONO_PROFILE_OK);
        if (status)
                *status = MONO_IMAGE_OK;
 
        return image;
 
 invalid_image:
+       mono_profiler_module_loaded (image, MONO_PROFILE_FAILED);
        mono_image_close (image);
                return NULL;
 }
@@ -816,9 +873,19 @@ do_mono_image_open (const char *fname, MonoImageOpenStatus *status,
        struct stat stat_buf;
 
        if ((filed = fopen (fname, "rb")) == NULL){
-               if (status)
-                       *status = MONO_IMAGE_ERROR_ERRNO;
-               return NULL;
+               if (IS_PORTABILITY_SET) {
+                       gchar *ffname = mono_portability_find_file (fname, FALSE);
+                       if (ffname) {
+                               filed = fopen (ffname, "rb");
+                               g_free (ffname);
+                       }
+               }
+
+               if (filed == NULL) {
+                       if (status)
+                               *status = MONO_IMAGE_ERROR_ERRNO;
+                       return NULL;
+               }
        }
 
        if (fstat (fileno (filed), &stat_buf)) {
@@ -828,7 +895,7 @@ do_mono_image_open (const char *fname, MonoImageOpenStatus *status,
                return NULL;
        }
        image = g_new0 (MonoImage, 1);
-       image->file_descr = filed;
+       image->raw_buffer_used = TRUE;
        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);
@@ -837,6 +904,8 @@ do_mono_image_open (const char *fname, MonoImageOpenStatus *status,
        image->ref_only = refonly;
        image->ref_count = 1;
 
+       fclose (filed);
+
        return do_mono_image_load (image, status, care_about_cli);
 }
 
@@ -866,16 +935,36 @@ mono_image_loaded (const char *name)
        return mono_image_loaded_full (name, FALSE);
 }
 
+typedef struct {
+       MonoImage *res;
+       const char* guid;
+} GuidData;
+
+static void
+find_by_guid (gpointer key, gpointer val, gpointer user_data)
+{
+       GuidData *data = user_data;
+       MonoImage *image;
+
+       if (data->res)
+               return;
+       image = val;
+       if (strcmp (data->guid, mono_image_get_guid (image)) == 0)
+               data->res = image;
+}
+
 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;
+       GuidData data;
+       GHashTable *loaded_images = refonly ? loaded_images_refonly_hash : loaded_images_hash;
+       data.res = NULL;
+       data.guid = guid;
 
        mono_images_lock ();
-       res = g_hash_table_lookup (loaded_images, guid);
+       g_hash_table_foreach (loaded_images, find_by_guid, &data);
        mono_images_unlock ();
-       return res;
+       return data.res;
 }
 
 MonoImage *
@@ -903,7 +992,6 @@ register_image (MonoImage *image)
        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;
@@ -1092,7 +1180,7 @@ void
 mono_image_close (MonoImage *image)
 {
        MonoImage *image2;
-       GHashTable *loaded_images, *loaded_images_guid;
+       GHashTable *loaded_images;
        int i;
 
        g_return_if_fail (image != NULL);
@@ -1100,30 +1188,42 @@ mono_image_close (MonoImage *image)
        if (InterlockedDecrement (&image->ref_count) > 0)
                return;
 
+       mono_profiler_module_event (image, MONO_PROFILE_START_UNLOAD);
+
        mono_trace (G_LOG_LEVEL_INFO, MONO_TRACE_ASSEMBLY, "Unloading image %s [%p].", image->name, image);
 
        mono_metadata_clean_for_image (image);
 
+       /*
+        * The caches inside a MonoImage might refer to metadata which is stored in referenced 
+        * assemblies, so we can't release these references in mono_assembly_close () since the
+        * MonoImage might outlive its associated MonoAssembly.
+        */
+       if (image->references) {
+               int i;
+
+               for (i = 0; image->references [i]; i++) {
+                       if (image->references [i])
+                               mono_assembly_close (image->references [i]);
+               }
+
+               g_free (image->references);
+               image->references = NULL;
+       }
+
        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, image->name);
-               g_hash_table_remove (loaded_images_guid, image->guid);
        }
        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);
-               image->file_descr = NULL;
+       if (image->raw_buffer_used) {
                if (image->raw_data != NULL)
                        mono_raw_buffer_free (image->raw_data);
        }
@@ -1270,6 +1370,8 @@ mono_image_close (MonoImage *image)
                }
                mono_mempool_destroy (image->mempool);
        }
+
+       mono_profiler_module_event (image, MONO_PROFILE_END_UNLOAD);
 }
 
 /** 
@@ -1511,8 +1613,11 @@ mono_image_load_file_for_image (MonoImage *image, int fileidx)
        if (fileidx < 1 || fileidx > t->rows)
                return NULL;
 
-       if (image->files && image->files [fileidx - 1])
+       mono_loader_lock ();
+       if (image->files && image->files [fileidx - 1]) {
+               mono_loader_unlock ();
                return image->files [fileidx - 1];
+       }
 
        if (!image->files)
                image->files = g_new0 (MonoImage*, t->rows);
@@ -1533,6 +1638,7 @@ mono_image_load_file_for_image (MonoImage *image, int fileidx)
 
                image->files [fileidx - 1] = res;
        }
+       mono_loader_unlock ();
        g_free (name);
        g_free (base_dir);
        return res;