mono/jit/emit-x86.c

   1 /*
   2  * emit-x86.c: Support functions for emitting x86 code
   3  *
   4  * Authors:
   5  *   Dietmar Maurer (dietmar@ximian.com)
   6  *   Miguel de Icaza (miguel@ximian.com)
   7  *
   8  * (C) 2001 Ximian, Inc.
   9  */
  10
  11 #include <config.h>
  12 #include <glib.h>
  13
  14 #include <mono/metadata/assembly.h>
  15 #include <mono/metadata/loader.h>
  16 #include <mono/metadata/cil-coff.h>
  17 #include <mono/metadata/tabledefs.h>
  18 #include <mono/metadata/class.h>
  19 #include <mono/metadata/mono-endian.h>
  20 #include <mono/arch/x86/x86-codegen.h>
  21
  22 #include "jit.h"
  23 #include "codegen.h"
  24 #include "debug.h"
  25
  26 static void
  27 enter_method (MonoMethod *method, gpointer ebp)
  28 {
  29         int i, j;
  30         MonoClass *class;
  31         MonoObject *o;
  32
  33         printf ("ENTER: %s.%s::%s\n(", method->klass->name_space,
  34                 method->klass->name, method->name);
  35
  36
  37         if (((int)ebp & 3) != 0) {
  38                 g_error ("unaligned stack detected (%p)", ebp);
  39         }
  40
  41         ebp += 8;
  42
  43         if (ISSTRUCT (method->signature->ret)) {
  44                 int size, align;
  45
  46                 g_assert (!method->signature->ret->byref);
  47
  48                 size = mono_type_stack_size (method->signature->ret, &align);
  49
  50                 printf ("VALUERET:%p, ", *((gpointer *)ebp));
  51                 ebp += sizeof (gpointer);
  52         }
  53
  54         if (method->signature->hasthis) {
  55                 if (method->klass->valuetype) {
  56                         printf ("value:%p, ", *((gpointer *)ebp));
  57                 } else {
  58                         o = *((MonoObject **)ebp);
  59
  60                         g_assert (o);
  61
  62                         class = o->vtable->klass;
  63
  64                         if (class == mono_defaults.string_class) {
  65                                 printf ("this:[STRING:%p:%s], ", o, mono_string_to_utf8 ((MonoString *)o));
  66                         } else {
  67                                 printf ("this:%p[%s.%s], ", o, class->name_space, class->name);
  68                         }
  69                 }
  70                 ebp += sizeof (gpointer);
  71         }
  72
  73         for (i = 0; i < method->signature->param_count; ++i) {
  74                 MonoType *type = method->signature->params [i];
  75                 int size, align;
  76                 size = mono_type_stack_size (type, &align);
  77
  78                 if (type->byref) {
  79                         printf ("[BYREF:%p], ", *((gpointer *)ebp));
  80                 } else switch (type->type) {
  81
  82                 case MONO_TYPE_BOOLEAN:
  83                 case MONO_TYPE_CHAR:
  84                 case MONO_TYPE_I1:
  85                 case MONO_TYPE_U1:
  86                 case MONO_TYPE_I2:
  87                 case MONO_TYPE_U2:
  88                 case MONO_TYPE_I4:
  89                 case MONO_TYPE_U4:
  90                 case MONO_TYPE_I:
  91                 case MONO_TYPE_U:
  92                         printf ("%d, ", *((int *)(ebp)));
  93                         break;
  94                 case MONO_TYPE_STRING: {
  95                         MonoString *s = *((MonoString **)ebp);
  96                         if (s) {
  97                                 g_assert (((MonoObject *)s)->vtable->klass == mono_defaults.string_class);
  98                                 printf ("[STRING:%p:%s], ", s, mono_string_to_utf8 (s));
  99                         } else
 100                                 printf ("[STRING:null], ");
 101                         break;
 102                 }
 103                 case MONO_TYPE_CLASS:
 104                 case MONO_TYPE_OBJECT: {
 105                         o = *((MonoObject **)ebp);
 106                         if (o) {
 107                                 class = o->vtable->klass;
 108
 109                                 if (class == mono_defaults.string_class) {
 110                                         printf ("[STRING:%p:%s], ", o, mono_string_to_utf8 ((MonoString *)o));
 111                                 } else if (class == mono_defaults.int32_class) {
 112                                         printf ("[INT32:%p:%d], ", o, *(gint32 *)((gpointer)o + sizeof (MonoObject)));
 113                                 } else
 114                                         printf ("[%s.%s:%p], ", class->name_space, class->name, o);
 115                         } else {
 116                                 printf ("%p, ", *((gpointer *)(ebp)));
 117                         }
 118                         break;
 119                 }
 120                 case MONO_TYPE_PTR:
 121                 case MONO_TYPE_FNPTR:
 122                 case MONO_TYPE_ARRAY:
 123                 case MONO_TYPE_SZARRAY:
 124                         printf ("%p, ", *((gpointer *)(ebp)));
 125                         break;
 126                 case MONO_TYPE_I8:
 127                         printf ("%lld, ", *((gint64 *)(ebp)));
 128                         break;
 129                 case MONO_TYPE_R4:
 130                         printf ("%f, ", *((float *)(ebp)));
 131                         break;
 132                 case MONO_TYPE_R8:
 133                         printf ("%f, ", *((double *)(ebp)));
 134                         break;
 135                 case MONO_TYPE_VALUETYPE:
 136                         printf ("[");
 137                         for (j = 0; j < size; j++)
 138                                 printf ("%02x,", *((guint8*)ebp +j));
 139                         printf ("], ");
 140                         break;
 141                 default:
 142                         printf ("XX, ");
 143                 }
 144
 145                 g_assert (align == 4);
 146                 ebp += size + 3;
 147                 ebp = (gpointer)((unsigned)ebp & ~(3));
 148         }
 149
 150         printf (")\n");
 151 }
 152
 153 static void
 154 leave_method (MonoMethod *method, int edx, int eax, double test)
 155 {
 156         gint64 l;
 157
 158         printf ("LEAVE: %s.%s::%s ", method->klass->name_space,
 159                 method->klass->name, method->name);
 160
 161         switch (method->signature->ret->type) {
 162         case MONO_TYPE_VOID:
 163                 break;
 164         case MONO_TYPE_BOOLEAN:
 165                 if (eax)
 166                         printf ("TRUE:%d", eax);
 167                 else
 168                         printf ("FALSE");
 169
 170                 break;
 171         case MONO_TYPE_CHAR:
 172         case MONO_TYPE_I1:
 173         case MONO_TYPE_U1:
 174         case MONO_TYPE_I2:
 175         case MONO_TYPE_U2:
 176         case MONO_TYPE_I4:
 177         case MONO_TYPE_U4:
 178         case MONO_TYPE_I:
 179         case MONO_TYPE_U:
 180                 printf ("EAX=%d", eax);
 181                 break;
 182         case MONO_TYPE_STRING: {
 183                 MonoString *s = (MonoString *)eax;
 184
 185                 if (s) {
 186                         g_assert (((MonoObject *)s)->vtable->klass == mono_defaults.string_class);
 187                         printf ("[STRING:%p:%s]", s, mono_string_to_utf8 (s));
 188                 } else
 189                         printf ("[STRING:null], ");
 190                 break;
 191         }
 192         case MONO_TYPE_OBJECT: {
 193                 MonoObject *o = (MonoObject *)eax;
 194
 195                 if (o) {
 196                         if (o->vtable->klass == mono_defaults.boolean_class) {
 197                                 printf ("[BOOLEAN:%p:%d]", o, *((guint8 *)o + sizeof (MonoObject)));
 198                         } else if  (o->vtable->klass == mono_defaults.int32_class) {
 199                                 printf ("[INT32:%p:%d]", o, *((gint32 *)((gpointer)o + sizeof (MonoObject))));
 200                         } else
 201                                 printf ("[%s.%s:%p]", o->vtable->klass->name_space, o->vtable->klass->name, o);
 202                 } else
 203                         printf ("[OBJECT:%p]", o);
 204
 205                 break;
 206         }
 207         case MONO_TYPE_CLASS:
 208         case MONO_TYPE_PTR:
 209         case MONO_TYPE_FNPTR:
 210         case MONO_TYPE_ARRAY:
 211         case MONO_TYPE_SZARRAY:
 212                 printf ("EAX=%p", (gpointer)eax);
 213                 break;
 214         case MONO_TYPE_I8:
 215                 *((gint32 *)&l) = eax;
 216                 *((gint32 *)&l + 1) = edx;
 217                 printf ("EAX/EDX=%lld", l);
 218                 break;
 219         case MONO_TYPE_R8:
 220                 printf ("FP=%f\n", test);
 221                 break;
 222         default:
 223                 printf ("(unknown return type)");
 224         }
 225
 226         printf ("\n");
 227 }
 228
 229 /**
 230  * arch_emit_prologue:
 231  * @cfg: pointer to status information
 232  *
 233  * Emits the function prolog.
 234  */
 235 static void
 236 arch_emit_prologue (MonoFlowGraph *cfg)
 237 {
 238         x86_push_reg (cfg->code, X86_EBP);
 239         x86_mov_reg_reg (cfg->code, X86_EBP, X86_ESP, 4);
 240
 241         if (cfg->locals_size)
 242                 x86_alu_reg_imm (cfg->code, X86_SUB, X86_ESP, cfg->locals_size);
 243
 244         if (mono_regset_reg_used (cfg->rs, X86_EBX))
 245                 x86_push_reg (cfg->code, X86_EBX);
 246
 247         if (mono_regset_reg_used (cfg->rs, X86_EDI))
 248                 x86_push_reg (cfg->code, X86_EDI);
 249
 250         if (mono_regset_reg_used (cfg->rs, X86_ESI))
 251                 x86_push_reg (cfg->code, X86_ESI);
 252
 253         if (mono_jit_trace_calls) {
 254                 x86_push_reg (cfg->code, X86_EBP);
 255                 x86_push_imm (cfg->code, cfg->method);
 256                 x86_mov_reg_imm (cfg->code, X86_EAX, enter_method);
 257                 x86_call_reg (cfg->code, X86_EAX);
 258                 x86_alu_reg_imm (cfg->code, X86_ADD, X86_ESP, 8);
 259         }
 260 }
 261
 262 /**
 263  * arch_emit_epilogue:
 264  * @cfg: pointer to status information
 265  *
 266  * Emits the function epilog.
 267  */
 268 static void
 269 arch_emit_epilogue (MonoFlowGraph *cfg)
 270 {
 271         if (mono_jit_trace_calls) {
 272                 x86_fld_reg (cfg->code, 0);
 273                 x86_alu_reg_imm (cfg->code, X86_SUB, X86_ESP, 8);
 274                 x86_fst_membase (cfg->code, X86_ESP, 0, TRUE, TRUE);
 275                 x86_push_reg (cfg->code, X86_EAX);
 276                 x86_push_reg (cfg->code, X86_EDX);
 277                 x86_push_imm (cfg->code, cfg->method);
 278                 x86_mov_reg_imm (cfg->code, X86_EAX, leave_method);
 279                 x86_call_reg (cfg->code, X86_EAX);
 280                 x86_alu_reg_imm (cfg->code, X86_ADD, X86_ESP, 4);
 281                 x86_pop_reg (cfg->code, X86_EDX);
 282                 x86_pop_reg (cfg->code, X86_EAX);
 283                 x86_alu_reg_imm (cfg->code, X86_ADD, X86_ESP, 8);
 284         }
 285
 286         if (mono_regset_reg_used (cfg->rs, X86_ESI))
 287                 x86_pop_reg (cfg->code, X86_ESI);
 288
 289         if (mono_regset_reg_used (cfg->rs, X86_EDI))
 290                 x86_pop_reg (cfg->code, X86_EDI);
 291
 292         if (mono_regset_reg_used (cfg->rs, X86_EBX))
 293                 x86_pop_reg (cfg->code, X86_EBX);
 294
 295         x86_leave (cfg->code);
 296         x86_ret (cfg->code);
 297 }
 298
 299 /*
 300  * get_unbox_trampoline:
 301  * @m: method pointer
 302  * @addr: pointer to native code for @m
 303  *
 304  * when value type methods are called through the vtable we need to unbox the
 305  * this argument. This method returns a pointer to a trampoline which does
 306  * unboxing before calling the method
 307  */
 308 static gpointer
 309 get_unbox_trampoline (MonoMethod *m, gpointer addr)
 310 {
 311         guint8 *code, *start;
 312         int this_pos = 4;
 313
 314         if (!m->signature->ret->byref && m->signature->ret->type == MONO_TYPE_VALUETYPE)
 315                 this_pos = 8;
 316
 317         start = code = g_malloc (16);
 318
 319         x86_alu_membase_imm (code, X86_ADD, X86_ESP, this_pos, sizeof (MonoObject));
 320         x86_jump_code (code, addr);
 321         g_assert ((code - start) < 16);
 322
 323         return start;
 324 }
 325
 326 /**
 327  * x86_magic_trampoline:
 328  * @eax: saved x86 register
 329  * @ecx: saved x86 register
 330  * @edx: saved x86 register
 331  * @esi: saved x86 register
 332  * @edi: saved x86 register
 333  * @ebx: saved x86 register
 334  * @code: pointer into caller code
 335  * @method: the method to translate
 336  *
 337  * This method is called by the trampoline functions for virtual
 338  * methods. It inspects the caller code to find the address of the
 339  * vtable slot, then calls the JIT compiler and writes the address
 340  * of the compiled method back to the vtable. All virtual methods
 341  * are called with: x86_call_membase (inst, basereg, disp). We always
 342  * use 32 bit displacement to ensure that the length of the call
 343  * instruction is 6 bytes. We need to get the value of the basereg
 344  * and the constant displacement.
 345  */
 346 static gpointer
 347 x86_magic_trampoline (int eax, int ecx, int edx, int esi, int edi,
 348                       int ebx, const guint8 *code, MonoMethod *m)
 349 {
 350         guint8 reg;
 351         gint32 disp;
 352         gpointer o;
 353         gpointer addr;
 354
 355         EnterCriticalSection (metadata_section);
 356         addr = arch_compile_method (m);
 357         LeaveCriticalSection (metadata_section);
 358         g_assert (addr);
 359
 360
 361         /* go to the start of the call instruction
 362          *
 363          * address_byte = (m << 6) | (o << 3) | reg
 364          * call opcode: 0xff address_byte displacement
 365          * 0xff m=1,o=2 imm8
 366          * 0xff m=2,o=2 imm32
 367          */
 368         code -= 6;
 369         if ((code [1] != 0xe8) && (code [3] == 0xff) && ((code [4] & 0x18) == 0x10) && ((code [4] >> 6) == 1)) {
 370                 reg = code [4] & 0x07;
 371                 disp = (signed char)code [5];
 372         } else {
 373                 if ((code [0] == 0xff) && ((code [1] & 0x18) == 0x10) && ((code [1] >> 6) == 2)) {
 374                         reg = code [1] & 0x07;
 375                         disp = *((gint32*)(code + 2));
 376                 } else if ((code [1] == 0xe8)) {
 377                         *((guint32*)(code + 2)) = (guint)addr - ((guint)code + 1) - 5;
 378                         return addr;
 379                 } else {
 380                         printf ("%x %x %x %x %x %x \n", code [0], code [1], code [2], code [3],
 381                                 code [4], code [5]);
 382                         g_assert_not_reached ();
 383                 }
 384         }
 385
 386         switch (reg) {
 387         case X86_EAX:
 388                 o = (gpointer)eax;
 389                 break;
 390         case X86_EDX:
 391                 o = (gpointer)edx;
 392                 break;
 393         case X86_ECX:
 394                 o = (gpointer)ecx;
 395                 break;
 396         case X86_ESI:
 397                 o = (gpointer)esi;
 398                 break;
 399         case X86_EDI:
 400                 o = (gpointer)edi;
 401                 break;
 402         case X86_EBX:
 403                 o = (gpointer)ebx;
 404                 break;
 405         default:
 406                 g_assert_not_reached ();
 407         }
 408
 409         o += disp;
 410
 411         if (m->klass->valuetype) {
 412                 return *((gpointer *)o) = get_unbox_trampoline (m, addr);
 413         } else {
 414                 return *((gpointer *)o) = addr;
 415         }
 416 }
 417
 418 /**
 419  * arch_create_jit_trampoline:
 420  * @method: pointer to the method info
 421  *
 422  * Creates a trampoline function for virtual methods. If the created
 423  * code is called it first starts JIT compilation of method,
 424  * and then calls the newly created method. I also replaces the
 425  * corresponding vtable entry (see x86_magic_trampoline).
 426  *
 427  * Returns: a pointer to the newly created code
 428  */
 429 gpointer
 430 arch_create_jit_trampoline (MonoMethod *method)
 431 {
 432         MonoDomain *domain = mono_domain_get ();
 433         guint8 *code, *buf;
 434         static guint8 *vc = NULL;
 435         GHashTable *jit_code_hash;
 436
 437         /* icalls use method->addr */
 438         if (method->addr)
 439                 return method->addr;
 440
 441         /* previously created trampoline code */
 442         if (method->info)
 443                 return method->info;
 444
 445         /* check if we already have JITed code */
 446         if (mono_jit_share_code)
 447                 jit_code_hash = mono_root_domain->jit_code_hash;
 448         else
 449                 jit_code_hash = domain->jit_code_hash;
 450
 451         if ((code = g_hash_table_lookup (jit_code_hash, method))) {
 452                 mono_jit_stats.methods_lookups++;
 453                 return code;
 454         }
 455
 456         if (!vc) {
 457                 vc = buf = g_malloc (256);
 458
 459                 /* save caller save regs because we need to do a call */
 460                 x86_push_reg (buf, X86_EDX);
 461                 x86_push_reg (buf, X86_EAX);
 462                 x86_push_reg (buf, X86_ECX);
 463
 464                 /* save LMF begin */
 465                 x86_push_reg (buf, X86_EBX);
 466                 x86_push_reg (buf, X86_EDI);
 467                 x86_push_reg (buf, X86_ESI);
 468                 x86_push_reg (buf, X86_EBP);
 469
 470                 /* save the IP (caller ip) */
 471                 x86_push_membase (buf, X86_ESP, 32);
 472
 473                 /* save method info */
 474                 x86_push_membase (buf, X86_ESP, 32);
 475                 /* get the address of lmf for the current thread */
 476                 x86_call_code (buf, arch_get_lmf_addr);
 477                 /* push lmf */
 478                 x86_push_reg (buf, X86_EAX);
 479                 /* push *lfm (previous_lmf) */
 480                 x86_push_membase (buf, X86_EAX, 0);
 481                 /* *(lmf) = ESP */
 482                 x86_mov_membase_reg (buf, X86_EAX, 0, X86_ESP, 4);
 483                 /* save LFM end */
 484
 485                 /* push the method info */
 486                 x86_push_membase (buf, X86_ESP, 44);
 487                 /* push the return address onto the stack */
 488                 x86_push_membase (buf, X86_ESP, 52);
 489
 490                 /* save all register values */
 491                 x86_push_reg (buf, X86_EBX);
 492                 x86_push_reg (buf, X86_EDI);
 493                 x86_push_reg (buf, X86_ESI);
 494                 x86_push_membase (buf, X86_ESP, 64); /* EDX */
 495                 x86_push_membase (buf, X86_ESP, 64); /* ECX */
 496                 x86_push_membase (buf, X86_ESP, 64); /* EAX */
 497
 498                 x86_call_code (buf, x86_magic_trampoline);
 499                 x86_alu_reg_imm (buf, X86_ADD, X86_ESP, 8*4);
 500
 501                 /* restore LMF start */
 502                 /* ebx = previous_lmf */
 503                 x86_pop_reg (buf, X86_EBX);
 504                 /* edi = lmf */
 505                 x86_pop_reg (buf, X86_EDI);
 506                 /* *(lmf) = previous_lmf */
 507                 x86_mov_membase_reg (buf, X86_EDI, 0, X86_EBX, 4);
 508                 /* discard method info */
 509                 x86_pop_reg (buf, X86_ESI);
 510                 /* discard save IP */
 511                 x86_pop_reg (buf, X86_ESI);
 512                 /* restore caller saved regs */
 513                 x86_pop_reg (buf, X86_EBP);
 514                 x86_pop_reg (buf, X86_ESI);
 515                 x86_pop_reg (buf, X86_EDI);
 516                 x86_pop_reg (buf, X86_EBX);
 517                 /* restore LMF end */
 518
 519                 x86_alu_reg_imm (buf, X86_ADD, X86_ESP, 16);
 520
 521                 /* call the compiled method */
 522                 x86_jump_reg (buf, X86_EAX);
 523
 524                 g_assert ((buf - vc) <= 256);
 525         }
 526
 527         code = buf = g_malloc (16);
 528         x86_push_imm (buf, method);
 529         x86_jump_code (buf, vc);
 530         g_assert ((buf - code) <= 16);
 531
 532         /* store trampoline address */
 533         method->info = code;
 534
 535         mono_jit_stats.method_trampolines++;
 536
 537         return code;
 538 }
 539
 540 static void
 541 mono_label_cfg (MonoFlowGraph *cfg)
 542 {
 543         int i, j;
 544
 545         for (i = 0; i < cfg->block_count; i++) {
 546                 GPtrArray *forest = cfg->bblocks [i].forest;
 547                 int top;
 548
 549                 if (!cfg->bblocks [i].reached) /* unreachable code */
 550                         continue;
 551
 552                 top = forest->len;
 553
 554                 for (j = 0; j < top; j++) {
 555                         MBTree *t1 = (MBTree *) g_ptr_array_index (forest, j);
 556                         MBState *mbstate;
 557
 558                         mbstate =  mono_burg_label (t1, cfg);
 559                         if (!mbstate) {
 560                                 cfg->invalid = 1;
 561                                 if (mono_debug_handle)
 562                                         return;
 563                                 g_warning ("tree does not match");
 564                                 mono_print_ctree (t1); printf ("\n\n");
 565
 566                                 mono_print_forest (forest);
 567                                 g_assert_not_reached ();
 568                         }
 569                 }
 570         }
 571 }
 572
 573 static void
 574 tree_preallocate_regs (MBTree *tree, int goal, MonoRegSet *rs)
 575 {
 576         switch (tree->op) {
 577         case MB_TERM_CALL_I4:
 578         case MB_TERM_CALL_I8:
 579         case MB_TERM_CALL_R8:
 580 //      case MB_TERM_CALL_VOID :
 581                 tree->reg1 = mono_regset_alloc_reg (rs, X86_EAX, tree->exclude_mask);
 582                 tree->reg2 = mono_regset_alloc_reg (rs, X86_EDX, tree->exclude_mask);
 583                 tree->reg3 = mono_regset_alloc_reg (rs, X86_ECX, tree->exclude_mask);
 584                 return;
 585         default: break;
 586         }
 587
 588         switch (goal) {
 589         case MB_NTERM_reg:
 590         case MB_NTERM_lreg: {
 591                 switch (tree->op) {
 592                 case MB_TERM_SHL:
 593                 case MB_TERM_SHR:
 594                 case MB_TERM_SHR_UN:
 595                         tree->exclude_mask |= (1 << X86_ECX);
 596                         tree->left->exclude_mask |= (1 << X86_ECX);
 597                         break;
 598                 case MB_TERM_MUL:
 599                 case MB_TERM_MUL_OVF:
 600                 case MB_TERM_MUL_OVF_UN:
 601                 case MB_TERM_DIV:
 602                 case MB_TERM_DIV_UN:
 603                 case MB_TERM_REM:
 604                 case MB_TERM_REM_UN:
 605                         tree->reg1 = mono_regset_alloc_reg (rs, X86_EAX, tree->exclude_mask);
 606                         tree->reg2 = mono_regset_alloc_reg (rs, X86_EDX, tree->exclude_mask);
 607                         if (goal == MB_NTERM_reg) {
 608                                 tree->left->exclude_mask |= (1 << X86_EDX);
 609                                 tree->right->exclude_mask |= (1 << X86_EDX) | (1 << X86_EAX);
 610                         }
 611                         break;
 612                 default:
 613                         break;
 614                 }
 615                 break;
 616         }
 617         default:
 618                 break;
 619         }
 620 }
 621
 622 static void
 623 tree_allocate_regs (MBTree *tree, int goal, MonoRegSet *rs)
 624 {
 625         MBTree *kids[10];
 626         int ern = mono_burg_rule (tree->state, goal);
 627         const guint16 *nts = mono_burg_nts [ern];
 628         int i;
 629
 630         mono_burg_kids (tree, ern, kids);
 631
 632         //printf ("RALLOC START %d %p %d\n",  tree->op, rs->free_mask, goal);
 633
 634         if (nts [0] && kids [0] == tree) {
 635                 /* chain rule */
 636                 tree_allocate_regs (kids [0], nts [0], rs);
 637                 return;
 638         }
 639
 640         for (i = 0; nts [i]; i++)
 641                 tree_preallocate_regs (kids [i], nts [i], rs);
 642
 643         for (i = 0; nts [i]; i++)
 644                 tree_allocate_regs (kids [i], nts [i], rs);
 645
 646         for (i = 0; nts [i]; i++) {
 647                 mono_regset_free_reg (rs, kids [i]->reg1);
 648                 mono_regset_free_reg (rs, kids [i]->reg2);
 649                 mono_regset_free_reg (rs, kids [i]->reg3);
 650         }
 651
 652         switch (goal) {
 653         case MB_NTERM_reg:
 654                 if (tree->reg1 < 0) {
 655                         tree->reg1 = mono_regset_alloc_reg (rs, -1, tree->exclude_mask);
 656                         g_assert (tree->reg1 != -1);
 657                 }
 658                 break;
 659
 660         case MB_NTERM_lreg:
 661                 if (tree->reg1 < 0) {
 662                         tree->reg1 = mono_regset_alloc_reg (rs, -1, tree->exclude_mask);
 663                         g_assert (tree->reg1 != -1);
 664                 }
 665                 if (tree->reg2 < 0) {
 666                         tree->reg2 = mono_regset_alloc_reg (rs, -1, tree->exclude_mask);
 667                         g_assert (tree->reg2 != -1);
 668                 }
 669                 break;
 670
 671         case MB_NTERM_freg:
 672                 /* fixme: allocate floating point registers */
 673                 break;
 674
 675         case MB_NTERM_addr:
 676                 if (tree->op == MB_TERM_ADD) {
 677                         tree->reg1 = mono_regset_alloc_reg (rs, tree->left->reg1, tree->exclude_mask);
 678                         tree->reg2 = mono_regset_alloc_reg (rs, tree->right->reg1, tree->exclude_mask);
 679                 }
 680                 break;
 681
 682         case MB_NTERM_base:
 683                 if (tree->op == MB_TERM_ADD) {
 684                         tree->reg1 = mono_regset_alloc_reg (rs, tree->left->reg1, tree->exclude_mask);
 685                 }
 686                 break;
 687
 688         case MB_NTERM_index:
 689                 if (tree->op == MB_TERM_SHL ||
 690                     tree->op == MB_TERM_MUL) {
 691                         tree->reg1 = mono_regset_alloc_reg (rs, tree->left->reg1, tree->exclude_mask);
 692                 }
 693                 break;
 694
 695         default:
 696                 /* do nothing */
 697         }
 698
 699         //printf ("RALLOC END %d %p\n",  tree->op, rs->free_mask);
 700         tree->emit = mono_burg_func [ern];
 701 }
 702
 703 static void
 704 arch_allocate_regs (MonoFlowGraph *cfg)
 705 {
 706         int i, j;
 707
 708         for (i = 0; i < cfg->block_count; i++) {
 709                 GPtrArray *forest = cfg->bblocks [i].forest;
 710                 int top;
 711
 712                 if (!cfg->bblocks [i].reached) /* unreachable code */
 713                         continue;
 714
 715                 top = forest->len;
 716
 717                 for (j = 0; j < top; j++) {
 718                         MBTree *t1 = (MBTree *) g_ptr_array_index (forest, j);
 719                         //printf ("AREGSTART %d:%d %p\n", i, j, cfg->rs->free_mask);
 720                         tree_allocate_regs (t1, 1, cfg->rs);
 721                         //printf ("AREGENDT %d:%d %p\n", i, j, cfg->rs->free_mask);
 722                         g_assert (cfg->rs->free_mask == 0xffffffff);
 723                 }
 724         }
 725 }
 726
 727 static void
 728 tree_emit (int goal, MonoFlowGraph *cfg, MBTree *tree)
 729 {
 730         MBTree *kids[10];
 731         int i, ern = mono_burg_rule (tree->state, goal);
 732         const guint16 *nts = mono_burg_nts [ern];
 733         MBEmitFunc emit;
 734         int offset;
 735
 736         mono_burg_kids (tree, ern, kids);
 737
 738         for (i = 0; nts [i]; i++)
 739                 tree_emit (nts [i], cfg, kids [i]);
 740
 741         tree->addr = offset = cfg->code - cfg->start;
 742
 743         // we assume an instruction uses a maximum of 128 bytes
 744         if ((cfg->code_size - offset) <= 128) {
 745                 int add = MIN (cfg->code_size, 128);
 746                 cfg->code_size += add;
 747                 mono_jit_stats.code_reallocs++;
 748                 cfg->start = g_realloc (cfg->start, cfg->code_size);
 749                 g_assert (cfg->start);
 750                 cfg->code = cfg->start + offset;
 751         }
 752
 753         if ((emit = mono_burg_func [ern]))
 754                 emit (tree, cfg);
 755
 756         g_assert ((cfg->code - cfg->start) < cfg->code_size);
 757 }
 758
 759 static void
 760 mono_emit_cfg (MonoFlowGraph *cfg)
 761 {
 762         int i, j;
 763
 764         for (i = 0; i < cfg->block_count; i++) {
 765                 MonoBBlock *bb = &cfg->bblocks [i];
 766                 GPtrArray *forest = bb->forest;
 767                 int top;
 768
 769                 if (!bb->reached) /* unreachable code */
 770                         continue;
 771
 772                 top = forest->len;
 773
 774                 bb->addr = cfg->code - cfg->start;
 775
 776                 for (j = 0; j < top; j++) {
 777                         MBTree *t1 = (MBTree *) g_ptr_array_index (forest, j);
 778
 779                         tree_emit (1, cfg, t1);
 780                 }
 781         }
 782
 783         cfg->epilog = cfg->code - cfg->start;
 784 }
 785
 786 static void
 787 mono_compute_branches (MonoFlowGraph *cfg)
 788 {
 789         MonoJumpInfo *ji;
 790         guint8 *end;
 791         int i, j;
 792
 793         end = cfg->code;
 794
 795         for (j = 0; j < cfg->block_count; j++) {
 796                 MonoBBlock *bb = &cfg->bblocks [j];
 797                 GPtrArray *forest = bb->forest;
 798                 int top;
 799
 800                 if (!bb->reached) /* unreachable code */
 801                         continue;
 802
 803                 top = forest->len;
 804
 805                 for (i = 0; i < top; i++) {
 806                         MBTree *t1 = (MBTree *) g_ptr_array_index (forest, i);
 807
 808                         if (t1->op == MB_TERM_SWITCH) {
 809                                 MonoBBlock **jt = (MonoBBlock **)t1->data.p;
 810                                 guint32 *rt = (guint32 *)t1->data.p;
 811
 812                                 int m = *((guint32 *)t1->data.p) + 1;
 813                                 int j;
 814
 815                                 for (j = 1; j <= m; j++)
 816                                         rt [j] = (int)(jt [j]->addr + cfg->start);
 817
 818                                 /* emit the switch instruction again to update addresses */
 819                                 cfg->code = cfg->start + t1->addr;
 820                                 ((MBEmitFunc)t1->emit) (t1, cfg);
 821                         }
 822                 }
 823         }
 824
 825         cfg->code = end;
 826
 827         for (ji = cfg->jump_info; ji; ji = ji->next) {
 828                 gpointer *ip = GUINT_TO_POINTER (GPOINTER_TO_UINT (ji->ip) + cfg->start);
 829                 gpointer target;
 830
 831                 if (ji->target) {
 832                         target = ji->target;
 833                 } else if (ji->bb) {
 834                         target = ji->bb->addr + cfg->start;
 835                 } else {
 836                         /* special case - jump to epilog */
 837                         target = cfg->epilog + cfg->start;
 838                 }
 839                 *ip = target - GPOINTER_TO_UINT(ip) - 4;
 840         }
 841 }
 842
 843 void
 844 mono_add_jump_info (MonoFlowGraph *cfg, gpointer ip, gpointer target, MonoBBlock *bb)
 845 {
 846         MonoJumpInfo *ji = mono_mempool_alloc (cfg->mp, sizeof (MonoJumpInfo));
 847
 848         ji->ip = GUINT_TO_POINTER (GPOINTER_TO_UINT (ip) - GPOINTER_TO_UINT (cfg->start));
 849         ji->target = target;
 850         ji->next = cfg->jump_info;
 851         ji->bb = bb;
 852
 853         cfg->jump_info = ji;
 854 }
 855
 856 static int
 857 match_debug_method (MonoMethod* method)
 858 {
 859         GList *tmp = mono_debug_methods;
 860
 861         for (; tmp; tmp = tmp->next) {
 862                 if (strcmp (method->name, tmp->data) == 0) {
 863                         return 1;
 864                 }
 865         }
 866         return 0;
 867 }
 868
 869 /**
 870  * arch_compile_method:
 871  * @method: pointer to the method info
 872  *
 873  * JIT compilation of a single method.
 874  *
 875  * Returns: a pointer to the newly created code.
 876  */
 877 gpointer
 878 arch_compile_method (MonoMethod *method)
 879 {
 880         MonoDomain *domain = mono_domain_get ();
 881         MonoFlowGraph *cfg;
 882         MonoMemPool *mp;
 883         guint8 *addr;
 884         GHashTable *jit_code_hash;
 885
 886         g_assert (!(method->iflags & METHOD_IMPL_ATTRIBUTE_INTERNAL_CALL));
 887         g_assert (!(method->flags & METHOD_ATTRIBUTE_PINVOKE_IMPL));
 888
 889         if (mono_jit_share_code)
 890                 jit_code_hash = mono_root_domain->jit_code_hash;
 891         else
 892                 jit_code_hash = domain->jit_code_hash;
 893
 894         if ((addr = g_hash_table_lookup (jit_code_hash, method))) {
 895                 mono_jit_stats.methods_lookups++;
 896                 return addr;
 897         }
 898
 899         mono_jit_stats.methods_compiled++;
 900
 901         mp = mono_mempool_new ();
 902
 903         if (mono_jit_trace_calls || mono_jit_dump_asm || mono_jit_dump_forest) {
 904                 printf ("Start JIT compilation of %s.%s:%s\n", method->klass->name_space,
 905                         method->klass->name, method->name);
 906         }
 907
 908         if (method->iflags & METHOD_IMPL_ATTRIBUTE_RUNTIME) {
 909                 MonoClassField *field;
 910                 const char *name = method->name;
 911                 static guint target_offset = 0;
 912                 static guint method_offset = 0;
 913                 guint8 *code;
 914                 gboolean delegate = FALSE;
 915
 916                 if (method->klass->parent &&
 917                     method->klass->parent->parent == mono_defaults.delegate_class)
 918                         delegate = TRUE;
 919
 920                 if (!target_offset) {
 921                         mono_class_init (mono_defaults.delegate_class);
 922
 923                         field = mono_class_get_field_from_name (mono_defaults.delegate_class, "m_target");
 924                         target_offset = field->offset;
 925                         field = mono_class_get_field_from_name (mono_defaults.delegate_class, "method_ptr");
 926                         method_offset = field->offset;
 927                 }
 928
 929                 if (delegate && *name == '.' && (strcmp (name, ".ctor") == 0)) {
 930                         addr = code = g_malloc (32);
 931                         x86_push_reg (code, X86_EBP);
 932                         x86_mov_reg_reg (code, X86_EBP, X86_ESP, 4);
 933
 934                         /* load the this pointer */
 935                         x86_mov_reg_membase (code, X86_EAX, X86_EBP, 8, 4);
 936                         /* load m_target arg */
 937                         x86_mov_reg_membase (code, X86_EDX, X86_EBP, 12, 4);
 938                         /* store mtarget */
 939                         x86_mov_membase_reg (code, X86_EAX, target_offset, X86_EDX, 4);
 940                         /* load method_ptr arg */
 941                         x86_mov_reg_membase (code, X86_EDX, X86_EBP, 16, 4);
 942                         /* store method_ptr */
 943                         x86_mov_membase_reg (code, X86_EAX, method_offset, X86_EDX, 4);
 944
 945                         x86_leave (code);
 946                         x86_ret (code);
 947
 948                         g_assert ((code - (guint8*)addr) < 32);
 949
 950                 } else if (delegate && *name == 'I' && (strcmp (name, "Invoke") == 0)) {
 951                         MonoMethodSignature *csig = method->signature;
 952                         int i, arg_size, target, this_pos = 4;
 953                         guint8 *source;
 954
 955                         if (csig->ret->type == MONO_TYPE_VALUETYPE) {
 956                                 g_assert (!csig->ret->byref);
 957                                 this_pos = 8;
 958                         }
 959
 960                         arg_size = 0;
 961                         if (csig->param_count) {
 962                                 int align;
 963
 964                                 for (i = 0; i < csig->param_count; ++i) {
 965                                         arg_size += mono_type_stack_size (csig->params [i], &align);
 966                                         g_assert (align == 4);
 967                                 }
 968                         }
 969
 970                         addr = g_malloc (64 + arg_size);
 971
 972                         for (i = 0; i < 2; i ++) {
 973                                 int j;
 974
 975                                 code = addr;
 976                                 /* load the this pointer */
 977                                 x86_mov_reg_membase (code, X86_EAX, X86_ESP, this_pos, 4);
 978                                 /* load mtarget */
 979                                 x86_mov_reg_membase (code, X86_EDX, X86_EAX, target_offset, 4);
 980                                 /* check if zero (static method call without this pointer) */
 981                                 x86_alu_reg_imm (code, X86_CMP, X86_EDX, 0);
 982                                 x86_branch32 (code, X86_CC_EQ, target, TRUE);
 983                                 source = code;
 984
 985                                 /* virtual delegate methods: we have to replace the this pointer
 986                                  * withe the actual target */
 987                                 x86_mov_membase_reg (code, X86_ESP, this_pos, X86_EDX, 4);
 988                                 /* jump to method_ptr() */
 989                                 x86_jump_membase (code, X86_EAX, method_offset);
 990
 991                                 /* static delegate methods: we have to remove the this pointer
 992                                  * from the activation frame - I do this do creating a new
 993                                  * stack frame an copy all arguments except the this pointer */
 994
 995                                 target = code - source;
 996                                 g_assert ((arg_size & 3) == 0);
 997                                 for (j = 0; j < (arg_size>>2); j++) {
 998                                         x86_push_membase (code, X86_ESP, (arg_size + this_pos));
 999                                 }
1000
1001                                 if (this_pos == 8)
1002                                         x86_push_membase (code, X86_ESP, (arg_size + 4));
1003
1004                                 x86_call_membase (code, X86_EAX, method_offset);
1005                                 if (this_pos == 8)
1006                                         x86_alu_reg_imm (code, X86_ADD, X86_ESP, arg_size + 4);
1007                                 else
1008                                         x86_alu_reg_imm (code, X86_ADD, X86_ESP, arg_size);
1009
1010                                 x86_ret (code);
1011
1012                         }
1013
1014                         g_assert ((code - (guint8*)addr) < (64 + arg_size));
1015
1016                 } else {
1017                         if (mono_debug_handle)
1018                                 return NULL;
1019
1020                         g_error ("Don't know how to exec runtime method %s.%s::%s",
1021                                  method->klass->name_space, method->klass->name, method->name);
1022                 }
1023
1024         } else {
1025                 MonoMethodHeader *header = ((MonoMethodNormal *)method)->header;
1026                 MonoJitInfo *ji = g_new0 (MonoJitInfo, 1);
1027                 gulong code_size_ratio;
1028
1029                 cfg = mono_cfg_new (method, mp);
1030
1031                 mono_analyze_flow (cfg);
1032                 if (cfg->invalid)
1033                         return NULL;
1034
1035                 mono_analyze_stack (cfg);
1036                 if (cfg->invalid)
1037                         return NULL;
1038
1039                 cfg->rs = mono_regset_new (X86_NREG);
1040                 mono_regset_reserve_reg (cfg->rs, X86_ESP);
1041                 mono_regset_reserve_reg (cfg->rs, X86_EBP);
1042
1043                 cfg->code_size = MAX (header->code_size * 5, 256);
1044                 cfg->start = cfg->code = g_malloc (cfg->code_size);
1045
1046                 if (match_debug_method (method))
1047                         x86_breakpoint (cfg->code);
1048
1049                 if (mono_jit_dump_forest) {
1050                         int i;
1051                         printf ("FOREST %s.%s:%s\n", method->klass->name_space,
1052                                 method->klass->name, method->name);
1053                         for (i = 0; i < cfg->block_count; i++) {
1054                                 printf ("BLOCK %d:\n", i);
1055                                 mono_print_forest (cfg->bblocks [i].forest);
1056                         }
1057                 }
1058
1059                 mono_label_cfg (cfg);
1060                 if (cfg->invalid)
1061                         return NULL;
1062
1063                 arch_allocate_regs (cfg);
1064
1065                 /* align to 8 byte boundary */
1066                 cfg->locals_size += 7;
1067                 cfg->locals_size &= ~7;
1068
1069                 arch_emit_prologue (cfg);
1070                 mono_emit_cfg (cfg);
1071                 arch_emit_epilogue (cfg);
1072
1073                 addr = cfg->start;
1074
1075                 mono_jit_stats.allocated_code_size += cfg->code_size;
1076
1077                 code_size_ratio = cfg->code - cfg->start;
1078                 if (code_size_ratio > mono_jit_stats.biggest_method_size) {
1079                         mono_jit_stats.biggest_method_size = code_size_ratio;
1080                         mono_jit_stats.biggest_method = method;
1081                 }
1082                 code_size_ratio = (code_size_ratio * 100) / header->code_size;
1083                 if (code_size_ratio > mono_jit_stats.max_code_size_ratio) {
1084                         mono_jit_stats.max_code_size_ratio = code_size_ratio;
1085                         mono_jit_stats.max_ratio_method = method;
1086                 }
1087
1088                 mono_compute_branches (cfg);
1089
1090                 if (mono_jit_dump_asm) {
1091                         char *id = g_strdup_printf ("%s.%s_%s", method->klass->name_space,
1092                                                     method->klass->name, method->name);
1093                         mono_disassemble_code (cfg->start, cfg->code - cfg->start, id);
1094                         g_free (id);
1095                 }
1096                 if (mono_debug_handle)
1097                         mono_debug_add_method (mono_debug_handle, cfg);
1098
1099                 ji->code_size = cfg->code - cfg->start;
1100                 ji->used_regs = cfg->rs->used_mask;
1101                 ji->method = method;
1102                 ji->code_start = addr;
1103                 mono_jit_info_table_add (mono_jit_info_table, ji);
1104
1105                 mono_jit_stats.native_code_size += ji->code_size;
1106
1107                 if (header->num_clauses) {
1108                         int i, start_block, end_block;
1109
1110                         ji->num_clauses = header->num_clauses;
1111                         ji->clauses = g_new0 (MonoJitExceptionInfo, header->num_clauses);
1112
1113                         for (i = 0; i < header->num_clauses; i++) {
1114                                 MonoExceptionClause *ec = &header->clauses [i];
1115                                 MonoJitExceptionInfo *ei = &ji->clauses [i];
1116
1117                                 ei->flags = ec->flags;
1118                                 ei->token_or_filter = ec->token_or_filter;
1119
1120                                 g_assert (cfg->bcinfo [ec->try_offset].is_block_start);
1121                                 start_block = cfg->bcinfo [ec->try_offset].block_id;
1122                                 end_block = cfg->bcinfo [ec->try_offset + ec->try_len].block_id;
1123                                 g_assert (cfg->bcinfo [ec->try_offset + ec->try_len].is_block_start);
1124
1125                                 ei->try_start = cfg->start + cfg->bblocks [start_block].addr;
1126                                 ei->try_end = cfg->start + cfg->bblocks [end_block].addr;
1127
1128                                 g_assert (cfg->bcinfo [ec->handler_offset].is_block_start);
1129                                 start_block = cfg->bcinfo [ec->handler_offset].block_id;
1130                                 ei->handler_start = cfg->start + cfg->bblocks [start_block].addr;
1131
1132                                 //printf ("TEST %x %x %x\n", ei->try_start, ei->try_end, ei->handler_start);
1133                         }
1134                 }
1135
1136                 mono_regset_free (cfg->rs);
1137
1138                 mono_cfg_free (cfg);
1139
1140                 mono_mempool_destroy (mp);
1141
1142         }
1143
1144         if (mono_jit_trace_calls || mono_jit_dump_asm || mono_jit_dump_forest) {
1145                 printf ("END JIT compilation of %s.%s:%s %p %p\n", method->klass->name_space,
1146                         method->klass->name, method->name, method, addr);
1147         }
1148
1149         g_hash_table_insert (jit_code_hash, method, addr);
1150
1151         return addr;
1152 }
1153
1154 /*
1155  * arch_get_restore_context:
1156  *
1157  * Returns a pointer to a method which restores a previously saved sigcontext.
1158  */
1159 static gpointer
1160 arch_get_restore_context ()
1161 {
1162         static guint8 *start = NULL;
1163         guint8 *code;
1164
1165         if (start)
1166                 return start;
1167
1168         /* restore_contect (struct sigcontext *ctx) */
1169         /* we do not restore X86_EAX, X86_EDX */
1170
1171         start = code = malloc (1024);
1172
1173         /* load ctx */
1174         x86_mov_reg_membase (code, X86_EAX, X86_ESP, 4, 4);
1175
1176         /* get return address, stored in EDX */
1177         x86_mov_reg_membase (code, X86_EDX, X86_EAX,  G_STRUCT_OFFSET (struct sigcontext, eip), 4);
1178
1179         /* restore EBX */
1180         x86_mov_reg_membase (code, X86_EBX, X86_EAX,  G_STRUCT_OFFSET (struct sigcontext, ebx), 4);
1181         /* restore EDI */
1182         x86_mov_reg_membase (code, X86_EDI, X86_EAX,  G_STRUCT_OFFSET (struct sigcontext, edi), 4);
1183         /* restore ESI */
1184         x86_mov_reg_membase (code, X86_ESI, X86_EAX,  G_STRUCT_OFFSET (struct sigcontext, esi), 4);
1185         /* restore ESP */
1186         x86_mov_reg_membase (code, X86_ESP, X86_EAX,  G_STRUCT_OFFSET (struct sigcontext, esp), 4);
1187         /* restore EBP */
1188         x86_mov_reg_membase (code, X86_EBP, X86_EAX,  G_STRUCT_OFFSET (struct sigcontext, ebp), 4);
1189         /* restore ECX. the exception object is passed here to the catch handler */
1190         x86_mov_reg_membase (code, X86_ECX, X86_EAX,  G_STRUCT_OFFSET (struct sigcontext, ecx), 4);
1191
1192         /* jump to the saved IP */
1193         x86_jump_reg (code, X86_EDX);
1194
1195         return start;
1196 }
1197
1198 /*
1199  * arch_get_call_finally:
1200  *
1201  * Returns a pointer to a method which calls a finally handler.
1202  */
1203 static gpointer
1204 arch_get_call_finally ()
1205 {
1206         static guint8 start [28];
1207         static int inited = 0;
1208         guint8 *code;
1209
1210         if (inited)
1211                 return start;
1212
1213         inited = 1;
1214         /* call_finally (struct sigcontext *ctx, unsigned long eip) */
1215         code = start;
1216
1217         x86_push_reg (code, X86_EBP);
1218         x86_mov_reg_reg (code, X86_EBP, X86_ESP, 4);
1219         x86_push_reg (code, X86_EBX);
1220         x86_push_reg (code, X86_EDI);
1221         x86_push_reg (code, X86_ESI);
1222
1223         /* load ctx */
1224         x86_mov_reg_membase (code, X86_EAX, X86_EBP, 8, 4);
1225         /* load eip */
1226         x86_mov_reg_membase (code, X86_ECX, X86_EBP, 12, 4);
1227         /* save EBP */
1228         x86_push_reg (code, X86_EBP);
1229         /* set new EBP */
1230         x86_mov_reg_membase (code, X86_EBP, X86_EAX,  G_STRUCT_OFFSET (struct sigcontext, ebp), 4);
1231         /* call the handler */
1232         x86_call_reg (code, X86_ECX);
1233         /* restore EBP */
1234         x86_pop_reg (code, X86_EBP);
1235         /* restore saved regs */
1236         x86_pop_reg (code, X86_ESI);
1237         x86_pop_reg (code, X86_EDI);
1238         x86_pop_reg (code, X86_EBX);
1239         x86_leave (code);
1240         x86_ret (code);
1241
1242         g_assert ((code - start) < 28);
1243         return start;
1244 }
1245
1246 /**
1247  * arch_handle_exception:
1248  * @ctx: saved processor state
1249  * @obj:
1250  */
1251 void
1252 arch_handle_exception (struct sigcontext *ctx, gpointer obj)
1253 {
1254         MonoDomain *domain = mono_domain_get ();
1255         MonoJitInfo *ji;
1256         gpointer ip = (gpointer)ctx->eip;
1257         static void (*restore_context) (struct sigcontext *);
1258         static void (*call_finally) (struct sigcontext *, unsigned long);
1259
1260         g_assert (ctx != NULL);
1261         g_assert (obj != NULL);
1262
1263         ji = mono_jit_info_table_find (mono_jit_info_table, ip);
1264
1265         if (!restore_context)
1266                 restore_context = arch_get_restore_context ();
1267
1268         if (!call_finally)
1269                 call_finally = arch_get_call_finally ();
1270
1271         if (ji) { /* we are inside managed code */
1272                 MonoMethod *m = ji->method;
1273                 int offset = 2;
1274
1275                 if (ji->num_clauses) {
1276                         int i;
1277
1278                         g_assert (ji->clauses);
1279
1280                         for (i = 0; i < ji->num_clauses; i++) {
1281                                 MonoJitExceptionInfo *ei = &ji->clauses [i];
1282
1283                                 if (ei->try_start <= ip && ip <= (ei->try_end)) {
1284                                         /* catch block */
1285                                         if (ei->flags == 0 && mono_object_isinst (obj,
1286                                                 mono_class_get (m->klass->image, ei->token_or_filter))) {
1287
1288                                                 ctx->eip = (unsigned long)ei->handler_start;
1289                                                 ctx->ecx = (unsigned long)obj;
1290                                                 restore_context (ctx);
1291                                                 g_assert_not_reached ();
1292                                         }
1293                                 }
1294                         }
1295
1296                         /* no handler found - we need to call all finally handlers */
1297                         for (i = 0; i < ji->num_clauses; i++) {
1298                                 MonoJitExceptionInfo *ei = &ji->clauses [i];
1299
1300                                 if (ei->try_start <= ip && ip < (ei->try_end) &&
1301                                     (ei->flags & MONO_EXCEPTION_CLAUSE_FINALLY)) {
1302                                         call_finally (ctx, (unsigned long)ei->handler_start);
1303                                 }
1304                         }
1305                 }
1306
1307                 if (mono_object_isinst (obj, mono_defaults.exception_class)) {
1308                         char  *strace = mono_string_to_utf8 (((MonoException*)obj)->stack_trace);
1309                         char  *tmp;
1310
1311                         if (!strcmp (strace, "TODO: implement stack traces")){
1312                                 g_free (strace);
1313                                 strace = g_strdup ("");
1314                         }
1315
1316                         tmp = g_strdup_printf ("%sin %s.%s:%s ()\n", strace, m->klass->name_space,
1317                                                m->klass->name, m->name);
1318
1319                         g_free (strace);
1320
1321                         ((MonoException*)obj)->stack_trace = mono_string_new (domain, tmp);
1322                         g_free (tmp);
1323                 }
1324
1325                 /* continue unwinding */
1326
1327                 /* restore caller saved registers */
1328                 if (ji->used_regs & X86_ESI_MASK) {
1329                         ctx->esi = *((int *)ctx->ebp + offset);
1330                         offset++;
1331                 }
1332                 if (ji->used_regs & X86_EDI_MASK) {
1333                         ctx->edi = *((int *)ctx->ebp + offset);
1334                         offset++;
1335                 }
1336                 if (ji->used_regs & X86_EBX_MASK) {
1337                         ctx->ebx = *((int *)ctx->ebp + offset);
1338                 }
1339
1340                 ctx->esp = ctx->ebp;
1341                 ctx->eip = *((int *)ctx->ebp + 1);
1342                 ctx->ebp = *((int *)ctx->ebp);
1343
1344                 if (ctx->ebp < (unsigned)mono_end_of_stack)
1345                         arch_handle_exception (ctx, obj);
1346                 else
1347                         mono_jit_abort (obj);
1348         } else {
1349                 gpointer *lmf_addr = TlsGetValue (lmf_thread_id);
1350                 MonoLMF *lmf;
1351                 MonoMethod *m;
1352
1353                 g_assert (lmf_addr);
1354                 lmf = *((MonoLMF **)lmf_addr);
1355
1356                 if (!lmf)
1357                         mono_jit_abort (obj);
1358
1359                 m = lmf->method;
1360
1361                 *lmf_addr = lmf->previous_lmf;
1362
1363                 ctx->esi = lmf->esi;
1364                 ctx->edi = lmf->edi;
1365                 ctx->ebx = lmf->ebx;
1366                 ctx->ebp = lmf->ebp;
1367                 ctx->eip = lmf->eip;
1368                 ctx->esp = (unsigned long)lmf;
1369
1370                 if (mono_object_isinst (obj, mono_defaults.exception_class)) {
1371                         char  *strace = mono_string_to_utf8 (((MonoException*)obj)->stack_trace);
1372                         char  *tmp;
1373
1374                         if (!strcmp (strace, "TODO: implement stack traces"))
1375                                 strace = g_strdup ("");
1376
1377                         tmp = g_strdup_printf ("%sin (unmanaged) %s.%s:%s ()\n", strace, m->klass->name_space,
1378                                                m->klass->name, m->name);
1379
1380                         g_free (strace);
1381
1382                         ((MonoException*)obj)->stack_trace = mono_string_new (domain, tmp);
1383                         g_free (tmp);
1384                 }
1385
1386                 if (ctx->eip < (unsigned)mono_end_of_stack)
1387                         arch_handle_exception (ctx, obj);
1388                 else
1389                         mono_jit_abort (obj);
1390         }
1391
1392         g_assert_not_reached ();
1393 }
1394
1395 static void
1396 throw_exception (unsigned long eax, unsigned long ecx, unsigned long edx, unsigned long ebx,
1397                  unsigned long esi, unsigned long edi, unsigned long ebp, MonoObject *exc,
1398                  unsigned long eip,  unsigned long esp)
1399 {
1400         struct sigcontext ctx;
1401
1402         ctx.esp = esp;
1403         ctx.eip = eip;
1404         ctx.ebp = ebp;
1405         ctx.edi = edi;
1406         ctx.esi = esi;
1407         ctx.ebx = ebx;
1408         ctx.edx = edx;
1409         ctx.ecx = ecx;
1410         ctx.eax = eax;
1411
1412         arch_handle_exception (&ctx, exc);
1413
1414         g_assert_not_reached ();
1415 }
1416
1417 /**
1418  * arch_get_throw_exception:
1419  *
1420  * Returns a function pointer which can be used to raise
1421  * exceptions. The returned function has the following
1422  * signature: void (*func) (MonoException *exc);
1423  * For example to raise an arithmetic exception you can use:
1424  *
1425  * x86_push_imm (code, mono_get_exception_arithmetic ());
1426  * x86_call_code (code, arch_get_throw_exception ());
1427  *
1428  */
1429 gpointer
1430 arch_get_throw_exception (void)
1431 {
1432         static guint8 start [24];
1433         static int inited = 0;
1434         guint8 *code;
1435
1436         if (inited)
1437                 return start;
1438
1439         inited = 1;
1440         code = start;
1441
1442         x86_push_reg (code, X86_ESP);
1443         x86_push_membase (code, X86_ESP, 4); /* IP */
1444         x86_push_membase (code, X86_ESP, 12); /* exception */
1445         x86_push_reg (code, X86_EBP);
1446         x86_push_reg (code, X86_EDI);
1447         x86_push_reg (code, X86_ESI);
1448         x86_push_reg (code, X86_EBX);
1449         x86_push_reg (code, X86_EDX);
1450         x86_push_reg (code, X86_ECX);
1451         x86_push_reg (code, X86_EAX);
1452         x86_call_code (code, throw_exception);
1453         /* we should never reach this breakpoint */
1454         x86_breakpoint (code);
1455
1456         g_assert ((code - start) < 24);
1457         return start;
1458 }
1459
1460 /**
1461  * arch_get_throw_exception_by_name:
1462  *
1463  * Returns a function pointer which can be used to raise
1464  * corlib exceptions. The returned function has the following
1465  * signature: void (*func) (char *exc_name);
1466  * For example to raise an arithmetic exception you can use:
1467  *
1468  * x86_push_imm (code, "ArithmeticException");
1469  * x86_call_code (code, arch_get_throw_exception ());
1470  *
1471  */
1472 gpointer
1473 arch_get_throw_exception_by_name ()
1474 {
1475         static guint8 start [32];
1476         static int inited = 0;
1477         guint8 *code;
1478
1479         if (inited)
1480                 return start;
1481
1482         inited = 1;
1483         code = start;
1484
1485         /* fixme: we do not save EAX, EDX, ECD - unsure if we need that */
1486
1487         x86_push_membase (code, X86_ESP, 4); /* exception name */
1488         x86_push_imm (code, "System");
1489         x86_push_imm (code, mono_defaults.exception_class->image);
1490         x86_call_code (code, mono_exception_from_name);
1491         x86_alu_reg_imm (code, X86_ADD, X86_ESP, 12);
1492         /* save the newly create object (overwrite exception name)*/
1493         x86_mov_membase_reg (code, X86_ESP, 4, X86_EAX, 4);
1494         x86_jump_code (code, arch_get_throw_exception ());
1495
1496         g_assert ((code - start) < 32);
1497
1498         return start;
1499 }
1500
1501 /*
1502  * this returns a helper method to invoke a method with a user supplied
1503  * stack frame. The returned method has the following signature:
1504  * invoke_method_with_frame ((gpointer code, gpointer frame, int frame_size);
1505  */
1506 static gpointer
1507 get_invoke_method_with_frame ()
1508 {
1509         static guint8 *start;
1510         guint8 *code;
1511
1512         if (start)
1513                 return start;
1514
1515         start = code = malloc (64);
1516
1517         /* Prolog */
1518         x86_push_reg (code, X86_EBP);
1519         x86_mov_reg_reg (code, X86_EBP, X86_ESP, 4);
1520         x86_push_reg (code, X86_EBX);
1521         x86_push_reg (code, X86_EDI);
1522         x86_push_reg (code, X86_ESI);
1523
1524         x86_mov_reg_membase (code, X86_EAX, X86_EBP, 16, 4);
1525         x86_alu_reg_reg (code, X86_SUB, X86_ESP, X86_EAX);
1526
1527         x86_push_membase (code, X86_EBP, 16);
1528         x86_push_membase (code, X86_EBP, 12);
1529         x86_lea_membase (code, X86_EAX, X86_ESP, 2*4);
1530         x86_push_reg (code, X86_EAX);
1531         x86_call_code (code, memcpy);
1532         x86_alu_reg_imm (code, X86_ADD, X86_ESP, 12);
1533
1534         x86_mov_reg_membase (code, X86_EAX, X86_EBP, 8, 4);
1535         x86_call_reg (code, X86_EAX);
1536
1537         x86_mov_reg_membase (code, X86_ECX, X86_EBP, 16, 4);
1538         x86_alu_reg_reg (code, X86_ADD, X86_ESP, X86_ECX);
1539
1540         /* Epilog */
1541         x86_pop_reg (code, X86_ESI);
1542         x86_pop_reg (code, X86_EDI);
1543         x86_pop_reg (code, X86_EBX);
1544         x86_leave (code);
1545         x86_ret (code);
1546
1547         g_assert ((code - start) < 64);
1548
1549         return start;
1550 }
1551
1552 /**
1553  * arch_runtime_invoke:
1554  * @method: the method to invoke
1555  * @obj: this pointer
1556  * @params: array of parameter values.
1557  *
1558  * TODO: very ugly piece of code. we should replace that with a method-specific
1559  * trampoline (as suggested by Paolo).
1560  */
1561 MonoObject*
1562 arch_runtime_invoke (MonoMethod *method, void *obj, void **params)
1563 {
1564         static guint64 (*invoke_int64) (gpointer code, gpointer frame, int frame_size) = NULL;
1565         static double (*invoke_double) (gpointer code, gpointer frame, int frame_size) = NULL;
1566         MonoObject *retval;
1567         MonoMethodSignature *sig = method->signature;
1568         int i, tmp, type, sp = 0;
1569         void *ret;
1570         int frame_size = 0;
1571         gpointer *frame;
1572         gpointer code;
1573
1574         /* allocate ret object. */
1575         if (sig->ret->type == MONO_TYPE_VOID) {
1576                 retval = NULL;
1577                 ret = NULL;
1578         } else {
1579                 MonoClass *klass = mono_class_from_mono_type (sig->ret);
1580                 if (klass->valuetype) {
1581                         retval = mono_object_new (mono_domain_get (), klass);
1582                         ret = ((char*)retval) + sizeof (MonoObject);
1583                 } else {
1584                         ret = &retval;
1585                 }
1586         }
1587
1588         if (ISSTRUCT (sig->ret))
1589                 frame_size += sizeof (gpointer);
1590
1591         if (sig->hasthis)
1592                 frame_size += sizeof (gpointer);
1593
1594         for (i = 0; i < sig->param_count; ++i) {
1595                 int align;
1596                 frame_size += mono_type_stack_size (sig->params [i], &align);
1597         }
1598
1599         frame = alloca (frame_size);
1600
1601         if (ISSTRUCT (sig->ret))
1602                 frame [sp++] = ret;
1603
1604         if (sig->hasthis)
1605                 frame [sp++] = obj;
1606
1607
1608         for (i = 0; i < sig->param_count; ++i) {
1609                 if (sig->params [i]->byref) {
1610                         frame [sp++] = params [i];
1611                         continue;
1612                 }
1613                 type = sig->params [i]->type;
1614 handle_enum:
1615                 switch (type) {
1616                 case MONO_TYPE_U1:
1617                 case MONO_TYPE_I1:
1618                 case MONO_TYPE_BOOLEAN:
1619                         tmp = *(MonoBoolean*)params [i];
1620                         frame [sp++] = (gpointer)tmp;
1621                         break;
1622                 case MONO_TYPE_U2:
1623                 case MONO_TYPE_I2:
1624                 case MONO_TYPE_CHAR:
1625                         tmp = *(gint16*)params [i];
1626                         frame [sp++] = (gpointer)tmp;
1627                         break;
1628 #if SIZEOF_VOID_P == 4
1629                 case MONO_TYPE_U:
1630                 case MONO_TYPE_I:
1631 #endif
1632                 case MONO_TYPE_U4:
1633                 case MONO_TYPE_I4:
1634                         frame [sp++] = (gpointer)*(gint32*)params [i];
1635                         break;
1636 #if SIZEOF_VOID_P == 8
1637                 case MONO_TYPE_U:
1638                 case MONO_TYPE_I:
1639 #endif
1640                 case MONO_TYPE_U8:
1641                 case MONO_TYPE_I8:
1642                         frame [sp++] = (gpointer)*(gint32*)params [i];
1643                         frame [sp++] = (gpointer)*(((gint32*)params [i]) + 1);
1644                         break;
1645                 case MONO_TYPE_VALUETYPE:
1646                         if (sig->params [i]->data.klass->enumtype) {
1647                                 type = sig->params [i]->data.klass->enum_basetype->type;
1648                                 goto handle_enum;
1649                         } else {
1650                                 g_warning ("generic valutype %s not handled in runtime invoke", sig->params [i]->data.klass->name);
1651                         }
1652                         break;
1653                 case MONO_TYPE_STRING:
1654                         frame [sp++] = params [i];
1655                         break;
1656                 default:
1657                         g_error ("type 0x%x not handled in invoke", sig->params [i]->type);
1658                 }
1659         }
1660
1661         if (method->addr)
1662                 code = method->addr;
1663         else
1664                 code = arch_compile_method (method);
1665
1666         if (!invoke_int64)
1667                 invoke_int64 = (gpointer)invoke_double = get_invoke_method_with_frame ();
1668
1669         type = sig->ret->type;
1670 handle_enum_2:
1671         switch (type) {
1672         case MONO_TYPE_VOID:
1673                 invoke_int64 (code, frame, frame_size);
1674                 break;
1675         case MONO_TYPE_U1:
1676         case MONO_TYPE_I1:
1677         case MONO_TYPE_BOOLEAN:
1678         case MONO_TYPE_U2:
1679         case MONO_TYPE_I2:
1680         case MONO_TYPE_CHAR:
1681 #if SIZEOF_VOID_P == 4
1682         case MONO_TYPE_U:
1683         case MONO_TYPE_I:
1684 #endif
1685         case MONO_TYPE_U4:
1686         case MONO_TYPE_I4:
1687         case MONO_TYPE_STRING:
1688                 *((guint32 *)ret) = invoke_int64 (code, frame, frame_size);
1689                 break;
1690 #if SIZEOF_VOID_P == 8
1691         case MONO_TYPE_U:
1692         case MONO_TYPE_I:
1693 #endif
1694         case MONO_TYPE_U8:
1695         case MONO_TYPE_I8:
1696                 *((guint64 *)ret) = invoke_int64 (code, frame, frame_size);
1697                 break;
1698         case MONO_TYPE_R4:
1699                 *((float *)ret) = invoke_double (code, frame, frame_size);
1700                 break;
1701         case MONO_TYPE_R8:
1702                 *((double *)ret) = invoke_double (code, frame, frame_size);
1703                 break;
1704         case MONO_TYPE_VALUETYPE:
1705                 if (sig->params [i]->data.klass->enumtype) {
1706                         type = sig->params [i]->data.klass->enum_basetype->type;
1707                         goto handle_enum_2;
1708                 } else
1709                         invoke_int64 (code, frame, frame_size);
1710                 break;
1711         default:
1712                 g_error ("type 0x%x not handled in invoke", sig->params [i]->type);
1713         }
1714
1715         return retval;
1716 }
1717
1718
1719