#include <glib.h>
#include <errno.h>
#include <string.h>
+#include <stdlib.h>
#include "assembly.h"
#include "image.h"
#include "cil-coff.h"
#include "rawbuffer.h"
-#define CSIZE(x) (sizeof (x) / 4)
+/* the default search path is just MONO_ASSEMBLIES */
+static const char*
+default_path [] = {
+ MONO_ASSEMBLIES,
+ NULL
+};
+static char **assemblies_path = NULL;
+static int env_checked = 0;
+
+static void
+check_env (void) {
+ const char *path;
+ char **splitted;
+
+ if (env_checked)
+ return;
+ env_checked = 1;
+
+ path = getenv ("MONO_PATH");
+ if (!path)
+ return;
+ splitted = g_strsplit (path, G_SEARCHPATH_SEPARATOR_S, 1000);
+ if (assemblies_path)
+ g_strfreev (assemblies_path);
+ assemblies_path = splitted;
+}
+
+/*
+ * keeps track of loaded assemblies
+ */
+static GList *loaded_assemblies = NULL;
+static MonoAssembly *corlib;
+
+static MonoAssembly*
+search_loaded (MonoAssemblyName* aname)
+{
+ GList *tmp;
+ MonoAssembly *ass;
+
+ for (tmp = loaded_assemblies; tmp; tmp = tmp->next) {
+ ass = tmp->data;
+ /* we just compare the name, but later we'll do all the checks */
+ /* g_print ("compare %s %s\n", aname->name, ass->aname.name); */
+ if (strcmp (aname->name, ass->aname.name))
+ continue;
+ /* g_print ("success\n"); */
+ return ass;
+ }
+ return NULL;
+}
+
+/**
+ * g_concat_dir_and_file:
+ * @dir: directory name
+ * @file: filename.
+ *
+ * returns a new allocated string that is the concatenation of dir and file,
+ * takes care of the exact details for concatenating them.
+ */
static char *
-default_assembly_name_resolver (const char *name)
+g_concat_dir_and_file (const char *dir, const char *file)
+{
+ g_return_val_if_fail (dir != NULL, NULL);
+ g_return_val_if_fail (file != NULL, NULL);
+
+ /*
+ * If the directory name doesn't have a / on the end, we need
+ * to add one so we get a proper path to the file
+ */
+ if (dir [strlen(dir) - 1] != G_DIR_SEPARATOR)
+ return g_strconcat (dir, G_DIR_SEPARATOR_S, file, NULL);
+ else
+ return g_strconcat (dir, file, NULL);
+}
+
+static MonoAssembly *
+load_in_path (const char *basename, const char** search_path, MonoImageOpenStatus *status)
+{
+ int i;
+ char *fullpath;
+ MonoAssembly *result;
+
+ for (i = 0; search_path [i]; ++i) {
+ fullpath = g_concat_dir_and_file (search_path [i], basename);
+ result = mono_assembly_open (fullpath, status);
+ g_free (fullpath);
+ if (result)
+ return result;
+ }
+ return NULL;
+}
+
+/**
+ * mono_assembly_setrootdir:
+ * @root_dir: The pathname of the root directory where we will locate assemblies
+ *
+ * This routine sets the internal default root directory for looking up
+ * assemblies. This is used by Windows installations to compute dynamically
+ * the place where the Mono assemblies are located.
+ *
+ */
+void
+mono_assembly_setrootdir (const char *root_dir)
{
- if (strcmp (name, "mscorlib") == 0)
- return g_strdup (MONO_ASSEMBLIES "/corlib.dll");
+ /*
+ * Override the MONO_ASSEMBLIES directory configured at compile time.
+ */
+ default_path [0] = g_strdup (root_dir);
+}
+
+static void
+load_references (MonoImage *image, MonoImageOpenStatus *status) {
+ MonoTableInfo *t;
+ guint32 cols [MONO_ASSEMBLYREF_SIZE];
+ const char *hash;
+ int i;
+
+ if (image->references)
+ return;
+
+ t = &image->tables [MONO_TABLE_ASSEMBLYREF];
+
+ image->references = g_new0 (MonoAssembly *, t->rows + 1);
+
+ /*
+ * Load any assemblies this image references
+ */
+ for (i = 0; i < t->rows; i++) {
+ MonoAssemblyName aname;
+
+ mono_metadata_decode_row (t, i, cols, MONO_ASSEMBLYREF_SIZE);
+
+ hash = mono_metadata_blob_heap (image, cols [MONO_ASSEMBLYREF_HASH_VALUE]);
+ aname.hash_len = mono_metadata_decode_blob_size (hash, &hash);
+ aname.hash_value = hash;
+ aname.name = mono_metadata_string_heap (image, cols [MONO_ASSEMBLYREF_NAME]);
+ aname.culture = mono_metadata_string_heap (image, cols [MONO_ASSEMBLYREF_CULTURE]);
+ aname.flags = cols [MONO_ASSEMBLYREF_FLAGS];
+ aname.major = cols [MONO_ASSEMBLYREF_MAJOR_VERSION];
+ aname.minor = cols [MONO_ASSEMBLYREF_MINOR_VERSION];
+ aname.build = cols [MONO_ASSEMBLYREF_BUILD_NUMBER];
+ aname.revision = cols [MONO_ASSEMBLYREF_REV_NUMBER];
+
+ image->references [i] = mono_assembly_load (&aname, image->assembly->basedir, status);
+
+ if (image->references [i] == NULL){
+ int j;
+
+ for (j = 0; j < i; j++)
+ mono_assembly_close (image->references [j]);
+ g_free (image->references);
+ image->references = NULL;
+
+ g_warning ("Could not find assembly %s", aname.name);
+ *status = MONO_IMAGE_MISSING_ASSEMBLYREF;
+ return;
+ }
+ }
+ image->references [i] = NULL;
- return g_strconcat (MONO_ASSEMBLIES "/", name, ".dll", NULL);
}
/**
* mono_assembly_open:
* @filename: Opens the assembly pointed out by this name
- * @resolver: A user provided function to resolve assembly references
* @status: where a status code can be returned
*
* mono_assembly_open opens the PE-image pointed by @filename, and
* it.
*/
MonoAssembly *
-mono_assembly_open (const char *filename, MonoAssemblyResolverFn resolver,
- enum MonoImageOpenStatus *status)
+mono_assembly_open (const char *filename, MonoImageOpenStatus *status)
{
- MonoAssembly *ass;
+ MonoAssembly *ass, *ass2;
MonoImage *image;
- metadata_tableinfo_t *t;
- cli_image_info_t *iinfo;
- metadata_t *m;
- int i, j;
- const char *basename = strrchr (filename, '/');
- static MonoAssembly *corlib;
+ MonoTableInfo *t;
+ guint32 cols [MONO_ASSEMBLY_SIZE];
+ int i;
+ char *base_dir;
+ MonoImageOpenStatus def_status;
- if (basename == NULL)
- basename = filename;
- else
- basename++;
-
- /*
- * Temporary hack until we have a complete corlib.dll
- */
- if (strcmp (basename, "corlib.dll") && corlib != NULL)
- return corlib;
-
g_return_val_if_fail (filename != NULL, NULL);
+ if (!status)
+ status = &def_status;
+ *status = MONO_IMAGE_OK;
+
+ /* g_print ("file loading %s\n", filename); */
image = mono_image_open (filename, status);
+
if (!image){
- if (status)
- *status = MONO_IMAGE_ERROR_ERRNO;
+ *status = MONO_IMAGE_ERROR_ERRNO;
return NULL;
}
- if (resolver == NULL)
- resolver = default_assembly_name_resolver;
-
- iinfo = image->image_info;
- m = &iinfo->cli_metadata;
- t = &m->tables [META_TABLE_ASSEMBLYREF];
-
- image->references = g_new (MonoAssembly *, t->rows + 1);
-
- ass = g_new (MonoAssembly, 1);
- ass->image = image;
+ base_dir = g_path_get_dirname (filename);
/*
- * Load any assemblies this image references
+ * Create assembly struct, and enter it into the assembly cache
*/
- for (i = j = 0; i < t->rows; i++){
- char *assembly_ref;
- const char *name;
- guint32 cols [9];
-
- mono_metadata_decode_row (t, i, cols, CSIZE (cols));
- name = mono_metadata_string_heap (m, cols [6]);
+ ass = g_new0 (MonoAssembly, 1);
+ ass->basedir = base_dir;
+ ass->image = image;
- /*
- * Special case until we have a passable corlib:
- *
- * ie, references to mscorlib from corlib.dll are ignored
- * and we do not load corlib twice.
- */
- if (strcmp (basename, "corlib.dll") == 0){
- if (strcmp (name, "mscorlib") == 0)
- continue;
+ t = &image->tables [MONO_TABLE_ASSEMBLY];
+ if (t->rows) {
+ mono_metadata_decode_row (t, 0, cols, MONO_ASSEMBLY_SIZE);
+
+ ass->aname.hash_len = 0;
+ ass->aname.hash_value = NULL;
+ ass->aname.name = mono_metadata_string_heap (image, cols [MONO_ASSEMBLY_NAME]);
+ ass->aname.culture = mono_metadata_string_heap (image, cols [MONO_ASSEMBLY_CULTURE]);
+ ass->aname.flags = cols [MONO_ASSEMBLY_FLAGS];
+ ass->aname.major = cols [MONO_ASSEMBLY_MAJOR_VERSION];
+ ass->aname.minor = cols [MONO_ASSEMBLY_MINOR_VERSION];
+ ass->aname.build = cols [MONO_ASSEMBLY_BUILD_NUMBER];
+ ass->aname.revision = cols [MONO_ASSEMBLY_REV_NUMBER];
- if (corlib == NULL)
- corlib = ass;
+ /* avoid loading the same assembly twixe for now... */
+ if ((ass2 = search_loaded (&ass->aname))) {
+ g_free (ass);
+ g_free (base_dir);
+ *status = MONO_IMAGE_OK;
+ return ass2;
}
-
- assembly_ref = (*resolver) (name);
+ }
- image->references [i] = mono_assembly_open (assembly_ref, resolver, status);
-
- if (image->references [i] == NULL){
- int j;
-
- for (j = 0; j < i; j++)
- mono_assembly_close (image->references [j]);
- g_free (image->references);
- mono_image_close (image);
+ image->assembly = ass;
- g_warning ("Could not find assembly %s %s", name, assembly_ref);
- g_free (assembly_ref);
- if (status)
- *status = MONO_IMAGE_MISSING_ASSEMBLYREF;
- g_free (ass);
- return NULL;
+ /* register right away to prevent loops */
+ loaded_assemblies = g_list_prepend (loaded_assemblies, ass);
+
+ load_references (image, status);
+ if (*status != MONO_IMAGE_OK) {
+ mono_assembly_close (ass);
+ return NULL;
+ }
+
+ t = &image->tables [MONO_TABLE_MODULEREF];
+ ass->modules = g_new0 (MonoImage *, t->rows);
+ for (i = 0; i < t->rows; i++){
+ char *module_ref;
+ const char *name;
+ guint32 cols [MONO_MODULEREF_SIZE];
+
+ mono_metadata_decode_row (t, i, cols, MONO_MODULEREF_SIZE);
+ name = mono_metadata_string_heap (image, cols [MONO_MODULEREF_NAME]);
+ module_ref = g_concat_dir_and_file (base_dir, name);
+ ass->modules [i] = mono_image_open (module_ref, status);
+ if (ass->modules [i]) {
+ ass->modules [i]->assembly = ass;
+ load_references (ass->modules [i], status);
}
- g_free (assembly_ref);
- j++;
+ /*
+ * FIXME: what do we do here? it could be a native dll...
+ * We should probably do lazy-loading of modules.
+ */
+ *status = MONO_IMAGE_OK;
+ g_free (module_ref);
}
- image->references [j] = NULL;
return ass;
}
+MonoAssembly*
+mono_assembly_load (MonoAssemblyName *aname, const char *basedir, MonoImageOpenStatus *status)
+{
+ MonoAssembly *result;
+ char *fullpath, *filename;
+
+ check_env ();
+
+ /* g_print ("loading %s\n", aname->name); */
+ /* special case corlib */
+ if ((strcmp (aname->name, "mscorlib") == 0) || (strcmp (aname->name, "corlib") == 0)) {
+ if (corlib) {
+ /* g_print ("corlib already loaded\n"); */
+ return corlib;
+ }
+ /* g_print ("corlib load\n"); */
+ if (assemblies_path) {
+ corlib = load_in_path (CORLIB_NAME, (const char**)assemblies_path, status);
+ if (corlib)
+ return corlib;
+ }
+ corlib = load_in_path (CORLIB_NAME, default_path, status);
+ return corlib;
+ }
+ result = search_loaded (aname);
+ if (result)
+ return result;
+ /* g_print ("%s not found in cache\n", aname->name); */
+ if (strstr (aname->name, ".dll"))
+ filename = g_strdup (aname->name);
+ else
+ filename = g_strconcat (aname->name, ".dll", NULL);
+ if (basedir) {
+ fullpath = g_concat_dir_and_file (basedir, filename);
+ result = mono_assembly_open (fullpath, status);
+ g_free (fullpath);
+ if (result) {
+ g_free (filename);
+ return result;
+ }
+ }
+ if (assemblies_path) {
+ result = load_in_path (filename, (const char**)assemblies_path, status);
+ if (result) {
+ g_free (filename);
+ return result;
+ }
+ }
+ result = load_in_path (filename, default_path, status);
+ g_free (filename);
+ return result;
+}
+
void
mono_assembly_close (MonoAssembly *assembly)
{
g_return_if_fail (assembly != NULL);
- for (i = 0; image->references [i] != NULL; i++)
- mono_image_close (assembly->image);
- g_free (image->references);
+ if (--assembly->ref_count != 0)
+ return;
+
+ loaded_assemblies = g_list_remove (loaded_assemblies, assembly);
+ image = assembly->image;
+ if (image->references) {
+ for (i = 0; image->references [i] != NULL; i++)
+ mono_image_close (image->references [i]->image);
+ g_free (image->references);
+ }
mono_image_close (assembly->image);
+
+ g_free (assembly->basedir);
g_free (assembly);
}
+void
+mono_assembly_foreach (GFunc func, gpointer user_data)
+{
+ /* In the future this can do locking of loaded_assemblies */
+
+ g_list_foreach (loaded_assemblies, func, user_data);
+}
+
+/* Holds the assembly of the application, for
+ * System.Diagnostics.Process::MainModule
+ */
+static MonoAssembly *main_assembly=NULL;
+
+void
+mono_assembly_set_main (MonoAssembly *assembly)
+{
+ main_assembly=assembly;
+}
+
+MonoAssembly *
+mono_assembly_get_main (void)
+{
+ return(main_assembly);
+}