in mono/mono/mini:
[mono.git] / mono / metadata / domain.c
index 1c5002bb92402d11779dbad29702d00d86c67d84..9eeb20e02d0f1a137845695d5e71bd9bd478cd5f 100644 (file)
@@ -17,6 +17,7 @@
 
 #include <mono/utils/mono-compiler.h>
 #include <mono/utils/mono-logger.h>
+#include <mono/utils/mono-membar.h>
 #include <mono/metadata/object.h>
 #include <mono/metadata/object-internals.h>
 #include <mono/metadata/domain-internals.h>
@@ -95,7 +96,7 @@ static MonoJitInfoFindInAot jit_info_find_in_aot_func = NULL;
 /*
  * Contains information about AOT loaded code.
  */
-static MonoJitInfoTable *aot_modules = NULL;
+static MonoAotModuleInfoTable *aot_modules = NULL;
 
 /* This is the list of runtime versions supported by this JIT.
  */
@@ -103,7 +104,8 @@ static const MonoRuntimeInfo supported_runtimes[] = {
        {"v1.0.3705", "1.0", { {1,0,5000,0}, {7,0,5000,0} }     },
        {"v1.1.4322", "1.0", { {1,0,5000,0}, {7,0,5000,0} }     },
        {"v2.0.50215","2.0", { {2,0,0,0},    {8,0,0,0} }        },
-       {"v2.0.50727","2.0", { {2,0,0,0},    {8,0,0,0} }        }
+       {"v2.0.50727","2.0", { {2,0,0,0},    {8,0,0,0} }        },
+       {"moonlight", "2.1", { {2,1,0,0},    {9,0,0,0} }    },
 };
 
 
@@ -135,64 +137,245 @@ mono_domain_get_tls_offset (void)
        return offset;
 }
 
+#define JIT_INFO_TABLE_FILL_RATIO_NOM          3
+#define JIT_INFO_TABLE_FILL_RATIO_DENOM                4
+#define JIT_INFO_TABLE_FILLED_NUM_ELEMENTS     (MONO_JIT_INFO_TABLE_CHUNK_SIZE * JIT_INFO_TABLE_FILL_RATIO_NOM / JIT_INFO_TABLE_FILL_RATIO_DENOM)
+
+#define JIT_INFO_TABLE_LOW_WATERMARK(n)                ((n) / 2)
+#define JIT_INFO_TABLE_HIGH_WATERMARK(n)       ((n) * 5 / 6)
+
+#define IS_JIT_INFO_TOMBSTONE(ji)      ((ji)->method == NULL)
+
+#define JIT_INFO_TABLE_HAZARD_INDEX            0
+#define JIT_INFO_HAZARD_INDEX                  1
+
+static int
+jit_info_table_num_elements (MonoJitInfoTable *table)
+{
+       int i;
+       int num_elements = 0;
+
+       for (i = 0; i < table->num_chunks; ++i) {
+               MonoJitInfoTableChunk *chunk = table->chunks [i];
+               int chunk_num_elements = chunk->num_elements;
+               int j;
+
+               for (j = 0; j < chunk_num_elements; ++j) {
+                       if (!IS_JIT_INFO_TOMBSTONE (chunk->data [j]))
+                               ++num_elements;
+               }
+       }
+
+       return num_elements;
+}
+
+static MonoJitInfoTableChunk*
+jit_info_table_new_chunk (void)
+{
+       MonoJitInfoTableChunk *chunk = g_new0 (MonoJitInfoTableChunk, 1);
+       chunk->refcount = 1;
+
+       return chunk;
+}
+
 static MonoJitInfoTable *
-mono_jit_info_table_new (void)
+jit_info_table_new (void)
 {
-       return g_array_new (FALSE, FALSE, sizeof (gpointer));
+       MonoJitInfoTable *table = g_malloc0 (sizeof (MonoJitInfoTable) + sizeof (MonoJitInfoTableChunk*));
+
+       table->num_chunks = 1;
+       table->chunks [0] = jit_info_table_new_chunk ();
+
+       return table;
 }
 
 static void
-mono_jit_info_table_free (MonoJitInfoTable *table)
+jit_info_table_free (MonoJitInfoTable *table)
 {
-       g_array_free (table, TRUE);
+       int i;
+       int num_chunks = table->num_chunks;
+
+       /* At this point we assume that there are no other threads
+          still accessing the table, so we don't have to worry about
+          hazardous pointers. */
+
+       for (i = 0; i < num_chunks; ++i) {
+               MonoJitInfoTableChunk *chunk = table->chunks [i];
+               int num_elements;
+               int j;
+
+               if (--chunk->refcount > 0)
+                       continue;
+
+               num_elements = chunk->num_elements;
+               for (j = 0; j < num_elements; ++j) {
+                       MonoJitInfo *ji = chunk->data [j];
+
+                       if (IS_JIT_INFO_TOMBSTONE (ji))
+                               g_free (ji);
+               }
+
+               g_free (chunk);
+       }
+
+       g_free (table);
 }
 
+/* Can be called with hp==NULL, in which case it acts as an ordinary
+   pointer fetch.  It's used that way indirectly from
+   mono_jit_info_table_add(), which doesn't have to care about hazards
+   because it holds the respective domain lock. */
+static gpointer
+get_hazardous_pointer (gpointer volatile *pp, MonoThreadHazardPointers *hp, int hazard_index)
+{
+       gpointer p;
+
+       for (;;) {
+               /* Get the pointer */
+               p = *pp;
+               /* If we don't have hazard pointers just return the
+                  pointer. */
+               if (!hp)
+                       return p;
+               /* Make it hazardous */
+               mono_hazard_pointer_set (hp, hazard_index, p);
+               /* Check that it's still the same.  If not, try
+                  again. */
+               if (*pp != p) {
+                       mono_hazard_pointer_clear (hp, hazard_index);
+                       continue;
+               }
+               break;
+       }
+
+       return p;
+}
+
+/* The jit_info_table is sorted in ascending order by the end
+ * addresses of the compiled methods.  The reason why we have to do
+ * this is that once we introduce tombstones, it becomes possible for
+ * code ranges to overlap, and if we sort by code start and insert at
+ * the back of the table, we cannot guarantee that we won't overlook
+ * an entry.
+ *
+ * There are actually two possible ways to do the sorting and
+ * inserting which work with our lock-free mechanism:
+ *
+ * 1. Sort by start address and insert at the front.  When looking for
+ * an entry, find the last one with a start address lower than the one
+ * you're looking for, then work your way to the front of the table.
+ *
+ * 2. Sort by end address and insert at the back.  When looking for an
+ * entry, find the first one with an end address higher than the one
+ * you're looking for, then work your way to the end of the table.
+ *
+ * We chose the latter out of convenience.
+ */
 static int
-mono_jit_info_table_index (MonoJitInfoTable *table, char *addr)
+jit_info_table_index (MonoJitInfoTable *table, gint8 *addr)
 {
-       int left = 0, right = table->len;
+       int left = 0, right = table->num_chunks;
 
-       while (left < right) {
+       g_assert (left < right);
+
+       do {
                int pos = (left + right) / 2;
-               MonoJitInfo *ji = g_array_index (table, gpointer, pos);
-               char *start = ji->code_start;
-               char *end = start + ji->code_size;
+               MonoJitInfoTableChunk *chunk = table->chunks [pos];
 
-               if (addr < start)
+               if (addr < chunk->last_code_end)
                        right = pos;
-               else if (addr >= end) 
+               else
                        left = pos + 1;
+       } while (left < right);
+       g_assert (left == right);
+
+       if (left >= table->num_chunks)
+               return table->num_chunks - 1;
+       return left;
+}
+
+static int
+jit_info_table_chunk_index (MonoJitInfoTableChunk *chunk, MonoThreadHazardPointers *hp, gint8 *addr)
+{
+       int left = 0, right = chunk->num_elements;
+
+       while (left < right) {
+               int pos = (left + right) / 2;
+               MonoJitInfo *ji = get_hazardous_pointer((gpointer volatile*)&chunk->data [pos], hp, JIT_INFO_HAZARD_INDEX);
+               gint8 *code_end = (gint8*)ji->code_start + ji->code_size;
+
+               if (addr < code_end)
+                       right = pos;
                else
-                       return pos;
+                       left = pos + 1;
        }
+       g_assert (left == right);
 
        return left;
 }
 
-MonoJitInfo *
+MonoJitInfo*
 mono_jit_info_table_find (MonoDomain *domain, char *addr)
 {
-       MonoJitInfoTable *table = domain->jit_info_table;
+       MonoJitInfoTable *table;
        MonoJitInfo *ji;
-       guint left = 0, right;
+       int chunk_pos, pos;
+       MonoThreadHazardPointers *hp = mono_hazard_pointer_get ();
+
+       ++mono_stats.jit_info_table_lookup_count;
+
+       /* First we have to get the domain's jit_info_table.  This is
+          complicated by the fact that a writer might substitute a
+          new table and free the old one.  What the writer guarantees
+          us is that it looks at the hazard pointers after it has
+          changed the jit_info_table pointer.  So, if we guard the
+          table by a hazard pointer and make sure that the pointer is
+          still there after we've made it hazardous, we don't have to
+          worry about the writer freeing the table. */
+       table = get_hazardous_pointer ((gpointer volatile*)&domain->jit_info_table, hp, JIT_INFO_TABLE_HAZARD_INDEX);
+
+       chunk_pos = jit_info_table_index (table, (gint8*)addr);
+       g_assert (chunk_pos < table->num_chunks);
+
+       pos = jit_info_table_chunk_index (table->chunks [chunk_pos], hp, (gint8*)addr);
+
+       /* We now have a position that's very close to that of the
+          first element whose end address is higher than the one
+          we're looking for.  If we don't have the exact position,
+          then we have a position below that one, so we'll just
+          search upward until we find our element. */
+       do {
+               MonoJitInfoTableChunk *chunk = table->chunks [chunk_pos];
+
+               while (pos < chunk->num_elements) {
+                       ji = get_hazardous_pointer ((gpointer volatile*)&chunk->data [pos], hp, JIT_INFO_HAZARD_INDEX);
+
+                       ++pos;
+
+                       if (IS_JIT_INFO_TOMBSTONE (ji)) {
+                               mono_hazard_pointer_clear (hp, JIT_INFO_HAZARD_INDEX);
+                               continue;
+                       }
+                       if ((gint8*)addr >= (gint8*)ji->code_start
+                                       && (gint8*)addr < (gint8*)ji->code_start + ji->code_size) {
+                               mono_hazard_pointer_clear (hp, JIT_INFO_TABLE_HAZARD_INDEX);
+                               mono_hazard_pointer_clear (hp, JIT_INFO_HAZARD_INDEX);
+                               return ji;
+                       }
 
-       mono_domain_lock (domain);
+                       /* If we find a non-tombstone element which is already
+                          beyond what we're looking for, we have to end the
+                          search. */
+                       if ((gint8*)addr < (gint8*)ji->code_start)
+                               break;
+               }
 
-       right = table->len;
-       while (left < right) {
-               guint pos = (left + right) / 2;
-               ji = g_array_index (table, gpointer, pos);
+               ++chunk_pos;
+               pos = 0;
+        } while (chunk_pos < table->num_chunks);
 
-               if (addr < (char*)ji->code_start)
-                       right = pos;
-               else if (addr >= (char*)ji->code_start + ji->code_size) 
-                       left = pos + 1;
-               else {
-                       mono_domain_unlock (domain);
-                       return ji;
-               }
-       }
-       mono_domain_unlock (domain);
+       mono_hazard_pointer_clear (hp, JIT_INFO_TABLE_HAZARD_INDEX);
+       mono_hazard_pointer_clear (hp, JIT_INFO_HAZARD_INDEX);
 
        /* maybe it is shared code, so we also search in the root domain */
        ji = NULL;
@@ -209,41 +392,394 @@ mono_jit_info_table_find (MonoDomain *domain, char *addr)
        return ji;
 }
 
+static void
+jit_info_table_check (MonoJitInfoTable *table)
+{
+       int i;
+
+       for (i = 0; i < table->num_chunks; ++i) {
+               MonoJitInfoTableChunk *chunk = table->chunks [i];
+               int j;
+
+               g_assert (chunk->refcount > 0 /* && chunk->refcount <= 8 */);
+               if (chunk->refcount > 10)
+                       printf("warning: chunk refcount is %d\n", chunk->refcount);
+               g_assert (chunk->num_elements <= MONO_JIT_INFO_TABLE_CHUNK_SIZE);
+
+               for (j = 0; j < chunk->num_elements; ++j) {
+                       MonoJitInfo *this = chunk->data [j];
+                       MonoJitInfo *next;
+
+                       g_assert ((gint8*)this->code_start + this->code_size <= chunk->last_code_end);
+
+                       if (j < chunk->num_elements - 1)
+                               next = chunk->data [j + 1];
+                       else if (i < table->num_chunks - 1) {
+                               int k;
+
+                               for (k = i + 1; k < table->num_chunks; ++k)
+                                       if (table->chunks [k]->num_elements > 0)
+                                               break;
+
+                               if (k >= table->num_chunks)
+                                       return;
+
+                               g_assert (table->chunks [k]->num_elements > 0);
+                               next = table->chunks [k]->data [0];
+                       } else
+                               return;
+
+                       g_assert ((gint8*)this->code_start + this->code_size <= (gint8*)next->code_start + next->code_size);
+               }
+       }
+}
+
+static MonoJitInfoTable*
+jit_info_table_realloc (MonoJitInfoTable *old)
+{
+       int i;
+       int num_elements = jit_info_table_num_elements (old);
+       int required_size;
+       int num_chunks;
+       int new_chunk, new_element;
+       MonoJitInfoTable *new;
+
+       /* number of needed places for elements needed */
+       required_size = (int)((long)num_elements * JIT_INFO_TABLE_FILL_RATIO_DENOM / JIT_INFO_TABLE_FILL_RATIO_NOM);
+       num_chunks = (required_size + MONO_JIT_INFO_TABLE_CHUNK_SIZE - 1) / MONO_JIT_INFO_TABLE_CHUNK_SIZE;
+
+       new = g_malloc (sizeof (MonoJitInfoTable) + sizeof (MonoJitInfoTableChunk*) * num_chunks);
+       new->num_chunks = num_chunks;
+
+       for (i = 0; i < num_chunks; ++i)
+               new->chunks [i] = jit_info_table_new_chunk ();
+
+       new_chunk = 0;
+       new_element = 0;
+       for (i = 0; i < old->num_chunks; ++i) {
+               MonoJitInfoTableChunk *chunk = old->chunks [i];
+               int chunk_num_elements = chunk->num_elements;
+               int j;
+
+               for (j = 0; j < chunk_num_elements; ++j) {
+                       if (!IS_JIT_INFO_TOMBSTONE (chunk->data [j])) {
+                               g_assert (new_chunk < num_chunks);
+                               new->chunks [new_chunk]->data [new_element] = chunk->data [j];
+                               if (++new_element >= JIT_INFO_TABLE_FILLED_NUM_ELEMENTS) {
+                                       new->chunks [new_chunk]->num_elements = new_element;
+                                       ++new_chunk;
+                                       new_element = 0;
+                               }
+                       }
+               }
+       }
+
+       if (new_chunk < num_chunks) {
+               g_assert (new_chunk == num_chunks - 1);
+               new->chunks [new_chunk]->num_elements = new_element;
+               g_assert (new->chunks [new_chunk]->num_elements > 0);
+       }
+
+       for (i = 0; i < num_chunks; ++i) {
+               MonoJitInfoTableChunk *chunk = new->chunks [i];
+               MonoJitInfo *ji = chunk->data [chunk->num_elements - 1];
+
+               new->chunks [i]->last_code_end = (gint8*)ji->code_start + ji->code_size;
+       }
+
+       return new;
+}
+
+static void
+jit_info_table_split_chunk (MonoJitInfoTableChunk *chunk, MonoJitInfoTableChunk **new1p, MonoJitInfoTableChunk **new2p)
+{
+       MonoJitInfoTableChunk *new1 = jit_info_table_new_chunk ();
+       MonoJitInfoTableChunk *new2 = jit_info_table_new_chunk ();
+
+       g_assert (chunk->num_elements == MONO_JIT_INFO_TABLE_CHUNK_SIZE);
+
+       new1->num_elements = MONO_JIT_INFO_TABLE_CHUNK_SIZE / 2;
+       new2->num_elements = MONO_JIT_INFO_TABLE_CHUNK_SIZE - new1->num_elements;
+
+       memcpy ((void*)new1->data, (void*)chunk->data, sizeof (MonoJitInfo*) * new1->num_elements);
+       memcpy ((void*)new2->data, (void*)(chunk->data + new1->num_elements), sizeof (MonoJitInfo*) * new2->num_elements);
+
+       new1->last_code_end = (gint8*)new1->data [new1->num_elements - 1]->code_start
+               + new1->data [new1->num_elements - 1]->code_size;
+       new2->last_code_end = (gint8*)new2->data [new2->num_elements - 1]->code_start
+               + new2->data [new2->num_elements - 1]->code_size;
+
+       *new1p = new1;
+       *new2p = new2;
+}
+
+static MonoJitInfoTable*
+jit_info_table_copy_and_split_chunk (MonoJitInfoTable *table, MonoJitInfoTableChunk *chunk)
+{
+       MonoJitInfoTable *new_table = g_malloc (sizeof (MonoJitInfoTable)
+               + sizeof (MonoJitInfoTableChunk*) * (table->num_chunks + 1));
+       int i, j;
+
+       new_table->num_chunks = table->num_chunks + 1;
+
+       j = 0;
+       for (i = 0; i < table->num_chunks; ++i) {
+               if (table->chunks [i] == chunk) {
+                       jit_info_table_split_chunk (chunk, &new_table->chunks [j], &new_table->chunks [j + 1]);
+                       j += 2;
+               } else {
+                       new_table->chunks [j] = table->chunks [i];
+                       ++new_table->chunks [j]->refcount;
+                       ++j;
+               }
+       }
+
+       g_assert (j == new_table->num_chunks);
+
+       return new_table;
+}
+
+static MonoJitInfoTableChunk*
+jit_info_table_purify_chunk (MonoJitInfoTableChunk *old)
+{
+       MonoJitInfoTableChunk *new = jit_info_table_new_chunk ();
+       int i, j;
+
+       j = 0;
+       for (i = 0; i < old->num_elements; ++i) {
+               if (!IS_JIT_INFO_TOMBSTONE (old->data [i]))
+                       new->data [j++] = old->data [i];
+       }
+
+       new->num_elements = j;
+       if (new->num_elements > 0)
+               new->last_code_end = (gint8*)new->data [j - 1]->code_start + new->data [j - 1]->code_size;
+       else
+               new->last_code_end = old->last_code_end;
+
+       return new;
+}
+
+static MonoJitInfoTable*
+jit_info_table_copy_and_purify_chunk (MonoJitInfoTable *table, MonoJitInfoTableChunk *chunk)
+{
+       MonoJitInfoTable *new_table = g_malloc (sizeof (MonoJitInfoTable)
+               + sizeof (MonoJitInfoTableChunk*) * table->num_chunks);
+       int i, j;
+
+       new_table->num_chunks = table->num_chunks;
+
+       j = 0;
+       for (i = 0; i < table->num_chunks; ++i) {
+               if (table->chunks [i] == chunk)
+                       new_table->chunks [j++] = jit_info_table_purify_chunk (table->chunks [i]);
+               else {
+                       new_table->chunks [j] = table->chunks [i];
+                       ++new_table->chunks [j]->refcount;
+                       ++j;
+               }
+       }
+
+       g_assert (j == new_table->num_chunks);
+
+       return new_table;
+}
+
+/* As we add an element to the table the case can arise that the chunk
+ * to which we need to add is already full.  In that case we have to
+ * allocate a new table and do something about that chunk.  We have
+ * several strategies:
+ *
+ * If the number of elements in the table is below the low watermark
+ * or above the high watermark, we reallocate the whole table.
+ * Otherwise we only concern ourselves with the overflowing chunk:
+ *
+ * If there are no tombstones in the chunk then we split the chunk in
+ * two, each half full.
+ *
+ * If the chunk does contain tombstones, we just make a new copy of
+ * the chunk without the tombstones, which will have room for at least
+ * the one element we have to add.
+ */
+static MonoJitInfoTable*
+jit_info_table_chunk_overflow (MonoJitInfoTable *table, MonoJitInfoTableChunk *chunk)
+{
+       int num_elements = jit_info_table_num_elements (table);
+       int i;
+
+       if (num_elements < JIT_INFO_TABLE_LOW_WATERMARK (table->num_chunks * MONO_JIT_INFO_TABLE_CHUNK_SIZE)
+                       || num_elements > JIT_INFO_TABLE_HIGH_WATERMARK (table->num_chunks * MONO_JIT_INFO_TABLE_CHUNK_SIZE)) {
+               //printf ("reallocing table\n");
+               return jit_info_table_realloc (table);
+       }
+
+       /* count the number of non-tombstone elements in the chunk */
+       num_elements = 0;
+       for (i = 0; i < chunk->num_elements; ++i) {
+               if (!IS_JIT_INFO_TOMBSTONE (chunk->data [i]))
+                       ++num_elements;
+       }
+
+       if (num_elements == MONO_JIT_INFO_TABLE_CHUNK_SIZE) {
+               //printf ("splitting chunk\n");
+               return jit_info_table_copy_and_split_chunk (table, chunk);
+       }
+
+       //printf ("purifying chunk\n");
+       return jit_info_table_copy_and_purify_chunk (table, chunk);
+}
+
+/* We add elements to the table by first making space for them by
+ * shifting the elements at the back to the right, one at a time.
+ * This results in duplicate entries during the process, but during
+ * all the time the table is in a sorted state.  Also, when an element
+ * is replaced by another one, the element that replaces it has an end
+ * address that is equal to or lower than that of the replaced
+ * element.  That property is necessary to guarantee that when
+ * searching for an element we end up at a position not higher than
+ * the one we're looking for (i.e. we either find the element directly
+ * or we end up to the left of it).
+ */
 void
 mono_jit_info_table_add (MonoDomain *domain, MonoJitInfo *ji)
 {
-       MonoJitInfoTable *table = domain->jit_info_table;
-       gpointer start = ji->code_start;
-       int pos;
+       MonoJitInfoTable *table;
+       int chunk_pos, pos;
+       MonoJitInfoTableChunk *chunk;
+       int num_elements;
+       int i;
+
+       g_assert (ji->method != NULL);
 
        mono_domain_lock (domain);
-       pos = mono_jit_info_table_index (table, start);
 
-       g_array_insert_val (table, pos, ji);
+       ++mono_stats.jit_info_table_insert_count;
+
+       table = domain->jit_info_table;
+
+ restart:
+       chunk_pos = jit_info_table_index (table, (gint8*)ji->code_start + ji->code_size);
+       g_assert (chunk_pos < table->num_chunks);
+       chunk = table->chunks [chunk_pos];
+
+       if (chunk->num_elements >= MONO_JIT_INFO_TABLE_CHUNK_SIZE) {
+               MonoJitInfoTable *new_table = jit_info_table_chunk_overflow (table, chunk);
+
+               /* Debugging code, should be removed. */
+               //jit_info_table_check (new_table);
+
+               domain->jit_info_table = new_table;
+               mono_memory_barrier ();
+               mono_thread_hazardous_free_or_queue (table, (MonoHazardousFreeFunc)jit_info_table_free);
+               table = new_table;
+
+               goto restart;
+       }
+
+       /* Debugging code, should be removed. */
+       //jit_info_table_check (table);
+
+       num_elements = chunk->num_elements;
+
+       pos = jit_info_table_chunk_index (chunk, NULL, (gint8*)ji->code_start + ji->code_size);
+
+       /* First we need to size up the chunk by one, by copying the
+          last item, or inserting the first one, if the table is
+          empty. */
+       if (num_elements > 0)
+               chunk->data [num_elements] = chunk->data [num_elements - 1];
+       else
+               chunk->data [0] = ji;
+       mono_memory_write_barrier ();
+       chunk->num_elements = ++num_elements;
+
+       /* Shift the elements up one by one. */
+       for (i = num_elements - 2; i >= pos; --i) {
+               mono_memory_write_barrier ();
+               chunk->data [i + 1] = chunk->data [i];
+       }
+
+       /* Now we have room and can insert the new item. */
+       mono_memory_write_barrier ();
+       chunk->data [pos] = ji;
+
+       /* Set the high code end address chunk entry. */
+       chunk->last_code_end = (gint8*)chunk->data [chunk->num_elements - 1]->code_start
+               + chunk->data [chunk->num_elements - 1]->code_size;
+
+       /* Debugging code, should be removed. */
+       //jit_info_table_check (table);
+
        mono_domain_unlock (domain);
 }
 
+static MonoJitInfo*
+mono_jit_info_make_tombstone (MonoJitInfo *ji)
+{
+       MonoJitInfo *tombstone = g_new0 (MonoJitInfo, 1);
+
+       tombstone->code_start = ji->code_start;
+       tombstone->code_size = ji->code_size;
+       tombstone->method = NULL;
+
+       return tombstone;
+}
+
 void
 mono_jit_info_table_remove (MonoDomain *domain, MonoJitInfo *ji)
 {
-       MonoJitInfoTable *table = domain->jit_info_table;
+       MonoJitInfoTable *table;
+       MonoJitInfoTableChunk *chunk;
        gpointer start = ji->code_start;
-       int pos;
+       int chunk_pos, pos;
 
        mono_domain_lock (domain);
-       pos = mono_jit_info_table_index (table, start);
-       if (g_array_index (table, gpointer, pos) != ji) {
-               MonoJitInfo *ji2 = g_array_index (table, gpointer, pos);
-               g_assert (ji == ji2);
-       }
-       g_assert (g_array_index (table, gpointer, pos) == ji);
+       table = domain->jit_info_table;
+
+       ++mono_stats.jit_info_table_remove_count;
+
+       chunk_pos = jit_info_table_index (table, start);
+       g_assert (chunk_pos < table->num_chunks);
+
+       pos = jit_info_table_chunk_index (table->chunks [chunk_pos], NULL, start);
+
+       do {
+               chunk = table->chunks [chunk_pos];
+
+               while (pos < chunk->num_elements) {
+                       if (chunk->data [pos] == ji)
+                               goto found;
+
+                       g_assert (IS_JIT_INFO_TOMBSTONE (chunk->data [pos]));
+                       g_assert ((guint8*)chunk->data [pos]->code_start + chunk->data [pos]->code_size
+                               <= (guint8*)ji->code_start + ji->code_size);
+
+                       ++pos;
+               }
+
+               ++chunk_pos;
+               pos = 0;
+       } while (chunk_pos < table->num_chunks);
+
+ found:
+       g_assert (chunk->data [pos] == ji);
+
+       chunk->data [pos] = mono_jit_info_make_tombstone (ji);
+
+       /* Debugging code, should be removed. */
+       //jit_info_table_check (table);
 
-       g_array_remove_index (table, pos);
        mono_domain_unlock (domain);
-}      
+}
+
+static MonoAotModuleInfoTable*
+mono_aot_module_info_table_new (void)
+{
+       return g_array_new (FALSE, FALSE, sizeof (gpointer));
+}
 
 static int
-aot_info_table_index (MonoJitInfoTable *table, char *addr)
+aot_info_table_index (MonoAotModuleInfoTable *table, char *addr)
 {
        int left = 0, right = table->len;
 
@@ -277,7 +813,7 @@ mono_jit_info_add_aot_module (MonoImage *image, gpointer start, gpointer end)
        mono_appdomains_lock ();
 
        if (!aot_modules)
-               aot_modules = mono_jit_info_table_new ();
+               aot_modules = mono_aot_module_info_table_new ();
 
        pos = aot_info_table_index (aot_modules, start);
 
@@ -519,7 +1055,7 @@ mono_domain_create (void)
        domain->static_data_array = NULL;
        mono_jit_code_hash_init (&domain->jit_code_hash);
        domain->ldstr_table = mono_g_hash_table_new ((GHashFunc)mono_string_hash, (GCompareFunc)mono_string_equal);
-       domain->jit_info_table = mono_jit_info_table_new ();
+       domain->jit_info_table = jit_info_table_new ();
        domain->class_init_trampoline_hash = g_hash_table_new (mono_aligned_addr_hash, NULL);
        domain->jump_trampoline_hash = g_hash_table_new (mono_aligned_addr_hash, NULL);
        domain->finalizable_objects_hash = g_hash_table_new (mono_aligned_addr_hash, NULL);
@@ -1163,7 +1699,7 @@ mono_domain_free (MonoDomain *domain, gboolean force)
        }
        mono_g_hash_table_destroy (domain->ldstr_table);
        domain->ldstr_table = NULL;
-       mono_jit_info_table_free (domain->jit_info_table);
+       jit_info_table_free (domain->jit_info_table);
        domain->jit_info_table = NULL;
 #ifdef DEBUG_DOMAIN_UNLOAD
        mono_mempool_invalidate (domain->mp);