2008-08-24 Zoltan Varga <vargaz@gmail.com>
[mono.git] / mono / mini / tramp-x86.c
index 52e6bfd58021d9883f0e9195cc54c62acd53f0e2..7eb361736e2c073f30777dcd15de4387024716a0 100644 (file)
@@ -29,6 +29,7 @@ static guint8* nullified_class_init_trampoline;
 
 /*
  * mono_arch_get_unbox_trampoline:
+ * @gsctx: the generic sharing context
  * @m: method pointer
  * @addr: pointer to native code for @m
  *
@@ -37,13 +38,13 @@ static guint8* nullified_class_init_trampoline;
  * unboxing before calling the method
  */
 gpointer
-mono_arch_get_unbox_trampoline (MonoMethod *m, gpointer addr)
+mono_arch_get_unbox_trampoline (MonoGenericSharingContext *gsctx, MonoMethod *m, gpointer addr)
 {
        guint8 *code, *start;
        int this_pos = 4;
        MonoDomain *domain = mono_domain_get ();
 
-       if (!mono_method_signature (m)->ret->byref && MONO_TYPE_ISSTRUCT (mono_method_signature (m)->ret))
+       if (MONO_TYPE_ISSTRUCT (mono_method_signature (m)->ret))
                this_pos = 8;
            
        mono_domain_lock (domain);
@@ -58,8 +59,16 @@ mono_arch_get_unbox_trampoline (MonoMethod *m, gpointer addr)
 }
 
 void
-mono_arch_patch_callsite (guint8 *code, guint8 *addr)
+mono_arch_patch_callsite (guint8 *method_start, guint8 *orig_code, guint8 *addr)
 {
+       guint8 *code;
+       guint8 buf [8];
+       gboolean can_write = mono_breakpoint_clean_code (method_start, orig_code, 8, buf, sizeof (buf));
+
+       code = buf + 8;
+       if (mono_running_on_valgrind ())
+               can_write = FALSE;
+
        /* go to the start of the call instruction
         *
         * address_byte = (m << 6) | (o << 3) | reg
@@ -68,9 +77,10 @@ mono_arch_patch_callsite (guint8 *code, guint8 *addr)
         * 0xff m=2,o=2 imm32
         */
        code -= 6;
+       orig_code -= 6;
        if ((code [1] == 0xe8)) {
-               if (!mono_running_on_valgrind ()) {
-                       InterlockedExchange ((gint32*)(code + 2), (guint)addr - ((guint)code + 1) - 5);
+               if (can_write) {
+                       InterlockedExchange ((gint32*)(orig_code + 2), (guint)addr - ((guint)orig_code + 1) - 5);
 
 #ifdef HAVE_VALGRIND_MEMCHECK_H
                                /* Tell valgrind to recompile the patched code */
@@ -79,8 +89,8 @@ mono_arch_patch_callsite (guint8 *code, guint8 *addr)
                }
        } else if (code [1] == 0xe9) {
                /* A PLT entry: jmp <DISP> */
-               if (!mono_running_on_valgrind ())
-                       InterlockedExchange ((gint32*)(code + 2), (guint)addr - ((guint)code + 1) - 5);
+               if (can_write)
+                       InterlockedExchange ((gint32*)(orig_code + 2), (guint)addr - ((guint)orig_code + 1) - 5);
        } else {
                printf ("Invalid trampoline sequence: %x %x %x %x %x %x %x\n", code [0], code [1], code [2], code [3],
                                code [4], code [5], code [6]);
@@ -101,6 +111,12 @@ mono_arch_patch_plt_entry (guint8 *code, guint8 *addr)
 void
 mono_arch_nullify_class_init_trampoline (guint8 *code, gssize *regs)
 {
+       guint8 buf [16];
+       gboolean can_write = mono_breakpoint_clean_code (NULL, code, 6, buf, sizeof (buf));
+
+       if (!can_write)
+               return;
+
        code -= 5;
        if (code [0] == 0xe8) {
                if (!mono_running_on_valgrind ()) {
@@ -169,35 +185,11 @@ mono_arch_nullify_plt_entry (guint8 *code)
        }
 }
 
-void
-mono_arch_patch_delegate_trampoline (guint8 *code, guint8 *tramp, gssize *regs, guint8 *addr)
-{
-       guint32 reg;
-       guint32 disp;
-
-       if ((code [-3] == 0xff) && (x86_modrm_reg (code [-2]) == 0x2) && (x86_modrm_mod (code [-2]) == 0x1)) {
-               /* call *[reg+disp8] */
-               reg = x86_modrm_rm (code [-2]);
-               disp = *(guint8*)(code - 1);
-               //printf ("B: [%%r%d+0x%x]\n", reg, disp);
-       }
-       else {
-               int i;
-
-               for (i = -16; i < 0; ++i)
-                       printf ("%d ", code [i]);
-               printf ("\n");
-               g_assert_not_reached ();
-       }
-
-       *(gpointer*)(((guint32)(regs [reg])) + disp) = addr;
-}
-
 guchar*
 mono_arch_create_trampoline_code (MonoTrampolineType tramp_type)
 {
-       guint8 *buf, *code;
-       int pushed_args;
+       guint8 *buf, *code, *tramp;
+       int pushed_args, pushed_args_caller_saved;
 
        code = buf = mono_global_codeman_reserve (256);
 
@@ -205,7 +197,11 @@ mono_arch_create_trampoline_code (MonoTrampolineType tramp_type)
         * and it is stored at: esp + pushed_args * sizeof (gpointer)
         * the ret address is at: esp + (pushed_args + 1) * sizeof (gpointer)
         */
-       /* Put all registers into an array on the stack */
+
+       /* Put all registers into an array on the stack
+        * If this code is changed, make sure to update the offset value in
+        * mono_arch_find_this_argument () in mini-x86.c.
+        */
        x86_push_reg (buf, X86_EDI);
        x86_push_reg (buf, X86_ESI);
        x86_push_reg (buf, X86_EBP);
@@ -215,7 +211,12 @@ mono_arch_create_trampoline_code (MonoTrampolineType tramp_type)
        x86_push_reg (buf, X86_ECX);
        x86_push_reg (buf, X86_EAX);
 
-       pushed_args = 8;
+       pushed_args_caller_saved = pushed_args = 8;
+
+       /* Align stack on apple */
+       x86_alu_reg_imm (buf, X86_SUB, X86_ESP, 4);
+
+       pushed_args ++;
 
        /* save LMF begin */
 
@@ -234,25 +235,36 @@ mono_arch_create_trampoline_code (MonoTrampolineType tramp_type)
 
        pushed_args += 4;
 
+       /* save ESP */
+       x86_push_reg (buf, X86_ESP);
+       /* Adjust ESP so it points to the previous frame */
+       x86_alu_membase_imm (buf, X86_ADD, X86_ESP, 0, (pushed_args + 2) * 4);
+
+       pushed_args ++;
+
        /* save method info */
-       x86_push_membase (buf, X86_ESP, pushed_args * sizeof (gpointer));
+       if ((tramp_type == MONO_TRAMPOLINE_JIT) || (tramp_type == MONO_TRAMPOLINE_JUMP))
+               x86_push_membase (buf, X86_ESP, pushed_args * sizeof (gpointer));
+       else
+               x86_push_imm (buf, 0);
 
        pushed_args++;
 
-       /* the stack is correctly aligned to 16 bytes because pushed_args is 14
-        * and there is the extra trampoline arg + the return ip pushed by call
+       /* On apple, the stack is correctly aligned to 16 bytes because pushed_args is
+        * 16 and there is the extra trampoline arg + the return ip pushed by call
         * FIXME: Note that if an exception happens while some args are pushed
         * on the stack, the stack will be misaligned.
         */
-#ifdef __APPLE__
-       g_assert (pushed_args == 14);
-#endif
+       g_assert (pushed_args == 16);
+
        /* get the address of lmf for the current thread */
        x86_call_code (buf, mono_get_lmf_addr);
        /* push lmf */
        x86_push_reg (buf, X86_EAX); 
        /* push *lfm (previous_lmf) */
        x86_push_membase (buf, X86_EAX, 0);
+       /* Signal to mono_arch_find_jit_info () that this is a trampoline frame */
+       x86_alu_membase_imm (buf, X86_ADD, X86_ESP, 0, 1);
        /* *(lmf) = ESP */
        x86_mov_membase_reg (buf, X86_EAX, 0, X86_ESP, 4);
        /* save LFM end */
@@ -260,11 +272,6 @@ mono_arch_create_trampoline_code (MonoTrampolineType tramp_type)
        pushed_args += 2;
 
        /* starting the call sequence */
-#ifdef __APPLE__
-       /* changing esp to keep the stack aligned */
-       x86_alu_reg_imm (buf, X86_SUB, X86_ESP, 8);
-       pushed_args += 2;
-#endif
 
        /* FIXME: Push the trampoline address */
        x86_push_imm (buf, 0);
@@ -297,51 +304,71 @@ mono_arch_create_trampoline_code (MonoTrampolineType tramp_type)
        x86_breakpoint (buf);*/
 #endif
 
-       if (tramp_type == MONO_TRAMPOLINE_CLASS_INIT)
-               x86_call_code (buf, mono_class_init_trampoline);
-       else if (tramp_type == MONO_TRAMPOLINE_AOT)
-               x86_call_code (buf, mono_aot_trampoline);
-       else if (tramp_type == MONO_TRAMPOLINE_AOT_PLT)
-               x86_call_code (buf, mono_aot_plt_trampoline);
-       else if (tramp_type == MONO_TRAMPOLINE_DELEGATE)
-               x86_call_code (buf, mono_delegate_trampoline);
-       else
-               x86_call_code (buf, mono_magic_trampoline);
+       tramp = (guint8*)mono_get_trampoline_func (tramp_type);
+       x86_call_code (buf, tramp);
 
-#ifdef __APPLE__
-       /* account for the alignment above */
-       x86_alu_reg_imm (buf, X86_ADD, X86_ESP, 6*4);
-#else
        x86_alu_reg_imm (buf, X86_ADD, X86_ESP, 4*4);
-#endif
 
-       /* restore LMF start */
+       pushed_args -= 4;
+
+       /* Check for thread interruption */
+       /* This is not perf critical code so no need to check the interrupt flag */
+       x86_push_reg (buf, X86_EAX);
+       x86_call_code (buf, (guint8*)mono_thread_force_interruption_checkpoint);
+       x86_pop_reg (buf, X86_EAX);
+
+       /* Restore LMF */
+
        /* ebx = previous_lmf */
        x86_pop_reg (buf, X86_EBX);
+       pushed_args--;
+       x86_alu_reg_imm (buf, X86_SUB, X86_EBX, 1);
+
        /* edi = lmf */
        x86_pop_reg (buf, X86_EDI);
+       pushed_args--;
+
        /* *(lmf) = previous_lmf */
        x86_mov_membase_reg (buf, X86_EDI, 0, X86_EBX, 4);
+
        /* discard method info */
        x86_pop_reg (buf, X86_ESI);
+       pushed_args--;
+
+       /* discard ESP */
+       x86_pop_reg (buf, X86_ESI);
+       pushed_args--;
+
        /* restore caller saved regs */
        x86_pop_reg (buf, X86_EBX);
        x86_pop_reg (buf, X86_EDI);
        x86_pop_reg (buf, X86_ESI);
        x86_pop_reg (buf, X86_EBP);
 
+       pushed_args -= 4;
+
        /* discard save IP */
-       x86_alu_reg_imm (buf, X86_ADD, X86_ESP, 4);             
+       x86_alu_reg_imm (buf, X86_ADD, X86_ESP, 4);
+       pushed_args--;
+
        /* restore LMF end */
 
        /* Restore caller saved registers */
-       x86_mov_reg_membase (buf, X86_ECX, X86_ESP, 1 * 4, 4);
-       x86_mov_reg_membase (buf, X86_EDX, X86_ESP, 2 * 4, 4);
+       x86_mov_reg_membase (buf, X86_ECX, X86_ESP, (pushed_args - pushed_args_caller_saved + X86_ECX) * 4, 4);
+       x86_mov_reg_membase (buf, X86_EDX, X86_ESP, (pushed_args - pushed_args_caller_saved + X86_EDX) * 4, 4);
 
-       /* Pop saved reg array + method ptr */
-       x86_alu_reg_imm (buf, X86_ADD, X86_ESP, 9 * 4);
+       /* Pop saved reg array + stack align + method ptr */
+       x86_alu_reg_imm (buf, X86_ADD, X86_ESP, 10 * 4);
 
-       if (tramp_type == MONO_TRAMPOLINE_CLASS_INIT)
+       pushed_args -= 10;
+
+       /* We've popped one more stack item than we've pushed (the
+          method ptr argument), so we must end up at -1. */
+       g_assert (pushed_args == -1);
+
+       if (tramp_type == MONO_TRAMPOLINE_CLASS_INIT ||
+                       tramp_type == MONO_TRAMPOLINE_GENERIC_CLASS_INIT ||
+                       tramp_type == MONO_TRAMPOLINE_RGCTX_LAZY_FETCH)
                x86_ret (buf);
        else
                /* call the compiled method */
@@ -368,7 +395,7 @@ mono_arch_create_specific_trampoline (gpointer arg1, MonoTrampolineType tramp_ty
        tramp = mono_get_trampoline_code (tramp_type);
 
        mono_domain_lock (domain);
-       code = buf = mono_code_manager_reserve (domain->code_mp, TRAMPOLINE_SIZE);
+       code = buf = mono_code_manager_reserve_align (domain->code_mp, TRAMPOLINE_SIZE, 4);
        mono_domain_unlock (domain);
 
        x86_push_imm (buf, arg1);
@@ -383,6 +410,129 @@ mono_arch_create_specific_trampoline (gpointer arg1, MonoTrampolineType tramp_ty
        return code;
 }
 
+gpointer
+mono_arch_create_rgctx_lazy_fetch_trampoline (guint32 slot)
+{
+       guint8 *tramp;
+       guint8 *code, *buf;
+       guint8 **rgctx_null_jumps;
+       int tramp_size;
+       int depth, index;
+       int i;
+       gboolean mrgctx;
+
+       mrgctx = MONO_RGCTX_SLOT_IS_MRGCTX (slot);
+       index = MONO_RGCTX_SLOT_INDEX (slot);
+       if (mrgctx)
+               index += sizeof (MonoMethodRuntimeGenericContext) / sizeof (gpointer);
+       for (depth = 0; ; ++depth) {
+               int size = mono_class_rgctx_get_array_size (depth, mrgctx);
+
+               if (index < size - 1)
+                       break;
+               index -= size - 1;
+       }
+
+       tramp_size = 36 + 6 * depth;
+
+       code = buf = mono_global_codeman_reserve (tramp_size);
+
+       rgctx_null_jumps = g_malloc (sizeof (guint8*) * (depth + 2));
+
+       /* load vtable/mrgctx ptr */
+       x86_mov_reg_membase (buf, X86_EAX, X86_ESP, 4, 4);
+       if (!mrgctx) {
+               /* load rgctx ptr from vtable */
+               x86_mov_reg_membase (buf, X86_EAX, X86_EAX, G_STRUCT_OFFSET (MonoVTable, runtime_generic_context), 4);
+               /* is the rgctx ptr null? */
+               x86_test_reg_reg (buf, X86_EAX, X86_EAX);
+               /* if yes, jump to actual trampoline */
+               rgctx_null_jumps [0] = buf;
+               x86_branch8 (buf, X86_CC_Z, -1, 1);
+       }
+
+       for (i = 0; i < depth; ++i) {
+               /* load ptr to next array */
+               if (mrgctx && i == 0)
+                       x86_mov_reg_membase (buf, X86_EAX, X86_EAX, sizeof (MonoMethodRuntimeGenericContext), 4);
+               else
+                       x86_mov_reg_membase (buf, X86_EAX, X86_EAX, 0, 4);
+               /* is the ptr null? */
+               x86_test_reg_reg (buf, X86_EAX, X86_EAX);
+               /* if yes, jump to actual trampoline */
+               rgctx_null_jumps [i + 1] = buf;
+               x86_branch8 (buf, X86_CC_Z, -1, 1);
+       }
+
+       /* fetch slot */
+       x86_mov_reg_membase (buf, X86_EAX, X86_EAX, sizeof (gpointer) * (index + 1), 4);
+       /* is the slot null? */
+       x86_test_reg_reg (buf, X86_EAX, X86_EAX);
+       /* if yes, jump to actual trampoline */
+       rgctx_null_jumps [depth + 1] = buf;
+       x86_branch8 (buf, X86_CC_Z, -1, 1);
+       /* otherwise return */
+       x86_ret (buf);
+
+       for (i = mrgctx ? 1 : 0; i <= depth + 1; ++i)
+               x86_patch (rgctx_null_jumps [i], buf);
+
+       g_free (rgctx_null_jumps);
+
+       x86_mov_reg_membase (buf, MONO_ARCH_VTABLE_REG, X86_ESP, 4, 4);
+
+       tramp = mono_arch_create_specific_trampoline (GUINT_TO_POINTER (slot), MONO_TRAMPOLINE_RGCTX_LAZY_FETCH, mono_get_root_domain (), NULL);
+
+       /* jump to the actual trampoline */
+       x86_jump_code (buf, tramp);
+
+       mono_arch_flush_icache (code, buf - code);
+
+       g_assert (buf - code <= tramp_size);
+
+       return code;
+}
+
+gpointer
+mono_arch_create_generic_class_init_trampoline (void)
+{
+       guint8 *tramp;
+       guint8 *code, *buf;
+       static int byte_offset = -1;
+       static guint8 bitmask;
+       guint8 *jump;
+       int tramp_size;
+
+       tramp_size = 64;
+
+       code = buf = mono_global_codeman_reserve (tramp_size);
+
+       if (byte_offset < 0)
+               mono_marshal_find_bitfield_offset (MonoVTable, initialized, &byte_offset, &bitmask);
+
+       x86_test_membase_imm (code, MONO_ARCH_VTABLE_REG, byte_offset, bitmask);
+       jump = code;
+       x86_branch8 (code, X86_CC_Z, -1, 1);
+
+       x86_ret (code);
+
+       x86_patch (jump, code);
+
+       /* Push the vtable so the stack is the same as in a specific trampoline */
+       x86_push_reg (code, MONO_ARCH_VTABLE_REG);
+
+       tramp = mono_get_trampoline_code (MONO_TRAMPOLINE_GENERIC_CLASS_INIT);
+
+       /* jump to the actual trampoline */
+       x86_jump_code (code, tramp);
+
+       mono_arch_flush_icache (code, code - buf);
+
+       g_assert (code - buf <= tramp_size);
+
+       return buf;
+}
+
 void
 mono_arch_invalidate_method (MonoJitInfo *ji, void *func, gpointer func_arg)
 {
@@ -392,17 +542,3 @@ mono_arch_invalidate_method (MonoJitInfo *ji, void *func, gpointer func_arg)
        x86_push_imm (code, func_arg);
        x86_call_code (code, (guint8*)func);
 }
-
-/*
- * This method is only called when running in the Mono Debugger.
- */
-guint8 *
-mono_debugger_create_notification_function (MonoCodeManager *codeman)
-{
-       guint8 *buf, *code;
-
-       code = buf = mono_code_manager_reserve (codeman, 2);
-       x86_breakpoint (buf);
-       x86_ret (buf);
-       return code;
-}