2010-02-02 Zoltan Varga <vargaz@gmail.com>
[mono.git] / mono / mini / mini-trampolines.c
index 535af07cd55ba7a8517efd9646068bbf92f5315b..9b32e4be74af885beabd7f5741dd79c10870e926 100644 (file)
@@ -8,10 +8,6 @@
 #include <mono/metadata/tabledefs.h>
 #include <mono/utils/mono-counters.h>
 
-#ifdef HAVE_VALGRIND_MEMCHECK_H
-#include <valgrind/memcheck.h>
-#endif
-
 #include "mini.h"
 #include "debug-mini.h"
 
@@ -47,6 +43,32 @@ get_unbox_trampoline (MonoGenericSharingContext *gsctx, MonoMethod *m, gpointer
 }
 
 #ifdef MONO_ARCH_HAVE_STATIC_RGCTX_TRAMPOLINE
+
+typedef struct {
+       MonoMethod *m;
+       gpointer addr;
+} RgctxTrampInfo;
+
+static gint
+rgctx_tramp_info_equal (gconstpointer ka, gconstpointer kb)
+{
+       const RgctxTrampInfo *i1 = ka;
+       const RgctxTrampInfo *i2 = kb;
+
+       if (i1->m == i2->m && i1->addr == i2->addr)
+               return 1;
+       else
+               return 0;
+}
+
+static guint
+rgctx_tramp_info_hash (gconstpointer data)
+{
+       const RgctxTrampInfo *info = data;
+
+       return GPOINTER_TO_UINT (info->m) ^ GPOINTER_TO_UINT (info->addr);
+}
+
 /*
  * mono_create_static_rgctx_trampoline:
  *
@@ -59,6 +81,7 @@ get_unbox_trampoline (MonoGenericSharingContext *gsctx, MonoMethod *m, gpointer
  * methods code. These trampolines are similar to the unbox trampolines, they
  * perform the same task as the static rgctx wrappers, but they are smaller/faster,
  * and can be made to work with full AOT.
+ * On PPC addr should be an ftnptr and the return value is an ftnptr too.
  */
 gpointer
 mono_create_static_rgctx_trampoline (MonoMethod *m, gpointer addr)
@@ -66,29 +89,46 @@ mono_create_static_rgctx_trampoline (MonoMethod *m, gpointer addr)
        gpointer ctx;
        gpointer res;
        MonoDomain *domain;
+       RgctxTrampInfo tmp_info;
+       RgctxTrampInfo *info;
+
+#ifdef PPC_USES_FUNCTION_DESCRIPTOR
+       g_assert (((gpointer*)addr) [2] == 0);
+#endif
 
        if (mini_method_get_context (m)->method_inst)
                ctx = mono_method_lookup_rgctx (mono_class_vtable (mono_domain_get (), m->klass), mini_method_get_context (m)->method_inst);
        else
                ctx = mono_class_vtable (mono_domain_get (), m->klass);
 
-       if (mono_aot_only)
-               return mono_aot_get_static_rgctx_trampoline (ctx, addr);
-
        domain = mono_domain_get ();
 
+       /* 
+        * In the AOT case, addr might point to either the method, or to an unbox trampoline,
+        * so make the hash keyed on the m+addr pair.
+        */
        mono_domain_lock (domain);
+       if (!domain_jit_info (domain)->static_rgctx_trampoline_hash)
+               domain_jit_info (domain)->static_rgctx_trampoline_hash = g_hash_table_new (rgctx_tramp_info_hash, rgctx_tramp_info_equal);
+       tmp_info.m = m;
+       tmp_info.addr = addr;
        res = g_hash_table_lookup (domain_jit_info (domain)->static_rgctx_trampoline_hash,
-                                                          m);
+                                                          &tmp_info);
        mono_domain_unlock (domain);
        if (res)
                return res;
 
-       res = mono_arch_get_static_rgctx_trampoline (m, ctx, addr);
+       if (mono_aot_only)
+               res = mono_aot_get_static_rgctx_trampoline (ctx, addr);
+       else
+               res = mono_arch_get_static_rgctx_trampoline (m, ctx, addr);
 
        mono_domain_lock (domain);
        /* Duplicates inserted while we didn't hold the lock are OK */
-       g_hash_table_insert (domain_jit_info (domain)->static_rgctx_trampoline_hash, m, res);
+       info = mono_domain_alloc (domain, sizeof (RgctxTrampInfo));
+       info->m = m;
+       info->addr = addr;
+       g_hash_table_insert (domain_jit_info (domain)->static_rgctx_trampoline_hash, info, res);
        mono_domain_unlock (domain);
 
        return res;
@@ -119,10 +159,9 @@ mono_get_vcall_slot_addr (guint8* code, mgreg_t *regs)
 #ifdef MONO_ARCH_HAVE_IMT
 
 static gpointer*
-mono_convert_imt_slot_to_vtable_slot (gpointer* slot, mgreg_t *regs, guint8 *code, MonoMethod *method, MonoMethod **impl_method, gboolean *need_rgctx_tramp)
+mono_convert_imt_slot_to_vtable_slot (gpointer* slot, mgreg_t *regs, guint8 *code, MonoMethod *method, MonoMethod **impl_method, gboolean *need_rgctx_tramp, gboolean *variance_used)
 {
-       MonoGenericSharingContext *gsctx = mono_get_generic_context_from_code (code);
-       MonoObject *this_argument = mono_arch_find_this_argument (regs, method, gsctx);
+       MonoObject *this_argument = mono_arch_get_this_arg_from_call (NULL, mono_method_signature (method), regs, code);
        MonoVTable *vt = this_argument->vtable;
        int displacement = slot - ((gpointer*)vt);
 
@@ -137,11 +176,11 @@ mono_convert_imt_slot_to_vtable_slot (gpointer* slot, mgreg_t *regs, guint8 *cod
                int interface_offset;
                int imt_slot = MONO_IMT_SIZE + displacement;
 
-               interface_offset = mono_class_interface_offset (vt->klass, imt_method->klass);
+               /*This has to be variance aware since imt_method can be from an interface that vt->klass doesn't directly implement*/
+               interface_offset = mono_class_interface_offset_with_variance (vt->klass, imt_method->klass, variance_used);
 
                if (interface_offset < 0) {
-                       g_print ("%s doesn't implement interface %s\n", mono_type_get_name_full (&vt->klass->byval_arg, 0), mono_type_get_name_full (&imt_method->klass->byval_arg, 0));
-                       g_assert_not_reached ();
+                       g_error ("%s doesn't implement interface %s\n", mono_type_get_name_full (&vt->klass->byval_arg, 0), mono_type_get_name_full (&imt_method->klass->byval_arg, 0));
                }
                mono_vtable_build_imt_slot (vt, mono_method_get_imt_slot (imt_method));
 
@@ -163,7 +202,7 @@ mono_convert_imt_slot_to_vtable_slot (gpointer* slot, mgreg_t *regs, guint8 *cod
                                context.method_inst = ((MonoMethodInflated*)imt_method)->context.method_inst;
                                impl = mono_class_inflate_generic_method (impl, &context);
                        } else {
-                               impl = mono_class_get_vtable_entry (vt->klass, interface_offset + imt_method->slot);
+                               impl = mono_class_get_vtable_entry (vt->klass, interface_offset + mono_method_get_vtable_slot (imt_method));
                        }
 
                        if (mono_method_needs_static_rgctx_invoke (impl, FALSE))
@@ -180,10 +219,11 @@ mono_convert_imt_slot_to_vtable_slot (gpointer* slot, mgreg_t *regs, guint8 *cod
                if (vt->imt_collisions_bitmap & (1 << imt_slot)) {
                        int slot = mono_method_get_vtable_index (imt_method);
                        int vtable_offset;
+                       gpointer *vtable_slot;
 
                        g_assert (slot != -1);
                        vtable_offset = interface_offset + slot;
-                       gpointer *vtable_slot = & (vt->vtable [vtable_offset]);
+                       vtable_slot = & (vt->vtable [vtable_offset]);
 #if DEBUG_IMT
                        printf ("mono_convert_imt_slot_to_vtable_slot: slot %p[%d] is in the IMT, and colliding becomes %p[%d] (interface_offset = %d, method->slot = %d)\n", slot, imt_slot, vtable_slot, vtable_offset, interface_offset, imt_method->slot);
 #endif
@@ -207,14 +247,15 @@ mono_convert_imt_slot_to_vtable_slot (gpointer* slot, mgreg_t *regs, guint8 *cod
 static gpointer
 common_call_trampoline (mgreg_t *regs, guint8 *code, gpointer arg, guint8* tramp, MonoVTable *vt, gpointer *vtable_slot, gboolean need_rgctx_tramp)
 {
-       gpointer addr;
+       gpointer addr, compiled_method;
        gboolean generic_shared = FALSE;
        MonoMethod *m;
        MonoMethod *declaring = NULL;
-       MonoMethod *generic_virtual = NULL;
+       MonoMethod *generic_virtual = NULL, *variant_iface = NULL;
        int context_used;
-       gboolean proxy = FALSE;
+       gboolean proxy = FALSE, variance_used = FALSE;
        gpointer *orig_vtable_slot;
+       MonoJitInfo *ji = NULL;
 
        m = arg;
 
@@ -269,7 +310,6 @@ common_call_trampoline (mgreg_t *regs, guint8 *code, gpointer arg, guint8* tramp
 #ifdef MONO_ARCH_HAVE_IMT
        if (m == MONO_FAKE_IMT_METHOD) {
                MonoMethod *impl_method;
-               MonoGenericSharingContext *gsctx;
                MonoObject *this_arg;
 
                /* we get the interface method because mono_convert_imt_slot_to_vtable_slot ()
@@ -279,15 +319,14 @@ common_call_trampoline (mgreg_t *regs, guint8 *code, gpointer arg, guint8* tramp
                vtable_slot = orig_vtable_slot;
                g_assert (vtable_slot);
 
-               gsctx = mono_get_generic_context_from_code (code);
-               this_arg = mono_arch_find_this_argument (regs, m, gsctx);
+               this_arg = mono_arch_get_this_arg_from_call (NULL, mono_method_signature (m), regs, code);
 
                if (this_arg->vtable->klass == mono_defaults.transparent_proxy_class) {
                        /* Use the slow path for now */
                        proxy = TRUE;
                    m = mono_object_get_virtual_method (this_arg, m);
                } else {
-                       vtable_slot = mono_convert_imt_slot_to_vtable_slot (vtable_slot, regs, code, m, &impl_method, &need_rgctx_tramp);
+                       vtable_slot = mono_convert_imt_slot_to_vtable_slot (vtable_slot, regs, code, m, &impl_method, &need_rgctx_tramp, &variance_used);
                        /* mono_convert_imt_slot_to_vtable_slot () also gives us the method that is supposed
                         * to be called, so we compile it and go ahead as usual.
                         */
@@ -295,11 +334,25 @@ common_call_trampoline (mgreg_t *regs, guint8 *code, gpointer arg, guint8* tramp
                        if (m->is_inflated && ((MonoMethodInflated*)m)->context.method_inst) {
                                /* Generic virtual method */
                                generic_virtual = m;
-                               m = impl_method;
                                need_rgctx_tramp = TRUE;
-                       } else {
-                               m = impl_method;
+                       } else if (variance_used && mono_class_has_variant_generic_params (m->klass)) {
+                               variant_iface = m;
                        }
+                       m = impl_method;
+               }
+       }
+#endif
+
+#ifdef MONO_ARCH_LLVM_SUPPORTED
+       if (!vtable_slot && code && !need_rgctx_tramp && mono_method_needs_static_rgctx_invoke (m, FALSE)) {
+               /*
+                * Call this only if the called method is shared, cause it is slow/loads a lot of
+                * data in AOT.
+                */
+               ji = mono_jit_info_table_find (mono_domain_get (), (char*)code);
+               if (ji && ji->from_llvm) {
+                       /* LLVM can't pass an rgctx arg */
+                       need_rgctx_tramp = TRUE;
                }
        }
 #endif
@@ -359,8 +412,7 @@ common_call_trampoline (mgreg_t *regs, guint8 *code, gpointer arg, guint8* tramp
 #endif
                } else {
 #ifdef MONO_ARCH_HAVE_IMT
-                       MonoObject *this_argument = mono_arch_find_this_argument (regs, m,
-                               mono_get_generic_context_from_code (code));
+                       MonoObject *this_argument = mono_arch_get_this_arg_from_call (NULL, mono_method_signature (m), regs, code);
 
                        vt = this_argument->vtable;
                        vtable_slot = orig_vtable_slot;
@@ -407,6 +459,7 @@ common_call_trampoline (mgreg_t *regs, guint8 *code, gpointer arg, guint8* tramp
                }
 
                g_assert (klass);
+               g_assert (actual_method);
                g_assert (actual_method->klass == klass);
 
                if (actual_method->is_inflated)
@@ -434,7 +487,7 @@ common_call_trampoline (mgreg_t *regs, guint8 *code, gpointer arg, guint8* tramp
        if (!code && mono_method_needs_static_rgctx_invoke (m, FALSE))
                need_rgctx_tramp = TRUE;
 
-       addr = mono_compile_method (m);
+       addr = compiled_method = mono_compile_method (m);
        g_assert (addr);
 
        mono_debugger_trampoline_compiled (code, m, addr);
@@ -442,16 +495,18 @@ common_call_trampoline (mgreg_t *regs, guint8 *code, gpointer arg, guint8* tramp
        if (need_rgctx_tramp)
                addr = mono_create_static_rgctx_trampoline (m, addr);
 
-       if (generic_virtual) {
+       if (generic_virtual || variant_iface) {
+               MonoMethod *target = generic_virtual ? generic_virtual : variant_iface;
+
                vtable_slot = orig_vtable_slot;
                g_assert (vtable_slot);
 
-               if (vt->klass->valuetype)
+               if (vt->klass->valuetype) /*FIXME is this required variant iface?*/
                        addr = get_unbox_trampoline (mono_get_generic_context_from_code (code), m, addr, need_rgctx_tramp);
 
                mono_method_add_generic_virtual_invocation (mono_domain_get (), 
                                                                                                        vt, vtable_slot,
-                                                                                                       generic_virtual, addr);
+                                                                                                       target, addr);
 
                return addr;
        }
@@ -487,13 +542,14 @@ common_call_trampoline (mgreg_t *regs, guint8 *code, gpointer arg, guint8* tramp
        vtable_slot = orig_vtable_slot;
 
        if (vtable_slot) {
+               gboolean variance_used = FALSE;
                if (m->klass->valuetype)
                        addr = get_unbox_trampoline (mono_get_generic_context_from_code (code), m, addr, need_rgctx_tramp);
                g_assert (*vtable_slot);
 
                if (!proxy && (mono_aot_is_got_entry (code, (guint8*)vtable_slot) || mono_domain_owns_vtable_slot (mono_domain_get (), vtable_slot))) {
 #ifdef MONO_ARCH_HAVE_IMT
-                       vtable_slot = mono_convert_imt_slot_to_vtable_slot (vtable_slot, regs, code, m, NULL, &need_rgctx_tramp);
+                       vtable_slot = mono_convert_imt_slot_to_vtable_slot (vtable_slot, regs, code, m, NULL, &need_rgctx_tramp, &variance_used);
 #endif
                        *vtable_slot = mono_get_addr_from_ftnptr (addr);
                }
@@ -515,10 +571,11 @@ common_call_trampoline (mgreg_t *regs, guint8 *code, gpointer arg, guint8* tramp
                        if (plt_entry) {
 
                        } else {
-                               MonoJitInfo *ji = 
-                                       mono_jit_info_table_find (mono_domain_get (), (char*)code);
                                MonoJitInfo *target_ji = 
-                                       mono_jit_info_table_find (mono_domain_get (), mono_get_addr_from_ftnptr (addr));
+                                       mono_jit_info_table_find (mono_domain_get (), mono_get_addr_from_ftnptr (compiled_method));
+
+                               if (!ji)
+                                       ji = mono_jit_info_table_find (mono_domain_get (), (char*)code);
 
                                if (mono_method_same_domain (ji, target_ji))
                                        mono_arch_patch_callsite (ji->code_start, code, addr);
@@ -680,9 +737,10 @@ mono_aot_trampoline (mgreg_t *regs, guint8 *code, guint8 *token_info,
         */
        if ((is_got_entry && (mono_domain_get () == mono_get_root_domain ())) || mono_domain_owns_vtable_slot (mono_domain_get (), vtable_slot)) {
 #ifdef MONO_ARCH_HAVE_IMT
+               gboolean variance_used = FALSE;
                if (!method)
                        method = mono_get_method (image, token, NULL);
-               vtable_slot = mono_convert_imt_slot_to_vtable_slot (vtable_slot, regs, code, method, NULL, &need_rgctx_tramp);
+               vtable_slot = mono_convert_imt_slot_to_vtable_slot (vtable_slot, regs, code, method, NULL, &need_rgctx_tramp, &variance_used);
 #endif
                *vtable_slot = addr;
        }
@@ -700,8 +758,17 @@ mono_aot_plt_trampoline (mgreg_t *regs, guint8 *code, guint8 *aot_module,
                                                 guint8* tramp)
 {
        guint32 plt_info_offset = mono_aot_get_plt_info_offset (regs, code);
+       gpointer res;
+
+       res = mono_aot_plt_resolve (aot_module, plt_info_offset, code);
+       if (!res) {
+               if (mono_loader_get_last_error ())
+                       mono_raise_exception (mono_loader_error_prepare_exception (mono_loader_get_last_error ()));
+               // FIXME: Error handling (how ?)
+               g_assert (res);
+       }
 
-       return mono_aot_plt_resolve (aot_module, plt_info_offset, code);
+       return res;
 }
 #endif
 
@@ -849,15 +916,17 @@ mono_delegate_trampoline (mgreg_t *regs, guint8 *code, gpointer *tramp_data, gui
                        delegate->method_ptr = *delegate->method_code;
                } else {
                        delegate->method_ptr = mono_compile_method (method);
+                       if (need_rgctx_tramp)
+                               delegate->method_ptr = mono_create_static_rgctx_trampoline (method, delegate->method_ptr);
                        if (delegate->method_code)
                                *delegate->method_code = delegate->method_ptr;
                        mono_debugger_trampoline_compiled (NULL, method, delegate->method_ptr);
                }
+       } else {
+               if (need_rgctx_tramp)
+                       delegate->method_ptr = mono_create_static_rgctx_trampoline (method, delegate->method_ptr);
        }
 
-       if (need_rgctx_tramp)
-               delegate->method_ptr = mono_create_static_rgctx_trampoline (method, delegate->method_ptr);
-
        multicast = ((MonoMulticastDelegate*)delegate)->prev != NULL;
        if (!multicast && !callvirt) {
                if (method && (method->flags & METHOD_ATTRIBUTE_STATIC) && mono_method_signature (method)->param_count == mono_method_signature (invoke)->param_count + 1)
@@ -1031,7 +1100,8 @@ mono_create_generic_class_init_trampoline (void)
 
        if (!code) {
                if (mono_aot_only)
-                       code = mono_aot_get_named_code ("generic_class_init_trampoline");
+                       /* get_named_code () might return an ftnptr, but our caller expects a direct pointer */
+                       code = mono_get_addr_from_ftnptr (mono_aot_get_named_code ("generic_class_init_trampoline"));
                else
                        code = mono_arch_create_generic_class_init_trampoline ();
        }