mono/mini/tramp-amd64.c

   1 /*
   2  * tramp-amd64.c: JIT trampoline code for amd64
   3  *
   4  * Authors:
   5  *   Dietmar Maurer (dietmar@ximian.com)
   6  *   Zoltan Varga (vargaz@gmail.com)
   7  *
   8  * (C) 2001 Ximian, Inc.
   9  */
  10
  11 #include <config.h>
  12 #include <glib.h>
  13
  14 #include <mono/metadata/appdomain.h>
  15 #include <mono/metadata/marshal.h>
  16 #include <mono/metadata/tabledefs.h>
  17 #include <mono/metadata/mono-debug-debugger.h>
  18 #include <mono/metadata/monitor.h>
  19 #include <mono/arch/amd64/amd64-codegen.h>
  20
  21 #ifdef HAVE_VALGRIND_MEMCHECK_H
  22 #include <valgrind/memcheck.h>
  23 #endif
  24
  25 #include "mini.h"
  26 #include "mini-amd64.h"
  27
  28 #define IS_REX(inst) (((inst) >= 0x40) && ((inst) <= 0x4f))
  29
  30 static guint8* nullified_class_init_trampoline;
  31
  32 /*
  33  * mono_arch_get_unbox_trampoline:
  34  * @gsctx: the generic sharing context
  35  * @m: method pointer
  36  * @addr: pointer to native code for @m
  37  *
  38  * when value type methods are called through the vtable we need to unbox the
  39  * this argument. This method returns a pointer to a trampoline which does
  40  * unboxing before calling the method
  41  */
  42 gpointer
  43 mono_arch_get_unbox_trampoline (MonoGenericSharingContext *gsctx, MonoMethod *m, gpointer addr)
  44 {
  45         guint8 *code, *start;
  46         int this_reg;
  47
  48         MonoDomain *domain = mono_domain_get ();
  49
  50         this_reg = mono_arch_get_this_arg_reg (mono_method_signature (m), gsctx, NULL);
  51
  52         mono_domain_lock (domain);
  53         start = code = mono_code_manager_reserve (domain->code_mp, 20);
  54         mono_domain_unlock (domain);
  55
  56         amd64_alu_reg_imm (code, X86_ADD, this_reg, sizeof (MonoObject));
  57         /* FIXME: Optimize this */
  58         amd64_mov_reg_imm (code, AMD64_RAX, addr);
  59         amd64_jump_reg (code, AMD64_RAX);
  60         g_assert ((code - start) < 20);
  61
  62         mono_arch_flush_icache (start, code - start);
  63
  64         return start;
  65 }
  66
  67 /*
  68  * mono_arch_patch_callsite:
  69  *
  70  *   Patch the callsite whose address is given by ORIG_CODE so it calls ADDR. ORIG_CODE
  71  * points to the pc right after the call.
  72  */
  73 void
  74 mono_arch_patch_callsite (guint8 *method_start, guint8 *orig_code, guint8 *addr)
  75 {
  76         guint8 *code;
  77         guint8 buf [16];
  78         gboolean can_write = mono_breakpoint_clean_code (method_start, orig_code, 14, buf, sizeof (buf));
  79
  80         code = buf + 14;
  81
  82         if (((code [-13] == 0x49) && (code [-12] == 0xbb)) || (code [-5] == 0xe8)) {
  83                 if (code [-5] != 0xe8) {
  84                         if (can_write) {
  85                                 InterlockedExchangePointer ((gpointer*)(orig_code - 11), addr);
  86 #ifdef HAVE_VALGRIND_MEMCHECK_H
  87                                 VALGRIND_DISCARD_TRANSLATIONS (orig_code - 11, sizeof (gpointer));
  88 #endif
  89                         }
  90                 } else {
  91                         if ((((guint64)(addr)) >> 32) != 0) {
  92                                 /* Print some diagnostics */
  93                                 MonoJitInfo *ji = mono_jit_info_table_find (mono_domain_get (), (char*)orig_code);
  94                                 if (ji)
  95                                         fprintf (stderr, "At %s, offset 0x%zx\n", mono_method_full_name (ji->method, TRUE), (guint8*)orig_code - (guint8*)ji->code_start);
  96                                 fprintf (stderr, "Addr: %p\n", addr);
  97                                 ji = mono_jit_info_table_find (mono_domain_get (), (char*)addr);
  98                                 if (ji)
  99                                         fprintf (stderr, "Callee: %s\n", mono_method_full_name (ji->method, TRUE));
 100                                 g_assert_not_reached ();
 101                         }
 102                         g_assert ((((guint64)(orig_code)) >> 32) == 0);
 103                         if (can_write) {
 104                                 InterlockedExchange ((gint32*)(orig_code - 4), ((gint64)addr - (gint64)orig_code));
 105 #ifdef HAVE_VALGRIND_MEMCHECK_H
 106                                 VALGRIND_DISCARD_TRANSLATIONS (orig_code - 5, 4);
 107 #endif
 108                         }
 109                 }
 110         }
 111         else if ((code [-7] == 0x41) && (code [-6] == 0xff) && (code [-5] == 0x15)) {
 112                 /* call *<OFFSET>(%rip) */
 113                 gpointer *got_entry = (gpointer*)((guint8*)orig_code + (*(guint32*)(orig_code - 4)));
 114                 if (can_write) {
 115                         InterlockedExchangePointer (got_entry, addr);
 116 #ifdef HAVE_VALGRIND_MEMCHECK_H
 117                         VALGRIND_DISCARD_TRANSLATIONS (orig_code - 5, sizeof (gpointer));
 118 #endif
 119                 }
 120         }
 121 }
 122
 123 void
 124 mono_arch_patch_plt_entry (guint8 *code, guint8 *addr)
 125 {
 126         gint32 disp;
 127         gpointer *plt_jump_table_entry;
 128
 129         /* A PLT entry: jmp *<DISP>(%rip) */
 130         g_assert (code [0] == 0xff);
 131         g_assert (code [1] == 0x25);
 132
 133         disp = *(gint32*)(code + 2);
 134
 135         plt_jump_table_entry = (gpointer*)(code + 6 + disp);
 136
 137         InterlockedExchangePointer (plt_jump_table_entry, addr);
 138 }
 139
 140 void
 141 mono_arch_nullify_class_init_trampoline (guint8 *code, gssize *regs)
 142 {
 143         guint8 buf [16];
 144         gboolean can_write = mono_breakpoint_clean_code (NULL, code, 7, buf, sizeof (buf));
 145
 146         if (!can_write)
 147                 return;
 148
 149         code -= 3;
 150
 151         /*
 152          * A given byte sequence can match more than case here, so we have to be
 153          * really careful about the ordering of the cases. Longer sequences
 154          * come first.
 155          */
 156         if ((code [-4] == 0x41) && (code [-3] == 0xff) && (code [-2] == 0x15)) {
 157                 gpointer *vtable_slot;
 158
 159                 /* call *<OFFSET>(%rip) */
 160                 vtable_slot = mono_arch_get_vcall_slot_addr (code + 3, (gpointer*)regs);
 161                 g_assert (vtable_slot);
 162
 163                 *vtable_slot = nullified_class_init_trampoline;
 164         } else if (code [-2] == 0xe8) {
 165                 /* call <TARGET> */
 166                 guint8 *buf = code - 2;
 167
 168                 buf [0] = 0x66;
 169                 buf [1] = 0x66;
 170                 buf [2] = 0x90;
 171                 buf [3] = 0x66;
 172                 buf [4] = 0x90;
 173         } else if ((code [0] == 0x41) && (code [1] == 0xff)) {
 174                 /* call <REG> */
 175                 /* happens on machines without MAP_32BIT like freebsd */
 176                 /* amd64_set_reg_template is 10 bytes long */
 177                 guint8* buf = code - 10;
 178
 179                 /* FIXME: Make this thread safe */
 180                 /* Padding code suggested by the AMD64 Opt Manual */
 181                 buf [0] = 0x66;
 182                 buf [1] = 0x66;
 183                 buf [2] = 0x66;
 184                 buf [3] = 0x90;
 185                 buf [4] = 0x66;
 186                 buf [5] = 0x66;
 187                 buf [6] = 0x66;
 188                 buf [7] = 0x90;
 189                 buf [8] = 0x66;
 190                 buf [9] = 0x66;
 191                 buf [10] = 0x90;
 192                 buf [11] = 0x66;
 193                 buf [12] = 0x90;
 194         } else if (code [0] == 0x90 || code [0] == 0xeb || code [0] == 0x66) {
 195                 /* Already changed by another thread */
 196                 ;
 197         } else {
 198                 printf ("Invalid trampoline sequence: %x %x %x %x %x %x %x\n", code [0], code [1], code [2], code [3],
 199                         code [4], code [5], code [6]);
 200                 g_assert_not_reached ();
 201         }
 202 }
 203
 204 void
 205 mono_arch_nullify_plt_entry (guint8 *code)
 206 {
 207         if (mono_aot_only && !nullified_class_init_trampoline)
 208                 nullified_class_init_trampoline = mono_aot_get_named_code ("nullified_class_init_trampoline");
 209
 210         mono_arch_patch_plt_entry (code, nullified_class_init_trampoline);
 211 }
 212
 213 guchar*
 214 mono_arch_create_trampoline_code (MonoTrampolineType tramp_type)
 215 {
 216         MonoJumpInfo *ji;
 217         guint32 code_size;
 218
 219         return mono_arch_create_trampoline_code_full (tramp_type, &code_size, &ji, FALSE);
 220 }
 221
 222 guchar*
 223 mono_arch_create_trampoline_code_full (MonoTrampolineType tramp_type, guint32 *code_size, MonoJumpInfo **ji, gboolean aot)
 224 {
 225         guint8 *buf, *code, *tramp, *br [2], *r11_save_code, *after_r11_save_code;
 226         int i, lmf_offset, offset, res_offset, arg_offset, rax_offset, tramp_offset, saved_regs_offset;
 227         int saved_fpregs_offset, rbp_offset, framesize, orig_rsp_to_rbp_offset;
 228         gboolean has_caller;
 229
 230         if (tramp_type == MONO_TRAMPOLINE_JUMP)
 231                 has_caller = FALSE;
 232         else
 233                 has_caller = TRUE;
 234
 235         code = buf = mono_global_codeman_reserve (524);
 236
 237         *ji = NULL;
 238
 239         framesize = 524 + sizeof (MonoLMF);
 240         framesize = (framesize + (MONO_ARCH_FRAME_ALIGNMENT - 1)) & ~ (MONO_ARCH_FRAME_ALIGNMENT - 1);
 241
 242         orig_rsp_to_rbp_offset = 0;
 243         r11_save_code = code;
 244         /* Reserve 5 bytes for the mov_membase_reg to save R11 */
 245         code += 5;
 246         after_r11_save_code = code;
 247
 248         /* Pop the return address off the stack */
 249         amd64_pop_reg (code, AMD64_R11);
 250         orig_rsp_to_rbp_offset += 8;
 251
 252         /*
 253          * Allocate a new stack frame
 254          */
 255         amd64_push_reg (code, AMD64_RBP);
 256         orig_rsp_to_rbp_offset -= 8;
 257         amd64_mov_reg_reg (code, AMD64_RBP, AMD64_RSP, 8);
 258         amd64_alu_reg_imm (code, X86_SUB, AMD64_RSP, framesize);
 259
 260         offset = 0;
 261         rbp_offset = - offset;
 262
 263         offset += 8;
 264         rax_offset = - offset;
 265
 266         offset += 8;
 267         tramp_offset = - offset;
 268
 269         offset += 8;
 270         arg_offset = - offset;
 271
 272         /* Compute the trampoline address from the return address */
 273         if (aot) {
 274                 /* 7 = length of call *<offset>(rip) */
 275                 amd64_alu_reg_imm (code, X86_SUB, AMD64_R11, 7);
 276         } else {
 277                 /* 5 = length of amd64_call_membase () */
 278                 amd64_alu_reg_imm (code, X86_SUB, AMD64_R11, 5);
 279         }
 280         amd64_mov_membase_reg (code, AMD64_RBP, tramp_offset, AMD64_R11, 8);
 281
 282         offset += 8;
 283         res_offset = - offset;
 284
 285         /* Save all registers */
 286
 287         offset += AMD64_NREG * 8;
 288         saved_regs_offset = - offset;
 289         for (i = 0; i < AMD64_NREG; ++i) {
 290                 if (i == AMD64_RBP) {
 291                         /* RAX is already saved */
 292                         amd64_mov_reg_membase (code, AMD64_RAX, AMD64_RBP, rbp_offset, 8);
 293                         amd64_mov_membase_reg (code, AMD64_RBP, saved_regs_offset + (i * 8), AMD64_RAX, 8);
 294                 } else if (i != AMD64_R11) {
 295                         amd64_mov_membase_reg (code, AMD64_RBP, saved_regs_offset + (i * 8), i, 8);
 296                 } else {
 297                         /* We have to save R11 right at the start of
 298                            the trampoline code because it's used as a
 299                            scratch register */
 300                         amd64_mov_membase_reg (r11_save_code, AMD64_RSP, saved_regs_offset + orig_rsp_to_rbp_offset + (i * 8), i, 8);
 301                         g_assert (r11_save_code == after_r11_save_code);
 302                 }
 303         }
 304         offset += 8 * 8;
 305         saved_fpregs_offset = - offset;
 306         for (i = 0; i < 8; ++i)
 307                 amd64_movsd_membase_reg (code, AMD64_RBP, saved_fpregs_offset + (i * 8), i);
 308
 309         if (tramp_type != MONO_TRAMPOLINE_GENERIC_CLASS_INIT &&
 310                         tramp_type != MONO_TRAMPOLINE_MONITOR_ENTER &&
 311                         tramp_type != MONO_TRAMPOLINE_MONITOR_EXIT) {
 312                 /* Obtain the trampoline argument which is encoded in the instruction stream */
 313                 if (aot) {
 314                         /* Load the GOT offset */
 315                         amd64_mov_reg_membase (code, AMD64_R11, AMD64_RBP, tramp_offset, 8);
 316                         amd64_mov_reg_membase (code, AMD64_RAX, AMD64_R11, 7, 4);
 317                         /* Compute the address of the GOT slot */
 318                         amd64_alu_reg_reg_size (code, X86_ADD, AMD64_R11, AMD64_RAX, 8);
 319                         /* Load the value */
 320                         amd64_mov_reg_membase (code, AMD64_R11, AMD64_R11, 0, 8);
 321                 } else {
 322                         amd64_mov_reg_membase (code, AMD64_R11, AMD64_RBP, tramp_offset, 8);
 323                         amd64_mov_reg_membase (code, AMD64_RAX, AMD64_R11, 5, 1);
 324                         amd64_widen_reg (code, AMD64_RAX, AMD64_RAX, TRUE, FALSE);
 325                         amd64_alu_reg_imm_size (code, X86_CMP, AMD64_RAX, 4, 1);
 326                         br [0] = code;
 327                         x86_branch8 (code, X86_CC_NE, 6, FALSE);
 328                         /* 32 bit immediate */
 329                         amd64_mov_reg_membase (code, AMD64_R11, AMD64_R11, 6, 4);
 330                         br [1] = code;
 331                         x86_jump8 (code, 10);
 332                         /* 64 bit immediate */
 333                         mono_amd64_patch (br [0], code);
 334                         amd64_mov_reg_membase (code, AMD64_R11, AMD64_R11, 6, 8);
 335                         mono_amd64_patch (br [1], code);
 336                 }
 337                 amd64_mov_membase_reg (code, AMD64_RBP, arg_offset, AMD64_R11, 8);
 338         } else {
 339                 amd64_mov_reg_membase (code, AMD64_R11, AMD64_RBP, saved_regs_offset + (MONO_AMD64_ARG_REG1 * 8), 8);
 340                 amd64_mov_membase_reg (code, AMD64_RBP, arg_offset, AMD64_R11, 8);
 341         }
 342
 343         /* Save LMF begin */
 344
 345         offset += sizeof (MonoLMF);
 346         lmf_offset = - offset;
 347
 348         /* Save ip */
 349         if (has_caller)
 350                 amd64_mov_reg_membase (code, AMD64_R11, AMD64_RBP, 8, 8);
 351         else
 352                 amd64_mov_reg_imm (code, AMD64_R11, 0);
 353         amd64_mov_membase_reg (code, AMD64_RBP, lmf_offset + G_STRUCT_OFFSET (MonoLMF, rip), AMD64_R11, 8);
 354         /* Save fp */
 355         amd64_mov_reg_membase (code, AMD64_R11, AMD64_RSP, framesize, 8);
 356         amd64_mov_membase_reg (code, AMD64_RBP, lmf_offset + G_STRUCT_OFFSET (MonoLMF, rbp), AMD64_R11, 8);
 357         /* Save sp */
 358         amd64_mov_reg_reg (code, AMD64_R11, AMD64_RSP, 8);
 359         amd64_alu_reg_imm (code, X86_ADD, AMD64_R11, framesize + 16);
 360         amd64_mov_membase_reg (code, AMD64_RBP, lmf_offset + G_STRUCT_OFFSET (MonoLMF, rsp), AMD64_R11, 8);
 361         /* Save method */
 362         if (tramp_type == MONO_TRAMPOLINE_JIT || tramp_type == MONO_TRAMPOLINE_JUMP) {
 363                 amd64_mov_reg_membase (code, AMD64_R11, AMD64_RBP, arg_offset, 8);
 364                 amd64_mov_membase_reg (code, AMD64_RBP, lmf_offset + G_STRUCT_OFFSET (MonoLMF, method), AMD64_R11, 8);
 365         } else {
 366                 amd64_mov_membase_imm (code, AMD64_RBP, lmf_offset + G_STRUCT_OFFSET (MonoLMF, method), 0, 8);
 367         }
 368         /* Save callee saved regs */
 369 #ifdef PLATFORM_WIN32
 370         amd64_mov_membase_reg (code, AMD64_RBP, lmf_offset + G_STRUCT_OFFSET (MonoLMF, rdi), AMD64_RDI, 8);
 371         amd64_mov_membase_reg (code, AMD64_RBP, lmf_offset + G_STRUCT_OFFSET (MonoLMF, rsi), AMD64_RSI, 8);
 372 #endif
 373         amd64_mov_membase_reg (code, AMD64_RBP, lmf_offset + G_STRUCT_OFFSET (MonoLMF, rbx), AMD64_RBX, 8);
 374         amd64_mov_membase_reg (code, AMD64_RBP, lmf_offset + G_STRUCT_OFFSET (MonoLMF, r12), AMD64_R12, 8);
 375         amd64_mov_membase_reg (code, AMD64_RBP, lmf_offset + G_STRUCT_OFFSET (MonoLMF, r13), AMD64_R13, 8);
 376         amd64_mov_membase_reg (code, AMD64_RBP, lmf_offset + G_STRUCT_OFFSET (MonoLMF, r14), AMD64_R14, 8);
 377         amd64_mov_membase_reg (code, AMD64_RBP, lmf_offset + G_STRUCT_OFFSET (MonoLMF, r15), AMD64_R15, 8);
 378
 379         if (aot) {
 380                 *ji = mono_patch_info_list_prepend (*ji, code - buf, MONO_PATCH_INFO_JIT_ICALL_ADDR, "mono_get_lmf_addr");
 381                 amd64_mov_reg_membase (code, AMD64_R11, AMD64_RIP, 0, 8);
 382         } else {
 383                 amd64_mov_reg_imm (code, AMD64_R11, mono_get_lmf_addr);
 384         }
 385         amd64_call_reg (code, AMD64_R11);
 386
 387         /* Save lmf_addr */
 388         amd64_mov_membase_reg (code, AMD64_RBP, lmf_offset + G_STRUCT_OFFSET (MonoLMF, lmf_addr), AMD64_RAX, 8);
 389         /* Save previous_lmf */
 390         /* Set the lowest bit to 1 to signal that this LMF has the ip field set */
 391         amd64_mov_reg_membase (code, AMD64_R11, AMD64_RAX, 0, 8);
 392         amd64_alu_reg_imm_size (code, X86_ADD, AMD64_R11, 1, 8);
 393         amd64_mov_membase_reg (code, AMD64_RBP, lmf_offset + G_STRUCT_OFFSET (MonoLMF, previous_lmf), AMD64_R11, 8);
 394         /* Set new lmf */
 395         amd64_lea_membase (code, AMD64_R11, AMD64_RBP, lmf_offset);
 396         amd64_mov_membase_reg (code, AMD64_RAX, 0, AMD64_R11, 8);
 397
 398         /* Save LMF end */
 399
 400         /* Arg1 is the pointer to the saved registers */
 401         amd64_lea_membase (code, AMD64_ARG_REG1, AMD64_RBP, saved_regs_offset);
 402
 403         /* Arg2 is the address of the calling code */
 404         if (has_caller)
 405                 amd64_mov_reg_membase (code, AMD64_ARG_REG2, AMD64_RBP, 8, 8);
 406         else
 407                 amd64_mov_reg_imm (code, AMD64_ARG_REG2, 0);
 408
 409         /* Arg3 is the method/vtable ptr */
 410         amd64_mov_reg_membase (code, AMD64_ARG_REG3, AMD64_RBP, arg_offset, 8);
 411
 412         /* Arg4 is the trampoline address */
 413         amd64_mov_reg_membase (code, AMD64_ARG_REG4, AMD64_RBP, tramp_offset, 8);
 414
 415         if (aot) {
 416                 char *icall_name = g_strdup_printf ("trampoline_func_%d", tramp_type);
 417                 *ji = mono_patch_info_list_prepend (*ji, code - buf, MONO_PATCH_INFO_JIT_ICALL_ADDR, icall_name);
 418                 amd64_mov_reg_membase (code, AMD64_RAX, AMD64_RIP, 0, 8);
 419         } else {
 420                 tramp = (guint8*)mono_get_trampoline_func (tramp_type);
 421                 amd64_mov_reg_imm (code, AMD64_RAX, tramp);
 422         }
 423         amd64_call_reg (code, AMD64_RAX);
 424
 425         /* Check for thread interruption */
 426         /* This is not perf critical code so no need to check the interrupt flag */
 427         /*
 428          * Have to call the _force_ variant, since there could be a protected wrapper on the top of the stack.
 429          */
 430         amd64_mov_membase_reg (code, AMD64_RBP, res_offset, AMD64_RAX, 8);
 431         if (aot) {
 432                 *ji = mono_patch_info_list_prepend (*ji, code - buf, MONO_PATCH_INFO_JIT_ICALL_ADDR, "mono_thread_force_interruption_checkpoint");
 433                 amd64_mov_reg_membase (code, AMD64_RAX, AMD64_RIP, 0, 8);
 434         } else {
 435                 amd64_mov_reg_imm (code, AMD64_RAX, (guint8*)mono_thread_force_interruption_checkpoint);
 436         }
 437         amd64_call_reg (code, AMD64_RAX);
 438         amd64_mov_reg_membase (code, AMD64_RAX, AMD64_RBP, res_offset, 8);
 439
 440         /* Restore LMF */
 441
 442         amd64_mov_reg_membase (code, AMD64_RCX, AMD64_RBP, lmf_offset + G_STRUCT_OFFSET (MonoLMF, previous_lmf), 8);
 443         amd64_alu_reg_imm_size (code, X86_SUB, AMD64_RCX, 1, 8);
 444         amd64_mov_reg_membase (code, AMD64_R11, AMD64_RBP, lmf_offset + G_STRUCT_OFFSET (MonoLMF, lmf_addr), 8);
 445         amd64_mov_membase_reg (code, AMD64_R11, 0, AMD64_RCX, 8);
 446
 447         /*
 448          * Save rax to the stack, after the leave instruction, this will become part of
 449          * the red zone.
 450          */
 451         amd64_mov_membase_reg (code, AMD64_RBP, rax_offset, AMD64_RAX, 8);
 452
 453         /* Restore argument registers, r10 (needed to pass rgctx to
 454            static shared generic methods), r11 (imt register for
 455            interface calls), and rax (needed for direct calls to C vararg functions). */
 456         for (i = 0; i < AMD64_NREG; ++i)
 457                 if (AMD64_IS_ARGUMENT_REG (i) || i == AMD64_R10 || i == AMD64_R11 || i == AMD64_RAX)
 458                         amd64_mov_reg_membase (code, i, AMD64_RBP, saved_regs_offset + (i * 8), 8);
 459
 460         for (i = 0; i < 8; ++i)
 461                 amd64_movsd_reg_membase (code, i, AMD64_RBP, saved_fpregs_offset + (i * 8));
 462
 463         /* Restore stack */
 464         amd64_leave (code);
 465
 466         if (MONO_TRAMPOLINE_TYPE_MUST_RETURN (tramp_type)) {
 467                 /* Load result */
 468                 amd64_mov_reg_membase (code, AMD64_RAX, AMD64_RSP, rax_offset - 0x8, 8);
 469                 amd64_ret (code);
 470         } else {
 471                 /* call the compiled method using the saved rax */
 472                 amd64_jump_membase (code, AMD64_RSP, rax_offset - 0x8);
 473         }
 474
 475         g_assert ((code - buf) <= 524);
 476
 477         mono_arch_flush_icache (buf, code - buf);
 478
 479         *code_size = code - buf;
 480
 481         if (tramp_type == MONO_TRAMPOLINE_CLASS_INIT) {
 482                 guint32 code_len;
 483
 484                 /* Initialize the nullified class init trampoline used in the AOT case */
 485                 nullified_class_init_trampoline = mono_arch_get_nullified_class_init_trampoline (&code_len);
 486         }
 487
 488         return buf;
 489 }
 490
 491 gpointer
 492 mono_arch_get_nullified_class_init_trampoline (guint32 *code_len)
 493 {
 494         guint8 *code, *buf;
 495
 496         code = buf = mono_global_codeman_reserve (16);
 497         amd64_ret (code);
 498
 499         mono_arch_flush_icache (buf, code - buf);
 500
 501         *code_len = code - buf;
 502
 503         return buf;
 504 }
 505
 506 gpointer
 507 mono_arch_create_specific_trampoline (gpointer arg1, MonoTrampolineType tramp_type, MonoDomain *domain, guint32 *code_len)
 508 {
 509         guint8 *code, *buf, *tramp;
 510         int size;
 511
 512         tramp = mono_get_trampoline_code (tramp_type);
 513
 514         if ((((guint64)arg1) >> 32) == 0)
 515                 size = 5 + 1 + 4;
 516         else
 517                 size = 5 + 1 + 8;
 518
 519         mono_domain_lock (domain);
 520         code = buf = mono_code_manager_reserve_align (domain->code_mp, size, 1);
 521         mono_domain_unlock (domain);
 522
 523         amd64_call_code (code, tramp);
 524         /* The trampoline code will obtain the argument from the instruction stream */
 525         if ((((guint64)arg1) >> 32) == 0) {
 526                 *code = 0x4;
 527                 *(guint32*)(code + 1) = (gint64)arg1;
 528                 code += 5;
 529         } else {
 530                 *code = 0x8;
 531                 *(guint64*)(code + 1) = (gint64)arg1;
 532                 code += 9;
 533         }
 534
 535         g_assert ((code - buf) <= size);
 536
 537         if (code_len)
 538                 *code_len = size;
 539
 540         mono_arch_flush_icache (buf, size);
 541
 542         return buf;
 543 }
 544
 545 gpointer
 546 mono_arch_create_rgctx_lazy_fetch_trampoline (guint32 slot)
 547 {
 548         guint32 code_size;
 549         MonoJumpInfo *ji;
 550
 551         return mono_arch_create_rgctx_lazy_fetch_trampoline_full (slot, &code_size, &ji, FALSE);
 552 }
 553
 554 gpointer
 555 mono_arch_create_rgctx_lazy_fetch_trampoline_full (guint32 slot, guint32 *code_size, MonoJumpInfo **ji, gboolean aot)
 556 {
 557         guint8 *tramp;
 558         guint8 *code, *buf;
 559         guint8 **rgctx_null_jumps;
 560         int tramp_size;
 561         int depth, index;
 562         int i;
 563         gboolean mrgctx;
 564
 565         *ji = NULL;
 566
 567         mrgctx = MONO_RGCTX_SLOT_IS_MRGCTX (slot);
 568         index = MONO_RGCTX_SLOT_INDEX (slot);
 569         if (mrgctx)
 570                 index += sizeof (MonoMethodRuntimeGenericContext) / sizeof (gpointer);
 571         for (depth = 0; ; ++depth) {
 572                 int size = mono_class_rgctx_get_array_size (depth, mrgctx);
 573
 574                 if (index < size - 1)
 575                         break;
 576                 index -= size - 1;
 577         }
 578
 579         tramp_size = 64 + 8 * depth;
 580
 581         code = buf = mono_global_codeman_reserve (tramp_size);
 582
 583         rgctx_null_jumps = g_malloc (sizeof (guint8*) * (depth + 2));
 584
 585         if (mrgctx) {
 586                 /* get mrgctx ptr */
 587                 amd64_mov_reg_reg (code, AMD64_RAX, AMD64_ARG_REG1, 8);
 588         } else {
 589                 /* load rgctx ptr from vtable */
 590                 amd64_mov_reg_membase (code, AMD64_RAX, AMD64_ARG_REG1, G_STRUCT_OFFSET (MonoVTable, runtime_generic_context), 8);
 591                 /* is the rgctx ptr null? */
 592                 amd64_test_reg_reg (code, AMD64_RAX, AMD64_RAX);
 593                 /* if yes, jump to actual trampoline */
 594                 rgctx_null_jumps [0] = code;
 595                 amd64_branch8 (code, X86_CC_Z, -1, 1);
 596         }
 597
 598         for (i = 0; i < depth; ++i) {
 599                 /* load ptr to next array */
 600                 if (mrgctx && i == 0)
 601                         amd64_mov_reg_membase (code, AMD64_RAX, AMD64_RAX, sizeof (MonoMethodRuntimeGenericContext), 8);
 602                 else
 603                         amd64_mov_reg_membase (code, AMD64_RAX, AMD64_RAX, 0, 8);
 604                 /* is the ptr null? */
 605                 amd64_test_reg_reg (code, AMD64_RAX, AMD64_RAX);
 606                 /* if yes, jump to actual trampoline */
 607                 rgctx_null_jumps [i + 1] = code;
 608                 amd64_branch8 (code, X86_CC_Z, -1, 1);
 609         }
 610
 611         /* fetch slot */
 612         amd64_mov_reg_membase (code, AMD64_RAX, AMD64_RAX, sizeof (gpointer) * (index + 1), 8);
 613         /* is the slot null? */
 614         amd64_test_reg_reg (code, AMD64_RAX, AMD64_RAX);
 615         /* if yes, jump to actual trampoline */
 616         rgctx_null_jumps [depth + 1] = code;
 617         amd64_branch8 (code, X86_CC_Z, -1, 1);
 618         /* otherwise return */
 619         amd64_ret (code);
 620
 621         for (i = mrgctx ? 1 : 0; i <= depth + 1; ++i)
 622                 x86_patch (rgctx_null_jumps [i], code);
 623
 624         g_free (rgctx_null_jumps);
 625
 626         /* move the rgctx pointer to the VTABLE register */
 627         amd64_mov_reg_reg (code, MONO_ARCH_VTABLE_REG, AMD64_ARG_REG1, 8);
 628
 629         if (aot) {
 630                 *ji = mono_patch_info_list_prepend (*ji, code - buf, MONO_PATCH_INFO_JIT_ICALL_ADDR, g_strdup_printf ("specific_trampoline_lazy_fetch_%u", slot));
 631                 amd64_mov_reg_membase (code, AMD64_R11, AMD64_RIP, 0, 8);
 632                 amd64_jump_reg (code, AMD64_R11);
 633         } else {
 634                 tramp = mono_arch_create_specific_trampoline (GUINT_TO_POINTER (slot), MONO_TRAMPOLINE_RGCTX_LAZY_FETCH, mono_get_root_domain (), NULL);
 635
 636                 /* jump to the actual trampoline */
 637                 amd64_jump_code (code, tramp);
 638         }
 639
 640         mono_arch_flush_icache (buf, code - buf);
 641
 642         g_assert (code - buf <= tramp_size);
 643
 644         *code_size = code - buf;
 645
 646         return buf;
 647 }
 648
 649 gpointer
 650 mono_arch_create_generic_class_init_trampoline (void)
 651 {
 652         guint8 *tramp;
 653         guint8 *code, *buf;
 654         static int byte_offset = -1;
 655         static guint8 bitmask;
 656         guint8 *jump;
 657         int tramp_size;
 658
 659         tramp_size = 64;
 660
 661         code = buf = mono_global_codeman_reserve (tramp_size);
 662
 663         if (byte_offset < 0)
 664                 mono_marshal_find_bitfield_offset (MonoVTable, initialized, &byte_offset, &bitmask);
 665
 666         amd64_test_membase_imm_size (code, MONO_AMD64_ARG_REG1, byte_offset, bitmask, 1);
 667         jump = code;
 668         amd64_branch8 (code, X86_CC_Z, -1, 1);
 669
 670         amd64_ret (code);
 671
 672         x86_patch (jump, code);
 673
 674         tramp = mono_arch_create_specific_trampoline (NULL, MONO_TRAMPOLINE_GENERIC_CLASS_INIT, mono_get_root_domain (), NULL);
 675
 676         /* jump to the actual trampoline */
 677         amd64_jump_code (code, tramp);
 678
 679         mono_arch_flush_icache (buf, code - buf);
 680
 681         g_assert (code - buf <= tramp_size);
 682
 683         return buf;
 684 }
 685
 686 #ifdef MONO_ARCH_MONITOR_OBJECT_REG
 687
 688 gpointer
 689 mono_arch_create_monitor_enter_trampoline (void)
 690 {
 691         guint32 code_size;
 692         MonoJumpInfo *ji;
 693
 694         return mono_arch_create_monitor_enter_trampoline_full (&code_size, &ji, FALSE);
 695 }
 696
 697 gpointer
 698 mono_arch_create_monitor_enter_trampoline_full (guint32 *code_size, MonoJumpInfo **ji, gboolean aot)
 699 {
 700         guint8 *tramp;
 701         guint8 *code, *buf;
 702         guint8 *jump_obj_null, *jump_sync_null, *jump_cmpxchg_failed, *jump_other_owner, *jump_tid;
 703         int tramp_size;
 704         int owner_offset, nest_offset, dummy;
 705
 706         g_assert (MONO_ARCH_MONITOR_OBJECT_REG == AMD64_RDI);
 707
 708         mono_monitor_threads_sync_members_offset (&owner_offset, &nest_offset, &dummy);
 709         g_assert (MONO_THREADS_SYNC_MEMBER_SIZE (owner_offset) == sizeof (gpointer));
 710         g_assert (MONO_THREADS_SYNC_MEMBER_SIZE (nest_offset) == sizeof (guint32));
 711         owner_offset = MONO_THREADS_SYNC_MEMBER_OFFSET (owner_offset);
 712         nest_offset = MONO_THREADS_SYNC_MEMBER_OFFSET (nest_offset);
 713
 714         tramp_size = 96;
 715
 716         code = buf = mono_global_codeman_reserve (tramp_size);
 717
 718         if (mono_thread_get_tls_offset () != -1) {
 719                 /* MonoObject* obj is in RDI */
 720                 /* is obj null? */
 721                 amd64_test_reg_reg (buf, AMD64_RDI, AMD64_RDI);
 722                 /* if yes, jump to actual trampoline */
 723                 jump_obj_null = buf;
 724                 amd64_branch8 (buf, X86_CC_Z, -1, 1);
 725
 726                 /* load obj->synchronization to RCX */
 727                 amd64_mov_reg_membase (buf, AMD64_RCX, AMD64_RDI, G_STRUCT_OFFSET (MonoObject, synchronisation), 8);
 728                 /* is synchronization null? */
 729                 amd64_test_reg_reg (buf, AMD64_RCX, AMD64_RCX);
 730                 /* if yes, jump to actual trampoline */
 731                 jump_sync_null = buf;
 732                 amd64_branch8 (buf, X86_CC_Z, -1, 1);
 733
 734                 /* load MonoThread* into RDX */
 735                 buf = mono_amd64_emit_tls_get (buf, AMD64_RDX, mono_thread_get_tls_offset ());
 736                 /* load TID into RDX */
 737                 amd64_mov_reg_membase (buf, AMD64_RDX, AMD64_RDX, G_STRUCT_OFFSET (MonoThread, tid), 8);
 738
 739                 /* is synchronization->owner null? */
 740                 amd64_alu_membase_imm_size (buf, X86_CMP, AMD64_RCX, owner_offset, 0, 8);
 741                 /* if not, jump to next case */
 742                 jump_tid = buf;
 743                 amd64_branch8 (buf, X86_CC_NZ, -1, 1);
 744
 745                 /* if yes, try a compare-exchange with the TID */
 746                 /* zero RAX */
 747                 amd64_alu_reg_reg (buf, X86_XOR, AMD64_RAX, AMD64_RAX);
 748                 /* compare and exchange */
 749                 amd64_prefix (buf, X86_LOCK_PREFIX);
 750                 amd64_cmpxchg_membase_reg_size (buf, AMD64_RCX, owner_offset, AMD64_RDX, 8);
 751                 /* if not successful, jump to actual trampoline */
 752                 jump_cmpxchg_failed = buf;
 753                 amd64_branch8 (buf, X86_CC_NZ, -1, 1);
 754                 /* if successful, return */
 755                 amd64_ret (buf);
 756
 757                 /* next case: synchronization->owner is not null */
 758                 x86_patch (jump_tid, buf);
 759                 /* is synchronization->owner == TID? */
 760                 amd64_alu_membase_reg_size (buf, X86_CMP, AMD64_RCX, owner_offset, AMD64_RDX, 8);
 761                 /* if not, jump to actual trampoline */
 762                 jump_other_owner = buf;
 763                 amd64_branch8 (buf, X86_CC_NZ, -1, 1);
 764                 /* if yes, increment nest */
 765                 amd64_inc_membase_size (buf, AMD64_RCX, nest_offset, 4);
 766                 /* return */
 767                 amd64_ret (buf);
 768
 769                 x86_patch (jump_obj_null, buf);
 770                 x86_patch (jump_sync_null, buf);
 771                 x86_patch (jump_cmpxchg_failed, buf);
 772                 x86_patch (jump_other_owner, buf);
 773         }
 774
 775         /* jump to the actual trampoline */
 776 #if MONO_AMD64_ARG_REG1 != AMD64_RDI
 777         amd64_mov_reg_reg (buf, MONO_AMD64_ARG_REG1, AMD64_RDI);
 778 #endif
 779
 780         if (aot) {
 781                 *ji = mono_patch_info_list_prepend (*ji, buf - code, MONO_PATCH_INFO_JIT_ICALL_ADDR, "specific_trampoline_monitor_enter");
 782                 amd64_mov_reg_membase (buf, AMD64_R11, AMD64_RIP, 0, 8);
 783                 amd64_jump_reg (buf, AMD64_R11);
 784         } else {
 785                 tramp = mono_arch_create_specific_trampoline (NULL, MONO_TRAMPOLINE_MONITOR_ENTER, mono_get_root_domain (), NULL);
 786
 787                 /* jump to the actual trampoline */
 788                 amd64_jump_code (buf, tramp);
 789         }
 790
 791         mono_arch_flush_icache (buf, buf - code);
 792         g_assert (buf - code <= tramp_size);
 793
 794         *code_size = buf - code;
 795
 796         return code;
 797 }
 798
 799 gpointer
 800 mono_arch_create_monitor_exit_trampoline (void)
 801 {
 802         guint32 code_size;
 803         MonoJumpInfo *ji;
 804
 805         return mono_arch_create_monitor_exit_trampoline_full (&code_size, &ji, FALSE);
 806 }
 807
 808 gpointer
 809 mono_arch_create_monitor_exit_trampoline_full (guint32 *code_size, MonoJumpInfo **ji, gboolean aot)
 810 {
 811         guint8 *tramp;
 812         guint8 *code, *buf;
 813         guint8 *jump_obj_null, *jump_have_waiters;
 814         guint8 *jump_next;
 815         int tramp_size;
 816         int owner_offset, nest_offset, entry_count_offset;
 817
 818         g_assert (MONO_ARCH_MONITOR_OBJECT_REG == AMD64_RDI);
 819
 820         mono_monitor_threads_sync_members_offset (&owner_offset, &nest_offset, &entry_count_offset);
 821         g_assert (MONO_THREADS_SYNC_MEMBER_SIZE (owner_offset) == sizeof (gpointer));
 822         g_assert (MONO_THREADS_SYNC_MEMBER_SIZE (nest_offset) == sizeof (guint32));
 823         g_assert (MONO_THREADS_SYNC_MEMBER_SIZE (entry_count_offset) == sizeof (gint32));
 824         owner_offset = MONO_THREADS_SYNC_MEMBER_OFFSET (owner_offset);
 825         nest_offset = MONO_THREADS_SYNC_MEMBER_OFFSET (nest_offset);
 826         entry_count_offset = MONO_THREADS_SYNC_MEMBER_OFFSET (entry_count_offset);
 827
 828         tramp_size = 94;
 829
 830         code = buf = mono_global_codeman_reserve (tramp_size);
 831
 832         if (mono_thread_get_tls_offset () != -1) {
 833                 /* MonoObject* obj is in RDI */
 834                 /* is obj null? */
 835                 amd64_test_reg_reg (buf, AMD64_RDI, AMD64_RDI);
 836                 /* if yes, jump to actual trampoline */
 837                 jump_obj_null = buf;
 838                 amd64_branch8 (buf, X86_CC_Z, -1, 1);
 839
 840                 /* load obj->synchronization to RCX */
 841                 amd64_mov_reg_membase (buf, AMD64_RCX, AMD64_RDI, G_STRUCT_OFFSET (MonoObject, synchronisation), 8);
 842                 /* is synchronization null? */
 843                 amd64_test_reg_reg (buf, AMD64_RCX, AMD64_RCX);
 844                 /* if not, jump to next case */
 845                 jump_next = buf;
 846                 amd64_branch8 (buf, X86_CC_NZ, -1, 1);
 847                 /* if yes, just return */
 848                 amd64_ret (buf);
 849
 850                 /* next case: synchronization is not null */
 851                 x86_patch (jump_next, buf);
 852                 /* load MonoThread* into RDX */
 853                 buf = mono_amd64_emit_tls_get (buf, AMD64_RDX, mono_thread_get_tls_offset ());
 854                 /* load TID into RDX */
 855                 amd64_mov_reg_membase (buf, AMD64_RDX, AMD64_RDX, G_STRUCT_OFFSET (MonoThread, tid), 8);
 856                 /* is synchronization->owner == TID */
 857                 amd64_alu_membase_reg_size (buf, X86_CMP, AMD64_RCX, owner_offset, AMD64_RDX, 8);
 858                 /* if yes, jump to next case */
 859                 jump_next = buf;
 860                 amd64_branch8 (buf, X86_CC_Z, -1, 1);
 861                 /* if not, just return */
 862                 amd64_ret (buf);
 863
 864                 /* next case: synchronization->owner == TID */
 865                 x86_patch (jump_next, buf);
 866                 /* is synchronization->nest == 1 */
 867                 amd64_alu_membase_imm_size (buf, X86_CMP, AMD64_RCX, nest_offset, 1, 4);
 868                 /* if not, jump to next case */
 869                 jump_next = buf;
 870                 amd64_branch8 (buf, X86_CC_NZ, -1, 1);
 871                 /* if yes, is synchronization->entry_count zero? */
 872                 amd64_alu_membase_imm_size (buf, X86_CMP, AMD64_RCX, entry_count_offset, 0, 4);
 873                 /* if not, jump to actual trampoline */
 874                 jump_have_waiters = buf;
 875                 amd64_branch8 (buf, X86_CC_NZ, -1 , 1);
 876                 /* if yes, set synchronization->owner to null and return */
 877                 amd64_mov_membase_imm (buf, AMD64_RCX, owner_offset, 0, 8);
 878                 amd64_ret (buf);
 879
 880                 /* next case: synchronization->nest is not 1 */
 881                 x86_patch (jump_next, buf);
 882                 /* decrease synchronization->nest and return */
 883                 amd64_dec_membase_size (buf, AMD64_RCX, nest_offset, 4);
 884                 amd64_ret (buf);
 885
 886                 x86_patch (jump_obj_null, buf);
 887                 x86_patch (jump_have_waiters, buf);
 888         }
 889
 890         /* jump to the actual trampoline */
 891 #if MONO_AMD64_ARG_REG1 != AMD64_RDI
 892         amd64_mov_reg_reg (buf, MONO_AMD64_ARG_REG1, AMD64_RDI);
 893 #endif
 894
 895         if (aot) {
 896                 *ji = mono_patch_info_list_prepend (*ji, buf - code, MONO_PATCH_INFO_JIT_ICALL_ADDR, "specific_trampoline_monitor_exit");
 897                 amd64_mov_reg_membase (buf, AMD64_R11, AMD64_RIP, 0, 8);
 898                 amd64_jump_reg (buf, AMD64_R11);
 899         } else {
 900                 tramp = mono_arch_create_specific_trampoline (NULL, MONO_TRAMPOLINE_MONITOR_EXIT, mono_get_root_domain (), NULL);
 901                 amd64_jump_code (buf, tramp);
 902         }
 903
 904         mono_arch_flush_icache (buf, buf - code);
 905         g_assert (buf - code <= tramp_size);
 906
 907         *code_size = buf - code;
 908
 909         return code;
 910 }
 911 #endif
 912
 913 void
 914 mono_arch_invalidate_method (MonoJitInfo *ji, void *func, gpointer func_arg)
 915 {
 916         /* FIXME: This is not thread safe */
 917         guint8 *code = ji->code_start;
 918
 919         amd64_mov_reg_imm (code, AMD64_ARG_REG1, func_arg);
 920         amd64_mov_reg_imm (code, AMD64_R11, func);
 921
 922         x86_push_imm (code, (guint64)func_arg);
 923         amd64_call_reg (code, AMD64_R11);
 924 }