New test.
[mono.git] / mono / mini / exceptions-amd64.c
index 5873c08507fd7b4f9201cca17017bec314ae6172..7ab4d62976dfa514b8a3756e945be6b1127a62c1 100644 (file)
@@ -11,8 +11,8 @@
 #include <glib.h>
 #include <signal.h>
 #include <string.h>
-#ifndef PLATFORM_WIN32
-#include <sys/ucontext.h>
+#ifdef HAVE_UCONTEXT_H
+#include <ucontext.h>
 #endif
 
 #include <mono/arch/amd64/amd64-codegen.h>
 
 #include "mini.h"
 #include "mini-amd64.h"
+#include "tasklets.h"
+#include "debug-mini.h"
 
 #define ALIGN_TO(val,align) (((val) + ((align) - 1)) & ~((align) - 1))
 
-#ifdef PLATFORM_WIN32
+#ifdef TARGET_WIN32
 static MonoW32ExceptionHandler fpe_handler;
 static MonoW32ExceptionHandler ill_handler;
 static MonoW32ExceptionHandler segv_handler;
@@ -39,7 +41,7 @@ static MonoW32ExceptionHandler segv_handler;
 static LPTOP_LEVEL_EXCEPTION_FILTER old_handler;
 
 #define W32_SEH_HANDLE_EX(_ex) \
-       if (_ex##_handler) _ex##_handler((int)sctx)
+       if (_ex##_handler) _ex##_handler(0, er, sctx)
 
 /*
  * Unhandled Exception Filter
@@ -93,15 +95,22 @@ LONG CALLBACK seh_handler(EXCEPTION_POINTERS* ep)
        }
 
        /* Copy context back */
-       ctx->Rax = sctx->rax;
-       ctx->Rbx = sctx->rbx;
-       ctx->Rcx = sctx->rcx;
-       ctx->Rdx = sctx->rdx;
+       /* Nonvolatile */
+       ctx->Rsp = sctx->rsp; 
+       ctx->Rdi = sctx->rdi; 
+       ctx->Rsi = sctx->rsi; 
+       ctx->Rbx = sctx->rbx; 
        ctx->Rbp = sctx->rbp;
-       ctx->Rsp = sctx->rsp;
-       ctx->Rsi = sctx->rsi;
-       ctx->Rdi = sctx->rdi;
-       ctx->Rip = sctx->rip;
+       ctx->R12 = sctx->r12; 
+       ctx->R13 = sctx->r13; 
+       ctx->R14 = sctx->r14;
+       ctx->R15 = sctx->r15;
+       ctx->Rip = sctx->rip; 
+
+       /* Volatile But should not matter?*/
+       ctx->Rax = sctx->rax; 
+       ctx->Rcx = sctx->rcx; 
+       ctx->Rdx = sctx->rdx;
 
        g_free (sctx);
 
@@ -135,7 +144,7 @@ void win32_seh_set_handler(int type, MonoW32ExceptionHandler handler)
        }
 }
 
-#endif /* PLATFORM_WIN32 */
+#endif /* TARGET_WIN32 */
 
 /*
  * mono_arch_get_restore_context:
@@ -243,7 +252,7 @@ mono_arch_get_call_filter_full (guint32 *code_size, MonoJumpInfo **ji, gboolean
        amd64_mov_reg_membase (code, AMD64_R13, AMD64_ARG_REG1, G_STRUCT_OFFSET (MonoContext, r13), 8);
        amd64_mov_reg_membase (code, AMD64_R14, AMD64_ARG_REG1, G_STRUCT_OFFSET (MonoContext, r14), 8);
        amd64_mov_reg_membase (code, AMD64_R15, AMD64_ARG_REG1, G_STRUCT_OFFSET (MonoContext, r15), 8);
-#ifdef PLATFORM_WIN32
+#ifdef TARGET_WIN32
        amd64_mov_reg_membase (code, AMD64_RDI, AMD64_ARG_REG1,  G_STRUCT_OFFSET (MonoContext, rdi), 8);
        amd64_mov_reg_membase (code, AMD64_RSI, AMD64_ARG_REG1,  G_STRUCT_OFFSET (MonoContext, rsi), 8);
 #endif
@@ -307,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);
 
@@ -349,6 +353,7 @@ get_throw_trampoline (gboolean rethrow, guint32 *code_size, MonoJumpInfo **ji, g
 {
        guint8* start;
        guint8 *code;
+       GSList *unwind_ops;
 
        start = code = mono_global_codeman_reserve (64);
 
@@ -356,6 +361,8 @@ get_throw_trampoline (gboolean rethrow, guint32 *code_size, MonoJumpInfo **ji, g
 
        *ji = NULL;
 
+       unwind_ops = mono_arch_get_cie_program ();
+
        amd64_mov_reg_reg (code, AMD64_R11, AMD64_RSP, 8);
 
        /* reverse order */
@@ -382,7 +389,9 @@ get_throw_trampoline (gboolean rethrow, guint32 *code_size, MonoJumpInfo **ji, g
        /* Exception */
        amd64_push_reg (code, AMD64_ARG_REG1);
 
-#ifdef PLATFORM_WIN32
+       mono_add_unwind_op_def_cfa_offset (unwind_ops, code, start, (15 + 1) * sizeof (gpointer));
+
+#ifdef TARGET_WIN32
        /* align stack */
        amd64_push_imm (code, 0);
        amd64_push_imm (code, 0);
@@ -407,6 +416,8 @@ get_throw_trampoline (gboolean rethrow, guint32 *code_size, MonoJumpInfo **ji, g
 
        *code_size = code - start;
 
+       mono_save_trampoline_xdebug_info ("throw_exception_trampoline", start, code - start, unwind_ops);
+
        return start;
 }
 
@@ -430,26 +441,6 @@ mono_arch_get_rethrow_exception_full (guint32 *code_size, MonoJumpInfo **ji, gbo
        return get_throw_trampoline (TRUE, code_size, ji, aot);
 }
 
-gpointer 
-mono_arch_get_throw_exception_by_name_full (guint32 *code_size, MonoJumpInfo **ji, gboolean aot)
-{      
-       guint8* start;
-       guint8 *code;
-
-       start = code = mono_global_codeman_reserve (64);
-
-       *ji = NULL;
-
-       /* Not used on amd64 */
-       amd64_breakpoint (code);
-
-       mono_arch_flush_icache (start, code - start);
-
-       *code_size = code - start;
-
-       return start;
-}
-
 /**
  * mono_arch_get_throw_corlib_exception:
  *
@@ -485,11 +476,11 @@ mono_arch_get_throw_corlib_exception_full (guint32 *code_size, MonoJumpInfo **ji
                amd64_mov_reg_imm (code, AMD64_ARG_REG1, mono_defaults.exception_class->image);
                amd64_mov_reg_imm (code, AMD64_R11, mono_exception_from_token);
        }
-#ifdef PLATFORM_WIN32
+#ifdef TARGET_WIN32
        amd64_alu_reg_imm (code, X86_SUB, AMD64_RSP, 32);
 #endif
        amd64_call_reg (code, AMD64_R11);
-#ifdef PLATFORM_WIN32
+#ifdef TARGET_WIN32
        amd64_alu_reg_imm (code, X86_ADD, AMD64_RSP, 32);
 #endif
 
@@ -524,126 +515,90 @@ mono_arch_get_throw_corlib_exception_full (guint32 *code_size, MonoJumpInfo **ji
        return start;
 }
 
-/* mono_arch_find_jit_info:
+/*
+ * mono_arch_find_jit_info_ext:
  *
- * This function is used to gather information from @ctx. It return the 
- * MonoJitInfo of the corresponding function, unwinds one stack frame and
- * stores the resulting context into @new_ctx. It also stores a string 
- * describing the stack location into @trace (if not NULL), and modifies
- * the @lmf if necessary. @native_offset return the IP offset from the 
- * start of the function or -1 if that info is not available.
+ * This function is used to gather information from @ctx, and store it in @frame_info.
+ * It unwinds one stack frame, and stores the resulting context into @new_ctx. @lmf
+ * is modified if needed.
+ * Returns TRUE on success, FALSE otherwise.
+ * This function is a version of mono_arch_find_jit_info () where all the results are
+ * returned in a StackFrameInfo structure.
  */
-MonoJitInfo *
-mono_arch_find_jit_info (MonoDomain *domain, MonoJitTlsData *jit_tls, MonoJitInfo *res, MonoJitInfo *prev_ji, MonoContext *ctx, 
-                        MonoContext *new_ctx, MonoLMF **lmf, gboolean *managed)
+gboolean
+mono_arch_find_jit_info_ext (MonoDomain *domain, MonoJitTlsData *jit_tls, 
+                                                        MonoJitInfo *ji, MonoContext *ctx, 
+                                                        MonoContext *new_ctx, MonoLMF **lmf, 
+                                                        StackFrameInfo *frame)
 {
-       MonoJitInfo *ji;
-       int i;
        gpointer ip = MONO_CONTEXT_GET_IP (ctx);
 
-       /* Avoid costly table lookup during stack overflow */
-       if (prev_ji && (ip > prev_ji->code_start && ((guint8*)ip < ((guint8*)prev_ji->code_start) + prev_ji->code_size)))
-               ji = prev_ji;
-       else
-               ji = mono_jit_info_table_find (domain, ip);
-
-       if (managed)
-               *managed = FALSE;
+       memset (frame, 0, sizeof (StackFrameInfo));
+       frame->ji = ji;
+       frame->managed = FALSE;
 
        *new_ctx = *ctx;
 
        if (ji != NULL) {
-               int offset;
-               gboolean omit_fp = (ji->used_regs & (1 << 31)) > 0;
-
-               if (managed)
-                       if (!ji->method->wrapper_type)
-                               *managed = TRUE;
-
-               /*
-                * If a method has save_lmf set, then register save/restore code is not generated 
-                * by the JIT, so we have to restore callee saved registers from the lmf.
-                */
-               if (ji->method->save_lmf) {
-                       MonoLMF *lmf_addr;
-
-                       /* 
-                        * *lmf might not point to the LMF pushed by this method, so compute the LMF
-                        * address instead.
-                        */
-                       if (omit_fp)
-                               lmf_addr = (MonoLMF*)ctx->rsp;
-                       else
-                               lmf_addr = (MonoLMF*)(ctx->rbp - sizeof (MonoLMF));
-
-                       new_ctx->rbp = lmf_addr->rbp;
-                       new_ctx->rbx = lmf_addr->rbx;
-                       new_ctx->r12 = lmf_addr->r12;
-                       new_ctx->r13 = lmf_addr->r13;
-                       new_ctx->r14 = lmf_addr->r14;
-                       new_ctx->r15 = lmf_addr->r15;
-               }
-               else {
-                       offset = omit_fp ? 0 : -1;
-                       /* restore caller saved registers */
-                       for (i = 0; i < AMD64_NREG; i ++)
-                               if (AMD64_IS_CALLEE_SAVED_REG (i) && (ji->used_regs & (1 << i))) {
-                                       guint64 reg;
-
-                                       if (omit_fp) {
-                                               reg = *((guint64*)ctx->rsp + offset);
-                                               offset ++;
-                                       }
-                                       else {
-                                               reg = *((guint64 *)ctx->rbp + offset);
-                                               offset --;
-                                       }
-
-                                       switch (i) {
-                                       case AMD64_RBX:
-                                               new_ctx->rbx = reg;
-                                               break;
-                                       case AMD64_R12:
-                                               new_ctx->r12 = reg;
-                                               break;
-                                       case AMD64_R13:
-                                               new_ctx->r13 = reg;
-                                               break;
-                                       case AMD64_R14:
-                                               new_ctx->r14 = reg;
-                                               break;
-                                       case AMD64_R15:
-                                               new_ctx->r15 = reg;
-                                               break;
-                                       case AMD64_RBP:
-                                               new_ctx->rbp = reg;
-                                               break;
-                                       default:
-                                               g_assert_not_reached ();
-                                       }
-                               }
-               }
+               gssize regs [MONO_MAX_IREGS + 1];
+               guint8 *cfa;
+               guint32 unwind_info_len;
+               guint8 *unwind_info;
+
+               frame->type = FRAME_TYPE_MANAGED;
+
+               if (!ji->method->wrapper_type || ji->method->wrapper_type == MONO_WRAPPER_DYNAMIC_METHOD)
+                       frame->managed = TRUE;
+
+               if (ji->from_aot)
+                       unwind_info = mono_aot_get_unwind_info (ji, &unwind_info_len);
+               else
+                       unwind_info = mono_get_cached_unwind_info (ji->used_regs, &unwind_info_len);
+               regs [AMD64_RAX] = new_ctx->rax;
+               regs [AMD64_RBX] = new_ctx->rbx;
+               regs [AMD64_RCX] = new_ctx->rcx;
+               regs [AMD64_RDX] = new_ctx->rdx;
+               regs [AMD64_RBP] = new_ctx->rbp;
+               regs [AMD64_RSP] = new_ctx->rsp;
+               regs [AMD64_RSI] = new_ctx->rsi;
+               regs [AMD64_RDI] = new_ctx->rdi;
+               regs [AMD64_RIP] = new_ctx->rip;
+               regs [AMD64_R12] = new_ctx->r12;
+               regs [AMD64_R13] = new_ctx->r13;
+               regs [AMD64_R14] = new_ctx->r14;
+               regs [AMD64_R15] = new_ctx->r15;
+
+               mono_unwind_frame (unwind_info, unwind_info_len, ji->code_start, 
+                                                  (guint8*)ji->code_start + ji->code_size,
+                                                  ip, regs, MONO_MAX_IREGS + 1, &cfa);
+
+               new_ctx->rax = regs [AMD64_RAX];
+               new_ctx->rbx = regs [AMD64_RBX];
+               new_ctx->rcx = regs [AMD64_RCX];
+               new_ctx->rdx = regs [AMD64_RDX];
+               new_ctx->rbp = regs [AMD64_RBP];
+               new_ctx->rsp = regs [AMD64_RSP];
+               new_ctx->rsi = regs [AMD64_RSI];
+               new_ctx->rdi = regs [AMD64_RDI];
+               new_ctx->rip = regs [AMD64_RIP];
+               new_ctx->r12 = regs [AMD64_R12];
+               new_ctx->r13 = regs [AMD64_R13];
+               new_ctx->r14 = regs [AMD64_R14];
+               new_ctx->r15 = regs [AMD64_R15];
+               /* The CFA becomes the new SP value */
+               new_ctx->rsp = (gssize)cfa;
+
+               /* Adjust IP */
+               new_ctx->rip --;
 
                if (*lmf && ((*lmf) != jit_tls->first_lmf) && (MONO_CONTEXT_GET_SP (ctx) >= (gpointer)(*lmf)->rsp)) {
                        /* remove any unused lmf */
-                       *lmf = (gpointer)(((guint64)(*lmf)->previous_lmf) & ~1);
-               }
-
-               if (omit_fp) {
-                       /* Pop frame */
-                       new_ctx->rsp += (ji->used_regs >> 16) & (0x7fff);
-                       new_ctx->rip = *((guint64 *)new_ctx->rsp) - 1;
-                       /* Pop return address */
-                       new_ctx->rsp += 8;
-               }
-               else {
-                       /* Pop EBP and the return address */
-                       new_ctx->rsp = ctx->rbp + (2 * sizeof (gpointer));
-                       /* we substract 1, so that the IP points into the call instruction */
-                       new_ctx->rip = *((guint64 *)ctx->rbp + 1) - 1;
-                       new_ctx->rbp = *((guint64 *)ctx->rbp);
+                       *lmf = (gpointer)(((guint64)(*lmf)->previous_lmf) & ~3);
                }
 
+#ifndef MONO_AMD64_NO_PUSHES
                /* Pop arguments off the stack */
                {
                        MonoJitArgumentInfo *arg_info = g_newa (MonoJitArgumentInfo, mono_method_signature (ji->method)->param_count + 1);
@@ -651,17 +606,36 @@ 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;
+               return TRUE;
        } else if (*lmf) {
                guint64 rip;
 
+               if (((guint64)(*lmf)->previous_lmf) & 2) {
+                       /* 
+                        * This LMF entry is created by the soft debug code to mark transitions to
+                        * managed code done during invokes.
+                        */
+                       MonoLMFExt *ext = (MonoLMFExt*)(*lmf);
+
+                       g_assert (ext->debugger_invoke);
+
+                       memcpy (new_ctx, &ext->ctx, sizeof (MonoContext));
+
+                       *lmf = (gpointer)(((guint64)(*lmf)->previous_lmf) & ~3);
+
+                       frame->type = FRAME_TYPE_DEBUGGER_INVOKE;
+
+                       return TRUE;
+               }
+
                if (((guint64)(*lmf)->previous_lmf) & 1) {
                        /* This LMF has the rip field set */
                        rip = (*lmf)->rip;
                } else if ((*lmf)->rsp == 0) {
                        /* Top LMF entry */
-                       return (gpointer)-1;
+                       return FALSE;
                } else {
                        /* 
                         * The rsp field is set just before the call which transitioned to native 
@@ -670,16 +644,15 @@ mono_arch_find_jit_info (MonoDomain *domain, MonoJitTlsData *jit_tls, MonoJitInf
                        rip = *(guint64*)((*lmf)->rsp - sizeof (gpointer));
                }
 
-               ji = mono_jit_info_table_find (domain, (gpointer)rip);
+               ji = mini_jit_info_table_find (domain, (gpointer)rip, NULL);
                if (!ji) {
-                       if (!(*lmf)->method)
-                               /* Top LMF entry */
-                               return (gpointer)-1;
-                       /* Trampoline lmf frame */
-                       memset (res, 0, sizeof (MonoJitInfo));
-                       res->method = (*lmf)->method;
+                       // FIXME: This can happen with multiple appdomains (bug #444383)
+                       return FALSE;
                }
 
+               frame->ji = ji;
+               frame->type = FRAME_TYPE_MANAGED_TO_NATIVE;
+
                new_ctx->rip = rip;
                new_ctx->rbp = (*lmf)->rbp;
                new_ctx->rsp = (*lmf)->rsp;
@@ -689,13 +662,17 @@ mono_arch_find_jit_info (MonoDomain *domain, MonoJitTlsData *jit_tls, MonoJitInf
                new_ctx->r13 = (*lmf)->r13;
                new_ctx->r14 = (*lmf)->r14;
                new_ctx->r15 = (*lmf)->r15;
+#ifdef TARGET_WIN32
+               new_ctx->rdi = (*lmf)->rdi;
+               new_ctx->rsi = (*lmf)->rsi;
+#endif
 
-               *lmf = (gpointer)(((guint64)(*lmf)->previous_lmf) & ~1);
+               *lmf = (gpointer)(((guint64)(*lmf)->previous_lmf) & ~3);
 
-               return ji ? ji : res;
+               return TRUE;
        }
 
-       return NULL;
+       return FALSE;
 }
 
 /**
@@ -711,6 +688,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);
@@ -718,23 +698,17 @@ mono_arch_handle_exception (void *sigctx, gpointer obj, gboolean test_only)
        return TRUE;
 }
 
-#ifdef MONO_ARCH_USE_SIGACTION
+#if defined(MONO_ARCH_USE_SIGACTION) && defined(UCONTEXT_GREGS)
 static inline guint64*
 gregs_from_ucontext (ucontext_t *ctx)
 {
-#ifdef __FreeBSD__
-    guint64 *gregs = (guint64 *) &ctx->uc_mcontext;
-#else
-    guint64 *gregs = (guint64 *) &ctx->uc_mcontext.gregs;
-#endif
-
-       return gregs;
+       return (guint64 *) UCONTEXT_GREGS (ctx);
 }
 #endif
 void
 mono_arch_sigctx_to_monoctx (void *sigctx, MonoContext *mctx)
 {
-#ifdef MONO_ARCH_USE_SIGACTION
+#if defined(MONO_ARCH_USE_SIGACTION) && defined(UCONTEXT_GREGS)
        ucontext_t *ctx = (ucontext_t*)sigctx;
 
     guint64 *gregs = gregs_from_ucontext (ctx);
@@ -752,6 +726,22 @@ mono_arch_sigctx_to_monoctx (void *sigctx, MonoContext *mctx)
        mctx->r13 = gregs [REG_R13];
        mctx->r14 = gregs [REG_R14];
        mctx->r15 = gregs [REG_R15];
+#elif defined(MONO_ARCH_USE_SIGACTION)
+       ucontext_t *ctx = (ucontext_t*)sigctx;
+
+       mctx->rax = UCONTEXT_REG_RAX (ctx);
+       mctx->rbx = UCONTEXT_REG_RBX (ctx);
+       mctx->rcx = UCONTEXT_REG_RCX (ctx);
+       mctx->rdx = UCONTEXT_REG_RDX (ctx);
+       mctx->rbp = UCONTEXT_REG_RBP (ctx);
+       mctx->rsp = UCONTEXT_REG_RSP (ctx);
+       mctx->rsi = UCONTEXT_REG_RSI (ctx);
+       mctx->rdi = UCONTEXT_REG_RDI (ctx);
+       mctx->rip = UCONTEXT_REG_RIP (ctx);
+       mctx->r12 = UCONTEXT_REG_R12 (ctx);
+       mctx->r13 = UCONTEXT_REG_R13 (ctx);
+       mctx->r14 = UCONTEXT_REG_R14 (ctx);
+       mctx->r15 = UCONTEXT_REG_R15 (ctx);
 #else
        MonoContext *ctx = (MonoContext *)sigctx;
 
@@ -774,7 +764,7 @@ mono_arch_sigctx_to_monoctx (void *sigctx, MonoContext *mctx)
 void
 mono_arch_monoctx_to_sigctx (MonoContext *mctx, void *sigctx)
 {
-#ifdef MONO_ARCH_USE_SIGACTION
+#if defined(MONO_ARCH_USE_SIGACTION) && defined(UCONTEXT_GREGS)
        ucontext_t *ctx = (ucontext_t*)sigctx;
 
     guint64 *gregs = gregs_from_ucontext (ctx);
@@ -792,6 +782,22 @@ mono_arch_monoctx_to_sigctx (MonoContext *mctx, void *sigctx)
        gregs [REG_R13] = mctx->r13;
        gregs [REG_R14] = mctx->r14;
        gregs [REG_R15] = mctx->r15;
+#elif defined(MONO_ARCH_USE_SIGACTION)
+       ucontext_t *ctx = (ucontext_t*)sigctx;
+
+       UCONTEXT_REG_RAX (ctx) = mctx->rax;
+       UCONTEXT_REG_RBX (ctx) = mctx->rbx;
+       UCONTEXT_REG_RCX (ctx) = mctx->rcx;
+       UCONTEXT_REG_RDX (ctx) = mctx->rdx;
+       UCONTEXT_REG_RBP (ctx) = mctx->rbp;
+       UCONTEXT_REG_RSP (ctx) = mctx->rsp;
+       UCONTEXT_REG_RSI (ctx) = mctx->rsi;
+       UCONTEXT_REG_RDI (ctx) = mctx->rdi;
+       UCONTEXT_REG_RIP (ctx) = mctx->rip;
+       UCONTEXT_REG_R12 (ctx) = mctx->r12;
+       UCONTEXT_REG_R13 (ctx) = mctx->r13;
+       UCONTEXT_REG_R14 (ctx) = mctx->r14;
+       UCONTEXT_REG_R15 (ctx) = mctx->r15;
 #else
        MonoContext *ctx = (MonoContext *)sigctx;
 
@@ -814,14 +820,16 @@ mono_arch_monoctx_to_sigctx (MonoContext *mctx, void *sigctx)
 gpointer
 mono_arch_ip_from_context (void *sigctx)
 {
-       
-#ifdef MONO_ARCH_USE_SIGACTION
-
+#if defined(MONO_ARCH_USE_SIGACTION) && defined(UCONTEXT_GREGS)
        ucontext_t *ctx = (ucontext_t*)sigctx;
 
     guint64 *gregs = gregs_from_ucontext (ctx);
 
        return (gpointer)gregs [REG_RIP];
+#elif defined(MONO_ARCH_USE_SIGACTION)
+       ucontext_t *ctx = (ucontext_t*)sigctx;
+
+       return (gpointer)UCONTEXT_REG_RIP (ctx);
 #else
        MonoContext *ctx = sigctx;
        return (gpointer)ctx->rip;
@@ -862,6 +870,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);
@@ -871,11 +886,11 @@ altstack_handle_and_restore (void *sigctx, gpointer obj, gboolean stack_ovf)
 void
 mono_arch_handle_altstack_exception (void *sigctx, gpointer fault_addr, gboolean stack_ovf)
 {
-#ifdef MONO_ARCH_USE_SIGACTION
+#if defined(MONO_ARCH_USE_SIGACTION) && defined(UCONTEXT_GREGS)
        MonoException *exc = NULL;
        ucontext_t *ctx = (ucontext_t*)sigctx;
        guint64 *gregs = gregs_from_ucontext (ctx);
-       MonoJitInfo *ji = mono_jit_info_table_find (mono_domain_get (), (gpointer)gregs [REG_RIP]);
+       MonoJitInfo *ji = mini_jit_info_table_find (mono_domain_get (), (gpointer)gregs [REG_RIP], NULL);
        gpointer *sp;
        int frame_size;
 
@@ -910,8 +925,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 ();
 
@@ -923,17 +938,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);
 
@@ -954,7 +966,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 */
@@ -969,7 +986,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 */
@@ -985,8 +1007,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);
 
@@ -994,7 +1021,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);
 
@@ -1010,11 +1042,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 
@@ -1026,6 +1060,10 @@ mono_arch_notify_pending_exc (void)
 {
        MonoLMF *lmf = mono_get_lmf ();
 
+       if (!lmf)
+               /* Not yet started */
+               return;
+
        if (lmf->rsp == 0)
                /* Initial LMF */
                return;
@@ -1039,10 +1077,24 @@ 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)
+{
+       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
+#ifdef TARGET_WIN32
 
 /*
  * The mono_arch_unwindinfo* methods are used to build and add
@@ -1231,6 +1283,27 @@ mono_arch_unwindinfo_get_size (gpointer monoui)
                (sizeof (UNWIND_CODE) * (MONO_MAX_UNWIND_CODES - unwindinfo->unwindInfo.CountOfCodes));
 }
 
+PRUNTIME_FUNCTION
+MONO_GET_RUNTIME_FUNCTION_CALLBACK ( DWORD64 ControlPc, IN PVOID Context )
+{
+       MonoJitInfo *ji;
+       guint64 pos;
+       PMonoUnwindInfo targetinfo;
+       MonoDomain *domain = mono_domain_get ();
+
+       ji = mini_jit_info_table_find (domain, (char*)ControlPc, NULL);
+       if (!ji)
+               return 0;
+
+       pos = (guint64)(((char*)ji->code_start) + ji->code_size);
+       
+       targetinfo = (PMonoUnwindInfo)ALIGN_TO (pos, 8);
+
+       targetinfo->runtimeFunction.UnwindData = ((DWORD64)&targetinfo->unwindInfo) - ((DWORD64)Context);
+
+       return &targetinfo->runtimeFunction;
+}
+
 void
 mono_arch_unwindinfo_install_unwind_info (gpointer* monoui, gpointer code, guint code_size)
 {
@@ -1258,10 +1331,63 @@ mono_arch_unwindinfo_install_unwind_info (gpointer* monoui, gpointer code, guint
        g_free (unwindinfo);
        *monoui = 0;
 
-       RtlAddFunctionTable (&targetinfo->runtimeFunction, 1, (DWORD64)code);
+       RtlInstallFunctionTableCallback (((DWORD64)code) | 0x3, (DWORD64)code, code_size, MONO_GET_RUNTIME_FUNCTION_CALLBACK, code, NULL);
 }
 
 #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 TARGET_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