2009-06-10 Zoltan Varga <vargaz@gmail.com>
[mono.git] / mono / mini / exceptions-amd64.c
index 40121611c7088c8362da4831f58e821770119b29..525ff9867045a5d2febeda428ddbd1eeed250126 100644 (file)
@@ -28,6 +28,8 @@
 
 #include "mini.h"
 #include "mini-amd64.h"
+#include "tasklets.h"
+#include "debug-mini.h"
 
 #define ALIGN_TO(val,align) (((val) + ((align) - 1)) & ~((align) - 1))
 
@@ -314,37 +316,32 @@ mono_amd64_throw_exception (guint64 dummy1, guint64 dummy2, guint64 dummy3, guin
        ctx.rcx = rcx;
        ctx.rdx = rdx;
 
-       if (!rethrow && mono_debugger_throw_exception ((gpointer)(rip - 8), (gpointer)rsp, exc)) {
-               /*
-                * The debugger wants us to stop on the `throw' instruction.
-                * By the time we get here, it already inserted a breakpoint on
-                * eip - 8 (which is the address of the `mov %r15,%rdi ; callq throw').
-                */
-
-               /* FIXME FIXME
-                *
-                * In case of a rethrow, the JIT is emitting code like this:
-                *
-                *    mov    0xffffffffffffffd0(%rbp),%rax'
-                *    mov    %rax,%rdi
-                *    callq  throw
-                *
-                * Here, restore_context() wouldn't restore the %rax register correctly.
-                */
-               ctx.rip = rip - 8;
-               ctx.rsp = rsp + 8;
-               restore_context (&ctx);
-               g_assert_not_reached ();
-       }
-
-       /* adjust eip so that it point into the call instruction */
-       ctx.rip -= 1;
-
        if (mono_object_isinst (exc, mono_defaults.exception_class)) {
                MonoException *mono_ex = (MonoException*)exc;
                if (!rethrow)
                        mono_ex->stack_trace = NULL;
        }
+
+       if (mono_debug_using_mono_debugger ()) {
+               guint8 buf [16], *code;
+
+               mono_breakpoint_clean_code (NULL, (gpointer)rip, 8, buf, sizeof (buf));
+               code = buf + 8;
+
+               if (buf [3] == 0xe8) {
+                       MonoContext ctx_cp = ctx;
+                       ctx_cp.rip = rip - 5;
+
+                       if (mono_debugger_handle_exception (&ctx_cp, exc)) {
+                               restore_context (&ctx_cp);
+                               g_assert_not_reached ();
+                       }
+               }
+       }
+
+       /* adjust eip so that it point into the call instruction */
+       ctx.rip -= 1;
+
        mono_handle_exception (&ctx, exc, (gpointer)rip, FALSE);
        restore_context (&ctx);
 
@@ -616,6 +613,7 @@ mono_arch_find_jit_info (MonoDomain *domain, MonoJitTlsData *jit_tls, MonoJitInf
                        *lmf = (gpointer)(((guint64)(*lmf)->previous_lmf) & ~1);
                }
 
+#ifndef MONO_AMD64_NO_PUSHES
                /* Pop arguments off the stack */
                {
                        MonoJitArgumentInfo *arg_info = g_newa (MonoJitArgumentInfo, mono_method_signature (ji->method)->param_count + 1);
@@ -623,6 +621,7 @@ mono_arch_find_jit_info (MonoDomain *domain, MonoJitTlsData *jit_tls, MonoJitInf
                        guint32 stack_to_pop = mono_arch_get_argument_info (mono_method_signature (ji->method), mono_method_signature (ji->method)->param_count, arg_info);
                        new_ctx->rsp += stack_to_pop;
                }
+#endif
 
                return ji;
        } else if (*lmf) {
@@ -683,6 +682,9 @@ mono_arch_handle_exception (void *sigctx, gpointer obj, gboolean test_only)
 
        mono_arch_sigctx_to_monoctx (sigctx, &mctx);
 
+       if (mono_debugger_handle_exception (&mctx, (MonoObject *)obj))
+               return TRUE;
+
        mono_handle_exception (&mctx, obj, MONO_CONTEXT_GET_IP (&mctx), test_only);
 
        mono_arch_monoctx_to_sigctx (&mctx, sigctx);
@@ -834,6 +836,13 @@ altstack_handle_and_restore (void *sigctx, gpointer obj, gboolean stack_ovf)
 
        restore_context = mono_get_restore_context ();
        mono_arch_sigctx_to_monoctx (sigctx, &mctx);
+
+       if (mono_debugger_handle_exception (&mctx, (MonoObject *)obj)) {
+               if (stack_ovf)
+                       prepare_for_guard_pages (&mctx);
+               restore_context (&mctx);
+       }
+
        mono_handle_exception (&mctx, obj, MONO_CONTEXT_GET_IP (&mctx), FALSE);
        if (stack_ovf)
                prepare_for_guard_pages (&mctx);
@@ -882,8 +891,8 @@ mono_arch_handle_altstack_exception (void *sigctx, gpointer fault_addr, gboolean
 #endif
 }
 
-static guint64
-get_original_ip (void)
+guint64
+mono_amd64_get_original_ip (void)
 {
        MonoLMF *lmf = mono_get_lmf ();
 
@@ -895,17 +904,14 @@ get_original_ip (void)
        return lmf->rip;
 }
 
-static gpointer 
-get_throw_pending_exception (void)
+gpointer
+mono_arch_get_throw_pending_exception_full (guint32 *code_size, MonoJumpInfo **ji, gboolean aot)
 {
-       static guint8* start;
-       static gboolean inited = FALSE;
-       guint8 *code;
+       guint8 *code, *start;
        guint8 *br[1];
        gpointer throw_trampoline;
 
-       if (inited)
-               return start;
+       *ji = NULL;
 
        start = code = mono_global_codeman_reserve (128);
 
@@ -926,7 +932,12 @@ get_throw_pending_exception (void)
        amd64_alu_reg_imm (code, X86_SUB, AMD64_RSP, 8);
 
        /* Obtain the pending exception */
-       amd64_mov_reg_imm (code, AMD64_R11, mono_thread_get_and_clear_pending_exception);
+       if (aot) {
+               *ji = mono_patch_info_list_prepend (*ji, code - start, MONO_PATCH_INFO_JIT_ICALL_ADDR, "mono_thread_get_and_clear_pending_exception");
+               amd64_mov_reg_membase (code, AMD64_R11, AMD64_RIP, 0, 8);
+       } else {
+               amd64_mov_reg_imm (code, AMD64_R11, mono_thread_get_and_clear_pending_exception);
+       }
        amd64_call_reg (code, AMD64_R11);
 
        /* Check if it is NULL, and branch */
@@ -941,7 +952,12 @@ get_throw_pending_exception (void)
        amd64_alu_reg_imm (code, X86_SUB, AMD64_RSP, 8);
 
        /* Obtain the original ip and clear the flag in previous_lmf */
-       amd64_mov_reg_imm (code, AMD64_R11, get_original_ip);
+       if (aot) {
+               *ji = mono_patch_info_list_prepend (*ji, code - start, MONO_PATCH_INFO_JIT_ICALL_ADDR, "mono_amd64_get_original_ip");
+               amd64_mov_reg_membase (code, AMD64_R11, AMD64_RIP, 0, 8);
+       } else {
+               amd64_mov_reg_imm (code, AMD64_R11, mono_amd64_get_original_ip);
+       }
        amd64_call_reg (code, AMD64_R11);       
 
        /* Load exc */
@@ -957,8 +973,13 @@ get_throw_pending_exception (void)
        amd64_push_reg (code, AMD64_RAX);
 
        /* Call the throw trampoline */
-       throw_trampoline = mono_get_throw_exception ();
-       amd64_mov_reg_imm (code, AMD64_R11, throw_trampoline);
+       if (aot) {
+               *ji = mono_patch_info_list_prepend (*ji, code - start, MONO_PATCH_INFO_JIT_ICALL_ADDR, "mono_amd64_throw_exception");
+               amd64_mov_reg_membase (code, AMD64_R11, AMD64_RIP, 0, 8);
+       } else {
+               throw_trampoline = mono_get_throw_exception ();
+               amd64_mov_reg_imm (code, AMD64_R11, throw_trampoline);
+       }
        /* We use a jump instead of a call so we can push the original ip on the stack */
        amd64_jump_reg (code, AMD64_R11);
 
@@ -966,7 +987,12 @@ get_throw_pending_exception (void)
        mono_amd64_patch (br [0], code);
 
        /* Obtain the original ip and clear the flag in previous_lmf */
-       amd64_mov_reg_imm (code, AMD64_R11, get_original_ip);
+       if (aot) {
+               *ji = mono_patch_info_list_prepend (*ji, code - start, MONO_PATCH_INFO_JIT_ICALL_ADDR, "mono_amd64_get_original_ip");
+               amd64_mov_reg_membase (code, AMD64_R11, AMD64_RIP, 0, 8);
+       } else {
+               amd64_mov_reg_imm (code, AMD64_R11, mono_amd64_get_original_ip);
+       }
        amd64_call_reg (code, AMD64_R11);       
        amd64_mov_reg_reg (code, AMD64_R11, AMD64_RAX, 8);
 
@@ -982,11 +1008,13 @@ get_throw_pending_exception (void)
 
        g_assert ((code - start) < 128);
 
-       inited = TRUE;
+       *code_size = code - start;
 
        return start;
 }
 
+static gpointer throw_pending_exception;
+
 /*
  * Called when a thread receives an async exception while executing unmanaged code.
  * Instead of checking for this exception in the managed-to-native wrapper, we hijack 
@@ -1011,14 +1039,21 @@ mono_arch_notify_pending_exc (void)
        /* Signal that lmf->rip is set */
        lmf->previous_lmf = (gpointer)((guint64)lmf->previous_lmf | 1);
 
-       *(gpointer*)(lmf->rsp - 8) = get_throw_pending_exception ();
+       *(gpointer*)(lmf->rsp - 8) = throw_pending_exception;
 }
 
 void
 mono_arch_exceptions_init (void)
 {
-       /* Call this to avoid initialization races */
-       get_throw_pending_exception ();
+       guint32 code_size;
+       MonoJumpInfo *ji;
+
+       if (mono_aot_only) {
+               throw_pending_exception = mono_aot_get_named_code ("throw_pending_exception");
+       } else {
+               /* Call this to avoid initialization races */
+               throw_pending_exception = mono_arch_get_throw_pending_exception_full (&code_size, &ji, FALSE);
+       }
 }
 
 #ifdef PLATFORM_WIN32
@@ -1263,5 +1298,58 @@ mono_arch_unwindinfo_install_unwind_info (gpointer* monoui, gpointer code, guint
 
 #endif
 
+#if MONO_SUPPORT_TASKLETS
+MonoContinuationRestore
+mono_tasklets_arch_restore (void)
+{
+       static guint8* saved = NULL;
+       guint8 *code, *start;
+       int cont_reg = AMD64_R9; /* register usable on both call conventions */
+
+       if (saved)
+               return (MonoContinuationRestore)saved;
+       code = start = mono_global_codeman_reserve (64);
+       /* the signature is: restore (MonoContinuation *cont, int state, MonoLMF **lmf_addr) */
+       /* cont is in AMD64_ARG_REG1 ($rcx or $rdi)
+        * state is in AMD64_ARG_REG2 ($rdx or $rsi)
+        * lmf_addr is in AMD64_ARG_REG3 ($r8 or $rdx)
+        * We move cont to cont_reg since we need both rcx and rdi for the copy
+        * state is moved to $rax so it's setup as the return value and we can overwrite $rsi
+        */
+       amd64_mov_reg_reg (code, cont_reg, MONO_AMD64_ARG_REG1, 8);
+       amd64_mov_reg_reg (code, AMD64_RAX, MONO_AMD64_ARG_REG2, 8);
+       /* setup the copy of the stack */
+       amd64_mov_reg_membase (code, AMD64_RCX, cont_reg, G_STRUCT_OFFSET (MonoContinuation, stack_used_size), sizeof (int));
+       amd64_shift_reg_imm (code, X86_SHR, AMD64_RCX, 3);
+       x86_cld (code);
+       amd64_mov_reg_membase (code, AMD64_RSI, cont_reg, G_STRUCT_OFFSET (MonoContinuation, saved_stack), sizeof (gpointer));
+       amd64_mov_reg_membase (code, AMD64_RDI, cont_reg, G_STRUCT_OFFSET (MonoContinuation, return_sp), sizeof (gpointer));
+       amd64_prefix (code, X86_REP_PREFIX);
+       amd64_movsl (code);
+
+       /* now restore the registers from the LMF */
+       amd64_mov_reg_membase (code, AMD64_RCX, cont_reg, G_STRUCT_OFFSET (MonoContinuation, lmf), 8);
+       amd64_mov_reg_membase (code, AMD64_RBX, AMD64_RCX, G_STRUCT_OFFSET (MonoLMF, rbx), 8);
+       amd64_mov_reg_membase (code, AMD64_RBP, AMD64_RCX, G_STRUCT_OFFSET (MonoLMF, rbp), 8);
+       amd64_mov_reg_membase (code, AMD64_R12, AMD64_RCX, G_STRUCT_OFFSET (MonoLMF, r12), 8);
+       amd64_mov_reg_membase (code, AMD64_R13, AMD64_RCX, G_STRUCT_OFFSET (MonoLMF, r13), 8);
+       amd64_mov_reg_membase (code, AMD64_R14, AMD64_RCX, G_STRUCT_OFFSET (MonoLMF, r14), 8);
+       amd64_mov_reg_membase (code, AMD64_R15, AMD64_RCX, G_STRUCT_OFFSET (MonoLMF, r15), 8);
+#ifdef PLATFORM_WIN32
+       amd64_mov_reg_membase (code, AMD64_RDI, AMD64_RCX, G_STRUCT_OFFSET (MonoLMF, rdi), 8);
+       amd64_mov_reg_membase (code, AMD64_RSI, AMD64_RCX, G_STRUCT_OFFSET (MonoLMF, rsi), 8);
+#endif
+       amd64_mov_reg_membase (code, AMD64_RSP, AMD64_RCX, G_STRUCT_OFFSET (MonoLMF, rsp), 8);
 
+       /* restore the lmf chain */
+       /*x86_mov_reg_membase (code, X86_ECX, X86_ESP, 12, 4);
+       x86_mov_membase_reg (code, X86_ECX, 0, X86_EDX, 4);*/
+
+       /* state is already in rax */
+       amd64_jump_membase (code, cont_reg, G_STRUCT_OFFSET (MonoContinuation, return_ip));
+       g_assert ((code - start) <= 64);
+       saved = start;
+       return (MonoContinuationRestore)saved;
+}
+#endif