2008-02-27 Zoltan Varga <vargaz@gmail.com>
[mono.git] / mono / metadata / loader.c
index ce103f4f8b7eb427cc1cfbe402c20d71555a1bac..708bd3ba24ed3e6d0494ae1dc2d3cfcb7a71346b 100644 (file)
@@ -75,7 +75,7 @@ mono_loader_cleanup (void)
  * with the appropriate arguments, then return NULL to report the failure. The error 
  * should be propagated until it reaches code which can throw managed exceptions. At that
  * point, an exception should be thrown based on the information returned by
- * mono_loader_get_error (). Then the error should be cleared by calling 
+ * mono_loader_get_last_error (). Then the error should be cleared by calling 
  * mono_loader_clear_error ().
  */
 
@@ -190,6 +190,27 @@ mono_loader_set_error_field_load (MonoClass *klass, const char *member_name)
        set_loader_error (error);
 }
 
+/*
+ * mono_loader_set_error_bad_image:
+ *
+ * Set the loader error for this thread. 
+ */
+void
+mono_loader_set_error_bad_image (char *msg)
+{
+       MonoLoaderError *error;
+
+       if (mono_loader_get_last_error ())
+               return;
+
+       error = g_new0 (MonoLoaderError, 1);
+       error->exception_type = MONO_EXCEPTION_BAD_IMAGE;
+       error->msg = msg;
+
+       set_loader_error (error);
+}      
+
+
 /*
  * mono_loader_get_last_error:
  *
@@ -215,6 +236,7 @@ mono_loader_clear_error (void)
        if (ex) {
                g_free (ex->class_name);
                g_free (ex->assembly_name);
+               g_free (ex->msg);
                g_free (ex);
        
                TlsSetValue (loader_error_thread_id, NULL);
@@ -244,10 +266,10 @@ mono_loader_error_prepare_exception (MonoLoaderError *error)
                
                class_name = mono_string_new (mono_domain_get (), cname);
 
-                ex = mono_get_exception_type_load (class_name, aname);
+               ex = mono_get_exception_type_load (class_name, aname);
                g_free (cname);
                g_free (aname);
-                break;
+               break;
         }
        case MONO_EXCEPTION_MISSING_METHOD: {
                char *cname = g_strdup (error->class_name);
@@ -270,11 +292,11 @@ mono_loader_error_prepare_exception (MonoLoaderError *error)
                class_name = g_strdup_printf ("%s%s%s", cnspace, cnspace ? "." : "", cname);
                
                ex = mono_get_exception_missing_field (class_name, cmembername);
-                g_free (class_name);
+               g_free (class_name);
                g_free (cname);
                g_free (cmembername);
                g_free (cnspace);
-                break;
+               break;
         }
        
        case MONO_EXCEPTION_FILE_NOT_FOUND: {
@@ -293,7 +315,15 @@ mono_loader_error_prepare_exception (MonoLoaderError *error)
                g_free (filename);
                break;
        }
-       
+
+       case MONO_EXCEPTION_BAD_IMAGE: {
+               char *msg = g_strdup (error->msg);
+               mono_loader_clear_error ();
+               ex = mono_get_exception_bad_image_format (msg);
+               g_free (msg);
+               break;
+       }
+
        default:
                g_assert_not_reached ();
        }
@@ -314,12 +344,6 @@ field_from_memberref (MonoImage *image, guint32 token, MonoClass **retklass,
        const char *ptr;
        guint32 idx = mono_metadata_token_index (token);
 
-       if (image->dynamic) {
-               MonoClassField *result = mono_lookup_dynamic_token (image, token, context);
-               *retklass = result->parent;
-               return result;
-       }
-
        mono_metadata_decode_row (&tables [MONO_TABLE_MEMBERREF], idx-1, cols, MONO_MEMBERREF_SIZE);
        nindex = cols [MONO_MEMBERREF_CLASS] >> MONO_MEMBERREF_PARENT_BITS;
        class = cols [MONO_MEMBERREF_CLASS] & MONO_MEMBERREF_PARENT_MASK;
@@ -334,8 +358,9 @@ field_from_memberref (MonoImage *image, guint32 token, MonoClass **retklass,
        case MONO_MEMBERREF_PARENT_TYPEDEF:
                klass = mono_class_get (image, MONO_TOKEN_TYPE_DEF | nindex);
                if (!klass) {
-                       char *name = mono_class_name_from_token (image, MONO_TOKEN_TYPE_REF | nindex);
-                       g_warning ("Missing field %s in class %s (typeref index %d)", fname, name, nindex);
+                       char *name = mono_class_name_from_token (image, MONO_TOKEN_TYPE_DEF | nindex);
+                       g_warning ("Missing field %s in class %s (typedef index %d)", fname, name, nindex);
+                       mono_loader_set_error_type_load (name, image->assembly_name);
                        g_free (name);
                        return NULL;
                }
@@ -348,7 +373,8 @@ field_from_memberref (MonoImage *image, guint32 token, MonoClass **retklass,
                klass = mono_class_from_typeref (image, MONO_TOKEN_TYPE_REF | nindex);
                if (!klass) {
                        char *name = mono_class_name_from_token (image, MONO_TOKEN_TYPE_REF | nindex);
-                       g_warning ("Missing field %s in class %s (typeref index %d)", fname, name, nindex);
+                       g_warning ("missing field %s in class %s (typeref index %d)", fname, name, nindex);
+                       mono_loader_set_error_type_load (name, image->assembly_name);
                        g_free (name);
                        return NULL;
                }
@@ -372,6 +398,7 @@ field_from_memberref (MonoImage *image, guint32 token, MonoClass **retklass,
                mono_class_init (klass);
                g_print ("type in sig: %s\n", klass->name);*/
                klass = mono_class_get_full (image, MONO_TOKEN_TYPE_SPEC | nindex, context);
+               //FIXME can't klass be null?
                mono_class_init (klass);
                if (retklass)
                        *retklass = klass;
@@ -398,7 +425,16 @@ mono_field_from_token (MonoImage *image, guint32 token, MonoClass **retklass,
        MonoClassField *field;
 
        if (image->dynamic) {
-               MonoClassField *result = mono_lookup_dynamic_token (image, token, context);
+               MonoClassField *result;
+               MonoClass *handle_class;
+
+               *retklass = NULL;
+               result = mono_lookup_dynamic_token_class (image, token, TRUE, &handle_class, context);
+               // This checks the memberref type as well
+               if (result && handle_class != mono_defaults.fieldhandle_class) {
+                       mono_loader_set_error_bad_image (g_strdup ("Bad field token."));
+                       return NULL;
+               }
                *retklass = result->parent;
                return result;
        }
@@ -427,7 +463,7 @@ mono_field_from_token (MonoImage *image, guint32 token, MonoClass **retklass,
        }
 
        mono_loader_lock ();
-       if (field && !field->parent->generic_class)
+       if (field && !field->parent->generic_class && !field->parent->generic_container)
                g_hash_table_insert (image->field_cache, GUINT_TO_POINTER (token), field);
        mono_loader_unlock ();
        return field;
@@ -566,7 +602,9 @@ inflate_generic_signature (MonoImage *image, MonoMethodSignature *sig, MonoGener
        if (!context)
                return sig;
 
-       res = mono_metadata_signature_alloc (image, sig->param_count);
+       res = g_malloc0 (sizeof (MonoMethodSignature) + ((gint32)sig->param_count - MONO_ZERO_LEN_ARRAY) * sizeof (MonoType*));
+       res->param_count = sig->param_count;
+       res->sentinelpos = -1;
        res->ret = mono_class_inflate_generic_type (sig->ret, context);
        is_open = mono_class_is_open_constructed_type (res->ret);
        for (i = 0; i < sig->param_count; ++i) {
@@ -609,6 +647,7 @@ inflate_generic_header (MonoMethodHeader *header, MonoGenericContext *context)
                                continue;
                        t = mono_class_inflate_generic_type (&clause->data.catch_class->byval_arg, context);
                        clause->data.catch_class = mono_class_from_mono_type (t);
+                       mono_metadata_free_type (t);
                }
        }
        return res;
@@ -660,7 +699,6 @@ mono_method_get_signature_full (MonoMethod *method, MonoImage *image, guint32 to
                prev_sig = g_hash_table_lookup (image->memberref_signatures, GUINT_TO_POINTER (token));
                if (prev_sig) {
                        /* Somebody got in before us */
-                       /* FIXME: Free sig */
                        sig = prev_sig;
                }
                else
@@ -668,7 +706,16 @@ mono_method_get_signature_full (MonoMethod *method, MonoImage *image, guint32 to
                mono_loader_unlock ();
        }
 
-       sig = inflate_generic_signature (image, sig, context);
+       if (context) {
+               MonoMethodSignature *cached;
+
+               /* This signature is not owned by a MonoMethod, so need to cache */
+               sig = inflate_generic_signature (image, sig, context);
+               cached = mono_metadata_get_inflated_signature (sig, context);
+               if (cached != sig)
+                       mono_metadata_free_inflated_signature (sig);
+               sig = cached;
+       }
 
        return sig;
 }
@@ -727,7 +774,7 @@ method_from_memberref (MonoImage *image, guint32 idx, MonoGenericContext *typesp
                if (!klass) {
                        char *name = mono_class_name_from_token (image, MONO_TOKEN_TYPE_REF | nindex);
                        g_warning ("Missing method %s in assembly %s, type %s", mname, image->name, name);
-                       mono_loader_set_error_method_load (name, mname);
+                       mono_loader_set_error_type_load (name, image->assembly_name);
                        g_free (name);
                        return NULL;
                }
@@ -740,7 +787,7 @@ method_from_memberref (MonoImage *image, guint32 idx, MonoGenericContext *typesp
                if (!klass) {
                        char *name = mono_class_name_from_token (image, MONO_TOKEN_TYPE_SPEC | nindex);
                        g_warning ("Missing method %s in assembly %s, type %s", mname, image->name, name);
-                       mono_loader_set_error_method_load (name, mname);
+                       mono_loader_set_error_type_load (name, image->assembly_name);
                        g_free (name);
                        return NULL;
                }
@@ -750,7 +797,7 @@ method_from_memberref (MonoImage *image, guint32 idx, MonoGenericContext *typesp
                if (!klass) {
                        char *name = mono_class_name_from_token (image, MONO_TOKEN_TYPE_DEF | nindex);
                        g_warning ("Missing method %s in assembly %s, type %s", mname, image->name, name);
-                       mono_loader_set_error_method_load (name, mname);
+                       mono_loader_set_error_type_load (name, image->assembly_name);
                        g_free (name);
                        return NULL;
                }
@@ -833,7 +880,7 @@ method_from_memberref (MonoImage *image, guint32 idx, MonoGenericContext *typesp
 static MonoMethod *
 method_from_methodspec (MonoImage *image, MonoGenericContext *context, guint32 idx)
 {
-       MonoMethod *method, *inflated;
+       MonoMethod *method;
        MonoClass *klass;
        MonoTableInfo *tables = image->tables;
        MonoGenericContext new_context;
@@ -853,32 +900,10 @@ method_from_methodspec (MonoImage *image, MonoGenericContext *context, guint32 i
        param_count = mono_metadata_decode_value (ptr, &ptr);
        g_assert (param_count);
 
-       /*
-        * Be careful with the two contexts here:
-        *
-        * ----------------------------------------
-        * class Foo<S> {
-        *   static void Hello<T> (S s, T t) { }
-        *
-        *   static void Test<U> (U u) {
-        *     Foo<U>.Hello<string> (u, "World");
-        *   }
-        * }
-        * ----------------------------------------
-        *
-        * Let's assume we're currently JITing Foo<int>.Test<long>
-        * (ie. `S' is instantiated as `int' and `U' is instantiated as `long').
-        *
-        * The call to Hello() is encoded with a MethodSpec with a TypeSpec as parent
-        * (MONO_MEMBERREF_PARENT_TYPESPEC).
-        *
-        * The TypeSpec is encoded as `Foo<!!0>', so we need to parse it in the current
-        * context (S=int, U=long) to get the correct `Foo<long>'.
-        * 
-        * After that, we parse the memberref signature in the new context
-        * (S=int, T=uninstantiated) and get the open generic method `Foo<long>.Hello<T>'.
-        *
-        */
+       inst = mono_metadata_parse_generic_inst (image, NULL, param_count, ptr, &ptr);
+       if (context && inst->is_open)
+               inst = mono_metadata_inflate_generic_inst (inst, context);
+
        if ((token & MONO_METHODDEFORREF_MASK) == MONO_METHODDEFORREF_METHODDEF)
                method = mono_get_method_full (image, MONO_TOKEN_METHOD_DEF | nindex, NULL, context);
        else
@@ -886,47 +911,15 @@ method_from_methodspec (MonoImage *image, MonoGenericContext *context, guint32 i
 
        klass = method->klass;
 
-       /* FIXME: Is this invalid metadata?  Should we throw an exception instead?  */
-       g_assert (!klass->generic_container);
-
        if (klass->generic_class) {
                g_assert (method->is_inflated);
                method = ((MonoMethodInflated *) method)->declaring;
        }
 
        new_context.class_inst = klass->generic_class ? klass->generic_class->context.class_inst : NULL;
-
-       /*
-        * When parsing the methodspec signature, we're in the old context again:
-        *
-        * ----------------------------------------
-        * class Foo {
-        *   static void Hello<T> (T t) { }
-        *
-        *   static void Test<U> (U u) {
-        *     Foo.Hello<U> (u);
-        *   }
-        * }
-        * ----------------------------------------
-        *
-        * Let's assume we're currently JITing "Foo.Test<float>".
-        *
-        * In this case, we already parsed the memberref as "Foo.Hello<T>" and the methodspec
-        * signature is "<!!0>".  This means that we must instantiate the method type parameter
-        * `T' from the new method with the method type parameter `U' from the current context;
-        * ie. instantiate the method as `Foo.Hello<float>.
-        */
-
-       inst = mono_metadata_parse_generic_inst (image, NULL, param_count, ptr, &ptr);
-
-       if (context && inst->is_open)
-               inst = mono_metadata_inflate_generic_inst (inst, context);
-
        new_context.method_inst = inst;
 
-       inflated = mono_class_inflate_generic_method_full (method, klass, &new_context);
-
-       return inflated;
+       return mono_class_inflate_generic_method_full (method, klass, &new_context);
 }
 
 struct _MonoDllMap {
@@ -1325,8 +1318,17 @@ mono_get_method_from_token (MonoImage *image, guint32 token, MonoClass *klass,
        int size, i;
        guint32 cols [MONO_TYPEDEF_SIZE];
 
-       if (image->dynamic)
-               return mono_lookup_dynamic_token (image, token, context);
+       if (image->dynamic) {
+               MonoClass *handle_class;
+
+               result = mono_lookup_dynamic_token_class (image, token, TRUE, &handle_class, context);
+               // This checks the memberref type as well
+               if (result && handle_class != mono_defaults.methodhandle_class) {
+                       mono_loader_set_error_bad_image (g_strdup ("Bad method token."));
+                       return NULL;
+               }
+               return result;
+       }
 
        if (table != MONO_TABLE_METHOD) {
                if (table == MONO_TABLE_METHODSPEC) {
@@ -1341,6 +1343,11 @@ mono_get_method_from_token (MonoImage *image, guint32 token, MonoClass *klass,
 
        if (used_context) *used_context = FALSE;
 
+       if (idx > image->tables [MONO_TABLE_METHOD].rows) {
+               mono_loader_set_error_bad_image (g_strdup ("Bad method token."));
+               return NULL;
+       }
+
        mono_metadata_decode_row (&image->tables [MONO_TABLE_METHOD], idx - 1, cols, 6);
 
        if ((cols [2] & METHOD_ATTRIBUTE_PINVOKE_IMPL) ||
@@ -1514,6 +1521,10 @@ mono_get_method_constrained (MonoImage *image, guint32 token, MonoClass *constra
 void
 mono_free_method  (MonoMethod *method)
 {
+       if (mono_profiler_get_events () & MONO_PROFILE_METHOD_EVENTS)
+               mono_profiler_method_free (method);
+       
+       /* FIXME: This hack will go away when the profiler will support freeing methods */
        if (mono_profiler_get_events () != MONO_PROFILE_NONE)
                return;
        
@@ -1548,17 +1559,21 @@ void
 mono_method_get_param_names (MonoMethod *method, const char **names)
 {
        int i, lastp;
-       MonoClass *klass = method->klass;
+       MonoClass *klass;
        MonoTableInfo *methodt;
        MonoTableInfo *paramt;
        guint32 idx;
 
+       if (method->is_inflated)
+               method = ((MonoMethodInflated *) method)->declaring;
+
        if (!mono_method_signature (method)->param_count)
                return;
        for (i = 0; i < mono_method_signature (method)->param_count; ++i)
                names [i] = "";
 
-       if (klass->generic_class || klass->rank) /* copy the names later */
+       klass = method->klass;
+       if (klass->rank)
                return;
 
        mono_class_init (klass);