compile.c: Drop iseq_list from ibf_dump

[Feature #16505]
This commit is contained in:
NagayamaRyoga 2020-02-06 21:31:28 +09:00 committed by Alan Wu
parent 6e5e6a40c4
commit e443f23576
Notes: git 2020-02-10 01:33:59 +09:00

View File

@ -9525,7 +9525,6 @@ struct ibf_dump_buffer {
}; };
struct ibf_dump { struct ibf_dump {
VALUE iseq_list; /* [iseqs] */
st_table *iseq_table; /* iseq -> iseq number */ st_table *iseq_table; /* iseq -> iseq number */
struct ibf_dump_buffer global_buffer; struct ibf_dump_buffer global_buffer;
struct ibf_dump_buffer *current_buffer; struct ibf_dump_buffer *current_buffer;
@ -9647,7 +9646,7 @@ ibf_table_lookup(struct st_table *table, st_data_t key)
} }
static int static int
ibf_table_index(struct st_table *table, st_data_t key) ibf_table_find_or_insert(struct st_table *table, st_data_t key)
{ {
int index = ibf_table_lookup(table, key); int index = ibf_table_lookup(table, key);
@ -9683,7 +9682,7 @@ ibf_dump_object(struct ibf_dump *dump, VALUE obj)
{ {
int obj_index = ibf_table_lookup(dump->current_buffer->obj_table, (st_data_t)obj); int obj_index = ibf_table_lookup(dump->current_buffer->obj_table, (st_data_t)obj);
if (obj_index < 0) { if (obj_index < 0) {
obj_index = ibf_table_index(dump->current_buffer->obj_table, (st_data_t)obj); obj_index = ibf_table_find_or_insert(dump->current_buffer->obj_table, (st_data_t)obj);
rb_ary_push(dump->current_buffer->obj_list, obj); rb_ary_push(dump->current_buffer->obj_list, obj);
} }
return obj_index; return obj_index;
@ -9725,12 +9724,7 @@ ibf_dump_iseq(struct ibf_dump *dump, const rb_iseq_t *iseq)
return -1; return -1;
} }
else { else {
int iseq_index = ibf_table_lookup(dump->iseq_table, (st_data_t)iseq); return ibf_table_find_or_insert(dump->iseq_table, (st_data_t)iseq);
if (iseq_index < 0) {
iseq_index = ibf_table_index(dump->iseq_table, (st_data_t)iseq);
rb_ary_push(dump->iseq_list, (VALUE)iseq);
}
return iseq_index;
} }
} }
@ -10668,22 +10662,41 @@ ibf_load_iseq_each(struct ibf_load *load, rb_iseq_t *iseq, ibf_offset_t offset)
verify_call_cache(iseq); verify_call_cache(iseq);
} }
struct ibf_dump_iseq_list_arg
{
struct ibf_dump *dump;
VALUE offset_list;
};
static int
ibf_dump_iseq_list_i(st_data_t key, st_data_t val, st_data_t ptr)
{
const rb_iseq_t *iseq = (const rb_iseq_t *)key;
struct ibf_dump_iseq_list_arg *args = (struct ibf_dump_iseq_list_arg *)ptr;
ibf_offset_t offset = ibf_dump_iseq_each(args->dump, iseq);
rb_ary_push(args->offset_list, UINT2NUM(offset));
return ST_CONTINUE;
}
static void static void
ibf_dump_iseq_list(struct ibf_dump *dump, struct ibf_header *header) ibf_dump_iseq_list(struct ibf_dump *dump, struct ibf_header *header)
{ {
VALUE list = rb_ary_tmp_new(RARRAY_LEN(dump->iseq_list)); VALUE offset_list = rb_ary_tmp_new(dump->iseq_table->num_entries);
long i;
for (i = 0; i < RARRAY_LEN(dump->iseq_list); i++) { struct ibf_dump_iseq_list_arg args;
ibf_offset_t offset = ibf_dump_iseq_each(dump, (rb_iseq_t *)RARRAY_AREF(dump->iseq_list, i)); args.dump = dump;
rb_ary_push(list, UINT2NUM(offset)); args.offset_list = offset_list;
}
long size = RARRAY_LEN(dump->iseq_list); st_foreach(dump->iseq_table, ibf_dump_iseq_list_i, (st_data_t)&args);
st_index_t i;
st_index_t size = dump->iseq_table->num_entries;
ibf_offset_t *offsets = ALLOCA_N(ibf_offset_t, size); ibf_offset_t *offsets = ALLOCA_N(ibf_offset_t, size);
for (i = 0; i < size; i++) { for (i = 0; i < size; i++) {
offsets[i] = NUM2UINT(RARRAY_AREF(list, i)); offsets[i] = NUM2UINT(RARRAY_AREF(offset_list, i));
} }
ibf_dump_align(dump, sizeof(ibf_offset_t)); ibf_dump_align(dump, sizeof(ibf_offset_t));
@ -11335,7 +11348,8 @@ ibf_dump_mark(void *ptr)
struct ibf_dump *dump = (struct ibf_dump *)ptr; struct ibf_dump *dump = (struct ibf_dump *)ptr;
rb_gc_mark(dump->global_buffer.str); rb_gc_mark(dump->global_buffer.str);
rb_gc_mark(dump->global_buffer.obj_list); rb_gc_mark(dump->global_buffer.obj_list);
rb_gc_mark(dump->iseq_list);
rb_mark_set(dump->iseq_table);
} }
static void static void
@ -11372,7 +11386,8 @@ static const rb_data_type_t ibf_dump_type = {
static void static void
ibf_dump_setup(struct ibf_dump *dump, VALUE dumper_obj) ibf_dump_setup(struct ibf_dump *dump, VALUE dumper_obj)
{ {
RB_OBJ_WRITE(dumper_obj, &dump->iseq_list, rb_ary_tmp_new(0)); dump->iseq_table = NULL;
RB_OBJ_WRITE(dumper_obj, &dump->global_buffer.obj_list, ibf_dump_object_list_new(&dump->global_buffer.obj_table)); RB_OBJ_WRITE(dumper_obj, &dump->global_buffer.obj_list, ibf_dump_object_list_new(&dump->global_buffer.obj_table));
RB_OBJ_WRITE(dumper_obj, &dump->global_buffer.str, rb_str_new(0, 0)); RB_OBJ_WRITE(dumper_obj, &dump->global_buffer.str, rb_str_new(0, 0));
dump->iseq_table = st_init_numtable(); /* need free */ dump->iseq_table = st_init_numtable(); /* need free */