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