2009-04-20 Rodrigo Kumpera <rkumpera@novell.com>
[mono.git] / mono / mini / mini-trampolines.c
index 068bc62dec8f2c08cd00f626b6605837cba7b175..d028d15e5f8e5754d487262e74ccbd21d0fbceba 100644 (file)
@@ -39,6 +39,66 @@ get_unbox_trampoline (MonoGenericSharingContext *gsctx, MonoMethod *m, gpointer
                return mono_arch_get_unbox_trampoline (gsctx, m, addr);
 }
 
+#ifdef MONO_ARCH_HAVE_STATIC_RGCTX_TRAMPOLINE
+/*
+ * mono_create_static_rgctx_trampoline:
+ *
+ *   Return a static rgctx trampoline for M which branches to ADDR which should
+ * point to the compiled code of M.
+ *
+ *   Static rgctx trampolines are used when a shared generic method which doesn't
+ * have a this argument is called indirectly, ie. from code which can't pass in
+ * the rgctx argument. The trampoline sets the rgctx argument and jumps to the
+ * 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.
+ */
+gpointer
+mono_create_static_rgctx_trampoline (MonoMethod *m, gpointer addr)
+{
+       gpointer ctx;
+       gpointer res;
+       MonoDomain *domain;
+
+       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 ();
+
+       mono_domain_lock (domain);
+       res = g_hash_table_lookup (domain_jit_info (domain)->static_rgctx_trampoline_hash,
+                                                          m);
+       mono_domain_unlock (domain);
+       if (res)
+               return res;
+
+       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);
+       mono_domain_unlock (domain);
+
+       return res;
+}
+#endif
+
+gpointer*
+mono_get_vcall_slot_addr (guint8* code, gpointer *regs)
+{
+       gpointer vt;
+       int displacement;
+       vt = mono_arch_get_vcall_slot (code, regs, &displacement);
+       if (!vt)
+               return NULL;
+       return (gpointer*)((char*)vt + displacement);
+}
+
 #ifdef MONO_ARCH_HAVE_IMT
 
 static gpointer*
@@ -60,7 +120,6 @@ mono_convert_imt_slot_to_vtable_slot (gpointer* slot, gpointer *regs, guint8 *co
                int interface_offset;
                int imt_slot = MONO_IMT_SIZE + displacement;
 
-               mono_class_setup_vtable (vt->klass);
                interface_offset = mono_class_interface_offset (vt->klass, imt_method->klass);
 
                if (interface_offset < 0) {
@@ -69,16 +128,37 @@ mono_convert_imt_slot_to_vtable_slot (gpointer* slot, gpointer *regs, guint8 *co
                }
                mono_vtable_build_imt_slot (vt, mono_method_get_imt_slot (imt_method));
 
-               if (impl_method)
-                       *impl_method = vt->klass->vtable [interface_offset + imt_method->slot];
+               if (impl_method) {
+                       MonoMethod *impl;
+
+                       if (imt_method->is_inflated && ((MonoMethodInflated*)imt_method)->context.method_inst) {
+                               MonoGenericContext context = { NULL, NULL };
+
+                               /* 
+                                * Generic virtual method, imt_method contains the inflated interface 
+                                * method, need to get the inflated impl method.
+                                */
+                               /* imt_method->slot might not be set */
+                               impl = mono_class_get_vtable_entry (vt->klass, interface_offset + mono_method_get_declaring_generic_method (imt_method)->slot);
+
+                               if (impl->klass->generic_class)
+                                       context.class_inst = impl->klass->generic_class->context.class_inst;
+                               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_method = impl;
 #if DEBUG_IMT
                printf ("mono_convert_imt_slot_to_vtable_slot: method = %s.%s.%s, imt_method = %s.%s.%s\n",
                                method->klass->name_space, method->klass->name, method->name, 
                                imt_method->klass->name_space, imt_method->klass->name, imt_method->name);
 #endif
+               }
                g_assert (imt_slot < MONO_IMT_SIZE);
                if (vt->imt_collisions_bitmap & (1 << imt_slot)) {
-                       int vtable_offset = interface_offset + imt_method->slot;
+                       int vtable_offset = interface_offset + mono_method_get_vtable_index (imt_method);
                        gpointer *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);
@@ -106,13 +186,29 @@ mono_magic_trampoline (gssize *regs, guint8 *code, MonoMethod *m, guint8* tramp)
        gpointer *vtable_slot;
        gboolean generic_shared = FALSE;
        MonoMethod *declaring = NULL;
+       MonoMethod *generic_virtual = NULL;
        int context_used;
+       gboolean proxy = FALSE;
+#ifdef MONO_ARCH_HAVE_STATIC_RGCTX_TRAMPOLINE
+       gboolean need_rgctx_tramp = FALSE;
+#endif
 
-#if MONO_ARCH_COMMON_VTABLE_TRAMPOLINE
        if (m == MONO_FAKE_VTABLE_METHOD) {
                int displacement;
                MonoVTable *vt = mono_arch_get_vcall_slot (code, (gpointer*)regs, &displacement);
-               g_assert (vt);
+               if (!vt) {
+                       int i;
+                       MonoJitInfo *ji;
+
+                       ji = mono_jit_info_table_find (mono_domain_get (), (char*)code);
+                       if (ji)
+                               printf ("Caller: %s\n", mono_method_full_name (ji->method, TRUE));
+                       /* Print some debug info */
+                       for (i = 0; i < 32; ++i)
+                               printf ("0x%x ", code [-32 + i]);
+                       printf ("\n");
+                       g_assert (vt);
+               }
                if (displacement > 0) {
                        displacement -= G_STRUCT_OFFSET (MonoVTable, vtable);
                        g_assert (displacement >= 0);
@@ -121,7 +217,7 @@ mono_magic_trampoline (gssize *regs, guint8 *code, MonoMethod *m, guint8* tramp)
                        /* Avoid loading metadata or creating a generic vtable if possible */
                        addr = mono_aot_get_method_from_vt_slot (mono_domain_get (), vt, displacement);
                        if (addr && !vt->klass->valuetype) {
-                               vtable_slot = mono_arch_get_vcall_slot_addr (code, (gpointer*)regs);
+                               vtable_slot = mono_get_vcall_slot_addr (code, (gpointer*)regs);
                                if (mono_aot_is_got_entry (code, (guint8*)vtable_slot) || mono_domain_owns_vtable_slot (mono_domain_get (), vtable_slot)) {
                                        *vtable_slot = mono_get_addr_from_ftnptr (addr);
                                }
@@ -129,8 +225,7 @@ mono_magic_trampoline (gssize *regs, guint8 *code, MonoMethod *m, guint8* tramp)
                                return addr;
                        }
 
-                       mono_class_setup_vtable (vt->klass);
-                       m = vt->klass->vtable [displacement];
+                       m = mono_class_get_vtable_entry (vt->klass, displacement);
                        /*g_print ("%s with disp %d: %s at %p\n", vt->klass->name, displacement, m->name, code);*/
                } else {
                        /* We got here from an interface method: redirect to IMT handling */
@@ -138,27 +233,80 @@ mono_magic_trampoline (gssize *regs, guint8 *code, MonoMethod *m, guint8* tramp)
                        /*g_print ("vtable with disp %d at %p\n", displacement, code);*/
                }
        }
-#endif
+
        /* this is the IMT trampoline */
 #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 ()
                 * needs the signature to be able to find the this argument
                 */
                m = mono_arch_find_imt_method ((gpointer*)regs, code);
-               vtable_slot = mono_arch_get_vcall_slot_addr (code, (gpointer*)regs);
+               vtable_slot = mono_get_vcall_slot_addr (code, (gpointer*)regs);
                g_assert (vtable_slot);
-               vtable_slot = mono_convert_imt_slot_to_vtable_slot (vtable_slot, (gpointer*)regs, code, m, &impl_method);
-               /* 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.
-                */
-               /*g_print ("imt found method %p (%s) at %p\n", impl_method, impl_method->name, code);*/
-               m = impl_method;
+
+               gsctx = mono_get_generic_context_from_code (code);
+               this_arg = mono_arch_find_this_argument ((gpointer*)regs, m, gsctx);
+
+               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, (gpointer*)regs, code, m, &impl_method);
+                       /* 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.
+                        */
+                       /*g_print ("imt found method %p (%s) at %p\n", impl_method, impl_method->name, code);*/
+                       if (m->is_inflated && ((MonoMethodInflated*)m)->context.method_inst) {
+                               /* Generic virtual method */
+                               generic_virtual = m;
+                               m = impl_method;
+#ifdef MONO_ARCH_HAVE_STATIC_RGCTX_TRAMPOLINE
+                               need_rgctx_tramp = TRUE;
+#else
+                               m = mono_marshal_get_static_rgctx_invoke (m);
+#endif
+                       } else {
+                               m = impl_method;
+                       }
+               }
        }
 #endif
 
-       if ((context_used = mono_method_check_context_used (m))) {
+       if (m->is_generic) {
+               MonoGenericContext context = { NULL, NULL };
+               MonoMethod *declaring;
+
+               if (m->is_inflated)
+                       declaring = mono_method_get_declaring_generic_method (m);
+               else
+                       declaring = m;
+
+               if (m->klass->generic_class)
+                       context.class_inst = m->klass->generic_class->context.class_inst;
+               else
+                       g_assert (!m->klass->generic_container);
+
+#ifdef MONO_ARCH_HAVE_IMT
+               generic_virtual = mono_arch_find_imt_method ((gpointer*)regs, code);
+#endif
+               if (generic_virtual) {
+                       g_assert (generic_virtual->is_inflated);
+                       context.method_inst = ((MonoMethodInflated*)generic_virtual)->context.method_inst;
+               }
+
+               m = mono_class_inflate_generic_method (declaring, &context);
+#ifdef MONO_ARCH_HAVE_STATIC_RGCTX_TRAMPOLINE
+               need_rgctx_tramp = TRUE;
+#else
+               /* FIXME: only do this if the method is sharable */
+               m = mono_marshal_get_static_rgctx_invoke (m);
+#endif
+       } else if ((context_used = mono_method_check_context_used (m))) {
                MonoClass *klass = NULL;
                MonoMethod *actual_method = NULL;
                MonoVTable *vt = NULL;
@@ -178,7 +326,7 @@ mono_magic_trampoline (gssize *regs, guint8 *code, MonoMethod *m, guint8* tramp)
 #else
                        g_assert_not_reached ();
 #endif
-               } else if (m->flags & METHOD_ATTRIBUTE_STATIC) {
+               } else if ((m->flags & METHOD_ATTRIBUTE_STATIC) || m->klass->valuetype) {
 #ifdef MONO_ARCH_RGCTX_REG
                        MonoVTable *vtable = mono_arch_find_static_call_vtable ((gpointer*)regs, code);
 
@@ -192,7 +340,7 @@ mono_magic_trampoline (gssize *regs, guint8 *code, MonoMethod *m, guint8* tramp)
                                mono_get_generic_context_from_code (code));
 
                        vt = this_argument->vtable;
-                       vtable_slot = mono_arch_get_vcall_slot_addr (code, (gpointer*)regs);
+                       vtable_slot = mono_get_vcall_slot_addr (code, (gpointer*)regs);
 
                        g_assert (this_argument->vtable->klass->inited);
                        //mono_class_init (this_argument->vtable->klass);
@@ -259,11 +407,37 @@ mono_magic_trampoline (gssize *regs, guint8 *code, MonoMethod *m, guint8* tramp)
                        m = mono_marshal_get_synchronized_wrapper (m);
        }
 
+       /* Calls made through delegates on platforms without delegate trampolines */
+       if (!code && mono_method_needs_static_rgctx_invoke (m, FALSE))
+               m = mono_marshal_get_static_rgctx_invoke (m);
+
        addr = mono_compile_method (m);
        g_assert (addr);
 
        mono_debugger_trampoline_compiled (m, addr);
 
+#ifdef MONO_ARCH_HAVE_STATIC_RGCTX_TRAMPOLINE
+       if (need_rgctx_tramp)
+               addr = mono_create_static_rgctx_trampoline (m, addr);
+#endif
+
+       if (generic_virtual) {
+               int displacement;
+               MonoVTable *vt = mono_arch_get_vcall_slot (code, (gpointer*)regs, &displacement);
+
+               vtable_slot = mono_get_vcall_slot_addr (code, (gpointer*)regs);
+               g_assert (vtable_slot);
+
+               if (vt->klass->valuetype)
+                       addr = get_unbox_trampoline (mono_get_generic_context_from_code (code), m, addr);                       
+
+               mono_method_add_generic_virtual_invocation (mono_domain_get (), 
+                                                                                                       vt, vtable_slot,
+                                                                                                       generic_virtual, addr);
+
+               return addr;
+       }
+
        /* the method was jumped to */
        if (!code) {
                MonoDomain *domain = mono_domain_get ();
@@ -273,17 +447,17 @@ mono_magic_trampoline (gssize *regs, guint8 *code, MonoMethod *m, guint8* tramp)
                 * We do this here instead of in mono_codegen () to cover the case when m
                 * was loaded from an aot image.
                 */
-               if (jit_domain_info (domain)->jump_target_got_slot_hash) {
+               if (domain_jit_info (domain)->jump_target_got_slot_hash) {
                        GSList *list, *tmp;
 
                        mono_domain_lock (domain);
-                       list = g_hash_table_lookup (jit_domain_info (domain)->jump_target_got_slot_hash, m);
+                       list = g_hash_table_lookup (domain_jit_info (domain)->jump_target_got_slot_hash, m);
                        if (list) {
                                for (tmp = list; tmp; tmp = tmp->next) {
                                        gpointer *got_slot = tmp->data;
                                        *got_slot = addr;
                                }
-                               g_hash_table_remove (jit_domain_info (domain)->jump_target_got_slot_hash, m);
+                               g_hash_table_remove (domain_jit_info (domain)->jump_target_got_slot_hash, m);
                                g_slist_free (list);
                        }
                        mono_domain_unlock (domain);
@@ -292,44 +466,148 @@ mono_magic_trampoline (gssize *regs, guint8 *code, MonoMethod *m, guint8* tramp)
                return addr;
        }
 
-       vtable_slot = mono_arch_get_vcall_slot_addr (code, (gpointer*)regs);
+       vtable_slot = mono_get_vcall_slot_addr (code, (gpointer*)regs);
 
        if (vtable_slot) {
                if (m->klass->valuetype)
                        addr = get_unbox_trampoline (mono_get_generic_context_from_code (code), m, addr);
-
                g_assert (*vtable_slot);
 
-               if (mono_aot_is_got_entry (code, (guint8*)vtable_slot) || mono_domain_owns_vtable_slot (mono_domain_get (), 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, (gpointer*)regs, code, m, NULL);
 #endif
                        *vtable_slot = mono_get_addr_from_ftnptr (addr);
                }
        }
-       else if (!generic_shared || (m->iflags & METHOD_IMPL_ATTRIBUTE_INTERNAL_CALL) ||
-                       mono_domain_lookup_shared_generic (mono_domain_get (), declaring)) {
+       else {
                guint8 *plt_entry = mono_aot_get_plt_entry (code);
 
-               if (generic_shared)
-                       g_assert (mono_method_is_generic_sharable_impl (m, FALSE));
-
-               /* Patch calling code */
                if (plt_entry) {
                        mono_arch_patch_plt_entry (plt_entry, addr);
-               } 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));
+               } else if (!generic_shared || (m->iflags & METHOD_IMPL_ATTRIBUTE_INTERNAL_CALL) ||
+                       mono_domain_lookup_shared_generic (mono_domain_get (), declaring)) {
+                       if (generic_shared) {
+                               if (m->wrapper_type != MONO_WRAPPER_NONE)
+                                       m = mono_marshal_method_from_wrapper (m);
+                               g_assert (mono_method_is_generic_sharable_impl (m, FALSE));
+                       }
+
+                       /* Patch calling code */
+                       if (plt_entry) {
 
-                       if (mono_method_same_domain (ji, target_ji))
-                               mono_arch_patch_callsite (ji->code_start, code, addr);
+                       } 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));
+
+                               if (mono_method_same_domain (ji, target_ji))
+                                       mono_arch_patch_callsite (ji->code_start, code, addr);
+                       }
                }
        }
 
        return addr;
 }
+#ifdef ENABLE_LLVM
+/*
+ * mono_llvm_vcall_trampoline:
+ *
+ *   This trampoline handles virtual calls when using LLVM.
+ */
+static gpointer
+mono_llvm_vcall_trampoline (gssize *regs, guint8 *code, MonoMethod *m, guint8 *tramp)
+{
+       MonoObject *this;
+       gpointer addr;
+       MonoVTable *vt;
+       gpointer *vtable_slot;
+       gboolean proxy = FALSE;
+
+       /* 
+        * We have the method which is called, we need to obtain the vtable slot without
+        * disassembly which is impossible with LLVM.
+        * So we use the this argument.
+        */
+       this = mono_arch_get_this_arg_from_call (NULL, mono_method_signature (m), regs, code);
+       g_assert (this);
+
+       vt = this->vtable;
+
+       /* This is a simplified version of mono_magic_trampoline () */
+       /* FIXME: Avoid code duplication */
+
+       if (m->iflags & METHOD_IMPL_ATTRIBUTE_SYNCHRONIZED) {
+               MonoJitInfo *ji;
+
+               if (code)
+                       ji = mono_jit_info_table_find (mono_domain_get (), (char*)code);
+               else
+                       ji = NULL;
+
+               /* Avoid recursion */
+               if (!(ji && ji->method->wrapper_type == MONO_WRAPPER_SYNCHRONIZED))
+                       m = mono_marshal_get_synchronized_wrapper (m);
+       }
+
+       addr = mono_compile_method (m);
+       g_assert (addr);
+
+       if (m->klass->valuetype)
+               addr = get_unbox_trampoline (mono_get_generic_context_from_code (code), m, addr);
+
+       vtable_slot = &(vt->vtable [mono_method_get_vtable_slot (m)]);
+       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, (gpointer*)regs, code, m, NULL);
+#endif
+               *vtable_slot = mono_get_addr_from_ftnptr (addr);
+         }
+
+       mono_debugger_trampoline_compiled (m, addr);
+
+       return addr;
+}
+#endif
+
+gpointer
+mono_generic_virtual_remoting_trampoline (gssize *regs, guint8 *code, MonoMethod *m, guint8 *tramp)
+{
+       MonoGenericContext context = { NULL, NULL };
+       MonoMethod *imt_method, *declaring;
+       gpointer addr;
+
+       g_assert (m->is_generic);
+
+       if (m->is_inflated)
+               declaring = mono_method_get_declaring_generic_method (m);
+       else
+               declaring = m;
+
+       if (m->klass->generic_class)
+               context.class_inst = m->klass->generic_class->context.class_inst;
+       else
+               g_assert (!m->klass->generic_container);
+
+#ifdef MONO_ARCH_HAVE_IMT
+       imt_method = mono_arch_find_imt_method ((gpointer*)regs, code);
+       if (imt_method->is_inflated)
+               context.method_inst = ((MonoMethodInflated*)imt_method)->context.method_inst;
+#endif
+       m = mono_class_inflate_generic_method (declaring, &context);
+       m = mono_marshal_get_remoting_invoke_with_check (m);
+
+       addr = mono_compile_method (m);
+       g_assert (addr);
+
+       mono_debugger_trampoline_compiled (m, addr);
+
+       return addr;
+}
 
 /*
  * mono_aot_trampoline:
@@ -348,6 +626,7 @@ mono_aot_trampoline (gssize *regs, guint8 *code, guint8 *token_info,
        gpointer addr;
        gpointer *vtable_slot;
        gboolean is_got_entry;
+       guint8 *plt_entry;
 
        image = *(gpointer*)(gpointer)token_info;
        token_info += sizeof (gpointer);
@@ -358,36 +637,20 @@ mono_aot_trampoline (gssize *regs, guint8 *code, guint8 *token_info,
                method = mono_get_method (image, token, NULL);
                g_assert (method);
 
-               //printf ("F: %s\n", mono_method_full_name (method, TRUE));
-
-               if (method->iflags & METHOD_IMPL_ATTRIBUTE_SYNCHRONIZED)
-                       method = mono_marshal_get_synchronized_wrapper (method);
-
-               addr = mono_compile_method (method);
-               g_assert (addr);
+               /* Use the generic code */
+               return mono_magic_trampoline (regs, code, method, tramp);
        }
 
-       vtable_slot = mono_arch_get_vcall_slot_addr (code, (gpointer*)regs);
-
-       if (vtable_slot) {
-               is_got_entry = mono_aot_is_got_entry (code, (guint8*)vtable_slot);
-
-               if (!is_got_entry) {
-                       if (!method)
-                               method = mono_get_method (image, token, NULL);
-                       if (method->klass->valuetype)
-                               addr = get_unbox_trampoline (mono_get_generic_context_from_code (code), method, addr);
-               }
-       } else {
-               /* This is a normal call through a PLT entry */
-               guint8 *plt_entry = mono_aot_get_plt_entry (code);
+       vtable_slot = mono_get_vcall_slot_addr (code, (gpointer*)regs);
+       g_assert (!vtable_slot);
 
-               g_assert (plt_entry);
+       /* This is a normal call through a PLT entry */
+       plt_entry = mono_aot_get_plt_entry (code);
+       g_assert (plt_entry);
 
-               mono_arch_patch_plt_entry (plt_entry, addr);
+       mono_arch_patch_plt_entry (plt_entry, addr);
 
-               is_got_entry = FALSE;
-       }
+       is_got_entry = FALSE;
 
        /*
         * Since AOT code is only used in the root domain, 
@@ -435,12 +698,10 @@ mono_class_init_trampoline (gssize *regs, guint8 *code, MonoVTable *vtable, guin
 
        mono_runtime_class_init (vtable);
 
-       if (!mono_running_on_valgrind ()) {
-               if (plt_entry) {
-                       mono_arch_nullify_plt_entry (plt_entry);
-               } else {
-                       mono_arch_nullify_class_init_trampoline (code, regs);
-               }
+       if (plt_entry) {
+               mono_arch_nullify_plt_entry (plt_entry);
+       } else {
+               mono_arch_nullify_class_init_trampoline (code, regs);
        }
 }
 
@@ -485,6 +746,18 @@ mono_rgctx_lazy_fetch_trampoline (gssize *regs, guint8 *code, gpointer data, gui
 #endif
 }
 
+void
+mono_monitor_enter_trampoline (gssize *regs, guint8 *code, MonoObject *obj, guint8 *tramp)
+{
+       mono_monitor_enter (obj);
+}
+
+void
+mono_monitor_exit_trampoline (gssize *regs, guint8 *code, MonoObject *obj, guint8 *tramp)
+{
+       mono_monitor_exit (obj);
+}
+
 #ifdef MONO_ARCH_HAVE_CREATE_DELEGATE_TRAMPOLINE
 
 /**
@@ -502,6 +775,9 @@ mono_delegate_trampoline (gssize *regs, guint8 *code, gpointer *tramp_data, guin
        MonoMethod *m;
        MonoMethod *method = NULL;
        gboolean multicast, callvirt;
+#ifdef MONO_ARCH_HAVE_STATIC_RGCTX_TRAMPOLINE
+       gboolean need_rgctx_tramp = FALSE;
+#endif
        MonoMethod *invoke = tramp_data [0];
        guint8 *impl_this = tramp_data [1];
        guint8 *impl_nothis = tramp_data [2];
@@ -514,16 +790,20 @@ mono_delegate_trampoline (gssize *regs, guint8 *code, gpointer *tramp_data, guin
         */
        delegate = mono_arch_get_this_arg_from_call (NULL, mono_method_signature (invoke), regs, code);
 
-       if (!delegate->method_ptr && delegate->method) {
-               /* The delegate was initialized by mini_delegate_ctor */
+       if (delegate->method) {
                method = delegate->method;
 
+               /*
+                * delegate->method_ptr == NULL means the delegate was initialized by 
+                * mini_delegate_ctor, while != NULL means it is initialized by 
+                * mono_delegate_ctor_with_method (). In both cases, we need to add wrappers
+                * (ctor_with_method () does this, but it doesn't store the wrapper back into
+                * delegate->method).
+                */
                if (delegate->target && delegate->target->vtable->klass == mono_defaults.transparent_proxy_class)
                        method = mono_marshal_get_remoting_invoke (method);
                else if (mono_method_signature (method)->hasthis && method->klass->valuetype)
                        method = mono_marshal_get_unbox_wrapper (method);
-       } else if (delegate->method) {
-               method = delegate->method;
        } else {
                ji = mono_jit_info_table_find (domain, mono_get_addr_from_ftnptr (delegate->method_ptr));
                if (ji)
@@ -534,6 +814,14 @@ mono_delegate_trampoline (gssize *regs, guint8 *code, gpointer *tramp_data, guin
        if (method && method->iflags & METHOD_IMPL_ATTRIBUTE_SYNCHRONIZED)
                method = mono_marshal_get_synchronized_wrapper (method);
 
+       if (method && mono_method_needs_static_rgctx_invoke (method, FALSE)) {
+#ifdef MONO_ARCH_HAVE_STATIC_RGCTX_TRAMPOLINE
+               need_rgctx_tramp = TRUE;
+#else
+               method = mono_marshal_get_static_rgctx_invoke (method);
+#endif
+       }
+
        /* 
         * If the called address is a trampoline, replace it with the compiled method so
         * further calls don't have to go through the trampoline.
@@ -550,12 +838,21 @@ mono_delegate_trampoline (gssize *regs, guint8 *code, gpointer *tramp_data, guin
                }
        }
 
+#ifdef MONO_ARCH_HAVE_STATIC_RGCTX_TRAMPOLINE
+       if (need_rgctx_tramp)
+               delegate->method_ptr = mono_create_static_rgctx_trampoline (method, delegate->method_ptr);
+#endif
+
        multicast = ((MonoMulticastDelegate*)delegate)->prev != NULL;
        if (!multicast && !callvirt) {
-               code = delegate->target ? impl_this : impl_nothis;
+               if (method && (method->flags & METHOD_ATTRIBUTE_STATIC) && mono_method_signature (method)->param_count == mono_method_signature (invoke)->param_count + 1)
+                       /* Closed static delegate */
+                       code = impl_this;
+               else
+                       code = delegate->target ? impl_this : impl_nothis;
 
                if (code) {
-                       delegate->invoke_impl = code;
+                       delegate->invoke_impl = mono_get_addr_from_ftnptr (code);
                        return code;
                }
        }
@@ -599,6 +896,18 @@ mono_get_trampoline_func (MonoTrampolineType tramp_type)
 #ifdef MONO_ARCH_HAVE_CREATE_DELEGATE_TRAMPOLINE
        case MONO_TRAMPOLINE_DELEGATE:
                return mono_delegate_trampoline;
+#endif
+       case MONO_TRAMPOLINE_RESTORE_STACK_PROT:
+               return mono_altstack_restore_prot;
+       case MONO_TRAMPOLINE_GENERIC_VIRTUAL_REMOTING:
+               return mono_generic_virtual_remoting_trampoline;
+       case MONO_TRAMPOLINE_MONITOR_ENTER:
+               return mono_monitor_enter_trampoline;
+       case MONO_TRAMPOLINE_MONITOR_EXIT:
+               return mono_monitor_exit_trampoline;
+#ifdef ENABLE_LLVM
+       case MONO_TRAMPOLINE_LLVM_VCALL:
+               return mono_llvm_vcall_trampoline;
 #endif
        default:
                g_assert_not_reached ();
@@ -625,6 +934,13 @@ mono_trampolines_init (void)
 #endif
 #ifdef MONO_ARCH_HAVE_CREATE_DELEGATE_TRAMPOLINE
        mono_trampoline_code [MONO_TRAMPOLINE_DELEGATE] = mono_arch_create_trampoline_code (MONO_TRAMPOLINE_DELEGATE);
+#endif
+       mono_trampoline_code [MONO_TRAMPOLINE_RESTORE_STACK_PROT] = mono_arch_create_trampoline_code (MONO_TRAMPOLINE_RESTORE_STACK_PROT);
+       mono_trampoline_code [MONO_TRAMPOLINE_GENERIC_VIRTUAL_REMOTING] = mono_arch_create_trampoline_code (MONO_TRAMPOLINE_GENERIC_VIRTUAL_REMOTING);
+       mono_trampoline_code [MONO_TRAMPOLINE_MONITOR_ENTER] = mono_arch_create_trampoline_code (MONO_TRAMPOLINE_MONITOR_ENTER);
+       mono_trampoline_code [MONO_TRAMPOLINE_MONITOR_EXIT] = mono_arch_create_trampoline_code (MONO_TRAMPOLINE_MONITOR_EXIT);
+#ifdef ENABLE_LLVM
+       mono_trampoline_code [MONO_TRAMPOLINE_LLVM_VCALL] = mono_arch_create_trampoline_code (MONO_TRAMPOLINE_LLVM_VCALL);
 #endif
 }
 
@@ -660,27 +976,28 @@ gpointer
 mono_create_class_init_trampoline (MonoVTable *vtable)
 {
        gpointer code, ptr;
+       MonoDomain *domain = vtable->domain;
 
        g_assert (!vtable->klass->generic_container);
 
        /* previously created trampoline code */
-       mono_domain_lock (vtable->domain);
+       mono_domain_lock (domain);
        ptr = 
-               g_hash_table_lookup (vtable->domain->class_init_trampoline_hash,
+               g_hash_table_lookup (domain_jit_info (domain)->class_init_trampoline_hash,
                                                                  vtable);
-       mono_domain_unlock (vtable->domain);
+       mono_domain_unlock (domain);
        if (ptr)
                return ptr;
 
-       code = mono_create_specific_trampoline (vtable, MONO_TRAMPOLINE_CLASS_INIT, vtable->domain, NULL);
+       code = mono_create_specific_trampoline (vtable, MONO_TRAMPOLINE_CLASS_INIT, domain, NULL);
 
-       ptr = mono_create_ftnptr (vtable->domain, code);
+       ptr = mono_create_ftnptr (domain, code);
 
        /* store trampoline address */
-       mono_domain_lock (vtable->domain);
-       g_hash_table_insert (vtable->domain->class_init_trampoline_hash,
+       mono_domain_lock (domain);
+       g_hash_table_insert (domain_jit_info (domain)->class_init_trampoline_hash,
                                                          vtable, ptr);
-       mono_domain_unlock (vtable->domain);
+       mono_domain_unlock (domain);
 
        mono_trampolines_lock ();
        if (!class_init_hash_addr)
@@ -691,6 +1008,29 @@ mono_create_class_init_trampoline (MonoVTable *vtable)
        return ptr;
 }
 
+gpointer
+mono_create_generic_class_init_trampoline (void)
+{
+#ifdef MONO_ARCH_VTABLE_REG
+       static gpointer code;
+
+       mono_trampolines_lock ();
+
+       if (!code) {
+               if (mono_aot_only)
+                       code = mono_aot_get_named_code ("generic_class_init_trampoline");
+               else
+                       code = mono_arch_create_generic_class_init_trampoline ();
+       }
+
+       mono_trampolines_unlock ();
+
+       return code;
+#else
+       g_assert_not_reached ();
+#endif
+}
+
 gpointer
 mono_create_jump_trampoline (MonoDomain *domain, MonoMethod *method, gboolean add_sync_wrapper)
 {
@@ -698,12 +1038,17 @@ mono_create_jump_trampoline (MonoDomain *domain, MonoMethod *method, gboolean ad
        gpointer code;
        guint32 code_size = 0;
 
-       code = mono_jit_find_compiled_method (domain, method);
-       if (code)
+       code = mono_jit_find_compiled_method_with_jit_info (domain, method, &ji);
+       /*
+        * We cannot recover the correct type of a shared generic
+        * method from its native code address, so we use the
+        * trampoline instead.
+        */
+       if (code && !ji->has_generic_jit_info)
                return code;
 
        mono_domain_lock (domain);
-       code = g_hash_table_lookup (domain->jump_trampoline_hash, method);
+       code = g_hash_table_lookup (domain_jit_info (domain)->jump_trampoline_hash, method);
        mono_domain_unlock (domain);
        if (code)
                return code;
@@ -711,9 +1056,7 @@ mono_create_jump_trampoline (MonoDomain *domain, MonoMethod *method, gboolean ad
        code = mono_create_specific_trampoline (method, MONO_TRAMPOLINE_JUMP, mono_domain_get (), &code_size);
        g_assert (code_size);
 
-       mono_domain_lock (domain);
-       ji = mono_mempool_alloc0 (domain->mp, sizeof (MonoJitInfo));
-       mono_domain_unlock (domain);
+       ji = mono_domain_alloc0 (domain, sizeof (MonoJitInfo));
        ji->code_start = code;
        ji->code_size = code_size;
        ji->method = method;
@@ -726,7 +1069,7 @@ mono_create_jump_trampoline (MonoDomain *domain, MonoMethod *method, gboolean ad
        mono_jit_info_table_add (domain, ji);
 
        mono_domain_lock (domain);
-       g_hash_table_insert (domain->jump_trampoline_hash, method, ji->code_start);
+       g_hash_table_insert (domain_jit_info (domain)->jump_trampoline_hash, method, ji->code_start);
        mono_domain_unlock (domain);
 
        return ji->code_start;
@@ -746,7 +1089,7 @@ mono_create_jit_trampoline_in_domain (MonoDomain *domain, MonoMethod *method)
        }
 
        mono_domain_lock (domain);
-       tramp = g_hash_table_lookup (domain->jit_trampoline_hash, method);
+       tramp = g_hash_table_lookup (domain_jit_info (domain)->jit_trampoline_hash, method);
        mono_domain_unlock (domain);
        if (tramp)
                return tramp;
@@ -754,7 +1097,7 @@ mono_create_jit_trampoline_in_domain (MonoDomain *domain, MonoMethod *method)
        tramp = mono_create_specific_trampoline (method, MONO_TRAMPOLINE_JIT, domain, NULL);
        
        mono_domain_lock (domain);
-       g_hash_table_insert (domain->jit_trampoline_hash, method, tramp);
+       g_hash_table_insert (domain_jit_info (domain)->jit_trampoline_hash, method, tramp);
        mono_domain_unlock (domain);
 
        mono_jit_stats.method_trampolines++;
@@ -777,9 +1120,7 @@ mono_create_jit_trampoline_from_token (MonoImage *image, guint32 token)
        MonoDomain *domain = mono_domain_get ();
        guint8 *buf, *start;
 
-       mono_domain_lock (domain);
-       buf = start = mono_code_manager_reserve (domain->code_mp, 2 * sizeof (gpointer));
-       mono_domain_unlock (domain);
+       buf = start = mono_domain_code_reserve (domain, 2 * sizeof (gpointer));
 
        *(gpointer*)(gpointer)buf = image;
        buf += sizeof (gpointer);
@@ -804,7 +1145,7 @@ mono_create_delegate_trampoline (MonoClass *klass)
        MonoMethod *invoke;
 
        mono_domain_lock (domain);
-       ptr = g_hash_table_lookup (domain->delegate_trampoline_hash, klass);
+       ptr = g_hash_table_lookup (domain_jit_info (domain)->delegate_trampoline_hash, klass);
        mono_domain_unlock (domain);
        if (ptr)
                return ptr;
@@ -813,24 +1154,17 @@ mono_create_delegate_trampoline (MonoClass *klass)
        invoke = mono_get_delegate_invoke (klass);
        g_assert (invoke);
 
-       mono_domain_lock (domain );
-       tramp_data = mono_mempool_alloc (domain->mp, sizeof (gpointer) * 3);
-       mono_domain_unlock (domain);
+       tramp_data = mono_domain_alloc (domain, sizeof (gpointer) * 3);
        tramp_data [0] = invoke;
-       if (mono_aot_only) {
-               tramp_data [1] = NULL;
-               tramp_data [2] = NULL;
-       } else {
-               tramp_data [1] = mono_arch_get_delegate_invoke_impl (mono_method_signature (invoke), TRUE);
-               tramp_data [2] = mono_arch_get_delegate_invoke_impl (mono_method_signature (invoke), FALSE);
-       }
+       tramp_data [1] = mono_arch_get_delegate_invoke_impl (mono_method_signature (invoke), TRUE);
+       tramp_data [2] = mono_arch_get_delegate_invoke_impl (mono_method_signature (invoke), FALSE);
 
        ptr = mono_create_specific_trampoline (tramp_data, MONO_TRAMPOLINE_DELEGATE, mono_domain_get (), &code_size);
        g_assert (code_size);
 
        /* store trampoline address */
        mono_domain_lock (domain);
-       g_hash_table_insert (domain->delegate_trampoline_hash,
+       g_hash_table_insert (domain_jit_info (domain)->delegate_trampoline_hash,
                                                          klass, ptr);
        mono_domain_unlock (domain);
 
@@ -854,6 +1188,9 @@ mono_create_rgctx_lazy_fetch_trampoline (guint32 offset)
 
        gpointer tramp, ptr;
 
+       if (mono_aot_only)
+               return mono_aot_get_lazy_fetch_trampoline (offset);
+
        mono_trampolines_lock ();
        if (rgctx_lazy_fetch_trampoline_hash)
                tramp = g_hash_table_lookup (rgctx_lazy_fetch_trampoline_hash, GUINT_TO_POINTER (offset));
@@ -886,6 +1223,94 @@ mono_create_rgctx_lazy_fetch_trampoline (guint32 offset)
        return ptr;
 }
 
+gpointer
+mono_create_monitor_enter_trampoline (void)
+{
+       static gpointer code;
+
+       if (mono_aot_only) {
+               if (!code)
+                       code = mono_aot_get_named_code ("monitor_enter_trampoline");
+               return code;
+       }
+
+#ifdef MONO_ARCH_MONITOR_OBJECT_REG
+       mono_trampolines_lock ();
+
+       if (!code)
+               code = mono_arch_create_monitor_enter_trampoline ();
+
+       mono_trampolines_unlock ();
+#else
+       code = NULL;
+       g_assert_not_reached ();
+#endif
+
+       return code;
+}
+
+gpointer
+mono_create_monitor_exit_trampoline (void)
+{
+       static gpointer code;
+
+       if (mono_aot_only) {
+               if (!code)
+                       code = mono_aot_get_named_code ("monitor_exit_trampoline");
+               return code;
+       }
+
+#ifdef MONO_ARCH_MONITOR_OBJECT_REG
+       mono_trampolines_lock ();
+
+       if (!code)
+               code = mono_arch_create_monitor_exit_trampoline ();
+
+       mono_trampolines_unlock ();
+#else
+       code = NULL;
+       g_assert_not_reached ();
+#endif
+       return code;
+}
+#ifdef ENABLE_LLVM
+/*
+ * mono_create_llvm_vcall_trampoline:
+ *
+ *  LLVM emits code for virtual calls which mono_get_vcall_slot is unable to
+ * decode, i.e. only the final branch address is available:
+ * mov <offset>(%rax), %rax
+ * <random code inserted by instruction scheduling>
+ * call *%rax
+ *
+ * To work around this problem, we don't use the common vtable trampoline when
+ * llvm is enabled. Instead, we use one trampoline per method.
+ */
+gpointer
+mono_create_llvm_vcall_trampoline (MonoMethod *method)
+{
+       MonoDomain *domain;
+       gpointer res;
+
+       domain = mono_domain_get ();
+
+       mono_domain_lock (domain);
+       res = g_hash_table_lookup (domain_jit_info (domain)->llvm_vcall_trampoline_hash, method);
+       mono_domain_unlock (domain);
+       if (res)
+               return res;
+
+       res = mono_create_specific_trampoline (method, MONO_TRAMPOLINE_LLVM_VCALL, domain, NULL);
+
+       mono_domain_lock (domain);
+       g_hash_table_insert (domain_jit_info (domain)->llvm_vcall_trampoline_hash, method, res);
+       mono_domain_unlock (domain);
+
+       return res;
+}
+#endif
+
 MonoVTable*
 mono_find_class_init_trampoline_by_addr (gconstpointer addr)
 {