2008-01-26 Zoltan Varga <vargaz@gmail.com>
[mono.git] / mono / mini / mini-trampolines.c
index 7ec5dc352d1c6cbf5c145341bd434606920b1e63..ea62178628da39cecf014177f4978ad0f4abac19 100644 (file)
 #endif
 
 #include "mini.h"
+#include "debug-mini.h"
 
 #ifdef MONO_ARCH_HAVE_IMT
 
 static gpointer*
-mono_convert_imt_slot_to_vtable_slot (gpointer* slot, gpointer *regs, guint8 *code, MonoMethod *method)
+mono_convert_imt_slot_to_vtable_slot (gpointer* slot, gpointer *regs, guint8 *code, MonoMethod *method, MonoMethod **impl_method)
 {
        MonoObject *this_argument = mono_arch_find_this_argument (regs, method);
        MonoVTable *vt = this_argument->vtable;
@@ -30,9 +31,20 @@ mono_convert_imt_slot_to_vtable_slot (gpointer* slot, gpointer *regs, guint8 *co
                return slot;
        } else {
                MonoMethod *imt_method = mono_arch_find_imt_method (regs, code);
-               int interface_offset = mono_class_interface_offset (vt->klass, imt_method->klass);
+               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) {
+                       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 ();
+               }
+               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 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, 
@@ -45,7 +57,6 @@ mono_convert_imt_slot_to_vtable_slot (gpointer* slot, gpointer *regs, guint8 *co
 #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
-                       g_assert (vtable_offset >= 0);
                        return vtable_slot;
                } else {
 #if DEBUG_IMT
@@ -68,9 +79,63 @@ mono_magic_trampoline (gssize *regs, guint8 *code, MonoMethod *m, guint8* tramp)
        gpointer addr;
        gpointer *vtable_slot;
 
+#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 (displacement > 0) {
+                       displacement -= G_STRUCT_OFFSET (MonoVTable, vtable);
+                       g_assert (displacement >= 0);
+                       displacement /= sizeof (gpointer);
+
+                       /* 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);
+                               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);
+                               }
+
+                               return addr;
+                       }
+
+                       mono_class_setup_vtable (vt->klass);
+                       m = vt->klass->vtable [displacement];
+                       if (m->iflags & METHOD_IMPL_ATTRIBUTE_SYNCHRONIZED)
+                               m = mono_marshal_get_synchronized_wrapper (m);
+                       /*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 */
+                       m = MONO_FAKE_IMT_METHOD;
+                       /*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;
+               /* 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);
+               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;
+       }
+#endif
+
        addr = mono_compile_method (m);
        g_assert (addr);
 
+       mono_debugger_trampoline_compiled (m, addr);
+
        /* the method was jumped to */
        if (!code)
                return addr;
@@ -85,7 +150,7 @@ mono_magic_trampoline (gssize *regs, guint8 *code, MonoMethod *m, guint8* tramp)
 
                if (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);
+                       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);
                }
@@ -178,7 +243,7 @@ mono_aot_trampoline (gssize *regs, guint8 *code, guint8 *token_info,
 #ifdef MONO_ARCH_HAVE_IMT
                if (!method)
                        method = mono_get_method (image, token, NULL);
-               vtable_slot = mono_convert_imt_slot_to_vtable_slot (vtable_slot, (gpointer*)regs, code, method);
+               vtable_slot = mono_convert_imt_slot_to_vtable_slot (vtable_slot, (gpointer*)regs, code, method, NULL);
 #endif
                *vtable_slot = addr;
        }
@@ -227,6 +292,22 @@ mono_class_init_trampoline (gssize *regs, guint8 *code, MonoVTable *vtable, guin
        }
 }
 
+/**
+ * mono_generic_class_init_trampoline:
+ *
+ * This method calls mono_runtime_class_init () to run the static constructor
+ * for the type.
+ */
+void
+mono_generic_class_init_trampoline (gssize *regs, guint8 *code, MonoVTable *vtable, guint8 *tramp)
+{
+       //g_print ("generic class init for class %s.%s\n", vtable->klass->name_space, vtable->klass->name);
+
+       mono_runtime_class_init (vtable);
+
+       //g_print ("done initing generic\n");
+}
+
 #ifdef MONO_ARCH_HAVE_CREATE_DELEGATE_TRAMPOLINE
 
 /**
@@ -240,69 +321,92 @@ mono_delegate_trampoline (gssize *regs, guint8 *code, MonoClass *klass, guint8*
        MonoDomain *domain = mono_domain_get ();
        MonoDelegate *delegate;
        MonoJitInfo *ji;
-       gpointer iter;
-       MonoMethod *invoke;
-       gboolean multicast;
-
-       /* Find the Invoke method */
-       iter = NULL;
-       while ((invoke = mono_class_get_methods (klass, &iter))) {
-               if (!strcmp (invoke->name, "Invoke"))
-                       break;
-       }
+       MonoMethod *invoke, *m;
+       MonoMethod *method = NULL;
+       gboolean multicast, callvirt;
+
+       invoke = mono_get_delegate_invoke (klass);
        g_assert (invoke);
 
        /* Obtain the delegate object according to the calling convention */
 
        delegate = mono_arch_get_this_arg_from_call (mono_method_signature (invoke), regs, code);
 
+       if (!delegate->method_ptr && delegate->method) {
+               /* The delegate was initialized by mini_delegate_ctor */
+               method = 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 {
+               ji = mono_jit_info_table_find (domain, mono_get_addr_from_ftnptr (delegate->method_ptr));
+               if (ji)
+                       method = ji->method;
+       }
+       callvirt = !delegate->target && method && mono_method_signature (method)->hasthis;
+
        /* 
         * If the called address is a trampoline, replace it with the compiled method so
         * further calls don't have to go through the trampoline.
         */
-       ji = mono_jit_info_table_find (domain, mono_get_addr_from_ftnptr (delegate->method_ptr));
-       if (ji)
-               delegate->method_ptr = mono_compile_method (ji->method);
+       if (method && !callvirt) {
+               delegate->method_ptr = mono_compile_method (method);
+               mono_debugger_trampoline_compiled (method, delegate->method_ptr);
+       }
 
        multicast = ((MonoMulticastDelegate*)delegate)->prev != NULL;
-       if (!multicast) {
-               guint8* code;
-               GHashTable *cache;
-
-               mono_domain_lock (domain);
-               if (delegate->target != NULL) {
-                       if (!domain->delegate_invoke_impl_with_target_hash)
-                               domain->delegate_invoke_impl_with_target_hash = g_hash_table_new (mono_aligned_addr_hash, NULL);
-                       cache = domain->delegate_invoke_impl_with_target_hash;
-               } else {
-                       if (!domain->delegate_invoke_impl_no_target_hash)
-                               domain->delegate_invoke_impl_no_target_hash = g_hash_table_new (mono_aligned_addr_hash, NULL);
-                       cache = domain->delegate_invoke_impl_no_target_hash;
-               }
-               code = g_hash_table_lookup (cache, mono_method_signature (invoke));
-               mono_domain_unlock (domain);
-               if (code) {
-                       delegate->invoke_impl = code;
-                       return code;
-               }
-
+       if (!multicast && !callvirt) {
                code = mono_arch_get_delegate_invoke_impl (mono_method_signature (invoke), delegate->target != NULL);
 
                if (code) {
-                       mono_domain_lock (domain);
-                       g_hash_table_insert (cache, mono_method_signature (invoke), code);
-                       mono_domain_unlock (domain);
-
                        delegate->invoke_impl = code;
                        return code;
                }
        }
 
        /* The general, unoptimized case */
-       delegate->invoke_impl = mono_compile_method (mono_marshal_get_delegate_invoke (invoke));
+       m = mono_marshal_get_delegate_invoke (invoke, delegate);
+       delegate->invoke_impl = mono_compile_method (m);
+       mono_debugger_trampoline_compiled (m, delegate->invoke_impl);
+
        return delegate->invoke_impl;
 }
 
 #endif
 
+/*
+ * mono_get_trampoline_func:
+ *
+ *   Return the C function which needs to be called by the generic trampoline of type
+ * TRAMP_TYPE.
+ */
+gconstpointer
+mono_get_trampoline_func (MonoTrampolineType tramp_type)
+{
+       switch (tramp_type) {
+       case MONO_TRAMPOLINE_GENERIC:
+       case MONO_TRAMPOLINE_JUMP:
+               return mono_magic_trampoline;
+       case MONO_TRAMPOLINE_CLASS_INIT:
+               return mono_class_init_trampoline;
+       case MONO_TRAMPOLINE_GENERIC_CLASS_INIT:
+               return mono_generic_class_init_trampoline;
+#ifdef MONO_ARCH_AOT_SUPPORTED
+       case MONO_TRAMPOLINE_AOT:
+               return mono_aot_trampoline;
+       case MONO_TRAMPOLINE_AOT_PLT:
+               return mono_aot_plt_trampoline;
+#endif
+#ifdef MONO_ARCH_HAVE_CREATE_DELEGATE_TRAMPOLINE
+       case MONO_TRAMPOLINE_DELEGATE:
+               return mono_delegate_trampoline;
+#endif
+       default:
+               g_assert_not_reached ();
+               return NULL;
+       }
+}
+