New tests, update.
[mono.git] / mono / mini / aot-compiler.c
1 /*
2  * aot.c: mono Ahead of Time compiler
3  *
4  * Author:
5  *   Dietmar Maurer (dietmar@ximian.com)
6  *   Zoltan Varga (vargaz@gmail.com)
7  *
8  * (C) 2002 Ximian, Inc.
9  */
10
11 /* Remaining AOT-only work:
12  * - reduce the length of the wrapper names.
13  * - aot IMT tables, so we don't have two kinds of aot code.
14  * - optimize the trampolines, generate more code in the arch files.
15  * - make things more consistent with how elf works, for example, use ELF 
16  *   relocations.
17  * Remaining generics sharing work:
18  * - optimize the size of the data which is encoded.
19  * - optimize the runtime loading of data:
20  *   - the trampoline code calls mono_jit_info_table_find () to find the rgctx, 
21  *     which loads the debugging+exception handling info for the method. This is a 
22  *     huge waste of time and code, since the rgctx structure is currently empty.
23  *   - every shared method has a MonoGenericJitInfo structure which is only really
24  *     used for handling catch clauses with open types, not a very common use case.
25  */
26
27 #include "config.h"
28 #include <sys/types.h>
29 #ifdef HAVE_UNISTD_H
30 #include <unistd.h>
31 #endif
32 #include <fcntl.h>
33 #include <ctype.h>
34 #include <string.h>
35 #ifndef PLATFORM_WIN32
36 #include <sys/mman.h>
37 #else
38 #include <winsock2.h>
39 #include <windows.h>
40 #endif
41
42 #include <errno.h>
43 #include <sys/stat.h>
44 #include <limits.h>    /* for PAGESIZE */
45 #ifndef PAGESIZE
46 #define PAGESIZE 4096
47 #endif
48
49 #include <mono/metadata/tabledefs.h>
50 #include <mono/metadata/class.h>
51 #include <mono/metadata/object.h>
52 #include <mono/metadata/tokentype.h>
53 #include <mono/metadata/appdomain.h>
54 #include <mono/metadata/debug-helpers.h>
55 #include <mono/metadata/assembly.h>
56 #include <mono/metadata/metadata-internals.h>
57 #include <mono/metadata/marshal.h>
58 #include <mono/metadata/gc-internal.h>
59 #include <mono/metadata/method-builder.h>
60 #include <mono/utils/mono-logger.h>
61 #include "mono/utils/mono-compiler.h"
62
63 #include "mini.h"
64 #include "version.h"
65
66 #ifndef DISABLE_AOT
67
68 #ifdef PLATFORM_WIN32
69 #define SHARED_EXT ".dll"
70 #elif defined(__ppc__) && defined(__MACH__)
71 #define SHARED_EXT ".dylib"
72 #else
73 #define SHARED_EXT ".so"
74 #endif
75
76 #if defined(sparc) || defined(__ppc__) || defined(__powerpc__) || defined(__MACH__)
77 #define AS_STRING_DIRECTIVE ".asciz"
78 #else
79 /* GNU as */
80 #define AS_STRING_DIRECTIVE ".string"
81 #endif
82
83
84 // __MACH__
85 // .byte generates 1 byte per expression.
86 // .short generates 2 bytes per expression.
87 // .long generates 4 bytes per expression.
88 // .quad generates 8 bytes per expression.
89
90 #define ALIGN_PTR_TO(ptr,align) (gpointer)((((gssize)(ptr)) + (align - 1)) & (~(align - 1)))
91 #define ROUND_DOWN(VALUE,SIZE)  ((VALUE) & ~((SIZE) - 1))
92
93 typedef struct MonoAotOptions {
94         char *outfile;
95         gboolean save_temps;
96         gboolean write_symbols;
97         gboolean metadata_only;
98         gboolean bind_to_runtime_version;
99         gboolean full_aot;
100         gboolean no_dlsym;
101         gboolean static_link;
102         gboolean asm_only;
103 } MonoAotOptions;
104
105 typedef struct MonoAotStats {
106         int ccount, mcount, lmfcount, abscount, wrappercount, gcount, ocount, genericcount;
107         int code_size, info_size, ex_info_size, got_size, class_info_size, got_info_size, got_info_offsets_size;
108         int methods_without_got_slots, direct_calls, all_calls;
109         int got_slots;
110         int got_slot_types [MONO_PATCH_INFO_NONE];
111 } MonoAotStats;
112
113 /*#define USE_ELF_WRITER 1*/
114
115 #if defined(USE_ELF_WRITER)
116 #define USE_BIN_WRITER 1
117 #endif
118
119 #ifdef USE_BIN_WRITER
120
121 typedef struct _BinSymbol BinSymbol;
122 typedef struct _BinReloc BinReloc;
123 typedef struct _BinSection BinSection;
124
125 #else
126
127 /* emit mode */
128 enum {
129         EMIT_NONE,
130         EMIT_BYTE,
131         EMIT_WORD,
132         EMIT_LONG
133 };
134
135 #endif
136
137 typedef struct MonoAotCompile {
138         MonoImage *image;
139         GPtrArray *methods;
140         GHashTable *method_indexes;
141         MonoCompile **cfgs;
142         GHashTable *patch_to_plt_offset;
143         GHashTable *plt_offset_to_patch;
144         GHashTable *patch_to_shared_got_offset;
145         GPtrArray *shared_patches;
146         GHashTable *image_hash;
147         GHashTable *method_to_cfg;
148         GHashTable *token_info_hash;
149         GPtrArray *image_table;
150         GPtrArray *globals;
151         GList *method_order;
152         /* Number of trampolines emitted into the AOT file */
153         guint32 num_aot_trampolines;
154         guint32 got_offset, plt_offset;
155         /* Number of GOT entries reserved for trampolines */
156         guint32 num_trampoline_got_entries;
157         guint32 *method_got_offsets;
158         MonoAotOptions aot_opts;
159         guint32 nmethods;
160         guint32 opts;
161         MonoMemPool *mempool;
162         MonoAotStats stats;
163         int method_index;
164         char *static_linking_symbol;
165 #ifdef USE_BIN_WRITER
166         BinSymbol *symbols;
167         BinSection *sections;
168         BinSection *cur_section;
169         BinReloc *relocations;
170         GHashTable *labels;
171         int num_relocs;
172 #else
173         FILE *fp;
174         char *tmpfname;
175         int mode; /* emit mode */
176         int col_count; /* bytes emitted per .byte line */
177 #endif
178 } MonoAotCompile;
179
180 #ifdef HAVE_ARRAY_ELEM_INIT
181 #define MSGSTRFIELD(line) MSGSTRFIELD1(line)
182 #define MSGSTRFIELD1(line) str##line
183 static const struct msgstr_t {
184 #define PATCH_INFO(a,b) char MSGSTRFIELD(__LINE__) [sizeof (b)];
185 #include "patch-info.h"
186 #undef PATCH_INFO
187 } opstr = {
188 #define PATCH_INFO(a,b) b,
189 #include "patch-info.h"
190 #undef PATCH_INFO
191 };
192 static const gint16 opidx [] = {
193 #define PATCH_INFO(a,b) [MONO_PATCH_INFO_ ## a] = offsetof (struct msgstr_t, MSGSTRFIELD(__LINE__)),
194 #include "patch-info.h"
195 #undef PATCH_INFO
196 };
197
198 static const char*
199 get_patch_name (int info)
200 {
201         return (const char*)&opstr + opidx [info];
202 }
203
204 #else
205 #define PATCH_INFO(a,b) b,
206 static const char* const
207 patch_types [MONO_PATCH_INFO_NUM + 1] = {
208 #include "patch-info.h"
209         NULL
210 };
211
212 static const char*
213 get_patch_name (int info)
214 {
215         return patch_types [info];
216 }
217
218 #endif
219
220 static void
221 emit_global (MonoAotCompile *acfg, const char *name, gboolean func);
222
223 static gboolean 
224 is_got_patch (MonoJumpInfoType patch_type)
225 {
226         return TRUE;
227 }
228
229 static G_GNUC_UNUSED int
230 ilog2(register int value)
231 {
232         int count = -1;
233         while (value & ~0xf) count += 4, value >>= 4;
234         while (value) count++, value >>= 1;
235         return count;
236 }
237
238 #ifdef USE_BIN_WRITER
239
240 typedef struct _BinLabel BinLabel;
241 struct _BinLabel {
242         char *name;
243         BinSection *section;
244         int offset;
245 };
246
247 struct _BinReloc {
248         BinReloc *next;
249         char *val1;
250         char *val2;
251         BinSection *val2_section;
252         int val2_offset;
253         int offset;
254         BinSection *section;
255         int section_offset;
256 };
257
258 struct _BinSymbol {
259         BinSymbol *next;
260         char *name;
261         BinSection *section;
262         int offset;
263         gboolean is_function;
264         gboolean is_global;
265 };
266
267 struct _BinSection {
268         BinSection *next;
269         BinSection *parent;
270         char *name;
271         int subsection;
272         guint8 *data;
273         int data_len;
274         int cur_offset;
275         int file_offset;
276         int virt_offset;
277         int shidx;
278 };
279
280 static void
281 emit_start (MonoAotCompile *acfg)
282 {
283         acfg->labels = g_hash_table_new (g_str_hash, g_str_equal);
284 }
285
286 static void
287 emit_section_change (MonoAotCompile *acfg, const char *section_name, int subsection_index)
288 {
289         BinSection *section;
290
291         if (acfg->cur_section && acfg->cur_section->subsection == subsection_index
292                         && strcmp (acfg->cur_section->name, section_name) == 0)
293                 return;
294         for (section = acfg->sections; section; section = section->next) {
295                 if (section->subsection == subsection_index && strcmp (section->name, section_name) == 0) {
296                         acfg->cur_section = section;
297                         return;
298                 }
299         }
300         if (!section) {
301                 section = g_new0 (BinSection, 1);
302                 section->name = g_strdup (section_name);
303                 section->subsection = subsection_index;
304                 section->next = acfg->sections;
305                 acfg->sections = section;
306                 acfg->cur_section = section;
307         }
308 }
309
310 static void
311 emit_global_inner (MonoAotCompile *acfg, const char *name, gboolean func)
312 {
313         BinSymbol *symbol = g_new0 (BinSymbol, 1);
314         symbol->name = g_strdup (name);
315         symbol->is_function = func;
316         symbol->is_global = TRUE;
317         symbol->section = acfg->cur_section;
318         /* FIXME: we align after this call... */
319         symbol->offset = symbol->section->cur_offset;
320         symbol->next = acfg->symbols;
321         acfg->symbols = symbol;
322 }
323
324 static void
325 emit_label (MonoAotCompile *acfg, const char *name)
326 {
327         BinLabel *label = g_new0 (BinLabel, 1);
328         label->name = g_strdup (name);
329         label->section = acfg->cur_section;
330         label->offset = acfg->cur_section->cur_offset;
331         g_hash_table_insert (acfg->labels, label->name, label);
332 }
333
334 static void
335 emit_ensure_buffer (BinSection *section, int size)
336 {
337         int new_offset = section->cur_offset + size;
338         if (new_offset >= section->data_len) {
339                 int new_size = section->data_len? section->data_len * 2: 256;
340                 guint8 *data;
341                 while (new_size <= new_offset)
342                         new_size *= 2;
343                 data = g_malloc0 (new_size);
344                 memcpy (data, section->data, section->data_len);
345                 g_free (section->data);
346                 section->data = data;
347                 section->data_len = new_size;
348         }
349 }
350
351 static void
352 emit_bytes (MonoAotCompile *acfg, const guint8* buf, int size)
353 {
354         emit_ensure_buffer (acfg->cur_section, size);
355         memcpy (acfg->cur_section->data + acfg->cur_section->cur_offset, buf, size);
356         acfg->cur_section->cur_offset += size;
357 }
358
359 static void
360 emit_string (MonoAotCompile *acfg, const char *value)
361 {
362         int size = strlen (value) + 1;
363         emit_bytes (acfg, (const guint8*)value, size);
364 }
365
366 static void
367 emit_line (MonoAotCompile *acfg)
368 {
369         /* Nothing to do in binary writer */
370 }
371
372 static void
373 emit_string_symbol (MonoAotCompile *acfg, const char *name, const char *value)
374 {
375         emit_section_change (acfg, ".text", 1);
376         emit_global (acfg, name, FALSE);
377         emit_label (acfg, name);
378         emit_string (acfg, value);
379 }
380
381 static void 
382 emit_alignment (MonoAotCompile *acfg, int size)
383 {
384         int offset = acfg->cur_section->cur_offset;
385         int add;
386         offset += (size - 1);
387         offset &= ~(size - 1);
388         add = offset - acfg->cur_section->cur_offset;
389         if (add) {
390                 emit_ensure_buffer (acfg->cur_section, add);
391                 acfg->cur_section->cur_offset += add;
392         }
393 }
394
395 static void
396 emit_pointer (MonoAotCompile *acfg, const char *target)
397 {
398         BinReloc *reloc;
399
400         if (!target)
401                 // FIXME:
402                 g_assert_not_reached ();
403         emit_alignment (acfg, sizeof (gpointer));
404         reloc = g_new0 (BinReloc, 1);
405         reloc->val1 = g_strdup (target);
406         reloc->section = acfg->cur_section;
407         reloc->section_offset = acfg->cur_section->cur_offset;
408         reloc->next = acfg->relocations;
409         acfg->relocations = reloc;
410         if (strcmp (reloc->section->name, ".data") == 0) {
411                 acfg->num_relocs++;
412                 g_print ("reloc: %s at %d\n", target, acfg->cur_section->cur_offset);
413         }
414         acfg->cur_section->cur_offset += sizeof (gpointer);
415 }
416
417 static void
418 emit_int16 (MonoAotCompile *acfg, int value)
419 {
420         guint8 *data;
421         emit_ensure_buffer (acfg->cur_section, 2);
422         data = acfg->cur_section->data + acfg->cur_section->cur_offset;
423         acfg->cur_section->cur_offset += 2;
424         /* FIXME: little endian */
425         data [0] = value;
426         data [1] = value >> 8;
427 }
428
429 static void
430 emit_int32 (MonoAotCompile *acfg, int value)
431 {
432         guint8 *data;
433         emit_ensure_buffer (acfg->cur_section, 4);
434         data = acfg->cur_section->data + acfg->cur_section->cur_offset;
435         acfg->cur_section->cur_offset += 4;
436         /* FIXME: little endian */
437         data [0] = value;
438         data [1] = value >> 8;
439         data [2] = value >> 16;
440         data [3] = value >> 24;
441 }
442
443 static void
444 emit_symbol_diff (MonoAotCompile *acfg, const char *end, const char* start, int offset)
445 {
446         BinReloc *reloc;
447         reloc = g_new0 (BinReloc, 1);
448         reloc->val1 = g_strdup (end);
449         if (strcmp (start, ".") == 0) {
450                 reloc->val2_section = acfg->cur_section;
451                 reloc->val2_offset = acfg->cur_section->cur_offset;
452         } else {
453                 reloc->val2 = g_strdup (start);
454         }
455         reloc->offset = offset;
456         reloc->section = acfg->cur_section;
457         reloc->section_offset = acfg->cur_section->cur_offset;
458         reloc->next = acfg->relocations;
459         acfg->relocations = reloc;
460         acfg->cur_section->cur_offset += 4;
461         /*if (strcmp (reloc->section->name, ".data") == 0) {
462                 acfg->num_relocs++;
463                 g_print ("reloc: %s - %s + %d at %d\n", end, start, offset, acfg->cur_section->cur_offset - 4);
464         }*/
465 }
466
467 static void
468 emit_zero_bytes (MonoAotCompile *acfg, int num)
469 {
470         emit_ensure_buffer (acfg->cur_section, num);
471         acfg->cur_section->cur_offset += num;
472 }
473
474 #ifdef USE_ELF_WRITER
475 enum {
476         SYM_LOCAL = 0 << 4,
477         SYM_GLOBAL = 1 << 4,
478         SYM_OBJECT = 1,
479         SYM_FUNC = 2,
480         SYM_SECTION = 3
481 };
482
483 enum {
484         SECT_NULL,
485         SECT_HASH,
486         SECT_DYNSYM,
487         SECT_DYNSTR,
488         SECT_REL_DYN,
489         SECT_TEXT,
490         SECT_DYNAMIC,
491         SECT_GOT_PLT,
492         SECT_DATA,
493         SECT_BSS,
494         SECT_SHSTRTAB,
495         SECT_SYMTAB,
496         SECT_STRTAB,
497         SECT_NUM
498 };
499
500 enum {
501         DYN_HASH = 4,
502         DYN_STRTAB = 5,
503         DYN_SYMTAB = 6,
504         DYN_STRSZ = 10,
505         DYN_SYMENT = 11,
506         DYN_REL = 17,
507         DYN_RELSZ = 18,
508         DYN_RELENT = 19,
509         DYN_RELCOUNT = 0x6ffffffa
510 };
511
512 static const char* section_names [] = {
513         "",
514         ".hash",
515         ".dynsym",
516         ".dynstr",
517         ".rel.dyn",
518         ".text",
519         ".dynamic",
520         ".got.plt",
521         ".data",
522         ".bss",
523         ".shstrtab",
524         ".symtab",
525         ".strtab"
526 };
527
528 static const guint8 section_type [] = {
529         0, 5, 11, 3, 9, 1,
530         6, 1, 1, 8, 3, 2, 3
531 };
532
533 static const guint8 section_link [] = {
534         0, 2, 3, 0, 2, 0, 3, 0, 0, 0, 0, 12, 0
535 };
536
537 static const guint8 section_esize [] = {
538         0, 4, 16, 0, 8, 0, 8, 4, 0, 0, 0, 16, 0
539 };
540
541 static const guint8 section_flags [] = {
542         0, 2, 2, 2, 2, 6, 3, 3, 3, 3, 0, 0, 0
543 };
544
545 static const guint16 section_align [] = {
546         0, 4, 4, 1, 4, 4096, 4, 4, 8, 8, 1, 4, 1
547 };
548
549 struct ElfHeader {
550         guint8  e_ident [16];
551         guint16 e_type;
552         guint16 e_machine;
553         guint32 e_version;
554         gsize   e_entry;
555         gsize   e_phoff;
556         gsize   e_shoff;
557         guint32 e_flags;
558         guint16 e_ehsize;
559         guint16 e_phentsize;
560         guint16 e_phnum;
561         guint16 e_shentsize;
562         guint16 e_shnum;
563         guint16 e_shstrndx;
564 };
565
566 struct ElfSectHeader {
567         guint32 sh_name;
568         guint32 sh_type;
569         gsize   sh_flags;
570         gsize   sh_addr;
571         gsize   sh_offset;
572         gsize   sh_size;
573         guint32 sh_link;
574         guint32 sh_info;
575         gsize   sh_addralign;
576         gsize   sh_entsize;
577 };
578
579 #if SIZEOF_VOID_P == 4
580
581 struct ElfProgHeader {
582         guint32 p_type;
583         guint32 p_offset;
584         guint32 p_vaddr;
585         guint32 p_paddr;
586         guint32 p_filesz;
587         guint32 p_memsz;
588         guint32 p_flags;
589         guint32 p_align;
590 };
591
592 typedef struct {
593         guint32 st_name;
594         guint32 st_value;
595         guint32 st_size;
596         guint8  st_info;
597         guint8  st_other;
598         guint16 st_shndx;
599 } ElfSymbol;
600
601 typedef struct {
602         guint32 addr;
603         guint32 value;
604 } ElfReloc;
605
606 typedef struct {
607         guint32 d_tag;
608         guint32 d_val;
609 } ElfDynamic;
610
611 #else
612
613 struct ElfProgHeader {
614         guint32 p_type;
615         guint32 p_flags;
616         guint64 p_offset;
617         guint64 p_vaddr;
618         guint64 p_paddr;
619         guint64 p_filesz;
620         guint64 p_memsz;
621         guint64 p_align;
622 };
623
624 typedef struct {
625         guint32 st_name;
626         guint8  st_info;
627         guint8  st_other;
628         guint16 st_shndx;
629         guint64 st_value;
630         guint64 st_size;
631 } ElfSymbol;
632
633 typedef struct {
634         guint64 addr;
635         guint64 value;
636 } ElfReloc;
637
638 typedef struct {
639         guint64 addr;
640         guint64 value;
641         guint64 addend;
642 } ElfRelocA;
643
644 typedef struct {
645         guint64 d_tag;
646         guint64 d_val;
647 } ElfDynamic;
648
649 #endif
650
651 typedef struct {
652         GString *data;
653         GHashTable *hash;
654 } ElfStrTable;
655
656 static int
657 str_table_add (ElfStrTable *table, const char* value)
658 {
659         int idx;
660         if (!table->data) {
661                 table->data = g_string_new_len ("", 1);
662                 table->hash = g_hash_table_new (g_str_hash, g_str_equal);
663         }
664         idx = GPOINTER_TO_UINT (g_hash_table_lookup (table->hash, value));
665         if (idx)
666                 return idx;
667         idx = table->data->len;
668         g_string_append (table->data, value);
669         g_string_append_c (table->data, 0);
670         g_hash_table_insert (table->hash, (void*)value, GUINT_TO_POINTER (idx));
671         return idx;
672 }
673
674 static void
675 append_subsection (MonoAotCompile *acfg, struct ElfSectHeader *sheaders, BinSection *sect, BinSection *add)
676 {
677         int offset = sect->cur_offset;
678         /*offset += (sheaders [sect->shidx].sh_addralign - 1);
679         offset &= ~(sheaders [sect->shidx].sh_addralign - 1);*/
680         offset += (8 - 1);
681         offset &= ~(8 - 1);
682         emit_ensure_buffer (sect, offset);
683         g_print ("section %s aligned to %d from %d\n", sect->name, offset, sect->cur_offset);
684         sect->cur_offset = offset;
685
686         emit_ensure_buffer (sect, add->cur_offset);
687         memcpy (sect->data + sect->cur_offset, add->data, add->cur_offset);
688         add->parent = sect;
689         sect->cur_offset += add->cur_offset;
690         add->cur_offset = offset; /* it becomes the offset in the parent section */
691         g_print ("subsection %d of %s added at offset %d (align: %d)\n", add->subsection, sect->name, add->cur_offset, sheaders [sect->shidx].sh_addralign);
692         add->data = NULL;
693         add->data_len = 0;
694 }
695
696 /* merge the subsections */
697 static int
698 collect_sections (MonoAotCompile *acfg, struct ElfSectHeader *sheaders, BinSection **out, int num)
699 {
700         int i, j, maxs, num_sections;
701         BinSection *sect;
702
703         num_sections = 0;
704         maxs = 0;
705         for (sect = acfg->sections; sect; sect = sect->next) {
706                 if (sect->subsection == 0) {
707                         out [num_sections++] = sect;
708                         g_assert (num_sections < num);
709                         if (strcmp (sect->name, ".text") == 0) {
710                                 sect->shidx = SECT_TEXT;
711                         } else if (strcmp (sect->name, ".data") == 0) {
712                                 sect->shidx = SECT_DATA;
713                         } else if (strcmp (sect->name, ".bss") == 0) {
714                                 sect->shidx = SECT_BSS;
715                         }
716                 }
717                 maxs = MAX (maxs, sect->subsection);
718         }
719         for (i = 0; i < num_sections; i++) {
720                 for (j = 1; j <= maxs; ++j) {
721                         for (sect = acfg->sections; sect; sect = sect->next) {
722                                 if (sect->subsection == j && strcmp (out [i]->name, sect->name) == 0) {
723                                         append_subsection (acfg, sheaders, out [i], sect);
724                                 }
725                         }
726                 }
727         }
728         return num_sections;
729 }
730
731 static unsigned long
732 elf_hash (const unsigned char *name)
733 {
734         unsigned long h = 0, g;
735         while (*name) {
736                 h = (h << 4) + *name++;
737                 if ((g = h & 0xf0000000))
738                         h ^= g >> 24;
739                 h &= ~g;
740         }
741         return h;
742 }
743
744 #define NUM_BUCKETS 17
745
746 static int*
747 build_hash (MonoAotCompile *acfg, int num_sections, ElfStrTable *dynstr)
748 {
749         int *data;
750         int num_symbols = 1 + num_sections + 3;
751         BinSymbol *symbol;
752
753         for (symbol = acfg->symbols; symbol; symbol = symbol->next) {
754                 if (!symbol->is_global)
755                         continue;
756                 num_symbols++;
757                 str_table_add (dynstr, symbol->name);
758                 /*g_print ("adding sym: %s\n", symbol->name);*/
759         }
760         str_table_add (dynstr, "__bss_start");
761         str_table_add (dynstr, "_edata");
762         str_table_add (dynstr, "_end");
763
764         data = g_new0 (int, num_symbols + 2 + NUM_BUCKETS);
765         data [0] = NUM_BUCKETS;
766         data [1] = num_symbols;
767
768         return data;
769 }
770
771 static gsize
772 get_label_addr (MonoAotCompile *acfg, const char *name)
773 {
774         int offset;
775         BinLabel *lab;
776         BinSection *section;
777         gsize value;
778
779         lab = g_hash_table_lookup (acfg->labels, name);
780         section = lab->section;
781         offset = lab->offset;
782         if (section->parent) {
783                 value = section->parent->file_offset + section->cur_offset + offset;
784         } else {
785                 value = section->file_offset + offset;
786         }
787         return value;
788 }
789
790 static ElfSymbol*
791 collect_syms (MonoAotCompile *acfg, int *hash, ElfStrTable *strtab, struct ElfSectHeader *sheaders, int *num_syms)
792 {
793         ElfSymbol *symbols;
794         BinSymbol *symbol;
795         BinSection *section;
796         int i;
797         int *bucket;
798         int *chain;
799         unsigned long hashc;
800
801         if (hash)
802                 symbols = g_new0 (ElfSymbol, hash [1]);
803         else
804                 symbols = g_new0 (ElfSymbol, *num_syms + SECT_NUM + 10); /* FIXME */
805
806         /* the first symbol is undef, all zeroes */
807         i = 1;
808         if (sheaders) {
809                 int j;
810                 for (j = 1; j < SECT_NUM; ++j) {
811                         symbols [i].st_info = SYM_LOCAL | SYM_SECTION;
812                         symbols [i].st_shndx = j;
813                         symbols [i].st_value = sheaders [j].sh_addr;
814                         ++i;
815                 }
816         } else {
817                 for (section = acfg->sections; section; section = section->next) {
818                         if (section->parent)
819                                 continue;
820                         symbols [i].st_info = SYM_LOCAL | SYM_SECTION;
821                         if (strcmp (section->name, ".text") == 0) {
822                                 symbols [i].st_shndx = SECT_TEXT;
823                                 section->shidx = SECT_TEXT;
824                                 section->file_offset = 4096;
825                                 symbols [i].st_value = section->file_offset;
826                         } else if (strcmp (section->name, ".data") == 0) {
827                                 symbols [i].st_shndx = SECT_DATA;
828                                 section->shidx = SECT_DATA;
829                                 section->file_offset = 4096 + 28; /* FIXME */
830                                 symbols [i].st_value = section->file_offset;
831                         } else if (strcmp (section->name, ".bss") == 0) {
832                                 symbols [i].st_shndx = SECT_BSS;
833                                 section->shidx = SECT_BSS;
834                                 section->file_offset = 4096 + 28 + 8; /* FIXME */
835                                 symbols [i].st_value = section->file_offset;
836                         }
837                         ++i;
838                 }
839         }
840         for (symbol = acfg->symbols; symbol; symbol = symbol->next) {
841                 int offset;
842                 BinLabel *lab;
843                 if (!symbol->is_global)
844                         continue;
845                 symbols [i].st_info = (symbol->is_function? SYM_FUNC : SYM_OBJECT) | SYM_GLOBAL;
846                 symbols [i].st_name = str_table_add (strtab, symbol->name);
847                 /*g_print ("sym name %s tabled to %d\n", symbol->name, symbols [i].st_name);*/
848                 section = symbol->section;
849                 symbols [i].st_shndx = section->parent? section->parent->shidx: section->shidx;
850                 lab = g_hash_table_lookup (acfg->labels, symbol->name);
851                 offset = lab->offset;
852                 if (section->parent) {
853                         symbols [i].st_value = section->parent->file_offset + section->cur_offset + offset;
854                 } else {
855                         symbols [i].st_value = section->file_offset + offset;
856                 }
857                 ++i;
858         }
859         /* add special symbols */
860         symbols [i].st_name = str_table_add (strtab, "__bss_start");
861         symbols [i].st_shndx = 0xfff1;
862         symbols [i].st_info = SYM_GLOBAL;
863         ++i;
864         symbols [i].st_name = str_table_add (strtab, "_edata");
865         symbols [i].st_shndx = 0xfff1;
866         symbols [i].st_info = SYM_GLOBAL;
867         ++i;
868         symbols [i].st_name = str_table_add (strtab, "_end");
869         symbols [i].st_shndx = 0xfff1;
870         symbols [i].st_info = SYM_GLOBAL;
871         ++i;
872
873         if (num_syms)
874                 *num_syms = i;
875
876         /* add to hash table */
877         if (hash) {
878                 bucket = hash + 2;
879                 chain = hash + 2 + hash [0];
880                 for (i = 0; i < hash [1]; ++i) {
881                         int slot;
882                         /*g_print ("checking %d '%s' (sym %d)\n", symbols [i].st_name, strtab->data->str + symbols [i].st_name, i);*/
883                         if (!symbols [i].st_name)
884                                 continue;
885                         hashc = elf_hash ((guint8*)strtab->data->str + symbols [i].st_name);
886                         slot = hashc % hash [0];
887                         /*g_print ("hashing '%s' at slot %d (sym %d)\n", strtab->data->str + symbols [i].st_name, slot, i);*/
888                         if (bucket [slot]) {
889                                 chain [i] = bucket [slot];
890                                 bucket [slot] = i;
891                         } else {
892                                 bucket [slot] = i;
893                         }
894                 }
895         }
896         return symbols;
897 }
898
899 static void
900 reloc_symbols (MonoAotCompile *acfg, ElfSymbol *symbols, struct ElfSectHeader *sheaders, ElfStrTable *strtab, gboolean dynamic)
901 {
902         BinSection *section;
903         BinSymbol *symbol;
904         int i;
905
906         i = 1;
907         if (dynamic) {
908                 for (section = acfg->sections; section; section = section->next) {
909                         if (section->parent)
910                                 continue;
911                         symbols [i].st_value = sheaders [section->shidx].sh_addr;
912                         ++i;
913                 }
914         } else {
915                 for (i = 1; i < SECT_NUM; ++i) {
916                         symbols [i].st_value = sheaders [i].sh_addr;
917                 }
918         }
919         for (symbol = acfg->symbols; symbol; symbol = symbol->next) {
920                 int offset;
921                 BinLabel *lab;
922                 if (dynamic && !symbol->is_global)
923                         continue;
924                 section = symbol->section;
925                 lab = g_hash_table_lookup (acfg->labels, symbol->name);
926                 offset = lab->offset;
927                 if (section->parent) {
928                         symbols [i].st_value = sheaders [section->parent->shidx].sh_addr + section->cur_offset + offset;
929                 } else {
930                         symbols [i].st_value = sheaders [section->shidx].sh_addr + offset;
931                 }
932                 ++i;
933         }
934         /* __bss_start */
935         symbols [i].st_value = sheaders [SECT_BSS].sh_addr;
936         ++i;
937         /* _edata */
938         symbols [i].st_value = sheaders [SECT_DATA].sh_addr + sheaders [SECT_DATA].sh_size;
939         ++i;
940         /* _end */
941         symbols [i].st_value = sheaders [SECT_BSS].sh_addr + sheaders [SECT_BSS].sh_size;
942         ++i;
943 }
944
945 static ElfReloc*
946 resolve_relocations (MonoAotCompile *acfg)
947 {
948         BinReloc *reloc;
949         guint8 *data;
950         gsize end_val, start_val;
951         ElfReloc *rr;
952         int i;
953         gsize vaddr;
954
955         rr = g_new0 (ElfReloc, acfg->num_relocs);
956         i = 0;
957
958         for (reloc = acfg->relocations; reloc; reloc = reloc->next) {
959                 end_val = get_label_addr (acfg, reloc->val1);
960                 if (reloc->val2) {
961                         start_val = get_label_addr (acfg, reloc->val2);
962                 } else if (reloc->val2_section) {
963                         start_val = reloc->val2_offset;
964                         if (reloc->val2_section->parent)
965                                 start_val += reloc->val2_section->parent->file_offset + reloc->val2_section->cur_offset;
966                         else
967                                 start_val += reloc->val2_section->file_offset;
968                 } else {
969                         start_val = 0;
970                 }
971                 end_val = end_val - start_val + reloc->offset;
972                 if (reloc->section->parent) {
973                         data = reloc->section->parent->data;
974                         data += reloc->section->cur_offset;
975                         data += reloc->section_offset;
976                         vaddr = reloc->section->parent->file_offset;
977                         vaddr += reloc->section->cur_offset;
978                         vaddr += reloc->section_offset;
979                 } else {
980                         data = reloc->section->data;
981                         data += reloc->section_offset;
982                         vaddr = reloc->section->file_offset;
983                         vaddr += reloc->section_offset;
984                 }
985                 /* FIXME: little endian */
986                 data [0] = end_val;
987                 data [1] = end_val >> 8;
988                 data [2] = end_val >> 16;
989                 data [3] = end_val >> 24;
990                 if (start_val == 0) {
991                         rr [i].addr = vaddr;
992                         rr [i].value = 8; /* FIXME: 386_RELATIVE */
993                         ++i;
994                         g_assert (i <= acfg->num_relocs);
995                 }
996         }
997         return rr;
998 }
999
1000 static int
1001 emit_writeout (MonoAotCompile *acfg)
1002 {
1003         char *outfile_name, *tmp_outfile_name;
1004         FILE *file;
1005         struct ElfHeader header;
1006         struct ElfProgHeader progh [3];
1007         struct ElfSectHeader secth [SECT_NUM];
1008         ElfReloc *relocs;
1009         ElfStrTable str_table = {NULL, NULL};
1010         ElfStrTable sh_str_table = {NULL, NULL};
1011         ElfStrTable dyn_str_table = {NULL, NULL};
1012         BinSection* sections [6];
1013         BinSection *text_section = NULL, *data_section = NULL, *bss_section = NULL;
1014         ElfSymbol *dynsym;
1015         ElfSymbol *symtab;
1016         ElfDynamic dynamic [14];
1017         int *hash;
1018         int i, num_sections, file_offset, virt_offset, size, num_symtab;
1019         int num_local_syms;
1020
1021         g_assert (!acfg->aot_opts.asm_only);
1022
1023         if (acfg->aot_opts.outfile)
1024                 outfile_name = g_strdup_printf ("%s", acfg->aot_opts.outfile);
1025         else
1026                 outfile_name = g_strdup_printf ("%s%s", acfg->image->name, SHARED_EXT);
1027
1028         tmp_outfile_name = g_strdup_printf ("%s.tmp", outfile_name);
1029
1030         unlink (tmp_outfile_name);
1031         file = fopen (tmp_outfile_name, "w");
1032         g_assert (file);
1033
1034         /* Section headers */
1035         memset (&secth, 0, sizeof (secth));
1036         memset (&dynamic, 0, sizeof (dynamic));
1037         memset (&header, 0, sizeof (header));
1038
1039         for (i = 1; i < SECT_NUM; ++i) {
1040                 secth [i].sh_name = str_table_add (&sh_str_table, section_names [i]);
1041                 secth [i].sh_type = section_type [i];
1042                 secth [i].sh_link = section_link [i];
1043                 secth [i].sh_addralign = section_align [i];
1044                 secth [i].sh_flags = section_flags [i];
1045                 secth [i].sh_entsize = section_esize [i];
1046         }
1047         secth [SECT_DYNSYM].sh_info = 4;
1048         secth [SECT_SYMTAB].sh_info = 20;
1049
1050         num_sections = collect_sections (acfg, secth, sections, 6);
1051         hash = build_hash (acfg, num_sections, &dyn_str_table);
1052         num_symtab = hash [1]; /* FIXME */
1053         g_print ("num_sections: %d\n", num_sections);
1054         g_print ("dynsym: %d, dynstr size: %d\n", hash [1], dyn_str_table.data->len);
1055         for (i = 0; i < num_sections; ++i) {
1056                 g_print ("section %s, size: %d, %x\n", sections [i]->name, sections [i]->cur_offset, sections [i]->cur_offset);
1057         }
1058
1059         /* at this point we know where in the file the first segment sections go */
1060         dynsym = collect_syms (acfg, hash, &dyn_str_table, NULL, NULL);
1061         num_local_syms = hash [1];
1062         symtab = collect_syms (acfg, NULL, &str_table, secth, &num_local_syms);
1063
1064         for (i = 0; i < num_sections; ++i) {
1065                 if (sections [i]->shidx == SECT_TEXT) {
1066                         text_section = sections [i];
1067                 } else if (sections [i]->shidx == SECT_DATA) {
1068                         data_section = sections [i];
1069                 } else if (sections [i]->shidx == SECT_BSS) {
1070                         bss_section = sections [i];
1071                 }
1072         }
1073
1074         file_offset = virt_offset = sizeof (header) + sizeof (progh);
1075         secth [SECT_HASH].sh_addr = secth [SECT_HASH].sh_offset = file_offset;
1076         size = sizeof (int) * (2 + hash [0] + hash [1]);
1077         virt_offset = (file_offset += size);
1078         secth [SECT_HASH].sh_size = size;
1079         secth [SECT_DYNSYM].sh_addr = secth [SECT_DYNSYM].sh_offset = file_offset;
1080         size = sizeof (ElfSymbol) * hash [1];
1081         virt_offset = (file_offset += size);
1082         secth [SECT_DYNSYM].sh_size = size;
1083         secth [SECT_DYNSTR].sh_addr = secth [SECT_DYNSTR].sh_offset = file_offset;
1084         size = dyn_str_table.data->len;
1085         virt_offset = (file_offset += size);
1086         secth [SECT_DYNSTR].sh_size = size;
1087         file_offset += 4-1;
1088         file_offset &= ~(4-1);
1089         secth [SECT_REL_DYN].sh_addr = secth [SECT_REL_DYN].sh_offset = file_offset;
1090         size = sizeof (ElfReloc) * acfg->num_relocs;
1091         secth [SECT_REL_DYN].sh_size = size;
1092         virt_offset = (file_offset += size);
1093         secth [SECT_REL_DYN].sh_size = size;
1094         file_offset += 4096-1;
1095         file_offset &= ~(4096-1);
1096         virt_offset = file_offset;
1097         secth [SECT_TEXT].sh_addr = secth [SECT_TEXT].sh_offset = file_offset;
1098         size = text_section->cur_offset;
1099         secth [SECT_TEXT].sh_size = size;
1100         file_offset += size;
1101         file_offset += 4-1;
1102         file_offset &= ~(4-1);
1103         virt_offset = file_offset;
1104         /* .dynamic, .got.plt, .data, .bss here */
1105         secth [SECT_DYNAMIC].sh_addr = virt_offset;
1106         secth [SECT_DYNAMIC].sh_offset = file_offset;
1107         size = sizeof (dynamic);
1108         secth [SECT_DYNAMIC].sh_size = size;
1109         size += 4-1;
1110         size &= ~(4-1);
1111         file_offset += size;
1112         virt_offset += size;
1113         secth [SECT_GOT_PLT].sh_addr = virt_offset;
1114         secth [SECT_GOT_PLT].sh_offset = file_offset;
1115         size = 12;
1116         secth [SECT_GOT_PLT].sh_size = size;
1117         size += 8-1;
1118         size &= ~(8-1);
1119         file_offset += size;
1120         virt_offset += size;
1121         secth [SECT_DATA].sh_addr = virt_offset;
1122         secth [SECT_DATA].sh_offset = file_offset;
1123         size = data_section->cur_offset;
1124         secth [SECT_DATA].sh_size = size;
1125         size += 8-1;
1126         size &= ~(8-1);
1127         file_offset += size;
1128         virt_offset += size;
1129         secth [SECT_BSS].sh_addr = virt_offset;
1130         secth [SECT_BSS].sh_offset = file_offset;
1131         size = bss_section->cur_offset;
1132         secth [SECT_BSS].sh_size = size;
1133
1134         /* virtual doesn't matter anymore */
1135         secth [SECT_SHSTRTAB].sh_offset = file_offset;
1136         size = sh_str_table.data->len;
1137         secth [SECT_SHSTRTAB].sh_size = size;
1138         size += 4-1;
1139         size &= ~(4-1);
1140         file_offset += size;
1141         secth [SECT_SYMTAB].sh_offset = file_offset;
1142         size = sizeof (ElfSymbol) * num_local_syms;
1143         secth [SECT_SYMTAB].sh_size = size;
1144         file_offset += size;
1145         secth [SECT_STRTAB].sh_offset = file_offset;
1146         size = str_table.data->len;
1147         secth [SECT_STRTAB].sh_size = size;
1148         file_offset += size;
1149         file_offset += 4-1;
1150         file_offset &= ~(4-1);
1151
1152         text_section->file_offset = secth [SECT_TEXT].sh_offset;
1153         data_section->file_offset = secth [SECT_DATA].sh_offset;
1154         bss_section->file_offset = secth [SECT_BSS].sh_offset;
1155
1156         header.e_ident [0] = 0x7f; header.e_ident [1] = 'E';
1157         header.e_ident [2] = 'L'; header.e_ident [3] = 'F';
1158         header.e_ident [4] = SIZEOF_VOID_P == 4? 1: 2;
1159         header.e_ident [5] = 1; /* FIXME: little endian, bigendian is 2 */
1160         header.e_ident [6] = 1; /* version */
1161         header.e_ident [7] = 0; /* FIXME: */
1162         header.e_ident [8] = 0; /* FIXME: */
1163         for (i = 9; i < 16; ++i)
1164                 header.e_ident [i] = 0;
1165
1166         header.e_type = 3; /* shared library */
1167         header.e_machine = 3; /* FIXME: 386 */
1168         header.e_version = 1; /* FIXME:  */
1169
1170         header.e_phoff = sizeof (header);
1171         header.e_ehsize = sizeof (header);
1172         header.e_phentsize = sizeof (struct ElfProgHeader);
1173         header.e_phnum = 3;
1174         header.e_entry = secth [SECT_TEXT].sh_addr;
1175         header.e_shstrndx = 10;
1176         header.e_shentsize = sizeof (struct ElfSectHeader);
1177         header.e_shnum = SECT_NUM;
1178         header.e_shoff = file_offset;
1179
1180         /* dynamic data */
1181         i = 0;
1182         dynamic [i].d_tag = DYN_HASH;
1183         dynamic [i].d_val = secth [SECT_HASH].sh_offset;
1184         ++i;
1185         dynamic [i].d_tag = DYN_STRTAB;
1186         dynamic [i].d_val = secth [SECT_DYNSTR].sh_offset;
1187         ++i;
1188         dynamic [i].d_tag = DYN_SYMTAB;
1189         dynamic [i].d_val = secth [SECT_DYNSYM].sh_offset;
1190         ++i;
1191         dynamic [i].d_tag = DYN_STRSZ;
1192         dynamic [i].d_val = dyn_str_table.data->len;
1193         ++i;
1194         dynamic [i].d_tag = DYN_SYMENT;
1195         dynamic [i].d_val = sizeof (ElfSymbol);
1196         ++i;
1197         dynamic [i].d_tag = DYN_REL;
1198         dynamic [i].d_val = secth [SECT_REL_DYN].sh_offset;
1199         ++i;
1200         dynamic [i].d_tag = DYN_RELSZ;
1201         dynamic [i].d_val = secth [SECT_REL_DYN].sh_size;
1202         ++i;
1203         dynamic [i].d_tag = DYN_RELENT;
1204         dynamic [i].d_val = sizeof (ElfReloc);
1205         ++i;
1206         dynamic [i].d_tag = DYN_RELCOUNT;
1207         dynamic [i].d_val = acfg->num_relocs;
1208         ++i;
1209
1210         /* Program header */
1211         memset (&progh, 0, sizeof (progh));
1212         progh [0].p_type = 1; /* LOAD */
1213         progh [0].p_filesz = progh [0].p_memsz = secth [SECT_DYNAMIC].sh_offset;
1214         progh [0].p_align = 4096;
1215         progh [0].p_flags = 5;
1216
1217         progh [1].p_type = 1;
1218         progh [1].p_offset = secth [SECT_DYNAMIC].sh_offset;
1219         progh [1].p_vaddr = progh [1].p_paddr = secth [SECT_DYNAMIC].sh_addr;
1220         progh [1].p_filesz = secth [SECT_BSS].sh_offset  - secth [SECT_DYNAMIC].sh_offset;
1221         progh [1].p_memsz = secth [SECT_BSS].sh_addr + secth [SECT_BSS].sh_size - secth [SECT_DYNAMIC].sh_addr;
1222         progh [1].p_align = 4096;
1223         progh [1].p_flags = 6;
1224
1225         progh [2].p_type = 2; /* DYNAMIC */
1226         progh [2].p_offset = secth [SECT_DYNAMIC].sh_offset;
1227         progh [2].p_vaddr = progh [2].p_paddr = secth [SECT_DYNAMIC].sh_addr;
1228         progh [2].p_filesz = progh [2].p_memsz = secth [SECT_DYNAMIC].sh_size;
1229         progh [2].p_align = 4;
1230         progh [2].p_flags = 6;
1231
1232         reloc_symbols (acfg, dynsym, secth, &dyn_str_table, TRUE);
1233         reloc_symbols (acfg, symtab, secth, &str_table, FALSE);
1234         relocs = resolve_relocations (acfg);
1235
1236         fwrite (&header, sizeof (header), 1, file);
1237         fwrite (&progh, sizeof (progh), 1, file);
1238         fwrite (hash, sizeof (int) * (hash [0] + hash [1] + 2), 1, file);
1239         fwrite (dynsym, sizeof (ElfSymbol) * hash [1], 1, file);
1240         fwrite (dyn_str_table.data->str, dyn_str_table.data->len, 1, file);
1241         /* .rel.dyn */
1242         fseek (file, secth [SECT_REL_DYN].sh_offset, SEEK_SET);
1243         fwrite (relocs, sizeof (ElfReloc), acfg->num_relocs, file);
1244
1245         fseek (file, secth [SECT_TEXT].sh_offset, SEEK_SET);
1246         /* write .text, .data, .bss sections */
1247         fwrite (text_section->data, text_section->cur_offset, 1, file);
1248
1249         /* .dynamic */
1250         fwrite (dynamic, sizeof (dynamic), 1, file);
1251         /* .got.plt */
1252         size = secth [SECT_DYNAMIC].sh_addr;
1253         fwrite (&size, sizeof (size), 1, file);
1254         fseek (file, secth [SECT_DATA].sh_offset, SEEK_SET);
1255         fwrite (data_section->data, data_section->cur_offset, 1, file);
1256
1257         fseek (file, secth [SECT_SHSTRTAB].sh_offset, SEEK_SET);
1258         fwrite (sh_str_table.data->str, sh_str_table.data->len, 1, file);
1259         fseek (file, secth [SECT_SYMTAB].sh_offset, SEEK_SET);
1260         fwrite (symtab, sizeof (ElfSymbol) * num_local_syms, 1, file);
1261         fseek (file, secth [SECT_STRTAB].sh_offset, SEEK_SET);
1262         fwrite (str_table.data->str, str_table.data->len, 1, file);
1263         /*g_print ("file_offset %d vs %d\n", file_offset, ftell (file));*/
1264         /*g_assert (file_offset >= ftell (file));*/
1265         fseek (file, file_offset, SEEK_SET);
1266         fwrite (&secth, sizeof (secth), 1, file);
1267         fclose (file);
1268         rename (tmp_outfile_name, outfile_name);
1269
1270         g_free (tmp_outfile_name);
1271         g_free (outfile_name);
1272
1273         return 0;
1274 }
1275
1276 #endif /* USE_ELF_WRITER */
1277
1278 #else
1279
1280 static void
1281 emit_start (MonoAotCompile *acfg)
1282 {
1283         if (acfg->aot_opts.asm_only) {
1284                 if (acfg->aot_opts.outfile)
1285                         acfg->tmpfname = g_strdup_printf ("%s", acfg->aot_opts.outfile);
1286                 else
1287                         acfg->tmpfname = g_strdup_printf ("%s.s", acfg->image->name);
1288                 acfg->fp = fopen (acfg->tmpfname, "w+");
1289         } else {
1290                 int i = g_file_open_tmp ("mono_aot_XXXXXX", &acfg->tmpfname, NULL);
1291                 acfg->fp = fdopen (i, "w+");
1292         }
1293         g_assert (acfg->fp);
1294 }
1295
1296 static void
1297 emit_unset_mode (MonoAotCompile *acfg)
1298 {
1299         if (acfg->mode == EMIT_NONE)
1300                 return;
1301         fprintf (acfg->fp, "\n");
1302         acfg->mode = EMIT_NONE;
1303 }
1304
1305 static void
1306 emit_section_change (MonoAotCompile *acfg, const char *section_name, int subsection_index)
1307 {
1308         emit_unset_mode (acfg);
1309 #if defined(PLATFORM_WIN32)
1310         fprintf (acfg->fp, ".section %s\n", section_name);
1311 #elif defined(__MACH__)
1312         if (strcmp(section_name, ".bss") == 0)
1313                 fprintf (acfg->fp, "%s\n", ".data");
1314         else
1315                 fprintf (acfg->fp, "%s\n", section_name);
1316 #elif defined(sparc) || defined(__arm__)
1317         /* For solaris as, GNU as should accept the same */
1318         fprintf (acfg->fp, ".section \"%s\"\n", section_name);
1319 #else
1320         fprintf (acfg->fp, "%s %d\n", section_name, subsection_index);
1321 #endif
1322 }
1323
1324 static void
1325 emit_symbol_type (MonoAotCompile *acfg, const char *name, gboolean func)
1326 {
1327         const char *stype;
1328
1329         if (func)
1330                 stype = "function";
1331         else
1332                 stype = "object";
1333
1334         emit_unset_mode (acfg);
1335 #if defined(__MACH__)
1336
1337 #elif defined(sparc) || defined(__arm__)
1338         fprintf (acfg->fp, "\t.type %s,#%s\n", name, stype);
1339 #elif defined(PLATFORM_WIN32)
1340
1341 #elif defined(__x86_64__) || defined(__i386__)
1342         fprintf (acfg->fp, "\t.type %s,@%s\n", name, stype);
1343 #else
1344         fprintf (acfg->fp, "\t.type %s,@%s\n", name, stype);
1345 #endif
1346 }
1347
1348 static void
1349 emit_global_inner (MonoAotCompile *acfg, const char *name, gboolean func)
1350 {
1351         emit_unset_mode (acfg);
1352 #if  (defined(__ppc__) && defined(__MACH__)) || defined(PLATFORM_WIN32)
1353     // mach-o always uses a '_' prefix.
1354         fprintf (acfg->fp, "\t.globl _%s\n", name);
1355 #else
1356         fprintf (acfg->fp, "\t.globl %s\n", name);
1357 #endif
1358
1359         emit_symbol_type (acfg, name, func);
1360 }
1361
1362 static void
1363 emit_label (MonoAotCompile *acfg, const char *name)
1364 {
1365         emit_unset_mode (acfg);
1366 #if (defined(__ppc__) && defined(__MACH__)) || defined(PLATFORM_WIN32)
1367     // mach-o always uses a '_' prefix.
1368         fprintf (acfg->fp, "_%s:\n", name);
1369 #else
1370         fprintf (acfg->fp, "%s:\n", name);
1371 #endif
1372
1373 #if defined(PLATFORM_WIN32)
1374         /* Emit a normal label too */
1375         fprintf (acfg->fp, "%s:\n", name);
1376 #endif
1377 }
1378
1379 static void
1380 emit_string (MonoAotCompile *acfg, const char *value)
1381 {
1382         emit_unset_mode (acfg);
1383         fprintf (acfg->fp, "\t%s \"%s\"\n", AS_STRING_DIRECTIVE, value);
1384 }
1385
1386 static void
1387 emit_line (MonoAotCompile *acfg)
1388 {
1389         emit_unset_mode (acfg);
1390         fprintf (acfg->fp, "\n");
1391 }
1392
1393 static void
1394 emit_string_symbol (MonoAotCompile *acfg, const char *name, const char *value)
1395 {
1396         emit_unset_mode (acfg);
1397         emit_section_change (acfg, ".text", 1);
1398         emit_global (acfg, name, FALSE);
1399         emit_label (acfg, name);
1400         emit_string (acfg, value);
1401 }
1402
1403 static void 
1404 emit_alignment (MonoAotCompile *acfg, int size)
1405 {
1406         emit_unset_mode (acfg);
1407 #if defined(__arm__)
1408         fprintf (acfg->fp, "\t.align %d\n", ilog2 (size));
1409 #elif defined(__ppc__) && defined(__MACH__)
1410         // the mach-o assembler specifies alignments as powers of 2.
1411         fprintf (acfg->fp, "\t.align %d\t; ilog2\n", ilog2(size));
1412 #elif defined(__powerpc__)
1413         /* ignore on linux/ppc */
1414 #else
1415         fprintf (acfg->fp, "\t.align %d\n", size);
1416 #endif
1417 }
1418
1419 static void
1420 emit_pointer (MonoAotCompile *acfg, const char *target)
1421 {
1422         emit_unset_mode (acfg);
1423         emit_alignment (acfg, sizeof (gpointer));
1424 #if defined(__x86_64__)
1425         fprintf (acfg->fp, "\t.quad %s\n", target ? target : "0");
1426 #elif defined(sparc) && SIZEOF_VOID_P == 8
1427         fprintf (acfg->fp, "\t.xword %s\n", target ? target : "0");
1428 #else
1429         fprintf (acfg->fp, "\t.long %s\n", target ? target : "0");
1430 #endif
1431 }
1432
1433 static void
1434 emit_bytes (MonoAotCompile *acfg, const guint8* buf, int size)
1435 {
1436         int i;
1437         if (acfg->mode != EMIT_BYTE) {
1438                 acfg->mode = EMIT_BYTE;
1439                 acfg->col_count = 0;
1440         }
1441         for (i = 0; i < size; ++i, ++acfg->col_count) {
1442                 if ((acfg->col_count % 32) == 0)
1443                         fprintf (acfg->fp, "\n\t.byte ");
1444                 else
1445                         fprintf (acfg->fp, ",");
1446                 fprintf (acfg->fp, "0x%x", buf [i]);
1447         }
1448 }
1449
1450 static inline void
1451 emit_int16 (MonoAotCompile *acfg, int value)
1452 {
1453         if (acfg->mode != EMIT_WORD) {
1454                 acfg->mode = EMIT_WORD;
1455                 acfg->col_count = 0;
1456         }
1457         if ((acfg->col_count++ % 8) == 0)
1458 #if defined(__MACH__)
1459                 fprintf (acfg->fp, "\n\t.short ");
1460 #elif defined(__arm__)
1461                 /* FIXME: Use .hword on other archs as well */
1462                 fprintf (acfg->fp, "\n\t.hword ");
1463 #else
1464                 fprintf (acfg->fp, "\n\t.word ");
1465 #endif
1466         else
1467                 fprintf (acfg->fp, ", ");
1468         fprintf (acfg->fp, "%d", value);
1469 }
1470
1471 static inline void
1472 emit_int32 (MonoAotCompile *acfg, int value)
1473 {
1474         if (acfg->mode != EMIT_LONG) {
1475                 acfg->mode = EMIT_LONG;
1476                 acfg->col_count = 0;
1477         }
1478         if ((acfg->col_count++ % 8) == 0)
1479                 fprintf (acfg->fp, "\n\t.long ");
1480         else
1481                 fprintf (acfg->fp, ",");
1482         fprintf (acfg->fp, "%d", value);
1483 }
1484
1485 static void
1486 emit_symbol_diff (MonoAotCompile *acfg, const char *end, const char* start, int offset)
1487 {
1488         if (acfg->mode != EMIT_LONG) {
1489                 acfg->mode = EMIT_LONG;
1490                 acfg->col_count = 0;
1491         }
1492         if ((acfg->col_count++ % 8) == 0)
1493                 fprintf (acfg->fp, "\n\t.long ");
1494         else
1495                 fprintf (acfg->fp, ",");
1496         if (offset > 0)
1497                 fprintf (acfg->fp, "%s - %s + %d", end, start, offset);
1498         else if (offset < 0)
1499                 fprintf (acfg->fp, "%s - %s %d", end, start, offset);
1500         else
1501                 fprintf (acfg->fp, "%s - %s", end, start);
1502 }
1503
1504 static void
1505 emit_zero_bytes (MonoAotCompile *acfg, int num)
1506 {
1507         emit_unset_mode (acfg);
1508 #if defined(__MACH__)
1509         fprintf (acfg->fp, "\t.space %d\n", num);
1510 #else
1511         fprintf (acfg->fp, "\t.skip %d\n", num);
1512 #endif
1513 }
1514
1515 static int
1516 emit_writeout (MonoAotCompile *acfg)
1517 {
1518         char *command, *objfile;
1519         char *outfile_name, *tmp_outfile_name;
1520
1521         fclose (acfg->fp);
1522
1523 #if defined(__x86_64__)
1524 #define AS_OPTIONS "--64"
1525 #elif defined(sparc) && SIZEOF_VOID_P == 8
1526 #define AS_OPTIONS "-xarch=v9"
1527 #else
1528 #define AS_OPTIONS ""
1529 #endif
1530
1531         if (acfg->aot_opts.asm_only) {
1532                 printf ("Output file: '%s'.\n", acfg->tmpfname);
1533                 if (acfg->aot_opts.static_link)
1534                         printf ("Linking symbol: '%s'.\n", acfg->static_linking_symbol);
1535                 return 0;
1536         }
1537
1538         if (acfg->aot_opts.static_link) {
1539                 if (acfg->aot_opts.outfile)
1540                         objfile = g_strdup_printf ("%s", acfg->aot_opts.outfile);
1541                 else
1542                         objfile = g_strdup_printf ("%s.o", acfg->image->name);
1543         } else {
1544                 objfile = g_strdup_printf ("%s.o", acfg->tmpfname);
1545         }
1546         command = g_strdup_printf ("as %s %s -o %s", AS_OPTIONS, acfg->tmpfname, objfile);
1547         printf ("Executing the native assembler: %s\n", command);
1548         if (system (command) != 0) {
1549                 g_free (command);
1550                 g_free (objfile);
1551                 return 1;
1552         }
1553
1554         g_free (command);
1555
1556         if (acfg->aot_opts.static_link) {
1557                 printf ("Output file: '%s'.\n", objfile);
1558                 printf ("Linking symbol: '%s'.\n", acfg->static_linking_symbol);
1559                 g_free (objfile);
1560                 return 0;
1561         }
1562
1563         if (acfg->aot_opts.outfile)
1564                 outfile_name = g_strdup_printf ("%s", acfg->aot_opts.outfile);
1565         else
1566                 outfile_name = g_strdup_printf ("%s%s", acfg->image->name, SHARED_EXT);
1567
1568         tmp_outfile_name = g_strdup_printf ("%s.tmp", outfile_name);
1569
1570 #if defined(sparc)
1571         command = g_strdup_printf ("ld -shared -G -o %s %s.o", outfile_name, acfg->tmpfname);
1572 #elif defined(__ppc__) && defined(__MACH__)
1573         command = g_strdup_printf ("gcc -dynamiclib -o %s %s.o", outfile_name, acfg->tmpfname);
1574 #elif defined(PLATFORM_WIN32)
1575         command = g_strdup_printf ("gcc -shared --dll -mno-cygwin -o %s %s.o", outfile_name, acfg->tmpfname);
1576 #else
1577         if (acfg->aot_opts.no_dlsym) {
1578                 /* 
1579                  * Need to link using gcc so our ctor function gets called.
1580                  */
1581                 command = g_strdup_printf ("gcc -shared -o %s %s.o", outfile_name, acfg->tmpfname);
1582         } else {
1583                 command = g_strdup_printf ("ld -shared -o %s %s.o", outfile_name, acfg->tmpfname);
1584         }
1585 #endif
1586         printf ("Executing the native linker: %s\n", command);
1587         if (system (command) != 0) {
1588                 g_free (tmp_outfile_name);
1589                 g_free (outfile_name);
1590                 g_free (command);
1591                 g_free (objfile);
1592                 return 1;
1593         }
1594
1595         g_free (command);
1596         unlink (objfile);
1597         /*com = g_strdup_printf ("strip --strip-unneeded %s%s", acfg->image->name, SHARED_EXT);
1598         printf ("Stripping the binary: %s\n", com);
1599         system (com);
1600         g_free (com);*/
1601
1602         rename (tmp_outfile_name, outfile_name);
1603
1604         g_free (tmp_outfile_name);
1605         g_free (outfile_name);
1606         g_free (objfile);
1607
1608         if (acfg->aot_opts.save_temps)
1609                 printf ("Retained input file.\n");
1610         else
1611                 unlink (acfg->tmpfname);
1612
1613         return 0;
1614 }
1615
1616 #endif /* ASM_WRITER */
1617
1618 static void
1619 emit_global (MonoAotCompile *acfg, const char *name, gboolean func)
1620 {
1621         if (acfg->aot_opts.no_dlsym) {
1622                 g_ptr_array_add (acfg->globals, g_strdup (name));
1623         } else {
1624                 emit_global_inner (acfg, name, func);
1625         }
1626 }
1627
1628 static void
1629 emit_byte (MonoAotCompile *acfg, guint8 val)
1630 {
1631         emit_bytes (acfg, &val, 1);
1632 }
1633
1634 static guint32
1635 mono_get_field_token (MonoClassField *field) 
1636 {
1637         MonoClass *klass = field->parent;
1638         int i;
1639
1640         for (i = 0; i < klass->field.count; ++i) {
1641                 if (field == &klass->fields [i])
1642                         return MONO_TOKEN_FIELD_DEF | (klass->field.first + 1 + i);
1643         }
1644
1645         g_assert_not_reached ();
1646         return 0;
1647 }
1648
1649 static inline void
1650 encode_value (gint32 value, guint8 *buf, guint8 **endbuf)
1651 {
1652         guint8 *p = buf;
1653
1654         //printf ("ENCODE: %d 0x%x.\n", value, value);
1655
1656         /* 
1657          * Same encoding as the one used in the metadata, extended to handle values
1658          * greater than 0x1fffffff.
1659          */
1660         if ((value >= 0) && (value <= 127))
1661                 *p++ = value;
1662         else if ((value >= 0) && (value <= 16383)) {
1663                 p [0] = 0x80 | (value >> 8);
1664                 p [1] = value & 0xff;
1665                 p += 2;
1666         } else if ((value >= 0) && (value <= 0x1fffffff)) {
1667                 p [0] = (value >> 24) | 0xc0;
1668                 p [1] = (value >> 16) & 0xff;
1669                 p [2] = (value >> 8) & 0xff;
1670                 p [3] = value & 0xff;
1671                 p += 4;
1672         }
1673         else {
1674                 p [0] = 0xff;
1675                 p [1] = (value >> 24) & 0xff;
1676                 p [2] = (value >> 16) & 0xff;
1677                 p [3] = (value >> 8) & 0xff;
1678                 p [4] = value & 0xff;
1679                 p += 5;
1680         }
1681         if (endbuf)
1682                 *endbuf = p;
1683 }
1684
1685 static guint32
1686 get_image_index (MonoAotCompile *cfg, MonoImage *image)
1687 {
1688         guint32 index;
1689
1690         index = GPOINTER_TO_UINT (g_hash_table_lookup (cfg->image_hash, image));
1691         if (index)
1692                 return index - 1;
1693         else {
1694                 index = g_hash_table_size (cfg->image_hash);
1695                 g_hash_table_insert (cfg->image_hash, image, GUINT_TO_POINTER (index + 1));
1696                 g_ptr_array_add (cfg->image_table, image);
1697                 return index;
1698         }
1699 }
1700
1701 static guint32
1702 find_typespec_for_class (MonoAotCompile *acfg, MonoClass *klass)
1703 {
1704         int i;
1705         MonoClass *k = NULL;
1706
1707         /* FIXME: Search referenced images as well */
1708         for (i = 0; i < acfg->image->tables [MONO_TABLE_TYPESPEC].rows; ++i) {
1709                 k = mono_class_get_full (acfg->image, MONO_TOKEN_TYPE_SPEC | (i + 1), NULL);
1710                 if (k == klass)
1711                         break;
1712         }
1713
1714         if (i < acfg->image->tables [MONO_TABLE_TYPESPEC].rows)
1715                 return MONO_TOKEN_TYPE_SPEC | (i + 1);
1716         else
1717                 return 0;
1718 }
1719
1720 static void
1721 encode_method_ref (MonoAotCompile *acfg, MonoMethod *method, guint8 *buf, guint8 **endbuf);
1722
1723 /*
1724  * encode_klass_ref:
1725  *
1726  *   Encode a reference to KLASS. We use our home-grown encoding instead of the
1727  * standard metadata encoding.
1728  */
1729 static void
1730 encode_klass_ref (MonoAotCompile *acfg, MonoClass *klass, guint8 *buf, guint8 **endbuf)
1731 {
1732         guint8 *p = buf;
1733
1734         if (klass->generic_class) {
1735                 guint32 token;
1736                 g_assert (klass->type_token);
1737
1738                 /* Find a typespec for a class if possible */
1739                 token = find_typespec_for_class (acfg, klass);
1740                 if (token) {
1741                         encode_value (token, p, &p);
1742                         encode_value (get_image_index (acfg, acfg->image), p, &p);
1743                 } else {
1744                         MonoClass *gclass = klass->generic_class->container_class;
1745                         MonoGenericInst *inst = klass->generic_class->context.class_inst;
1746                         int i;
1747
1748                         /* Encode it ourselves */
1749                         /* Marker */
1750                         encode_value (MONO_TOKEN_TYPE_SPEC, p, &p);
1751                         encode_value (MONO_TYPE_GENERICINST, p, &p);
1752                         encode_klass_ref (acfg, gclass, p, &p);
1753                         encode_value (inst->type_argc, p, &p);
1754                         for (i = 0; i < inst->type_argc; ++i)
1755                                 encode_klass_ref (acfg, mono_class_from_mono_type (inst->type_argv [i]), p, &p);
1756                 }
1757         } else if (klass->type_token) {
1758                 g_assert (mono_metadata_token_code (klass->type_token) == MONO_TOKEN_TYPE_DEF);
1759                 encode_value (klass->type_token - MONO_TOKEN_TYPE_DEF, p, &p);
1760                 encode_value (get_image_index (acfg, klass->image), p, &p);
1761         } else if ((klass->byval_arg.type == MONO_TYPE_VAR) || (klass->byval_arg.type == MONO_TYPE_MVAR)) {
1762                 MonoGenericParam *param = klass->byval_arg.data.generic_param;
1763
1764                 /* Marker */
1765                 encode_value (MONO_TOKEN_TYPE_SPEC, p, &p);
1766                 encode_value (klass->byval_arg.type, p, &p);
1767
1768                 encode_value (param->num, p, &p);
1769                 
1770                 g_assert (param->owner);
1771                 encode_value (param->owner->is_method, p, &p);
1772                 if (param->owner->is_method)
1773                         encode_method_ref (acfg, param->owner->owner.method, p, &p);
1774                 else
1775                         encode_klass_ref (acfg, param->owner->owner.klass, p, &p);
1776         } else {
1777                 /* Array class */
1778                 g_assert (klass->rank > 0);
1779                 encode_value (MONO_TOKEN_TYPE_DEF, p, &p);
1780                 encode_value (get_image_index (acfg, klass->image), p, &p);
1781                 encode_value (klass->rank, p, &p);
1782                 encode_klass_ref (acfg, klass->element_class, p, &p);
1783         }
1784         *endbuf = p;
1785 }
1786
1787 static void
1788 encode_field_info (MonoAotCompile *cfg, MonoClassField *field, guint8 *buf, guint8 **endbuf)
1789 {
1790         guint32 token = mono_get_field_token (field);
1791         guint8 *p = buf;
1792
1793         encode_klass_ref (cfg, field->parent, p, &p);
1794         g_assert (mono_metadata_token_code (token) == MONO_TOKEN_FIELD_DEF);
1795         encode_value (token - MONO_TOKEN_FIELD_DEF, p, &p);
1796         *endbuf = p;
1797 }
1798
1799 #define MAX_IMAGE_INDEX 250
1800
1801 static void
1802 encode_method_ref (MonoAotCompile *acfg, MonoMethod *method, guint8 *buf, guint8 **endbuf)
1803 {
1804         guint32 image_index = get_image_index (acfg, method->klass->image);
1805         guint32 token = method->token;
1806         MonoJumpInfoToken *ji;
1807         guint8 *p = buf;
1808
1809         g_assert (image_index < MAX_IMAGE_INDEX);
1810
1811         /* Mark methods which can't use aot trampolines because they need the further 
1812          * processing in mono_magic_trampoline () which requires a MonoMethod*.
1813          */
1814         if (method->is_generic && (method->flags & METHOD_ATTRIBUTE_VIRTUAL))
1815                 encode_value ((252 << 24), p, &p);
1816
1817         if (method->wrapper_type) {
1818                 /* Marker */
1819                 encode_value ((253 << 24), p, &p);
1820
1821                 encode_value (method->wrapper_type, p, &p);
1822
1823                 switch (method->wrapper_type) {
1824                 case MONO_WRAPPER_REMOTING_INVOKE:
1825                 case MONO_WRAPPER_REMOTING_INVOKE_WITH_CHECK:
1826                 case MONO_WRAPPER_XDOMAIN_INVOKE: {
1827                         MonoMethod *m;
1828
1829                         m = mono_marshal_method_from_wrapper (method);
1830                         g_assert (m);
1831                         encode_method_ref (acfg, m, p, &p);
1832                         break;
1833                 }
1834                 case MONO_WRAPPER_PROXY_ISINST:
1835                 case MONO_WRAPPER_LDFLD:
1836                 case MONO_WRAPPER_LDFLDA:
1837                 case MONO_WRAPPER_STFLD:
1838                 case MONO_WRAPPER_ISINST: {
1839                         MonoClass *proxy_class = (MonoClass*)mono_marshal_method_from_wrapper (method);
1840                         encode_klass_ref (acfg, proxy_class, p, &p);
1841                         break;
1842                 }
1843                 case MONO_WRAPPER_LDFLD_REMOTE:
1844                 case MONO_WRAPPER_STFLD_REMOTE:
1845                         break;
1846                 case MONO_WRAPPER_ALLOC: {
1847                         int alloc_type = mono_gc_get_managed_allocator_type (method);
1848                         g_assert (alloc_type != -1);
1849                         encode_value (alloc_type, p, &p);
1850                         break;
1851                 }
1852                 case MONO_WRAPPER_STELEMREF:
1853                         break;
1854                 case MONO_WRAPPER_STATIC_RGCTX_INVOKE: {
1855                         MonoMethod *m;
1856
1857                         m = mono_marshal_method_from_wrapper (method);
1858                         g_assert (m);
1859                         encode_method_ref (acfg, m, p, &p);
1860                         break;
1861                 }
1862                 default:
1863                         g_assert_not_reached ();
1864                 }
1865         } else if (mono_method_signature (method)->is_inflated) {
1866                 /* 
1867                  * This is a generic method, find the original token which referenced it and
1868                  * encode that.
1869                  * Obtain the token from information recorded by the JIT.
1870                  */
1871                 ji = g_hash_table_lookup (acfg->token_info_hash, method);
1872                 if (ji) {
1873                         image_index = get_image_index (acfg, ji->image);
1874                         g_assert (image_index < MAX_IMAGE_INDEX);
1875                         token = ji->token;
1876
1877                         /* Marker */
1878                         encode_value ((255 << 24), p, &p);
1879                         encode_value (image_index, p, &p);
1880                         encode_value (token, p, &p);
1881                 } else {
1882                         MonoMethod *declaring;
1883                         MonoGenericContext *context = mono_method_get_context (method);
1884                         MonoGenericInst *inst;
1885                         int i;
1886
1887                         g_assert (method->is_inflated);
1888                         declaring = ((MonoMethodInflated*)method)->declaring;
1889
1890                         /*
1891                          * This might be a non-generic method of a generic instance, which 
1892                          * doesn't have a token since the reference is generated by the JIT 
1893                          * like Nullable:Box/Unbox, or by generic sharing.
1894                          */
1895
1896                         /* Marker */
1897                         encode_value ((254 << 24), p, &p);
1898                         /* Encode the klass */
1899                         encode_klass_ref (acfg, method->klass, p, &p);
1900                         /* Encode the method */
1901                         image_index = get_image_index (acfg, method->klass->image);
1902                         g_assert (image_index < MAX_IMAGE_INDEX);
1903                         g_assert (declaring->token);
1904                         token = declaring->token;
1905                         g_assert (mono_metadata_token_table (token) == MONO_TABLE_METHOD);
1906                         encode_value (image_index, p, &p);
1907                         encode_value (token, p, &p);
1908
1909                         /* Encode the context */
1910                         inst = context->class_inst;
1911                         encode_value (inst ? 1 : 0, p, &p);
1912                         if (inst) {
1913                                 encode_value (inst->type_argc, p, &p);
1914                                 for (i = 0; i < inst->type_argc; ++i)
1915                                         encode_klass_ref (acfg, mono_class_from_mono_type (inst->type_argv [i]), p, &p);
1916                         }
1917                         inst = context->method_inst;
1918                         encode_value (inst ? 1 : 0, p, &p);
1919                         if (inst) {
1920                                 encode_value (inst->type_argc, p, &p);
1921                                 for (i = 0; i < inst->type_argc; ++i)
1922                                         encode_klass_ref (acfg, mono_class_from_mono_type (inst->type_argv [i]), p, &p);
1923                         }
1924                 }
1925         } else if (token == 0) {
1926                 /* This might be a method of a constructed type like int[,].Set */
1927                 /* Obtain the token from information recorded by the JIT */
1928                 ji = g_hash_table_lookup (acfg->token_info_hash, method);
1929                 g_assert (ji);
1930                 image_index = get_image_index (acfg, ji->image);
1931                 g_assert (image_index < MAX_IMAGE_INDEX);
1932                 token = ji->token;
1933
1934                 /* Marker */
1935                 encode_value ((255 << 24), p, &p);
1936                 encode_value (image_index, p, &p);
1937                 encode_value (token, p, &p);
1938         } else {
1939                 g_assert (mono_metadata_token_table (token) == MONO_TABLE_METHOD);
1940                 encode_value ((image_index << 24) | mono_metadata_token_index (token), p, &p);
1941         }
1942         *endbuf = p;
1943 }
1944
1945 static gint
1946 compare_patches (gconstpointer a, gconstpointer b)
1947 {
1948         int i, j;
1949
1950         i = (*(MonoJumpInfo**)a)->ip.i;
1951         j = (*(MonoJumpInfo**)b)->ip.i;
1952
1953         if (i < j)
1954                 return -1;
1955         else
1956                 if (i > j)
1957                         return 1;
1958         else
1959                 return 0;
1960 }
1961
1962 /*
1963  * is_plt_patch:
1964  *
1965  *   Return whenever PATCH_INFO refers to a direct call, and thus requires a
1966  * PLT entry.
1967  */
1968 static inline gboolean
1969 is_plt_patch (MonoJumpInfo *patch_info)
1970 {
1971         switch (patch_info->type) {
1972         case MONO_PATCH_INFO_METHOD:
1973         case MONO_PATCH_INFO_INTERNAL_METHOD:
1974         case MONO_PATCH_INFO_JIT_ICALL_ADDR:
1975         case MONO_PATCH_INFO_CLASS_INIT:
1976         case MONO_PATCH_INFO_RGCTX_FETCH:
1977         case MONO_PATCH_INFO_GENERIC_CLASS_INIT:
1978                 return TRUE;
1979         default:
1980                 return FALSE;
1981         }
1982 }
1983
1984 /*
1985  * is_shared_got_patch:
1986  *
1987  *   Return whenever PATCH_INFO refers to a patch which needs a shared GOT
1988  * entry.
1989  * Keep it in sync with the version in aot-runtime.c.
1990  */
1991 static inline gboolean
1992 is_shared_got_patch (MonoJumpInfo *patch_info)
1993 {
1994         switch (patch_info->type) {
1995         case MONO_PATCH_INFO_VTABLE:
1996         case MONO_PATCH_INFO_CLASS:
1997         case MONO_PATCH_INFO_IID:
1998         case MONO_PATCH_INFO_ADJUSTED_IID:
1999         case MONO_PATCH_INFO_FIELD:
2000         case MONO_PATCH_INFO_SFLDA:
2001         case MONO_PATCH_INFO_DECLSEC:
2002         case MONO_PATCH_INFO_LDTOKEN:
2003         case MONO_PATCH_INFO_TYPE_FROM_HANDLE:
2004         case MONO_PATCH_INFO_RVA:
2005         case MONO_PATCH_INFO_METHODCONST:
2006                 return TRUE;
2007         default:
2008                 return FALSE;
2009         }
2010 }
2011
2012 static int
2013 get_plt_offset (MonoAotCompile *acfg, MonoJumpInfo *patch_info)
2014 {
2015         int res = -1;
2016
2017         if (is_plt_patch (patch_info)) {
2018                 int idx = GPOINTER_TO_UINT (g_hash_table_lookup (acfg->patch_to_plt_offset, patch_info));
2019
2020                 if (idx) {
2021                         res = idx;
2022                 } else {
2023                         MonoJumpInfo *new_ji = mono_patch_info_dup_mp (acfg->mempool, patch_info);
2024
2025                         res = acfg->plt_offset;
2026                         g_hash_table_insert (acfg->plt_offset_to_patch, GUINT_TO_POINTER (res), new_ji);
2027                         g_hash_table_insert (acfg->patch_to_plt_offset, new_ji, GUINT_TO_POINTER (res));
2028                         acfg->plt_offset ++;
2029                 }
2030         }
2031
2032         return res;
2033 }
2034
2035 /**
2036  * get_got_offset:
2037  *
2038  *   Returns the offset of the GOT slot where the runtime object resulting from resolving
2039  * JI could be found if it exists, otherwise allocates a new one.
2040  */
2041 static guint32
2042 get_got_offset (MonoAotCompile *acfg, MonoJumpInfo *ji)
2043 {
2044         guint32 got_offset;
2045
2046         got_offset = GPOINTER_TO_UINT (g_hash_table_lookup (acfg->patch_to_shared_got_offset, ji));
2047         if (got_offset)
2048                 return got_offset - 1;
2049
2050         got_offset = acfg->got_offset;
2051         acfg->got_offset ++;
2052
2053         acfg->stats.got_slots ++;
2054         acfg->stats.got_slot_types [ji->type] ++;
2055
2056         return got_offset;
2057 }
2058
2059 static guint32
2060 get_shared_got_offset (MonoAotCompile *acfg, MonoJumpInfo *ji)
2061 {
2062         MonoJumpInfo *copy;
2063         guint32 got_offset;
2064
2065         if (!g_hash_table_lookup (acfg->patch_to_shared_got_offset, ji)) {
2066                 got_offset = get_got_offset (acfg, ji);
2067                 copy = mono_patch_info_dup_mp (acfg->mempool, ji);
2068                 g_hash_table_insert (acfg->patch_to_shared_got_offset, copy, GUINT_TO_POINTER (got_offset + 1));
2069                 g_ptr_array_add (acfg->shared_patches, copy);
2070         }
2071
2072         return get_got_offset (acfg, ji);
2073 }
2074
2075 /* Add a method to the list of methods which need to be emitted */
2076 static void
2077 add_method_with_index (MonoAotCompile *acfg, MonoMethod *method, int index)
2078 {
2079         if (!g_hash_table_lookup (acfg->method_indexes, method)) {
2080                 g_ptr_array_add (acfg->methods, method);
2081                 g_hash_table_insert (acfg->method_indexes, method, GUINT_TO_POINTER (index + 1));
2082         }
2083 }
2084
2085 static guint32
2086 get_method_index (MonoAotCompile *acfg, MonoMethod *method)
2087 {
2088         int index = GPOINTER_TO_UINT (g_hash_table_lookup (acfg->method_indexes, method));
2089         
2090         g_assert (index);
2091
2092         return index - 1;
2093 }
2094
2095 static int
2096 add_method (MonoAotCompile *acfg, MonoMethod *method)
2097 {
2098         int index;
2099
2100         index = GPOINTER_TO_UINT (g_hash_table_lookup (acfg->method_indexes, method));
2101         if (index)
2102                 return index - 1;
2103
2104         index = acfg->method_index;
2105         add_method_with_index (acfg, method, index);
2106
2107         /* FIXME: Fix quadratic behavior */
2108         acfg->method_order = g_list_append (acfg->method_order, GUINT_TO_POINTER (index));
2109
2110         acfg->method_index ++;
2111
2112         return index;
2113 }
2114
2115 static void
2116 add_jit_icall_wrapper (gpointer key, gpointer value, gpointer user_data)
2117 {
2118         MonoAotCompile *acfg = user_data;
2119         MonoJitICallInfo *callinfo = value;
2120         MonoMethod *wrapper;
2121         char *name;
2122
2123         if (!callinfo->sig)
2124                 return;
2125
2126         name = g_strdup_printf ("__icall_wrapper_%s", callinfo->name);
2127         wrapper = mono_marshal_get_icall_wrapper (callinfo->sig, name, callinfo->func, check_for_pending_exc);
2128         g_free (name);
2129
2130         add_method (acfg, wrapper);
2131 }
2132
2133 static MonoMethod*
2134 get_runtime_invoke_sig (MonoMethodSignature *sig)
2135 {
2136         MonoMethodBuilder *mb;
2137         MonoMethod *m;
2138
2139         mb = mono_mb_new (mono_defaults.object_class, "FOO", MONO_WRAPPER_NONE);
2140         m = mono_mb_create_method (mb, sig, 16);
2141         return mono_marshal_get_runtime_invoke (m);
2142 }
2143
2144 static void
2145 add_wrappers (MonoAotCompile *acfg)
2146 {
2147         MonoMethod *method, *m;
2148         int i, j, nallocators;
2149         MonoMethodSignature *sig, *csig;
2150         guint32 token;
2151
2152         /* 
2153          * FIXME: Instead of AOTing all the wrappers, it might be better to redesign them
2154          * so there is only one wrapper of a given type, or inlining their contents into their
2155          * callers.
2156          */
2157
2158         /* 
2159          * FIXME: This depends on the fact that different wrappers have different 
2160          * names.
2161          */
2162
2163         /* FIXME: Collect these automatically */
2164
2165         /* Runtime invoke wrappers */
2166
2167         /* void runtime-invoke () [.cctor] */
2168         csig = mono_metadata_signature_alloc (mono_defaults.corlib, 0);
2169         csig->ret = &mono_defaults.void_class->byval_arg;
2170         add_method (acfg, get_runtime_invoke_sig (csig));
2171
2172         /* void runtime-invoke () [Finalize] */
2173         csig = mono_metadata_signature_alloc (mono_defaults.corlib, 0);
2174         csig->hasthis = 1;
2175         csig->ret = &mono_defaults.void_class->byval_arg;
2176         add_method (acfg, get_runtime_invoke_sig (csig));
2177
2178         /* void runtime-invoke (string) [exception ctor] */
2179         csig = mono_metadata_signature_alloc (mono_defaults.corlib, 1);
2180         csig->hasthis = 1;
2181         csig->ret = &mono_defaults.void_class->byval_arg;
2182         csig->params [0] = &mono_defaults.string_class->byval_arg;
2183         add_method (acfg, get_runtime_invoke_sig (csig));
2184
2185         /* void runtime-invoke (string, string) [exception ctor] */
2186         csig = mono_metadata_signature_alloc (mono_defaults.corlib, 2);
2187         csig->hasthis = 1;
2188         csig->ret = &mono_defaults.void_class->byval_arg;
2189         csig->params [0] = &mono_defaults.string_class->byval_arg;
2190         csig->params [1] = &mono_defaults.string_class->byval_arg;
2191         add_method (acfg, get_runtime_invoke_sig (csig));
2192
2193         /* string runtime-invoke () [Exception.ToString ()] */
2194         csig = mono_metadata_signature_alloc (mono_defaults.corlib, 0);
2195         csig->hasthis = 1;
2196         csig->ret = &mono_defaults.string_class->byval_arg;
2197         add_method (acfg, get_runtime_invoke_sig (csig));
2198
2199         /* void runtime-invoke (string, Exception) [exception ctor] */
2200         csig = mono_metadata_signature_alloc (mono_defaults.corlib, 2);
2201         csig->hasthis = 1;
2202         csig->ret = &mono_defaults.void_class->byval_arg;
2203         csig->params [0] = &mono_defaults.string_class->byval_arg;
2204         csig->params [1] = &mono_defaults.exception_class->byval_arg;
2205         add_method (acfg, get_runtime_invoke_sig (csig));
2206
2207         /* Assembly runtime-invoke (string, bool) [DoAssemblyResolve] */
2208         csig = mono_metadata_signature_alloc (mono_defaults.corlib, 2);
2209         csig->hasthis = 1;
2210         csig->ret = &(mono_class_from_name (
2211                                                                                 mono_defaults.corlib, "System.Reflection", "Assembly"))->byval_arg;
2212         csig->params [0] = &mono_defaults.string_class->byval_arg;
2213         csig->params [1] = &mono_defaults.boolean_class->byval_arg;
2214         add_method (acfg, get_runtime_invoke_sig (csig));
2215
2216         for (i = 0; i < acfg->image->tables [MONO_TABLE_METHOD].rows; ++i) {
2217                 MonoMethod *method;
2218                 guint32 token = MONO_TOKEN_METHOD_DEF | (i + 1);
2219                 gboolean skip = FALSE;
2220
2221                 method = mono_get_method (acfg->image, token, NULL);
2222
2223                 if ((method->flags & METHOD_ATTRIBUTE_PINVOKE_IMPL) ||
2224                         (method->iflags & METHOD_IMPL_ATTRIBUTE_RUNTIME) ||
2225                         (method->flags & METHOD_ATTRIBUTE_ABSTRACT))
2226                         skip = TRUE;
2227
2228                 if (method->is_generic || method->klass->generic_container)
2229                         skip = TRUE;
2230
2231                 /* Skip methods which can not be handled by get_runtime_invoke () */
2232                 sig = mono_method_signature (method);
2233                 if ((sig->ret->type == MONO_TYPE_PTR) ||
2234                         (sig->ret->type == MONO_TYPE_TYPEDBYREF))
2235                         skip = TRUE;
2236
2237                 for (j = 0; j < sig->param_count; j++) {
2238                         if (sig->params [j]->type == MONO_TYPE_TYPEDBYREF)
2239                                 skip = TRUE;
2240                 }
2241
2242                 if (!skip)
2243                         add_method (acfg, mono_marshal_get_runtime_invoke (method));
2244         }
2245
2246         if (strcmp (acfg->image->assembly->aname.name, "mscorlib") == 0) {
2247                 /* JIT icall wrappers */
2248                 /* FIXME: locking */
2249                 g_hash_table_foreach (mono_get_jit_icall_info (), add_jit_icall_wrapper, acfg);
2250
2251                 /* Managed Allocators */
2252                 nallocators = mono_gc_get_managed_allocator_types ();
2253                 for (i = 0; i < nallocators; ++i) {
2254                         m = mono_gc_get_managed_allocator_by_type (i);
2255                         if (m)
2256                                 add_method (acfg, m);
2257                 }
2258
2259                 /* stelemref */
2260                 add_method (acfg, mono_marshal_get_stelemref ());
2261         }
2262
2263         /* remoting-invoke wrappers */
2264         for (i = 0; i < acfg->image->tables [MONO_TABLE_METHOD].rows; ++i) {
2265                 MonoMethodSignature *sig;
2266                 
2267                 token = MONO_TOKEN_METHOD_DEF | (i + 1);
2268                 method = mono_get_method (acfg->image, token, NULL);
2269
2270                 sig = mono_method_signature (method);
2271
2272                 if (sig->hasthis && (method->klass->marshalbyref || method->klass == mono_defaults.object_class) && 
2273                         !(method->flags & METHOD_ATTRIBUTE_VIRTUAL)) {
2274                         m = mono_marshal_get_remoting_invoke_with_check (method);
2275
2276                         add_method (acfg, m);
2277                 }
2278         }
2279
2280         /* delegate-invoke wrappers */
2281         for (i = 0; i < acfg->image->tables [MONO_TABLE_TYPEDEF].rows; ++i) {
2282                 MonoClass *klass;
2283                 
2284                 token = MONO_TOKEN_TYPE_DEF | (i + 1);
2285                 klass = mono_class_get (acfg->image, token);
2286
2287                 if (klass->delegate && klass != mono_defaults.delegate_class && klass != mono_defaults.multicastdelegate_class) {
2288                         method = mono_get_delegate_invoke (klass);
2289
2290                         m = mono_marshal_get_delegate_invoke (method, NULL);
2291
2292                         add_method (acfg, m);
2293                 }
2294         }
2295
2296 #if 0
2297         /* static rgctx wrappers */
2298         /* FIXME: Each wrapper belongs to a given instantiation of a generic method */
2299         for (i = 0; i < acfg->image->tables [MONO_TABLE_METHOD].rows; ++i) {
2300                 token = MONO_TOKEN_METHOD_DEF | (i + 1);
2301                 method = mono_get_method (acfg->image, token, NULL);
2302
2303                 if (((method->flags & METHOD_ATTRIBUTE_STATIC) ||
2304                          (method->is_inflated && mono_method_get_context (method)->method_inst)) &&
2305                         mono_class_generic_sharing_enabled (method->klass) &&
2306                         mono_method_is_generic_sharable_impl (method, FALSE)) {
2307                         m = mono_marshal_get_static_rgctx_invoke (method);
2308                         add_method (acfg, m);
2309                 }
2310         }
2311 #endif
2312 }
2313
2314 static void
2315 emit_and_reloc_code (MonoAotCompile *acfg, MonoMethod *method, guint8 *code, guint32 code_len, MonoJumpInfo *relocs, gboolean got_only)
2316 {
2317         int i, pindex, start_index, method_index;
2318         GPtrArray *patches;
2319         MonoJumpInfo *patch_info;
2320         MonoMethodHeader *header;
2321         gboolean skip;
2322         guint32 got_slot;
2323
2324         if (method) {
2325                 header = mono_method_get_header (method);
2326
2327                 method_index = get_method_index (acfg, method);
2328         }
2329
2330         /* Collect and sort relocations */
2331         patches = g_ptr_array_new ();
2332         for (patch_info = relocs; patch_info; patch_info = patch_info->next)
2333                 g_ptr_array_add (patches, patch_info);
2334         g_ptr_array_sort (patches, compare_patches);
2335
2336         start_index = 0;
2337         for (i = 0; i < code_len; i++) {
2338                 patch_info = NULL;
2339                 for (pindex = start_index; pindex < patches->len; ++pindex) {
2340                         patch_info = g_ptr_array_index (patches, pindex);
2341                         if (patch_info->ip.i >= i)
2342                                 break;
2343                 }
2344
2345 #ifdef MONO_ARCH_AOT_SUPPORTED
2346                 skip = FALSE;
2347                 if (patch_info && (patch_info->ip.i == i) && (pindex < patches->len)) {
2348                         start_index = pindex;
2349
2350                         switch (patch_info->type) {
2351                         case MONO_PATCH_INFO_NONE:
2352                                 break;
2353                         case MONO_PATCH_INFO_GOT_OFFSET: {
2354                                 guint32 offset = mono_arch_get_patch_offset (code + i);
2355                                 emit_bytes (acfg, code + i, offset);
2356                                 emit_symbol_diff (acfg, "got", ".", offset);
2357
2358                                 i += offset + 4 - 1;
2359                                 skip = TRUE;
2360                                 break;
2361                         }
2362                         default: {
2363                                 char *direct_call_target;
2364
2365                                 if (!is_got_patch (patch_info->type))
2366                                         break;
2367
2368                                 /*
2369                                  * If this patch is a call, try emitting a direct call instead of
2370                                  * through a PLT entry. This is possible if the called method is in
2371                                  * the same assembly and requires no initialization.
2372                                  */
2373                                 direct_call_target = NULL;
2374                                 if (!got_only && (patch_info->type == MONO_PATCH_INFO_METHOD) && (patch_info->data.method->klass->image == method->klass->image)) {
2375                                         MonoCompile *callee_cfg = g_hash_table_lookup (acfg->method_to_cfg, patch_info->data.method);
2376                                         if (callee_cfg) {
2377                                                 if (!callee_cfg->has_got_slots && (callee_cfg->method->klass->flags & TYPE_ATTRIBUTE_BEFORE_FIELD_INIT)) {
2378                                                         //printf ("DIRECT: %s %s\n", mono_method_full_name (cfg->method, TRUE), mono_method_full_name (callee_cfg->method, TRUE));
2379                                                         direct_call_target = g_strdup_printf (".Lm_%x", get_method_index (acfg, callee_cfg->orig_method));
2380                                                         patch_info->type = MONO_PATCH_INFO_NONE;
2381                                                         acfg->stats.direct_calls ++;
2382                                                 }
2383                                         }
2384
2385                                         acfg->stats.all_calls ++;
2386                                 }
2387
2388                                 if (!got_only && !direct_call_target) {
2389                                         int plt_offset = get_plt_offset (acfg, patch_info);
2390                                         if (plt_offset != -1) {
2391                                                 /* This patch has a PLT entry, so we must emit a call to the PLT entry */
2392                                                 direct_call_target = g_strdup_printf (".Lp_%d", plt_offset);
2393                 
2394                                                 /* Nullify the patch */
2395                                                 patch_info->type = MONO_PATCH_INFO_NONE;
2396                                         }
2397                                 }
2398
2399                                 if (direct_call_target) {
2400 #if defined(__i386__) || defined(__x86_64__)
2401                                         g_assert (code [i] == 0xe8);
2402                                         /* Need to make sure this is exactly 5 bytes long */
2403                                         emit_byte (acfg, '\xe8');
2404                                         emit_symbol_diff (acfg, direct_call_target, ".", -4);
2405                                         i += 4;
2406 #elif defined(__arm__)
2407 #ifdef USE_BIN_WRITER
2408                                         /* FIXME: Can't encode this using the current symbol writer functions */
2409                                         g_assert_not_reached ();
2410 #else
2411                                         emit_unset_mode (acfg);
2412                                         fprintf (acfg->fp, "bl %s\n", direct_call_target);
2413                                         i += 4 - 1;
2414 #endif
2415 #endif
2416
2417                                         g_free (direct_call_target);
2418                                 } else {
2419                                         got_slot = get_got_offset (acfg, patch_info);
2420
2421                                         emit_bytes (acfg, code + i, mono_arch_get_patch_offset (code + i));
2422 #ifdef __x86_64__
2423                                         emit_symbol_diff (acfg, "got", ".", (unsigned int) ((got_slot * sizeof (gpointer)) - 4));
2424 #elif defined(__i386__)
2425                                         emit_int32 (acfg, (unsigned int) ((got_slot * sizeof (gpointer))));
2426 #elif defined(__arm__)
2427                                         emit_symbol_diff (acfg, "got", ".", (unsigned int) ((got_slot * sizeof (gpointer))) - 12);
2428 #else
2429                                         g_assert_not_reached ();
2430 #endif
2431                                         
2432                                         i += mono_arch_get_patch_offset (code + i) + 4 - 1;
2433                                 }
2434                                 skip = TRUE;
2435                         }
2436                         }
2437                 }
2438 #endif /* MONO_ARCH_AOT_SUPPORTED */
2439
2440                 if (!skip)
2441                         emit_bytes (acfg, code + i, 1);
2442         }
2443 }
2444
2445 static void
2446 emit_method_code (MonoAotCompile *acfg, MonoCompile *cfg)
2447 {
2448         MonoMethod *method;
2449         int method_index;
2450         guint8 *code;
2451         char *symbol;
2452         int func_alignment = 16;
2453         MonoMethodHeader *header;
2454
2455         method = cfg->orig_method;
2456         code = cfg->native_code;
2457         header = mono_method_get_header (method);
2458
2459         method_index = get_method_index (acfg, method);
2460
2461         /* Make the labels local */
2462         symbol = g_strdup_printf (".Lm_%x", method_index);
2463
2464         emit_alignment (acfg, func_alignment);
2465         emit_label (acfg, symbol);
2466         if (acfg->aot_opts.write_symbols)
2467                 emit_global (acfg, symbol, TRUE);
2468
2469         if (cfg->verbose_level > 0)
2470                 g_print ("Method %s emitted as %s\n", mono_method_full_name (method, TRUE), symbol);
2471         g_free (symbol);
2472
2473         acfg->stats.code_size += cfg->code_len;
2474
2475         acfg->method_got_offsets [method_index] = acfg->got_offset;
2476
2477         emit_and_reloc_code (acfg, method, code, cfg->code_len, cfg->patch_info, FALSE);
2478
2479         emit_line (acfg);
2480 }
2481
2482 /**
2483  * encode_patch:
2484  *
2485  *  Encode PATCH_INFO into its disk representation.
2486  */
2487 static void
2488 encode_patch (MonoAotCompile *acfg, MonoJumpInfo *patch_info, guint8 *buf, guint8 **endbuf)
2489 {
2490         guint8 *p = buf;
2491
2492         switch (patch_info->type) {
2493         case MONO_PATCH_INFO_NONE:
2494                 break;
2495         case MONO_PATCH_INFO_IMAGE:
2496                 encode_value (get_image_index (acfg, patch_info->data.image), p, &p);
2497                 break;
2498         case MONO_PATCH_INFO_METHOD_REL:
2499                 encode_value ((gint)patch_info->data.offset, p, &p);
2500                 break;
2501         case MONO_PATCH_INFO_SWITCH: {
2502                 gpointer *table = (gpointer *)patch_info->data.table->table;
2503                 int k;
2504
2505                 encode_value (patch_info->data.table->table_size, p, &p);
2506                 for (k = 0; k < patch_info->data.table->table_size; k++)
2507                         encode_value ((int)(gssize)table [k], p, &p);
2508                 break;
2509         }
2510         case MONO_PATCH_INFO_METHODCONST:
2511         case MONO_PATCH_INFO_METHOD:
2512         case MONO_PATCH_INFO_METHOD_JUMP:
2513         case MONO_PATCH_INFO_ICALL_ADDR:
2514         case MONO_PATCH_INFO_METHOD_RGCTX:
2515                 encode_method_ref (acfg, patch_info->data.method, p, &p);
2516                 break;
2517         case MONO_PATCH_INFO_INTERNAL_METHOD:
2518         case MONO_PATCH_INFO_JIT_ICALL_ADDR: {
2519                 guint32 len = strlen (patch_info->data.name);
2520
2521                 encode_value (len, p, &p);
2522
2523                 memcpy (p, patch_info->data.name, len);
2524                 p += len;
2525                 *p++ = '\0';
2526                 break;
2527         }
2528         case MONO_PATCH_INFO_LDSTR: {
2529                 guint32 image_index = get_image_index (acfg, patch_info->data.token->image);
2530                 guint32 token = patch_info->data.token->token;
2531                 g_assert (mono_metadata_token_code (token) == MONO_TOKEN_STRING);
2532                 encode_value (image_index, p, &p);
2533                 encode_value (patch_info->data.token->token - MONO_TOKEN_STRING, p, &p);
2534                 break;
2535         }
2536         case MONO_PATCH_INFO_RVA:
2537         case MONO_PATCH_INFO_DECLSEC:
2538         case MONO_PATCH_INFO_LDTOKEN:
2539         case MONO_PATCH_INFO_TYPE_FROM_HANDLE:
2540                 encode_value (get_image_index (acfg, patch_info->data.token->image), p, &p);
2541                 encode_value (patch_info->data.token->token, p, &p);
2542                 break;
2543         case MONO_PATCH_INFO_EXC_NAME: {
2544                 MonoClass *ex_class;
2545
2546                 ex_class =
2547                         mono_class_from_name (mono_defaults.exception_class->image,
2548                                                                   "System", patch_info->data.target);
2549                 g_assert (ex_class);
2550                 encode_klass_ref (acfg, ex_class, p, &p);
2551                 break;
2552         }
2553         case MONO_PATCH_INFO_R4:
2554                 encode_value (*((guint32 *)patch_info->data.target), p, &p);
2555                 break;
2556         case MONO_PATCH_INFO_R8:
2557                 encode_value (*((guint32 *)patch_info->data.target), p, &p);
2558                 encode_value (*(((guint32 *)patch_info->data.target) + 1), p, &p);
2559                 break;
2560         case MONO_PATCH_INFO_VTABLE:
2561         case MONO_PATCH_INFO_CLASS:
2562         case MONO_PATCH_INFO_IID:
2563         case MONO_PATCH_INFO_ADJUSTED_IID:
2564                 encode_klass_ref (acfg, patch_info->data.klass, p, &p);
2565                 break;
2566         case MONO_PATCH_INFO_CLASS_INIT:
2567         case MONO_PATCH_INFO_DELEGATE_TRAMPOLINE:
2568                 encode_klass_ref (acfg, patch_info->data.klass, p, &p);
2569                 break;
2570         case MONO_PATCH_INFO_FIELD:
2571         case MONO_PATCH_INFO_SFLDA:
2572                 encode_field_info (acfg, patch_info->data.field, p, &p);
2573                 break;
2574         case MONO_PATCH_INFO_INTERRUPTION_REQUEST_FLAG:
2575                 break;
2576         case MONO_PATCH_INFO_RGCTX_FETCH: {
2577                 MonoJumpInfoRgctxEntry *entry = patch_info->data.rgctx_entry;
2578
2579                 encode_method_ref (acfg, entry->method, p, &p);
2580                 encode_value (entry->in_mrgctx, p, &p);
2581                 encode_value (entry->info_type, p, &p);
2582                 encode_value (entry->data->type, p, &p);
2583                 encode_patch (acfg, entry->data, p, &p);
2584                 break;
2585         }
2586         case MONO_PATCH_INFO_GENERIC_CLASS_INIT:
2587                 break;
2588         default:
2589                 g_warning ("unable to handle jump info %d", patch_info->type);
2590                 g_assert_not_reached ();
2591         }
2592
2593         *endbuf = p;
2594 }
2595
2596 static void
2597 encode_patch_list (MonoAotCompile *acfg, GPtrArray *patches, int n_patches, int first_got_offset, guint8 *buf, guint8 **endbuf)
2598 {
2599         guint8 *p = buf;
2600         guint32 last_offset, j, pindex;
2601         MonoJumpInfo *patch_info;
2602
2603         encode_value (n_patches, p, &p);
2604
2605         if (n_patches)
2606                 encode_value (first_got_offset, p, &p);
2607
2608         /* First encode the type+position table */
2609         last_offset = 0;
2610         j = 0;
2611         for (pindex = 0; pindex < patches->len; ++pindex) {
2612                 guint32 offset;
2613                 patch_info = g_ptr_array_index (patches, pindex);
2614                 
2615                 if (patch_info->type == MONO_PATCH_INFO_NONE)
2616                         /* Nothing to do */
2617                         continue;
2618
2619                 j ++;
2620                 //printf ("T: %d O: %d.\n", patch_info->type, patch_info->ip.i);
2621                 offset = patch_info->ip.i - last_offset;
2622                 last_offset = patch_info->ip.i;
2623
2624                 /* Only the type is needed */
2625                 *p = patch_info->type;
2626                 p++;
2627         }
2628
2629         /* Then encode the other info */
2630         for (pindex = 0; pindex < patches->len; ++pindex) {
2631                 patch_info = g_ptr_array_index (patches, pindex);
2632
2633                 if (is_shared_got_patch (patch_info)) {
2634                         guint32 offset = get_got_offset (acfg, patch_info);
2635                         encode_value (offset, p, &p);
2636                 } else {
2637                         encode_patch (acfg, patch_info, p, &p);
2638                 }
2639         }
2640
2641         *endbuf = p;
2642 }
2643
2644 static void
2645 emit_method_info (MonoAotCompile *acfg, MonoCompile *cfg)
2646 {
2647         MonoMethod *method;
2648         GList *l;
2649         int pindex, buf_size, n_patches;
2650         guint8 *code;
2651         char *symbol;
2652         GPtrArray *patches;
2653         MonoJumpInfo *patch_info;
2654         MonoMethodHeader *header;
2655         guint32 method_index;
2656         guint8 *p, *buf;
2657         guint32 first_got_offset;
2658
2659         method = cfg->orig_method;
2660         code = cfg->native_code;
2661         header = mono_method_get_header (method);
2662
2663         method_index = get_method_index (acfg, method);
2664
2665         /* Make the labels local */
2666         symbol = g_strdup_printf (".Lm_%x_p", method_index);
2667
2668         /* Sort relocations */
2669         patches = g_ptr_array_new ();
2670         for (patch_info = cfg->patch_info; patch_info; patch_info = patch_info->next)
2671                 g_ptr_array_add (patches, patch_info);
2672         g_ptr_array_sort (patches, compare_patches);
2673
2674         first_got_offset = acfg->method_got_offsets [method_index];
2675
2676         /**********************/
2677         /* Encode method info */
2678         /**********************/
2679
2680         buf_size = (patches->len < 1000) ? 40960 : 40960 + (patches->len * 64);
2681         p = buf = g_malloc (buf_size);
2682
2683         if (mono_class_get_cctor (method->klass))
2684                 encode_klass_ref (acfg, method->klass, p, &p);
2685         else
2686                 /* Not needed when loading the method */
2687                 encode_value (0, p, &p);
2688
2689         /* String table */
2690         if (cfg->opt & MONO_OPT_SHARED) {
2691                 encode_value (g_list_length (cfg->ldstr_list), p, &p);
2692                 for (l = cfg->ldstr_list; l; l = l->next) {
2693                         encode_value ((long)l->data, p, &p);
2694                 }
2695         }
2696         else
2697                 /* Used only in shared mode */
2698                 g_assert (!cfg->ldstr_list);
2699
2700         n_patches = 0;
2701         for (pindex = 0; pindex < patches->len; ++pindex) {
2702                 patch_info = g_ptr_array_index (patches, pindex);
2703                 
2704                 if ((patch_info->type == MONO_PATCH_INFO_GOT_OFFSET) ||
2705                         (patch_info->type == MONO_PATCH_INFO_NONE)) {
2706                         patch_info->type = MONO_PATCH_INFO_NONE;
2707                         /* Nothing to do */
2708                         continue;
2709                 }
2710
2711                 if ((patch_info->type == MONO_PATCH_INFO_IMAGE) && (patch_info->data.image == acfg->image)) {
2712                         /* Stored in a GOT slot initialized at module load time */
2713                         patch_info->type = MONO_PATCH_INFO_NONE;
2714                         continue;
2715                 }
2716
2717                 if (is_plt_patch (patch_info)) {
2718                         /* Calls are made through the PLT */
2719                         patch_info->type = MONO_PATCH_INFO_NONE;
2720                         continue;
2721                 }
2722
2723                 n_patches ++;
2724         }
2725
2726         if (n_patches)
2727                 g_assert (cfg->has_got_slots);
2728
2729         encode_patch_list (acfg, patches, n_patches, first_got_offset, p, &p);
2730
2731         acfg->stats.info_size += p - buf;
2732
2733         /* Emit method info */
2734
2735         emit_label (acfg, symbol);
2736         g_free (symbol);
2737
2738         g_assert (p - buf < buf_size);
2739         emit_bytes (acfg, buf, p - buf);
2740         g_free (buf);
2741 }
2742
2743 static void
2744 emit_exception_debug_info (MonoAotCompile *acfg, MonoCompile *cfg)
2745 {
2746         MonoMethod *method;
2747         int k, buf_size, method_index;
2748         guint32 debug_info_size;
2749         guint8 *code;
2750         char *symbol;
2751         MonoMethodHeader *header;
2752         guint8 *p, *buf, *debug_info;
2753         MonoJitInfo *jinfo = cfg->jit_info;
2754
2755         method = cfg->orig_method;
2756         code = cfg->native_code;
2757         header = mono_method_get_header (method);
2758
2759         method_index = get_method_index (acfg, method);
2760
2761         /* Make the labels local */
2762         symbol = g_strdup_printf (".Le_%x_p", method_index);
2763
2764         mono_debug_serialize_debug_info (cfg, &debug_info, &debug_info_size);
2765
2766         buf_size = header->num_clauses * 256 + debug_info_size + 128;
2767         p = buf = g_malloc (buf_size);
2768
2769         encode_value (jinfo->code_size, p, &p);
2770         encode_value (jinfo->used_regs, p, &p);
2771         encode_value (jinfo->has_generic_jit_info, p, &p);
2772
2773         /* Exception table */
2774         if (header->num_clauses) {
2775                 for (k = 0; k < header->num_clauses; ++k) {
2776                         MonoJitExceptionInfo *ei = &jinfo->clauses [k];
2777
2778                         encode_value (ei->exvar_offset, p, &p);
2779
2780                         if (ei->flags == MONO_EXCEPTION_CLAUSE_FILTER)
2781                                 encode_value ((gint)((guint8*)ei->data.filter - code), p, &p);
2782
2783                         encode_value ((gint)((guint8*)ei->try_start - code), p, &p);
2784                         encode_value ((gint)((guint8*)ei->try_end - code), p, &p);
2785                         encode_value ((gint)((guint8*)ei->handler_start - code), p, &p);
2786                 }
2787         }
2788
2789         if (jinfo->has_generic_jit_info) {
2790                 MonoGenericJitInfo *gi = mono_jit_info_get_generic_jit_info (jinfo);
2791
2792                 encode_value (gi->has_this ? 1 : 0, p, &p);
2793                 encode_value (gi->this_reg, p, &p);
2794                 encode_value (gi->this_offset, p, &p);
2795
2796                 /* 
2797                  * Need to encode jinfo->method too, since it is not equal to 'method'
2798                  * when using generic sharing.
2799                  */
2800                 encode_method_ref (acfg, jinfo->method, p, &p);
2801         }
2802
2803         g_assert (debug_info_size < buf_size);
2804
2805         encode_value (debug_info_size, p, &p);
2806         if (debug_info_size) {
2807                 memcpy (p, debug_info, debug_info_size);
2808                 p += debug_info_size;
2809                 g_free (debug_info);
2810         }
2811
2812         acfg->stats.ex_info_size += p - buf;
2813
2814         /* Emit info */
2815
2816         emit_label (acfg, symbol);
2817         g_free (symbol);
2818
2819         g_assert (p - buf < buf_size);
2820         emit_bytes (acfg, buf, p - buf);
2821         g_free (buf);
2822 }
2823
2824 static void
2825 emit_klass_info (MonoAotCompile *acfg, guint32 token)
2826 {
2827         MonoClass *klass = mono_class_get (acfg->image, token);
2828         guint8 *p, *buf;
2829         int i, buf_size;
2830         char *symbol;
2831         gboolean no_special_static, cant_encode;
2832         gpointer iter = NULL;
2833
2834         buf_size = 10240 + (klass->vtable_size * 16);
2835         p = buf = g_malloc (buf_size);
2836
2837         g_assert (klass);
2838
2839         mono_class_init (klass);
2840
2841         mono_class_get_nested_types (klass, &iter);
2842         g_assert (klass->nested_classes_inited);
2843
2844         mono_class_setup_vtable (klass);
2845
2846         /* 
2847          * Emit all the information which is required for creating vtables so
2848          * the runtime does not need to create the MonoMethod structures which
2849          * take up a lot of space.
2850          */
2851
2852         no_special_static = !mono_class_has_special_static_fields (klass);
2853
2854         /* Check whenever we have enough info to encode the vtable */
2855         cant_encode = FALSE;
2856         for (i = 0; i < klass->vtable_size; ++i) {
2857                 MonoMethod *cm = klass->vtable [i];
2858
2859                 if (cm && mono_method_signature (cm)->is_inflated && !g_hash_table_lookup (acfg->token_info_hash, cm))
2860                         cant_encode = TRUE;
2861         }
2862
2863         if (klass->generic_container || cant_encode) {
2864                 encode_value (-1, p, &p);
2865         } else {
2866                 encode_value (klass->vtable_size, p, &p);
2867                 encode_value ((no_special_static << 7) | (klass->has_static_refs << 6) | (klass->has_references << 5) | ((klass->blittable << 4) | (klass->nested_classes ? 1 : 0) << 3) | (klass->has_cctor << 2) | (klass->has_finalize << 1) | klass->ghcimpl, p, &p);
2868                 if (klass->has_cctor)
2869                         encode_method_ref (acfg, mono_class_get_cctor (klass), p, &p);
2870                 if (klass->has_finalize)
2871                         encode_method_ref (acfg, mono_class_get_finalizer (klass), p, &p);
2872  
2873                 encode_value (klass->instance_size, p, &p);
2874                 encode_value (mono_class_data_size (klass), p, &p);
2875                 encode_value (klass->packing_size, p, &p);
2876                 encode_value (klass->min_align, p, &p);
2877
2878                 for (i = 0; i < klass->vtable_size; ++i) {
2879                         MonoMethod *cm = klass->vtable [i];
2880
2881                         if (cm)
2882                                 encode_method_ref (acfg, cm, p, &p);
2883                         else
2884                                 encode_value (0, p, &p);
2885                 }
2886         }
2887
2888         acfg->stats.class_info_size += p - buf;
2889
2890         /* Emit the info */
2891         symbol = g_strdup_printf (".LK_I_%x", token - MONO_TOKEN_TYPE_DEF - 1);
2892         emit_label (acfg, symbol);
2893         g_free (symbol);
2894
2895         g_assert (p - buf < buf_size);
2896         emit_bytes (acfg, buf, p - buf);
2897         g_free (buf);
2898 }
2899
2900 /*
2901  * Calls made from AOTed code are routed through a table of jumps similar to the
2902  * ELF PLT (Program Linkage Table). The differences are the following:
2903  * - the ELF PLT entries make an indirect jump though the GOT so they expect the
2904  *   GOT pointer to be in EBX. We want to avoid this, so our table contains direct
2905  *   jumps. This means the jumps need to be patched when the address of the callee is
2906  *   known. Initially the PLT entries jump to code which transfers control to the
2907  *   AOT runtime through the first PLT entry.
2908  */
2909 static void
2910 emit_plt (MonoAotCompile *acfg)
2911 {
2912         char *symbol;
2913         int i, buf_size;
2914         guint8 *p, *buf;
2915         guint32 *plt_info_offsets;
2916
2917         /*
2918          * Encode info need to resolve PLT entries.
2919          */
2920         buf_size = acfg->plt_offset * 128;
2921         p = buf = g_malloc (buf_size);
2922
2923         plt_info_offsets = g_new0 (guint32, acfg->plt_offset);
2924
2925         for (i = 1; i < acfg->plt_offset; ++i) {
2926                 MonoJumpInfo *patch_info = g_hash_table_lookup (acfg->plt_offset_to_patch, GUINT_TO_POINTER (i));
2927
2928                 plt_info_offsets [i] = p - buf;
2929                 encode_value (patch_info->type, p, &p);
2930                 encode_patch (acfg, patch_info, p, &p);
2931         }
2932
2933         emit_line (acfg);
2934         symbol = g_strdup_printf ("plt");
2935
2936         /* This section will be made read-write by the AOT loader */
2937         emit_section_change (acfg, ".text", 0);
2938         emit_global (acfg, symbol, TRUE);
2939         emit_alignment (acfg, PAGESIZE);
2940         emit_label (acfg, symbol);
2941         g_free (symbol);
2942
2943 #if defined(USE_BIN_WRITER) && defined(__arm__)
2944         /* FIXME: */
2945         g_assert_not_reached ();
2946 #endif
2947
2948         for (i = 0; i < acfg->plt_offset; ++i) {
2949                 char *label;
2950
2951                 label = g_strdup_printf (".Lp_%d", i);
2952                 emit_label (acfg, label);
2953                 g_free (label);
2954
2955                 /* 
2956                  * The first plt entry is used to transfer code to the AOT loader. 
2957                  */
2958
2959 #if defined(__i386__)
2960                 if (i == 0) {
2961                         /* It is filled up during loading by the AOT loader. */
2962                         emit_zero_bytes (acfg, 16);
2963                 } else {
2964                         /* Need to make sure this is 9 bytes long */
2965                         emit_byte (acfg, '\xe9');
2966                         emit_symbol_diff (acfg, "plt", ".", -4);
2967                         emit_int32 (acfg, plt_info_offsets [i]);
2968                 }
2969 #elif defined(__x86_64__)
2970                 /*
2971                  * We can't emit jumps because they are 32 bits only so they can't be patched.
2972                  * So we emit a jump table instead whose entries are patched by the AOT loader to
2973                  * point to .Lpd entries. ELF stores these in the GOT too, but we don't, since
2974                  * methods with GOT entries can't be called directly.
2975                  * We also emit the default PLT code here since the PLT code will not be patched.
2976                  * An x86_64 plt entry is 10 bytes long, init_plt () depends on this.
2977                  */
2978                 /* jmpq *<offset>(%rip) */
2979                 emit_byte (acfg, '\xff');
2980                 emit_byte (acfg, '\x25');
2981                 emit_symbol_diff (acfg, "plt_jump_table", ".", (i * sizeof (gpointer)) -4);
2982                 /* Used by mono_aot_get_plt_info_offset */
2983                 emit_int32 (acfg, plt_info_offsets [i]);
2984 #elif defined(__arm__)
2985                 /* FIXME:
2986                  * - optimize OP_AOTCONST implementation
2987                  * - optimize the PLT entries
2988                  * - optimize SWITCH AOT implementation
2989                  * - implement IMT support
2990                  */
2991                 emit_unset_mode (acfg);
2992                 fprintf (acfg->fp, "\tldr ip, [pc, #4]\n");
2993                 fprintf (acfg->fp, "\tadd ip, pc, ip\n");
2994                 fprintf (acfg->fp, "\tldr pc, [ip, #0]\n");
2995                 emit_symbol_diff (acfg, "plt_jump_table", ".", (i * sizeof (gpointer)));
2996                 /* Used by mono_aot_get_plt_info_offset */
2997     #if defined(__MACH__)
2998                 fprintf (acfg->fp, "\n\t.long %d\n", plt_info_offsets [i]);
2999     #else
3000                 fprintf (acfg->fp, "\n\t.word %d\n", plt_info_offsets [i]);
3001     #endif
3002
3003 #else
3004                 g_assert_not_reached ();
3005 #endif
3006         }
3007
3008         g_free (plt_info_offsets);
3009
3010         symbol = g_strdup_printf ("plt_end");
3011         emit_global (acfg, symbol, TRUE);
3012         emit_label (acfg, symbol);
3013         g_free (symbol);
3014
3015         /* Emit PLT info */
3016         symbol = g_strdup_printf ("plt_info");
3017         emit_global (acfg, symbol, FALSE);
3018         emit_label (acfg, symbol);
3019         g_free (symbol);
3020
3021         g_assert (p - buf < buf_size);
3022         emit_bytes (acfg, buf, p - buf);
3023         g_free (buf);
3024
3025         symbol = g_strdup_printf ("plt_jump_table_addr");
3026         emit_section_change (acfg, ".data", 0);
3027         emit_global (acfg, symbol, FALSE);
3028         emit_alignment (acfg, 8);
3029         emit_label (acfg, symbol);
3030         emit_pointer (acfg, "plt_jump_table");
3031         g_free (symbol);
3032
3033         symbol = g_strdup_printf ("plt_jump_table_size");
3034         emit_section_change (acfg, ".data", 0);
3035         emit_global (acfg, symbol, FALSE);
3036         emit_alignment (acfg, 8);
3037         emit_label (acfg, symbol);
3038         emit_symbol_diff (acfg, "plt_jump_table_end", "plt_jump_table", 0);
3039         g_free (symbol);
3040
3041         /* Don't make this a global so accesses don't need relocations */
3042         symbol = g_strdup_printf ("plt_jump_table");
3043         emit_section_change (acfg, ".bss", 0);
3044         emit_label (acfg, symbol);
3045         g_free (symbol);
3046
3047 #if defined(__x86_64__) || defined(__arm__)
3048         emit_zero_bytes (acfg, (int)(acfg->plt_offset * sizeof (gpointer)));
3049 #endif  
3050
3051         symbol = g_strdup_printf ("plt_jump_table_end");
3052         emit_label (acfg, symbol);
3053         g_free (symbol);
3054 }
3055
3056 static G_GNUC_UNUSED void
3057 emit_named_code (MonoAotCompile *acfg, const char *name, guint8 *code, 
3058                                  guint32 code_size, int got_offset, MonoJumpInfo *ji)
3059 {
3060         char *symbol;
3061         guint32 buf_size;
3062         MonoJumpInfo *patch_info;
3063         guint8 *buf, *p;
3064         GPtrArray *patches;
3065
3066         /* Emit code */
3067
3068         symbol = g_strdup_printf ("%s", name);
3069
3070         emit_section_change (acfg, ".text", 0);
3071         emit_global (acfg, symbol, TRUE);
3072         emit_alignment (acfg, 16);
3073         emit_label (acfg, symbol);
3074         g_free (symbol);
3075
3076         /* 
3077          * The code should access everything through the GOT, so we pass
3078          * TRUE here.
3079          */
3080         emit_and_reloc_code (acfg, NULL, code, code_size, ji, TRUE);
3081
3082         /* Emit info */
3083
3084         /* Sort relocations */
3085         patches = g_ptr_array_new ();
3086         for (patch_info = ji; patch_info; patch_info = patch_info->next)
3087                 g_ptr_array_add (patches, patch_info);
3088         g_ptr_array_sort (patches, compare_patches);
3089
3090         buf_size = patches->len * 128 + 128;
3091         buf = g_malloc (buf_size);
3092         p = buf;
3093
3094         encode_patch_list (acfg, patches, patches->len, got_offset, p, &p);
3095         g_assert (p - buf < buf_size);
3096
3097         symbol = g_strdup_printf ("%s_p", name);
3098
3099         emit_section_change (acfg, ".text", 0);
3100         emit_global (acfg, symbol, FALSE);
3101         emit_label (acfg, symbol);
3102         g_free (symbol);
3103                 
3104         emit_bytes (acfg, buf, p - buf);
3105 }
3106
3107 /*
3108  * When running in aot-only mode, we can't create trampolines at runtime, so we create 
3109  * a few, and save them in the AOT file. Normal trampolines embed their argument as a 
3110  * literal inside the trampoline code, we can't do that here, so instead we embed an offset
3111  * which needs to be added to the trampoline address to get the address of the GOT slot
3112  * which contains the argument value.
3113  * The generated trampolines jump to the generic trampolines using another GOT slot, which
3114  * will be setup by the AOT loader to point to the generic trampoline code of the given 
3115  * type.
3116  */
3117 static void
3118 emit_trampolines (MonoAotCompile *acfg)
3119 {
3120         char *symbol;
3121         int i, offset;
3122 #ifdef MONO_ARCH_HAVE_FULL_AOT_TRAMPOLINES
3123         int tramp_type;
3124         guint32 code_size;
3125         MonoJumpInfo *ji;
3126         guint8 *code;
3127 #endif
3128
3129         if (!acfg->aot_opts.full_aot)
3130                 return;
3131         
3132         g_assert (acfg->image->assembly);
3133
3134         /* Currently, we only emit most trampolines into the mscorlib AOT image. */
3135         if (strcmp (acfg->image->assembly->aname.name, "mscorlib") == 0) {
3136 #ifdef MONO_ARCH_HAVE_FULL_AOT_TRAMPOLINES
3137                 /*
3138                  * Emit the generic trampolines.
3139                  *
3140                  * We could save some code by treating the generic trampolines as a wrapper
3141                  * method, but that approach has its own complexities, so we choose the simpler
3142                  * method.
3143                  */
3144                 for (tramp_type = 0; tramp_type < MONO_TRAMPOLINE_NUM; ++tramp_type) {
3145                         code = mono_arch_create_trampoline_code_full (tramp_type, &code_size, &ji, TRUE);
3146
3147                         /* Emit trampoline code */
3148
3149                         symbol = g_strdup_printf ("generic_trampoline_%d", tramp_type);
3150
3151                         emit_named_code (acfg, symbol, code, code_size, acfg->got_offset, ji);
3152
3153                         g_free (symbol);
3154                 }
3155
3156                 code = mono_arch_get_nullified_class_init_trampoline (&code_size);
3157                 emit_named_code (acfg, "nullified_class_init_trampoline", code, code_size, acfg->got_offset, NULL);
3158
3159                 /* Emit the exception related code pieces */
3160                 code = mono_arch_get_restore_context_full (&code_size, &ji, TRUE);
3161                 emit_named_code (acfg, "restore_context", code, code_size, acfg->got_offset, ji);
3162                 code = mono_arch_get_call_filter_full (&code_size, &ji, TRUE);
3163                 emit_named_code (acfg, "call_filter", code, code_size, acfg->got_offset, ji);
3164                 code = mono_arch_get_throw_exception_full (&code_size, &ji, TRUE);
3165                 emit_named_code (acfg, "throw_exception", code, code_size, acfg->got_offset, ji);
3166                 code = mono_arch_get_rethrow_exception_full (&code_size, &ji, TRUE);
3167                 emit_named_code (acfg, "rethrow_exception", code, code_size, acfg->got_offset, ji);
3168                 code = mono_arch_get_throw_exception_by_name_full (&code_size, &ji, TRUE);
3169                 emit_named_code (acfg, "throw_exception_by_name", code, code_size, acfg->got_offset, ji);
3170                 code = mono_arch_get_throw_corlib_exception_full (&code_size, &ji, TRUE);
3171                 emit_named_code (acfg, "throw_corlib_exception", code, code_size, acfg->got_offset, ji);
3172
3173 #ifdef __x86_64__
3174                 for (i = 0; i < 128; ++i) {
3175                         int offset;
3176
3177                         offset = MONO_RGCTX_SLOT_MAKE_RGCTX (i);
3178                         code = mono_arch_create_rgctx_lazy_fetch_trampoline_full (offset, &code_size, &ji, TRUE);
3179                         symbol = g_strdup_printf ("rgctx_fetch_trampoline_%u", offset);
3180                         emit_named_code (acfg, symbol, code, code_size, acfg->got_offset, ji);
3181                         g_free (symbol);
3182
3183                         offset = MONO_RGCTX_SLOT_MAKE_MRGCTX (i);
3184                         code = mono_arch_create_rgctx_lazy_fetch_trampoline_full (offset, &code_size, &ji, TRUE);
3185                         symbol = g_strdup_printf ("rgctx_fetch_trampoline_%u", offset);
3186                         emit_named_code (acfg, symbol, code, code_size, acfg->got_offset, ji);
3187                         g_free (symbol);
3188                 }
3189 #endif
3190 #endif
3191
3192                 /*
3193                  * FIXME: Maybe we should use more specific trampolines (i.e. one class init for
3194                  * each class).
3195                  */
3196
3197                 /* Reserve some entries at the end of the GOT for our use */
3198                 acfg->num_trampoline_got_entries = acfg->num_aot_trampolines * 2;
3199
3200                 symbol = g_strdup_printf ("trampolines");
3201
3202                 emit_section_change (acfg, ".text", 0);
3203                 emit_global (acfg, symbol, TRUE);
3204                 emit_alignment (acfg, 16);
3205                 emit_label (acfg, symbol);
3206
3207                 for (i = 0; i < acfg->num_aot_trampolines; ++i) {
3208                         offset = acfg->got_offset + (i * 2);
3209
3210                         /*
3211                          * The trampolines created here are variations of the specific 
3212                          * trampolines created in mono_arch_create_specific_trampoline (). The 
3213                          * differences are:
3214                          * - the generic trampoline address is taken from a got slot.
3215                          * - the offset of the got slot where the trampoline argument is stored
3216                          *   is embedded in the instruction stream, and the generic trampoline
3217                          *   can load the argument by loading the offset, adding it to the
3218                          *   address of the trampoline to get the address of the got slot, and
3219                          *   loading the argument from the there.
3220                          */
3221 #if defined(__x86_64__)
3222                         /* This should be exactly 16 bytes long */
3223                         /* It should work together with the generic trampoline code in tramp-amd64.c */
3224                         /* call *<offset>(%rip) */
3225                         emit_byte (acfg, '\x41');
3226                         emit_byte (acfg, '\xff');
3227                         emit_byte (acfg, '\x15');
3228                         emit_symbol_diff (acfg, "got", ".", (offset * sizeof (gpointer)) - 4);
3229                         /* This should be relative to the start of the trampoline */
3230                         emit_symbol_diff (acfg, "got", ".", (offset * sizeof (gpointer)) - 4 + 19);
3231                         emit_zero_bytes (acfg, 5);
3232 #elif defined(__arm__)
3233                         {
3234                                 guint8 buf [128];
3235
3236                                 /* Generate the trampoline code */
3237                                 /* This should be exactly 28 bytes long */
3238
3239                                 code = buf;
3240                                 ARM_PUSH (code, 0x5fff);
3241                                 ARM_LDR_IMM (code, ARMREG_R1, ARMREG_PC, 8);
3242                                 /* Load the value from the GOT */
3243                                 ARM_LDR_REG_REG (code, ARMREG_R1, ARMREG_PC, ARMREG_R1);
3244                                 /* Branch to it */
3245                                 ARM_MOV_REG_REG (code, ARMREG_LR, ARMREG_PC);
3246                                 ARM_MOV_REG_REG (code, ARMREG_PC, ARMREG_R1);
3247
3248                                 g_assert (code - buf == 20);
3249
3250                                 /* Emit it */
3251                                 emit_bytes (acfg, buf, code - buf);
3252                                 emit_symbol_diff (acfg, "got", ".", (offset * sizeof (gpointer)) - 4 + 8);
3253                                 emit_symbol_diff (acfg, "got", ".", ((offset + 1) * sizeof (gpointer)) - 4 + 8);
3254                         }
3255 #else
3256                         g_assert_not_reached ();
3257 #endif
3258                 }
3259         }
3260
3261         /* Unbox trampolines */
3262
3263         for (i = 0; i < acfg->image->tables [MONO_TABLE_METHOD].rows; ++i) {
3264                 MonoMethod *method;
3265                 guint32 token = MONO_TOKEN_METHOD_DEF | (i + 1);
3266                 MonoCompile *cfg;
3267                 char *call_target;
3268
3269                 method = mono_get_method (acfg->image, token, NULL);
3270
3271                 cfg = g_hash_table_lookup (acfg->method_to_cfg, method);
3272                 if (!cfg || !cfg->orig_method->klass->valuetype || !(method->flags & METHOD_ATTRIBUTE_VIRTUAL))
3273                         continue;
3274
3275                 symbol = g_strdup_printf ("unbox_trampoline_%d", i);
3276
3277                 emit_section_change (acfg, ".text", 0);
3278                 emit_global (acfg, symbol, TRUE);
3279                 emit_label (acfg, symbol);
3280
3281                 call_target = g_strdup_printf (".Lm_%x", get_method_index (acfg, cfg->orig_method));
3282
3283 #if defined(__x86_64__)
3284                 {
3285                         guint8 buf [32];
3286                         int this_reg;
3287
3288                         this_reg = mono_arch_get_this_arg_reg (mono_method_signature (cfg->orig_method), cfg->generic_sharing_context, NULL);
3289                         code = buf;
3290                         amd64_alu_reg_imm (code, X86_ADD, this_reg, sizeof (MonoObject));
3291
3292                         emit_bytes (acfg, buf, code - buf);
3293                         /* jump <method> */
3294                         emit_byte (acfg, '\xe9');
3295                         emit_symbol_diff (acfg, call_target, ".", -4);
3296                 }
3297 #elif defined(__arm__)
3298                 {
3299                         guint8 buf [128];
3300                         int this_pos = 0;
3301
3302                         code = buf;
3303
3304                         if (MONO_TYPE_ISSTRUCT (mono_method_signature (cfg->orig_method)->ret))
3305                                 this_pos = 1;
3306
3307                         ARM_ADD_REG_IMM8 (code, this_pos, this_pos, sizeof (MonoObject));
3308
3309                         emit_bytes (acfg, buf, code - buf);
3310                         /* jump to method */
3311 #if defined(USE_BIN_WRITER)
3312                         /* FIXME: */
3313                         g_assert_not_reached ();
3314 #endif
3315                         fprintf (acfg->fp, "\n\tb %s\n", call_target);
3316                 }
3317 #else
3318                 g_assert_not_reached ();
3319 #endif
3320         }
3321
3322         symbol = g_strdup_printf ("trampolines_info");
3323
3324         emit_section_change (acfg, ".text", 0);
3325         emit_global (acfg, symbol, TRUE);
3326         emit_alignment (acfg, PAGESIZE);
3327         emit_label (acfg, symbol);
3328
3329         emit_int32 (acfg, acfg->num_aot_trampolines);
3330         emit_int32 (acfg, acfg->got_offset);
3331 }
3332
3333 static gboolean
3334 str_begins_with (const char *str1, const char *str2)
3335 {
3336         int len = strlen (str2);
3337         return strncmp (str1, str2, len) == 0;
3338 }
3339
3340 static void
3341 mono_aot_parse_options (const char *aot_options, MonoAotOptions *opts)
3342 {
3343         gchar **args, **ptr;
3344
3345         memset (opts, 0, sizeof (*opts));
3346
3347         args = g_strsplit (aot_options ? aot_options : "", ",", -1);
3348         for (ptr = args; ptr && *ptr; ptr ++) {
3349                 const char *arg = *ptr;
3350
3351                 if (str_begins_with (arg, "outfile=")) {
3352                         opts->outfile = g_strdup (arg + strlen ("outfile="));
3353                 } else if (str_begins_with (arg, "save-temps")) {
3354                         opts->save_temps = TRUE;
3355                 } else if (str_begins_with (arg, "keep-temps")) {
3356                         opts->save_temps = TRUE;
3357                 } else if (str_begins_with (arg, "write-symbols")) {
3358                         opts->write_symbols = TRUE;
3359                 } else if (str_begins_with (arg, "metadata-only")) {
3360                         opts->metadata_only = TRUE;
3361                 } else if (str_begins_with (arg, "bind-to-runtime-version")) {
3362                         opts->bind_to_runtime_version = TRUE;
3363                 } else if (str_begins_with (arg, "full")) {
3364                         opts->full_aot = TRUE;
3365                         /*
3366                          * The no-dlsym option is only useful on the iphone, and even there,
3367                          * do to other limitations of the dynamic linker, it doesn't seem to
3368                          * work. So disable it for now so we don't have to support it.
3369                          */
3370                         /*
3371                 } else if (str_begins_with (arg, "no-dlsym")) {
3372                         opts->no_dlsym = TRUE;
3373                         */
3374                 } else if (str_begins_with (arg, "static")) {
3375                         opts->static_link = TRUE;
3376                         opts->no_dlsym = TRUE;
3377                 } else if (str_begins_with (arg, "asmonly")) {
3378                         opts->asm_only = TRUE;
3379                 } else {
3380                         fprintf (stderr, "AOT : Unknown argument '%s'.\n", arg);
3381                         exit (1);
3382                 }
3383         }
3384
3385         g_strfreev (args);
3386 }
3387
3388 static void
3389 add_token_info_hash (gpointer key, gpointer value, gpointer user_data)
3390 {
3391         MonoMethod *method = (MonoMethod*)key;
3392         MonoJumpInfoToken *ji = (MonoJumpInfoToken*)value;
3393         MonoJumpInfoToken *new_ji = g_new0 (MonoJumpInfoToken, 1);
3394         MonoAotCompile *acfg = user_data;
3395
3396         new_ji->image = ji->image;
3397         new_ji->token = ji->token;
3398         g_hash_table_insert (acfg->token_info_hash, method, new_ji);
3399 }
3400
3401 static void
3402 compile_method (MonoAotCompile *acfg, MonoMethod *method)
3403 {
3404         MonoCompile *cfg;
3405         MonoJumpInfo *patch_info;
3406         gboolean skip;
3407         int index;
3408         MonoMethod *wrapped;
3409
3410         if (acfg->aot_opts.metadata_only)
3411                 return;
3412
3413         index = get_method_index (acfg, method);
3414
3415         /* fixme: maybe we can also precompile wrapper methods */
3416         if ((method->flags & METHOD_ATTRIBUTE_PINVOKE_IMPL) ||
3417                 (method->iflags & METHOD_IMPL_ATTRIBUTE_RUNTIME) ||
3418                 (method->flags & METHOD_ATTRIBUTE_ABSTRACT)) {
3419                 //printf ("Skip (impossible): %s\n", mono_method_full_name (method, TRUE));
3420                 return;
3421         }
3422
3423         if (method->iflags & METHOD_IMPL_ATTRIBUTE_INTERNAL_CALL)
3424                 return;
3425
3426         wrapped = mono_marshal_method_from_wrapper (method);
3427         if (wrapped && (wrapped->iflags & METHOD_IMPL_ATTRIBUTE_INTERNAL_CALL) && wrapped->is_generic)
3428                 // FIXME: The wrapper should be generic too, but it is not
3429                 return;
3430
3431         acfg->stats.mcount++;
3432
3433         if (method->iflags & METHOD_IMPL_ATTRIBUTE_SYNCHRONIZED) {
3434                 /* 
3435                  * FIXME: Enabling this causes virtual-sync.exe to fail, since the trampoline
3436                  * code can't determine that it needs to insert a sync wrapper in the AOT case.
3437                  */
3438                 return;
3439         }
3440
3441 #if 0
3442         if (method->is_generic || method->klass->generic_container) {
3443                 acfg->stats.genericcount ++;
3444                 return;
3445         }
3446 #endif
3447
3448         if (acfg->aot_opts.full_aot)
3449                 mono_use_imt = FALSE;
3450
3451         /*
3452          * Since these methods are the only ones which are compiled with
3453          * AOT support, and they are not used by runtime startup/shutdown code,
3454          * the runtime will not see AOT methods during AOT compilation,so it
3455          * does not need to support them by creating a fake GOT etc.
3456          */
3457         cfg = mini_method_compile (method, acfg->opts, mono_get_root_domain (), FALSE, TRUE, 0);
3458         if (cfg->exception_type == MONO_EXCEPTION_GENERIC_SHARING_FAILED) {
3459                 //printf ("F: %s\n", mono_method_full_name (method, TRUE));
3460                 acfg->stats.genericcount ++;
3461                 return;
3462         }
3463         if (cfg->exception_type != MONO_EXCEPTION_NONE) {
3464                 /* Let the exception happen at runtime */
3465                 return;
3466         }
3467
3468         if (cfg->disable_aot) {
3469                 //printf ("Skip (other): %s\n", mono_method_full_name (method, TRUE));
3470                 acfg->stats.ocount++;
3471                 mono_destroy_compile (cfg);
3472                 return;
3473         }
3474
3475         /* Nullify patches which need no aot processing */
3476         for (patch_info = cfg->patch_info; patch_info; patch_info = patch_info->next) {
3477                 switch (patch_info->type) {
3478                 case MONO_PATCH_INFO_LABEL:
3479                 case MONO_PATCH_INFO_BB:
3480                         patch_info->type = MONO_PATCH_INFO_NONE;
3481                         break;
3482                 default:
3483                         break;
3484                 }
3485         }
3486
3487         /* Collect method->token associations from the cfg */
3488         g_hash_table_foreach (cfg->token_info_hash, add_token_info_hash, acfg);
3489
3490         /*
3491          * Check for absolute addresses.
3492          */
3493         skip = FALSE;
3494         for (patch_info = cfg->patch_info; patch_info; patch_info = patch_info->next) {
3495                 switch (patch_info->type) {
3496                 case MONO_PATCH_INFO_ABS:
3497                         /* unable to handle this */
3498                         //printf ("Skip (abs addr):   %s %d\n", mono_method_full_name (method, TRUE), patch_info->type);
3499                         skip = TRUE;    
3500                         break;
3501                 default:
3502                         break;
3503                 }
3504         }
3505
3506         if (skip) {
3507                 acfg->stats.abscount++;
3508                 mono_destroy_compile (cfg);
3509                 return;
3510         }
3511
3512         /*
3513          * Check for wrapper methods we can't encode.
3514          */
3515         skip = FALSE;
3516         for (patch_info = cfg->patch_info; patch_info; patch_info = patch_info->next) {
3517                 if ((patch_info->type == MONO_PATCH_INFO_METHODCONST) || (patch_info->type == MONO_PATCH_INFO_METHOD)) {
3518                         switch (patch_info->data.method->wrapper_type) {
3519                         case MONO_WRAPPER_NONE:
3520                         case MONO_WRAPPER_REMOTING_INVOKE_WITH_CHECK:
3521                         case MONO_WRAPPER_XDOMAIN_INVOKE:
3522                         case MONO_WRAPPER_STFLD:
3523                         case MONO_WRAPPER_LDFLD:
3524                         case MONO_WRAPPER_LDFLDA:
3525                         case MONO_WRAPPER_LDFLD_REMOTE:
3526                         case MONO_WRAPPER_STFLD_REMOTE:
3527                         case MONO_WRAPPER_STELEMREF:
3528                         case MONO_WRAPPER_ISINST:
3529                         case MONO_WRAPPER_PROXY_ISINST:
3530                         case MONO_WRAPPER_ALLOC:
3531                         case MONO_WRAPPER_REMOTING_INVOKE:
3532                         case MONO_WRAPPER_STATIC_RGCTX_INVOKE:
3533                                 break;
3534                         default:
3535                                 /* unable to handle this */
3536                                 //printf ("Skip (wrapper call):   %s %d -> %s\n", mono_method_full_name (method, TRUE), patch_info->type, mono_method_full_name (patch_info->data.method, TRUE));
3537                                 skip = TRUE;
3538                                 break;
3539                         }
3540                 } else if (patch_info->type == MONO_PATCH_INFO_RGCTX_FETCH) {
3541                         MonoJumpInfo *child = patch_info->data.rgctx_entry->data;
3542
3543                         if (child->type == MONO_PATCH_INFO_METHODCONST) {
3544                                 switch (child->data.method->wrapper_type) {
3545                                 case MONO_WRAPPER_NONE:
3546                                 case MONO_WRAPPER_STATIC_RGCTX_INVOKE:
3547                                         break;
3548                                 default:
3549                                         skip = TRUE;
3550                                 }
3551                         }
3552                 }
3553         }
3554
3555         if (skip) {
3556                 acfg->stats.wrappercount++;
3557                 mono_destroy_compile (cfg);
3558                 return;
3559         }
3560
3561         /*
3562          * Check for methods/klasses we can't encode.
3563          */
3564         skip = FALSE;
3565         for (patch_info = cfg->patch_info; patch_info; patch_info = patch_info->next) {
3566                 switch (patch_info->type) {
3567                 case MONO_PATCH_INFO_METHOD:
3568                 case MONO_PATCH_INFO_METHODCONST:
3569                         if (patch_info->data.method->wrapper_type)
3570                                 break;
3571                         if (!patch_info->data.method->token) {
3572                                 /* The method is part of a constructed type like Int[,].Set (). */
3573                                 if (!g_hash_table_lookup (acfg->token_info_hash, patch_info->data.method))
3574                                         skip = TRUE;
3575                         }
3576                         break;
3577                 case MONO_PATCH_INFO_VTABLE:
3578                 case MONO_PATCH_INFO_CLASS_INIT:
3579                 case MONO_PATCH_INFO_DELEGATE_TRAMPOLINE:
3580                 case MONO_PATCH_INFO_CLASS:
3581                 case MONO_PATCH_INFO_IID:
3582                 case MONO_PATCH_INFO_ADJUSTED_IID:
3583                         if (!patch_info->data.klass->type_token)
3584                                 if (!patch_info->data.klass->element_class->type_token && !(patch_info->data.klass->element_class->rank && patch_info->data.klass->element_class->element_class->type_token))
3585                                         skip = TRUE;
3586                         break;
3587                 default:
3588                         break;
3589                 }
3590         }
3591
3592         if (skip) {
3593                 acfg->stats.ocount++;
3594                 mono_destroy_compile (cfg);
3595                 return;
3596         }
3597
3598         /* Determine whenever the method has GOT slots */
3599         for (patch_info = cfg->patch_info; patch_info; patch_info = patch_info->next) {
3600                 switch (patch_info->type) {
3601                 case MONO_PATCH_INFO_GOT_OFFSET:
3602                 case MONO_PATCH_INFO_NONE:
3603                         break;
3604                 case MONO_PATCH_INFO_IMAGE:
3605                         /* The assembly is stored in GOT slot 0 */
3606                         if (patch_info->data.image != acfg->image)
3607                                 cfg->has_got_slots = TRUE;
3608                         break;
3609                 default:
3610                         if (!is_plt_patch (patch_info))
3611                                 cfg->has_got_slots = TRUE;
3612                         break;
3613                 }
3614         }
3615
3616         if (!cfg->has_got_slots)
3617                 acfg->stats.methods_without_got_slots ++;
3618
3619         /* Make a copy of the patch info which is in the mempool */
3620         {
3621                 MonoJumpInfo *patches = NULL, *patches_end = NULL;
3622
3623                 for (patch_info = cfg->patch_info; patch_info; patch_info = patch_info->next) {
3624                         MonoJumpInfo *new_patch_info = mono_patch_info_dup_mp (acfg->mempool, patch_info);
3625
3626                         if (!patches)
3627                                 patches = new_patch_info;
3628                         else
3629                                 patches_end->next = new_patch_info;
3630                         patches_end = new_patch_info;
3631                 }
3632                 cfg->patch_info = patches;
3633         }
3634
3635         /* Free some fields used by cfg to conserve memory */
3636         mono_mempool_destroy (cfg->mempool);
3637         cfg->mempool = NULL;
3638         g_free (cfg->varinfo);
3639         cfg->varinfo = NULL;
3640         g_free (cfg->vars);
3641         cfg->vars = NULL;
3642         if (cfg->rs) {
3643                 mono_regstate_free (cfg->rs);
3644                 cfg->rs = NULL;
3645         }
3646
3647         //printf ("Compile:           %s\n", mono_method_full_name (method, TRUE));
3648
3649         acfg->cfgs [index] = cfg;
3650
3651         g_hash_table_insert (acfg->method_to_cfg, cfg->orig_method, cfg);
3652
3653         acfg->stats.ccount++;
3654 }
3655
3656 static void
3657 load_profile_files (MonoAotCompile *acfg)
3658 {
3659         FILE *infile;
3660         char *tmp;
3661         int file_index, res, method_index, i;
3662         char ver [256];
3663         guint32 token;
3664         GList *unordered;
3665
3666         file_index = 0;
3667         while (TRUE) {
3668                 tmp = g_strdup_printf ("%s/.mono/aot-profile-data/%s-%s-%d", g_get_home_dir (), acfg->image->assembly_name, acfg->image->guid, file_index);
3669
3670                 if (!g_file_test (tmp, G_FILE_TEST_IS_REGULAR)) {
3671                         g_free (tmp);
3672                         break;
3673                 }
3674
3675                 infile = fopen (tmp, "r");
3676                 g_assert (infile);
3677
3678                 printf ("Using profile data file '%s'\n", tmp);
3679                 g_free (tmp);
3680
3681                 file_index ++;
3682
3683                 res = fscanf (infile, "%32s\n", ver);
3684                 if ((res != 1) || strcmp (ver, "#VER:1") != 0) {
3685                         printf ("Profile file has wrong version or invalid.\n");
3686                         fclose (infile);
3687                         continue;
3688                 }
3689
3690                 while (TRUE) {
3691                         res = fscanf (infile, "%d\n", &token);
3692                         if (res < 1)
3693                                 break;
3694
3695                         method_index = mono_metadata_token_index (token) - 1;
3696
3697                         if (!g_list_find (acfg->method_order, GUINT_TO_POINTER (method_index)))
3698                                 acfg->method_order = g_list_append (acfg->method_order, GUINT_TO_POINTER (method_index));
3699                 }
3700                 fclose (infile);
3701         }
3702
3703         /* Add missing methods */
3704         unordered = NULL;
3705         for (i = 0; i < acfg->image->tables [MONO_TABLE_METHOD].rows; ++i) {
3706                 if (!g_list_find (acfg->method_order, GUINT_TO_POINTER (i)))
3707                         unordered = g_list_prepend (unordered, GUINT_TO_POINTER (i));
3708         }
3709         unordered = g_list_reverse (unordered);
3710         if (acfg->method_order)
3711                 g_list_last (acfg->method_order)->next = unordered;
3712         else
3713                 acfg->method_order = unordered;
3714 }
3715
3716 /**
3717  * alloc_got_slots:
3718  *
3719  *  Collect all patches which have shared GOT entries and alloc entries for them. The
3720  * rest will get entries allocated during emit_code ().
3721  */
3722 static void
3723 alloc_got_slots (MonoAotCompile *acfg)
3724 {
3725         int i;
3726         GList *l;
3727         MonoJumpInfo *ji;
3728
3729         /* Slot 0 is reserved for the address of the current assembly */
3730         ji = mono_mempool_alloc0 (acfg->mempool, sizeof (MonoAotCompile));
3731         ji->type = MONO_PATCH_INFO_IMAGE;
3732         ji->data.image = acfg->image;
3733
3734         get_shared_got_offset (acfg, ji);
3735
3736         for (l = acfg->method_order; l != NULL; l = l->next) {
3737                 i = GPOINTER_TO_UINT (l->data);
3738
3739                 if (acfg->cfgs [i]) {
3740                         MonoCompile *cfg = acfg->cfgs [i];
3741
3742                         for (ji = cfg->patch_info; ji; ji = ji->next) {
3743                                 if (is_shared_got_patch (ji))
3744                                         get_shared_got_offset (acfg, ji);
3745                         }
3746                 }
3747         }
3748 }
3749
3750 static void
3751 emit_code (MonoAotCompile *acfg)
3752 {
3753         int i;
3754         char *symbol;
3755         GList *l;
3756
3757         symbol = g_strdup_printf ("methods");
3758         emit_section_change (acfg, ".text", 0);
3759         emit_global (acfg, symbol, TRUE);
3760         emit_alignment (acfg, 8);
3761         emit_label (acfg, symbol);
3762         g_free (symbol);
3763
3764         for (l = acfg->method_order; l != NULL; l = l->next) {
3765                 i = GPOINTER_TO_UINT (l->data);
3766
3767                 if (acfg->cfgs [i])
3768                         emit_method_code (acfg, acfg->cfgs [i]);
3769         }
3770
3771         symbol = g_strdup_printf ("methods_end");
3772         emit_section_change (acfg, ".text", 0);
3773         emit_global (acfg, symbol, FALSE);
3774         emit_alignment (acfg, 8);
3775         emit_label (acfg, symbol);
3776         g_free (symbol);
3777
3778         symbol = g_strdup_printf ("method_offsets");
3779         emit_section_change (acfg, ".text", 1);
3780         emit_global (acfg, symbol, FALSE);
3781         emit_alignment (acfg, 8);
3782         emit_label (acfg, symbol);
3783         g_free (symbol);
3784
3785         for (i = 0; i < acfg->nmethods; ++i) {
3786                 if (acfg->cfgs [i]) {
3787                         symbol = g_strdup_printf (".Lm_%x", i);
3788                         emit_symbol_diff (acfg, symbol, "methods", 0);
3789                         g_free (symbol);
3790                 } else {
3791                         emit_int32 (acfg, 0xffffffff);
3792                 }
3793         }
3794         emit_line (acfg);
3795 }
3796
3797 static void
3798 emit_info (MonoAotCompile *acfg)
3799 {
3800         int i;
3801         char *symbol;
3802         GList *l;
3803
3804         /* Emit method info */
3805         symbol = g_strdup_printf ("method_info");
3806         emit_section_change (acfg, ".text", 1);
3807         emit_global (acfg, symbol, FALSE);
3808         emit_alignment (acfg, 8);
3809         emit_label (acfg, symbol);
3810         g_free (symbol);
3811
3812         /* To reduce size of generated assembly code */
3813         symbol = g_strdup_printf ("mi");
3814         emit_label (acfg, symbol);
3815         g_free (symbol);
3816
3817         for (l = acfg->method_order; l != NULL; l = l->next) {
3818                 i = GPOINTER_TO_UINT (l->data);
3819
3820                 if (acfg->cfgs [i])
3821                         emit_method_info (acfg, acfg->cfgs [i]);
3822         }
3823
3824         symbol = g_strdup_printf ("method_info_offsets");
3825         emit_section_change (acfg, ".text", 1);
3826         emit_global (acfg, symbol, FALSE);
3827         emit_alignment (acfg, 8);
3828         emit_label (acfg, symbol);
3829         g_free (symbol);
3830
3831         for (i = 0; i < acfg->nmethods; ++i) {
3832                 if (acfg->cfgs [i]) {
3833                         symbol = g_strdup_printf (".Lm_%x_p", i);
3834                         emit_symbol_diff (acfg, symbol, "mi", 0);
3835                         g_free (symbol);
3836                 } else {
3837                         emit_int32 (acfg, 0);
3838                 }
3839         }
3840         emit_line (acfg);
3841 }
3842
3843 static void
3844 emit_wrapper_info (MonoAotCompile *acfg)
3845 {
3846         int i, index;
3847         char *symbol;
3848         char *name;
3849
3850         /* Emit method info */
3851         symbol = g_strdup_printf ("wrapper_info");
3852         emit_section_change (acfg, ".text", 1);
3853         emit_global (acfg, symbol, FALSE);
3854         emit_alignment (acfg, 8);
3855         emit_label (acfg, symbol);
3856         g_free (symbol);
3857
3858         if (!acfg->aot_opts.full_aot)
3859                 return;
3860
3861         for (i = 0; i < acfg->nmethods; ++i) {
3862                 MonoCompile *cfg = acfg->cfgs [i];
3863                 MonoMethod *method;
3864
3865                 if (!cfg || !cfg->orig_method->wrapper_type)
3866                         continue;
3867
3868                 method = cfg->orig_method;
3869                 index = get_method_index (acfg, method);
3870
3871                 // FIXME: Optimize disk usage and lookup speed
3872                 if (method->wrapper_type == MONO_WRAPPER_RUNTIME_INVOKE) {
3873                         char *tmpsig = mono_signature_get_desc (mono_method_signature (method), TRUE);
3874                         name = g_strdup_printf ("(wrapper runtime-invoke):%s (%s)", method->name, tmpsig);
3875                         g_free (tmpsig);
3876                 } else {
3877                         name = mono_method_full_name (cfg->orig_method, TRUE);
3878                 }
3879                 emit_string (acfg, name);
3880                 emit_alignment (acfg, 4);
3881                 emit_int32 (acfg, index);
3882         }
3883
3884         emit_byte (acfg, 0);
3885
3886         emit_line (acfg);
3887 }       
3888
3889 static void
3890 emit_method_order (MonoAotCompile *acfg)
3891 {
3892         int i, index, len;
3893         char *symbol;
3894         GList *l;
3895
3896         symbol = g_strdup_printf ("method_order");
3897         emit_section_change (acfg, ".text", 1);
3898         emit_global (acfg, symbol, FALSE);
3899         emit_alignment (acfg, 8);
3900         emit_label (acfg, symbol);
3901         g_free (symbol);
3902
3903         /* First emit an index table */
3904         index = 0;
3905         len = 0;
3906         for (l = acfg->method_order; l != NULL; l = l->next) {
3907                 i = GPOINTER_TO_UINT (l->data);
3908
3909                 if (acfg->cfgs [i]) {
3910                         if ((index % 1024) == 0) {
3911                                 emit_int32 (acfg, i);
3912                         }
3913
3914                         index ++;
3915                 }
3916
3917                 len ++;
3918         }
3919         emit_int32 (acfg, 0xffffff);
3920
3921         /* Then emit the whole method order */
3922         for (l = acfg->method_order; l != NULL; l = l->next) {
3923                 i = GPOINTER_TO_UINT (l->data);
3924
3925                 if (acfg->cfgs [i]) {
3926                         emit_int32 (acfg, i);
3927                 }
3928         }       
3929         emit_line (acfg);
3930
3931         symbol = g_strdup_printf ("method_order_end");
3932         emit_section_change (acfg, ".text", 1);
3933         emit_global (acfg, symbol, FALSE);
3934         emit_label (acfg, symbol);
3935         g_free (symbol);
3936 }
3937
3938 static void
3939 emit_exception_info (MonoAotCompile *acfg)
3940 {
3941         int i;
3942         char *symbol;
3943
3944         symbol = g_strdup_printf ("ex_info");
3945         emit_section_change (acfg, ".text", 1);
3946         emit_global (acfg, symbol, FALSE);
3947         emit_alignment (acfg, 8);
3948         emit_label (acfg, symbol);
3949         g_free (symbol);
3950
3951         /* To reduce size of generated assembly */
3952         symbol = g_strdup_printf ("ex");
3953         emit_label (acfg, symbol);
3954         g_free (symbol);
3955
3956         for (i = 0; i < acfg->nmethods; ++i) {
3957                 if (acfg->cfgs [i])
3958                         emit_exception_debug_info (acfg, acfg->cfgs [i]);
3959         }
3960
3961         symbol = g_strdup_printf ("ex_info_offsets");
3962         emit_section_change (acfg, ".text", 1);
3963         emit_global (acfg, symbol, FALSE);
3964         emit_alignment (acfg, 8);
3965         emit_label (acfg, symbol);
3966         g_free (symbol);
3967
3968         for (i = 0; i < acfg->nmethods; ++i) {
3969                 if (acfg->cfgs [i]) {
3970                         symbol = g_strdup_printf (".Le_%x_p", i);
3971                         emit_symbol_diff (acfg, symbol, "ex", 0);
3972                         g_free (symbol);
3973                 } else {
3974                         emit_int32 (acfg, 0);
3975                 }
3976         }
3977         emit_line (acfg);
3978 }
3979
3980 static void
3981 emit_class_info (MonoAotCompile *acfg)
3982 {
3983         int i;
3984         char *symbol;
3985
3986         symbol = g_strdup_printf ("class_info");
3987         emit_section_change (acfg, ".text", 1);
3988         emit_global (acfg, symbol, FALSE);
3989         emit_alignment (acfg, 8);
3990         emit_label (acfg, symbol);
3991         g_free (symbol);
3992
3993         for (i = 0; i < acfg->image->tables [MONO_TABLE_TYPEDEF].rows; ++i)
3994                 emit_klass_info (acfg, MONO_TOKEN_TYPE_DEF | (i + 1));
3995
3996         symbol = g_strdup_printf ("class_info_offsets");
3997         emit_section_change (acfg, ".text", 1);
3998         emit_global (acfg, symbol, FALSE);
3999         emit_alignment (acfg, 8);
4000         emit_label (acfg, symbol);
4001         g_free (symbol);
4002
4003         for (i = 0; i < acfg->image->tables [MONO_TABLE_TYPEDEF].rows; ++i) {
4004                 symbol = g_strdup_printf (".LK_I_%x", i);
4005                 emit_symbol_diff (acfg, symbol, "class_info", 0);
4006                 g_free (symbol);
4007         }
4008         emit_line (acfg);
4009 }
4010
4011 typedef struct ClassNameTableEntry {
4012         guint32 token, index;
4013         struct ClassNameTableEntry *next;
4014 } ClassNameTableEntry;
4015
4016 static void
4017 emit_class_name_table (MonoAotCompile *acfg)
4018 {
4019         int i, table_size;
4020         guint32 token, hash;
4021         MonoClass *klass;
4022         GPtrArray *table;
4023         char *full_name;
4024         char *symbol;
4025         ClassNameTableEntry *entry, *new_entry;
4026
4027         /*
4028          * Construct a chained hash table for mapping class names to typedef tokens.
4029          */
4030         table_size = g_spaced_primes_closest ((int)(acfg->image->tables [MONO_TABLE_TYPEDEF].rows * 1.5));
4031         table = g_ptr_array_sized_new (table_size);
4032         for (i = 0; i < table_size; ++i)
4033                 g_ptr_array_add (table, NULL);
4034         for (i = 0; i < acfg->image->tables [MONO_TABLE_TYPEDEF].rows; ++i) {
4035                 token = MONO_TOKEN_TYPE_DEF | (i + 1);
4036                 klass = mono_class_get (acfg->image, token);
4037                 full_name = mono_type_get_name_full (mono_class_get_type (klass), MONO_TYPE_NAME_FORMAT_FULL_NAME);
4038                 hash = g_str_hash (full_name) % table_size;
4039                 g_free (full_name);
4040
4041                 /* FIXME: Allocate from the mempool */
4042                 new_entry = g_new0 (ClassNameTableEntry, 1);
4043                 new_entry->token = token;
4044
4045                 entry = g_ptr_array_index (table, hash);
4046                 if (entry == NULL) {
4047                         new_entry->index = hash;
4048                         g_ptr_array_index (table, hash) = new_entry;
4049                 } else {
4050                         while (entry->next)
4051                                 entry = entry->next;
4052                         
4053                         entry->next = new_entry;
4054                         new_entry->index = table->len;
4055                         g_ptr_array_add (table, new_entry);
4056                 }
4057         }
4058
4059         /* Emit the table */
4060         symbol = g_strdup_printf ("class_name_table");
4061         emit_section_change (acfg, ".text", 0);
4062         emit_global (acfg, symbol, FALSE);
4063         emit_alignment (acfg, 8);
4064         emit_label (acfg, symbol);
4065         g_free (symbol);
4066
4067         /* FIXME: Optimize memory usage */
4068         g_assert (table_size < 65000);
4069         emit_int16 (acfg, table_size);
4070         g_assert (table->len < 65000);
4071         for (i = 0; i < table->len; ++i) {
4072                 ClassNameTableEntry *entry = g_ptr_array_index (table, i);
4073
4074                 if (entry == NULL) {
4075                         emit_int16 (acfg, 0);
4076                         emit_int16 (acfg, 0);
4077                 } else {
4078                         emit_int16 (acfg, mono_metadata_token_index (entry->token));
4079                         if (entry->next)
4080                                 emit_int16 (acfg, entry->next->index);
4081                         else
4082                                 emit_int16 (acfg, 0);
4083                 }
4084         }
4085 }
4086
4087 static void
4088 emit_image_table (MonoAotCompile *acfg)
4089 {
4090         int i;
4091         char *symbol;
4092
4093         /*
4094          * The image table is small but referenced in a lot of places.
4095          * So we emit it at once, and reference its elements by an index.
4096          */
4097
4098         symbol = g_strdup_printf ("mono_image_table");
4099         emit_section_change (acfg, ".text", 1);
4100         emit_global (acfg, symbol, FALSE);
4101         emit_alignment (acfg, 8);
4102         emit_label (acfg, symbol);
4103         g_free (symbol);
4104
4105         emit_int32 (acfg, acfg->image_table->len);
4106         for (i = 0; i < acfg->image_table->len; i++) {
4107                 MonoImage *image = (MonoImage*)g_ptr_array_index (acfg->image_table, i);
4108                 MonoAssemblyName *aname = &image->assembly->aname;
4109
4110                 /* FIXME: Support multi-module assemblies */
4111                 g_assert (image->assembly->image == image);
4112
4113                 emit_string (acfg, image->assembly_name);
4114                 emit_string (acfg, image->guid);
4115                 emit_string (acfg, aname->culture ? aname->culture : "");
4116                 emit_string (acfg, (const char*)aname->public_key_token);
4117
4118                 emit_alignment (acfg, 8);
4119                 emit_int32 (acfg, aname->flags);
4120                 emit_int32 (acfg, aname->major);
4121                 emit_int32 (acfg, aname->minor);
4122                 emit_int32 (acfg, aname->build);
4123                 emit_int32 (acfg, aname->revision);
4124         }
4125 }
4126
4127 static void
4128 emit_got_info (MonoAotCompile *acfg)
4129 {
4130         char *symbol;
4131         int i, buf_size;
4132         guint8 *p, *buf;
4133         guint32 *got_info_offsets;
4134
4135         /**
4136          * FIXME: 
4137          * - optimize offsets table.
4138          * - reduce number of exported symbols.
4139          * - emit info for a klass only once.
4140          * - determine when a method uses a GOT slot which is guaranteed to be already 
4141          *   initialized.
4142          * - clean up and document the code.
4143          * - use String.Empty in class libs.
4144          */
4145
4146         /* Encode info required to decode shared GOT entries */
4147         buf_size = acfg->shared_patches->len * 64;
4148         p = buf = mono_mempool_alloc (acfg->mempool, buf_size);
4149         got_info_offsets = mono_mempool_alloc (acfg->mempool, acfg->shared_patches->len * sizeof (guint32));
4150         for (i = 0; i < acfg->shared_patches->len; ++i) {
4151                 MonoJumpInfo *ji = g_ptr_array_index (acfg->shared_patches, i);
4152
4153                 /* No need to encode the patch type */
4154                 got_info_offsets [i] = p - buf;
4155                 encode_patch (acfg, ji, p, &p);
4156         }
4157
4158         g_assert (p - buf <= buf_size);
4159
4160         acfg->stats.got_info_size = p - buf;
4161
4162         /* Emit got_info table */
4163         symbol = g_strdup_printf ("got_info");
4164         emit_section_change (acfg, ".text", 1);
4165         emit_global (acfg, symbol, FALSE);
4166         emit_alignment (acfg, 8);
4167         emit_label (acfg, symbol);
4168         g_free (symbol);
4169
4170         emit_bytes (acfg, buf, p - buf);
4171
4172         /* Emit got_info_offsets table */
4173         symbol = g_strdup_printf ("got_info_offsets");
4174         emit_section_change (acfg, ".text", 1);
4175         emit_global (acfg, symbol, FALSE);
4176         emit_alignment (acfg, 8);
4177         emit_label (acfg, symbol);
4178         g_free (symbol);
4179
4180         for (i = 0; i < acfg->shared_patches->len; ++i)
4181                 emit_int32 (acfg, got_info_offsets [i]);
4182
4183         acfg->stats.got_info_offsets_size = acfg->shared_patches->len * 4;
4184 }
4185
4186 static void
4187 emit_got (MonoAotCompile *acfg)
4188 {
4189         char *symbol;
4190
4191         /* Don't make GOT global so accesses to it don't need relocations */
4192         symbol = g_strdup_printf ("got");
4193         emit_section_change (acfg, ".bss", 1);
4194         emit_alignment (acfg, 8);
4195         emit_label (acfg, symbol);
4196         if ((acfg->got_offset + acfg->num_trampoline_got_entries) > 0)
4197                 emit_zero_bytes (acfg, (int)((acfg->got_offset + acfg->num_trampoline_got_entries) * sizeof (gpointer)));
4198         g_free (symbol);
4199
4200         symbol = g_strdup_printf ("got_addr");
4201         emit_section_change (acfg, ".data", 1);
4202         emit_global (acfg, symbol, FALSE);
4203         emit_alignment (acfg, 8);
4204         emit_label (acfg, symbol);
4205         emit_pointer (acfg, "got");
4206         g_free (symbol);
4207
4208         symbol = g_strdup_printf ("got_size");
4209         emit_section_change (acfg, ".data", 1);
4210         emit_global (acfg, symbol, FALSE);
4211         emit_alignment (acfg, 8);
4212         emit_label (acfg, symbol);
4213         emit_int32 (acfg, (int)(acfg->got_offset * sizeof (gpointer)));
4214         g_free (symbol);
4215 }
4216
4217 static void
4218 emit_globals (MonoAotCompile *acfg)
4219 {
4220         char *opts_str;
4221
4222         emit_string_symbol (acfg, "mono_assembly_guid" , acfg->image->guid);
4223
4224         emit_string_symbol (acfg, "mono_aot_version", MONO_AOT_FILE_VERSION);
4225
4226         opts_str = g_strdup_printf ("%d", acfg->opts);
4227         emit_string_symbol (acfg, "mono_aot_opt_flags", opts_str);
4228         g_free (opts_str);
4229
4230         emit_string_symbol (acfg, "mono_aot_full_aot", acfg->aot_opts.full_aot ? "TRUE" : "FALSE");
4231
4232         if (acfg->aot_opts.bind_to_runtime_version)
4233                 emit_string_symbol (acfg, "mono_runtime_version", FULL_VERSION);
4234         else
4235                 emit_string_symbol (acfg, "mono_runtime_version", "");
4236
4237         /*
4238          * Some platforms like the iphone have no working dlsym (). To work around this,
4239          * we create an ELF ctor function which will be invoked by dlopen, and which
4240          * will call a function in the AOT loader to register the symbols used by the
4241          * image.
4242          * When static linking, we emit a global which will point to the symbol table.
4243          */
4244         if (acfg->aot_opts.no_dlsym) {
4245                 int i;
4246                 char *symbol;
4247
4248                 if (acfg->aot_opts.static_link)
4249                         /* Emit a string holding the assembly name */
4250                         emit_string_symbol (acfg, "mono_aot_assembly_name", acfg->image->assembly->aname.name);
4251
4252                 /* Emit the names */
4253                 for (i = 0; i < acfg->globals->len; ++i) {
4254                         char *name = g_ptr_array_index (acfg->globals, i);
4255
4256                         symbol = g_strdup_printf ("name_%d", i);
4257                         emit_section_change (acfg, ".text", 1);
4258                         emit_label (acfg, symbol);
4259                         emit_string (acfg, name);
4260                         g_free (symbol);
4261                 }
4262
4263                 /* Emit the globals table */
4264                 symbol = g_strdup_printf ("globals");
4265                 emit_section_change (acfg, ".data", 0);
4266                 /* This is not a global, since it is accessed by the init function */
4267                 emit_alignment (acfg, 8);
4268                 emit_label (acfg, symbol);
4269
4270                 for (i = 0; i < acfg->globals->len; ++i) {
4271                         char *name = g_ptr_array_index (acfg->globals, i);
4272
4273                         symbol = g_strdup_printf ("name_%d", i);
4274                         emit_pointer (acfg, symbol);
4275                         g_free (symbol);
4276
4277                         symbol = g_strdup_printf ("%s", name);
4278                         emit_pointer (acfg, symbol);
4279                         g_free (symbol);
4280                 }
4281                 /* Null terminate the table */
4282                 emit_pointer (acfg, NULL);
4283                 emit_pointer (acfg, NULL);
4284
4285                 if (acfg->aot_opts.static_link) {
4286                         char *p;
4287
4288                         /* 
4289                          * Emit a global symbol which can be passed by an embedding app to
4290                          * mono_aot_register_module ().
4291                          */
4292 #if defined(__MACH__)
4293                         symbol = g_strdup_printf ("_mono_aot_module_%s_info", acfg->image->assembly->aname.name);
4294 #else
4295                         symbol = g_strdup_printf ("mono_aot_module_%s_info", acfg->image->assembly->aname.name);
4296 #endif
4297
4298                         /* Get rid of characters which cannot occur in symbols */
4299                         p = symbol;
4300                         for (p = symbol; *p; ++p) {
4301                                 if (!(isalnum (*p) || *p == '_'))
4302                                         *p = '_';
4303                         }
4304                         acfg->static_linking_symbol = g_strdup (symbol);
4305                         emit_global_inner (acfg, symbol, FALSE);
4306                         emit_alignment (acfg, 8);
4307                         emit_label (acfg, symbol);
4308                         emit_pointer (acfg, "globals");
4309                 } else {
4310                         symbol = g_strdup_printf ("init_%s", acfg->image->assembly->aname.name);
4311                         emit_section_change (acfg, ".text", 1);
4312                         emit_alignment (acfg, 8);
4313                         emit_label (acfg, symbol);
4314 #ifdef USE_BIN_WRITER
4315                         g_assert_not_reached ();
4316 #else
4317 #ifdef __x86_64__
4318                         fprintf (acfg->fp, "leaq globals(%%rip), %%rdi\n");
4319                         fprintf (acfg->fp, "call mono_aot_register_globals@PLT\n");
4320                         fprintf (acfg->fp, "ret\n");
4321                         fprintf (acfg->fp, ".section .ctors,\"aw\",@progbits\n");
4322                         emit_alignment (acfg, 8);
4323                         emit_pointer (acfg, symbol);
4324 #elif defined(__arm__) && defined(__MACH__)
4325                                 
4326                         fprintf (acfg->fp, ".text\n");
4327                         fprintf (acfg->fp, ".align   3\n");
4328                 
4329                         fprintf (acfg->fp, "ldr r0, .L5\n");
4330                         fprintf (acfg->fp, ".LPIC0:\n");
4331                         fprintf (acfg->fp, "add r0, pc, r0\n");
4332                         fprintf (acfg->fp, "ldr r0, [r0]\n");
4333                         fprintf (acfg->fp, "b   _mono_aot_register_globals@PLT\n");
4334                         fprintf (acfg->fp, ".align 2\n");
4335
4336                         fprintf (acfg->fp, ".L5:\n");
4337                         fprintf (acfg->fp, ".long       globals_ptr-(.LPIC0+8)\n");
4338                         
4339                         fprintf (acfg->fp, ".data\n");
4340                         fprintf (acfg->fp, ".align      2\n");
4341                         fprintf (acfg->fp, "globals_ptr:\n");
4342                         fprintf (acfg->fp, ".long       globals\n");
4343                         
4344                         fprintf (acfg->fp, ".mod_init_func\n");
4345                         fprintf (acfg->fp, ".align      2\n");
4346                         fprintf (acfg->fp, ".long       %s@target1\n", symbol);
4347
4348 #elif defined(__arm__)
4349                         /* 
4350                          * Taken from gcc generated code for:
4351                          * static int i;
4352                          * void foo () { bar (&i); }
4353                          * gcc --shared -fPIC -O2
4354                          */
4355                         fprintf (acfg->fp, "ldr r3, .L5\n");
4356                         fprintf (acfg->fp, "ldr r0, .L5+4\n");
4357                         fprintf (acfg->fp, ".LPIC0:\n");
4358                         fprintf (acfg->fp, "add r3, pc, r3\n");
4359                         fprintf (acfg->fp, "add r0, r3, r0\n");
4360                         fprintf (acfg->fp, "b   mono_aot_register_globals(PLT)\n");
4361
4362                         fprintf (acfg->fp, ".L5:\n");
4363                         fprintf (acfg->fp, ".word       _GLOBAL_OFFSET_TABLE_-(.LPIC0+8)\n");
4364                         fprintf (acfg->fp, ".word       globals(GOTOFF)\n");
4365
4366                         fprintf (acfg->fp, ".section    .init_array,\"aw\",%%init_array\n");
4367                         fprintf (acfg->fp, ".align      2\n");
4368                         fprintf (acfg->fp, ".word       %s(target1)\n", symbol);
4369 #else
4370                         g_assert_not_reached ();
4371 #endif
4372 #endif
4373                         g_free (symbol);
4374                 }
4375         }
4376 }
4377
4378 static void
4379 acfg_free (MonoAotCompile *acfg)
4380 {
4381         int i;
4382
4383         for (i = 0; i < acfg->nmethods; ++i)
4384                 if (acfg->cfgs [i])
4385                         g_free (acfg->cfgs [i]);
4386         g_free (acfg->cfgs);
4387         g_free (acfg->method_got_offsets);
4388         g_free (acfg->static_linking_symbol);
4389         g_ptr_array_free (acfg->methods, TRUE);
4390         g_ptr_array_free (acfg->shared_patches, TRUE);
4391         g_ptr_array_free (acfg->image_table, TRUE);
4392         g_ptr_array_free (acfg->globals, TRUE);
4393         g_hash_table_destroy (acfg->method_indexes);
4394         g_hash_table_destroy (acfg->plt_offset_to_patch);
4395         g_hash_table_destroy (acfg->patch_to_plt_offset);
4396         g_hash_table_destroy (acfg->patch_to_shared_got_offset);
4397         g_hash_table_destroy (acfg->method_to_cfg);
4398         g_hash_table_destroy (acfg->token_info_hash);
4399         g_hash_table_destroy (acfg->image_hash);
4400         mono_mempool_destroy (acfg->mempool);
4401         g_free (acfg);
4402 }
4403
4404 int
4405 mono_compile_assembly (MonoAssembly *ass, guint32 opts, const char *aot_options)
4406 {
4407         MonoImage *image = ass->image;
4408         char *symbol;
4409         int i, res;
4410         MonoAotCompile *acfg;
4411
4412         printf ("Mono Ahead of Time compiler - compiling assembly %s\n", image->name);
4413
4414         acfg = g_new0 (MonoAotCompile, 1);
4415         acfg->methods = g_ptr_array_new ();
4416         acfg->method_indexes = g_hash_table_new (NULL, NULL);
4417         acfg->plt_offset_to_patch = g_hash_table_new (NULL, NULL);
4418         acfg->patch_to_plt_offset = g_hash_table_new (mono_patch_info_hash, mono_patch_info_equal);
4419         acfg->patch_to_shared_got_offset = g_hash_table_new (mono_patch_info_hash, mono_patch_info_equal);
4420         acfg->shared_patches = g_ptr_array_new ();
4421         acfg->method_to_cfg = g_hash_table_new (NULL, NULL);
4422         acfg->token_info_hash = g_hash_table_new_full (NULL, NULL, NULL, g_free);
4423         acfg->image_hash = g_hash_table_new (NULL, NULL);
4424         acfg->image_table = g_ptr_array_new ();
4425         acfg->globals = g_ptr_array_new ();
4426         acfg->image = image;
4427         acfg->opts = opts;
4428         acfg->mempool = mono_mempool_new ();
4429
4430         mono_aot_parse_options (aot_options, &acfg->aot_opts);
4431
4432         load_profile_files (acfg);
4433
4434         emit_start (acfg);
4435
4436         acfg->num_aot_trampolines = acfg->aot_opts.full_aot ? 10240 : 0;
4437
4438         acfg->method_index = 1;
4439
4440         /* Collect methods */
4441         for (i = 0; i < image->tables [MONO_TABLE_METHOD].rows; ++i) {
4442                 MonoMethod *method;
4443                 guint32 token = MONO_TOKEN_METHOD_DEF | (i + 1);
4444
4445                 method = mono_get_method (acfg->image, token, NULL);
4446
4447                 if (acfg->aot_opts.full_aot && method->iflags & METHOD_IMPL_ATTRIBUTE_INTERNAL_CALL) {
4448                         /* Compile the wrapper instead */
4449                         /* We do this here instead of add_wrappers () because it is easy to do it here */
4450                         MonoMethod *wrapper = mono_marshal_get_native_wrapper (method, check_for_pending_exc, TRUE);
4451                         method = wrapper;
4452                 }
4453
4454                 /* Since we add the normal methods first, their index will be equal to their zero based token index */
4455                 add_method_with_index (acfg, method, i);
4456                 acfg->method_index ++;
4457         }
4458
4459         if (acfg->aot_opts.full_aot)
4460                 add_wrappers (acfg);
4461
4462         acfg->nmethods = acfg->methods->len + 1;
4463         acfg->cfgs = g_new0 (MonoCompile*, acfg->nmethods + 32);
4464         acfg->method_got_offsets = g_new0 (guint32, acfg->nmethods + 32);
4465
4466         /* PLT offset 0 is reserved for the PLT trampoline */
4467         acfg->plt_offset = 1;
4468
4469         /* Compile methods */
4470         for (i = 0; i < acfg->methods->len; ++i) {
4471                 compile_method (acfg, g_ptr_array_index (acfg->methods, i));
4472         }
4473
4474         alloc_got_slots (acfg);
4475
4476         emit_code (acfg);
4477
4478         emit_info (acfg);
4479
4480         emit_wrapper_info (acfg);
4481
4482         emit_method_order (acfg);
4483
4484         emit_trampolines (acfg);
4485
4486         emit_class_name_table (acfg);
4487
4488         emit_got_info (acfg);
4489
4490         emit_exception_info (acfg);
4491
4492         emit_class_info (acfg);
4493
4494         emit_plt (acfg);
4495
4496         emit_image_table (acfg);
4497
4498         emit_got (acfg);
4499
4500         emit_globals (acfg);
4501
4502         symbol = g_strdup_printf ("mem_end");
4503         emit_section_change (acfg, ".text", 1);
4504         emit_global (acfg, symbol, FALSE);
4505         emit_alignment (acfg, 8);
4506         emit_label (acfg, symbol);
4507         g_free (symbol);
4508
4509         printf ("Code: %d Info: %d Ex Info: %d Class Info: %d PLT: %d GOT Info: %d GOT Info Offsets: %d GOT: %d\n", acfg->stats.code_size, acfg->stats.info_size, acfg->stats.ex_info_size, acfg->stats.class_info_size, acfg->plt_offset, acfg->stats.got_info_size, acfg->stats.got_info_offsets_size, (int)(acfg->got_offset * sizeof (gpointer)));
4510
4511         res = emit_writeout (acfg);
4512         if (res != 0) {
4513                 acfg_free (acfg);
4514                 return res;
4515         }
4516
4517         printf ("Compiled %d out of %d methods (%d%%)\n", acfg->stats.ccount, acfg->stats.mcount, acfg->stats.mcount ? (acfg->stats.ccount * 100) / acfg->stats.mcount : 100);
4518         if (acfg->stats.genericcount)
4519                 printf ("%d methods are generic (%d%%)\n", acfg->stats.genericcount, acfg->stats.mcount ? (acfg->stats.genericcount * 100) / acfg->stats.mcount : 100);
4520         if (acfg->stats.abscount)
4521                 printf ("%d methods contain absolute addresses (%d%%)\n", acfg->stats.abscount, acfg->stats.mcount ? (acfg->stats.abscount * 100) / acfg->stats.mcount : 100);
4522         if (acfg->stats.wrappercount)
4523                 printf ("%d methods contain wrapper references (%d%%)\n", acfg->stats.wrappercount, acfg->stats.mcount ? (acfg->stats.wrappercount * 100) / acfg->stats.mcount : 100);
4524         if (acfg->stats.lmfcount)
4525                 printf ("%d methods contain lmf pointers (%d%%)\n", acfg->stats.lmfcount, acfg->stats.mcount ? (acfg->stats.lmfcount * 100) / acfg->stats.mcount : 100);
4526         if (acfg->stats.ocount)
4527                 printf ("%d methods have other problems (%d%%)\n", acfg->stats.ocount, acfg->stats.mcount ? (acfg->stats.ocount * 100) / acfg->stats.mcount : 100);
4528         printf ("Methods without GOT slots: %d (%d%%)\n", acfg->stats.methods_without_got_slots, acfg->stats.mcount ? (acfg->stats.methods_without_got_slots * 100) / acfg->stats.mcount : 100);
4529         printf ("Direct calls: %d (%d%%)\n", acfg->stats.direct_calls, acfg->stats.all_calls ? (acfg->stats.direct_calls * 100) / acfg->stats.all_calls : 100);
4530
4531         printf ("GOT slot distribution:\n");
4532         for (i = 0; i < MONO_PATCH_INFO_NONE; ++i)
4533                 if (acfg->stats.got_slot_types [i])
4534                         printf ("\t%s: %d\n", get_patch_name (i), acfg->stats.got_slot_types [i]);
4535
4536         acfg_free (acfg);
4537         
4538         return 0;
4539 }
4540
4541 #else
4542
4543 /* AOT disabled */
4544
4545 int
4546 mono_compile_assembly (MonoAssembly *ass, guint32 opts, const char *aot_options)
4547 {
4548         return 0;
4549 }
4550
4551 #endif