Increase the number of trampoline entries reserved for trampolines.
[mono.git] / mono / mini / tramp-amd64.c
index 381e1b968601b0a7c4791cae67a62369962090a5..3c108f307cf0cd91a1141a645ff4eef6f9295454 100644 (file)
@@ -6,6 +6,8 @@
  *   Zoltan Varga (vargaz@gmail.com)
  *
  * (C) 2001 Ximian, Inc.
+ * Copyright 2003-2011 Novell, Inc (http://www.novell.com)
+ * Copyright 2011 Xamarin, Inc (http://www.xamarin.com)
  */
 
 #include <config.h>
@@ -61,9 +63,7 @@ mono_arch_get_unbox_trampoline (MonoMethod *m, gpointer addr)
        amd64_jump_reg (code, AMD64_RAX);
        g_assert ((code - start) < 20);
 
-#if defined(__native_client_codegen__) && defined(__native_client__)
        nacl_domain_code_validate (domain, &start, 20, &code);
-#endif
 
        mono_arch_flush_icache (start, code - start);
 
@@ -99,9 +99,7 @@ mono_arch_get_static_rgctx_trampoline (MonoMethod *m, MonoMethodRuntimeGenericCo
        amd64_jump_code (code, addr);
        g_assert ((code - start) < buf_len);
 
-#if defined(__native_client_codegen__) && defined(__native_client__)
        nacl_domain_code_validate (domain, &start, buf_len, &code);
-#endif
        mono_arch_flush_icache (start, code - start);
 
        return start;
@@ -129,9 +127,7 @@ mono_arch_get_llvm_imt_trampoline (MonoDomain *domain, MonoMethod *m, int vt_off
 
        g_assert ((code - start) < buf_len);
 
-#if defined(__native_client_codegen__) && defined(__native_client__)
        nacl_domain_code_validate (domain, &start, buf_len, &code);
-#endif
 
        mono_arch_flush_icache (start, code - start);
 
@@ -162,7 +158,9 @@ mono_arch_patch_callsite (guint8 *method_start, guint8 *orig_code, guint8 *addr)
                                VALGRIND_DISCARD_TRANSLATIONS (orig_code - 11, sizeof (gpointer));
                        }
                } else {
-                       if ((((guint64)(addr)) >> 32) != 0) {
+                       gboolean disp_32bit = ((((gint64)addr - (gint64)orig_code)) < (1 << 30)) && ((((gint64)addr - (gint64)orig_code)) > -(1 << 30));
+
+                       if ((((guint64)(addr)) >> 32) != 0 && !disp_32bit) {
 #ifdef MONO_ARCH_NOMAP32BIT
                                /* Print some diagnostics */
                                MonoJitInfo *ji = mono_jit_info_table_find (mono_domain_get (), (char*)orig_code);
@@ -187,7 +185,6 @@ mono_arch_patch_callsite (guint8 *method_start, guint8 *orig_code, guint8 *addr)
                                mono_arch_flush_icache (thunk_start, thunk_code - thunk_start);
 #endif
                        }
-                       g_assert ((((guint64)(orig_code)) >> 32) == 0);
                        if (can_write) {
                                InterlockedExchange ((gint32*)(orig_code - 4), ((gint64)addr - (gint64)orig_code));
                                VALGRIND_DISCARD_TRANSLATIONS (orig_code - 5, 4);
@@ -386,6 +383,13 @@ mono_arch_nullify_plt_entry (guint8 *code, mgreg_t *regs)
        mono_arch_patch_plt_entry (code, NULL, regs, nullified_class_init_trampoline);
 }
 
+static void
+stack_unaligned (MonoTrampolineType tramp_type)
+{
+       printf ("%d\n", tramp_type);
+       g_assert_not_reached ();
+}
+
 guchar*
 mono_arch_create_generic_trampoline (MonoTrampolineType tramp_type, MonoTrampInfo **info, gboolean aot)
 {
@@ -395,13 +399,13 @@ mono_arch_create_generic_trampoline (MonoTrampolineType tramp_type, MonoTrampInf
        gboolean has_caller;
        GSList *unwind_ops = NULL;
        MonoJumpInfo *ji = NULL;
-       const guint kMaxCodeSize = NACL_SIZE (548, 548*2);
+       const guint kMaxCodeSize = NACL_SIZE (600, 600*2);
 
 #if defined(__native_client_codegen__)
        const guint kNaClTrampOffset = 17;
 #endif
 
-       if (tramp_type == MONO_TRAMPOLINE_JUMP)
+       if (tramp_type == MONO_TRAMPOLINE_JUMP || tramp_type == MONO_TRAMPOLINE_HANDLER_BLOCK_GUARD)
                has_caller = FALSE;
        else
                has_caller = TRUE;
@@ -425,31 +429,31 @@ mono_arch_create_generic_trampoline (MonoTrampolineType tramp_type, MonoTrampInf
 
        /* Pop the return address off the stack */
        amd64_pop_reg (code, AMD64_R11);
-       orig_rsp_to_rbp_offset += SIZEOF_REGISTER;
+       orig_rsp_to_rbp_offset += sizeof(mgreg_t);
 
-       cfa_offset -= SIZEOF_REGISTER;
+       cfa_offset -= sizeof(mgreg_t);
        mono_add_unwind_op_def_cfa_offset (unwind_ops, code, buf, cfa_offset);
 
        /* 
         * Allocate a new stack frame
         */
        amd64_push_reg (code, AMD64_RBP);
-       cfa_offset += SIZEOF_REGISTER;
+       cfa_offset += sizeof(mgreg_t);
        mono_add_unwind_op_def_cfa_offset (unwind_ops, code, buf, cfa_offset);
        mono_add_unwind_op_offset (unwind_ops, code, buf, AMD64_RBP, - cfa_offset);
 
-       orig_rsp_to_rbp_offset -= SIZEOF_REGISTER;
-       amd64_mov_reg_reg (code, AMD64_RBP, AMD64_RSP, SIZEOF_REGISTER);
+       orig_rsp_to_rbp_offset -= sizeof(mgreg_t);
+       amd64_mov_reg_reg (code, AMD64_RBP, AMD64_RSP, sizeof(mgreg_t));
        mono_add_unwind_op_def_cfa_reg (unwind_ops, code, buf, AMD64_RBP);
        amd64_alu_reg_imm (code, X86_SUB, AMD64_RSP, framesize);
 
        offset = 0;
        rbp_offset = - offset;
 
-       offset += SIZEOF_REGISTER;
+       offset += sizeof(mgreg_t);
        rax_offset = - offset;
 
-       offset += sizeof(gpointer);
+       offset += sizeof(mgreg_t);
        tramp_offset = - offset;
 
        offset += sizeof(gpointer);
@@ -469,36 +473,56 @@ mono_arch_create_generic_trampoline (MonoTrampolineType tramp_type, MonoTrampInf
        }
        amd64_mov_membase_reg (code, AMD64_RBP, tramp_offset, AMD64_R11, sizeof(gpointer));
 
-       offset += SIZEOF_REGISTER;
+       offset += sizeof(mgreg_t);
        res_offset = - offset;
 
        /* Save all registers */
 
-       offset += AMD64_NREG * SIZEOF_REGISTER;
+       offset += AMD64_NREG * sizeof(mgreg_t);
        saved_regs_offset = - offset;
        for (i = 0; i < AMD64_NREG; ++i) {
                if (i == AMD64_RBP) {
                        /* RAX is already saved */
-                       amd64_mov_reg_membase (code, AMD64_RAX, AMD64_RBP, rbp_offset, SIZEOF_REGISTER);
-                       amd64_mov_membase_reg (code, AMD64_RBP, saved_regs_offset + (i * 8), AMD64_RAX, SIZEOF_REGISTER);
+                       amd64_mov_reg_membase (code, AMD64_RAX, AMD64_RBP, rbp_offset, sizeof(mgreg_t));
+                       amd64_mov_membase_reg (code, AMD64_RBP, saved_regs_offset + (i * sizeof(mgreg_t)), AMD64_RAX, sizeof(mgreg_t));
                } else if (i != AMD64_R11) {
-                       amd64_mov_membase_reg (code, AMD64_RBP, saved_regs_offset + (i * 8), i, SIZEOF_REGISTER);
+                       amd64_mov_membase_reg (code, AMD64_RBP, saved_regs_offset + (i * sizeof(mgreg_t)), i, sizeof(mgreg_t));
                } else {
                        /* We have to save R11 right at the start of
                           the trampoline code because it's used as a
                           scratch register */
-                       amd64_mov_membase_reg (r11_save_code, AMD64_RSP, saved_regs_offset + orig_rsp_to_rbp_offset + (i * SIZEOF_REGISTER), i, SIZEOF_REGISTER);
+                       amd64_mov_membase_reg (r11_save_code, AMD64_RSP, saved_regs_offset + orig_rsp_to_rbp_offset + (i * sizeof(mgreg_t)), i, sizeof(mgreg_t));
                        g_assert (r11_save_code == after_r11_save_code);
                }
        }
-       offset += 8 * SIZEOF_REGISTER;
+       offset += 8 * sizeof(mgreg_t);
        saved_fpregs_offset = - offset;
        for (i = 0; i < 8; ++i)
-               amd64_movsd_membase_reg (code, AMD64_RBP, saved_fpregs_offset + (i * SIZEOF_REGISTER), i);
+               amd64_movsd_membase_reg (code, AMD64_RBP, saved_fpregs_offset + (i * sizeof(mgreg_t)), i);
+
+       /* Check that the stack is aligned */
+#if defined(__default_codegen__)
+       amd64_mov_reg_reg (code, AMD64_R11, AMD64_RSP, sizeof (mgreg_t));
+       amd64_alu_reg_imm (code, X86_AND, AMD64_R11, 15);
+       amd64_alu_reg_imm (code, X86_CMP, AMD64_R11, 0);
+       br [0] = code;
+       amd64_branch_disp (code, X86_CC_Z, 0, FALSE);
+       if (aot) {
+               amd64_mov_reg_imm (code, AMD64_R11, 0);
+               amd64_mov_reg_membase (code, AMD64_R11, AMD64_R11, 0, 8);
+       } else {
+               amd64_mov_reg_imm (code, AMD64_RDI, tramp_type);
+               amd64_mov_reg_imm (code, AMD64_R11, stack_unaligned);
+               amd64_call_reg (code, AMD64_R11);
+       }
+       mono_amd64_patch (br [0], code);
+       //amd64_breakpoint (code);
+#endif
 
        if (tramp_type != MONO_TRAMPOLINE_GENERIC_CLASS_INIT &&
-                       tramp_type != MONO_TRAMPOLINE_MONITOR_ENTER &&
-                       tramp_type != MONO_TRAMPOLINE_MONITOR_EXIT) {
+               tramp_type != MONO_TRAMPOLINE_MONITOR_ENTER &&
+               tramp_type != MONO_TRAMPOLINE_MONITOR_EXIT &&
+               tramp_type != MONO_TRAMPOLINE_HANDLER_BLOCK_GUARD) {
                /* Obtain the trampoline argument which is encoded in the instruction stream */
                if (aot) {
                        /* Load the GOT offset */
@@ -537,7 +561,7 @@ mono_arch_create_generic_trampoline (MonoTrampolineType tramp_type, MonoTrampInf
                }
                amd64_mov_membase_reg (code, AMD64_RBP, arg_offset, AMD64_R11, sizeof(gpointer));
        } else {
-               amd64_mov_reg_membase (code, AMD64_R11, AMD64_RBP, saved_regs_offset + (MONO_AMD64_ARG_REG1 * SIZEOF_REGISTER), SIZEOF_REGISTER);
+               amd64_mov_reg_membase (code, AMD64_R11, AMD64_RBP, saved_regs_offset + (MONO_AMD64_ARG_REG1 * sizeof(mgreg_t)), sizeof(mgreg_t));
                amd64_mov_membase_reg (code, AMD64_RBP, arg_offset, AMD64_R11, sizeof(gpointer));
        }
 
@@ -551,14 +575,14 @@ mono_arch_create_generic_trampoline (MonoTrampolineType tramp_type, MonoTrampInf
                amd64_mov_reg_membase (code, AMD64_R11, AMD64_RBP, 8, sizeof(gpointer));
        else
                amd64_mov_reg_imm (code, AMD64_R11, 0);
-       amd64_mov_membase_reg (code, AMD64_RBP, lmf_offset + G_STRUCT_OFFSET (MonoLMF, rip), AMD64_R11, SIZEOF_REGISTER);
+       amd64_mov_membase_reg (code, AMD64_RBP, lmf_offset + G_STRUCT_OFFSET (MonoLMF, rip), AMD64_R11, sizeof(mgreg_t));
        /* Save fp */
-       amd64_mov_reg_membase (code, AMD64_R11, AMD64_RSP, framesize, SIZEOF_REGISTER);
-       amd64_mov_membase_reg (code, AMD64_RBP, lmf_offset + G_STRUCT_OFFSET (MonoLMF, rbp), AMD64_R11, SIZEOF_REGISTER);
+       amd64_mov_reg_membase (code, AMD64_R11, AMD64_RSP, framesize, sizeof(mgreg_t));
+       amd64_mov_membase_reg (code, AMD64_RBP, lmf_offset + G_STRUCT_OFFSET (MonoLMF, rbp), AMD64_R11, sizeof(mgreg_t));
        /* Save sp */
-       amd64_mov_reg_reg (code, AMD64_R11, AMD64_RSP, SIZEOF_REGISTER);
+       amd64_mov_reg_reg (code, AMD64_R11, AMD64_RSP, sizeof(mgreg_t));
        amd64_alu_reg_imm (code, X86_ADD, AMD64_R11, framesize + 16);
-       amd64_mov_membase_reg (code, AMD64_RBP, lmf_offset + G_STRUCT_OFFSET (MonoLMF, rsp), AMD64_R11, SIZEOF_REGISTER);
+       amd64_mov_membase_reg (code, AMD64_RBP, lmf_offset + G_STRUCT_OFFSET (MonoLMF, rsp), AMD64_R11, sizeof(mgreg_t));
        /* Save method */
        if (tramp_type == MONO_TRAMPOLINE_JIT || tramp_type == MONO_TRAMPOLINE_JUMP) {
                amd64_mov_reg_membase (code, AMD64_R11, AMD64_RBP, arg_offset, sizeof(gpointer));
@@ -568,14 +592,14 @@ mono_arch_create_generic_trampoline (MonoTrampolineType tramp_type, MonoTrampInf
        }
        /* Save callee saved regs */
 #ifdef TARGET_WIN32
-       amd64_mov_membase_reg (code, AMD64_RBP, lmf_offset + G_STRUCT_OFFSET (MonoLMF, rdi), AMD64_RDI, SIZEOF_REGISTER);
-       amd64_mov_membase_reg (code, AMD64_RBP, lmf_offset + G_STRUCT_OFFSET (MonoLMF, rsi), AMD64_RSI, SIZEOF_REGISTER);
+       amd64_mov_membase_reg (code, AMD64_RBP, lmf_offset + G_STRUCT_OFFSET (MonoLMF, rdi), AMD64_RDI, sizeof(mgreg_t));
+       amd64_mov_membase_reg (code, AMD64_RBP, lmf_offset + G_STRUCT_OFFSET (MonoLMF, rsi), AMD64_RSI, sizeof(mgreg_t));
 #endif
-       amd64_mov_membase_reg (code, AMD64_RBP, lmf_offset + G_STRUCT_OFFSET (MonoLMF, rbx), AMD64_RBX, SIZEOF_REGISTER);
-       amd64_mov_membase_reg (code, AMD64_RBP, lmf_offset + G_STRUCT_OFFSET (MonoLMF, r12), AMD64_R12, SIZEOF_REGISTER);
-       amd64_mov_membase_reg (code, AMD64_RBP, lmf_offset + G_STRUCT_OFFSET (MonoLMF, r13), AMD64_R13, SIZEOF_REGISTER);
-       amd64_mov_membase_reg (code, AMD64_RBP, lmf_offset + G_STRUCT_OFFSET (MonoLMF, r14), AMD64_R14, SIZEOF_REGISTER);
-       amd64_mov_membase_reg (code, AMD64_RBP, lmf_offset + G_STRUCT_OFFSET (MonoLMF, r15), AMD64_R15, SIZEOF_REGISTER);
+       amd64_mov_membase_reg (code, AMD64_RBP, lmf_offset + G_STRUCT_OFFSET (MonoLMF, rbx), AMD64_RBX, sizeof(mgreg_t));
+       amd64_mov_membase_reg (code, AMD64_RBP, lmf_offset + G_STRUCT_OFFSET (MonoLMF, r12), AMD64_R12, sizeof(mgreg_t));
+       amd64_mov_membase_reg (code, AMD64_RBP, lmf_offset + G_STRUCT_OFFSET (MonoLMF, r13), AMD64_R13, sizeof(mgreg_t));
+       amd64_mov_membase_reg (code, AMD64_RBP, lmf_offset + G_STRUCT_OFFSET (MonoLMF, r14), AMD64_R14, sizeof(mgreg_t));
+       amd64_mov_membase_reg (code, AMD64_RBP, lmf_offset + G_STRUCT_OFFSET (MonoLMF, r15), AMD64_R15, sizeof(mgreg_t));
 
        if (aot) {
                code = mono_arch_emit_load_aotconst (buf, code, &ji, MONO_PATCH_INFO_JIT_ICALL_ADDR, "mono_get_lmf_addr");
@@ -626,7 +650,7 @@ mono_arch_create_generic_trampoline (MonoTrampolineType tramp_type, MonoTrampInf
        /* 
         * Have to call the _force_ variant, since there could be a protected wrapper on the top of the stack.
         */
-       amd64_mov_membase_reg (code, AMD64_RBP, res_offset, AMD64_RAX, SIZEOF_REGISTER);
+       amd64_mov_membase_reg (code, AMD64_RBP, res_offset, AMD64_RAX, sizeof(mgreg_t));
        if (aot) {
                code = mono_arch_emit_load_aotconst (buf, code, &ji, MONO_PATCH_INFO_JIT_ICALL_ADDR, "mono_thread_force_interruption_checkpoint");
        } else {
@@ -634,7 +658,7 @@ mono_arch_create_generic_trampoline (MonoTrampolineType tramp_type, MonoTrampInf
        }
        amd64_call_reg (code, AMD64_R11);
 
-       amd64_mov_reg_membase (code, AMD64_RAX, AMD64_RBP, res_offset, SIZEOF_REGISTER);        
+       amd64_mov_reg_membase (code, AMD64_RAX, AMD64_RBP, res_offset, sizeof(mgreg_t));        
 
        /* Restore LMF */
 
@@ -647,34 +671,32 @@ mono_arch_create_generic_trampoline (MonoTrampolineType tramp_type, MonoTrampInf
         * Save rax to the stack, after the leave instruction, this will become part of
         * the red zone.
         */
-       amd64_mov_membase_reg (code, AMD64_RBP, rax_offset, AMD64_RAX, SIZEOF_REGISTER);
+       amd64_mov_membase_reg (code, AMD64_RBP, rax_offset, AMD64_RAX, sizeof(mgreg_t));
 
        /* Restore argument registers, r10 (imt method/rgxtx)
           and rax (needed for direct calls to C vararg functions). */
        for (i = 0; i < AMD64_NREG; ++i)
                if (AMD64_IS_ARGUMENT_REG (i) || i == AMD64_R10 || i == AMD64_RAX)
-                       amd64_mov_reg_membase (code, i, AMD64_RBP, saved_regs_offset + (i * SIZEOF_REGISTER), SIZEOF_REGISTER);
+                       amd64_mov_reg_membase (code, i, AMD64_RBP, saved_regs_offset + (i * sizeof(mgreg_t)), sizeof(mgreg_t));
 
        for (i = 0; i < 8; ++i)
-               amd64_movsd_reg_membase (code, i, AMD64_RBP, saved_fpregs_offset + (i * SIZEOF_REGISTER));
+               amd64_movsd_reg_membase (code, i, AMD64_RBP, saved_fpregs_offset + (i * sizeof(mgreg_t)));
 
        /* Restore stack */
        amd64_leave (code);
 
        if (MONO_TRAMPOLINE_TYPE_MUST_RETURN (tramp_type)) {
                /* Load result */
-               amd64_mov_reg_membase (code, AMD64_RAX, AMD64_RSP, rax_offset - SIZEOF_REGISTER, SIZEOF_REGISTER);
+               amd64_mov_reg_membase (code, AMD64_RAX, AMD64_RSP, rax_offset - sizeof(mgreg_t), sizeof(mgreg_t));
                amd64_ret (code);
        } else {
                /* call the compiled method using the saved rax */
-               amd64_jump_membase (code, AMD64_RSP, rax_offset - SIZEOF_REGISTER);
+               amd64_jump_membase (code, AMD64_RSP, rax_offset - sizeof(mgreg_t));
        }
 
        g_assert ((code - buf) <= kMaxCodeSize);
 
-#if defined(__native_client_codegen__) && defined(__native_client__)
        nacl_global_codeman_validate (&buf, kMaxCodeSize, &code);
-#endif
 
        mono_arch_flush_icache (buf, code - buf);
 
@@ -697,9 +719,7 @@ mono_arch_get_nullified_class_init_trampoline (MonoTrampInfo **info)
        code = buf = mono_global_codeman_reserve (16);
        amd64_ret (code);
 
-#if defined(__native_client_codegen__) && defined(__native_client__)
        nacl_global_codeman_validate(&buf, 16, &code);
-#endif
 
        mono_arch_flush_icache (buf, code - buf);
 
@@ -717,6 +737,7 @@ mono_arch_create_specific_trampoline (gpointer arg1, MonoTrampolineType tramp_ty
 {
        guint8 *code, *buf, *tramp;
        int size;
+       gboolean far_addr = FALSE;
 
        tramp = mono_get_trampoline_code (tramp_type);
 
@@ -727,6 +748,15 @@ mono_arch_create_specific_trampoline (gpointer arg1, MonoTrampolineType tramp_ty
                size = 5 + 1 + 8;
 
        code = buf = mono_domain_code_reserve_align (domain, size, 1);
+
+       if (((gint64)tramp - (gint64)code) >> 31 != 0 && ((gint64)tramp - (gint64)code) >> 31 != -1) {
+#ifndef MONO_ARCH_NOMAP32BIT
+               g_assert_not_reached ();
+#endif
+               far_addr = TRUE;
+               size += 16;
+               code = buf = mono_domain_code_reserve_align (domain, size, 1);
+       }
 #elif defined(__native_client_codegen__)
        size = 5 + 1 + 4;
        /* Aligning the call site below could */
@@ -736,7 +766,12 @@ mono_arch_create_specific_trampoline (gpointer arg1, MonoTrampolineType tramp_ty
        code = buf;
 #endif
 
-       amd64_call_code (code, tramp);
+       if (far_addr) {
+               amd64_mov_reg_imm (code, AMD64_R11, tramp);
+               amd64_call_reg (code, AMD64_R11);
+       } else {
+               amd64_call_code (code, tramp);
+       }
        /* The trampoline code will obtain the argument from the instruction stream */
 #if defined(__default_codegen__)
        if ((((guint64)arg1) >> 32) == 0) {
@@ -760,9 +795,7 @@ mono_arch_create_specific_trampoline (gpointer arg1, MonoTrampolineType tramp_ty
        if (code_len)
                *code_len = size;
 
-#if defined(__native_client_codegen__) && defined(__native_client__)
        nacl_domain_code_validate(domain, &buf, size, &code);
-#endif
 
        mono_arch_flush_icache (buf, size);
 
@@ -856,9 +889,7 @@ mono_arch_create_rgctx_lazy_fetch_trampoline (guint32 slot, MonoTrampInfo **info
                amd64_jump_code (code, tramp);
        }
 
-#if defined(__native_client_codegen__) && defined(__native_client__)
        nacl_global_codeman_validate (&buf, tramp_size, &code);
-#endif
        mono_arch_flush_icache (buf, code - buf);
 
        g_assert (code - buf <= tramp_size);
@@ -906,9 +937,7 @@ mono_arch_create_generic_class_init_trampoline (MonoTrampInfo **info, gboolean a
                amd64_jump_code (code, tramp);
        }
 
-#if defined(__native_client_codegen__) && defined(__native_client__)
        nacl_global_codeman_validate (&buf, tramp_size, &code);
-#endif
 
        mono_arch_flush_icache (buf, code - buf);
 
@@ -1033,9 +1062,7 @@ mono_arch_create_monitor_enter_trampoline (MonoTrampInfo **info, gboolean aot)
                amd64_jump_code (code, tramp);
        }
 
-#if defined(__native_client_codegen__) && defined(__native_client__)
        nacl_global_codeman_validate (&buf, tramp_size, &code);
-#endif
 
        mono_arch_flush_icache (code, code - buf);
        g_assert (code - buf <= tramp_size);
@@ -1153,9 +1180,7 @@ mono_arch_create_monitor_exit_trampoline (MonoTrampInfo **info, gboolean aot)
                amd64_jump_code (code, tramp);
        }
 
-#if defined(__native_client_codegen__) && defined(__native_client__)
        nacl_global_codeman_validate (&buf, tramp_size, &code);
-#endif
 
        mono_arch_flush_icache (code, code - buf);
        g_assert (code - buf <= tramp_size);
@@ -1184,7 +1209,7 @@ mono_arch_invalidate_method (MonoJitInfo *ji, void *func, gpointer func_arg)
 static void
 handler_block_trampoline_helper (gpointer *ptr)
 {
-       MonoJitTlsData *jit_tls = TlsGetValue (mono_jit_tls_id);
+       MonoJitTlsData *jit_tls = mono_native_tls_get_value (mono_jit_tls_id);
        *ptr = jit_tls->handler_block_return_address;
 }
 
@@ -1202,8 +1227,9 @@ mono_arch_create_handler_block_trampoline (void)
 
        if (mono_get_jit_tls_offset () != -1) {
                code = mono_amd64_emit_tls_get (code, AMD64_RDI, mono_get_jit_tls_offset ());
-               /*simulate a call*/
                amd64_mov_reg_membase (code, AMD64_RDI, AMD64_RDI, G_STRUCT_OFFSET (MonoJitTlsData, handler_block_return_address), 8);
+               /* Simulate a call */
+               amd64_push_reg (code, AMD64_RAX);
                amd64_jump_code (code, tramp);
        } else {
                /*Slow path uses a c helper*/