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, 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         tramp_offset = - offset;
 265
 266         offset += 8;
 267         arg_offset = - offset;
 268
 269         /* Compute the trampoline address from the return address */
 270         if (aot) {
 271                 /* 7 = length of call *<offset>(rip) */
 272                 amd64_alu_reg_imm (code, X86_SUB, AMD64_R11, 7);
 273         } else {
 274                 /* 5 = length of amd64_call_membase () */
 275                 amd64_alu_reg_imm (code, X86_SUB, AMD64_R11, 5);
 276         }
 277         amd64_mov_membase_reg (code, AMD64_RBP, tramp_offset, AMD64_R11, 8);
 278
 279         offset += 8;
 280         res_offset = - offset;
 281
 282         /* Save all registers */
 283
 284         offset += AMD64_NREG * 8;
 285         saved_regs_offset = - offset;
 286         for (i = 0; i < AMD64_NREG; ++i) {
 287                 if (i == AMD64_RBP) {
 288                         /* RAX is already saved */
 289                         amd64_mov_reg_membase (code, AMD64_RAX, AMD64_RBP, rbp_offset, 8);
 290                         amd64_mov_membase_reg (code, AMD64_RBP, saved_regs_offset + (i * 8), AMD64_RAX, 8);
 291                 } else if (i != AMD64_R11) {
 292                         amd64_mov_membase_reg (code, AMD64_RBP, saved_regs_offset + (i * 8), i, 8);
 293                 } else {
 294                         /* We have to save R11 right at the start of
 295                            the trampoline code because it's used as a
 296                            scratch register */
 297                         amd64_mov_membase_reg (r11_save_code, AMD64_RSP, saved_regs_offset + orig_rsp_to_rbp_offset + (i * 8), i, 8);
 298                         g_assert (r11_save_code == after_r11_save_code);
 299                 }
 300         }
 301         offset += 8 * 8;
 302         saved_fpregs_offset = - offset;
 303         for (i = 0; i < 8; ++i)
 304                 amd64_movsd_membase_reg (code, AMD64_RBP, saved_fpregs_offset + (i * 8), i);
 305
 306         if (tramp_type != MONO_TRAMPOLINE_GENERIC_CLASS_INIT &&
 307                         tramp_type != MONO_TRAMPOLINE_MONITOR_ENTER &&
 308                         tramp_type != MONO_TRAMPOLINE_MONITOR_EXIT) {
 309                 /* Obtain the trampoline argument which is encoded in the instruction stream */
 310                 if (aot) {
 311                         /* Load the GOT offset */
 312                         amd64_mov_reg_membase (code, AMD64_R11, AMD64_RBP, tramp_offset, 8);
 313                         amd64_mov_reg_membase (code, AMD64_RAX, AMD64_R11, 7, 4);
 314                         /* Compute the address of the GOT slot */
 315                         amd64_alu_reg_reg_size (code, X86_ADD, AMD64_R11, AMD64_RAX, 8);
 316                         /* Load the value */
 317                         amd64_mov_reg_membase (code, AMD64_R11, AMD64_R11, 0, 8);
 318                 } else {
 319                         amd64_mov_reg_membase (code, AMD64_R11, AMD64_RBP, tramp_offset, 8);
 320                         amd64_mov_reg_membase (code, AMD64_RAX, AMD64_R11, 5, 1);
 321                         amd64_widen_reg (code, AMD64_RAX, AMD64_RAX, TRUE, FALSE);
 322                         amd64_alu_reg_imm_size (code, X86_CMP, AMD64_RAX, 4, 1);
 323                         br [0] = code;
 324                         x86_branch8 (code, X86_CC_NE, 6, FALSE);
 325                         /* 32 bit immediate */
 326                         amd64_mov_reg_membase (code, AMD64_R11, AMD64_R11, 6, 4);
 327                         br [1] = code;
 328                         x86_jump8 (code, 10);
 329                         /* 64 bit immediate */
 330                         mono_amd64_patch (br [0], code);
 331                         amd64_mov_reg_membase (code, AMD64_R11, AMD64_R11, 6, 8);
 332                         mono_amd64_patch (br [1], code);
 333                 }
 334                 amd64_mov_membase_reg (code, AMD64_RBP, arg_offset, AMD64_R11, 8);
 335         } else {
 336                 amd64_mov_reg_membase (code, AMD64_R11, AMD64_RBP, saved_regs_offset + (MONO_AMD64_ARG_REG1 * 8), 8);
 337                 amd64_mov_membase_reg (code, AMD64_RBP, arg_offset, AMD64_R11, 8);
 338         }
 339
 340         /* Save LMF begin */
 341
 342         offset += sizeof (MonoLMF);
 343         lmf_offset = - offset;
 344
 345         /* Save ip */
 346         if (has_caller)
 347                 amd64_mov_reg_membase (code, AMD64_R11, AMD64_RBP, 8, 8);
 348         else
 349                 amd64_mov_reg_imm (code, AMD64_R11, 0);
 350         amd64_mov_membase_reg (code, AMD64_RBP, lmf_offset + G_STRUCT_OFFSET (MonoLMF, rip), AMD64_R11, 8);
 351         /* Save fp */
 352         amd64_mov_reg_membase (code, AMD64_R11, AMD64_RSP, framesize, 8);
 353         amd64_mov_membase_reg (code, AMD64_RBP, lmf_offset + G_STRUCT_OFFSET (MonoLMF, rbp), AMD64_R11, 8);
 354         /* Save sp */
 355         amd64_mov_reg_reg (code, AMD64_R11, AMD64_RSP, 8);
 356         amd64_alu_reg_imm (code, X86_ADD, AMD64_R11, framesize + 16);
 357         amd64_mov_membase_reg (code, AMD64_RBP, lmf_offset + G_STRUCT_OFFSET (MonoLMF, rsp), AMD64_R11, 8);
 358         /* Save method */
 359         if (tramp_type == MONO_TRAMPOLINE_JIT || tramp_type == MONO_TRAMPOLINE_JUMP) {
 360                 amd64_mov_reg_membase (code, AMD64_R11, AMD64_RBP, arg_offset, 8);
 361                 amd64_mov_membase_reg (code, AMD64_RBP, lmf_offset + G_STRUCT_OFFSET (MonoLMF, method), AMD64_R11, 8);
 362         } else {
 363                 amd64_mov_membase_imm (code, AMD64_RBP, lmf_offset + G_STRUCT_OFFSET (MonoLMF, method), 0, 8);
 364         }
 365         /* Save callee saved regs */
 366 #ifdef PLATFORM_WIN32
 367         amd64_mov_membase_reg (code, AMD64_RBP, lmf_offset + G_STRUCT_OFFSET (MonoLMF, rdi), AMD64_RDI, 8);
 368         amd64_mov_membase_reg (code, AMD64_RBP, lmf_offset + G_STRUCT_OFFSET (MonoLMF, rsi), AMD64_RSI, 8);
 369 #endif
 370         amd64_mov_membase_reg (code, AMD64_RBP, lmf_offset + G_STRUCT_OFFSET (MonoLMF, rbx), AMD64_RBX, 8);
 371         amd64_mov_membase_reg (code, AMD64_RBP, lmf_offset + G_STRUCT_OFFSET (MonoLMF, r12), AMD64_R12, 8);
 372         amd64_mov_membase_reg (code, AMD64_RBP, lmf_offset + G_STRUCT_OFFSET (MonoLMF, r13), AMD64_R13, 8);
 373         amd64_mov_membase_reg (code, AMD64_RBP, lmf_offset + G_STRUCT_OFFSET (MonoLMF, r14), AMD64_R14, 8);
 374         amd64_mov_membase_reg (code, AMD64_RBP, lmf_offset + G_STRUCT_OFFSET (MonoLMF, r15), AMD64_R15, 8);
 375
 376         if (aot) {
 377                 *ji = mono_patch_info_list_prepend (*ji, code - buf, MONO_PATCH_INFO_JIT_ICALL_ADDR, "mono_get_lmf_addr");
 378                 amd64_mov_reg_membase (code, AMD64_R11, AMD64_RIP, 0, 8);
 379         } else {
 380                 amd64_mov_reg_imm (code, AMD64_R11, mono_get_lmf_addr);
 381         }
 382         amd64_call_reg (code, AMD64_R11);
 383
 384         /* Save lmf_addr */
 385         amd64_mov_membase_reg (code, AMD64_RBP, lmf_offset + G_STRUCT_OFFSET (MonoLMF, lmf_addr), AMD64_RAX, 8);
 386         /* Save previous_lmf */
 387         /* Set the lowest bit to 1 to signal that this LMF has the ip field set */
 388         amd64_mov_reg_membase (code, AMD64_R11, AMD64_RAX, 0, 8);
 389         amd64_alu_reg_imm_size (code, X86_ADD, AMD64_R11, 1, 8);
 390         amd64_mov_membase_reg (code, AMD64_RBP, lmf_offset + G_STRUCT_OFFSET (MonoLMF, previous_lmf), AMD64_R11, 8);
 391         /* Set new lmf */
 392         amd64_lea_membase (code, AMD64_R11, AMD64_RBP, lmf_offset);
 393         amd64_mov_membase_reg (code, AMD64_RAX, 0, AMD64_R11, 8);
 394
 395         /* Save LMF end */
 396
 397         /* Arg1 is the pointer to the saved registers */
 398         amd64_lea_membase (code, AMD64_ARG_REG1, AMD64_RBP, saved_regs_offset);
 399
 400         /* Arg2 is the address of the calling code */
 401         if (has_caller)
 402                 amd64_mov_reg_membase (code, AMD64_ARG_REG2, AMD64_RBP, 8, 8);
 403         else
 404                 amd64_mov_reg_imm (code, AMD64_ARG_REG2, 0);
 405
 406         /* Arg3 is the method/vtable ptr */
 407         amd64_mov_reg_membase (code, AMD64_ARG_REG3, AMD64_RBP, arg_offset, 8);
 408
 409         /* Arg4 is the trampoline address */
 410         amd64_mov_reg_membase (code, AMD64_ARG_REG4, AMD64_RBP, tramp_offset, 8);
 411
 412         if (aot) {
 413                 char *icall_name = g_strdup_printf ("trampoline_func_%d", tramp_type);
 414                 *ji = mono_patch_info_list_prepend (*ji, code - buf, MONO_PATCH_INFO_JIT_ICALL_ADDR, icall_name);
 415                 amd64_mov_reg_membase (code, AMD64_RAX, AMD64_RIP, 0, 8);
 416         } else {
 417                 tramp = (guint8*)mono_get_trampoline_func (tramp_type);
 418                 amd64_mov_reg_imm (code, AMD64_RAX, tramp);
 419         }
 420         amd64_call_reg (code, AMD64_RAX);
 421
 422         /* Check for thread interruption */
 423         /* This is not perf critical code so no need to check the interrupt flag */
 424         /*
 425          * Have to call the _force_ variant, since there could be a protected wrapper on the top of the stack.
 426          */
 427         amd64_mov_membase_reg (code, AMD64_RBP, res_offset, AMD64_RAX, 8);
 428         if (aot) {
 429                 *ji = mono_patch_info_list_prepend (*ji, code - buf, MONO_PATCH_INFO_JIT_ICALL_ADDR, "mono_thread_force_interruption_checkpoint");
 430                 amd64_mov_reg_membase (code, AMD64_RAX, AMD64_RIP, 0, 8);
 431         } else {
 432                 amd64_mov_reg_imm (code, AMD64_RAX, (guint8*)mono_thread_force_interruption_checkpoint);
 433         }
 434         amd64_call_reg (code, AMD64_RAX);
 435         amd64_mov_reg_membase (code, AMD64_RAX, AMD64_RBP, res_offset, 8);
 436
 437         /* Restore LMF */
 438
 439         amd64_mov_reg_membase (code, AMD64_RCX, AMD64_RBP, lmf_offset + G_STRUCT_OFFSET (MonoLMF, previous_lmf), 8);
 440         amd64_alu_reg_imm_size (code, X86_SUB, AMD64_RCX, 1, 8);
 441         amd64_mov_reg_membase (code, AMD64_R11, AMD64_RBP, lmf_offset + G_STRUCT_OFFSET (MonoLMF, lmf_addr), 8);
 442         amd64_mov_membase_reg (code, AMD64_R11, 0, AMD64_RCX, 8);
 443
 444         /* Restore argument registers, r10 (needed to pass rgctx to
 445            static shared generic methods) and r11 (imt register for
 446            interface calls). */
 447         for (i = 0; i < AMD64_NREG; ++i)
 448                 if (AMD64_IS_ARGUMENT_REG (i) || i == AMD64_R10 || i == AMD64_R11)
 449                         amd64_mov_reg_membase (code, i, AMD64_RBP, saved_regs_offset + (i * 8), 8);
 450
 451         /*
 452          * FIXME: When using aot-only, the called code might be a C vararg function
 453          * which uses %rax as well.
 454          * We could restore it, but we would have to use another register to store the
 455          * target address, and we don't have any left.
 456          * Also, the default AOT plt trampolines overwrite 'rax'.
 457          */
 458
 459         for (i = 0; i < 8; ++i)
 460                 amd64_movsd_reg_membase (code, i, AMD64_RBP, saved_fpregs_offset + (i * 8));
 461
 462         if (tramp_type == MONO_TRAMPOLINE_RESTORE_STACK_PROT)
 463                 amd64_mov_reg_membase (code, AMD64_RAX, AMD64_RBP, saved_regs_offset + (AMD64_RAX * 8), 8);
 464
 465         /* Restore stack */
 466         amd64_leave (code);
 467
 468         if (MONO_TRAMPOLINE_TYPE_MUST_RETURN (tramp_type)) {
 469                 amd64_ret (code);
 470         } else {
 471                 /* call the compiled method */
 472                 amd64_jump_reg (code, AMD64_RAX);
 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         return buf;
 645 }
 646
 647 gpointer
 648 mono_arch_create_generic_class_init_trampoline (void)
 649 {
 650         guint8 *tramp;
 651         guint8 *code, *buf;
 652         static int byte_offset = -1;
 653         static guint8 bitmask;
 654         guint8 *jump;
 655         int tramp_size;
 656
 657         tramp_size = 64;
 658
 659         code = buf = mono_global_codeman_reserve (tramp_size);
 660
 661         if (byte_offset < 0)
 662                 mono_marshal_find_bitfield_offset (MonoVTable, initialized, &byte_offset, &bitmask);
 663
 664         amd64_test_membase_imm_size (code, MONO_AMD64_ARG_REG1, byte_offset, bitmask, 1);
 665         jump = code;
 666         amd64_branch8 (code, X86_CC_Z, -1, 1);
 667
 668         amd64_ret (code);
 669
 670         x86_patch (jump, code);
 671
 672         tramp = mono_arch_create_specific_trampoline (NULL, MONO_TRAMPOLINE_GENERIC_CLASS_INIT, mono_get_root_domain (), NULL);
 673
 674         /* jump to the actual trampoline */
 675         amd64_jump_code (code, tramp);
 676
 677         mono_arch_flush_icache (buf, code - buf);
 678
 679         g_assert (code - buf <= tramp_size);
 680
 681         return buf;
 682 }
 683
 684 #ifdef MONO_ARCH_MONITOR_OBJECT_REG
 685 gpointer
 686 mono_arch_create_monitor_enter_trampoline (void)
 687 {
 688         guint8 *tramp = mono_arch_create_specific_trampoline (NULL, MONO_TRAMPOLINE_MONITOR_ENTER, mono_get_root_domain (), NULL);
 689         guint8 *code, *buf;
 690         guint8 *jump_obj_null, *jump_sync_null, *jump_cmpxchg_failed, *jump_other_owner, *jump_tid;
 691         int tramp_size;
 692         int owner_offset, nest_offset, dummy;
 693
 694         g_assert (MONO_ARCH_MONITOR_OBJECT_REG == AMD64_RDI);
 695
 696         mono_monitor_threads_sync_members_offset (&owner_offset, &nest_offset, &dummy);
 697         g_assert (MONO_THREADS_SYNC_MEMBER_SIZE (owner_offset) == sizeof (gpointer));
 698         g_assert (MONO_THREADS_SYNC_MEMBER_SIZE (nest_offset) == sizeof (guint32));
 699         owner_offset = MONO_THREADS_SYNC_MEMBER_OFFSET (owner_offset);
 700         nest_offset = MONO_THREADS_SYNC_MEMBER_OFFSET (nest_offset);
 701
 702         tramp_size = 96;
 703
 704         code = buf = mono_global_codeman_reserve (tramp_size);
 705
 706         if (mono_thread_get_tls_offset () != -1) {
 707                 /* MonoObject* obj is in RDI */
 708                 /* is obj null? */
 709                 amd64_test_reg_reg (buf, AMD64_RDI, AMD64_RDI);
 710                 /* if yes, jump to actual trampoline */
 711                 jump_obj_null = buf;
 712                 amd64_branch8 (buf, X86_CC_Z, -1, 1);
 713
 714                 /* load obj->synchronization to RCX */
 715                 amd64_mov_reg_membase (buf, AMD64_RCX, AMD64_RDI, G_STRUCT_OFFSET (MonoObject, synchronisation), 8);
 716                 /* is synchronization null? */
 717                 amd64_test_reg_reg (buf, AMD64_RCX, AMD64_RCX);
 718                 /* if yes, jump to actual trampoline */
 719                 jump_sync_null = buf;
 720                 amd64_branch8 (buf, X86_CC_Z, -1, 1);
 721
 722                 /* load MonoThread* into RDX */
 723                 buf = mono_amd64_emit_tls_get (buf, AMD64_RDX, mono_thread_get_tls_offset ());
 724                 /* load TID into RDX */
 725                 amd64_mov_reg_membase (buf, AMD64_RDX, AMD64_RDX, G_STRUCT_OFFSET (MonoThread, tid), 8);
 726
 727                 /* is synchronization->owner null? */
 728                 amd64_alu_membase_imm_size (buf, X86_CMP, AMD64_RCX, owner_offset, 0, 8);
 729                 /* if not, jump to next case */
 730                 jump_tid = buf;
 731                 amd64_branch8 (buf, X86_CC_NZ, -1, 1);
 732
 733                 /* if yes, try a compare-exchange with the TID */
 734                 /* zero RAX */
 735                 amd64_alu_reg_reg (buf, X86_XOR, AMD64_RAX, AMD64_RAX);
 736                 /* compare and exchange */
 737                 amd64_prefix (buf, X86_LOCK_PREFIX);
 738                 amd64_cmpxchg_membase_reg_size (buf, AMD64_RCX, owner_offset, AMD64_RDX, 8);
 739                 /* if not successful, jump to actual trampoline */
 740                 jump_cmpxchg_failed = buf;
 741                 amd64_branch8 (buf, X86_CC_NZ, -1, 1);
 742                 /* if successful, return */
 743                 amd64_ret (buf);
 744
 745                 /* next case: synchronization->owner is not null */
 746                 x86_patch (jump_tid, buf);
 747                 /* is synchronization->owner == TID? */
 748                 amd64_alu_membase_reg_size (buf, X86_CMP, AMD64_RCX, owner_offset, AMD64_RDX, 8);
 749                 /* if not, jump to actual trampoline */
 750                 jump_other_owner = buf;
 751                 amd64_branch8 (buf, X86_CC_NZ, -1, 1);
 752                 /* if yes, increment nest */
 753                 amd64_inc_membase_size (buf, AMD64_RCX, nest_offset, 4);
 754                 /* return */
 755                 amd64_ret (buf);
 756
 757                 x86_patch (jump_obj_null, buf);
 758                 x86_patch (jump_sync_null, buf);
 759                 x86_patch (jump_cmpxchg_failed, buf);
 760                 x86_patch (jump_other_owner, buf);
 761         }
 762
 763         /* jump to the actual trampoline */
 764 #if MONO_AMD64_ARG_REG1 != AMD64_RDI
 765         amd64_mov_reg_reg (buf, MONO_AMD64_ARG_REG1, AMD64_RDI);
 766 #endif
 767         amd64_jump_code (buf, tramp);
 768
 769         mono_arch_flush_icache (buf, buf - code);
 770         g_assert (buf - code <= tramp_size);
 771
 772         return code;
 773 }
 774
 775 gpointer
 776 mono_arch_create_monitor_exit_trampoline (void)
 777 {
 778         guint8 *tramp = mono_arch_create_specific_trampoline (NULL, MONO_TRAMPOLINE_MONITOR_EXIT, mono_get_root_domain (), NULL);
 779         guint8 *code, *buf;
 780         guint8 *jump_obj_null, *jump_have_waiters;
 781         guint8 *jump_next;
 782         int tramp_size;
 783         int owner_offset, nest_offset, entry_count_offset;
 784
 785         g_assert (MONO_ARCH_MONITOR_OBJECT_REG == AMD64_RDI);
 786
 787         mono_monitor_threads_sync_members_offset (&owner_offset, &nest_offset, &entry_count_offset);
 788         g_assert (MONO_THREADS_SYNC_MEMBER_SIZE (owner_offset) == sizeof (gpointer));
 789         g_assert (MONO_THREADS_SYNC_MEMBER_SIZE (nest_offset) == sizeof (guint32));
 790         g_assert (MONO_THREADS_SYNC_MEMBER_SIZE (entry_count_offset) == sizeof (gint32));
 791         owner_offset = MONO_THREADS_SYNC_MEMBER_OFFSET (owner_offset);
 792         nest_offset = MONO_THREADS_SYNC_MEMBER_OFFSET (nest_offset);
 793         entry_count_offset = MONO_THREADS_SYNC_MEMBER_OFFSET (entry_count_offset);
 794
 795         tramp_size = 94;
 796
 797         code = buf = mono_global_codeman_reserve (tramp_size);
 798
 799         if (mono_thread_get_tls_offset () != -1) {
 800                 /* MonoObject* obj is in RDI */
 801                 /* is obj null? */
 802                 amd64_test_reg_reg (buf, AMD64_RDI, AMD64_RDI);
 803                 /* if yes, jump to actual trampoline */
 804                 jump_obj_null = buf;
 805                 amd64_branch8 (buf, X86_CC_Z, -1, 1);
 806
 807                 /* load obj->synchronization to RCX */
 808                 amd64_mov_reg_membase (buf, AMD64_RCX, AMD64_RDI, G_STRUCT_OFFSET (MonoObject, synchronisation), 8);
 809                 /* is synchronization null? */
 810                 amd64_test_reg_reg (buf, AMD64_RCX, AMD64_RCX);
 811                 /* if not, jump to next case */
 812                 jump_next = buf;
 813                 amd64_branch8 (buf, X86_CC_NZ, -1, 1);
 814                 /* if yes, just return */
 815                 amd64_ret (buf);
 816
 817                 /* next case: synchronization is not null */
 818                 x86_patch (jump_next, buf);
 819                 /* load MonoThread* into RDX */
 820                 buf = mono_amd64_emit_tls_get (buf, AMD64_RDX, mono_thread_get_tls_offset ());
 821                 /* load TID into RDX */
 822                 amd64_mov_reg_membase (buf, AMD64_RDX, AMD64_RDX, G_STRUCT_OFFSET (MonoThread, tid), 8);
 823                 /* is synchronization->owner == TID */
 824                 amd64_alu_membase_reg_size (buf, X86_CMP, AMD64_RCX, owner_offset, AMD64_RDX, 8);
 825                 /* if yes, jump to next case */
 826                 jump_next = buf;
 827                 amd64_branch8 (buf, X86_CC_Z, -1, 1);
 828                 /* if not, just return */
 829                 amd64_ret (buf);
 830
 831                 /* next case: synchronization->owner == TID */
 832                 x86_patch (jump_next, buf);
 833                 /* is synchronization->nest == 1 */
 834                 amd64_alu_membase_imm_size (buf, X86_CMP, AMD64_RCX, nest_offset, 1, 4);
 835                 /* if not, jump to next case */
 836                 jump_next = buf;
 837                 amd64_branch8 (buf, X86_CC_NZ, -1, 1);
 838                 /* if yes, is synchronization->entry_count zero? */
 839                 amd64_alu_membase_imm_size (buf, X86_CMP, AMD64_RCX, entry_count_offset, 0, 4);
 840                 /* if not, jump to actual trampoline */
 841                 jump_have_waiters = buf;
 842                 amd64_branch8 (buf, X86_CC_NZ, -1 , 1);
 843                 /* if yes, set synchronization->owner to null and return */
 844                 amd64_mov_membase_imm (buf, AMD64_RCX, owner_offset, 0, 8);
 845                 amd64_ret (buf);
 846
 847                 /* next case: synchronization->nest is not 1 */
 848                 x86_patch (jump_next, buf);
 849                 /* decrease synchronization->nest and return */
 850                 amd64_dec_membase_size (buf, AMD64_RCX, nest_offset, 4);
 851                 amd64_ret (buf);
 852
 853                 x86_patch (jump_obj_null, buf);
 854                 x86_patch (jump_have_waiters, buf);
 855         }
 856
 857         /* jump to the actual trampoline */
 858 #if MONO_AMD64_ARG_REG1 != AMD64_RDI
 859         amd64_mov_reg_reg (buf, MONO_AMD64_ARG_REG1, AMD64_RDI);
 860 #endif
 861         amd64_jump_code (buf, tramp);
 862
 863         mono_arch_flush_icache (buf, buf - code);
 864         g_assert (buf - code <= tramp_size);
 865
 866         return code;
 867 }
 868 #endif
 869
 870 void
 871 mono_arch_invalidate_method (MonoJitInfo *ji, void *func, gpointer func_arg)
 872 {
 873         /* FIXME: This is not thread safe */
 874         guint8 *code = ji->code_start;
 875
 876         amd64_mov_reg_imm (code, AMD64_ARG_REG1, func_arg);
 877         amd64_mov_reg_imm (code, AMD64_R11, func);
 878
 879         x86_push_imm (code, (guint64)func_arg);
 880         amd64_call_reg (code, AMD64_R11);
 881 }