Merge pull request #2247 from ivmai/match-ext-libgc-api
[mono.git] / mono / mini / mini-trampolines.c
1 /*
2  * (C) 2003 Ximian, Inc.
3  * (C) 2003-2011 Novell, Inc.
4  * Copyright 2011 Xamarin, Inc (http://www.xamarin.com)
5  */
6 #include <config.h>
7 #include <glib.h>
8
9 #include <mono/metadata/appdomain.h>
10 #include <mono/metadata/metadata-internals.h>
11 #include <mono/metadata/marshal.h>
12 #include <mono/metadata/tabledefs.h>
13 #include <mono/utils/mono-counters.h>
14 #include <mono/utils/mono-error-internals.h>
15 #include <mono/utils/mono-membar.h>
16 #include <mono/utils/mono-compiler.h>
17
18 #include "mini.h"
19
20 /*
21  * Address of the trampoline code.  This is used by the debugger to check
22  * whether a method is a trampoline.
23  */
24 guint8* mono_trampoline_code [MONO_TRAMPOLINE_NUM];
25
26 static GHashTable *rgctx_lazy_fetch_trampoline_hash;
27 static GHashTable *rgctx_lazy_fetch_trampoline_hash_addr;
28 static guint32 trampoline_calls, jit_trampolines, unbox_trampolines, static_rgctx_trampolines;
29
30 #define mono_trampolines_lock() mono_os_mutex_lock (&trampolines_mutex)
31 #define mono_trampolines_unlock() mono_os_mutex_unlock (&trampolines_mutex)
32 static mono_mutex_t trampolines_mutex;
33
34 #ifdef MONO_ARCH_GSHARED_SUPPORTED
35
36 typedef struct {
37         MonoMethod *m;
38         gpointer addr;
39 } RgctxTrampInfo;
40
41 static gint
42 rgctx_tramp_info_equal (gconstpointer ka, gconstpointer kb)
43 {
44         const RgctxTrampInfo *i1 = ka;
45         const RgctxTrampInfo *i2 = kb;
46
47         if (i1->m == i2->m && i1->addr == i2->addr)
48                 return 1;
49         else
50                 return 0;
51 }
52
53 static guint
54 rgctx_tramp_info_hash (gconstpointer data)
55 {
56         const RgctxTrampInfo *info = data;
57
58         return GPOINTER_TO_UINT (info->m) ^ GPOINTER_TO_UINT (info->addr);
59 }
60
61 /**
62  * mono_create_static_rgctx_trampoline:
63  * @m: the mono method to create a trampoline for
64  * @addr: the address to jump to (where the compiled code for M lives)
65  *
66  * Creates a static rgctx trampoline for M which branches to ADDR which should
67  * point to the compiled code of M.
68  *
69  * Static rgctx trampolines are used when a shared generic method which doesn't
70  * have a this argument is called indirectly, ie. from code which can't pass in
71  * the rgctx argument. The trampoline sets the rgctx argument and jumps to the
72  * methods code. These trampolines are similar to the unbox trampolines, they
73  * perform the same task as the static rgctx wrappers, but they are smaller/faster,
74  * and can be made to work with full AOT.
75  *
76  * On PPC addr should be an ftnptr and the return value is an ftnptr too.
77  *
78  * Returns the generated static rgctx trampoline.
79  */
80 gpointer
81 mono_create_static_rgctx_trampoline (MonoMethod *m, gpointer addr)
82 {
83         gpointer ctx;
84         gpointer res;
85         MonoDomain *domain;
86         RgctxTrampInfo tmp_info;
87         RgctxTrampInfo *info;
88
89 #ifdef PPC_USES_FUNCTION_DESCRIPTOR
90         g_assert (((gpointer*)addr) [2] == 0);
91 #endif
92
93         ctx = mini_method_get_rgctx (m);
94
95         domain = mono_domain_get ();
96
97         /* 
98          * In the AOT case, addr might point to either the method, or to an unbox trampoline,
99          * so make the hash keyed on the m+addr pair.
100          */
101         mono_domain_lock (domain);
102         if (!domain_jit_info (domain)->static_rgctx_trampoline_hash)
103                 domain_jit_info (domain)->static_rgctx_trampoline_hash = g_hash_table_new (rgctx_tramp_info_hash, rgctx_tramp_info_equal);
104         tmp_info.m = m;
105         tmp_info.addr = addr;
106         res = g_hash_table_lookup (domain_jit_info (domain)->static_rgctx_trampoline_hash,
107                                                            &tmp_info);
108         mono_domain_unlock (domain);
109         if (res)
110                 return res;
111
112         if (mono_aot_only)
113                 res = mono_aot_get_static_rgctx_trampoline (ctx, addr);
114         else
115                 res = mono_arch_get_static_rgctx_trampoline (m, ctx, addr);
116
117         mono_domain_lock (domain);
118         /* Duplicates inserted while we didn't hold the lock are OK */
119         info = mono_domain_alloc (domain, sizeof (RgctxTrampInfo));
120         info->m = m;
121         info->addr = addr;
122         g_hash_table_insert (domain_jit_info (domain)->static_rgctx_trampoline_hash, info, res);
123         mono_domain_unlock (domain);
124
125         static_rgctx_trampolines ++;
126
127         return res;
128 }
129 #else
130 gpointer
131 mono_create_static_rgctx_trampoline (MonoMethod *m, gpointer addr)
132 {
133        /* 
134         * This shouldn't happen as all arches which support generic sharing support
135         * static rgctx trampolines as well.
136         */
137        g_assert_not_reached ();
138 }
139 #endif
140
141 #if 0
142 #define DEBUG_IMT(stmt) do { stmt; } while (0)
143 #else
144 #define DEBUG_IMT(stmt) do { } while (0)
145 #endif
146
147 /*
148  * mini_resolve_imt_method:
149  *
150  *   Resolve the actual method called when making an IMT call through VTABLE_SLOT with IMT_METHOD as the interface method.
151  *
152  * Either IMPL_METHOD or OUT_AOT_ADDR will be set on return.
153  */
154 gpointer*
155 mini_resolve_imt_method (MonoVTable *vt, gpointer *vtable_slot, MonoMethod *imt_method, MonoMethod **impl_method, gpointer *out_aot_addr, gboolean *out_need_rgctx_tramp, MonoMethod **variant_iface)
156 {
157         MonoMethod *impl = NULL, *generic_virtual = NULL;
158         gboolean lookup_aot, variance_used = FALSE, need_rgctx_tramp = FALSE;
159         gpointer addr;
160         guint8 *aot_addr = NULL;
161         int displacement = vtable_slot - ((gpointer*)vt);
162         int interface_offset;
163         int imt_slot = MONO_IMT_SIZE + displacement;
164
165         g_assert (imt_slot < MONO_IMT_SIZE);
166
167         /* This has to be variance aware since imt_method can be from an interface that vt->klass doesn't directly implement */
168         interface_offset = mono_class_interface_offset_with_variance (vt->klass, imt_method->klass, &variance_used);
169         if (interface_offset < 0)
170                 g_error ("%s doesn't implement interface %s\n", mono_type_get_name_full (&vt->klass->byval_arg, 0), mono_type_get_name_full (&imt_method->klass->byval_arg, 0));
171
172         *variant_iface = NULL;
173         if (imt_method->is_inflated && ((MonoMethodInflated*)imt_method)->context.method_inst) {
174                 /* Generic virtual method */
175                 generic_virtual = imt_method;
176                 need_rgctx_tramp = TRUE;
177         } else if (variance_used && mono_class_has_variant_generic_params (imt_method->klass)) {
178                 *variant_iface = imt_method;
179         }
180
181         addr = NULL;
182         /* We can only use the AOT compiled code if we don't require further processing */
183         lookup_aot = !generic_virtual & !variant_iface;
184
185         mono_vtable_build_imt_slot (vt, mono_method_get_imt_slot (imt_method));
186
187         if (imt_method->is_inflated && ((MonoMethodInflated*)imt_method)->context.method_inst) {
188                 MonoError error;
189                 MonoGenericContext context = { NULL, NULL };
190
191                 /*
192                  * Generic virtual method, imt_method contains the inflated interface
193                  * method, need to get the inflated impl method.
194                  */
195                 /* imt_method->slot might not be set */
196                 impl = mono_class_get_vtable_entry (vt->klass, interface_offset + mono_method_get_declaring_generic_method (imt_method)->slot);
197
198                 if (impl->klass->generic_class)
199                         context.class_inst = impl->klass->generic_class->context.class_inst;
200                 context.method_inst = ((MonoMethodInflated*)imt_method)->context.method_inst;
201                 impl = mono_class_inflate_generic_method_checked (impl, &context, &error);
202                 g_assert (mono_error_ok (&error)); /* FIXME don't swallow the error */
203         } else {
204                 /* Avoid loading metadata or creating a generic vtable if possible */
205                 if (lookup_aot && !vt->klass->valuetype)
206                         aot_addr = mono_aot_get_method_from_vt_slot (mono_domain_get (), vt, interface_offset + mono_method_get_vtable_slot (imt_method));
207                 else
208                         aot_addr = NULL;
209                 if (aot_addr)
210                         impl = NULL;
211                 else
212                         impl = mono_class_get_vtable_entry (vt->klass, interface_offset + mono_method_get_vtable_slot (imt_method));
213         }
214
215         if (impl && mono_method_needs_static_rgctx_invoke (impl, FALSE))
216                 need_rgctx_tramp = TRUE;
217         if (impl && impl->wrapper_type == MONO_WRAPPER_MANAGED_TO_MANAGED) {
218                 WrapperInfo *info = mono_marshal_get_wrapper_info (impl);
219
220                 if (info && info->subtype == WRAPPER_SUBTYPE_GENERIC_ARRAY_HELPER)
221                         need_rgctx_tramp = TRUE;
222         }
223         *impl_method = impl;
224         *out_need_rgctx_tramp = need_rgctx_tramp;
225         *out_aot_addr = aot_addr;
226
227         DEBUG_IMT (printf ("mono_convert_imt_slot_to_vtable_slot: method = %s.%s.%s, imt_method = %s.%s.%s\n",
228                                            method->klass->name_space, method->klass->name, method->name,
229                                            imt_method->klass->name_space, imt_method->klass->name, imt_method->name));
230
231         if (vt->imt_collisions_bitmap & (1 << imt_slot)) {
232                 int slot = mono_method_get_vtable_index (imt_method);
233                 int vtable_offset;
234
235                 g_assert (slot != -1);
236                 vtable_offset = interface_offset + slot;
237                 vtable_slot = & (vt->vtable [vtable_offset]);
238                 DEBUG_IMT (printf ("mono_convert_imt_slot_to_vtable_slot: slot %p[%d] is in the IMT, and colliding becomes %p[%d] (interface_offset = %d, method->slot = %d)\n", slot, imt_slot, vtable_slot, vtable_offset, interface_offset, imt_method->slot));
239                 return vtable_slot;
240         } else {
241                 DEBUG_IMT (printf ("mono_convert_imt_slot_to_vtable_slot: slot %p[%d] is in the IMT, but not colliding\n", slot, imt_slot));
242                 return vtable_slot;
243         }
244 }
245
246 /*
247  * This is a super-ugly hack to fix bug #616463.
248  *
249  * The problem is that we don't always set is_generic for generic
250  * method definitions.  See the comment at the end of
251  * mono_class_inflate_generic_method_full_checked() in class.c.
252  */
253 static gboolean
254 is_generic_method_definition (MonoMethod *m)
255 {
256         MonoGenericContext *context;
257         if (m->is_generic)
258                 return TRUE;
259         if (!m->is_inflated)
260                 return FALSE;
261
262         context = mono_method_get_context (m);
263         if (!context->method_inst)
264                 return FALSE;
265         if (context->method_inst == mono_method_get_generic_container (((MonoMethodInflated*)m)->declaring)->context.method_inst)
266                 return TRUE;
267         return FALSE;
268 }
269
270 gboolean
271 mini_jit_info_is_gsharedvt (MonoJitInfo *ji)
272 {
273         if (ji && ji->has_generic_jit_info && (mono_jit_info_get_generic_sharing_context (ji)->is_gsharedvt))
274                 return TRUE;
275         else
276                 return FALSE;
277 }
278
279 /**
280  * mini_add_method_trampoline:
281  * @m: 
282  * @compiled_method:
283  * @add_static_rgctx_tramp: adds a static rgctx trampoline
284  * @add_unbox_tramp: adds an unboxing trampoline
285  *
286  * Add static rgctx/gsharedvt_in/unbox trampolines to
287  * M/COMPILED_METHOD if needed.
288  *
289  * Returns the trampoline address, or COMPILED_METHOD if no trampoline
290  * is needed.
291  */
292 gpointer
293 mini_add_method_trampoline (MonoMethod *m, gpointer compiled_method, gboolean add_static_rgctx_tramp, gboolean add_unbox_tramp)
294 {
295         gpointer addr = compiled_method;
296         gboolean callee_gsharedvt, callee_array_helper;
297         MonoMethod *jmethod = NULL;
298         MonoJitInfo *ji;
299
300         // FIXME: This loads information from AOT (perf problem)
301         ji = mini_jit_info_table_find (mono_domain_get (), mono_get_addr_from_ftnptr (compiled_method), NULL);
302         callee_gsharedvt = mini_jit_info_is_gsharedvt (ji);
303
304         callee_array_helper = FALSE;
305         if (m->wrapper_type == MONO_WRAPPER_MANAGED_TO_MANAGED) {
306                 WrapperInfo *info = mono_marshal_get_wrapper_info (m);
307
308                 /*
309                  * generic array helpers.
310                  * Have to replace the wrappers with the original generic instances.
311                  */
312                 if (info && info->subtype == WRAPPER_SUBTYPE_GENERIC_ARRAY_HELPER) {
313                         callee_array_helper = TRUE;
314                         m = info->d.generic_array_helper.method;
315                 }
316         } else if (m->wrapper_type == MONO_WRAPPER_UNKNOWN) {
317                 WrapperInfo *info = mono_marshal_get_wrapper_info (m);
318
319                 /* Same for synchronized inner wrappers */
320                 if (info && info->subtype == WRAPPER_SUBTYPE_SYNCHRONIZED_INNER) {
321                         m = info->d.synchronized_inner.method;
322                 }
323         }
324
325         if (callee_gsharedvt)
326                 g_assert (m->is_inflated);
327
328         addr = compiled_method;
329
330         if (add_unbox_tramp) {
331                 /* 
332                  * The unbox trampolines call the method directly, so need to add
333                  * an rgctx tramp before them.
334                  */
335                 if (mono_aot_only) {
336                         addr = mono_aot_get_unbox_trampoline (m);
337                 } else {
338                         unbox_trampolines ++;
339                         addr = mono_arch_get_unbox_trampoline (m, addr);
340                 }
341         }
342
343         if (ji && !ji->is_trampoline)
344                 jmethod = jinfo_get_method (ji);
345         if (callee_gsharedvt && mini_is_gsharedvt_variable_signature (mono_method_signature (jmethod))) {
346                 MonoMethodSignature *sig, *gsig;
347
348                 /* Here m is a generic instance, while ji->method is the gsharedvt method implementing it */
349
350                 /* Call from normal/gshared code to gsharedvt code with variable signature */
351                 sig = mono_method_signature (m);
352                 gsig = mono_method_signature (jmethod);
353
354                 addr = mini_get_gsharedvt_wrapper (TRUE, addr, sig, gsig, -1, FALSE);
355
356                 //printf ("IN: %s\n", mono_method_full_name (m, TRUE));
357         }
358
359         if (callee_array_helper) {
360                 add_static_rgctx_tramp = FALSE;
361                 /* FIXME: ji->from_aot is not set for llvm methods */
362                 if (ji && (ji->from_aot || mono_aot_only)) {
363                         /* In AOT mode, compiled_method points to one of the InternalArray methods in Array. */
364                         if (!mono_llvm_only && mono_method_needs_static_rgctx_invoke (jinfo_get_method (ji), TRUE))
365                                 add_static_rgctx_tramp = TRUE;
366                 }
367         }
368
369         if (mono_llvm_only)
370                 add_static_rgctx_tramp = FALSE;
371
372         if (add_static_rgctx_tramp)
373                 addr = mono_create_static_rgctx_trampoline (m, addr);
374
375         return addr;
376 }
377
378 /**
379  * common_call_trampoline:
380  *
381  *   The code to handle normal, virtual, and interface method calls and jumps, both
382  * from JITted and LLVM compiled code.
383  */
384 static gpointer
385 common_call_trampoline_inner (mgreg_t *regs, guint8 *code, MonoMethod *m, MonoVTable *vt, gpointer *vtable_slot)
386 {
387         gpointer addr, compiled_method;
388         gboolean generic_shared = FALSE;
389         gboolean need_unbox_tramp = FALSE;
390         gboolean need_rgctx_tramp = FALSE;
391         MonoMethod *declaring = NULL;
392         MonoMethod *generic_virtual = NULL, *variant_iface = NULL;
393         int context_used;
394         gboolean imt_call, virtual;
395         gpointer *orig_vtable_slot, *vtable_slot_to_patch = NULL;
396         MonoJitInfo *ji = NULL;
397
398         virtual = vt && (gpointer)vtable_slot > (gpointer)vt;
399         imt_call = vt && (gpointer)vtable_slot < (gpointer)vt;
400
401         /*
402          * rgctx trampolines are needed when the call is indirect so the caller can't pass
403          * the rgctx argument needed by the callee.
404          */
405         if (virtual && m)
406                 need_rgctx_tramp = mono_method_needs_static_rgctx_invoke (m, FALSE);
407
408         orig_vtable_slot = vtable_slot;
409         vtable_slot_to_patch = vtable_slot;
410
411         /* IMT call */
412         if (imt_call) {
413                 MonoMethod *imt_method = NULL, *impl_method = NULL;
414                 MonoObject *this_arg;
415
416                 g_assert (vtable_slot);
417
418                 imt_method = mono_arch_find_imt_method (regs, code);
419                 this_arg = mono_arch_get_this_arg_from_call (regs, code);
420
421                 if (mono_object_is_transparent_proxy (this_arg)) {
422                         /* Use the slow path for now */
423                     m = mono_object_get_virtual_method (this_arg, imt_method);
424                         vtable_slot_to_patch = NULL;
425                 } else {
426                         if (imt_method->is_inflated && ((MonoMethodInflated*)imt_method)->context.method_inst) {
427                                 /* Generic virtual method */
428                                 generic_virtual = imt_method;
429                                 need_rgctx_tramp = TRUE;
430                         }
431
432                         vtable_slot = mini_resolve_imt_method (vt, vtable_slot, imt_method, &impl_method, &addr, &need_rgctx_tramp, &variant_iface);
433                         /* This is the vcall slot which gets called through the IMT thunk */
434                         vtable_slot_to_patch = vtable_slot;
435
436                         if (addr) {
437                                 /*
438                                  * We found AOT compiled code for the method, skip the rest.
439                                  */
440                                 if (mono_domain_owns_vtable_slot (mono_domain_get (), vtable_slot))
441                                         *vtable_slot = addr;
442
443                                 return mono_create_ftnptr (mono_domain_get (), addr);
444                         }
445
446                         m = impl_method;
447                 }
448         }
449
450         /*
451          * The virtual check is needed because is_generic_method_definition (m) could
452          * return TRUE for methods used in IMT calls too.
453          */
454         if (virtual && is_generic_method_definition (m)) {
455                 MonoError error;
456                 MonoGenericContext context = { NULL, NULL };
457                 MonoMethod *declaring;
458
459                 if (m->is_inflated)
460                         declaring = mono_method_get_declaring_generic_method (m);
461                 else
462                         declaring = m;
463
464                 if (m->klass->generic_class)
465                         context.class_inst = m->klass->generic_class->context.class_inst;
466                 else
467                         g_assert (!m->klass->generic_container);
468
469                 generic_virtual = mono_arch_find_imt_method (regs, code);
470                 g_assert (generic_virtual);
471                 g_assert (generic_virtual->is_inflated);
472                 context.method_inst = ((MonoMethodInflated*)generic_virtual)->context.method_inst;
473
474                 m = mono_class_inflate_generic_method_checked (declaring, &context, &error);
475                 g_assert (mono_error_ok (&error)); /* FIXME don't swallow the error */
476                 /* FIXME: only do this if the method is sharable */
477                 need_rgctx_tramp = TRUE;
478         } else if ((context_used = mono_method_check_context_used (m))) {
479                 MonoClass *klass = NULL;
480                 MonoMethod *actual_method = NULL;
481                 MonoVTable *vt = NULL;
482                 MonoGenericInst *method_inst = NULL;
483
484                 vtable_slot = NULL;
485                 generic_shared = TRUE;
486
487                 g_assert (code);
488
489                 /*
490                  * The caller is gshared code, compute the actual method to call from M and this/rgctx.
491                  */
492                 if (m->is_inflated && mono_method_get_context (m)->method_inst) {
493                         MonoMethodRuntimeGenericContext *mrgctx = (MonoMethodRuntimeGenericContext*)mono_arch_find_static_call_vtable (regs, code);
494
495                         klass = mrgctx->class_vtable->klass;
496                         method_inst = mrgctx->method_inst;
497                 } else if ((m->flags & METHOD_ATTRIBUTE_STATIC) || m->klass->valuetype) {
498                         MonoVTable *vtable = mono_arch_find_static_call_vtable (regs, code);
499
500                         klass = vtable->klass;
501                 } else {
502                         MonoObject *this_argument = mono_arch_get_this_arg_from_call (regs, code);
503
504                         vt = this_argument->vtable;
505                         vtable_slot = orig_vtable_slot;
506
507                         g_assert (this_argument->vtable->klass->inited);
508
509                         if (!vtable_slot) {
510                                 mono_class_setup_supertypes (this_argument->vtable->klass);
511                                 klass = this_argument->vtable->klass->supertypes [m->klass->idepth - 1];
512                         }
513                 }
514
515                 g_assert (vtable_slot || klass);
516
517                 if (vtable_slot) {
518                         int displacement = vtable_slot - ((gpointer*)vt);
519
520                         g_assert_not_reached ();
521
522                         g_assert (displacement > 0);
523
524                         actual_method = vt->klass->vtable [displacement];
525                 }
526
527                 if (method_inst || m->wrapper_type) {
528                         MonoError error;
529                         MonoGenericContext context = { NULL, NULL };
530
531                         if (m->is_inflated)
532                                 declaring = mono_method_get_declaring_generic_method (m);
533                         else
534                                 declaring = m;
535
536                         if (klass->generic_class)
537                                 context.class_inst = klass->generic_class->context.class_inst;
538                         else if (klass->generic_container)
539                                 context.class_inst = klass->generic_container->context.class_inst;
540                         context.method_inst = method_inst;
541
542                         actual_method = mono_class_inflate_generic_method_checked (declaring, &context, &error);
543                         g_assert (mono_error_ok (&error)); /* FIXME don't swallow the error */
544                 } else {
545                         actual_method = mono_class_get_method_generic (klass, m);
546                 }
547
548                 g_assert (klass);
549                 g_assert (actual_method);
550                 g_assert (actual_method->klass == klass);
551
552                 if (actual_method->is_inflated)
553                         declaring = mono_method_get_declaring_generic_method (actual_method);
554                 else
555                         declaring = NULL;
556
557                 m = actual_method;
558         }
559
560         if (m->iflags & METHOD_IMPL_ATTRIBUTE_SYNCHRONIZED) {
561                 m = mono_marshal_get_synchronized_wrapper (m);
562                 need_rgctx_tramp = FALSE;
563         }
564
565         /* Calls made through delegates on platforms without delegate trampolines */
566         if (!code && mono_method_needs_static_rgctx_invoke (m, FALSE))
567                 need_rgctx_tramp = TRUE;
568
569         addr = compiled_method = mono_compile_method (m);
570         g_assert (addr);
571
572         if (generic_virtual || variant_iface) {
573                 if (vt->klass->valuetype) /*FIXME is this required variant iface?*/
574                         need_unbox_tramp = TRUE;
575         } else if (orig_vtable_slot) {
576                 if (m->klass->valuetype)
577                         need_unbox_tramp = TRUE;
578         }
579
580         addr = mini_add_method_trampoline (m, compiled_method, need_rgctx_tramp, need_unbox_tramp);
581
582         if (generic_virtual || variant_iface) {
583                 MonoMethod *target = generic_virtual ? generic_virtual : variant_iface;
584
585                 vtable_slot = orig_vtable_slot;
586                 g_assert (vtable_slot);
587
588                 mono_method_add_generic_virtual_invocation (mono_domain_get (), 
589                                                                                                         vt, vtable_slot,
590                                                                                                         target, addr);
591
592                 return addr;
593         }
594
595         /* the method was jumped to */
596         if (!code) {
597                 MonoDomain *domain = mono_domain_get ();
598
599                 /* Patch the got entries pointing to this method */
600                 /* 
601                  * We do this here instead of in mono_codegen () to cover the case when m
602                  * was loaded from an aot image.
603                  */
604                 if (domain_jit_info (domain)->jump_target_got_slot_hash) {
605                         GSList *list, *tmp;
606
607                         mono_domain_lock (domain);
608                         list = g_hash_table_lookup (domain_jit_info (domain)->jump_target_got_slot_hash, m);
609                         if (list) {
610                                 for (tmp = list; tmp; tmp = tmp->next) {
611                                         gpointer *got_slot = tmp->data;
612                                         *got_slot = addr;
613                                 }
614                                 g_hash_table_remove (domain_jit_info (domain)->jump_target_got_slot_hash, m);
615                                 g_slist_free (list);
616                         }
617                         mono_domain_unlock (domain);
618                 }
619
620                 return addr;
621         }
622
623         vtable_slot = orig_vtable_slot;
624
625         if (vtable_slot) {
626                 if (vtable_slot_to_patch && (mono_aot_is_got_entry (code, (guint8*)vtable_slot_to_patch) || mono_domain_owns_vtable_slot (mono_domain_get (), vtable_slot_to_patch))) {
627                         g_assert (*vtable_slot_to_patch);
628                         *vtable_slot_to_patch = mono_get_addr_from_ftnptr (addr);
629                 }
630         } else {
631                 guint8 *plt_entry = mono_aot_get_plt_entry (code);
632                 gboolean no_patch = FALSE;
633                 MonoJitInfo *target_ji;
634
635                 if (plt_entry) {
636                         if (generic_shared) {
637                                 target_ji =
638                                         mini_jit_info_table_find (mono_domain_get (), mono_get_addr_from_ftnptr (compiled_method), NULL);
639                                 if (!ji)
640                                         ji = mini_jit_info_table_find (mono_domain_get (), (char*)code, NULL);
641
642                                 if (ji && target_ji && generic_shared && ji->has_generic_jit_info && !target_ji->has_generic_jit_info) {
643                                         no_patch = TRUE;
644                                 }
645                         }
646                         if (!no_patch)
647                                 mono_aot_patch_plt_entry (code, plt_entry, NULL, regs, addr);
648                 } else {
649                         if (generic_shared) {
650                                 if (m->wrapper_type != MONO_WRAPPER_NONE)
651                                         m = mono_marshal_method_from_wrapper (m);
652                                 //g_assert (mono_method_is_generic_sharable (m, FALSE));
653                         }
654
655                         /* Patch calling code */
656                         target_ji =
657                                 mini_jit_info_table_find (mono_domain_get (), mono_get_addr_from_ftnptr (compiled_method), NULL);
658                         if (!ji)
659                                 ji = mini_jit_info_table_find (mono_domain_get (), (char*)code, NULL);
660
661                         if (ji && target_ji && generic_shared && ji->has_generic_jit_info && !target_ji->has_generic_jit_info) {
662                                 /* 
663                                  * Can't patch the call as the caller is gshared, but the callee is not. Happens when
664                                  * generic sharing fails.
665                                  * FIXME: Performance problem.
666                                  */
667                                 no_patch = TRUE;
668                         }
669
670                         if (!no_patch && mono_method_same_domain (ji, target_ji))
671                                 mono_arch_patch_callsite (ji->code_start, code, addr);
672                 }
673         }
674
675         return addr;
676 }
677
678 static gpointer
679 common_call_trampoline (mgreg_t *regs, guint8 *code, MonoMethod *m, MonoVTable *vt, gpointer *vtable_slot)
680 {
681         gpointer res;
682         MONO_PREPARE_RESET_BLOCKING;
683         res = common_call_trampoline_inner (regs, code, m, vt, vtable_slot);
684         MONO_FINISH_RESET_BLOCKING;
685         return res;
686 }
687
688 /**
689  * mono_magic_trampoline:
690  *
691  * This trampoline handles normal calls from JITted code.
692  */
693 gpointer
694 mono_magic_trampoline (mgreg_t *regs, guint8 *code, gpointer arg, guint8* tramp)
695 {
696         trampoline_calls ++;
697
698         return common_call_trampoline (regs, code, arg, NULL, NULL);
699 }
700
701 /**
702  * mono_vcall_trampoline:
703  *
704  * This trampoline handles virtual calls.
705  */
706 static gpointer
707 mono_vcall_trampoline (mgreg_t *regs, guint8 *code, int slot, guint8 *tramp)
708 {
709         MonoObject *this_arg;
710         MonoVTable *vt;
711         gpointer *vtable_slot;
712         MonoMethod *m;
713         gpointer addr;
714
715         trampoline_calls ++;
716
717         /*
718          * We need to obtain the following pieces of information:
719          * - the method which needs to be compiled.
720          * - the vtable slot.
721          * We use one vtable trampoline per vtable slot index, so we need only the vtable,
722          * the other two can be computed from the vtable + the slot index.
723          */
724
725         /*
726          * Obtain the vtable from the 'this' arg.
727          */
728         this_arg = mono_arch_get_this_arg_from_call (regs, code);
729         g_assert (this_arg);
730
731         vt = this_arg->vtable;
732
733         if (slot >= 0) {
734                 /* Normal virtual call */
735                 vtable_slot = &(vt->vtable [slot]);
736
737                 /* Avoid loading metadata or creating a generic vtable if possible */
738                 addr = mono_aot_get_method_from_vt_slot (mono_domain_get (), vt, slot);
739                 if (addr && !vt->klass->valuetype) {
740                         if (mono_domain_owns_vtable_slot (mono_domain_get (), vtable_slot))
741                                 *vtable_slot = addr;
742
743                         return mono_create_ftnptr (mono_domain_get (), addr);
744                 }
745
746                 /*
747                  * Bug #616463 (see
748                  * is_generic_method_definition() above) also
749                  * goes away if we do a
750                  * mono_class_setup_vtable (vt->klass) here,
751                  * because we then inflate the method
752                  * correctly, put it in the cache, and the
753                  * "wrong" inflation invocation still looks up
754                  * the correctly inflated method.
755                  *
756                  * The hack above seems more stable and
757                  * trustworthy.
758                  */
759                 m = mono_class_get_vtable_entry (vt->klass, slot);
760         } else {
761                 /* IMT call */
762                 vtable_slot = &(((gpointer*)vt) [slot]);
763
764                 m = NULL;
765         }
766
767         return common_call_trampoline (regs, code, m, vt, vtable_slot);
768 }
769
770 #ifndef DISABLE_REMOTING
771 gpointer
772 mono_generic_virtual_remoting_trampoline (mgreg_t *regs, guint8 *code, MonoMethod *m, guint8 *tramp)
773 {
774         MonoError error;
775         MonoGenericContext context = { NULL, NULL };
776         MonoMethod *imt_method, *declaring;
777         gpointer addr;
778
779         trampoline_calls ++;
780
781         g_assert (m->is_generic);
782
783         if (m->is_inflated)
784                 declaring = mono_method_get_declaring_generic_method (m);
785         else
786                 declaring = m;
787
788         if (m->klass->generic_class)
789                 context.class_inst = m->klass->generic_class->context.class_inst;
790         else
791                 g_assert (!m->klass->generic_container);
792
793         imt_method = mono_arch_find_imt_method (regs, code);
794         if (imt_method->is_inflated)
795                 context.method_inst = ((MonoMethodInflated*)imt_method)->context.method_inst;
796         m = mono_class_inflate_generic_method_checked (declaring, &context, &error);
797         g_assert (mono_error_ok (&error)); /* FIXME don't swallow the error */;
798         m = mono_marshal_get_remoting_invoke_with_check (m);
799
800         addr = mono_compile_method (m);
801         g_assert (addr);
802
803         return addr;
804 }
805 #endif
806
807 /**
808  * mono_aot_trampoline:
809  *
810  * This trampoline handles calls made from AOT code. We try to bypass the 
811  * normal JIT compilation logic to avoid loading the metadata for the method.
812  */
813 #ifdef MONO_ARCH_AOT_SUPPORTED
814 gpointer
815 mono_aot_trampoline (mgreg_t *regs, guint8 *code, guint8 *token_info, 
816                                          guint8* tramp)
817 {
818         MonoImage *image;
819         guint32 token;
820         MonoMethod *method = NULL;
821         gpointer addr;
822         guint8 *plt_entry;
823
824         trampoline_calls ++;
825
826         image = *(gpointer*)(gpointer)token_info;
827         token_info += sizeof (gpointer);
828         token = *(guint32*)(gpointer)token_info;
829
830         addr = mono_aot_get_method_from_token (mono_domain_get (), image, token);
831         if (!addr) {
832                 method = mono_get_method (image, token, NULL);
833                 g_assert (method);
834
835                 /* Use the generic code */
836                 return mono_magic_trampoline (regs, code, method, tramp);
837         }
838
839         addr = mono_create_ftnptr (mono_domain_get (), addr);
840
841         /* This is a normal call through a PLT entry */
842         plt_entry = mono_aot_get_plt_entry (code);
843         g_assert (plt_entry);
844
845         mono_aot_patch_plt_entry (code, plt_entry, NULL, regs, addr);
846
847         return addr;
848 }
849
850 /*
851  * mono_aot_plt_trampoline:
852  *
853  *   This trampoline handles calls made from AOT code through the PLT table.
854  */
855 gpointer
856 mono_aot_plt_trampoline (mgreg_t *regs, guint8 *code, guint8 *aot_module, 
857                                                  guint8* tramp)
858 {
859         guint32 plt_info_offset = mono_aot_get_plt_info_offset (regs, code);
860         gpointer res;
861
862         trampoline_calls ++;
863
864         res = mono_aot_plt_resolve (aot_module, plt_info_offset, code);
865         if (!res) {
866                 if (mono_loader_get_last_error ())
867                         mono_raise_exception (mono_loader_error_prepare_exception (mono_loader_get_last_error ()));
868                 // FIXME: Error handling (how ?)
869                 g_assert (res);
870         }
871
872         return res;
873 }
874 #endif
875
876 static gpointer
877 mono_rgctx_lazy_fetch_trampoline (mgreg_t *regs, guint8 *code, gpointer data, guint8 *tramp)
878 {
879         static gboolean inited = FALSE;
880         static int num_lookups = 0;
881         guint32 slot = GPOINTER_TO_UINT (data);
882         mgreg_t *r = (mgreg_t*)regs;
883         gpointer arg = (gpointer)(gssize)r [MONO_ARCH_VTABLE_REG];
884         guint32 index = MONO_RGCTX_SLOT_INDEX (slot);
885         gboolean mrgctx = MONO_RGCTX_SLOT_IS_MRGCTX (slot);
886
887         trampoline_calls ++;
888
889         if (!inited) {
890                 mono_counters_register ("RGCTX unmanaged lookups", MONO_COUNTER_GENERICS | MONO_COUNTER_INT, &num_lookups);
891                 inited = TRUE;
892         }
893
894         num_lookups++;
895
896         if (mrgctx)
897                 return mono_method_fill_runtime_generic_context (arg, index);
898         else
899                 return mono_class_fill_runtime_generic_context (arg, index);
900 }
901
902 /*
903  * Precompute data to speed up mono_delegate_trampoline ().
904  * METHOD might be NULL.
905  */
906 static MonoDelegateTrampInfo*
907 create_delegate_trampoline_data (MonoDomain *domain, MonoClass *klass, MonoMethod *method)
908 {
909         MonoDelegateTrampInfo *tramp_data;
910         MonoMethod *invoke;
911         MonoError err;
912
913         // Precompute the delegate invoke impl and pass it to the delegate trampoline
914         invoke = mono_get_delegate_invoke (klass);
915         g_assert (invoke);
916
917         tramp_data = mono_domain_alloc0 (domain, sizeof (MonoDelegateTrampInfo));
918         tramp_data->invoke = invoke;
919         tramp_data->invoke_sig = mono_method_signature (invoke);
920         tramp_data->impl_this = mono_arch_get_delegate_invoke_impl (mono_method_signature (invoke), TRUE);
921         tramp_data->impl_nothis = mono_arch_get_delegate_invoke_impl (mono_method_signature (invoke), FALSE);
922         tramp_data->method = method;
923         if (method) {
924                 mono_error_init (&err);
925                 tramp_data->sig = mono_method_signature_checked (method, &err);
926                 tramp_data->need_rgctx_tramp = mono_method_needs_static_rgctx_invoke (method, FALSE);
927         }
928
929         return tramp_data;
930 }
931
932 /**
933  * mono_delegate_trampoline:
934  *
935  *   This trampoline handles calls made to Delegate:Invoke ().
936  * This is called once the first time a delegate is invoked, so it must be fast.
937  */
938 gpointer
939 mono_delegate_trampoline (mgreg_t *regs, guint8 *code, gpointer *arg, guint8* tramp)
940 {
941         MonoDomain *domain = mono_domain_get ();
942         MonoDelegate *delegate;
943         MonoJitInfo *ji;
944         MonoMethod *m;
945         MonoMethod *method = NULL;
946         gboolean multicast, callvirt = FALSE, closed_over_null = FALSE;
947         gboolean need_rgctx_tramp = FALSE;
948         gboolean need_unbox_tramp = FALSE;
949         gboolean enable_caching = TRUE;
950         MonoDelegateTrampInfo *tramp_info = (MonoDelegateTrampInfo*)arg;
951         MonoMethod *invoke = tramp_info->invoke;
952         guint8 *impl_this = tramp_info->impl_this;
953         guint8 *impl_nothis = tramp_info->impl_nothis;
954         MonoError err;
955         MonoMethodSignature *sig;
956         gpointer addr, compiled_method;
957         gboolean is_remote = FALSE;
958
959         trampoline_calls ++;
960
961         /* Obtain the delegate object according to the calling convention */
962         delegate = mono_arch_get_this_arg_from_call (regs, code);
963         g_assert (mono_class_has_parent (mono_object_class (delegate), mono_defaults.multicastdelegate_class));
964
965         if (delegate->method) {
966                 method = delegate->method;
967
968                 /*
969                  * delegate->method_ptr == NULL means the delegate was initialized by 
970                  * mini_delegate_ctor, while != NULL means it is initialized by 
971                  * mono_delegate_ctor_with_method (). In both cases, we need to add wrappers
972                  * (ctor_with_method () does this, but it doesn't store the wrapper back into
973                  * delegate->method).
974                  */
975 #ifndef DISABLE_REMOTING
976                 if (delegate->target && delegate->target->vtable->klass == mono_defaults.transparent_proxy_class) {
977                         is_remote = TRUE;
978 #ifndef DISABLE_COM
979                         if (((MonoTransparentProxy *)delegate->target)->remote_class->proxy_class != mono_class_get_com_object_class () &&
980                            !mono_class_is_com_object (((MonoTransparentProxy *)delegate->target)->remote_class->proxy_class))
981 #endif
982                                 method = mono_marshal_get_remoting_invoke (method);
983                 }
984 #endif
985                 if (!is_remote) {
986                         sig = tramp_info->sig;
987                         if (!(sig && method == tramp_info->method)) {
988                                 mono_error_init (&err);
989                                 sig = mono_method_signature_checked (method, &err);
990                                 if (!sig)
991                                         mono_error_raise_exception (&err);
992                         }
993
994                         if (sig->hasthis && method->klass->valuetype) {
995                                 gboolean need_unbox = TRUE;
996
997                                 if (tramp_info->invoke_sig->param_count > sig->param_count && tramp_info->invoke_sig->params [0]->byref)
998                                         need_unbox = FALSE;
999
1000                                 if (need_unbox) {
1001                                         if (mono_aot_only)
1002                                                 need_unbox_tramp = TRUE;
1003                                         else
1004                                                 method = mono_marshal_get_unbox_wrapper (method);
1005                                 }
1006                         }
1007                 }
1008         // If "delegate->method_ptr" is null mono_get_addr_from_ftnptr will fail if
1009         // ftnptrs are being used.  "method" would end up null on archtitectures without
1010         // ftnptrs so we can just skip this.
1011         } else if (delegate->method_ptr) {
1012                 ji = mono_jit_info_table_find (domain, mono_get_addr_from_ftnptr (delegate->method_ptr));
1013                 if (ji)
1014                         method = jinfo_get_method (ji);
1015         }
1016
1017         if (method) {
1018                 sig = tramp_info->sig;
1019                 if (!(sig && method == tramp_info->method)) {
1020                         mono_error_init (&err);
1021                         sig = mono_method_signature_checked (method, &err);
1022                         if (!sig)
1023                                 mono_error_raise_exception (&err);
1024                 }
1025
1026                 callvirt = !delegate->target && sig->hasthis;
1027                 if (callvirt)
1028                         closed_over_null = tramp_info->invoke_sig->param_count == sig->param_count;
1029
1030                 if (callvirt && !closed_over_null) {
1031                         /*
1032                          * The delegate needs to make a virtual call to the target method using its
1033                          * first argument as the receiver. This is hard to support in full-aot, so
1034                          * optimize it in some cases if possible.
1035                          * If the target method is not virtual or is in a sealed class,
1036                          * the vcall will call it directly.
1037                          * If the call doesn't return a valuetype, then the vcall uses the same calling
1038                          * convention as a normal call.
1039                          */
1040                         if (((method->klass->flags & TYPE_ATTRIBUTE_SEALED) || !(method->flags & METHOD_ATTRIBUTE_VIRTUAL)) && !MONO_TYPE_ISSTRUCT (sig->ret)) {
1041                                 callvirt = FALSE;
1042                                 enable_caching = FALSE;
1043                         }
1044                 }
1045
1046                 if (delegate->target && 
1047                         method->flags & METHOD_ATTRIBUTE_VIRTUAL && 
1048                         method->flags & METHOD_ATTRIBUTE_ABSTRACT &&
1049                         method->klass->flags & TYPE_ATTRIBUTE_ABSTRACT) {
1050                         method = mono_object_get_virtual_method (delegate->target, method);
1051                         enable_caching = FALSE;
1052                 }
1053
1054                 if (method->iflags & METHOD_IMPL_ATTRIBUTE_SYNCHRONIZED)
1055                         method = mono_marshal_get_synchronized_wrapper (method);
1056
1057                 if (method == tramp_info->method)
1058                         need_rgctx_tramp = tramp_info->need_rgctx_tramp;
1059                 else if (mono_method_needs_static_rgctx_invoke (method, FALSE))
1060                         need_rgctx_tramp = TRUE;
1061         }
1062
1063         /* 
1064          * If the called address is a trampoline, replace it with the compiled method so
1065          * further calls don't have to go through the trampoline.
1066          */
1067         if (method && !callvirt) {
1068                 /* Avoid the overhead of looking up an already compiled method if possible */
1069                 if (enable_caching && delegate->method_code && *delegate->method_code) {
1070                         delegate->method_ptr = *delegate->method_code;
1071                 } else {
1072                         compiled_method = addr = mono_compile_method (method);
1073                         addr = mini_add_method_trampoline (method, compiled_method, need_rgctx_tramp, need_unbox_tramp);
1074                         delegate->method_ptr = addr;
1075                         if (enable_caching && delegate->method_code)
1076                                 *delegate->method_code = delegate->method_ptr;
1077                 }
1078         } else {
1079                 if (need_rgctx_tramp)
1080                         delegate->method_ptr = mono_create_static_rgctx_trampoline (method, delegate->method_ptr);
1081         }
1082
1083         /* Necessary for !code condition to fallback to slow path */
1084         code = NULL;
1085
1086         multicast = ((MonoMulticastDelegate*)delegate)->delegates != NULL;
1087         if (!multicast && !callvirt) {
1088                 if (method && (method->flags & METHOD_ATTRIBUTE_STATIC) && mono_method_signature (method)->param_count == mono_method_signature (invoke)->param_count + 1)
1089                         /* Closed static delegate */
1090                         code = impl_this;
1091                 else
1092                         code = delegate->target ? impl_this : impl_nothis;
1093         }
1094
1095         if (!code) {
1096                 /* The general, unoptimized case */
1097                 m = mono_marshal_get_delegate_invoke (invoke, delegate);
1098                 code = mono_compile_method (m);
1099                 code = mini_add_method_trampoline (m, code, mono_method_needs_static_rgctx_invoke (m, FALSE), FALSE);
1100         }
1101
1102         delegate->invoke_impl = mono_get_addr_from_ftnptr (code);
1103         if (enable_caching && !callvirt && tramp_info->method) {
1104                 tramp_info->method_ptr = delegate->method_ptr;
1105                 tramp_info->invoke_impl = delegate->invoke_impl;
1106         }
1107
1108         return code;
1109 }
1110
1111 #ifdef MONO_ARCH_HAVE_HANDLER_BLOCK_GUARD
1112 static gpointer
1113 mono_handler_block_guard_trampoline (mgreg_t *regs, guint8 *code, gpointer *tramp_info, guint8* tramp)
1114 {
1115         MonoContext ctx;
1116         MonoException *exc;
1117         MonoJitTlsData *jit_tls = mono_native_tls_get_value (mono_jit_tls_id);
1118         gpointer resume_ip = jit_tls->handler_block_return_address;
1119
1120         memcpy (&ctx, &jit_tls->handler_block_context, sizeof (MonoContext));
1121         MONO_CONTEXT_SET_IP (&ctx, jit_tls->handler_block_return_address);
1122
1123         jit_tls->handler_block_return_address = NULL;
1124         jit_tls->handler_block = NULL;
1125
1126         if (!resume_ip) /*this should not happen, but we should avoid crashing */
1127                 exc = mono_get_exception_execution_engine ("Invalid internal state, resuming abort after handler block but no resume ip found");
1128         else
1129                 exc = mono_thread_resume_interruption ();
1130
1131         if (exc) {
1132                 mono_handle_exception (&ctx, exc);
1133                 mono_restore_context (&ctx);
1134         }
1135
1136         return resume_ip;
1137 }
1138
1139 gpointer
1140 mono_create_handler_block_trampoline (void)
1141 {
1142         static gpointer code;
1143         if (code) {
1144                 mono_memory_barrier ();
1145                 return code;
1146         }
1147
1148         g_assert (!mono_aot_only);
1149
1150         mono_trampolines_lock ();
1151
1152         if (!code) {
1153                 MonoTrampInfo *info;
1154                 gpointer tmp;
1155
1156                 tmp = mono_arch_create_handler_block_trampoline (&info, FALSE);
1157                 mono_tramp_info_register (info, NULL);
1158                 mono_memory_barrier ();
1159                 code = tmp;
1160         }
1161         mono_trampolines_unlock ();
1162
1163         return code;
1164 }
1165 #endif
1166
1167 /*
1168  * mono_get_trampoline_func:
1169  *
1170  *   Return the C function which needs to be called by the generic trampoline of type
1171  * TRAMP_TYPE.
1172  */
1173 gconstpointer
1174 mono_get_trampoline_func (MonoTrampolineType tramp_type)
1175 {
1176         switch (tramp_type) {
1177         case MONO_TRAMPOLINE_JIT:
1178         case MONO_TRAMPOLINE_JUMP:
1179                 return mono_magic_trampoline;
1180         case MONO_TRAMPOLINE_RGCTX_LAZY_FETCH:
1181                 return mono_rgctx_lazy_fetch_trampoline;
1182 #ifdef MONO_ARCH_AOT_SUPPORTED
1183         case MONO_TRAMPOLINE_AOT:
1184                 return mono_aot_trampoline;
1185         case MONO_TRAMPOLINE_AOT_PLT:
1186                 return mono_aot_plt_trampoline;
1187 #endif
1188         case MONO_TRAMPOLINE_DELEGATE:
1189                 return mono_delegate_trampoline;
1190         case MONO_TRAMPOLINE_RESTORE_STACK_PROT:
1191                 return mono_altstack_restore_prot;
1192 #ifndef DISABLE_REMOTING
1193         case MONO_TRAMPOLINE_GENERIC_VIRTUAL_REMOTING:
1194                 return mono_generic_virtual_remoting_trampoline;
1195 #endif
1196         case MONO_TRAMPOLINE_VCALL:
1197                 return mono_vcall_trampoline;
1198 #ifdef MONO_ARCH_HAVE_HANDLER_BLOCK_GUARD
1199         case MONO_TRAMPOLINE_HANDLER_BLOCK_GUARD:
1200                 return mono_handler_block_guard_trampoline;
1201 #endif
1202         default:
1203                 g_assert_not_reached ();
1204                 return NULL;
1205         }
1206 }
1207
1208 static guchar*
1209 create_trampoline_code (MonoTrampolineType tramp_type)
1210 {
1211         MonoTrampInfo *info;
1212         guchar *code;
1213
1214         code = mono_arch_create_generic_trampoline (tramp_type, &info, FALSE);
1215         mono_tramp_info_register (info, NULL);
1216
1217         return code;
1218 }
1219
1220 void
1221 mono_trampolines_init (void)
1222 {
1223         mono_os_mutex_init_recursive (&trampolines_mutex);
1224
1225         if (mono_aot_only)
1226                 return;
1227
1228         mono_trampoline_code [MONO_TRAMPOLINE_JIT] = create_trampoline_code (MONO_TRAMPOLINE_JIT);
1229         mono_trampoline_code [MONO_TRAMPOLINE_JUMP] = create_trampoline_code (MONO_TRAMPOLINE_JUMP);
1230         mono_trampoline_code [MONO_TRAMPOLINE_RGCTX_LAZY_FETCH] = create_trampoline_code (MONO_TRAMPOLINE_RGCTX_LAZY_FETCH);
1231 #ifdef MONO_ARCH_AOT_SUPPORTED
1232         mono_trampoline_code [MONO_TRAMPOLINE_AOT] = create_trampoline_code (MONO_TRAMPOLINE_AOT);
1233         mono_trampoline_code [MONO_TRAMPOLINE_AOT_PLT] = create_trampoline_code (MONO_TRAMPOLINE_AOT_PLT);
1234 #endif
1235         mono_trampoline_code [MONO_TRAMPOLINE_DELEGATE] = create_trampoline_code (MONO_TRAMPOLINE_DELEGATE);
1236         mono_trampoline_code [MONO_TRAMPOLINE_RESTORE_STACK_PROT] = create_trampoline_code (MONO_TRAMPOLINE_RESTORE_STACK_PROT);
1237 #ifndef DISABLE_REMOTING
1238         mono_trampoline_code [MONO_TRAMPOLINE_GENERIC_VIRTUAL_REMOTING] = create_trampoline_code (MONO_TRAMPOLINE_GENERIC_VIRTUAL_REMOTING);
1239 #endif
1240         mono_trampoline_code [MONO_TRAMPOLINE_VCALL] = create_trampoline_code (MONO_TRAMPOLINE_VCALL);
1241 #ifdef MONO_ARCH_HAVE_HANDLER_BLOCK_GUARD
1242         mono_trampoline_code [MONO_TRAMPOLINE_HANDLER_BLOCK_GUARD] = create_trampoline_code (MONO_TRAMPOLINE_HANDLER_BLOCK_GUARD);
1243         mono_create_handler_block_trampoline ();
1244 #endif
1245
1246         mono_counters_register ("Calls to trampolines", MONO_COUNTER_JIT | MONO_COUNTER_INT, &trampoline_calls);
1247         mono_counters_register ("JIT trampolines", MONO_COUNTER_JIT | MONO_COUNTER_INT, &jit_trampolines);
1248         mono_counters_register ("Unbox trampolines", MONO_COUNTER_JIT | MONO_COUNTER_INT, &unbox_trampolines);
1249         mono_counters_register ("Static rgctx trampolines", MONO_COUNTER_JIT | MONO_COUNTER_INT, &static_rgctx_trampolines);
1250 }
1251
1252 void
1253 mono_trampolines_cleanup (void)
1254 {
1255         if (rgctx_lazy_fetch_trampoline_hash)
1256                 g_hash_table_destroy (rgctx_lazy_fetch_trampoline_hash);
1257         if (rgctx_lazy_fetch_trampoline_hash_addr)
1258                 g_hash_table_destroy (rgctx_lazy_fetch_trampoline_hash_addr);
1259
1260         mono_os_mutex_destroy (&trampolines_mutex);
1261 }
1262
1263 guint8 *
1264 mono_get_trampoline_code (MonoTrampolineType tramp_type)
1265 {
1266         g_assert (mono_trampoline_code [tramp_type]);
1267
1268         return mono_trampoline_code [tramp_type];
1269 }
1270
1271 gpointer
1272 mono_create_specific_trampoline (gpointer arg1, MonoTrampolineType tramp_type, MonoDomain *domain, guint32 *code_len)
1273 {
1274         if (mono_aot_only)
1275                 return mono_aot_create_specific_trampoline (mono_defaults.corlib, arg1, tramp_type, domain, code_len);
1276         else
1277                 return mono_arch_create_specific_trampoline (arg1, tramp_type, domain, code_len);
1278 }
1279
1280 gpointer
1281 mono_create_jump_trampoline (MonoDomain *domain, MonoMethod *method, gboolean add_sync_wrapper)
1282 {
1283         MonoJitInfo *ji;
1284         gpointer code;
1285         guint32 code_size = 0;
1286
1287         code = mono_jit_find_compiled_method_with_jit_info (domain, method, &ji);
1288         /*
1289          * We cannot recover the correct type of a shared generic
1290          * method from its native code address, so we use the
1291          * trampoline instead.
1292          * For synchronized methods, the trampoline adds the wrapper.
1293          */
1294         if (code && !ji->has_generic_jit_info && !(method->iflags & METHOD_IMPL_ATTRIBUTE_SYNCHRONIZED))
1295                 return code;
1296
1297         if (mono_llvm_only)
1298                 return mono_jit_compile_method (method);
1299
1300         mono_domain_lock (domain);
1301         code = g_hash_table_lookup (domain_jit_info (domain)->jump_trampoline_hash, method);
1302         mono_domain_unlock (domain);
1303         if (code)
1304                 return code;
1305
1306         code = mono_create_specific_trampoline (method, MONO_TRAMPOLINE_JUMP, mono_domain_get (), &code_size);
1307         g_assert (code_size);
1308
1309         ji = mono_domain_alloc0 (domain, MONO_SIZEOF_JIT_INFO);
1310         ji->code_start = code;
1311         ji->code_size = code_size;
1312         ji->d.method = method;
1313
1314         /*
1315          * mono_delegate_ctor needs to find the method metadata from the 
1316          * trampoline address, so we save it here.
1317          */
1318
1319         mono_jit_info_table_add (domain, ji);
1320
1321         mono_domain_lock (domain);
1322         g_hash_table_insert (domain_jit_info (domain)->jump_trampoline_hash, method, ji->code_start);
1323         mono_domain_unlock (domain);
1324
1325         return ji->code_start;
1326 }
1327
1328 static void
1329 method_not_found (void)
1330 {
1331         g_assert_not_reached ();
1332 }
1333
1334 gpointer
1335 mono_create_jit_trampoline_in_domain (MonoDomain *domain, MonoMethod *method)
1336 {
1337         gpointer tramp;
1338
1339         if (mono_aot_only) {
1340                 /* Avoid creating trampolines if possible */
1341                 gpointer code = mono_jit_find_compiled_method (domain, method);
1342                 
1343                 if (code)
1344                         return code;
1345                 if (mono_llvm_only) {
1346                         if (method->wrapper_type == MONO_WRAPPER_PROXY_ISINST || method->wrapper_type == MONO_WRAPPER_LDFLD_REMOTE ||
1347                                 method->wrapper_type == MONO_WRAPPER_STFLD_REMOTE)
1348                                 /* These wrappers are not generated */
1349                                 return method_not_found;
1350                         /* Methods are lazily initialized on first call, so this can't lead recursion */
1351                         return mono_compile_method (method);
1352                 }
1353         }
1354
1355         mono_domain_lock (domain);
1356         tramp = g_hash_table_lookup (domain_jit_info (domain)->jit_trampoline_hash, method);
1357         mono_domain_unlock (domain);
1358         if (tramp)
1359                 return tramp;
1360
1361         tramp = mono_create_specific_trampoline (method, MONO_TRAMPOLINE_JIT, domain, NULL);
1362         
1363         mono_domain_lock (domain);
1364         g_hash_table_insert (domain_jit_info (domain)->jit_trampoline_hash, method, tramp);
1365         mono_domain_unlock (domain);
1366
1367         jit_trampolines++;
1368
1369         return tramp;
1370 }       
1371
1372 gpointer
1373 mono_create_jit_trampoline (MonoMethod *method)
1374 {
1375         return mono_create_jit_trampoline_in_domain (mono_domain_get (), method);
1376 }
1377
1378 gpointer
1379 mono_create_jit_trampoline_from_token (MonoImage *image, guint32 token)
1380 {
1381         gpointer tramp;
1382
1383         MonoDomain *domain = mono_domain_get ();
1384         guint8 *buf, *start;
1385
1386         buf = start = mono_domain_alloc0 (domain, 2 * sizeof (gpointer));
1387
1388         *(gpointer*)(gpointer)buf = image;
1389         buf += sizeof (gpointer);
1390         *(guint32*)(gpointer)buf = token;
1391
1392         tramp = mono_create_specific_trampoline (start, MONO_TRAMPOLINE_AOT, domain, NULL);
1393
1394         jit_trampolines++;
1395
1396         return tramp;
1397 }       
1398
1399
1400 /*
1401  * mono_create_delegate_trampoline_info:
1402  *
1403  *   Create a delegate trampoline for the KLASS+METHOD pair.
1404  */
1405 MonoDelegateTrampInfo*
1406 mono_create_delegate_trampoline_info (MonoDomain *domain, MonoClass *klass, MonoMethod *method)
1407 {
1408         MonoDelegateTrampInfo *tramp_info;
1409         MonoClassMethodPair pair, *dpair;
1410         guint32 code_size = 0;
1411
1412         pair.klass = klass;
1413         pair.method = method;
1414         mono_domain_lock (domain);
1415         tramp_info = g_hash_table_lookup (domain_jit_info (domain)->delegate_trampoline_hash, &pair);
1416         mono_domain_unlock (domain);
1417         if (tramp_info)
1418                 return tramp_info;
1419
1420         tramp_info = create_delegate_trampoline_data (domain, klass, method);
1421
1422         tramp_info->invoke_impl = mono_create_specific_trampoline (tramp_info, MONO_TRAMPOLINE_DELEGATE, domain, &code_size);
1423         g_assert (code_size);
1424
1425         dpair = mono_domain_alloc0 (domain, sizeof (MonoClassMethodPair));
1426         memcpy (dpair, &pair, sizeof (MonoClassMethodPair));
1427
1428         /* store trampoline address */
1429         mono_domain_lock (domain);
1430         g_hash_table_insert (domain_jit_info (domain)->delegate_trampoline_hash, dpair, tramp_info);
1431         mono_domain_unlock (domain);
1432
1433         return tramp_info;
1434 }
1435
1436 static void
1437 no_delegate_trampoline (void)
1438 {
1439         g_assert_not_reached ();
1440 }
1441
1442 gpointer
1443 mono_create_delegate_trampoline (MonoDomain *domain, MonoClass *klass)
1444 {
1445         if (mono_llvm_only)
1446                 return no_delegate_trampoline;
1447
1448         return mono_create_delegate_trampoline_info (domain, klass, NULL)->invoke_impl;
1449 }
1450
1451 gpointer
1452 mono_create_delegate_virtual_trampoline (MonoDomain *domain, MonoClass *klass, MonoMethod *method)
1453 {
1454         MonoMethod *invoke = mono_get_delegate_invoke (klass);
1455         g_assert (invoke);
1456
1457         return mono_get_delegate_virtual_invoke_impl (mono_method_signature (invoke), method);
1458 }
1459
1460 gpointer
1461 mono_create_rgctx_lazy_fetch_trampoline (guint32 offset)
1462 {
1463         static gboolean inited = FALSE;
1464         static int num_trampolines = 0;
1465         MonoTrampInfo *info;
1466
1467         gpointer tramp, ptr;
1468
1469         mono_trampolines_lock ();
1470         if (rgctx_lazy_fetch_trampoline_hash)
1471                 tramp = g_hash_table_lookup (rgctx_lazy_fetch_trampoline_hash, GUINT_TO_POINTER (offset));
1472         else
1473                 tramp = NULL;
1474         mono_trampolines_unlock ();
1475         if (tramp)
1476                 return tramp;
1477
1478         if (mono_aot_only) {
1479                 ptr = mono_aot_get_lazy_fetch_trampoline (offset);
1480         } else {
1481                 tramp = mono_arch_create_rgctx_lazy_fetch_trampoline (offset, &info, FALSE);
1482                 mono_tramp_info_register (info, NULL);
1483                 ptr = mono_create_ftnptr (mono_get_root_domain (), tramp);
1484         }
1485
1486         mono_trampolines_lock ();
1487         if (!rgctx_lazy_fetch_trampoline_hash) {
1488                 rgctx_lazy_fetch_trampoline_hash = g_hash_table_new (NULL, NULL);
1489                 rgctx_lazy_fetch_trampoline_hash_addr = g_hash_table_new (NULL, NULL);
1490         }
1491         g_hash_table_insert (rgctx_lazy_fetch_trampoline_hash, GUINT_TO_POINTER (offset), ptr);
1492         g_assert (offset != -1);
1493         g_hash_table_insert (rgctx_lazy_fetch_trampoline_hash_addr, ptr, GUINT_TO_POINTER (offset + 1));
1494         mono_trampolines_unlock ();
1495
1496         if (!inited) {
1497                 mono_counters_register ("RGCTX num lazy fetch trampolines",
1498                                 MONO_COUNTER_GENERICS | MONO_COUNTER_INT, &num_trampolines);
1499                 inited = TRUE;
1500         }
1501         num_trampolines++;
1502
1503         return ptr;
1504 }
1505
1506 guint32
1507 mono_find_rgctx_lazy_fetch_trampoline_by_addr (gconstpointer addr)
1508 {
1509         int offset;
1510
1511         mono_trampolines_lock ();
1512         if (rgctx_lazy_fetch_trampoline_hash_addr) {
1513                 /* We store the real offset + 1 so we can detect when the lookup fails */
1514                 offset = GPOINTER_TO_INT (g_hash_table_lookup (rgctx_lazy_fetch_trampoline_hash_addr, addr));
1515                 if (offset)
1516                         offset -= 1;
1517                 else
1518                         offset = -1;
1519         } else {
1520                 offset = -1;
1521         }
1522         mono_trampolines_unlock ();
1523         return offset;
1524 }
1525
1526 static const char*tramp_names [MONO_TRAMPOLINE_NUM] = {
1527         "jit",
1528         "jump",
1529         "rgctx_lazy_fetch",
1530         "aot",
1531         "aot_plt",
1532         "delegate",
1533         "restore_stack_prot",
1534         "generic_virtual_remoting",
1535         "vcall",
1536         "handler_block_guard"
1537 };
1538
1539 /*
1540  * mono_get_generic_trampoline_simple_name:
1541  *
1542  */
1543 const char*
1544 mono_get_generic_trampoline_simple_name (MonoTrampolineType tramp_type)
1545 {
1546         return tramp_names [tramp_type];
1547 }
1548
1549 /*
1550  * mono_get_generic_trampoline_name:
1551  *
1552  *   Returns a pointer to malloc-ed memory.
1553  */
1554 char*
1555 mono_get_generic_trampoline_name (MonoTrampolineType tramp_type)
1556 {
1557         return g_strdup_printf ("generic_trampoline_%s", tramp_names [tramp_type]);
1558 }
1559
1560 /*
1561  * mono_get_rgctx_fetch_trampoline_name:
1562  *
1563  *   Returns a pointer to malloc-ed memory.
1564  */
1565 char*
1566 mono_get_rgctx_fetch_trampoline_name (int slot)
1567 {
1568         gboolean mrgctx;
1569         int index;
1570
1571         mrgctx = MONO_RGCTX_SLOT_IS_MRGCTX (slot);
1572         index = MONO_RGCTX_SLOT_INDEX (slot);
1573
1574         return g_strdup_printf ("rgctx_fetch_trampoline_%s_%d", mrgctx ? "mrgctx" : "rgctx", index);
1575 }
1576
1577 /*
1578  * mini_get_single_step_trampoline:
1579  *
1580  *   Return a trampoline which calls debugger_agent_single_step_from_context ().
1581  */
1582 gpointer
1583 mini_get_single_step_trampoline (void)
1584 {
1585         static gpointer trampoline;
1586
1587         if (!trampoline) {
1588                 gpointer tramp;
1589
1590                 if (mono_aot_only) {
1591                         tramp = mono_aot_get_trampoline ("sdb_single_step_trampoline");
1592                 } else {
1593 #ifdef MONO_ARCH_HAVE_SDB_TRAMPOLINES
1594                         MonoTrampInfo *info;
1595                         tramp = mono_arch_create_sdb_trampoline (TRUE, &info, FALSE);
1596                         mono_tramp_info_register (info, NULL);
1597 #else
1598                         tramp = NULL;
1599                         g_assert_not_reached ();
1600 #endif
1601                 }
1602                 mono_memory_barrier ();
1603                 trampoline = tramp;
1604         }
1605
1606         return trampoline;
1607 }
1608
1609 /*
1610  * mini_get_breakpoint_trampoline:
1611  *
1612  *   Return a trampoline which calls debugger_agent_breakpoint_from_context ().
1613  */
1614 gpointer
1615 mini_get_breakpoint_trampoline (void)
1616 {
1617         static gpointer trampoline;
1618
1619         if (!trampoline) {
1620                 gpointer tramp;
1621
1622                 if (mono_aot_only) {
1623                         tramp = mono_aot_get_trampoline ("sdb_breakpoint_trampoline");
1624                 } else {
1625 #ifdef MONO_ARCH_HAVE_SDB_TRAMPOLINES
1626                         MonoTrampInfo *info;
1627                         tramp = mono_arch_create_sdb_trampoline (FALSE, &info, FALSE);
1628                         mono_tramp_info_register (info, NULL);
1629 #else
1630                         tramp = NULL;
1631                         g_assert_not_reached ();
1632 #endif
1633                 }
1634                 mono_memory_barrier ();
1635                 trampoline = tramp;
1636         }
1637
1638         return trampoline;
1639 }